From be956cd27353a4bb585b1a648e8469cf7adb5edf Mon Sep 17 00:00:00 2001 From: Daniel Baumann Date: Fri, 14 Jan 2022 16:03:48 +0100 Subject: Adding upstream version 0.2.0. Signed-off-by: Daniel Baumann --- src/dnswire/reader.h | 91 ++++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 91 insertions(+) create mode 100644 src/dnswire/reader.h (limited to 'src/dnswire/reader.h') diff --git a/src/dnswire/reader.h b/src/dnswire/reader.h new file mode 100644 index 0000000..5fdfad5 --- /dev/null +++ b/src/dnswire/reader.h @@ -0,0 +1,91 @@ +/* + * Author Jerry Lundström + * Copyright (c) 2019, OARC, Inc. + * All rights reserved. + * + * This file is part of the dnswire library. + * + * dnswire library is free software: you can redistribute it and/or modify + * it under the terms of the GNU Lesser General Public License as published by + * the Free Software Foundation, either version 3 of the License, or + * (at your option) any later version. + * + * dnswire library is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public License + * along with dnswire library. If not, see . + */ + +#include +#include +#include + +#include + +#ifndef __dnswire_h_reader +#define __dnswire_h_reader 1 + +enum dnswire_reader_state { + dnswire_reader_reading_control = 0, + dnswire_reader_decoding_control = 1, + dnswire_reader_encoding_accept = 2, + dnswire_reader_writing_accept = 3, + dnswire_reader_reading = 4, + dnswire_reader_decoding = 5, + dnswire_reader_encoding_finish = 6, + dnswire_reader_writing_finish = 7, + dnswire_reader_done = 8, +}; +extern const char* const dnswire_reader_state_string[]; + +/* + * Attributes: + * - size: The current size of the buffer + * - inc: How much the buffer will be increased by if more spare is needed + * - max: The maximum size the buffer is allowed to have + * - at: Where in the buffer we are decoding (start of data) + * - left: How much data that is still left in the buffer from `at` + * - pushed: How much data that was pushed to the buffer by `dnswire_reader_push()` + */ +struct dnswire_reader { + enum dnswire_reader_state state; + + struct dnswire_decoder decoder; + uint8_t* buf; + size_t size, inc, max, at, left, pushed; + + struct dnswire_encoder encoder; + uint8_t* write_buf; + size_t write_size, write_inc, write_max, write_at, write_left; + + bool allow_bidirectional, is_bidirectional; +}; + +enum dnswire_result dnswire_reader_init(struct dnswire_reader*); + +#define dnswire_reader_pushed(r) (r).pushed +#define dnswire_reader_dnstap(r) (&(r).decoder.dnstap) +#define dnswire_reader_cleanup(r) \ + dnswire_decoder_cleanup((r).decoder) +#define dnswire_reader_destroy(r) \ + free((r).buf); \ + free((r).write_buf); \ + dnswire_decoder_cleanup((r).decoder) +#define dnswire_reader_is_bidirectional(r) (r).is_bidirectional + +enum dnswire_result dnswire_reader_allow_bidirectional(struct dnswire_reader*, bool); +enum dnswire_result dnswire_reader_set_bufsize(struct dnswire_reader*, size_t); +enum dnswire_result dnswire_reader_set_bufinc(struct dnswire_reader*, size_t); +enum dnswire_result dnswire_reader_set_bufmax(struct dnswire_reader*, size_t); + +enum dnswire_result dnswire_reader_push(struct dnswire_reader*, const uint8_t*, size_t, uint8_t*, size_t*); +enum dnswire_result dnswire_reader_read(struct dnswire_reader*, int); +static inline enum dnswire_result dnswire_reader_fread(struct dnswire_reader* handle, FILE* fp) +{ + return dnswire_reader_read(handle, fileno(fp)); +} + +#endif -- cgit v1.2.3