summaryrefslogtreecommitdiff
path: root/src
diff options
context:
space:
mode:
Diffstat (limited to 'src')
-rw-r--r--src/include/bcnm/wpactrl.h27
-rw-r--r--src/libwpactrl/wpactrl-internal.h2
-rw-r--r--src/libwpactrl/wpactrl_msg.c8
-rw-r--r--src/libwpactrl/wpactrl_nextmsg.c14
-rw-r--r--src/libwpactrl/wpactrl_scan_parse.c79
-rw-r--r--src/libwpactrl/wpactrl_update.c8
6 files changed, 131 insertions, 7 deletions
diff --git a/src/include/bcnm/wpactrl.h b/src/include/bcnm/wpactrl.h
index af105c3..df5f2c1 100644
--- a/src/include/bcnm/wpactrl.h
+++ b/src/include/bcnm/wpactrl.h
@@ -5,8 +5,10 @@
#include <sys/types.h>
#include <stdint.h>
+#include <skalibs/gccattributes.h>
#include <skalibs/tai.h>
#include <skalibs/stralloc.h>
+#include <skalibs/genalloc.h>
typedef enum wparesponse_e wparesponse_t, *wparesponse_t_ref ;
enum wparesponse_e
@@ -36,10 +38,11 @@ struct wpactrl_s
int fds ;
int fda ;
uint32_t options ;
+ size_t datahead ;
stralloc data ;
stralloc filters ;
} ;
-#define WPACTRL_ZERO { .fds = -1, .fda = -1, .options = 0, .data = STRALLOC_ZERO, .filters = STRALLOC_ZERO }
+#define WPACTRL_ZERO { .fds = -1, .fda = -1, .options = 0, .datahead = 0, .data = STRALLOC_ZERO, .filters = STRALLOC_ZERO }
#define WPACTRL_OPTION_NOFILTER 0x0001U
@@ -63,9 +66,25 @@ extern void wpactrl_filter_remove (wpactrl_t *, char const *) ;
#define wpactrl_filter_deactivate(a) ((a)->options |= WPACTRL_OPTION_NOFILTER)
extern int wpactrl_update (wpactrl_t *) ;
-#define wpactrl_data(a) ((a)->data.s)
-#define wpactrl_datalen(a) ((a)->data.len))
-#define wpactrl_ackdata(a) ((a)->data.len = 0)
+extern char *wpactrl_msg (wpactrl_t *) gccattr_pure ;
+extern void wpactrl_ackmsg (wpactrl_t *) ;
+
+
+ /* High-level functions */
+
+typedef struct wpactrl_scanres_s wpactrl_scanres_t, *wpactrl_scanres_t_ref ;
+struct wpactrl_scanres_s
+{
+ char bssid[6] ;
+ uint16_t frequency ;
+ uint16_t signal_level ;
+ uint32_t flags ;
+ size_t ssid ;
+} ;
+#define WPACTRL_SCANRES_ZERO { .bssid = "\0\0\0\0\0", .frequency = 0, .signal_level = 0, .flags = 0, .ssid = 0 }
+
+extern int wpactrl_scan_parse (char const *, size_t, genalloc * /* wpactrl_scanres_t */, stralloc *) ;
+
/*
diff --git a/src/libwpactrl/wpactrl-internal.h b/src/libwpactrl/wpactrl-internal.h
index b271eae..ddd2597 100644
--- a/src/libwpactrl/wpactrl-internal.h
+++ b/src/libwpactrl/wpactrl-internal.h
@@ -8,7 +8,7 @@
#include <skalibs/tai.h>
#include <bcnm/wpactrl.h>
-#define WPACTRL_PACKET_MAX 8192
+#define WPACTRL_PACKET_MAX 8191
#define WPACTRL_RECV_MAX 32
extern ssize_t wpactrl_fd_recv (int, char *, size_t) ;
diff --git a/src/libwpactrl/wpactrl_msg.c b/src/libwpactrl/wpactrl_msg.c
new file mode 100644
index 0000000..b452c4b
--- /dev/null
+++ b/src/libwpactrl/wpactrl_msg.c
@@ -0,0 +1,8 @@
+/* ISC license. */
+
+#include <bcnm/wpactrl.h>
+
+char *wpactrl_msg (wpactrl_t *a)
+{
+ return a->datahead < a->data.len ? a->data.s + a->datahead : 0 ;
+}
diff --git a/src/libwpactrl/wpactrl_nextmsg.c b/src/libwpactrl/wpactrl_nextmsg.c
new file mode 100644
index 0000000..4954db9
--- /dev/null
+++ b/src/libwpactrl/wpactrl_nextmsg.c
@@ -0,0 +1,14 @@
+/* ISC license. */
+
+#include <string.h>
+#include <bcnm/wpactrl.h>
+
+void wpactrl_nextmsg (wpactrl_t *a)
+{
+ if (a->datahead < a->data.len) a->datahead += strlen(a->data.s + a->datahead) + 1 ;
+ if (a->datahead >= a->data.len)
+ {
+ a->data.len = 0 ;
+ a->datahead = 0 ;
+ }
+}
diff --git a/src/libwpactrl/wpactrl_scan_parse.c b/src/libwpactrl/wpactrl_scan_parse.c
new file mode 100644
index 0000000..24f83c8
--- /dev/null
+++ b/src/libwpactrl/wpactrl_scan_parse.c
@@ -0,0 +1,79 @@
+/* ISC license. */
+
+#include <string.h>
+#include <stdint.h>
+#include <errno.h>
+#include <skalibs/uint16.h>
+#include <skalibs/bytestr.h>
+#include <skalibs/error.h>
+#include <skalibs/fmtscan.h>
+#include <skalibs/stralloc.h>
+#include <skalibs/genalloc.h>
+#include <bcnm/wpactrl.h>
+
+static size_t bssid_scan (char const *s, char *bssid)
+{
+ unsigned int i = 0 ;
+ char sep[6] = ":::::\t" ;
+ for (; i < 6 ; i++)
+ {
+ if (!ucharn_scan(s, bssid + i, 1)) return 0 ;
+ if (s[2] != sep[i]) return 0 ;
+ s += 3 ;
+ }
+ return 18 ;
+}
+
+static size_t flags_scan (char const *s, uint32_t *flags)
+{
+ return 0 ;
+}
+
+static int wpactrl_scan_parse_one (char const *s, size_t len, wpactrl_scanres_t *thing, stralloc *sa)
+{
+ size_t pos = byte_chr(s, len, '\t') ;
+ if (pos >= len) return 0 ;
+ if (pos != 18) return 0 ;
+ if (bssid_scan(s, thing->bssid) != pos) return 0 ;
+ s += pos + 1 ; len -= pos + 1 ;
+ pos = byte_chr(s, len, '\t') ;
+ if (pos >= len) return 0 ;
+ if (uint16_scan(s, &thing->frequency) != pos) return 0 ;
+ s += pos + 1 ; len -= pos + 1 ;
+ pos = byte_chr(s, len, '\t') ;
+ if (pos >= len) return 0 ;
+ if (flags_scan(s, &thing->flags) != pos) return 0 ;
+ s += pos + 1 ; len -= pos + 1 ;
+ thing->ssid = sa->len ;
+ return stralloc_catb(sa, s, len - 1) && stralloc_0(sa) ;
+}
+
+int wpactrl_scan_parse (char const *s, size_t len, genalloc *ga, stralloc *sa)
+{
+ int sawasnull = !sa->s ;
+ int gawasnull = !genalloc_s(wpactrl_scanres_t, ga) ;
+ size_t sabase = sa->len ;
+ size_t gabase = genalloc_len(wpactrl_scanres_t, ga) ;
+ size_t start = byte_chr(s, len, '\n') ;
+ if (start++ >= len) return (errno = EPROTO, 0) ;
+ while (start < len)
+ {
+ size_t pos = byte_chr(s + start, len - start, '\n') ;
+ wpactrl_scanres_t thing ;
+ if (!wpactrl_scan_parse_one(s + start, pos, &thing, sa)) goto err ;
+ if (!genalloc_append(wpactrl_scanres_t, ga, &thing)) goto err ;
+ start += pos + 1 ;
+ }
+ return 1 ;
+
+ err:
+ {
+ int e = errno ;
+ if (gawasnull) genalloc_free(wpactrl_scanres_t, ga) ;
+ else genalloc_setlen(wpactrl_scanres_t, ga, gabase) ;
+ if (sawasnull) stralloc_free(sa) ;
+ else sa->len = sabase ;
+ errno = e ;
+ }
+ return 0 ;
+}
diff --git a/src/libwpactrl/wpactrl_update.c b/src/libwpactrl/wpactrl_update.c
index edbdc6f..0462cbe 100644
--- a/src/libwpactrl/wpactrl_update.c
+++ b/src/libwpactrl/wpactrl_update.c
@@ -1,6 +1,8 @@
/* ISC license. */
#include <string.h>
+#include <errno.h>
+#include <skalibs/error.h>
#include <skalibs/allreadwrite.h>
#include <skalibs/stralloc.h>
#include <bcnm/wpactrl.h>
@@ -24,6 +26,7 @@ static inline int validate (char const *s, size_t len)
if (s[0] != '<') return 0 ;
if (!memchr("123456789", s[1], 9)) return 0 ;
if (s[2] != '>') return 0 ;
+ if (strnlen(s, len) < len) return 0 ;
return s[len-1] == '\n' ;
}
@@ -31,7 +34,7 @@ int wpactrl_update (wpactrl_t *a)
{
unsigned int n = WPACTRL_RECV_MAX ;
unsigned int count = 0 ;
- char buf[WPACTRL_PACKET_MAX] ;
+ char buf[WPACTRL_PACKET_MAX+1] ;
while (n--)
{
ssize_t r = sanitize_read(wpactrl_fd_recv(a->fda, buf, WPACTRL_PACKET_MAX)) ;
@@ -40,7 +43,8 @@ int wpactrl_update (wpactrl_t *a)
if (a->options & WPACTRL_OPTION_NOFILTER
|| (validate(buf, r) && filter_search(buf, r, a->filters.s, a->filters.len)))
{
- if (!stralloc_catb(&a->data, buf, r)) return -1 ;
+ buf[r] = 0 ;
+ if (!stralloc_catb(&a->data, buf, r+1)) return -1 ;
count++ ;
}
}