conn.h

Go to the documentation of this file.

 
#ifndef CONN_H
#define CONN_H
 
#include <stdint.h>
#include <sys/uio.h>
#include <sys/socket.h>
#include <time.h>
 
#include <uthash/uthash.h>
 
#include "query.h"
 
#define CONN_IS_UDP_LISTENER(conn) \
    conn->lc == 0 && conn->proto == 0
 
#define CONN_IS_TCP_LISTENER(conn) \
    conn->lc == 0 && conn->proto == 1
 
#define CONN_IS_TCP_CONN(conn) \
    conn->lc == 1 && conn->proto == 1
 
typedef enum listener_start_error_e {
    LISTENER_ERR_SOCKET                      = -1,
 
    LISTENER_ERR_SOCKET_OPT_REUSEADDR        = -2,
     
    LISTENER_ERR_SOCKET_OPT_REUSEPORT        = -3,
     
    LISTENER_ERR_BIND                        = -4,
     
    LISTENER_ERR_LISTEN                      = -5,
     
    LISTENER_ERR_SOCKET_OPT_IP_PKTINFO       = -6,
     
    LISTENER_ERR_SOCKET_OPT_IPV6_V6ONLY      = -7,
     
    LISTENER_ERR_SOCKET_OPT_IPV6_RECVPKTINFO = -8,
 
    LISTENER_ERR_SOCKET_OPT_RCVBUF           = -9,
 
    LISTENER_ERR_SOCKET_OPT_SNDBUF           = -10,
} listener_start_error_t;
 
typedef enum conn_tcp_state_e {
    /*8 Error assigning connection ID. */
    TCP_CONN_ST_ASSIGN_CONN_ID_ERR = 0,
 
    TCP_CONN_ST_WAIT_FOR_QUERY,
 
    /* Wait for query data, meaning some data was read in yet not enough to be
     * a complete query request.
     */
    TCP_CONN_ST_WAIT_FOR_QUERY_DATA,
 
    TCP_CONN_ST_WAIT_FOR_WRITE,
 
    TCP_CONN_ST_CLOSED_FOR_READ,
 
    TCP_CONN_ST_READ_ERR,
 
    TCP_CONN_ST_CLOSED_FOR_WRITE,
 
    TCP_CONN_ST_WRITE_ERR,
 
    TCP_CONN_ST_QUERY_SIZE_TOOLARGE,
 
} conn_tcp_state_t;
 
typedef struct conn_tcp_s {   
    struct sockaddr_storage client_ip;
 
    struct sockaddr_storage local_ip;
 
    unsigned char *read_buffer;
 
    size_t read_buffer_size;
 
    size_t read_buffer_len;
 
    size_t query_write_index;
 
    size_t write_index;
 
    query_t *queries;
 
    size_t queries_size;
 
    size_t queries_count;
 
    size_t queries_total_count;
 
    size_t tcp_keepalive;
 
    conn_tcp_state_t state;
 
    struct timespec start_time;
 
    struct timespec timeout;
 
    struct timespec end_time;
} conn_tcp_t;
 
typedef struct conn_udp_s {
    unsigned int vector_len;
 
    struct mmsghdr *read_vector;
 
    unsigned int read_vector_count;
 
    query_t *queries;
 
    struct mmsghdr *write_vector;
 
    unsigned int write_vector_write_index;
 
    unsigned int write_vector_count;
 
} conn_udp_t;
 
typedef struct conn_s {
    UT_hash_handle hh;
 
    uint64_t cid;
 
    uint8_t lc: 1;
 
    uint8_t proto: 1;
 
    uint8_t ip_version: 1;
 
    uint8_t waiting_for_read: 1;
 
    uint8_t waiting_for_write: 1;
 
    uint8_t in_read_queue: 1;
 
    uint8_t in_write_queue: 1;
 
    uint8_t in_release_queue: 1;
 
    struct conn_s *read_q_handle;
 
    struct conn_s *write_q_handle;
 
    struct conn_s *gen_q_handle;
 
    int fd;
 
    union {
        conn_udp_t *udp;
        conn_tcp_t *tcp;
    } conn;
 
} conn_t;
 
typedef struct conn_fifo_queue_s {
    conn_t *head;
 
    conn_t *tail;
 
} conn_fifo_queue_t;
 
 
void     conn_fifo_enqueue_read(conn_fifo_queue_t *queue, conn_t *entry);
conn_t * conn_fifo_dequeue_read(conn_fifo_queue_t *queue);
void     conn_fifo_enqueue_write(conn_fifo_queue_t *queue, conn_t *entry);
conn_t * conn_fifo_dequeue_write(conn_fifo_queue_t *queue);
void     conn_fifo_enqueue_gen(conn_fifo_queue_t *queue, conn_t *conn);
conn_t * conn_fifo_dequeue_gen(conn_fifo_queue_t *queue);
void     conn_fifo_remove_from_read_queue(conn_fifo_queue_t *queue, conn_t *conn_rm);
void     conn_fifo_remove_from_write_queue(conn_fifo_queue_t *queue, conn_t *conn_rm);
void     conn_fifo_enqueue_release(conn_fifo_queue_t *queue, conn_t *conn);
conn_t * conn_fifo_dequeue_release(conn_fifo_queue_t *queue);
 
void         conn_tcp_release(conn_tcp_t *conn_tcp);
void         conn_udp_release(conn_udp_t *conn_udp);
void         conn_release(conn_t *conn);
void         conn_udp_vectors_reset(conn_udp_t *conn_udp);
conn_udp_t * conn_udp_new(config_t *cfg, int family);
conn_t     * conn_new_tcp(int fd, config_t *cfg, int ip_version,
                          struct sockaddr_storage *client_ip,
                          struct sockaddr_storage *local_ip);
 
conn_t * conn_listener_provision(config_t *cfg, int family, int protocol,
                                 char *err_buf, size_t err_buf_len);
 
conn_t * conn_lru_cache_get(conn_t **lru, uint64_t id);
 
bool conn_tcp_id_assign(uint64_t *id, conn_t **lru, uint64_t *base);
 
void conn_tcp_report_metrics(conn_tcp_t *conn_tcp, metrics_t *metrics);
 
#endif /* End of CONN_H */