Skip to content

Commit

Permalink
Merge pull request #3 from kanzure/properties
Browse files Browse the repository at this point in the history 
Use properties and some python formatting changes
  • Loading branch information
@antitree
antitree committed Jul 16, 2012
2 parents 0d26575 + 7ffa0c7 commit a18f8a1
Show file tree
Hide file tree
Showing 9 changed files with 424 additions and 416 deletions.
7 changes: 7 additions & 0 deletions .gitignore
View file
@@ -1 +1,8 @@
# ignore precompiled python files
*.pyc

# ignore vim editor leftovers
.*.swp
.*.swo
.*~
*~
6 changes: 3 additions & 3 deletions README
View file
@@ -1,17 +1,17 @@
# -------------------------------------------------------------------
# AndroidXMLParser was created by Anthony Desnos and is taken from
# AndroidXMLParser was created by Anthony Desnos and is taken from
# the Androguard Project.
# -------------------------------------------------------------------
# This program is free software: you can redistribute it and/or
# modify it under the terms of the GNU General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
#
# You should have received a copy of the GNU General Public License
# along with this program.
# If not, see <http://www.gnu.org/licenses/>.
Expand Down
218 changes: 113 additions & 105 deletions apk.py
View file
Expand Up @@ -25,253 +25,261 @@
from struct import pack, unpack
from xml.dom import minidom

try :
try:
import chilkat
ZIPMODULE = 0
# UNLOCK : change it with your valid key !
try :
# UNLOCK: change it with your valid key !
try:
CHILKAT_KEY = open("key.txt", "rb").read()
except Exception :
except Exception:
CHILKAT_KEY = "testme"

except ImportError :
except ImportError:
ZIPMODULE = 1


######################################################## APK FORMAT ########################################################
class APK :
class APK:
"""APK manages apk file format"""
def __init__(self, filename, raw=False) :
def __init__(self, filename, raw=False):
"""
@param filename : specify the path of the file, or raw data
@param raw : specify (boolean) if the filename is a path or raw data
@param filename: specify the path of the file, or raw data
@param raw: specify (boolean) if the filename is a path or raw data
"""
self.filename = filename

self.xml = {}
self.package = ""
self.androidversion = {}
self.permissions = []
self.validAPK = False
self.valid_apk = False

if raw == True :
if raw == True:
self.__raw = filename
else :
fd = open( filename, "rb" )
else:
fd = open(filename, "rb")
self.__raw = fd.read()
fd.close()

if ZIPMODULE == 0:
self.zip = ChilkatZip(self.__raw)
else:
self.zip = zipfile.ZipFile(StringIO.StringIO(self.__raw))

if ZIPMODULE == 0 :
self.zip = ChilkatZip( self.__raw )
else :
self.zip = zipfile.ZipFile( StringIO.StringIO( self.__raw ) )

# CHECK if there is only one embedded file
#self._reload_apk()

for i in self.zip.namelist() :
if i == "AndroidManifest.xml" :
self.xml[i] = minidom.parseString( AXMLPrinter( self.zip.read( i ) ).getBuff() )
for i in self.zip.namelist():
if i == "AndroidManifest.xml":
self.xml[i] = minidom.parseString(AXMLPrinter(self.zip.read(i)).getBuff())

self.package = self.xml[i].documentElement.getAttribute( "package" )
self.androidversion["Code"] = self.xml[i].documentElement.getAttribute( "android:versionCode" )
self.androidversion["Name"] = self.xml[i].documentElement.getAttribute( "android:versionName")
self.package = self.xml[i].documentElement.getAttribute("package")
self.androidversion["Code"] = self.xml[i].documentElement.getAttribute("android:versionCode")
self.androidversion["Name"] = self.xml[i].documentElement.getAttribute("android:versionName")

for item in self.xml[i].getElementsByTagName('uses-permission') :
self.permissions.append( str( item.getAttribute("android:name") ) )
for item in self.xml[i].getElementsByTagName("uses-permission"):
self.permissions.append(str(item.getAttribute("android:name")))

self.validAPK = True
self.valid_apk = True

def is_valid_APK(self) :
return self.validAPK
def is_valid_apk(self):
return self.valid_apk

#def _reload_apk(self) :
# if len(files) == 1 :
# if ".apk" in files[0] :
# self.__raw = self.zip.read( files[0] )
# if ZIPMODULE == 0 :
# self.zip = ChilkatZip( self.__raw )
# else :
# self.zip = zipfile.ZipFile( StringIO.StringIO( self.__raw ) )
#def _reload_apk(self):
# if len(files) == 1:
# if ".apk" in files[0]:
# self.__raw = self.zip.read(files[0])
# if ZIPMODULE == 0:
# self.zip = ChilkatZip(self.__raw)
# else:
# self.zip = zipfile.ZipFile(StringIO.StringIO(self.__raw))

def get_filename(self) :
def get_filename(self):
"""
Return the filename of the APK
"""
return self.filename

def get_package(self) :
def get_package(self):
"""
Return the name of the package
"""
return self.package

def get_androidversion_code(self) :
def get_androidversion_code(self):
"""
Return the android version code
"""
return self.androidversion["Code"]
androidversion_code = property(get_androidversion_code)

def get_androidversion_name(self) :
def get_androidversion_name(self):
"""
Return the android version name
Return the android version name
"""
return self.androidversion["Name"]
androidversion_name = property(get_androidversion_name)

