Line data    Source code

       1             : /***
       2             :   SPDX-License-Identifier: LGPL-2.1+
       3             : ***/
       4             : 
       5             : #include "fd-util.h"
       6             : #include "fileio.h"
       7             : #include "log.h"
       8             : #include "macro.h"
       9             : #include "network-internal.h"
      10             : #include "networkd-manager.h"
      11             : #include "string-util.h"
      12             : #include "tests.h"
      13             : #include "tmpfile-util.h"
      14             : 
      15           7 : static void test_rule_serialization(const char *title, const char *ruleset, const char *expected) {
      16           7 :         char pattern[] = "/tmp/systemd-test-routing-policy-rule.XXXXXX",
      17           7 :              pattern2[] = "/tmp/systemd-test-routing-policy-rule.XXXXXX",
      18           7 :              pattern3[] = "/tmp/systemd-test-routing-policy-rule.XXXXXX";
      19             :         const char *cmd;
      20             :         int fd, fd2, fd3;
      21           7 :         _cleanup_fclose_ FILE *f = NULL, *f2 = NULL, *f3 = NULL;
      22           7 :         Set *rules = NULL;
      23           7 :         _cleanup_free_ char *buf = NULL;
      24             :         size_t buf_size;
      25             : 
      26           7 :         log_info("========== %s ==========", title);
      27           7 :         log_info("put:\n%s\n", ruleset);
      28             : 
      29           7 :         fd = mkostemp_safe(pattern);
      30           7 :         assert_se(fd >= 0);
      31           7 :         assert_se(f = fdopen(fd, "a+"));
      32           7 :         assert_se(write_string_stream(f, ruleset, 0) == 0);
      33             : 
      34           7 :         assert_se(routing_policy_load_rules(pattern, &rules) == 0);
      35             : 
      36           7 :         fd2 = mkostemp_safe(pattern2);
      37           7 :         assert_se(fd2 >= 0);
      38           7 :         assert_se(f2 = fdopen(fd2, "a+"));
      39             : 
      40           7 :         assert_se(routing_policy_serialize_rules(rules, f2) == 0);
      41           7 :         assert_se(fflush_and_check(f2) == 0);
      42             : 
      43           7 :         assert_se(read_full_file(pattern2, &buf, &buf_size) == 0);
      44             : 
      45           7 :         log_info("got:\n%s", buf);
      46             : 
      47           7 :         fd3 = mkostemp_safe(pattern3);
      48           7 :         assert_se(fd3 >= 0);
      49           7 :         assert_se(f3 = fdopen(fd3, "w"));
      50           7 :         assert_se(write_string_stream(f3, expected ?: ruleset, 0) == 0);
      51             : 
      52          63 :         cmd = strjoina("diff -u ", pattern3, " ", pattern2);
      53           7 :         log_info("$ %s", cmd);
      54           7 :         assert_se(system(cmd) == 0);
      55             : 
      56          14 :         set_free_with_destructor(rules, routing_policy_rule_free);
      57           7 : }
      58             : 
      59           1 : int main(int argc, char **argv) {
      60           2 :         _cleanup_free_ char *p = NULL;
      61             : 
      62           1 :         test_setup_logging(LOG_DEBUG);
      63             : 
      64           1 :         test_rule_serialization("basic parsing",
      65             :                                 "RULE=from=1.2.3.4/32 to=2.3.4.5/32 family=AF_INET tos=5 priority=0 fwmark=1/2 table=10", NULL);
      66             : 
      67           1 :         test_rule_serialization("ignored values",
      68             :                                 "RULE=something=to=ignore from=1.2.3.4/32 from=1.2.3.4/32"
      69             :                                 "   \t  to=2.3.4.5/24 to=2.3.4.5/32 tos=5 fwmark=2 fwmark=1 table=10 table=20",
      70             :                                 "RULE=from=1.2.3.4/32"
      71             :                                 " to=2.3.4.5/32 family=AF_INET tos=5 priority=0 fwmark=1/0 table=20");
      72             : 
      73           1 :         test_rule_serialization("ipv6",
      74             :                                 "RULE=from=1::2/64 to=2::3/64 family=AF_INET6 priority=0 table=6", NULL);
      75             : 
      76           1 :         assert_se(asprintf(&p, "RULE=from=1::2/64 to=2::3/64 family=AF_INET6 priority=0 table=%d", RT_TABLE_MAIN) >= 0);
      77           1 :         test_rule_serialization("default table",
      78             :                                 "RULE=from=1::2/64 to=2::3/64", p);
      79             : 
      80           1 :         test_rule_serialization("incoming interface",
      81             :                                 "RULE=from=1::2/64 to=2::3/64 table=1 iif=lo",
      82             :                                 "RULE=from=1::2/64 to=2::3/64 family=AF_INET6 priority=0 iif=lo table=1");
      83             : 
      84           1 :         test_rule_serialization("outgoing interface",
      85             :                                 "RULE=from=1::2/64 to=2::3/64 family=AF_INET6 priority=0 oif=eth0 table=1", NULL);
      86             : 
      87           1 :         test_rule_serialization("freeing interface names",
      88             :                                 "RULE=from=1::2/64 to=2::3/64 family=AF_INET6 iif=e0 iif=e1 oif=e0 oif=e1 table=1",
      89             :                                 "RULE=from=1::2/64 to=2::3/64 family=AF_INET6 priority=0 iif=e1 oif=e1 table=1");
      90             : 
      91           1 :         return 0;
      92             : }