From 45082064cbc498a1f4287f411512a8183f0065d6 Mon Sep 17 00:00:00 2001
From: David Lamparter <equinox@opensourcerouting.org>
Date: Tue, 11 Jul 2017 16:07:41 +0200
Subject: [PATCH] lib: grammar_sandbox: don't pre-setup nodegraph

This shows up as still reachable allocation on exit, which is somewhat
misleading.  Note this only affects --enable-dev-build.

Signed-off-by: David Lamparter <equinox@opensourcerouting.org>
---
 lib/grammar_sandbox.c | 30 +++++++++++++++++++-----------
 1 file changed, 19 insertions(+), 11 deletions(-)

diff --git a/lib/grammar_sandbox.c b/lib/grammar_sandbox.c
index 9bb672dc53..454e076f06 100644
--- a/lib/grammar_sandbox.c
+++ b/lib/grammar_sandbox.c
@@ -50,6 +50,12 @@ init_cmdgraph (struct vty *, struct graph **);
 /** shim interface commands **/
 struct graph *nodegraph = NULL, *nodegraph_free = NULL;
 
+#define check_nodegraph() \
+  do { if (!nodegraph) { \
+      vty_outln(vty, "nodegraph not initialized"); \
+      return CMD_WARNING; \
+    } } while (0)
+
 DEFUN (grammar_test,
        grammar_test_cmd,
        "grammar parse LINE...",
@@ -57,6 +63,8 @@ DEFUN (grammar_test,
        "parse a command\n"
        "command to pass to new parser\n")
 {
+  check_nodegraph();
+
   int idx_command = 2;
   // make a string from tokenized command line
   char *command = argv_concat (argv, argc, idx_command);
@@ -85,6 +93,8 @@ DEFUN (grammar_test_complete,
        "attempt to complete input on DFA\n"
        "command to complete\n")
 {
+  check_nodegraph();
+
   int idx_command = 2;
   char *cmdstr = argv_concat (argv, argc, idx_command);
   if (!cmdstr)
@@ -143,6 +153,8 @@ DEFUN (grammar_test_match,
        "attempt to match input on DFA\n"
        "command to match\n")
 {
+  check_nodegraph();
+
   int idx_command = 2;
   if (argv[2]->arg[0] == '#')
     return CMD_SUCCESS;
@@ -209,6 +221,8 @@ DEFUN (grammar_test_doc,
        "Test function for docstring\n"
        "Command end\n")
 {
+  check_nodegraph();
+
   // create cmd_element with docstring
   struct cmd_element *cmd = XCALLOC (MTYPE_CMD_TOKENS, sizeof (struct cmd_element));
   cmd->string = XSTRDUP (MTYPE_CMD_TOKENS, "test docstring <example|selector follow> (1-255) end VARIABLE [OPTION|set lol] . VARARG");
@@ -243,12 +257,10 @@ DEFUN (grammar_test_show,
        "print current accumulated DFA\n"
        "include docstrings\n")
 {
-  struct graph_node *stack[MAXDEPTH];
+  check_nodegraph();
 
-  if (!nodegraph)
-    vty_out(vty, "nodegraph uninitialized\r\n");
-  else
-    pretty_print_graph (vty, vector_slot (nodegraph->nodes, 0), 0, argc >= 3, stack, 0);
+  struct graph_node *stack[MAXDEPTH];
+  pretty_print_graph (vty, vector_slot (nodegraph->nodes, 0), 0, argc >= 3, stack, 0);
   return CMD_SUCCESS;
 }
 
@@ -259,14 +271,12 @@ DEFUN (grammar_test_dot,
        "print current graph for dot\n"
        ".dot filename\n")
 {
+  check_nodegraph();
+
   struct graph_node *stack[MAXDEPTH];
   struct graph_node *visited[MAXDEPTH*MAXDEPTH];
   size_t vpos = 0;
 
-  if (!nodegraph) {
-    vty_out(vty, "nodegraph uninitialized\r\n");
-    return CMD_SUCCESS;
-  }
   FILE *ofd = fopen(argv[2]->arg, "w");
   if (!ofd) {
     vty_out(vty, "%s: %s\r\n", argv[2]->arg, strerror(errno));
@@ -476,8 +486,6 @@ DEFUN (grammar_access,
 
 /* this is called in vtysh.c to set up the testing shim */
 void grammar_sandbox_init(void) {
-  init_cmdgraph (NULL, &nodegraph);
-
   // install all enable elements
   install_element (ENABLE_NODE, &grammar_test_cmd);
   install_element (ENABLE_NODE, &grammar_test_show_cmd);
-- 
2.39.5