mfork_test.c

Go to the documentation of this file.

/* $Id: mfork_test.c,v 1.10 2005/11/10 20:46:10 seymour Exp $ */
/* $UTK_Copyright: $ */

#include <stdio.h>
#include <stdlib.h>
#include <signal.h>
#include <sys/types.h>
#include <unistd.h>

#include "mfork.h"

void
child_action(void **args)
{
  int foo;

  foo = *((int *)args[0]);

  printf("child [%d], foo = %d\n", (int)getpid(), foo);

  return;
}

void
child_action_loop(void **args)
{
  int foo;

  foo = *((int *)args[0]);

  while(1) {
    printf("child [%d], foo = %d\n", (int)getpid(), foo);

    if(!mfork_check_parent()) {
      ERRPRINTF("Parent died, so I am exiting\n");
      _exit(0);
    }

    sleep(1);
  }
}

void
generic_cleanup(void **args)
{
  int foo;

  foo = *((int *)args[0]);

  printf("child [%d], cleaning up.  arg was %d\n", (int)getpid(), foo);

  return;
}

void
generic_pre(void **args)
{
  int foo;

  foo = *((int *)args[0]);

  printf("child [%d], pre-fork.  arg was %d\n", (int)getpid(), foo);

  return;
}

void
generic_post(void **args)
{
  int foo;

  foo = *((int *)args[0]);

  printf("child [%d], post-fork.  arg was %d\n", (int)getpid(), foo);

  return;
}

void
parent_loop(int foo)
{
  int i;

  for(i=0;i<60;i++) {
    printf("parent [%d]...\n", (int)getpid());
    sleep(1);
  }
}

int main(int argc, char** argv)
{
  int gs_server1 = 666, gs_server2 = 1234, gs_server3 = 777, gs_server4 = 222;
  void **child_args1, **child_args2, **child_args3, **child_args4;
  pid_t pid;

  child_args1 = (void **)malloc(sizeof(void*));
  child_args1[0] = &gs_server1;

  pid = mfork(child_action, 1, child_args1, generic_pre, 
    generic_post, generic_cleanup, 10);

  if(pid < 0) {
    fprintf(stderr,"Failed to fork workload manager process.\n");
    exit(-1);
  }

  child_args2 = (void **)malloc(sizeof(void*));
  child_args2[0] = &gs_server2;

  pid = mfork(child_action, 1, child_args2, generic_pre,
    generic_post, generic_cleanup, 10);

  if(pid < 0) {
    fprintf(stderr,"Failed to fork workload manager process.\n");
    exit(-1);
  }

  child_args3 = (void **)malloc(sizeof(void*));
  child_args3[0] = &gs_server3;

  pid = mfork(child_action, 1, child_args3, generic_pre, 
    generic_post, generic_cleanup, 10);

  if(pid < 0) {
    fprintf(stderr,"Failed to fork workload manager process.\n");
    exit(-1);
  }

  child_args4 = (void **)malloc(sizeof(void*));
  child_args4[0] = &gs_server4;

  pid = mfork(child_action_loop, -1, child_args4, NULL, NULL, NULL, 10);

  if(pid < 0) {
    fprintf(stderr,"Failed to fork workload manager process.\n");
    exit(-1);
  }

  printf("[%d] after forking last child...\n",(int)getpid());
  parent_loop(gs_server1);

  mfork_finalize();
  exit(EXIT_SUCCESS);
}