[thirdparty/systemd.git] / src / libsystemd-network / test-lldp-rx.c

/* SPDX-License-Identifier: LGPL-2.1-or-later */

#include <arpa/inet.h>
#include <errno.h>
#include <net/ethernet.h>
#include <stdio.h>
#include <unistd.h>

#include "sd-event.h"
#include "sd-lldp-rx.h"

#include "alloc-util.h"
#include "fd-util.h"
#include "lldp-network.h"
#include "macro.h"
#include "string-util.h"
#include "tests.h"

#define TEST_LLDP_PORT "em1"
#define TEST_LLDP_TYPE_SYSTEM_NAME "systemd-lldp"
#define TEST_LLDP_TYPE_SYSTEM_DESC "systemd-lldp-desc"

static int test_fd[2] = PIPE_EBADF;
static int lldp_rx_handler_calls;

int lldp_network_bind_raw_socket(int ifindex) {
        if (socketpair(AF_UNIX, SOCK_DGRAM | SOCK_CLOEXEC | SOCK_NONBLOCK, 0, test_fd) < 0)
                return -errno;

        return test_fd[0];
}

static void lldp_rx_handler(sd_lldp_rx *lldp_rx, sd_lldp_rx_event_t event, sd_lldp_neighbor *n, void *userdata) {
        lldp_rx_handler_calls++;
}

static int start_lldp_rx(sd_lldp_rx **lldp_rx, sd_event *e, sd_lldp_rx_callback_t cb, void *cb_data) {
        int r;

        r = sd_lldp_rx_new(lldp_rx);
        if (r < 0)
                return r;

        r = sd_lldp_rx_set_ifindex(*lldp_rx, 42);
        if (r < 0)
                return r;

        r = sd_lldp_rx_set_callback(*lldp_rx, cb, cb_data);
        if (r < 0)
                return r;

        r = sd_lldp_rx_attach_event(*lldp_rx, e, 0);
        if (r < 0)
                return r;

        r = sd_lldp_rx_start(*lldp_rx);
        if (r < 0)
                return r;

        return 0;
}

static int stop_lldp_rx(sd_lldp_rx *lldp_rx) {
        int r;

        r = sd_lldp_rx_stop(lldp_rx);
        if (r < 0)
                return r;

        r = sd_lldp_rx_detach_event(lldp_rx);
        if (r < 0)
                return r;

        sd_lldp_rx_unref(lldp_rx);
        safe_close(test_fd[1]);

        return 0;
}

static void test_receive_basic_packet(sd_event *e) {

        static const uint8_t frame[] = {
                /* Ethernet header */
                0x01, 0x80, 0xc2, 0x00, 0x00, 0x03,     /* Destination MAC */
                0x01, 0x02, 0x03, 0x04, 0x05, 0x06,     /* Source MAC */
                0x88, 0xcc,                             /* Ethertype */
                /* LLDP mandatory TLVs */
                0x02, 0x07, 0x04, 0x00, 0x01, 0x02,     /* Chassis: MAC, 00:01:02:03:04:05 */
                0x03, 0x04, 0x05,
                0x04, 0x04, 0x05, 0x31, 0x2f, 0x33,     /* Port: interface name, "1/3" */
                0x06, 0x02, 0x00, 0x78,                 /* TTL: 120 seconds */
                /* LLDP optional TLVs */
                0x08, 0x04, 0x50, 0x6f, 0x72, 0x74,     /* Port Description: "Port" */
                0x0a, 0x03, 0x53, 0x59, 0x53,           /* System Name: "SYS" */
                0x0c, 0x04, 0x66, 0x6f, 0x6f, 0x00,     /* System Description: "foo" (NULL-terminated) */
                0x00, 0x00                              /* End Of LLDPDU */
        };

        sd_lldp_rx *lldp_rx;
        sd_lldp_neighbor **neighbors;
        uint8_t type;
        const void *data;
        uint16_t ttl;
        size_t length;
        const char *str;

        lldp_rx_handler_calls = 0;
        assert_se(start_lldp_rx(&lldp_rx, e, lldp_rx_handler, NULL) == 0);

        assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
        sd_event_run(e, 0);
        assert_se(lldp_rx_handler_calls == 1);
        assert_se(sd_lldp_rx_get_neighbors(lldp_rx, &neighbors) == 1);

        assert_se(sd_lldp_neighbor_get_chassis_id(neighbors[0], &type, &data, &length) == 0);
        assert_se(type == SD_LLDP_CHASSIS_SUBTYPE_MAC_ADDRESS);
        assert_se(length == ETH_ALEN);
        assert_se(!memcmp(data, "\x00\x01\x02\x03\x04\x05", ETH_ALEN));

        assert_se(sd_lldp_neighbor_get_port_id(neighbors[0], &type, &data, &length) == 0);
        assert_se(type == SD_LLDP_PORT_SUBTYPE_INTERFACE_NAME);
        assert_se(length == 3);
        assert_se(!memcmp(data, "1/3", 3));

        assert_se(sd_lldp_neighbor_get_port_description(neighbors[0], &str) == 0);
        assert_se(streq(str, "Port"));

        assert_se(sd_lldp_neighbor_get_system_name(neighbors[0], &str) == 0);
        assert_se(streq(str, "SYS"));

        assert_se(sd_lldp_neighbor_get_system_description(neighbors[0], &str) == 0);
        assert_se(streq(str, "foo"));

        assert_se(sd_lldp_neighbor_get_ttl(neighbors[0], &ttl) == 0);
        assert_se(ttl == 120);

        sd_lldp_neighbor_unref(neighbors[0]);
        free(neighbors);

        assert_se(stop_lldp_rx(lldp_rx) == 0);
}

