From 514a0365feb5f893687aa3f86fef1b956ba7c315 Mon Sep 17 00:00:00 2001
From: Yasuyuki Tanaka <yatch1.tanaka@toshiba.co.jp>
Date: Thu, 14 Sep 2017 21:48:08 +0900
Subject: [PATCH] shell: add 6top command

---
 os/services/shell/shell-commands.c | 34 ++++++++++++++++++++++++++++++
 os/services/shell/shell-commands.h |  7 ++++++
 2 files changed, 41 insertions(+)

diff --git a/os/services/shell/shell-commands.c b/os/services/shell/shell-commands.c
index a48162616..736cc68d1 100644
--- a/os/services/shell/shell-commands.c
+++ b/os/services/shell/shell-commands.c
@@ -72,6 +72,9 @@ static shell_output_func *curr_ping_output_func = NULL;
 static struct process *curr_ping_process;
 static uint8_t curr_ping_ttl;
 static uint16_t curr_ping_datalen;
+#if TSCH_WITH_SIXTOP
+static shell_command_6top_sub_cmd_t sixtop_sub_cmd = NULL;
+#endif /* TSCH_WITH_SIXTOP */
 
 /*---------------------------------------------------------------------------*/
 static const char *
@@ -608,6 +611,34 @@ PT_THREAD(cmd_tsch_schedule(struct pt *pt, shell_output_func output, char *args)
   PT_END(pt);
 }
 /*---------------------------------------------------------------------------*/
+#if TSCH_WITH_SIXTOP
+void
+shell_commands_set_6top_sub_cmd(shell_command_6top_sub_cmd_t sub_cmd)
+{
+  sixtop_sub_cmd = sub_cmd;
+}
+/*---------------------------------------------------------------------------*/
+static
+PT_THREAD(cmd_6top(struct pt *pt, shell_output_func output, char *args))
+{
+  char *next_args;
+
+  PT_BEGIN(pt);
+
+  SHELL_ARGS_INIT(args, next_args);
+
+  if(sixtop_sub_cmd == NULL) {
+    SHELL_OUTPUT(output, "6top command is unavailable:\n");
+  } else {
+    SHELL_OUTPUT(output, "6top: ");
+    sixtop_sub_cmd(output, args);
+  }
+  SHELL_ARGS_NEXT(args, next_args);
+
+  PT_END(pt);
+}
+#endif /* TSCH_WITH_SIXTOP */
+/*---------------------------------------------------------------------------*/
 void
 shell_commands_init(void)
 {
@@ -630,6 +661,9 @@ struct shell_command_t shell_commands[] = {
   { "tsch-schedule",        cmd_tsch_schedule,        "'> tsch-schedule': Shows the current TSCH schedule" },
   { "tsch-status",          cmd_tsch_status,          "'> tsch-status': Shows a summary of the current TSCH state" },
   { "reboot",               cmd_reboot,               "'> reboot': Reboot the board by watchdog_reboot()" },
+#if TSCH_WITH_SIXTOP
+  { "6top",                 cmd_6top,                 "'> 6top help': Shows 6top command usage" },
+#endif /* TSCH_WITH_SIXTOP */
   { NULL, NULL, NULL },
 };
 
diff --git a/os/services/shell/shell-commands.h b/os/services/shell/shell-commands.h
index 4e6aa0f40..f89703273 100644
--- a/os/services/shell/shell-commands.h
+++ b/os/services/shell/shell-commands.h
@@ -61,6 +61,13 @@ extern struct shell_command_t shell_commands[];
  */
 void shell_commands_init(void);
 
+#include "net/mac/tsch/tsch.h"
+#if TSCH_WITH_SIXTOP
+typedef void (*shell_command_6top_sub_cmd_t)(shell_output_func output,
+                                             char *args);
+void shell_commands_set_6top_sub_cmd(shell_command_6top_sub_cmd_t sub_cmd);
+#endif /* TSCH_WITH_SIXTOP */
+
 #endif /* _SHELL_COMMANDS_H_ */
 
 /** @} */