Application.mm

   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
  40 #define _GNU_SOURCE
  41
  42 #include <substrate.h>
  43 #include "cycript.hpp"
  44
  45 #include <cstdio>
  46 #include <sstream>
  47
  48 #include <setjmp.h>
  49
  50 #include <readline/readline.h>
  51 #include <readline/history.h>
  52
  53 #include <sys/mman.h>
  54
  55 #include <errno.h>
  56 #include <unistd.h>
  57
  58 #include <sys/types.h>
  59 #include <sys/stat.h>
  60 #include <fcntl.h>
  61
  62 #include "Cycript.tab.hh"
  63
  64 static jmp_buf ctrlc_;
  65
  66 static void sigint(int) {
  67     longjmp(ctrlc_, 1);
  68 }
  69
  70 static JSStringRef Result_;
  71
  72 void Run(const char *code, FILE *fout) { _pooled
  73     JSStringRef script(JSStringCreateWithUTF8CString(code));
  74
  75     JSContextRef context(CYGetJSContext());
  76
  77     JSValueRef exception(NULL);
  78     JSValueRef result(JSEvaluateScript(context, script, NULL, NULL, 0, &exception));
  79     JSStringRelease(script);
  80
  81     if (exception != NULL) { error:
  82         result = exception;
  83         exception = NULL;
  84     }
  85
  86     if (!JSValueIsUndefined(context, result)) {
  87         CYPool pool;
  88         const char *json;
  89
  90         json = CYPoolCYONString(pool, context, result, &exception);
  91         if (exception != NULL)
  92             goto error;
  93
  94         CYSetProperty(context, CYGetGlobalObject(context), Result_, result);
  95
  96         if (fout != NULL) {
  97             fputs(json, fout);
  98             fputs("\n", fout);
  99             fflush(fout);
 100         }
 101     }
 102 }
 103
 104 static void Console() {
 105     bool bypass(false);
 106     bool debug(false);
 107
 108     FILE *fout(stdout);
 109
 110     rl_bind_key('\t', rl_insert);
 111
 112     struct sigaction action;
 113     sigemptyset(&action.sa_mask);
 114     action.sa_handler = &sigint;
 115     action.sa_flags = 0;
 116     sigaction(SIGINT, &action, NULL);
 117
 118     restart: for (;;) {
 119         std::string command;
 120         std::vector<std::string> lines;
 121
 122         bool extra(false);
 123         const char *prompt("cy# ");
 124
 125         if (setjmp(ctrlc_) != 0) {
 126             fputs("\n", fout);
 127             fflush(fout);
 128             goto restart;
 129         }
 130
 131       read:
 132         char *line(readline(prompt));
 133         if (line == NULL)
 134             break;
 135
 136         if (!extra) {
 137             extra = true;
 138             if (line[0] == '\\') {
 139                 std::string data(line + 1);
 140                 if (data == "bypass") {
 141                     bypass = !bypass;
 142                     fprintf(fout, "bypass == %s\n", bypass ? "true" : "false");
 143                     fflush(fout);
 144                 } else if (data == "debug") {
 145                     debug = !debug;
 146                     fprintf(fout, "debug == %s\n", debug ? "true" : "false");
 147                     fflush(fout);
 148                 }
 149                 add_history(line);
 150                 goto restart;
 151             }
 152         }
 153
 154         lines.push_back(line);
 155         command += line;
 156         free(line);
 157
 158         std::string code;
 159
 160         if (bypass)
 161             code = command;
 162         else {
 163             CYDriver driver("");
 164             cy::parser parser(driver);
 165
 166             driver.data_ = command.c_str();
 167             driver.size_ = command.size();
 168
 169             if (parser.parse() != 0 || !driver.errors_.empty()) {
 170                 for (CYDriver::Errors::const_iterator i(driver.errors_.begin()); i != driver.errors_.end(); ++i) {
 171                     cy::position begin(i->location_.begin);
 172                     if (begin.line != lines.size() || begin.column - 1 != lines.back().size()) {
 173                         std::cerr << i->message_ << std::endl;
 174                         add_history(command.c_str());
 175                         goto restart;
 176                     }
 177                 }
 178
 179                 driver.errors_.clear();
 180
 181                 command += '\n';
 182                 prompt = "cy> ";
 183                 goto read;
 184             }
 185
 186             if (driver.source_ == NULL)
 187                 goto restart;
 188
 189             std::ostringstream str;
 190             driver.source_->Show(str);
 191             code = str.str();
 192         }
 193
 194         add_history(command.c_str());
 195
 196         if (debug)
 197             std::cout << code << std::endl;
 198
 199         Run(code.c_str(), fout);
 200     }
 201
 202     fputs("\n", fout);
 203     fflush(fout);
 204 }
 205
 206 static void *Map(const char *path, size_t *psize) {
 207     int fd;
 208     _syscall(fd = open(path, O_RDONLY));
 209
 210     struct stat stat;
 211     _syscall(fstat(fd, &stat));
 212     size_t size(stat.st_size);
 213
 214     *psize = size;
 215
 216     void *base;
 217     _syscall(base = mmap(NULL, size, PROT_READ, MAP_SHARED, fd, 0));
 218
 219     _syscall(close(fd));
 220     return base;
 221 }
 222
 223 int main(int argc, const char *argv[]) {
 224     const char *script;
 225
 226     if (argc == 1)
 227         script = NULL;
 228     else {
 229         CYSetArgs(argc - 1, argv + 1);
 230         script = argv[1];
 231     }
 232
 233     Result_ = CYCopyJSString("_");
 234
 235     if (script == NULL || strcmp(script, "-") == 0)
 236         Console();
 237     else {
 238         CYDriver driver(script);
 239         cy::parser parser(driver);
 240
 241         size_t size;
 242         char *start(reinterpret_cast<char *>(Map(script, &size)));
 243         char *end(start + size);
 244
 245         if (size >= 2 && start[0] == '#' && start[1] == '!') {
 246             start += 2;
 247
 248             if (void *line = memchr(start, '\n', end - start))
 249                 start = reinterpret_cast<char *>(line);
 250             else
 251                 start = end;
 252         }
 253
 254         driver.data_ = start;
 255         driver.size_ = end - start;
 256
 257         if (parser.parse() != 0 || !driver.errors_.empty()) {
 258             for (CYDriver::Errors::const_iterator i(driver.errors_.begin()); i != driver.errors_.end(); ++i)
 259                 std::cerr << i->location_.begin << ": " << i->message_ << std::endl;
 260         } else if (driver.source_ != NULL) {
 261             std::ostringstream str;
 262             driver.source_->Show(str);
 263             std::string code(str.str());
 264             std::cout << code << std::endl;
 265             Run(code.c_str(), stdout);
 266         }
 267     }
 268
 269     return 0;
 270 }