← Back to branch summary

~ubuntu-branches/ubuntu/wily/bluez/wily

« back to all changes in this revision

Viewing changes to test/scotest.c

  • Committer: Bazaar Package Importer
  • Author(s): Mario Limonciello
  • Date: 2008-10-07 12:10:29 UTC
  • Revision ID: james.westby@ubuntu.com-20081007121029-4gup4fmmh2vfo5nh
Tags: upstream-4.12
Import upstream version 4.12

Show diffs side-by-side

added added

removed removed

Lines of Context:
test/scotest.c
 
1
/*
 
2
 *
 
3
 *  BlueZ - Bluetooth protocol stack for Linux
 
4
 *
 
5
 *  Copyright (C) 2002-2003  Maxim Krasnyansky <maxk@qualcomm.com>
 
6
 *  Copyright (C) 2002-2008  Marcel Holtmann <marcel@holtmann.org>
 
7
 *
 
8
 *
 
9
 *  This program is free software; you can redistribute it and/or modify
 
10
 *  it under the terms of the GNU General Public License as published by
 
11
 *  the Free Software Foundation; either version 2 of the License, or
 
12
 *  (at your option) any later version.
 
13
 *
 
14
 *  This program is distributed in the hope that it will be useful,
 
15
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 
16
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
17
 *  GNU General Public License for more details.
 
18
 *
 
19
 *  You should have received a copy of the GNU General Public License
 
20
 *  along with this program; if not, write to the Free Software
 
21
 *  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
 
22
 *
 
23
 */
 
24
 
 
25
#ifdef HAVE_CONFIG_H
 
26
#include <config.h>
 
27
#endif
 
28
 
 
29
#include <stdio.h>
 
30
#include <errno.h>
 
31
#include <ctype.h>
 
32
#include <unistd.h>
 
33
#include <stdlib.h>
 
34
#include <getopt.h>
 
35
#include <syslog.h>
 
36
#include <signal.h>
 
37
#include <sys/time.h>
 
38
#include <sys/socket.h>
 
39
 
 
40
#include <bluetooth/bluetooth.h>
 
41
#include <bluetooth/sco.h>
 
42
 
 
43
/* Test modes */
 
44
enum {
 
45
        SEND,
 
46
        RECV,
 
47
        RECONNECT,
 
48
        MULTY,
 
49
        DUMP,
 
50
        CONNECT
 
51
};
 
52
 
 
53
static unsigned char *buf;
 
54
 
 
55
/* Default data size */
 
56
static long data_size = 672;
 
57
 
 
58
static bdaddr_t bdaddr;
 
59
 
 
60
static float tv2fl(struct timeval tv)
 
61
{
 
62
        return (float)tv.tv_sec + (float)(tv.tv_usec/1000000.0);
 
63
}
 
64
 
 
65
static int do_connect(char *svr)
 