static void test_receive_incomplete_packet(sd_event *e) {
        sd_lldp_rx *lldp_rx;
        sd_lldp_neighbor **neighbors;
        uint8_t frame[] = {
                /* Ethernet header */
                0x01, 0x80, 0xc2, 0x00, 0x00, 0x03,     /* Destination MAC */
                0x01, 0x02, 0x03, 0x04, 0x05, 0x06,     /* Source MAC */
                0x88, 0xcc,                             /* Ethertype */
                /* LLDP mandatory TLVs */
                0x02, 0x07, 0x04, 0x00, 0x01, 0x02,     /* Chassis: MAC, 00:01:02:03:04:05 */
                0x03, 0x04, 0x05,
                0x04, 0x04, 0x05, 0x31, 0x2f, 0x33,     /* Port: interface name, "1/3" */
                                                        /* Missing TTL */
                0x00, 0x00                              /* End Of LLDPDU */
        };

        lldp_rx_handler_calls = 0;
        assert_se(start_lldp_rx(&lldp_rx, e, lldp_rx_handler, NULL) == 0);

        assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
        sd_event_run(e, 0);
        assert_se(lldp_rx_handler_calls == 0);
        assert_se(sd_lldp_rx_get_neighbors(lldp_rx, &neighbors) == 0);

        assert_se(stop_lldp_rx(lldp_rx) == 0);
}

static void test_receive_oui_packet(sd_event *e) {
        sd_lldp_rx *lldp_rx;
        sd_lldp_neighbor **neighbors;
        uint8_t frame[] = {
                /* Ethernet header */
                0x01, 0x80, 0xc2, 0x00, 0x00, 0x03,     /* Destination MAC */
                0x01, 0x02, 0x03, 0x04, 0x05, 0x06,     /* Source MAC */
                0x88, 0xcc,                             /* Ethertype */
                /* LLDP mandatory TLVs */
                0x02, 0x07, 0x04, 0x00, 0x01, 0x02,     /* Chassis: MAC, 00:01:02:03:04:05 */
                0x03, 0x04, 0x05,
                0x04, 0x04, 0x05, 0x31, 0x2f, 0x33,     /* Port TLV: interface name, "1/3" */
                0x06, 0x02, 0x00, 0x78,                 /* TTL: 120 seconds */
                /* LLDP optional TLVs */
                0xfe, 0x06, 0x00, 0x80, 0xc2, 0x01,     /* Port VLAN ID: 0x1234 */
                0x12, 0x34,
                0xfe, 0x07, 0x00, 0x80, 0xc2, 0x02,     /* Port and protocol: flag 1, PPVID 0x7788 */
                0x01, 0x77, 0x88,
                0xfe, 0x0d, 0x00, 0x80, 0xc2, 0x03,     /* VLAN Name: ID 0x1234, name "Vlan51" */
                0x12, 0x34, 0x06, 0x56, 0x6c, 0x61,
                0x6e, 0x35, 0x31,
                0xfe, 0x06, 0x00, 0x80, 0xc2, 0x06,     /* Management VID: 0x0102 */
                0x01, 0x02,
                0xfe, 0x09, 0x00, 0x80, 0xc2, 0x07,     /* Link aggregation: status 1, ID 0x00140012 */
                0x01, 0x00, 0x14, 0x00, 0x12,
                0xfe, 0x07, 0x00, 0x12, 0x0f, 0x02,     /* 802.3 Power via MDI: PSE, MDI enabled */
                0x07, 0x01, 0x00,
                0x00, 0x00                              /* End of LLDPDU */
        };

        lldp_rx_handler_calls = 0;
        assert_se(start_lldp_rx(&lldp_rx, e, lldp_rx_handler, NULL) == 0);

        assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
        sd_event_run(e, 0);
        assert_se(lldp_rx_handler_calls == 1);
        assert_se(sd_lldp_rx_get_neighbors(lldp_rx, &neighbors) == 1);

        assert_se(sd_lldp_neighbor_tlv_rewind(neighbors[0]) >= 0);
        assert_se(sd_lldp_neighbor_tlv_is_type(neighbors[0], SD_LLDP_TYPE_CHASSIS_ID) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
        assert_se(sd_lldp_neighbor_tlv_is_type(neighbors[0], SD_LLDP_TYPE_PORT_ID) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
        assert_se(sd_lldp_neighbor_tlv_is_type(neighbors[0], SD_LLDP_TYPE_TTL) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
        assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_PORT_VLAN_ID) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
        assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_PORT_PROTOCOL_VLAN_ID) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
        assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_VLAN_NAME) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
        assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_MANAGEMENT_VID) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
        assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_LINK_AGGREGATION) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
        assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_3, SD_LLDP_OUI_802_3_SUBTYPE_POWER_VIA_MDI) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
        assert_se(sd_lldp_neighbor_tlv_is_type(neighbors[0], SD_LLDP_TYPE_END) > 0);
        assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) == 0);

        sd_lldp_neighbor_unref(neighbors[0]);
        free(neighbors);

        assert_se(stop_lldp_rx(lldp_rx) == 0);
}

