1 files changed, 141 insertions, 0 deletions
diff --git a/src/test/test2.c b/src/test/test2.c
new file mode 100644
index 0000000..c25198b
--- /dev/null
+++ b/src/test/test2.c
@@ -0,0 +1,141 @@
+/*
+ * Author Jerry Lundström <jerry@dns-oarc.net>
+ * Copyright (c) 2020, OARC, Inc.
+ * All rights reserved.
+ *
+ * This file is part of the tinyframe library.
+ *
+ * tinyframe library is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Lesser General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * tinyframe library 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 Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public License
+ * along with tinyframe library.  If not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include <tinyframe/tinyframe.h>
+
+#include <stdio.h>
+#include <string.h>
+#include <stdlib.h>
+
+static char content_type[] = "tinyframe.test";
+
+int main(int argc, const char* argv[])
+{
+    if (argc < 3) {
+        return 1;
+    }
+
+    if (argv[1][0] == 'w') {
+        FILE* fp = fopen(argv[2], "w");
+        if (!fp) {
+            return 1;
+        }
+
+        struct tinyframe_writer writer = TINYFRAME_WRITER_INITIALIZER;
+
+        uint8_t out[sizeof(content_type) + (TINYFRAME_CONTROL_FRAME_LENGTH_MAX * 3)];
+        size_t  len   = sizeof(out);
+        size_t  wrote = 0;
+
+        if (tinyframe_write_control_start(&writer, &out[wrote], len, content_type, sizeof(content_type) - 1) != tinyframe_ok) {
+            return 1;
+        }
+        wrote += writer.bytes_wrote;
+        len -= writer.bytes_wrote;
+
+        if (tinyframe_write_frame(&writer, &out[wrote], len, (uint8_t*)content_type, sizeof(content_type)) != tinyframe_ok) {
+            return 1;
+        }
+        wrote += writer.bytes_wrote;
+        len -= writer.bytes_wrote;
+
+        if (tinyframe_write_control_stop(&writer, &out[wrote], len) != tinyframe_ok) {
+            return 1;
+        }
+        wrote += writer.bytes_wrote;
+        len -= writer.bytes_wrote;
+
+        if (fwrite(out, 1, wrote, fp) != wrote) {
+            return 1;
+        }
+
+        fclose(fp);
+        return 0;
+    } else if (argv[1][0] == 'r') {
+        FILE* fp = fopen(argv[2], "r");
+        if (!fp) {
+            return 1;
+        }
+
+        long content_length;
+
+        if (fseek(fp, 0, SEEK_END)) {
+            return 1;
+        }
+        content_length = ftell(fp);
+        if (fseek(fp, 0, SEEK_SET)) {
+            return 1;
+        }
+
+        uint8_t* content = malloc(content_length);
+        if (!content) {
+            return 1;
+        }
+
+        if (fread(content, 1, content_length, fp) != content_length) {
+            return 1;
+        }
+
+        fclose(fp);
+
+        struct tinyframe_reader reader   = TINYFRAME_READER_INITIALIZER;
+        uint8_t*                decoding = content;
+        size_t                  left     = content_length;
+
+        if (tinyframe_read(&reader, decoding, left) != tinyframe_have_control
+            || reader.control.type != TINYFRAME_CONTROL_START) {
+            return 1;
+        }
+        decoding += reader.bytes_read;
+        left -= reader.bytes_read;
+
+        if (tinyframe_read(&reader, decoding, left) != tinyframe_have_control_field
+            || reader.control_field.type != TINYFRAME_CONTROL_FIELD_CONTENT_TYPE
+            || strncmp(content_type, (const char*)reader.control_field.data, reader.control_field.length)) {
+            return 1;
+        }
+        decoding += reader.bytes_read;
+        left -= reader.bytes_read;
+
+        while (1) {
+            switch (tinyframe_read(&reader, decoding, left)) {
+            case tinyframe_have_frame: {
+                if (strncmp(content_type, (char*)reader.frame.data, reader.frame.length)) {
+                    return 1;
+                }
+                break;
+            }
+
+            case tinyframe_stopped:
+            case tinyframe_finished:
+                return 0;
+
+            default:
+                return 1;
+            }
+
+            decoding += reader.bytes_read;
+            left -= reader.bytes_read;
+        }
+        return 0;
+    }
+    return 1;
+}