static int _handle_kvs_put(int fd, int lrank, client_req_t *req) { int rc = SLURM_SUCCESS; client_resp_t *resp; char *key = NULL, *val = NULL; debug3("mpi/pmi2: in _handle_kvs_put"); client_req_parse_body(req); client_req_get_str(req, KEY_KEY, &key); client_req_get_str(req, VALUE_KEY, &val); /* no need to add k-v to hash. just get it ready to be up-forward */ rc = temp_kvs_add(key, val); xfree(key); xfree(val); resp = client_resp_new(); client_resp_append(resp, CMD_KEY"="KVSPUTRESP_CMD";" RC_KEY"=%d;", rc); rc = client_resp_send(resp, fd); client_resp_free(resp); debug3("mpi/pmi2: out _handle_kvs_put"); return rc; }
static int _handle_ring(int fd, int lrank, client_req_t *req) { int rc = SLURM_SUCCESS; int count = 0; char *left = NULL; char *right = NULL; debug3("mpi/pmi2: in _handle_ring"); /* extract left, right, and count values from ring payload */ client_req_parse_body(req); client_req_get_int(req, RING_COUNT_KEY, &count); client_req_get_str(req, RING_LEFT_KEY, &left); client_req_get_str(req, RING_RIGHT_KEY, &right); /* compute ring_id, we list all application tasks first, * followed by stepds, so here we just use the application * process rank */ int ring_id = lrank; rc = pmix_ring_in(ring_id, count, left, right); xfree(left); xfree(right); /* the repsonse is sent back to client from the pmix_ring_out call */ debug3("mpi/pmi2: out _handle_ring"); return rc; }
static int _handle_get(int fd, int lrank, client_req_t *req) { int rc; client_resp_t *resp; char *kvsname = NULL, *key = NULL, *val = NULL; debug3("mpi/pmi2: in _handle_get"); client_req_parse_body(req); client_req_get_str(req, KVSNAME_KEY, &kvsname); /* not used */ client_req_get_str(req, KEY_KEY, &key); val = kvs_get(key); resp = client_resp_new(); if (val != NULL) { client_resp_append(resp, CMD_KEY"="GETRESULT_CMD" " RC_KEY"=0 " VALUE_KEY"=%s\n", val); } else { client_resp_append(resp, CMD_KEY"="GETRESULT_CMD" " RC_KEY"=1\n"); } rc = client_resp_send(resp, fd); client_resp_free(resp); debug3("mpi/pmi2: out _handle_get"); return rc; }
static int _handle_put(int fd, int lrank, client_req_t *req) { int rc = SLURM_SUCCESS; client_resp_t *resp; char *kvsname = NULL, *key = NULL, *val = NULL; debug3("mpi/pmi2: in _handle_put"); client_req_parse_body(req); client_req_get_str(req, KVSNAME_KEY, &kvsname); /* not used */ client_req_get_str(req, KEY_KEY, &key); client_req_get_str(req, VALUE_KEY, &val); /* no need to add k-v to hash. just get it ready to be up-forward */ rc = temp_kvs_add(key, val); if (rc == SLURM_SUCCESS) rc = 0; else rc = 1; resp = client_resp_new(); client_resp_append(resp, CMD_KEY"="PUTRESULT_CMD" " RC_KEY"=%d\n", rc); rc = client_resp_send(resp, fd); client_resp_free(resp); debug3("mpi/pmi2: out _handle_put"); return rc; }
extern spawn_subcmd_t * client_req_parse_spawn_subcmd(client_req_t *req) { spawn_subcmd_t *subcmd = NULL; char buf[PMI2_MAX_KEYLEN]; int i = 0; subcmd = xmalloc(sizeof(spawn_subcmd_t)); client_req_get_str(req, EXECNAME_KEY, &subcmd->cmd); client_req_get_int(req, NPROCS_KEY, (int *)&subcmd->max_procs); client_req_get_int(req, ARGCNT_KEY, (int *)&subcmd->argc); subcmd->argv = xmalloc(subcmd->argc * sizeof(char *)); for (i = 0; i < subcmd->argc; i ++) { snprintf(buf, PMI2_MAX_KEYLEN, "arg%d", i + 1); client_req_get_str(req, buf, &(subcmd->argv[i])); } client_req_get_int(req, INFONUM_KEY, (int *)&subcmd->info_cnt); subcmd->info_keys = xmalloc(subcmd->info_cnt * sizeof(char *)); subcmd->info_vals = xmalloc(subcmd->info_cnt * sizeof(char *)); for (i = 0; i < subcmd->info_cnt; i ++) { snprintf(buf, PMI2_MAX_KEYLEN, "info_key_%d", i); client_req_get_str(req, buf, &(subcmd->info_keys[i])); snprintf(buf, PMI2_MAX_KEYLEN, "info_val_%d", i); client_req_get_str(req, buf, &(subcmd->info_vals[i])); } return subcmd; }
static int _handle_mcmd(int fd, int lrank, client_req_t *req) { spawn_subcmd_t *subcmd = NULL; spawn_resp_t *spawn_resp = NULL; client_resp_t *task_resp = NULL; int spawnssofar = 0, rc = SLURM_SUCCESS, i; char buf[64]; debug3("mpi/pmi2: in _handle_mcmd"); client_req_parse_body(req); subcmd = client_req_parse_spawn_subcmd(req); debug3("mpi/pmi2: got subcmd"); client_req_get_int(req, SPAWNSSOFAR_KEY, &spawnssofar); if (spawnssofar == 1) { pmi1_spawn = spawn_req_new(); client_req_get_int(req, TOTSPAWNS_KEY, (int *)&pmi1_spawn->subcmd_cnt); pmi1_spawn->subcmds = xmalloc(pmi1_spawn->subcmd_cnt * sizeof(spawn_subcmd_t *)); client_req_get_int(req, PREPUTNUM_KEY, (int *)&pmi1_spawn->preput_cnt); pmi1_spawn->pp_keys = xmalloc(pmi1_spawn->preput_cnt * sizeof(char *)); pmi1_spawn->pp_vals = xmalloc(pmi1_spawn->preput_cnt * sizeof(char *)); for (i = 0; i < pmi1_spawn->preput_cnt; i ++) { snprintf(buf, 64, PREPUTKEY_KEY"%d", i); client_req_get_str(req, buf, &pmi1_spawn->pp_keys[i]); snprintf(buf, 64, PREPUTVAL_KEY"%d", i); client_req_get_str(req, buf, &pmi1_spawn->pp_vals[i]); } } pmi1_spawn->subcmds[spawnssofar - 1] = subcmd; if (spawnssofar == pmi1_spawn->subcmd_cnt) { debug3("mpi/pmi2: got whole spawn req"); /* a resp will be send back from srun. this will not be forwarded to the tasks */ rc = spawn_req_send_to_srun(pmi1_spawn, &spawn_resp); if (spawn_resp->rc != SLURM_SUCCESS) { task_resp = client_resp_new(); client_resp_append(task_resp, CMD_KEY"="SPAWNRESP_CMD";" RC_KEY"=%d;" ERRMSG_KEY"=spawn failed;", spawn_resp->rc); client_resp_send(task_resp, fd); client_resp_free(task_resp); spawn_resp_free(spawn_resp); spawn_req_free(pmi1_spawn); pmi1_spawn = NULL; error("mpi/pmi2: spawn failed"); rc = SLURM_ERROR; goto out; } debug("mpi/pmi2: spawn request sent to srun"); spawn_psr_enqueue(spawn_resp->seq, fd, lrank, NULL); spawn_resp_free(spawn_resp); spawn_req_free(pmi1_spawn); pmi1_spawn = NULL; } out: debug3("mpi/pmi2: out _handle_mcmd"); return rc; }