Add basic file creation, write, then read test and fix bug in tcp.c

src/client.c

@@ -36,8 +36,8 @@ typedef pthread_mutex_t Mutex;
 
 #define PARALLEL_LIMIT 5
 
-#define CLIENT_TRACE(fmt, ...) {}
-//#define CLIENT_TRACE(fmt, ...) fprintf(stderr, "CLIENT: " fmt "\n", ##__VA_ARGS__);
+//#define CLIENT_TRACE(fmt, ...) {}
+#define CLIENT_TRACE(fmt, ...) fprintf(stderr, "CLIENT: " fmt "\n", ##__VA_ARGS__);
 
 typedef struct {
     SHA256   hash;

src/metadata_server.c

@@ -9,8 +9,8 @@
 #include "message.h"
 #include "metadata_server.h"
 
-#define MS_TRACE(fmt, ...) {}
-//#define MS_TRACE(fmt, ...) fprintf(stderr, "MS: " fmt "\n", ##__VA_ARGS__);
+//#define MS_TRACE(fmt, ...) {}
+#define MS_TRACE(fmt, ...) fprintf(stderr, "MS: " fmt "\n", ##__VA_ARGS__);
 
 static void chunk_server_peer_init(ChunkServerPeer *chunk_server, Time current_time)
 {
@@ -1174,6 +1174,8 @@ int metadata_server_tick(void *state_, void **ctxs,
 
             case EVENT_MESSAGE:
             {
+                MS_TRACE("TCP EVENT: message");
+
                 // We don't trace message events from chunk servers
                 // as it would become very verbose
                 for (;;) {

src/tcp.c

@@ -316,9 +316,11 @@ int tcp_translate_events(TCP *tcp, Event *events, void **contexts, struct pollfd
 
         if (polled[i].fd == tcp->wait_fd) {
 
-            char buf[100];
-            recv(tcp->wait_fd, buf, sizeof(buf), 0); // TODO: Make sure all bytes are consumed
-            events[num_events++] = (Event) { EVENT_WAKEUP, -1, -1 };
+            if (polled[i].revents & POLLIN) {
+                char buf[100];
+                recv(tcp->wait_fd, buf, sizeof(buf), 0); // TODO: Make sure all bytes are consumed
+                events[num_events++] = (Event) { EVENT_WAKEUP, -1, -1 };
+            }
 
         } else if (polled[i].fd == tcp->listen_fd) {
 

src/test_client.c

@@ -51,6 +51,8 @@ int test_client_init(void *state_, int argc, char **argv,
     if (client->toasty == NULL)
         return -1;
 
+    client->state = TEST_CLIENT_STATE_0;
+
     printf("Client set up (remote=%s:%d)\n", addr, port);
 
     *timeout = 0;
@@ -75,7 +77,92 @@ int test_client_tick(void *state_, void **ctxs,
     // Process any pending events from the network and get new poll descriptors
     *pnum = toasty_process_events(client->toasty, ctxs, pdata, *pnum);
 
-    assert(0); // TODO
+    // This must be static as write operations will refer to this data
+    // after the function has returned
+    static char msg[] =
+        "Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do "
+        "eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut "
+        "enim ad minim veniam, quis nostrud exercitation ullamco laboris "
+        "nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor "
+        "in reprehenderit in voluptate velit esse cillum dolore eu fugiat "
+        "nulla pariatur. Excepteur sint occaecat cupidatat non proident, "
+        "sunt in culpa qui officia deserunt mollit anim id est laborum.";
+
+    ToastyString file_path = TOASTY_STR("some_file.txt");
+
+    switch (client->state) {
+        int ret;
+        ToastyResult result;
+    case TEST_CLIENT_STATE_0:
+        client->handle = toasty_begin_create_file(client->toasty, file_path, 128);
+        if (client->handle == TOASTY_INVALID) {
+            assert(0); // TODO
+        }
+        printf(">>> Create started <<<<<<<<<<<<\n");
+        client->state = TEST_CLIENT_STATE_1;
+        break;
+    case TEST_CLIENT_STATE_1:
+        ret = toasty_get_result(client->toasty, client->handle, &result);
+        if (ret < 0) {
+            assert(0); // TODO
+        }
+        if (ret == 1) {
+            break;
+        }
+        printf(">>> Create completed <<<<<<<<<<<<\n");
+        assert(ret == 0);
+        if (result.type != TOASTY_RESULT_CREATE_SUCCESS) {
+            assert(0); // TODO
+        }
+        client->handle = toasty_begin_write(client->toasty, file_path, 0, msg, sizeof(msg)-1, TOASTY_VERSION_TAG_EMPTY, 0);
+        if (client->handle == TOASTY_INVALID) {
+            assert(0); // TODO
+        }
+        printf(">>> Write started <<<<<<<<<<<<\n");
+        client->state = TEST_CLIENT_STATE_2;
+        break;
+    case TEST_CLIENT_STATE_2:
+        ret = toasty_get_result(client->toasty, client->handle, &result);
+        if (ret < 0) {
+            assert(0); // TODO
+        }
+        if (ret == 1) {
+            break;
+        }
+        printf(">>> Write completed <<<<<<<<<<<<\n");
+        assert(ret == 0);
+        if (result.type != TOASTY_RESULT_WRITE_SUCCESS) {
+            assert(0); // TODO
+        }
+        client->handle = toasty_begin_read(client->toasty, file_path, 0, client->buf, sizeof(client->buf), TOASTY_VERSION_TAG_EMPTY);
+        if (client->handle == TOASTY_INVALID) {
+            assert(0); // TODO
+        }
+        printf(">>> Read started <<<<<<<<<<<<\n");
+        client->state = TEST_CLIENT_STATE_3;
+        break;
+    case TEST_CLIENT_STATE_3:
+        ret = toasty_get_result(client->toasty, client->handle, &result);
+        if (ret < 0) {
+            assert(0); // TODO
+        }
+        if (ret == 1) {
+            break;
+        }
+        printf(">>> Read completed <<<<<<<<<<<<\n");
+        assert(ret == 0);
+        if (result.type != TOASTY_RESULT_READ_SUCCESS) {
+            assert(0); // TODO
+        }
+        if (result.bytes_read != sizeof(msg)-1) {
+            assert(0); // TODO
+        }
+        if (memcmp(client->buf, msg, sizeof(msg)-1)) {
+            assert(0); // TODO
+        }
+        assert(0); // TODO
+        break;
+    }
 
     *timeout = -1;
     if (pcap < TCP_POLL_CAPACITY)

src/test_client.h

@@ -3,8 +3,18 @@
 
 #include "ToastyFS.h"
 
+typedef enum {
+    TEST_CLIENT_STATE_0,
+    TEST_CLIENT_STATE_1,
+    TEST_CLIENT_STATE_2,
+    TEST_CLIENT_STATE_3,
+} TestClientState;
+
 typedef struct {
-    ToastyFS *toasty;
+    ToastyFS*       toasty;
+    ToastyHandle    handle;
+    TestClientState state;
+    char buf[1<<10];
 } TestClient;
 
 struct pollfd;