如何使用 Libevent 检测客户端连接到服务器

Posted

技术标签:

【中文标题】如何使用 Libevent 检测客户端连接到服务器【英文标题】:How to detect a client gets connected to the server using Libevent 【发布时间】:2015-12-05 23:28:36 【问题描述】:

我正在编写使用libevent 设置本地主机服务器和本地主机客户端的程序。我需要知道客户端何时连接到服务器。 Libevent为事件回调提供了六个事件信号,分别是BEV_EVENT_READINGBEV_EVENT_WRITINGBEV_EVENT_ERRORBEV_EVENT_TIMEOUTBEV_EVENT_EOFBEV_EVENT_CONNECTED。基本上我希望客户端进入事件循环并继续等待服务器在线。如果我只是在没有服务器的情况下在线运行客户端程序,我有时会同时获得BEV_EVENT_READINGBEV_EVENT_CONNECTED 标志。 BEV_EVENT_READING 的错误是 connection refused,这是有道理的,因为服务器处于脱机状态。但是,在这种情况下,标志BEV_EVENT_CONNECTED 也已设置,即使我调度它,程序也会立即离开。下面是代码片段

address.sin_family = AF_INET;
address.sin_addr.s_addr = inet_addr("127.0.0.1");
address.sin_port = htons(7777);
buffev_ptr = bufferevent_socket_new(_evbase_ptr, -1, BEV_OPT_CLOSE_ON_FREE);

bufferevent_setcb(buffev_ptr, myread, mywrite, mysignal, NULL);
bufferevent_enable(buffev_ptr, EV_READ | EV_WRITE);

bufferevent_socket_connect(buffev_ptr, (sockaddr_pt)&address, sizeof(address));

那么对于mysignal回调,究竟应该写什么来处理客户端连接的情况呢?

void_t Client::mysignal(buffev_pt bev, short flag, void_pt user_data) 

  // An event occurred during a read operation on the bufferevent.
  if(flag & BEV_EVENT_READING) 
  

  // An event occurred during a write operation on the bufferevent.
  if(flag & BEV_EVENT_WRITING) 
  

  // An error occurred during a bufferevent operation.
  if(flag & BEV_EVENT_ERROR) 
  

  // A timeout expired on the bufferevent.
  if(flag & BEV_EVENT_TIMEOUT) 
  

  // We got an end-of-line indication on the bufferevent.
  if(flag & BEV_EVENT_EOF) 
  

  // We finished a requested connection on the bufferevent.
  if(flag & BEV_EVENT_CONNECTED) 
  

如果我在没有服务器联机的情况下运行客户端程序,BEV_EVENT_CONNECTEDBEV_EVENT_ERROR 可能会同时发生。我假设 BEV_EVENT_CONNECTED 只会在客户端连接到服务器时设置,但似乎没有。

【问题讨论】:

我认为您需要先发布相关代码,然后才能有人为您提供答案 谢谢你,克雷格。我发布了代码片段。如果你能看一看,将不胜感激。 我现在正在查看 libevent 的源代码,我有一些想法。您能否发布更多代码?具体来说,您在致电bufferevent_socket_connect 后所做的事情。你打电话给event_base_dispatch吗?另外,您在什么操作系统上运行(例如 WinX、BSD、linux)? cmets 意味着一个或多个操作系统的事件排序行为略有不同。 是的。我调用 event_base_dispatch 并在 linux 上运行。谢谢 看看这个例子:github.com/nitrogenlogic/cliserver/blob/master/cliserver.c,尤其是setup_connection(....)cmd_read(....) 【参考方案1】:

这是一个很好的示例页面:http://www.wangafu.net/~nickm/libevent-book/Ref6_bufferevent.html

一些事情:bufferevent_socket_connect 可以立即失败 [旁注:套接字是非阻塞的]。那没有正确处理。 可以同时获得 CONNECT 和 ERROR。我在您的代码中添加了一些内容并对其进行了注释[请原谅无偿的样式清理]。

您的代码只是处理事件而没有跟踪任何“状态”信息,因此我还添加了一个“状态”结构,您可以使用它来确定如何处理传入事件,以确定连接是否发生以及是否可用。远非完美,但它可能会有所帮助:

