forked from abrignoni/ALEAPP
-
-
Notifications
You must be signed in to change notification settings - Fork 1
/
aleappGUI.py
executable file
·241 lines (204 loc) · 10.8 KB
/
aleappGUI.py
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
import json
import pathlib
import typing
import aleapp
import PySimpleGUI as sg
import webbrowser
import plugin_loader
from scripts.ilapfuncs import *
from scripts.version_info import aleapp_version
from time import process_time, gmtime, strftime
from scripts.search_files import *
MODULE_START_INDEX = 1000
def ValidateInput(values, window):
'''Returns tuple (success, extraction_type)'''
global module_end_index
i_path = values[0] # input file/folder
o_path = values[1] # output folder
ext_type = ''
if len(i_path) == 0:
sg.PopupError('No INPUT file or folder selected!')
return False, ext_type
elif not os.path.exists(i_path):
sg.PopupError('INPUT file/folder does not exist!')
return False, ext_type
elif os.path.isdir(i_path):
ext_type = 'fs'
else: # must be an existing file then
if not i_path.lower().endswith('.tar') and not i_path.lower().endswith('.zip') and not i_path.lower().endswith('.gz'):
sg.PopupError('Input file is not a supported archive! ', i_path)
return False, ext_type
else:
ext_type = Path(i_path).suffix[1:].lower()
# check output now
if len(o_path) == 0 : # output folder
sg.PopupError('No OUTPUT folder selected!')
return False, ext_type
one_element_is_selected = False
for x in range(1000, module_end_index):
if window.FindElement(x).Get():
one_element_is_selected = True
break
if not one_element_is_selected:
sg.PopupError('No module selected for processing!')
return False, ext_type
return True, ext_type
# initialize CheckBox control with module name
def CheckList(mtxt, lkey, mdstring, disable=False):
if mdstring == 'test1' or mdstring == 'test2' : #items in the if are modules that take a long time to run. Deselects them by default.
dstate = False
else:
dstate = True
return [sg.CBox(mtxt, default=dstate, key=lkey, metadata=mdstring, disabled=disable)]
def pickModules():
global module_end_index
global mlist
global loader
loader = plugin_loader.PluginLoader()
module_end_index = MODULE_START_INDEX # arbitrary number to not interfere with other controls
for plugin in sorted(loader.plugins, key=lambda p: p.category.upper()):
disabled = plugin.module_name == 'usagestatsVersion'
mlist.append(CheckList(f'{plugin.category} [{plugin.name} - {plugin.module_name}.py]', module_end_index, plugin.name, disabled))
module_end_index = module_end_index + 1
sg.theme('LightGreen5') # Add a touch of color
# All the stuff inside your window.
normal_font = ("Helvetica", 12)
loader: typing.Optional[plugin_loader.PluginLoader] = None
mlist = []
# go through list of available modules and confirm they exist on the disk
pickModules()
GuiWindow.progress_bar_total = len(loader)
layout = [ [sg.Text('Android Logs, Events, And Protobuf Parser', font=("Helvetica", 22))],
[sg.Text('https://github.com/abrignoni/ALEAPP', font=("Helvetica", 14))],
[sg.Frame(layout=[
[sg.Input(size=(97,1)),
sg.FileBrowse(font=normal_font, button_text='Browse File', key='INPUTFILEBROWSE'),
sg.FolderBrowse(font=normal_font, button_text='Browse Folder', target=(sg.ThisRow, -2), key='INPUTFOLDERBROWSE')
]
],
title='Select a file (tar/zip/gz) or directory of the target Android full file system extraction for parsing:')],
[sg.Frame(layout=[
[sg.Input(size=(112, 1)), sg.FolderBrowse(font=normal_font, button_text='Browse Folder')]
],
title='Select Output Folder:')],
[sg.Text('Available Modules')],
[sg.Button('Select All', key='SELECT ALL'), sg.Button('Deselect All', key='DESELECT ALL'),
sg.Button('Load Profile', key='LOAD PROFILE'), sg.Button('Save Profile', key='SAVE PROFILE')
# sg.FileBrowse(
# button_text='Load Profile', key='LOADPROFILE', enable_events=True, target='LOADPROFILE',
# file_types=(('ALEAPP Profile (*.alprofile)', '*.alprofile'), ('All Files', '*'))),
# sg.FileSaveAs(
# button_text='Save Profile', key='SAVEPROFILE', enable_events=True, target='SAVEPROFILE',
# file_types=(('ALEAPP Profile (*.alprofile)', '*.alprofile'), ('All Files', '*')),
# default_extension='.alprofile')
],
[sg.Column(mlist, size=(300,310), scrollable=True), sg.Output(size=(85,20))] ,
[sg.ProgressBar(max_value=GuiWindow.progress_bar_total, orientation='h', size=(86, 7), key='PROGRESSBAR', bar_color=('DarkGreen', 'White'))],
[sg.Submit('Process', font=normal_font), sg.Button('Close', font=normal_font)] ]
# Create the Window
window = sg.Window(f'ALEAPP version {aleapp_version}', layout)
GuiWindow.progress_bar_handle = window['PROGRESSBAR']
# Event Loop to process "events" and get the "values" of the inputs
while True:
event, values = window.read()
if event in (None, 'Close'): # if user closes window or clicks cancel
break
if event == "SELECT ALL":
# mark all modules
for x in range(MODULE_START_INDEX, module_end_index):
window[x].Update(True)
if event == "DESELECT ALL":
# none modules
for x in range(MODULE_START_INDEX, module_end_index):
window[x].Update(False if window[x].metadata != 'usagestatsVersion' else True) # usagestatsVersion.py is REQUIRED
if event == "SAVE PROFILE":
destination_path = sg.popup_get_file(
"Save a profile", save_as=True,
file_types=(('ALEAPP Profile (*.alprofile)', '*.alprofile'),),
default_extension='.alprofile', no_window=True)
if destination_path:
ticked = []
for x in range(MODULE_START_INDEX, module_end_index):
if window.FindElement(x).Get():
key = window[x].metadata
ticked.append(key)
with open(destination_path, "wt", encoding="utf-8") as profile_out:
json.dump({"leapp": "aleapp", "format_version": 1, "plugins": ticked}, profile_out)
sg.Popup(f"Profile saved: {destination_path}")
if event == "LOAD PROFILE":
destination_path = sg.popup_get_file(
"Load a profile", save_as=False,
file_types=(('ALEAPP Profile (*.alprofile)', '*.alprofile'), ('All Files', '*')),
default_extension='.alprofile', no_window=True)
if destination_path and os.path.exists(destination_path):
profile_load_error = None
with open(destination_path, "rt", encoding="utf-8") as profile_in:
try:
profile = json.load(profile_in)
except json.JSONDecodeError as json_ex:
profile_load_error = f"File was not a valid profile file: {json_ex}"
if not profile_load_error:
if isinstance(profile, dict):
if profile.get("leapp") != "aleapp" or profile.get("format_version") != 1:
profile_load_error = "File was not a valid profile file: incorrect LEAPP or version"
else:
ticked = set(profile.get("plugins", []))
ticked.add("usagestatsVersion") # always
for x in range(MODULE_START_INDEX, module_end_index):
if window[x].metadata in ticked:
window[x].update(True)
else:
window[x].update(False)
else:
profile_load_error = "File was not a valid profile file: invalid format"
if profile_load_error:
sg.popup(profile_load_error)
else:
sg.popup(f"Loaded profile: {destination_path}")
if event == 'Process':
#check is selections made properly; if not we will return to input form without exiting app altogether
is_valid, extracttype = ValidateInput(values, window)
if is_valid:
GuiWindow.window_handle = window
input_path = values[0]
output_folder = values[1]
# File system extractions can contain paths > 260 char, which causes problems
# This fixes the problem by prefixing \\?\ on each windows path.
if is_platform_windows():
if input_path[1] == ':' and extracttype =='fs': input_path = '\\\\?\\' + input_path.replace('/', '\\')
if output_folder[1] == ':': output_folder = '\\\\?\\' + output_folder.replace('/', '\\')
# re-create modules list based on user selection
#search_list = { 'usagestatsVersion' : tosearch['usagestatsVersion'] } # hardcode usagestatsVersion as first item
search_list = [loader['usagestatsVersion']] # hardcode usagestatsVersion as first item
s_items = 0
for x in range(MODULE_START_INDEX, module_end_index):
if window.FindElement(x).Get():
key = window[x].metadata
if key in loader and key != 'usagestatsVersion':
search_list.append(loader[key])
s_items = s_items + 1 # for progress bar
# no more selections allowed
window[x].Update(disabled=True)
window['SELECT ALL'].update(disabled=True)
window['DESELECT ALL'].update(disabled=True)
GuiWindow.window_handle = window
out_params = OutputParameters(output_folder)
wrap_text = True
crunch_successful = aleapp.crunch_artifacts(
search_list, extracttype, input_path, out_params, len(loader)/s_items, wrap_text)
if crunch_successful:
report_path = os.path.join(out_params.report_folder_base, 'index.html')
if report_path.startswith('\\\\?\\'): # windows
report_path = report_path[4:]
if report_path.startswith('\\\\'): # UNC path
report_path = report_path[2:]
locationmessage = 'Report name: ' + report_path
sg.Popup('Processing completed', locationmessage)
webbrowser.open_new_tab('file://' + report_path)
else:
log_path = out_params.screen_output_file_path
if log_path.startswith('\\\\?\\'): # windows
log_path = log_path[4:]
sg.Popup('Processing failed :( ', f'See log for error details..\nLog file located at {log_path}')
break
window.close()