/* Modify a job:
* CMD=MODIFYJOB ARG=<jobid>
* [BANK=<name>;]
* [COMMENT=<whatever>;]
* [DEPEND=afterany:<jobid>;]
* [JOBNAME=<name>;]
* [MINSTARTTIME=<uts>;]
* [NODES=<number>;]
* [PARTITION=<name>;]
* [RFEATURES=<features>;]
* [TIMELIMT=<seconds>;]
* [VARIABLELIST=<env_vars>;]
* [GRES=<name:value>;]
* [WCKEY=<name>;]
*
* RET 0 on success, -1 on failure */
extern int job_modify_wiki(char *cmd_ptr, int *err_code, char **err_msg)
{
char *arg_ptr, *bank_ptr, *depend_ptr, *nodes_ptr, *start_ptr;
char *host_ptr, *name_ptr, *part_ptr, *time_ptr, *tmp_char;
char *comment_ptr, *feature_ptr, *env_ptr, *gres_ptr, *wckey_ptr;
int i, slurm_rc;
uint32_t jobid, new_node_cnt = 0, new_time_limit = 0;
static char reply_msg[128];
/* Locks: write job, read node and partition info */
slurmctld_lock_t job_write_lock = {
NO_LOCK, WRITE_LOCK, READ_LOCK, READ_LOCK };
arg_ptr = strstr(cmd_ptr, "ARG=");
if (arg_ptr == NULL) {
*err_code = -300;
*err_msg = "MODIFYJOB lacks ARG=";
error("wiki: MODIFYJOB lacks ARG=");
return -1;
}
/* Change all parsed "=" to ":" then search for remaining "="
* and report results as unrecognized options */
arg_ptr[3] = ':';
arg_ptr += 4;
jobid = strtoul(arg_ptr, &tmp_char, 10);
if ((tmp_char[0] != '\0') && (!isspace(tmp_char[0]))) {
*err_code = -300;
*err_msg = "Invalid ARG value";
error("wiki: MODIFYJOB has invalid jobid");
return -1;
}
bank_ptr = strstr(cmd_ptr, "BANK=");
comment_ptr = strstr(cmd_ptr, "COMMENT=");
depend_ptr = strstr(cmd_ptr, "DEPEND=");
gres_ptr = strstr(cmd_ptr, "GRES=");
host_ptr = strstr(cmd_ptr, "HOSTLIST=");
name_ptr = strstr(cmd_ptr, "JOBNAME=");
start_ptr = strstr(cmd_ptr, "MINSTARTTIME=");
nodes_ptr = strstr(cmd_ptr, "NODES=");
part_ptr = strstr(cmd_ptr, "PARTITION=");
feature_ptr = strstr(cmd_ptr, "RFEATURES=");
time_ptr = strstr(cmd_ptr, "TIMELIMIT=");
env_ptr = strstr(cmd_ptr, "VARIABLELIST=");
wckey_ptr = strstr(cmd_ptr, "WCKEY=");
if (bank_ptr) {
bank_ptr[4] = ':';
bank_ptr += 5;
null_term(bank_ptr);
}
if (comment_ptr) {
comment_ptr[7] = ':';
comment_ptr += 8;
if (comment_ptr[0] == '\"') {
comment_ptr++;
for (i=0; ; i++) {
if (comment_ptr[i] == '\0')
break;
if (comment_ptr[i] == '\"') {
comment_ptr[i] = '\0';
break;
}
}
} else if (comment_ptr[0] == '\'') {
comment_ptr++;
for (i=0; ; i++) {
if (comment_ptr[i] == '\0')
break;
if (comment_ptr[i] == '\'') {
comment_ptr[i] = '\0';
break;
}
}
} else
null_term(comment_ptr);
}
if (depend_ptr) {
depend_ptr[6] = ':';
depend_ptr += 7;
null_term(depend_ptr);
}
if (feature_ptr) {
feature_ptr[9] = ':';
feature_ptr += 10;
null_term(feature_ptr);
}
if (gres_ptr) {
gres_ptr[4] = ':';
gres_ptr += 5;
null_term(gres_ptr);
}
if (host_ptr) {
host_ptr[8] = ':';
host_ptr += 9;
null_term(host_ptr);
}
if (name_ptr) {
name_ptr[7] = ':';
name_ptr += 8;
if (name_ptr[0] == '\"') {
name_ptr++;
for (i=0; ; i++) {
if (name_ptr[i] == '\0')
break;
if (name_ptr[i] == '\"') {
name_ptr[i] = '\0';
break;
}
}
} else if (name_ptr[0] == '\'') {
name_ptr++;
for (i=0; ; i++) {
if (name_ptr[i] == '\0')
break;
if (name_ptr[i] == '\'') {
name_ptr[i] = '\0';
break;
}
}
} else
null_term(name_ptr);
}
if (start_ptr) {
start_ptr[12] = ':';
start_ptr += 13;
null_term(start_ptr);
}
if (nodes_ptr) {
nodes_ptr[5] = ':';
nodes_ptr += 6;
new_node_cnt = strtoul(nodes_ptr, NULL, 10);
}
if (part_ptr) {
part_ptr[9] = ':';
part_ptr += 10;
null_term(part_ptr);
}
if (time_ptr) {
time_ptr[9] = ':';
time_ptr += 10;
new_time_limit = strtoul(time_ptr, NULL, 10);
}
if (env_ptr) {
env_ptr[12] = ':';
env_ptr += 13;
null_term(env_ptr);
}
if (wckey_ptr) {
wckey_ptr[5] = ':';
wckey_ptr += 6;
null_term(wckey_ptr);
}
/* Look for any un-parsed "=" ignoring anything after VARIABLELIST
* which is expected to contain "=" in its value*/
tmp_char = strchr(cmd_ptr, '=');
if (tmp_char && (!env_ptr || (env_ptr > tmp_char))) {
tmp_char[0] = '\0';
while (tmp_char[-1] && (!isspace(tmp_char[-1])))
tmp_char--;
error("wiki: Invalid MODIFYJOB option %s", tmp_char);
}
lock_slurmctld(job_write_lock);
slurm_rc = _job_modify(jobid, bank_ptr, depend_ptr, host_ptr,
new_node_cnt, part_ptr, new_time_limit, name_ptr,
start_ptr, feature_ptr, env_ptr, comment_ptr,
gres_ptr, wckey_ptr);
unlock_slurmctld(job_write_lock);
if (slurm_rc != SLURM_SUCCESS) {
*err_code = -700;
*err_msg = slurm_strerror(slurm_rc);
error("wiki: Failed to modify job %u (%m)", jobid);
return -1;
}
snprintf(reply_msg, sizeof(reply_msg),
"job %u modified successfully", jobid);
*err_msg = reply_msg;
return 0;
}
/* Modify a job:
* CMD=MODIFYJOB ARG=<jobid> PARTITION=<name> NODES=<number>
* DEPEND=afterany:<jobid> TIMELIMT=<seconds> BANK=<name>
* RET 0 on success, -1 on failure */
extern int job_modify_wiki(char *cmd_ptr, int *err_code, char **err_msg)
{
char *arg_ptr, *bank_ptr, *depend_ptr, *nodes_ptr;
char *host_ptr, *part_ptr, *time_ptr, *tmp_char;
int slurm_rc;
uint32_t jobid, new_node_cnt = 0, new_time_limit = 0;
static char reply_msg[128];
/* Locks: write job, read node and partition info */
slurmctld_lock_t job_write_lock = {
NO_LOCK, WRITE_LOCK, READ_LOCK, READ_LOCK, NO_LOCK };
arg_ptr = strstr(cmd_ptr, "ARG=");
if (arg_ptr == NULL) {
*err_code = -300;
*err_msg = "MODIFYJOB lacks ARG=";
error("wiki: MODIFYJOB lacks ARG=");
return -1;
}
/* Change all parsed "=" to ":" then search for remaining "="
* and report results as unrecognized options */
arg_ptr[3] = ':';
arg_ptr += 4;
jobid = strtoul(arg_ptr, &tmp_char, 10);
if ((tmp_char[0] != '\0') && (!isspace(tmp_char[0]))) {
*err_code = -300;
*err_msg = "Invalid ARG value";
error("wiki: MODIFYJOB has invalid jobid");
return -1;
}
bank_ptr = strstr(cmd_ptr, "BANK=");
depend_ptr = strstr(cmd_ptr, "DEPEND=");
host_ptr = strstr(cmd_ptr, "HOSTLIST=");
nodes_ptr = strstr(cmd_ptr, "NODES=");
part_ptr = strstr(cmd_ptr, "PARTITION=");
time_ptr = strstr(cmd_ptr, "TIMELIMIT=");
if (bank_ptr) {
bank_ptr[4] = ':';
bank_ptr += 5;
null_term(bank_ptr);
}
if (depend_ptr) {
depend_ptr[6] = ':';
depend_ptr += 7;
null_term(depend_ptr);
}
if (host_ptr) {
host_ptr[8] = ':';
host_ptr += 9;
null_term(bank_ptr);
}
if (nodes_ptr) {
nodes_ptr[5] = ':';
nodes_ptr += 6;
new_node_cnt = strtoul(nodes_ptr, NULL, 10);
}
if (part_ptr) {
part_ptr[9] = ':';
part_ptr += 10;
null_term(part_ptr);
}
if (time_ptr) {
time_ptr[9] = ':';
time_ptr += 10;
new_time_limit = strtoul(time_ptr, NULL, 10);
}
/* Look for any un-parsed "=" */
tmp_char = strchr(cmd_ptr, '=');
if (tmp_char) {
tmp_char[0] = '\0';
while (tmp_char[-1] && (!isspace(tmp_char[-1])))
tmp_char--;
error("wiki: Invalid MODIFYJOB option %s", tmp_char);
}
lock_slurmctld(job_write_lock);
slurm_rc = _job_modify(jobid, bank_ptr, depend_ptr, host_ptr,
new_node_cnt, part_ptr, new_time_limit);
unlock_slurmctld(job_write_lock);
if (slurm_rc != SLURM_SUCCESS) {
*err_code = -700;
*err_msg = slurm_strerror(slurm_rc);
error("wiki: Failed to modify job %u (%m)", jobid);
return -1;
}
snprintf(reply_msg, sizeof(reply_msg),
"job %u modified successfully", jobid);
*err_msg = reply_msg;
return 0;
}