def get_files(self) :
def get_files(self):
"""
Return the files inside the APK
"""
return self.zip.namelist()
files = property(get_files)

def get_files_types(self) :
def get_files_types(self):
"""
Return the files inside the APK with their types (by using python-magic)
"""
try :
try:
import magic
except ImportError :
except ImportError:
return {}

l = {}

builtin_magic = 0
try :
try:
getattr(magic, "Magic")
except AttributeError :
except AttributeError:
builtin_magic = 1
if builtin_magic :

if builtin_magic:
ms = magic.open(magic.MAGIC_NONE)
ms.load()
for i in self.get_files() :
l[ i ] = ms.buffer( self.zip.read( i ) )
else :

for i in self.get_files():
l[ i ] = ms.buffer(self.zip.read(i))
else:
m = magic.Magic()
for i in self.get_files() :
l[ i ] = m.from_buffer( self.zip.read( i ) )
for i in self.get_files():
l[ i ] = m.from_buffer(self.zip.read(i))

return l
files_types = property(get_files_types)

def get_raw(self) :
"""
def get_raw(self):
"""
Return raw bytes of the APK
"""
return self.__raw
raw = property(get_raw)

def get_file(self, filename) :
def get_file(self, filename):
"""
Return the raw data of the specified filename
"""
try :
return self.zip.read( filename )
except KeyError :
try:
return self.zip.read(filename)
except KeyError:
return ""

def get_dex(self) :
def get_dex(self):
"""
Return the raw data of the classes dex file
"""
return self.get_file( "classes.dex" )
return self.get_file("classes.dex")
dex = property(get_dex)

def get_elements(self, tag_name, attribute) :
def get_elements(self, tag_name, attribute):
"""
Return elements in xml files which match with the tag name and the specific attribute
@param tag_name : a string which specify the tag name
@param attribute : a string which specify the attribute
@param tag_name: a string which specify the tag name
@param attribute: a string which specify the attribute
"""
l = []
for i in self.xml :
for item in self.xml[i].getElementsByTagName(tag_name) :
for i in self.xml:
for item in self.xml[i].getElementsByTagName(tag_name):
value = item.getAttribute(attribute)
if len(value) > 0 :
if value[0] == "." :

if len(value) > 0:
if value[0] == ".":
value = self.package + value
else :
else:
v_dot = value.find(".")
if v_dot == 0 :
if v_dot == 0:
value = self.package + "." + value
elif v_dot == -1 :
elif v_dot == -1:
value = self.package + "." + value

l.append( str( value ) )
l.append(str(value))
return l

def get_element(self, tag_name, attribute) :
def get_element(self, tag_name, attribute):
"""
Return element in xml files which match with the tag name and the specific attribute
@param tag_name : a string which specify the tag name
@param attribute : a string which specify the attribute
@param tag_name: a string which specify the tag name
@param attribute: a string which specify the attribute
"""
l = []
for i in self.xml :
for item in self.xml[i].getElementsByTagName(tag_name) :
for i in self.xml:
for item in self.xml[i].getElementsByTagName(tag_name):
value = item.getAttribute(attribute)

if len(value) > 0 :
if len(value) > 0:
return value
return None

def get_activities(self) :
def get_activities(self):
"""
Return the android:name attribute of all activities
"""
return self.get_elements("activity", "android:name")
activities = property(get_activities)

def get_services(self) :
def get_services(self):
"""
Return the android:name attribute of all services
"""
return self.get_elements("service", "android:name")
services = property(get_services)

def get_receivers(self) :
def get_receivers(self):
"""
Return the android:name attribute of all receivers
"""
return self.get_elements("receiver", "android:name")
receivers = property(get_receivers)

def get_providers(self) :
def get_providers(self):
"""
Return the android:name attribute of all providers
"""
return self.get_elements("provider", "android:name")
providers = property(get_providers)

def get_permissions(self) :
def get_permissions(self):
"""
Return permissions
"""
return self.permissions
permissions = property(get_permissions)

def get_min_sdk_version(self) :
def get_min_sdk_version(self):
"""
Return the android:minSdkVersion attribute
"""
return self.get_element( "uses-sdk", "android:minSdkVersion" )
return self.get_element("uses-sdk", "android:minSdkVersion")
min_sdk_version = property(get_min_sdk_version)

def get_target_sdk_version(self) :
def get_target_sdk_version(self):
"""
Return the android:targetSdkVersion attribute
"""
return self.get_element( "uses-sdk", "android:targetSdkVersion" )
return self.get_element("uses-sdk", "android:targetSdkVersion")
target_sdk_version = property(get_target_sdk_version)

def get_libraries(self) :
def get_libraries(self):
"""
Return the android:name attributes for libraries
"""
return self.get_elements( "uses-library", "android:name" )

def show(self) :
print "FILES : ", self.get_files_types()

print "ACTIVITIES : ", self.get_activities()
print "SERVICES : ", self.get_services()
print "RECEIVERS : ", self.get_receivers()
print "PROVIDERS : ", self.get_providers()
return self.get_elements("uses-library", "android:name")
libraries = property(get_libraries)

def show(self):
print "FILES: ", self.get_files_types()

print "ACTIVITIES: ", self.get_activities()
print "SERVICES: ", self.get_services()
print "RECEIVERS: ", self.get_receivers()
print "PROVIDERS: ", self.get_providers()

0 comments on commit a18f8a1

Please sign in to comment.