teshsuite/msg/task_destroy_cancel.c

   1 /* Copyright (c) 2010-2014. The SimGrid Team.
   2  * All rights reserved.                                                     */
   3
   4 /* This program is free software; you can redistribute it and/or modify it
   5  * under the terms of the license (GNU LGPL) which comes with this package. */
   6
   7 #include <stdio.h>
   8 #include "msg/msg.h"            /* Yeah! If you want to use msg, you need to include msg/msg.h */
   9 #include "xbt/sysdep.h"         /* calloc, printf */
  10
  11 /* Create a log channel to have nice outputs. */
  12 #include "xbt/log.h"
  13 #include "xbt/asserts.h"
  14 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,
  15                              "Messages specific for this msg example");
  16
  17 int master(int argc, char *argv[]);
  18 int slave(int argc, char *argv[]);
  19
  20 /** Emitter function  */
  21 int master(int argc, char *argv[])
  22 {
  23   double task_comp_size = 5E7;
  24   double task_comm_size = 1E6;
  25   double timeout = 1;
  26
  27   char mailbox[256];
  28   msg_task_t task = NULL;
  29   msg_comm_t comm = NULL;
  30   xbt_ex_t ex;
  31
  32   sprintf(mailbox, "jupi");
  33
  34   task = MSG_task_create("normal", task_comp_size, task_comm_size, NULL);
  35   XBT_INFO("Sending task: \"%s\"", task->name);
  36   MSG_task_send_with_timeout(task, mailbox, timeout);
  37
  38   task = MSG_task_create("cancel directly", task_comp_size, task_comm_size, NULL);
  39   XBT_INFO("Canceling task \"%s\" directly", task->name);
  40   MSG_task_cancel(task);
  41
  42   task = MSG_task_create("destroy directly", task_comp_size, task_comm_size, NULL);
  43   XBT_INFO("Destroying task \"%s\" directly", task->name);
  44   MSG_task_destroy(task);
  45
  46   task = MSG_task_create("cancel", task_comp_size, task_comm_size, NULL);
  47   comm = MSG_task_isend(task, mailbox);
  48   XBT_INFO("Canceling task \"%s\" during comm", task->name);
  49   MSG_task_cancel(task);
  50   TRY {
  51     MSG_comm_wait(comm, -1);
  52   }
  53   CATCH (ex) {
  54     xbt_ex_free(ex);
  55     MSG_comm_destroy(comm);
  56   }
  57
  58   task = MSG_task_create("finalize", task_comp_size, task_comm_size, NULL);
  59   comm = MSG_task_isend(task, mailbox);
  60   XBT_INFO("Destroying task \"%s\" during comm", task->name);
  61   MSG_task_destroy(task);
  62   TRY {
  63     MSG_comm_wait(comm, -1);
  64   }
  65   CATCH (ex) {
  66     xbt_ex_free(ex);
  67     MSG_comm_destroy(comm);
  68   }
  69
  70   task = MSG_task_create("cancel", task_comp_size, task_comm_size, NULL);
  71   MSG_task_send_with_timeout(task, mailbox, timeout);
  72
  73   task = MSG_task_create("finalize", task_comp_size, task_comm_size, NULL);
  74   MSG_task_send_with_timeout(task, mailbox, timeout);
  75
  76   XBT_INFO("Goodbye now!");
  77   return 0;
  78 }                               /* end_of_master */
  79
  80 static int worker_main(int argc, char *argv[])
  81 {
  82   msg_task_t task = MSG_process_get_data(MSG_process_self());
  83   XBT_INFO("Start %s", task->name);
  84   MSG_task_execute(task);
  85   XBT_INFO("Task done");
  86   return 0;
  87 }
  88
  89 /** Receiver function  */
  90 int slave(int argc, char *argv[])
  91 {
  92   msg_task_t task;
  93   _XBT_GNUC_UNUSED int res;
  94   int id = -1;
  95   char mailbox[80];
  96   double start, end;
  97   sprintf(mailbox, "jupi");
  98
  99   while (1) {
 100     task = NULL;
 101     res = MSG_task_receive(&(task), mailbox);
 102     xbt_assert(res == MSG_OK, "MSG_task_get failed");
 103     XBT_INFO("Handling task \"%s\"", MSG_task_get_name(task));
 104
 105     if (!strcmp(MSG_task_get_name(task), "finalize")) {
 106       XBT_INFO("Destroying task \"%s\"", task->name);
 107       MSG_task_destroy(task);
 108       break;
 109     }
 110
 111     if (!strcmp(MSG_task_get_name(task), "cancel")) {
 112       MSG_process_create("worker1", worker_main, task, MSG_host_self());
 113       XBT_INFO("Canceling task \"%s\"", task->name);
 114       MSG_task_cancel(task);
 115       continue;
 116     }
 117
 118     start = MSG_get_clock();
 119     MSG_task_execute(task);
 120     end = MSG_get_clock();
 121     XBT_INFO("Task \"%s\" done in %f (amount %f)"
 122                 , MSG_task_get_name(task)
 123                 , end - start
 124                 , MSG_task_get_remaining_computation(task));
 125
 126     MSG_task_destroy(task);
 127     task = NULL;
 128     id--;
 129   }
 130   XBT_INFO("I'm done. See you!");
 131   return 0;
 132 }                               /* end_of_slave */
 133
 134 /** Main function */
 135 int main(int argc, char *argv[])
 136 {
 137   msg_error_t res;
 138   const char *platform_file;
 139   const char *application_file;
 140
 141   MSG_init(&argc, argv);
 142   if (argc != 3) {
 143     printf("Usage: %s platform_file deployment_file\n", argv[0]);
 144     printf("example: %s msg_platform.xml msg_deployment.xml\n", argv[0]);
 145     exit(1);
 146   }
 147   platform_file = argv[1];
 148   application_file = argv[2];
 149
 150   /* MSG_config("workstation/model","KCCFLN05"); */
 151   {                             /*  Simulation setting */
 152     MSG_create_environment(platform_file);
 153   }
 154   {                             /*   Application deployment */
 155     MSG_function_register("master", master);
 156     MSG_function_register("slave", slave);
 157
 158     MSG_launch_application(application_file);
 159   }
 160   res = MSG_main();
 161
 162   XBT_INFO("Simulation time %g", MSG_get_clock());
 163
 164   if (res == MSG_OK)
 165     return 0;
 166   else
 167     return 1;
 168 }                               /* end_of_main */