#include <unistd.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "pvm3.h"
#include "pvm_lib.h"

#define EXEC "hello_clean"
#define BUFF 100

int main(int argc, char **argv){
  int cc=0,*tids, mytid, ptid, ntask, i;
  char buf[BUFF];

  mytid = pvm_error_check(pvm_mytid(), "pvm_mytid");
  //ptid = pvm_error_check(pvm_parent(), "pvm_parent");
  ptid = pvm_parent();

  if(ptid == PvmNoParent){
    if((argc != 2) || ((ntask = atoi(argv[1])) == 0)){
      fprintf(stderr,"Usage: %s <ntask>\n", argv[0]);
      exit(0);
    }
    tids = (int*)calloc(sizeof(int), ntask);

    for(i=0;i<ntask;i++){
      char host[BUFF];
      sprintf(host, "b%d",i+1);
      pvm_error_check(pvm_spawn(EXEC, &argv[1], 0, host, 1, &tids[i]), "pvm_spawn");
    }

    for(i=0;i<ntask;i++){
      cc = pvm_error_check(pvm_recv(-1, -1), "pvm_recv");
      pvm_error_check(pvm_bufinfo(cc, (int*)0, (int*)0, &tids[i]), "pvm_bufinfo");
      pvm_error_check(pvm_upkstr(buf), "pvm_upkstr");
      fprintf(stderr,"from t%x: %s\n", tids[i], buf);
    }
  }
  else{
    strcpy(buf, "hello, world from ");
    gethostname(buf + strlen(buf), 64);
    pvm_error_check(pvm_initsend(PvmDataDefault), "pvm_initsend");
    pvm_error_check(pvm_pkstr(buf), "pvm_pkstr");
    pvm_error_check(pvm_send(ptid, 1), "pvm_send");
  }

  pvm_exit();
  exit(0);
}