Library.cpp

   1 /* Cycript - Optimizing JavaScript Compiler/Runtime
   2  * Copyright (C) 2009-2013  Jay Freeman (saurik)
   3 */
   4
   5 /* GNU General Public License, Version 3 {{{ */
   6 /*
   7  * Cycript is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published
   9  * by the Free Software Foundation, either version 3 of the License,
  10  * or (at your option) any later version.
  11  *
  12  * Cycript is distributed in the hope that it will be useful, but
  13  * WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License
  18  * along with Cycript.  If not, see <http://www.gnu.org/licenses/>.
  19 **/
  20 /* }}} */
  21
  22 #include <dlfcn.h>
  23
  24 #include "cycript.hpp"
  25
  26 #include "Pooling.hpp"
  27
  28 #include <sys/mman.h>
  29
  30 #include <iostream>
  31 #include <set>
  32 #include <map>
  33 #include <iomanip>
  34 #include <sstream>
  35 #include <cmath>
  36
  37 #include "Parser.hpp"
  38 #include "Cycript.tab.hh"
  39
  40 #include "Error.hpp"
  41 #include "String.hpp"
  42 #include "Execute.hpp"
  43 #include "JavaScript.hpp"
  44
  45 #include "ConvertUTF.h"
  46
  47 /* C Strings {{{ */
  48 CYUTF8String CYPoolUTF8String(CYPool &pool, CYUTF16String utf16) {
  49     // XXX: this is wrong
  50     size_t size(utf16.size * 5);
  51     char *temp(new(pool) char[size]);
  52
  53     const uint16_t *lhs(utf16.data);
  54     uint8_t *rhs(reinterpret_cast<uint8_t *>(temp));
  55     _assert(ConvertUTF16toUTF8(&lhs, lhs + utf16.size, &rhs, rhs + size, lenientConversion) == conversionOK);
  56
  57     *rhs = 0;
  58     return CYUTF8String(temp, reinterpret_cast<char *>(rhs) - temp);
  59 }
  60
  61 CYUTF16String CYPoolUTF16String(CYPool &pool, CYUTF8String utf8) {
  62     // XXX: this is wrong
  63     size_t size(utf8.size * 5);
  64     uint16_t *temp(new (pool) uint16_t[size]);
  65
  66     const uint8_t *lhs(reinterpret_cast<const uint8_t *>(utf8.data));
  67     uint16_t *rhs(temp);
  68     _assert(ConvertUTF8toUTF16(&lhs, lhs + utf8.size, &rhs, rhs + size, lenientConversion) == conversionOK);
  69
  70     *rhs = 0;
  71     return CYUTF16String(temp, rhs - temp);
  72 }
  73 /* }}} */
  74 /* Index Offsets {{{ */
  75 size_t CYGetIndex(const CYUTF8String &value) {
  76     if (value.data[0] != '0') {
  77         size_t index(0);
  78         for (size_t i(0); i != value.size; ++i) {
  79             if (!DigitRange_[value.data[i]])
  80                 return _not(size_t);
  81             index *= 10;
  82             index += value.data[i] - '0';
  83         }
  84         return index;
  85     } else if (value.size == 1)
  86         return 0;
  87     else
  88         return _not(size_t);
  89 }
  90
  91 // XXX: this isn't actually right
  92 bool CYGetOffset(const char *value, ssize_t &index) {
  93     if (value[0] != '0') {
  94         char *end;
  95         index = strtol(value, &end, 10);
  96         if (value + strlen(value) == end)
  97             return true;
  98     } else if (value[1] == '\0') {
  99         index = 0;
 100         return true;
 101     }
 102
 103     return false;
 104 }
 105 /* }}} */
 106 /* JavaScript *ify {{{ */
 107 void CYStringify(std::ostringstream &str, const char *data, size_t size) {
 108     unsigned quot(0), apos(0);
 109     for (const char *value(data), *end(data + size); value != end; ++value)
 110         if (*value == '"')
 111             ++quot;
 112         else if (*value == '\'')
 113             ++apos;
 114
 115     bool single(quot > apos);
 116
 117     str << (single ? '\'' : '"');
 118
 119     for (const char *value(data), *end(data + size); value != end; ++value)
 120         switch (*value) {
 121             case '\\': str << "\\\\"; break;
 122             case '\b': str << "\\b"; break;
 123             case '\f': str << "\\f"; break;
 124             case '\n': str << "\\n"; break;
 125             case '\r': str << "\\r"; break;
 126             case '\t': str << "\\t"; break;
 127             case '\v': str << "\\v"; break;
 128
 129             case '"':
 130                 if (!single)
 131                     str << "\\\"";
 132                 else goto simple;
 133             break;
 134
 135             case '\'':
 136                 if (single)
 137                     str << "\\'";
 138                 else goto simple;
 139             break;
 140
 141             default:
 142                 // this test is designed to be "awesome", generating neither warnings nor incorrect results
 143                 if (*value < 0x20 || *value >= 0x7f)
 144                     str << "\\x" << std::setbase(16) << std::setw(2) << std::setfill('0') << unsigned(uint8_t(*value));
 145                 else simple:
 146                     str << *value;
 147         }
 148
 149     str << (single ? '\'' : '"');
 150 }
 151
 152 void CYNumerify(std::ostringstream &str, double value) {
 153     char string[32];
 154     // XXX: I want this to print 1e3 rather than 1000
 155     sprintf(string, "%.17g", value);
 156     str << string;
 157 }
 158
 159 bool CYIsKey(CYUTF8String value) {
 160     const char *data(value.data);
 161     size_t size(value.size);
 162
 163     if (size == 0)
 164         return false;
 165
 166     if (DigitRange_[data[0]]) {
 167         size_t index(CYGetIndex(value));
 168         if (index == _not(size_t))
 169             return false;
 170     } else {
 171         if (!WordStartRange_[data[0]])
 172             return false;
 173         for (size_t i(1); i != size; ++i)
 174             if (!WordEndRange_[data[i]])
 175                 return false;
 176     }
 177
 178     return true;
 179 }
 180 /* }}} */
 181
 182 double CYCastDouble(const char *value, size_t size) {
 183     char *end;
 184     double number(strtod(value, &end));
 185     if (end != value + size)
 186         return NAN;
 187     return number;
 188 }
 189
 190 double CYCastDouble(const char *value) {
 191     return CYCastDouble(value, strlen(value));
 192 }
 193
 194 size_t CYArrayLength(JSContextRef context, JSObjectRef array) {
 195     return CYCastDouble(context, CYGetProperty(context, array, length_s));
 196 }
 197
 198 JSValueRef CYArrayGet(JSContextRef context, JSObjectRef array, size_t index) {
 199     JSValueRef exception(NULL);
 200     JSValueRef value(JSObjectGetPropertyAtIndex(context, array, index, &exception));
 201     CYThrow(context, exception);
 202     return value;
 203 }
 204
 205 void CYArrayPush(JSContextRef context, JSObjectRef array, JSValueRef value) {
 206     JSValueRef exception(NULL);
 207     JSValueRef arguments[1];
 208     arguments[0] = value;
 209     JSObjectRef Array(CYGetCachedObject(context, CYJSString("Array_prototype")));
 210     JSObjectCallAsFunction(context, CYCastJSObject(context, CYGetProperty(context, Array, push_s)), array, 1, arguments, &exception);
 211     CYThrow(context, exception);
 212 }
 213
 214 extern "C" void CydgetMemoryParse(const uint16_t **data, size_t *size) {
 215     CYLocalPool local;
 216
 217     CYUTF8String utf8(CYPoolUTF8String(local, CYUTF16String(*data, *size)));
 218     CYStream stream(utf8.data, utf8.data + utf8.size);
 219     CYDriver driver(stream);
 220
 221     cy::parser parser(driver);
 222     if (parser.parse() != 0 || !driver.errors_.empty())
 223         return;
 224
 225     CYOptions options;
 226     CYContext context(options);
 227     driver.program_->Replace(context);
 228     std::ostringstream str;
 229     CYOutput out(str, options);
 230     out << *driver.program_;
 231     std::string code(str.str());
 232
 233     CYPool pool;
 234     CYUTF16String utf16(CYPoolUTF16String(pool, CYUTF8String(code.c_str(), code.size())));
 235
 236     size_t bytes(utf16.size * sizeof(uint16_t));
 237     uint16_t *copy(reinterpret_cast<uint16_t *>(malloc(bytes)));
 238     memcpy(copy, utf16.data, bytes);
 239
 240     *data = copy;
 241     *size = utf16.size;
 242 }
 243
 244 CYPool &CYGetGlobalPool() {
 245     static CYPool pool;
 246     return pool;
 247 }
 248
 249 void CYThrow(const char *format, ...) {
 250     va_list args;
 251     va_start(args, format);
 252     throw CYPoolError(format, args);
 253     // XXX: does this matter? :(
 254     va_end(args);
 255 }
 256
 257 const char *CYPoolError::PoolCString(CYPool &pool) const {
 258     return pool.strdup(message_);
 259 }
 260
 261 CYPoolError::CYPoolError(const CYPoolError &rhs) :
 262     message_(pool_.strdup(rhs.message_))
 263 {
 264 }
 265
 266 CYPoolError::CYPoolError(const char *format, ...) {
 267     va_list args;
 268     va_start(args, format);
 269     // XXX: there might be a beter way to think about this
 270     message_ = pool_.vsprintf(64, format, args);
 271     va_end(args);
 272 }
 273
 274 CYPoolError::CYPoolError(const char *format, va_list args) {
 275     // XXX: there might be a beter way to think about this
 276     message_ = pool_.vsprintf(64, format, args);
 277 }