- 爱易网页
-
Linux
- 关于拔高socket效率,select+多进程/多线程 付源码
日期:2014-05-16 浏览次数:20756 次
关于提高socket效率,select+多进程/多线程 付源码
hi,
小弟刚玩socket,日前完成了一个单个线程+select 的程序,大致功能就是可以将一个服务器A发送过来的内容转发到另一个服务器B同时将另一台服务器B发送过来的内容转发到这个服务器A。测试发现:
1.单向传输1G文件,速率有8MB(百兆网口),但是双向同时传输文件,速率只有4MB
2.希望能将程序改成多进程的或者多线程的来提高速率(即吞吐量)
源码如下,小弟尝试在connect之后fork了2个子程序来实现(即一个进程只负责A-B,另一个只负责B-A)的操作,但是发现传文件的时候传不完就程序就断了,下面是我单个线程+select 的程序,有没高手指点一下如何修改???
#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <errno.h>
#include <time.h>
#include <sys/types.h>
#include <sys/ioctl.h>
#include <string.h>
#include <stdarg.h>
#include <netdb.h>
#define SERVPORT 8090
#define MAXDATASIZE 1024
int do_con_n(char *address, int port)
{
int s;
struct sockaddr_in sin;
fd_set set;
struct timeval timeo = {15, 0}; //time ou struct
socklen_t len = sizeof(timeo);
int retval;
int error;
int code;
s=socket(AF_INET, SOCK_STREAM, 0);
// x=fcntl(s,F_GETFL,0); // Get socket flags
fcntl(s, F_SETFL, fcntl(s, F_GETFL) | O_NONBLOCK); //设置为非阻塞模式
memset(&sin, 0, sizeof(struct sockaddr_in));
sin.sin_family=AF_INET;
sin.sin_port=htons(port);
//sin.sin_addr.s_addr=name_resolve(address);
sin.sin_addr.s_addr = inet_addr(address);
if(!sin.sin_addr.s_addr) return(-1);
//printf("ip: %s\n",inet_ntoa(sin.sin_addr));
if(connect(s, (struct sockaddr *)&sin, sizeof(sin))!=0)
{
if (errno != EINPROGRESS)
{
printf("connect:normal network unreach!!");
return -1;
}
FD_ZERO(&set);/*将set清零使集合中不含任何fd*/
FD_SET(s,&set); /*将一个给定的文件描述符加入集合之中*/
retval = select(s + 1, NULL, &set, NULL, &timeo);
if (retval == -1)
{
printf("select");
return -1;
}
else if(retval == 0)
{
printf("timeout\n"); //这样的select等于是变成了再timeout时间内是阻塞模式,超过timeout就直接返回
printf("%d\n", time(NULL));
return 0;
}
else
{
//printf("connected--->:[%d]\n",retval);
if (FD_ISSET(s, &set))
{
error=0;
len = sizeof(error);