ui/ncurses: Safely handle lost terminal control commands

[petitboot] / test / ui / console-sequence.c

diff --git a/test/ui/console-sequence.c b/test/ui/console-sequence.c
new file mode 100644 (file)
index 0000000..88e0faf
--- /dev/null
+++ b/test/ui/console-sequence.c
@@ -0,0 +1,112 @@
+/*
+ *  Copyright (C) 2017 IBM Corporation
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License as published by
+ *  the Free Software Foundation; version 2 of the License.
+ *
+ *  This program is distributed in the hope that it will be useful,
+ *  but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ *  GNU General Public License for more details.
+ *
+ */
+
+#include <stdlib.h>
+#include <string.h>
+#include <assert.h>
+#include <stdbool.h>
+#include <stdio.h>
+
+#include "talloc/talloc.h"
+
+#define ERR     (-1)
+#define pb_log(...)    printf(__VA_ARGS__)
+#define pb_debug(...)  printf(__VA_ARGS__)
+
+/*
+ * Several example terminal commands, see:
+ * https://vt100.net/docs/vt100-ug/chapter3.html
+ */
+static char identify_rsp[] = {033, 0133, 077, 061, 073, 060, 0143, '\0'};
+static char decdwl[] = {033, 043, 066, '\0'};
+static char attrs[] = {033, 0133, 060, 073, 064, 073, 065, 0155, '\0'};
+static char cursor[] = {033, 070, '\0'};
+static char conf_test[] = {033, 0133, 062, 073, 061, 0171, '\0'};
+static char status[] = {033, 0133, 060, 0156, '\0'};
+static char erase_screen[] = {033, 0133, 062, 0112, '\0'};
+static char status_ok_rsp[] = {033, 0133, 064, 0156, '\0'};
+static char garbage[] = {001, 002, 003, 004, '\0'};
+static char esc_garbage[] = {033, 002, 003, 004, '\0'};
+static char *ptr;
+
+static signed char getch(void)
+{
+       if (!ptr || *ptr == '\0')
+               return -ERR;
+       return *ptr++;
+}
+
+#include "ui/ncurses/console-codes.c"
+
+int main(void)
+{
+       void *ctx;
+       char *seq, *confused;
+
+       ctx = talloc_new(NULL);
+
+       ptr = &identify_rsp[1];
+       printf("Identity response\n");
+       seq = handle_control_sequence(ctx, identify_rsp[0]);
+       assert(strncmp(seq, identify_rsp, strlen(identify_rsp)) == 0);
+
+       printf("DECDWL\n");
+       ptr = &decdwl[1];
+       seq = handle_control_sequence(ctx, decdwl[0]);
+       assert(strncmp(seq, decdwl, strlen(decdwl)) == 0);
+
+       printf("Attributes\n");
+       ptr = &attrs[1];
+       seq = handle_control_sequence(ctx, attrs[0]);
+       assert(strncmp(seq, attrs, strlen(attrs)) == 0);
+
+       printf("Reset Cursor\n");
+       ptr = &cursor[1];
+       seq = handle_control_sequence(ctx, cursor[0]);
+       assert(strncmp(seq, cursor, strlen(cursor)) == 0);
+
+       printf("Confidence Test\n");
+       ptr = &conf_test[1];
+       seq = handle_control_sequence(ctx, conf_test[0]);
+       assert(strncmp(seq, conf_test, strlen(conf_test)) == 0);
+
+       printf("Status\n");
+       ptr = &status[1];
+       seq = handle_control_sequence(ctx, status[0]);
+       assert(strncmp(seq, status, strlen(status)) == 0);
+
+       printf("Erase Screen\n");
+       ptr = &erase_screen[1];
+       seq = handle_control_sequence(ctx, erase_screen[0]);
+       assert(strncmp(seq, erase_screen, strlen(erase_screen)) == 0);
+
+       printf("Status Response\n");
+       ptr = &status_ok_rsp[1];
+       seq = handle_control_sequence(ctx, status_ok_rsp[0]);
+       assert(strncmp(seq, status_ok_rsp, strlen(status_ok_rsp)) == 0);
+
+       printf("Garbage\n");
+       ptr = &garbage[1];
+       seq = handle_control_sequence(ctx, garbage[0]);
+       assert(seq == NULL);
+
+       printf("ESC then Garbage\n");
+       ptr = &esc_garbage[1];
+       confused = talloc_asprintf(ctx, "%c%c...", 033, 002);
+       seq = handle_control_sequence(ctx, esc_garbage[0]);
+       assert(strncmp(seq, confused, strlen(confused)) == 0);
+
+       talloc_free(ctx);
+       return EXIT_SUCCESS;
+}