Updated with module support
This commit is contained in:
Kevin Rode
parent
741ca20c35
commit
2d8f137cb6
108
amt.py
108
amt.py
|
|
@ -4,100 +4,8 @@ Code to interact with an android device using ADB
|
||||||
Written by Kevin Rode
|
Written by Kevin Rode
|
||||||
Last Updated Sep 28 2019
|
Last Updated Sep 28 2019
|
||||||
"""
|
"""
|
||||||
import adbutils
|
|
||||||
import os
|
|
||||||
import sys
|
import sys
|
||||||
import re
|
from helpers import root
|
||||||
import time
|
|
||||||
import requests
|
|
||||||
from lxml import html
|
|
||||||
|
|
||||||
|
|
||||||
def adb_start():
|
|
||||||
if "platform-tools" in os.environ['PATH']:
|
|
||||||
print("ADB found in PATH")
|
|
||||||
else:
|
|
||||||
os.environ['PATH'] += ';'+os.getcwd()+'\\platform-tools'
|
|
||||||
|
|
||||||
|
|
||||||
def adb_connect():
|
|
||||||
adb = adbutils.AdbClient(host="127.0.0.1", port=5037)
|
|
||||||
d = adb.device()
|
|
||||||
if d != None:
|
|
||||||
print("Device connected.")
|
|
||||||
return d
|
|
||||||
|
|
||||||
|
|
||||||
def twrp_download(d):
|
|
||||||
found = 0
|
|
||||||
files = os.listdir(os.curdir)
|
|
||||||
for file in files:
|
|
||||||
if "twrp" in file:
|
|
||||||
found = 1
|
|
||||||
if found != 1:
|
|
||||||
cpu = d.shell('cat /proc/cpuinfo | grep Hardware')
|
|
||||||
cpu = cpu.replace(" ", "")
|
|
||||||
cpu = re.sub(r'(.+:)', '', cpu)
|
|
||||||
r = requests.get('https://dl.twrp.me/'+cpu)
|
|
||||||
tree = html.fromstring(r.text)
|
|
||||||
urls = tree.xpath('//a/@href')
|
|
||||||
downloads = []
|
|
||||||
for i in urls:
|
|
||||||
if "img" in i:
|
|
||||||
downloads.append(i)
|
|
||||||
url_to_download = "https://dl.twrp.me"+downloads[0]
|
|
||||||
url_to_download = url_to_download.replace('.html', '')
|
|
||||||
print("Use this link to download twrp for your connected device: "+url_to_download)
|
|
||||||
print("Ensure that the downloaded file is moved to the same folder as the script before continuing")
|
|
||||||
input("Press Enter to continue...")
|
|
||||||
files = os.listdir(os.curdir)
|
|
||||||
for file in files:
|
|
||||||
if "twrp" in file:
|
|
||||||
found = 1
|
|
||||||
while found != 1:
|
|
||||||
print("File not found. Please confirm it has been moved to the correct directory")
|
|
||||||
input("Press Enter to continue...")
|
|
||||||
files = os.listdir(os.curdir)
|
|
||||||
for file in files:
|
|
||||||
if "twrp" in file:
|
|
||||||
found = 1
|
|
||||||
else:
|
|
||||||
print("twrp already downloaded")
|
|
||||||
|
|
||||||
|
|
||||||
def push_files(d):
|
|
||||||
initcheck_magisk = d.shell("cd /sdcard && ls | grep Magisk")
|
|
||||||
if initcheck_magisk == None and initcheck_twrp == None:
|
|
||||||
d.sync.push("Magisk-v19.3.zip", "/sdcard/Magisk.zip")
|
|
||||||
check_magisk = d.shell("cd /sdcard && ls | grep Magisk")
|
|
||||||
if check != None:
|
|
||||||
print("File copied successfully.")
|
|
||||||
else:
|
|
||||||
print("Something went wrong. Please try again.")
|
|
||||||
else:
|
|
||||||
print("Magisk already copied")
|
|
||||||
|
|
||||||
|
|
||||||
def reboot_bootloader():
|
|
||||||
adb = "platform-tools\\adb.exe"
|
|
||||||
fastboot = "platform-tools\\fastboot.exe"
|
|
||||||
os.system(adb+" reboot bootloader")
|
|
||||||
input("Press Enter when the device has rebooted")
|
|
||||||
os.system(fastboot+" devices")
|
|
||||||
|
|
||||||
|
|
||||||
def root_device():
|
|
||||||
adb = "platform-tools\\adb.exe"
|
|
||||||
fastboot = "platform-tools\\fastboot.exe"
|
|
||||||
files = os.listdir(os.curdir)
|
|
||||||
for file in files:
|
|
||||||
if "twrp" in file:
|
|
||||||
twrp = file
|
|
||||||
os.system(fastboot + " boot "+twrp)
|
|
||||||
input("Press Enter when TWRP has booted")
|
|
||||||
print("Follow the onscreen directions to install Magisk (Located at the bottom of the install window)")
|
|
||||||
print("After Magisk installs click [Reboot] then [Do Not Install]")
|
|
||||||
input("Press Enter when the device has rebooted")
|
|
||||||
|
|
||||||
|
|
||||||
def menu():
|
def menu():
|
||||||
|
|
@ -105,11 +13,7 @@ def menu():
|
||||||
print("[1] Root Device (WIP)\n[2] Extract Data (Coming Soon)\n[99] Quit")
|
print("[1] Root Device (WIP)\n[2] Extract Data (Coming Soon)\n[99] Quit")
|
||||||
choice = input("Please select a number: ")
|
choice = input("Please select a number: ")
|
||||||
if int(choice) == 1:
|
if int(choice) == 1:
|
||||||
device = adb_connect()
|
root.root_device()
|
||||||
twrp_download(device)
|
|
||||||
push_files(device)
|
|
||||||
reboot_bootloader()
|
|
||||||
root_device()
|
|
||||||
elif int(choice) == 2:
|
elif int(choice) == 2:
|
||||||
print("Data extraction is coming soon.")
|
print("Data extraction is coming soon.")
|
||||||
time.sleep(2)
|
time.sleep(2)
|
||||||
|
|
@ -135,14 +39,10 @@ def show_help():
|
||||||
|
|
||||||
def main():
|
def main():
|
||||||
if "--interactive" in sys.argv[1:] or "-i" in sys.argv[1:]:
|
if "--interactive" in sys.argv[1:] or "-i" in sys.argv[1:]:
|
||||||
adb_start()
|
root.adb_start()
|
||||||
menu()
|
menu()
|
||||||
elif "--root" in sys.argv[1:] or "-r" in sys.argv[1:]:
|
elif "--root" in sys.argv[1:] or "-r" in sys.argv[1:]:
|
||||||
device = adb_connect()
|
root.root_device()
|
||||||
twrp_download(device)
|
|
||||||
push_files(device)
|
|
||||||
reboot_bootloader()
|
|
||||||
root_device()
|
|
||||||
elif "--help" in sys.argv[1:] or "-h" in sys.argv[1:]:
|
elif "--help" in sys.argv[1:] or "-h" in sys.argv[1:]:
|
||||||
show_help()
|
show_help()
|
||||||
else:
|
else:
|
||||||
|
|
|
||||||
BIN
helpers/__pycache__/root.cpython-37.pyc
Normal file
BIN
helpers/__pycache__/root.cpython-37.pyc
Normal file
Binary file not shown.
97
helpers/root.py
Normal file
97
helpers/root.py
Normal file
|
|
@ -0,0 +1,97 @@
|
||||||
|
import adbutils
|
||||||
|
import os
|
||||||
|
import sys
|
||||||
|
import time
|
||||||
|
import re
|
||||||
|
import requests
|
||||||
|
from lxml import html
|
||||||
|
|
||||||
|
def adb_start():
|
||||||
|
if "platform-tools" in os.environ['PATH']:
|
||||||
|
print("ADB found in PATH")
|
||||||
|
else:
|
||||||
|
os.environ['PATH'] += ';'+os.getcwd()+'\\platform-tools'
|
||||||
|
|
||||||
|
|
||||||
|
def adb_connect():
|
||||||
|
adb = adbutils.AdbClient(host="127.0.0.1", port=5037)
|
||||||
|
d = adb.device()
|
||||||
|
if d != None:
|
||||||
|
print("Device connected.")
|
||||||
|
return d
|
||||||
|
|
||||||
|
|
||||||
|
def twrp_download(d):
|
||||||
|
found = 0
|
||||||
|
files = os.listdir(os.curdir)
|
||||||
|
for file in files:
|
||||||
|
if "twrp" in file:
|
||||||
|
found = 1
|
||||||
|
if found != 1:
|
||||||
|
cpu = d.shell('cat /proc/cpuinfo | grep Hardware')
|
||||||
|
cpu = cpu.replace(" ", "")
|
||||||
|
cpu = re.sub(r'(.+:)', '', cpu)
|
||||||
|
r = requests.get('https://dl.twrp.me/'+cpu)
|
||||||
|
tree = html.fromstring(r.text)
|
||||||
|
urls = tree.xpath('//a/@href')
|
||||||
|
downloads = []
|
||||||
|
for i in urls:
|
||||||
|
if "img" in i:
|
||||||
|
downloads.append(i)
|
||||||
|
url_to_download = "https://dl.twrp.me"+downloads[0]
|
||||||
|
url_to_download = url_to_download.replace('.html', '')
|
||||||
|
print("Use this link to download twrp for your connected device: "+url_to_download)
|
||||||
|
print("Ensure that the downloaded file is moved to the same folder as the script before continuing")
|
||||||
|
input("Press Enter to continue...")
|
||||||
|
files = os.listdir(os.curdir)
|
||||||
|
for file in files:
|
||||||
|
if "twrp" in file:
|
||||||
|
found = 1
|
||||||
|
while found != 1:
|
||||||
|
print("File not found. Please confirm it has been moved to the correct directory")
|
||||||
|
input("Press Enter to continue...")
|
||||||
|
files = os.listdir(os.curdir)
|
||||||
|
for file in files:
|
||||||
|
if "twrp" in file:
|
||||||
|
found = 1
|
||||||
|
else:
|
||||||
|
print("twrp already downloaded")
|
||||||
|
|
||||||
|
|
||||||
|
def push_files(d):
|
||||||
|
initcheck_magisk = d.shell("cd /sdcard && ls | grep Magisk")
|
||||||
|
if initcheck_magisk == None and initcheck_twrp == None:
|
||||||
|
d.sync.push("Magisk-v19.3.zip", "/sdcard/Magisk.zip")
|
||||||
|
check_magisk = d.shell("cd /sdcard && ls | grep Magisk")
|
||||||
|
if check != None:
|
||||||
|
print("File copied successfully.")
|
||||||
|
else:
|
||||||
|
print("Something went wrong. Please try again.")
|
||||||
|
else:
|
||||||
|
print("Magisk already copied")
|
||||||
|
|
||||||
|
|
||||||
|
def reboot_bootloader():
|
||||||
|
adb = "platform-tools\\adb.exe"
|
||||||
|
fastboot = "platform-tools\\fastboot.exe"
|
||||||
|
os.system(adb+" reboot bootloader")
|
||||||
|
input("Press Enter when the device has rebooted")
|
||||||
|
os.system(fastboot+" devices")
|
||||||
|
|
||||||
|
|
||||||
|
def root_device():
|
||||||
|
device = adb_connect()
|
||||||
|
twrp_download(device)
|
||||||
|
push_files(device)
|
||||||
|
reboot_bootloader()
|
||||||
|
adb = "platform-tools\\adb.exe"
|
||||||
|
fastboot = "platform-tools\\fastboot.exe"
|
||||||
|
files = os.listdir(os.curdir)
|
||||||
|
for file in files:
|
||||||
|
if "twrp" in file:
|
||||||
|
twrp = file
|
||||||
|
os.system(fastboot + " boot "+twrp)
|
||||||
|
input("Press Enter when TWRP has booted")
|
||||||
|
print("Follow the onscreen directions to install Magisk (Located at the bottom of the install window)")
|
||||||
|
print("After Magisk installs click [Reboot] then [Do Not Install]")
|
||||||
|
input("Press Enter when the device has rebooted")
|
||||||
Reference in New Issue
Block a user