bridger/main.c at master · nbd168/bridger · GitHub

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
112
113
114
115
116
117
118
119
120
121
122
123
124
// SPDX-License-Identifier: GPL-2.0+
/*
 * Copyright (C) 2022 Felix Fietkau <nbd@nbd.name>
 */

#include <stdarg.h>
#include "bridger.h"

static int debug_level = 0;

#ifdef UBUS_SUPPORT
static struct udebug ud;
static struct udebug_buf udb_log;
struct udebug_buf udb_nl;
static const struct udebug_buf_meta meta_log = {
	.name = "bridger_log",
	.format = UDEBUG_FORMAT_STRING,
};
static const struct udebug_buf_meta meta_nl = {
	.name = "bridger_nl",
	.format = UDEBUG_FORMAT_PACKET,
	.sub_format = UDEBUG_DLT_NETLINK,
};
static struct udebug_ubus_ring rings[] = {
	{
		.buf = &udb_log,
		.meta = &meta_log,
		.default_entries = 1024,
		.default_size = 64 * 1024,
	},
	{
		.buf = &udb_nl,
		.meta = &meta_nl,
		.default_entries = 1024,
		.default_size = 64 * 1024,
	},
};

void bridger_udebug_config(struct udebug_ubus *ctx, struct blob_attr *data,
			   bool enabled)
{
    udebug_ubus_apply_config(&ud, rings, ARRAY_SIZE(rings), data, enabled);
}
#endif


void bridger_dprintf(const char *format, ...)
{
	va_list ap;

#ifdef UBUS_SUPPORT
	if (udebug_buf_valid(&udb_log)) {
		va_start(ap, format);
		udebug_entry_init(&udb_log);
		udebug_entry_vprintf(&udb_log, format, ap);
		udebug_entry_add(&udb_log);
		va_end(ap);
	}
#endif

	if (!debug_level)
		return;

	va_start(ap, format);
	vfprintf(stderr, format, ap);
	va_end(ap);
}

const char *format_macaddr(const uint8_t *mac)
{
	static char str[sizeof("ff:ff:ff:ff:ff:ff ")];

	snprintf(str, sizeof(str), "%02x:%02x:%02x:%02x:%02x:%02x",
	mac[0], mac[1], mac[2], mac[3], mac[4], mac[5]);

	return str;
}

int main(int argc, char **argv)
{
	int ch;

	while ((ch = getopt(argc, argv, "d")) != -1) {
		switch (ch) {
		case 'd':
			debug_level++;
			break;
		default:
			break;
		}
	}

	uloop_init();

#ifdef UBUS_SUPPORT
	udebug_init(&ud);
	udebug_auto_connect(&ud, NULL);
	for (size_t i = 0; i < ARRAY_SIZE(rings); i++)
		udebug_ubus_ring_init(&ud, &rings[i]);
#endif

	if (bridger_bpf_init()) {
		perror("bridger_bpf_init");
		return 1;
	}

	if (bridger_nl_init()) {
		perror("bridger_nl_init");
		return 1;
	}

	bridger_device_init();
	bridger_flow_init();
	bridger_ubus_init();

	uloop_run();

	bridger_ubus_stop();
	bridger_device_stop();

	uloop_done();

	return 0;
}