Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
- Reduce the number of system headers loaded, overload some more system
[simgrid.git] / src / gras / Transport / transport.c
index f799f4c..5d440fc 100644 (file)
@@ -8,21 +8,15 @@
 /* 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 <time.h>       /* time() */
+#include "gras/Transport/transport_private.h"
 
-#include "Transport/transport_private.h"
-
-GRAS_LOG_NEW_DEFAULT_SUBCATEGORY(transport,GRAS);
+GRAS_LOG_NEW_DEFAULT_SUBCATEGORY(transport,gras,"Conveying bytes over the network");
 
 static gras_dict_t  *_gras_trp_plugins;     /* All registered plugins */
 static void gras_trp_plugin_free(void *p); /* free one of the plugins */
 
-
-gras_dynar_t *_gras_trp_sockets; /* all existing sockets */
 static void gras_trp_socket_free(void *s); /* free one socket */
 
-static fd_set FDread;
-
 gras_error_t
 gras_trp_plugin_new(const char *name, gras_trp_setup_t setup);
 
@@ -30,24 +24,23 @@ gras_error_t
 gras_trp_plugin_new(const char *name, gras_trp_setup_t setup) {
   gras_error_t errcode;
 
-  gras_trp_plugin_t *plug = malloc(sizeof(gras_trp_plugin_t));
+  gras_trp_plugin_t *plug = gras_new0(gras_trp_plugin_t, 1);
   
   DEBUG1("Create plugin %s",name);
 
   if (!plug) 
     RAISE_MALLOC;
 
-  memset(plug,0,sizeof(gras_trp_plugin_t));
-
   plug->name=strdup(name);
   if (!plug->name)
     RAISE_MALLOC;
 
   errcode = setup(plug);
-  switch (setup(plug)) {
+  switch (errcode) {
   case mismatch_error:
     /* SG plugin return mismatch when in RL mode (and vice versa) */
-    free(plug);
+    gras_free(plug->name);
+    gras_free(plug);
     break;
 
   case no_error:
@@ -56,7 +49,7 @@ gras_trp_plugin_new(const char *name, gras_trp_setup_t setup) {
     break;
 
   default:
-    free(plug);
+    gras_free(plug);
     return errcode;
   }
   return no_error;
@@ -66,12 +59,6 @@ gras_error_t
 gras_trp_init(void){
   gras_error_t errcode;
   
-  /* make room for all socket ownership descriptions */
-  TRY(gras_dynar_new(&_gras_trp_sockets, sizeof(gras_socket_t*), NULL));
-
-  /* We do not ear for any socket for now */
-  FD_ZERO(&FDread);
-  
   /* make room for all plugins */
   TRY(gras_dict_new(&_gras_trp_plugins));
 
@@ -80,13 +67,15 @@ gras_trp_init(void){
   TRY(gras_trp_plugin_new("file",gras_trp_file_setup));
   TRY(gras_trp_plugin_new("sg",gras_trp_sg_setup));
 
+  /* buf is composed, so it must come after the others */
+  TRY(gras_trp_plugin_new("buf", gras_trp_buf_setup));
+
   return no_error;
 }
 
 void
 gras_trp_exit(void){
   gras_dict_free(&_gras_trp_plugins);
-  gras_dynar_free(_gras_trp_sockets);
 }
 
 
@@ -94,13 +83,15 @@ void gras_trp_plugin_free(void *p) {
   gras_trp_plugin_t *plug = p;
 
   if (plug) {
-    if (plug->exit)
+    if (plug->exit) {
       plug->exit(plug);
-    else if (plug->data) 
-      free(plug->data);
+    } else if (plug->data) {
+      DEBUG1("Plugin %s lacks exit(). Free data anyway.",plug->name);
+      gras_free(plug->data);
+    }
 
-    free(plug->name);
-    free(plug);
+    gras_free(plug->name);
+    gras_free(plug);
   }
 }
 
@@ -115,11 +106,13 @@ gras_error_t gras_trp_socket_new(int incoming,
 
   gras_socket_t *sock;
 
-  if (! (sock=malloc(sizeof(gras_socket_t))) )
+  if (! (sock=gras_new(gras_socket_t,1)) )
     RAISE_MALLOC;
+  DEBUG1("Create a new socket (%p)", sock);
 
   sock->plugin = NULL;
   sock->sd     = -1;
+  sock->data   = NULL;
 
   sock->incoming  = incoming ? 1:0;
   sock->outgoing  = incoming ? 0:1;
@@ -131,7 +124,8 @@ gras_error_t gras_trp_socket_new(int incoming,
   sock->raw = 0;
 
   *dst = sock;
-  return no_error;
+
+  return gras_dynar_push(gras_socketset_get(),dst);
 }
 
 
@@ -152,8 +146,7 @@ gras_socket_server(unsigned short port,
   *dst = NULL;
 
   DEBUG1("Create a server socket from plugin %s",gras_if_RL() ? "tcp" : "sg");
-  TRY(gras_trp_plugin_get_by_name(gras_if_RL() ? "tcp" : "sg",
-                                 &trp));
+  TRY(gras_trp_plugin_get_by_name("buf",&trp));
 
   /* defaults settings */
   TRY(gras_trp_socket_new(1,&sock));
@@ -168,18 +161,11 @@ gras_socket_server(unsigned short port,
         sock->accepting?'y':'n');
 
   if (errcode != no_error) {
-    free(sock);
+    gras_free(sock);
     return errcode;
   }
 
   *dst = sock;
-  /* Register this socket */
-  errcode = gras_dynar_push(_gras_trp_sockets,dst);
-  if (errcode != no_error) {
-    free(sock);
-    *dst = NULL;
-    return errcode;
-  }
 
   return no_error;
 }
@@ -201,9 +187,9 @@ gras_socket_client(const char *host,
 
   *dst = NULL;
 
-  TRY(gras_trp_plugin_get_by_name(gras_if_RL() ? "tcp" : "sg",
-                                 &trp));
+  TRY(gras_trp_plugin_get_by_name("buf",&trp));
 
+  DEBUG1("Create a client socket from plugin %s",gras_if_RL() ? "tcp" : "sg");
   /* defaults settings */
   TRY(gras_trp_socket_new(0,&sock));
   sock->plugin= trp;
@@ -211,46 +197,41 @@ gras_socket_client(const char *host,
   sock->peer_name = strdup(host?host:"localhost");
 
   /* plugin-specific */
-  errcode= (* trp->socket_client)(trp, 
-                                 host ? host : "localhost", port,
-                                 sock);
+  errcode= (*trp->socket_client)(trp, 
+                                host ? host : "localhost", port,
+                                sock);
   DEBUG3("in=%c out=%c accept=%c",
         sock->incoming?'y':'n', 
         sock->outgoing?'y':'n',
         sock->accepting?'y':'n');
 
   if (errcode != no_error) {
-    free(sock);
+    gras_free(sock);
     return errcode;
   }
 
-  /* register socket */
   *dst = sock;
-  errcode = gras_dynar_push(_gras_trp_sockets,dst);
-  if (errcode != no_error) {
-    free(sock);
-    *dst = NULL;
-    return errcode;
-  }
 
   return no_error;
 }
 
-void gras_socket_close(gras_socket_t **sock) {
+void gras_socket_close(gras_socket_t *sock) {
+  gras_dynar_t *sockets = gras_socketset_get();
   gras_socket_t *sock_iter;
   int cursor;
 
   /* FIXME: Issue an event when the socket is closed */
-  if (sock && *sock) {
-    gras_dynar_foreach(_gras_trp_sockets,cursor,sock_iter) {
-      if (*sock == sock_iter) {
-       gras_dynar_cursor_rm(_gras_trp_sockets,&cursor);
-       if ( (*sock)->plugin->socket_close) 
-         (* (*sock)->plugin->socket_close)(*sock);
+  if (sock) {
+    gras_dynar_foreach(sockets,cursor,sock_iter) {
+      if (sock == sock_iter) {
+       gras_dynar_cursor_rm(sockets,&cursor);
+       if ( sock->plugin->socket_close) 
+         (* sock->plugin->socket_close)(sock);
 
        /* free the memory */
-       free(*sock);
-       *sock=NULL;
+       if (sock->peer_name)
+         gras_free(sock->peer_name);
+       gras_free(sock);
        return;
       }
     }
@@ -266,7 +247,7 @@ void gras_socket_close(gras_socket_t **sock) {
 gras_error_t
 gras_trp_chunk_send(gras_socket_t *sd,
                    char *data,
-                   size_t size) {
+                   long int size) {
   gras_assert1(sd->outgoing,
               "Socket not suited for data send (outgoing=%c)",
               sd->outgoing?'y':'n');
@@ -283,7 +264,7 @@ gras_trp_chunk_send(gras_socket_t *sd,
 gras_error_t 
 gras_trp_chunk_recv(gras_socket_t *sd,
                    char *data,
-                   size_t size) {
+                   long int size) {
   gras_assert0(sd->incoming,
               "Socket not suited for data receive");
   gras_assert1(sd->plugin->chunk_recv,
@@ -292,6 +273,15 @@ gras_trp_chunk_recv(gras_socket_t *sd,
   return (sd->plugin->chunk_recv)(sd,data,size);
 }
 
+/**
+ * gras_trp_flush:
+ *
+ * Make sure all pending communications are done
+ */
+gras_error_t 
+gras_trp_flush(gras_socket_t *sd) {
+  return (sd->plugin->flush)(sd);
+}
 
 gras_error_t
 gras_trp_plugin_get_by_name(const char *name,