Fixed time command segfault with no arguments
[busybox4maemo] / networking / brctl.c
1 /* vi: set sw=4 ts=4: */
2 /*
3  * Small implementation of brctl for busybox.
4  *
5  * Copyright (C) 2008 by Bernhard Fischer
6  *
7  * Some helper functions from bridge-utils are
8  * Copyright (C) 2000 Lennert Buytenhek
9  *
10  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
11  */
12 /* This applet currently uses only the ioctl interface and no sysfs at all.
13  * At the time of this writing this was considered a feature.
14  */
15 #include "libbb.h"
16 #include <linux/sockios.h>
17 #include <net/if.h>
18
19 /* Maximum number of ports supported per bridge interface.  */
20 #ifndef MAX_PORTS
21 #define MAX_PORTS 32
22 #endif
23
24 /* Use internal number parsing and not the "exact" conversion.  */
25 /* #define BRCTL_USE_INTERNAL 0 */ /* use exact conversion */
26 #define BRCTL_USE_INTERNAL 1
27
28 #ifdef ENABLE_FEATURE_BRCTL_SHOW
29 #error Remove these
30 #endif
31 #define ENABLE_FEATURE_BRCTL_SHOW 0
32 #define USE_FEATURE_BRCTL_SHOW(...)
33
34 #if ENABLE_FEATURE_BRCTL_FANCY
35 #include <linux/if_bridge.h>
36
37 /* FIXME: These 4 funcs are not really clean and could be improved */
38 static ALWAYS_INLINE void strtotimeval(struct timeval *tv,
39                 const char *time_str)
40 {
41         double secs;
42 #if BRCTL_USE_INTERNAL
43         secs = /*bb_*/strtod(time_str, NULL);
44         if (!secs)
45 #else
46         if (sscanf(time_str, "%lf", &secs) != 1)
47 #endif
48                 bb_error_msg_and_die (bb_msg_invalid_arg, time_str, "timespec");
49         tv->tv_sec = secs;
50         tv->tv_usec = 1000000 * (secs - tv->tv_sec);
51 }
52
53 static ALWAYS_INLINE unsigned long __tv_to_jiffies(const struct timeval *tv)
54 {
55         unsigned long long jif;
56
57         jif = 1000000ULL * tv->tv_sec + tv->tv_usec;
58
59         return jif/10000;
60 }
61 # if 0
62 static void __jiffies_to_tv(struct timeval *tv, unsigned long jiffies)
63 {
64         unsigned long long tvusec;
65
66         tvusec = 10000ULL*jiffies;
67         tv->tv_sec = tvusec/1000000;
68         tv->tv_usec = tvusec - 1000000 * tv->tv_sec;
69 }
70 # endif
71 static unsigned long str_to_jiffies(const char *time_str)
72 {
73         struct timeval tv;
74         strtotimeval(&tv, time_str);
75         return __tv_to_jiffies(&tv);
76 }
77
78 static void arm_ioctl(unsigned long *args,
79                 unsigned long arg0, unsigned long arg1, unsigned long arg2)
80 {
81         args[0] = arg0;
82         args[1] = arg1;
83         args[2] = arg2;
84         args[3] = 0;
85 }
86 #endif
87
88
89 int brctl_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
90 int brctl_main(int argc ATTRIBUTE_UNUSED, char **argv)
91 {
92         static const char keywords[] ALIGN1 =
93                 "addbr\0" "delbr\0" "addif\0" "delif\0"
94         USE_FEATURE_BRCTL_FANCY(
95                 "stp\0"
96                 "setageing\0" "setfd\0" "sethello\0" "setmaxage\0"
97                 "setpathcost\0" "setportprio\0" "setbridgeprio\0"
98         )
99                 USE_FEATURE_BRCTL_SHOW("showmacs\0" "show\0");
100
101         enum { ARG_addbr = 0, ARG_delbr, ARG_addif, ARG_delif
102                 USE_FEATURE_BRCTL_FANCY(,
103                    ARG_stp,
104                    ARG_setageing, ARG_setfd, ARG_sethello, ARG_setmaxage,
105                    ARG_setpathcost, ARG_setportprio, ARG_setbridgeprio
106                 )
107                   USE_FEATURE_BRCTL_SHOW(, ARG_showmacs, ARG_show)
108         };
109
110         int fd;
111         smallint key;
112         struct ifreq ifr;
113         char *br, *brif;
114 #if ENABLE_FEATURE_BRCTL_FANCY
115         unsigned long args[4] = {0, 0, 0, 0};
116         int port;
117         int tmp;
118 #endif
119
120         argv++;
121         while (*argv) {
122                 key = index_in_strings(keywords, *argv);
123                 if (key == -1) /* no match found in keywords array, bail out. */
124                         bb_error_msg_and_die(bb_msg_invalid_arg, *argv, applet_name);
125                 argv++;
126 #if ENABLE_FEATURE_BRCTL_SHOW
127                 if (key == ARG_show) { /* show */
128                         goto out; /* FIXME: implement me! :) */
129                 }
130 #endif
131                 fd = xsocket(AF_INET, SOCK_STREAM, 0);
132                 br = *argv++;
133
134                 if (key == ARG_addbr || key == ARG_delbr) { /* addbr or delbr */
135                         ioctl_or_perror_and_die(fd,
136                                         key == ARG_addbr ? SIOCBRADDBR : SIOCBRDELBR,
137                                         br, "bridge %s", br);
138                         goto done;
139                 }
140                 if (!*argv) /* all but 'show' need at least one argument */
141                         bb_show_usage();
142                 safe_strncpy(ifr.ifr_name, br, IFNAMSIZ);
143                 if (key == ARG_addif || key == ARG_delif) { /* addif or delif */
144                         brif = *argv++;
145                         ifr.ifr_ifindex = if_nametoindex(brif);
146                         if (!ifr.ifr_ifindex) {
147                                 bb_perror_msg_and_die("iface %s", brif);
148                         }
149                         ioctl_or_perror_and_die(fd,
150                                         key == ARG_addif ? SIOCBRADDIF : SIOCBRDELIF,
151                                         &ifr, "bridge %s", br);
152                         goto done;
153                 }
154 #if ENABLE_FEATURE_BRCTL_FANCY
155                 ifr.ifr_data = (char *) &args;
156                 if (key == ARG_stp) { /* stp */
157                         /* FIXME: parsing yes/y/on/1 versus no/n/off/0 is too involved */
158                         arm_ioctl(args, BRCTL_SET_BRIDGE_STP_STATE,
159                                           (unsigned)(**argv - '0'), 0);
160                         goto fire;
161                 }
162                 if ((unsigned)(key - ARG_stp) < 5) { /* time related ops */
163                         unsigned long op = (key == ARG_setageing) ? BRCTL_SET_AGEING_TIME :
164                                            (key == ARG_setfd) ? BRCTL_SET_BRIDGE_FORWARD_DELAY :
165                                            (key == ARG_sethello) ? BRCTL_SET_BRIDGE_HELLO_TIME :
166                                            /*key == ARG_setmaxage*/ BRCTL_SET_BRIDGE_MAX_AGE;
167                         arm_ioctl(args, op, str_to_jiffies(*argv), 0);
168                         goto fire;
169                 }
170                 port = -1;
171                 if (key == ARG_setpathcost || key == ARG_setportprio) {/* get portnum */
172                         int ifidx[MAX_PORTS];
173                         unsigned i;
174
175                         port = if_nametoindex(*argv);
176                         if (!port)
177                                 bb_error_msg_and_die(bb_msg_invalid_arg, *argv, "port");
178                         argv++;
179                         memset(ifidx, 0, sizeof ifidx);
180                         arm_ioctl(args, BRCTL_GET_PORT_LIST, (unsigned long)ifidx,
181                                           MAX_PORTS);
182                         xioctl(fd, SIOCDEVPRIVATE, &ifr);
183                         for (i = 0; i < MAX_PORTS; i++) {
184                                 if (ifidx[i] == port) {
185                                         port = i;
186                                         break;
187                                 }
188                         }
189                 }
190                 if (key == ARG_setpathcost
191                  || key == ARG_setportprio
192                  || key == ARG_setbridgeprio
193                 ) {
194                         unsigned long op = (key == ARG_setpathcost) ? BRCTL_SET_PATH_COST :
195                                            (key == ARG_setportprio) ? BRCTL_SET_PORT_PRIORITY :
196                                            /*key == ARG_setbridgeprio*/ BRCTL_SET_BRIDGE_PRIORITY;
197                         unsigned long arg1 = port;
198                         unsigned long arg2;
199 # if BRCTL_USE_INTERNAL
200                         tmp = xatoi(*argv);
201 # else
202                         if (sscanf(*argv, "%i", &tmp) != 1)
203                                 bb_error_msg_and_die(bb_msg_invalid_arg, *argv,
204                                                 key == ARG_setpathcost ? "cost" : "prio");
205 # endif
206                         if (key == ARG_setbridgeprio) {
207                                 arg1 = tmp;
208                                 arg2 = 0;
209                         } else
210                                 arg2 = tmp;
211                         arm_ioctl(args, op, arg1, arg2);
212                 }
213  fire:
214                 /* Execute the previously set command.  */
215                 xioctl(fd, SIOCDEVPRIVATE, &ifr);
216                 argv++;
217 #endif
218  done:
219                 if (ENABLE_FEATURE_CLEAN_UP)
220                         close(fd);
221         }
222  USE_FEATURE_BRCTL_SHOW(out:)
223         return EXIT_SUCCESS;
224 }