Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
used during the build process
[simgrid.git] / examples / gras / mmrpc / mmrpc.c
1 /* $Id$ */
2
3 /* GridRPC - Fake Grid RPC thingy doing matrix multiplications (as expected)*/
4
5 /* Copyright (c) 2005 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 #define GRAS_DEFINE_TYPE_EXTERN
11 #include "xbt/matrix.h"
12 #include "mmrpc.h"
13
14 XBT_LOG_NEW_DEFAULT_CATEGORY(MatMult,"Messages specific to this example");
15
16 /* register messages which may be sent and their payload
17    (common to client and server) */
18 void mmrpc_register_messages(void) {
19   gras_datadesc_type_t matrix_type, request_type;
20
21   matrix_type=gras_datadesc_matrix(gras_datadesc_by_name("double"),
22                                    NULL);
23   request_type=gras_datadesc_array_fixed("s_matrix_t(double)[2]",matrix_type,2);
24   
25   gras_msgtype_declare("answer", matrix_type);
26   gras_msgtype_declare("request", request_type);
27 }
28
29 typedef xbt_matrix_t request_t[2];
30 static int server_cb_request_handler(gras_msg_cb_ctx_t ctx, 
31                                      void *payload_data) {
32
33   gras_socket_t expeditor=gras_msg_cb_ctx_from(ctx);
34                              
35   /* 1. Get the payload into the data variable */
36   xbt_matrix_t *request = (xbt_matrix_t*)payload_data;
37   xbt_matrix_t result;
38   
39   /* 2. Do the computation */
40   result = xbt_matrix_double_new_mult(request[0], request[1]);
41
42   /* 3. Send it back as payload of a pong message to the expeditor */
43   gras_msg_send(expeditor, "answer", &result);
44
45   /* 4. Cleanups */
46   xbt_matrix_free(request[0]);
47   xbt_matrix_free(request[1]);
48   xbt_matrix_free(result);
49   gras_socket_close(expeditor);
50    
51   return 0;
52 } /* end_of_server_cb_request_handler */
53
54 int server (int argc,char *argv[]) {
55   xbt_ex_t e; 
56   gras_socket_t sock=NULL;
57   int port = 4000;
58   
59   /* 1. Init the GRAS infrastructure */
60   gras_init(&argc,argv);
61    
62   /* 2. Get the port I should listen on from the command line, if specified */
63   if (argc == 2) {
64     port=atoi(argv[1]);
65   }
66
67   /* 3. Create my master socket */
68   INFO1("Launch server (port=%d)", port);
69   TRY {
70     sock = gras_socket_server(port);
71   } CATCH(e) {
72     RETHROW0("Unable to establish a server socket: %s");
73   }
74
75   /* 4. Register the known messages and payloads. */
76   mmrpc_register_messages();
77    
78   /* 5. Register my callback */
79   gras_cb_register("request",&server_cb_request_handler);
80
81   /* 6. Wait up to 10 minutes for an incomming message to handle */
82   gras_msg_handle(600.0);
83    
84   /* 7. Free the allocated resources, and shut GRAS down */
85   gras_socket_close(sock);
86   gras_exit();
87    
88   INFO0("Done.");
89   return 0;
90 } /* end_of_server */
91
92
93 int client(int argc,char *argv[]) {
94   xbt_ex_t e; 
95   gras_socket_t toserver=NULL; /* peer */
96
97   gras_socket_t from;
98   xbt_matrix_t request[2], answer;
99
100   int i,j;
101
102   const char *host = "127.0.0.1";
103         int   port = 4000;
104
105   /* 1. Init the GRAS's infrastructure */
106   gras_init(&argc, argv);
107    
108   /* 2. Get the server's address. The command line override defaults when specified */
109   if (argc == 3) {
110     host=argv[1];
111     port=atoi(argv[2]);
112   } 
113
114   INFO2("Launch client (server on %s:%d)",host,port);
115    
116   /* 3. Wait for the server startup */
117   gras_os_sleep(1);
118    
119   /* 4. Create a socket to speak to the server */
120   TRY {
121     toserver=gras_socket_client(host,port);
122   } CATCH(e) {
123     RETHROW0("Unable to connect to the server: %s");
124   }
125   INFO2("Connected to %s:%d.",host,port);    
126
127
128   /* 5. Register the messages (before use) */
129   mmrpc_register_messages();
130
131   /* 6. Keep the user informed of what's going on */
132   INFO2(">>>>>>>> Connected to server which is on %s:%d <<<<<<<<", 
133         gras_socket_peer_name(toserver),gras_socket_peer_port(toserver));
134
135   /* 7. Prepare and send the request to the server */
136
137   request[0] = xbt_matrix_double_new_id(MATSIZE,MATSIZE);
138   request[1] = xbt_matrix_double_new_rand(MATSIZE,MATSIZE);
139
140   /*
141   xbt_matrix_dump(request[0],"C:sent0",0,xbt_matrix_dump_display_double);
142   xbt_matrix_dump(request[1],"C:sent1",0,xbt_matrix_dump_display_double);
143   */
144
145   gras_msg_send(toserver, "request", &request);
146
147   xbt_matrix_free(request[0]);
148
149   INFO2(">>>>>>>> Request sent to %s:%d <<<<<<<<",
150         gras_socket_peer_name(toserver),gras_socket_peer_port(toserver));
151
152   /* 8. Wait for the answer from the server, and deal with issues */
153   gras_msg_wait(6000,"answer",&from,&answer);
154
155   /*
156   xbt_matrix_dump(answer,"C:answer",0,xbt_matrix_dump_display_double);
157   */
158   for (i=0; i<MATSIZE; i++) 
159     for (j=0; i<MATSIZE; i++) 
160       xbt_assert4(xbt_matrix_get_as(answer,i,j,double)==xbt_matrix_get_as(request[1],i,j,double),
161                   "Answer does not match expectations. Found %f at cell %d,%d instead of %f",
162                   xbt_matrix_get_as(answer,i,j,double),i,j,
163                   xbt_matrix_get_as(request[1],i,j,double));
164
165   /* 9. Keep the user informed of what's going on, again */
166   INFO2(">>>>>>>> Got answer from %s:%d (values are right) <<<<<<<<", 
167         gras_socket_peer_name(from),gras_socket_peer_port(from));
168
169   /* 10. Free the allocated resources, and shut GRAS down */
170   xbt_matrix_free(request[1]);
171   xbt_matrix_free(answer);
172   gras_socket_close(toserver);
173   gras_exit();
174   INFO0("Done.");
175   return 0;
176 } /* end_of_client */