X-Git-Url: http://info.iut-bm.univ-fcomte.fr/pub/gitweb/simgrid.git/blobdiff_plain/63a8fb284a051f3748a3eb0cdf73882a47c00c9d..8a0e560ec45d87b09644b07ea26ddb95d589c670:/teshsuite/msg/app-chainsend/broadcaster.c

diff --git a/teshsuite/msg/app-chainsend/broadcaster.c b/teshsuite/msg/app-chainsend/broadcaster.c
new file mode 100644
index 0000000000..ff258f267f
--- /dev/null
+++ b/teshsuite/msg/app-chainsend/broadcaster.c
@@ -0,0 +1,130 @@
+/* Copyright (c) 2012-2017. The SimGrid Team. All rights reserved.          */
+
+/* This program is free software; you can redistribute it and/or modify it
+ * under the terms of the license (GNU LGPL) which comes with this package. */
+
+#include "broadcaster.h"
+
+XBT_LOG_NEW_DEFAULT_CATEGORY(msg_broadcaster, "Messages specific for the broadcaster");
+
+xbt_dynar_t build_hostlist_from_hostcount(int hostcount)
+{
+  xbt_dynar_t host_list = xbt_dynar_new(sizeof(char*), xbt_free_ref);
+  for (int i = 1; i <= hostcount; i++) {
+    char* hostname = bprintf("host%d", i);
+    XBT_DEBUG("%s", hostname);
+    xbt_dynar_push(host_list, &hostname);
+  }
+  return host_list;
+}
+
+int broadcaster_build_chain(broadcaster_t bc)
+{
+  msg_task_t task          = NULL;
+  char** cur               = (char**)xbt_dynar_iterator_next(bc->it);
+  const char* me           = MSG_host_get_name(MSG_host_self());
+  const char* current_host = NULL;
+  const char* prev         = NULL;
+  const char* next         = NULL;
+  const char* last         = NULL;
+
+  /* Build the chain if there's at least one peer */
+  if (cur != NULL) {
+    /* init: prev=NULL, host=current cur, next=next cur */
+    next      = *cur;
+    bc->first = next;
+
+    /* This iterator iterates one step ahead: cur is current iterated element, but is actually next in the chain */
+    do {
+      /* following steps: prev=last, host=next, next=cur */
+      cur          = (char**)xbt_dynar_iterator_next(bc->it);
+      prev         = last;
+      current_host = next;
+      if (cur != NULL)
+        next = *cur;
+      else
+        next = NULL;
+      XBT_DEBUG("Building chain--broadcaster:\"%s\" dest:\"%s\" prev:\"%s\" next:\"%s\"", me, current_host, prev, next);
+
+      /* Send message to current peer */
+      task = task_message_chain_new(prev, next, bc->piece_count);
+      MSG_task_send(task, current_host);
+
+      last = current_host;
+    } while (cur != NULL);
+  }
+
+  return MSG_OK;
+}
+
+int broadcaster_send_file(broadcaster_t bc)
+{
+  const char* me  = MSG_host_get_name(MSG_host_self());
+  msg_task_t task = NULL;
+
+  bc->current_piece = 0;
+
+  while (bc->current_piece < bc->piece_count) {
+    task = task_message_data_new(NULL, PIECE_SIZE);
+    XBT_DEBUG("Sending (send) piece %d from %s into mailbox %s", bc->current_piece, me, bc->first);
+    MSG_task_send(task, bc->first);
+    bc->current_piece++;
+  }
+
+  return MSG_OK;
+}
+
+broadcaster_t broadcaster_init(xbt_dynar_t host_list, unsigned int piece_count)
+{
+  int status;
+  broadcaster_t bc = xbt_new(s_broadcaster_t, 1);
+
+  bc->first             = NULL;
+  bc->piece_count       = piece_count;
+  bc->current_piece     = 0;
+  bc->host_list         = host_list;
+  bc->it                = xbt_dynar_iterator_new(bc->host_list, forward_indices_list);
+  bc->max_pending_sends = MAX_PENDING_SENDS;
+  bc->pending_sends     = xbt_dynar_new(sizeof(msg_comm_t), NULL);
+
+  status = broadcaster_build_chain(bc);
+  xbt_assert(status == MSG_OK, "Chain initialization failed");
+
+  return bc;
+}
+
+static void broadcaster_destroy(broadcaster_t bc)
+{
+  /* Destroy iterator and hostlist */
+  xbt_dynar_iterator_delete(bc->it);
+  xbt_dynar_free(&bc->pending_sends);
+  xbt_dynar_free(&bc->host_list);
+  xbt_free(bc);
+}
+
+/** Emitter function  */
+int broadcaster(int argc, char* argv[])
+{
+  unsigned int piece_count = PIECE_COUNT;
+
+  XBT_DEBUG("broadcaster");
+
+  /* Add every mailbox given by the hostcount in argv[1] to a dynamic array */
+  xbt_dynar_t host_list = build_hostlist_from_hostcount(xbt_str_parse_int(argv[1], "Invalid number of peers: %s"));
+
+  /* argv[2] is the number of pieces */
+  if (argc > 2) {
+    piece_count = xbt_str_parse_int(argv[2], "Invalid number of pieces: %s");
+    XBT_DEBUG("piece_count set to %u", piece_count);
+  } else {
+    XBT_DEBUG("No piece_count specified, defaulting to %u", piece_count);
+  }
+  broadcaster_t bc = broadcaster_init(host_list, piece_count);
+
+  /* TODO: Error checking */
+  int status = broadcaster_send_file(bc);
+
+  broadcaster_destroy(bc);
+
+  return status;
+}