static void test_multiple_neighbors_sorted(sd_event *e) {

        static const uint8_t frame1[] = {
                /* Ethernet header */
                0x01, 0x80, 0xc2, 0x00, 0x00, 0x03,     /* Destination MAC */
                0x01, 0x02, 0x03, 0x04, 0x05, 0x06,     /* Source MAC */
                0x88, 0xcc,                             /* Ethertype */
                /* LLDP mandatory TLVs */
                0x02, 0x04, 0x01, '1', '/', '2',        /* Chassis component: "1/2" */
                0x04, 0x04, 0x02, '2', '/', '3',        /* Port component: "2/3" */
                0x06, 0x02, 0x00, 0x78,                 /* TTL: 120 seconds */
                0x00, 0x00                              /* End Of LLDPDU */
        };
        static const uint8_t frame2[] = {
                /* Ethernet header */
                0x01, 0x80, 0xc2, 0x00, 0x00, 0x03,     /* Destination MAC */
                0x01, 0x02, 0x03, 0x04, 0x05, 0x06,     /* Source MAC */
                0x88, 0xcc,                             /* Ethertype */
                /* LLDP mandatory TLVs */
                0x02, 0x04, 0x01, '2', '/', '1',        /* Chassis component: "2/1" */
                0x04, 0x04, 0x02, '1', '/', '3',        /* Port component: "1/3" */
                0x06, 0x02, 0x00, 0x78,                 /* TTL: 120 seconds */
                0x00, 0x00                              /* End Of LLDPDU */
        };
        static const uint8_t frame3[] = {
                /* Ethernet header */
                0x01, 0x80, 0xc2, 0x00, 0x00, 0x03,     /* Destination MAC */
                0x01, 0x02, 0x03, 0x04, 0x05, 0x06,     /* Source MAC */
                0x88, 0xcc,                             /* Ethertype */
                /* LLDP mandatory TLVs */
                0x02, 0x05, 0x01, '2', '/', '1', '0',   /* Chassis component: "2/10" */
                0x04, 0x04, 0x02, '1', '/', '0',        /* Port component: "1/0" */
                0x06, 0x02, 0x00, 0x78,                 /* TTL: 120 seconds */
                0x00, 0x00                              /* End Of LLDPDU */
        };
        static const uint8_t frame4[] = {
                /* Ethernet header */
                0x01, 0x80, 0xc2, 0x00, 0x00, 0x03,     /* Destination MAC */
                0x01, 0x02, 0x03, 0x04, 0x05, 0x06,     /* Source MAC */
                0x88, 0xcc,                             /* Ethertype */
                /* LLDP mandatory TLVs */
                0x02, 0x05, 0x01, '2', '/', '1', '9',   /* Chassis component: "2/19" */
                0x04, 0x04, 0x02, '1', '/', '0',        /* Port component: "1/0" */
                0x06, 0x02, 0x00, 0x78,                 /* TTL: 120 seconds */
                0x00, 0x00                              /* End Of LLDPDU */
        };
        static const uint8_t frame5[] = {
                /* Ethernet header */
                0x01, 0x80, 0xc2, 0x00, 0x00, 0x03,     /* Destination MAC */
                0x01, 0x02, 0x03, 0x04, 0x05, 0x06,     /* Source MAC */
                0x88, 0xcc,                             /* Ethertype */
                /* LLDP mandatory TLVs */
                0x02, 0x04, 0x01, '1', '/', '2',        /* Chassis component: "1/2" */
                0x04, 0x05, 0x02, '2', '/', '1', '0',   /* Port component: "2/10" */
                0x06, 0x02, 0x00, 0x78,                 /* TTL: 120 seconds */
                0x00, 0x00                              /* End Of LLDPDU */
        };
        static const uint8_t frame6[] = {
                /* Ethernet header */
                0x01, 0x80, 0xc2, 0x00, 0x00, 0x03,     /* Destination MAC */
                0x01, 0x02, 0x03, 0x04, 0x05, 0x06,     /* Source MAC */
                0x88, 0xcc,                             /* Ethertype */
                /* LLDP mandatory TLVs */
                0x02, 0x04, 0x01, '1', '/', '2',        /* Chassis component: "1/2" */
                0x04, 0x05, 0x02, '2', '/', '3', '9',   /* Port component: "2/10" */
                0x06, 0x02, 0x00, 0x78,                 /* TTL: 120 seconds */
                0x00, 0x00                              /* End Of LLDPDU */
        };
        static const char* expected[] = {
                /* ordered pairs of Chassis+Port */
                "1/2", "2/10",
                "1/2", "2/3",
                "1/2", "2/39",
                "2/1", "1/3",
                "2/10", "1/0",
                "2/19", "1/0",
        };

        sd_lldp_rx *lldp_rx;
        sd_lldp_neighbor **neighbors;
        int i;
        uint8_t type;
        const void *data;
        size_t length, expected_length;
        uint16_t ttl;

        lldp_rx_handler_calls = 0;
        assert_se(start_lldp_rx(&lldp_rx, e, lldp_rx_handler, NULL) == 0);

        assert_se(write(test_fd[1], frame1, sizeof(frame1)) == sizeof(frame1));
        sd_event_run(e, 0);
        assert_se(write(test_fd[1], frame2, sizeof(frame2)) == sizeof(frame2));
        sd_event_run(e, 0);
        assert_se(write(test_fd[1], frame3, sizeof(frame3)) == sizeof(frame3));
        sd_event_run(e, 0);
        assert_se(write(test_fd[1], frame4, sizeof(frame4)) == sizeof(frame4));
        sd_event_run(e, 0);
        assert_se(write(test_fd[1], frame5, sizeof(frame5)) == sizeof(frame5));
        sd_event_run(e, 0);
        assert_se(write(test_fd[1], frame6, sizeof(frame6)) == sizeof(frame6));
        sd_event_run(e, 0);
        assert_se(lldp_rx_handler_calls == 6);

        assert_se(sd_lldp_rx_get_neighbors(lldp_rx, &neighbors) == 6);

        for (i = 0; i < 6; i++) {
                assert_se(sd_lldp_neighbor_get_chassis_id(neighbors[i], &type, &data, &length) == 0);
                assert_se(type == SD_LLDP_CHASSIS_SUBTYPE_CHASSIS_COMPONENT);
                expected_length = strlen(expected[2 * i]);
                assert_se(length == expected_length);
                assert_se(memcmp(data, expected[2 * i], expected_length) == 0);

                assert_se(sd_lldp_neighbor_get_port_id(neighbors[i], &type, &data, &length) == 0);
                assert_se(type == SD_LLDP_PORT_SUBTYPE_PORT_COMPONENT);
                expected_length = strlen(expected[2 * i + 1]);
                assert_se(length == expected_length);
                assert_se(memcmp(data, expected[2 * i + 1], expected_length) == 0);

                assert_se(sd_lldp_neighbor_get_ttl(neighbors[i], &ttl) == 0);
                assert_se(ttl == 120);
        }

        for (i = 0; i < 6; i++)
                sd_lldp_neighbor_unref(neighbors[i]);
        free(neighbors);

        assert_se(stop_lldp_rx(lldp_rx) == 0);
}

