/**
* Start a new ttcp transfer. It should be possible to call this function
* multiple times in order to get multiple ttcp streams. done_cb() will be
* invoked upon completion.
*
*/
int
ttcp_start(struct ip_addr addr, uint16_t port, void *opaque,
ttcp_done_cb_t *done_cb,
int mode, uint16_t nbuf, uint16_t buflen, int udp, int verbose)
{
struct ttcp* ttcp;
int status;
if (mode != TTCP_MODE_TRANSMIT && mode != TTCP_MODE_RECEIVE) {
printk("TTCP [-]: invalid mode\n");
return -1;
}
if (nbuf == 0) {
printk("TTCP [-]: invalid nbuf\n");
return -1;
}
if (buflen == 0) {
printk("TTCP [-]: invalid buflen\n");
return -1;
}
ttcp = calloc(1, sizeof(struct ttcp));
if (ttcp == NULL) {
printk("TTCP [-]: could not allocate memory for ttcp\n");
return -1;
}
ttcp->addr = addr;
ttcp->port = port;
ttcp->nbuf = nbuf;
ttcp->mode = mode;
ttcp->left = nbuf * buflen;
ttcp->done_cb = done_cb;
ttcp->opaque = opaque;
ttcp->udp = udp;
ttcp->verbose = verbose;
ttcp->buflen = buflen;
printk("TTCP [%p]: nbuf=%d, buflen=%d, port=%d (%s/%s)\n",
ttcp, ttcp->nbuf, ttcp->buflen, ttcp->port,
ttcp->udp ? "udp" : "tcp",
ttcp->mode == TTCP_MODE_TRANSMIT ? "tx" : "rx");
if (ttcp->udp)
status = udp_start(ttcp);
else
status = tcp_start(ttcp);
if (status)
goto fail;
return 0;
fail:
ttcp_destroy(ttcp);
return -1;
}
static int start_udp(Cfg *cfg)
{
static int started = 0;
if (started) return 0;
udp_start(cfg);
start_wap(cfg);
started = 1;
return 0;
}
int main(int argc, char **argv)
{
int opt;
Cfg *cfg = NULL;
gwlib_init();
server_port = CONNECTION_ORIENTED_PORT;
while ((opt = getopt(argc, argv, "v:meti:p:")) != EOF) {
switch (opt) {
case 'v':
log_set_output_level(atoi(optarg));
break;
case 'm':
verbose += 1;
break;
case 'e':
verbose += 2;
break;
case 't':
verbose += 4;
break;
case 'h':
help();
exit(0);
case 'i':
interface_name = octstr_create(optarg);
break;
case 'p':
server_port = atoi(optarg);
break;
case '?':
default:
error(0, "Invalid option %c", opt);
help();
panic(0, "Stopping.");
}
}
if (optind == argc) {
help();
exit(0);
}
/* get the host or IP of the real wap gw to forward the WDP packets */
wapgw = octstr_create(argv[optind]);
/* if no interface was given use 0.0.0.0 */
if (!interface_name)
interface_name = octstr_create("*");
report_versions("wapproxy");
/* initialize main inbound and outbound queues */
outgoing_wdp = gwlist_create();
incoming_wdp = gwlist_create();
flow_threads = gwlist_create();
outgoing_wdp_counter = counter_create();
incoming_wdp_counter = counter_create();
/* start the main UDP listening threads */
udp_start(cfg);
gwlist_add_producer(outgoing_wdp);
debug("bb", 0, "starting WDP routers");
if (gwthread_create(service_router, NULL) == -1)
panic(0, "Failed to start a new thread for inbound WDP routing");
if (gwthread_create(wdp_router, NULL) == -1)
panic(0, "Failed to start a new thread for outbound WDP routing");
gwthread_sleep(5.0); /* give time to threads to register themselves */
while (gwlist_consume(flow_threads) != NULL)
;
udp_shutdown();
gwlist_remove_producer(outgoing_wdp);
gwlist_destroy(flow_threads, NULL);
gwlist_destroy(incoming_wdp, NULL);
gwlist_destroy(outgoing_wdp, NULL);
counter_destroy(incoming_wdp_counter);
counter_destroy(outgoing_wdp_counter);
octstr_destroy(interface_name);
octstr_destroy(wapgw);
gwlib_shutdown();
return 0;
}
/**
* Start a new ttcp transfer. It should be possible to call this function
* multiple times in order to get multiple ttcp streams. done_cb() will be
* invoked upon completion.
*
*/
int ard_tcp_start(struct ip_addr addr, uint16_t port, void *opaque,
ard_tcp_done_cb_t *done_cb, int mode, uint16_t nbuf, uint16_t buflen,
int udp, int verbose, uint8_t sock, void** _ttcp) {
struct ttcp* ttcp;
int status;
if (mode != TTCP_MODE_TRANSMIT && mode != TTCP_MODE_RECEIVE) {
WARN("TTCP [-]: invalid mode\n");
return -1;
}
if (nbuf == 0) {
WARN("TTCP [-]: invalid nbuf\n");
return -1;
}
if (buflen == 0) {
WARN("TTCP [-]: invalid buflen\n");
return -1;
}
ttcp = calloc(1, sizeof(struct ttcp));
if (ttcp == NULL) {
WARN("TTCP [-]: could not allocate memory for ttcp\n");
return -1;
}
ttcp->addr = addr;
ttcp->port = port;
ttcp->nbuf = nbuf;
ttcp->mode = mode;
ttcp->left = nbuf * buflen;
ttcp->done_cb = done_cb;
ttcp->opaque = opaque;
ttcp->udp = udp;
ttcp->verbose = verbose;
ttcp->buflen = buflen;
if (ttcp->udp)
status = udp_start(ttcp);
else
status = atcp_start(ttcp);
if (status) {
WARN("Start server FAILED!\n");
goto fail;
}
INFO_TCP("TTCP [%p-%p]: nbuf=%d, buflen=%d, port=%d (%s/%s)\n", ttcp,
ttcp->tpcb, ttcp->nbuf, ttcp->buflen, ttcp->port, ttcp->udp ? "udp"
: "tcp", ttcp->mode == TTCP_MODE_TRANSMIT ? "tx" : "rx");
*_ttcp = (void*) ttcp;
ttcp->sock = sock;
ttcp->buff_sent = 1;
return 0;
//fail: ard_tcp_abort(ttcp);
fail: ard_tcp_destroy(ttcp);
return -1;
}
