2005-11-14 Carl Worth <cworth@cworth.org>

[ttt] / src / ttt-socket.c

diff --git a/src/ttt-socket.c b/src/ttt-socket.c
index 765cab1104d7c568b22758a76e6c7c1a508289c8..199535f300e200d0970341ea6177e53cf35bbf70 100644 (file)
--- a/src/ttt-socket.c
+++ b/src/ttt-socket.c
@@ -21,7 +21,7 @@
 
 #include "ttt-socket.h"
 
-#include <netinet/ip.h>
+#include <netinet/in.h>
 #include <netdb.h>
 
 /* Initialize a TCP/IP address structure with the given host and port.
@@ -146,11 +146,45 @@ ttt_socket_create_server (const char *host, const char *port)
 
 /* Exported: see ttt-socket.h for documentation. */
 void
-ttt_socket_accept (int listen_socket, ttt_socket_dispatch_func_t dispatch)
+ttt_socket_accept (int                          listen_socket,
+                  ttt_socket_accept_func_t      accept,
+                  void                         *closure)
 {
     int connected_socket;
 
     connected_socket = _wait_for_connection (listen_socket);
 
-    (dispatch) (connected_socket);
+    (accept) (closure, connected_socket);
+}
+
+/* Exported: see ttt-socket.h for documentation. */
+void
+ttt_socket_read (int    socket,
+                void   *buf,
+                size_t  count)
+{
+    int bytes_read;
+    char *cbuf = buf;
+
+    while (count) {
+       bytes_read = xread (socket, cbuf, count);
+       cbuf += bytes_read;
+       count -= bytes_read;
+    }
+}
+
+/* Exported: see ttt-socket.h for documentation. */
+void
+ttt_socket_write (int           socket,
+                 const void    *buf,
+                 size_t         count)
+{
+    int written;
+    const char *cbuf = buf;
+
+    while (count) {
+       written = xwrite (socket, cbuf, count);
+       cbuf += written;
+       count -= written;
+    }
 }