int main(int argc, char *argv[]) {
        _cleanup_(sd_event_unrefp) sd_event *e = NULL;

        test_setup_logging(LOG_DEBUG);

        /* LLDP reception tests */
        assert_se(sd_event_new(&e) == 0);
        test_receive_basic_packet(e);
        test_receive_incomplete_packet(e);
        test_receive_oui_packet(e);
        test_multiple_neighbors_sorted(e);

        return 0;
}
Commit	Line	Data
db9ecf05	1	/* SPDX-License-Identifier: LGPL-2.1-or-later */
ad1ad5c8	2
07630cea	3	#include <arpa/inet.h>
dccca82b	4	#include <errno.h>
07630cea	5	#include <net/ethernet.h>
ad1ad5c8	6	#include <stdio.h>
34437b4f	7	#include <unistd.h>
ad1ad5c8	8
6fd255cf	9	#include "sd-event.h"
3a2ee855	10	#include "sd-lldp-rx.h"
07630cea	11
b5efdb8a	12	#include "alloc-util.h"
3ffd4af2	13	#include "fd-util.h"
6fd255cf	14	#include "lldp-network.h"
07630cea LP	15	#include "macro.h"
07630cea LP	16	#include "string-util.h"
f68a2622	17	#include "tests.h"
ad1ad5c8 SS	18
	19	#define TEST_LLDP_PORT "em1"
	20	#define TEST_LLDP_TYPE_SYSTEM_NAME "systemd-lldp"
	21	#define TEST_LLDP_TYPE_SYSTEM_DESC "systemd-lldp-desc"
	22
19ee48a6	23	static int test_fd[2] = PIPE_EBADF;
35778343	24	static int lldp_rx_handler_calls;
6fd255cf BG	25
6fd255cf BG	26	int lldp_network_bind_raw_socket(int ifindex) {
3e29b889	27	if (socketpair(AF_UNIX, SOCK_DGRAM \| SOCK_CLOEXEC \| SOCK_NONBLOCK, 0, test_fd) < 0)
6fd255cf BG	28	return -errno;
	29
	30	return test_fd[0];
	31	}
	32
