oui.c

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