linux非阻塞connect（续）

加入Epoll监听后的代码：

#include <sys/socket.h>
#include <sys/types.h>
#include <arpa/inet.h>

#include <stdio.h>
#include <errno.h>
#include <fcntl.h>
#include <unistd.h>

#include <iostream>
#include <fstream>
#include <string>
#include <cstring>
#include <cstdlib>

#include <openssl/ssl.h>
#include <openssl/err.h>
#include <sys/epoll.h>

#define READ
#define WRITE

#define OUT(str) do{cout<<str<<endl;}while(0)
#define TEST do{printf("line:%d\n",__LINE__);}while(0)

using namespace std;

static int EpollCreate(){

int epfd;

epfd = epoll_create(1024);

if(epfd== -1){
return -1;
}else{
return epfd;
}
}

static bool EpollAdd(int epfd,int *fd){

struct epoll_event ev;
int ret;

ev.events = EPOLLIN | EPOLLOUT;
ev.data.ptr=fd;

ret=epoll_ctl(epfd, EPOLL_CTL_ADD, *fd, &ev);
if(ret==-1){
return false;
}

return true;
}

static bool EpollDel(int epfd, int fd){

struct epoll_event ev;
int ret;

ev.data.ptr=NULL;
ret=epoll_ctl(epfd, EPOLL_CTL_DEL, fd, &ev);
if(ret==-1){
return false;
}

return true;
}

static void Check(int fd){

int ret,error=1;
socklen_t len;
len=sizeof(error);
ret=getsockopt(fd,SOL_SOCKET,SO_ERROR,&error,&len);
if(error == 0){
OUT("SUCESS");
}else{
OUT("FAILED");
}
return;

}
static void EpollRun(int epfd,int timeout){

const int nevents = 1024;
struct epoll_event evs[nevents];
int *fd;
int n = epoll_wait(epfd, evs, nevents, timeout);
for(int i=0;i<n;i++){

fd=(int *)evs[i].data.ptr;

if(evs[i].events & EPOLLIN){

#ifdef READ
OUT("Read event note!");
OUT(*fd);
Check(*fd);
OUT("\n\n");
#endif

}
if(evs[i].events & EPOLLOUT){

#ifdef WRITE
OUT("Write event note!");
OUT(*fd);
Check(*fd);
OUT("\n\n");
#endif

}
}
}

static int CreateScok(const char * ip){

struct sockaddr_in serv_addr;
serv_addr.sin_family=AF_INET;
serv_addr.sin_port=htons(443);
serv_addr.sin_addr.s_addr=inet_addr(ip);
bzero(&(serv_addr.sin_zero),8);

SSL_CTX *ctx;
SSL *ssl;

SSL_library_init();
OpenSSL_add_all_algorithms();
SSL_load_error_strings();
ctx= SSL_CTX_new(SSLv23_client_method());
if(ctx == NULL) {
return -1;
}

int sockfd;
if((sockfd=socket(AF_INET,SOCK_STREAM,0))==-1){

OUT("socket create error");

return -1;
}

int flags=fcntl(sockfd,F_GETFL,0);
fcntl(sockfd,F_SETFL,flags|O_NONBLOCK);

int n=connect(sockfd,(struct sockaddr *)&serv_addr, sizeof(struct sockaddr));
sleep(2);

if(n<0){

if(errno !=EINPROGRESS){
perror(strerror(errno));
return -1;
}else{
OUT(ip);
OUT("EINPROGRESS");
}
}else{
OUT(ip);
}

ssl= SSL_new(ctx);
if(SSL_set_fd(ssl, sockfd) == 0){
return -1;
}
SSL_set_connect_state(ssl);

if(n==0){

OUT("connect sucess immediately");
return sockfd;
}

return sockfd;

}

int main(){

ifstream ipin("./iplist.txt");

string line;

int epfd=EpollCreate();

while(getline(ipin,line)){

int sock=CreateScok(line.c_str());

if(sock==-1){

OUT("connect error");
return 0;

}else{

EpollAdd(epfd,&sock);

}

EpollRun(epfd,100);
//close(sock);
}

return 0;
}