From 6aa2a55aaa84d92eefa170602c5f6028b4a10d0f Mon Sep 17 00:00:00 2001
From: Vincent Dupont <vincent@otakeys.com>
Date: Fri, 22 Jun 2018 15:20:45 -0700
Subject: [PATCH] tests/driver_at: add urc test commands
---
tests/driver_at/Makefile | 1 +
tests/driver_at/main.c | 90 ++++++++++++++++++++++++++++++++++++++++
2 files changed, 91 insertions(+)
diff --git a/tests/driver_at/Makefile b/tests/driver_at/Makefile
index 43ed6a582c..f68ba06428 100644
--- a/tests/driver_at/Makefile
+++ b/tests/driver_at/Makefile
@@ -4,5 +4,6 @@ BOARD_INSUFFICIENT_MEMORY += nucleo-f031k6
USEMODULE += shell
USEMODULE += at
+USEMODULE += at_urc
include $(RIOTBASE)/Makefile.include
diff --git a/tests/driver_at/main.c b/tests/driver_at/main.c
index 026d966d1b..7353bb5bec 100644
--- a/tests/driver_at/main.c
+++ b/tests/driver_at/main.c
@@ -22,6 +22,7 @@
#include <stdio.h>
#include <stdlib.h>
+#include <string.h>
#include "at.h"
#include "shell.h"
@@ -114,12 +115,101 @@ static int drain(int argc, char **argv)
return 0;
}
+#ifdef MODULE_AT_URC
+#ifndef MAX_URC_NB
+#define MAX_URC_NB 5
+#endif
+
+#ifndef MAX_URC_LEN
+#define MAX_URC_LEN 32
+#endif
+
+static at_urc_t urc_list[MAX_URC_NB];
+static char urc_str[MAX_URC_NB][MAX_URC_LEN];
+static bool urc_used[MAX_URC_NB];
+
+static void _urc_cb(void *arg, const char *urc)
+{
+ (void)arg;
+ printf("urc received: %s\n", urc);
+}
+
+static int add_urc(int argc, char **argv)
+{
+ if (argc < 2) {
+ printf("Usage: %s <urc>\n", argv[0]);
+ return 1;
+ }
+
+ if (strlen(argv[1]) > MAX_URC_LEN - 1) {
+ puts("urc is too long");
+ return 1;
+ }
+
+ for (size_t i = 0; i < MAX_URC_NB; i++) {
+ if (!urc_used[i]) {
+ strcpy(urc_str[i], argv[1]);
+ urc_list[i].code = urc_str[i];
+ urc_list[i].arg = NULL;
+ urc_list[i].cb = _urc_cb;
+ urc_used[i] = true;
+ at_add_urc(&at_dev, &urc_list[i]);
+ puts("urc registered");
+ return 0;
+ }
+ }
+
+ puts("Not enough memory, urc is not registered");
+ return 1;
+}
+
+static int process_urc(int argc, char **argv)
+{
+ if (argc < 2) {
+ printf("Usage: %s <timeout>\n", argv[0]);
+ return 1;
+ }
+
+ uint32_t timeout = strtoul(argv[1], NULL, 0);
+ at_process_urc(&at_dev, timeout);
+
+ puts("urc processed");
+
+ return 0;
+}
+
+static int remove_urc(int argc, char **argv)
+{
+ if (argc < 2) {
+ printf("Usage: %s <urc>\n", argv[0]);
+ return 1;
+ }
+
+ for (size_t i = 0; i < MAX_URC_NB; i++) {
+ if (urc_used[i] && strcmp(urc_list[i].code, argv[1]) == 0) {
+ at_remove_urc(&at_dev, &urc_list[i]);
+ urc_used[i] = false;
+ puts("urc removed");
+ return 0;
+ }
+ }
+
+ puts("urc not found");
+ return 1;
+}
+#endif
+
static const shell_command_t shell_commands[] = {
{ "init", "Initialize AT device", init },
{ "send", "Send a command and wait response", send },
{ "send_ok", "Send a command and wait OK", send_ok },
{ "send_lines", "Send a command and wait lines", send_lines },
{ "drain", "Drain AT device", drain },
+#ifdef MODULE_AT_URC
+ { "add_urc", "Register an URC", add_urc },
+ { "remove_urc", "De-register an URC", remove_urc },
+ { "process_urc", "Process the URCs", process_urc },
+#endif
{ NULL, NULL, NULL },
};
--
GitLab