mirror of
http://git.haproxy.org/git/haproxy.git
synced 2026-02-14 00:59:12 +02:00
2da156fe5e
Use a single tcp_connect_probe() instead of tcp_connect_write() and tcp_connect_read(). We call this one only when no data layer function have been processed, so this is a fallback to test for completion of a connection attempt. With this done, we don't have the need for any direct I/O callback anymore. The function still relies on ->write() to wake the stream interface up, so it's not finished.
67 lines
1.6 KiB
C
67 lines
1.6 KiB
C
/*
|
|
* Connection management functions
|
|
*
|
|
* Copyright 2000-2012 Willy Tarreau <w@1wt.eu>
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License
|
|
* as published by the Free Software Foundation; either version
|
|
* 2 of the License, or (at your option) any later version.
|
|
*
|
|
*/
|
|
|
|
#include <common/compat.h>
|
|
#include <common/config.h>
|
|
|
|
#include <types/connection.h>
|
|
|
|
#include <proto/proto_tcp.h>
|
|
#include <proto/stream_interface.h>
|
|
|
|
/* I/O callback for fd-based connections. It calls the read/write handlers
|
|
* provided by the connection's sock_ops, which must be valid. It returns
|
|
* FD_WAIT_*.
|
|
*/
|
|
int conn_fd_handler(int fd)
|
|
{
|
|
struct connection *conn = fdtab[fd].owner;
|
|
int ret = 0;
|
|
|
|
if (!conn)
|
|
goto leave;
|
|
|
|
if (conn->flags & CO_FL_ERROR)
|
|
goto leave;
|
|
|
|
if (conn->flags & CO_FL_SI_SEND_PROXY)
|
|
if ((ret = conn_si_send_proxy(conn, CO_FL_SI_SEND_PROXY)))
|
|
goto leave;
|
|
|
|
if (fdtab[fd].ev & (FD_POLL_IN | FD_POLL_HUP | FD_POLL_ERR))
|
|
if (!conn->data->read(fd))
|
|
ret |= FD_WAIT_READ;
|
|
|
|
if (conn->flags & CO_FL_ERROR)
|
|
goto leave;
|
|
|
|
if (fdtab[fd].ev & (FD_POLL_OUT | FD_POLL_ERR))
|
|
if (!conn->data->write(fd))
|
|
ret |= FD_WAIT_WRITE;
|
|
|
|
if (conn->flags & CO_FL_ERROR)
|
|
goto leave;
|
|
|
|
if (conn->flags & CO_FL_WAIT_L4_CONN) {
|
|
/* still waiting for a connection to establish and no data to
|
|
* send in order to probe it ? Then let's retry the connect().
|
|
*/
|
|
if (!tcp_connect_probe(fd))
|
|
ret |= FD_WAIT_WRITE;
|
|
}
|
|
|
|
leave:
|
|
/* remove the events before leaving */
|
|
fdtab[fd].ev &= ~(FD_POLL_IN | FD_POLL_OUT | FD_POLL_HUP | FD_POLL_ERR);
|
|
return ret;
|
|
}
|