src/oui.c

   1 /*
   2  * netsniff-ng - the packet sniffing beast
   3  * By Daniel Borkmann <daniel@netsniff-ng.org>
   4  * Copyright 2009, 2010 Daniel Borkmann.
   5  * Subject to the GPL, version 2.
   6  */
   7
   8 #include <stdint.h>
   9
  10 #include "hash.h"
  11 #include "xmalloc.h"
  12 #include "xutils.h"
  13 #include "oui.h"
  14
  15 static struct hash_table oui;
  16 static int initialized = 0;
  17
  18 struct vendor_id {
  19         unsigned int id;
  20         char *vendor;
  21         struct vendor_id *next;
  22 };
  23
  24 const char *lookup_vendor(unsigned int id)
  25 {
  26         struct vendor_id *entry = lookup_hash(id, &oui);
  27
  28         while (entry && id != entry->id)
  29                 entry = entry->next;
  30
  31         return (entry && id == entry->id ? entry->vendor : NULL);
  32 }
  33
  34 void dissector_init_oui(void)
  35 {
  36         FILE *fp;
  37         char buff[512], *ptr;
  38         struct vendor_id *ven;
  39         void **pos;
  40
  41         if (initialized)
  42                 return;
  43
  44         fp = fopen("/etc/netsniff-ng/oui.conf", "r");
  45         if (!fp)
  46                 panic("No /etc/netsniff-ng/oui.conf found!\n");
  47
  48         memset(buff, 0, sizeof(buff));
  49         while (fgets(buff, sizeof(buff), fp) != NULL) {
  50                 buff[sizeof(buff) - 1] = 0;
  51
  52                 ven = xmalloc(sizeof(*ven));
  53                 ptr = buff;
  54                 ptr = skips(ptr);
  55                 ptr = getuint(ptr, &ven->id);
  56                 ptr = skips(ptr);
  57                 ptr = skipchar(ptr, ',');
  58                 ptr = skips(ptr);
  59                 ptr = strtrim_right(ptr, '\n');
  60                 ptr = strtrim_right(ptr, ' ');
  61                 ven->vendor = xstrdup(ptr);
  62                 ven->next = NULL;
  63                 pos = insert_hash(ven->id, ven, &oui);
  64                 if (pos) {
  65                         ven->next = *pos;
  66                         *pos = ven;
  67                 }
  68                 memset(buff, 0, sizeof(buff));
  69         }
  70         fclose(fp);
  71
  72         initialized = 1;
  73 }
  74
  75 static int __dissector_cleanup_oui(void *ptr)
  76 {
  77         struct vendor_id *tmp, *v = ptr;
  78         if (!ptr)
  79                 return 0;
  80         while ((tmp = v->next)) {
  81                 xfree(v->vendor);
  82                 xfree(v);
  83                 v = tmp;
  84         }
  85         xfree(v->vendor);
  86         xfree(v);
  87
  88         return 0;
  89 }
  90
  91 void dissector_cleanup_oui(void)
  92 {
  93         for_each_hash(&oui, __dissector_cleanup_oui);
  94         free_hash(&oui);
  95
  96         initialized = 0;
  97 }