few.c

来自「MPI stands for the Message Passing Inter」· C语言 代码 · 共 128 行

/* 
 * Copyright (c) 2004-2005 The Trustees of Indiana University and Indiana
 *                         University Research and Technology
 *                         Corporation.  All rights reserved.
 * Copyright (c) 2004-2005 The University of Tennessee and The University
 *                         of Tennessee Research Foundation.  All rights
 *                         reserved.
 * Copyright (c) 2004-2005 High Performance Computing Center Stuttgart, 
 *                         University of Stuttgart.  All rights reserved.
 * Copyright (c) 2004-2005 The Regents of the University of California.
 *                         All rights reserved.
 * $COPYRIGHT$
 * 
 * Additional copyrights may follow
 * 
 * $HEADER$
 */


#include "opal_config.h"

#include <stdio.h>
#include <errno.h>
#ifdef HAVE_SYS_WAIT_H
#include <sys/wait.h>
#endif
#include <stdlib.h>
#ifdef HAVE_UNISTD_H
#include <unistd.h>
#endif

#include "opal/util/few.h"
#include "opal/util/basename.h"
#include "opal/util/argv.h"
#include "opal/constants.h"

int opal_few(char *argv[], int *status)
{
#ifndef __WINDOWS__
#if defined(HAVE_FORK) && defined(HAVE_EXECVE) && defined(HAVE_WAITPID)
    pid_t pid, ret;

    if ((pid = fork()) < 0) {
      return OPAL_ERR_IN_ERRNO;
    }

    /* Child execs.  If it fails to exec, exit. */

    else if (0 == pid) {    
      execvp(argv[0], argv);
      exit(errno);
    }

    /* Parent loops waiting for the child to die. */

    else {  
      do {
        /* If the child exited, return */

        if (pid == (ret = waitpid(pid, status, 0))) {
          break;
        } 

        /* If waitpid was interrupted, loop around again */

        else if (ret < 0) {
          if (EINTR == errno) {
            continue;
          }

          /* Otherwise, some bad juju happened -- need to quit */

          return OPAL_ERR_IN_ERRNO;
        }
      } while (true);
    }

    /* Return the status to the caller */

    return OPAL_SUCCESS;
#else
    return OPAL_ERR_NOT_SUPPORTED;
#endif

#else

    STARTUPINFO si;
    PROCESS_INFORMATION pi;
    DWORD process_stat;
    char* command  = argv[0];
    char* exec_command;

    ZeroMemory (&si, sizeof(si));
    ZeroMemory (&pi, sizeof(pi));

    _flushall();  /* Push all output */

    GetStartupInfo (&si);
    argv[0] = opal_basename( command );
    exec_command = opal_argv_join( argv, ' ' );
    free( argv[0] );
    argv[0] = command;
    if (!CreateProcess (argv[0],
                        (LPSTR)exec_command,
                        NULL,
                        NULL,
                        TRUE,
                        0,
                        NULL,
                        NULL,
                        &si,
                        &pi)){
        *status = (int)GetLastError();
        return OPAL_ERROR;
    }

    /* wait for child to die */
    WaitForSingleObject(pi.hProcess, INFINITE);
    if( 0 == GetExitCodeProcess(pi.hProcess, &process_stat) ) {
        *status = (int)GetLastError();
        return OPAL_ERROR;
    }
    *status = (int)process_stat;

    return OPAL_SUCCESS;
#endif
}