source: rtems/testsuites/samples/loopback/init.c @ c499856

4.11
Last change on this file since c499856 was c499856, checked in by Chris Johns <chrisj@…>, on Mar 20, 2014 at 9:10:47 PM

Change all references of rtems.com to rtems.org.

  • Property mode set to 100644
File size: 8.4 KB
Line 
1/*
2 *  The license and distribution terms for this file may be
3 *  found in the file LICENSE in this distribution or at
4 *  http://www.rtems.org/license/LICENSE.
5 */
6
7#ifdef HAVE_CONFIG_H
8#include "config.h"
9#endif
10
11#include <rtems/test.h>
12
13#include <bsp.h>
14
15const char rtems_test_name[] = "LOOPBACK";
16
17#define CONFIGURE_APPLICATION_NEEDS_CONSOLE_DRIVER
18#define CONFIGURE_APPLICATION_NEEDS_CLOCK_DRIVER
19#define CONFIGURE_RTEMS_INIT_TASKS_TABLE
20
21#define CONFIGURE_EXECUTIVE_RAM_SIZE        (512*1024)
22#define CONFIGURE_MAXIMUM_SEMAPHORES        20
23#define CONFIGURE_MAXIMUM_MESSAGE_QUEUES    20
24#define CONFIGURE_MAXIMUM_TASKS            20
25
26#define CONFIGURE_MICROSECONDS_PER_TICK    10000
27#define CONFIGURE_LIBIO_MAXIMUM_FILE_DESCRIPTORS 50
28#define CONFIGURE_USE_IMFS_AS_BASE_FILESYSTEM
29
30#define CONFIGURE_INIT_TASK_STACK_SIZE    (10*1024)
31#define CONFIGURE_INIT_TASK_PRIORITY    50
32#define CONFIGURE_INIT_TASK_INITIAL_MODES (RTEMS_PREEMPT | \
33                                           RTEMS_NO_TIMESLICE | \
34                                           RTEMS_NO_ASR | \
35                                           RTEMS_INTERRUPT_LEVEL(0))
36
37#define CONFIGURE_INIT
38rtems_task Init(rtems_task_argument argument);
39
40#define CONFIGURE_INITIAL_EXTENSIONS RTEMS_TEST_INITIAL_EXTENSION
41
42#include <rtems/confdefs.h>
43
44#if !BSP_SMALL_MEMORY
45
46#include <rtems/rtems_bsdnet.h>
47#include <rtems/error.h>
48#include <stdio.h>
49#include <stdarg.h>
50#include <stdlib.h>
51#include <string.h>
52#include <errno.h>
53#include <sys/socket.h>
54#include <netinet/in.h>
55
56/*
57 * Network configuration
58 */
59
60struct rtems_bsdnet_config rtems_bsdnet_config = {
61    NULL,                   /* Network interface */
62    NULL,                   /* Use fixed network configuration */
63    0,                      /* Default network task priority */
64    0,                      /* Default mbuf capacity */
65    0,                      /* Default mbuf cluster capacity */
66    "testSystem",           /* Host name */
67    "nowhere.com",          /* Domain name */
68    "127.0.0.1",            /* Gateway */
69    "127.0.0.1",            /* Log host */
70    {"127.0.0.1" },         /* Name server(s) */
71    {"127.0.0.1" },         /* NTP server(s) */
72    0,
73    0,
74    0,
75    0,
76    0
77};
78
79/*
80 * Thread-safe output routines
81 */
82static rtems_id printMutex;
83static void printSafe(const char *fmt, ...)
84{
85    va_list args;
86    va_start(args, fmt);
87    rtems_semaphore_obtain(printMutex, RTEMS_WAIT, RTEMS_NO_TIMEOUT);
88    vprintf(fmt, args);
89    rtems_semaphore_release(printMutex);
90    va_end(args);
91}
92#define printf printSafe
93
94/*
95 * Spawn a task
96 */
97static void spawnTask(rtems_task_entry entryPoint, rtems_task_priority priority, rtems_task_argument arg)
98{
99    rtems_status_code sc;
100    rtems_id tid;
101
102    sc = rtems_task_create(rtems_build_name('t','a','s','k'),
103            priority,
104            RTEMS_MINIMUM_STACK_SIZE+(8*1024),
105            RTEMS_PREEMPT|RTEMS_TIMESLICE|RTEMS_NO_ASR|RTEMS_INTERRUPT_LEVEL(0),
106            RTEMS_FLOATING_POINT|RTEMS_LOCAL,
107            &tid);
108    if (sc != RTEMS_SUCCESSFUL)
109        rtems_panic("Can't create task: %s", rtems_status_text(sc));
110    sc = rtems_task_start(tid, entryPoint, arg);
111    if (sc != RTEMS_SUCCESSFUL)
112        rtems_panic("Can't start task: %s", rtems_status_text(sc));
113}
114
115/*
116 * Server subtask
117 */
118static rtems_task workerTask(rtems_task_argument arg)
119{
120    int s = arg;
121    char msg[80];
122    char reply[100];
123    int i;
124
125    for (;;) {
126        if ((i = read(s, msg, sizeof msg)) < 0) {
127            printf("Server couldn't read message from client: %s\n", strerror(errno));
128            break;
129        }
130        if (i == 0)
131            break;
132        rtems_task_wake_after(20); /* Simulate some processing delay */
133        i = sprintf(reply, "Server received %d (%s)", i, msg);
134        if ((i = write(s, reply, i+1)) < 0) {
135            printf("Server couldn't write message to client: %s\n", strerror(errno));
136            break;
137        }
138    }
139    if (close(s) < 0)
140        printf("Can't close worker task socket: %s\n", strerror(errno));
141    printf("Worker task terminating.\n");
142    rtems_task_delete(RTEMS_SELF);
143}
144
145/*
146 * Server Task
147 */
148static rtems_task serverTask(rtems_task_argument arg)
149{
150    int s, s1;
151    socklen_t addrlen;
152        struct sockaddr_in myAddr, farAddr;
153    rtems_task_priority myPriority;
154
155        printf("Create socket.\n");
156        s = socket(AF_INET, SOCK_STREAM, 0);
157        if (s < 0)
158                rtems_panic("Can't create socket: %s\n", strerror(errno));
159    memset(&myAddr, 0, sizeof myAddr);
160        myAddr.sin_family = AF_INET;
161        myAddr.sin_port = htons(1234);
162        myAddr.sin_addr.s_addr = htonl(INADDR_ANY);
163        printf("Bind socket.\n");
164        if (bind(s, (struct sockaddr *)&myAddr, sizeof myAddr) < 0)
165                rtems_panic("Can't bind socket: %s\n", strerror(errno));
166        if (listen(s, 5) < 0)
167                printf("Can't listen on socket: %s\n", strerror(errno));
168    rtems_task_set_priority(RTEMS_SELF, RTEMS_CURRENT_PRIORITY, &myPriority);
169    for(;;) {
170        addrlen = sizeof farAddr;
171        s1 = accept(s, (struct sockaddr *)&farAddr, &addrlen);
172        if (s1 < 0)
173            rtems_panic("Can't accept connection: %s", strerror(errno));
174        else
175            printf("ACCEPTED:%lX\n", ntohl(farAddr.sin_addr.s_addr));
176        spawnTask(workerTask, myPriority, s1);
177    }
178}
179
180/*
181 * The real part of the client
182 */
183static rtems_task clientWorker(int arg)
184{
185    int s;
186        struct sockaddr_in myAddr, farAddr;
187    char cbuf[50];
188    int i;
189
190        s = socket(AF_INET, SOCK_STREAM, 0);
191        if (s < 0) {
192                printf("Can't create client socket: %s\n", strerror(errno));
193        return;
194    }
195    memset(&myAddr, 0, sizeof myAddr);
196        myAddr.sin_family = AF_INET;
197        myAddr.sin_port = htons(0);
198        myAddr.sin_addr.s_addr = htonl(INADDR_ANY);
199        if (bind(s, (struct sockaddr *)&myAddr, sizeof myAddr) < 0) {
200                printf("Can't bind socket: %s\n", strerror(errno));
201        goto close;
202    }
203    memset(&farAddr, 0, sizeof farAddr);
204        farAddr.sin_family = AF_INET;
205        farAddr.sin_port = htons(1234);
206        farAddr.sin_addr.s_addr = htonl(INADDR_ANY);
207        printf("Connect to server.\n");
208        if (connect(s, (struct sockaddr *)&farAddr, sizeof farAddr) < 0) {
209                printf("Can't connect to server: %s\n", strerror(errno));
210        goto close;
211    }
212    rtems_task_wake_after(20); /* Simulate client delay */
213    i = sprintf(cbuf, "Hi there, server (%d).", arg);
214    i++;    /* Send the '\0', too */
215    printf("Write %d-byte message to server.\n", i);
216    if (write(s, cbuf, i) < 0) {
217                printf("Can't write to server: %s\n", strerror(errno));
218        goto close;
219    }
220    if ((i = read(s, cbuf, sizeof cbuf)) < 0) {
221                printf("Can't read from server: %s\n", strerror(errno));
222        goto close;
223    }
224    printf("Read %d from server: %.*s\n", i, i, cbuf);
225    rtems_task_wake_after(20); /* Simulate client delay */
226  close:
227    printf("Client closing connection.\n");
228    if (close(s) < 0)
229        printf("Can't close client task socket: %s\n", strerror(errno));
230}
231
232/*
233 * Client Task
234 */
235static rtems_task clientTask(rtems_task_argument arg)
236{
237    clientWorker(arg);
238    printf("Client task terminating.\n");
239    rtems_task_delete( RTEMS_SELF );
240}
241
242/*
243 * RTEMS Startup Task
244 */
245rtems_task
246Init (rtems_task_argument ignored)
247{
248    rtems_status_code sc;
249
250    rtems_test_begin();
251
252    sc = rtems_semaphore_create(rtems_build_name('P','m','t','x'),
253                1,
254                RTEMS_PRIORITY|RTEMS_BINARY_SEMAPHORE|RTEMS_INHERIT_PRIORITY|
255                                    RTEMS_NO_PRIORITY_CEILING|RTEMS_LOCAL,
256                0,
257                &printMutex);
258    if (sc != RTEMS_SUCCESSFUL)
259        rtems_panic("Can't create printf mutex:", rtems_status_text(sc));
260    printf("\"Network\" initializing!\n");
261    rtems_bsdnet_initialize_network();
262    printf("\"Network\" initialized!\n");
263
264    printf("Try running client with no server present.\n");
265    printf("Should fail with `connection refused'.\n");
266    clientWorker(0);
267
268    printf("\nStart server.\n");
269    spawnTask(serverTask, 150, 0);
270
271    printf("\nTry running client with server present.\n");
272    spawnTask(clientTask, 120, 1);
273    rtems_task_wake_after(500);
274
275    printf("\nTry running two clients.\n");
276    spawnTask(clientTask, 120, 2);
277    spawnTask(clientTask, 120, 3);
278    rtems_task_wake_after(500);
279
280    printf("\nTry running three clients.\n");
281    spawnTask(clientTask, 120, 4);
282    spawnTask(clientTask, 120, 5);
283    spawnTask(clientTask, 120, 6);
284
285    rtems_task_wake_after(500);
286    rtems_test_end();
287    exit( 0 );
288}
289#else
290#include <stdio.h>
291/*
292 * RTEMS Startup Task
293 */
294rtems_task
295Init (rtems_task_argument ignored)
296{
297  printf("NO NETWORKING. MEMORY TOO SMALL");
298}
299#endif
Note: See TracBrowser for help on using the repository browser.