// connection state control
struct privdata 
    u32 priv_evseen;                    // events previously seen
    struct trace_buffer *priv_trace;    // trace buffer to aid debug (e.g.)
    ...
;

int
start_connection(...)

    struct privdata *data;
    int err;

    address.sin_family = AF_INET;
    address.sin_addr.s_addr = inet_addr("127.0.0.1");
    address.sin_port = htons(7777);
    buffev_ptr = bufferevent_socket_new(_evbase_ptr, -1, BEV_OPT_CLOSE_ON_FREE);

    data = calloc(1,sizeof(struct privdata));

    bufferevent_setcb(buffev_ptr, myread, mywrite, mysignal, data);
    bufferevent_enable(buffev_ptr, EV_READ | EV_WRITE);

    // NOTE: this can fail immediately
    err = bufferevent_socket_connect(buffev_ptr,
        (sockaddr_pt) &address,sizeof(address));

    // NOTE: if it fails above, the connection attempt will _not_ be retried by
    // libevent -- we _must_ tear down and recreate from scratch (i.e. libevent
    // has already released the socket it tried to connect with)
    //
    // we'll have to decide how/when/if we want to reenter this function to
    // try again (e.g. server is offline and we requeue externally to call us
    // [say] every 5 minutes, with message "server offline -- will retry in
    // 5 minutes")
    if (err < 0) 
        err = errno;

        event_base_free(_evbase_ptr);
        free(data);

        errno = err;
        return -1;
    

    event_base_dispatch(base);

    return 0;


void_t
Client::mysignal(buffev_pt bev, short flag, void_pt user_data)

    struct privdata *priv;

    priv = user_data;

    // NOTE: it _is_ possible to get an event that has _both_ CONNNECTED and
    // ERROR because when the connection occurs, libevent issues a getsockopt
    // to check for errors
    //
    // we can use our private data to determine if an ERROR occurs at the
    // connection point (e.g. priv_evseen does _not_ have CONNECTED set) or
    // later during normal socket operation

    do 
        // wait for connection
        // NOTE: this should also handle other errors like timeout
        if ((priv->priv_evseen & CONNECTED) == 0) 
            switch (flag & (ERROR | CONNECTED)) 
            case CONNECTED: // got clean connection -- puppy is happy!
                break;

            case ERROR: // got error first -- CONNECTED event unlikely
                break;

            case (CONNECTED | ERROR):  // connected with error -- how bad is it?
                break;

            case 0:  // huh? -- the only excuse ...
                if (flag & TIMEOUT)
                    ...
                break;
            
            break;
        

        // events that occur during normal operation ...

        // An error occurred during a bufferevent operation.
        if (flag & ERROR) 
        

        // A timeout expired on the bufferevent.
        if (flag & TIMEOUT) 
        

        // We got an end-of-line indication on the bufferevent.
        if (flag & EOF) 
        

        // We finished a requested connection on the bufferevent.
        // NOTE: this should _never_ happen now
        if (flag & CONNECTED) 
        

        // An event occurred during a read operation on the bufferevent.
        if (flag & READING) 
        

        // An event occurred during a write operation on the bufferevent.
        if (flag & WRITING) 
        
     while (0);

    // remember types we've seen before
    priv->priv_seen |= flag;


以下是一些相关的 libevent 来源:

int
bufferevent_socket_connect(struct bufferevent *bev, struct sockaddr *sa, int socklen)

    struct bufferevent_private *bufev_p = EVUTIL_UPCAST(bev, struct bufferevent_private, bev);

    evutil_socket_t fd;
    int r = 0;
    int result = -1;
    int ownfd = 0;

    _bufferevent_incref_and_lock(bev);

    if (!bufev_p)
        goto done;

    fd = bufferevent_getfd(bev);
    if (fd < 0) 
        if (!sa)
            goto done;
        fd = socket(sa->sa_family, SOCK_STREAM, 0);
        if (fd < 0)
            goto done;
        if (evutil_make_socket_nonblocking(fd) < 0)
            goto done;
        ownfd = 1;
    
    if (sa) 
#ifdef WIN32
        if (bufferevent_async_can_connect(bev)) 
            bufferevent_setfd(bev, fd);
            r = bufferevent_async_connect(bev, fd, sa, socklen);
            if (r < 0)
                goto freesock;
            bufev_p->connecting = 1;
            result = 0;
            goto done;
        
        else
