def NumToNameForOneFile(path):
if Smali.isSmali(path):
path = Smali.getSmaliMainPath(path)
else:
return
if os.path.exists(path):
fDir = os.path.dirname(path)
if cmp(fDir, "") == 0:
fDir = "."
name = Smali.getSmaliRoot(path)
else:
return
if os.path.exists(os.path.join(fDir, name)+".data"):
if False: print "NumToName: "+os.path.join(fDir, name)+".data is exist, ignore!"
return os.path.join(fDir, name) + ".data"
java = Java(fDir, name)
#java.printJava()
if java.getListLen() == 0:
if False: print "Can not find smali file: "+os.path.join(java.path, java.name)+"*.smali"
return
toName = NumToName(java.name, java.path, java.smaliList)
#toName.printAccessSmaliSet()
toName.doNumToName()
toName.dumpMap()
return os.path.join(java.path, java.name) + ".data"
def NameToNumForOneFile(path):
if Smali.isSmali(path):
path = Smali.getDataFilePath(
path) #change smali path to data file path
if DataFile.isDataFile(path) and os.path.exists(path):
fDir = os.path.dirname(path)
if cmp(fDir, "") == 0:
fDir = "."
name = DataFile.getDataFileName(path)
else:
return
java = Java(fDir, name)
#java.printJava()
if java.getListLen() == 0:
print "Can not find data file: " + os.path.join(java.path,
java.name) + ".data"
return
if False:
print "NameToNum: " + os.path.join(java.path, java.name) + ".data"
toNum = NameToNum(java.name, java.path, java.smaliList)
#toNum.printAccessSmaliSet()
toNum.doNameToNum()
os.remove(path)
def getAccessSmaliSet(self):
""" check smalilist to build access num to name map """
for s in self.smaliList:
sName = Smali.getSmaliName(s)
aSmali = AccessSmali(sName)
mode = 0
method = ""
for line in open(os.path.join(self.path, s), 'r'):
if (mode == 0):
match = patternMethodBegin.match(line)
if(match):
method = Method(match.group("accessnum"), match.group("parameterlist"), match.group("returntype"))
mode = 1
elif (mode == 1):
match = patternMethodEnd.match(line)
if(match):
method.endMethod()
mode = 0
aSmali.addMethod(method.accessNum, method.accessName)
else:
match = patternMethodAnnotation.match(line)
if(match):
continue
match = patternMethodCall.match(line)
if(match):
method.addCallLine(match.group("operation"), match.group("operationobject"), match.group(0))
else:
method.addMethodLine(line)
if (aSmali.getMethodNumSetLen() > 0):
aSmali.createNumMap()
self.accessSmaliSet[sName] = aSmali
def NameToNumForOneFile(path):
if Smali.isSmali(path):
path = Smali.getDataFilePath(path) #change smali path to data file path
if DataFile.isDataFile(path) and os.path.exists(path):
fDir = os.path.dirname(path)
if cmp(fDir, "") == 0:
fDir = "."
name = DataFile.getDataFileName(path)
else:
return
java = Java(fDir, name)
#java.printJava()
if java.getListLen() == 0:
print "Can not find data file: "+os.path.join(java.path, java.name)+".data"
return
if False: print "NameToNum: "+os.path.join(java.path, java.name)+".data"
toNum = NameToNum(java.name, java.path, java.smaliList)
#toNum.printAccessSmaliSet()
toNum.doNameToNum()
os.remove(path)
def doNumToName(self):
allMethodCallNumMap = {}
for sName in self.accessSmaliSet.keys():
callNumMap = self.accessSmaliSet[sName].methodCallNumMap
for callNum in callNumMap.keys():
if callNum not in allMethodCallNumMap.keys():
allMethodCallNumMap[callNum] = callNumMap[callNum]
else:
raise ValueError("method call num map duplicate")
for s in self.smaliList:
sFile = File(os.path.join(self.path, s))
sName = Smali.getSmaliName(s)
if sName in self.accessSmaliSet.keys():
sFile.replaces(self.accessSmaliSet[sName].methodDefNumMap)
sFile.replaces(allMethodCallNumMap)
def doNameToNum(self):
allMethodCallNameMap = {}
for aSmali in self.accessSmaliSet.keys():
self.accessSmaliSet[aSmali].createNameMap()
callNameMap = self.accessSmaliSet[aSmali].methodCallNameMap
for callName in callNameMap.keys():
if callName not in allMethodCallNameMap.keys():
allMethodCallNameMap[callName] = callNameMap[callName]
else:
raise ValueError("method call name map duplicate")
for s in self.smaliList:
sFile = File(os.path.join(self.path, s))
sName = Smali.getSmaliName(s)
if sName in self.accessSmaliSet.keys():
sFile.replaces(self.accessSmaliSet[sName].methodDefNameMap)
sFile.replaces(allMethodCallNameMap)
def doNameToNum(self):
allMethodCallNameMap = {}
for aSmali in self.accessSmaliSet.keys():
self.accessSmaliSet[aSmali].createNameMap()
callNameMap = self.accessSmaliSet[aSmali].methodCallNameMap
for callName in callNameMap.keys():
if callName not in allMethodCallNameMap.keys():
allMethodCallNameMap[callName] = callNameMap[callName]
else:
raise ValueError("method call name map duplicate")
for s in self.smaliList:
sFile = File(os.path.join(self.path, s))
sName = Smali.getSmaliName(s)
if sName in self.accessSmaliSet.keys():
sFile.replaces(self.accessSmaliSet[sName].methodDefNameMap)
sFile.replaces(allMethodCallNameMap)
toName.dumpMap()
return os.path.join(java.path, java.name) + ".data"
def Usage():
print "Usage: num2name.py aa/bb/A.smali"
print " num2name.py aa/bb/B$1.smali"
print " num2name.py aa/bb"
if __name__ == '__main__':
argLen = len(sys.argv)
if argLen == 2:
path = sys.argv[1]
if Smali.isSmali(path):
NumToNameForOneFile(path)
elif os.path.isdir(path):
for root, dirs, files in os.walk(path):
for sfile in files:
fPath = os.path.join(root, sfile)
if Smali.isRootSmali(fPath):
NumToNameForOneFile(fPath)
else:
Usage()
else:
Usage()
toNum.doNameToNum()
os.remove(path)
def Usage():
print "Usage: name2num.py aa/bb/A.data"
print " name2num.py aa/bb/A.smali"
print " name2num.py aa/bb"
if __name__ == '__main__':
argLen = len(sys.argv)
if argLen == 2:
path = sys.argv[1]
if os.path.isfile(path) and (DataFile.isDataFile(path)
or Smali.isSmali(path)):
NameToNumForOneFile(path)
elif os.path.isdir(path):
for root, dirs, files in os.walk(path):
for sfile in files:
fPath = os.path.join(root, sfile)
if DataFile.isDataFile(fPath):
NameToNumForOneFile(fPath)
else:
Usage()
else:
Usage()
toNum = NameToNum(java.name, java.path, java.smaliList)
#toNum.printAccessSmaliSet()
toNum.doNameToNum()
os.remove(path)
def Usage():
print "Usage: name2num.py aa/bb/A.data"
print " name2num.py aa/bb/A.smali"
print " name2num.py aa/bb"
if __name__ == '__main__':
argLen = len(sys.argv)
if argLen == 2:
path = sys.argv[1]
if os.path.isfile(path) and (DataFile.isDataFile(path) or Smali.isSmali(path)):
NameToNumForOneFile(path)
elif os.path.isdir(path):
for root, dirs, files in os.walk(path):
for sfile in files:
fPath = os.path.join(root, sfile)
if DataFile.isDataFile(fPath):
NameToNumForOneFile(fPath)
else:
Usage()
else:
Usage()