mirror of
https://git.kernel.org/pub/scm/bluetooth/bluez.git
synced 2024-11-25 05:04:18 +08:00
782 lines
15 KiB
C
782 lines
15 KiB
C
/*
|
|
*
|
|
* BlueZ - Bluetooth protocol stack for Linux
|
|
*
|
|
* Copyright (C) 2002-2003 Maxim Krasnyansky <maxk@qualcomm.com>
|
|
* Copyright (C) 2002-2009 Marcel Holtmann <marcel@holtmann.org>
|
|
*
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
|
|
*
|
|
*/
|
|
|
|
#ifdef HAVE_CONFIG_H
|
|
#include <config.h>
|
|
#endif
|
|
|
|
#include <stdio.h>
|
|
#include <errno.h>
|
|
#include <ctype.h>
|
|
#include <fcntl.h>
|
|
#include <unistd.h>
|
|
#include <stdlib.h>
|
|
#include <getopt.h>
|
|
#include <syslog.h>
|
|
#include <signal.h>
|
|
#include <sys/time.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/socket.h>
|
|
|
|
#include <bluetooth/bluetooth.h>
|
|
#include <bluetooth/hci.h>
|
|
#include <bluetooth/hci_lib.h>
|
|
#include <bluetooth/rfcomm.h>
|
|
#include <bluetooth/sdp.h>
|
|
#include <bluetooth/sdp_lib.h>
|
|
|
|
/* Test modes */
|
|
enum {
|
|
SEND,
|
|
RECV,
|
|
RECONNECT,
|
|
MULTY,
|
|
DUMP,
|
|
CONNECT,
|
|
CRECV,
|
|
LSEND
|
|
};
|
|
|
|
static unsigned char *buf;
|
|
|
|
/* Default data size */
|
|
static long data_size = 127;
|
|
static long num_frames = -1;
|
|
|
|
/* Default number of consecutive frames before the delay */
|
|
static int count = 1;
|
|
|
|
/* Default delay after sending count number of frames */
|
|
static unsigned long delay = 0;
|
|
|
|
/* Default addr and channel */
|
|
static bdaddr_t bdaddr;
|
|
static uint16_t uuid = 0x0000;
|
|
static uint8_t channel = 10;
|
|
|
|
static char *filename = NULL;
|
|
|
|
static int master = 0;
|
|
static int auth = 0;
|
|
static int encrypt = 0;
|
|
static int secure = 0;
|
|
static int socktype = SOCK_STREAM;
|
|
static int linger = 0;
|
|
static int timestamp = 0;
|
|
static int defer_setup = 0;
|
|
|
|
static float tv2fl(struct timeval tv)
|
|
{
|
|
return (float)tv.tv_sec + (float)(tv.tv_usec/1000000.0);
|
|
}
|
|
|
|
static uint8_t get_channel(const char *svr, uint16_t uuid)
|
|
{
|
|
sdp_session_t *sdp;
|
|
sdp_list_t *srch, *attrs, *rsp;
|
|
uuid_t svclass;
|
|
uint16_t attr;
|
|
bdaddr_t dst;
|
|
uint8_t channel = 0;
|
|
int err;
|
|
|
|
str2ba(svr, &dst);
|
|
|
|
sdp = sdp_connect(&bdaddr, &dst, SDP_RETRY_IF_BUSY);
|
|
if (!sdp)
|
|
return 0;
|
|
|
|
sdp_uuid16_create(&svclass, uuid);
|
|
srch = sdp_list_append(NULL, &svclass);
|
|
|
|
attr = SDP_ATTR_PROTO_DESC_LIST;
|
|
attrs = sdp_list_append(NULL, &attr);
|
|
|
|
err = sdp_service_search_attr_req(sdp, srch,
|
|
SDP_ATTR_REQ_INDIVIDUAL, attrs, &rsp);
|
|
if (err)
|
|
goto done;
|
|
|
|
for (; rsp; rsp = rsp->next) {
|
|
sdp_record_t *rec = (sdp_record_t *) rsp->data;
|
|
sdp_list_t *protos;
|
|
|
|
if (!sdp_get_access_protos(rec, &protos)) {
|
|
channel = sdp_get_proto_port(protos, RFCOMM_UUID);
|
|
if (channel > 0)
|
|
break;
|
|
}
|
|
}
|
|
|
|
done:
|
|
sdp_close(sdp);
|
|
|
|
return channel;
|
|
}
|
|
|
|
static int do_connect(const char *svr)
|
|
{
|
|
struct sockaddr_rc addr;
|
|
struct rfcomm_conninfo conn;
|
|
socklen_t optlen;
|
|
int sk, opt;
|
|
|
|
if (uuid != 0x0000)
|
|
channel = get_channel(svr, uuid);
|
|
|
|
if (channel == 0) {
|
|
syslog(LOG_ERR, "Can't get channel number");
|
|
return -1;
|
|
}
|
|
|
|
/* Create socket */
|
|
sk = socket(PF_BLUETOOTH, socktype, BTPROTO_RFCOMM);
|
|
if (sk < 0) {
|
|
syslog(LOG_ERR, "Can't create socket: %s (%d)",
|
|
strerror(errno), errno);
|
|
return -1;
|
|
}
|
|
|
|
/* Bind to local address */
|
|
memset(&addr, 0, sizeof(addr));
|
|
addr.rc_family = AF_BLUETOOTH;
|
|
bacpy(&addr.rc_bdaddr, &bdaddr);
|
|
|
|
if (bind(sk, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
|
|
syslog(LOG_ERR, "Can't bind socket: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
|
|
#if 0
|
|
/* Enable SO_TIMESTAMP */
|
|
if (timestamp) {
|
|
int t = 1;
|
|
|
|
if (setsockopt(sk, SOL_SOCKET, SO_TIMESTAMP, &t, sizeof(t)) < 0) {
|
|
syslog(LOG_ERR, "Can't enable SO_TIMESTAMP: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
/* Enable SO_LINGER */
|
|
if (linger) {
|
|
struct linger l = { .l_onoff = 1, .l_linger = linger };
|
|
|
|
if (setsockopt(sk, SOL_SOCKET, SO_LINGER, &l, sizeof(l)) < 0) {
|
|
syslog(LOG_ERR, "Can't enable SO_LINGER: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
/* Set link mode */
|
|
opt = 0;
|
|
if (master)
|
|
opt |= RFCOMM_LM_MASTER;
|
|
if (auth)
|
|
opt |= RFCOMM_LM_AUTH;
|
|
if (encrypt)
|
|
opt |= RFCOMM_LM_ENCRYPT;
|
|
if (secure)
|
|
opt |= RFCOMM_LM_SECURE;
|
|
|
|
if (opt && setsockopt(sk, SOL_RFCOMM, RFCOMM_LM, &opt, sizeof(opt)) < 0) {
|
|
syslog(LOG_ERR, "Can't set RFCOMM link mode: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
|
|
/* Connect to remote device */
|
|
memset(&addr, 0, sizeof(addr));
|
|
addr.rc_family = AF_BLUETOOTH;
|
|
str2ba(svr, &addr.rc_bdaddr);
|
|
addr.rc_channel = channel;
|
|
|
|
if (connect(sk, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
|
|
syslog(LOG_ERR, "Can't connect: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
|
|
/* Get connection information */
|
|
memset(&conn, 0, sizeof(conn));
|
|
optlen = sizeof(conn);
|
|
|
|
if (getsockopt(sk, SOL_RFCOMM, RFCOMM_CONNINFO, &conn, &optlen) < 0) {
|
|
syslog(LOG_ERR, "Can't get RFCOMM connection information: %s (%d)",
|
|
strerror(errno), errno);
|
|
//goto error;
|
|
}
|
|
|
|
syslog(LOG_INFO, "Connected [handle %d, class 0x%02x%02x%02x]",
|
|
conn.hci_handle,
|
|
conn.dev_class[2], conn.dev_class[1], conn.dev_class[0]);
|
|
|
|
return sk;
|
|
|
|
error:
|
|
close(sk);
|
|
return -1;
|
|
}
|
|
|
|
static void do_listen(void (*handler)(int sk))
|
|
{
|
|
struct sockaddr_rc addr;
|
|
struct rfcomm_conninfo conn;
|
|
socklen_t optlen;
|
|
int sk, nsk, opt;
|
|
char ba[18];
|
|
|
|
/* Create socket */
|
|
sk = socket(PF_BLUETOOTH, socktype, BTPROTO_RFCOMM);
|
|
if (sk < 0) {
|
|
syslog(LOG_ERR, "Can't create socket: %s (%d)",
|
|
strerror(errno), errno);
|
|
exit(1);
|
|
}
|
|
|
|
/* Bind to local address */
|
|
memset(&addr, 0, sizeof(addr));
|
|
addr.rc_family = AF_BLUETOOTH;
|
|
bacpy(&addr.rc_bdaddr, &bdaddr);
|
|
addr.rc_channel = channel;
|
|
|
|
if (bind(sk, (struct sockaddr *) &addr, sizeof(addr)) < 0) {
|
|
syslog(LOG_ERR, "Can't bind socket: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
|
|
/* Set link mode */
|
|
opt = 0;
|
|
if (master)
|
|
opt |= RFCOMM_LM_MASTER;
|
|
if (auth)
|
|
opt |= RFCOMM_LM_AUTH;
|
|
if (encrypt)
|
|
opt |= RFCOMM_LM_ENCRYPT;
|
|
if (secure)
|
|
opt |= RFCOMM_LM_SECURE;
|
|
|
|
if (opt && setsockopt(sk, SOL_RFCOMM, RFCOMM_LM, &opt, sizeof(opt)) < 0) {
|
|
syslog(LOG_ERR, "Can't set RFCOMM link mode: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
|
|
/* Enable deferred setup */
|
|
opt = defer_setup;
|
|
|
|
if (opt && setsockopt(sk, SOL_BLUETOOTH, BT_DEFER_SETUP,
|
|
&opt, sizeof(opt)) < 0) {
|
|
syslog(LOG_ERR, "Can't enable deferred setup : %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
|
|
/* Listen for connections */
|
|
if (listen(sk, 10)) {
|
|
syslog(LOG_ERR,"Can not listen on the socket: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
|
|
/* Check for socket address */
|
|
memset(&addr, 0, sizeof(addr));
|
|
optlen = sizeof(addr);
|
|
|
|
if (getsockname(sk, (struct sockaddr *) &addr, &optlen) < 0) {
|
|
syslog(LOG_ERR, "Can't get socket name: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
|
|
channel = addr.rc_channel;
|
|
|
|
syslog(LOG_INFO, "Waiting for connection on channel %d ...", channel);
|
|
|
|
while (1) {
|
|
memset(&addr, 0, sizeof(addr));
|
|
optlen = sizeof(addr);
|
|
|
|
nsk = accept(sk, (struct sockaddr *) &addr, &optlen);
|
|
if (nsk < 0) {
|
|
syslog(LOG_ERR,"Accept failed: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
if (fork()) {
|
|
/* Parent */
|
|
close(nsk);
|
|
continue;
|
|
}
|
|
/* Child */
|
|
close(sk);
|
|
|
|
/* Get connection information */
|
|
memset(&conn, 0, sizeof(conn));
|
|
optlen = sizeof(conn);
|
|
|
|
if (getsockopt(nsk, SOL_RFCOMM, RFCOMM_CONNINFO, &conn, &optlen) < 0) {
|
|
syslog(LOG_ERR, "Can't get RFCOMM connection information: %s (%d)",
|
|
strerror(errno), errno);
|
|
//close(nsk);
|
|
//goto error;
|
|
}
|
|
|
|
ba2str(&addr.rc_bdaddr, ba);
|
|
syslog(LOG_INFO, "Connect from %s [handle %d, class 0x%02x%02x%02x]",
|
|
ba, conn.hci_handle,
|
|
conn.dev_class[2], conn.dev_class[1], conn.dev_class[0]);
|
|
|
|
#if 0
|
|
/* Enable SO_TIMESTAMP */
|
|
if (timestamp) {
|
|
int t = 1;
|
|
|
|
if (setsockopt(nsk, SOL_SOCKET, SO_TIMESTAMP, &t, sizeof(t)) < 0) {
|
|
syslog(LOG_ERR, "Can't enable SO_TIMESTAMP: %s (%d)",
|
|
strerror(errno), errno);
|
|
goto error;
|
|
}
|
|
}
|
|
#endif
|
|
|
|
/* Enable SO_LINGER */
|
|
if (linger) {
|
|
struct linger l = { .l_onoff = 1, .l_linger = linger };
|
|
|
|
if (setsockopt(nsk, SOL_SOCKET, SO_LINGER, &l, sizeof(l)) < 0) {
|
|
syslog(LOG_ERR, "Can't enable SO_LINGER: %s (%d)",
|
|
strerror(errno), errno);
|
|
close(nsk);
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
/* Handle deferred setup */
|
|
if (defer_setup) {
|
|
syslog(LOG_INFO, "Waiting for %d seconds",
|
|
abs(defer_setup) - 1);
|
|
sleep(abs(defer_setup) - 1);
|
|
|
|
if (defer_setup < 0) {
|
|
close(nsk);
|
|
goto error;
|
|
}
|
|
}
|
|
|
|
handler(nsk);
|
|
|
|
syslog(LOG_INFO, "Disconnect: %m");
|
|
exit(0);
|
|
}
|
|
|
|
return;
|
|
|
|
error:
|
|
close(sk);
|
|
exit(1);
|
|
}
|
|
|
|
static void dump_mode(int sk)
|
|
{
|
|
int len;
|
|
|
|
syslog(LOG_INFO, "Receiving ...");
|
|
while ((len = read(sk, buf, data_size)) > 0)
|
|
syslog(LOG_INFO, "Recevied %d bytes", len);
|
|
}
|
|
|
|
static void recv_mode(int sk)
|
|
{
|
|
struct timeval tv_beg, tv_end, tv_diff;
|
|
char ts[30];
|
|
long total;
|
|
uint32_t seq;
|
|
|
|
syslog(LOG_INFO, "Receiving ...");
|
|
|
|
memset(ts, 0, sizeof(ts));
|
|
|
|
seq = 0;
|
|
while (1) {
|
|
gettimeofday(&tv_beg,NULL);
|
|
total = 0;
|
|
while (total < data_size) {
|
|
//uint32_t sq;
|
|
//uint16_t l;
|
|
int r;
|
|
|
|
if ((r = recv(sk, buf, data_size, 0)) < 0) {
|
|
if (r < 0)
|
|
syslog(LOG_ERR, "Read failed: %s (%d)",
|
|
strerror(errno), errno);
|
|
return;
|
|
}
|
|
|
|
if (timestamp) {
|
|
struct timeval tv;
|
|
|
|
if (ioctl(sk, SIOCGSTAMP, &tv) < 0) {
|
|
timestamp = 0;
|
|
memset(ts, 0, sizeof(ts));
|
|
} else {
|
|
sprintf(ts, "[%ld.%ld] ",
|
|
tv.tv_sec, tv.tv_usec);
|
|
}
|
|
}
|
|
|
|
#if 0
|
|
/* Check sequence */
|
|
sq = btohl(*(uint32_t *) buf);
|
|
if (seq != sq) {
|
|
syslog(LOG_INFO, "seq missmatch: %d -> %d", seq, sq);
|
|
seq = sq;
|
|
}
|
|
seq++;
|
|
|
|
/* Check length */
|
|
l = btohs(*(uint16_t *) (buf + 4));
|
|
if (r != l) {
|
|
syslog(LOG_INFO, "size missmatch: %d -> %d", r, l);
|
|
continue;
|
|
}
|
|
|
|
/* Verify data */
|
|
for (i = 6; i < r; i++) {
|
|
if (buf[i] != 0x7f)
|
|
syslog(LOG_INFO, "data missmatch: byte %d 0x%2.2x", i, buf[i]);
|
|
}
|
|
#endif
|
|
total += r;
|
|
}
|
|
gettimeofday(&tv_end,NULL);
|
|
|
|
timersub(&tv_end,&tv_beg,&tv_diff);
|
|
|
|
syslog(LOG_INFO,"%s%ld bytes in %.2f sec, %.2f kB/s", ts, total,
|
|
tv2fl(tv_diff), (float)(total / tv2fl(tv_diff) ) / 1024.0);
|
|
}
|
|
}
|
|
|
|
static void do_send(int sk)
|
|
{
|
|
uint32_t seq;
|
|
int i, fd, len;
|
|
|
|
syslog(LOG_INFO,"Sending ...");
|
|
|
|
if (filename) {
|
|
fd = open(filename, O_RDONLY);
|
|
if (fd < 0) {
|
|
syslog(LOG_ERR, "Open failed: %s (%d)",
|
|
strerror(errno), errno);
|
|
exit(1);
|
|
}
|
|
len = read(fd, buf, data_size);
|
|
send(sk, buf, len, 0);
|
|
return;
|
|
} else {
|
|
for (i = 6; i < data_size; i++)
|
|
buf[i] = 0x7f;
|
|
}
|
|
|
|
seq = 0;
|
|
while ((num_frames == -1) || (num_frames-- > 0)) {
|
|
*(uint32_t *) buf = htobl(seq);
|
|
*(uint16_t *) (buf + 4) = htobs(data_size);
|
|
seq++;
|
|
|
|
if (send(sk, buf, data_size, 0) <= 0) {
|
|
syslog(LOG_ERR, "Send failed: %s (%d)",
|
|
strerror(errno), errno);
|
|
exit(1);
|
|
}
|
|
|
|
if (num_frames && delay && count && !(seq % count))
|
|
usleep(delay);
|
|
}
|
|
}
|
|
|
|
static void send_mode(int sk)
|
|
{
|
|
do_send(sk);
|
|
|
|
syslog(LOG_INFO, "Closing channel ...");
|
|
if (shutdown(sk, SHUT_RDWR) < 0)
|
|
syslog(LOG_INFO, "Close failed: %m");
|
|
else
|
|
syslog(LOG_INFO, "Done");
|
|
}
|
|
|
|
static void reconnect_mode(char *svr)
|
|
{
|
|
while(1) {
|
|
int sk = do_connect(svr);
|
|
close(sk);
|
|
}
|
|
}
|
|
|
|
static void multi_connect_mode(int argc, char *argv[])
|
|
{
|
|
int i, n, sk;
|
|
|
|
while (1) {
|
|
for (n = 0; n < argc; n++) {
|
|
for (i = 0; i < count; i++) {
|
|
if (fork())
|
|
continue;
|
|
|
|
/* Child */
|
|
sk = do_connect(argv[n]);
|
|
usleep(500);
|
|
close(sk);
|
|
exit(0);
|
|
}
|
|
}
|
|
sleep(4);
|
|
}
|
|
}
|
|
|
|
static void usage(void)
|
|
{
|
|
printf("rctest - RFCOMM testing\n"
|
|
"Usage:\n");
|
|
printf("\trctest <mode> [options] [bdaddr]\n");
|
|
printf("Modes:\n"
|
|
"\t-r listen and receive\n"
|
|
"\t-w listen and send\n"
|
|
"\t-d listen and dump incoming data\n"
|
|
"\t-s connect and send\n"
|
|
"\t-u connect and receive\n"
|
|
"\t-n connect and be silent\n"
|
|
"\t-c connect, disconnect, connect, ...\n"
|
|
"\t-m multiple connects\n");
|
|
|
|
printf("Options:\n"
|
|
"\t[-b bytes] [-i device] [-P channel] [-U uuid]\n"
|
|
"\t[-L seconds] enabled SO_LINGER option\n"
|
|
"\t[-W seconds] enable deferred setup\n"
|
|
"\t[-B filename] use data packets from file\n"
|
|
"\t[-N num] number of frames to send\n"
|
|
"\t[-C num] send num frames before delay (default = 1)\n"
|
|
"\t[-D milliseconds] delay after sending num frames (default = 0)\n"
|
|
"\t[-A] request authentication\n"
|
|
"\t[-E] request encryption\n"
|
|
"\t[-S] secure connection\n"
|
|
"\t[-M] become master\n"
|
|
"\t[-T] enable timestamps\n");
|
|
}
|
|
|
|
int main(int argc, char *argv[])
|
|
{
|
|
struct sigaction sa;
|
|
int opt, sk, mode = RECV, need_addr = 0;
|
|
|
|
bacpy(&bdaddr, BDADDR_ANY);
|
|
|
|
while ((opt=getopt(argc,argv,"rdscuwmnb:i:P:U:B:N:MAESL:W:C:D:T")) != EOF) {
|
|
switch (opt) {
|
|
case 'r':
|
|
mode = RECV;
|
|
break;
|
|
|
|
case 's':
|
|
mode = SEND;
|
|
need_addr = 1;
|
|
break;
|
|
|
|
case 'w':
|
|
mode = LSEND;
|
|
break;
|
|
|
|
case 'u':
|
|
mode = CRECV;
|
|
need_addr = 1;
|
|
break;
|
|
|
|
case 'd':
|
|
mode = DUMP;
|
|
break;
|
|
|
|
case 'c':
|
|
mode = RECONNECT;
|
|
need_addr = 1;
|
|
break;
|
|
|
|
case 'n':
|
|
mode = CONNECT;
|
|
need_addr = 1;
|
|
break;
|
|
|
|
case 'm':
|
|
mode = MULTY;
|
|
need_addr = 1;
|
|
break;
|
|
|
|
case 'b':
|
|
data_size = atoi(optarg);
|
|
break;
|
|
|
|
case 'i':
|
|
if (!strncasecmp(optarg, "hci", 3))
|
|
hci_devba(atoi(optarg + 3), &bdaddr);
|
|
else
|
|
str2ba(optarg, &bdaddr);
|
|
break;
|
|
|
|
case 'P':
|
|
channel = atoi(optarg);
|
|
break;
|
|
|
|
case 'U':
|
|
if (!strcasecmp(optarg, "spp"))
|
|
uuid = SERIAL_PORT_SVCLASS_ID;
|
|
else if (!strncasecmp(optarg, "0x", 2))
|
|
uuid = strtoul(optarg + 2, NULL, 16);
|
|
else
|
|
uuid = atoi(optarg);
|
|
break;
|
|
|
|
case 'M':
|
|
master = 1;
|
|
break;
|
|
|
|
case 'A':
|
|
auth = 1;
|
|
break;
|
|
|
|
case 'E':
|
|
encrypt = 1;
|
|
break;
|
|
|
|
case 'S':
|
|
secure = 1;
|
|
break;
|
|
|
|
case 'L':
|
|
linger = atoi(optarg);
|
|
break;
|
|
|
|
case 'W':
|
|
defer_setup = atoi(optarg);
|
|
break;
|
|
|
|
case 'B':
|
|
filename = strdup(optarg);
|
|
break;
|
|
|
|
case 'N':
|
|
num_frames = atoi(optarg);
|
|
break;
|
|
|
|
case 'C':
|
|
count = atoi(optarg);
|
|
break;
|
|
|
|
case 'D':
|
|
delay = atoi(optarg) * 1000;
|
|
break;
|
|
|
|
case 'T':
|
|
timestamp = 1;
|
|
break;
|
|
|
|
default:
|
|
usage();
|
|
exit(1);
|
|
}
|
|
}
|
|
|
|
if (need_addr && !(argc - optind)) {
|
|
usage();
|
|
exit(1);
|
|
}
|
|
|
|
if (!(buf = malloc(data_size))) {
|
|
perror("Can't allocate data buffer");
|
|
exit(1);
|
|
}
|
|
|
|
memset(&sa, 0, sizeof(sa));
|
|
sa.sa_handler = SIG_IGN;
|
|
sa.sa_flags = SA_NOCLDSTOP;
|
|
sigaction(SIGCHLD, &sa, NULL);
|
|
|
|
openlog("rctest", LOG_PERROR | LOG_PID, LOG_LOCAL0);
|
|
|
|
switch (mode) {
|
|
case RECV:
|
|
do_listen(recv_mode);
|
|
break;
|
|
|
|
case CRECV:
|
|
sk = do_connect(argv[optind]);
|
|
if (sk < 0)
|
|
exit(1);
|
|
recv_mode(sk);
|
|
break;
|
|
|
|
case DUMP:
|
|
do_listen(dump_mode);
|
|
break;
|
|
|
|
case SEND:
|
|
sk = do_connect(argv[optind]);
|
|
if (sk < 0)
|
|
exit(1);
|
|
send_mode(sk);
|
|
break;
|
|
|
|
case LSEND:
|
|
do_listen(send_mode);
|
|
break;
|
|
|
|
case RECONNECT:
|
|
reconnect_mode(argv[optind]);
|
|
break;
|
|
|
|
case MULTY:
|
|
multi_connect_mode(argc - optind, argv + optind);
|
|
break;
|
|
|
|
case CONNECT:
|
|
sk = do_connect(argv[optind]);
|
|
if (sk < 0)
|
|
exit(1);
|
|
dump_mode(sk);
|
|
break;
|
|
}
|
|
|
|
syslog(LOG_INFO, "Exit");
|
|
|
|
closelog();
|
|
|
|
return 0;
|
|
}
|