Branch data     Line data    Source code

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