summaryrefslogtreecommitdiff
path: root/oui.c
blob: 5eea9fbd7a30841fc8d166b00b07ffe371e95b3f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
/*
 * netsniff-ng - the packet sniffing beast
 * Copyright 2009 - 2013 Daniel Borkmann.
 * Subject to the GPL, version 2.
 */

#include <stdint.h>
#include <stdbool.h>

#include "hash.h"
#include "xmalloc.h"
#include "oui.h"
#include "str.h"

static struct hash_table oui;

static bool initialized = false;

struct vendor_id {
	unsigned int id;
	char *vendor;
	struct vendor_id *next;
};

const char *lookup_vendor(unsigned int id)
{
	struct vendor_id *v;

	v = lookup_hash(id, &oui);
	while (v && id != v->id)
		v = v->next;

	return (v && id == v->id ? v->vendor : NULL);
}

void dissector_init_oui(void)
{
	FILE *fp;
	char buff[128], *ptr, *end;
	struct vendor_id *v;
	void **pos;

	if (initialized)
		return;

	fp = fopen(PREFIX_STRING "/etc/netsniff-ng/oui.conf", "r");
	if (!fp)
		panic("No oui.conf found!\n");

	memset(buff, 0, sizeof(buff));

	while (fgets(buff, sizeof(buff), fp) != NULL) {
		buff[sizeof(buff) - 1] = 0;
		ptr = buff;

		v = xmalloc(sizeof(*v));
		v->id = strtol(ptr, &end, 0);
		/* not a valid line, skip */
		if (v->id == 0 && end == ptr)
			continue;

		ptr = strstr(buff, ", ");
		/* likewise */
		if (!ptr)
			continue;

		ptr += strlen(", ");
		ptr = strtrim_right(ptr, '\n');
		ptr = strtrim_right(ptr, ' ');

		v->vendor = xstrdup(ptr);
		v->next = NULL;

		pos = insert_hash(v->id, v, &oui);
		if (pos) {
			v->next = *pos;
			*pos = v;
		}

		memset(buff, 0, sizeof(buff));
	}

	fclose(fp);
	initialized = true;
}

static int dissector_cleanup_oui_hash(void *ptr)
{
	struct vendor_id *tmp, *v = ptr;

	if (!ptr)
		return 0;

	while ((tmp = v->next)) {
		xfree(v->vendor);
		xfree(v);
		v = tmp;
	}

	xfree(v->vendor);
	xfree(v);

	return 0;
}

void dissector_cleanup_oui(void)
{
	for_each_hash(&oui, dissector_cleanup_oui_hash);
	free_hash(&oui);
	initialized = false;
}