pyxis/config.c at main · NVIDIA/pyxis · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/*
 * Copyright (c) 2020-2025, NVIDIA CORPORATION. All rights reserved.
 */

#include <stdio.h>
#include <string.h>

#include <slurm/spank.h>

#include "config.h"

int parse_bool(const char *s)
{
	if (strcmp(s, "1") == 0 || strcmp(s, "true") == 0 || strcmp(s, "yes") == 0 || strcmp(s, "y") == 0)
		return (true);

	if (strcmp(s, "0") == 0 || strcmp(s, "false") == 0 || strcmp(s, "no") == 0 || strcmp(s, "n") == 0)
		return (false);

	return (-1);
}

int pyxis_config_parse(struct plugin_config *config, int ac, char **av)
{
	int ret;
	const char *optarg;

	memset(config, 0, sizeof(*config));

	/*
	 * Since Slurm might not be configured to integrate with PAM and
	 * logind, we can't assume /run/user/<uid> will be present.
	 * Instead, we default to using a new directory under an existing tmpfs: /run/pyxis.
	 */
	strcpy(config->runtime_path, "/run/pyxis");
	config->execute_entrypoint = false;
	config->container_scope = SCOPE_GLOBAL;
	config->sbatch_support = true;
	config->use_enroot_load = false;
	config->importer_path[0] = '\0';
	config->use_squashfuse = false;

	for (int i = 0; i < ac; ++i) {
		if (strncmp("runtime_path=", av[i], 13) == 0) {
			optarg = av[i] + 13;
			ret = snprintf(config->runtime_path, sizeof(config->runtime_path), "%s", optarg);
			if (ret < 0 || ret >= sizeof(config->runtime_path)) {
				slurm_error("pyxis: runtime_path: path too long: %s", optarg);
				return (-1);
			}
		} else if (strncmp("execute_entrypoint=", av[i], 19) == 0) {
			optarg = av[i] + 19;
			ret = parse_bool(optarg);
			if (ret < 0) {
				slurm_error("pyxis: execute_entrypoint: invalid value: %s", optarg);
				return (-1);
			}
			config->execute_entrypoint = ret;
		} else if (strncmp("container_scope=", av[i], 16) == 0) {
			optarg = av[i] + 16;
			if (strcmp(optarg, "job") == 0)
				config->container_scope = SCOPE_JOB;
			else if (strcmp(optarg, "global") == 0)
				config->container_scope = SCOPE_GLOBAL;
			else {
				slurm_error("pyxis: container_scope: invalid value: %s", optarg);
				return (-1);
			}
		} else if (strncmp("sbatch_support=", av[i], 15) == 0) {
			optarg = av[i] + 15;
			ret = parse_bool(optarg);
			if (ret < 0) {
				slurm_error("pyxis: sbatch_support: invalid value: %s", optarg);
				return (-1);
			}
			config->sbatch_support = ret;
		} else if (strncmp("use_enroot_load=", av[i], 16) == 0) {
			optarg = av[i] + 16;
			ret = parse_bool(optarg);
			if (ret < 0) {
				slurm_error("pyxis: use_enroot_load: invalid value: %s", optarg);
				return (-1);
			}
			config->use_enroot_load = ret;
		} else if (strncmp("importer=", av[i], 9) == 0) {
			optarg = av[i] + 9;
			ret = snprintf(config->importer_path, sizeof(config->importer_path), "%s", optarg);
			if (ret < 0 || ret >= sizeof(config->importer_path)) {
				slurm_error("pyxis: importer: path too long: %s", optarg);
				return (-1);
			}
		} else if (strncmp("use_squashfuse=", av[i], 15) == 0) {
			optarg = av[i] + 15;
			ret = parse_bool(optarg);
			if (ret < 0) {
				slurm_error("pyxis: use_squashfuse: invalid value: %s", optarg);
				return (-1);
			}
			config->use_squashfuse = ret;
		} else {
			slurm_error("pyxis: unknown configuration option: %s", av[i]);
			return (-1);
		}
	}

	return (0);
}