]>
Commit | Line | Data |
---|---|---|
46c1ad1f | 1 | #include <net/if.h> |
2 | #include <errno.h> | |
3 | #include <string.h> | |
4 | ||
5 | #include <netlink/genl/genl.h> | |
6 | #include <netlink/genl/family.h> | |
7 | #include <netlink/genl/ctrl.h> | |
8 | #include <netlink/msg.h> | |
9 | #include <netlink/attr.h> | |
10 | ||
11 | #include "nl80211.h" | |
12 | #include "iw.h" | |
13 | ||
14 | typedef struct _any_t { | |
15 | union { | |
16 | uint32_t as_32; | |
17 | uint16_t as_16; | |
18 | uint8_t as_8; | |
19 | } u; | |
20 | } _any; | |
21 | ||
22 | /* describes a mesh parameter */ | |
23 | struct mesh_param_descr { | |
24 | const char *name; | |
25 | enum nl80211_meshconf_params mesh_param_num; | |
26 | int (*nla_put_fn)(struct nl_msg*, int, _any*); | |
27 | uint32_t (*parse_fn)(const char*, _any*); | |
28 | void (*nla_print_fn)(struct nlattr *); | |
29 | }; | |
30 | ||
31 | /* utility functions for manipulating and printing u8/u16/u32 values and | |
32 | * timesouts. */ | |
33 | static int _my_nla_put_u8(struct nl_msg *n, int mesh_param_num, _any *value) | |
34 | { | |
35 | return nla_put(n, mesh_param_num, sizeof(uint8_t), &value->u.as_8); | |
36 | } | |
37 | ||
38 | static int _my_nla_put_u16(struct nl_msg *n, int mesh_param_num, _any *value) | |
39 | { | |
40 | return nla_put(n, mesh_param_num, sizeof(uint16_t), &value->u.as_16); | |
41 | } | |
42 | ||
43 | static int _my_nla_put_u32(struct nl_msg *n, int mesh_param_num, _any *value) | |
44 | { | |
45 | return nla_put(n, mesh_param_num, sizeof(uint32_t), &value->u.as_32); | |
46 | } | |
47 | ||
48 | static uint32_t _parse_u8(const char *str, _any *ret) | |
49 | { | |
50 | char *endptr = NULL; | |
51 | unsigned long int v = strtoul(str, &endptr, 10); | |
52 | if (*endptr != '\0') | |
53 | return 0xff; | |
54 | if (v > 0xff) | |
55 | return 0xff; | |
56 | ret->u.as_8 = (uint8_t)v; | |
57 | return 0; | |
58 | } | |
59 | ||
60 | static uint32_t _parse_u8_as_bool(const char *str, _any *ret) | |
61 | { | |
62 | char *endptr = NULL; | |
63 | unsigned long int v = strtoul(str, &endptr, 10); | |
64 | if (*endptr != '\0') | |
65 | return 0x1; | |
66 | if (v > 0x1) | |
67 | return 0x1; | |
68 | ret->u.as_8 = (uint8_t)v; | |
69 | return 0; | |
70 | } | |
71 | ||
72 | static uint32_t _parse_u16(const char *str, _any *ret) | |
73 | { | |
74 | char *endptr = NULL; | |
75 | long int v = strtol(str, &endptr, 10); | |
76 | if (*endptr != '\0') | |
77 | return 0xffff; | |
78 | if ((v < 0) || (v > 0xffff)) | |
79 | return 0xffff; | |
80 | ret->u.as_16 = (uint16_t)v; | |
81 | return 0; | |
82 | } | |
83 | ||
84 | static uint32_t _parse_u32(const char *str, _any *ret) | |
85 | { | |
86 | char *endptr = NULL; | |
87 | long long int v = strtoll(str, &endptr, 10); | |
88 | if (*endptr != '\0') | |
89 | return 0xffffffff; | |
90 | if ((v < 0) || (v > 0xffffffff)) | |
91 | return 0xffffffff; | |
92 | ret->u.as_32 = (uint32_t)v; | |
93 | return 0; | |
94 | } | |
95 | ||
656aa246 | 96 | static void _print_u8(struct nlattr *a) |
46c1ad1f | 97 | { |
98 | printf("%d", nla_get_u8(a)); | |
99 | } | |
100 | ||
656aa246 | 101 | static void _print_u16(struct nlattr *a) |
46c1ad1f | 102 | { |
103 | printf("%d", nla_get_u16(a)); | |
104 | } | |
105 | ||
656aa246 | 106 | static void _print_u16_timeout(struct nlattr *a) |
46c1ad1f | 107 | { |
108 | printf("%d milliseconds", nla_get_u16(a)); | |
109 | } | |
110 | ||
656aa246 | 111 | static void _print_u16_in_TUs(struct nlattr *a) |
46c1ad1f | 112 | { |
113 | printf("%d TUs", nla_get_u16(a)); | |
114 | } | |
115 | ||
656aa246 | 116 | static void _print_u32_timeout(struct nlattr *a) |
46c1ad1f | 117 | { |
118 | printf("%u milliseconds", nla_get_u32(a)); | |
119 | } | |
120 | ||
656aa246 | 121 | static void _print_u32_in_TUs(struct nlattr *a) |
46c1ad1f | 122 | { |
123 | printf("%d TUs", nla_get_u32(a)); | |
124 | } | |
125 | ||
126 | /* The current mesh parameters */ | |
127 | const static struct mesh_param_descr _mesh_param_descrs[] = | |
128 | { | |
129 | {"mesh_retry_timeout", | |
130 | NL80211_MESHCONF_RETRY_TIMEOUT, | |
131 | _my_nla_put_u16, _parse_u16, _print_u16_timeout}, | |
132 | {"mesh_confirm_timeout", | |
133 | NL80211_MESHCONF_CONFIRM_TIMEOUT, | |
134 | _my_nla_put_u16, _parse_u16, _print_u16_timeout}, | |
135 | {"mesh_holding_timeout", | |
136 | NL80211_MESHCONF_HOLDING_TIMEOUT, | |
137 | _my_nla_put_u16, _parse_u16, _print_u16_timeout}, | |
138 | {"mesh_max_peer_links", | |
139 | NL80211_MESHCONF_MAX_PEER_LINKS, | |
140 | _my_nla_put_u16, _parse_u16, _print_u16}, | |
141 | {"mesh_max_retries", | |
142 | NL80211_MESHCONF_MAX_RETRIES, | |
143 | _my_nla_put_u8, _parse_u8, _print_u8}, | |
144 | {"mesh_ttl", | |
145 | NL80211_MESHCONF_TTL, | |
146 | _my_nla_put_u8, _parse_u8, _print_u8}, | |
147 | {"mesh_auto_open_plinks", | |
148 | NL80211_MESHCONF_AUTO_OPEN_PLINKS, | |
149 | _my_nla_put_u8, _parse_u8_as_bool, _print_u8}, | |
150 | {"mesh_hwmp_max_preq_retries", | |
151 | NL80211_MESHCONF_HWMP_MAX_PREQ_RETRIES, | |
152 | _my_nla_put_u8, _parse_u8, _print_u8}, | |
153 | {"mesh_path_refresh_time", | |
154 | NL80211_MESHCONF_PATH_REFRESH_TIME, | |
155 | _my_nla_put_u32, _parse_u32, _print_u32_timeout}, | |
156 | {"mesh_min_discovery_timeout", | |
157 | NL80211_MESHCONF_MIN_DISCOVERY_TIMEOUT, | |
158 | _my_nla_put_u16, _parse_u16, _print_u16_timeout}, | |
159 | {"mesh_hwmp_active_path_timeout", | |
160 | NL80211_MESHCONF_HWMP_ACTIVE_PATH_TIMEOUT, | |
161 | _my_nla_put_u32, _parse_u32, _print_u32_in_TUs}, | |
162 | {"mesh_hwmp_preq_min_interval", | |
163 | NL80211_MESHCONF_HWMP_PREQ_MIN_INTERVAL, | |
164 | _my_nla_put_u16, _parse_u16, _print_u16_in_TUs}, | |
165 | {"mesh_hwmp_net_diameter_traversal_time", | |
166 | NL80211_MESHCONF_HWMP_NET_DIAM_TRVS_TIME, | |
167 | _my_nla_put_u16, _parse_u16, _print_u16_in_TUs}, | |
168 | }; | |
169 | ||
170 | static void print_all_mesh_param_descr(void) | |
171 | { | |
172 | int i; | |
46c1ad1f | 173 | |
f089471a JB |
174 | printf("Possible mesh parameters are:\n"); |
175 | ||
176 | for (i = 0; i < ARRAY_SIZE(_mesh_param_descrs); i++) | |
177 | printf(" - %s\n", _mesh_param_descrs[i].name); | |
46c1ad1f | 178 | } |
179 | ||
180 | static const struct mesh_param_descr* find_mesh_param(int argc, char **argv, | |
181 | const char *action_name) | |
182 | { | |
183 | int i; | |
184 | const struct mesh_param_descr *mdescr; | |
185 | ||
186 | if (argc < 1) { | |
187 | printf("You must specify which mesh parameter to %s.\n", | |
188 | action_name); | |
f089471a | 189 | print_all_mesh_param_descr(); |
46c1ad1f | 190 | return NULL; |
191 | } | |
192 | ||
193 | /* Find out what mesh parameter we want to change. */ | |
194 | mdescr = NULL; | |
4d0d2ea7 | 195 | for (i = 0; i < ARRAY_SIZE(_mesh_param_descrs); i++) { |
46c1ad1f | 196 | if (!strcmp(_mesh_param_descrs[i].name, argv[0])) |
197 | return _mesh_param_descrs + i; | |
4d0d2ea7 | 198 | } |
46c1ad1f | 199 | |
200 | if (!mdescr) { | |
46c1ad1f | 201 | print_all_mesh_param_descr(); |
46c1ad1f | 202 | return NULL; |
203 | } | |
204 | return mdescr; | |
205 | } | |
206 | ||
207 | /* Setter */ | |
7c37a24d JB |
208 | static int set_interface_meshparam(struct nl80211_state *state, |
209 | struct nl_cb *cb, | |
210 | struct nl_msg *msg, | |
211 | int argc, char **argv) | |
46c1ad1f | 212 | { |
213 | int err; | |
214 | uint32_t ret; | |
215 | const struct mesh_param_descr *mdescr; | |
656aa246 | 216 | struct nlattr *container; |
46c1ad1f | 217 | _any any; |
218 | ||
219 | mdescr = find_mesh_param(argc, argv, "change"); | |
220 | if (!mdescr) | |
221 | return 2; | |
222 | if (argc != 2) { | |
223 | printf("Must specify a value for %s.\n", mdescr->name); | |
224 | return 2; | |
225 | } | |
226 | ||
227 | /* Parse the new value */ | |
228 | memset(&any, 0, sizeof(_any)); | |
229 | ret = mdescr->parse_fn(argv[1], &any); | |
230 | if (ret != 0) { | |
231 | printf("%s must be set to a number " | |
232 | "between 0 and %u\n", mdescr->name, ret); | |
233 | return 2; | |
234 | } | |
235 | ||
236 | /* Construct a netlink message */ | |
656aa246 | 237 | container = nla_nest_start(msg, NL80211_ATTR_MESH_PARAMS); |
46c1ad1f | 238 | if (!container) |
239 | return -ENOBUFS; | |
240 | err = mdescr->nla_put_fn(msg, mdescr->mesh_param_num, &any); | |
241 | nla_nest_end(msg, container); | |
242 | ||
243 | return err; | |
244 | } | |
245 | ||
246 | COMMAND(set, mesh_param, "<param> <value>", | |
247 | NL80211_CMD_SET_MESH_PARAMS, 0, CIB_NETDEV, set_interface_meshparam); | |
248 | ||
249 | /* Getter */ | |
250 | static int print_mesh_param_handler(struct nl_msg *msg, void *arg) | |
251 | { | |
252 | const struct mesh_param_descr *mdescr = arg; | |
253 | struct nlattr *attrs[NL80211_ATTR_MAX + 1]; | |
254 | struct nlattr *parent_attr; | |
255 | struct nlattr *mesh_params[NL80211_MESHCONF_ATTR_MAX + 1]; | |
256 | struct genlmsghdr *gnlh = nlmsg_data(nlmsg_hdr(msg)); | |
257 | ||
258 | /* locate NL80211_ATTR_MESH_PARAMS */ | |
259 | nla_parse(attrs, NL80211_ATTR_MAX, genlmsg_attrdata(gnlh, 0), | |
260 | genlmsg_attrlen(gnlh, 0), NULL); | |
261 | parent_attr = attrs[NL80211_ATTR_MESH_PARAMS]; | |
262 | if (!parent_attr) | |
263 | return -EINVAL; | |
264 | ||
265 | /* unpack the mesh parameters */ | |
266 | if (nla_parse_nested(mesh_params, NL80211_MESHCONF_ATTR_MAX, | |
267 | parent_attr, NULL)) | |
268 | return -EINVAL; | |
269 | ||
270 | /* print out the mesh parameter */ | |
271 | mdescr->nla_print_fn(mesh_params[mdescr->mesh_param_num]); | |
272 | printf("\n"); | |
273 | return NL_SKIP; | |
274 | } | |
275 | ||
7c37a24d JB |
276 | static int get_interface_meshparam(struct nl80211_state *state, |
277 | struct nl_cb *cb, | |
278 | struct nl_msg *msg, | |
279 | int argc, char **argv) | |
46c1ad1f | 280 | { |
281 | const struct mesh_param_descr *mdescr; | |
282 | ||
283 | mdescr = find_mesh_param(argc, argv, "get"); | |
284 | if (!mdescr) | |
285 | return 2; | |
286 | ||
287 | nl_cb_set(cb, NL_CB_VALID, NL_CB_CUSTOM, | |
288 | print_mesh_param_handler, (void *)mdescr); | |
289 | return 0; | |
290 | } | |
291 | ||
292 | COMMAND(get, mesh_param, "<param>", | |
293 | NL80211_CMD_GET_MESH_PARAMS, 0, CIB_NETDEV, get_interface_meshparam); |