one side of cleint connections works for recv

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

1 files changed, 82 insertions, 30 deletions

diff --git a/src/clients.c b/src/clients.c
index ab82e28..a4b58b3 100644
--- a/src/clients.c
+++ b/src/clients.c
@@ -32,12 +32,9 @@
 #include <unistd.h>
 #include <string.h>
 #include <stdio.h>
-/* #include <netdb.h> */
-/* #include <sys/types.h> */
-/* #include <sys/socket.h> */
-/* #include <arpa/inet.h> */
-/* #include <netinet/in.h> */
-/* #include <sys/select.h> */
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <sys/select.h>
 
 #include "clients.h"
 #include "tcp.h"
@@ -49,7 +46,8 @@ void clients_delete_element(void* e)
     return;
   
   client_t* element = (client_t*)e;
-  close(element->fd_);
+  close(element->fd_[0]);
+//  close(element->fd_[1]);
 
   free(e);
 }
@@ -64,16 +62,31 @@ void clients_clear(clients_t* list)
   slist_clear(list);
 }
 
-int clients_add(clients_t* list, int fd, tcp_endpoint_t remote_end)
+int clients_add(clients_t* list, int fd, const tcp_endpoint_t* remote_end)
 {
+
   if(!list)
     return -1;
 
-  int ret = 0;
+  client_t* element = malloc(sizeof(client_t));
+  if(!element) {
+    close(fd);
+    return -2;
+  }
 
-// TODO: connect to remote end and setup write buffers
+  element->fd_[0] = fd;
+  element->fd_[1] = 0;
+// TODO: open new socket
+//  element->fd_[1] = socket(...);
 
-  return ret;
+  if(slist_add(list, element) == NULL) {
+    close(element->fd_[0]);
+//    close(element->fd_[1]);
+    free(element);
+    return -2;
+  }
+
+  return 0;
 }
 
 void clients_remove(clients_t* list, int fd)
@@ -88,9 +101,9 @@ client_t* clients_find(clients_t* list, int fd)
 
   slist_element_t* tmp = list->first_;
   while(tmp) {
-    client_t* l = (client_t*)tmp->data_;
-    if(l && l->fd_ == fd)
-      return l;
+    client_t* c = (client_t*)tmp->data_;
+    if(c && (c->fd_[0] == fd || c->fd_[1] == fd))
+      return c;
     tmp = tmp->next_;
   }
 
@@ -104,10 +117,10 @@ void clients_print(clients_t* list)
   
   slist_element_t* tmp = list->first_;
   while(tmp) {
-    client_t* l = (client_t*)tmp->data_;
-    if(l) {
+    client_t* c = (client_t*)tmp->data_;
+    if(c) {
           // print useful info
-      printf("clients #%d: tba...\n", l->fd_);
+      printf("client %d <-> %d: tba...\n", c->fd_[0], c->fd_[1]);
     }
     tmp = tmp->next_;
   }
@@ -120,10 +133,12 @@ void clients_read_fds(clients_t* list, fd_set* set, int* max_fd)
 
   slist_element_t* tmp = list->first_;
   while(tmp) {
-    client_t* l = (client_t*)tmp->data_;
-    if(l) {
-      FD_SET(l->fd_, set);
-      *max_fd = *max_fd > l->fd_ ? *max_fd : l->fd_;
+    client_t* c = (client_t*)tmp->data_;
+    if(c) {
+      FD_SET(c->fd_[0], set);
+//      FD_SET(c->fd_[1], set);
+      *max_fd = *max_fd > c->fd_[0] ? *max_fd : c->fd_[0];
+//      *max_fd = *max_fd > c->fd_[1] ? *max_fd : c->fd_[1];
     }
     tmp = tmp->next_;
   }
@@ -135,13 +150,50 @@ void clients_write_fds(clients_t* list, fd_set* set, int* max_fd)
     return;
 
       // TODO: add all clients with pending data
-/*   slist_element_t* tmp = list->first_; */
-/*   while(tmp) { */
-/*     client_t* l = (client_t*)tmp->data_; */
-/*     if(l) { */
-/*       FD_SET(l->fd_, set); */
-/*       *max_fd = *max_fd > l->fd_ ? *max_fd : l->fd_; */
-/*     } */
-/*     tmp = tmp->next_; */
-/*   } */
+}
+
+int clients_read(clients_t* list, fd_set* set)
+{
+  if(!list)
+    return -1;
+  
+  slist_element_t* tmp = list->first_;
+  while(tmp) {
+    client_t* c = (client_t*)tmp->data_;
+    tmp = tmp->next_;
+    if(c) {
+      int in, out;
+      if(FD_ISSET(c->fd_[0], set)) {
+        in = 0;
+        out = 1;
+      }
+      else if(FD_ISSET(c->fd_[1], set)) {
+        in = 1;
+        out = 0;
+      }
+      else continue;
+
+      u_int8_t* buffer[1024];
+      int len = recv(c->fd_[in], buffer, sizeof(buffer), 0);
+      if(len < 0) {
+        log_printf(INFO, "Error on recv(): %s, removing client %d", strerror(errno), c->fd_[0]);
+        slist_remove(list, c);
+      }
+      else if(!len) {
+        log_printf(INFO, "client %d closed connection", c->fd_[0]);
+        slist_remove(list, c);
+      }       
+      else {
+        log_printf(INFO, "client %d: read %d bytes", c->fd_[0], len);
+            // TODO: add data to write buffer of l->fd_[out]
+      }
+    }
+  }
+  
+  return 0;
+}
+
+int clients_write(clients_t* list, fd_set* set)
+{
+  return 0;
 }