summaryrefslogtreecommitdiffstats
path: root/src/dnswire/encoder.h
blob: c5fbc6aa894afbc11a3f4f27825ba7ce2aab31d9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
/*
 * Author Jerry Lundström <jerry@dns-oarc.net>
 * 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 <http://www.gnu.org/licenses/>.
 */

#include <dnswire/dnswire.h>
#include <dnswire/dnstap.h>

#include <tinyframe/tinyframe.h>

#ifndef __dnswire_h_encoder
#define __dnswire_h_encoder 1

enum dnswire_encoder_state {
    dnswire_encoder_control_ready  = 0,
    dnswire_encoder_control_start  = 1,
    dnswire_encoder_control_accept = 2,
    dnswire_encoder_control_finish = 3,
    dnswire_encoder_frames         = 4,
    dnswire_encoder_control_stop   = 5,
    dnswire_encoder_done           = 6,
};
extern const char* const dnswire_encoder_state_string[];

struct dnswire_encoder {
    enum dnswire_encoder_state state;
    struct tinyframe_writer    writer;
    const struct dnstap*       dnstap;
};

#define DNSWIRE_ENCODER_INITIALIZER              \
    {                                            \
        .state  = dnswire_encoder_control_start, \
        .writer = TINYFRAME_WRITER_INITIALIZER,  \
        .dnstap = 0,                             \
    }

#define dnswire_encoder_set_dnstap(e, d) (e).dnstap = d
#define dnswire_encoder_encoded(e) (e).writer.bytes_wrote

enum dnswire_result dnswire_encoder_encode(struct dnswire_encoder*, uint8_t*, size_t);
enum dnswire_result dnswire_encoder_stop(struct dnswire_encoder*);

#endif