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

4.11
Last change on this file since e313551 was e313551, checked in by Ralf Corsepius <ralf.corsepius@…>, on Feb 22, 2011 at 10:58:44 AM

Add HAVE_CONFIG_H.

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