def generate_all_apks_method_graph(apk_directory_path, apk_type):
try:
file = Path(apk_directory_path)
my_abs_path = file.resolve()
except FileNotFoundError:
# 不存在
print("file or dir not exist")
return
else:
typeApkFiles = os.listdir(apk_directory_path)
for typeApkFile in typeApkFiles:
print(typeApkFile)
smali_loc = apk_directory_path + '\\' + typeApkFile
apk_name = typeApkFile[:-4]
print(apk_name)
apk = Apk(apk_name)
# 生成FCG图
fcg_start_time = datetime.datetime.now()
graph = generate_apk_method_graph(smali_loc, apk)
print("generate apk fcg successfully")
fcg_end_time = datetime.datetime.now()
apk.fcg_graph_time = (fcg_end_time - fcg_start_time).seconds
# test_api(apk)
# 先创建一个存储apk信息的目录
os.makedirs('..\data\{}\{}'.format(apk_type, apk_name))
pattern_start_time = datetime.datetime.now()
# 寻找三点子图的模式
find_three_point_pattern(apk, graph, apk_type)
# 寻找四点子图的模式
find_four_point_pattern(apk, graph, apk_type)
# 寻找五点子图的模式
find_five_point_pattern(apk, graph, apk_type)
pattern_end_time = datetime.datetime.now()
apk.pattern_find_time = (pattern_end_time -
pattern_start_time).seconds
print("find apk pattern successfully")
# 统计信息，输出到文件
message_path = '..\data\{}\{}\message.txt'.format(
apk_type, apk_name)
write_apk_message(apk, message_path)
# 展示一下生成的子图
print_opcode_graph(apk)
def main():
import config_reader as config
from apk import Apk
from adb_settings import AdbSettings
import emulator_manager
from adb_logcat import Logcat, TestType
# print("resetting emulator")
import os
import api_commands
import os
dir = os.path.dirname(__file__)
activity = os.path.join(dir, 'test/activity')
activity_list = os.path.join(dir, 'test/activity_list')
activities = []
apk = Apk(config.APK_FULL_PATH)
emulator = emulator_manager.get_adb_instance_from_emulators(
config.EMULATOR_NAME)
# file = open("test/activity", "w")
file = open(activity, 'w')
file.write(api_commands.adb_get_activity_list(emulator, apk))
file.close()
# file = open('test/activity', 'r')
file = open(activity, 'r')
# file2 = open('test/activity_list', 'w')
file2 = open(activity_list, 'w')
for l in file.readlines():
if 'A: android:name' in l and 'Activity' in l:
arr = l.split('"')
activities.append(arr[1])
file2.write(arr[1] + "\n")
print(arr[1])
file.close()
file2.close()
os.remove('test/activity')
# log = Logcat(emulator, apk, TestType.MobileMonkey)
# log.start_logcat()
AdbMonkey(emulator, apk, config.SEED, config.DURATION * 10).start_monkey()
def find_sen_api(apk_directory_path, api_dict):
try:
file = Path(apk_directory_path)
my_abs_path = file.resolve()
except FileNotFoundError:
# 不存在
print("file or dir not exist")
return
else:
typeApkFiles = os.listdir(apk_directory_path)
for typeApkFile in typeApkFiles:
print(typeApkFile)
smali_loc = apk_directory_path + '\\' + typeApkFile
apk_name = typeApkFile[:-4]
print(apk_name)
apk = Apk(apk_name)
# 生成FCG图
start_time = datetime.datetime.now()
api_set = find_apk_smali(smali_loc, apk, api_dict)
for sen_api in api_set:
api_dict[sen_api] = api_dict[sen_api] + 1
end_time = datetime.datetime.now()
print('{}:time:{}'.format(typeApkFile,
(start_time - end_time).seconds))
# log_thread.join()
# thread_monkey.start()
# thread_fuzz_rotation.start()
# thread_fuzz_delay.start()
# thread_fuzz_profile.start()
# thread_fuzz_speed.start()
# thread_fuzz_airplane.start()
# thread_test.start()
# thread_test.join()
# thread_fuzz_delay.join()
# thread_fuzz_profile.join()
# thread_fuzz_speed.join()
# thread_fuzz_rotation.join()
# thread_fuzz_airplane.join()
# thread_monkey.join()
# telnet_connector.kill_avd()
api_commands.adb_stop_activity_of_apk(emulator, apk)
log.stop_logcat()
api_commands.adb_uninstall_apk(emulator, apk)
util.debug_print(
'Finished testing and uninstalling app at {}'.format(time.ctime()), flag=TIME_PRINT_FLAG)
print(Analyzer(log.file_address))
if wipe_after_finish:
print("successfully completed testing app. Closing emulator")
telnet_connector.kill_avd()
emulator_manager.emulator_wipe_data(emulator)
if __name__ == '__main__':
run(Apk(config.APK_FULL_PATH), config.EMULATOR_NAME, config.EMULATOR_PORT)
def start_test():
apk = Apk(config.APK_FULL_PATH)
emulator = emulator_manager.get_adb_instance_from_emulators(
config.EMULATOR_NAME)
adb_settings = AdbSettings("emulator-" + str(emulator.port))
activities = []
log = Logcat(emulator, apk, TestType.MobileMonkey)
if config.GUIDED_APPROACH == 3:
monkey.run_contexts_only(emulator, config.EMULATOR_NAME,
config.EMULATOR_PORT, config.SEED, log)
return
log.start_logcat()
activity = os.path.join(dir, 'test/activity')
activity_list = os.path.join(dir, 'test/activity_list')
if config.GUIDED_APPROACH == 2:
# file = open("test/activity", "w")
file = open(activity, "w")
file.write(api_commands.adb_get_activity_list(emulator, apk))
file.close()
# file = open('test/activity', 'r')
file = open(activity, 'r')
# file2 = open('test/activity_list', 'w')
file2 = open(activity_list, 'w')
for l in file.readlines():
if 'A: android:name' in l and 'Activity' in l:
arr = l.split('"')
activities.append(arr[1])
file2.write(arr[1] + "\n")
print(arr[1])
file.close()
file2.close()
# os.remove('test/activity')
os.remove(activity)
print(len(activities))
seed = config.SEED
for activity in activities:
try:
api_commands.adb_start_activity(emulator, apk, activity)
except Exception:
print(Exception)
threads = []
threads.append(
Thread(target=monkey.run,
args=(emulator, apk, config.EMULATOR_NAME,
config.EMULATOR_PORT, seed, log)))
# monkey.run(emulator, apk, config.EMULATOR_NAME,
# config.EMULATOR_PORT, seed, log)
# threads.append(Thread(target=test_ui, args=(
# activity, emulator, adb_settings, display_height)))
from adb_monkey import AdbMonkey
ad_monkey = AdbMonkey(emulator, apk, seed, config.DURATION)
# ad_monkey.start_monkey()
threads.append(Thread(target=ad_monkey.start_monkey))
[thread.start() for thread in threads]
[thread.join() for thread in threads]
seed = seed + 1
log.stop_logcat()
eventlog.close()
return
if config.GUIDED_APPROACH == 1:
# file = open('test/activity_list', 'r')
file = open(activity_list, 'r')
for l in file.readlines():
activities.append(l.strip())
print(l.strip())
file.close()
else:
# file = open("test/activity", "w")
file = open(activity, "w")
file.write(api_commands.adb_get_activity_list(emulator, apk))
file.close()
# file = open('test/activity', 'r')
# file2 = open('test/activity_list', 'w')
file = open(activity, 'r')
file2 = open(activity_list, 'w')
for l in file.readlines():
if 'A: android:name' in l and 'Activity' in l:
arr = l.split('"')
activities.append(arr[1])
file2.write(arr[1] + "\n")
print(arr[1])
file.close()
file2.close()
# os.remove('test/activity')
os.remove(activity)
print(len(activities))
display_properties = api_commands.adb_display_properties().decode()
if 'DisplayDeviceInfo' in display_properties:
arr = display_properties.split('height=')[1]
display_height = arr.split(',')[0]
seed = config.SEED
for activity in activities:
try:
api_commands.adb_start_activity(emulator, apk, activity)
except Exception:
print(Exception)
threads = []
threads.append(
Thread(target=monkey.run,
args=(emulator, apk, config.EMULATOR_NAME,
config.EMULATOR_PORT, seed, log)))
# monkey.run(emulator, apk, config.EMULATOR_NAME,
# config.EMULATOR_PORT, seed, log)
threads.append(
Thread(target=test_ui,
args=(activity, emulator, adb_settings, display_height)))
[thread.start() for thread in threads]
[thread.join() for thread in threads]
seed = seed + 1
log.stop_logcat()
eventlog.close()
def __init__(self, apk_id, apk, package_name, apk_conf_files,
apk_conf_gzip, apk_launch_activity, av_start_scan_activity):
Apk.__init__(self, apk_id, apk, package_name, apk_conf_files,
apk_conf_gzip, apk_launch_activity)
self.av_start_scan_activity = av_start_scan_activity
import os
import api_commands
import emulator_manager
import config_reader as config
from apk import Apk
import api_commands
from adb_logcat import Logcat, TestType
activities = []
apk = Apk(config.APK_FULL_PATH)
emulator = emulator_manager.get_adb_instance_from_emulators(
config.EMULATOR_NAME)
# file = open("test/activity", "w")
# file.write(api_commands.adb_get_activity_list(emulator, apk))
# file.close()
# file = open('test/activity', 'r')
# file2 = open('test/activity_list', 'w')
# for l in file.readlines():
# if 'A: android:name' in l and 'Activity' in l:
# arr = l.split('"')
# activities.append(arr[1])
# file2.write(arr[1] + "\n")
# print(arr[1])
# file.close()
# file2.close()
# os.remove('test/activity')
from telnet_connector import TelnetAdb
import adb_monkey
from adb_logcat import Logcat
def get_apk(apk_id):
return Apk(apk_id, apksConf[apk_id]['apk_path'],
apksConf[apk_id]['package_name'], apksConf[apk_id]['conf_file'],
apksConf[apk_id]['conf_gzip'],
apksConf[apk_id]['launch_activity'])
def __init__(self, apk_id, apk, package_name, apk_conf_files, apk_conf_gzip, apk_launch_activity, av_start_scan_activity):
Apk.__init__(self, apk_id, apk, package_name, apk_conf_files, apk_conf_gzip, apk_launch_activity)
self.av_start_scan_activity = av_start_scan_activity
# coding=utf-8
import telebot
from telebot import types
import os
from flask import Flask, request
from apk import Apk
import messages
bot = telebot.TeleBot(os.environ['telegram_token'])
server = Flask(__name__)
apk = Apk()
@bot.message_handler(commands=['start'])
def start(message):
# bot.send_message(message.chat.id, messages.hello, reply_markup=types.ReplyKeyboardHide())
bot.send_message(message.chat.id, messages.hello)
@bot.message_handler(commands=['help'])
def help(message):
bot.send_message(message.chat.id, messages.help)
# bot.send_message(message.chat.id, messages.help, reply_markup=types.ReplyKeyboardHide())
@bot.message_handler(commands=['about'])
def about(message):
bot.send_message(message.chat.id, messages.about)
@bot.message_handler(content_types=['text'])