X-Git-Url: https://git.saurik.com/cycript.git/blobdiff_plain/8d9b5eed2cbd4dd74437558400833a9931ad3078..930aa21ba5c097f9ee6ebfc418c321de0729e076:/Application.mm?ds=sidebyside diff --git a/Application.mm b/Application.mm index 106c7dd..46b24ac 100644 --- a/Application.mm +++ b/Application.mm @@ -1,7 +1,266 @@ +/* Cycript - Remove Execution Server and Disassembler + * Copyright (C) 2009 Jay Freeman (saurik) +*/ + +/* Modified BSD License {{{ */ +/* + * Redistribution and use in source and binary + * forms, with or without modification, are permitted + * provided that the following conditions are met: + * + * 1. Redistributions of source code must retain the + * above copyright notice, this list of conditions + * and the following disclaimer. + * 2. Redistributions in binary form must reproduce the + * above copyright notice, this list of conditions + * and the following disclaimer in the documentation + * and/or other materials provided with the + * distribution. + * 3. The name of the author may not be used to endorse + * or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, + * BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF + * MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHOR BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, + * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT + * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR + * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF + * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR + * TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN + * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF + * ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. +*/ +/* }}} */ + +#define _GNU_SOURCE + +#include +#include "cycript.hpp" + #include +#include + +#include + +#include +#include + +#include + +#include +#include + +#include +#include +#include + +#include "Cycript.tab.hh" + +static jmp_buf ctrlc_; + +static void sigint(int) { + longjmp(ctrlc_, 1); +} + +static JSStringRef Result_; + +void Run(const char *code, FILE *fout) { _pooled + JSStringRef script(JSStringCreateWithUTF8CString(code)); + + JSContextRef context(CYGetJSContext()); + + JSValueRef exception(NULL); + JSValueRef result(JSEvaluateScript(context, script, NULL, NULL, 0, &exception)); + JSStringRelease(script); + + if (exception != NULL) { error: + result = exception; + exception = NULL; + } + + if (!JSValueIsUndefined(context, result)) { + CYPool pool; + const char *json; + + json = CYPoolCYONString(pool, context, result, &exception); + if (exception != NULL) + goto error; + + CYSetProperty(context, CYGetGlobalObject(context), Result_, result); + + if (fout != NULL) { + fputs(json, fout); + fputs("\n", fout); + fflush(fout); + } + } +} + +static void Console() { + bool bypass(false); + bool debug(false); + + FILE *fout(stdout); + + rl_bind_key('\t', rl_insert); + + struct sigaction action; + sigemptyset(&action.sa_mask); + action.sa_handler = &sigint; + action.sa_flags = 0; + sigaction(SIGINT, &action, NULL); + + restart: for (;;) { + std::string command; + std::vector lines; + + bool extra(false); + const char *prompt("cy# "); + + if (setjmp(ctrlc_) != 0) { + fputs("\n", fout); + fflush(fout); + goto restart; + } + + read: + char *line(readline(prompt)); + if (line == NULL) + break; + + if (!extra) { + extra = true; + if (line[0] == '\\') { + std::string data(line + 1); + if (data == "bypass") { + bypass = !bypass; + fprintf(fout, "bypass == %s\n", bypass ? "true" : "false"); + fflush(fout); + } else if (data == "debug") { + debug = !debug; + fprintf(fout, "debug == %s\n", debug ? "true" : "false"); + fflush(fout); + } + add_history(line); + goto restart; + } + } + + lines.push_back(line); + command += line; + free(line); + + std::string code; + + if (bypass) + code = command; + else { + CYDriver driver(""); + cy::parser parser(driver); + + driver.data_ = command.c_str(); + driver.size_ = command.size(); + + if (parser.parse() != 0 || !driver.errors_.empty()) { + for (CYDriver::Errors::const_iterator i(driver.errors_.begin()); i != driver.errors_.end(); ++i) { + cy::position begin(i->location_.begin); + if (begin.line != lines.size() || begin.column - 1 != lines.back().size()) { + std::cerr << i->message_ << std::endl; + add_history(command.c_str()); + goto restart; + } + } + + driver.errors_.clear(); + + command += '\n'; + prompt = "cy> "; + goto read; + } + + if (driver.source_ == NULL) + goto restart; + + std::ostringstream str; + driver.source_->Show(str); + code = str.str(); + } + + add_history(command.c_str()); + + if (debug) + std::cout << code << std::endl; + + Run(code.c_str(), fout); + } + + fputs("\n", fout); + fflush(fout); +} + +static void *Map(const char *path, size_t *psize) { + int fd; + _syscall(fd = open(path, O_RDONLY)); + + struct stat stat; + _syscall(fstat(fd, &stat)); + size_t size(stat.st_size); + + *psize = size; + + void *base; + _syscall(base = mmap(NULL, size, PROT_READ, MAP_SHARED, fd, 0)); + + _syscall(close(fd)); + return base; +} + +int main(int argc, const char *argv[]) { + const char *script; + + if (argc == 1) + script = NULL; + else { + CYSetArgs(argc - 1, argv + 1); + script = argv[1]; + } + + Result_ = CYCopyJSString("_"); + + if (script == NULL || strcmp(script, "-") == 0) + Console(); + else { + CYDriver driver(script); + cy::parser parser(driver); + + size_t size; + char *start(reinterpret_cast(Map(script, &size))); + char *end(start + size); + + if (size >= 2 && start[0] == '#' && start[1] == '!') { + start += 2; + while (start != end && *start++ != '\n'); + } + + driver.data_ = start; + driver.size_ = end - start; -int CYConsole(FILE *in, FILE *out, FILE *err); + if (parser.parse() != 0 || !driver.errors_.empty()) { + for (CYDriver::Errors::const_iterator i(driver.errors_.begin()); i != driver.errors_.end(); ++i) + std::cerr << i->location_.begin << ": " << i->message_ << std::endl; + } else if (driver.source_ != NULL) { + std::ostringstream str; + driver.source_->Show(str); + std::string code(str.str()); + std::cout << code << std::endl; + Run(code.c_str(), stdout); + } + } -int main() { - return CYConsole(stdin, stdout, stderr); + return 0; }