Exemplo n.º 1
0
    def open(self):
        if self.request.headers['Origin'] != 'http://%s' % (
                self.request.headers['Host']):
            self.close()
            return

        self.uuid = str(uuid4())
        syncwebsockets.add(self.uuid, self)
        if not LibPythonWatcher:
            syncwebsockets.send(self.uuid, 'StartLoop')
Exemplo n.º 2
0
    def open(self):
        if self.request.headers['Origin'] != 'http://%s' % (
                self.request.headers['Host']):
            self.close()
            return

        self.uuid = str(uuid4())
        syncwebsockets.add(self.uuid, self)
        if not LibPythonWatcher:
            syncwebsockets.send(self.uuid, 'StartLoop')
Exemplo n.º 3
0
def refresh_process(uuid=None):
    if uuid is not None:
        send = lambda cmd, data: syncwebsockets.send(uuid, cmd, data)
    else:
        send = syncwebsockets.broadcast

    remaining_pids = []
    remaining_tids = []
    for proc in psutil.process_iter():
        cl = proc.cmdline()
        if len(cl) == 0:
            continue
        binary = cl[0].split('/')[-1]
        if (('python' in binary or 'pypy' in binary) and proc.is_running()
                and proc.status() != psutil.STATUS_ZOMBIE):
            try:
                send(
                    'AddProcess', {
                        'pid': proc.pid,
                        'user': proc.username(),
                        'cmd': ' '.join(proc.cmdline()),
                        'threads': proc.num_threads(),
                        'time': proc.create_time(),
                        'mem': proc.memory_percent(),
                        'cpu': proc.cpu_percent(interval=.01)
                    })
                remaining_pids.append(proc.pid)
                for thread in proc.threads():
                    send('AddThread', {'id': thread.id, 'of': proc.pid})
                    remaining_tids.append(thread.id)
            except:
                log.warn('', exc_info=True)
                continue
    send('KeepProcess', remaining_pids)
    send('KeepThreads', remaining_tids)
Exemplo n.º 4
0
    def on_message(self, message):
        if '|' in message:
            cmd, data = message.split('|', 1)
        else:
            cmd, data = message, ''

        if cmd == 'ListSockets':
            for uuid in sockets.uuids:
                syncwebsockets.send(self.uuid, 'AddSocket', uuid)
        elif cmd == 'ListWebsockets':
            for uuid in websockets.uuids:
                syncwebsockets.send(self.uuid, 'AddWebSocket', uuid)
        elif cmd == 'ListBreaks':
            for brk in breakpoints.get():
                syncwebsockets.send(self.uuid, 'AddBreak', brk)
        elif cmd == 'RemoveBreak':
            brk = json.loads(data)
            breakpoints.remove(brk)
            # If it was here, it wasn't temporary
            brk['temporary'] = False
            sockets.broadcast('Unbreak', brk)
        elif cmd == 'RemoveUUID':
            sockets.close(data)
            sockets.remove(data)
            websockets.close(data)
            websockets.remove(data)
        elif cmd == 'ListProcesses':
            refresh_process(self.uuid)
        elif cmd == 'Pause':
            if int(data) == os.getpid():
                log.debug('Pausing self')

                def self_shell(variables):
                    # Debugging self
                    import wdb
                    wdb.set_trace()

                Process(target=self_shell, args=(globals(),)).start()

            else:
                log.debug('Pausing %s' % data)
                tornado.process.Subprocess([
                    'gdb', '-p', data,
                    '-batch'] + [
                        "-eval-command=call %s" % hook
                        for hook in [
                            'PyGILState_Ensure()',
                            'PyRun_SimpleString('
                            '"import wdb; wdb.set_trace(skip=1)"'
                            ')',
                            'PyGILState_Release($1)',
                        ]])
        elif cmd == 'RunFile':
            file_name = data

            def run():
                from wdb import Wdb
                Wdb.get().run_file(file_name)

            Process(target=run).start()
