socket select

#define RELAY_IP	"0.0.0.0"
#define RELAY_PORT	"502"

struct RelayDeviceInfo
{
	unsigned int addr;
	short port;
	int sockfd;
	int useflag;
};

int relay_listen_sock = 0;
struct RelayDeviceInfo relaydevice[MAX_RELAY_STATION_NUM];

int relay_create_socket(void)
{
	struct addrinfo hint, *addr;
	int opt = 1;
	int ret = 0; 
	
	relay_listen_sock = socket(AF_INET, SOCK_STREAM, IPPROTO_TCP);
	if (relay_listen_sock < 0)
	{
		DBG(DBG_CRITICAL, "Create relay socket fails:%s", strerror(errno));
		return VOIP_FAIL;
	}

	memset(&addr, 0, sizeof(addr));
	hint.ai_flags = 0;
	hint.ai_family = AF_INET;
	hint.ai_socktype = SOCK_STREAM;
	hint.ai_protocol = IPPROTO_TCP;

	ret = setsockopt(relay_listen_sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt));
	if (ret==-1) {
		DBG(DBG_ERROR, "setsockopt socket fails:%s", strerror(errno));
		return VOIP_FAIL;
	}
	
	ret = getaddrinfo(RELAY_IP, RELAY_PORT, &hint, &addr);
	if (ret!=0) {
		fprintf(stderr, "getaddrinfo(): %s\n", gai_strerror(ret));
		exit(1);
	}
	
	if (bind(relay_listen_sock, addr->ai_addr, addr->ai_addrlen) == -1)
	{		
		DBG(DBG_CRITICAL, "Bind relay socket fails:%s", strerror(errno));
		close(relay_listen_sock);
		return VOIP_FAIL;		
	}
	
	freeaddrinfo(addr);

	ret = listen(relay_listen_sock, 255);
	if (ret==-1) {
		DBG(DBG_ERROR, "listen socket fails:%s", strerror(errno));
		return VOIP_FAIL;
	}
    
	return VOIP_OK;
}

void relay_accept_connect()
{
	struct sockaddr_in peer;
	socklen_t peer_len;
    int relay_sock;
    int i = 0;   
    
	peer_len = sizeof(peer);
	relay_sock = accept(relay_listen_sock, (struct sockaddr*)&peer, &peer_len);
	if (relay_sock==-1) {
		DBG(DBG_ERROR, "accept socket fails:%s", strerror(errno));
		return VOIP_FAIL;
	}
	fcntl(relay_sock, F_SETFL, O_NONBLOCK);
    DBG(DBG_CRITICAL,"Accept relay connect, addr:%s, port:%d",inet_ntoa(peer.sin_addr),ntohs(peer.sin_port));

    //add relay socket connection
    for (i = 0 ; i < MAX_RELAY_STATION_NUM; i++)
    {
        if (relaydevice[i].useflag && relaydevice[i].addr == ntohl(peer.sin_addr.s_addr))
        {
            if (relaydevice[i].sockfd > 0)
            {
                close(relaydevice[i].sockfd);
            }   
            relaydevice[i].sockfd  = relay_sock;
            relaydevice[i].addr = ntohl(peer.sin_addr.s_addr);
            relaydevice[i].port = ntohs(peer.sin_port);
            break;            
        }
    }

    if (i == MAX_RELAY_STATION_NUM)
    {
        for (i = 0; i < MAX_RELAY_STATION_NUM; i++)
        {        
            if (!relaydevice[i].useflag)
            {
                relaydevice[i].useflag = 1;
                relaydevice[i].sockfd  = relay_sock;
                relaydevice[i].addr = ntohl(peer.sin_addr.s_addr);;
                relaydevice[i].port = ntohs(peer.sin_port);
                break;
            }
        }
    }
    
    if (i == MAX_RELAY_STATION_NUM)
    {
        DBG(DBG_ERROR, "accept sockfd overflow");
    }    
}

