Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
Tweak gras_trp_*_recv() prototype. This is now (sock, char*data,int size, int bufsize...
[simgrid.git] / src / gras / Transport / transport_plugin_file.c
1 /* $Id$ */
2
3 /* File transport - send/receive a bunch of bytes from a file               */
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 "portable.h"
11 #include "transport_private.h"
12 #include "xbt/ex.h"
13
14 XBT_LOG_NEW_DEFAULT_SUBCATEGORY(trp_file,transport,
15         "Pseudo-transport to write to/read from a file");
16
17 /***
18  *** Prototypes 
19  ***/
20 void gras_trp_file_close(gras_socket_t sd);
21   
22 void gras_trp_file_chunk_send(gras_socket_t sd,
23                               const char *data,
24                               unsigned long int size);
25
26 void gras_trp_file_chunk_recv(gras_socket_t sd,
27                               char *data,
28                               unsigned long int size,
29                               unsigned long int bufsize);
30
31
32 /***
33  *** Specific plugin part
34  ***/
35
36 typedef struct {
37   fd_set incoming_socks;
38 } gras_trp_file_plug_data_t;
39
40 /***
41  *** Specific socket part
42  ***/
43
44
45
46 /***
47  *** Code
48  ***/
49 void
50 gras_trp_file_setup(gras_trp_plugin_t plug) {
51
52   gras_trp_file_plug_data_t *file = xbt_new(gras_trp_file_plug_data_t,1);
53
54   FD_ZERO(&(file->incoming_socks));
55
56   plug->socket_close = gras_trp_file_close;
57   plug->chunk_send   = gras_trp_file_chunk_send;
58   plug->chunk_recv   = gras_trp_file_chunk_recv;
59   plug->data         = (void*)file;
60 }
61
62 /**
63  * gras_socket_client_from_file:
64  *
65  * Create a client socket from a file path.
66  *
67  * This only possible in RL, and is mainly for debugging.
68  */
69 gras_socket_t
70 gras_socket_client_from_file(const char*path) {
71   gras_socket_t res;
72
73   xbt_assert0(gras_if_RL(),
74                "Cannot use file as socket in the simulator");
75
76   gras_trp_socket_new(0,&res);
77
78   res->plugin=gras_trp_plugin_get_by_name("file");
79
80   if (strcmp("-", path)) {
81     res->sd = open(path, O_WRONLY|O_CREAT | O_BINARY, S_IRUSR|S_IWUSR|S_IRGRP );
82     
83     if ( res->sd < 0) {
84       THROW2(system_error,0,
85              "Cannot create a client socket from file %s: %s",
86              path, strerror(errno));
87     }
88   } else {
89     res->sd = 1; /* stdout */
90   }
91
92   DEBUG5("sock_client_from_file(%s): sd=%d in=%c out=%c accept=%c",
93          path,
94          res->sd,
95          res->incoming?'y':'n', 
96          res->outgoing?'y':'n',
97          res->accepting?'y':'n');
98
99   return res;
100 }
101
102 /**
103  * gras_socket_server_from_file:
104  *
105  * Create a server socket from a file path.
106  *
107  * This only possible in RL, and is mainly for debugging.
108  */
109 gras_socket_t gras_socket_server_from_file(const char*path) {
110   gras_socket_t res;
111
112   xbt_assert0(gras_if_RL(),
113                "Cannot use file as socket in the simulator");
114
115   gras_trp_socket_new(1,&res);
116
117   res->plugin=gras_trp_plugin_get_by_name("file");
118
119
120   if (strcmp("-", path)) {
121     res->sd = open(path, O_RDONLY | O_BINARY);
122
123     if ( res->sd < 0) {
124       THROW2(system_error,0,
125              "Cannot create a server socket from file %s: %s",
126              path, strerror(errno));
127     }
128   } else {
129     res->sd = 0; /* stdin */
130   }
131
132   DEBUG4("sd=%d in=%c out=%c accept=%c",
133          res->sd,
134          res->incoming?'y':'n', 
135          res->outgoing?'y':'n',
136          res->accepting?'y':'n');
137
138   return res;
139 }
140
141 void gras_trp_file_close(gras_socket_t sock){
142   gras_trp_file_plug_data_t *data;
143   
144   if (!sock) return; /* close only once */
145   data=sock->plugin->data;
146
147   if (sock->sd == 0) {
148     DEBUG0("Do not close stdin");
149   } else if (sock->sd == 1) {
150     DEBUG0("Do not close stdout");
151   } else {
152     DEBUG1("close file connection %d", sock->sd);
153
154     /* forget about the socket */
155     FD_CLR(sock->sd, &(data->incoming_socks));
156
157     /* close the socket */
158     if(close(sock->sd) < 0) {
159       WARN2("error while closing file %d: %s", 
160                sock->sd, strerror(errno));
161     }
162   }
163 }
164
165 /**
166  * gras_trp_file_chunk_send:
167  *
168  * Send data on a file pseudo-socket
169  */
170 void
171 gras_trp_file_chunk_send(gras_socket_t sock,
172                          const char *data,
173                          unsigned long int size) {
174   
175   xbt_assert0(sock->outgoing, "Cannot write on client file socket");
176   xbt_assert0(size >= 0, "Cannot send a negative amount of data");
177
178   while (size) {
179     int status = 0;
180     
181     DEBUG3("write(%d, %p, %ld);", sock->sd, data, (long int)size);
182     status = write(sock->sd, data, (long int)size);
183     
184     if (status == -1) {
185       THROW4(system_error,0,"write(%d,%p,%d) failed: %s",
186              sock->sd, data, (int)size,
187              strerror(errno));
188     }
189     
190     if (status) {
191       size  -= status;
192       data  += status;
193     } else {
194       THROW0(system_error,0,"file descriptor closed");
195     }
196   }
197 }
198 /**
199  * gras_trp_file_chunk_recv:
200  *
201  * Receive data on a file pseudo-socket.
202  */
203 void
204 gras_trp_file_chunk_recv(gras_socket_t sock,
205                          char *data,
206                          unsigned long int size,
207                          unsigned long int bufsize) {
208
209   xbt_assert0(sock, "Cannot recv on an NULL socket");
210   xbt_assert0(sock->incoming, "Cannot recv on client file socket");
211   xbt_assert0(size >= 0, "Cannot receive a negative amount of data");
212   xbt_assert0(bufsize>=size,"Not enough buffer size to receive that much data");
213
214   while (size) {
215     int status = 0;
216     
217     status = read(sock->sd, data, (long int)bufsize);
218     DEBUG3("read(%d, %p, %ld);", sock->sd, data, size);
219     
220     if (status == -1) {
221       THROW4(system_error,0,"read(%d,%p,%d) failed: %s",
222              sock->sd, data, (int)size,
223              strerror(errno));
224     }
225     
226     if (status) {
227       size    -= status;
228       bufsize -= status;
229       data    += status;
230     } else {
231       THROW0(system_error,0,"file descriptor closed");
232     }
233   }
234 }
235