1 files changed, 105 insertions, 0 deletions
diff --git a/epan/test_epan.c b/epan/test_epan.c
new file mode 100644
index 0000000000..452c53211b
--- /dev/null
+++ b/epan/test_epan.c
@@ -0,0 +1,105 @@
+/*
+ * Wireshark - Network traffic analyzer
+ * By Gerald Combs <gerald@wireshark.org>
+ * Copyright 1998 Gerald Combs
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+#include "config.h"
+
+#include "strutil.h"
+#include <wsutil/utf8_entities.h>
+
+/*
+ * FIXME: LABEL_LENGTH includes the nul byte terminator.
+ * This is confusing but matches ITEM_LABEL_LENGTH.
+ */
+#define LABEL_LENGTH 8
+
+void test_label_strcat(void)
+{
+    char label[LABEL_LENGTH];
+    const char *src;
+    size_t pos;
+
+    src = "ABCD";
+    pos = 0;
+    pos = ws_label_strcpy(label, sizeof(label), pos, src, 0);
+    g_assert_cmpstr(label, ==, "ABCD");
+    g_assert_cmpuint(pos, ==, 4);
+
+    src = "EFGH";
+    pos = ws_label_strcpy(label, sizeof(label), pos, src, 0);
+    g_assert_cmpstr(label, ==, "ABCDEFG");
+    g_assert_cmpuint(pos, ==, 8);
+
+    src = "IJKL";
+    pos = 7;
+    pos = ws_label_strcpy(label, sizeof(label), pos, src, 0);
+    g_assert_cmpstr(label, ==, "ABCDEFG");
+    g_assert_cmpuint(pos, ==, 11);
+
+    /* UTF-8 multibyte does not fit, do not truncate. */
+    src = "ABCDEF"UTF8_MIDDLE_DOT;
+    pos = 0;
+    pos = ws_label_strcpy(label, sizeof(label), pos, src, 0);
+    g_assert_cmpstr(label, ==, "ABCDEF");
+    g_assert_cmpuint(pos, ==, 8); /* Tried to write 8 bytes. */
+}
+
+void test_label_strcat_escape_whitespace(void)
+{
+    char label[128];
+    const char *src, *dst;
+    size_t pos;
+
+    src = "ABCD\n\t\f\r\aE"UTF8_MIDDLE_DOT"Z";
+    dst = "ABCD\\n\\t\\f\\r\\aE"UTF8_MIDDLE_DOT"Z";
+    pos = ws_label_strcpy(label, sizeof(label), 0, src, 0);
+    g_assert_cmpstr(label, ==, dst);
+    g_assert_cmpuint(pos, ==, strlen(dst));
+}
+
+void test_label_escape_control(void)
+{
+    char label[128];
+    const char *src, *dst;
+    size_t pos;
+
+    src = "ABCD \x04\x17\xC2\x80 EFG \xC2\x90 HIJ \xC2\x9F Z";
+    dst = "ABCD \\x04\\x17\\u0080 EFG \\u0090 HIJ \\u009F Z";
+    pos = ws_label_strcpy(label, sizeof(label), 0, src, 0);
+    g_assert_cmpstr(label, ==, dst);
+    g_assert_cmpuint(pos, ==, strlen(dst));
+}
+
+int main(int argc, char **argv)
+{
+    int ret;
+
+    ws_log_init("test_proto", NULL);
+
+    g_test_init(&argc, &argv, NULL);
+
+    g_test_add_func("/label/strcat", test_label_strcat);
+    g_test_add_func("/label/escape_whitespace", test_label_strcat_escape_whitespace);
+    g_test_add_func("/label/escape_control", test_label_escape_control);
+
+    ret = g_test_run();
+
+    return ret;
+}
+
+/*
+ * Editor modelines  -  https://www.wireshark.org/tools/modelines.html
+ *
+ * Local variables:
+ * c-basic-offset: 4
+ * tab-width: 8
+ * indent-tabs-mode: nil
+ * End:
+ *
+ * vi: set shiftwidth=4 tabstop=8 expandtab:
+ * :indentSize=4:tabSize=8:noTabs=true:
+ */