sqlite_header.h

Go to the documentation of this file.

 
#ifndef SQLITE_HEADER_H
#define SQLITE_HEADER_H
 
#include <stdint.h>
#include <pcap.h>
#include <sqlite3.h>
 
#include "../../../utils/allocs.h"
#include "../../../utils/os.h"
 
#include "../../capture_config.h"
 
#include "packet_decoder.h"
 
#define MAX_DB_NAME 100
 
#define ETH_CREATE_TABLE                                                       \
  "CREATE TABLE IF NOT EXISTS eth (timestamp INTEGER NOT NULL, "               \
  "id TEXT NOT NULL, caplen INTEGER, length INTEGER, ifname TEXT, "            \
  "ether_dhost TEXT, ether_shost TEXT, ether_type INTEGER, PRIMARY KEY "       \
  "(timestamp, id));"
 
#define ARP_CREATE_TABLE                                                       \
  "CREATE TABLE IF NOT EXISTS arp (id TEXT, "                                  \
  "ar_hrd INTEGER, ar_pro INTEGER, ar_hln INTEGER, "                           \
  "ar_pln INTEGER, ar_op INTEGER, arp_sha TEXT, arp_spa TEXT, "                \
  "arp_tha TEXT, arp_tpa TEXT, PRIMARY KEY (id));"
 
#define IP4_CREATE_TABLE                                                       \
  "CREATE TABLE IF NOT EXISTS ip4 (id TEXT NOT NULL, "                         \
  "ip_hl INTEGER, ip_v INTEGER, ip_tos INTEGER, ip_len INTEGER, ip_id "        \
  "INTEGER, "                                                                  \
  "ip_off INTEGER, ip_ttl INTEGER, ip_p INTEGER, ip_sum INTEGER, ip_src "      \
  "TEXT, "                                                                     \
  "ip_dst TEXT, PRIMARY KEY (id));"
 
#define IP6_CREATE_TABLE                                                       \
  "CREATE TABLE IF NOT EXISTS ip6 (id TEXT NOT NULL, "                         \
  "ip6_un1_flow INTEGER, ip6_un1_plen INTEGER, ip6_un1_nxt INTEGER, "          \
  "ip6_un1_hlim INTEGER, "                                                     \
  "ip6_un2_vfc INTEGER, ip6_src TEXT, ip6_dst TEXT, PRIMARY KEY (id));"
 
#define TCP_CREATE_TABLE                                                       \
  "CREATE TABLE IF NOT EXISTS tcp (id TEXT NOT NULL, "                         \
  "source INTEGER, dest INTEGER, seq INTEGER, ack_seq INTEGER, res1 INTEGER, " \
  "doff INTEGER, fin INTEGER, "                                                \
  "syn INTEGER, rst INTEGER, psh INTEGER, ack INTEGER, urg INTEGER, window "   \
  "INTEGER, check_p INTEGER, "                                                 \
  "urg_ptr INTEGER, PRIMARY KEY (id));"
 
#define UDP_CREATE_TABLE                                                       \
  "CREATE TABLE IF NOT EXISTS udp (id TEXT NOT NULL, "                         \
  "source INTEGER, dest INTEGER, len INTEGER, check_p INTEGER, PRIMARY KEY "   \
  "(id));"
 
#define ICMP4_CREATE_TABLE                                                     \
  "CREATE TABLE IF NOT EXISTS icmp4 (id TEXT NOT NULL, "                       \
  "type INTEGER, code INTEGER, checksum INTEGER, gateway INTEGER, PRIMARY "    \
  "KEY (id));"
 
#define ICMP6_CREATE_TABLE                                                     \
  "CREATE TABLE IF NOT EXISTS icmp6 (id TEXT NOT NULL, "                       \
  "icmp6_type INTEGER, icmp6_code INTEGER, icmp6_cksum INTEGER, "              \
  "icmp6_un_data32 INTEGER, PRIMARY KEY (id));"
 