66
{
 
67
        struct sockaddr_sco addr;
 
68
        struct sco_conninfo conn;
 
69
        socklen_t optlen;
 
70
        int sk;
 
71
 
 
72
        /* Create socket */
 
73
        sk = socket(PF_BLUETOOTH, SOCK_SEQPACKET, BTPROTO_SCO);
 
74
        if (sk < 0) {
 
75
                syslog(LOG_ERR, "Can't create socket: %s (%d)",
 
76
                                                        strerror(errno), errno);
 
77
                return -1;
 
78
        }
 
79
 
 
80
        /* Bind to local address */
 
81
        memset(&addr, 0, sizeof(addr));
 
82
        addr.sco_family = AF_BLUETOOTH;
 
83
        bacpy(&addr.sco_bdaddr, &bdaddr);
 
84
 
 
85
        if (bind(sk, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
 
86
                syslog(LOG_ERR, "Can't bind socket: %s (%d)",
 
87
                                                        strerror(errno), errno);
 
88
                goto error;
 
89
        }
 
90
 
 
91
        /* Connect to remote device */
 
92
        memset(&addr, 0, sizeof(addr));
 
93
        addr.sco_family = AF_BLUETOOTH;
 
94
        str2ba(svr, &addr.sco_bdaddr);
 
95
 
 
96
        if (connect(sk, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
 
97
                syslog(LOG_ERR, "Can't connect: %s (%d)",
 
98
                                                        strerror(errno), errno);
 
99
                goto error;
 
100
        }
 
101
 
 
102
        /* Get connection information */
 
103
        memset(&conn, 0, sizeof(conn));
 
104
        optlen = sizeof(conn);
 
105
 
 
106
        if (getsockopt(sk, SOL_SCO, SCO_CONNINFO, &conn, &optlen) < 0) {
 
107
                syslog(LOG_ERR, "Can't get SCO connection information: %s (%d)",
 
108
                                                        strerror(errno), errno);
 
109
                goto error;
 
110
        }
 
111
 
 
112
        syslog(LOG_INFO, "Connected [handle %d, class 0x%02x%02x%02x]",
 
113
                conn.hci_handle,
 
114
                conn.dev_class[2], conn.dev_class[1], conn.dev_class[0]);
 
115
 
 
116
        return sk;
 
117
 
 
118
error:
 
119
        close(sk);
 
120
        return -1;
 
121
}
 
122
 
 
123
static void do_listen(void (*handler)(int sk))
 
124
{
 
125
        struct sockaddr_sco addr;
 
126
        struct sco_conninfo conn;
 
127
        socklen_t optlen;
 
128
        int sk, nsk;
 
129
        char ba[18];
 
130
 
 
131
        /* Create socket */
 
132
        sk = socket(PF_BLUETOOTH, SOCK_SEQPACKET, BTPROTO_SCO);
 
133
        if (sk < 0) {
 
134
                syslog(LOG_ERR, "Can't create socket: %s (%d)",
 
135
                                                        strerror(errno), errno);
 
136
                exit(1);
 
137
        }
 
138
 
 
139
        /* Bind to local address */
 
140
        memset(&addr, 0, sizeof(addr));
 
141
        addr.sco_family = AF_BLUETOOTH;
 
142
        bacpy(&addr.sco_bdaddr, &bdaddr);
 
143
 
 
144
        if (bind(sk, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
 
145
                syslog(LOG_ERR, "Can't bind socket: %s (%d)",
 
146
                                                        strerror(errno), errno);
 
147
                goto error;
 
148
        }
 
149
 
 
150
        /* Listen for connections */
 
151
        if (listen(sk, 10)) {
 
152
                syslog(LOG_ERR,"Can not listen on the socket: %s (%d)",
 
153
                                                        strerror(errno), errno);
 
154
                goto error;
 
155
        }
 
156
 
 
157
        syslog(LOG_INFO,"Waiting for connection ...");
 
158
 
 
159
        while (1) {
 
160
                memset(&addr, 0, sizeof(addr));
 
161
                optlen = sizeof(addr);
 
162
 
 
163
                nsk = accept(sk, (struct sockaddr *) &addr, &optlen);
 
164
                if (nsk < 0) {
 
165
                        syslog(LOG_ERR,"Accept failed: %s (%d)",
 
166
                                                        strerror(errno), errno);
 
167
                        goto error;
 
168
                }
 
169
                if (fork()) {
 
170
                        /* Parent */
 
171
                        close(nsk);
 
172
                        continue;
 
173
                }
 
174
                /* Child */
 
175
                close(sk);
 
176
 
 
177
                /* Get connection information */
 
178
                memset(&conn, 0, sizeof(conn));
 
179
                optlen = sizeof(conn);
 
180
 
 
181
                if (getsockopt(nsk, SOL_SCO, SCO_CONNINFO, &conn, &optlen) < 0) {
 
182
                        syslog(LOG_ERR, "Can't get SCO connection information: %s (%d)",
 
183
                                                        strerror(errno), errno);
 
184
                        close(nsk);
 
185
                        goto error;
 
186
                }
 
187
 
 
188
                ba2str(&addr.sco_bdaddr, ba);
 
189
                syslog(LOG_INFO, "Connect from %s [handle %d, class 0x%02x%02x%02x]",
 
190
                        ba, conn.hci_handle,
 
191
                        conn.dev_class[2], conn.dev_class[1], conn.dev_class[0]);
 
192
 
 
193
                handler(nsk);
 
194
 
 
195
                syslog(LOG_INFO, "Disconnect");
 
196
                exit(0);
 
197
        }
 
198
 
 
199
        return;
 
200
 
 
201
error:
 
202
        close(sk);
 
203
        exit(1);
 
204
}
 
205
 
 
206
static void dump_mode(int sk)
 
207
{
 
208
        int len;
 
209
 
 
210
        syslog(LOG_INFO,"Receiving ...");
 
211
        while ((len = read(sk, buf, data_size)) > 0)
 
212
                syslog(LOG_INFO, "Recevied %d bytes", len);
 
213
}
 
214
 
 
215
static void recv_mode(int sk)
 
216
{
 
217
        struct timeval tv_beg,tv_end,tv_diff;
 
218
        long total;
 
219
        uint32_t seq;
 
220
 
 
221
        syslog(LOG_INFO, "Receiving ...");
 
222
 
 
223
        seq = 0;
 
224
        while (1) {
 
225
                gettimeofday(&tv_beg, NULL);
 
226
                total = 0;
 
227
                while (total < data_size) {
 
228
                        int r;
 
229
                        if ((r = recv(sk, buf, data_size, 0)) <= 0) {
 
230
                                if (r < 0)
 
231
                                        syslog(LOG_ERR, "Read failed: %s (%d)",
 
232
                                                        strerror(errno), errno);
 
233
                                return; 
 
234
                        }
 
235
                        total += r;
 
236
                }
 
237
                gettimeofday(&tv_end, NULL);
 
238
 
 
239
                timersub(&tv_end, &tv_beg, &tv_diff);
 
240
 
 
241
                syslog(LOG_INFO,"%ld bytes in %.2fm speed %.2f kb", total,
 
242
                        tv2fl(tv_diff) / 60.0,
 
243
                        (float)( total / tv2fl(tv_diff) ) / 1024.0 );
 
244
        }
 
245
}
 
246
 
 
247
static void send_mode(char *svr)
 
248
{
 
249
        struct sco_options so;
 
250
        socklen_t len;
 
251
        uint32_t seq;
 
252
        int i, sk;
 
253
 
 
254
        if ((sk = do_connect(svr)) < 0) {
 
255
                syslog(LOG_ERR, "Can't connect to the server: %s (%d)",
 
256
                                                        strerror(errno), errno);
 
257
                exit(1);
 
258
        }
 
259
 
 
260
        len = sizeof(so);
 
261
        if (getsockopt(sk, SOL_SCO, SCO_OPTIONS, &so, &len) < 0) {
 
262
                syslog(LOG_ERR, "Can't get SCO options: %s (%d)",
 
263
                                                        strerror(errno), errno);
 
264
                exit(1);
 
265
        }
 
266
 
 
267
        syslog(LOG_INFO,"Sending ...");
 
268
 
 
269
        for (i = 6; i < so.mtu; i++)
 
270
                buf[i] = 0x7f;
 
271
 
 
272
        seq = 0;
 
273
        while (1) {
 
274
                *(uint32_t *) buf = htobl(seq);
 
275
                *(uint16_t *) (buf + 4) = htobs(data_size);
 
276
                seq++;
 
277
 
 
278
                if (send(sk, buf, so.mtu, 0) <= 0) {
 
279
                        syslog(LOG_ERR, "Send failed: %s (%d)",
 
280
                                                        strerror(errno), errno);
 
281
                        exit(1);
 
282
                }
 
283
 
 
284
                usleep(1);
 
285
        }
 
286
}
 
287
 
 
288
static void reconnect_mode(char *svr)
 
289
{
 
290
        while (1) {
 
291
                int sk;
 
292
 
 
293
                if ((sk = do_connect(svr)) < 0) {
 
294
                        syslog(LOG_ERR, "Can't connect to the server: %s (%d)",
 
295
                                                        strerror(errno), errno);
 
296
                        exit(1);
 
297
                }
 
298
 
 
299
                close(sk);
 
300
 
 
301
                sleep(5);
 
302
        }
 
303
}
 
304
 
 
305
static void multy_connect_mode(char *svr)
 
306
{
 
307
        while (1) {
 
308
                int i, sk;
 
309
 
 
310
                for (i = 0; i < 10; i++){
 
311
                        if (fork())
 
312
                                continue;
 
313
 
 
314
                        /* Child */
 
315
                        sk = do_connect(svr);
 
316
                        if (sk < 0) {
 
317
                                syslog(LOG_ERR, "Can't connect to the server: %s (%d)",
 
318
                                                        strerror(errno), errno);
 
319
                        }
 
320
                        close(sk);
 
321
                        exit(0);
 
322
                }
 
323
 
 
324
                sleep(19);
 
325
        }
 
326
}
 
327
 
 
328
static void usage(void)
 
329
{
 
330
        printf("scotest - SCO testing\n"
 
331
                "Usage:\n");
 
332
        printf("\tscotest <mode> [-b bytes] [bd_addr]\n");
 
333
        printf("Modes:\n"
 
334
                "\t-d dump (server)\n"
 
335
                "\t-c reconnect (client)\n"
 
336
                "\t-m multiple connects (client)\n"
 
337
                "\t-r receive (server)\n"
 
338
                "\t-s connect and send (client)\n"
 
339
                "\t-n connect and be silent (client)\n");
 
340
}
 
341
 
 
342
int main(int argc ,char *argv[])
 
343
{
 
344
        struct sigaction sa;
 
345
        int opt, sk, mode = RECV;
 
346
 
 
347
        while ((opt=getopt(argc,argv,"rdscmnb:")) != EOF) {
 
348
                switch(opt) {
 
349
                case 'r':
 
350
                        mode = RECV;
 
351
                        break;
 
352
 
 
353
                case 's':
 
354
                        mode = SEND;
 
355
                        break;
 
356
 
 
357
                case 'd':
 
358
                        mode = DUMP;
 
359
                        break;
 
360
 
 
361
                case 'c':
 
362
                        mode = RECONNECT;
 
363
                        break;
 
364
 
 
365
                case 'm':
 
366
                        mode = MULTY;
 
367
                        break;
 
368
 
 
369
                case 'n':
 
370
                        mode = CONNECT;
 
371
                        break;
 
372
 
 
373
                case 'b':
 
374
                        data_size = atoi(optarg);
 
375
                        break;
 
376
 
 
377
                default:
 
378
                        usage();
 
379
                        exit(1);
 
380
                }
 
381
        }
 
382
 
 
383
        if (!(argc - optind) && (mode != RECV && mode != DUMP)) {
 
384
                usage();
 
385
                exit(1);
 
386
        }
 
387
 
 
388
        if (!(buf = malloc(data_size))) {
 
389
                perror("Can't allocate data buffer");
 
390
                exit(1);
 
391
        }
 
392
 
 
393
        memset(&sa, 0, sizeof(sa));
 
394
        sa.sa_handler = SIG_IGN;
 
395
        sa.sa_flags   = SA_NOCLDSTOP;
 
396
        sigaction(SIGCHLD, &sa, NULL);
 
397
 
 
398
        openlog("scotest", LOG_PERROR | LOG_PID, LOG_LOCAL0);
 
399
 
 
400
        switch( mode ){
 
401
                case RECV:
 
402
                        do_listen(recv_mode);
 
403
                        break;
 
404
 
 
405
                case DUMP:
 
406
                        do_listen(dump_mode);
 
407
                        break;
 
408
 
 
409
                case SEND:
 
410
                        send_mode(argv[optind]);
 
411
                        break;
 
412
 
 
413
                case RECONNECT:
 
414
                        reconnect_mode(argv[optind]);
 
415
                        break;
 
416
 
 
417
                case MULTY:
 
418
                        multy_connect_mode(argv[optind]);
 
419
                        break;
 
420
 
 
421
                case CONNECT:
 
422
                        sk = do_connect(argv[optind]);
 
423
                        if (sk < 0)
 
424
                                exit(1);
 
425
                        dump_mode(sk);
 
426
                        break;
 
427
        }
 
428
 
 
429
        syslog(LOG_INFO, "Exit");
 
430
 
 
431
        closelog();
 
432
 
 
433
        return 0;
 
434
}