src/test/test-socket-netlink.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include "alloc-util.h"
   4 #include "missing_network.h"
   5 #include "tests.h"
   6 #include "socket-netlink.h"
   7 #include "string-util.h"
   8
   9 static void test_socket_address_parse_one(const char *in, int ret, int family, const char *expected) {
  10         SocketAddress a;
  11         _cleanup_free_ char *out = NULL;
  12         int r;
  13
  14         r = socket_address_parse(&a, in);
  15         if (r >= 0) {
  16                 r = socket_address_print(&a, &out);
  17                 if (r < 0)
  18                         log_error_errno(r, "Printing failed for \"%s\": %m", in);
  19                 assert(r >= 0);
  20         }
  21
  22         log_info("\"%s\" → %s %d → \"%s\" (expect %d / \"%s\")",
  23                  in,
  24                  r >= 0 ? "✓" : "✗", r,
  25                  empty_to_dash(out),
  26                  ret,
  27                  ret >= 0 ? expected ?: in : "-");
  28         assert_se(r == ret);
  29         if (r >= 0) {
  30                 assert_se(a.sockaddr.sa.sa_family == family);
  31                 assert_se(streq(out, expected ?: in));
  32         }
  33 }
  34
  35 static void test_socket_address_parse(void) {
  36         log_info("/* %s */", __func__);
  37
  38         test_socket_address_parse_one("junk", -EINVAL, 0, NULL);
  39         test_socket_address_parse_one("192.168.1.1", -EINVAL, 0, NULL);
  40         test_socket_address_parse_one(".168.1.1", -EINVAL, 0, NULL);
  41         test_socket_address_parse_one("989.168.1.1", -EINVAL, 0, NULL);
  42         test_socket_address_parse_one("192.168.1.1:65536", -ERANGE, 0, NULL);
  43         test_socket_address_parse_one("192.168.1.1:0", -EINVAL, 0, NULL);
  44         test_socket_address_parse_one("0", -EINVAL, 0, NULL);
  45         test_socket_address_parse_one("65536", -ERANGE, 0, NULL);
  46
  47         const int default_family = socket_ipv6_is_supported() ? AF_INET6 : AF_INET;
  48
  49         test_socket_address_parse_one("65535", 0, default_family, "[::]:65535");
  50
  51         /* The checks below will pass even if ipv6 is disabled in
  52          * kernel. The underlying glibc's inet_pton() is just a string
  53          * parser and doesn't make any syscalls. */
  54
  55         test_socket_address_parse_one("[::1]", -EINVAL, 0, NULL);
  56         test_socket_address_parse_one("[::1]8888", -EINVAL, 0, NULL);
  57         test_socket_address_parse_one("::1", -EINVAL, 0, NULL);
  58         test_socket_address_parse_one("[::1]:0", -EINVAL, 0, NULL);
  59         test_socket_address_parse_one("[::1]:65536", -ERANGE, 0, NULL);
  60         test_socket_address_parse_one("[a:b:1]:8888", -EINVAL, 0, NULL);
  61
  62         test_socket_address_parse_one("8888", 0, default_family, "[::]:8888");
  63         test_socket_address_parse_one("[2001:0db8:0000:85a3:0000:0000:ac1f:8001]:8888", 0, AF_INET6,
  64                                       "[2001:db8:0:85a3::ac1f:8001]:8888");
  65         test_socket_address_parse_one("[::1]:8888", 0, AF_INET6, NULL);
  66         test_socket_address_parse_one("192.168.1.254:8888", 0, AF_INET, NULL);
  67         test_socket_address_parse_one("/foo/bar", 0, AF_UNIX, NULL);
  68         test_socket_address_parse_one("/", 0, AF_UNIX, NULL);
  69         test_socket_address_parse_one("@abstract", 0, AF_UNIX, NULL);
  70
  71         {
  72                 char aaa[SUN_PATH_LEN + 1] = "@";
  73
  74                 memset(aaa + 1, 'a', SUN_PATH_LEN - 1);
  75                 char_array_0(aaa);
  76
  77                 test_socket_address_parse_one(aaa, -EINVAL, 0, NULL);
  78
  79                 aaa[SUN_PATH_LEN - 1] = '\0';
  80                 test_socket_address_parse_one(aaa, 0, AF_UNIX, NULL);
  81         }
  82
  83         test_socket_address_parse_one("vsock:2:1234", 0, AF_VSOCK, NULL);
  84         test_socket_address_parse_one("vsock::1234", 0, AF_VSOCK, NULL);
  85         test_socket_address_parse_one("vsock:2:1234x", -EINVAL, 0, NULL);
  86         test_socket_address_parse_one("vsock:2x:1234", -EINVAL, 0, NULL);
  87         test_socket_address_parse_one("vsock:2", -EINVAL, 0, NULL);
  88 }
  89
  90 static void test_socket_address_parse_netlink(void) {
  91         SocketAddress a;
  92
  93         log_info("/* %s */", __func__);
  94
  95         assert_se(socket_address_parse_netlink(&a, "junk") < 0);
  96         assert_se(socket_address_parse_netlink(&a, "") < 0);
  97
  98         assert_se(socket_address_parse_netlink(&a, "route") >= 0);
  99         assert_se(a.sockaddr.nl.nl_family == AF_NETLINK);
 100         assert_se(a.sockaddr.nl.nl_groups == 0);
 101         assert_se(a.protocol == NETLINK_ROUTE);
 102         assert_se(socket_address_parse_netlink(&a, "route") >= 0);
 103         assert_se(socket_address_parse_netlink(&a, "route 10") >= 0);
 104         assert_se(a.sockaddr.nl.nl_family == AF_NETLINK);
 105         assert_se(a.sockaddr.nl.nl_groups == 10);
 106         assert_se(a.protocol == NETLINK_ROUTE);
 107
 108         /* With spaces and tabs */
 109         assert_se(socket_address_parse_netlink(&a, " kobject-uevent ") >= 0);
 110         assert_se(a.sockaddr.nl.nl_family == AF_NETLINK);
 111         assert_se(a.sockaddr.nl.nl_groups == 0);
 112         assert_se(a.protocol == NETLINK_KOBJECT_UEVENT);
 113         assert_se(socket_address_parse_netlink(&a, " \t kobject-uevent \t 10") >= 0);
 114         assert_se(a.sockaddr.nl.nl_family == AF_NETLINK);
 115         assert_se(a.sockaddr.nl.nl_groups == 10);
 116         assert_se(a.protocol == NETLINK_KOBJECT_UEVENT);
 117         assert_se(socket_address_parse_netlink(&a, "kobject-uevent\t10") >= 0);
 118         assert_se(a.sockaddr.nl.nl_family == AF_NETLINK);
 119         assert_se(a.sockaddr.nl.nl_groups == 10);
 120         assert_se(a.protocol == NETLINK_KOBJECT_UEVENT);
 121
 122         /* trailing space is not supported */
 123         assert_se(socket_address_parse_netlink(&a, "kobject-uevent\t10 ") < 0);
 124
 125         /* Group must be unsigned */
 126         assert_se(socket_address_parse_netlink(&a, "kobject-uevent -1") < 0);
 127
 128         /* oss-fuzz #6884 */
 129         assert_se(socket_address_parse_netlink(&a, "\xff") < 0);
 130 }
 131
 132 static void test_socket_address_equal(void) {
 133         SocketAddress a, b;
 134
 135         log_info("/* %s */", __func__);
 136
 137         assert_se(socket_address_parse(&a, "192.168.1.1:8888") >= 0);
 138         assert_se(socket_address_parse(&b, "192.168.1.1:888") >= 0);
 139         assert_se(!socket_address_equal(&a, &b));
 140
 141         assert_se(socket_address_parse(&a, "192.168.1.1:8888") >= 0);
 142         assert_se(socket_address_parse(&b, "192.16.1.1:8888") >= 0);
 143         assert_se(!socket_address_equal(&a, &b));
 144
 145         assert_se(socket_address_parse(&a, "192.168.1.1:8888") >= 0);
 146         assert_se(socket_address_parse(&b, "8888") >= 0);
 147         assert_se(!socket_address_equal(&a, &b));
 148
 149         assert_se(socket_address_parse(&a, "192.168.1.1:8888") >= 0);
 150         assert_se(socket_address_parse(&b, "/foo/bar/") >= 0);
 151         assert_se(!socket_address_equal(&a, &b));
 152
 153         assert_se(socket_address_parse(&a, "192.168.1.1:8888") >= 0);
 154         assert_se(socket_address_parse(&b, "192.168.1.1:8888") >= 0);
 155         assert_se(socket_address_equal(&a, &b));
 156
 157         assert_se(socket_address_parse(&a, "/foo/bar") >= 0);
 158         assert_se(socket_address_parse(&b, "/foo/bar") >= 0);
 159         assert_se(socket_address_equal(&a, &b));
 160
 161         assert_se(socket_address_parse(&a, "[::1]:8888") >= 0);
 162         assert_se(socket_address_parse(&b, "[::1]:8888") >= 0);
 163         assert_se(socket_address_equal(&a, &b));
 164
 165         assert_se(socket_address_parse(&a, "@abstract") >= 0);
 166         assert_se(socket_address_parse(&b, "@abstract") >= 0);
 167         assert_se(socket_address_equal(&a, &b));
 168
 169         assert_se(socket_address_parse_netlink(&a, "firewall") >= 0);
 170         assert_se(socket_address_parse_netlink(&b, "firewall") >= 0);
 171         assert_se(socket_address_equal(&a, &b));
 172
 173         assert_se(socket_address_parse(&a, "vsock:2:1234") >= 0);
 174         assert_se(socket_address_parse(&b, "vsock:2:1234") >= 0);
 175         assert_se(socket_address_equal(&a, &b));
 176         assert_se(socket_address_parse(&b, "vsock:2:1235") >= 0);
 177         assert_se(!socket_address_equal(&a, &b));
 178         assert_se(socket_address_parse(&b, "vsock:3:1234") >= 0);
 179         assert_se(!socket_address_equal(&a, &b));
 180 }
 181
 182 static void test_socket_address_get_path(void) {
 183         SocketAddress a;
 184
 185         log_info("/* %s */", __func__);
 186
 187         assert_se(socket_address_parse(&a, "192.168.1.1:8888") >= 0);
 188         assert_se(!socket_address_get_path(&a));
 189
 190         assert_se(socket_address_parse(&a, "@abstract") >= 0);
 191         assert_se(!socket_address_get_path(&a));
 192
 193         assert_se(socket_address_parse(&a, "[::1]:8888") >= 0);
 194         assert_se(!socket_address_get_path(&a));
 195
 196         assert_se(socket_address_parse(&a, "/foo/bar") >= 0);
 197         assert_se(streq(socket_address_get_path(&a), "/foo/bar"));
 198
 199         assert_se(socket_address_parse(&a, "vsock:2:1234") >= 0);
 200         assert_se(!socket_address_get_path(&a));
 201 }
 202
 203 static void test_socket_address_is(void) {
 204         SocketAddress a;
 205
 206         log_info("/* %s */", __func__);
 207
 208         assert_se(socket_address_parse(&a, "192.168.1.1:8888") >= 0);
 209         assert_se(socket_address_is(&a, "192.168.1.1:8888", SOCK_STREAM));
 210         assert_se(!socket_address_is(&a, "route", SOCK_STREAM));
 211         assert_se(!socket_address_is(&a, "192.168.1.1:8888", SOCK_RAW));
 212 }
 213
 214 static void test_socket_address_is_netlink(void) {
 215         SocketAddress a;
 216
 217         log_info("/* %s */", __func__);
 218
 219         assert_se(socket_address_parse_netlink(&a, "route 10") >= 0);
 220         assert_se(socket_address_is_netlink(&a, "route 10"));
 221         assert_se(!socket_address_is_netlink(&a, "192.168.1.1:8888"));
 222         assert_se(!socket_address_is_netlink(&a, "route 1"));
 223 }
 224
 225 static void test_in_addr_ifindex_to_string_one(int f, const char *a, int ifindex, const char *b) {
 226         _cleanup_free_ char *r = NULL;
 227         union in_addr_union ua, uuaa;
 228         int ff, ifindex2;
 229
 230         assert_se(in_addr_from_string(f, a, &ua) >= 0);
 231         assert_se(in_addr_ifindex_to_string(f, &ua, ifindex, &r) >= 0);
 232         printf("test_in_addr_ifindex_to_string_one: %s == %s\n", b, r);
 233         assert_se(streq(b, r));
 234
 235         assert_se(in_addr_ifindex_from_string_auto(b, &ff, &uuaa, &ifindex2) >= 0);
 236         assert_se(ff == f);
 237         assert_se(in_addr_equal(f, &ua, &uuaa));
 238         assert_se(ifindex2 == ifindex || ifindex2 == 0);
 239 }
 240
 241 static void test_in_addr_ifindex_to_string(void) {
 242         log_info("/* %s */", __func__);
 243
 244         test_in_addr_ifindex_to_string_one(AF_INET, "192.168.0.1", 7, "192.168.0.1");
 245         test_in_addr_ifindex_to_string_one(AF_INET, "10.11.12.13", 9, "10.11.12.13");
 246         test_in_addr_ifindex_to_string_one(AF_INET6, "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff", 10, "ffff:ffff:ffff:ffff:ffff:ffff:ffff:ffff");
 247         test_in_addr_ifindex_to_string_one(AF_INET6, "::1", 11, "::1");
 248         test_in_addr_ifindex_to_string_one(AF_INET6, "fe80::", 12, "fe80::%12");
 249         test_in_addr_ifindex_to_string_one(AF_INET6, "fe80::", 0, "fe80::");
 250         test_in_addr_ifindex_to_string_one(AF_INET6, "fe80::14", 12, "fe80::14%12");
 251         test_in_addr_ifindex_to_string_one(AF_INET6, "fe80::15", -7, "fe80::15");
 252         test_in_addr_ifindex_to_string_one(AF_INET6, "fe80::16", LOOPBACK_IFINDEX, "fe80::16%1");
 253 }
 254
 255 static void test_in_addr_ifindex_from_string_auto(void) {
 256         int family, ifindex;
 257         union in_addr_union ua;
 258
 259         log_info("/* %s */", __func__);
 260         /* Most in_addr_ifindex_from_string_auto() invocations have already been tested above, but let's test some more */
 261
 262         assert_se(in_addr_ifindex_from_string_auto("fe80::17", &family, &ua, &ifindex) >= 0);
 263         assert_se(family == AF_INET6);
 264         assert_se(ifindex == 0);
 265
 266         assert_se(in_addr_ifindex_from_string_auto("fe80::18%19", &family, &ua, &ifindex) >= 0);
 267         assert_se(family == AF_INET6);
 268         assert_se(ifindex == 19);
 269
 270         assert_se(in_addr_ifindex_from_string_auto("fe80::18%lo", &family, &ua, &ifindex) >= 0);
 271         assert_se(family == AF_INET6);
 272         assert_se(ifindex == LOOPBACK_IFINDEX);
 273
 274         assert_se(in_addr_ifindex_from_string_auto("fe80::19%thisinterfacecantexist", &family, &ua, &ifindex) == -ENODEV);
 275 }
 276
 277 static void test_in_addr_ifindex_name_from_string_auto_one(const char *a, const char *expected) {
 278         int family, ifindex;
 279         union in_addr_union ua;
 280         _cleanup_free_ char *server_name = NULL;
 281
 282         assert_se(in_addr_ifindex_name_from_string_auto(a, &family, &ua, &ifindex, &server_name) >= 0);
 283         assert_se(streq_ptr(server_name, expected));
 284 }
 285
 286 static void test_in_addr_ifindex_name_from_string_auto(void) {
 287         log_info("/* %s */", __func__);
 288
 289         test_in_addr_ifindex_name_from_string_auto_one("192.168.0.1", NULL);
 290         test_in_addr_ifindex_name_from_string_auto_one("192.168.0.1#test.com", "test.com");
 291         test_in_addr_ifindex_name_from_string_auto_one("fe80::18%19", NULL);
 292         test_in_addr_ifindex_name_from_string_auto_one("fe80::18%19#another.test.com", "another.test.com");
 293 }
 294
 295 static void test_in_addr_port_ifindex_name_from_string_auto_one(const char *str, int family, uint16_t port, int ifindex, const char *server_name) {
 296         union in_addr_union a;
 297         uint16_t p;
 298         int f, i;
 299         char *fake;
 300
 301         log_info("%s: %s", __func__, str);
 302
 303         {
 304                 _cleanup_free_ char *name = NULL, *x = NULL;
 305                 assert_se(in_addr_port_ifindex_name_from_string_auto(str, &f, &a, &p, &i, &name) == 0);
 306                 assert_se(family == f);
 307                 assert_se(port == p);
 308                 assert_se(ifindex == i);
 309                 assert_se(streq_ptr(server_name, name));
 310                 assert_se(in_addr_port_ifindex_name_to_string(f, &a, p, i, name, &x) >= 0);
 311                 assert_se(streq(str, x));
 312         }
 313
 314         if (port > 0)
 315                 assert_se(in_addr_port_ifindex_name_from_string_auto(str, &f, &a, NULL, &i, &fake) == -EINVAL);
 316         else {
 317                 _cleanup_free_ char *name = NULL, *x = NULL;
 318                 assert_se(in_addr_port_ifindex_name_from_string_auto(str, &f, &a, NULL, &i, &name) == 0);
 319                 assert_se(family == f);
 320                 assert_se(ifindex == i);
 321                 assert_se(streq_ptr(server_name, name));
 322                 assert_se(in_addr_port_ifindex_name_to_string(f, &a, 0, i, name, &x) >= 0);
 323                 assert_se(streq(str, x));
 324         }
 325
 326         if (ifindex > 0)
 327                 assert_se(in_addr_port_ifindex_name_from_string_auto(str, &f, &a, &p, NULL, &fake) == -EINVAL);
 328         else {
 329                 _cleanup_free_ char *name = NULL, *x = NULL;
 330                 assert_se(in_addr_port_ifindex_name_from_string_auto(str, &f, &a, &p, NULL, &name) == 0);
 331                 assert_se(family == f);
 332                 assert_se(port == p);
 333                 assert_se(streq_ptr(server_name, name));
 334                 assert_se(in_addr_port_ifindex_name_to_string(f, &a, p, 0, name, &x) >= 0);
 335                 assert_se(streq(str, x));
 336         }
 337
 338         if (server_name)
 339                 assert_se(in_addr_port_ifindex_name_from_string_auto(str, &f, &a, &p, &i, NULL) == -EINVAL);
 340         else {
 341                 _cleanup_free_ char *x = NULL;
 342                 assert_se(in_addr_port_ifindex_name_from_string_auto(str, &f, &a, &p, &i, NULL) == 0);
 343                 assert_se(family == f);
 344                 assert_se(port == p);
 345                 assert_se(ifindex == i);
 346                 assert_se(in_addr_port_ifindex_name_to_string(f, &a, p, i, NULL, &x) >= 0);
 347                 assert_se(streq(str, x));
 348         }
 349 }
 350
 351 static void test_in_addr_port_ifindex_name_from_string_auto(void) {
 352         log_info("/* %s */", __func__);
 353
 354         test_in_addr_port_ifindex_name_from_string_auto_one("192.168.0.1", AF_INET, 0, 0, NULL);
 355         test_in_addr_port_ifindex_name_from_string_auto_one("192.168.0.1#test.com", AF_INET, 0, 0, "test.com");
 356         test_in_addr_port_ifindex_name_from_string_auto_one("192.168.0.1:53", AF_INET, 53, 0, NULL);
 357         test_in_addr_port_ifindex_name_from_string_auto_one("192.168.0.1:53#example.com", AF_INET, 53, 0, "example.com");
 358         test_in_addr_port_ifindex_name_from_string_auto_one("fe80::18", AF_INET6, 0, 0, NULL);
 359         test_in_addr_port_ifindex_name_from_string_auto_one("fe80::18#hoge.com", AF_INET6, 0, 0, "hoge.com");
 360         test_in_addr_port_ifindex_name_from_string_auto_one("fe80::18%19", AF_INET6, 0, 19, NULL);
 361         test_in_addr_port_ifindex_name_from_string_auto_one("[fe80::18]:53", AF_INET6, 53, 0, NULL);
 362         test_in_addr_port_ifindex_name_from_string_auto_one("fe80::18%19#hoge.com", AF_INET6, 0, 19, "hoge.com");
 363         test_in_addr_port_ifindex_name_from_string_auto_one("[fe80::18]:53#hoge.com", AF_INET6, 53, 0, "hoge.com");
 364         test_in_addr_port_ifindex_name_from_string_auto_one("[fe80::18]:53%19", AF_INET6, 53, 19, NULL);
 365         test_in_addr_port_ifindex_name_from_string_auto_one("[fe80::18]:53%19#hoge.com", AF_INET6, 53, 19, "hoge.com");
 366 }
 367
 368 int main(int argc, char *argv[]) {
 369         test_setup_logging(LOG_DEBUG);
 370
 371         test_socket_address_parse();
 372         test_socket_address_parse_netlink();
 373         test_socket_address_equal();
 374         test_socket_address_get_path();
 375         test_socket_address_is();
 376         test_socket_address_is_netlink();
 377
 378         test_in_addr_ifindex_to_string();
 379         test_in_addr_ifindex_from_string_auto();
 380         test_in_addr_ifindex_name_from_string_auto();
 381         test_in_addr_port_ifindex_name_from_string_auto();
 382
 383         return 0;
 384 }