]>
Commit | Line | Data |
---|---|---|
cad53b3f JB |
1 | /* |
2 | * nl80211 userspace tool | |
3 | * | |
2a1fced2 | 4 | * Copyright 2007, 2008 Johannes Berg <johannes@sipsolutions.net> |
cad53b3f JB |
5 | */ |
6 | ||
7 | #include <errno.h> | |
8 | #include <stdio.h> | |
d5ac8ad3 | 9 | #include <string.h> |
bd396f2a JB |
10 | #include <net/if.h> |
11 | #include <sys/types.h> | |
12 | #include <sys/stat.h> | |
13 | #include <fcntl.h> | |
14 | #include <unistd.h> | |
15 | ||
cad53b3f JB |
16 | #include <netlink/genl/genl.h> |
17 | #include <netlink/genl/family.h> | |
18 | #include <netlink/genl/ctrl.h> | |
19 | #include <netlink/msg.h> | |
20 | #include <netlink/attr.h> | |
21 | #include <linux/nl80211.h> | |
22 | ||
23 | #include "iw.h" | |
24 | ||
25 | ||
26 | static int nl80211_init(struct nl80211_state *state) | |
27 | { | |
28 | int err; | |
29 | ||
30 | state->nl_handle = nl_handle_alloc(); | |
31 | if (!state->nl_handle) { | |
32 | fprintf(stderr, "Failed to allocate netlink handle.\n"); | |
33 | return -ENOMEM; | |
34 | } | |
35 | ||
36 | if (genl_connect(state->nl_handle)) { | |
37 | fprintf(stderr, "Failed to connect to generic netlink.\n"); | |
38 | err = -ENOLINK; | |
39 | goto out_handle_destroy; | |
40 | } | |
41 | ||
42 | state->nl_cache = genl_ctrl_alloc_cache(state->nl_handle); | |
43 | if (!state->nl_cache) { | |
44 | fprintf(stderr, "Failed to allocate generic netlink cache.\n"); | |
45 | err = -ENOMEM; | |
46 | goto out_handle_destroy; | |
47 | } | |
48 | ||
49 | state->nl80211 = genl_ctrl_search_by_name(state->nl_cache, "nl80211"); | |
50 | if (!state->nl80211) { | |
51 | fprintf(stderr, "nl80211 not found.\n"); | |
52 | err = -ENOENT; | |
53 | goto out_cache_free; | |
54 | } | |
55 | ||
56 | return 0; | |
57 | ||
58 | out_cache_free: | |
59 | nl_cache_free(state->nl_cache); | |
60 | out_handle_destroy: | |
61 | nl_handle_destroy(state->nl_handle); | |
62 | return err; | |
63 | } | |
64 | ||
65 | static void nl80211_cleanup(struct nl80211_state *state) | |
66 | { | |
67 | genl_family_put(state->nl80211); | |
68 | nl_cache_free(state->nl_cache); | |
69 | nl_handle_destroy(state->nl_handle); | |
70 | } | |
71 | ||
bd396f2a JB |
72 | static void usage(const char *argv0) |
73 | { | |
74 | struct cmd *cmd; | |
75 | ||
76 | fprintf(stderr, "Usage:\t%s help\n", argv0); | |
77 | for (cmd = &__start___cmd; cmd < &__stop___cmd; cmd++) { | |
78 | switch (cmd->idby) { | |
79 | case CIB_NONE: | |
d631650b JB |
80 | fprintf(stderr, "\t%s ", argv0); |
81 | /* fall through */ | |
bd396f2a | 82 | case CIB_PHY: |
d631650b JB |
83 | if (cmd->idby == CIB_PHY) |
84 | fprintf(stderr, "\t%s phy <phyname> ", argv0); | |
bd396f2a JB |
85 | /* fall through */ |
86 | case CIB_NETDEV: | |
87 | if (cmd->idby == CIB_NETDEV) | |
88 | fprintf(stderr, "\t%s dev <devname> ", argv0); | |
89 | if (cmd->section) | |
90 | fprintf(stderr, "%s ", cmd->section); | |
91 | fprintf(stderr, "%s", cmd->name); | |
92 | if (cmd->args) | |
93 | fprintf(stderr, " %s", cmd->args); | |
94 | fprintf(stderr, "\n"); | |
95 | break; | |
96 | } | |
97 | } | |
98 | } | |
99 | ||
100 | static int phy_lookup(char *name) | |
101 | { | |
102 | char buf[200]; | |
103 | int fd, pos; | |
104 | ||
105 | snprintf(buf, sizeof(buf), "/sys/class/ieee80211/%s/index", name); | |
106 | ||
107 | fd = open(buf, O_RDONLY); | |
108 | pos = read(fd, buf, sizeof(buf) - 1); | |
109 | if (pos < 0) | |
110 | return -1; | |
111 | buf[pos] = '\0'; | |
112 | return atoi(buf); | |
113 | } | |
114 | ||
d631650b JB |
115 | static int handle_cmd(struct nl80211_state *state, |
116 | enum command_identify_by idby, | |
117 | int argc, char **argv) | |
45c7212c | 118 | { |
bd396f2a JB |
119 | struct cmd *cmd; |
120 | struct nl_msg *msg; | |
121 | int devidx = 0; | |
122 | const char *command, *section; | |
45c7212c | 123 | |
d631650b | 124 | if (argc <= 1 && idby != CIB_NONE) |
bd396f2a | 125 | return -1; |
45c7212c | 126 | |
bd396f2a JB |
127 | switch (idby) { |
128 | case CIB_PHY: | |
129 | devidx = phy_lookup(*argv); | |
130 | argc--; | |
131 | argv++; | |
132 | break; | |
133 | case CIB_NETDEV: | |
134 | devidx = if_nametoindex(*argv); | |
135 | argc--; | |
136 | argv++; | |
137 | break; | |
138 | default: | |
139 | break; | |
140 | } | |
141 | ||
142 | section = command = *argv; | |
143 | argc--; | |
144 | argv++; | |
145 | ||
146 | for (cmd = &__start___cmd; cmd < &__stop___cmd; cmd++) { | |
147 | if (cmd->idby != idby) | |
148 | continue; | |
149 | if (cmd->section) { | |
150 | if (strcmp(cmd->section, section)) | |
151 | continue; | |
152 | /* this is a bit icky ... */ | |
153 | if (command == section) { | |
154 | if (argc <= 0) | |
155 | return -1; | |
156 | command = *argv; | |
157 | argc--; | |
158 | argv++; | |
159 | } | |
160 | } else if (section != command) | |
161 | continue; | |
162 | if (strcmp(cmd->name, command)) | |
163 | continue; | |
164 | if (argc && !cmd->args) | |
165 | continue; | |
166 | break; | |
167 | } | |
45c7212c | 168 | |
bd396f2a JB |
169 | if (cmd == &__stop___cmd) |
170 | return -1; | |
45c7212c | 171 | |
bd396f2a JB |
172 | msg = nlmsg_alloc(); |
173 | if (!msg) { | |
174 | fprintf(stderr, "out of memory\n"); | |
45c7212c | 175 | return 1; |
bd396f2a | 176 | } |
45c7212c | 177 | |
bd396f2a JB |
178 | genlmsg_put(msg, 0, 0, genl_family_get_id(state->nl80211), 0, |
179 | cmd->nl_msg_flags, cmd->cmd, 0); | |
180 | ||
181 | switch (idby) { | |
182 | case CIB_PHY: | |
183 | NLA_PUT_U32(msg, NL80211_ATTR_WIPHY, devidx); | |
184 | break; | |
185 | case CIB_NETDEV: | |
186 | NLA_PUT_U32(msg, NL80211_ATTR_IFINDEX, devidx); | |
187 | break; | |
188 | default: | |
189 | break; | |
190 | } | |
191 | ||
192 | return cmd->handler(state, msg, argc, argv); | |
193 | nla_put_failure: | |
194 | fprintf(stderr, "building message failed\n"); | |
195 | return 1; | |
45c7212c JB |
196 | } |
197 | ||
cad53b3f JB |
198 | int main(int argc, char **argv) |
199 | { | |
200 | struct nl80211_state nlstate; | |
bd396f2a JB |
201 | int err; |
202 | const char *argv0; | |
cad53b3f JB |
203 | |
204 | err = nl80211_init(&nlstate); | |
205 | if (err) | |
206 | return 1; | |
207 | ||
45c7212c JB |
208 | /* strip off self */ |
209 | argc--; | |
1cdd9016 MK |
210 | argv0 = *argv++; |
211 | ||
bd396f2a | 212 | if (argc == 0 || strcmp(*argv, "help") == 0) { |
1cdd9016 MK |
213 | usage(argv0); |
214 | goto out; | |
215 | } | |
45c7212c | 216 | |
bd396f2a | 217 | if (strcmp(*argv, "dev") == 0) { |
14a0380d LR |
218 | argc--; |
219 | argv++; | |
d631650b | 220 | err = handle_cmd(&nlstate, CIB_NETDEV, argc, argv); |
bd396f2a JB |
221 | } else if (strcmp(*argv, "phy") == 0) { |
222 | argc--; | |
223 | argv++; | |
d631650b | 224 | err = handle_cmd(&nlstate, CIB_PHY, argc, argv); |
bd396f2a | 225 | } else |
d631650b | 226 | err = handle_cmd(&nlstate, CIB_NONE, argc, argv); |
45c7212c | 227 | |
bd396f2a JB |
228 | if (err < 0) |
229 | usage(argv0); | |
cad53b3f | 230 | |
45c7212c | 231 | out: |
cad53b3f JB |
232 | nl80211_cleanup(&nlstate); |
233 | ||
45c7212c | 234 | return err; |
cad53b3f | 235 | } |