In trampoline, make $strlcpy not crash given NULL.

[cycript.git] / Library.cpp

diff --git a/Library.cpp b/Library.cpp
index f79711b5ba5e1382fcf55801be9212697e089406..816dab6cebb44a0d27d86eccec185797e45e1702 100644 (file)
--- a/Library.cpp
+++ b/Library.cpp
@@ -191,38 +191,22 @@ double CYCastDouble(const char *value) {
     return CYCastDouble(value, strlen(value));
 }
 
-extern "C" bool CydgetMemoryParse(const uint16_t **data, size_t *size) {
+CYUTF8String CYPoolCode(CYPool &pool, std::istream &stream) {
     CYLocalPool local;
-
-    CYUTF8String utf8(CYPoolUTF8String(local, CYUTF16String(*data, *size)));
-    CYStream stream(utf8.data, utf8.data + utf8.size);
     CYDriver driver(stream);
 
     cy::parser parser(driver);
-    if (parser.parse() != 0 || !driver.errors_.empty()) {
-        *data = NULL;
-        *size = 0;
-        return false;
-    }
+    _assert(parser.parse() == 0);
+    _assert(driver.errors_.empty());
 
     CYOptions options;
     CYContext context(options);
     driver.program_->Replace(context);
+
     std::ostringstream str;
     CYOutput out(str, options);
     out << *driver.program_;
-    std::string code(str.str());
-
-    CYPool pool;
-    CYUTF16String utf16(CYPoolUTF16String(pool, CYUTF8String(code.c_str(), code.size())));
-
-    size_t bytes(utf16.size * sizeof(uint16_t));
-    uint16_t *copy(reinterpret_cast<uint16_t *>(malloc(bytes)));
-    memcpy(copy, utf16.data, bytes);
-
-    *data = copy;
-    *size = utf16.size;
-    return true;
+    return $pool.strdup(str.str().c_str());
 }
 
 CYPool &CYGetGlobalPool() {