teshsuite/s4u/pid/pid.cpp

   1 /* Copyright (c) 2009-2010, 2013-2015. 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 "simgrid/s4u.hpp"
   8
   9 XBT_LOG_NEW_DEFAULT_CATEGORY(s4u_test, "Messages specific for this msg example");
  10
  11 simgrid::s4u::MailboxPtr mailbox = simgrid::s4u::Mailbox::byName("mailbox");
  12 double comp_size                 = 1000;
  13 double comm_size                 = 100000;
  14
  15 static int my_onexit(smx_process_exit_status_t status, int* pid)
  16 {
  17   XBT_INFO("Process \"%d\" killed.", *pid);
  18   return 0;
  19 }
  20
  21 static void sendpid()
  22 {
  23   int pid = simgrid::s4u::this_actor::pid();
  24   simgrid::s4u::this_actor::onExit((int_f_pvoid_pvoid_t)my_onexit, &pid);
  25
  26   XBT_INFO("Sending pid of \"%d\".", pid);
  27   simgrid::s4u::this_actor::send(mailbox, &pid, comm_size);
  28   XBT_INFO("Send of pid \"%d\" done.", pid);
  29
  30   simgrid::s4u::this_actor::suspend();
  31 }
  32
  33 static void killall()
  34 {
  35   for (int i = 0; i < 3; i++) {
  36     int* pid = static_cast<int*>(simgrid::s4u::this_actor::recv(mailbox));
  37     XBT_INFO("Killing process \"%d\".", *pid);
  38     simgrid::s4u::Actor::byPid(*pid)->kill();
  39   }
  40 }
  41
  42 int main(int argc, char* argv[])
  43 {
  44   simgrid::s4u::Engine* e = new simgrid::s4u::Engine(&argc, argv);
  45   e->loadPlatform(argv[1]);
  46
  47   if (argc > 2)
  48     simgrid::s4u::Actor::killAll(atoi(argv[2]));
  49   else
  50     simgrid::s4u::Actor::killAll();
  51
  52   simgrid::s4u::Actor::createActor("sendpid", simgrid::s4u::Host::by_name("Tremblay"), sendpid);
  53   simgrid::s4u::Actor::createActor("sendpid", simgrid::s4u::Host::by_name("Tremblay"), sendpid);
  54   simgrid::s4u::Actor::createActor("sendpid", simgrid::s4u::Host::by_name("Tremblay"), sendpid);
  55   simgrid::s4u::Actor::createActor("killall", simgrid::s4u::Host::by_name("Tremblay"), killall);
  56
  57   e->run();
  58
  59   return 0;
  60 }