blob: 1bf7e21ab437bfd137995e0e59ed4c51278ca145 (
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
80
81
82
83
84
85
|
#include "../lib/env/env.h"
#include "../lib/cstr/cstr.h"
#include "../lib/malloc/malloc.h"
#include "../lib/io/io.h"
#include "env.h"
void clear_buf(char *buf, u64 size)
{
for (int i = 0; i < size; ++i) buf[i] = 0;
}
void init_arg_env(int argc, const char **argv)
{
char buf[32];
u64_to_cstr(argc - 1, buf, 32);
setenv("@", buf);
for (int i = 1; i < argc; ++i) {
clear_buf(buf, 32);
u64_to_cstr(i - 1, buf, 32);
setenv(buf, argv[i]);
}
}
char* new_line_and_replace_vars(const char *line, u64 n)
{
char var[64];
int size;
char *replaced;
for (int i = 0; i < n && line[i]; ++i) {
if (line[i] == '$') {
clear_buf(var, 64);
++i;
for (int j = 0; line[i] && line[i] != ' ';) {
var[j] = line[i];
++j;
++i;
}
if (cstr_length(var) == 0) {
size++;
} else {
const char *e = getenv(var);
if (e) size += cstr_length(e);
}
} else {
size++;
}
}
replaced = malloc(size + 1);
int line_pos = 0;
for (int i = 0; i < size && line[line_pos]; ++i) {
if (line[line_pos] == '$') {
clear_buf(var, 64);
++line_pos;
for (int j = 0; line[line_pos] && line[line_pos] != ' ';) {
var[j] = line[line_pos];
++j;
++line_pos;
}
if (cstr_length(var)) {
const char *e = getenv(var);
while (*e) {
replaced[i++] = *(e++);
}
}
} else {
replaced[i] = line[line_pos++];
}
}
replaced[size] = 0;
return replaced;
}
|
