Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
If the timeout is negative, the user wants to block
[simgrid.git] / src / gras / Transport / sg_transport.c
1 /* $Id$ */
2
3 /* sg_transport - SG specific functions for transport                       */
4
5 /* Copyright (c) 2004 Martin Quinson. All rights reserved.                  */
6
7 /* This program is free software; you can redistribute it and/or modify it
8  * under the terms of the license (GNU LGPL) which comes with this package. */
9
10 #include "xbt/ex.h"
11 #include "gras/Transport/transport_private.h"
12 #include "gras/Virtu/virtu_sg.h"
13
14 XBT_LOG_EXTERNAL_DEFAULT_CATEGORY(gras_trp);
15
16 /* check transport_private.h for an explanation of this variable; this just need to be defined to NULL in SG */
17 gras_socket_t _gras_lastly_selected_socket = NULL;
18
19 /**     
20  * gras_trp_select:
21  *
22  * Returns the next socket to service having a message awaiting.
23  *
24  * if timeout<0, we ought to implement the adaptative timeout (FIXME)
25  *
26  * if timeout=0, do not wait for new message, only handle the ones already there.
27  *
28  * if timeout>0 and no message there, wait at most that amount of time before giving up.
29  */
30 gras_socket_t gras_trp_select(double timeout) {
31   gras_socket_t res;
32   gras_trp_procdata_t pd = 
33     (gras_trp_procdata_t) gras_libdata_by_id(gras_trp_libdata_id);
34   gras_trp_sg_sock_data_t *sockdata;
35   gras_trp_plugin_t trp;
36   gras_socket_t active_socket = NULL;
37   gras_trp_sg_sock_data_t *active_socket_data;
38   gras_socket_t sock_iter; /* iterating over all sockets */
39   int cursor;
40
41   DEBUG3("select on %s@%s with timeout=%f",
42          SIMIX_process_get_name(SIMIX_process_self()),
43          SIMIX_host_get_name(SIMIX_host_self()),
44          timeout);
45    if (timeout>=0) {
46       xbt_queue_shift_timed(pd->msg_selectable_sockets,
47                             &active_socket, timeout);
48    } else {
49       xbt_queue_shift(pd->msg_selectable_sockets, &active_socket);
50    }
51    
52   if (active_socket == NULL) {
53     DEBUG0("TIMEOUT");
54     THROW0(timeout_error,0,"Timeout");
55   }
56   active_socket_data = (gras_trp_sg_sock_data_t*)active_socket->data;
57   
58   /* Ok, got something. Open a socket back to the expeditor */
59
60   /* Try to reuse an already openned socket to that expeditor */
61   DEBUG1("Open sockets size %lu",xbt_dynar_length(pd->sockets));
62   xbt_dynar_foreach(pd->sockets,cursor,sock_iter) {
63     gras_trp_sg_sock_data_t *sock_data;
64     DEBUG1("Consider %p as outgoing socket to expeditor",sock_iter);
65     
66     if (sock_iter->meas || !sock_iter->outgoing)
67       continue;
68     sock_data = ((gras_trp_sg_sock_data_t*)sock_iter->data);
69
70     if ( (sock_data->to_socket == active_socket) && 
71          (sock_data->to_host == SIMIX_process_get_host(active_socket_data->from_process)) ) {
72                         xbt_dynar_cursor_unlock(pd->sockets);   
73       return sock_iter;
74     }
75   }
76
77   /* Socket to expeditor not created yet */
78   DEBUG0("Create a socket to the expeditor");
79
80   trp = gras_trp_plugin_get_by_name("sg");
81
82   gras_trp_socket_new(1,&res);
83   res->plugin   = trp;
84
85   res->incoming  = 1;
86   res->outgoing  = 1;
87   res->accepting = 0;
88   res->sd        = -1;
89
90   res->port = -1;
91
92   /* initialize the ports */
93   //res->peer_port = active_socket->port;
94   res->port = active_socket->peer_port;
95   
96   /* create sockdata */
97   sockdata = xbt_new(gras_trp_sg_sock_data_t,1);
98   sockdata->from_process = SIMIX_process_self();
99   sockdata->to_process   = active_socket_data->from_process;
100         
101   res->peer_port = 
102     ((gras_trp_procdata_t)gras_libdata_by_name_from_remote("gras_trp",sockdata->to_process))->myport;
103   sockdata->to_socket = active_socket;
104   /*update the peer to_socket  variable */
105   active_socket_data->to_socket = res;
106   sockdata->cond = SIMIX_cond_init();
107   sockdata->mutex = SIMIX_mutex_init();
108
109   sockdata->to_host = SIMIX_process_get_host(active_socket_data->from_process);
110
111   res->data = sockdata;
112   res->peer_name = strdup(SIMIX_host_get_name(sockdata->to_host));
113
114   gras_trp_buf_init_sock(res);
115
116   DEBUG4("Create socket to process:%s(Port %d) from process: %s(Port %d)",
117          SIMIX_process_get_name(sockdata->from_process),
118          res->peer_port,
119          SIMIX_process_get_name(sockdata->to_process), res->port);
120
121   return res;
122 }
123
124   
125 /* dummy implementations of the functions used in RL mode */
126
127 void gras_trp_tcp_setup(gras_trp_plugin_t plug) {  THROW0(mismatch_error,0,NULL); }
128 void gras_trp_file_setup(gras_trp_plugin_t plug){  THROW0(mismatch_error,0,NULL); }
129 void gras_trp_iov_setup(gras_trp_plugin_t plug) {  THROW0(mismatch_error,0,NULL); }
130
131 gras_socket_t gras_trp_buf_init_sock(gras_socket_t sock) { return sock;}
132