创建非阻塞connect

#include <stdio.h>
#include <sys/time.h>
#include <sys/types.h>
#include <sys/file.h>
#include <unistd.h>
#include <getopt.h>
#include <stdlib.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <signal.h>
#include <time.h>
#include <string.h>
#include <sys/ioctl.h>
#include <net/if.h>
#include <errno.h>


int create_socket_conn(const char * des,int nblock,int nsec);
int main(int argc,char ** argv)
{
int fd = -1;
int nblock = 1;
char buf[20];
int n;


if (argc != 2)
{
printf("error ,please input desaddr\n");
return -1;
}


if ((fd = create_socket_conn(argv[1],nblock,10)) < 0)
{
perror("create socket error");
return -1;
}


printf("create success\n");
return 0;
}


int create_socket_conn(const char * des,int nblock,int nsec)
{
struct sockaddr_in desaddr;
int fd;
int flags;
int ret;
fd_set rset,wset;
struct timeval tval;


if ((fd = socket(AF_INET,SOCK_STREAM,0)) < 0)
return fd;


memset(&desaddr,0,sizeof(desaddr));
desaddr.sin_family = AF_INET;
desaddr.sin_port = htons(80);
if(inet_pton(AF_INET,des,&desaddr.sin_addr) <= 0)
{
close(fd);
return -1;
}


if (nblock)
{
flags = fcntl(fd,F_GETFL,0);
fcntl(fd,F_SETFL,flags | O_NONBLOCK);
}


if((ret = connect(fd,(struct sockaddr *)&desaddr,sizeof(desaddr))) < 0)
{
if (errno != EINPROGRESS)
{
close(fd);
return -1;
}
}


if (ret == 0 && !nblock)
goto done;
if (nblock)
{
FD_ZERO(&rset);
FD_SET(fd,&rset);
wset = rset;
tval.tv_sec = nsec;
tval.tv_usec = 0;
if((ret = select(fd+1,&rset,&wset,NULL,nsec?&tval:NULL)) == 0)
{
close(fd);
return -1;
}

if (FD_ISSET(fd,&rset) || FD_ISSET(fd,&wset))
{
int error;
int len = sizeof(error);
if (getsockopt(fd,SOL_SOCKET,SO_ERROR,&error,&len) < 0)
{
close(fd);
return -1;
}
}
else
{
close(fd);
return -1;
}
}


done:
if (nblock)
{
fcntl(fd,F_SETFL,flags);
}
return fd;
}