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