cstats, rstats - Try to eliminate traffic peaks in graphs
[tomato.git] / release / src / router / shared / misc.c
blobefcc5d98df516cbf9fd7eeab47e3d811c57ceb3c
1 /*
3 Tomato Firmware
4 Copyright (C) 2006-2009 Jonathan Zarate
6 */
7 #include <string.h>
8 #include <stdio.h>
9 #include <stdlib.h>
10 #include <unistd.h>
11 #include <fcntl.h>
12 #include <sys/stat.h>
13 #include <stdarg.h>
14 #include <syslog.h>
15 #include <sys/ioctl.h>
16 #include <net/if.h>
17 #include <netinet/in.h>
18 #include <sys/socket.h>
19 #include <arpa/inet.h>
20 #include <ifaddrs.h>
21 #include <sys/sysinfo.h>
22 #include <sys/types.h>
23 #include <sys/wait.h>
25 #include <bcmnvram.h>
26 #include <bcmdevs.h>
27 #include <wlutils.h>
29 #include "shutils.h"
30 #include "shared.h"
33 int get_wan_proto(void)
35 const char *names[] = { // order must be synced with def at shared.h
36 "static",
37 "dhcp",
38 "l2tp",
39 "pppoe",
40 "pptp",
41 "ppp3g",
42 "lte",
43 NULL
45 int i;
46 const char *p;
48 p = nvram_safe_get("wan_proto");
49 for (i = 0; names[i] != NULL; ++i) {
50 if (strcmp(p, names[i]) == 0) return i + 1;
52 return WP_DISABLED;
55 #ifdef TCONFIG_IPV6
56 int get_ipv6_service(void)
58 const char *names[] = { // order must be synced with def at shared.h
59 "native", // IPV6_NATIVE
60 "native-pd", // IPV6_NATIVE_DHCP
61 "6to4", // IPV6_ANYCAST_6TO4
62 "sit", // IPV6_6IN4
63 "other", // IPV6_MANUAL
64 "6rd", // IPV6_6RD
65 "6rd-pd", // IPV6_6RD_DHCP
66 NULL
68 int i;
69 const char *p;
71 p = nvram_safe_get("ipv6_service");
72 for (i = 0; names[i] != NULL; ++i) {
73 if (strcmp(p, names[i]) == 0) return i + 1;
75 return IPV6_DISABLED;
78 const char *ipv6_router_address(struct in6_addr *in6addr)
80 char *p;
81 struct in6_addr addr;
82 static char addr6[INET6_ADDRSTRLEN];
84 addr6[0] = '\0';
86 if ((p = nvram_get("ipv6_rtr_addr")) && *p) {
87 inet_pton(AF_INET6, p, &addr);
89 else if ((p = nvram_get("ipv6_prefix")) && *p) {
90 inet_pton(AF_INET6, p, &addr);
91 addr.s6_addr16[7] = htons(0x0001);
93 else {
94 return addr6;
97 inet_ntop(AF_INET6, &addr, addr6, sizeof(addr6));
98 if (in6addr)
99 memcpy(in6addr, &addr, sizeof(addr));
101 return addr6;
104 int calc_6rd_local_prefix(const struct in6_addr *prefix,
105 int prefix_len, int relay_prefix_len,
106 const struct in_addr *local_ip,
107 struct in6_addr *local_prefix, int *local_prefix_len)
109 // the following code is based on ipv6calc's code
110 uint32_t local_ip_bits, j;
111 int i;
113 if (!prefix || !local_ip || !local_prefix || !local_prefix_len) {
114 return 0;
117 *local_prefix_len = prefix_len + 32 - relay_prefix_len;
118 if (*local_prefix_len > 64) {
119 return 0;
122 local_ip_bits = ntohl(local_ip->s_addr) << relay_prefix_len;
124 for (i=0; i<4; i++) {
125 local_prefix->s6_addr32[i] = prefix->s6_addr32[i];
128 for (j = 0x80000000, i = prefix_len; i < *local_prefix_len; i++, j>>=1)
130 if (local_ip_bits & j)
131 local_prefix->s6_addr[i>>3] |= (0x80 >> (i & 0x7));
134 return 1;
136 #endif
138 int using_dhcpc(void)
140 switch (get_wan_proto()) {
141 case WP_DHCP:
142 case WP_LTE:
143 return 1;
144 case WP_L2TP:
145 case WP_PPTP:
146 return nvram_get_int("pptp_dhcp");
148 return 0;
151 int wl_client(int unit, int subunit)
153 char *mode = nvram_safe_get(wl_nvname("mode", unit, subunit));
155 return ((strcmp(mode, "sta") == 0) || (strcmp(mode, "wet") == 0));
158 int foreach_wif(int include_vifs, void *param,
159 int (*func)(int idx, int unit, int subunit, void *param))
161 char ifnames[256];
162 char name[64], ifname[64], *next = NULL;
163 int unit = -1, subunit = -1;
164 int i;
165 int ret = 0;
167 snprintf(ifnames, sizeof(ifnames), "%s %s %s %s %s %s %s %s %s %s",
168 nvram_safe_get("lan_ifnames"),
169 nvram_safe_get("lan1_ifnames"),
170 nvram_safe_get("lan2_ifnames"),
171 nvram_safe_get("lan3_ifnames"),
172 nvram_safe_get("wan_ifnames"),
173 nvram_safe_get("wl_ifname"),
174 nvram_safe_get("wl0_ifname"),
175 nvram_safe_get("wl0_vifs"),
176 nvram_safe_get("wl1_ifname"),
177 nvram_safe_get("wl1_vifs"));
178 remove_dups(ifnames, sizeof(ifnames));
179 sort_list(ifnames, sizeof(ifnames));
181 i = 0;
182 foreach(name, ifnames, next) {
183 if (nvifname_to_osifname(name, ifname, sizeof(ifname)) != 0)
184 continue;
186 if (wl_probe(ifname) || wl_ioctl(ifname, WLC_GET_INSTANCE, &unit, sizeof(unit)))
187 continue;
189 // Convert eth name to wl name
190 if (osifname_to_nvifname(name, ifname, sizeof(ifname)) != 0)
191 continue;
193 // Slave intefaces have a '.' in the name
194 if (strchr(ifname, '.') && !include_vifs)
195 continue;
197 if (get_ifname_unit(ifname, &unit, &subunit) < 0)
198 continue;
200 ret |= func(i++, unit, subunit, param);
202 return ret;
205 void notice_set(const char *path, const char *format, ...)
207 char p[256];
208 char buf[2048];
209 va_list args;
211 va_start(args, format);
212 vsnprintf(buf, sizeof(buf), format, args);
213 va_end(args);
215 mkdir("/var/notice", 0755);
216 snprintf(p, sizeof(p), "/var/notice/%s", path);
217 f_write_string(p, buf, 0, 0);
218 if (buf[0]) syslog(LOG_INFO, "notice[%s]: %s", path, buf);
222 // #define _x_dprintf(args...) syslog(LOG_DEBUG, args);
223 #define _x_dprintf(args...) do { } while (0);
225 //function for rstats & cstats
226 long check_wanup_time(void){
228 long wanuptime = 0; // wanuptime in seconds
229 struct sysinfo si;
230 long uptime;
232 sysinfo(&si); //get time
233 if(f_read("/var/lib/misc/wantime", &uptime, sizeof(uptime)) == sizeof(uptime)){
234 wanuptime = si.uptime - uptime; //calculate the difference
235 if(wanuptime < 0) wanuptime = 0; //something wrong?
237 else{
238 wanuptime = 0; //something wrong? f_read()?
241 return wanuptime;
244 int check_wanup(void)
246 int up = 0;
247 int proto;
248 char buf1[64];
249 char buf2[64];
250 const char *name;
251 int f;
252 struct ifreq ifr;
254 proto = get_wan_proto();
255 if (proto == WP_DISABLED)
257 if (nvram_match("boardrev", "0x11")) { // Ovislink 1600GL - led "connected" off
258 led(LED_WHITE,LED_OFF);
260 return 0;
263 if ((proto == WP_PPTP) || (proto == WP_L2TP) || (proto == WP_PPPOE) || (proto == WP_PPP3G)) {
264 if (f_read_string("/tmp/ppp/link", buf1, sizeof(buf1)) > 0) {
265 // contains the base name of a file in /var/run/ containing pid of a daemon
266 snprintf(buf2, sizeof(buf2), "/var/run/%s.pid", buf1);
267 if (f_read_string(buf2, buf1, sizeof(buf1)) > 0) {
268 name = psname(atoi(buf1), buf2, sizeof(buf2));
269 if (strcmp(name, "pppd") == 0) up = 1;
271 else {
272 _dprintf("%s: error reading %s\n", __FUNCTION__, buf2);
274 if (!up) {
275 unlink("/tmp/ppp/link");
276 _x_dprintf("required daemon not found, assuming link is dead\n");
279 else {
280 _x_dprintf("%s: error reading %s\n", __FUNCTION__, "/tmp/ppp/link");
283 else if (!nvram_match("wan_ipaddr", "0.0.0.0")) {
284 up = 1;
286 else {
287 _x_dprintf("%s: default !up\n", __FUNCTION__);
290 if ((up) && ((f = socket(AF_INET, SOCK_DGRAM, 0)) >= 0)) {
291 strlcpy(ifr.ifr_name, nvram_safe_get("wan_iface"), sizeof(ifr.ifr_name));
292 if (ioctl(f, SIOCGIFFLAGS, &ifr) < 0) {
293 up = 0;
294 _x_dprintf("%s: SIOCGIFFLAGS\n", __FUNCTION__);
296 close(f);
297 if ((ifr.ifr_flags & IFF_UP) == 0) {
298 up = 0;
299 _x_dprintf("%s: !IFF_UP\n", __FUNCTION__);
302 if (nvram_match("boardrev", "0x11")) { // Ovislink 1600GL - led "connected" on
303 led(LED_WHITE,up);
306 return up;
310 const dns_list_t *get_dns(void)
312 static dns_list_t dns;
313 char s[512];
314 int n;
315 int i, j;
316 struct in_addr ia;
317 char d[7][22];
318 unsigned short port;
319 char *c;
321 dns.count = 0;
323 strlcpy(s, nvram_safe_get("wan_dns"), sizeof(s));
324 if ((nvram_get_int("dns_addget")) || (s[0] == 0)) {
325 n = strlen(s);
326 snprintf(s + n, sizeof(s) - n, " %s", nvram_safe_get("wan_get_dns"));
329 n = sscanf(s, "%21s %21s %21s %21s %21s %21s %21s", d[0], d[1], d[2], d[3], d[4], d[5], d[6]);
330 for (i = 0; i < n; ++i) {
331 port = 53;
333 if ((c = strchr(d[i], ':')) != NULL) {
334 *c++ = 0;
335 if (((j = atoi(c)) < 1) || (j > 0xFFFF)) continue;
336 port = j;
339 if (inet_pton(AF_INET, d[i], &ia) > 0) {
340 for (j = dns.count - 1; j >= 0; --j) {
341 if ((dns.dns[j].addr.s_addr == ia.s_addr) && (dns.dns[j].port == port)) break;
343 if (j < 0) {
344 dns.dns[dns.count].port = port;
345 dns.dns[dns.count++].addr.s_addr = ia.s_addr;
346 if (dns.count == 6) break;
351 return &dns;
354 // -----------------------------------------------------------------------------
356 void set_action(int a)
358 int r = 3;
359 while (f_write("/var/lock/action", &a, sizeof(a), 0, 0) != sizeof(a)) {
360 sleep(1);
361 if (--r == 0) return;
363 if (a != ACT_IDLE) sleep(2);
366 int check_action(void)
368 int a;
369 int r = 3;
371 while (f_read("/var/lock/action", &a, sizeof(a)) != sizeof(a)) {
372 sleep(1);
373 if (--r == 0) return ACT_UNKNOWN;
375 return a;
378 int wait_action_idle(int n)
380 while (n-- > 0) {
381 if (check_action() == ACT_IDLE) return 1;
382 sleep(1);
384 return 0;
387 // -----------------------------------------------------------------------------
389 const wanface_list_t *get_wanfaces(void)
391 static wanface_list_t wanfaces;
392 char *ip, *iface;
393 int proto;
395 wanfaces.count = 0;
397 switch ((proto = get_wan_proto())) {
398 case WP_PPTP:
399 case WP_L2TP:
400 while (wanfaces.count < 2) {
401 if (wanfaces.count == 0) {
402 ip = nvram_safe_get("ppp_get_ip");
403 iface = nvram_safe_get("wan_iface");
404 if (!(*iface)) iface = "ppp+";
406 else /* if (wanfaces.count == 1) */ {
407 ip = nvram_safe_get("wan_ipaddr");
408 if ((!(*ip) || strcmp(ip, "0.0.0.0") == 0) && (wanfaces.count > 0))
409 iface = "";
410 else
411 iface = nvram_safe_get("wan_ifname");
413 strlcpy(wanfaces.iface[wanfaces.count].ip, ip, sizeof(wanfaces.iface[0].ip));
414 strlcpy(wanfaces.iface[wanfaces.count].name, iface, IFNAMSIZ);
415 ++wanfaces.count;
417 break;
418 default:
419 ip = (proto == WP_DISABLED) ? "0.0.0.0" : nvram_safe_get("wan_ipaddr");
420 if ((proto == WP_PPPOE) || (proto == WP_PPP3G)) {
421 iface = nvram_safe_get("wan_iface");
422 if (!(*iface)) iface = "ppp+";
424 else if (proto == WP_LTE) {
425 iface = nvram_safe_get("wan_4g");
426 } else {
427 iface = nvram_safe_get("wan_ifname");
429 strlcpy(wanfaces.iface[wanfaces.count].ip, ip, sizeof(wanfaces.iface[0].ip));
430 strlcpy(wanfaces.iface[wanfaces.count++].name, iface, IFNAMSIZ);
431 break;
434 return &wanfaces;
437 const char *get_wanface(void)
439 return (*get_wanfaces()).iface[0].name;
442 #ifdef TCONFIG_IPV6
443 const char *get_wan6face(void)
445 switch (get_ipv6_service()) {
446 case IPV6_NATIVE:
447 case IPV6_NATIVE_DHCP:
448 return get_wanface();
449 case IPV6_ANYCAST_6TO4:
450 return "v6to4";
451 case IPV6_6IN4:
452 return "v6in4";
454 return nvram_safe_get("ipv6_ifname");
456 #endif
458 const char *get_wanip(void)
460 if (!check_wanup()) return "0.0.0.0";
462 return (*get_wanfaces()).iface[0].ip;
465 const char *getifaddr(char *ifname, int family, int linklocal)
467 static char buf[INET6_ADDRSTRLEN];
468 void *addr = NULL;
469 struct ifaddrs *ifap, *ifa;
471 if (getifaddrs(&ifap) != 0) {
472 _dprintf("getifaddrs failed: %s\n", strerror(errno));
473 return NULL;
476 for (ifa = ifap; ifa; ifa = ifa->ifa_next) {
477 if ((ifa->ifa_addr == NULL) ||
478 (strncmp(ifa->ifa_name, ifname, IFNAMSIZ) != 0) ||
479 (ifa->ifa_addr->sa_family != family))
480 continue;
482 #ifdef TCONFIG_IPV6
483 if (ifa->ifa_addr->sa_family == AF_INET6) {
484 struct sockaddr_in6 *s6 = (struct sockaddr_in6 *)(ifa->ifa_addr);
485 if (IN6_IS_ADDR_LINKLOCAL(&s6->sin6_addr) ^ linklocal)
486 continue;
487 addr = (void *)&(s6->sin6_addr);
489 else
490 #endif
492 struct sockaddr_in *s = (struct sockaddr_in *)(ifa->ifa_addr);
493 addr = (void *)&(s->sin_addr);
496 if ((addr) && inet_ntop(ifa->ifa_addr->sa_family, addr, buf, sizeof(buf)) != NULL) {
497 freeifaddrs(ifap);
498 return buf;
502 freeifaddrs(ifap);
503 return NULL;
506 // -----------------------------------------------------------------------------
508 long get_uptime(void)
510 struct sysinfo si;
511 sysinfo(&si);
512 return si.uptime;
515 char *wl_nvname(const char *nv, int unit, int subunit)
517 static char tmp[128];
518 char prefix[] = "wlXXXXXXXXXX_";
520 if (unit < 0)
521 strcpy(prefix, "wl_");
522 else if (subunit > 0)
523 snprintf(prefix, sizeof(prefix), "wl%d.%d_", unit, subunit);
524 else
525 snprintf(prefix, sizeof(prefix), "wl%d_", unit);
526 return strcat_r(prefix, nv, tmp);
529 int get_radio(int unit)
531 uint32 n;
533 return (wl_ioctl(nvram_safe_get(wl_nvname("ifname", unit, 0)), WLC_GET_RADIO, &n, sizeof(n)) == 0) &&
534 ((n & WL_RADIO_SW_DISABLE) == 0);
537 void set_radio(int on, int unit)
539 uint32 n;
541 #ifndef WL_BSS_INFO_VERSION
542 #error WL_BSS_INFO_VERSION
543 #endif
545 #if WL_BSS_INFO_VERSION >= 108
546 n = on ? (WL_RADIO_SW_DISABLE << 16) : ((WL_RADIO_SW_DISABLE << 16) | 1);
547 wl_ioctl(nvram_safe_get(wl_nvname("ifname", unit, 0)), WLC_SET_RADIO, &n, sizeof(n));
548 if (!on) {
549 led(LED_WLAN, 0);
550 led(LED_DIAG, 0);
552 #else
553 n = on ? 0 : WL_RADIO_SW_DISABLE;
554 wl_ioctl(nvram_safe_get(wl_nvname("ifname", unit, 0)), WLC_SET_RADIO, &n, sizeof(n));
555 if (!on) {
556 led(LED_DIAG, 0);
558 #endif
561 // -----------------------------------------------------------------------------
563 int mtd_getinfo(const char *mtdname, int *part, int *size)
565 FILE *f;
566 char s[256];
567 char t[256];
568 int r;
570 r = 0;
571 if ((strlen(mtdname) < 128) && (strcmp(mtdname, "pmon") != 0)) {
572 sprintf(t, "\"%s\"", mtdname);
573 if ((f = fopen("/proc/mtd", "r")) != NULL) {
574 while (fgets(s, sizeof(s), f) != NULL) {
575 if ((sscanf(s, "mtd%d: %x", part, size) == 2) && (strstr(s, t) != NULL)) {
576 // don't accidentally mess with bl (0)
577 if (*part > 0) r = 1;
578 break;
581 fclose(f);
584 if (!r) {
585 *size = 0;
586 *part = -1;
588 return r;
591 // -----------------------------------------------------------------------------
593 int nvram_get_int(const char *key)
595 return atoi(nvram_safe_get(key));
599 long nvram_xget_long(const char *name, long min, long max, long def)
601 const char *p;
602 char *e;
603 long n;
605 p = nvram_get(name);
606 if ((p != NULL) && (*p != 0)) {
607 n = strtol(p, &e, 0);
608 if ((e != p) && ((*e == 0) || (*e == ' ')) && (n > min) && (n < max)) {
609 return n;
612 return def;
616 int nvram_get_file(const char *key, const char *fname, int max)
618 int n;
619 char *p;
620 char *b;
621 int r;
623 r = 0;
624 p = nvram_safe_get(key);
625 n = strlen(p);
626 if (n <= max) {
627 if ((b = malloc(base64_decoded_len(n) + 128)) != NULL) {
628 n = base64_decode(p, b, n);
629 if (n > 0) r = (f_write(fname, b, n, 0, 0644) == n);
630 free(b);
633 return r;
635 char b[2048];
636 int n;
637 char *p;
639 p = nvram_safe_get(key);
640 n = strlen(p);
641 if (n <= max) {
642 n = base64_decode(p, b, n);
643 if (n > 0) return (f_write(fname, b, n, 0, 0700) == n);
645 return 0;
649 int nvram_set_file(const char *key, const char *fname, int max)
651 char *in;
652 char *out;
653 long len;
654 int n;
655 int r;
657 if ((len = f_size(fname)) > max) return 0;
658 max = (int)len;
659 r = 0;
660 if (f_read_alloc(fname, &in, max) == max) {
661 if ((out = malloc(base64_encoded_len(max) + 128)) != NULL) {
662 n = base64_encode(in, out, max);
663 out[n] = 0;
664 nvram_set(key, out);
665 free(out);
666 r = 1;
668 free(in);
670 return r;
672 char a[2048];
673 char b[4096];
674 int n;
676 if (((n = f_read(fname, &a, sizeof(a))) > 0) && (n <= max)) {
677 n = base64_encode(a, b, n);
678 b[n] = 0;
679 nvram_set(key, b);
680 return 1;
682 return 0;
686 int nvram_contains_word(const char *key, const char *word)
688 return (find_word(nvram_safe_get(key), word) != NULL);
691 int nvram_is_empty(const char *key)
693 char *p;
694 return (((p = nvram_get(key)) == NULL) || (*p == 0));
697 void nvram_commit_x(void)
699 if (!nvram_get_int("debug_nocommit")) nvram_commit();
702 int connect_timeout(int fd, const struct sockaddr *addr, socklen_t len, int timeout)
704 fd_set fds;
705 struct timeval tv;
706 int flags;
707 int n;
708 int r;
710 if (((flags = fcntl(fd, F_GETFL, 0)) < 0) ||
711 (fcntl(fd, F_SETFL, flags | O_NONBLOCK) < 0)) {
712 _dprintf("%s: error in F_*ETFL %d\n", __FUNCTION__, fd);
713 return -1;
716 if (connect(fd, addr, len) < 0) {
717 // _dprintf("%s: connect %d = <0\n", __FUNCTION__, fd);
719 if (errno != EINPROGRESS) {
720 _dprintf("%s: error in connect %d errno=%d\n", __FUNCTION__, fd, errno);
721 return -1;
724 while (1) {
725 tv.tv_sec = timeout;
726 tv.tv_usec = 0;
727 FD_ZERO(&fds);
728 FD_SET(fd, &fds);
729 r = select(fd + 1, NULL, &fds, NULL, &tv);
730 if (r == 0) {
731 _dprintf("%s: timeout in select %d\n", __FUNCTION__, fd);
732 return -1;
734 else if (r < 0) {
735 if (errno != EINTR) {
736 _dprintf("%s: error in select %d\n", __FUNCTION__, fd);
737 return -1;
739 // loop
741 else {
742 r = 0;
743 n = sizeof(r);
744 if ((getsockopt(fd, SOL_SOCKET, SO_ERROR, &r, &n) < 0) || (r != 0)) {
745 _dprintf("%s: error in SO_ERROR %d\n", __FUNCTION__, fd);
746 return -1;
748 break;
753 if (fcntl(fd, F_SETFL, flags) < 0) {
754 _dprintf("%s: error in F_*ETFL %d\n", __FUNCTION__, fd);
755 return -1;
758 // _dprintf("%s: OK %d\n", __FUNCTION__, fd);
759 return 0;
762 void chld_reap(int sig)
764 while (waitpid(-1, NULL, WNOHANG) > 0) {}
768 int time_ok(void)
770 return time(0) > Y2K;