oui.c

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