Commit | Line | Data |
---|---|---|
b4aa79af JF |
1 | /* Cycript - Remove Execution Server and Disassembler |
2 | * Copyright (C) 2009 Jay Freeman (saurik) | |
3 | */ | |
4 | ||
5 | /* Modified BSD License {{{ */ | |
6 | /* | |
7 | * Redistribution and use in source and binary | |
8 | * forms, with or without modification, are permitted | |
9 | * provided that the following conditions are met: | |
10 | * | |
11 | * 1. Redistributions of source code must retain the | |
12 | * above copyright notice, this list of conditions | |
13 | * and the following disclaimer. | |
14 | * 2. Redistributions in binary form must reproduce the | |
15 | * above copyright notice, this list of conditions | |
16 | * and the following disclaimer in the documentation | |
17 | * and/or other materials provided with the | |
18 | * distribution. | |
19 | * 3. The name of the author may not be used to endorse | |
20 | * or promote products derived from this software | |
21 | * without specific prior written permission. | |
22 | * | |
23 | * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' | |
24 | * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, | |
25 | * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF | |
26 | * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE | |
27 | * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE | |
28 | * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, | |
29 | * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT | |
30 | * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR | |
31 | * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS | |
32 | * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF | |
33 | * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR | |
34 | * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN | |
35 | * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF | |
36 | * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. | |
37 | */ | |
38 | /* }}} */ | |
39 | ||
057f943f JF |
40 | #define _GNU_SOURCE |
41 | ||
30ddc20c | 42 | #include "cycript.hpp" |
057f943f JF |
43 | |
44 | #include <cstdio> | |
45 | #include <sstream> | |
46 | ||
47 | #include <setjmp.h> | |
48 | ||
49 | #include <readline/readline.h> | |
50 | #include <readline/history.h> | |
51 | ||
b09da87b JF |
52 | #include <sys/mman.h> |
53 | ||
54 | #include <errno.h> | |
55 | #include <unistd.h> | |
56 | ||
57 | #include <sys/types.h> | |
58 | #include <sys/stat.h> | |
59 | #include <fcntl.h> | |
60 | ||
057f943f JF |
61 | #include "Cycript.tab.hh" |
62 | ||
967067aa JF |
63 | #include <sys/types.h> |
64 | #include <sys/socket.h> | |
65 | #include <netinet/in.h> | |
66 | #include <sys/un.h> | |
67 | ||
4e39dc0b JF |
68 | static volatile enum { |
69 | Working, | |
70 | Parsing, | |
71 | Running, | |
72 | Sending, | |
73 | Waiting, | |
74 | } mode_; | |
75 | ||
057f943f JF |
76 | static jmp_buf ctrlc_; |
77 | ||
579ed526 | 78 | static void sigint(int) { |
4e39dc0b JF |
79 | switch (mode_) { |
80 | case Working: | |
81 | return; | |
82 | case Parsing: | |
83 | longjmp(ctrlc_, 1); | |
84 | case Running: | |
85 | throw "*** Ctrl-C"; | |
86 | case Sending: | |
87 | return; | |
88 | case Waiting: | |
89 | return; | |
90 | } | |
057f943f JF |
91 | } |
92 | ||
cac61857 JF |
93 | #if YYDEBUG |
94 | static bool bison_; | |
95 | #endif | |
b10bd496 | 96 | static bool strict_; |
11c1cc16 | 97 | static bool pretty_; |
cac61857 | 98 | |
b10bd496 | 99 | void Setup(CYDriver &driver, cy::parser &parser) { |
cac61857 JF |
100 | #if YYDEBUG |
101 | if (bison_) | |
102 | parser.set_debug_level(1); | |
103 | #endif | |
b10bd496 JF |
104 | if (strict_) |
105 | driver.strict_ = true; | |
cac61857 JF |
106 | } |
107 | ||
3b52fd1a | 108 | void Setup(CYOutput &out, CYDriver &driver) { |
11c1cc16 | 109 | out.pretty_ = pretty_; |
3b52fd1a JF |
110 | |
111 | CYContext context(driver.pool_); | |
112 | driver.program_->Replace(context); | |
11c1cc16 JF |
113 | } |
114 | ||
48e3be8a | 115 | void Run(int socket, const char *data, size_t size, FILE *fout = NULL, bool expand = false) { |
967067aa | 116 | CYPool pool; |
b09da87b | 117 | |
967067aa | 118 | const char *json; |
6ec85c5b | 119 | if (socket == -1) { |
4e39dc0b | 120 | mode_ = Running; |
2aa77fd8 | 121 | #ifdef CY_EXECUTE |
fcc64bb5 | 122 | json = CYExecute(pool, data); |
2aa77fd8 JF |
123 | #else |
124 | json = NULL; | |
125 | #endif | |
4e39dc0b | 126 | mode_ = Working; |
6ec85c5b JF |
127 | if (json != NULL) |
128 | size = strlen(json); | |
129 | } else { | |
4e39dc0b | 130 | mode_ = Sending; |
967067aa JF |
131 | CYSendAll(socket, &size, sizeof(size)); |
132 | CYSendAll(socket, data, size); | |
4e39dc0b | 133 | mode_ = Waiting; |
967067aa JF |
134 | CYRecvAll(socket, &size, sizeof(size)); |
135 | if (size == _not(size_t)) | |
136 | json = NULL; | |
137 | else { | |
138 | char *temp(new(pool) char[size + 1]); | |
139 | CYRecvAll(socket, temp, size); | |
140 | temp[size] = '\0'; | |
141 | json = temp; | |
142 | } | |
4e39dc0b | 143 | mode_ = Working; |
4cf49641 | 144 | } |
b09da87b | 145 | |
967067aa | 146 | if (json != NULL && fout != NULL) { |
6ec85c5b JF |
147 | if (!expand || json[0] != '"' && json[0] != '\'') |
148 | fputs(json, fout); | |
149 | else for (size_t i(0); i != size; ++i) | |
150 | if (json[i] != '\\') | |
151 | fputc(json[i], fout); | |
152 | else switch(json[++i]) { | |
153 | case '\0': goto done; | |
154 | case '\\': fputc('\\', fout); break; | |
155 | case '\'': fputc('\'', fout); break; | |
156 | case '"': fputc('"', fout); break; | |
157 | case 'b': fputc('\b', fout); break; | |
158 | case 'f': fputc('\f', fout); break; | |
159 | case 'n': fputc('\n', fout); break; | |
160 | case 'r': fputc('\r', fout); break; | |
161 | case 't': fputc('\t', fout); break; | |
162 | case 'v': fputc('\v', fout); break; | |
163 | default: fputc('\\', fout); --i; break; | |
164 | } | |
165 | ||
166 | done: | |
967067aa JF |
167 | fputs("\n", fout); |
168 | fflush(fout); | |
b09da87b JF |
169 | } |
170 | } | |
171 | ||
48e3be8a | 172 | void Run(int socket, std::string &code, FILE *fout = NULL, bool expand = false) { |
6ec85c5b | 173 | Run(socket, code.c_str(), code.size(), fout, expand); |
fcc64bb5 JF |
174 | } |
175 | ||
967067aa | 176 | static void Console(int socket) { |
1f8eae40 JF |
177 | bool bypass(false); |
178 | bool debug(false); | |
6ec85c5b | 179 | bool expand(false); |
1f8eae40 | 180 | |
057f943f JF |
181 | FILE *fout(stdout); |
182 | ||
183 | rl_bind_key('\t', rl_insert); | |
184 | ||
185 | struct sigaction action; | |
186 | sigemptyset(&action.sa_mask); | |
187 | action.sa_handler = &sigint; | |
188 | action.sa_flags = 0; | |
189 | sigaction(SIGINT, &action, NULL); | |
190 | ||
191 | restart: for (;;) { | |
192 | std::string command; | |
193 | std::vector<std::string> lines; | |
194 | ||
931b816a JF |
195 | bool extra(false); |
196 | const char *prompt("cy# "); | |
197 | ||
057f943f | 198 | if (setjmp(ctrlc_) != 0) { |
4e39dc0b | 199 | mode_ = Working; |
057f943f JF |
200 | fputs("\n", fout); |
201 | fflush(fout); | |
202 | goto restart; | |
203 | } | |
204 | ||
057f943f | 205 | read: |
4e39dc0b | 206 | mode_ = Parsing; |
057f943f | 207 | char *line(readline(prompt)); |
4e39dc0b | 208 | mode_ = Working; |
057f943f JF |
209 | if (line == NULL) |
210 | break; | |
931b816a JF |
211 | |
212 | if (!extra) { | |
213 | extra = true; | |
6ec85c5b | 214 | if (line[0] == '?') { |
1f8eae40 JF |
215 | std::string data(line + 1); |
216 | if (data == "bypass") { | |
217 | bypass = !bypass; | |
218 | fprintf(fout, "bypass == %s\n", bypass ? "true" : "false"); | |
219 | fflush(fout); | |
220 | } else if (data == "debug") { | |
221 | debug = !debug; | |
222 | fprintf(fout, "debug == %s\n", debug ? "true" : "false"); | |
223 | fflush(fout); | |
6ec85c5b JF |
224 | } else if (data == "expand") { |
225 | expand = !expand; | |
226 | fprintf(fout, "expand == %s\n", expand ? "true" : "false"); | |
227 | fflush(fout); | |
1f8eae40 | 228 | } |
d35a3b07 | 229 | add_history(line); |
931b816a JF |
230 | goto restart; |
231 | } | |
232 | } | |
233 | ||
057f943f | 234 | command += line; |
e818f0e0 JF |
235 | |
236 | char *begin(line), *end(line + strlen(line)); | |
237 | while (char *nl = reinterpret_cast<char *>(memchr(begin, '\n', end - begin))) { | |
238 | *nl = '\0'; | |
239 | lines.push_back(begin); | |
240 | begin = nl + 1; | |
241 | } | |
242 | ||
243 | lines.push_back(begin); | |
244 | ||
057f943f JF |
245 | free(line); |
246 | ||
1f8eae40 JF |
247 | std::string code; |
248 | ||
249 | if (bypass) | |
250 | code = command; | |
251 | else { | |
252 | CYDriver driver(""); | |
253 | cy::parser parser(driver); | |
b10bd496 | 254 | Setup(driver, parser); |
1f8eae40 JF |
255 | |
256 | driver.data_ = command.c_str(); | |
257 | driver.size_ = command.size(); | |
258 | ||
259 | if (parser.parse() != 0 || !driver.errors_.empty()) { | |
f0360d51 JF |
260 | for (CYDriver::Errors::const_iterator error(driver.errors_.begin()); error != driver.errors_.end(); ++error) { |
261 | cy::position begin(error->location_.begin); | |
b10bd496 | 262 | if (begin.line != lines.size() || begin.column - 1 != lines.back().size() || error->warning_) { |
f0360d51 | 263 | cy::position end(error->location_.end); |
48e3be8a | 264 | |
f0360d51 JF |
265 | if (begin.line != lines.size()) { |
266 | std::cerr << " | "; | |
267 | std::cerr << lines[begin.line - 1] << std::endl; | |
268 | } | |
48e3be8a | 269 | |
c52a09b8 | 270 | std::cerr << "...."; |
f0360d51 JF |
271 | for (size_t i(0); i != begin.column - 1; ++i) |
272 | std::cerr << '.'; | |
48e3be8a | 273 | if (begin.line != end.line || begin.column == end.column) |
f0360d51 JF |
274 | std::cerr << '^'; |
275 | else for (size_t i(0), e(end.column - begin.column); i != e; ++i) | |
276 | std::cerr << '^'; | |
277 | std::cerr << std::endl; | |
48e3be8a | 278 | |
f0360d51 JF |
279 | std::cerr << " | "; |
280 | std::cerr << error->message_ << std::endl; | |
48e3be8a | 281 | |
1f8eae40 JF |
282 | add_history(command.c_str()); |
283 | goto restart; | |
284 | } | |
285 | } | |
057f943f | 286 | |
1f8eae40 | 287 | driver.errors_.clear(); |
057f943f | 288 | |
1f8eae40 JF |
289 | command += '\n'; |
290 | prompt = "cy> "; | |
291 | goto read; | |
057f943f JF |
292 | } |
293 | ||
b10bd496 | 294 | if (driver.program_ == NULL) |
1f8eae40 | 295 | goto restart; |
057f943f | 296 | |
967067aa JF |
297 | if (socket != -1) |
298 | code = command; | |
299 | else { | |
300 | std::ostringstream str; | |
652ec1ba | 301 | CYOutput out(str); |
3b52fd1a JF |
302 | Setup(out, driver); |
303 | out << *driver.program_; | |
967067aa JF |
304 | code = str.str(); |
305 | } | |
057f943f JF |
306 | } |
307 | ||
057f943f JF |
308 | add_history(command.c_str()); |
309 | ||
1f8eae40 JF |
310 | if (debug) |
311 | std::cout << code << std::endl; | |
057f943f | 312 | |
6ec85c5b | 313 | Run(socket, code, fout, expand); |
b09da87b | 314 | } |
057f943f | 315 | |
b09da87b JF |
316 | fputs("\n", fout); |
317 | fflush(fout); | |
318 | } | |
057f943f | 319 | |
579ed526 | 320 | static void *Map(const char *path, size_t *psize) { |
b09da87b JF |
321 | int fd; |
322 | _syscall(fd = open(path, O_RDONLY)); | |
057f943f | 323 | |
b09da87b JF |
324 | struct stat stat; |
325 | _syscall(fstat(fd, &stat)); | |
326 | size_t size(stat.st_size); | |
057f943f | 327 | |
b09da87b | 328 | *psize = size; |
057f943f | 329 | |
b09da87b JF |
330 | void *base; |
331 | _syscall(base = mmap(NULL, size, PROT_READ, MAP_SHARED, fd, 0)); | |
057f943f | 332 | |
b09da87b JF |
333 | _syscall(close(fd)); |
334 | return base; | |
335 | } | |
057f943f | 336 | |
967067aa | 337 | int main(int argc, char *argv[]) { |
48e3be8a | 338 | bool tty(isatty(STDIN_FILENO)); |
75b0a457 | 339 | bool compile(false); |
967067aa | 340 | |
2aa77fd8 JF |
341 | #ifdef CY_ATTACH |
342 | pid_t pid(_not(pid_t)); | |
343 | #endif | |
344 | ||
345 | for (;;) switch (getopt(argc, argv, | |
346 | "cg:n:" | |
347 | #ifdef CY_ATTACH | |
348 | "p:" | |
349 | #endif | |
350 | "s" | |
351 | )) { | |
967067aa JF |
352 | case -1: |
353 | goto getopt; | |
354 | case '?': | |
2aa77fd8 JF |
355 | fprintf(stderr, "usage: cycript [-c]" |
356 | #ifdef CY_ATTACH | |
357 | " [-p <pid>]" | |
358 | #endif | |
359 | " [<script> [<arg>...]]\n"); | |
967067aa JF |
360 | return 1; |
361 | ||
75b0a457 JF |
362 | case 'c': |
363 | compile = true; | |
364 | break; | |
365 | ||
cac61857 JF |
366 | case 'g': |
367 | if (false); | |
368 | #if YYDEBUG | |
369 | else if (strcmp(optarg, "bison") == 0) | |
370 | bison_ = true; | |
371 | #endif | |
372 | else { | |
373 | fprintf(stderr, "invalid name for -g\n"); | |
374 | return 1; | |
375 | } | |
376 | break; | |
377 | ||
11c1cc16 JF |
378 | case 'n': |
379 | if (false); | |
380 | else if (strcmp(optarg, "minify") == 0) | |
381 | pretty_ = true; | |
382 | else { | |
383 | fprintf(stderr, "invalid name for -n\n"); | |
384 | return 1; | |
385 | } | |
386 | break; | |
387 | ||
2aa77fd8 | 388 | #ifdef CY_ATTACH |
967067aa JF |
389 | case 'p': { |
390 | size_t size(strlen(optarg)); | |
391 | char *end; | |
392 | pid = strtoul(optarg, &end, 0); | |
393 | if (optarg + size != end) { | |
394 | fprintf(stderr, "invalid pid for -p\n"); | |
395 | return 1; | |
396 | } | |
397 | } break; | |
2aa77fd8 | 398 | #endif |
b10bd496 JF |
399 | |
400 | case 's': | |
401 | strict_ = true; | |
402 | break; | |
967067aa JF |
403 | } getopt:; |
404 | ||
b09da87b JF |
405 | const char *script; |
406 | ||
2aa77fd8 | 407 | #ifdef CY_ATTACH |
75b0a457 | 408 | if (pid != _not(pid_t) && optind < argc - 1) { |
fcc64bb5 JF |
409 | fprintf(stderr, "-p cannot set argv\n"); |
410 | return 1; | |
411 | } | |
412 | ||
75b0a457 JF |
413 | if (pid != _not(pid_t) && compile) { |
414 | fprintf(stderr, "-p conflicts with -c\n"); | |
415 | return 1; | |
416 | } | |
2aa77fd8 | 417 | #endif |
75b0a457 | 418 | |
967067aa | 419 | if (optind == argc) |
b09da87b JF |
420 | script = NULL; |
421 | else { | |
967067aa JF |
422 | // XXX: const_cast?! wtf gcc :( |
423 | CYSetArgs(argc - optind - 1, const_cast<const char **>(argv + optind + 1)); | |
424 | script = argv[optind]; | |
425 | if (strcmp(script, "-") == 0) | |
426 | script = NULL; | |
b09da87b JF |
427 | } |
428 | ||
2aa77fd8 JF |
429 | #ifdef CY_ATTACH |
430 | if (pid != _not(pid_t) && script == NULL && !tty) { | |
431 | fprintf(stderr, "non-terminal attaching to remote console\n"); | |
48e3be8a JF |
432 | return 1; |
433 | } | |
2aa77fd8 | 434 | #endif |
48e3be8a | 435 | |
967067aa JF |
436 | int socket; |
437 | ||
2aa77fd8 | 438 | #ifdef CY_ATTACH |
967067aa JF |
439 | if (pid == _not(pid_t)) |
440 | socket = -1; | |
441 | else { | |
442 | socket = _syscall(::socket(PF_UNIX, SOCK_STREAM, 0)); | |
443 | ||
444 | struct sockaddr_un address; | |
445 | memset(&address, 0, sizeof(address)); | |
446 | address.sun_family = AF_UNIX; | |
447 | sprintf(address.sun_path, "/tmp/.s.cy.%u", pid); | |
448 | ||
449 | _syscall(connect(socket, reinterpret_cast<sockaddr *>(&address), SUN_LEN(&address))); | |
450 | } | |
2aa77fd8 JF |
451 | #else |
452 | socket = -1; | |
453 | #endif | |
579ed526 | 454 | |
48e3be8a | 455 | if (script == NULL && tty) |
967067aa | 456 | Console(socket); |
b09da87b | 457 | else { |
48e3be8a | 458 | CYDriver driver(script ?: "<stdin>"); |
b09da87b | 459 | cy::parser parser(driver); |
b10bd496 | 460 | Setup(driver, parser); |
b09da87b | 461 | |
48e3be8a | 462 | char *start, *end; |
b09da87b | 463 | |
48e3be8a JF |
464 | if (script == NULL) { |
465 | start = NULL; | |
466 | end = NULL; | |
2b1245e4 | 467 | |
48e3be8a JF |
468 | driver.file_ = stdin; |
469 | } else { | |
470 | size_t size; | |
471 | start = reinterpret_cast<char *>(Map(script, &size)); | |
472 | end = start + size; | |
473 | ||
474 | if (size >= 2 && start[0] == '#' && start[1] == '!') { | |
475 | start += 2; | |
057f943f | 476 | |
48e3be8a JF |
477 | if (void *line = memchr(start, '\n', end - start)) |
478 | start = reinterpret_cast<char *>(line); | |
479 | else | |
480 | start = end; | |
481 | } | |
482 | ||
483 | driver.data_ = start; | |
484 | driver.size_ = end - start; | |
485 | } | |
62ca2b82 | 486 | |
b09da87b JF |
487 | if (parser.parse() != 0 || !driver.errors_.empty()) { |
488 | for (CYDriver::Errors::const_iterator i(driver.errors_.begin()); i != driver.errors_.end(); ++i) | |
489 | std::cerr << i->location_.begin << ": " << i->message_ << std::endl; | |
b10bd496 | 490 | } else if (driver.program_ != NULL) |
fcc64bb5 JF |
491 | if (socket != -1) |
492 | Run(socket, start, end - start, stdout); | |
493 | else { | |
494 | std::ostringstream str; | |
652ec1ba | 495 | CYOutput out(str); |
3b52fd1a JF |
496 | Setup(out, driver); |
497 | out << *driver.program_; | |
fcc64bb5 | 498 | std::string code(str.str()); |
75b0a457 JF |
499 | if (compile) |
500 | std::cout << code; | |
501 | else | |
c0bc320e | 502 | Run(socket, code, stdout); |
fcc64bb5 | 503 | } |
057f943f | 504 | } |
62ca2b82 | 505 | |
057f943f | 506 | return 0; |
62ca2b82 | 507 | } |