src/test/test-ip-protocol-list.c

   1 /* SPDX-License-Identifier: LGPL-2.1+ */
   2
   3 #include <netinet/in.h>
   4
   5 #include "macro.h"
   6 #include "ip-protocol-list.h"
   7 #include "stdio-util.h"
   8 #include "string-util.h"
   9
  10 static void test_int(int i) {
  11         char str[DECIMAL_STR_MAX(int)];
  12
  13         assert_se(ip_protocol_from_name(ip_protocol_to_name(i)) == i);
  14
  15         xsprintf(str, "%i", i);
  16         assert_se(ip_protocol_from_name(ip_protocol_to_name(parse_ip_protocol(str))) == i);
  17 }
  18
  19 static void test_int_fail(int i) {
  20         char str[DECIMAL_STR_MAX(int)];
  21
  22         assert_se(!ip_protocol_to_name(i));
  23
  24         xsprintf(str, "%i", i);
  25         assert_se(parse_ip_protocol(str) == -EINVAL);
  26 }
  27
  28 static void test_str(const char *s) {
  29         assert_se(streq(ip_protocol_to_name(ip_protocol_from_name(s)), s));
  30         assert_se(streq(ip_protocol_to_name(parse_ip_protocol(s)), s));
  31 }
  32
  33 static void test_str_fail(const char *s) {
  34         assert_se(ip_protocol_from_name(s) == -EINVAL);
  35         assert_se(parse_ip_protocol(s) == -EINVAL);
  36 }
  37
  38 static void test_parse_ip_protocol(const char *s, int expected) {
  39         assert_se(parse_ip_protocol(s) == expected);
  40 }
  41
  42 int main(int argc, const char *argv[]) {
  43         test_int(IPPROTO_TCP);
  44         test_int(IPPROTO_DCCP);
  45         test_int_fail(-1);
  46         test_int_fail(1024 * 1024);
  47
  48         test_str("sctp");
  49         test_str("udp");
  50         test_str_fail("hoge");
  51         test_str_fail("-1");
  52         test_str_fail("1000000000");
  53
  54         test_parse_ip_protocol("sctp", IPPROTO_SCTP);
  55         test_parse_ip_protocol("ScTp", IPPROTO_SCTP);
  56         test_parse_ip_protocol("ip", IPPROTO_IP);
  57         test_parse_ip_protocol("", IPPROTO_IP);
  58         test_parse_ip_protocol("1", 1);
  59         test_parse_ip_protocol("0", 0);
  60         test_parse_ip_protocol("-10", -EINVAL);
  61         test_parse_ip_protocol("100000000", -EINVAL);
  62
  63         return 0;
  64 }