void dt_writer_free(dt_writer_t *writer)
{
if (writer == NULL) {
return;
}
fstrm_writer_destroy(&writer->fw);
free(writer->version);
free(writer);
}
struct dt_env *
dt_create(const char *socket_path, unsigned num_workers)
{
fstrm_res res;
struct dt_env *env;
struct fstrm_iothr_options *fopt;
struct fstrm_unix_writer_options *fuwopt;
struct fstrm_writer *fw;
struct fstrm_writer_options *fwopt;
verbose(VERB_OPS, "attempting to connect to dnstap socket %s",
socket_path);
log_assert(socket_path != NULL);
log_assert(num_workers > 0);
env = (struct dt_env *) calloc(1, sizeof(struct dt_env));
if (!env)
return NULL;
fwopt = fstrm_writer_options_init();
res = fstrm_writer_options_add_content_type(fwopt,
DNSTAP_CONTENT_TYPE, sizeof(DNSTAP_CONTENT_TYPE) - 1);
log_assert(res == fstrm_res_success);
fuwopt = fstrm_unix_writer_options_init();
fstrm_unix_writer_options_set_socket_path(fuwopt, socket_path);
fw = fstrm_unix_writer_init(fuwopt, fwopt);
log_assert(fw != NULL);
fopt = fstrm_iothr_options_init();
fstrm_iothr_options_set_num_input_queues(fopt, num_workers);
env->iothr = fstrm_iothr_init(fopt, &fw);
if (env->iothr == NULL) {
verbose(VERB_DETAIL, "dt_create: fstrm_iothr_init() failed");
fstrm_writer_destroy(&fw);
free(env);
env = NULL;
}
fstrm_iothr_options_destroy(&fopt);
fstrm_unix_writer_options_destroy(&fuwopt);
fstrm_writer_options_destroy(&fwopt);
return env;
}
int main(int argc, char **argv)
{
const char *input_fname = NULL;
const char *output_fname = NULL;
fstrm_res res = fstrm_res_failure;
struct fstrm_file_options *fopt = NULL;
struct fstrm_writer_options *wopt = NULL;
struct fstrm_reader *r = NULL;
struct fstrm_writer *w = NULL;
int rv = EXIT_FAILURE;
/* Args. */
if (argc != 2 && argc != 3) {
fprintf(stderr, "Usage: %s <INPUT FILE> [<OUTPUT FILE>]\n", argv[0]);
fprintf(stderr, "Dumps a Frame Streams formatted input file.\n\n");
return EXIT_FAILURE;
}
input_fname = argv[1];
if (argc == 3)
output_fname = argv[2];
/* Line buffering. */
setvbuf(stdout, NULL, _IOLBF, 0);
setvbuf(stderr, NULL, _IOLBF, 0);
/* Setup file reader options. */
fopt = fstrm_file_options_init();
fstrm_file_options_set_file_path(fopt, input_fname);
/* Initialize file reader. */
r = fstrm_file_reader_init(fopt, NULL);
if (r == NULL) {
fputs("Error: fstrm_file_reader_init() failed.\n", stderr);
goto out;
}
res = fstrm_reader_open(r);
if (res != fstrm_res_success) {
fputs("Error: fstrm_reader_open() failed.\n", stderr);
goto out;
}
if (output_fname != NULL) {
/* Setup file writer options. */
fstrm_file_options_set_file_path(fopt, output_fname);
/* Setup writer options. */
wopt = fstrm_writer_options_init();
/* Copy "content type" from the reader's START frame. */
res = process_start_frame(r, wopt);
if (res != fstrm_res_success) {
fputs("Error: process_start_frame() failed.\n", stderr);
goto out;
}
/* Initialize file writer. */
w = fstrm_file_writer_init(fopt, wopt);
if (w == NULL) {
fputs("Error: fstrm_file_writer_init() failed.\n", stderr);
goto out;
}
res = fstrm_writer_open(w);
if (res != fstrm_res_success) {
fstrm_writer_destroy(&w);
fputs("Error: fstrm_writer_open() failed.\n", stderr);
goto out;
}
} else {
/* Process the START frame. */
res = process_start_frame(r, NULL);
if (res != fstrm_res_success) {
fprintf(stderr, "Error: process_start_frame() failed.\n");
goto out;
}
}
/* Loop over data frames. */
for (;;) {
const uint8_t *data;
size_t len_data;
res = fstrm_reader_read(r, &data, &len_data);
if (res == fstrm_res_success) {
/* Got a data frame. */
res = print_data_frame(data, len_data);
if (res != fstrm_res_success) {
fprintf(stderr, "Error: print_data_frame() failed.\n");
goto out;
}
if (w != NULL) {
/* Write the data frame. */
res = fstrm_writer_write(w, data, len_data);
if (res != fstrm_res_success) {
fprintf(stderr, "Error: write_data_frame() failed.\n");
goto out;
}
}
} else if (res == fstrm_res_stop) {
/* Normal end of data stream. */
res = print_stop_frame(r);
if (res != fstrm_res_success) {
fprintf(stderr, "Error: unable to read STOP frame.\n");
goto out;
}
rv = EXIT_SUCCESS;
break;
} else {
/* Abnormal end. */
fprintf(stderr, "Error: fstrm_reader_read() failed.\n");
goto out;
}
}
out:
/* Cleanup options. */
fstrm_file_options_destroy(&fopt);
fstrm_writer_options_destroy(&wopt);
/* Cleanup reader. */
fstrm_reader_destroy(&r);
/* Cleanup writer. */
if (w != NULL) {
res = fstrm_writer_close(w);
if (res != fstrm_res_success) {
fprintf(stderr, "Error: fstrm_writer_close() failed.\n");
rv = EXIT_FAILURE;
}
fstrm_writer_destroy(&w);
}
return rv;
}
