1*86d7f5d3SJohn Marino /*-
2*86d7f5d3SJohn Marino * Copyright (c) 2006, Andrea Bittau <a.bittau@cs.ucl.ac.uk>
3*86d7f5d3SJohn Marino * All rights reserved.
4*86d7f5d3SJohn Marino *
5*86d7f5d3SJohn Marino * Redistribution and use in source and binary forms, with or without
6*86d7f5d3SJohn Marino * modification, are permitted provided that the following conditions
7*86d7f5d3SJohn Marino * are met:
8*86d7f5d3SJohn Marino * 1. Redistributions of source code must retain the above copyright
9*86d7f5d3SJohn Marino * notice, this list of conditions and the following disclaimer.
10*86d7f5d3SJohn Marino * 2. Redistributions in binary form must reproduce the above copyright
11*86d7f5d3SJohn Marino * notice, this list of conditions and the following disclaimer in the
12*86d7f5d3SJohn Marino * documentation and/or other materials provided with the distribution.
13*86d7f5d3SJohn Marino *
14*86d7f5d3SJohn Marino * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15*86d7f5d3SJohn Marino * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16*86d7f5d3SJohn Marino * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17*86d7f5d3SJohn Marino * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18*86d7f5d3SJohn Marino * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19*86d7f5d3SJohn Marino * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20*86d7f5d3SJohn Marino * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21*86d7f5d3SJohn Marino * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22*86d7f5d3SJohn Marino * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23*86d7f5d3SJohn Marino * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24*86d7f5d3SJohn Marino * SUCH DAMAGE.
25*86d7f5d3SJohn Marino *
26*86d7f5d3SJohn Marino * $FreeBSD: src/tools/tools/net80211/wesside/udps/udps.c,v 1.1 2006/08/07 00:05:03 sam Exp $
27*86d7f5d3SJohn Marino */
28*86d7f5d3SJohn Marino #include <sys/types.h>
29*86d7f5d3SJohn Marino #include <sys/socket.h>
30*86d7f5d3SJohn Marino #include <sys/select.h>
31*86d7f5d3SJohn Marino #include <sys/time.h>
32*86d7f5d3SJohn Marino #include <netinet/in.h>
33*86d7f5d3SJohn Marino #include <arpa/inet.h>
34*86d7f5d3SJohn Marino #include <stdio.h>
35*86d7f5d3SJohn Marino #include <unistd.h>
36*86d7f5d3SJohn Marino #include <stdlib.h>
37*86d7f5d3SJohn Marino #include <string.h>
38*86d7f5d3SJohn Marino #include <assert.h>
39*86d7f5d3SJohn Marino #include <time.h>
40*86d7f5d3SJohn Marino #include <err.h>
41*86d7f5d3SJohn Marino
42*86d7f5d3SJohn Marino int poll_rate = 5;
43*86d7f5d3SJohn Marino int pps = 10;
44*86d7f5d3SJohn Marino
45*86d7f5d3SJohn Marino //#define INSANE
46*86d7f5d3SJohn Marino
own(int s,struct sockaddr_in * s_in)47*86d7f5d3SJohn Marino void own(int s, struct sockaddr_in* s_in) {
48*86d7f5d3SJohn Marino char buf[64];
49*86d7f5d3SJohn Marino int times = 10;
50*86d7f5d3SJohn Marino int i;
51*86d7f5d3SJohn Marino int delay = 10*1000;
52*86d7f5d3SJohn Marino unsigned int sent = 0;
53*86d7f5d3SJohn Marino struct timeval start, end;
54*86d7f5d3SJohn Marino struct timespec ts;
55*86d7f5d3SJohn Marino int dont_sleep_times = 1;
56*86d7f5d3SJohn Marino int dont_sleep;
57*86d7f5d3SJohn Marino
58*86d7f5d3SJohn Marino delay = (int) ((double)1.0/pps*1000.0*1000.0);
59*86d7f5d3SJohn Marino
60*86d7f5d3SJohn Marino if (delay <= 5000) {
61*86d7f5d3SJohn Marino dont_sleep_times = 10;
62*86d7f5d3SJohn Marino /*
63*86d7f5d3SJohn Marino printf("delay is %d... sleeping every %d packets\n",
64*86d7f5d3SJohn Marino delay, dont_sleep_times);
65*86d7f5d3SJohn Marino */
66*86d7f5d3SJohn Marino delay *= dont_sleep_times;
67*86d7f5d3SJohn Marino
68*86d7f5d3SJohn Marino delay = (int) (0.90*delay);
69*86d7f5d3SJohn Marino }
70*86d7f5d3SJohn Marino
71*86d7f5d3SJohn Marino dont_sleep = dont_sleep_times;
72*86d7f5d3SJohn Marino times = poll_rate*pps;
73*86d7f5d3SJohn Marino // times *= dont_sleep;
74*86d7f5d3SJohn Marino
75*86d7f5d3SJohn Marino
76*86d7f5d3SJohn Marino
77*86d7f5d3SJohn Marino ts.tv_sec = 0;
78*86d7f5d3SJohn Marino ts.tv_nsec = delay*1000;
79*86d7f5d3SJohn Marino
80*86d7f5d3SJohn Marino // printf("times=%d delay=%d\n", times, delay);
81*86d7f5d3SJohn Marino if (gettimeofday(&start, NULL) == -1) {
82*86d7f5d3SJohn Marino perror("gettimeofday()");
83*86d7f5d3SJohn Marino exit(1);
84*86d7f5d3SJohn Marino }
85*86d7f5d3SJohn Marino
86*86d7f5d3SJohn Marino for(i = 0; i < times; i++) {
87*86d7f5d3SJohn Marino if( sendto(s, buf, 6, 0, (struct sockaddr *)s_in, sizeof(*s_in)) != 6) {
88*86d7f5d3SJohn Marino printf("messed up a bit\n");
89*86d7f5d3SJohn Marino return;
90*86d7f5d3SJohn Marino }
91*86d7f5d3SJohn Marino
92*86d7f5d3SJohn Marino #ifndef INSANE
93*86d7f5d3SJohn Marino
94*86d7f5d3SJohn Marino #if 0
95*86d7f5d3SJohn Marino if (usleep(delay) == -1) {
96*86d7f5d3SJohn Marino perror("usleep()");
97*86d7f5d3SJohn Marino exit(1);
98*86d7f5d3SJohn Marino }
99*86d7f5d3SJohn Marino #endif
100*86d7f5d3SJohn Marino dont_sleep--;
101*86d7f5d3SJohn Marino
102*86d7f5d3SJohn Marino if (!dont_sleep) {
103*86d7f5d3SJohn Marino if (nanosleep(&ts, NULL) == -1) {
104*86d7f5d3SJohn Marino perror("nanosleep()");
105*86d7f5d3SJohn Marino exit(1);
106*86d7f5d3SJohn Marino }
107*86d7f5d3SJohn Marino
108*86d7f5d3SJohn Marino dont_sleep = dont_sleep_times;
109*86d7f5d3SJohn Marino }
110*86d7f5d3SJohn Marino
111*86d7f5d3SJohn Marino #endif
112*86d7f5d3SJohn Marino sent++;
113*86d7f5d3SJohn Marino }
114*86d7f5d3SJohn Marino
115*86d7f5d3SJohn Marino if (gettimeofday(&end, NULL) == -1) {
116*86d7f5d3SJohn Marino perror("gettimeofday()");
117*86d7f5d3SJohn Marino exit(1);
118*86d7f5d3SJohn Marino }
119*86d7f5d3SJohn Marino
120*86d7f5d3SJohn Marino printf ("Sent %.03f p/s\n", ((double)sent)/(((double)end.tv_sec) - start.tv_sec));
121*86d7f5d3SJohn Marino
122*86d7f5d3SJohn Marino // printf("Sent %d packets\n", i);
123*86d7f5d3SJohn Marino }
124*86d7f5d3SJohn Marino
main(int argc,char * argv[])125*86d7f5d3SJohn Marino int main(int argc, char* argv[]) {
126*86d7f5d3SJohn Marino int port = 6969;
127*86d7f5d3SJohn Marino struct sockaddr_in s_in;
128*86d7f5d3SJohn Marino int s;
129*86d7f5d3SJohn Marino int rd;
130*86d7f5d3SJohn Marino int len;
131*86d7f5d3SJohn Marino char buf[64];
132*86d7f5d3SJohn Marino struct timeval tv;
133*86d7f5d3SJohn Marino int do_it = 0;
134*86d7f5d3SJohn Marino fd_set rfds;
135*86d7f5d3SJohn Marino char ip[17];
136*86d7f5d3SJohn Marino
137*86d7f5d3SJohn Marino if( argc > 1)
138*86d7f5d3SJohn Marino pps = atoi(argv[1]);
139*86d7f5d3SJohn Marino
140*86d7f5d3SJohn Marino printf("Packets per second=%d\n", pps);
141*86d7f5d3SJohn Marino
142*86d7f5d3SJohn Marino s = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP);
143*86d7f5d3SJohn Marino if( s < 0)
144*86d7f5d3SJohn Marino err(1, "socket()");
145*86d7f5d3SJohn Marino
146*86d7f5d3SJohn Marino s_in.sin_family = PF_INET;
147*86d7f5d3SJohn Marino s_in.sin_port = htons(port);
148*86d7f5d3SJohn Marino s_in.sin_addr.s_addr = INADDR_ANY;
149*86d7f5d3SJohn Marino
150*86d7f5d3SJohn Marino if( bind(s, (struct sockaddr*)&s_in, sizeof(s_in)) < 0) {
151*86d7f5d3SJohn Marino perror("bind()");
152*86d7f5d3SJohn Marino exit(1);
153*86d7f5d3SJohn Marino }
154*86d7f5d3SJohn Marino
155*86d7f5d3SJohn Marino while(1) {
156*86d7f5d3SJohn Marino assert(do_it >= 0);
157*86d7f5d3SJohn Marino len = sizeof(struct sockaddr_in);
158*86d7f5d3SJohn Marino
159*86d7f5d3SJohn Marino memset(&tv, 0, sizeof(tv));
160*86d7f5d3SJohn Marino tv.tv_usec = 1000*10;
161*86d7f5d3SJohn Marino FD_ZERO(&rfds);
162*86d7f5d3SJohn Marino FD_SET(s, &rfds);
163*86d7f5d3SJohn Marino rd = select(s + 1, &rfds, NULL ,NULL ,&tv);
164*86d7f5d3SJohn Marino if (rd == -1) {
165*86d7f5d3SJohn Marino perror("select()");
166*86d7f5d3SJohn Marino exit(1);
167*86d7f5d3SJohn Marino }
168*86d7f5d3SJohn Marino if (rd == 1 && FD_ISSET(s, &rfds)) {
169*86d7f5d3SJohn Marino rd = recvfrom(s, buf, 64, 0, (struct sockaddr*)&s_in, &len);
170*86d7f5d3SJohn Marino
171*86d7f5d3SJohn Marino if(rd < 0) {
172*86d7f5d3SJohn Marino perror("read died");
173*86d7f5d3SJohn Marino exit(1);
174*86d7f5d3SJohn Marino }
175*86d7f5d3SJohn Marino
176*86d7f5d3SJohn Marino if(rd == 5 && memcmp(buf, "sorbo", 5) == 0) {
177*86d7f5d3SJohn Marino sprintf(ip, "%s", inet_ntoa(s_in.sin_addr));
178*86d7f5d3SJohn Marino printf("Got signal from %s\n", ip);
179*86d7f5d3SJohn Marino #ifdef INSANE
180*86d7f5d3SJohn Marino do_it = 10;
181*86d7f5d3SJohn Marino #else
182*86d7f5d3SJohn Marino do_it = 2;
183*86d7f5d3SJohn Marino #endif
184*86d7f5d3SJohn Marino }
185*86d7f5d3SJohn Marino }
186*86d7f5d3SJohn Marino
187*86d7f5d3SJohn Marino if (do_it) {
188*86d7f5d3SJohn Marino printf("Sending stuff to %s\n", ip);
189*86d7f5d3SJohn Marino
190*86d7f5d3SJohn Marino own(s, &s_in);
191*86d7f5d3SJohn Marino do_it--;
192*86d7f5d3SJohn Marino
193*86d7f5d3SJohn Marino if(do_it == 0)
194*86d7f5d3SJohn Marino printf("Stopping send\n");
195*86d7f5d3SJohn Marino }
196*86d7f5d3SJohn Marino }
197*86d7f5d3SJohn Marino }
198