/*
* fork_process() - fork process if user has requested such behavior
*
* If the user passed the -b option, we will need to fork this process.
* Also, arguments for the soon-to-be created new process are gathered.
*/
int fork_process(
int argc,
char **argv,
int DoBackground,
int &this_pid,
char *JobID,
char **Args)
{
int aindex = 0;
int rc = -100;
/* gather a.out and other arguments */
aindex = 0;
for (;optind < argc;optind++)
{
Args[aindex++] = strdup(argv[optind]);
printf("Got arg: %s\n",
Args[aindex-1]);
}
Args[aindex] = NULL;
/* decide if we should fork or not */
this_pid = 1;
if (DoBackground == 1)
{
printf("FORKING!\n");
this_pid = fork();
}
if ((DoBackground == 0) || (this_pid == 0))
{
rc = tm_adopt(JobID, TM_ADOPT_JOBID, getpid());
}
else if (this_pid > 0)
{
/* parent*/
fclose(stdin);
fclose(stdout);
fclose(stderr);
}
else if (this_pid < 0)
{
fprintf(stderr, "pbs_track: could not fork (%d:%s)\n",
errno,
strerror(errno));
}
return rc;
}
/*
* adopt_process() - adopt a running process into a running PBS job
*
* This function will only be called if the calling function (main)
* determines that the user wants to adopt an existing process
*/
int adopt_process(
char *JobID,
const std::string &tmpAdopteeID)
{
int rc;
int adoptee_pid;
adoptee_pid = strtol(tmpAdopteeID.c_str(), NULL, 10);
if (errno == ERANGE || tmpAdopteeID.find_first_not_of("0123456789") != std::string::npos)
{
fprintf(stderr, "Invalid PID to adopt: %s\n", tmpAdopteeID.c_str());
return PBSE_RMBADPARAM;
}
rc = tm_adopt(JobID, TM_ADOPT_JOBID, adoptee_pid);
return rc;
}
int main(
int argc,
char **argv) /* pbs_track */
{
int ArgIndex;
int NumErrs = 0;
char *Args[MAXARGS];
int aindex = 0;
int rc;
int pid;
char tmpJobID[PBS_MAXCLTJOBID]; /* from the command line */
char JobID[PBS_MAXCLTJOBID]; /* modified job ID for MOM/server consumption */
char ServerName[MAXSERVERNAME];
int DoBackground = 0;
tmpJobID[0] = '\0';
initialize_network_info();
/* USAGE: pbs_track [-j <JOBID>] -- a.out arg1 arg2 ... argN */
#define GETOPT_ARGS "bj:"
while ((ArgIndex = getopt(argc, argv, GETOPT_ARGS)) != EOF)
{
switch (ArgIndex)
{
case 'b':
/* background process */
DoBackground = 1;
break;
case 'j':
strncpy(tmpJobID, optarg, sizeof(tmpJobID));
if (tmpJobID[PBS_MAXCLTJOBID-1] != '\0')
{
/* truncation occurred! */
fprintf(stderr, "pbs_track: given job ID too large (> %d)\n",
PBS_MAXCLTJOBID);
exit(-1);
}
break;
default:
NumErrs++;
break;
}
}
if ((NumErrs > 0) ||
(optind >= argc) ||
(tmpJobID[0] == '\0'))
{
static char Usage[] = "USAGE: pbs_track [-j <JOBID>] [-b] -- a.out arg1 arg2 ... argN\n";
fprintf(stderr, "%s", Usage);
exit(2);
}
if (getenv(NO_SERVER_SUFFIX) != NULL)
{
snprintf(JobID, sizeof(JobID), "%s", tmpJobID);
}
else
{
if (get_server(tmpJobID, JobID, sizeof(JobID), ServerName, sizeof(ServerName)))
{
fprintf(stderr, "pbs_track: illegally formed job identifier: '%s'\n", JobID);
exit(1);
}
}
/* gather a.out and other arguments */
aindex = 0;
for (;optind < argc;optind++)
{
Args[aindex++] = strdup(argv[optind]);
printf("Got arg: %s\n",
Args[aindex-1]);
}
Args[aindex] = NULL;
/* decide if we should fork or not */
pid = 1;
if (DoBackground == 1)
{
printf("FORKING!\n");
pid = fork();
}
if ((DoBackground == 0) || (pid == 0))
{
/* either parent or child, depending on the setting */
/* call tm_adopt() to start tracking this process */
rc = tm_adopt(JobID, TM_ADOPT_JOBID, getpid());
switch (rc)
{
case TM_SUCCESS:
/* success! */
break;
case TM_ENOTFOUND:
fprintf(stderr, "pbs_track: MOM could not find job %s\n",
JobID);
break;
case TM_ESYSTEM:
case TM_ENOTCONNECTED:
fprintf(stderr, "pbs_track: error occurred while trying to communication with pbs_mom: %s (%d)\n",
pbse_to_txt(rc),
rc);
break;
default:
/* Unexpected error occurred */
fprintf(stderr, "pbs_track: unexpected error %s (%d) occurred\n",
pbse_to_txt(rc),
rc);
break;
} /* END switch(rc) */
if (rc != TM_SUCCESS)
{
exit(-1);
}
/* do the exec */
if (execvp(Args[0], Args) == -1)
{
fprintf(stderr,"execvp failed with error %d, message:\n%s\n",
errno,
strerror(errno));
}
} /* END if ((DoBackground == 0) || (pid == 0)) */
else if (pid > 0)
{
/* parent*/
fclose(stdin);
fclose(stdout);
fclose(stderr);
}
else if (pid < 0)
{
fprintf(stderr, "pbs_track: could not fork (%d:%s)\n",
errno,
strerror(errno));
}
exit(0);
} /* END main() */