35778343 YW	33	static void lldp_rx_handler(sd_lldp_rx lldp_rx, sd_lldp_rx_event_t event, sd_lldp_neighbor n, void *userdata) {
35778343 YW	34	lldp_rx_handler_calls++;
6fd255cf BG	35	}
6fd255cf BG	36
35778343	37	static int start_lldp_rx(sd_lldp_rx *lldp_rx, sd_event e, sd_lldp_rx_callback_t cb, void *cb_data) {
6fd255cf BG	38	int r;
6fd255cf BG	39
35778343	40	r = sd_lldp_rx_new(lldp_rx);
bd8650e9	41	if (r < 0)
6fd255cf BG	42	return r;
6fd255cf BG	43
35778343	44	r = sd_lldp_rx_set_ifindex(*lldp_rx, 42);
bd8650e9	45	if (r < 0)
6fd255cf BG	46	return r;
6fd255cf BG	47
35778343	48	r = sd_lldp_rx_set_callback(*lldp_rx, cb, cb_data);
bd8650e9	49	if (r < 0)
6fd255cf BG	50	return r;
6fd255cf BG	51
35778343	52	r = sd_lldp_rx_attach_event(*lldp_rx, e, 0);
fc6a313b LP	53	if (r < 0)
	54	return r;
	55
35778343	56	r = sd_lldp_rx_start(*lldp_rx);
bd8650e9	57	if (r < 0)
6fd255cf BG	58	return r;
	59
	60	return 0;
	61	}
	62
