#include <stdint.h>
#include "all.h"
-#include "queue.h"
+
+// Macros to make the YAJL API a bit easier to use.
+#define y(x, ...) yajl_gen_ ## x (command_output.json_gen, ##__VA_ARGS__)
+#define ystr(str) yajl_gen_string(command_output.json_gen, (unsigned char*)str, strlen(str))
/*******************************************************************************
* The data structures used for parsing. Essentially the current state and a
/* When we arrive here, the stack is full. This should not happen and
* means there’s either a bug in this parser or the specification
* contains a command with more than 10 identified tokens. */
- printf("argh! stack full\n");
+ fprintf(stderr, "BUG: commands_parser stack full. This means either a bug "
+ "in the code, or a new command which contains more than "
+ "10 identified tokens.\n");
exit(1);
}
return NULL;
}
-static void clear_stack() {
+static void clear_stack(void) {
DLOG("clearing stack.\n");
for (int c = 0; c < 10; c++) {
if (stack[c].str != NULL)
#ifndef TEST_PARSER
static Match current_match;
#endif
-static char *json_output;
+static struct CommandResult subcommand_output;
+static struct CommandResult command_output;
#include "GENERATED_call.h"
if (token->next_state == __CALL) {
DLOG("should call stuff, yay. call_id = %d\n",
token->extra.call_identifier);
- json_output = GENERATED_call(token->extra.call_identifier);
+ subcommand_output.json_gen = command_output.json_gen;
+ subcommand_output.needs_tree_render = false;
+ GENERATED_call(token->extra.call_identifier, &subcommand_output);
+ /* If any subcommand requires a tree_render(), we need to make the
+ * whole parser result request a tree_render(). */
+ if (subcommand_output.needs_tree_render)
+ command_output.needs_tree_render = true;
clear_stack();
return;
}
}
/* TODO: Return parsing errors via JSON. */
-char *parse_command(const char *input) {
+struct CommandResult *parse_command(const char *input) {
DLOG("new parser handling: %s\n", input);
state = INITIAL;
- json_output = NULL;
+
+/* A YAJL JSON generator used for formatting replies. */
+#if YAJL_MAJOR >= 2
+ command_output.json_gen = yajl_gen_alloc(NULL);
+#else
+ command_output.json_gen = yajl_gen_alloc(NULL, NULL);
+#endif
+
+ y(array_open);
+ command_output.needs_tree_render = false;
const char *walk = input;
const size_t len = strlen(input);
// TODO: make this testable
#ifndef TEST_PARSER
- cmd_criteria_init(¤t_match);
+ cmd_criteria_init(¤t_match, &subcommand_output);
#endif
/* The "<=" operator is intentional: We also handle the terminating 0-byte
if (strncasecmp(walk, token->name + 1, strlen(token->name) - 1) == 0) {
DLOG("found literal, moving to next state\n");
if (token->identifier != NULL)
- push_string(token->identifier, strdup(token->name + 1));
+ push_string(token->identifier, sstrdup(token->name + 1));
walk += strlen(token->name) - 1;
next_state(token);
token_handled = true;
if (*walk == '"') {
beginning++;
walk++;
- while (*walk != '"' || *(walk-1) == '\\')
+ while (*walk != '\0' && (*walk != '"' || *(walk-1) == '\\'))
walk++;
} else {
if (token->name[0] == 's') {
}
}
if (walk != beginning) {
- char *str = calloc(walk-beginning + 1, 1);
- strncpy(str, beginning, walk-beginning);
+ char *str = scalloc(walk-beginning + 1);
+ /* We copy manually to handle escaping of characters. */
+ int inpos, outpos;
+ for (inpos = 0, outpos = 0;
+ inpos < (walk-beginning);
+ inpos++, outpos++) {
+ /* We only handle escaped double quotes to not break
+ * backwards compatibility with people using \w in
+ * regular expressions etc. */
+ if (beginning[inpos] == '\\' && beginning[inpos+1] == '"')
+ inpos++;
+ str[outpos] = beginning[inpos];
+ }
if (token->identifier)
push_string(token->identifier, str);
DLOG("str is \"%s\"\n", str);
// TODO: make this testable
#ifndef TEST_PARSER
if (*walk == '\0' || *walk == ';')
- cmd_criteria_init(¤t_match);
+ cmd_criteria_init(¤t_match, &subcommand_output);
#endif
walk++;
break;
* full input, and underline the position where the parser
* currently is. */
char *errormessage;
- char *possible_tokens = malloc(tokenlen + 1);
+ char *possible_tokens = smalloc(tokenlen + 1);
char *tokenwalk = possible_tokens;
for (c = 0; c < ptr->n; c++) {
token = &(ptr->array[c]);
}
}
*tokenwalk = '\0';
- asprintf(&errormessage, "Expected one of these tokens: %s",
- possible_tokens);
+ sasprintf(&errormessage, "Expected one of these tokens: %s",
+ possible_tokens);
free(possible_tokens);
/* Contains the same amount of characters as 'input' has, but with
* the unparseable part highlighted using ^ characters. */
- char *position = malloc(len + 1);
+ char *position = smalloc(len + 1);
for (const char *copywalk = input; *copywalk != '\0'; copywalk++)
position[(copywalk - input)] = (copywalk >= walk ? '^' : ' ');
position[len] = '\0';
printf("Your command: %s\n", input);
printf(" %s\n", position);
+ /* Format this error message as a JSON reply. */
+ y(map_open);
+ ystr("success");
+ y(bool, false);
+ ystr("error");
+ ystr(errormessage);
+ ystr("input");
+ ystr(input);
+ ystr("errorposition");
+ ystr(position);
+ y(map_close);
+
free(position);
free(errormessage);
+ clear_stack();
break;
}
}
- DLOG("json_output = %s\n", json_output);
- return json_output;
+ y(array_close);
+
+ DLOG("command_output.needs_tree_render = %d\n", command_output.needs_tree_render);
+ return &command_output;
}
/*******************************************************************************