add mac80211 plugin

2 files changed, 210 insertions, 1 deletions

diff --git a/Makefile b/Makefile
index 7a6dd0a..c4b13b9 100644
--- a/Makefile
+++ b/Makefile
@@ -9,7 +9,7 @@ LIBS = -L. -lparsevent -ldl
 %.o: %.c
         $(CC) -c $(CFLAGS) $(EXT) $(INCLUDES) $< -o $@
 
-TARGETS = libparsevent.a trace-cmd plugin_hrtimer.so
+TARGETS = libparsevent.a trace-cmd plugin_hrtimer.so plugin_mac80211.so
 
 all: $(TARGETS)
 
@@ -41,6 +41,12 @@ plugin_hrtimer.o: plugin_hrtimer.c parse-events.h
 plugin_hrtimer.so: plugin_hrtimer.o
         $(CC) -shared -nostartfiles -o $@ $<
 
+plugin_mac80211.o: plugin_mac80211.c parse-events.h
+        $(CC) -c $(CFLAGS) -fPIC -o $@ $<
+
+plugin_mac80211.so: plugin_mac80211.o
+        $(CC) -shared -nostartfiles -o $@ $<
+
 .PHONY: force
 force:
 
diff --git a/plugin_mac80211.c b/plugin_mac80211.c
new file mode 100644
index 0000000..791bd63
--- /dev/null
+++ b/plugin_mac80211.c
@@ -0,0 +1,203 @@
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+
+#include "parse-events.h"
+
+#define INDENT 65
+
+/* return -1 (field not found/not valid number), 0 (ok), 1 (buffer full) */
+static int _print_field(struct trace_seq *s, const char *fmt,
+                        struct event *event, const char *name, const void *data)
+{
+        struct format_field *f = pevent_find_field(event, name);
+        unsigned long long val;
+
+        if (!f)
+                return -1;
+
+        if (pevent_read_number_field(f, data, &val))
+                return -1;
+
+        return trace_seq_printf(s, fmt, val);
+}
+
+/* return 0 (ok), 1 (buffer full) */
+static void print_field(struct trace_seq *s, const char *fmt,
+                        struct event *event, const char *name, const void *data)
+{
+        int ret = _print_field(s, fmt, event, name, data);
+
+        if (ret == -1)
+                trace_seq_printf(s, "NOTFOUND:%s", name);
+}
+
+static void print_string(struct trace_seq *s, struct event *event,
+                         const char *name, const void *data)
+{
+        struct format_field *f = pevent_find_field(event, name);
+        int offset;
+        int length;
+
+        if (!f) {
+                trace_seq_printf(s, "NOTFOUND:%s", name);
+                return;
+        }
+
+        offset = f->offset;
+        length = f->size;
+
+        if (!strncmp(f->type, "__data_loc", 10)) {
+                unsigned long long v;
+                if (pevent_read_number_field(f, data, &v)) {
+                        trace_seq_printf(s, "invalid_data_loc");
+                        return;
+                }
+                offset = v & 0xffff;
+                length = v >> 16;
+        }
+
+        trace_seq_printf(s, "%.*s", length, (char *)data + offset);
+}
+
+struct value_name {
+        unsigned long long value;
+        const char *name;
+};
+
+static void _print_enum(struct trace_seq *s, struct event *event,
+                        const char *name, const void *data,
+                        const struct value_name *names, int n_names)
+{
+        struct format_field *f = pevent_find_field(event, name);
+        unsigned long long val;
+        int i;
+
+        if (!f) {
+                trace_seq_puts(s, "field-not-found");
+                return;
+        }
+
+        if (pevent_read_number_field(f, data, &val)) {
+                trace_seq_puts(s, "field-invalid");
+                return;
+        }
+
+        for (i = 0; i < n_names; i++) {
+                if (names[i].value == val) {
+                        trace_seq_puts(s, names[i].name);
+                        return;
+                }
+        }
+
+        trace_seq_printf(s, "%llu", val);
+}
+
+#define print_enum(s, ev, name, data, enums...)                                 \
+        ({ static const struct value_name __n[] = { enums };                    \
+        _print_enum(s, ev, name, data, __n, sizeof(__n)/sizeof(__n[0]));        \
+        })
+
+static void _print_flag(struct trace_seq *s, struct event *event,
+                        const char *name, const void *data,
+                        const struct value_name *names, int n_names)
+{
+        struct format_field *f = pevent_find_field(event, name);
+        unsigned long long val;
+        int i, j, found, first = 1;
+
+        if (!f) {
+                trace_seq_puts(s, "field-not-found");
+                return;
+        }
+
+        if (pevent_read_number_field(f, data, &val)) {
+                trace_seq_puts(s, "field-invalid");
+                return;
+        }
+
+        for (i = 0; i < 64; i++) {
+                if (!(val & (1ULL<<i)))
+                        continue;
+                if (!first)
+                        trace_seq_putc(s, '|');
+                first = 0;
+
+                found = 0;
+                for (j = 0; j < n_names; j++) {
+                        if (j == names[i].value) {
+                                trace_seq_puts(s, names[i].name);
+                                found = 1;
+                                break;
+                        }
+                }
+                if (!found)
+                        trace_seq_printf(s, "b%d", i);
+        }
+}
+
+#define print_flag(s, ev, name, data, enums...)                                 \
+        ({ static const struct value_name __n[] = { enums };                    \
+        _print_flag(s, ev, name, data, __n, sizeof(__n)/sizeof(__n[0]));        \
+        })
+
+#define SF(fn)  _print_field(s, fn ":%d", event, fn, data)
+#define SFX(fn) _print_field(s, fn ":%#x", event, fn, data)
+#define SP()    trace_seq_putc(s, ' ')
+
+static int drv_bss_info_changed(struct trace_seq *s, void *data, int size,
+                                struct event *event)
+{
+        print_string(s, event, "wiphy_name", data);
+        trace_seq_printf(s, " vif:");
+        print_string(s, event, "vif_name", data);
+        print_field(s, "(%d)", event, "vif_type", data);
+
+        trace_seq_printf(s, "\n%*s", INDENT, "");
+        SF("assoc"); SP();
+        SF("aid"); SP();
+        SF("cts"); SP();
+        SF("shortpre"); SP();
+        SF("shortslot"); SP();
+        SF("dtimper"); SP();
+        trace_seq_printf(s, "\n%*s", INDENT, "");
+        SF("bcnint"); SP();
+        SFX("assoc_cap"); SP();
+        SFX("basic_rates"); SP();
+        SF("enable_beacon");
+        trace_seq_printf(s, "\n%*s", INDENT, "");
+        SF("ht_operation_mode");
+
+        return 0;
+}
+
+static int drv_config(struct trace_seq *s, void *data,
+                      int size, struct event *event)
+{
+        print_string(s, event, "wiphy_name", data);
+        trace_seq_putc(s, ' ');
+        print_flag(s, event, "flags", data,
+                { 0, "MONITOR" },
+                { 1, "PS" },
+                { 2, "IDLE" });
+        print_field(s, " chan:%d/", event, "center_freq", data);
+        print_enum(s, event, "channel_type", data,
+                { 0, "noht" },
+                { 1, "ht20" },
+                { 2, "ht40-" },
+                { 3, "ht40+" });
+        trace_seq_putc(s, ' ');
+        SF("power_level");
+
+        return 0;
+}
+
+int PEVENT_PLUGIN_LOADER(void)
+{
+        pevent_register_event_handler(-1, "mac80211", "drv_bss_info_changed",
+                                      drv_bss_info_changed);
+        pevent_register_event_handler(-1, "mac80211", "drv_config",
+                                      drv_config);
+
+        return 0;
+}