src/lib/ct.c

   1 /*
   2  * src/lib/ct.c         CLI Conntrack Helpers
   3  *
   4  *      This library is free software; you can redistribute it and/or
   5  *      modify it under the terms of the GNU Lesser General Public
   6  *      License as published by the Free Software Foundation version 2.1
   7  *      of the License.
   8  *
   9  * Copyright (c) 2008-2009 Thomas Graf <tgraf@suug.ch>
  10  */
  11
  12 /**
  13  * @ingroup cli
  14  * @defgroup cli_ct Connection Tracking
  15  *
  16  * @{
  17  */
  18
  19 #include <netlink/cli/utils.h>
  20 #include <netlink/cli/ct.h>
  21
  22 struct nfnl_ct *nl_cli_ct_alloc(void)
  23 {
  24         struct nfnl_ct *ct;
  25
  26         ct = nfnl_ct_alloc();
  27         if (!ct)
  28                 nl_cli_fatal(ENOMEM, "Unable to allocate conntrack object");
  29
  30         return ct;
  31 }
  32
  33 struct nl_cache *nl_cli_ct_alloc_cache(struct nl_sock *sk)
  34 {
  35         return nl_cli_alloc_cache(sk, "conntrack", nfnl_ct_alloc_cache);
  36 }
  37
  38 void nl_cli_ct_parse_family(struct nfnl_ct *ct, char *arg)
  39 {
  40         int family;
  41
  42         if ((family = nl_str2af(arg)) == AF_UNSPEC)
  43                 nl_cli_fatal(EINVAL,
  44                              "Unable to nl_cli_ct_parse family \"%s\": %s",
  45                              arg, nl_geterror(NLE_INVAL));
  46
  47         nfnl_ct_set_family(ct, family);
  48 }
  49
  50 void nl_cli_ct_parse_protocol(struct nfnl_ct *ct, char *arg)
  51 {
  52         int proto;
  53
  54         if ((proto = nl_str2ip_proto(arg)) < 0)
  55                 nl_cli_fatal(proto,
  56                              "Unable to nl_cli_ct_parse protocol \"%s\": %s",
  57                              arg, nl_geterror(proto));
  58
  59         nfnl_ct_set_proto(ct, proto);
  60 }
  61
  62 void nl_cli_ct_parse_mark(struct nfnl_ct *ct, char *arg)
  63 {
  64         uint32_t mark = nl_cli_parse_u32(arg);
  65         nfnl_ct_set_mark(ct, mark);
  66 }
  67
  68 void nl_cli_ct_parse_timeout(struct nfnl_ct *ct, char *arg)
  69 {
  70         uint32_t timeout = nl_cli_parse_u32(arg);
  71         nfnl_ct_set_timeout(ct, timeout);
  72 }
  73
  74 void nl_cli_ct_parse_id(struct nfnl_ct *ct, char *arg)
  75 {
  76         uint32_t id = nl_cli_parse_u32(arg);
  77         nfnl_ct_set_id(ct, id);
  78 }
  79
  80 void nl_cli_ct_parse_use(struct nfnl_ct *ct, char *arg)
  81 {
  82         uint32_t use = nl_cli_parse_u32(arg);
  83         nfnl_ct_set_use(ct, use);
  84 }
  85
  86 void nl_cli_ct_parse_src(struct nfnl_ct *ct, int reply, char *arg)
  87 {
  88         int err;
  89         struct nl_addr *a = nl_cli_addr_parse(arg, nfnl_ct_get_family(ct));
  90         if ((err = nfnl_ct_set_src(ct, reply, a)) < 0)
  91                 nl_cli_fatal(err, "Unable to set source address: %s",
  92                              nl_geterror(err));
  93 }
  94
  95 void nl_cli_ct_parse_dst(struct nfnl_ct *ct, int reply, char *arg)
  96 {
  97         int err;
  98         struct nl_addr *a = nl_cli_addr_parse(arg, nfnl_ct_get_family(ct));
  99         if ((err = nfnl_ct_set_dst(ct, reply, a)) < 0)
 100                 nl_cli_fatal(err, "Unable to set destination address: %s",
 101                              nl_geterror(err));
 102 }
 103
 104 void nl_cli_ct_parse_src_port(struct nfnl_ct *ct, int reply, char *arg)
 105 {
 106         uint32_t port = nl_cli_parse_u32(arg);
 107         nfnl_ct_set_src_port(ct, reply, port);
 108 }
 109
 110 void nl_cli_ct_parse_dst_port(struct nfnl_ct *ct, int reply, char *arg)
 111 {
 112         uint32_t port = nl_cli_parse_u32(arg);
 113         nfnl_ct_set_dst_port(ct, reply, port);
 114 }
 115
 116 void nl_cli_ct_parse_tcp_state(struct nfnl_ct *ct, char *arg)
 117 {
 118         int state;
 119
 120         if ((state = nfnl_ct_str2tcp_state(arg)) < 0)
 121                 nl_cli_fatal(state,
 122                              "Unable to nl_cli_ct_parse tcp state \"%s\": %s",
 123                              arg, nl_geterror(state));
 124
 125         nfnl_ct_set_tcp_state(ct, state);
 126 }
 127
 128 void nl_cli_ct_parse_status(struct nfnl_ct *ct, char *arg)
 129 {
 130         int status;
 131
 132         if ((status = nfnl_ct_str2status(arg)) < 0)
 133                 nl_cli_fatal(status,
 134                              "Unable to nl_cli_ct_parse flags \"%s\": %s",
 135                              arg, nl_geterror(status));
 136
 137         nfnl_ct_set_status(ct, status);
 138 }
 139
 140 #if 0
 141                 } else if (arg_match("origicmpid")) {
 142                         if (argc > ++idx)
 143                                 nfnl_ct_set_icmp_id(ct, 0, strtoul(argv[idx++], NULL, 0));
 144                 } else if (arg_match("origicmptype")) {
 145                         if (argc > ++idx)
 146                                 nfnl_ct_set_icmp_type(ct, 0, strtoul(argv[idx++], NULL, 0));
 147                 } else if (arg_match("origicmpcode")) {
 148                         if (argc > ++idx)
 149                                 nfnl_ct_set_icmp_code(ct, 0, strtoul(argv[idx++], NULL, 0));
 150                 } else if (arg_match("replyicmpid")) {
 151                         if (argc > ++idx)
 152                                 nfnl_ct_set_icmp_id(ct, 1, strtoul(argv[idx++], NULL, 0));
 153                 } else if (arg_match("replyicmptype")) {
 154                         if (argc > ++idx)
 155                                 nfnl_ct_set_icmp_type(ct, 1, strtoul(argv[idx++], NULL, 0));
 156                 } else if (arg_match("replyicmpcode")) {
 157                         if (argc > ++idx)
 158                                 nfnl_ct_set_icmp_code(ct, 1, strtoul(argv[idx++], NULL, 0));
 159                 }
 160 #endif
 161
 162 /** @} */