长话短说了 。。TCP 连接成功后, 连接的 socket 变成可写状态,那么我们就监听 EPOLLOUT 事件,来判断是不是连接成功了。但是我们需要先调用 connect 但是调用 connect 的时候会阻塞,好吧 还需要先把 socket 设置成非阻塞,这样调用完 connect 里面在线程中进入对 epoll 事件的查询。对于 connect 失败的情况,返回的事件是 EPOLLERR .简单的测试代码:
#include <iostream>
#include <sys/types.h>
#include <stdio.h>
#include <stdlib.h>
#include <sys/stat.h>
#include <sys/fcntl.h>
#include <unistd.h>
#include <sys/epoll.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <pthread.h>
#include <sys/ioctl.h>
#include <errno.h>
#include <string.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
using namespace std;
typedef void (* PF_CALLBACK)(int fd);
class CEpoll
{
public:
CEpoll(PF_CALLBACK pfOnRead,PF_CALLBACK pfOnWrite,PF_CALLBACK pfOnError)
{
m_epoll = epoll_create(100);//
if(m_epoll == -1)
{
throw "error create epoll ~";
}
m_OnRead = pfOnRead;
m_OnWrite = pfOnWrite;
m_OnError = pfOnError;
}
~CEpoll()
{
if(m_epoll >= 0)
{
close(m_epoll);
}
}
public:
bool AddFdOnWrite(int fd)
{
struct epoll_event ev;
ev.data.fd = fd;
//ev.data.ptr = "i am sincoder";//
ev.events = EPOLLOUT|EPOLLET;
int ret = epoll_ctl(m_epoll,EPOLL_CTL_ADD,fd,&ev);
if(ret == -1)
{
return false;
}
return true;
}
bool AddFdOnRead(int fd)
{
struct epoll_event ev;
ev.data.fd = fd;
//ev.data.ptr = "i am sincoder";//
ev.events = EPOLLIN|EPOLLET;
int ret =epoll_ctl(m_epoll,EPOLL_CTL_ADD,fd,&ev);
if(ret == -1)
{
return false;
}
return true;
}
bool ProcessMsg()
{
int i;
int nr_events = 0;
int ret;
char buff[1024];
struct epoll_event events[64];
printf("enter wait !\n");
nr_events = epoll_wait (m_epoll,events,64,-1);
printf("out wait%d\n",nr_events);
if (nr_events < 0)
{
perror("epoll_wait");
return false;
}
for (i = 0; i < nr_events; i++)
{
printf("event=%ld on fd=%d \n",events[i].events, events[i].data.fd);
if(events[i].events & EPOLLIN )
{
m_OnRead(events[i].data.fd);
}
if(events[i].events & EPOLLOUT )
{
m_OnWrite(events[i].data.fd);
}
if(events[i].events & EPOLLERR )
{
m_OnError(events[i].data.fd);
}
}
return true;
}
private:
int m_epoll;
PF_CALLBACK m_OnRead;
PF_CALLBACK m_OnWrite;
PF_CALLBACK m_OnError;
};
void OnRead(int fd);
int client_socket;
CEpoll *g_epoll = NULL;
void OnRead(int fd)
{
printf("enter call back ~\n");
if(fd == client_socket){
}else{
char buff[1024];
int ret = recv(fd,buff,sizeof(buff),0);
if(ret <= 0)
{
printf("close fd %d \n",fd);
close(fd);
}
else{
printf("recv %d bytes from fd %d \n",ret,fd);
}
}
}
void OnWrite(int fd)
{
printf("enter OnWrite \n");
//看看 socket 是不是链接成功了
int result;
socklen_t result_len = sizeof(result);
if (getsockopt(fd, SOL_SOCKET, SO_ERROR, &result, &result_len) < 0) {
// error, fail somehow, close socket
close(fd);
return ;
}
if (result != 0) {
// connection failed; error code is in 'result'
printf("connect failed ~!! \n");
close(fd);
return ;
}
printf("connect OK ~!! \n");
//将 fd 设置成非阻塞 并重新加入到 epoll 中
}
void OnError(int fd)
{
if(fd == client_socket)
{
//connect error ~!
printf("connect Error ~! \n");
close(fd);
return;
}
}
int main(int argc,char **argv) {
if(argc < 2 )
{
printf("usage : <ip> <port> \n");
return 0;
}
struct sockaddr_in serv_addr;
client_socket = socket(AF_INET,SOCK_STREAM,IPPROTO_TCP);
if(client_socket == -1)
{
perror("socket():");
return -1;
}
try {
g_epoll = new CEpoll(OnRead,OnWrite ,OnError);
}catch(...)
{
close(client_socket);
return -1;
}
printf("get socket fd %d", client_socket);
bzero((char *) &serv_addr, sizeof(serv_addr));
serv_addr.sin_family = AF_INET;
serv_addr.sin_addr.s_addr = inet_addr(argv[1]) ;
serv_addr.sin_port = htons(atoi(argv[2]));
//设置 connect 的 socket 为非阻塞
long on = 1L;
if (ioctl(client_socket, (int)FIONBIO, (char *)&on))
{
printf("ioctl FIONBIO call failed\n");
}
g_epoll->AddFdOnWrite(client_socket);
if (connect(client_socket, (struct sockaddr *) &serv_addr, sizeof(serv_addr)) < 0) {
if(errno != EINPROGRESS)
{
printf("%s", "ERROR connecting");
}
//close(client_socket);
//throw "can not connect to remote";
}
while(g_epoll->ProcessMsg());
return 0;
}