Select.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 <cstring>
  23
  24 #include <stdint.h>
  25
  26 #include <dlfcn.h>
  27 #include <syslog.h>
  28 #include <unistd.h>
  29
  30 #include <mach-o/dyld.h>
  31
  32 extern "C" void CYHandleServer(pid_t pid) {
  33     Dl_info addr;
  34     if (dladdr(reinterpret_cast<void *>(&CYHandleServer), &addr) == 0)
  35         return;
  36
  37     const char *fname(addr.dli_fname);
  38     size_t length(strlen(fname));
  39
  40     const char *ext(strrchr(fname, '.'));
  41     size_t prefix(ext - fname);
  42
  43     const char *addition;
  44 #ifdef __APPLE__
  45     // XXX: THIS IS HORRIBLE OMG I NEED TO FIX THIS ASAP
  46     bool simulator(false);
  47     for (uint32_t i(0), e(_dyld_image_count()); i != e; ++i) {
  48         if (strstr(_dyld_get_image_name(i), "/SDKs/iPhoneSimulator") != NULL)
  49             simulator = true;
  50     }
  51     if (simulator)
  52         addition = "-sim";
  53     else
  54 #endif
  55     // someone threw a fit about dangling #endif + else
  56     // the idea that this bothers someone gives me glee
  57     addition = "-sys";
  58
  59     char library[length + 5];
  60     memcpy(library, fname, prefix);
  61     memcpy(library + prefix, addition, 4);
  62     memcpy(library + prefix + 4, fname + prefix, length - prefix);
  63     library[length + 4] = '\0';
  64
  65     void *handle(dlopen(library, RTLD_LOCAL | RTLD_LAZY));
  66     if (handle == NULL) {
  67         syslog(LOG_ERR, "dlopen() -> %s", dlerror());
  68         return;
  69     }
  70
  71     void *symbol(dlsym(handle, "CYHandleServer"));
  72     if (symbol == NULL)
  73         return;
  74
  75     reinterpret_cast<void (*)(pid_t)>(symbol)(pid);
  76 }