X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/ae66e43b95b26467c1cb9df271e83f51d3d7147b..1a2592601e29b027872580cbfc364194f8b01ba6:/src/gras/Transport/transport_plugin_file.c

diff --git a/src/gras/Transport/transport_plugin_file.c b/src/gras/Transport/transport_plugin_file.c
index a09864ed16..4139999450 100644
--- a/src/gras/Transport/transport_plugin_file.c
+++ b/src/gras/Transport/transport_plugin_file.c
@@ -1,8 +1,7 @@
-/* $Id$ */
-
 /* File transport - send/receive a bunch of bytes from a file               */
 
-/* Copyright (c) 2004 Martin Quinson. All rights reserved.                  */
+/* Copyright (c) 2004, 2005, 2006, 2007, 2009, 2010. The SimGrid Team.
+ * All rights reserved.                                                     */
 
 /* This program is free software; you can redistribute it and/or modify it
  * under the terms of the license (GNU LGPL) which comes with this package. */
@@ -10,26 +9,24 @@
 #include "portable.h"
 #include "gras/Transport/transport_private.h"
 #include "xbt/ex.h"
+#include "gras/Msg/msg_interface.h"     /* gras_msg_listener_awake */
 
-XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_trp_file,gras_trp,
-	"Pseudo-transport to write to/read from a file");
+XBT_LOG_NEW_DEFAULT_SUBCATEGORY(gras_trp_file, gras_trp,
+                                "Pseudo-transport to write to/read from a file");
 
 /***
- *** Prototypes 
+ *** Prototypes
  ***/
 void gras_trp_file_close(gras_socket_t sd);
-  
+
 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);
+                                  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);
+                             char *data, unsigned long int size);
 
 /***
  *** Specific plugin part
@@ -43,18 +40,41 @@ typedef struct {
  *** Specific socket part
  ***/
 
-
+/***
+ *** Info about who's speaking
+ ***/
+static int gras_trp_file_my_port(gras_socket_t s) {
+  THROW_UNIMPLEMENTED;
+}
+static int gras_trp_file_peer_port(gras_socket_t s) {
+  THROW_UNIMPLEMENTED;
+}
+static char* gras_trp_file_peer_name(gras_socket_t s) {
+  THROW_UNIMPLEMENTED;
+}
+static char* gras_trp_file_peer_proc(gras_socket_t s) {
+  THROW_UNIMPLEMENTED;
+}
+static void gras_trp_file_peer_proc_set(gras_socket_t s,char *name) {
+  THROW_UNIMPLEMENTED;
+}
 
 /***
  *** Code
  ***/
-void
-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);
+  gras_trp_file_plug_data_t *file = xbt_new(gras_trp_file_plug_data_t, 1);
 
   FD_ZERO(&(file->incoming_socks));
 
+  plug->my_port = gras_trp_file_my_port;
+  plug->peer_port = gras_trp_file_peer_port;
+  plug->peer_name = gras_trp_file_peer_name;
+  plug->peer_proc = gras_trp_file_peer_proc;
+  plug->peer_proc_set = gras_trp_file_peer_proc_set;
+
   plug->socket_close = gras_trp_file_close;
 
   plug->raw_send = gras_trp_file_chunk_send_raw;
@@ -62,7 +82,7 @@ gras_trp_file_setup(gras_trp_plugin_t plug) {
 
   plug->raw_recv = plug->recv = gras_trp_file_chunk_recv;
 
-  plug->data         = (void*)file;
+  plug->data = (void *) file;
 }
 
 /**
@@ -72,37 +92,38 @@ gras_trp_file_setup(gras_trp_plugin_t plug) {
  *
  * This only possible in RL, and is mainly for debugging.
  */
-gras_socket_t
-gras_socket_client_from_file(const char*path) {
+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");
+  xbt_assert0(gras_if_RL(), "Cannot use file as socket in the simulator");
 
-  gras_trp_socket_new(0,&res);
+  gras_trp_socket_new(0, &res);
 
-  res->plugin=gras_trp_plugin_get_by_name("file");
+  res->plugin = gras_trp_plugin_get_by_name("file");
 
   if (strcmp("-", path)) {
-    res->sd = open(path, O_WRONLY|O_CREAT | O_BINARY, S_IRUSR|S_IWUSR|S_IRGRP );
-    
-    if ( res->sd < 0) {
-      THROW2(system_error,0,
-	     "Cannot create a client socket from file %s: %s",
-	     path, strerror(errno));
+    res->sd =
+        open(path, O_TRUNC | O_WRONLY | O_CREAT | O_BINARY,
+             S_IRUSR | S_IWUSR | S_IRGRP);
+
+    if (res->sd < 0) {
+      THROW2(system_error, 0,
+             "Cannot create a client socket from file %s: %s",
+             path, strerror(errno));
     }
   } else {
-    res->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,
-	 res->sd,
-	 res->incoming?'y':'n', 
-	 res->outgoing?'y':'n',
-	 res->accepting?'y':'n');
+         path,
+         res->sd,
+         res->incoming ? 'y' : 'n',
+         res->outgoing ? 'y' : 'n', res->accepting ? 'y' : 'n');
 
+  xbt_dynar_push(((gras_trp_procdata_t)
+                  gras_libdata_by_id(gras_trp_libdata_id))->sockets, &res);
   return res;
 }
 