void Relay_process_msg_thread()
{
	unsigned char read_buf[MAX_DATA_LENGTH_RELAY_INTF+1] = {'\0'};
	struct timeval next_interval;
	fd_set fd_read;
	int nfds;
	int ret;
	int peer_addr;
	short peer_port;
	struct timeval curr_time;
	struct timezone time_zone;
    int i = 0;

	memset(&relaydevice,0,sizeof(struct RelayDeviceInfo)*MAX_RELAY_STATION_NUM);
	relay_create_socket();
	
	while(1)
	{
		gettimeofday(&curr_time,&time_zone);	
		
		FD_ZERO(&fd_read);
		nfds = 0;
        
        FD_SET(relay_listen_sock, &fd_read);
        if(relay_listen_sock > nfds)                  
        {                    
            nfds = relay_listen_sock;              
        }
        
        for (i = 0; i < MAX_RELAY_STATION_NUM; i++)
        {
    		if(relaydevice[i].useflag && relaydevice[i].sockfd > 0)
    		{
    			FD_SET(relaydevice[i].sockfd, &fd_read);
    			if(relaydevice[i].sockfd > nfds)				  
    			{					 
    				nfds = relaydevice[i].sockfd; 			   
    			}
    		}
        }

		next_interval.tv_sec = 0;
		next_interval.tv_usec = 100000;  //100ms
		ret = select(nfds+1, &fd_read, NULL, NULL, &next_interval);        

		if (ret < 0)        
		{            
			continue;        
		}

        if(FD_ISSET(relay_listen_sock, &fd_read))
        {
            relay_accept_connect();
        }
	
        for (i = 0; i < MAX_RELAY_STATION_NUM; i++)
        {
    		if(FD_ISSET(relaydevice[i].sockfd, &fd_read))
    		{
    			memset(read_buf, 0, MAX_DATA_LENGTH_RELAY_INTF);
    			ret = Relay_Receive_Packet(relaydevice[i].sockfd,read_buf,MAX_DATA_LENGTH_RELAY_INTF,&peer_addr,&peer_port);
    			if (ret > 0)
    			{
    				CheakRelayData(relaydevice[i].addr,read_buf,ret);
    			}
                else
                {
                    close(relaydevice[i].sockfd);
                    memset(&relaydevice[i],0,sizeof(relaydevice[i]));
                }
    		}
        }

	}
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
SelectSocket编程中还是比较重要的,可是对于初学Socket的人来说都不太爱用Select写程序,他们只是习惯写诸如connect、accept、recv或recvfrom这样的阻塞程序(所谓阻塞方式block,顾名思义,就是进程或是线程执行到这些函数时必须等待某个事件的发生,如果事件没有发生,进程或线程就被阻塞,函数不能立即返回)。可是使用Select就可以完成非阻塞(所谓非阻塞方式non-block,就是进程或线程执行此函数时不必非要等待事件的发生,一旦执行肯定返回,以返回值的不同来反映函数的执行情况,如果事件发生则与阻塞方式相同,若事件没有发生则返回一个代码来告知事件未发生,而进程或线程继续执行,所以效率较高)方式工作的程序,它能够监视我们需要监视的文件描述符的变化情况——读写或是异常。下面详细介绍一下! Select的函数格式(我所说的是Unix系统下的伯克利socket编程,和windows下的有区别,一会儿说明): int select(int maxfdp,fd_set *readfds,fd_set *writefds,fd_set *errorfds,struct timeval *timeout); 先说明两个结构体: 第一,struct fd_set可以理解为一个集合,这个集合中存放的是文件描述符(file descriptor),即文件句柄,这可以是我们所说的普通意义的文件,当然Unix下任何设备、管道、FIFO等都是文件形式,全部包括在内,所以毫无疑问一个socket就是一个文件,socket句柄就是一个文件描述符。fd_set集合可以通过一些宏由人为来操作,比如清空集合FD_ZERO(fd_set *),将一个给定的文件描述符加入集合之中FD_SET(int ,fd_set *),将一个给定的文件描述符从集合中删除FD_CLR(int ,fd_set*),检查集合中指定的文件描述符是否可以读写FD_ISSET(int ,fd_set* )。一会儿举例说明。 第二,struct timeval是一个大家常用的结构,用来代表时间值,有两个成员,一个是秒数,另一个是毫秒数。 具体解释select的参数: int maxfdp是一个整数值,是指集合中所有文件描述符的范围,即所有文件描述符的最大值加1,不能错!在Windows中这个参数的值无所谓,可以设置不正确。 fd_set *readfds是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的读变化的,即我们关心是否可以从这些文件中读取数据了,如果这个集合中有一个文件可读,select就会返回一个大于0的值,表示有文件可读,如果没有可读的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的读变化。 fd_set *writefds是指向fd_set结构的指针,这个集合中应该包括文件描述符,我们是要监视这些文件描述符的写变化的,即我们关心是否可以向这些文件中写入数据了,如果这个集合中有一个文件可写,select就会返回一个大于0的值,表示有文件可写,如果没有可写的文件,则根据timeout参数再判断是否超时,若超出timeout的时间,select返回0,若发生错误返回负值。可以传入NULL值,表示不关心任何文件的写变化。 fd_set *errorfds同上面两个参数的意图,用来监视文件错误异常。 struct timeval* timeout是select的超时时间,这个参数至关重要,它可以使select处于三种状态,第一,若将NULL以形参传入,即不传入时间结构,就是将select置于阻塞状态,一定等到监视文件描述符集合中某个文件描述符发生变化为止;第二,若将时间值设为0秒0毫秒,就变成一个纯粹的非阻塞函数,不管文件描述符是否有变化,都立刻返回继续执行,文件无变化返回0,有变化返回一个正值;第三,timeout的值大于0,这就是等待的超时时间,即select在timeout时间内阻塞,超时时间之内有事件到来就返回了,否则在超时后不管怎样一定返回,返回值同上述。 返回值: 负值:select错误 正值:某些文件可读写或出错 0:等待超时,没有可读写或错误的文件

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值