fix div by zero
[oweals/gnunet.git] / src / transport / gnunet-transport-wlan-sender.c
1 /*
2  This file is part of GNUnet
3  Copyright (C) 2011 GNUnet e.V.
4
5  GNUnet is free software: you can redistribute it and/or modify it
6  under the terms of the GNU Affero General Public License as published
7  by the Free Software Foundation, either version 3 of the License,
8  or (at your option) any later version.
9
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  Affero General Public License for more details.
14
15  You should have received a copy of the GNU Affero General Public License
16  along with this program.  If not, see <http://www.gnu.org/licenses/>.
17  */
18
19 /**
20  * @file transport/gnunet-transport-wlan-sender.c
21  * @brief program to send via WLAN as much as possible (to test physical/theoretical throughput)
22  * @author David Brodski
23  */
24 #include "platform.h"
25 #include "plugin_transport_wlan.h"
26 #include "gnunet_protocols.h"
27
28 #define WLAN_MTU 1500
29
30 /**
31  * LLC fields for better compatibility
32  */
33 #define WLAN_LLC_DSAP_FIELD 0x1f
34 #define WLAN_LLC_SSAP_FIELD 0x1f
35
36 #define IEEE80211_ADDR_LEN      6       /* size of 802.11 address */
37
38 #define IEEE80211_FC0_VERSION_MASK              0x03
39 #define IEEE80211_FC0_VERSION_SHIFT             0
40 #define IEEE80211_FC0_VERSION_0                 0x00
41 #define IEEE80211_FC0_TYPE_MASK                 0x0c
42 #define IEEE80211_FC0_TYPE_SHIFT                2
43 #define IEEE80211_FC0_TYPE_MGT                  0x00
44 #define IEEE80211_FC0_TYPE_CTL                  0x04
45 #define IEEE80211_FC0_TYPE_DATA                 0x08
46
47
48 /**
49  * function to fill the radiotap header
50  * @param header pointer to the radiotap header
51  * @param size total message size
52  * @return GNUNET_YES at success
53  */
54 static int
55 getRadiotapHeader (struct GNUNET_TRANSPORT_WLAN_RadiotapSendMessage *header,
56                    uint16_t size)
57 {
58   header->header.size = htons (size);
59   header->header.type = htons (GNUNET_MESSAGE_TYPE_WLAN_DATA_TO_HELPER);
60   header->rate = 255;
61   header->tx_power = 0;
62   header->antenna = 0;
63   return GNUNET_YES;
64 }
65
66 /**
67  * function to generate the wlan hardware header for one packet
68  * @param Header address to write the header to
69  * @param to_mac_addr pointer to the address of the recipient
70  * @param mac pointer to the mac address to send from (normally overwritten over by helper)
71  * @param size size of the whole packet, needed to calculate the time to send the packet
72  * @return GNUNET_YES if there was no error
73  */
74 static int
75 getWlanHeader (struct GNUNET_TRANSPORT_WLAN_Ieee80211Frame *Header,
76                const struct GNUNET_TRANSPORT_WLAN_MacAddress *to_mac_addr,
77                const struct GNUNET_TRANSPORT_WLAN_MacAddress *mac, unsigned int size)
78 {
79   const int rate = 11000000;
80
81   Header->frame_control = htons (IEEE80211_FC0_TYPE_DATA);
82   Header->addr3 = mac_bssid_gnunet;
83   Header->addr2 = *mac;
84   Header->addr1 = *to_mac_addr;
85   Header->duration = GNUNET_htole16 ((size * 1000000) / rate + 290);
86   Header->llc[0] = WLAN_LLC_DSAP_FIELD;
87   Header->llc[1] = WLAN_LLC_SSAP_FIELD;
88   Header->llc[2] = 0; // FIXME
89   Header->llc[3] = 0; // FIXME
90   return GNUNET_YES;
91 }
92
93
94 int
95 main (int argc, char *argv[])
96 {
97   char msg_buf[WLAN_MTU];
98   struct GNUNET_TRANSPORT_WLAN_RadiotapSendMessage *radiotap;
99   unsigned int temp[6];
100   struct GNUNET_TRANSPORT_WLAN_MacAddress inmac;
101   struct GNUNET_TRANSPORT_WLAN_MacAddress outmac;
102   struct GNUNET_TRANSPORT_WLAN_HelperControlMessage hcm;
103   unsigned long long count;
104   double bytes_per_s;
105   time_t start;
106   time_t akt;
107   int i;
108   ssize_t ret;
109   pid_t pid;
110   int commpipe[2];              /* This holds the fd for the input & output of the pipe */
111   int macpipe[2];              /* This holds the fd for the input & output of the pipe */
112
113   if (4 != argc)
114   {
115     fprintf (stderr,
116              "This program must be started with the interface and the targets and source mac as argument.\n");
117     fprintf (stderr,
118              "Usage: interface-name mac-DST mac-SRC\n"
119              "e.g. mon0 11-22-33-44-55-66 12-34-56-78-90-ab\n");
120     return 1;
121   }
122   if (6 !=
123       SSCANF (argv[2], "%x-%x-%x-%x-%x-%x", &temp[0], &temp[1], &temp[2],
124               &temp[3], &temp[4], &temp[5]))
125   {
126     fprintf (stderr,
127              "Usage: interface-name mac-DST mac-SRC\n"
128              "e.g. mon0 11-22-33-44-55-66 12-34-56-78-90-ab\n");
129     return 1;
130   }
131   for (i = 0; i < 6; i++)
132     outmac.mac[i] = temp[i];
133   if (6 !=
134       SSCANF (argv[3], "%x-%x-%x-%x-%x-%x", &temp[0], &temp[1], &temp[2],
135               &temp[3], &temp[4], &temp[5]))
136   {
137     fprintf (stderr,
138              "Usage: interface-name mac-DST mac-SRC\n"
139              "e.g. mon0 11-22-33-44-55-66 12-34-56-78-90-ab\n");
140     return 1;
141   }
142   for (i = 0; i < 6; i++)
143     inmac.mac[i] = temp[i];
144
145
146   /* Setup communication pipeline first */
147   if (pipe (commpipe))
148   {
149     fprintf (stderr,
150              "Failed to create pipe: %s\n",
151              STRERROR (errno));
152     exit (1);
153   }
154   if (pipe (macpipe))
155   {
156     fprintf (stderr,
157              "Failed to create pipe: %s\n",
158              STRERROR (errno));
159     exit (1);
160   }
161
162   /* Attempt to fork and check for errors */
163   if ((pid = fork ()) == -1)
164   {
165     fprintf (stderr, "Failed to fork: %s\n",
166              STRERROR (errno));
167     exit (1);
168   }
169   memset (msg_buf, 0x42, sizeof (msg_buf));
170   if (pid)
171   {
172     /* A positive (non-negative) PID indicates the parent process */
173     if (0 != close (commpipe[0]))        /* Close unused side of pipe (in side) */
174       fprintf (stderr,
175                "Failed to close fd: %s\n",
176                strerror (errno));
177     setvbuf (stdout, (char *) NULL, _IONBF, 0); /* Set non-buffered output on stdout */
178
179     if (0 != close (macpipe[1]))
180       fprintf (stderr,
181                "Failed to close fd: %s\n",
182                strerror (errno));
183     if (sizeof (hcm) != read (macpipe[0], &hcm, sizeof (hcm)))
184       fprintf (stderr,
185                "Failed to read hcm...\n");
186     fprintf (stderr,
187              "Got MAC %.2X:%.2X:%.2X:%.2X:%.2X:%.2X\n",
188              hcm.mac.mac[0], hcm.mac.mac[1],
189              hcm.mac.mac[2], hcm.mac.mac[3], hcm.mac.mac[4], hcm.mac.mac[5]);                           
190     radiotap = (struct GNUNET_TRANSPORT_WLAN_RadiotapSendMessage *) msg_buf;
191     getRadiotapHeader (radiotap, WLAN_MTU);
192     getWlanHeader (&radiotap->frame, &outmac, &inmac,
193                    WLAN_MTU);
194     start = time (NULL);
195     count = 0;
196     while (1)
197     {
198       ret = write (commpipe[1], msg_buf, WLAN_MTU);
199       if (0 > ret)
200       {
201         fprintf (stderr, "write failed: %s\n", strerror (errno));
202         break;
203       }
204       count += ret;
205       akt =  time (NULL);
206       if (akt - start > 30)
207       {
208         bytes_per_s = count / (akt - start);
209         bytes_per_s /= 1024;
210         printf ("send %f kbytes/s\n", bytes_per_s);
211         start = akt;
212         count = 0;
213       }
214     }
215   }
216   else
217   {
218     /* A zero PID indicates that this is the child process */
219     (void) close (0);
220     (void) close (1);
221     if (-1 == dup2 (commpipe[0], 0))    /* Replace stdin with the in side of the pipe */
222       fprintf (stderr, "dup2 failed: %s\n", strerror (errno));
223     if (-1 == dup2 (macpipe[1], 1))    /* Replace stdout with the out side of the pipe */
224       fprintf (stderr, "dup2 failed: %s\n", strerror (errno));
225     (void) close (commpipe[1]); /* Close unused side of pipe (out side) */
226     (void) close (macpipe[0]); /* Close unused side of pipe (in side) */
227     /* Replace the child fork with a new process */
228     if (execlp
229         ("gnunet-helper-transport-wlan", "gnunet-helper-transport-wlan",
230          argv[1], NULL) == -1)
231     {
232       fprintf (stderr, "Could not start gnunet-helper-transport-wlan!");
233       _exit (1);
234     }
235   }
236   return 0;
237 }