query.h

Go to the documentation of this file.

 
#ifndef QUERY_H
#define QUERY_H
 
#include <stdbool.h>
#include <sys/socket.h>
#include <time.h>
 
#include "channel.h"
#include "config.h"
#include "constants.h"
#include "metrics.h"
#include "rip_ns_utils.h"
#include "rr_record.h"
 
 
typedef struct edns_client_subnet_s {
    unsigned char *edns_cs_raw_buf;
 
    uint16_t edns_cs_raw_buf_len;
 
    bool edns_cs_valid;
 
    uint16_t family;
 
    struct sockaddr_storage ip;
 
    uint8_t source_mask;
    
    uint8_t scope_mask;
} edns_client_subnet_t;
 
typedef struct edns_s {
    unsigned char *edns_raw_buf;
 
    uint16_t edns_raw_buf_len;
 
    bool edns_valid;
 
    uint8_t extended_rcode;
 
    uint8_t version;
 
    uint16_t udp_resp_len;
 
    bool dnssec;
 
    edns_client_subnet_t client_subnet; 
} edns_t;
 
typedef struct query_s {
    uint8_t protocol;
 
    struct sockaddr_storage *client_ip;
 
    struct sockaddr_storage *local_ip;
 
    /*** REQUEST ***/
    unsigned char *request_buffer;
 
    size_t request_buffer_size;
 
    size_t request_buffer_len;
 
    rip_ns_header_t *request_hdr;
 
    unsigned char *query_label;
 
    uint16_t query_label_size;
 
    uint16_t query_label_len;
 
    uint16_t query_q_type;
 
    uint16_t query_q_class;
 
    edns_t edns;
 
    /*** RESPONSE ***/
    unsigned char *response_buffer;
 
    size_t response_buffer_size;
 
    size_t response_buffer_len;
    
    rip_ns_header_t *response_hdr;
 
    rr_record_t *answer_section[RIP_NS_RESP_MAX_ANSW];
 
    uint8_t answer_section_count;
 
    rr_record_t *authority_section[RIP_NS_RESP_MAX_NS];
 
    uint8_t authority_section_count;
 
    rr_record_t *additional_section[RIP_NS_RESP_MAX_ADDL];
 
    uint8_t additional_section_count;
 
    struct timespec start_time;
 
    struct timespec end_time;
 
    int end_code;
 
    char error_message[ERR_MSG_LENGTH];
 
    const unsigned char *dnptrs[DNS_RESPONSE_COMPRESSED_NAMES_MAX];
} query_t;
 
 
void query_init(query_t *q, config_t *cfg, uint8_t protocol);
void query_reset(query_t *q);
void query_clean(query_t *q);
int  query_tcp_response_buffer_increase(query_t *q);
 
int  query_parse_edns_ext_cs(edns_client_subnet_t *cs);
int  query_parse_edns_ext(query_t *q, unsigned char *buf, unsigned char *eobuf);
int  query_parse_request_rr_additional_edns(query_t *q, unsigned char *ptr);
int  query_parse_request_rr_question(query_t *q);
void query_parse(query_t *q);
 
void query_resolve(query_t *q);
 
int  query_pack_edns(uint8_t *buf, uint16_t buf_len, edns_t *edns);
int  query_pack_rr(const unsigned char *name, rr_record_t *rr, unsigned char *buf, uint16_t buf_len,
                   const unsigned char **dnptrs, const unsigned char **lastdnptr);
int  query_response_pack(query_t *q);
 
typedef struct query_log_s {
    size_t buf_size;
 
    char *a_buf;
 
    size_t a_buf_len;
 
    char *b_buf;
    size_t b_buf_len;
 
    char *buf;
 
    size_t buf_len;
} query_log_t;
 
typedef struct query_log_loop_args_s {
    config_t *cfg;
 
    channel_bss_t *query_log_channels;
 
    size_t query_log_channel_count;
 
    channel_log_t *app_log_channel;
 
    metrics_t *metrics;
 
} query_log_loop_args_t;
 
int  query_log(char *buf, size_t buf_len, query_t *q);
void query_log_rotate(query_log_t *query_log);
 
void * query_log_loop(void *args);
 
 
void query_report_metrics(query_t *, metrics_t *metrics);
 
#endif /* End of QUERY_H */