Example #1
0
 def __init__(self):
     super(DevtoolsClient, self).__init__()
     self.offloader = MessageOffloader()
Example #2
0
	def __init__(self):
		super(DevtoolsClient, self).__init__()
		self.offloader = MessageOffloader()
Example #3
0
class DevtoolsClient(DevtoolsApp):
    def __init__(self):
        super(DevtoolsClient, self).__init__()
        self.offloader = MessageOffloader()

    #override
    def handleUnknown(self, mess):
        self.writeMessage("Received: " + mess)

    #override
    def handleEvent(self, mess):
        if mess["method"] == "Forensics.recordingAdd":
            self.offloader.messages.put(mess)
        else:
            self.writeMessage(json.dumps(mess, separators=(',', ':')))

    #override
    def handleResponse(self, idval):
        cmd = self.cmdHistory[idval]
        if (cmd.cmd == Command.DumpRecording
                or cmd.cmd == Command.DumpReplayInfo) and "file" in cmd.params:
            with open(cmd.params["file"], 'w') as outfile:
                outfile.write(
                    json.dumps(cmd.resp["result"],
                               sort_keys=True,
                               indent=4,
                               separators=(',', ':')))
                outfile.write("\n")
            self.writeMessage("Saved to " + cmd.params["file"])
        else:
            self.writeMessage("Received: " +
                              json.dumps(cmd.resp, separators=(',', ':')))

    #override
    def onError(self, ws, error):
        self.writeMessage(error)

    #override
    def onClose(self, ws):
        self.offloader.stop()
        self.offloader.join()
        print "Exiting"

    #override
    def onOpen(self, ws):
        def run(*args):
            self.offloader.start()
            # Input processing loop
            while True:
                self.writePrompt()
                user_input = sys.stdin.readline().strip()
                try:
                    if not self.handleClientCommand(user_input):
                        cmd = self.sendCommand(self.parseCommand(user_input))
                        print "Sent: ", cmd
                except ClientTermination:
                    break
                except Exception as e:
                    print "Error: ", e
            self.stop()

        thread.start_new_thread(run, ())

    def handleClientCommand(self, cmd_str):
        retval = False
        if cmd_str == "":
            retval = True
        elif cmd_str == "exit":
            raise ClientTermination
        elif cmd_str == "help":
            retval = True
            print "Available commands: ", Command.cmds
        return retval

    #override
    def pickUrl(self, data):
        selection = -1
        print "Select the page. (Enter the page's number)"
        while selection < 0 or selection > len(data) - 1:
            self.writePageChoices(data)
            self.writePrompt()
            user_input = sys.stdin.readline().strip()
            selection = int(user_input)
        return data[selection]["webSocketDebuggerUrl"]

    def writePageChoices(self, data):
        count = 0
        for d in data:
            print "(%i) Title: %s Url: %s ID: %s" % (count, d["title"].encode(
                'utf-8'), d["url"].encode('utf-8'), d["id"].encode('utf-8'))
            count += 1

    def writeMessage(self, message):
        sys.stdout.write("\n" + message + "\n")
        sys.stdout.flush()
        self.writePrompt()

    def writePrompt(self, inval=None):
        prompt = ':> '
        if inval is not None:
            prompt += inval
        sys.stdout.write(prompt)
        sys.stdout.flush()

    def parseCommand(self, cmd_str):
        cmd_str_comps = cmd_str.split()
        cmd = cmd_str_comps[0]
        params = self.parseParams(cmd_str)
        return Command(cmd, params)

    def parseParams(self, cmd_str):
        retval = {}
        cmd_str_comps = shlex.split(cmd_str)
        if len(cmd_str_comps) > 1:
            cmd = cmd_str_comps[0]
            if cmd == Command.StartReplay:
                retval["speed"] = float(cmd_str_comps[1])
            elif cmd == Command.DumpRecording or cmd == Command.DumpReplayInfo or cmd == Command.LoadRecording:
                retval["file"] = cmd_str_comps[1]

        return retval
Example #4
0
class DevtoolsClient(DevtoolsApp):
	def __init__(self):
		super(DevtoolsClient, self).__init__()
		self.offloader = MessageOffloader()
	
	#override
	def handleUnknown(self, mess):
		self.writeMessage("Received: " + mess)
	
	#override
	def handleEvent(self, mess):
		if mess["method"] == "Forensics.recordingAdd":
			self.offloader.messages.put(mess)
		else:
			self.writeMessage(json.dumps(mess, separators=(',', ':')))
	
	#override
	def handleResponse(self, idval):
		cmd = self.cmdHistory[idval]
		if (cmd.cmd == Command.DumpRecording or cmd.cmd == Command.DumpReplayInfo) and "file" in cmd.params:
			with open(cmd.params["file"], 'w') as outfile:
				outfile.write(json.dumps(cmd.resp["result"], sort_keys=True,
										indent=4, separators=(',', ':')))
				outfile.write("\n")
			self.writeMessage("Saved to " + cmd.params["file"]);
		else:
			self.writeMessage("Received: " + json.dumps(cmd.resp, separators=(',', ':')))
	
	#override
	def onError(self, ws, error):
		self.writeMessage(error)
	
	#override
	def onClose(self, ws):
		self.offloader.stop()
		self.offloader.join()
		print "Exiting"
	
	#override    
	def onOpen(self, ws):
		def run(*args):
			self.offloader.start()
			# Input processing loop
			while True:
				self.writePrompt()
				user_input = sys.stdin.readline().strip()
				try:
					if not self.handleClientCommand(user_input):
						cmd = self.sendCommand(self.parseCommand(user_input))
						print "Sent: ", cmd
				except ClientTermination:
					break
				except Exception as e:
					print "Error: ", e
			self.stop()

		thread.start_new_thread(run, ())
		
	def handleClientCommand(self, cmd_str):
		retval = False
		if cmd_str == "":
			retval = True
		elif cmd_str == "exit":
			raise ClientTermination
		elif cmd_str == "help":
			retval = True
			print "Available commands: ", Command.cmds
		return retval
	
	#override    
	def pickUrl(self, data):
		selection = -1
		print "Select the page. (Enter the page's number)"
		while selection < 0 or selection > len(data) - 1:
			self.writePageChoices(data)
			self.writePrompt()
			user_input = sys.stdin.readline().strip()
			selection = int(user_input)
		return data[selection]["webSocketDebuggerUrl"]

	def writePageChoices(self, data):
		count = 0
		for d in data:
			print "(%i) Title: %s Url: %s ID: %s" % (count, d["title"], d["url"], d["id"])
			count += 1
	
	def writeMessage(self, message):	
		sys.stdout.write("\n" + message + "\n")
		sys.stdout.flush()
		self.writePrompt()

	def writePrompt(self, inval=None):
		prompt = ':> '
		if inval is not None:
			prompt += inval
		sys.stdout.write(prompt);
		sys.stdout.flush()
		
	def parseCommand(self, cmd_str):
		cmd_str_comps = cmd_str.split()
		cmd = cmd_str_comps[0]
		params = self.parseParams(cmd_str)
		return Command(cmd, params)

	def parseParams(self, cmd_str):
		retval = {};
		cmd_str_comps = shlex.split(cmd_str)
		if len(cmd_str_comps) > 1 :
			cmd = cmd_str_comps[0]
			if cmd == Command.StartReplay:
				retval["speed"] = float(cmd_str_comps[1])
			elif cmd == Command.DumpRecording or cmd == Command.DumpReplayInfo or cmd == Command.LoadRecording:
				retval["file"] = cmd_str_comps[1]
				
		return retval;