71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
import subprocess
|
|
import logging
|
|
import subprocess
|
|
import logging
|
|
|
|
|
|
class UserInputMonitor(object):
|
|
"""
|
|
A connection with the target device through `getevent`.
|
|
`getevent` is able to get raw user input from device.
|
|
"""
|
|
|
|
def __init__(self, device=None):
|
|
"""
|
|
initialize connection
|
|
:param device: a Device instance
|
|
"""
|
|
self.logger = logging.getLogger(self.__class__.__name__)
|
|
|
|
if device is None:
|
|
from DroidBot.platforms.android import AndroidDevice as Device
|
|
device = Device()
|
|
self.device = device
|
|
self.connected = False
|
|
self.process = None
|
|
if device.output_dir is None:
|
|
self.out_file = None
|
|
else:
|
|
self.out_file = "%s/user_input.txt" % device.output_dir
|
|
|
|
def connect(self):
|
|
self.process = subprocess.Popen(["adb", "-s", self.device.serial, "shell", "getevent", "-lt"],
|
|
stdin=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
stdout=subprocess.PIPE)
|
|
import threading
|
|
listen_thread = threading.Thread(target=self.handle_output)
|
|
listen_thread.start()
|
|
|
|
def disconnect(self):
|
|
self.connected = False
|
|
if self.process is not None:
|
|
self.process.terminate()
|
|
|
|
def check_connectivity(self):
|
|
return self.connected
|
|
|
|
def handle_output(self):
|
|
self.connected = True
|
|
|
|
f = None
|
|
if self.out_file is not None:
|
|
f = open(self.out_file, 'w')
|
|
|
|
while self.connected:
|
|
if self.process is None:
|
|
continue
|
|
line = self.process.stdout.readline()
|
|
if not isinstance(line, str):
|
|
line = line.decode()
|
|
self.parse_line(line)
|
|
if f is not None:
|
|
f.write(line)
|
|
|
|
if f is not None:
|
|
f.close()
|
|
print("[CONNECTION] %s is disconnected" % self.__class__.__name__)
|
|
|
|
def parse_line(self, _getevent_line):
|
|
pass
|