35778343	63	static int stop_lldp_rx(sd_lldp_rx *lldp_rx) {
6fd255cf BG	64	int r;
6fd255cf BG	65
35778343	66	r = sd_lldp_rx_stop(lldp_rx);
bd8650e9	67	if (r < 0)
6fd255cf BG	68	return r;
6fd255cf BG	69
35778343	70	r = sd_lldp_rx_detach_event(lldp_rx);
bd8650e9	71	if (r < 0)
6fd255cf BG	72	return r;
6fd255cf BG	73
35778343	74	sd_lldp_rx_unref(lldp_rx);
6fd255cf BG	75	safe_close(test_fd[1]);
	76
	77	return 0;
	78	}
	79
	80	static void test_receive_basic_packet(sd_event *e) {
34437b4f LP	81
34437b4f LP	82	static const uint8_t frame[] = {
6fd255cf	83	/* Ethernet header */
13e785f7	84	0x01, 0x80, 0xc2, 0x00, 0x00, 0x03, /* Destination MAC */
6fd255cf BG	85	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, /* Source MAC */
	86	0x88, 0xcc, /* Ethertype */
	87	/* LLDP mandatory TLVs */
	88	0x02, 0x07, 0x04, 0x00, 0x01, 0x02, /* Chassis: MAC, 00:01:02:03:04:05 */
	89	0x03, 0x04, 0x05,
	90	0x04, 0x04, 0x05, 0x31, 0x2f, 0x33, /* Port: interface name, "1/3" */
13e785f7	91	0x06, 0x02, 0x00, 0x78, /* TTL: 120 seconds */
6fd255cf BG	92	/* LLDP optional TLVs */
	93	0x08, 0x04, 0x50, 0x6f, 0x72, 0x74, /* Port Description: "Port" */
	94	0x0a, 0x03, 0x53, 0x59, 0x53, /* System Name: "SYS" */
	95	0x0c, 0x04, 0x66, 0x6f, 0x6f, 0x00, /* System Description: "foo" (NULL-terminated) */
	96	0x00, 0x00 /* End Of LLDPDU */
	97	};
	98
35778343	99	sd_lldp_rx *lldp_rx;
34437b4f LP	100	sd_lldp_neighbor **neighbors;
	101	uint8_t type;
	102	const void *data;
	103	uint16_t ttl;
	104	size_t length;
	105	const char *str;
	106
35778343 YW	107	lldp_rx_handler_calls = 0;
35778343 YW	108	assert_se(start_lldp_rx(&lldp_rx, e, lldp_rx_handler, NULL) == 0);
6fd255cf BG	109
	110	assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
	111	sd_event_run(e, 0);
35778343 YW	112	assert_se(lldp_rx_handler_calls == 1);
35778343 YW	113	assert_se(sd_lldp_rx_get_neighbors(lldp_rx, &neighbors) == 1);
6fd255cf	114
34437b4f	115	assert_se(sd_lldp_neighbor_get_chassis_id(neighbors[0], &type, &data, &length) == 0);
6afa6767	116	assert_se(type == SD_LLDP_CHASSIS_SUBTYPE_MAC_ADDRESS);
6fd255cf BG	117	assert_se(length == ETH_ALEN);
	118	assert_se(!memcmp(data, "\x00\x01\x02\x03\x04\x05", ETH_ALEN));
	119
34437b4f	120	assert_se(sd_lldp_neighbor_get_port_id(neighbors[0], &type, &data, &length) == 0);
6afa6767	121	assert_se(type == SD_LLDP_PORT_SUBTYPE_INTERFACE_NAME);
6fd255cf	122	assert_se(length == 3);
ce691f31	123	assert_se(!memcmp(data, "1/3", 3));
6fd255cf	124
34437b4f LP	125	assert_se(sd_lldp_neighbor_get_port_description(neighbors[0], &str) == 0);
34437b4f LP	126	assert_se(streq(str, "Port"));
6fd255cf	127
34437b4f LP	128	assert_se(sd_lldp_neighbor_get_system_name(neighbors[0], &str) == 0);
34437b4f LP	129	assert_se(streq(str, "SYS"));
6fd255cf	130
34437b4f LP	131	assert_se(sd_lldp_neighbor_get_system_description(neighbors[0], &str) == 0);
34437b4f LP	132	assert_se(streq(str, "foo"));
6fd255cf	133
34437b4f	134	assert_se(sd_lldp_neighbor_get_ttl(neighbors[0], &ttl) == 0);
6fd255cf BG	135	assert_se(ttl == 120);
6fd255cf BG	136
34437b4f LP	137	sd_lldp_neighbor_unref(neighbors[0]);
34437b4f LP	138	free(neighbors);
6fd255cf	139
35778343	140	assert_se(stop_lldp_rx(lldp_rx) == 0);
6fd255cf BG	141	}
	142
	143	static void test_receive_incomplete_packet(sd_event *e) {
35778343	144	sd_lldp_rx *lldp_rx;
34437b4f	145	sd_lldp_neighbor **neighbors;
6fd255cf BG	146	uint8_t frame[] = {
6fd255cf BG	147	/* Ethernet header */
13e785f7	148	0x01, 0x80, 0xc2, 0x00, 0x00, 0x03, /* Destination MAC */
6fd255cf BG	149	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, /* Source MAC */
	150	0x88, 0xcc, /* Ethertype */
	151	/* LLDP mandatory TLVs */
	152	0x02, 0x07, 0x04, 0x00, 0x01, 0x02, /* Chassis: MAC, 00:01:02:03:04:05 */
	153	0x03, 0x04, 0x05,
	154	0x04, 0x04, 0x05, 0x31, 0x2f, 0x33, /* Port: interface name, "1/3" */
	155	/* Missing TTL */
	156	0x00, 0x00 /* End Of LLDPDU */
	157	};
	158
35778343 YW	159	lldp_rx_handler_calls = 0;
35778343 YW	160	assert_se(start_lldp_rx(&lldp_rx, e, lldp_rx_handler, NULL) == 0);
6fd255cf BG	161
	162	assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
	163	sd_event_run(e, 0);
35778343 YW	164	assert_se(lldp_rx_handler_calls == 0);
35778343 YW	165	assert_se(sd_lldp_rx_get_neighbors(lldp_rx, &neighbors) == 0);
6fd255cf	166
35778343	167	assert_se(stop_lldp_rx(lldp_rx) == 0);
6fd255cf BG	168	}
	169
	170	static void test_receive_oui_packet(sd_event *e) {
35778343	171	sd_lldp_rx *lldp_rx;
34437b4f	172	sd_lldp_neighbor **neighbors;
6fd255cf BG	173	uint8_t frame[] = {
6fd255cf BG	174	/* Ethernet header */
13e785f7	175	0x01, 0x80, 0xc2, 0x00, 0x00, 0x03, /* Destination MAC */
6fd255cf BG	176	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, /* Source MAC */
	177	0x88, 0xcc, /* Ethertype */
	178	/* LLDP mandatory TLVs */
	179	0x02, 0x07, 0x04, 0x00, 0x01, 0x02, /* Chassis: MAC, 00:01:02:03:04:05 */
	180	0x03, 0x04, 0x05,
	181	0x04, 0x04, 0x05, 0x31, 0x2f, 0x33, /* Port TLV: interface name, "1/3" */
13e785f7	182	0x06, 0x02, 0x00, 0x78, /* TTL: 120 seconds */
6fd255cf BG	183	/* LLDP optional TLVs */
	184	0xfe, 0x06, 0x00, 0x80, 0xc2, 0x01, /* Port VLAN ID: 0x1234 */
	185	0x12, 0x34,
	186	0xfe, 0x07, 0x00, 0x80, 0xc2, 0x02, /* Port and protocol: flag 1, PPVID 0x7788 */
	187	0x01, 0x77, 0x88,
	188	0xfe, 0x0d, 0x00, 0x80, 0xc2, 0x03, /* VLAN Name: ID 0x1234, name "Vlan51" */
	189	0x12, 0x34, 0x06, 0x56, 0x6c, 0x61,
	190	0x6e, 0x35, 0x31,
	191	0xfe, 0x06, 0x00, 0x80, 0xc2, 0x06, /* Management VID: 0x0102 */
	192	0x01, 0x02,
	193	0xfe, 0x09, 0x00, 0x80, 0xc2, 0x07, /* Link aggregation: status 1, ID 0x00140012 */
	194	0x01, 0x00, 0x14, 0x00, 0x12,
dc8dc0cf BG	195	0xfe, 0x07, 0x00, 0x12, 0x0f, 0x02, /* 802.3 Power via MDI: PSE, MDI enabled */
dc8dc0cf BG	196	0x07, 0x01, 0x00,
6fd255cf BG	197	0x00, 0x00 /* End of LLDPDU */
	198	};
	199
35778343 YW	200	lldp_rx_handler_calls = 0;
35778343 YW	201	assert_se(start_lldp_rx(&lldp_rx, e, lldp_rx_handler, NULL) == 0);
6fd255cf BG	202
	203	assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
	204	sd_event_run(e, 0);
35778343 YW	205	assert_se(lldp_rx_handler_calls == 1);
35778343 YW	206	assert_se(sd_lldp_rx_get_neighbors(lldp_rx, &neighbors) == 1);
34437b4f LP	207
34437b4f LP	208	assert_se(sd_lldp_neighbor_tlv_rewind(neighbors[0]) >= 0);
6afa6767	209	assert_se(sd_lldp_neighbor_tlv_is_type(neighbors[0], SD_LLDP_TYPE_CHASSIS_ID) > 0);
34437b4f	210	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
6afa6767	211	assert_se(sd_lldp_neighbor_tlv_is_type(neighbors[0], SD_LLDP_TYPE_PORT_ID) > 0);
34437b4f	212	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
6afa6767	213	assert_se(sd_lldp_neighbor_tlv_is_type(neighbors[0], SD_LLDP_TYPE_TTL) > 0);
34437b4f	214	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
6afa6767	215	assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_PORT_VLAN_ID) > 0);
34437b4f	216	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
6afa6767	217	assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_PORT_PROTOCOL_VLAN_ID) > 0);
34437b4f	218	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
6afa6767	219	assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_VLAN_NAME) > 0);
34437b4f	220	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
6afa6767	221	assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_MANAGEMENT_VID) > 0);
34437b4f	222	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
6afa6767	223	assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_1, SD_LLDP_OUI_802_1_SUBTYPE_LINK_AGGREGATION) > 0);
34437b4f	224	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
dc8dc0cf BG	225	assert_se(sd_lldp_neighbor_tlv_is_oui(neighbors[0], SD_LLDP_OUI_802_3, SD_LLDP_OUI_802_3_SUBTYPE_POWER_VIA_MDI) > 0);
dc8dc0cf BG	226	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) > 0);
6afa6767	227	assert_se(sd_lldp_neighbor_tlv_is_type(neighbors[0], SD_LLDP_TYPE_END) > 0);
34437b4f LP	228	assert_se(sd_lldp_neighbor_tlv_next(neighbors[0]) == 0);
	229
	230	sd_lldp_neighbor_unref(neighbors[0]);
	231	free(neighbors);
