Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Detect socket closed by peer in the transport layer, so that such things don't polute...
[simgrid.git] / src / gras / Transport / transport_plugin_file.c
index a4bb759..1b55969 100644 (file)
@@ -9,32 +9,27 @@
 
 #include "portable.h"
 #include "transport_private.h"
+#include "xbt/ex.h"
 
-#if 0
-#  include <errno.h>
-#  include <sys/time.h>
-#  include <sys/types.h>
-#  include <sys/stat.h>
-#  include <fcntl.h>
-#  include <unistd.h>
-#endif 
-
-XBT_LOG_NEW_DEFAULT_SUBCATEGORY(trp_file,transport,
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_trp_file,gras_trp,
        "Pseudo-transport to write to/read from a file");
 
 /***
  *** Prototypes 
  ***/
-void         gras_trp_file_close(gras_socket_t sd);
+void gras_trp_file_close(gras_socket_t sd);
   
-xbt_error_t gras_trp_file_chunk_send(gras_socket_t sd,
-                                     const char *data,
-                                     long int size);
-
-xbt_error_t gras_trp_file_chunk_recv(gras_socket_t sd,
-                                     char *data,
-                                     long int size);
-
+void gras_trp_file_chunk_send_raw(gras_socket_t sd,
+                                 const char *data,
+                                 unsigned long int size);
+void gras_trp_file_chunk_send(gras_socket_t sd,
+                             const char *data,
+                             unsigned long int size,
+                             int stable_ignored);
+
+int gras_trp_file_chunk_recv(gras_socket_t sd,
+                            char *data,
+                            unsigned long int size);
 
 /***
  *** Specific plugin part
@@ -53,19 +48,21 @@ typedef struct {
 /***
  *** Code
  ***/
