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