-
Notifications
You must be signed in to change notification settings - Fork 1
/
para_scanner.py
48 lines (35 loc) · 1.33 KB
/
para_scanner.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
# This is a parallel scanner to process all apk files in a folder.
from multiprocessing import Queue, Process, pool, Lock, current_process
from depriv_tools.apk_processor import ApkProcessor
import os
class ParaScanner(object):
def __init__(self):
self.file_queue = Queue()
self.max_process = 10
def set_max_process(self, n):
self.max_process = n
def find_apk_files(self, folder_path):
for root, dirs, files in os.walk(folder_path):
for f in files:
file_name, extension = os.path.splitext(f)
if extension == '.apk':
self.file_queue.put(os.path.join(root, f))
def process_files(self):
''' Create a process pool to process each file.
'''
processes = []
for i in range(self.max_process):
p = ApkProcessor(self.file_queue)
# poison pill for each process and kill threads when no more tasks
self.file_queue.put('STOP')
p.start()
processes.append(p)
for p in processes:
p.join()
def scan(self, folder='.'):
self.find_apk_files(folder)
self.process_files()
if __name__ == '__main__':
scanner = ParaScanner()
g920a = '/home/xzhou/data/Samsung/depriviallage/priv-app/priv_app_G920A'
scanner.scan(g920a)