diff options
author | Dimitri Staessens <dimitri.staessens@ugent.be> | 2017-10-23 23:25:27 +0200 |
---|---|---|
committer | Sander Vrijders <sander.vrijders@ugent.be> | 2017-10-24 10:50:57 +0200 |
commit | 3608d822b1c8b39f135fc056b864f86286d17530 (patch) | |
tree | 3910379c7d82084187f87c0b11b4c0d56dee332a /src/tools | |
parent | 3d9501007a24bf5422335238960f6777d2684d35 (diff) | |
download | ouroboros-3608d822b1c8b39f135fc056b864f86286d17530.tar.gz ouroboros-3608d822b1c8b39f135fc056b864f86286d17530.zip |
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 <dimitri.staessens@ugent.be>
Signed-off-by: Sander Vrijders <sander.vrijders@ugent.be>
Diffstat (limited to 'src/tools')
-rw-r--r-- | src/tools/echo/echo.c | 76 | ||||
-rw-r--r-- | src/tools/echo/echo_client.c | 57 | ||||
-rw-r--r-- | src/tools/echo/echo_server.c | 67 |
3 files changed, 73 insertions, 127 deletions
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 <ouroboros/dev.h> + #include <stdio.h> #include <string.h> #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; diff --git a/src/tools/echo/echo_client.c b/src/tools/echo/echo_client.c deleted file mode 100644 index 4df5904d..00000000 --- a/src/tools/echo/echo_client.c +++ /dev/null @@ -1,57 +0,0 @@ -/* - * Ouroboros - Copyright (C) 2016 - 2017 - * - * A simple echo application - * - * Dimitri Staessens <dimitri.staessens@ugent.be> - * Sander Vrijders <sander.vrijders@ugent.be> - * - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License version 2 as - * published by the Free Software Foundation. - * - * 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. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., http://www.fsf.org/about/contact/. - */ - -#include <ouroboros/dev.h> -#include <stdlib.h> - -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; -} diff --git a/src/tools/echo/echo_server.c b/src/tools/echo/echo_server.c deleted file mode 100644 index 14e0aa58..00000000 --- a/src/tools/echo/echo_server.c +++ /dev/null @@ -1,67 +0,0 @@ -/* - * Ouroboros - Copyright (C) 2016 - 2017 - * - * A simple echo application - * - * Dimitri Staessens <dimitri.staessens@ugent.be> - * Sander Vrijders <sander.vrijders@ugent.be> - * - * This program is free software; you can redistribute it and/or modify - * it under the terms of the GNU General Public License version 2 as - * published by the Free Software Foundation. - * - * 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. - * - * You should have received a copy of the GNU General Public License - * along with this program; if not, write to the Free Software - * Foundation, Inc., http://www.fsf.org/about/contact/. - */ - -#include <stdbool.h> -#include <signal.h> -#include <unistd.h> -#include <stdlib.h> - -#include <ouroboros/dev.h> - -int server_main(void) -{ - int fd = 0; - char buf[BUF_SIZE]; - ssize_t count = 0; - qosspec_t qs; - - printf("Starting the server.\n"); - - while (true) { - fd = flow_accept(&qs, 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; -} |