Implement str_to_addr() and unit test.
[darkstat] / decode.h
index 5128ec9..d4c8bc0 100644 (file)
--- a/decode.h
+++ b/decode.h
@@ -1,20 +1,28 @@
 /* darkstat 3
- * copyright (c) 2001-2007 Emil Mikulic.
+ * copyright (c) 2001-2009 Emil Mikulic.
  *
  * decode.h: packet decoding.
  *
  * You may use, modify and redistribute this file under the terms of the
  * GNU General Public License version 2. (see COPYING.GPL)
  */
+#ifndef __DARKSTAT_DECODE_H
+#define __DARKSTAT_DECODE_H
 
 #include <pcap.h>
-#include <netinet/in.h> /* in_addr_t */
+#include <netinet/in_systm.h>  /* n_time */
+#define __USE_GNU 1
+#include <netinet/in.h> /* for <netinet/ip.h>  */
+#include <netinet/ip.h> /* struct ip */
+
+#include "hosts_db.h"   /* addr46 */
 
 #define PPP_HDR_LEN     4
 #define FDDI_HDR_LEN    21
-#define IP_HDR_LEN      20
-#define TCP_HDR_LEN     20
-#define UDP_HDR_LEN     8
+#define IP_HDR_LEN      sizeof(struct ip)
+#define IPV6_HDR_LEN    sizeof(struct ip6_hdr)
+#define TCP_HDR_LEN     sizeof(struct tcphdr)
+#define UDP_HDR_LEN     sizeof(struct udphdr)
 #define NULL_HDR_LEN    4
 #define PPPOE_HDR_LEN   8
 #define SLL_HDR_LEN     16
 #define ETHER_ADDR_LEN 6
 #endif
 
-typedef struct {
+#ifndef IPPROTO_OSPF
+#  define IPPROTO_OSPF 89
+#endif
+
+struct linkhdr {
    int linktype;
-   int hdrlen;
+   unsigned int hdrlen;
    pcap_handler handler;
-} linkhdr_t;
+};
 
-const linkhdr_t *getlinkhdr(int linktype);
-int getcaplen(const linkhdr_t *lh);
-char *ip_to_str(const in_addr_t ip);
+const struct linkhdr *getlinkhdr(const int linktype);
+int getsnaplen(const struct linkhdr *lh);
+char *ip_to_str(const struct addr46 *const ip);
+char *ip_to_str_af(const void *const addr, sa_family_t af);
 
 typedef struct {
    /* Fields are in host byte order (except IPs) */
-   in_addr_t src_ip, dest_ip;
+   union {
+      struct in_addr src_ip;
+      struct in6_addr src_ip6;
+   };
+   union {
+      struct in_addr dest_ip;
+      struct in6_addr dest_ip6;
+   };
    time_t time;
    uint16_t len;
+   sa_family_t af;               /* AF_{UNSPEC, INET, INET6} */
    uint8_t proto;                /* IPPROTO_{TCP, UDP, ICMP} */
    uint8_t tcp_flags;            /* only for TCP */
    uint16_t src_port, dest_port; /* only for TCP, UDP */
@@ -46,4 +67,5 @@ typedef struct {
            dst_mac[ETHER_ADDR_LEN]; /* only for Ethernet */
 } pktsummary;
 
+#endif /* __DARKSTAT_DECODE_H */
 /* vim:set ts=3 sw=3 tw=78 expandtab: */