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

/* SPDX-License-Identifier: LGPL-2.1+ */

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

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

#include "alloc-util.h"
#include "fd-util.h"
#include "lldp-network.h"
#include "macro.h"
#include "string-util.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] = { -1, -1 };
static int lldp_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_handler(sd_lldp *lldp, sd_lldp_event event, sd_lldp_neighbor *n, void *userdata) {
        lldp_handler_calls++;
}

static int start_lldp(sd_lldp **lldp, sd_event *e, sd_lldp_callback_t cb, void *cb_data) {
        int r;

        r = sd_lldp_new(lldp);
        if (r < 0)
                return r;

        r = sd_lldp_set_ifindex(*lldp, 42);
        if (r < 0)
                return r;

        r = sd_lldp_set_callback(*lldp, cb, cb_data);
        if (r < 0)
                return r;

        r = sd_lldp_attach_event(*lldp, e, 0);
        if (r < 0)
                return r;

        r = sd_lldp_start(*lldp);
        if (r < 0)
                return r;

        return 0;
}

static int stop_lldp(sd_lldp *lldp) {
        int r;

        r = sd_lldp_stop(lldp);
        if (r < 0)
                return r;

        r = sd_lldp_detach_event(lldp);
        if (r < 0)
                return r;

        sd_lldp_unref(lldp);
        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 *lldp;
        sd_lldp_neighbor **neighbors;
        uint8_t type;
        const void *data;
        uint16_t ttl;
        size_t length;
        const char *str;

        lldp_handler_calls = 0;
        assert_se(start_lldp(&lldp, e, lldp_handler, NULL) == 0);

        assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
        sd_event_run(e, 0);
        assert_se(lldp_handler_calls == 1);
        assert_se(sd_lldp_get_neighbors(lldp, &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(lldp) == 0);
}

static void test_receive_incomplete_packet(sd_event *e) {
        sd_lldp *lldp;
        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_handler_calls = 0;
        assert_se(start_lldp(&lldp, e, lldp_handler, NULL) == 0);

        assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
        sd_event_run(e, 0);
        assert_se(lldp_handler_calls == 0);
        assert_se(sd_lldp_get_neighbors(lldp, &neighbors) == 0);

        assert_se(stop_lldp(lldp) == 0);
}

static void test_receive_oui_packet(sd_event *e) {
        sd_lldp *lldp;
        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_handler_calls = 0;
        assert_se(start_lldp(&lldp, e, lldp_handler, NULL) == 0);

        assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
        sd_event_run(e, 0);
        assert_se(lldp_handler_calls == 1);
        assert_se(sd_lldp_get_neighbors(lldp, &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(lldp) == 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 *lldp;
        sd_lldp_neighbor **neighbors;
        int i;
        uint8_t type;
        const void *data;
        size_t length, expected_length;
        uint16_t ttl;

        lldp_handler_calls = 0;
        assert_se(start_lldp(&lldp, e, lldp_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_handler_calls == 6);

        assert_se(sd_lldp_get_neighbors(lldp, &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(lldp) == 0);
}

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

        log_set_max_level(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
53e1b683	1	/* SPDX-License-Identifier: LGPL-2.1+ */
ad1ad5c8	2
07630cea	3	#include <arpa/inet.h>
dccca82b	4	#include <errno.h>
07630cea	5	#include <net/ethernet.h>
ad1ad5c8	6	#include <stdio.h>
ad1ad5c8	7	#include <string.h>
34437b4f	8	#include <unistd.h>
ad1ad5c8	9
6fd255cf	10	#include "sd-event.h"
07630cea LP	11	#include "sd-lldp.h"
07630cea LP	12
b5efdb8a	13	#include "alloc-util.h"
3ffd4af2	14	#include "fd-util.h"
6fd255cf	15	#include "lldp-network.h"
07630cea LP	16	#include "macro.h"
07630cea LP	17	#include "string-util.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
34437b4f LP	23	static int test_fd[2] = { -1, -1 };
34437b4f LP	24	static int lldp_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
90dffb22	33	static void lldp_handler(sd_lldp lldp, sd_lldp_event event, sd_lldp_neighbor n, void *userdata) {
6fd255cf BG	34	lldp_handler_calls++;
	35	}
	36
ccf86354	37	static int start_lldp(sd_lldp *lldp, sd_event e, sd_lldp_callback_t cb, void *cb_data) {
6fd255cf BG	38	int r;
6fd255cf BG	39
fc6a313b	40	r = sd_lldp_new(lldp);
bd8650e9	41	if (r < 0)
6fd255cf BG	42	return r;
6fd255cf BG	43
fc6a313b	44	r = sd_lldp_set_ifindex(*lldp, 42);
bd8650e9	45	if (r < 0)
6fd255cf BG	46	return r;
	47
	48	r = sd_lldp_set_callback(*lldp, cb, cb_data);
bd8650e9	49	if (r < 0)
6fd255cf BG	50	return r;
6fd255cf BG	51
fc6a313b LP	52	r = sd_lldp_attach_event(*lldp, e, 0);
	53	if (r < 0)
	54	return r;
	55
6fd255cf	56	r = sd_lldp_start(*lldp);
bd8650e9	57	if (r < 0)
6fd255cf BG	58	return r;
	59
	60	return 0;
	61	}
	62
	63	static int stop_lldp(sd_lldp *lldp) {
	64	int r;
	65
	66	r = sd_lldp_stop(lldp);
bd8650e9	67	if (r < 0)
6fd255cf BG	68	return r;
	69
	70	r = sd_lldp_detach_event(lldp);
bd8650e9	71	if (r < 0)
6fd255cf BG	72	return r;
6fd255cf BG	73
4afd3348	74	sd_lldp_unref(lldp);
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
34437b4f LP	99	sd_lldp *lldp;
	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
6fd255cf BG	107	lldp_handler_calls = 0;
	108	assert_se(start_lldp(&lldp, e, lldp_handler, NULL) == 0);
	109
	110	assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
	111	sd_event_run(e, 0);
	112	assert_se(lldp_handler_calls == 1);
34437b4f	113	assert_se(sd_lldp_get_neighbors(lldp, &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 BG	139
	140	assert_se(stop_lldp(lldp) == 0);
	141	}
	142
	143	static void test_receive_incomplete_packet(sd_event *e) {
	144	sd_lldp *lldp;
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
	159	lldp_handler_calls = 0;
	160	assert_se(start_lldp(&lldp, e, lldp_handler, NULL) == 0);
	161
	162	assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
	163	sd_event_run(e, 0);
	164	assert_se(lldp_handler_calls == 0);
34437b4f	165	assert_se(sd_lldp_get_neighbors(lldp, &neighbors) == 0);
6fd255cf BG	166
	167	assert_se(stop_lldp(lldp) == 0);
	168	}
	169
	170	static void test_receive_oui_packet(sd_event *e) {
	171	sd_lldp *lldp;
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
	200	lldp_handler_calls = 0;
	201	assert_se(start_lldp(&lldp, e, lldp_handler, NULL) == 0);
	202
	203	assert_se(write(test_fd[1], frame, sizeof(frame)) == sizeof(frame));
	204	sd_event_run(e, 0);
	205	assert_se(lldp_handler_calls == 1);
34437b4f LP	206	assert_se(sd_lldp_get_neighbors(lldp, &neighbors) == 1);
	207
	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 BG	232
	233	assert_se(stop_lldp(lldp) == 0);
	234	}
	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
314	sd_lldp *lldp;
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
322	lldp_handler_calls = 0;
323	assert_se(start_lldp(&lldp, e, lldp_handler, NULL) == 0);
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);
337	assert_se(lldp_handler_calls == 6);
338
339	assert_se(sd_lldp_get_neighbors(lldp, &neighbors) == 6);
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
362	assert_se(stop_lldp(lldp) == 0);
363	}
364
6fd255cf	365	int main(int argc, char *argv[]) {
4afd3348	366	_cleanup_(sd_event_unrefp) sd_event *e = NULL;
6fd255cf	367
34437b4f	368	log_set_max_level(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	}