OILS / cpp / stdlib.cc View on Github | oilshell.org

229 lines, 125 significant
1// stdlib.cc: Replacement for standard library modules
2// and native/posixmodule.c
3
4#include "stdlib.h"
5
6#include <dirent.h> // closedir(), opendir(), readdir()
7#include <errno.h>
8#include <fcntl.h> // open
9#include <signal.h> // kill
10#include <sys/stat.h> // umask
11#include <sys/types.h> // umask
12#include <sys/wait.h> // WUNTRACED
13#include <time.h>
14#include <unistd.h>
15
16#include "mycpp/runtime.h"
17// To avoid circular dependency with e_die()
18#include "prebuilt/core/error.mycpp.h"
19
20using error::e_die;
21
22namespace fcntl_ {
23
24int fcntl(int fd, int cmd) {
25 int result = ::fcntl(fd, cmd);
26 if (result < 0) {
27 throw Alloc<IOError>(errno);
28 }
29 return result;
30}
31
32int fcntl(int fd, int cmd, int arg) {
33 int result = ::fcntl(fd, cmd, arg);
34 if (result < 0) {
35 throw Alloc<IOError>(errno);
36 }
37 return result;
38}
39
40} // namespace fcntl_
41
42namespace posix {
43
44mode_t umask(mode_t mask) {
45 // No error case: always succeeds
46 return ::umask(mask);
47}
48
49int open(BigStr* path, int flags, int perms) {
50 int result = ::open(path->data_, flags, perms);
51 if (result < 0) {
52 throw Alloc<OSError>(errno);
53 }
54 return result;
55}
56
57void dup2(int oldfd, int newfd) {
58 if (::dup2(oldfd, newfd) < 0) {
59 throw Alloc<OSError>(errno);
60 }
61}
62void putenv(BigStr* name, BigStr* value) {
63 int overwrite = 1;
64 int ret = ::setenv(name->data_, value->data_, overwrite);
65 if (ret < 0) {
66 throw Alloc<IOError>(errno);
67 }
68}
69
70mylib::File* fdopen(int fd, BigStr* c_mode) {
71 // CPython checks if it's a directory first
72 struct stat buf;
73 if (fstat(fd, &buf) == 0 && S_ISDIR(buf.st_mode)) {
74 throw Alloc<OSError>(EISDIR);
75 }
76
77 // CPython does some fcntl() stuff with mode == 'a', which we don't support
78 DCHECK(c_mode->data_[0] != 'a');
79
80 FILE* f = ::fdopen(fd, c_mode->data_);
81 if (f == nullptr) {
82 throw Alloc<OSError>(errno);
83 }
84
85 return Alloc<mylib::CFile>(f);
86}
87
88void execve(BigStr* argv0, List<BigStr*>* argv,
89 Dict<BigStr*, BigStr*>* environ) {
90 int n_args = len(argv);
91 // never deallocated
92 char** _argv = static_cast<char**>(malloc((n_args + 1) * sizeof(char*)));
93
94 // Annoying const_cast
95 // https://stackoverflow.com/questions/190184/execv-and-const-ness
96 for (int i = 0; i < n_args; ++i) {
97 _argv[i] = const_cast<char*>(argv->at(i)->data_);
98 }
99 _argv[n_args] = nullptr;
100
101 // Convert environ into an array of pointers to strings of the form: "k=v".
102 int n_env = len(environ);
103 char** envp = static_cast<char**>(malloc((n_env + 1) * sizeof(char*)));
104
105 int env_index = 0;
106 for (DictIter<BigStr*, BigStr*> it(environ); !it.Done(); it.Next()) {
107 BigStr* k = it.Key();
108 BigStr* v = it.Value();
109
110 int joined_len = len(k) + len(v) + 1;
111 char* buf = static_cast<char*>(malloc(joined_len + 1));
112 memcpy(buf, k->data_, len(k));
113 buf[len(k)] = '=';
114 memcpy(buf + len(k) + 1, v->data_, len(v));
115 buf[joined_len] = '\0';
116
117 envp[env_index++] = buf;
118 }
119 envp[n_env] = nullptr;
120
121 int ret = ::execve(argv0->data_, _argv, envp);
122 if (ret == -1) {
123 throw Alloc<OSError>(errno);
124 }
125
126 // ::execve() never returns on success
127 FAIL(kShouldNotGetHere);
128}
129
130void kill(int pid, int sig) {
131 if (::kill(pid, sig) != 0) {
132 throw Alloc<OSError>(errno);
133 }
134}
135
136void killpg(int pgid, int sig) {
137 if (::killpg(pgid, sig) != 0) {
138 throw Alloc<OSError>(errno);
139 }
140}
141
142List<BigStr*>* listdir(BigStr* path) {
143 DIR* dirp = opendir(path->data());
144 if (dirp == NULL) {
145 throw Alloc<OSError>(errno);
146 }
147
148 auto* ret = Alloc<List<BigStr*>>();
149 while (true) {
150 errno = 0;
151 struct dirent* ep = readdir(dirp);
152 if (ep == NULL) {
153 if (errno != 0) {
154 closedir(dirp);
155 throw Alloc<OSError>(errno);
156 }
157 break; // no more files
158 }
159 // Skip . and ..
160 int name_len = strlen(ep->d_name);
161 if (ep->d_name[0] == '.' &&
162 (name_len == 1 || (ep->d_name[1] == '.' && name_len == 2))) {
163 continue;
164 }
165 ret->append(StrFromC(ep->d_name, name_len));
166 }
167
168 closedir(dirp);
169
170 return ret;
171}
172
173} // namespace posix
174
175namespace time_ {
176
177void tzset() {
178 // No error case: no return value
179 ::tzset();
180}
181
182time_t time() {
183 time_t result = ::time(nullptr);
184 if (result < 0) {
185 throw Alloc<IOError>(errno);
186 }
187 return result;
188}
189
190// NOTE(Jesse): time_t is specified to be an arithmetic type by C++. On most
191// systems it's a 64-bit integer. 64 bits is used because 32 will overflow in
192// 2038. Someone on a committee somewhere thought of that when moving to
193// 64-bit architectures to prevent breaking ABI again; on 32-bit systems it's
194// usually 32 bits. Point being, using anything but the time_t typedef here
195// could (unlikely, but possible) produce weird behavior.
196time_t localtime(time_t ts) {
197 // localtime returns a pointer to a static buffer
198 tm* loc_time = ::localtime(&ts);
199
200 time_t result = mktime(loc_time);
201 if (result < 0) {
202 throw Alloc<IOError>(errno);
203 }
204 return result;
205}
206
207BigStr* strftime(BigStr* s, time_t ts) {
208 tm* loc_time = ::localtime(&ts);
209
210 const int max_len = 1024;
211 BigStr* result = OverAllocatedStr(max_len);
212 int n = strftime(result->data(), max_len, s->data_, loc_time);
213 if (n == 0) {
214 // bash silently truncates on large format string like
215 // printf '%(%Y)T'
216 // Oil doesn't mask errors
217 // Leaving out location info points to 'printf' builtin
218
219 e_die(StrFromC("strftime() result exceeds 1024 bytes"));
220 }
221 result->MaybeShrink(n);
222 return result;
223}
224
225void sleep(int seconds) {
226 ::sleep(seconds);
227}
228
229} // namespace time_