sbcast - Add --nodelist/-w option

Issue #2

Author: benglines

src/bcast/file_bcast.c

@@ -151,6 +151,22 @@ static int _get_cred_for_job(struct bcast_parameters *params)
 		      job_id_str, slurm_strerror(errno));
 		return rc;
 	}
+
+	if (params->node_list) {
+		hostset_t *job_nodes = hostset_create(sbcast_cred->node_list);
+
+		if (!hostset_within(job_nodes, params->node_list)) {
+			error("Not all nodes in --nodelist/-w '%s' are in job %s.",
+			      params->node_list, job_id_str);
+			FREE_NULL_HOSTSET(job_nodes);
+			return SLURM_ERROR;
+		}
+		FREE_NULL_HOSTSET(job_nodes);
+
+		xfree(sbcast_cred->node_list);
+		sbcast_cred->node_list = xstrdup(params->node_list);
+	}
+
 	verbose("jobid      = %s", job_id_str);
 	verbose("node_list  = %s", sbcast_cred->node_list);
 

src/bcast/file_bcast.h

@@ -52,6 +52,7 @@ struct bcast_parameters {
 	char *dst_fname;
 	char *exe_fname;
 	uint16_t flags;
+	char *node_list;
 	slurm_selected_step_t *selected_step;
 	char *src_fname;
 	uint32_t step_id;

src/sbcast/opts.c

@@ -87,6 +87,7 @@ extern void parse_command_line(int argc, char **argv)
 		{"treewidth",    required_argument, 0, OPT_LONG_TREE_WIDTH},
 		{"force",     no_argument,       0, 'f'},
 		{"jobid",     required_argument, 0, 'j'},
+		{"nodelist",  optional_argument, 0, 'w'},
 		{"send-libs", optional_argument, 0, OPT_LONG_SEND_LIBS},
 		{"preserve",  no_argument,       0, 'p'},
 		{"size",      required_argument, 0, 's'},
@@ -148,7 +149,7 @@ extern void parse_command_line(int argc, char **argv)
 		params.timeout = (atoi(env_val) * 1000);
 
 	optind = 0;
-	while ((opt_char = getopt_long(argc, argv, "C::fF:j:ps:t:vV",
+	while ((opt_char = getopt_long(argc, argv, "C::fF:j:ps:t:vVw:",
 			long_options, &option_index)) != -1) {
 		switch (opt_char) {
 		case (int)'?':
@@ -201,6 +202,9 @@ extern void parse_command_line(int argc, char **argv)
 		case (int) 'V':
 			print_slurm_version();
 			exit(0);
+		case (int) 'w':
+			params.node_list = xstrdup(optarg);
+			break;
 		case (int) OPT_LONG_HELP:
 			_help();
 			exit(0);