2 This file is part of GNUnet.
3 (C) 2010, 2012 Christian Grothoff (and other contributing authors)
5 GNUnet is free software; you can redistribute it and/or modify
6 it under the terms of the GNU General Public License as published
7 by the Free Software Foundation; either version 3, or (at your
8 option) any later version.
10 GNUnet is distributed in the hope that it will be useful, but
11 WITHOUT ANY WARRANTY; without even the implied warranty of
12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
13 General Public License for more details.
15 You should have received a copy of the GNU General Public License
16 along with GNUnet; see the file COPYING. If not, write to the
17 Free Software Foundation, Inc., 59 Temple Place - Suite 330,
18 Boston, MA 02111-1307, USA.
21 * @file transport/gnunet-helper-transport-wlan-dummy.c
22 * @brief helper for the testcases for plugin_transport_wlan.c
23 * @author David Brodski
26 #include "gnunet_protocols.h"
27 #include "gnunet_util_lib.h"
28 #include "plugin_transport_wlan.h"
31 * Name of the fifo to use for IPC with the other dummy process.
33 #define FIFO_FILE1 "/tmp/test-transport/api-wlan-p1/WLAN_FIFO_in"
36 * Name of the fifo to use for IPC with the other dummy process.
38 #define FIFO_FILE2 "/tmp/test-transport/api-wlan-p1/WLAN_FIFO_out"
41 * Maximum size of a message allowed in either direction
42 * (used for our receive and sent buffers).
48 * IO buffer used for buffering data in transit.
54 * How many bytes that were stored in 'buf' did we already write to the
55 * destination? Always smaller than 'size'.
60 * How many bytes of data are stored in 'buf' for transmission right now?
61 * Data always starts at offset 0 and extends to 'size'.
66 * Buffered data; twice the maximum allowed message size as we add some
69 char buf[MAXLINE * 2];
74 * Flag set to 1 if we are to terminate, otherwise 0.
80 * We're being killed, clean up.
82 * @param sig killing signal
88 (void) unlink (FIFO_FILE1);
89 (void) unlink (FIFO_FILE2);
94 * Create control message for plugin
96 * @param buffer pointer to buffer for the message
97 * @param mac pointer to the mac address
98 * @return number of bytes written
101 send_mac_to_plugin (char *buffer, struct GNUNET_TRANSPORT_WLAN_MacAddress *mac)
104 struct GNUNET_TRANSPORT_WLAN_HelperControlMessage macmsg;
106 memcpy (&macmsg.mac, (char *) mac, sizeof (struct GNUNET_TRANSPORT_WLAN_MacAddress));
107 macmsg.hdr.size = htons (sizeof (struct GNUNET_TRANSPORT_WLAN_HelperControlMessage));
108 macmsg.hdr.type = htons (GNUNET_MESSAGE_TYPE_WLAN_HELPER_CONTROL);
109 memcpy (buffer, &macmsg, sizeof (struct GNUNET_TRANSPORT_WLAN_HelperControlMessage));
110 return sizeof (struct GNUNET_TRANSPORT_WLAN_HelperControlMessage);
115 * We got a message from the FIFO, check it, convert the message
116 * type to the output forward and copy it to the buffer for stdout.
118 * @param cls the 'struct SendBuffer' to copy the converted message to
119 * @param client unused
120 * @param hdr inbound message from the FIFO
123 stdin_send (void *cls, void *client, const struct GNUNET_MessageHeader *hdr)
125 struct SendBuffer *write_pout = cls;
126 const struct GNUNET_TRANSPORT_WLAN_RadiotapSendMessage *in;
128 struct GNUNET_TRANSPORT_WLAN_RadiotapReceiveMessage newheader;
131 sendsize = ntohs (hdr->size);
132 in = (const struct GNUNET_TRANSPORT_WLAN_RadiotapSendMessage *) hdr;
133 if ( (GNUNET_MESSAGE_TYPE_WLAN_DATA_TO_HELPER != ntohs (hdr->type)) ||
134 (sizeof (struct GNUNET_TRANSPORT_WLAN_RadiotapSendMessage) > sendsize) )
136 FPRINTF (stderr, "%s", "Received malformed message\n");
139 payload_size = sendsize - sizeof (struct GNUNET_TRANSPORT_WLAN_RadiotapSendMessage);
140 if ((payload_size + sizeof (struct GNUNET_TRANSPORT_WLAN_RadiotapReceiveMessage) + write_pout->size) > MAXLINE * 2)
142 FPRINTF (stderr, "%s", "Packet too big for buffer\n");
145 memset (&newheader, 0, sizeof (newheader));
146 newheader.header.size = htons (payload_size + sizeof (newheader));
147 newheader.header.type = htons (GNUNET_MESSAGE_TYPE_WLAN_DATA_FROM_HELPER);
148 newheader.frame = in->frame;
149 memcpy (write_pout->buf + write_pout->size,
152 write_pout->size += sizeof (newheader);
153 memcpy (write_pout->buf + write_pout->size,
156 write_pout->size += payload_size;
162 * We read a full message from stdin. Copy it to our send buffer.
164 * @param cls the 'struct SendBuffer' to copy to
165 * @param client unused
166 * @param hdr the message we received to copy to the buffer
169 file_in_send (void *cls, void *client, const struct GNUNET_MessageHeader *hdr)
171 struct SendBuffer *write_std = cls;
174 sendsize = ntohs (hdr->size);
175 if ((sendsize + write_std->size) > MAXLINE * 2)
177 FPRINTF (stderr, "%s", "Packet too big for buffer\n");
180 memcpy (write_std->buf + write_std->size, hdr, sendsize);
181 write_std->size += sendsize;
187 * Main function of a program that pretends to be a WLAN card.
189 * @param argc should be 2
190 * @param argv either '1' or '2', depending on which of the two cards this dummy is to emulate
191 * @return 1 on error, 0 if terminated normally via signal
194 main (int argc, char *argv[])
202 char readbuf[MAXLINE];
204 struct SendBuffer write_std;
205 struct SendBuffer write_pout;
212 struct GNUNET_SERVER_MessageStreamTokenizer *stdin_mst = NULL;
213 struct GNUNET_SERVER_MessageStreamTokenizer *file_in_mst = NULL;
214 struct GNUNET_TRANSPORT_WLAN_MacAddress macaddr;
218 ((0 != strcmp (argv[1], "1")) && (0 != strcmp (argv[1], "2"))) )
222 "This program must be started with the operating mode (1 or 2) as the only argument.\n");
226 /* make the fifos if needed */
228 if ( (GNUNET_OK != GNUNET_DISK_directory_create_for_file (FIFO_FILE1)) ||
229 (GNUNET_OK != GNUNET_DISK_directory_create_for_file (FIFO_FILE2)) )
231 FPRINTF (stderr, "Failed to create directory for file `%s'\n", FIFO_FILE1);
234 if (0 == strcmp (argv[1], "1") )
236 if (0 != stat (FIFO_FILE1, &st))
238 erg = mkfifo (FIFO_FILE1, 0666);
239 if ( (0 != erg) && (EEXIST != errno) )
240 FPRINTF (stderr, "Error in mkfifo(%s): %s\n", FIFO_FILE1,
246 if (0 != stat (FIFO_FILE2, &st))
248 GNUNET_break (0 == (erg = mkfifo (FIFO_FILE2, 0666)));
249 if ( (0 != erg) && (EEXIST != errno) )
250 FPRINTF (stderr, "Error in mkfifo(%s): %s\n", FIFO_FILE2,
255 if (0 == strcmp (argv[1], "1"))
258 fpin = fopen (FIFO_FILE1, "r");
261 FPRINTF (stderr, "fopen of read FIFO_FILE1 failed: %s\n", STRERROR (errno));
264 if (NULL == (fpout = fopen (FIFO_FILE2, "w")))
266 GNUNET_break (0 == mkfifo (FIFO_FILE2, 0666));
267 fpout = fopen (FIFO_FILE2, "w");
271 FPRINTF (stderr, "fopen of write FIFO_FILE2 failed: %s\n", STRERROR (errno));
278 if (NULL == (fpout = fopen (FIFO_FILE1, "w")))
280 GNUNET_break (0 == mkfifo (FIFO_FILE1, 0666));
281 fpout = fopen (FIFO_FILE1, "w");
285 FPRINTF (stderr, "fopen of write FIFO_FILE1 failed: %s\n", STRERROR (errno));
288 fpin = fopen (FIFO_FILE2, "r");
291 FPRINTF (stderr, "fopen of read FIFO_FILE2 failed: %s\n", STRERROR (errno));
296 fdpin = fileno (fpin);
297 GNUNET_assert (fpin >= 0);
298 if (fdpin >= FD_SETSIZE)
300 FPRINTF (stderr, "File fdpin number too large (%d > %u)\n", fdpin,
301 (unsigned int) FD_SETSIZE);
305 fdpout = fileno (fpout);
306 GNUNET_assert (fdpout >= 0);
308 if (fdpout >= FD_SETSIZE)
310 FPRINTF (stderr, "File fdpout number too large (%d > %u)\n", fdpout,
311 (unsigned int) FD_SETSIZE);
315 signal (SIGINT, &sigfunc);
316 signal (SIGTERM, &sigfunc);
322 stdin_mst = GNUNET_SERVER_mst_create (&stdin_send, &write_pout);
323 file_in_mst = GNUNET_SERVER_mst_create (&file_in_send, &write_std);
325 /* Send 'random' mac address */
326 macaddr.mac[0] = 0x13;
327 macaddr.mac[1] = 0x22;
328 macaddr.mac[2] = 0x33;
329 macaddr.mac[3] = 0x44;
330 macaddr.mac[4] = GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_STRONG, 256);
331 macaddr.mac[5] = GNUNET_CRYPTO_random_u32 (GNUNET_CRYPTO_QUALITY_NONCE, 256);
332 write_std.size = send_mac_to_plugin (write_std.buf, &macaddr);
334 while (0 == closeprog)
342 /* if output queue is empty, read */
343 if (0 == write_pout.size)
345 FD_SET (STDIN_FILENO, &rfds);
346 maxfd = MAX (STDIN_FILENO, maxfd);
348 if (0 == write_std.size)
350 FD_SET (fdpin, &rfds);
351 maxfd = MAX (fdpin, maxfd);
354 /* if there is something to write, try to write */
355 if (0 < write_std.size)
357 FD_SET (STDOUT_FILENO, &wfds);
358 maxfd = MAX (maxfd, STDOUT_FILENO);
360 if (0 < write_pout.size)
362 FD_SET (fdpout, &wfds);
363 maxfd = MAX (maxfd, fdpout);
366 retval = select (maxfd + 1, &rfds, &wfds, NULL, &tv);
367 if ((-1 == retval) && (EINTR == errno))
371 FPRINTF (stderr, "select failed: %s\n", STRERROR (errno));
376 if (FD_ISSET (STDOUT_FILENO, &wfds))
379 write (STDOUT_FILENO, write_std.buf + write_std.pos,
380 write_std.size - write_std.pos);
384 FPRINTF (stderr, "Write ERROR to STDOUT_FILENO: %s\n",
390 write_std.pos += ret;
391 /* check if finished writing */
392 if (write_std.pos == write_std.size)
400 if (FD_ISSET (fdpout, &wfds))
403 write (fdpout, write_pout.buf + write_pout.pos,
404 write_pout.size - write_pout.pos);
409 FPRINTF (stderr, "Write ERROR to fdpout failed: %s\n", STRERROR (errno));
413 write_pout.pos += ret;
414 /* check if finished writing */
415 if (write_pout.pos == write_pout.size)
423 if (FD_ISSET (STDIN_FILENO, &rfds))
425 readsize = read (STDIN_FILENO, readbuf, sizeof (readbuf));
430 FPRINTF (stderr, "Error reading from STDIN_FILENO: %s\n",
433 else if (0 < readsize)
435 GNUNET_SERVER_mst_receive (stdin_mst, NULL, readbuf, readsize,
436 GNUNET_NO, GNUNET_NO);
446 if (FD_ISSET (fdpin, &rfds))
448 readsize = read (fdpin, readbuf, sizeof (readbuf));
452 FPRINTF (stderr, "Error reading from fdpin: %s\n", STRERROR (errno));
455 else if (0 < readsize)
457 GNUNET_SERVER_mst_receive (file_in_mst, NULL, readbuf, readsize,
458 GNUNET_NO, GNUNET_NO);
470 if (NULL != stdin_mst)
471 GNUNET_SERVER_mst_destroy (stdin_mst);
472 if (NULL != file_in_mst)
473 GNUNET_SERVER_mst_destroy (file_in_mst);
481 (void) unlink (FIFO_FILE1);
482 (void) unlink (FIFO_FILE2);