6fd255cf	232
35778343	233	assert_se(stop_lldp_rx(lldp_rx) == 0);
6fd255cf BG	234	}
6fd255cf BG	235
7f099205 FB	236	static void test_multiple_neighbors_sorted(sd_event *e) {
	237
	238	static const uint8_t frame1[] = {
	239	/* Ethernet header */
	240	0x01, 0x80, 0xc2, 0x00, 0x00, 0x03, /* Destination MAC */
	241	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, /* Source MAC */
	242	0x88, 0xcc, /* Ethertype */
	243	/* LLDP mandatory TLVs */
	244	0x02, 0x04, 0x01, '1', '/', '2', /* Chassis component: "1/2" */
	245	0x04, 0x04, 0x02, '2', '/', '3', /* Port component: "2/3" */
	246	0x06, 0x02, 0x00, 0x78, /* TTL: 120 seconds */
	247	0x00, 0x00 /* End Of LLDPDU */
	248	};
	249	static const uint8_t frame2[] = {
	250	/* Ethernet header */
	251	0x01, 0x80, 0xc2, 0x00, 0x00, 0x03, /* Destination MAC */
	252	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, /* Source MAC */
	253	0x88, 0xcc, /* Ethertype */
	254	/* LLDP mandatory TLVs */
	255	0x02, 0x04, 0x01, '2', '/', '1', /* Chassis component: "2/1" */
	256	0x04, 0x04, 0x02, '1', '/', '3', /* Port component: "1/3" */
	257	0x06, 0x02, 0x00, 0x78, /* TTL: 120 seconds */
	258	0x00, 0x00 /* End Of LLDPDU */
	259	};
	260	static const uint8_t frame3[] = {
	261	/* Ethernet header */
	262	0x01, 0x80, 0xc2, 0x00, 0x00, 0x03, /* Destination MAC */
	263	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, /* Source MAC */
	264	0x88, 0xcc, /* Ethertype */
	265	/* LLDP mandatory TLVs */
	266	0x02, 0x05, 0x01, '2', '/', '1', '0', /* Chassis component: "2/10" */
	267	0x04, 0x04, 0x02, '1', '/', '0', /* Port component: "1/0" */
	268	0x06, 0x02, 0x00, 0x78, /* TTL: 120 seconds */
	269	0x00, 0x00 /* End Of LLDPDU */
	270	};
	271	static const uint8_t frame4[] = {
	272	/* Ethernet header */
	273	0x01, 0x80, 0xc2, 0x00, 0x00, 0x03, /* Destination MAC */
	274	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, /* Source MAC */
	275	0x88, 0xcc, /* Ethertype */
	276	/* LLDP mandatory TLVs */
	277	0x02, 0x05, 0x01, '2', '/', '1', '9', /* Chassis component: "2/19" */
	278	0x04, 0x04, 0x02, '1', '/', '0', /* Port component: "1/0" */
	279	0x06, 0x02, 0x00, 0x78, /* TTL: 120 seconds */
	280	0x00, 0x00 /* End Of LLDPDU */
	281	};
	282	static const uint8_t frame5[] = {
	283	/* Ethernet header */
	284	0x01, 0x80, 0xc2, 0x00, 0x00, 0x03, /* Destination MAC */
	285	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, /* Source MAC */
	286	0x88, 0xcc, /* Ethertype */
	287	/* LLDP mandatory TLVs */
	288	0x02, 0x04, 0x01, '1', '/', '2', /* Chassis component: "1/2" */
	289	0x04, 0x05, 0x02, '2', '/', '1', '0', /* Port component: "2/10" */
	290	0x06, 0x02, 0x00, 0x78, /* TTL: 120 seconds */
	291	0x00, 0x00 /* End Of LLDPDU */
	292	};
	293	static const uint8_t frame6[] = {
	294	/* Ethernet header */
	295	0x01, 0x80, 0xc2, 0x00, 0x00, 0x03, /* Destination MAC */
	296	0x01, 0x02, 0x03, 0x04, 0x05, 0x06, /* Source MAC */
	297	0x88, 0xcc, /* Ethertype */
	298	/* LLDP mandatory TLVs */
	299	0x02, 0x04, 0x01, '1', '/', '2', /* Chassis component: "1/2" */
300	0x04, 0x05, 0x02, '2', '/', '3', '9', /* Port component: "2/10" */
301	0x06, 0x02, 0x00, 0x78, /* TTL: 120 seconds */
302	0x00, 0x00 /* End Of LLDPDU */
303	};
304	static const char* expected[] = {
305	/* ordered pairs of Chassis+Port */
306	"1/2", "2/10",
307	"1/2", "2/3",
308	"1/2", "2/39",
309	"2/1", "1/3",
310	"2/10", "1/0",
311	"2/19", "1/0",
312	};
313
35778343	314	sd_lldp_rx *lldp_rx;
7f099205 FB	315	sd_lldp_neighbor **neighbors;
	316	int i;
	317	uint8_t type;
	318	const void *data;
	319	size_t length, expected_length;
	320	uint16_t ttl;
	321
35778343 YW	322	lldp_rx_handler_calls = 0;
35778343 YW	323	assert_se(start_lldp_rx(&lldp_rx, e, lldp_rx_handler, NULL) == 0);
7f099205 FB	324
	325	assert_se(write(test_fd[1], frame1, sizeof(frame1)) == sizeof(frame1));
	326	sd_event_run(e, 0);
	327	assert_se(write(test_fd[1], frame2, sizeof(frame2)) == sizeof(frame2));
	328	sd_event_run(e, 0);
	329	assert_se(write(test_fd[1], frame3, sizeof(frame3)) == sizeof(frame3));
	330	sd_event_run(e, 0);
	331	assert_se(write(test_fd[1], frame4, sizeof(frame4)) == sizeof(frame4));
	332	sd_event_run(e, 0);
	333	assert_se(write(test_fd[1], frame5, sizeof(frame5)) == sizeof(frame5));
	334	sd_event_run(e, 0);
	335	assert_se(write(test_fd[1], frame6, sizeof(frame6)) == sizeof(frame6));
	336	sd_event_run(e, 0);
35778343	337	assert_se(lldp_rx_handler_calls == 6);
7f099205	338
35778343	339	assert_se(sd_lldp_rx_get_neighbors(lldp_rx, &neighbors) == 6);
7f099205 FB	340
	341	for (i = 0; i < 6; i++) {
	342	assert_se(sd_lldp_neighbor_get_chassis_id(neighbors[i], &type, &data, &length) == 0);
	343	assert_se(type == SD_LLDP_CHASSIS_SUBTYPE_CHASSIS_COMPONENT);
	344	expected_length = strlen(expected[2 * i]);
	345	assert_se(length == expected_length);
	346	assert_se(memcmp(data, expected[2 * i], expected_length) == 0);
	347
	348	assert_se(sd_lldp_neighbor_get_port_id(neighbors[i], &type, &data, &length) == 0);
	349	assert_se(type == SD_LLDP_PORT_SUBTYPE_PORT_COMPONENT);
	350	expected_length = strlen(expected[2 * i + 1]);
	351	assert_se(length == expected_length);
	352	assert_se(memcmp(data, expected[2 * i + 1], expected_length) == 0);
	353
	354	assert_se(sd_lldp_neighbor_get_ttl(neighbors[i], &ttl) == 0);
	355	assert_se(ttl == 120);
	356	}
	357
	358	for (i = 0; i < 6; i++)
	359	sd_lldp_neighbor_unref(neighbors[i]);
	360	free(neighbors);
	361
35778343	362	assert_se(stop_lldp_rx(lldp_rx) == 0);
7f099205 FB	363	}
7f099205 FB	364
6fd255cf	365	int main(int argc, char *argv[]) {
4afd3348	366	_cleanup_(sd_event_unrefp) sd_event *e = NULL;
6fd255cf	367
f68a2622	368	test_setup_logging(LOG_DEBUG);
6fd255cf BG	369
	370	/* LLDP reception tests */
	371	assert_se(sd_event_new(&e) == 0);
	372	test_receive_basic_packet(e);
	373	test_receive_incomplete_packet(e);
	374	test_receive_oui_packet(e);
7f099205	375	test_multiple_neighbors_sorted(e);
ad1ad5c8 SS	376
	377	return 0;
	378	}