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