@@ -113,44 +134,47 @@ gras_socket_client_from_file(const char*path) {
  *
  * This only possible in RL, and is mainly for debugging.
  */
-gras_socket_t gras_socket_server_from_file(const char*path) {
+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");
+  xbt_assert0(gras_if_RL(), "Cannot use file as socket in the simulator");
 
-  gras_trp_socket_new(1,&res);
+  gras_trp_socket_new(1, &res);
 
-  res->plugin=gras_trp_plugin_get_by_name("file");
+  res->plugin = gras_trp_plugin_get_by_name("file");
 
 
   if (strcmp("-", path)) {
     res->sd = open(path, O_RDONLY | O_BINARY);
 
-    if ( res->sd < 0) {
-      THROW2(system_error,0,
-	     "Cannot create a server socket from file %s: %s",
-	     path, strerror(errno));
+    if (res->sd < 0) {
+      THROW2(system_error, 0,
+             "Cannot create a server socket from file %s: %s",
+             path, strerror(errno));
     }
   } else {
-    res->sd = 0; /* stdin */
+    res->sd = 0;                /* stdin */
   }
 
   DEBUG4("sd=%d in=%c out=%c accept=%c",
-	 res->sd,
-	 res->incoming?'y':'n', 
-	 res->outgoing?'y':'n',
-	 res->accepting?'y':'n');
+         res->sd,
+         res->incoming ? 'y' : 'n',
+         res->outgoing ? 'y' : 'n', res->accepting ? 'y' : 'n');
 
-  res->recv_ok=0;
+  xbt_dynar_push(((gras_trp_procdata_t)
+                  gras_libdata_by_id(gras_trp_libdata_id))->sockets, &res);
+  gras_msg_listener_awake();
   return res;
 }
 
-void gras_trp_file_close(gras_socket_t sock){
+void gras_trp_file_close(gras_socket_t sock)
+{
   gras_trp_file_plug_data_t *data;
-  
-  if (!sock) return; /* close only once */
-  data=sock->plugin->data;
+
+  if (!sock)
+    return;                     /* close only once */
+  data = sock->plugin->data;
 
   if (sock->sd == 0) {
     DEBUG0("Do not close stdin");
@@ -163,9 +187,8 @@ void gras_trp_file_close(gras_socket_t sock){
     FD_CLR(sock->sd, &(data->incoming_socks));
 
     /* close the socket */
-    if(close(sock->sd) < 0) {
-      WARN2("error while closing file %d: %s", 
-	       sock->sd, strerror(errno));
+    if (close(sock->sd) < 0) {
+      WARN2("error while closing file %d: %s", sock->sd, strerror(errno));
     }
   }
 }
@@ -177,39 +200,40 @@ void gras_trp_file_close(gras_socket_t sock){
  */
 void
 gras_trp_file_chunk_send(gras_socket_t sock,
-			 const char *data,
-			 unsigned long int size,
-			 int stable_ignored) {
-  gras_trp_file_chunk_send_raw(sock,data,size);
+                         const char *data,
+                         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) {
-  
+                             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");
 
   while (size) {
     int status = 0;
-    
-    DEBUG3("write(%d, %p, %ld);", sock->sd, data, (long int)size);
-    status = write(sock->sd, data, (long int)size);
-    
+
+    DEBUG3("write(%d, %p, %ld);", sock->sd, data, (long int) size);
+    status = write(sock->sd, data, (long int) size);
+
     if (status == -1) {
-      THROW4(system_error,0,"write(%d,%p,%d) failed: %s",
-	     sock->sd, data, (int)size,
-	     strerror(errno));
+      THROW4(system_error, 0, "write(%d,%p,%d) failed: %s",
+             sock->sd, data, (int) size, strerror(errno));
     }
-    
+
     if (status) {
-      size  -= status;
-      data  += status;
+      size -= status;
+      data += status;
     } else {
-      THROW0(system_error,0,"file descriptor closed");
+      THROW0(system_error, 0, "file descriptor closed");
     }
   }
 }
+
 /**
  * gras_trp_file_chunk_recv:
  *
@@ -217,8 +241,8 @@ gras_trp_file_chunk_send_raw(gras_socket_t sock,
  */
 int
 gras_trp_file_chunk_recv(gras_socket_t sock,
-			 char *data,
-			 unsigned long int size) {
+                         char *data, unsigned long int size)
+{
 
   int got = 0;
 
@@ -226,25 +250,31 @@ gras_trp_file_chunk_recv(gras_socket_t sock,
   xbt_assert0(sock->incoming, "Cannot recv on client file socket");
   xbt_assert0(size >= 0, "Cannot receive a negative amount of data");
 
+  if (sock->recvd) {
+    data[0] = sock->recvd_val;
+    sock->recvd = 0;
+    got++;
+    size--;
+  }
+
   while (size) {
     int status = 0;
-    
-    status = read(sock->sd, data+got, (long int)size);
-    DEBUG3("read(%d, %p, %ld);", sock->sd, data+got, size);
-    
-    if (status == -1) {
-      THROW4(system_error,0,"read(%d,%p,%d) failed: %s",
-	     sock->sd, data+got, (int)size,
-	     strerror(errno));
+
+    status = read(sock->sd, data + got, (long int) size);
+    DEBUG3("read(%d, %p, %ld);", sock->sd, data + got, size);
+
+    if (status < 0) {
+      THROW4(system_error, 0, "read(%d,%p,%d) failed: %s",
+             sock->sd, data + got, (int) size, strerror(errno));
     }
-    
+
     if (status) {
-      size    -= status;
-      got    += status;
+      size -= status;
+      got += status;
     } else {
-      THROW0(system_error,0,"file descriptor closed");
+      THROW1(system_error, errno, "file descriptor closed after %d bytes",
+             got);
     }
   }
   return got;
 }
-