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 "oui.h"
  13 #include "str.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, *end;
  40         struct vendor_id *v;
  41         void **pos;
  42
  43         if (initialized)
  44                 return;
  45
  46         fp = fopen(PREFIX_STRING "/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, &end, 0);
  58                 /* No valid line, skip */
  59                 if (v->id == 0 && end == ptr)
  60                         continue;
  61
  62                 ptr = strstr(buff, ", ");
  63                 if (!ptr)
  64                         continue;
  65
  66                 ptr += strlen(", ");
  67                 ptr = strtrim_right(ptr, '\n');
  68                 ptr = strtrim_right(ptr, ' ');
  69
  70                 v->vendor = xstrdup(ptr);
  71                 v->next = NULL;
  72
  73                 pos = insert_hash(v->id, v, &oui);
  74                 if (pos) {
  75                         v->next = *pos;
  76                         *pos = v;
  77                 }
  78
  79                 memset(buff, 0, sizeof(buff));
  80         }
  81
  82         fclose(fp);
  83         initialized = true;
  84 }
  85
  86 static int dissector_cleanup_oui_hash(void *ptr)
  87 {
  88         struct vendor_id *tmp, *v = ptr;
  89
  90         if (!ptr)
  91                 return 0;
  92
  93         while ((tmp = v->next)) {
  94                 xfree(v->vendor);
  95                 xfree(v);
  96                 v = tmp;
  97         }
  98
  99         xfree(v->vendor);
 100         xfree(v);
 101
 102         return 0;
 103 }
 104
 105 void dissector_cleanup_oui(void)
 106 {
 107         for_each_hash(&oui, dissector_cleanup_oui_hash);
 108         free_hash(&oui);
 109         initialized = false;
 110 }