Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Initial structure to implement gras using simix.
[simgrid.git] / src / gras_simix / Transport / gras_simix_transport_plugin_file.c
diff --git a/src/gras_simix/Transport/gras_simix_transport_plugin_file.c b/src/gras_simix/Transport/gras_simix_transport_plugin_file.c
new file mode 100644 (file)
index 0000000..1b55969
--- /dev/null
@@ -0,0 +1,250 @@
+/* $Id$ */
+
+/* File transport - send/receive a bunch of bytes from a file               */
+
+/* Copyright (c) 2004 Martin Quinson. 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. */
+
+#include "portable.h"
+#include "transport_private.h"
+#include "xbt/ex.h"
+
+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_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
+ ***/
+
+typedef struct {
+  fd_set incoming_socks;
+} gras_trp_file_plug_data_t;
+
+/***
+ *** Specific socket part
+ ***/
+
+
+
+/***
+ *** Code
+ ***/
+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->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;
+}
+
+/**
+ * gras_socket_client_from_file:
+ *
+ * Create a client socket from a file path.
+ *
+ * This only possible in RL, and is mainly for debugging.
+ */
+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,&res);
+
+  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));
+    }
+  } else {
+    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');
+
+  return res;
+}
+
+/**
+ * gras_socket_server_from_file:
+ *
+ * Create a server socket from a file 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 res;
+
+  xbt_assert0(gras_if_RL(),
+              "Cannot use file as socket in the simulator");
+
+  gras_trp_socket_new(1,&res);
+
+  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));
+    }
+  } else {
+    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->recv_ok=0;
+  return res;
+}
+
+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->sd == 0) {
+    DEBUG0("Do not close stdin");
+  } else if (sock->sd == 1) {
+    DEBUG0("Do not close stdout");
+  } else {
+    DEBUG1("close file connection %d", sock->sd);
+
+    /* forget about the socket */
+    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));
+    }
+  }
+}
+
+/**
+ * gras_trp_file_chunk_send:
+ *
+ * Send data on a file pseudo-socket
+ */
+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);
+}
+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");
+
+  while (size) {
+    int status = 0;
+    
+    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));
+    }
+    
+    if (status) {
+      size  -= status;
+      data  += status;
+    } else {
+      THROW0(system_error,0,"file descriptor closed");
+    }
+  }
+}
+/**
+ * gras_trp_file_chunk_recv:
+ *
+ * Receive data on a file pseudo-socket.
+ */
+int
+gras_trp_file_chunk_recv(gras_socket_t sock,
+                        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+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));
+    }
+    
+    if (status) {
+      size    -= status;
+      got    += status;
+    } else {
+      THROW0(system_error,0,"file descriptor closed");
+    }
+  }
+  return got;
+}
+