first minial version working

git-svn-id: https://svn.spreadspace.org/tcpproxy/trunk@12 e61f0598-a718-4e21-a8f0-0aadfa62ad6b

1 files changed, 7 insertions, 2 deletions

diff --git a/src/tcpproxy.c b/src/tcpproxy.c
index fce2e2c..6a55ea1 100644
--- a/src/tcpproxy.c
+++ b/src/tcpproxy.c
@@ -52,13 +52,15 @@ int main_loop(options_t* opt, listeners_t* listeners)
   int return_value = clients_init(&clients);
 
   while(!return_value) {
-    fd_set readfds;
+    fd_set readfds, writefds;
     FD_ZERO(&readfds);
+    FD_ZERO(&writefds);
     FD_SET(sig_fd, &readfds);
     int nfds = sig_fd;
     listener_read_fds(listeners, &readfds, &nfds);
     clients_read_fds(&clients, &readfds, &nfds);
-    int ret = select(nfds + 1, &readfds, NULL, NULL, NULL);
+    clients_write_fds(&clients, &writefds, &nfds);
+    int ret = select(nfds + 1, &readfds, &writefds, NULL, NULL);
     if(ret == -1 && errno != EINTR) {
       log_printf(ERROR, "select returned with error: %s", strerror(errno));
       return_value = -1;
@@ -77,6 +79,9 @@ int main_loop(options_t* opt, listeners_t* listeners)
     return_value = listener_handle_accept(listeners, &clients, &readfds);
     if(return_value) break;
 
+    return_value = clients_write(&clients, &writefds);
+    if(return_value) break;
+
     return_value = clients_read(&clients, &readfds);
   }