Fix ompi-server operations

Signed-off-by: Ralph Castain <[email protected]>

Author: Ralph Castain

orte/mca/rml/base/rml_base_contact.c

@@ -9,7 +9,7 @@
  *                         University of Stuttgart.  All rights reserved.
  * Copyright (c) 2004-2005 The Regents of the University of California.
  *                         All rights reserved.
- * Copyright (c) 2016      Intel, Inc.  All rights reserved.
+ * Copyright (c) 2016-2017 Intel, Inc. All rights reserved.
  * $COPYRIGHT$
  *
  * Additional copyrights may follow
@@ -72,6 +72,7 @@ int orte_rml_base_get_contact_info(orte_jobid_t job, opal_buffer_t *data)
 int orte_rml_base_update_contact_info(opal_buffer_t* data)
 {
     orte_std_cntr_t cnt;
+    orte_process_name_t peer;
     orte_vpid_t num_procs;
     char *rml_uri;
     int rc;
@@ -89,11 +90,18 @@ int orte_rml_base_update_contact_info(opal_buffer_t* data)
         if (NULL != rml_uri) {
             /* set the contact info into the hash table */
             orte_rml.set_contact_info(rml_uri);
+            /* if this was an update to my own job, then
+             * track how many procs were in the message */
+            if (ORTE_SUCCESS != (rc = orte_rml_base_parse_uris(rml_uri, &peer, NULL))) {
+                ORTE_ERROR_LOG(rc);
+                free(rml_uri);
+                return rc;
+            }
+            if (peer.jobid == ORTE_PROC_MY_NAME->jobid) {
+                ++num_procs;
+            }
             free(rml_uri);
         }
-
-        /* track how many procs were in the message */
-        ++num_procs;
     }
     if (ORTE_ERR_UNPACK_READ_PAST_END_OF_BUFFER != rc) {
         ORTE_ERROR_LOG(rc);

orte/orted/help-orted.txt

@@ -80,3 +80,10 @@ This is usually caused by a large job that encounters significant
 delays across the cluster when starting the application processes.
 Your job may terminate as a result of this problem. You may want to
 adjust the MCA parameter pmix_server_max_reqs and try again.
+#
+[noserver]
+A publish/lookup server was provided, but we were unable to connect
+to it - please check the connection info and ensure the server
+is alive:
+
+  Connection:  %s

orte/orted/pmix/pmix_server.c

@@ -296,94 +296,6 @@ int pmix_server_init(void)
     }
     OPAL_LIST_DESTRUCT(&info);
 
