From 4a3e48913b351cdc3144d61aa17233b11ea44fe4 Mon Sep 17 00:00:00 2001 From: Joel Sherrill Date: Thu, 26 Jul 2012 08:17:53 -0500 Subject: loopback01: New test from samples/loopback --- testsuite/loopback01/Makefile | 34 +++++ testsuite/loopback01/test_main.c | 276 +++++++++++++++++++++++++++++++++++++++ 2 files changed, 310 insertions(+) create mode 100644 testsuite/loopback01/Makefile create mode 100644 testsuite/loopback01/test_main.c diff --git a/testsuite/loopback01/Makefile b/testsuite/loopback01/Makefile new file mode 100644 index 00000000..7998aba6 --- /dev/null +++ b/testsuite/loopback01/Makefile @@ -0,0 +1,34 @@ +# +# $Id$ +# + +include ../../config.inc + +PGM=${ARCH}/loopback01.exe + +# optional managers required +MANAGERS=all + +# C source names +C_FILES = init.c test_main.c +C_O_FILES = $(C_FILES:%.c=${ARCH}/%.o) + +AM_CPPFLAGS += -I $(INSTALL_BASE)/include +AM_CPPFLAGS += -I ../init01 +LINK_LIBS += $(INSTALL_BASE)/libbsdc.a +LINK_LIBS += $(INSTALL_BASE)/libbsd.a ${REL_ARGS} + +include $(RTEMS_MAKEFILE_PATH)/Makefile.inc +include $(RTEMS_CUSTOM) +include $(PROJECT_ROOT)/make/leaf.cfg + +OBJS= $(C_O_FILES) +CLEAN_ADDITIONS += init.c + +all: init.c ${ARCH} $(PGM) + +init.c: ../init01/init.c + cp ../init01/init.c . + +$(PGM): $(OBJS) + -$(make-exe) diff --git a/testsuite/loopback01/test_main.c b/testsuite/loopback01/test_main.c new file mode 100644 index 00000000..f8b5b3b0 --- /dev/null +++ b/testsuite/loopback01/test_main.c @@ -0,0 +1,276 @@ +/* + * This is the body of the test. It does not do much except ensure + * that the target is alive after initializing the TCP/IP stack. + */ + +#include +#include + +#include +#include +#include +#include + +void print_test_name(void) +{ + printf( "\n\n*** LIBFREEBSD LOOPBACK SOCKET01 TEST ***\n" ); +} + + +#define CONFIGURE_INIT_TASK_STACK_SIZE (10*1024) +#define CONFIGURE_INIT_TASK_PRIORITY 50 +#define CONFIGURE_INIT_TASK_INITIAL_MODES (RTEMS_PREEMPT | \ + RTEMS_NO_TIMESLICE | \ + RTEMS_NO_ASR | \ + RTEMS_INTERRUPT_LEVEL(0)) + +#include +#include +#include +#include +#include +#include +#include +#include + +/* for old configuration structure */ +#include + +/* + * Network configuration + */ +struct rtems_bsdnet_config rtems_bsdnet_config = { + NULL, /* Network interface */ + NULL, /* Use fixed network configuration */ + 0, /* Default network task priority */ + 0, /* Default mbuf capacity */ + 0, /* Default mbuf cluster capacity */ + "testSystem", /* Host name */ + "nowhere.com", /* Domain name */ + "127.0.0.1", /* Gateway */ + "127.0.0.1", /* Log host */ + {"127.0.0.1" }, /* Name server(s) */ + {"127.0.0.1" }, /* NTP server(s) */ + 0, + 0, + 0, + 0, + 0 +}; + +/* + * Thread-safe output routines + */ +static rtems_id printMutex; +static void printSafe(const char *fmt, ...) +{ + va_list args; + va_start(args, fmt); + rtems_semaphore_obtain(printMutex, RTEMS_WAIT, RTEMS_NO_TIMEOUT); + vprintf(fmt, args); + rtems_semaphore_release(printMutex); + va_end(args); +} +#define printf printSafe + +/* + * Spawn a task + */ +static void spawnTask(rtems_task_entry entryPoint, rtems_task_priority priority, rtems_task_argument arg) +{ + rtems_status_code sc; + rtems_id tid; + + sc = rtems_task_create(rtems_build_name('t','a','s','k'), + priority, + RTEMS_MINIMUM_STACK_SIZE+(8*1024), + RTEMS_PREEMPT|RTEMS_TIMESLICE|RTEMS_NO_ASR|RTEMS_INTERRUPT_LEVEL(0), + RTEMS_FLOATING_POINT|RTEMS_LOCAL, + &tid); + if (sc != RTEMS_SUCCESSFUL) + rtems_panic("Can't create task: %s", rtems_status_text(sc)); + sc = rtems_task_start(tid, entryPoint, arg); + if (sc != RTEMS_SUCCESSFUL) + rtems_panic("Can't start task: %s", rtems_status_text(sc)); +} + +/* + * Server subtask + */ +static rtems_task workerTask(rtems_task_argument arg) +{ + int s = arg; + char msg[80]; + char reply[100]; + int i; + + for (;;) { + if ((i = read(s, msg, sizeof msg)) < 0) { + printf("Server couldn't read message from client: %s\n", strerror(errno)); + break; + } + if (i == 0) + break; + rtems_task_wake_after(20); /* Simulate some processing delay */ + i = sprintf(reply, "Server received %d (%s)", i, msg); + if ((i = write(s, reply, i+1)) < 0) { + printf("Server couldn't write message to client: %s\n", strerror(errno)); + break; + } + } + if (close(s) < 0) + printf("Can't close worker task socket: %s\n", strerror(errno)); + printf("Worker task terminating.\n"); + rtems_task_delete(RTEMS_SELF); +} + +/* + * Server Task + */ +static rtems_task serverTask(rtems_task_argument arg) +{ + int s, s1; + socklen_t addrlen; + struct sockaddr_in myAddr, farAddr; + rtems_task_priority myPriority; + + printf("Create socket.\n"); + s = socket(AF_INET, SOCK_STREAM, 0); + if (s < 0) + rtems_panic("Can't create socket: %s\n", strerror(errno)); + memset(&myAddr, 0, sizeof myAddr); + myAddr.sin_family = AF_INET; + myAddr.sin_port = htons(1234); + myAddr.sin_addr.s_addr = htonl(INADDR_ANY); + printf("Bind socket.\n"); + if (bind(s, (struct sockaddr *)&myAddr, sizeof myAddr) < 0) + rtems_panic("Can't bind socket: %s\n", strerror(errno)); + if (listen(s, 5) < 0) + printf("Can't listen on socket: %s\n", strerror(errno)); + rtems_task_set_priority(RTEMS_SELF, RTEMS_CURRENT_PRIORITY, &myPriority); + for(;;) { + addrlen = sizeof farAddr; + s1 = accept(s, (struct sockaddr *)&farAddr, &addrlen); + if (s1 < 0) + rtems_panic("Can't accept connection: %s", strerror(errno)); + else + printf("ACCEPTED:%lX\n", ntohl(farAddr.sin_addr.s_addr)); + spawnTask(workerTask, myPriority, s1); + } +} + +/* + * The real part of the client + */ +static rtems_task clientWorker(int arg) +{ + int s; + struct sockaddr_in myAddr, farAddr; + char cbuf[50]; + int i; + + s = socket(AF_INET, SOCK_STREAM, 0); + if (s < 0) { + printf("Can't create client socket: %s\n", strerror(errno)); + return; + } + memset(&myAddr, 0, sizeof myAddr); + myAddr.sin_family = AF_INET; + myAddr.sin_port = htons(0); + myAddr.sin_addr.s_addr = htonl(INADDR_ANY); + if (bind(s, (struct sockaddr *)&myAddr, sizeof myAddr) < 0) { + printf("Can't bind socket: %s\n", strerror(errno)); + goto close; + } + + memset(&farAddr, 0, sizeof farAddr); + farAddr.sin_family = AF_INET; + farAddr.sin_port = htons(1234); + farAddr.sin_addr.s_addr = htonl(INADDR_ANY); + printf("Connect to server.\n"); + if (connect(s, (struct sockaddr *)&farAddr, sizeof farAddr) < 0) { + printf("Can't connect to server: %s\n", strerror(errno)); + goto close; + } + rtems_task_wake_after(20); /* Simulate client delay */ + i = sprintf(cbuf, "Hi there, server (%d).", arg); + i++; /* Send the '\0', too */ + printf("Write %d-byte message to server.\n", i); + if (write(s, cbuf, i) < 0) { + printf("Can't write to server: %s\n", strerror(errno)); + goto close; + } + if ((i = read(s, cbuf, sizeof cbuf)) < 0) { + printf("Can't read from server: %s\n", strerror(errno)); + goto close; + } + printf("Read %d from server: %.*s\n", i, i, cbuf); + rtems_task_wake_after(20); /* Simulate client delay */ + close: + printf("Client closing connection.\n"); + if (close(s) < 0) + printf("Can't close client task socket: %s\n", strerror(errno)); +} + +/* + * Client Task + */ +static rtems_task clientTask(rtems_task_argument arg) +{ + clientWorker(arg); + printf("Client task terminating.\n"); + rtems_task_delete( RTEMS_SELF ); +} + +/* + * RTEMS Startup Task + */ +void test_main(void) +{ + rtems_status_code sc; + rtems_task_priority old; + + sc = rtems_semaphore_create( + rtems_build_name('P','m','t','x'), + 1, + RTEMS_PRIORITY|RTEMS_BINARY_SEMAPHORE|RTEMS_INHERIT_PRIORITY| + RTEMS_NO_PRIORITY_CEILING|RTEMS_LOCAL, + 0, + &printMutex + ); + if (sc != RTEMS_SUCCESSFUL) + rtems_panic("Can't create printf mutex:", rtems_status_text(sc)); + + sc = rtems_task_set_priority( RTEMS_SELF, 50, &old ); + if (sc != RTEMS_SUCCESSFUL) + rtems_panic("Can't create printf mutex:", rtems_status_text(sc)); + + /* + * Network is initialized when we get her + */ + + printf("Try running client with no server present.\n"); + printf("Should fail with `connection refused'.\n"); + clientWorker(0); + + printf("\nStart server.\n"); + spawnTask(serverTask, 150, 0); + + printf("\nTry running client with server present.\n"); + spawnTask(clientTask, 120, 1); + rtems_task_wake_after(500); + + printf("\nTry running two clients.\n"); + spawnTask(clientTask, 120, 2); + spawnTask(clientTask, 120, 3); + rtems_task_wake_after(500); + + printf("\nTry running three clients.\n"); + spawnTask(clientTask, 120, 4); + spawnTask(clientTask, 120, 5); + spawnTask(clientTask, 120, 6); + + rtems_task_wake_after(500); + puts( "*** END OF LOOPBACK TEST ***" ); + exit( 0 ); +} -- cgit v1.2.3