#endif
            r = evutil_socket_connect(&fd, sa, socklen);
        if (r < 0)
            goto freesock;
    
#ifdef WIN32
    /* ConnectEx() isn't always around, even when IOCP is enabled. Here, we borrow the socket object's write handler to fall back on a non-blocking connect() when ConnectEx() is unavailable. */
    if (BEV_IS_ASYNC(bev)) 
        event_assign(&bev->ev_write, bev->ev_base, fd, EV_WRITE | EV_PERSIST, bufferevent_writecb, bev);
    
#endif
    bufferevent_setfd(bev, fd);
    if (r == 0) 
        if (!be_socket_enable(bev, EV_WRITE)) 
            bufev_p->connecting = 1;
            result = 0;
            goto done;
        
    
    else if (r == 1) 
        /* The connect succeeded already. How very BSD of it. */
        result = 0;
        bufev_p->connecting = 1;
        event_active(&bev->ev_write, EV_WRITE, 1);
    
    else 
        /* The connect failed already.  How very BSD of it. */
        bufev_p->connection_refused = 1;
        bufev_p->connecting = 1;
        result = 0;
        event_active(&bev->ev_write, EV_WRITE, 1);
    

    goto done;

  freesock:
    _bufferevent_run_eventcb(bev, BEV_EVENT_ERROR);
    if (ownfd)
        evutil_closesocket(fd);
    /* do something about the error? */
  done:
    _bufferevent_decref_and_unlock(bev);
    return result;


/* XXX we should use an enum here. */
/* 2 for connection refused, 1 for connected, 0 for not yet, -1 for error. */
int
evutil_socket_connect(evutil_socket_t * fd_ptr, struct sockaddr *sa, int socklen)

    int made_fd = 0;

    if (*fd_ptr < 0) 
        if ((*fd_ptr = socket(sa->sa_family, SOCK_STREAM, 0)) < 0)
            goto err;
        made_fd = 1;
        if (evutil_make_socket_nonblocking(*fd_ptr) < 0) 
            goto err;
        
    

    if (connect(*fd_ptr, sa, socklen) < 0) 
        int e = evutil_socket_geterror(*fd_ptr);

        if (EVUTIL_ERR_CONNECT_RETRIABLE(e))
            return 0;
        if (EVUTIL_ERR_CONNECT_REFUSED(e))
            return 2;
        goto err;
    
    else 
        return 1;
    

  err:
    if (made_fd) 
        evutil_closesocket(*fd_ptr);
        *fd_ptr = -1;
    
    return -1;


/* Check whether a socket on which we called connect() is done
   connecting. Return 1 for connected, 0 for not yet, -1 for error.  In the
   error case, set the current socket errno to the error that happened during
   the connect operation. */
int
evutil_socket_finished_connecting(evutil_socket_t fd)

    int e;
    ev_socklen_t elen = sizeof(e);

    if (getsockopt(fd, SOL_SOCKET, SO_ERROR, (void *) &e, &elen) < 0)
        return -1;

    if (e) 
        if (EVUTIL_ERR_CONNECT_RETRIABLE(e))
            return 0;
        EVUTIL_SET_SOCKET_ERROR(e);
        return -1;
    

    return 1;


struct bufferevent *
bufferevent_socket_new(struct event_base *base, evutil_socket_t fd, int options)

    struct bufferevent_private *bufev_p;
    struct bufferevent *bufev;

#ifdef WIN32
    if (base && event_base_get_iocp(base))
        return bufferevent_async_new(base, fd, options);
#endif

    if ((bufev_p = mm_calloc(1, sizeof(struct bufferevent_private))) == NULL)
        return NULL;

    if (bufferevent_init_common(bufev_p, base, &bufferevent_ops_socket, options) < 0) 
        mm_free(bufev_p);
        return NULL;
    
    bufev = &bufev_p->bev;
    evbuffer_set_flags(bufev->output, EVBUFFER_FLAG_DRAINS_TO_FD);

    event_assign(&bufev->ev_read, bufev->ev_base, fd, EV_READ | EV_PERSIST, bufferevent_readcb, bufev);
    event_assign(&bufev->ev_write, bufev->ev_base, fd, EV_WRITE | EV_PERSIST, bufferevent_writecb, bufev);

    evbuffer_add_cb(bufev->output, bufferevent_socket_outbuf_cb, bufev);

    evbuffer_freeze(bufev->input, 0);
    evbuffer_freeze(bufev->output, 1);

    return bufev;