-    /* if the universal server wasn't specified, then we use
-     * our own HNP for that purpose */
-    if (NULL == orte_pmix_server_globals.server_uri) {
-        orte_pmix_server_globals.server = *ORTE_PROC_MY_HNP;
-    } else {
-        char *server;
-        opal_buffer_t buf;
-        if (0 == strncmp(orte_pmix_server_globals.server_uri, "file", strlen("file")) ||
-            0 == strncmp(orte_pmix_server_globals.server_uri, "FILE", strlen("FILE"))) {
-            char input[1024], *filename;
-            FILE *fp;
-
-            /* it is a file - get the filename */
-            filename = strchr(orte_pmix_server_globals.server_uri, ':');
-            if (NULL == filename) {
-                /* filename is not correctly formatted */
-                orte_show_help("help-orterun.txt", "orterun:ompi-server-filename-bad", true,
-                               orte_basename, orte_pmix_server_globals.server_uri);
-                return ORTE_ERR_BAD_PARAM;
-            }
-            ++filename; /* space past the : */
-
-            if (0 >= strlen(filename)) {
-                /* they forgot to give us the name! */
-                orte_show_help("help-orterun.txt", "orterun:ompi-server-filename-missing", true,
-                               orte_basename, orte_pmix_server_globals.server_uri);
-                return ORTE_ERR_BAD_PARAM;
-            }
-
-            /* open the file and extract the uri */
-            fp = fopen(filename, "r");
-            if (NULL == fp) { /* can't find or read file! */
-                orte_show_help("help-orterun.txt", "orterun:ompi-server-filename-access", true,
-                               orte_basename, orte_pmix_server_globals.server_uri);
-                return ORTE_ERR_BAD_PARAM;
-            }
-            if (NULL == fgets(input, 1024, fp)) {
-                /* something malformed about file */
-                fclose(fp);
-                orte_show_help("help-orterun.txt", "orterun:ompi-server-file-bad", true,
-                               orte_basename, orte_pmix_server_globals.server_uri,
-                               orte_basename);
-                return ORTE_ERR_BAD_PARAM;
-            }
-            fclose(fp);
-            input[strlen(input)-1] = '\0';  /* remove newline */
-            server = strdup(input);
-        } else {
-            server = strdup(orte_pmix_server_globals.server_uri);
-        }
-        /* setup our route to the server */
-        OBJ_CONSTRUCT(&buf, opal_buffer_t);
-        opal_dss.pack(&buf, &server, 1, OPAL_STRING);
-        if (ORTE_SUCCESS != (rc = orte_rml_base_update_contact_info(&buf))) {
-            ORTE_ERROR_LOG(rc);
-            ORTE_UPDATE_EXIT_STATUS(ORTE_ERROR_DEFAULT_EXIT_CODE);
-            return rc;
-        }
-        OBJ_DESTRUCT(&buf);
-        /* parse the URI to get the server's name */
-        if (ORTE_SUCCESS != (rc = orte_rml_base_parse_uris(server, &orte_pmix_server_globals.server, NULL))) {
-            ORTE_ERROR_LOG(rc);
-            return rc;
-        }
-        /* check if we are to wait for the server to start - resolves
-         * a race condition that can occur when the server is run
-         * as a background job - e.g., in scripts
-         */
-        if (orte_pmix_server_globals.wait_for_server) {
-            /* ping the server */
-            struct timeval timeout;
-            timeout.tv_sec = orte_pmix_server_globals.timeout;
-            timeout.tv_usec = 0;
-            if (ORTE_SUCCESS != (rc = orte_rml.ping(orte_mgmt_conduit, server, &timeout))) {
-                /* try it one more time */
-                if (ORTE_SUCCESS != (rc = orte_rml.ping(orte_mgmt_conduit, server, &timeout))) {
-                    /* okay give up */
-                    orte_show_help("help-orterun.txt", "orterun:server-not-found", true,
-                                   orte_basename, server,
-                                   (long)orte_pmix_server_globals.timeout,
-                                   ORTE_ERROR_NAME(rc));
-                    ORTE_UPDATE_EXIT_STATUS(ORTE_ERROR_DEFAULT_EXIT_CODE);
-                    return rc;
-                }
-            }
-        }
-    }
-
     return rc;
 }
 
@@ -716,8 +628,9 @@ OBJ_CLASS_INSTANCE(orte_pmix_server_op_caddy_t,
 static void rqcon(pmix_server_req_t *p)
 {
     p->operation = NULL;
-    p->target = *ORTE_NAME_INVALID;
+    p->range = OPAL_PMIX_RANGE_SESSION;
     p->proxy = *ORTE_NAME_INVALID;
+    p->target = *ORTE_NAME_INVALID;
     p->timeout = orte_pmix_server_globals.timeout;
     p->jdata = NULL;
     OBJ_CONSTRUCT(&p->msg, opal_buffer_t);

orte/orted/pmix/pmix_server_internal.h

@@ -67,8 +67,9 @@
     int timeout;
     int room_num;
     int remote_room_num;
+    opal_pmix_data_range_t range;
     orte_process_name_t proxy;
-    opal_process_name_t target;
+    orte_process_name_t target;
     orte_job_t *jdata;
     opal_buffer_t msg;
     opal_pmix_op_cbfunc_t opcbfunc;
@@ -255,6 +256,7 @@ typedef struct {
     bool wait_for_server;
     orte_process_name_t server;
     opal_list_t notifications;
+    bool pubsub_init;
 } pmix_server_globals_t;
 
 extern pmix_server_globals_t orte_pmix_server_globals;