#define DNS_CREATE_TABLE                                                       \
  "CREATE TABLE IF NOT EXISTS dns (id TEXT NOT NULL, "                         \
  "tid INTEGER, flags INTEGER, nqueries INTEGER, nanswers INTEGER, nauth "     \
  "INTEGER, "                                                                  \
  "nother INTEGER, qname TEXT, PRIMARY KEY (id));"
 
#define MDNS_CREATE_TABLE                                                      \
  "CREATE TABLE IF NOT EXISTS mdns (id TEXT NOT NULL, "                        \
  "tid INTEGER, flags INTEGER, nqueries INTEGER, nanswers INTEGER, nauth "     \
  "INTEGER, "                                                                  \
  "nother INTEGER, qname TEXT, PRIMARY KEY (id));"
 
#define DHCP_CREATE_TABLE                                                      \
  "CREATE TABLE IF NOT EXISTS dhcp (id TEXT NOT NULL, "                        \
  "op INTEGER, htype INTEGER, hlen INTEGER, hops INTEGER, xid INTEGER, secs "  \
  "INTEGER, flags INTEGER, "                                                   \
  "ciaddr TEXT, yiaddr TEXT, siaddr TEXT, giaddr TEXT, chaddr TEXT, "          \
  "PRIMARY KEY (id));"
 
#define ETH_INSERT_INTO                                                        \
  "INSERT INTO eth VALUES(@timestamp, @id, @caplen, @length, @ifname, "        \
  "@ether_dhost, @ether_shost, @ether_type);"
#define ARP_INSERT_INTO                                                        \
  "INSERT INTO arp VALUES(@id, "                                               \
  "@ar_hrd, @ar_pro, @ar_hln, @ar_pln, @ar_op, @arp_sha, @arp_spa, "           \
  "@arp_tha, @arp_tpa);"
#define IP4_INSERT_INTO                                                        \
  "INSERT INTO ip4 VALUES(@id, @ip_hl, @ip_v, "                                \
  "@ip_tos, @ip_len, @ip_id, "                                                 \
  "@ip_off, @ip_ttl, @ip_p, @ip_sum, @ip_src, @ip_dst);"
#define IP6_INSERT_INTO                                                        \
  "INSERT INTO ip6 VALUES(@id, "                                               \
  "@ip6_un1_flow, @ip6_un1_plen, @ip6_un1_nxt, @ip6_un1_hlim, @ip6_un2_vfc, "  \
  "@ip6_src, @ip6_dst);"
#define TCP_INSERT_INTO                                                        \
  "INSERT INTO tcp VALUES(@id, "                                               \
  "@source, @dest, @seq, @ack_seq, @res1, @doff, @fin, "                       \
  "@syn, @rst, @psh, @ack, @urg, @window, @check_p, @urg_ptr);"
#define UDP_INSERT_INTO                                                        \
  "INSERT INTO udp VALUES(@id, @source, @dest, @len, @check_p);"
#define ICMP4_INSERT_INTO                                                      \
  "INSERT INTO icmp4 VALUES(@id, @type, @code, @checksum, @gateway);"
#define ICMP6_INSERT_INTO                                                      \
  "INSERT INTO icmp6 VALUES(@id, "                                             \
  "@icmp6_type, @icmp6_code, @icmp6_cksum, @icmp6_un_data32);"
#define DNS_INSERT_INTO                                                        \
  "INSERT INTO dns VALUES(@id, "                                               \
  "@tid, @flags, @nqueries, @nanswers, @nauth, @nother, @qname);"
#define MDNS_INSERT_INTO                                                       \
  "INSERT INTO mdns VALUES(@id, "                                              \
  "@tid, @flags, @nqueries, @nanswers, @nauth, @nother, @qname);"
#define DHCP_INSERT_INTO                                                       \
  "INSERT INTO dhcp VALUES(@id, "                                              \
  "@op, @htype, @hlen, @hops, @xid, @secs, @flags, "                           \
  "@ciaddr, @yiaddr, @siaddr, @giaddr, @chaddr);"
 
int save_packet_statement(sqlite3 *db, struct tuple_packet *tp);
 
int init_sqlite_header_db(sqlite3 *db);
#endif