int main(int argc,char **argv)
{
void *txbuff,*rxbuff,*pdesc;
int sock,listener,len;
char *p;
int size = 0,ringset_idx;
int sockets_connected = 0;
int selector;
int ready_socket_count;
int i,tx_space;
int max_total_length = 0;
unsigned long received_count = 0;
unsigned long transmitted_count = 0;
struct timeval tm_out, *p_timeout = NULL;
struct ipaugenblick_fdset readfdset,writefdset, excfdset;
struct sockaddr addr;
struct sockaddr_in *in_addr = (struct sockaddr_in *)&addr;
ipaugenblick_fdzero(&readfdset);
ipaugenblick_fdzero(&writefdset);
ipaugenblick_fdzero(&excfdset);
if(ipaugenblick_app_init(argc,argv,"tcp_connect") != 0) {
printf("cannot initialize memory\n");
return 0;
}
printf("memory initialized\n");
selector = ipaugenblick_open_select();
if(selector != -1) {
printf("selector opened %d\n",selector);
}
if((sock = ipaugenblick_open_socket(AF_INET,SOCK_STREAM,selector)) < 0) {
printf("cannot open tcp client socket\n");
return 0;
}
printf("opened socket %d\n",sock);
ipaugenblick_fdset (sock, &readfdset);
ipaugenblick_fdset (sock, &writefdset);
// int bufsize = 1024*1024*1000;
// ipaugenblick_setsockopt(sock, SOL_SOCKET,SO_SNDBUFFORCE,(char *)&bufsize,sizeof(bufsize));
//ipaugenblick_setsockopt(sock, SOL_SOCKET,SO_RCVBUFFORCE,(char *)&bufsize,sizeof(bufsize));
in_addr->sin_family = AF_INET;
in_addr->sin_addr.s_addr = inet_addr("192.168.150.63");
in_addr->sin_port = htons(7777);
ipaugenblick_connect(sock,&addr, sizeof(addr));
p_timeout = &tm_out;
while(1) {
memset(&tm_out,0,sizeof(tm_out));
p_timeout = /*&tm_out*/NULL;
ready_socket_count = ipaugenblick_select(selector,&readfdset,&writefdset,&excfdset, p_timeout);
if(ready_socket_count == -1) {
continue;
}
if (ready_socket_count == 0) {
memset(&tm_out,0,sizeof(tm_out));
p_timeout = &tm_out;
continue;
}
//printf("readfd returned %d\n",readfdset.returned_idx);
for (sock = 0; sock < readfdset.returned_idx; sock++) {
if (!ipaugenblick_fdisset(ipaugenblick_fd_idx2sock(&readfdset,sock),&readfdset))
continue;
#if USE_RX
int first_seg_len = 0;
int len = 0;
p_timeout = NULL;
while(ipaugenblick_receive(ipaugenblick_fd_idx2sock(&readfdset,sock),&rxbuff,&len,&first_seg_len,&pdesc) == 0) {
int segs = 0;
void *porigdesc = pdesc;
#if 1
while(rxbuff) {
if(len > 0) {
/* do something */
/* don't release buf, release rxbuff */
}
segs++;
if(segs > 100000) {
printf("segs!!!!\n");exit(0);
}
rxbuff = ipaugenblick_get_next_buffer_segment(&pdesc,&len);
}
received_count+=segs;
#else
received_count++;
#endif
if(!(received_count%10)) {
printf("received %u transmitted_count %u\n", received_count, transmitted_count);
print_stats();
}
ipaugenblick_release_rx_buffer(porigdesc,ipaugenblick_fd_idx2sock(&readfdset,sock));
len = 0;
}
#endif
}
#if USE_TX
// printf("writefd returned %d\n",writefdset.returned_idx);
for (sock = 0; sock < writefdset.returned_idx; sock++) {
// printf("mask %x sock %d\n",writefdset.returned_mask[sock],writefdset.returned_sockets[sock]);
if (!ipaugenblick_fdisset(ipaugenblick_fd_idx2sock(&writefdset,sock),&writefdset))
continue;
p_timeout = NULL;
tx_space = ipaugenblick_get_socket_tx_space(ipaugenblick_fd_idx2sock(&writefdset,sock));
// printf("tx_space %d\n",tx_space);
#if 1
for(i = 0;i < tx_space;i++) {
txbuff = ipaugenblick_get_buffer(1448,ipaugenblick_fd_idx2sock(&writefdset,sock),&pdesc);
if(!txbuff) {
break;
}
//strcpy(txbuff,"VADIM");
if(ipaugenblick_send(ipaugenblick_fd_idx2sock(&writefdset,sock),pdesc,0,1448)) {
ipaugenblick_release_tx_buffer(pdesc);
break;
}
transmitted_count++;
if(!(transmitted_count%1000)) {
printf("received %u transmitted_count %u\n", received_count, transmitted_count);
print_stats();
}
}
#else
if(tx_space == 0)
continue;
struct data_and_descriptor bulk_bufs[tx_space];
int offsets[tx_space];
int lengths[tx_space];
if(!ipaugenblick_get_buffers_bulk(1448,ready_socket,tx_space,bulk_bufs)) {
for(i = 0;i < tx_space;i++) {
offsets[i] = 0;
lengths[i] = 1448;
}
if(ipaugenblick_send_bulk(ready_socket,bulk_bufs,offsets,lengths,tx_space)) {
for(i = 0;i < tx_space;i++)
ipaugenblick_release_tx_buffer(bulk_bufs[i].pdesc);
printf("%s %d\n",__FILE__,__LINE__);
}
else {
transmitted_count += tx_space;
/*if(!(transmitted_count%1000))*/ {
printf("transmitted %u received_count %u\n",transmitted_count, received_count);
print_stats();
}
}
}
#endif
int iter = 0;
while(ipaugenblick_socket_kick(ipaugenblick_fd_idx2sock(&writefdset,sock)) == -1) {
iter++;
if(!(iter%1000000)) {
printf("iter!\n");exit(0);
}
}
}
// ipaugenblick_socket_kick(ready_socket);
#endif
}
return 0;
}
int main(int argc,char **argv)
{
void *txbuff,*rxbuff,*pdesc;
int sock,len;
char *p;
int size = 0,ringset_idx,tx_space = 0;
int socket_connected = -1;
int i;
if(ipaugenblick_app_init(argc,argv,"tcp_listener") != 0) {
printf("cannot initialize memory\n");
return 0;
}
printf("memory initialized\n");
if((sock = ipaugenblick_open_socket(AF_INET,SOCK_STREAM,-1)) < 0) {
printf("cannot open tcp client socket\n");
return 0;
}
ipaugenblick_v4_connect_bind_socket(sock,inet_addr("192.168.150.63"),7777,0);
ipaugenblick_listen_socket(sock);
printf("listener socket opened\n");
while(1) {
if(socket_connected == -1) {
unsigned int ipaddr;
unsigned short port;
socket_connected = ipaugenblick_accept(sock,&ipaddr,&port);
if(socket_connected != -1)
printf("socket accepted\n");
}
if(socket_connected == -1) {
continue;
}
int first_seg_len = 0;
if(ipaugenblick_receive(socket_connected,&rxbuff,&len,&first_seg_len,&pdesc) == 0) {
ipaugenblick_release_rx_buffer(pdesc,socket_connected);
}
#if USE_TX
tx_space = ipaugenblick_get_socket_tx_space(socket_connected);
#if 0
for(i = 0;i < tx_space;i++) {
txbuff = ipaugenblick_get_buffer(1448,socket_connected,&pdesc);
if(!txbuff) {
break;
}
if(ipaugenblick_send(socket_connected,pdesc,0,1448)) {
ipaugenblick_release_tx_buffer(pdesc);
break;
}
}
#else
if(tx_space == 0) {
usleep(1);
continue;
}
struct data_and_descriptor bulk_bufs[tx_space];
int offsets[tx_space];
int lengths[tx_space];
if(!ipaugenblick_get_buffers_bulk(1448,socket_connected,tx_space,bulk_bufs)) {
for(i = 0;i < tx_space;i++) {
offsets[i] = 0;
lengths[i] = 1448;
}
if(ipaugenblick_send_bulk(socket_connected,bulk_bufs,offsets,lengths,tx_space)) {
for(i = 0;i < tx_space;i++)
ipaugenblick_release_tx_buffer(bulk_bufs[i].pdesc);
printf("%s %d\n",__FILE__,__LINE__);
}
else {
// printf("%s %d %d\n",__FILE__,__LINE__,tx_space);
// sent = 1;
// ipaugenblick_socket_kick(socket_connected);
}
}
#endif
ipaugenblick_socket_kick(socket_connected);
#endif
}
return 0;
}
