/* Alter a job's actual attributes. */
int
schd_alterjob(int sv_conn, Job *job, char *name, char *value, char *rsrc)
{
char *id = "schd_alterjob";
int err = 0;
AttrList atp;
/* Fill the attribute structure with function parameters */
atp.resource = rsrc;
atp.value = value;
atp.name = name;
atp.next = NULL;
err = pbs_alterjob(sv_conn, job->jobid, &atp, NULL);
if (err)
{
(void)sprintf(log_buffer,
"pbs_alterjob(%s, %s, %s, %s) failed: %d",
job->jobid, name, value, rsrc, pbs_errno);
log_record(PBSEVENT_SYSTEM, PBS_EVENTCLASS_SERVER, id, log_buffer);
}
return (err);
}
/* Alter a job's actual attributes. */
int
schd_alterjob(int sv_conn, Job *job, char *name, char *value, char *rsrc)
{
char *id = "schd_alterjob";
int err;
AttrList atp;
if (schd_TEST_ONLY)
{
DBPRT(("%s: Would have altered job %s:\n", id, job->jobid));
DBPRT(("%s: rsrc: %s, name: %s, value: %s\n", id,
rsrc ? rsrc : "(NULL)",
name ? name : "(NULL)",
value ? value : "(NULL)"));
return 0;
}
/* Fill the attribute structure with function parameters */
atp.resource = rsrc;
atp.value = value;
atp.name = name;
atp.next = NULL;
err = pbs_alterjob(sv_conn, job->jobid, &atp, NULL);
if (err)
{
(void)sprintf(log_buffer,
"pbs_alterjob(%s, %s, %s, %s) failed: %d",
job->jobid, name, value, rsrc, pbs_errno);
log_record(PBSEVENT_SYSTEM, PBS_EVENTCLASS_SERVER, id, log_buffer);
}
return (err);
}
int main(int argc, char **argv) {
char *server = NULL;
char *jobid = NULL;
char *var = NULL;
char *value = NULL;
int server_fd = 0;
int ret = 0;
int c = 0;
struct batch_status *job = NULL;
struct attrl *attribute = NULL;
char *var_string = NULL;
struct option prg_options[] = {
{"help", no_argument, 0, 'h'},
{"version", no_argument, 0, 'V'},
};
for ( ; ; ) {
int option_index = 0;
c = getopt_long(argc, argv, "s:hV",
prg_options, &option_index
);
if (c == -1) break;
switch (c) {
case 'h':
usage(0);
break;
case 'V':
printf("qsetenv version: %s; for torque version %s\n", QSETENV_VERSION, TORQUE_VERSION);
exit(0);
break;
case 's':
server = optarg;
break;
}
}
for (c = optind; c != argc; c++) {
switch (c-optind) {
case 0:
jobid = argv[c];
break;
case 1:
var = argv[c];
break;
case 2:
value = argv[c];
break;
default:
printf("Too many arguments!\n");
usage(1);
break;
}
}
if (value == NULL) {
printf("Too few arguments!\n");
usage(1);
}
if (server == NULL) {
server = pbs_get_server_list();
}
char *tok_server = server;
char *tgt_server = NULL;
while ((tgt_server = strtok(tok_server, ",")) != NULL) {
tok_server = NULL;
server_fd = pbs_connect(tgt_server);
if (server_fd > 0) {
break;
}
}
if (server_fd <= 0) {
fprintf(stderr, "Failed to connect to PBS server!\n");
exit(1);
}
printf("Querying job %s\n", jobid);
job = pbs_statjob(server_fd, jobid, NULL, 0);
if (job != NULL) {
printf("job name: %s\n", job->name);
var_string = job_setenv_varstr(job, var, value);
attribute = (struct attrl *) malloc(sizeof(struct attrl));
memset(attribute, 0, sizeof(struct attrl));
attribute->name = ATTR_v;
attribute->value = var_string;
attribute->next = NULL;
ret = pbs_alterjob(server_fd, jobid, attribute, NULL);
if (ret != 0) {
printf("Got error: %s\n", pbs_strerror(pbs_errno));
}
free(attribute);
attribute = NULL;
}
if (var_string != NULL) {
free(var_string);
}
if (job != NULL) {
pbs_statfree(job);
job = NULL;
}
pbs_disconnect(server_fd);
if (ret != 0) {
return 1;
}
return 0;
}
int
main(int argc, char **argv, char **envp) /* qalter */
{
int c;
int errflg=0;
int any_failed=0;
char *pc;
int i;
struct attrl *attrib = NULL;
char *keyword;
char *valuewd;
char *erplace;
time_t after;
char a_value[80];
char job_id[PBS_MAXCLTJOBID];
char job_id_out[PBS_MAXCLTJOBID];
char server_out[MAXSERVERNAME];
char rmt_server[MAXSERVERNAME];
struct ecl_attribute_errors *err_list;
#ifdef WIN32
struct attrl *ap = NULL;
short int nSizeofHostName = 0;
char* orig_apvalue = NULL;
char* temp_apvalue = NULL;
#endif
#define GETOPT_ARGS "a:A:c:e:h:j:k:l:m:M:N:o:p:r:R:S:u:W:P:"
/*test for real deal or just version and exit*/
execution_mode(argc, argv);
#ifdef WIN32
winsock_init();
#endif
while ((c = getopt(argc, argv, GETOPT_ARGS)) != EOF)
switch (c) {
case 'a':
if ((after = cvtdate(optarg)) < 0) {
fprintf(stderr, "qalter: illegal -a value\n");
errflg++;
break;
}
sprintf(a_value, "%ld", (long)after);
set_attr(&attrib, ATTR_a, a_value);
break;
case 'A':
set_attr(&attrib, ATTR_A, optarg);
break;
case 'P':
set_attr(&attrib, ATTR_project, optarg);
break;
case 'c':
while (isspace((int)*optarg)) optarg++;
pc = optarg;
if ((pc[0] == 'u') && (pc[1] == '\0')) {
fprintf(stderr, "qalter: illegal -c value\n");
errflg++;
break;
}
set_attr(&attrib, ATTR_c, optarg);
break;
case 'e':
set_attr(&attrib, ATTR_e, optarg);
break;
case 'h':
while (isspace((int)*optarg)) optarg++;
set_attr(&attrib, ATTR_h, optarg);
break;
case 'j':
set_attr(&attrib, ATTR_j, optarg);
break;
case 'k':
set_attr(&attrib, ATTR_k, optarg);
break;
case 'l':
if ((i = set_resources(&attrib, optarg, TRUE, &erplace)) != 0) {
if (i > 1) {
pbs_prt_parse_err("qalter: illegal -l value\n", optarg,
erplace-optarg, i);
} else
fprintf(stderr, "qalter: illegal -l value\n");
errflg++;
}
break;
case 'm':
while (isspace((int)*optarg)) optarg++;
set_attr(&attrib, ATTR_m, optarg);
break;
case 'M':
set_attr(&attrib, ATTR_M, optarg);
break;
case 'N':
set_attr(&attrib, ATTR_N, optarg);
break;
case 'o':
set_attr(&attrib, ATTR_o, optarg);
break;
case 'p':
while (isspace((int)*optarg)) optarg++;
set_attr(&attrib, ATTR_p, optarg);
break;
case 'r':
if (strlen(optarg) != 1) {
fprintf(stderr, "qalter: illegal -r value\n");
errflg++;
break;
}
if (*optarg != 'y' && *optarg != 'n') {
fprintf(stderr, "qalter: illegal -r value\n");
errflg++;
break;
}
set_attr(&attrib, ATTR_r, optarg);
break;
case 'R':
set_attr(&attrib, ATTR_R, optarg);
break;
case 'S':
set_attr(&attrib, ATTR_S, optarg);
break;
case 'u':
set_attr(&attrib, ATTR_u, optarg);
break;
case 'W':
while (isspace((int)*optarg)) optarg++;
if (strlen(optarg) == 0) {
fprintf(stderr, "qalter: illegal -W value\n");
errflg++;
break;
}
#ifdef WIN32
back2forward_slash2(optarg);
#endif
i = parse_equal_string(optarg, &keyword, &valuewd);
while (i == 1) {
set_attr(&attrib, keyword, valuewd);
i = parse_equal_string(NULL, &keyword, &valuewd);
}
if (i == -1) {
fprintf(stderr, "qalter: illegal -W value\n");
errflg++;
}
break;
case '?':
default :
errflg++;
break;
}
if (errflg || optind == argc) {
print_usage();
exit(2);
}
/*perform needed security library initializations (including none)*/
if (CS_client_init() != CS_SUCCESS) {
fprintf(stderr, "qalter: unable to initialize security library.\n");
exit(1);
}
for (; optind < argc; optind++) {
int connect;
int stat=0;
int located = FALSE;
strcpy(job_id, argv[optind]);
if (get_server(job_id, job_id_out, server_out)) {
fprintf(stderr, "qalter: illegally formed job identifier: %s\n", job_id);
any_failed = 1;
continue;
}
cnt:
connect = cnt2server(server_out);
if (connect <= 0) {
fprintf(stderr, "qalter: cannot connect to server %s (errno=%d)\n",
pbs_server, pbs_errno);
any_failed = pbs_errno;
continue;
}
stat = pbs_alterjob(connect, job_id_out, attrib, NULL);
if (stat && (pbs_errno != PBSE_UNKJOBID)) {
if ((err_list = pbs_get_attributes_in_error(connect)))
handle_attribute_errors(connect, err_list, job_id_out);
prt_job_err("qalter", connect, job_id_out);
any_failed = pbs_errno;
} else if (stat && (pbs_errno == PBSE_UNKJOBID) && !located) {
located = TRUE;
if (locate_job(job_id_out, server_out, rmt_server)) {
pbs_disconnect(connect);
strcpy(server_out, rmt_server);
goto cnt;
}
prt_job_err("qalter", connect, job_id_out);
any_failed = pbs_errno;
}
pbs_disconnect(connect);
}
CS_close_app();
exit(any_failed);
}