python_static_analyzer/main_badzipfile.py at master · CMUChimpsLab/python_static_analyzer · GitHub

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
'''
Created on Sep 9, 2012

@author: psachdev
'''
#import manager
import namespaceanalyzer
import permission
import SearchIntents
import os
import DbManager
import logging
import signal
import sys
# coding: utf-8
from androguard.core.bytecodes import apk
from androguard.core.bytecodes import dvm
from androguard.core.analysis.analysis import *

chilkat=True
def handler (signum, sigframe):
    raise Exception ("Killed");


def naiveSearch(aDict, searchString):
    for key in aDict:
        if searchString in key:
            return 1
    return 0

if __name__ == '__main__':
    DIR = sys.argv[1]
    '''
    Database Handle used to insert fields
    '''
    dbMgr = DbManager.DBManagerClass()

    '''
    Example of how the various entrie are made into the database
    dbMgr.insert3rdPartyPackageInfo("testpackage", "testfilename", "testexternalpackage")
    dbMgr.insertPermissionInfo('testpackage', 'testfilename', 'testpermission', True, 'testdest', 'testexternalpackagename', 'testsrc')
    dbMgr.insertLinkInfo('testpackage', 'testfilename', 'testlink', True, 'testdest', 'testexternalpackagename')
    '''
    # Make a global logging object.
    logObject = logging.getLogger("logfile")
    logObject.setLevel(logging.DEBUG)
    # create file handler which logs even debug messages
    logFileHandler = logging.FileHandler('/home/ubuntu/out/exceptions_withZipModule.log')
    logFileHandler.setLevel(logging.DEBUG)
    logFormat = logging.Formatter("%(levelname)s %(asctime)s %(funcName)s %(lineno)d %(message)s")
    logFileHandler.setFormatter(logFormat)
    logObject.addHandler(logFileHandler)

    #signal.signal(signal.SIGKILL, handler)

    #DIR = '<Directory-Name-where all apks reside>'
    #OUT = '<Directory-Name where output files are expected>'
    OUT = '/home/ubuntu/out/'

    filesread = {}
    filelistname = '/home/ubuntu/out/exceptions.log'
    filelisthandle = open (filelistname, 'r+')
    lines = filelisthandle.readlines()
    for line in lines:
	if "<module> 106 Main : Exception" in line:
        	tokens = line.split()
        	tokens.reverse()
        	filesread[tokens[0]]=1
    filelisthandle.close()
    filelistname = '/home/ubuntu/out/filelist_withZipModule.txt'
    filelisthandle = open (filelistname, 'a')

    for path, dirs, files in os.walk(DIR):
        for fileName in files:
            try:
                print "FileName Analyzed :" + path + '/' + fileName
                tokens = namespaceanalyzer.NameSpaceMgr.GetTokensStatic (path, '/')
                category =  tokens [len (tokens) - 1]
                #print category
                filename = path + '/' + fileName
                if fileName != 'links.txt' and fileName != 'permissions.txt' and fileName != 'packages.txt' and fileName != '.dropbox' and fileName != 'list.csv' and fileName != 'externalpackcntt.csv' and fileName != 'externalpackpopularitycnt.csv' and fileName != 'packages.txt':

		    x = naiveSearch (filesread, fileName)
		    if x == 1:
		    	filelisthandle.write( str(fileName) )
		    	filelisthandle.write("\n")
			filelisthandle.flush()
		    else:
			continue
                    outFileName = '/package.txt'
                    outFileName = OUT + outFileName
                    instance = namespaceanalyzer.NameSpaceMgr()

                    a = apk.APK(filename)
                    d = dvm.DalvikVMFormat (a.get_dex())
                    dx = uVMAnalysis (d)

                    #packages = instance.execute (filename, outFileName, dbMgr, fileName, category, a, d, dx)

                    #outfile_perm = '/permissions.txt'
                    #outfile_perm = OUT + outfile_perm
                    #x = permission.StaticAnalyzer (filename, outfile_perm, packages, dbMgr, fileName, a, d, dx)

                    #outfile_links = '/links.txt'
                    #outfile_links = OUT + outfile_links
                    #SearchIntents.Intents(filename, outfile_links, packages, dbMgr, fileName, a, d, dx);
            except:
                logObject.error("\n")
                logObject.error("=======================================================================")
                logObject.error("\n")
                logObject.exception("Main : Exception occured for " + fileName)

    filelisthandle.close()