summaryrefslogtreecommitdiffstats
path: root/src/dnswire/decoder.h
blob: 60ab44e21412c053d2f063faadd8ea664e725690 (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
61
62
63
64
/*
 * 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_decoder
#define __dnswire_h_decoder 1

enum dnswire_decoder_state {
    dnswire_decoder_reading_control = 0,
    dnswire_decoder_checking_ready  = 1,
    dnswire_decoder_checking_accept = 2,
    dnswire_decoder_reading_start   = 3,
    dnswire_decoder_checking_start  = 4,
    dnswire_decoder_reading_frames  = 5,
    dnswire_decoder_checking_finish = 6,
    dnswire_decoder_done            = 7,
};
extern const char* const dnswire_decoder_state_string[];

struct dnswire_decoder {
    enum dnswire_decoder_state state;
    struct tinyframe_reader    reader;
    struct dnstap              dnstap;

    unsigned ready_support_dnstap_protobuf : 1;
    unsigned accept_support_dnstap_protobuf : 1;
};

#define DNSWIRE_DECODER_INITIALIZER                \
    {                                              \
        .state  = dnswire_decoder_reading_control, \
        .reader = TINYFRAME_READER_INITIALIZER,    \
        .dnstap = DNSTAP_INITIALIZER,              \
    }

#define dnswire_decoder_decoded(d) (d).reader.bytes_read
#define dnswire_decoder_dnstap(d) (&(d).dnstap)
#define dnswire_decoder_cleanup(d) dnstap_cleanup(&(d).dnstap)

enum dnswire_result dnswire_decoder_decode(struct dnswire_decoder*, const uint8_t*, size_t);

#endif