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 *v;
  29
  30         v = lookup_hash(id, &oui);
  31         while (v && id != v->id)
  32                 v = v->next;
  33
  34         return (v && id == v->id ? v->vendor : NULL);
  35 }
  36
  37 void dissector_init_oui(void)
  38 {
  39         FILE *fp;
  40         char buff[128], *ptr;
  41         struct vendor_id *v;
  42         void **pos;
  43
  44         if (initialized)
  45                 return;
  46
  47         fp = fopen("/etc/netsniff-ng/oui.conf", "r");
  48         if (!fp)
  49                 panic("No oui.conf found!\n");
  50
  51         memset(buff, 0, sizeof(buff));
  52
  53         while (fgets(buff, sizeof(buff), fp) != NULL) {
  54                 buff[sizeof(buff) - 1] = 0;
  55                 ptr = buff;
  56
  57                 v = xmalloc(sizeof(*v));
  58                 v->id = strtol(ptr, &ptr, 0);
  59
  60                 if ((ptr = strstr(buff, ", ")))
  61                         ptr += strlen(", ");
  62                 ptr = strtrim_right(ptr, '\n');
  63                 ptr = strtrim_right(ptr, ' ');
  64
  65                 v->vendor = xstrdup(ptr);
  66                 v->next = NULL;
  67
  68                 pos = insert_hash(v->id, v, &oui);
  69                 if (pos) {
  70                         v->next = *pos;
  71                         *pos = v;
  72                 }
  73
  74                 memset(buff, 0, sizeof(buff));
  75         }
  76
  77         fclose(fp);
  78         initialized = true;
  79 }
  80
  81 static int dissector_cleanup_oui_hash(void *ptr)
  82 {
  83         struct vendor_id *tmp, *v = ptr;
  84
  85         if (!ptr)
  86                 return 0;
  87
  88         while ((tmp = v->next)) {
  89                 xfree(v->vendor);
  90                 xfree(v);
  91                 v = tmp;
  92         }
  93
  94         xfree(v->vendor);
  95         xfree(v);
  96
  97         return 0;
  98 }
  99
 100 void dissector_cleanup_oui(void)
 101 {
 102         for_each_hash(&oui, dissector_cleanup_oui_hash);
 103         free_hash(&oui);
 104         initialized = false;
 105 }