import sys
import os
import argparse
import platform

def parseArgument():
	parser = argparse.ArgumentParser(description='AndroBugs Framework: Android APK Vulnerability Scanner - Massive Tool')
	parser.add_argument("-d", "--input_apk_dir", help="APK input directory to analyze", type=str, required=True)
	parser.add_argument("-b", "--analyze_engine_build", help="Analysis build number.", type=int, required=True)
	parser.add_argument("-t", "--analyze_tag", help="Analysis tag to uniquely distinguish this time of analysis.", type=str, required=True)
	parser.add_argument("-o", "--report_output_dir", help="Analysis Report Output Directory.", type=str, required=True)
	parser.add_argument("-e", "--extra", help="1)Do not check(default)  2)Check  security class names, method names and native methods", type=int, required=False, default=1)
	parser.add_argument("-i", "--ignore_duplicated_scanning", help="If you specify this argument, APKs with the same \"package_name\", \"analyze_engine_build\" and \"analyze_tag\" will not be analyzed again.", action="store_true")
	args = parser.parse_args()
	return args

def main() :

	args = parseArgument()

	print
	print("## AndroBugs Framework: Android APK Vulnerability Scanner - Massive Tool ##")
	print

	ANALYZE_MODE_MASSIVE = "massive"

	if args.ignore_duplicated_scanning :

		from pymongo import MongoClient
		from ConfigParser import SafeConfigParser

		if platform.system().lower() == "windows" :
			import sys
			db_config_file = os.path.join(os.path.dirname(sys.executable), 'androbugs-db.cfg')
		else :
			db_config_file = os.path.join(os.path.dirname(os.path.abspath(__file__)), 'androbugs-db.cfg')

		if not os.path.isfile(db_config_file) :
			print("[ERROR] AndroBugs Framework DB config file not found: " + db_config_file)
			traceback.print_exc()

		configParser = SafeConfigParser()
		configParser.read(db_config_file)

		MongoDB_Hostname = configParser.get('DB_Config', 'MongoDB_Hostname')
		MongoDB_Port = configParser.getint('DB_Config', 'MongoDB_Port')
		MongoDB_Database = configParser.get('DB_Config', 'MongoDB_Database')

		Collection_Analyze_Result = configParser.get('DB_Collections', 'Collection_Analyze_Result')

		client = MongoClient(MongoDB_Hostname, MongoDB_Port)
		db = client[MongoDB_Database]	# Name is case-sensitive
		collection_AppInfo = db[Collection_Analyze_Result]		# Name is case-sensitive

		print("[Notice] APK with the same \"package_name\", \"analyze_engine_build\" and \"analyze_tag\" will not be analyzed again.")
		print

	input_dir = os.path.realpath(args.input_apk_dir)
	output_dir = os.path.realpath(args.report_output_dir)

	if (not os.path.isdir(input_dir)) :
		print("APK input directory does not exist.")
		sys.exit()

	dir_names = os.listdir(input_dir)
	total_dir = len(dir_names)
	current_file = 0

	for filename in dir_names:
		if filename.endswith(".apk") :
			current_file = current_file + 1

			package_name = filename[:-4]

			print("Analyzing APK(" + str(current_file) + "/" + str(total_dir) + "): " + filename)

			if args.ignore_duplicated_scanning :  #check if already scanned

				query_condition = { "analyze_mode" : ANALYZE_MODE_MASSIVE, 
									"package_name": package_name, 
									"analyze_engine_build": args.analyze_engine_build,
									"analyze_tag": args.analyze_tag }

				boolHasResult = False

				query_result = collection_AppInfo.find(query_condition)
				
				for result in query_result :
					boolHasResult = True
					break

				if boolHasResult :
					print(" ->Package name [" + package_name + "] has already in DB. Ignore analyzing it.")
					continue
			
			try:

				if platform.system().lower() == "windows" :
					main_cmd = "androbugs.exe"
				else :
					main_cmd = "python androbugs.py"

				cmd = main_cmd + " -s -v -e " + str(args.extra) + " -f " + os.path.join(input_dir, filename) + " -o " + output_dir + " -m " + ANALYZE_MODE_MASSIVE + " -b " + str(args.analyze_engine_build) + " -t " + str(args.analyze_tag)
				#print(cmd)
				process = os.popen(cmd)
				preprocessed = process.read()
				process.close()

			except KeyboardInterrupt :
				print("Stopped.")
				break
			except Exception as err:
				print(err)
				pass


if __name__ == "__main__":
	main()