/* * 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 (c) 2006 Cisco Systems, Inc. All rights reserved. * Copyright (c) 2006 Sun Microsystems, Inc. All rights reserved. * Use is subject to license terms. * $COPYRIGHT$ * * Additional copyrights may follow * * $HEADER$ * * These symbols are in a file by themselves to provide nice linker * semantics. Since linkers generally pull in symbols by object * files, keeping these symbols as the only symbols in this file * prevents utility programs such as "ompi_info" from having to import * entire components just to query their version and parameters. */ /** * @file: * Part of the gridengine launcher. * See pls_gridengine.h for an overview of how it works. */ #include "orte_config.h" #include "orte/orte_constants.h" #include #ifdef HAVE_UNISTD_H #include #endif #include #include #ifdef HAVE_SYS_SELECT_H #include #endif #ifdef HAVE_SYS_TIME_H #include #endif #ifdef HAVE_SYS_TYPES_H #include #endif #ifdef HAVE_SYS_STAT_H #include #endif #ifdef HAVE_SYS_WAIT_H #include #endif #include #include #ifdef HAVE_PWD_H #include #endif #include "opal/mca/installdirs/installdirs.h" #include "opal/mca/base/mca_base_param.h" #include "opal/util/if.h" #include "opal/util/os_path.h" #include "opal/util/path.h" #include "opal/event/event.h" #include "opal/util/show_help.h" #include "opal/util/argv.h" #include "opal/util/opal_environ.h" #include "opal/util/output.h" #include "opal/util/basename.h" #include "orte/util/univ_info.h" #include "orte/util/session_dir.h" #include "orte/util/sys_info.h" #include "orte/runtime/orte_wait.h" #include "orte/mca/ns/ns.h" #include "orte/mca/rml/rml.h" #include "orte/mca/gpr/gpr.h" #include "orte/mca/errmgr/errmgr.h" #include "orte/mca/ras/ras_types.h" #include "orte/mca/rmaps/rmaps.h" #include "orte/mca/smr/smr.h" #include "orte/mca/pls/pls.h" #include "orte/mca/pls/base/base.h" #include "orte/mca/pls/base/pls_private.h" #include "orte/mca/pls/gridengine/pls_gridengine.h" #if !defined(__WINDOWS__) extern char **environ; #endif /* !defined(__WINDOWS__) */ orte_pls_base_module_t orte_pls_gridengine_module = { orte_pls_gridengine_launch_job, orte_pls_gridengine_terminate_job, orte_pls_gridengine_terminate_orteds, orte_pls_gridengine_terminate_proc, orte_pls_gridengine_signal_job, orte_pls_gridengine_signal_proc, orte_pls_gridengine_cancel_operation, orte_pls_gridengine_finalize }; static void set_handler_default(int sig); #if 0 static int update_slot_keyval(orte_ras_node_t* node, int* slot_cnt); #endif /** * Fill the orted_path variable with the directory to the orted */ static int orte_pls_gridengine_fill_orted_path(char** orted_path) { struct stat buf; asprintf(orted_path, "%s/orted", opal_install_dirs.bindir); if (0 != stat(*orted_path, &buf)) { char *path = getenv("PATH"); if (NULL == path) { path = ("PATH is empty!"); } opal_show_help("help-pls-gridengine.txt", "no-local-orted", true, path, opal_install_dirs.bindir); return ORTE_ERR_NOT_FOUND; } return ORTE_SUCCESS; } /** * Callback on daemon exit. */ static void orte_pls_gridengine_wait_daemon(pid_t pid, int status, void* cbdata) { orte_pls_daemon_info_t *info = (orte_pls_daemon_info_t*) cbdata; int rc; /* if qrsh exited abnormally, set the daemon's state to aborted and print something useful to the user. The usual reasons for qrsh to exit abnormally all are a pretty good indication that the child processes aren't going to start up properly, so this will signal the system to kill the job. This should somehow be pushed up to the calling level, but we don't really have a way to do that just yet. */ if (! WIFEXITED(status) || ! WEXITSTATUS(status) == 0) { /* tell the user something went wrong. We need to do this BEFORE we * set the state to ABORTED as that action will cause a trigger to * fire that will kill the job before any output would get printed! */ opal_output(0, "ERROR: A daemon on node %s failed to start as expected.", info->nodename); opal_output(0, "ERROR: There may be more information available from"); opal_output(0, "ERROR: the 'qstat -t' command on the Grid Engine tasks."); opal_output(0, "ERROR: If the problem persists, please restart the"); opal_output(0, "ERROR: Grid Engine PE job"); if (WIFEXITED(status)) { opal_output(0, "ERROR: The daemon exited unexpectedly with status %d.", WEXITSTATUS(status)); } else if (WIFSIGNALED(status)) { #ifdef WCOREDUMP if (WCOREDUMP(status)) { opal_output(0, "The daemon received a signal %d (with core).", WTERMSIG(status)); } else { opal_output(0, "The daemon received a signal %d.", WTERMSIG(status)); } #else opal_output(0, "The daemon received a signal %d.", WTERMSIG(status)); #endif /* WCOREDUMP */ } else { opal_output(0, "No extra status information is available: %d.", status); } /* now set the state to aborted */ rc = orte_smr.set_proc_state(info->name, ORTE_PROC_STATE_ABORTED, status); if (ORTE_SUCCESS != rc) { ORTE_ERROR_LOG(rc); } } /* cleanup */ OBJ_RELEASE(info); } /** * Launch a daemon (bootproxy) on each node. The daemon will be responsible * for launching the application. */ int orte_pls_gridengine_launch_job(orte_jobid_t jobid) { orte_job_map_t *map; opal_list_item_t *n_item; orte_std_cntr_t num_nodes; orte_vpid_t vpid; int node_name_index1; int node_name_index2; int proc_name_index; int orted_index; char *jobid_string; char *prefix_dir; char *param; char **argv; char **env; int argc; int rc; sigset_t sigs; char *lib_base = NULL, *bin_base = NULL; char *sge_root, *sge_arch; opal_list_t daemons; orte_pls_daemon_info_t *dmn; /* setup a list that will contain the info for all the daemons * so we can store it on the registry when done */ OBJ_CONSTRUCT(&daemons, opal_list_t); /* Get the map for this job. * We need the entire mapping for a couple of reasons: * - need the prefix to start with. * - need to know if we are launching on a subset of the allocated nodes * All other mapping responsibilities fall to orted in the fork PLS */ rc = orte_rmaps.get_job_map(&map, jobid); if (ORTE_SUCCESS != rc) { ORTE_ERROR_LOG(rc); OBJ_DESTRUCT(&daemons); return rc; } /* if the user requested that we re-use daemons, * launch the procs on any existing, re-usable daemons */ if (orte_pls_base.reuse_daemons) { if (ORTE_SUCCESS != (rc = orte_pls_base_launch_on_existing_daemons(map))) { ORTE_ERROR_LOG(rc); OBJ_RELEASE(map); OBJ_DESTRUCT(&daemons); return rc; } } num_nodes = (orte_std_cntr_t)opal_list_get_size(&map->nodes); if (num_nodes == 0) { /* job must have been launched on existing daemons - just return */ OBJ_RELEASE(map); OBJ_DESTRUCT(&daemons); return ORTE_SUCCESS; } /* * Allocate a range of vpids for the daemons. */ rc = orte_ns.reserve_range(0, num_nodes, &vpid); if (ORTE_SUCCESS != rc) { ORTE_ERROR_LOG(rc); goto cleanup; } /* setup the orted triggers for passing their launch info */ if (ORTE_SUCCESS != (rc = orte_smr.init_orted_stage_gates(jobid, num_nodes, NULL, NULL))) { ORTE_ERROR_LOG(rc); goto cleanup; } /* need integer value for command line parameter */ if (ORTE_SUCCESS != (rc = orte_ns.convert_jobid_to_string(&jobid_string, jobid))) { ORTE_ERROR_LOG(rc); goto cleanup; } /* * Build argv array */ argv = opal_argv_split("qrsh", ' '); argc = opal_argv_count(argv); /* gridengine specific flags */ opal_argv_append(&argc, &argv, "-inherit");/*run tasks within curr job*/ opal_argv_append(&argc, &argv, "-noshell");/*execute w/o wrapping shell*/ opal_argv_append(&argc, &argv, "-nostdin");/*suppress input stream stdin*/ opal_argv_append(&argc, &argv, "-V"); /*task to have the env as job*/ if (mca_pls_gridengine_component.verbose) { opal_argv_append(&argc, &argv, "-verbose"); } node_name_index1 = argc; opal_argv_append(&argc, &argv, "