1 /* Cycript - Optimizing JavaScript Compiler/Runtime
2 * Copyright (C) 2009-2010 Jay Freeman (saurik)
5 /* GNU Lesser General Public License, Version 3 {{{ */
7 * Cycript is free software: you can redistribute it and/or modify it under
8 * the terms of the GNU Lesser General Public License as published by the
9 * Free Software Foundation, either version 3 of the License, or (at your
10 * option) any later version.
12 * Cycript is distributed in the hope that it will be useful, but WITHOUT
13 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
14 * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public
15 * License for more details.
17 * You should have received a copy of the GNU Lesser General Public License
18 * along with Cycript. If not, see <http://www.gnu.org/licenses/>.
22 #include "cycript.hpp"
25 #include "JavaScript.hpp"
33 #ifdef HAVE_READLINE_H
36 #include <readline/readline.h>
42 #include <readline/history.h>
50 #include <sys/types.h>
54 #include "Cycript.tab.hh"
56 #include <sys/types.h>
57 #include <sys/socket.h>
58 #include <netinet/in.h>
62 #include <apr_getopt.h>
66 #include "Replace.hpp"
68 static volatile enum {
76 static jmp_buf ctrlc_
;
78 static void sigint(int) {
99 void Setup(CYDriver
&driver
, cy::parser
&parser
) {
102 parser
.set_debug_level(1);
105 driver
.strict_
= true;
108 void Setup(CYOutput
&out
, CYDriver
&driver
, CYOptions
&options
) {
109 out
.pretty_
= pretty_
;
110 CYContext
context(options
);
111 driver
.program_
->Replace(context
);
114 static CYUTF8String
Run(CYPool
&pool
, int client
, CYUTF8String code
) {
121 json
= CYExecute(pool
, code
);
133 CYSendAll(client
, &size
, sizeof(size
));
134 CYSendAll(client
, code
.data
, code
.size
);
136 CYRecvAll(client
, &size
, sizeof(size
));
137 if (size
== _not(size_t))
140 char *temp(new(pool
) char[size
+ 1]);
141 CYRecvAll(client
, temp
, size
);
148 return CYUTF8String(json
, size
);
151 static CYUTF8String
Run(CYPool
&pool
, int client
, const std::string
&code
) {
152 return Run(pool
, client
, CYUTF8String(code
.c_str(), code
.size()));
157 static void Output(CYUTF8String json
, FILE *fout
, bool expand
= false) {
158 const char *data(json
.data
);
159 size_t size(json
.size
);
161 if (data
== NULL
|| fout
== NULL
)
164 if (!expand
|| data
[0] != '"' && data
[0] != '\'')
166 else for (size_t i(0); i
!= size
; ++i
)
168 fputc(data
[i
], fout
);
169 else switch(data
[++i
]) {
170 case '\0': goto done
;
171 case '\\': fputc('\\', fout
); break;
172 case '\'': fputc('\'', fout
); break;
173 case '"': fputc('"', fout
); break;
174 case 'b': fputc('\b', fout
); break;
175 case 'f': fputc('\f', fout
); break;
176 case 'n': fputc('\n', fout
); break;
177 case 'r': fputc('\r', fout
); break;
178 case 't': fputc('\t', fout
); break;
179 case 'v': fputc('\v', fout
); break;
180 default: fputc('\\', fout
); --i
; break;
188 static void Run(int client
, const char *data
, size_t size
, FILE *fout
= NULL
, bool expand
= false) {
190 Output(Run(pool
, client
, CYUTF8String(data
, size
)), fout
, expand
);
193 static void Run(int client
, std::string
&code
, FILE *fout
= NULL
, bool expand
= false) {
194 Run(client
, code
.c_str(), code
.size(), fout
, expand
);
197 int (*append_history$
)(int, const char *);
199 static std::string command_
;
201 static CYExpression
*ParseExpression(CYUTF8String code
) {
202 std::ostringstream str
;
203 str
<< '(' << code
<< ')';
204 std::string
string(str
.str());
207 driver
.data_
= string
.c_str();
208 driver
.size_
= string
.size();
210 cy::parser
parser(driver
);
211 Setup(driver
, parser
);
213 if (parser
.parse() != 0 || !driver
.errors_
.empty())
216 CYExpress
*express(dynamic_cast<CYExpress
*>(driver
.program_
->statements_
));
217 _assert(express
!= NULL
);
218 return express
->expression_
;
223 static char **Complete(const char *word
, int start
, int end
) {
224 rl_attempted_completion_over
= TRUE
;
228 cy::parser
parser(driver
);
229 Setup(driver
, parser
);
231 std::string
line(rl_line_buffer
, start
);
232 std::string
command(command_
+ line
);
234 driver
.data_
= command
.c_str();
235 driver
.size_
= command
.size();
239 if (parser
.parse() != 0 || !driver
.errors_
.empty())
242 if (driver
.mode_
== CYDriver::AutoNone
)
245 CYExpression
*expression
;
248 CYContext
context(options
);
250 std::ostringstream prefix
;
252 switch (driver
.mode_
) {
253 case CYDriver::AutoPrimary
:
254 expression
= $
CYThis();
257 case CYDriver::AutoDirect
:
258 expression
= driver
.context_
;
261 case CYDriver::AutoIndirect
:
262 expression
= $
CYIndirect(driver
.context_
);
265 case CYDriver::AutoMessage
: {
266 CYDriver::Context
&thing(driver
.contexts_
.back());
267 expression
= $
M($
M($
CYIndirect(thing
.context_
), $
S("isa")), $
S("messages"));
268 for (CYDriver::Context::Words::const_iterator
part(thing
.words_
.begin()); part
!= thing
.words_
.end(); ++part
)
269 prefix
<< (*part
)->word_
<< ':';
276 std::string
begin(prefix
.str());
278 driver
.program_
= $
CYProgram($
CYExpress($
C3(ParseExpression(
279 " function(object, prefix, word) {\n"
281 " var pattern = '^' + prefix + word;\n"
282 " var length = prefix.length;\n"
283 " for (name in object)\n"
284 " if (name.match(pattern) != null)\n"
285 " names.push(name.substr(length));\n"
288 ), expression
, $
S(begin
.c_str()), $
S(word
))));
290 driver
.program_
->Replace(context
);
292 std::ostringstream str
;
293 CYOutput
out(str
, options
);
294 out
<< *driver
.program_
;
296 std::string
code(str
.str());
297 CYUTF8String
json(Run(pool
, client_
, code
));
299 CYExpression
*result(ParseExpression(json
));
300 CYArray
*array(dynamic_cast<CYArray
*>(result
));
303 fprintf(fout_
, "\n");
305 rl_forced_update_display();
309 // XXX: use an std::set?
310 typedef std::vector
<std::string
> Completions
;
311 Completions completions
;
316 CYForEach (element
, array
->elements_
) {
317 CYString
*string(dynamic_cast<CYString
*>(element
->value_
));
318 _assert(string
!= NULL
);
320 std::string completion
;
321 if (string
->size_
!= 0)
322 completion
.assign(string
->value_
, string
->size_
);
323 else if (driver
.mode_
== CYDriver::AutoMessage
)
328 completions
.push_back(completion
);
334 size_t limit(completion
.size()), size(common
.size());
336 common
= common
.substr(0, limit
);
339 for (limit
= 0; limit
!= size
; ++limit
)
340 if (common
[limit
] != completion
[limit
])
343 common
= common
.substr(0, limit
);
347 size_t count(completions
.size());
351 size_t colon(common
.find(':'));
352 if (colon
!= std::string::npos
)
353 common
= common
.substr(0, colon
+ 1);
354 if (completions
.size() == 1)
357 char **results(reinterpret_cast<char **>(malloc(sizeof(char *) * (count
+ 2))));
359 results
[0] = strdup(common
.c_str());
361 for (Completions::const_iterator
i(completions
.begin()); i
!= completions
.end(); ++i
)
362 results
[++index
] = strdup(i
->c_str());
363 results
[count
+ 1] = NULL
;
368 // need char *, not const char *
369 static char name_
[] = "cycript";
370 static char break_
[] = " \t\n\"\\'`@$><=;|&{(" ")}" ".:[]";
372 static void Console(CYOptions
&options
) {
376 if (const char *username
= getenv("LOGNAME"))
377 passwd
= getpwnam(username
);
379 passwd
= getpwuid(getuid());
381 const char *basedir(apr_psprintf(pool
, "%s/.cycript", passwd
->pw_dir
));
382 const char *histfile(apr_psprintf(pool
, "%s/history", basedir
));
386 rl_readline_name
= name_
;
388 mkdir(basedir
, 0700);
389 read_history(histfile
);
397 // rl_completer_word_break_characters is broken in libedit
398 rl_basic_word_break_characters
= break_
;
400 rl_completer_word_break_characters
= break_
;
401 rl_attempted_completion_function
= &Complete
;
402 rl_bind_key('\t', rl_complete
);
404 struct sigaction action
;
405 sigemptyset(&action
.sa_mask
);
406 action
.sa_handler
= &sigint
;
408 sigaction(SIGINT
, &action
, NULL
);
412 std::vector
<std::string
> lines
;
415 const char *prompt("cy# ");
417 if (setjmp(ctrlc_
) != 0) {
426 char *line(readline(prompt
));
435 if (line
[0] == '?') {
436 std::string
data(line
+ 1);
437 if (data
== "bypass") {
439 fprintf(fout_
, "bypass == %s\n", bypass
? "true" : "false");
441 } else if (data
== "debug") {
443 fprintf(fout_
, "debug == %s\n", debug
? "true" : "false");
445 } else if (data
== "expand") {
447 fprintf(fout_
, "expand == %s\n", expand
? "true" : "false");
458 char *begin(line
), *end(line
+ strlen(line
));
459 while (char *nl
= reinterpret_cast<char *>(memchr(begin
, '\n', end
- begin
))) {
461 lines
.push_back(begin
);
465 lines
.push_back(begin
);
476 cy::parser
parser(driver
);
477 Setup(driver
, parser
);
479 driver
.data_
= command_
.c_str();
480 driver
.size_
= command_
.size();
482 if (parser
.parse() != 0 || !driver
.errors_
.empty()) {
483 for (CYDriver::Errors::const_iterator
error(driver
.errors_
.begin()); error
!= driver
.errors_
.end(); ++error
) {
484 cy::position
begin(error
->location_
.begin
);
485 if (begin
.line
!= lines
.size() || begin
.column
- 1 != lines
.back().size() || error
->warning_
) {
486 cy::position
end(error
->location_
.end
);
488 if (begin
.line
!= lines
.size()) {
490 std::cerr
<< lines
[begin
.line
- 1] << std::endl
;
494 for (size_t i(0); i
!= begin
.column
- 1; ++i
)
496 if (begin
.line
!= end
.line
|| begin
.column
== end
.column
)
498 else for (size_t i(0), e(end
.column
- begin
.column
); i
!= e
; ++i
)
500 std::cerr
<< std::endl
;
503 std::cerr
<< error
->message_
<< std::endl
;
505 add_history(command_
.c_str());
511 driver
.errors_
.clear();
518 if (driver
.program_
== NULL
)
524 std::ostringstream str
;
525 CYOutput
out(str
, options
);
526 Setup(out
, driver
, options
);
527 out
<< *driver
.program_
;
532 add_history(command_
.c_str());
536 std::cout
<< code
<< std::endl
;
538 Run(client_
, code
, fout_
, expand
);
541 if (append_history$
!= NULL
) {
542 _syscall(close(_syscall(open(histfile
, O_CREAT
| O_WRONLY
, 0600))));
543 (*append_history$
)(histlines
, histfile
);
545 write_history(histfile
);
552 static void *Map(const char *path
, size_t *psize
) {
554 _syscall(fd
= open(path
, O_RDONLY
));
557 _syscall(fstat(fd
, &stat
));
558 size_t size(stat
.st_size
);
563 _syscall(base
= mmap(NULL
, size
, PROT_READ
, MAP_SHARED
, fd
, 0));
569 void InjectLibrary(pid_t pid
);
571 int Main(int argc
, char const * const argv
[], char const * const envp
[]) {
572 bool tty(isatty(STDIN_FILENO
));
576 append_history$
= reinterpret_cast<int (*)(int, const char *)>(dlsym(RTLD_DEFAULT
, "append_history"));
579 pid_t
pid(_not(pid_t
));
584 _aprcall(apr_getopt_init(&state
, pool
, argc
, argv
));
590 apr_status_t
status(apr_getopt(state
,
604 "usage: cycript [-c]"
608 " [<script> [<arg>...]]\n"
622 else if (strcmp(arg
, "rename") == 0)
623 options
.verbose_
= true;
625 else if (strcmp(arg
, "bison") == 0)
629 fprintf(stderr
, "invalid name for -g\n");
636 else if (strcmp(arg
, "minify") == 0)
639 fprintf(stderr
, "invalid name for -n\n");
646 size_t size(strlen(arg
));
649 pid
= strtoul(arg
, &end
, 0);
650 if (arg
+ size
!= end
) {
651 // XXX: arg needs to be escaped in some horrendous way of doom
652 const char *command(apr_psprintf(pool
, "ps axc|sed -e '/^ *[0-9]/{s/^ *\\([0-9]*\\)\\( *[^ ]*\\)\\{3\\} *-*\\([^ ]*\\)/\\3 \\1/;/^%s /{s/^[^ ]* //;q;};};d'", arg
));
654 if (FILE *pids
= popen(command
, "r")) {
659 size_t read(fread(value
+ size
, 1, sizeof(value
) - size
, pids
));
664 if (size
== sizeof(value
)) {
674 if (value
[size
- 1] == '\n') {
680 size
= strlen(value
);
681 pid
= strtoul(value
, &end
, 0);
682 if (value
+ size
!= end
) fail
:
684 _syscall(pclose(pids
));
687 if (pid
== _not(pid_t
)) {
688 fprintf(stderr
, "invalid pid for -p\n");
705 if (pid
!= _not(pid_t
) && ind
< argc
- 1) {
706 fprintf(stderr
, "-p cannot set argv\n");
710 if (pid
!= _not(pid_t
) && compile
) {
711 fprintf(stderr
, "-p conflicts with -c\n");
720 // XXX: const_cast?! wtf gcc :(
721 CYSetArgs(argc
- ind
- 1, const_cast<const char **>(argv
+ ind
+ 1));
724 if (strcmp(script
, "-") == 0)
729 if (pid
!= _not(pid_t
) && script
== NULL
&& !tty
) {
730 fprintf(stderr
, "non-terminal attaching to remote console\n");
736 if (pid
== _not(pid_t
))
739 int server(_syscall(socket(PF_UNIX
, SOCK_STREAM
, 0))); try {
740 struct sockaddr_un address
;
741 memset(&address
, 0, sizeof(address
));
742 address
.sun_family
= AF_UNIX
;
744 sprintf(address
.sun_path
, "/tmp/.s.cy.%u", getpid());
746 _syscall(bind(server
, reinterpret_cast<sockaddr
*>(&address
), SUN_LEN(&address
)));
747 _syscall(chmod(address
.sun_path
, 0777));
750 _syscall(listen(server
, 1));
752 client_
= _syscall(accept(server
, NULL
, NULL
));
755 unlink(address
.sun_path
);
759 _syscall(close(server
));
767 if (script
== NULL
&& tty
)
771 CYDriver
driver(script
?: "<stdin>");
772 cy::parser
parser(driver
);
773 Setup(driver
, parser
);
777 if (script
== NULL
) {
781 driver
.file_
= stdin
;
784 start
= reinterpret_cast<char *>(Map(script
, &size
));
787 if (size
>= 2 && start
[0] == '#' && start
[1] == '!') {
790 if (void *line
= memchr(start
, '\n', end
- start
))
791 start
= reinterpret_cast<char *>(line
);
796 driver
.data_
= start
;
797 driver
.size_
= end
- start
;
800 if (parser
.parse() != 0 || !driver
.errors_
.empty()) {
801 for (CYDriver::Errors::const_iterator
i(driver
.errors_
.begin()); i
!= driver
.errors_
.end(); ++i
)
802 std::cerr
<< i
->location_
.begin
<< ": " << i
->message_
<< std::endl
;
803 } else if (driver
.program_
!= NULL
)
805 std::string
code(start
, end
-start
);
806 Run(client_
, code
, stdout
);
808 std::ostringstream str
;
809 CYOutput
out(str
, options
);
810 Setup(out
, driver
, options
);
811 out
<< *driver
.program_
;
812 std::string
code(str
.str());
816 Run(client_
, code
, stdout
);
823 int main(int argc
, char const * const argv
[], char const * const envp
[]) {
824 apr_status_t
status(apr_app_initialize(&argc
, &argv
, &envp
));
826 if (status
!= APR_SUCCESS
) {
827 fprintf(stderr
, "apr_app_initialize() != APR_SUCCESS\n");
830 return Main(argc
, argv
, envp
);
831 } catch (const CYException
&error
) {
833 fprintf(stderr
, "%s\n", error
.PoolCString(pool
));