teshsuite/msg/pid.c

   1 /* Copyright (c) 2009, 2010. 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 "msg/msg.h"
   8 #include "xbt/sysdep.h"
   9
  10 XBT_LOG_NEW_DEFAULT_CATEGORY(msg_test,
  11                              "Messages specific for this msg example");
  12 const char* mailbox = "mailbox";
  13 #define task_comp_size 1000
  14 #define task_comm_size 100000
  15
  16 static int onexit(void* data){
  17   XBT_INFO("Process \"%d\" killed.", *((int*)data));
  18   return 0;
  19 }
  20
  21 static int sendpid(int argc, char *argv[])
  22 {
  23   int pid = MSG_process_self_PID();
  24   MSG_process_on_exit(onexit, &pid);
  25   msg_task_t task = MSG_task_create("pid", task_comp_size, task_comm_size, &pid);
  26   XBT_INFO("Sending pid of \"%d\".", pid);
  27   MSG_task_send(task, mailbox);
  28   XBT_INFO("Send of pid \"%d\" done.", pid);
  29   MSG_process_suspend(MSG_process_self());
  30   return 0;
  31 }
  32
  33 static int killall(int argc, char *argv[]){
  34   msg_task_t task = NULL;
  35   _XBT_GNUC_UNUSED int res;
  36   int i;
  37   for (i=0; i<3;i++) {
  38     res = MSG_task_receive(&(task), mailbox);
  39     int pid = *((int*)MSG_task_get_data(task));
  40     XBT_INFO("Killing process \"%d\".", pid);
  41     MSG_process_kill(MSG_process_from_PID(pid));
  42     task = NULL;
  43   }
  44   return 0;
  45 }
  46
  47 /** Main function */
  48 int main(int argc, char *argv[])
  49 {
  50   msg_error_t res = MSG_OK;
  51
  52   MSG_init(&argc, argv);
  53
  54   /*   Application deployment */
  55   MSG_function_register("sendpid", &sendpid);
  56   MSG_function_register("killall", &killall);
  57
  58   MSG_process_killall(atoi(argv[2]));
  59
  60   MSG_create_environment(argv[1]);
  61   MSG_launch_application(argv[1]);
  62   res = MSG_main();
  63
  64   if (res == MSG_OK)
  65     return 0;
  66   else
  67     return 1;
  68 }