Exemplo n.º 5
0
    def on_message(self, message):
        if '|' in message:
            cmd, data = message.split('|', 1)
        else:
            cmd, data = message, ''

        if cmd == 'ListSockets':
            for uuid in sockets.uuids:
                syncwebsockets.send(self.uuid, 'AddSocket', uuid)
        elif cmd == 'ListWebsockets':
            for uuid in websockets.uuids:
                syncwebsockets.send(self.uuid, 'AddWebSocket', uuid)
        elif cmd == 'ListBreaks':
            for brk in breakpoints.get():
                syncwebsockets.send(self.uuid, 'AddBreak', brk)
        elif cmd == 'RemoveBreak':
            brk = json.loads(data)
            breakpoints.remove(brk)
            # If it was here, it wasn't temporary
            brk['temporary'] = False
            sockets.broadcast('Unbreak', brk)
        elif cmd == 'RemoveUUID':
            sockets.close(data)
            sockets.remove(data)
            websockets.close(data)
            websockets.remove(data)
        elif cmd == 'ListProcesses':
            refresh_process(self.uuid)
        elif cmd == 'Pause':
            if int(data) == os.getpid():
                log.debug('Pausing self')

                def self_shell(variables):
                    # Debugging self
                    import wdb
                    wdb.set_trace()

                Process(target=self_shell, args=(globals(),)).start()

            else:
                log.debug('Pausing %s' % data)
                tornado.process.Subprocess([
                    'gdb', '-p', data,
                    '-batch'] + [
                        "-eval-command=call %s" % hook
                        for hook in [
                            'PyGILState_Ensure()',
                            'PyRun_SimpleString('
                            '"import wdb; wdb.set_trace(skip=1)"'
                            ')',
                            'PyGILState_Release($1)',
                        ]])
        elif cmd == 'RunFile':
            file_name = data

            def run():
                from wdb import Wdb
                Wdb.get().run_file(file_name)

            Process(target=run).start()
Exemplo n.º 6
0
def refresh_process(uuid=None):
    if uuid is not None:
        send = lambda cmd, data: syncwebsockets.send(uuid, cmd, data)
    else:
        send = syncwebsockets.broadcast

    remaining_pids = []
    remaining_tids = []
    for proc in psutil.process_iter():
        try:
            cl = proc.cmdline()
        except (psutil.ZombieProcess,
                psutil.AccessDenied,
                psutil.NoSuchProcess):
            continue
        else:
            if len(cl) == 0:
                continue

        binary = cl[0].split('/')[-1]
        if (
                ('python' in binary or 'pypy' in binary) and
                proc.is_running() and
                proc.status() != psutil.STATUS_ZOMBIE):
            try:
                try:
                    cpu = proc.cpu_percent(interval=.01)
                    send('AddProcess', {
                        'pid': proc.pid,
                        'user': proc.username(),
                        'cmd': ' '.join(proc.cmdline()),
                        'threads': proc.num_threads(),
                        'time': proc.create_time(),
                        'mem': proc.memory_percent(),
                        'cpu': cpu
                    })
                    remaining_pids.append(proc.pid)
                    for thread in proc.threads():
                        send('AddThread', {
                            'id': thread.id,
                            'of': proc.pid
                        })
                        remaining_tids.append(thread.id)
                except psutil.NoSuchProcess:
                    pass
            except Exception:
                log.warn('', exc_info=True)
                continue

    send('KeepProcess', remaining_pids)
    send('KeepThreads', remaining_tids)
Exemplo n.º 7
0
 def on_open(self):
     self.uuid = str(uuid4())
     syncwebsockets.add(self.uuid, self)
     if not LibPythonWatcher:
         syncwebsockets.send(self.uuid, 'StartLoop')
Exemplo n.º 8
0
 def on_open(self):
     self.uuid = str(uuid4())
     syncwebsockets.add(self.uuid, self)
     if not LibPythonWatcher:
         syncwebsockets.send(self.uuid, 'StartLoop')