日期:2014-05-16  浏览次数:20748 次

关于提高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);