-xbt_error_t
-gras_trp_file_setup(gras_trp_plugin_t *plug) {
+void
+gras_trp_file_setup(gras_trp_plugin_t plug) {
 
   gras_trp_file_plug_data_t *file = xbt_new(gras_trp_file_plug_data_t,1);
 
   FD_ZERO(&(file->incoming_socks));
 
   plug->socket_close = gras_trp_file_close;
-  plug->chunk_send   = gras_trp_file_chunk_send;
-  plug->chunk_recv   = gras_trp_file_chunk_recv;
-  plug->data         = (void*)file;
 
-  return no_error;
+  plug->raw_send = gras_trp_file_chunk_send_raw;
+  plug->send = gras_trp_file_chunk_send;
+
+  plug->raw_recv = plug->recv = gras_trp_file_chunk_recv;
+
+  plug->data         = (void*)file;
 }
 
 /**
@@ -75,40 +72,38 @@ gras_trp_file_setup(gras_trp_plugin_t *plug) {
  *
  * This only possible in RL, and is mainly for debugging.
  */
-xbt_error_t
-gras_socket_client_from_file(const char*path,
-                            /* OUT */ gras_socket_t *dst) {
-  xbt_error_t errcode;
-  gras_trp_plugin_t *trp;
+gras_socket_t
+gras_socket_client_from_file(const char*path) {
+  gras_socket_t res;
 
   xbt_assert0(gras_if_RL(),
               "Cannot use file as socket in the simulator");
 
-  gras_trp_socket_new(0,dst);
+  gras_trp_socket_new(0,&res);
 
-  TRY(gras_trp_plugin_get_by_name("file",&trp));
-  (*dst)->plugin=trp;
+  res->plugin=gras_trp_plugin_get_by_name("file");
 
   if (strcmp("-", path)) {
-    (*dst)->sd = open(path, O_WRONLY|O_CREAT | O_BINARY, S_IRUSR|S_IWUSR|S_IRGRP );
+    res->sd = open(path, O_WRONLY|O_CREAT | O_BINARY, S_IRUSR|S_IWUSR|S_IRGRP );
     
-    if ( (*dst)->sd < 0) {
-      RAISE2(system_error,
+    if ( res->sd < 0) {
+      THROW2(system_error,0,
             "Cannot create a client socket from file %s: %s",
             path, strerror(errno));
     }
   } else {
-    (*dst)->sd = 1; /* stdout */
+    res->sd = 1; /* stdout */
   }
 
+  res->recv_ok=0;
   DEBUG5("sock_client_from_file(%s): sd=%d in=%c out=%c accept=%c",
         path,
-        (*dst)->sd,
-        (*dst)->incoming?'y':'n', 
-        (*dst)->outgoing?'y':'n',
-        (*dst)->accepting?'y':'n');
-   
-  return no_error;
+        res->sd,
+        res->incoming?'y':'n', 
+        res->outgoing?'y':'n',
+        res->accepting?'y':'n');
+
+  return res;
 }
 
 /**
@@ -118,40 +113,37 @@ gras_socket_client_from_file(const char*path,
  *
  * This only possible in RL, and is mainly for debugging.
  */
-xbt_error_t
-gras_socket_server_from_file(const char*path,
-                            /* OUT */ gras_socket_t *dst) {
-  xbt_error_t errcode;
-  gras_trp_plugin_t *trp;
+gras_socket_t gras_socket_server_from_file(const char*path) {
+  gras_socket_t res;
 
   xbt_assert0(gras_if_RL(),
               "Cannot use file as socket in the simulator");
 
-  gras_trp_socket_new(1,dst);
+  gras_trp_socket_new(1,&res);
 
-  TRY(gras_trp_plugin_get_by_name("file",&trp));
-  (*dst)->plugin=trp;
+  res->plugin=gras_trp_plugin_get_by_name("file");
 
 
   if (strcmp("-", path)) {
-    (*dst)->sd = open(path, O_RDONLY | O_BINARY);
+    res->sd = open(path, O_RDONLY | O_BINARY);
 
-    if ( (*dst)->sd < 0) {
-      RAISE2(system_error,
+    if ( res->sd < 0) {
+      THROW2(system_error,0,
             "Cannot create a server socket from file %s: %s",
             path, strerror(errno));
     }
   } else {
-    (*dst)->sd = 0; /* stdin */
+    res->sd = 0; /* stdin */
   }
 
   DEBUG4("sd=%d in=%c out=%c accept=%c",
-        (*dst)->sd,
-        (*dst)->incoming?'y':'n', 
-        (*dst)->outgoing?'y':'n',
-        (*dst)->accepting?'y':'n');
+        res->sd,
+        res->incoming?'y':'n', 
+        res->outgoing?'y':'n',
+        res->accepting?'y':'n');
 
-  return no_error;
+  res->recv_ok=0;
+  return res;
 }
 
 void gras_trp_file_close(gras_socket_t sock){
@@ -183,10 +175,17 @@ void gras_trp_file_close(gras_socket_t sock){
  *
  * Send data on a file pseudo-socket
  */
-xbt_error_t 
+void
 gras_trp_file_chunk_send(gras_socket_t sock,
                         const char *data,
-                        long int size) {
+                        unsigned long int size,
+                        int stable_ignored) {
+  gras_trp_file_chunk_send_raw(sock,data,size);
+}
+void
+gras_trp_file_chunk_send_raw(gras_socket_t sock,
+                            const char *data,
+                            unsigned long int size) {
   
   xbt_assert0(sock->outgoing, "Cannot write on client file socket");
   xbt_assert0(size >= 0, "Cannot send a negative amount of data");
@@ -198,7 +197,7 @@ gras_trp_file_chunk_send(gras_socket_t sock,
     status = write(sock->sd, data, (long int)size);
     
     if (status == -1) {
-      RAISE4(system_error,"write(%d,%p,%d) failed: %s",
+      THROW4(system_error,0,"write(%d,%p,%d) failed: %s",
             sock->sd, data, (int)size,
             strerror(errno));
     }
@@ -207,46 +206,45 @@ gras_trp_file_chunk_send(gras_socket_t sock,
       size  -= status;
       data  += status;
     } else {
-      RAISE0(system_error,"file descriptor closed");
+      THROW0(system_error,0,"file descriptor closed");
     }
   }
-
-  return no_error;
 }
 /**
  * gras_trp_file_chunk_recv:
  *
  * Receive data on a file pseudo-socket.
  */
-xbt_error_t 
+int
 gras_trp_file_chunk_recv(gras_socket_t sock,
-                       char *data,
-                       long int size) {
+                        char *data,
+                        unsigned long int size) {
+
+  int got = 0;
 
   xbt_assert0(sock, "Cannot recv on an NULL socket");
   xbt_assert0(sock->incoming, "Cannot recv on client file socket");
   xbt_assert0(size >= 0, "Cannot receive a negative amount of data");
-  
+
   while (size) {
     int status = 0;
     
-    status = read(sock->sd, data, (long int)size);
-    DEBUG3("read(%d, %p, %ld);", sock->sd, data, size);
+    status = read(sock->sd, data+got, (long int)size);
+    DEBUG3("read(%d, %p, %ld);", sock->sd, data+got, size);
     
     if (status == -1) {
-      RAISE4(system_error,"read(%d,%p,%d) failed: %s",
-            sock->sd, data, (int)size,
+      THROW4(system_error,0,"read(%d,%p,%d) failed: %s",
+            sock->sd, data+got, (int)size,
             strerror(errno));
     }
     
     if (status) {
-      size  -= status;
-      data  += status;
+      size    -= status;
+      got    += status;
     } else {
-      RAISE0(system_error,"file descriptor closed");
+      THROW0(system_error,0,"file descriptor closed");
     }
   }
-  
-  return no_error;
+  return got;
 }