+/* Copyright (c) 2012-2014. 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,
xbt_dynar_t build_hostlist_from_hostcount(int hostcount)
{
- xbt_dynar_t host_list = xbt_dynar_new(sizeof(char*), NULL);
- char *hostname = NULL;
- msg_host_t h = NULL;
- int i = 1;
+ xbt_dynar_t host_list = xbt_dynar_new(sizeof(char*), xbt_free_ref);
+ int i;
- for (; i < hostcount+1; i++) {
- hostname = xbt_new(char, HOSTNAME_LENGTH);
- snprintf(hostname, HOSTNAME_LENGTH, "host%d", i);
+ for (i = 1; i <= hostcount; i++) {
+ char *hostname = bprintf("host%d", i);
XBT_DEBUG("%s", hostname);
xbt_dynar_push(host_list, &hostname);
}
return host_list;
}
-void delete_hostlist(xbt_dynar_t h)
-{
- xbt_dynar_free(&h);
-}
-
-int broadcaster_build_chain(const char **first, xbt_dynar_t host_list, xbt_dynar_iterator_t it)
+int broadcaster_build_chain(broadcaster_t bc)
{
msg_task_t task = NULL;
- char **cur = (char**)xbt_dynar_iterator_next(it);
+ char **cur = (char**)xbt_dynar_iterator_next(bc->it);
const char *me = "host0"; /* FIXME: hardcoded*/ /*MSG_host_get_name(MSG_host_self());*/
const char *current_host = NULL;
const char *prev = NULL;
if (cur != NULL) {
/* init: prev=NULL, host=current cur, next=next cur */
next = *cur;
- *first = next;
+ bc->first = next;
/* This iterator iterates one step ahead: cur is current iterated element,
but it's actually the next one in the chain */
do {
/* following steps: prev=last, host=next, next=cur */
- cur = (char**)xbt_dynar_iterator_next(it);
+ cur = (char**)xbt_dynar_iterator_next(bc->it);
prev = last;
current_host = next;
if (cur != 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(me, current_host, prev, next);
- //MSG_task_set_category(task, current_host);
+ task = task_message_chain_new(prev, next, bc->piece_count);
MSG_task_send(task, current_host);
last = current_host;
return MSG_OK;
}
-int broadcaster_send_file(const char *first)
+int broadcaster_send_file(broadcaster_t bc)
{
- const char *me = MSG_host_get_name(MSG_host_self());
+ const char *me = "host0"; /* FIXME: hardcoded*/ /*MSG_host_get_name(MSG_host_self());*/
+ //msg_comm_t comm = NULL;
msg_task_t task = NULL;
- msg_comm_t comm = NULL;
- int status;
- int piece_count = PIECE_COUNT;
- int cur = 0;
+ bc->current_piece = 0;
- for (; cur < piece_count; cur++) {
- task = task_message_data_new(me, first, NULL, 0);
- XBT_INFO("Sending (send) from %s into mailbox %s", me, first);
- status = MSG_task_send(task, first);
-
- xbt_assert(status == MSG_OK, "broadcaster_send_file() failed");
+ 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;
}
-int broadcaster_finish(xbt_dynar_iterator_t it)
+broadcaster_t broadcaster_init(xbt_dynar_t host_list, unsigned int piece_count)
{
- msg_task_t task = NULL;
- const char *me = "host0"; /* FIXME: hardcoded*/ /*MSG_host_get_name(MSG_host_self());*/
- const char *current_host = NULL;
- char **cur = NULL;
+ int status;
+ broadcaster_t bc = xbt_new(s_broadcaster_t, 1);
- xbt_dynar_iterator_seek(it, 0);
+ 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);
- /* Send goodbye message to every peer in the order generated by iterator it */
- for (cur = (char**)xbt_dynar_iterator_next(it); cur != NULL; cur = (char**)xbt_dynar_iterator_next(it)) {
- /* Send message to current peer */
- current_host = *cur;
- task = task_message_end_data_new(me, current_host);
- //MSG_task_set_category(task, current_host);
- MSG_task_send(task, current_host);
- }
+ status = broadcaster_build_chain(bc);
+ xbt_assert(status == MSG_OK, "Chain initialization failed");
- return MSG_OK;
+ 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); /* FIXME: host names are not free'd */
+ xbt_free(bc);
+}
/** Emitter function */
int broadcaster(int argc, char *argv[])
{
+ broadcaster_t bc = NULL;
xbt_dynar_t host_list = NULL;
- const char *first = NULL;
- int status = !MSG_OK;
+ int status;
+ unsigned int piece_count = PIECE_COUNT;
- XBT_INFO("broadcaster");
+ XBT_DEBUG("broadcaster");
/* Add every mailbox given by the hostcount in argv[1] to a dynamic array */
host_list = build_hostlist_from_hostcount(atoi(argv[1]));
- /*host_list = build_hostlist_from_argv(argc, argv);*/
-
- /* Initialize iterator */
- xbt_dynar_iterator_t it = xbt_dynar_iterator_new(host_list, forward_indices_list);
+
+ /* argv[2] is the number of pieces */
+ if (argc > 2) {
+ piece_count = atoi(argv[2]);
+ XBT_DEBUG("piece_count set to %d", piece_count);
+ } else {
+ XBT_DEBUG("No piece_count specified, defaulting to %d", piece_count);
+ }
+ bc = broadcaster_init(host_list, piece_count);
/* TODO: Error checking */
- status = broadcaster_build_chain(&first, host_list, it);
- status = broadcaster_send_file(first);
- status = broadcaster_finish(it);
+ status = broadcaster_send_file(bc);
- /* Destroy iterator and hostlist */
- xbt_dynar_iterator_delete(it);
- delete_hostlist(host_list);
+ broadcaster_destroy(bc);
return status;
}