Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Reintroduce raw sockets.
[simgrid.git] / src / gras / Transport / transport.c
index cb207eb..d5d62b4 100644 (file)
@@ -8,11 +8,9 @@
 /* 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 */
@@ -26,16 +24,14 @@ 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);
+  plug->name=(char*)strdup(name);
   if (!plug->name)
     RAISE_MALLOC;
 
@@ -43,8 +39,8 @@ gras_trp_plugin_new(const char *name, gras_trp_setup_t setup) {
   switch (errcode) {
   case mismatch_error:
     /* SG plugin return mismatch when in RL mode (and vice versa) */
-    free(plug->name);
-    free(plug);
+    gras_free(plug->name);
+    gras_free(plug);
     break;
 
   case no_error:
@@ -53,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;
@@ -91,11 +87,11 @@ void gras_trp_plugin_free(void *p) {
       plug->exit(plug);
     } else if (plug->data) {
       DEBUG1("Plugin %s lacks exit(). Free data anyway.",plug->name);
-      free(plug->data);
+      gras_free(plug->data);
     }
 
-    free(plug->name);
-    free(plug);
+    gras_free(plug->name);
+    gras_free(plug);
   }
 }
 
@@ -110,9 +106,9 @@ 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);
+  DEBUG1("Create a new socket (%p)", (void*)sock);
 
   sock->plugin = NULL;
   sock->sd     = -1;
@@ -134,14 +130,18 @@ gras_error_t gras_trp_socket_new(int incoming,
 
 
 /**
- * gras_socket_server:
+ * gras_socket_server_ext:
  *
  * Opens a server socket and make it ready to be listened to.
  * In real life, you'll get a TCP socket.
  */
 gras_error_t
-gras_socket_server(unsigned short port,
-                  /* OUT */ gras_socket_t **dst) {
+gras_socket_server_ext(unsigned short port,
+                      
+                      unsigned long int bufSize,
+                      int raw,
+                      
+                      /* OUT */ gras_socket_t **dst) {
  
   gras_error_t errcode;
   gras_trp_plugin_t *trp;
@@ -156,16 +156,18 @@ gras_socket_server(unsigned short port,
   TRY(gras_trp_socket_new(1,&sock));
   sock->plugin= trp;
   sock->port=port;
+  sock->bufSize = bufSize;
+  sock->raw = raw;
 
   /* Call plugin socket creation function */
-  errcode = trp->socket_server(trp, port, sock);
+  errcode = trp->socket_server(trp, 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;
   }
 
@@ -173,17 +175,21 @@ gras_socket_server(unsigned short port,
 
   return no_error;
 }
-
+   
 /**
- * gras_socket_client:
+ * gras_socket_client_ext:
  *
  * Opens a client socket to a remote host.
  * In real life, you'll get a TCP socket.
  */
 gras_error_t
-gras_socket_client(const char *host,
-                  unsigned short port,
-                  /* OUT */ gras_socket_t **dst) {
+gras_socket_client_ext(const char *host,
+                      unsigned short port,
+                      
+                      unsigned long int bufSize,
+                      int raw,
+                      
+                      /* OUT */ gras_socket_t **dst) {
  
   gras_error_t errcode;
   gras_trp_plugin_t *trp;
@@ -198,19 +204,19 @@ gras_socket_client(const char *host,
   TRY(gras_trp_socket_new(0,&sock));
   sock->plugin= trp;
   sock->peer_port = port;
-  sock->peer_name = strdup(host?host:"localhost");
+  sock->peer_name = (char*)strdup(host?host:"localhost");
+  sock->bufSize = bufSize;
+  sock->raw = raw;
 
   /* plugin-specific */
-  errcode= (*trp->socket_client)(trp, 
-                                host ? host : "localhost", port,
-                                sock);
+  errcode= (*trp->socket_client)(trp, 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;
   }
 
@@ -219,6 +225,32 @@ gras_socket_client(const char *host,
   return no_error;
 }
 
+/**
+ * gras_socket_server:
+ *
+ * Opens a server socket and make it ready to be listened to.
+ * In real life, you'll get a TCP socket.
+ */
+gras_error_t
+gras_socket_server(unsigned short port,
+                  /* OUT */ gras_socket_t **dst) {
+   return gras_socket_server_ext(port,32,0,dst);
+}
+
+/**
+ * gras_socket_client:
+ *
+ * Opens a client socket to a remote host.
+ * In real life, you'll get a TCP socket.
+ */
+gras_error_t
+gras_socket_client(const char *host,
+                  unsigned short port,
+                  /* OUT */ gras_socket_t **dst) {
+   return gras_socket_client_ext(host,port,32,0,dst);
+}
+
+
 void gras_socket_close(gras_socket_t *sock) {
   gras_dynar_t *sockets = gras_socketset_get();
   gras_socket_t *sock_iter;
@@ -234,8 +266,8 @@ void gras_socket_close(gras_socket_t *sock) {
 
        /* free the memory */
        if (sock->peer_name)
-         free(sock->peer_name);
-       free(sock);
+         gras_free(sock->peer_name);
+       gras_free(sock);
        return;
       }
     }
@@ -303,3 +335,21 @@ int   gras_socket_peer_port(gras_socket_t *sock) {
 char *gras_socket_peer_name(gras_socket_t *sock) {
   return sock->peer_name;
 }
+
+gras_error_t gras_socket_raw_send(gras_socket_t *peer, 
+                                 unsigned int timeout,
+                                 unsigned long int expSize, 
+                                 unsigned long int msgSize) {
+  
+  gras_assert0(peer->raw,"Asked to send raw data on a regular socket\n");
+  return gras_socket_raw_exchange(peer,1,timeout,expSize,msgSize);   
+}
+
+gras_error_t gras_socket_raw_recv(gras_socket_t *peer, 
+                                 unsigned int timeout,
+                                 unsigned long int expSize, 
+                                 unsigned long int msgSize){
+  
+  gras_assert0(peer->raw,"Asked to recveive raw data on a regular socket\n");
+  return gras_socket_raw_exchange(peer,0,timeout,expSize,msgSize);   
+}