static void
bufferevent_writecb(evutil_socket_t fd, short event, void *arg)

    struct bufferevent *bufev = arg;
    struct bufferevent_private *bufev_p = EVUTIL_UPCAST(bufev, struct bufferevent_private, bev);
    int res = 0;
    short what = BEV_EVENT_WRITING;
    int connected = 0;
    ev_ssize_t atmost = -1;

    _bufferevent_incref_and_lock(bufev);

    if (event == EV_TIMEOUT) 
        /* Note that we only check for event==EV_TIMEOUT. If event==EV_TIMEOUT|EV_WRITE, we can safely ignore the timeout, since a read has occurred */
        what |= BEV_EVENT_TIMEOUT;
        goto error;
    
    if (bufev_p->connecting) 
        int c = evutil_socket_finished_connecting(fd);

        /* we need to fake the error if the connection was refused immediately - usually connection to localhost on BSD */
        if (bufev_p->connection_refused) 
            bufev_p->connection_refused = 0;
            c = -1;
        

        if (c == 0)
            goto done;

        bufev_p->connecting = 0;
        if (c < 0) 
            event_del(&bufev->ev_write);
            event_del(&bufev->ev_read);
            _bufferevent_run_eventcb(bufev, BEV_EVENT_ERROR);
            goto done;
        
        else 
            connected = 1;
#ifdef WIN32
            if (BEV_IS_ASYNC(bufev)) 
                event_del(&bufev->ev_write);
                bufferevent_async_set_connected(bufev);
                _bufferevent_run_eventcb(bufev, BEV_EVENT_CONNECTED);
                goto done;
            
#endif
            _bufferevent_run_eventcb(bufev, BEV_EVENT_CONNECTED);
            if (!(bufev->enabled & EV_WRITE) || bufev_p->write_suspended) 
                event_del(&bufev->ev_write);
                goto done;
            
        
    

    atmost = _bufferevent_get_write_max(bufev_p);

    if (bufev_p->write_suspended)
        goto done;

    if (evbuffer_get_length(bufev->output)) 
        evbuffer_unfreeze(bufev->output, 1);
        res = evbuffer_write_atmost(bufev->output, fd, atmost);
        evbuffer_freeze(bufev->output, 1);
        if (res == -1) 
            int err = evutil_socket_geterror(fd);

            if (EVUTIL_ERR_RW_RETRIABLE(err))
                goto reschedule;
            what |= BEV_EVENT_ERROR;
        
        else if (res == 0) 
            /* eof case XXXX Actually, a 0 on write doesn't indicate an EOF. An ECONNRESET might be more typical. */
            what |= BEV_EVENT_EOF;
        
        if (res <= 0)
            goto error;

        _bufferevent_decrement_write_buckets(bufev_p, res);
    

    if (evbuffer_get_length(bufev->output) == 0) 
        event_del(&bufev->ev_write);
    

    /*
     * Invoke the user callback if our buffer is drained or below the
     * low watermark.
     */
    if ((res || !connected) && evbuffer_get_length(bufev->output) <= bufev->wm_write.low) 
        _bufferevent_run_writecb(bufev);
    

    goto done;

  reschedule:
    if (evbuffer_get_length(bufev->output) == 0) 
        event_del(&bufev->ev_write);
    
    goto done;

  error:
    bufferevent_disable(bufev, EV_WRITE);
    _bufferevent_run_eventcb(bufev, what);

  done:
    _bufferevent_decref_and_unlock(bufev);

【讨论】:

以上是关于如何使用 Libevent 检测客户端连接到服务器的主要内容,如果未能解决你的问题,请参考以下文章

如何以编程方式通过 ipad 检测并连接到 3G 网络

QTcpSocket false 连接到 QTcpServer

如何检测 JMS 主题连接丢失

在 XMPP 中检测使用相同帐户的其他客户端

如何检测套接字连接何时丢失?

当计算机连接到 *** 时,有没有办法检测计算机的真实外部 IP?