Logo AND Algorithmique Numérique Distribuée

Public GIT Repository
1827df869cee0f4f9c4257d5cf72e4ba02c00384
[simgrid.git] / examples / msg / chainsend / peer.c
1 #include "peer.h"
2
3 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_peer,
4                              "Messages specific for the peer");
5
6 /*******************************************************
7  *                     Peer                            *
8  *******************************************************/
9
10 void peer_init_chain(peer_t peer, message_t msg)
11 {
12   peer->prev = msg->prev_hostname;
13   peer->next = msg->next_hostname;
14   peer->total_pieces = msg->num_pieces;
15   peer->init = 1;
16 }
17
18 static void peer_forward_msg(peer_t peer, message_t msg)
19 {
20   msg_task_t task = task_message_data_new(peer->me, peer->next, NULL, msg->data_length);
21   msg_comm_t comm = NULL;
22   XBT_DEBUG("Sending (isend) from %s into mailbox %s", peer->me, peer->next);
23   comm = MSG_task_isend(task, peer->next);
24   queue_pending_connection(comm, peer->pending_sends);
25 }
26
27 int peer_execute_task(peer_t peer, msg_task_t task)
28 {
29   int done = 0;
30   message_t msg = MSG_task_get_data(task);
31   
32   XBT_DEBUG("Peer %s got message of type %d\n", peer->me, msg->type);
33   switch (msg->type) {
34     case MESSAGE_BUILD_CHAIN:
35       peer_init_chain(peer, msg);
36       break;
37     case MESSAGE_SEND_DATA:
38       xbt_assert(peer->init, "peer_execute_task() failed: got msg_type %d before initialization", msg->type);
39       if (peer->next != NULL)
40         peer_forward_msg(peer, msg);
41       peer->pieces++;
42       peer->bytes += msg->data_length;
43       if (peer->pieces >= peer->total_pieces) {
44         XBT_DEBUG("%d pieces receieved", peer->pieces);
45         done = 1;
46       }
47       break;
48   }
49
50   MSG_task_execute(task);
51
52   return done;
53 }
54
55 msg_error_t peer_wait_for_message(peer_t peer)
56 {
57   msg_error_t status;
58   msg_comm_t comm = NULL;
59   msg_task_t task = NULL;
60   int idx = -1;
61   int done = 0;
62
63   while (!done) {
64     comm = MSG_task_irecv(&task, peer->me);
65     queue_pending_connection(comm, peer->pending_recvs);
66
67     if ((idx = MSG_comm_waitany(peer->pending_recvs)) != -1) {
68       comm = xbt_dynar_get_as(peer->pending_recvs, idx, msg_comm_t);
69       status = MSG_comm_get_status(comm);
70       XBT_DEBUG("peer_wait_for_message: error code = %d", status);
71       xbt_assert(status == MSG_OK, "peer_wait_for_message() failed");
72
73       task = MSG_comm_get_task(comm);
74       MSG_comm_destroy(comm);
75       xbt_dynar_cursor_rm(peer->pending_recvs, (unsigned int*)&idx);
76       done = peer_execute_task(peer, task);
77
78       task_message_delete(task);
79       task = NULL;
80     }
81     process_pending_connections(peer->pending_sends);
82   }
83
84   return status;
85 }
86
87 void peer_init(peer_t p, int argc, char *argv[])
88 {
89   p->init = 0;
90   p->prev = NULL;
91   p->next = NULL;
92   p->pieces = 0;
93   p->bytes = 0;
94   p->pending_recvs = xbt_dynar_new(sizeof(msg_comm_t), NULL);
95   p->pending_sends = xbt_dynar_new(sizeof(msg_comm_t), NULL);
96   p->me = xbt_new(char, HOSTNAME_LENGTH);
97   /* Set mailbox name: use host number from argv or hostname if no argument given */
98   if (argc > 1) {
99     snprintf(p->me, HOSTNAME_LENGTH, "host%s", argv[1]);
100   } else {
101     strncpy(p->me, MSG_host_get_name(MSG_host_self()), HOSTNAME_LENGTH);
102   }
103 }
104
105 void peer_shutdown(peer_t p)
106 {
107   float start_time = MSG_get_clock();
108   float end_time = start_time + PEER_SHUTDOWN_DEADLINE;
109
110   XBT_DEBUG("Waiting for sends to finish before shutdown...");
111   /* MSG_comm_waitall(p->pending_sends, PEER_SHUTDOWN_DEADLINE); FIXME: this doesn't work */
112   while (xbt_dynar_length(p->pending_sends) && MSG_get_clock() < end_time) {
113     process_pending_connections(p->pending_sends);
114     MSG_process_sleep(1);
115   }
116
117   xbt_assert(xbt_dynar_length(p->pending_sends) == 0, "Shutdown failed, sends still pending after deadline");
118 }
119
120 void peer_delete(peer_t p)
121 {
122   xbt_dynar_free(&p->pending_recvs);
123   xbt_dynar_free(&p->pending_sends);
124   xbt_free(p->me);
125
126   xbt_free(p);
127 }
128
129 void peer_print_stats(peer_t p, float elapsed_time)
130 {
131   XBT_INFO("### %f %llu bytes (Avg %f MB/s); copy finished (simulated).", elapsed_time, p->bytes, p->bytes / 1024.0 / 1024.0 / elapsed_time); 
132 }
133
134 /** Peer function  */
135 int peer(int argc, char *argv[])
136 {
137   float start_time, end_time;
138   peer_t p = xbt_new(s_peer_t, 1);
139   msg_error_t status;
140
141   XBT_DEBUG("peer");
142
143   peer_init(p, argc, argv);
144   start_time = MSG_get_clock();
145   status = peer_wait_for_message(p);
146   peer_shutdown(p);
147   end_time = MSG_get_clock();
148   peer_print_stats(p, end_time - start_time);
149   peer_delete(p);
150
151   return status;
152 }                               /* end_of_receiver */