]>
Commit | Line | Data |
---|---|---|
43c02e7b VB |
1 | /* |
2 | * Copyright (c) 2008 Vincent Bernat <bernat@luffy.cx> | |
3 | * | |
4 | * Permission to use, copy, modify, and distribute this software for any | |
5 | * purpose with or without fee is hereby granted, provided that the above | |
6 | * copyright notice and this permission notice appear in all copies. | |
7 | * | |
8 | * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES | |
9 | * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF | |
10 | * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR | |
11 | * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES | |
12 | * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN | |
13 | * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF | |
14 | * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE. | |
15 | */ | |
16 | ||
17 | #include "lldpd.h" | |
18 | ||
19 | #include <stdio.h> | |
20 | #include <unistd.h> | |
21 | #include <errno.h> | |
22 | #include <signal.h> | |
23 | #include <sys/stat.h> | |
24 | #include <fcntl.h> | |
43c02e7b | 25 | #include <time.h> |
a2993d83 | 26 | #include <libgen.h> |
43c02e7b VB |
27 | #include <sys/utsname.h> |
28 | #include <sys/types.h> | |
29 | #include <sys/socket.h> | |
30 | #include <sys/select.h> | |
31 | #include <sys/time.h> | |
32 | #include <sys/ioctl.h> | |
33 | #include <arpa/inet.h> | |
43c02e7b | 34 | #include <net/if_arp.h> |
43c02e7b VB |
35 | |
36 | #ifdef USE_SNMP | |
37 | #include <net-snmp/net-snmp-config.h> | |
38 | #include <net-snmp/net-snmp-includes.h> | |
39 | #include <net-snmp/agent/net-snmp-agent-includes.h> | |
40 | #include <net-snmp/agent/snmp_vars.h> | |
41 | #endif /* USE_SNMP */ | |
42 | ||
8888d191 | 43 | static void usage(void); |
43c02e7b | 44 | |
8888d191 | 45 | static struct protocol protos[] = |
43c02e7b VB |
46 | { |
47 | { LLDPD_MODE_LLDP, 1, "LLDP", ' ', lldp_send, lldp_decode, NULL, | |
f2dcb180 | 48 | LLDP_MULTICAST_ADDR }, |
4bad1937 | 49 | #ifdef ENABLE_CDP |
43c02e7b | 50 | { LLDPD_MODE_CDPV1, 0, "CDPv1", 'c', cdpv1_send, cdp_decode, cdpv1_guess, |
f2dcb180 | 51 | CDP_MULTICAST_ADDR }, |
43c02e7b | 52 | { LLDPD_MODE_CDPV2, 0, "CDPv2", 'c', cdpv2_send, cdp_decode, cdpv2_guess, |
f2dcb180 | 53 | CDP_MULTICAST_ADDR }, |
4bad1937 VB |
54 | #endif |
55 | #ifdef ENABLE_SONMP | |
43c02e7b | 56 | { LLDPD_MODE_SONMP, 0, "SONMP", 's', sonmp_send, sonmp_decode, NULL, |
f2dcb180 | 57 | SONMP_MULTICAST_ADDR }, |
4bad1937 VB |
58 | #endif |
59 | #ifdef ENABLE_EDP | |
43c02e7b | 60 | { LLDPD_MODE_EDP, 0, "EDP", 'e', edp_send, edp_decode, NULL, |
f2dcb180 | 61 | EDP_MULTICAST_ADDR }, |
4bad1937 VB |
62 | #endif |
63 | #ifdef ENABLE_FDP | |
031118c4 | 64 | { LLDPD_MODE_FDP, 0, "FDP", 'f', fdp_send, cdp_decode, NULL, |
f2dcb180 | 65 | FDP_MULTICAST_ADDR }, |
4bad1937 | 66 | #endif |
43c02e7b | 67 | { 0, 0, "any", ' ', NULL, NULL, NULL, |
f2dcb180 | 68 | {0,0,0,0,0,0} } |
43c02e7b VB |
69 | }; |
70 | ||
6e75df87 VB |
71 | static void lldpd_update_localchassis(struct lldpd *); |
72 | static void lldpd_update_localports(struct lldpd *); | |
73 | static void lldpd_cleanup(struct lldpd *); | |
8888d191 VB |
74 | static void lldpd_loop(struct lldpd *); |
75 | static void lldpd_shutdown(int); | |
76 | static void lldpd_exit(); | |
77 | static void lldpd_send_all(struct lldpd *); | |
78 | static void lldpd_recv_all(struct lldpd *); | |
79 | static int lldpd_guess_type(struct lldpd *, char *, int); | |
80 | static void lldpd_decode(struct lldpd *, char *, int, | |
0bc32943 | 81 | struct lldpd_hardware *); |
16f910e1 VB |
82 | static void lldpd_update_chassis(struct lldpd_chassis *, |
83 | const struct lldpd_chassis *); | |
89840df0 | 84 | #ifdef ENABLE_LLDPMED |
8888d191 | 85 | static void lldpd_med(struct lldpd_chassis *); |
89840df0 | 86 | #endif |
43c02e7b | 87 | |
8888d191 | 88 | static char **saved_argv; |
43c02e7b | 89 | |
8888d191 | 90 | static void |
43c02e7b VB |
91 | usage(void) |
92 | { | |
93 | extern const char *__progname; | |
e809a587 VB |
94 | fprintf(stderr, "usage: %s [options]\n", __progname); |
95 | fprintf(stderr, "see manual page lldpd(8) for more information\n"); | |
43c02e7b VB |
96 | exit(1); |
97 | } | |
98 | ||
6e75df87 | 99 | struct lldpd_hardware * |
849954d7 | 100 | lldpd_get_hardware(struct lldpd *cfg, char *name, struct lldpd_ops *ops) |
43c02e7b | 101 | { |
6e75df87 VB |
102 | struct lldpd_hardware *hardware; |
103 | TAILQ_FOREACH(hardware, &cfg->g_hardware, h_entries) { | |
849954d7 VB |
104 | if ((strcmp(hardware->h_ifname, name) == 0) && |
105 | ((!ops) || (ops == hardware->h_ops))) | |
6e75df87 | 106 | break; |
43c02e7b | 107 | } |
6e75df87 | 108 | return hardware; |
43c02e7b VB |
109 | } |
110 | ||
6e75df87 VB |
111 | struct lldpd_hardware * |
112 | lldpd_alloc_hardware(struct lldpd *cfg, char *name) | |
43c02e7b | 113 | { |
6e75df87 | 114 | struct lldpd_hardware *hardware; |
43c02e7b | 115 | |
6e75df87 VB |
116 | if ((hardware = (struct lldpd_hardware *) |
117 | calloc(1, sizeof(struct lldpd_hardware))) == NULL) | |
118 | return NULL; | |
43c02e7b | 119 | |
6e75df87 VB |
120 | strlcpy(hardware->h_ifname, name, sizeof(hardware->h_ifname)); |
121 | hardware->h_lport.p_chassis = LOCAL_CHASSIS(cfg); | |
122 | TAILQ_INIT(&hardware->h_rports); | |
43c02e7b | 123 | |
6e75df87 VB |
124 | #ifdef ENABLE_LLDPMED |
125 | if (LOCAL_CHASSIS(cfg)->c_med_cap_available) { | |
126 | hardware->h_lport.p_med_cap_enabled = LLDPMED_CAP_CAP; | |
127 | if (!cfg->g_noinventory) | |
128 | hardware->h_lport.p_med_cap_enabled |= LLDPMED_CAP_IV; | |
129 | } | |
130 | #endif | |
131 | #ifdef ENABLE_DOT1 | |
132 | TAILQ_INIT(&hardware->h_lport.p_vlans); | |
133 | #endif | |
134 | return hardware; | |
43c02e7b VB |
135 | } |
136 | ||
a1347cd8 | 137 | #ifdef ENABLE_DOT1 |
43c02e7b VB |
138 | void |
139 | lldpd_vlan_cleanup(struct lldpd_port *port) | |
140 | { | |
141 | struct lldpd_vlan *vlan, *vlan_next; | |
142 | for (vlan = TAILQ_FIRST(&port->p_vlans); | |
143 | vlan != NULL; | |
144 | vlan = vlan_next) { | |
145 | free(vlan->v_name); | |
146 | vlan_next = TAILQ_NEXT(vlan, v_entries); | |
147 | TAILQ_REMOVE(&port->p_vlans, vlan, v_entries); | |
148 | free(vlan); | |
149 | } | |
150 | } | |
a1347cd8 | 151 | #endif |
43c02e7b | 152 | |
a0edeaf8 | 153 | /* If `all' is true, clear all information, including information that |
4e624dc2 | 154 | are not refreshed periodically. Port should be freed manually. */ |
43c02e7b | 155 | void |
a0edeaf8 | 156 | lldpd_port_cleanup(struct lldpd_port *port, int all) |
43c02e7b | 157 | { |
740593ff VB |
158 | #ifdef ENABLE_LLDPMED |
159 | int i; | |
a0edeaf8 VB |
160 | if (all) |
161 | for (i=0; i < LLDPMED_LOCFORMAT_LAST; i++) | |
162 | free(port->p_med_location[i].data); | |
740593ff | 163 | #endif |
a1347cd8 | 164 | #ifdef ENABLE_DOT1 |
43c02e7b | 165 | lldpd_vlan_cleanup(port); |
a1347cd8 | 166 | #endif |
43c02e7b VB |
167 | free(port->p_id); |
168 | free(port->p_descr); | |
77507b69 VB |
169 | if (all) { |
170 | free(port->p_lastframe); | |
171 | if (port->p_chassis) /* chassis may not have been attributed, yet */ | |
172 | port->p_chassis->c_refcount--; | |
77507b69 | 173 | } |
43c02e7b VB |
174 | } |
175 | ||
176 | void | |
77507b69 | 177 | lldpd_chassis_cleanup(struct lldpd_chassis *chassis, int all) |
43c02e7b | 178 | { |
89840df0 VB |
179 | #ifdef ENABLE_LLDPMED |
180 | free(chassis->c_med_hw); | |
517d524b | 181 | free(chassis->c_med_sw); |
89840df0 VB |
182 | free(chassis->c_med_fw); |
183 | free(chassis->c_med_sn); | |
184 | free(chassis->c_med_manuf); | |
185 | free(chassis->c_med_model); | |
186 | free(chassis->c_med_asset); | |
187 | #endif | |
43c02e7b VB |
188 | free(chassis->c_id); |
189 | free(chassis->c_name); | |
190 | free(chassis->c_descr); | |
77507b69 VB |
191 | if (all) |
192 | free(chassis); | |
43c02e7b VB |
193 | } |
194 | ||
195 | void | |
77507b69 | 196 | lldpd_remote_cleanup(struct lldpd *cfg, struct lldpd_hardware *hardware, int all) |
43c02e7b | 197 | { |
77507b69 VB |
198 | struct lldpd_port *port, *port_next; |
199 | int del; | |
200 | for (port = TAILQ_FIRST(&hardware->h_rports); | |
201 | port != NULL; | |
202 | port = port_next) { | |
203 | port_next = TAILQ_NEXT(port, p_entries); | |
204 | del = all; | |
205 | if (!del && | |
206 | (time(NULL) - port->p_lastupdate > port->p_chassis->c_ttl)) { | |
207 | hardware->h_rx_ageout_cnt++; | |
208 | del = 1; | |
209 | } | |
210 | if (del) { | |
211 | TAILQ_REMOVE(&hardware->h_rports, port, p_entries); | |
212 | lldpd_port_cleanup(port, 1); | |
4e624dc2 | 213 | free(port); |
77507b69 | 214 | } |
43c02e7b | 215 | } |
43c02e7b VB |
216 | } |
217 | ||
d9be8ea0 | 218 | void |
6e75df87 | 219 | lldpd_hardware_cleanup(struct lldpd *cfg, struct lldpd_hardware *hardware) |
d9be8ea0 | 220 | { |
6e75df87 | 221 | int i; |
a0edeaf8 | 222 | lldpd_port_cleanup(&hardware->h_lport, 1); |
6e75df87 VB |
223 | /* If we have a dedicated cleanup function, use it. Otherwise, |
224 | we just free the hardware-dependent data and close all FD | |
225 | in h_recvfds and h_sendfd. */ | |
226 | if (hardware->h_ops->cleanup) | |
227 | hardware->h_ops->cleanup(cfg, hardware); | |
228 | else { | |
229 | free(hardware->h_data); | |
230 | for (i=0; i < FD_SETSIZE; i++) | |
231 | if (FD_ISSET(i, &hardware->h_recvfds)) | |
232 | close(i); | |
233 | if (hardware->h_sendfd) close(hardware->h_sendfd); | |
234 | } | |
d9be8ea0 VB |
235 | free(hardware); |
236 | } | |
237 | ||
6e75df87 | 238 | static void |
43c02e7b VB |
239 | lldpd_cleanup(struct lldpd *cfg) |
240 | { | |
241 | struct lldpd_hardware *hardware, *hardware_next; | |
242 | ||
243 | for (hardware = TAILQ_FIRST(&cfg->g_hardware); hardware != NULL; | |
244 | hardware = hardware_next) { | |
245 | hardware_next = TAILQ_NEXT(hardware, h_entries); | |
6e75df87 | 246 | if (!hardware->h_flags) { |
43c02e7b | 247 | TAILQ_REMOVE(&cfg->g_hardware, hardware, h_entries); |
77507b69 | 248 | lldpd_remote_cleanup(cfg, hardware, 1); |
6e75df87 | 249 | lldpd_hardware_cleanup(cfg, hardware); |
77507b69 VB |
250 | } else |
251 | lldpd_remote_cleanup(cfg, hardware, 0); | |
43c02e7b VB |
252 | } |
253 | } | |
254 | ||
8888d191 | 255 | static int |
43c02e7b VB |
256 | lldpd_guess_type(struct lldpd *cfg, char *frame, int s) |
257 | { | |
258 | int i; | |
259 | if (s < ETH_ALEN) | |
260 | return -1; | |
261 | for (i=0; cfg->g_protocols[i].mode != 0; i++) { | |
262 | if (!cfg->g_protocols[i].enabled) | |
263 | continue; | |
264 | if (cfg->g_protocols[i].guess == NULL) { | |
265 | if (memcmp(frame, cfg->g_protocols[i].mac, ETH_ALEN) == 0) | |
266 | return cfg->g_protocols[i].mode; | |
267 | } else { | |
268 | if (cfg->g_protocols[i].guess(frame, s)) | |
269 | return cfg->g_protocols[i].mode; | |
270 | } | |
271 | } | |
272 | return -1; | |
273 | } | |
274 | ||
8888d191 | 275 | static void |
43c02e7b | 276 | lldpd_decode(struct lldpd *cfg, char *frame, int s, |
0bc32943 | 277 | struct lldpd_hardware *hardware) |
43c02e7b | 278 | { |
77507b69 VB |
279 | int i, result; |
280 | struct lldpd_chassis *chassis, *ochassis = NULL; | |
281 | struct lldpd_port *port, *oport = NULL; | |
43c02e7b VB |
282 | int guess = LLDPD_MODE_LLDP; |
283 | ||
50a89ca7 | 284 | /* Discard VLAN frames */ |
a8105c1b VB |
285 | if ((s >= sizeof(struct ethhdr)) && |
286 | (((struct ethhdr*)frame)->h_proto == htons(ETHERTYPE_VLAN))) | |
50a89ca7 VB |
287 | return; |
288 | ||
77507b69 VB |
289 | TAILQ_FOREACH(oport, &hardware->h_rports, p_entries) { |
290 | if ((oport->p_lastframe != NULL) && | |
291 | (oport->p_lastframe->size == s) && | |
292 | (memcmp(oport->p_lastframe->frame, frame, s) == 0)) { | |
293 | /* Already received the same frame */ | |
294 | oport->p_lastupdate = time(NULL); | |
295 | return; | |
296 | } | |
43c02e7b VB |
297 | } |
298 | ||
f2dcb180 VB |
299 | guess = lldpd_guess_type(cfg, frame, s); |
300 | for (i=0; cfg->g_protocols[i].mode != 0; i++) { | |
301 | if (!cfg->g_protocols[i].enabled) | |
302 | continue; | |
303 | if (cfg->g_protocols[i].mode == guess) { | |
304 | if ((result = cfg->g_protocols[i].decode(cfg, frame, | |
305 | s, hardware, &chassis, &port)) == -1) | |
306 | return; | |
77507b69 VB |
307 | chassis->c_protocol = port->p_protocol = |
308 | cfg->g_protocols[i].mode; | |
f2dcb180 | 309 | break; |
43c02e7b | 310 | } |
f2dcb180 VB |
311 | } |
312 | if (cfg->g_protocols[i].mode == 0) { | |
313 | LLOG_INFO("unable to guess frame type"); | |
43c02e7b | 314 | return; |
f2dcb180 | 315 | } |
43c02e7b | 316 | |
77507b69 VB |
317 | /* Do we already have the same MSAP somewhere? */ |
318 | TAILQ_FOREACH(oport, &hardware->h_rports, p_entries) { | |
319 | if ((port->p_protocol == oport->p_protocol) && | |
320 | (port->p_id_subtype == oport->p_id_subtype) && | |
321 | (port->p_id_len == oport->p_id_len) && | |
322 | (memcmp(port->p_id, oport->p_id, port->p_id_len) == 0) && | |
323 | (chassis->c_id_subtype == oport->p_chassis->c_id_subtype) && | |
324 | (chassis->c_id_len == oport->p_chassis->c_id_len) && | |
325 | (memcmp(chassis->c_id, oport->p_chassis->c_id, | |
326 | chassis->c_id_len) == 0)) { | |
327 | ochassis = oport->p_chassis; | |
328 | break; | |
329 | } | |
43c02e7b | 330 | } |
77507b69 VB |
331 | /* No, but do we already know the system? */ |
332 | if (!oport) { | |
333 | TAILQ_FOREACH(ochassis, &cfg->g_chassis, c_entries) { | |
334 | if ((chassis->c_protocol == ochassis->c_protocol) && | |
335 | (chassis->c_id_subtype == ochassis->c_id_subtype) && | |
336 | (chassis->c_id_len == ochassis->c_id_len) && | |
337 | (memcmp(chassis->c_id, ochassis->c_id, | |
338 | chassis->c_id_len) == 0)) | |
339 | break; | |
43c02e7b | 340 | } |
43c02e7b | 341 | } |
43c02e7b | 342 | |
77507b69 VB |
343 | if (oport) { |
344 | /* The port is known, remove it before adding it back */ | |
345 | TAILQ_REMOVE(&hardware->h_rports, oport, p_entries); | |
346 | lldpd_port_cleanup(oport, 1); | |
4e624dc2 | 347 | free(oport); |
77507b69 VB |
348 | } |
349 | if (ochassis) { | |
16f910e1 | 350 | lldpd_update_chassis(ochassis, chassis); |
77507b69 VB |
351 | free(chassis); |
352 | chassis = ochassis; | |
353 | } else { | |
354 | /* Chassis not known, add it */ | |
355 | chassis->c_index = ++cfg->g_lastrid; | |
356 | port->p_chassis = chassis; | |
357 | chassis->c_refcount = 0; | |
358 | TAILQ_INSERT_TAIL(&cfg->g_chassis, chassis, c_entries); | |
f4c43902 VB |
359 | i = 0; TAILQ_FOREACH(ochassis, &cfg->g_chassis, c_entries) i++; |
360 | LLOG_DEBUG("Currently, we know %d different systems", i); | |
77507b69 VB |
361 | } |
362 | /* Add port */ | |
363 | port->p_lastchange = port->p_lastupdate = time(NULL); | |
364 | if ((port->p_lastframe = (struct lldpd_frame *)malloc(s + | |
365 | sizeof(int))) != NULL) { | |
366 | port->p_lastframe->size = s; | |
367 | memcpy(port->p_lastframe->frame, frame, s); | |
368 | } | |
369 | TAILQ_INSERT_TAIL(&hardware->h_rports, port, p_entries); | |
370 | port->p_chassis = chassis; | |
371 | port->p_chassis->c_refcount++; | |
f4c43902 VB |
372 | i = 0; TAILQ_FOREACH(oport, &hardware->h_rports, p_entries) i++; |
373 | LLOG_DEBUG("Currently, %s known %d neighbors", | |
374 | hardware->h_ifname, i); | |
43c02e7b VB |
375 | return; |
376 | } | |
377 | ||
16f910e1 VB |
378 | /* Update chassis `ochassis' with values from `chassis'. */ |
379 | static void | |
380 | lldpd_update_chassis(struct lldpd_chassis *ochassis, | |
381 | const struct lldpd_chassis *chassis) { | |
382 | TAILQ_ENTRY(lldpd_chassis) entries; | |
383 | /* We want to keep refcount, index and list stuff from the current | |
384 | * chassis */ | |
385 | int refcount = ochassis->c_refcount; | |
386 | int index = ochassis->c_index; | |
387 | memcpy(&entries, &ochassis->c_entries, | |
388 | sizeof(entries)); | |
389 | /* Make the copy */ | |
390 | lldpd_chassis_cleanup(ochassis, 0); | |
391 | memcpy(ochassis, chassis, sizeof(struct lldpd_chassis)); | |
392 | /* Restore saved values */ | |
393 | ochassis->c_refcount = refcount; | |
394 | ochassis->c_index = index; | |
395 | memcpy(&ochassis->c_entries, &entries, sizeof(entries)); | |
396 | } | |
397 | ||
398 | ||
8888d191 | 399 | static void |
43c02e7b VB |
400 | lldpd_recv_all(struct lldpd *cfg) |
401 | { | |
402 | struct lldpd_hardware *hardware; | |
403 | struct lldpd_client *client, *client_next; | |
404 | fd_set rfds; | |
405 | struct timeval tv; | |
43c02e7b VB |
406 | #ifdef USE_SNMP |
407 | int fakeblock = 0; | |
408 | struct timeval *tvp = &tv; | |
409 | #endif | |
0bc32943 | 410 | int rc, nfds, n; |
43c02e7b VB |
411 | char *buffer; |
412 | ||
413 | do { | |
414 | tv.tv_sec = cfg->g_delay - (time(NULL) - cfg->g_lastsent); | |
415 | if (tv.tv_sec < 0) | |
416 | tv.tv_sec = LLDPD_TX_DELAY; | |
417 | if (tv.tv_sec >= cfg->g_delay) | |
418 | tv.tv_sec = cfg->g_delay; | |
419 | tv.tv_usec = 0; | |
420 | ||
421 | FD_ZERO(&rfds); | |
422 | nfds = -1; | |
423 | ||
424 | TAILQ_FOREACH(hardware, &cfg->g_hardware, h_entries) { | |
425 | /* Ignore if interface is down */ | |
6e75df87 | 426 | if ((hardware->h_flags & IFF_RUNNING) == 0) |
43c02e7b | 427 | continue; |
6e75df87 VB |
428 | /* This is quite expensive but we don't rely on internal |
429 | * structure of fd_set. */ | |
430 | for (n = 0; n < FD_SETSIZE; n++) | |
431 | if (FD_ISSET(n, &hardware->h_recvfds)) { | |
432 | FD_SET(n, &rfds); | |
433 | if (nfds < n) | |
434 | nfds = n; | |
435 | } | |
43c02e7b VB |
436 | } |
437 | TAILQ_FOREACH(client, &cfg->g_clients, next) { | |
438 | FD_SET(client->fd, &rfds); | |
439 | if (nfds < client->fd) | |
440 | nfds = client->fd; | |
441 | } | |
442 | FD_SET(cfg->g_ctl, &rfds); | |
443 | if (nfds < cfg->g_ctl) | |
444 | nfds = cfg->g_ctl; | |
445 | ||
446 | #ifdef USE_SNMP | |
447 | if (cfg->g_snmp) | |
448 | snmp_select_info(&nfds, &rfds, tvp, &fakeblock); | |
449 | #endif /* USE_SNMP */ | |
450 | if (nfds == -1) { | |
451 | sleep(cfg->g_delay); | |
452 | return; | |
453 | } | |
454 | ||
455 | rc = select(nfds + 1, &rfds, NULL, NULL, &tv); | |
456 | if (rc == -1) { | |
457 | if (errno == EINTR) | |
458 | continue; | |
459 | LLOG_WARN("failure on select"); | |
460 | break; | |
461 | } | |
462 | #ifdef USE_SNMP | |
463 | if (cfg->g_snmp) { | |
464 | if (rc > 0) | |
465 | snmp_read(&rfds); | |
466 | else if (rc == 0) | |
467 | snmp_timeout(); | |
468 | } | |
469 | #endif /* USE_SNMP */ | |
470 | TAILQ_FOREACH(hardware, &cfg->g_hardware, h_entries) { | |
6e75df87 VB |
471 | for (n = 0; n < FD_SETSIZE; n++) |
472 | if ((FD_ISSET(n, &hardware->h_recvfds)) && | |
473 | (FD_ISSET(n, &rfds))) break; | |
474 | if (n == FD_SETSIZE) continue; | |
475 | if ((buffer = (char *)malloc( | |
476 | hardware->h_mtu)) == NULL) { | |
477 | LLOG_WARN("failed to alloc reception buffer"); | |
478 | continue; | |
479 | } | |
480 | if ((n = hardware->h_ops->recv(cfg, hardware, | |
481 | n, buffer, hardware->h_mtu)) == -1) { | |
43c02e7b | 482 | free(buffer); |
6e75df87 | 483 | continue; |
43c02e7b | 484 | } |
6e75df87 VB |
485 | hardware->h_rx_cnt++; |
486 | lldpd_decode(cfg, buffer, n, hardware); | |
487 | free(buffer); | |
488 | break; | |
43c02e7b VB |
489 | } |
490 | if (FD_ISSET(cfg->g_ctl, &rfds)) { | |
491 | if (ctl_accept(cfg, cfg->g_ctl) == -1) | |
492 | LLOG_WARN("unable to accept new client"); | |
493 | } | |
494 | for (client = TAILQ_FIRST(&cfg->g_clients); | |
495 | client != NULL; | |
496 | client = client_next) { | |
497 | client_next = TAILQ_NEXT(client, next); | |
498 | if (FD_ISSET(client->fd, &rfds)) { | |
499 | /* Got a message */ | |
500 | if ((buffer = (char *)malloc(MAX_HMSGSIZE)) == | |
501 | NULL) { | |
502 | LLOG_WARN("failed to alloc reception buffer"); | |
503 | continue; | |
504 | } | |
505 | if ((n = recv(client->fd, buffer, | |
506 | MAX_HMSGSIZE, 0)) == -1) { | |
507 | LLOG_WARN("error while receiving message"); | |
508 | free(buffer); | |
509 | continue; | |
510 | } | |
511 | if (n > 0) | |
a552a72e | 512 | client_handle_client(cfg, client, buffer, n); |
43c02e7b VB |
513 | else |
514 | ctl_close(cfg, client->fd); /* Will use TAILQ_REMOVE ! */ | |
515 | free(buffer); | |
516 | } | |
517 | } | |
518 | ||
519 | #ifdef USE_SNMP | |
520 | if (cfg->g_snmp) { | |
521 | run_alarms(); | |
522 | netsnmp_check_outstanding_agent_requests(); | |
523 | } | |
524 | #endif /* USE_SNMP */ | |
525 | } while ((rc != 0) || (time(NULL) - cfg->g_lastsent < cfg->g_delay)); | |
526 | } | |
527 | ||
8888d191 | 528 | static void |
43c02e7b VB |
529 | lldpd_send_all(struct lldpd *cfg) |
530 | { | |
531 | struct lldpd_hardware *hardware; | |
77507b69 VB |
532 | struct lldpd_port *port; |
533 | int i, sent = 0; | |
f7db0dd8 | 534 | |
43c02e7b VB |
535 | cfg->g_lastsent = time(NULL); |
536 | TAILQ_FOREACH(hardware, &cfg->g_hardware, h_entries) { | |
537 | /* Ignore if interface is down */ | |
6e75df87 | 538 | if ((hardware->h_flags & IFF_RUNNING) == 0) |
43c02e7b VB |
539 | continue; |
540 | ||
541 | for (i=0; cfg->g_protocols[i].mode != 0; i++) { | |
542 | if (!cfg->g_protocols[i].enabled) | |
543 | continue; | |
77507b69 VB |
544 | /* We send only if we have at least one remote system |
545 | * speaking this protocol */ | |
546 | TAILQ_FOREACH(port, &hardware->h_rports, p_entries) { | |
547 | if (port->p_protocol == | |
548 | cfg->g_protocols[i].mode) { | |
549 | cfg->g_protocols[i].send(cfg, | |
550 | hardware); | |
551 | sent = 1; | |
552 | break; | |
553 | } | |
554 | } | |
43c02e7b | 555 | } |
77507b69 VB |
556 | |
557 | if (!sent) | |
558 | /* Nothing was sent for this port, let's speak LLDP */ | |
559 | cfg->g_protocols[0].send(cfg, | |
560 | hardware); | |
43c02e7b VB |
561 | } |
562 | } | |
563 | ||
89840df0 | 564 | #ifdef ENABLE_LLDPMED |
8888d191 | 565 | static void |
89840df0 VB |
566 | lldpd_med(struct lldpd_chassis *chassis) |
567 | { | |
568 | free(chassis->c_med_hw); | |
569 | free(chassis->c_med_fw); | |
570 | free(chassis->c_med_sn); | |
571 | free(chassis->c_med_manuf); | |
572 | free(chassis->c_med_model); | |
573 | free(chassis->c_med_asset); | |
574 | chassis->c_med_hw = dmi_hw(); | |
575 | chassis->c_med_fw = dmi_fw(); | |
576 | chassis->c_med_sn = dmi_sn(); | |
577 | chassis->c_med_manuf = dmi_manuf(); | |
578 | chassis->c_med_model = dmi_model(); | |
579 | chassis->c_med_asset = dmi_asset(); | |
580 | } | |
581 | #endif | |
582 | ||
8888d191 | 583 | static void |
6e75df87 | 584 | lldpd_update_localchassis(struct lldpd *cfg) |
43c02e7b | 585 | { |
6e75df87 VB |
586 | struct utsname un; |
587 | char *hp; | |
43c02e7b VB |
588 | int f; |
589 | char status; | |
6e75df87 | 590 | struct lldpd_hardware *hardware; |
43c02e7b VB |
591 | |
592 | /* Set system name and description */ | |
6e75df87 | 593 | if (uname(&un) != 0) |
43c02e7b | 594 | fatal("failed to get system information"); |
b5562b23 | 595 | if ((hp = priv_gethostbyname()) == NULL) |
43c02e7b | 596 | fatal("failed to get system name"); |
77507b69 VB |
597 | free(LOCAL_CHASSIS(cfg)->c_name); |
598 | free(LOCAL_CHASSIS(cfg)->c_descr); | |
599 | if ((LOCAL_CHASSIS(cfg)->c_name = strdup(hp)) == NULL) | |
249644a4 | 600 | fatal(NULL); |
77507b69 | 601 | if (asprintf(&LOCAL_CHASSIS(cfg)->c_descr, "%s %s %s %s", |
6e75df87 | 602 | un.sysname, un.release, un.version, un.machine) == -1) |
43c02e7b | 603 | fatal("failed to set system description"); |
43c02e7b VB |
604 | |
605 | /* Check forwarding */ | |
b5562b23 | 606 | if ((f = priv_open("/proc/sys/net/ipv4/ip_forward")) >= 0) { |
4afe659e | 607 | if ((read(f, &status, 1) == 1) && (status == '1')) { |
77507b69 | 608 | LOCAL_CHASSIS(cfg)->c_cap_enabled = LLDP_CAP_ROUTER; |
4afe659e | 609 | } |
43c02e7b VB |
610 | close(f); |
611 | } | |
89840df0 | 612 | #ifdef ENABLE_LLDPMED |
77507b69 VB |
613 | if (LOCAL_CHASSIS(cfg)->c_cap_available & LLDP_CAP_TELEPHONE) |
614 | LOCAL_CHASSIS(cfg)->c_cap_enabled |= LLDP_CAP_TELEPHONE; | |
615 | lldpd_med(LOCAL_CHASSIS(cfg)); | |
616 | free(LOCAL_CHASSIS(cfg)->c_med_sw); | |
6e75df87 | 617 | LOCAL_CHASSIS(cfg)->c_med_sw = strdup(un.release); |
89840df0 | 618 | #endif |
43c02e7b | 619 | |
6e75df87 VB |
620 | /* Set chassis ID if needed */ |
621 | if ((LOCAL_CHASSIS(cfg)->c_id == NULL) && | |
622 | (hardware = TAILQ_FIRST(&cfg->g_hardware))) { | |
623 | if ((LOCAL_CHASSIS(cfg)->c_id = | |
624 | malloc(sizeof(hardware->h_lladdr))) == NULL) | |
625 | fatal(NULL); | |
626 | LOCAL_CHASSIS(cfg)->c_id_subtype = LLDP_CHASSISID_SUBTYPE_LLADDR; | |
627 | LOCAL_CHASSIS(cfg)->c_id_len = sizeof(hardware->h_lladdr); | |
628 | memcpy(LOCAL_CHASSIS(cfg)->c_id, | |
629 | hardware->h_lladdr, sizeof(hardware->h_lladdr)); | |
630 | } | |
631 | } | |
632 | ||
633 | static void | |
634 | lldpd_update_localports(struct lldpd *cfg) | |
635 | { | |
636 | struct ifaddrs *ifap; | |
637 | struct lldpd_hardware *hardware; | |
638 | lldpd_ifhandlers ifhs[] = { | |
849954d7 | 639 | lldpd_ifh_bond, /* Handle bond */ |
6e75df87 | 640 | lldpd_ifh_eth, /* Handle classic ethernet interfaces */ |
5994b27d | 641 | #ifdef ENABLE_DOT1 |
6e75df87 | 642 | lldpd_ifh_vlan, /* Handle VLAN */ |
5994b27d | 643 | #endif |
6e75df87 VB |
644 | lldpd_ifh_mgmt, /* Handle management address (if not already handled) */ |
645 | NULL | |
646 | }; | |
647 | lldpd_ifhandlers *ifh; | |
648 | ||
649 | /* h_flags is set to 0 for each port. If the port is updated, h_flags | |
650 | * will be set to a non-zero value. This will allow us to clean up any | |
651 | * non up-to-date port */ | |
43c02e7b VB |
652 | TAILQ_FOREACH(hardware, &cfg->g_hardware, h_entries) |
653 | hardware->h_flags = 0; | |
654 | ||
77507b69 | 655 | LOCAL_CHASSIS(cfg)->c_mgmt.s_addr = INADDR_ANY; |
6e75df87 VB |
656 | if (getifaddrs(&ifap) != 0) |
657 | fatal("lldpd_update_localports: failed to get interface list"); | |
658 | ||
659 | /* We will run the list of interfaces through a list of interface | |
660 | * handlers. Each handler will create or update some hardware port (and | |
661 | * will set h_flags to a non zero value. The handler can use the list of | |
662 | * interfaces but this is not mandatory. If the interface handler | |
663 | * handles an interface from the list, it should set ifa_flags to 0 to | |
664 | * let know the other handlers that it took care of this interface. This | |
665 | * means that more specific handlers should be before less specific | |
666 | * ones. */ | |
667 | for (ifh = ifhs; *ifh != NULL; ifh++) | |
668 | (*ifh)(cfg, ifap); | |
43c02e7b | 669 | freeifaddrs(ifap); |
6e75df87 | 670 | } |
43c02e7b | 671 | |
6e75df87 VB |
672 | static void |
673 | lldpd_loop(struct lldpd *cfg) | |
674 | { | |
675 | /* Main loop. | |
676 | ||
677 | 1. Update local ports information | |
678 | 2. Clean unwanted (removed) local ports | |
679 | 3. Update local chassis information | |
680 | 4. Send packets | |
681 | 5. Receive packets | |
682 | */ | |
849954d7 | 683 | LOCAL_CHASSIS(cfg)->c_cap_enabled = 0; |
6e75df87 | 684 | lldpd_update_localports(cfg); |
43c02e7b | 685 | lldpd_cleanup(cfg); |
6e75df87 | 686 | lldpd_update_localchassis(cfg); |
43c02e7b VB |
687 | lldpd_send_all(cfg); |
688 | lldpd_recv_all(cfg); | |
689 | } | |
690 | ||
8888d191 | 691 | static void |
43c02e7b VB |
692 | lldpd_shutdown(int sig) |
693 | { | |
694 | LLOG_INFO("signal received, exiting"); | |
695 | exit(0); | |
696 | } | |
697 | ||
698 | /* For signal handling */ | |
8888d191 | 699 | static struct lldpd *gcfg = NULL; |
43c02e7b | 700 | |
8888d191 | 701 | static void |
43c02e7b VB |
702 | lldpd_exit() |
703 | { | |
6e75df87 | 704 | struct lldpd_hardware *hardware, *hardware_next; |
b5562b23 VB |
705 | close(gcfg->g_ctl); |
706 | priv_ctl_cleanup(); | |
6e75df87 VB |
707 | for (hardware = TAILQ_FIRST(&gcfg->g_hardware); hardware != NULL; |
708 | hardware = hardware_next) { | |
709 | hardware_next = TAILQ_NEXT(hardware, h_entries); | |
710 | lldpd_hardware_cleanup(gcfg, hardware); | |
43c02e7b VB |
711 | } |
712 | #ifdef USE_SNMP | |
713 | if (gcfg->g_snmp) | |
714 | agent_shutdown(); | |
715 | #endif /* USE_SNMP */ | |
716 | } | |
717 | ||
718 | int | |
719 | main(int argc, char *argv[]) | |
720 | { | |
721 | struct lldpd *cfg; | |
77507b69 | 722 | struct lldpd_chassis *lchassis; |
e809a587 VB |
723 | int ch, debug = 0; |
724 | #ifdef USE_SNMP | |
725 | int snmp = 0; | |
726 | #endif | |
43c02e7b | 727 | char *mgmtp = NULL; |
6a2fa591 VB |
728 | char *popt, opts[] = "vdxm:p:M:i@ "; |
729 | int i, found, vlan = 0; | |
89840df0 | 730 | #ifdef ENABLE_LLDPMED |
e809a587 | 731 | int lldpmed = 0, noinventory = 0; |
89840df0 | 732 | #endif |
43c02e7b VB |
733 | |
734 | saved_argv = argv; | |
735 | ||
736 | /* | |
737 | * Get and parse command line options | |
738 | */ | |
739 | popt = index(opts, '@'); | |
740 | for (i=0; protos[i].mode != 0; i++) { | |
741 | if (protos[i].enabled == 1) continue; | |
742 | *(popt++) = protos[i].arg; | |
743 | } | |
744 | *popt = '\0'; | |
745 | while ((ch = getopt(argc, argv, opts)) != -1) { | |
746 | switch (ch) { | |
6a2fa591 VB |
747 | case 'v': |
748 | vlan = 1; | |
749 | break; | |
43c02e7b VB |
750 | case 'd': |
751 | debug++; | |
752 | break; | |
753 | case 'm': | |
754 | mgmtp = optarg; | |
755 | break; | |
e809a587 | 756 | #ifdef ENABLE_LLDPMED |
115ff55c | 757 | case 'M': |
89840df0 | 758 | lldpmed = atoi(optarg); |
e809a587 VB |
759 | if ((lldpmed < 1) || (lldpmed > 4)) { |
760 | fprintf(stderr, "-M requires an argument between 1 and 4\n"); | |
89840df0 | 761 | usage(); |
e809a587 | 762 | } |
89840df0 | 763 | break; |
e809a587 | 764 | case 'i': |
e809a587 | 765 | noinventory = 1; |
115ff55c | 766 | break; |
e809a587 | 767 | #else |
115ff55c VB |
768 | case 'M': |
769 | case 'i': | |
115ff55c | 770 | case 'P': |
e809a587 VB |
771 | fprintf(stderr, "LLDP-MED support is not built-in\n"); |
772 | usage(); | |
e809a587 | 773 | break; |
115ff55c | 774 | #endif |
43c02e7b | 775 | case 'x': |
e809a587 | 776 | #ifdef USE_SNMP |
43c02e7b | 777 | snmp = 1; |
e809a587 VB |
778 | #else |
779 | fprintf(stderr, "SNMP support is not built-in\n"); | |
780 | usage(); | |
781 | #endif | |
43c02e7b VB |
782 | break; |
783 | default: | |
784 | found = 0; | |
785 | for (i=0; protos[i].mode != 0; i++) { | |
786 | if (protos[i].enabled) continue; | |
787 | if (ch == protos[i].arg) { | |
788 | protos[i].enabled = 1; | |
789 | found = 1; | |
790 | } | |
791 | } | |
792 | if (!found) | |
793 | usage(); | |
794 | } | |
795 | } | |
115ff55c | 796 | |
43c02e7b | 797 | log_init(debug); |
a2993d83 | 798 | |
eac2f38a VB |
799 | if (!debug) { |
800 | int pid; | |
801 | char *spid; | |
802 | if (daemon(0, 0) != 0) | |
803 | fatal("failed to detach daemon"); | |
804 | if ((pid = open(LLDPD_PID_FILE, | |
0aa5f676 | 805 | O_TRUNC | O_CREAT | O_WRONLY, 0644)) == -1) |
eac2f38a VB |
806 | fatal("unable to open pid file " LLDPD_PID_FILE); |
807 | if (asprintf(&spid, "%d\n", getpid()) == -1) | |
808 | fatal("unable to create pid file " LLDPD_PID_FILE); | |
809 | if (write(pid, spid, strlen(spid)) == -1) | |
810 | fatal("unable to write pid file " LLDPD_PID_FILE); | |
811 | free(spid); | |
812 | close(pid); | |
813 | } | |
814 | ||
a2993d83 | 815 | priv_init(PRIVSEP_CHROOT); |
43c02e7b | 816 | |
43c02e7b VB |
817 | if ((cfg = (struct lldpd *) |
818 | calloc(1, sizeof(struct lldpd))) == NULL) | |
819 | fatal(NULL); | |
820 | ||
766f32b3 | 821 | cfg->g_mgmt_pattern = mgmtp; |
6a2fa591 | 822 | cfg->g_listen_vlans = vlan; |
43c02e7b VB |
823 | |
824 | /* Get ioctl socket */ | |
825 | if ((cfg->g_sock = socket(AF_INET, SOCK_DGRAM, 0)) == -1) | |
826 | fatal("failed to get ioctl socket"); | |
827 | cfg->g_delay = LLDPD_TX_DELAY; | |
828 | ||
829 | /* Set system capabilities */ | |
77507b69 VB |
830 | if ((lchassis = (struct lldpd_chassis*) |
831 | calloc(1, sizeof(struct lldpd_chassis))) == NULL) | |
832 | fatal(NULL); | |
833 | lchassis->c_cap_available = LLDP_CAP_BRIDGE | LLDP_CAP_WLAN | | |
43c02e7b | 834 | LLDP_CAP_ROUTER; |
89840df0 VB |
835 | #ifdef ENABLE_LLDPMED |
836 | if (lldpmed > 0) { | |
837 | if (lldpmed == LLDPMED_CLASS_III) | |
77507b69 VB |
838 | lchassis->c_cap_available |= LLDP_CAP_TELEPHONE; |
839 | lchassis->c_med_type = lldpmed; | |
840 | lchassis->c_med_cap_available = LLDPMED_CAP_CAP | | |
115ff55c | 841 | LLDPMED_CAP_IV | LLDPMED_CAP_LOCATION; |
740593ff | 842 | cfg->g_noinventory = noinventory; |
42bddd41 VB |
843 | } else |
844 | cfg->g_noinventory = 1; | |
89840df0 | 845 | #endif |
43c02e7b VB |
846 | |
847 | /* Set TTL */ | |
77507b69 | 848 | lchassis->c_ttl = LLDPD_TTL; |
43c02e7b VB |
849 | |
850 | cfg->g_protocols = protos; | |
43c02e7b VB |
851 | for (i=0; protos[i].mode != 0; i++) |
852 | if (protos[i].enabled) { | |
43c02e7b VB |
853 | LLOG_INFO("protocol %s enabled", protos[i].name); |
854 | } else | |
855 | LLOG_INFO("protocol %s disabled", protos[i].name); | |
43c02e7b VB |
856 | |
857 | TAILQ_INIT(&cfg->g_hardware); | |
77507b69 VB |
858 | TAILQ_INIT(&cfg->g_chassis); |
859 | TAILQ_INSERT_TAIL(&cfg->g_chassis, lchassis, c_entries); | |
860 | lchassis->c_refcount++; | |
43c02e7b VB |
861 | |
862 | #ifdef USE_SNMP | |
863 | if (snmp) { | |
864 | cfg->g_snmp = 1; | |
865 | agent_init(cfg, debug); | |
866 | } | |
867 | #endif /* USE_SNMP */ | |
868 | ||
869 | /* Create socket */ | |
b5562b23 VB |
870 | if ((cfg->g_ctl = priv_ctl_create(cfg)) == -1) |
871 | fatalx("unable to create control socket " LLDPD_CTL_SOCKET); | |
872 | TAILQ_INIT(&cfg->g_clients); | |
43c02e7b | 873 | |
43c02e7b VB |
874 | gcfg = cfg; |
875 | if (atexit(lldpd_exit) != 0) { | |
b5562b23 VB |
876 | close(cfg->g_ctl); |
877 | priv_ctl_cleanup(); | |
43c02e7b VB |
878 | fatal("unable to set exit function"); |
879 | } | |
43c02e7b VB |
880 | |
881 | /* Signal handling */ | |
b5562b23 | 882 | signal(SIGHUP, lldpd_shutdown); |
43c02e7b VB |
883 | signal(SIGINT, lldpd_shutdown); |
884 | signal(SIGTERM, lldpd_shutdown); | |
885 | ||
886 | for (;;) | |
887 | lldpd_loop(cfg); | |
888 | ||
889 | return (0); | |
890 | } |