AppMonitor/monitor.py at master · xdek42/AppMonitor · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
import frida
import sys
import os
import time
import logging
import analysis
import subprocess
import os
from subprocess import call, PIPE, Popen

logging.basicConfig(level=logging.INFO,
                    format = "%(asctime)s %(message)s",
                    datefmt = '%Y-%m-%d %H:%M',
                    filename = "E:\AppMonitor\log.txt",
                    filemode = "w")

def build_monitor_script(dir, topdown = True):
    script = ""
    for root, dirs, files in os.walk(dir, topdown):
        for name in files:
            script += open(os.path.join(root,name)).read()
    return script


def on_message(message, data):
    if message['type'] == 'send':
        logging.info(message['payload'])
    elif message['type'] == 'error':
        logging.info(message['stack'])

def start_app(apkFile, packageName, launcherActivity):
    #install apk
    cmd = "adb install -r " + apkFile
    ret = subprocess.call(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    if (ret == 1):
        print("[ERROR]: failed install apk")
        sys.exit(1)
    print("successfully installed apk")
    '''
    #launch app
    cmd = "adb shell am start -n " + packageName + "/" + launcherActivity
    ret = subprocess.call(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    if (ret == 1):
        print("[ERROR]: failed to launch app")
        sys.exit(1)
    print("successfully launched app")
    '''
    #prepare for frida
    cmd = "adb forward tcp:27042 tcp:27042"
    ret = subprocess.call(cmd, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)


def main():

    if len(sys.argv) != 2:
        print("usage: monitor.py example.apk")
        sys.exit(1)

    #get static info
    apkFile = sys.argv[1]
    app = analysis.Application(apkFile)
    packageName = app.getPackageName()
    launcherActivity = app.getMainActivity()

    #install apk and launch app
    start_app(apkFile, packageName, launcherActivity)
    #wait for app
    time.sleep(2)

    pid = None
    device = None
    session = None
    try:
        device = frida.get_usb_device()
        pid = device.spawn([packageName])
        session = device.attach(pid)
    except Exception as e:
        print("[ERROR]: %s" % str(e))
        sys.exit(1)
    print("successfully attached to app")
    script_dir = os.path.join(".", "scripts")
    script_content = build_monitor_script(script_dir)
    script = session.create_script(script_content)
    script.on("message", on_message)
    script.load()
    device.resume(pid)

    #prevent the python script from terminating
    start = time.clock()
    while True:
        end = time.clock()
        if int(end - start) > 300:
            session.detach()
            break

if __name__ == "__main__":
    main()