diff options
Diffstat (limited to 'qa/src/stripmark.c')
-rw-r--r-- | qa/src/stripmark.c | 78 |
1 files changed, 78 insertions, 0 deletions
diff --git a/qa/src/stripmark.c b/qa/src/stripmark.c new file mode 100644 index 0000000..0809c3f --- /dev/null +++ b/qa/src/stripmark.c @@ -0,0 +1,78 @@ +#include <stdio.h> +#include <string.h> +#include <errno.h> +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> +#include <pcp/pmapi.h> +#include <pcp/impl.h> + +/* + * filter to copy archive.0 and strip mark records + */ + +int +main(int argc, char *argv[]) +{ + int *len; + int buflen; + char *buf ; + int in; + int out; + int nb; + int sts; + + if (argc != 3) { + fprintf(stderr, "Usage: stripmark in.0 out.0\n"); + exit(1); + } + + if ((in = open(argv[1], O_RDONLY)) < 0) { + fprintf(stderr, "Failed to open %s: %s\n", argv[1], strerror(errno)); + exit(1); + } + if ((out = open(argv[2], O_WRONLY|O_CREAT|O_TRUNC, 0644)) < 0) { + fprintf(stderr, "Failed to create %s: %s\n", argv[2], strerror(errno)); + exit(1); + } + buflen = sizeof(*len); + buf = (char *)malloc(buflen); + len = (int *)buf; + + for ( ; ; ) { + if ((nb = read(in, buf, sizeof(*len))) != sizeof(*len)) { + if (nb == 0) break; + if (nb < 0) + fprintf(stderr, "read error: %s\n", strerror(errno)); + else + fprintf(stderr, "read error: expected %d bytes, got %d\n", + (int)sizeof(*len), nb); + exit(1); + } + if (htonl(*len) > buflen) { + buflen = htonl(*len); + buf = (char *)realloc(buf, buflen); + len = (int *)buf; + } + if ((nb = read(in, &buf[sizeof(*len)], htonl(*len)-sizeof(*len))) != htonl(*len)-sizeof(*len)) { + if (nb == 0) + fprintf(stderr, "read error: end of file\n"); + else if (nb < 0) + fprintf(stderr, "read error: %s\n", strerror(errno)); + else + fprintf(stderr, "read error: expected %d bytes, got %d\n", + (int)(htonl(*len)-sizeof(*len)), nb); + exit(1); + } + if (htonl(*len) > sizeof(__pmPDUHdr) - sizeof(*len) + sizeof(__pmTimeval) + sizeof(int)) { + sts = write(out, buf, htonl(*len)); + if (sts != htonl(*len)) { + fprintf(stderr, "Error: write %d returns %d\n", htonl(*len), sts); + exit(1); + } + } + else + fprintf(stderr, "Skip mark @ byte %d into input\n", (int)(lseek(in, 0, SEEK_CUR) - sizeof(*len))); + } + return 0; +} |