aboutsummaryrefslogtreecommitdiff
path: root/core/env.c
blob: 0d014a848dda88bc0c30932ec0ad8d1df24fd56d (plain)
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
#include "../lib/io/io.h"
#include "../lib/sys/sizes.h"
#include "../lib/sys/execve.h"
#include "../lib/env/env.h"
#include "../lib/sys/types.h"
#include "../lib/cstr/cstr.h"

char buf[BUFSIZ] = "";

void add_to_env(char *line, u64 linenumber)
{
	char *begin = line;
	char *key;
	char *value;

	key = line;

	while (*line != '=' && *line) ++line;

	if (!*line) {
		wff(STDERR_FD, "error: '%s' %i no valid environment line\n", begin, linenumber);
		return;
	}

	*(line++) = 0;
	value = line;

	while (*line != '\n' && *line) ++line;

	if (*line == '\n') *line = 0;

	setenv(key, value);
}


int main(int argc, const char **argv)
{
	if (argc == 1) {
		const char**envp = getenvp();
		
		while (*envp) {
			wstdf("%s\n", *envp);
			++envp;
		}

		return 0;
	}

	u64 fd = STDIN_FD;
	i64 size;
	char *p;
	u64 linenumber = 1;
	const char **execargv = argv + 1;

	if (argv[1][0] == '-' && argv[1][1] == 'f') {
		if (argc > 3) {
			fd = open(argv[2], OPEN_READ_ONLY, 0);
			if (fd < 0) {
				wff(STDERR_FD, "venv: cannot open file '%s'\n", argv[2]);
				return -1;
			}
			execargv = argv + 3;
		} else {
			wf(STDERR_FD, "venv [-f file] command [args...]\n");
			return -1;
		}
	}

	while ((size = get_next_line_from_fd(fd, buf)) >= 0) {
		if (cstr_length(buf)) {
			add_to_env(buf, linenumber++);
		}
	}

	execve(execargv[0], execargv, getenvp());

	wff(STDERR_FD, "cannot execute '%s'\n", execargv[0]);
	return -1;
}