Connector.cpp

   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 #include <substrate.h>
  41
  42 #include "cycript.hpp"
  43 #include "Pooling.hpp"
  44
  45 #include <sys/types.h>
  46 #include <sys/socket.h>
  47 #include <netinet/in.h>
  48 #include <sys/un.h>
  49
  50 #include <apr_thread_proc.h>
  51
  52 #include <fcntl.h>
  53 #include <unistd.h>
  54
  55 #import <CoreFoundation/CFLogUtilities.h>
  56
  57 void CYThrow(const char *format, ...) {
  58     CYPool pool;
  59
  60     va_list args;
  61     va_start (args, format);
  62     const char *message(apr_pvsprintf(pool, format, args));
  63     va_end (args);
  64
  65     fprintf(stderr, "%s\n", message);
  66     throw std::string(message);
  67 }
  68
  69 struct CYServer :
  70     CYData
  71 {
  72     int socket_;
  73 };
  74
  75 apr_status_t CYPoolDLClose_(void *data) {
  76     // XXX: this is an interesting idea
  77     /* void *handle(reinterpret_cast<void *>(data));
  78     dlclose(handle); */
  79     return APR_SUCCESS;
  80 }
  81
  82 static void * APR_THREAD_FUNC Cyrver(apr_thread_t *thread, void *data) {
  83     CYServer *server(reinterpret_cast<CYServer *>(data));
  84
  85     for (;;) {
  86         int socket(_syscall(accept(server->socket_, NULL, NULL)));
  87
  88         if (void *handle = dlopen("/usr/lib/libcycript.dylib", RTLD_LAZY | RTLD_LOCAL)) {
  89             apr_pool_t *pool;
  90             _aprcall(apr_pool_create(&pool, NULL));
  91
  92             apr_pool_cleanup_register(pool, handle, &CYPoolDLClose_, &apr_pool_cleanup_null);
  93
  94             if (void (*CYHandleClient_)(apr_pool_t *, int) = reinterpret_cast<void (*)(apr_pool_t *, int)>(dlsym(handle, "CYHandleClient")))
  95                 (*CYHandleClient_)(pool, socket);
  96             else
  97                 apr_pool_destroy(pool);
  98         } else
  99             CFLog(kCFLogLevelError, CFSTR("CY:Error: cannot load: %s"), dlerror());
 100     }
 101
 102     delete server;
 103     return NULL;
 104 }
 105
 106 static void Unlink() {
 107     pid_t pid(getpid());
 108     char path[104];
 109     sprintf(path, "/tmp/.s.cy.%u", pid);
 110     unlink(path);
 111 }
 112
 113 MSInitialize {
 114     _aprcall(apr_initialize());
 115
 116     CYServer *server(new CYServer());
 117     server->socket_ = _syscall(socket(PF_UNIX, SOCK_STREAM, 0));
 118
 119     struct sockaddr_un address;
 120     memset(&address, 0, sizeof(address));
 121     address.sun_family = AF_UNIX;
 122
 123     pid_t pid(getpid());
 124     sprintf(address.sun_path, "/tmp/.s.cy.%u", pid);
 125
 126     try {
 127         _syscall(bind(server->socket_, reinterpret_cast<sockaddr *>(&address), SUN_LEN(&address)));
 128         atexit(&Unlink);
 129         _syscall(listen(server->socket_, 0));
 130
 131         apr_threadattr_t *attr;
 132         _aprcall(apr_threadattr_create(&attr, server->pool_));
 133
 134         apr_thread_t *thread;
 135         _aprcall(apr_thread_create(&thread, attr, &Cyrver, server, server->pool_));
 136     } catch (...) {
 137         CFLog(kCFLogLevelError, CFSTR("CY:Error: cannot bind unix domain socket"));
 138     }
 139 }