libnl  3.2.24-rc1
nf-exp-add.c
1 /*
2  * src/nf-exp-add.c Create an expectation
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) 2003-2009 Thomas Graf <tgraf@suug.ch>
10  * Copyright (c) 2007 Philip Craig <philipc@snapgear.com>
11  * Copyright (c) 2007 Secure Computing Corporation
12  * Copyright (c) 2012 Rich Fought <rich.fought@watchguard.com>
13  *
14  */
15 
16 #include <netlink/cli/utils.h>
17 #include <netlink/cli/exp.h>
18 
19 static int quiet = 0;
20 
21 static void print_usage(void)
22 {
23  printf(
24  "Usage: nf-exp-list [OPTION]... [CONNTRACK ENTRY]\n"
25  "\n"
26  "Options\n"
27  " --replace Replace the address if it exists.\n"
28  " -q, --quiet Do not print informal notifications.\n"
29  " -h, --help Show this help\n"
30  " -v, --version Show versioning information\n"
31  "\n"
32  "Expectation Selection\n"
33  " -i, --id=NUM Identifier\n"
34  " --expect-proto=PROTOCOL Expectation protocol\n"
35  " --expect-src=ADDR Expectation source address\n"
36  " --expect-sport=PORT Expectation source port\n"
37  " --expect-dst=ADDR Expectation destination address\n"
38  " --expect-dport=PORT Expectation destination port\n"
39  " --master-proto=PROTOCOL Master conntrack protocol\n"
40  " --master-src=ADDR Master conntrack source address\n"
41  " --master-sport=PORT Master conntrack source port\n"
42  " --master-dst=ADDR Master conntrack destination address\n"
43  " --master-dport=PORT Master conntrack destination port\n"
44  " --mask-proto=PROTOCOL Mask protocol\n"
45  " --mask-src=ADDR Mask source address\n"
46  " --mask-sport=PORT Mask source port\n"
47  " --mask-dst=ADDR Mask destination address\n"
48  " --mask-dport=PORT Mask destination port\n"
49  " -F, --family=FAMILY Address family\n"
50  " --timeout=NUM Timeout value\n"
51  " --helper=STRING Helper Name\n"
52  " --flags Flags (Kernel 2.6.37)\n"
53  );
54  exit(0);
55 }
56 
57 int main(int argc, char *argv[])
58 {
59  struct nl_sock *sock;
60  struct nfnl_exp *exp;
61  struct nl_dump_params params = {
62  .dp_type = NL_DUMP_LINE,
63  .dp_fd = stdout,
64  };
65  int err, nlflags = NLM_F_CREATE;
66 
67  exp = nl_cli_exp_alloc();
68 
69  for (;;) {
70  int c, optidx = 0;
71  enum {
72  ARG_MARK = 270,
73  ARG_TCP_STATE = 271,
74  ARG_EXPECT_PROTO,
75  ARG_EXPECT_SRC,
76  ARG_EXPECT_SPORT,
77  ARG_EXPECT_DST,
78  ARG_EXPECT_DPORT,
79  ARG_MASTER_PROTO,
80  ARG_MASTER_SRC,
81  ARG_MASTER_SPORT,
82  ARG_MASTER_DST,
83  ARG_MASTER_DPORT,
84  ARG_MASK_PROTO,
85  ARG_MASK_SRC,
86  ARG_MASK_SPORT,
87  ARG_MASK_DST,
88  ARG_MASK_DPORT,
89  ARG_NAT_PROTO,
90  ARG_NAT_SRC,
91  ARG_NAT_SPORT,
92  ARG_NAT_DST,
93  ARG_NAT_DPORT,
94  ARG_NAT_DIR,
95  ARG_TIMEOUT,
96  ARG_HELPER_NAME,
97  ARG_REPLACE,
98  ARG_FLAGS,
99  };
100  static struct option long_opts[] = {
101  { "replace", 1, 0, ARG_REPLACE },
102  { "quiet", 0, 0, 'q' },
103  { "help", 0, 0, 'h' },
104  { "version", 0, 0, 'v' },
105  { "id", 1, 0, 'i' },
106  { "expect-proto", 1, 0, ARG_EXPECT_PROTO },
107  { "expect-src", 1, 0, ARG_EXPECT_SRC },
108  { "expect-sport", 1, 0, ARG_EXPECT_SPORT },
109  { "expect-dst", 1, 0, ARG_EXPECT_DST },
110  { "expect-dport", 1, 0, ARG_EXPECT_DPORT },
111  { "master-proto", 1, 0, ARG_MASTER_PROTO },
112  { "master-src", 1, 0, ARG_MASTER_SRC },
113  { "master-sport", 1, 0, ARG_MASTER_SPORT },
114  { "master-dst", 1, 0, ARG_MASTER_DST },
115  { "master-dport", 1, 0, ARG_MASTER_DPORT },
116  { "mask-proto", 1, 0, ARG_MASK_PROTO },
117  { "mask-src", 1, 0, ARG_MASK_SRC },
118  { "mask-sport", 1, 0, ARG_MASK_SPORT },
119  { "mask-dst", 1, 0, ARG_MASK_DST },
120  { "mask-dport", 1, 0, ARG_MASK_DPORT },
121  { "nat-proto", 1, 0, ARG_NAT_PROTO },
122  { "nat-src", 1, 0, ARG_NAT_SRC },
123  { "nat-sport", 1, 0, ARG_NAT_SPORT },
124  { "nat-dst", 1, 0, ARG_NAT_DST },
125  { "nat-dport", 1, 0, ARG_NAT_DPORT },
126  { "nat-dir", 1, 0, ARG_NAT_DIR },
127  { "family", 1, 0, 'F' },
128  { "timeout", 1, 0, ARG_TIMEOUT },
129  { "helper", 1, 0, ARG_HELPER_NAME },
130  { "flags", 1, 0, ARG_FLAGS},
131  { 0, 0, 0, 0 }
132  };
133 
134  c = getopt_long(argc, argv, "46f:hvi:p:F:", long_opts, &optidx);
135  if (c == -1)
136  break;
137 
138  switch (c) {
139  case '?': exit(NLE_INVAL);
140  case ARG_REPLACE: nlflags |= NLM_F_REPLACE; break;
141  case 'q': quiet = 1; break;
142  case '4': nfnl_exp_set_family(exp, AF_INET); break;
143  case '6': nfnl_exp_set_family(exp, AF_INET6); break;
144  case 'h': print_usage(); break;
145  case 'v': nl_cli_print_version(); break;
146  case 'i': nl_cli_exp_parse_id(exp, optarg); break;
147  case ARG_EXPECT_PROTO: nl_cli_exp_parse_l4protonum(exp, NFNL_EXP_TUPLE_EXPECT, optarg); break;
148  case ARG_EXPECT_SRC: nl_cli_exp_parse_src(exp, NFNL_EXP_TUPLE_EXPECT, optarg); break;
149  case ARG_EXPECT_SPORT: nl_cli_exp_parse_src_port(exp, NFNL_EXP_TUPLE_EXPECT, optarg); break;
150  case ARG_EXPECT_DST: nl_cli_exp_parse_dst(exp, NFNL_EXP_TUPLE_EXPECT, optarg); break;
151  case ARG_EXPECT_DPORT: nl_cli_exp_parse_dst_port(exp, NFNL_EXP_TUPLE_EXPECT, optarg); break;
152  case ARG_MASTER_PROTO: nl_cli_exp_parse_l4protonum(exp, NFNL_EXP_TUPLE_MASTER, optarg); break;
153  case ARG_MASTER_SRC: nl_cli_exp_parse_src(exp, NFNL_EXP_TUPLE_MASTER, optarg); break;
154  case ARG_MASTER_SPORT: nl_cli_exp_parse_src_port(exp, NFNL_EXP_TUPLE_MASTER, optarg); break;
155  case ARG_MASTER_DST: nl_cli_exp_parse_dst(exp, NFNL_EXP_TUPLE_MASTER, optarg); break;
156  case ARG_MASTER_DPORT: nl_cli_exp_parse_dst_port(exp, NFNL_EXP_TUPLE_MASTER, optarg); break;
157  case ARG_MASK_PROTO: nl_cli_exp_parse_l4protonum(exp, NFNL_EXP_TUPLE_MASK, optarg); break;
158  case ARG_MASK_SRC: nl_cli_exp_parse_src(exp, NFNL_EXP_TUPLE_MASK, optarg); break;
159  case ARG_MASK_SPORT: nl_cli_exp_parse_src_port(exp, NFNL_EXP_TUPLE_MASK, optarg); break;
160  case ARG_MASK_DST: nl_cli_exp_parse_dst(exp, NFNL_EXP_TUPLE_MASK, optarg); break;
161  case ARG_MASK_DPORT: nl_cli_exp_parse_dst_port(exp, NFNL_EXP_TUPLE_MASK, optarg); break;
162  case ARG_NAT_PROTO: nl_cli_exp_parse_l4protonum(exp, NFNL_EXP_TUPLE_NAT, optarg); break;
163  case ARG_NAT_SRC: nl_cli_exp_parse_src(exp, NFNL_EXP_TUPLE_NAT, optarg); break;
164  case ARG_NAT_SPORT: nl_cli_exp_parse_src_port(exp, NFNL_EXP_TUPLE_NAT, optarg); break;
165  case ARG_NAT_DST: nl_cli_exp_parse_dst(exp, NFNL_EXP_TUPLE_NAT, optarg); break;
166  case ARG_NAT_DPORT: nl_cli_exp_parse_dst_port(exp, NFNL_EXP_TUPLE_NAT, optarg); break;
167  case ARG_NAT_DIR: nl_cli_exp_parse_nat_dir(exp, optarg); break;
168  case 'F': nl_cli_exp_parse_family(exp, optarg); break;
169  case ARG_TIMEOUT: nl_cli_exp_parse_timeout(exp, optarg); break;
170  case ARG_HELPER_NAME: nl_cli_exp_parse_helper_name(exp, optarg); break;
171  case ARG_FLAGS: nl_cli_exp_parse_flags(exp, optarg); break;
172  }
173  }
174 
175  sock = nl_cli_alloc_socket();
176  nl_cli_connect(sock, NETLINK_NETFILTER);
177 
178  if ((err = nfnl_exp_add(sock, exp, nlflags)) < 0)
179  nl_cli_fatal(err, "Unable to add expectation: %s", nl_geterror(err));
180 
181  if (!quiet) {
182  printf("Added ");
183  nl_object_dump(OBJ_CAST(exp), &params);
184  }
185 
186  return 0;
187 }