diff options
Diffstat (limited to 'package/wprobe/src/user/wprobe-util.c')
| -rw-r--r-- | package/wprobe/src/user/wprobe-util.c | 450 |
1 files changed, 0 insertions, 450 deletions
diff --git a/package/wprobe/src/user/wprobe-util.c b/package/wprobe/src/user/wprobe-util.c deleted file mode 100644 index 654442f9c..000000000 --- a/package/wprobe/src/user/wprobe-util.c +++ /dev/null @@ -1,450 +0,0 @@ -/* - * wprobe-test.c: Wireless probe user space test code - * Copyright (C) 2008-2009 Felix Fietkau <nbd@openwrt.org> - * - * This program is free software; you can redistribute it and/or - * modify it under the terms of the GNU General Public License - * as published by the Free Software Foundation; either version 2 - * of the License, or (at your option) any later version. - * - * This program is distributed in the hope that it will be useful, - * but WITHOUT ANY WARRANTY; without even the implied warranty of - * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - * GNU General Public License for more details. - */ - -#include <sys/types.h> -#include <sys/socket.h> -#include <sys/wait.h> -#include <arpa/inet.h> -#include <stdio.h> -#include <string.h> -#include <stdlib.h> -#include <inttypes.h> -#include <errno.h> -#include <stdint.h> -#include <getopt.h> -#include <stdbool.h> -#include <unistd.h> -#include <netdb.h> -#include <fcntl.h> -#include <signal.h> - -#include <linux/wprobe.h> -#include "wprobe.h" - -static bool simple_mode = false; - -static const char * -wprobe_dump_value(struct wprobe_attribute *attr) -{ - static char buf[128]; - -#define HANDLE_TYPE(_type, _format) \ - case WPROBE_VAL_##_type: \ - snprintf(buf, sizeof(buf), _format, attr->val._type); \ - break - - switch(attr->type) { - HANDLE_TYPE(S8, "%d"); - HANDLE_TYPE(S16, "%d"); - HANDLE_TYPE(S32, "%d"); - HANDLE_TYPE(S64, "%lld"); - HANDLE_TYPE(U8, "%d"); - HANDLE_TYPE(U16, "%d"); - HANDLE_TYPE(U32, "%d"); - HANDLE_TYPE(U64, "%lld"); - case WPROBE_VAL_STRING: - /* FIXME: implement this */ - default: - strncpy(buf, "<unknown>", sizeof(buf)); - break; - } - if ((attr->flags & WPROBE_F_KEEPSTAT) && - (attr->val.n > 0)) { - int len = strlen(buf); - if (simple_mode) - snprintf(buf + len, sizeof(buf) - len, ";%.02f;%.02f;%d;%lld;%lld", attr->val.avg, attr->val.stdev, attr->val.n, attr->val.s, attr->val.ss); - else - snprintf(buf + len, sizeof(buf) - len, " (avg: %.02f; stdev: %.02f, n=%d)", attr->val.avg, attr->val.stdev, attr->val.n); - } -#undef HANDLE_TYPE - - return buf; -} - - -static void -wprobe_dump_data(struct wprobe_iface *dev) -{ - struct wprobe_attribute *attr; - struct wprobe_link *link; - bool first = true; - - if (!simple_mode) - fprintf(stdout, "\n"); - wprobe_request_data(dev, NULL); - list_for_each_entry(attr, &dev->global_attr, list) { - if (simple_mode) { - if (first) - fprintf(stdout, "[global]\n"); - fprintf(stdout, "%s=%s\n", attr->name, wprobe_dump_value(attr)); - } else { - fprintf(stdout, (first ? - "Global: %s=%s\n" : - " %s=%s\n"), - attr->name, - wprobe_dump_value(attr) - ); - } - first = false; - } - - list_for_each_entry(link, &dev->links, list) { - first = true; - wprobe_request_data(dev, link->addr); - list_for_each_entry(attr, &dev->link_attr, list) { - if (first) { - fprintf(stdout, - (simple_mode ? - "[%02x:%02x:%02x:%02x:%02x:%02x]\n%s=%s\n" : - "%02x:%02x:%02x:%02x:%02x:%02x: %s=%s\n"), - link->addr[0], link->addr[1], link->addr[2], - link->addr[3], link->addr[4], link->addr[5], - attr->name, - wprobe_dump_value(attr)); - first = false; - } else { - fprintf(stdout, - (simple_mode ? "%s=%s\n" : - " %s=%s\n"), - attr->name, - wprobe_dump_value(attr)); - } - } - } - fflush(stdout); -} - -static const char *attr_typestr[] = { - [0] = "Unknown", - [WPROBE_VAL_STRING] = "String", - [WPROBE_VAL_U8] = "Unsigned 8 bit", - [WPROBE_VAL_U16] = "Unsigned 16 bit", - [WPROBE_VAL_U32] = "Unsigned 32 bit", - [WPROBE_VAL_U64] = "Unsigned 64 bit", - [WPROBE_VAL_S8] = "Signed 8 bit", - [WPROBE_VAL_S16] = "Signed 16 bit", - [WPROBE_VAL_S32] = "Signed 32 bit", - [WPROBE_VAL_S64] = "Signed 64 bit", -}; - -static int usage(const char *prog) -{ - fprintf(stderr, -#ifndef NO_LOCAL_ACCESS - "Usage: %s <interface>|<host>:<device>|-P [options]\n" -#else - "Usage: %s <host>:<device> [options]\n" -#endif - "\n" - "Options:\n" - " -a: Print attributes\n" - " -c: Only apply configuration\n" - " -d: Delay between measurement dumps (in milliseconds, default: 1000)\n" - " A value of 0 (zero) prints once and exits; useful for scripts\n" - " -f: Dump contents of layer 2 filter counters during measurement\n" - " -F <file>: Apply layer 2 filters from <file>\n" - " -h: This help text\n" - " -i <interval>: Set measurement interval\n" - " -m: Run measurement loop\n" - " -p: Set the TCP port for server/client (default: 17990)\n" -#ifndef NO_LOCAL_ACCESS - " -P: Run in proxy mode (listen on network)\n" -#endif - "\n" - , prog); - exit(1); -} - -static void show_attributes(struct wprobe_iface *dev) -{ - struct wprobe_attribute *attr; - if (simple_mode) - return; - list_for_each_entry(attr, &dev->global_attr, list) { - fprintf(stdout, "Global attribute: '%s' (%s)\n", - attr->name, attr_typestr[attr->type]); - } - list_for_each_entry(attr, &dev->link_attr, list) { - fprintf(stdout, "Link attribute: '%s' (%s)\n", - attr->name, attr_typestr[attr->type]); - } -} - -static void show_filter_simple(void *arg, const char *group, struct wprobe_filter_item *items, int n_items) -{ - int i; - - fprintf(stdout, "[filter:%s]\n", group); - for (i = 0; i < n_items; i++) { - fprintf(stdout, "%s=%lld;%lld\n", - items[i].name, items[i].tx, items[i].rx); - } - fflush(stdout); -} - - -static void show_filter(void *arg, const char *group, struct wprobe_filter_item *items, int n_items) -{ - int i; - fprintf(stdout, "Filter group: '%s' (tx/rx)\n", group); - for (i = 0; i < n_items; i++) { - fprintf(stdout, " - %s (%lld/%lld)\n", - items[i].name, items[i].tx, items[i].rx); - } -} - -static void loop_measurement(struct wprobe_iface *dev, bool print_filters, unsigned long delay) -{ - do { - wprobe_update_links(dev); - wprobe_dump_data(dev); - if (print_filters) - wprobe_dump_filters(dev, simple_mode ? show_filter_simple : show_filter, NULL); - usleep(delay * 1000); - } - while (delay); -} - -static void set_filter(struct wprobe_iface *dev, const char *filename) -{ - unsigned char *buf = NULL; - unsigned int buflen = 0; - unsigned int len = 0; - int fd; - - /* clear filter */ - if (filename[0] == 0) { - dev->filter_len = -1; - return; - } - - fd = open(filename, O_RDONLY); - if (fd < 0) { - perror("open filter"); - return; - } - - do { - int rlen; - - if (!buf) { - len = 0; - buflen = 1024; - buf = malloc(1024); - } else { - buflen *= 2; - buf = realloc(buf, buflen); - } - rlen = read(fd, buf + len, buflen - len); - if (rlen < 0) - break; - - len += rlen; - } while (len == buflen); - - dev->filter = buf; - dev->filter_len = len; - close(fd); -} - -#ifndef NO_LOCAL_ACCESS - -static void sigchld_handler(int s) -{ - while (waitpid(-1, NULL, WNOHANG) > 0); -} - -static int run_proxy(int port) -{ - struct sockaddr_in sa; - struct sigaction sig; - int v = 1; - int s; - - s = socket(AF_INET, SOCK_STREAM, 0); - if (s < 0) { - perror("socket"); - return 1; - } - - sig.sa_handler = sigchld_handler; // Signal Handler fuer Zombie Prozesse - sigemptyset(&sig.sa_mask); - sig.sa_flags = SA_RESTART; - sigaction(SIGCHLD, &sig, NULL); - - memset(&sa, 0, sizeof(sa)); - sa.sin_family = AF_INET; - sa.sin_addr.s_addr = htonl(INADDR_ANY); - sa.sin_port = htons(wprobe_port); - - setsockopt(s, SOL_SOCKET, SO_REUSEADDR, &v, sizeof(v)); - if (bind(s, (struct sockaddr *) &sa, sizeof(sa)) < 0) { - perror("bind"); - return 1; - } - if (listen(s, 10)) { - perror("listen"); - return 1; - } - while(1) { - unsigned int addrlen = sizeof(struct sockaddr_in); - int ret, c; - - c = accept(s, (struct sockaddr *)&sa, &addrlen); - if (c < 0) { - if (errno == EINTR) - continue; - - perror("accept"); - return 1; - } - if (fork() == 0) { - /* close server socket, stdin, stdout, stderr */ - close(s); - close(0); - close(1); - close(2); - - wprobe_server_init(c); - do { - ret = wprobe_server_handle(c); - } while (ret >= 0); - wprobe_server_done(); - close(c); - exit(0); - } - close(c); - } - - return 0; -} -#endif - -int main(int argc, char **argv) -{ - struct wprobe_iface *dev = NULL; - const char *ifname; - const char *prog = argv[0]; - char *err = NULL; - enum { - CMD_NONE, - CMD_CONFIG, - CMD_MEASURE, - CMD_PROXY, - } cmd = CMD_NONE; - const char *filter = NULL; - bool print_attributes = false; - bool print_filters = false; - unsigned long delay = 1000; - int interval = -1; - int ch; - - if (argc < 2) - return usage(prog); - -#ifndef NO_LOCAL_ACCESS - if (!strcmp(argv[1], "-P")) { - while ((ch = getopt(argc - 1, argv + 1, "p:")) != -1) { - switch(ch) { - case 'p': - /* set port */ - wprobe_port = strtoul(optarg, NULL, 0); - break; - default: - return usage(prog); - } - } - return run_proxy(wprobe_port); - } -#endif - - if (argv[1][0] == '-') - return usage(prog); - - ifname = argv[1]; - argv++; - argc--; - - while ((ch = getopt(argc, argv, "acd:fF:hi:msp:")) != -1) { - switch(ch) { - case 'a': - print_attributes = true; - break; - case 'c': - cmd = CMD_CONFIG; - break; - case 'd': - delay = strtoul(optarg, NULL, 10); - break; - case 'm': - cmd = CMD_MEASURE; - break; - case 'i': - interval = strtoul(optarg, NULL, 10); - break; - case 'f': - print_filters = true; - break; - case 'F': - if (filter) { - fprintf(stderr, "Cannot set multiple filters\n"); - return usage(prog); - } - filter = optarg; - break; - case 's': - simple_mode = true; - break; - case 'p': - /* set port */ - wprobe_port = strtoul(optarg, NULL, 0); - break; - case 'h': - default: - usage(prog); - break; - } - } - - dev = wprobe_get_auto(ifname, &err); - if (!dev || (list_empty(&dev->global_attr) && - list_empty(&dev->link_attr))) { - if (err) - fprintf(stdout, "%s\n", err); - else - fprintf(stderr, "Interface '%s' not found\n", ifname); - return 1; - } - - if (filter || interval >= 0) { - if (filter) - set_filter(dev, filter); - if (interval >= 0) - dev->interval = interval; - - wprobe_apply_config(dev); - } - - if (cmd != CMD_CONFIG) { - if (print_attributes) - show_attributes(dev); - } - if (cmd == CMD_MEASURE) - loop_measurement(dev, print_filters, delay); - - wprobe_free_dev(dev); - - return 0; -} |