From 3608d822b1c8b39f135fc056b864f86286d17530 Mon Sep 17 00:00:00 2001 From: Dimitri Staessens Date: Mon, 23 Oct 2017 23:25:27 +0200 Subject: tools: Refactor echo to single source file This simplifies the echo-app tool implementation to a single source file since it's meant to be a small example application. Signed-off-by: Dimitri Staessens Signed-off-by: Sander Vrijders --- src/tools/echo/echo.c | 76 +++++++++++++++++++++++++++++++++++++++++++++++++-- 1 file changed, 73 insertions(+), 3 deletions(-) (limited to 'src/tools/echo/echo.c') diff --git a/src/tools/echo/echo.c b/src/tools/echo/echo.c index 91ab552d..27dad38b 100644 --- a/src/tools/echo/echo.c +++ b/src/tools/echo/echo.c @@ -22,14 +22,13 @@ #define _POSIX_C_SOURCE 199309L +#include + #include #include #define BUF_SIZE 256 -#include "echo_client.c" -#include "echo_server.c" - static void usage(void) { printf("Usage: echo-app [OPTION]...\n" @@ -38,6 +37,77 @@ static void usage(void) " --help Display this help text and exit\n"); } +int server_main(void) +{ + int fd = 0; + char buf[BUF_SIZE]; + ssize_t count = 0; + + printf("Starting the server.\n"); + + while (true) { + fd = flow_accept(NULL, NULL); + if (fd < 0) { + printf("Failed to accept flow.\n"); + break; + } + + printf("New flow.\n"); + + count = flow_read(fd, &buf, BUF_SIZE); + if (count < 0) { + printf("Failed to read SDU.\n"); + flow_dealloc(fd); + continue; + } + + printf("Message from client is %.*s.\n", (int) count, buf); + + if (flow_write(fd, buf, count) == -1) { + printf("Failed to write SDU.\n"); + flow_dealloc(fd); + continue; + } + + flow_dealloc(fd); + } + + return 0; +} + +int client_main(void) +{ + int fd = 0; + char buf[BUF_SIZE]; + char * message = "Client says hi!"; + ssize_t count = 0; + + fd = flow_alloc("echo", NULL, NULL); + if (fd < 0) { + printf("Failed to allocate flow.\n"); + return -1; + } + + if (flow_write(fd, message, strlen(message) + 1) < 0) { + printf("Failed to write SDU.\n"); + flow_dealloc(fd); + return -1; + } + + count = flow_read(fd, buf, BUF_SIZE); + if (count < 0) { + printf("Failed to read SDU.\n"); + flow_dealloc(fd); + return -1; + } + + printf("Server replied with %.*s\n", (int) count, buf); + + flow_dealloc(fd); + + return 0; +} + int main(int argc, char ** argv) { int ret = -1; -- cgit v1.2.3