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->init = 1;
  15 }
  16
  17 static void peer_forward_msg(peer_t peer, message_t msg)
  18 {
  19   msg_task_t task = task_message_data_new(peer->me, peer->next, NULL, msg->data_length);
  20   msg_comm_t comm = NULL;
  21   XBT_DEBUG("Sending (isend) from %s into mailbox %s", peer->me, peer->next);
  22   comm = MSG_task_isend(task, peer->next);
  23   queue_pending_connection(comm, peer->pending_sends);
  24 }
  25
  26 int peer_execute_task(peer_t peer, msg_task_t task)
  27 {
  28   int done = 0;
  29   message_t msg = MSG_task_get_data(task);
  30
  31   XBT_DEBUG("Peer %s got message of type %d\n", peer->me, msg->type);
  32   switch (msg->type) {
  33     case MESSAGE_BUILD_CHAIN:
  34       peer_init_chain(peer, msg);
  35       break;
  36     case MESSAGE_SEND_DATA:
  37       xbt_assert(peer->init, "peer_execute_task() failed: got msg_type %d before initialization", msg->type);
  38       if (peer->next != NULL)
  39         peer_forward_msg(peer, msg);
  40       peer->pieces++;
  41       peer->bytes += msg->data_length;
  42       break;
  43     case MESSAGE_END_DATA:
  44       xbt_assert(peer->init, "peer_execute_task() failed: got msg_type %d before initialization", msg->type);
  45       done = 1;
  46       XBT_DEBUG("%d pieces receieved", peer->pieces);
  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 done = 0;
  61
  62   while (!done) {
  63     if (comm == NULL)
  64       comm = MSG_task_irecv(&task, peer->me);
  65
  66     if (MSG_comm_test(comm)) {
  67       status = MSG_comm_get_status(comm);
  68       XBT_DEBUG("peer_wait_for_message: error code = %d", status);
  69       xbt_assert(status == MSG_OK, "peer_wait_for_message() failed");
  70       MSG_comm_destroy(comm);
  71       comm = NULL;
  72       done = peer_execute_task(peer, task);
  73       task_message_delete(task);
  74       task = NULL;
  75     } else {
  76       process_pending_connections(peer->pending_sends);
  77       MSG_process_sleep(0.01);
  78     }
  79   }
  80
  81   return status;
  82 }
  83
  84 void peer_init(peer_t p, int argc, char *argv[])
  85 {
  86   p->init = 0;
  87   p->prev = NULL;
  88   p->next = NULL;
  89   p->pieces = 0;
  90   p->bytes = 0;
  91   p->close_asap = 0;
  92   p->pending_sends = xbt_dynar_new(sizeof(msg_comm_t), NULL);
  93   p->me = xbt_new(char, HOSTNAME_LENGTH);
  94   /* Set mailbox name: use host number from argv or hostname if no argument given */
  95   if (argc > 1) {
  96     snprintf(p->me, HOSTNAME_LENGTH, "host%s", argv[1]);
  97   } else {
  98     strncpy(p->me, MSG_host_get_name(MSG_host_self()), HOSTNAME_LENGTH);
  99   }
 100 }
 101
 102 void peer_shutdown(peer_t p)
 103 {
 104   float start_time = MSG_get_clock();
 105   float end_time = start_time + PEER_SHUTDOWN_DEADLINE;
 106
 107   XBT_DEBUG("Waiting for sends to finish before shutdown...");
 108   while (xbt_dynar_length(p->pending_sends) && MSG_get_clock() < end_time) {
 109     process_pending_connections(p->pending_sends);
 110     MSG_process_sleep(1);
 111   }
 112
 113   xbt_assert(xbt_dynar_length(p->pending_sends) == 0, "Shutdown failed, sends still pending after deadline");
 114 }
 115
 116 void peer_delete(peer_t p)
 117 {
 118   xbt_dynar_free(&p->pending_sends);
 119   xbt_free(p->me);
 120
 121   xbt_free(p);
 122 }
 123
 124 void peer_print_stats(peer_t p, float elapsed_time)
 125 {
 126   XBT_INFO("### %f %llu bytes (Avg %f MB/s); copy finished (simulated).", elapsed_time, p->bytes, p->bytes / 1024.0 / 1024.0 / elapsed_time);
 127 }
 128
 129 /** Peer function  */
 130 int peer(int argc, char *argv[])
 131 {
 132   float start_time, end_time;
 133   peer_t p = xbt_new(s_peer_t, 1);
 134   msg_error_t status;
 135
 136   XBT_DEBUG("peer");
 137
 138   peer_init(p, argc, argv);
 139   start_time = MSG_get_clock();
 140   status = peer_wait_for_message(p);
 141   peer_shutdown(p);
 142   end_time = MSG_get_clock();
 143   peer_print_stats(p, end_time - start_time);
 144   peer_delete(p);
 145
 146   return status;
 147 }                               /* end_of_receiver */
 148
 149