class Padding: def __init__(self): self.mixer = Mixer() self.utils = Utils() self.simpleSpace = " " def ScriptsGenerator(self, randomClassesFunctions, mixerLengthArg): varRandom1 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom2 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom3 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom4 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom5 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom6 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom7 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom8 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom9 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom10 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom11 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom12 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom13 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) varRandom14 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) # ---------- Python random snippets code ---------- # if randomClassesFunctions == True: rand = random.randint(1, 15) else: rand = random.randint(1, 9) # -- snippet 1 -- # if rand == 1: scriptPadding1 = textwrap.dedent(""" {0} = '{5}' {1} = '{6}' {2} = '{7}' {3} = '{8}' {4} = '{9}' if {0} in {1}: {0} = {4} if {1} in {2}: {1} = {3} elif {1} in {0}: {2} = {1} if {2} in {1}: {1} = {4} """).format(varRandom1, varRandom2, varRandom3, varRandom4, varRandom5, \ varRandom6, varRandom7, varRandom8, varRandom9, varRandom10) return scriptPadding1 # -- snippet 2 -- # elif rand == 2: scriptPadding2 = textwrap.dedent(""" {0} = '{2}' {1} = '{3}' if {0} != {1}: {0} = '{3}' {1} = {0} {0} = '{2}' """).format( varRandom1, varRandom2, varRandom3, varRandom4) return scriptPadding2 # -- snippet 3 -- # elif rand == 3: scriptPadding3 = textwrap.dedent(""" {0} = '{6}' {1} = '{7}' {2} = '{8}' {3} = '{9}' {4} = '{10}' {5} = '{11}' if {0} != {3}: {1} = {2} for {5} in {3}: if {5} != {2}: {1} = {1} else: {4} = {0} else: {2} = {0} {0} = {4} if {2} == {0}: for {5} in {0}: if {5} == {2}: {2} = {0} else: {2} = {4} """).format(varRandom1, varRandom2, varRandom3, varRandom4, varRandom5, \ varRandom6, varRandom7, varRandom8, varRandom9, varRandom10, \ varRandom11, varRandom12) return scriptPadding3 # -- snippet 4 -- # elif rand == 4: scriptPadding4 = textwrap.dedent(""" {0} = '{3}' {1} = '{4}' {2} = '{5}' if {0} == {1}: {2} = '{5}' {2} = {0} else: {2} = '{5}' {2} = '{3}' """).format(varRandom1, varRandom2, varRandom3, varRandom4, \ varRandom5, varRandom6,) return scriptPadding4 # -- snippet 5 -- # elif rand == 5: scriptPadding5 = textwrap.dedent(""" {0} = '{6}' {1} = '{7}' {2} = '{8}' {3} = '{9}' {4} = '{10}' {5} = '{11}' if {2} == {3}: for {5} in {4}: if {5} == {3}: {4} = {0} else: {3} = {1} """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8, varRandom9, \ varRandom10, varRandom11, varRandom12) return scriptPadding5 # -- snippet 6 -- # elif rand == 6: scriptPadding6 = textwrap.dedent(""" {0} = '{4}' {1} = '{5}' {2} = '{6}' {3} = '{7}' if {1} == {0}: for {0} in {1}: if {1} == {1}: {2} = '{3}' elif {2} == {3}: {3} = {0} else: {0} = {1} elif {2} == {2}: for {2} in {1}: if {3} == {1}: {2} = '{3}' elif {2} == {3}: {3} = {0} else: {0} = {1} for {2} in {1}: if {3} == {1}: {2} = '{3}' elif {2} == {3}: {3} = {0} else: {0} = {3} else: {0} = {1} """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8) return scriptPadding6 # -- snippet 7 -- # elif rand == 7: scriptPadding7 = textwrap.dedent(""" {0} = '{7}' {1} = '{8}' {2} = '{9}' {3} = '{10}' {4} = '{11}' {5} = '{12}' {6} = [ '{7}', '{9}', '{11}', '{13}' ] for {0} in {5}: for {1} in {2}: if {3} == {4}: {1} = {0} elif {4} == {1}: {1} = {5} else: {4} = {5} for {1} in {6}: {2} = {1} """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8, varRandom9, \ varRandom10, varRandom11, varRandom12, \ varRandom13, varRandom14) return scriptPadding7 # -- snippet 8 -- # elif rand == 8: scriptPadding8 = textwrap.dedent(""" {0} = '{1}' """).format( varRandom1, varRandom2) return scriptPadding8 # -- snippet 9 -- # elif rand == 9: scriptPadding9 = textwrap.dedent(""" {0} = '{1}' {2} = '{3} {1}' """).format(varRandom1, varRandom2, varRandom3, \ varRandom4,) return scriptPadding9 # ---------- Python random functions ---------- # # -- snippet 10 -- # elif rand == 10: scriptPadding10 = textwrap.dedent(""" def {0}(): {1} = '{2}' {3} = True """).format(varRandom1, varRandom2, varRandom3, \ varRandom4,) return scriptPadding10 # -- snippet 11 -- # elif rand == 11: scriptPadding11 = textwrap.dedent(""" def {0}({1}): {1} = '{2}' {3} = '{4}' if {1} == {3}: for {5} in {3}: if {5} == {3}: continue else: {6} = False """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7) return scriptPadding11 # -- snippet 12 -- # elif rand == 12: scriptPadding12 = textwrap.dedent(""" def {0}({1}, {2}): {1} = '{3}' {2} = '{4}' try: for {5} in {1}: if {5} == {2}: {6} = [] {7} = None else: pass except: pass """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8) return scriptPadding12 # -- snippet 13 -- # elif rand == 13: scriptPadding13 = textwrap.dedent(""" def {0}({1}, {2}, {3}): {1} = '{4}' {2} = '{5}' {6} = True while {6}: break for {6} in {2}: {3} = False """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7) return scriptPadding13 # ---------- Python random classes ---------- # # -- snippet 14 -- # elif rand == 14: scriptPadding14 = textwrap.dedent(""" class {0}: def __init__(self): self.{1} = '{2}' self.{3} = '{4}' def {5}(self): {6} = True while {6}: break """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7) return scriptPadding14 # -- snippet 15 -- # elif rand == 15: scriptPadding15 = textwrap.dedent(""" class {0}: def __init__(self): self.{1} = '{2}' self.{3} = '{4}' self.{5} = [] def {6}(self): self.{5} = True self.{3} = '{7}' """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8) return scriptPadding15 def AddRandomScripts(self, outputArg, mixerLengthArg, basicIndentArg, verboseArg): countScriptsAdded = 0 countLineAdded = 0 countLine = 0 checkLine = 0 countRecursFiles = 0 multipleLinesQuotes = 0 comma = 0 openSymb = 0 basicIndentArg = int(basicIndentArg) recursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False) if verboseArg: print("\n[*] Check basic indentation...\n") if verboseArg: print("[*] Basic indentation : {} spaces".format(basicIndentArg)) for number in recursFiles: countRecursFiles += 1 print("\n[+] Running add of random scripts in {} file(s)...\n".format( countRecursFiles)) # -- Count the number of lines that will be checked before filling -- # with Bar("Setting up ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: if not eachLine: continue countLine += 1 bar.next(1) bar.finish() # -- Add padding script -- # with Bar("Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: if eachLine == "\n": continue else: sys.stdout.write(eachLine) spaces = len(eachLine) - len(eachLine.lstrip()) if comma == 1: if re.match(Reg.detectComma, eachLine) == None: comma = 0 continue else: pass elif openSymb == 1: # -- If value of list, tuple and dict on multiple line isn't alone when detect \ # # the first line, feature need to check if comma is present -- # if re.match(Reg.detectComma, eachLine): comma = 1 openSymb = 0 continue elif multipleLinesQuotes == 1: if re.match( Reg.checkIfEndVarStdoutMultipleQuotes, eachLine): if self.utils.DetectMultipleLinesQuotes( eachLine) == True: multipleLinesQuotes = 0 pass else: continue else: continue elif re.match(Reg.checkIfVarMultipleQuotes, eachLine) \ or re.match(Reg.checkIfStdoutMultipleQuotes, eachLine): if self.utils.DetectMultipleLinesQuotes( eachLine) == False: pass else: multipleLinesQuotes = 1 continue elif re.match( Reg.detectComma, eachLine ): # Args of function on multiple lines for example comma = 1 # -- Avoid to add padding if the values of dict or list is alone on multiple lines and does \ # # not contain any blacklisted chars -- # elif re.match(Reg.detectOpenSymb, eachLine): openSymb = 1 else: pass if re.match(Reg.noAddScript, eachLine): continue # -- Adding scripts -- # elif re.match(Reg.addIndentScript, eachLine): spaces = spaces + basicIndentArg # add indentation sys.stdout.write( textwrap.indent( Padding.ScriptsGenerator( self, randomClassesFunctions=False, mixerLengthArg=mixerLengthArg), self.simpleSpace * spaces)) countScriptsAdded += 1 else: if spaces == 0: sys.stdout.write( textwrap.indent( Padding.ScriptsGenerator( self, randomClassesFunctions=True, mixerLengthArg=mixerLengthArg), self.simpleSpace * spaces)) countScriptsAdded += 1 else: sys.stdout.write( textwrap.indent( Padding.ScriptsGenerator( self, randomClassesFunctions=False, mixerLengthArg=mixerLengthArg), self.simpleSpace * spaces)) countScriptsAdded += 1 bar.next(1) bar.finish() # -- Check if padding has added in output script -- # with Bar("Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: if not eachLine: continue checkLine += 1 bar.next(1) bar.finish() countLineAdded = checkLine - countLine if checkLine > countLine: print("\n-> {} scripts added in {} file(s)\n".format( countScriptsAdded, countRecursFiles)) print("-> {} lines added in {} file(s)\n".format( countLineAdded, countRecursFiles)) return 1 else: return 0 def EmptyClasses(self, outputArg, mixerLengthArg, basicIndentArg, verboseArg): countRecursFiles = 0 counterToCheckIndent = 0 numberLine = 0 numberLineInFile = 0 emptyClassInfo = {} emptyClassInfoCheck = {} basicIndentArg = int(basicIndentArg) recursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False) for number in recursFiles: countRecursFiles += 1 with Bar("Correction ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLineInFile = 0 numberLine = 0 # -- Count all line(s) in file -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLineInFile += 1 # -- Find empty class(es) in dict -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if counterToCheckIndent == 1: spacesAfterClass = len(eachLine) - len( eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterClass == spacesClass: if search: emptyClassInfo[search.group(1)] = file numberLineInFile += 1 # Adding one line because padding will be added numberLine += 1 # Adding one line because padding will be added if re.match(Reg.checkClassInLine, eachLine): spacesClass = len(eachLine) - len( eachLine.lstrip()) if numberLine == numberLineInFile: # If empty class in last line search = re.search(Reg.detectClasses, eachLine) if search: emptyClassInfo[search.group(1)] = file else: counterToCheckIndent += 1 search = re.search(Reg.detectClasses, eachLine) # -- Add padding in empty class(es) -- # numberLine = 0 with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: numberLine += 1 if counterToCheckIndent == 1: spacesAfterClass = len(eachLine) - len( eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterClass == spacesClass: paddingVar1 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) paddingVar2 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) finalVarPadding = "{} = '{}'\n".format( paddingVar1, paddingVar2) spacesClass = spacesClass + basicIndentArg sys.stdout.write( textwrap.indent( finalVarPadding, self.simpleSpace * spacesClass)) numberLine += 1 sys.stdout.write(eachLine) if re.match(Reg.checkClassInLine, eachLine): spacesClass = len(eachLine) - len( eachLine.lstrip()) if numberLine == numberLineInFile: # If empty class in last line paddingVar1 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) paddingVar2 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) finalVarPadding = "{} = '{}'\n".format( paddingVar1, paddingVar2) spacesClass = spacesClass + basicIndentArg sys.stdout.write( textwrap.indent( finalVarPadding, self.simpleSpace * spacesClass)) else: counterToCheckIndent += 1 bar.next(1) bar.finish() # -- Check if class(es) is still empty -- # if emptyClassInfo != {}: with Bar("Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLineInFile = 0 numberLine = 0 with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if counterToCheckIndent == 1: spacesAfterClass = len(eachLine) - len( eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterClass == spacesClass: if search: emptyClassInfo[search.group(1)] = file numberLineInFile += 1 numberLine += 1 if re.match(Reg.checkClassInLine, eachLine): spacesClass = len(eachLine) - len( eachLine.lstrip()) if numberLine == numberLineInFile: # If empty class in last line search = re.search(Reg.detectClasses, eachLine) if search: emptyClassInfo[search.group(1)] = file else: counterToCheckIndent += 1 search = re.search(Reg.detectClasses, eachLine) bar.next(1) bar.finish() if emptyClassInfoCheck == {}: for key, value in emptyClassInfo.items(): print("\n-> File : {}".format(value)) print( "-> Padding added in : {} ( empty class )".format(key)) return 1 else: if verboseArg: print("\n[!] No padding added to empty class(es)... :\n") for key, value in emptyClassInfoCheck.items(): print("\n-> File : {}".format(value)) print("-> Class : {}".format(key)) return 0 else: print("[!] No empty class found in {}".format(outputArg)) return 1 def EmptyFunctions(self, outputArg, mixerLengthArg, basicIndentArg, verboseArg): countRecursFiles = 0 counterToCheckIndent = 0 numberLine = 0 numberLineInFile = 0 emptyFuncInfo = {} emptyFuncInfoCheck = {} basicIndentArg = int(basicIndentArg) recursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False) for number in recursFiles: countRecursFiles += 1 with Bar("Correction ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLineInFile = 0 numberLine = 0 # -- Count all line(s) in file -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLineInFile += 1 # -- Find empty function(s) in dict -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if counterToCheckIndent == 1: spacesAfterFunc = len(eachLine) - len( eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterFunc == spacesFunc: if search: emptyFuncInfo[search.group(1)] = file numberLineInFile += 1 # Adding one line because padding will be added numberLine += 1 # Adding one line because padding will be added if re.match(Reg.checkFunctionInLine, eachLine): spacesFunc = len(eachLine) - len(eachLine.lstrip()) if numberLine == numberLineInFile: # If empty function last line search = re.search(Reg.detectFunctions, eachLine) if search: emptyFuncInfo[search.group(1)] = file else: counterToCheckIndent += 1 search = re.search(Reg.detectFunctions, eachLine) # -- Add padding in empty function(s) -- # numberLine = 0 with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: numberLine += 1 if counterToCheckIndent == 1: spacesAfterFunc = len(eachLine) - len( eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterFunc == spacesFunc: paddingVar1 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) paddingVar2 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) finalVarPadding = "{} = '{}'\n".format( paddingVar1, paddingVar2) spacesFunc = spacesFunc + basicIndentArg sys.stdout.write( textwrap.indent( finalVarPadding, self.simpleSpace * spacesFunc)) numberLine += 1 sys.stdout.write(eachLine) if re.match(Reg.checkFunctionInLine, eachLine): spacesFunc = len(eachLine) - len(eachLine.lstrip()) if numberLine == numberLineInFile: # IIf empty function last line paddingVar1 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) paddingVar2 = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) finalVarPadding = "{} = '{}'\n".format( paddingVar1, paddingVar2) spacesFunc = spacesFunc + basicIndentArg sys.stdout.write( textwrap.indent( finalVarPadding, self.simpleSpace * spacesFunc)) else: counterToCheckIndent += 1 bar.next(1) bar.finish() # -- Check if function(s) is still empty -- # if emptyFuncInfo != {}: with Bar("Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLineInFile = 0 numberLine = 0 with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if counterToCheckIndent == 1: spacesAfterFunc = len(eachLine) - len( eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterFunc == spacesFunc: if search: emptyFuncInfoCheck[search.group( 1)] = file numberLineInFile += 1 numberLine += 1 if re.match(Reg.checkFunctionInLine, eachLine): spacesFunc = len(eachLine) - len( eachLine.lstrip()) if numberLine == numberLineInFile: # If empty function last line search = re.search(Reg.detectFunctions, eachLine) if search: emptyFuncInfoCheck[search.group( 1)] = file else: counterToCheckIndent += 1 search = re.search(Reg.detectFunctions, eachLine) bar.next(1) bar.finish() if emptyFuncInfoCheck == {}: for key, value in emptyFuncInfo.items(): print("\n-> File : {}".format(value)) print("-> Padding added in : {} ( empty function )".format( key)) return 1 else: if verboseArg: print("\n[!] No padding added to empty function(s)... :\n") for key, value in emptyFuncInfoCheck.items(): print("\n-> File : {}".format(value)) print("-> Function : {}".format(key)) return 0 else: print("[!] No empty function found in {}".format(outputArg)) return 1
class Delete: def __init__(self): self.utils = Utils() def LinesSpaces(self, outputArg, verboseArg): checkLinesSpace = {} checkEmptyLineOutput = 0 checkEmptyLineInput = 0 countRecursFiles = 0 numberLine = 0 recursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False ) for file in recursFiles: countRecursFiles += 1 # -- Delete all empty lines -- # with Bar("Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with fileinput.FileInput(file, inplace=True) as inputFile: for eachLine in inputFile: if re.match(Reg.detectLineEmpty, eachLine): checkEmptyLineInput += 1 pass else: sys.stdout.write(eachLine) bar.next(1) bar.finish() with Bar("Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLine = 0 with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if re.match(Reg.detectLineEmpty, eachLine): checkLinesSpace[numberLine] = file checkEmptyLineOutput += 1 bar.next(1) bar.finish() if checkEmptyLineOutput == 0: return 0 else: if verboseArg: print("\n[*] Empty line that deleted : {}\n".format(checkEmptyLineInput)) print("\n[!] Empty line that not been deleted... :\n") for key, value in checkLinesSpace.items(): print("\n-> File : {}".format(value)) print("-> Line : {}".format(key)) return 1 def Comments(self, outputArg, verboseArg): getIndexList = [] filesConcerned = [] countLineCommentOutput = 0 countLineCommentInput = 0 multipleLinesComments = 0 countRecursFiles = 0 noCommentsQuotes = 0 detectIntoSimpleQuotes = None recursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False ) for i in recursFiles: countRecursFiles += 1 # -- Delete comments and count comments will be deleted -- # print("\n[+] Running delete comments in {} file(s)...\n".format(countRecursFiles)) with Bar("Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: if re.match(Reg.pythonFileHeader, eachLine): sys.stdout.write(eachLine) else: if multipleLinesComments == 1: if re.match(Reg.quotesCommentsEndMultipleLines, eachLine): if self.utils.VerifyMultipleLinesComments(eachLine) == True: if multipleLinesComments == 1: countLineCommentInput += 1 multipleLinesComments = 0 else: countLineCommentInput += 1 else: countLineCommentInput += 1 elif noCommentsQuotes == 1: if re.match(Reg.checkIfEndVarStdoutMultipleQuotes, eachLine): sys.stdout.write(eachLine) noCommentsQuotes = 0 else: sys.stdout.write(eachLine) else: if re.match(Reg.quotesCommentsOneLine, eachLine): countLineCommentInput += 1 else: if re.match(Reg.quotesCommentsMultipleLines, eachLine): if self.utils.VerifyMultipleLinesComments(eachLine) == True: countLineCommentInput += 1 multipleLinesComments = 1 else: sys.stdout.write(eachLine) else: if re.match(Reg.checkIfStdoutMultipleQuotes, eachLine) \ or re.match(Reg.checkIfVarMultipleQuotes, eachLine): sys.stdout.write(eachLine) noCommentsQuotes = 1 elif re.match(Reg.checkIfRegexMultipleQuotes, eachLine): sys.write.stdout(eachLine) else: sys.stdout.write(eachLine) with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: if re.match(Reg.pythonFileHeader, eachLine): sys.stdout.write(eachLine) else: if re.match(Reg.hashCommentsBeginLine, eachLine): countLineCommentInput += 1 elif re.match(Reg.hashCommentsAfterLine, eachLine): eachLineList = list(eachLine) getIndexList = [] for i, v in enumerate(eachLineList): if v == "#": getIndexList.append(i) for i in getIndexList: if self.utils.DetectIntoSimpleQuotes(eachLine, maxIndexLine=i) == False: countLineCommentInput += 1 detectIntoSimpleQuotes = False break else: continue if detectIntoSimpleQuotes == False: eachLineList = eachLineList[:getIndexList[-1]] eachLineList.append("\n") eachLine = "".join(eachLineList) sys.stdout.write(eachLine) detectIntoSimpleQuotes = None continue else: sys.stdout.write(eachLine) else: sys.stdout.write(eachLine) bar.next(1) bar.finish() # -- Check if all comments are deleted -- # with Bar("Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: if re.match(Reg.pythonFileHeader, eachLine): continue else: if multipleLinesComments == 1: if re.match(Reg.quotesCommentsEndMultipleLines, eachLine): if self.utils.VerifyMultipleLinesComments(eachLine) == True: if multipleLinesComments == 1: countLineCommentOutput += 1 multipleLinesComments = 0 filesConcerned.append(file) else: countLineCommentOutput += 1 filesConcerned.append(file) else: countLineCommentOutput += 1 filesConcerned.append(file) elif noCommentsQuotes == 1: if re.match(Reg.checkIfEndVarStdoutMultipleQuotes, eachLine): noCommentsQuotes = 0 else: continue else: if re.match(Reg.quotesCommentsOneLine, eachLine): countLineCommentOutput += 1 filesConcerned.append(file) else: if re.match(Reg.quotesCommentsMultipleLines, eachLine): if self.utils.VerifyMultipleLinesComments(eachLine) == True: countLineCommentOutput += 1 multipleLinesComments = 1 filesConcerned.append(file) else: continue else: if re.match(Reg.checkIfStdoutMultipleQuotes, eachLine) \ or re.match(Reg.checkIfVarMultipleQuotes, eachLine): noCommentsQuotes = 1 elif re.match(Reg.checkIfRegexMultipleQuotes, eachLine): continue else: continue with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: if re.match(Reg.pythonFileHeader, eachLine): continue else: if re.match(Reg.hashCommentsBeginLine, eachLine): countLineCommentOutput += 1 filesConcerned.append(file) elif re.match(Reg.hashCommentsAfterLine, eachLine): eachLineList = list(eachLine) getIndexList = [] for i, v in enumerate(eachLineList): if v == "#": getIndexList.append(i) for i in getIndexList: if self.utils.DetectIntoSimpleQuotes(eachLine, maxIndexLine=i) == False: countLineCommentOutput += 1 detectIntoSimpleQuotes = False filesConcerned.append(file) break else: continue if detectIntoSimpleQuotes == False: eachLineList = eachLineList[:getIndexList[-1]] eachLineList.append("\n") eachLine = "".join(eachLineList) sys.stdout.write(eachLine) detectIntoSimpleQuotes = None continue else: continue else: continue bar.next(1) bar.finish() if countLineCommentOutput == 0: print("\n-> {} lines of comments deleted\n".format(countLineCommentInput)) return 0 else: if verboseArg: filesConcerned = self.utils.RemoveDuplicatesValuesInList(filesConcerned) print("\nFiles concerned of comments no deleted :\n") for f in filesConcerned: print("-> {}".format(f)) print("\n-> {} lines of comments no deleted\n".format(countLineCommentOutput)) return 1 def TrashFiles(self, outputArg, verboseArg): countRecursFiles = 0 deleteFiles = 0 checkPycFile = [] currentPosition = os.getcwd() recursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="pyc", blockDir="__pycache__", blockFile=False, dirOnly=False ) for number in recursFiles: countRecursFiles += 1 if countRecursFiles == 0: print("[!] No .pyc file(s) found in {}".format(outputArg)) return 0 print("\n[+] Running delete {} .pyc file(s)...\n".format(countRecursFiles)) # -- Check if .pyc file(s) exists and delete it -- # with Bar("Setting up ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: if re.match(Reg.detectPycFiles, file): deleteFiles += 1 checkPycFile.append(file) bar.next(1) bar.finish() # -- Delete pyc file(s) -- # with Bar("Correction ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: if re.match(Reg.detectPycFiles, file): extractPycFiles = re.search(r".*\.pyc$", file) moveFolder = re.sub(r".*\.pyc$", "", file) os.chdir(moveFolder) os.remove(extractPycFiles.group(0)) os.chdir(currentPosition) bar.next(1) bar.finish() checkRecursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="pyc", blockDir="__pycache__", blockFile=False, dirOnly=False ) if checkRecursFiles != []: if verboseArg: for pycFile in checkRecursFiles: print("-> .pyc file no deleted : {}".format(pycFile)) return 1 else: if verboseArg: for pycFile in checkPycFile: print("-> .pyc file deleted : {}".format(pycFile)) print("\n-> {} .pyc file(s) deleted".format(deleteFiles)) return 0
class Replace: def __init__(self): self.mixer = Mixer() self.utils = Utils() self.pythonExcludeDefaultString = "exclude/string_to_string_mixed/exclude_word_do_not_modify.txt" self.pythonExcludeUserString = "exclude/string_to_string_mixed/exclude_word_by_user.txt" self.pythonExcludeUserFileName = "exclude/file_name/exclude_file_name_by_user.txt" def EachLine(self, line, dictionary, fileNameImport, listModuleImport): getIndexLineList = [] returnLine = [] charValue = [] checkCharAfterWord = 1 wordSetter = 0 checkGetKey = "" checkGetWord = "" getLine = "" breakLine = "" if listModuleImport == True: detectSpecialChars = Reg.detectSpecialChars else: detectSpecialChars = Reg.detectSpecialCharsWihtoutQuotes # -- Get list of all letters in line -- # for indexLine, letterLine in enumerate(line): getIndexLineList.append(letterLine) # -- Loop in each letter of line -- # for indexLine, letterLine in enumerate(line): # -- Add in final line list all chars mixed -- # if charValue != []: for obfIndex, obfValue in enumerate(charValue): if obfIndex == 0: # First letter in string mixed are already added in the final line continue returnLine.append(obfValue) charValue = [] # -- If the variable is only a letter, check if the next character is specific so as not to replace it -- # if re.match(detectSpecialChars, letterLine): returnLine.append(letterLine) # -- Count indexes of word to move after it --# countDeleteIndex = 0 for i in getWord: countDeleteIndex += 1 wordSetter = countDeleteIndex - 2 # -2 Is to letter already append and the letter in progress else: # -- The index numbers of variable is decremented to add the mixed letters that be replaced -- # if wordSetter > 0: wordSetter -= 1 continue else: try: # -- Loop in the dictionary with already mixed values-- # for key, value in dictionary: for indexKey, letterKey in enumerate(key): for letterValue in value: # -- Check if letter of word is equal to letter of key -- # if letterKey == letterLine: # -- Begin process to check -- # if indexKey == 0: # if equal to the first letter of word on <key> variable indexExplore = indexLine + len( key ) # Place index position after the word # -- If indexError return to next loop -- # try: getIndexLineList[indexExplore] except IndexError: continue # -- Check the char after and before the word -- # if re.match( detectSpecialChars, getIndexLineList[ indexExplore]): # Index check if word found is not into the other word indexExploreBefore = indexLine - 1 # Index check if char after the end of string is found with 'detectSpecialChars' regex indexExploreAfter = indexExplore try: if not re.match( r"\w|\\|\%", getIndexLineList[ indexExploreBefore] ): # -- Check if it's 'from' and 'import' file in line to avoid replace \ # name of file if variable is identical name to file -- # getLine = "".join( getIndexLineList) if fileNameImport == False: if "import" in getLine: if "from" in getLine: # -- Cut the line from the current index and check if it is \ # not there is the keyword "import" in the line -- # breakLine = getIndexLineList[: indexLine] breakLine = "".join( breakLine ) if not "import" in breakLine: # -- It's a file because only 'from'key word -- # checkCharAfterWord = 1 else: checkCharAfterWord = 0 else: checkCharAfterWord = 1 # -- Check if after char of the word found by 'detectSpecialChars' is \ # not ' or " -- # elif re.match( r"\"|\'", getIndexLineList[ indexExploreAfter] ): if re.match( r"\[|\(|\{", getIndexLineList[ indexExploreAfter - 1] ): checkCharAfterWord = 0 else: checkCharAfterWord = 1 else: checkCharAfterWord = 0 # -- Only for [-rfn, --replacefilsname] feature -- # else: # -- check if file name is imported - # breakLine = getIndexLineList[: indexLine] breakLine = "".join( breakLine) # -- If file name is imported after 'import', the file name is not \ # replaced -- # if "import" in breakLine: checkCharAfterWord = 1 else: checkCharAfterWord = 0 else: checkCharAfterWord = 1 except IndexError: checkCharAfterWord = 0 pass else: checkCharAfterWord = 1 if checkCharAfterWord == 0: # -- Initialize vars -- # getCharAllInKey = [] getWord = [] indexExploreStart = indexLine # -- Delete -1, first letter is already increment -- # indexExploreEnd = indexLine + len( key) - 1 # -- List contain all letters of key -- # for getLetterKey in key: getCharAllInKey.append( getLetterKey) # -- Check if all letters of key is equal to all letters of word -- # for indexCheckLetter, checkLetter in enumerate( getIndexLineList): if indexCheckLetter >= indexExploreStart and \ indexCheckLetter <= indexExploreEnd: getWord.append( checkLetter) # -- Check if number of chars in key equal number of chars in word -- # if list( set(getCharAllInKey) - set(getWord)) == []: checkGetWord = "".join( getWord) checkGetKey = "".join( getCharAllInKey) # -- Check if key == word -- # if checkGetWord == checkGetKey: # -- Check if word is not in strings quotes and if a variable # is in format() in end of multiple line -- # if self.utils.DetectIntoSimpleQuotes(getLine, indexLine) == False \ or self.utils.DetectMultipleLinesQuotes(getLine) == True: for obfChar in value: charValue.append( obfChar) letterLine = letterValue raise BreakLoop else: break else: break else: break else: break else: break else: break raise BreakLoop except BreakLoop: returnLine.append(letterLine) # -- Rewrite the line -- # returnLine = "".join(returnLine) return returnLine[:] def StringToString(self, outputArg, mixerLengthArg, verboseArg): variablesDict = {} classesDict = {} functionsDict = {} allDict = {} classFuncDict = {} checkWordsMixed = [] wordsExcludedUser = [] wordsExcludedUserFound = [] wordsExcludedDefault = [] wordsExcludedDefaultFound = [] checkAllWords = [] checkWordsError = [] checkKeyWordsMixed = [] checkCountWordsMixed = 0 checkCountWordsValue = 0 countRecursFiles = 0 multipleLinesQuotes = 0 recursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False) for file in recursFiles: countRecursFiles += 1 print("\n[+] Running replacement of variables/classes/functions in " + str(countRecursFiles) + \ " file(s), he can be long... you have time to make a coffee :)\n") # -- Replace variables/classes/functions to random strings with length defined -- # with Bar("Setting up ", fill="=", max=100, suffix="%(percent)d%%") as bar: for file in recursFiles: with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: # -- Variables -- # search = re.search(Reg.detectSimpleVars, eachLine) if search: if "," in search.group(1): modifySearch = search.group(1).replace( ",", " ") modifySearch = modifySearch.split() for i in modifySearch: if i not in variablesDict: mixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg ) i = i.strip() variablesDict[i] = mixer else: if search.group(1) not in variablesDict: mixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) modifySearch = search.group(1).strip() variablesDict[modifySearch] = mixer # -- Error variables -- # search = re.search(Reg.detectErrorVars, eachLine) if search: mixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) if search.group(2) not in variablesDict: variablesDict[search.group(2)] = mixer # -- Loop variables -- # search = re.search(Reg.detectLoopVars, eachLine) if search: if "," in search.group(1): modifySearch = search.group(1).replace( ",", " ") modifySearch = modifySearch.split() for i in modifySearch: if i not in variablesDict: mixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg ) variablesDict[i] = mixer else: if search.group(1) not in variablesDict: mixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) variablesDict[search.group(1)] = mixer # -- Function(s) -- # search = re.search(Reg.detectFunctions, eachLine) if search: mixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) if search.group(1) not in functionsDict: if not "__init__" in search.group(1): functionsDict[search.group(1)] = mixer # -- Class(es) -- # search = re.search(Reg.detectClasses, eachLine) if search: mixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) if search.group(1) not in classesDict: classesDict[search.group(1)] = mixer bar.next(40) # -- Delete excluded variables/classes/functions defined from \ # 'exclude/string_to_string_mixed/exclude_word_do_not_modify.txt' -- # if os.path.exists(self.pythonExcludeDefaultString) == True: with open(self.pythonExcludeDefaultString, "r") as readFile: for word in readFile: if "#" in word or word == "\n": continue else: word = word.rstrip() wordsExcludedDefault.append(word) else: print(Colors.ERROR + "[-] '{}' file not found".format(self.pythonExcludeDefaultString) + Colors.DISABLE \ + "\n") bar.next(10) # -- Delete excluded variables/classes/functions defined from \ # 'exclude/string_to_string_mixed/exclude_word_by_user.txt' -- # if os.path.exists(self.pythonExcludeUserString) == True: with open(self.pythonExcludeUserString, "r") as readFile: for word in readFile: if "#" in word or word == "\n": continue else: word = word.rstrip() wordsExcludedUser.append(word) else: print(Colors.ERROR + "[-] '{}' file not found".format(self.pythonExcludeUserString) + Colors.DISABLE \ + "\n") bar.next(10) for word in wordsExcludedUser: if word in variablesDict.keys(): wordsExcludedUserFound.append(word) if word in classesDict.keys(): wordsExcludedUserFound.append(word) if word in functionsDict.keys(): wordsExcludedUserFound.append(word) bar.next(20) for word in wordsExcludedDefault: if word in variablesDict.keys(): wordsExcludedDefaultFound.append(word) if word in classesDict.keys(): wordsExcludedDefaultFound.append(word) if word in functionsDict.keys(): wordsExcludedDefaultFound.append(word) for word in wordsExcludedUserFound: if word in variablesDict.keys(): del variablesDict[word] if word in classesDict.keys(): del classesDict[word] if word in functionsDict.keys(): del functionsDict[word] for word in wordsExcludedDefaultFound: if word in variablesDict.keys(): del variablesDict[word] if word in classesDict.keys(): del classesDict[word] if word in functionsDict.keys(): del functionsDict[word] bar.next(20) bar.finish() # -- Display variables/classes/functions found -- # if verboseArg: print("\n[+] Variable(s) found :\n") if variablesDict == {}: print("-> No result") else: for key, value in variablesDict.items(): print("-> {} : {}".format(key, value)) print("\n[+] Class(es) found :\n") if classesDict == {}: print("-> No result") else: for key, value in classesDict.items(): print("-> {} : {}".format(key, value)) print("\n[+] Function(s) found :\n") if functionsDict == {}: print("-> No result") else: for key, value in functionsDict.items(): print("-> {} : {}".format(key, value)) print( "\n[+] String excluded found in '{}' that have been matched from '{}' :\n" .format(self.pythonExcludeUserString, outputArg)) if wordsExcludedUserFound == []: print("-> No result") else: for word in wordsExcludedUserFound: print("-> {} : excluded by user".format(word)) print( "\n[+] String excluded found in '{}' that have been matched from '{}' :\n" .format(self.pythonExcludeDefaultString, outputArg)) if wordsExcludedDefaultFound == []: print("-> No result") else: for word in wordsExcludedDefaultFound: print("-> {} : excluded by default".format(word)) print("") # -- Merge all dicts -- # allDict = self.utils.DictMerge(dict1=allDict, dict2=variablesDict) allDict = self.utils.DictMerge(dict1=allDict, dict2=functionsDict) allDict = self.utils.DictMerge(dict1=allDict, dict2=classesDict) classFuncDict = self.utils.DictMerge(dict1=classFuncDict, dict2=classesDict) classFuncDict = self.utils.DictMerge(dict1=classFuncDict, dict2=functionsDict) # -- Change variables/classes/functions to mixed values -- # with Bar("Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: # -- Replace variable(s) only -- # with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: if not eachLine: continue else: if multipleLinesQuotes == 1: if re.match( Reg.checkIfEndVarStdoutMultipleQuotes, eachLine): if self.utils.DetectMultipleLinesQuotes( eachLine) == True: eachLine = Replace.EachLine( self, line=eachLine, dictionary=allDict.items(), fileNameImport=False, listModuleImport=False) sys.stdout.write(eachLine) multipleLinesQuotes = 0 else: sys.stdout.write(eachLine) else: sys.stdout.write(eachLine) elif re.match(Reg.checkIfVarMultipleQuotes, eachLine) \ or re.match(Reg.checkIfStdoutMultipleQuotes, eachLine): if self.utils.DetectMultipleLinesQuotes( eachLine) == False: pass else: multipleLinesQuotes = 1 eachLine = Replace.EachLine( self, line=eachLine, dictionary=allDict.items(), fileNameImport=False, listModuleImport=False) sys.stdout.write(eachLine) continue else: eachLine = Replace.EachLine( self, line=eachLine, dictionary=allDict.items(), fileNameImport=False, listModuleImport=False) sys.stdout.write(eachLine) bar.next(1) bar.finish() with Bar("Check ", fill="=", max=100, suffix="%(percent)d%%") as bar: for file in recursFiles: # -- Check if variables/classes/functions have been mixed -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: for key, value in allDict.items(): if value in eachLine: if re.match(r"\w+" + re.escape(value) + r"\w+", eachLine): pass else: checkWordsMixed.append(value) checkKeyWordsMixed.append(key) bar.next(70) # -- Delete duplicated words -- # checkListWordsMixed = list(dict.fromkeys(checkWordsMixed)) checkKeyWordsMixed = list(dict.fromkeys(checkKeyWordsMixed)) bar.next(15) for i in checkListWordsMixed: checkCountWordsMixed += 1 for i in allDict.values(): checkCountWordsValue += 1 bar.next(15) bar.finish() if checkCountWordsMixed == checkCountWordsValue: print( "\n-> {} variable(s)/class(es)/function(s) replaced in {} file(s)\n" .format(checkCountWordsValue, countRecursFiles)) return 1 else: if verboseArg: for key in allDict.keys(): checkAllWords.append(key) checkWordsError = list( set(checkAllWords) - set(checkKeyWordsMixed)) print("\n[!] Word(s) that not been replaced, check if an error will appear when will launch your " + \ "obfuscated code... :\n") if checkWordsError != []: for wordNoReplaced in checkWordsError: print("-> Word : {}".format(wordNoReplaced)) return 0 def StringsToHex(self, outputArg, mixerLengthArg, verboseArg): checkHexError = {} splitLetter = [] getLetterLineList = [] countRecursFiles = 0 checkPrint = 0 numberLine = 0 checkError = False hexLine = "" recursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False) for number in recursFiles: countRecursFiles += 1 print( "\n[+] Running replace all strings to their hexadecimal value in {} file(s)...\n" .format(countRecursFiles)) # -- Replace all strings to their hexadecimal value -- # with Bar("Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: # -- Add a new first random line and move the old first line to the second line to avoid replacing it -- # checkPrint = 0 # initialize check print() func at the begining of each file with open(file, "r") as inputFile: stringRandomMixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) firstLine = "{}\n".format(stringRandomMixer) line = inputFile.readlines() line.insert(0, firstLine) with open(file, "w") as inputFile: inputFile.writelines(line) # -- Replace all lines-- # with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: if checkPrint == 0: varMixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) sys.stdout.write(varMixer + "=\"\"\"") checkPrint = 1 else: getLetterLineList = [] for letterLine in eachLine: if len(letterLine.encode("utf-8").hex()) > 2: splitLetter = [] for i in letterLine.encode("utf-8").hex(): splitLetter.append(i) if len(splitLetter) == 2: letterLine = "".join(splitLetter) letterToHex = "\\x" + str( letterLine) getLetterLineList.append( letterToHex) splitLetter = [] else: continue else: letterToHex = "\\x" + str( letterLine.encode("utf-8").hex()) getLetterLineList.append( letterToHex ) # Get list of all letters in line hexLine = "".join(getLetterLineList) sys.stdout.write(hexLine) # -- Add exec funtions to interpret hex code in strings -- # with open(file, "a") as inputFile: inputFile.write("\"\"\"") inputFile.write("\nexec({})".format(varMixer)) bar.next(1) bar.finish() with Bar("Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLine = 0 # -- Check if all lines are replaced of hexadecimal value -- # with open(file, "r") as inputFile: for eachLine in inputFile: numberLine += 1 if not eachLine: continue else: if not "\\x" in eachLine: if re.match(Reg.detectExecFunction, eachLine): continue else: checkHexError[numberLine] = file checkError = True else: continue bar.next(1) bar.finish() if checkError == False: return 1 else: if verboseArg: print( "\n[!] Line(s) that have not been replaced by their hexadecimal values... :\n" ) for key, value in checkHexError.items(): print("\n-> File : {}".format(value)) print("-> Line : {}".format(key)) else: print("\n" + Colors.ERROR + "[!] Launch Intensio-Obfuscator with verbose mode because line(s) have not been " + \ "replaced by their hexadecimal values" + Colors.DISABLE + "\n") return 0 def FilesName(self, outputArg, mixerLengthArg, verboseArg): checkFilesFoundCompare = {} filesNameDict = {} filesNameDictNoExt = {} filesNameFound = [] filesNameFoundNoExt = [] filesNameMixed = [] fileNameExcluded = [] fileNameExcludedFound = [] fileNameExcludedByUser = [] importNoCompliantFound = [] badNameDir = [] numberLine = 0 countRecursFiles = 0 checkCountRecursFiles = 0 currentPosition = os.getcwd() recursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile="__init__", dirOnly=False) recursFilesWithInit = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False) recursDirs = self.utils.CheckFileDir(output=outputArg, detectFiles="", blockDir="__pycache__", blockFile=False, dirOnly=True) for file in recursFiles: countRecursFiles += 1 print("\n[+] Running replace files name in {} file(s)...\n".format( countRecursFiles)) with Bar("Setting up ", fill="=", max=100, suffix="%(percent)d%%") as bar: for file in recursFiles: parseFilePath = file.split( self.utils.Platform(getOS=False, getPathType=True)) mixer = self.mixer.GetStringMixer( mixerLengthArgDefined=mixerLengthArg) filesNameDict[parseFilePath[-1]] = mixer + ".py" filesNameDictNoExt[parseFilePath[-1].replace(".py", "")] = mixer deleteExt = parseFilePath[-1].replace(".py", "") filesNameFound.append(parseFilePath[-1]) filesNameFoundNoExt.append(deleteExt) filesNameMixed.append(mixer + ".py") bar.next(30) # -- Check if directory have the same name for directory in recursDirs: for fileName in filesNameFoundNoExt: if re.match(r".*" + re.escape(self.utils.Platform(getOS=False, getPathType=True)) + r"{1}" + \ re.escape(fileName) + re.escape(self.utils.Platform(getOS=False, getPathType=True)) + \ r"{1}.*", directory): fileNameExcluded.append(fileName) badNameDir.append(fileName) bar.next(20) # -- Delete if file name excluded by user from exclude/file_name/exclude_file_name_by_user.txt' -- # if os.path.exists(self.pythonExcludeUserFileName) == True: with open(self.pythonExcludeUserFileName) as readFile: for fileName in readFile: if "#" in fileName or fileName == "\n" or re.match( r"yourFileName[0-9]{1}", fileName): continue else: fileName = fileName.rstrip() fileNameExcluded.append(fileName) fileNameExcludedByUser.append(fileName) else: print(Colors.ERROR + "[-] '{}' file not found".format( self.pythonExcludeUserFileName) + Colors.DISABLE + "\n") bar.next(10) # -- Delete file name excluded in dictionnary -- # for word in fileNameExcluded: for fileNameNoExt in filesNameFoundNoExt: if fileNameNoExt == word: fileNameExcludedFound.append(word) filesNameFoundNoExt.remove(word) for fileName in filesNameFound: if fileName == word: filesNameFound.remove(word) if word in filesNameDictNoExt.keys(): del filesNameDictNoExt[word] word = word + ".py" if word in filesNameDict.keys(): del filesNameDict[word] bar.next(10) # -- Check if file name in code is after 'import' native python function --# for file in recursFiles: with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: if re.match(Reg.detectPythonImport, eachLine): searchFileName = re.search(r"(import\s+)(.*)", eachLine) if searchFileName.group(2): searchFileName = searchFileName.group( 2).replace(",", "") searchFileName = searchFileName.split() for i in searchFileName: i = i.strip() for fileNameNoExt in filesNameFoundNoExt: if fileNameNoExt == i: importNoCompliantFound.append(i) filesNameFoundNoExt.remove(i) for fileName in filesNameFound: fileName = fileName.replace(".py", "") if fileName == i: i = i + ".py" filesNameFound.remove(i) if i in filesNameDictNoExt.keys(): del filesNameDictNoExt[i] if i in filesNameDict.keys(): del filesNameDict[i] bar.next(30) bar.finish() # -- Diplay all file name(s) found with their mixed values if verbose arg is actived -- # if verboseArg: print("\n[+] File name(s) found with their mixed values :\n") if filesNameDict == {}: print("-> No result") else: for key, value in filesNameDict.items(): print("-> {} : {}".format(key, value)) print( "\n[+] File name(s) excluded found in '{}' that have been matched from '{}' :\n" .format(self.pythonExcludeUserFileName, outputArg)) if fileNameExcludedByUser == []: print("-> No result") else: for i in fileNameExcludedByUser: print("-> {} : excluded by user".format(i)) print( "\n[+] File name(s) no compliant for 'replace file name' feature :\n" ) if importNoCompliantFound == []: print("-> No result") else: for i in importNoCompliantFound: print( "-> {} : no compliant ( file name excluded automatically )" .format(i)) print("\n[+] Directory that have same name of python file(s) :\n") if badNameDir == []: print("-> No result") else: for i in badNameDir: print( "-> {} : no compliant ( file name excluded automatically )" .format(i)) i = i.rstrip() print("") # -- Replace all file names to random strings with length and obfuscation level defined -- # with Bar("Obfuscation ", fill="=", max=100, suffix="%(percent)d%%") as bar: for fileInCode in recursFilesWithInit: # -- Rename all files in python code -- # with fileinput.input(fileInCode, inplace=True) as inputFile: for eachLine in inputFile: if re.match(Reg.detectPythonImport, eachLine): eachLine = Replace.EachLine( self, line=eachLine, dictionary=filesNameDictNoExt.items(), fileNameImport=True, listModuleImport=False) sys.stdout.write(eachLine) continue else: sys.stdout.write(eachLine) bar.next(50) for file in recursFiles: parseFilePath = file.split( self.utils.Platform(getOS=False, getPathType=True)) # -- Rename all files in their directories -- # for key, value in filesNameDict.items(): if key == parseFilePath[-1]: parseFilePath.remove(parseFilePath[-1]) parseFilePathToMove = self.utils.Platform( getOS=False, getPathType=True).join(parseFilePath) os.chdir(parseFilePathToMove ) # Move in directory to rename python file os.rename(key, value) else: continue os.chdir(currentPosition) bar.next(50) bar.finish() checkRecursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile="__init__", dirOnly=False) for file in checkRecursFiles: checkCountRecursFiles += 1 # -- Check if all files name are been replaced to random strings -- # with Bar("Check ", fill="=", max=checkCountRecursFiles, suffix="%(percent)d%%") as bar: for file in checkRecursFiles: numberLine = 0 # -- Check for file name in directory -- # for key, value in filesNameDict.items(): if key in file: checkFilesFoundCompare[key] = value bar.next(1) bar.finish() if checkFilesFoundCompare != {}: if verboseArg: if checkFilesFoundCompare != {}: print( "\n[!] File name that have not been replaced by their random string value... :\n" ) for key, value in checkFilesFoundCompare.items(): print("\n-> File : {}".format(key)) print("-> Value mixed : {}".format(value)) else: print("\n" + Colors.ERROR + "[-] Launch intensio-obfuscatior with verbose mode [-v, --verbose] because " + \ "file name(s) have not been replaced by their random string value" + Colors.DISABLE) return 0 else: return 1
class Padding: def __init__(self): self.mixer = Mixer() self.utils = Utils() self.simpleSpace = " " def CheckBasicIndentation(self, recursPythonFiles): # -- Check basic indentation (2 - 4 - 8) of python files -- # allNumIndentations = [] headerPythonFile = r"\s*#!.*[python|1-9]|\s*#.*-*-$" for file in recursPythonFiles: with open(file, "r") as inputFile: for eachLine in inputFile: if re.match(headerPythonFile, eachLine): continue else: spaces = len(eachLine) - len(eachLine.lstrip()) if spaces == 0 or spaces == 1: continue else: allNumIndentations.append(spaces) if allNumIndentations == []: return 0 else: allNumIndentations.sort() return allNumIndentations[0] def ScriptsGenerator(self, randomClassesFunctions, mixerLengthArg): varRandom1 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom2 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom3 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom4 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom5 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom6 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom7 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom8 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom9 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom10 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom11 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom12 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom13 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) varRandom14 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) # ---------- Python random snippets code ---------- # if randomClassesFunctions == True: rand = random.randint(1, 15) else: rand = random.randint(1, 9) # -- snippet 1 -- # if rand == 1: scriptPadding1 = textwrap.dedent(""" {0} = '{5}' {1} = '{6}' {2} = '{7}' {3} = '{8}' {4} = '{9}' if {0} in {1}: {0} = {4} if {1} in {2}: {1} = {3} elif {1} in {0}: {2} = {1} if {2} in {1}: {1} = {4} """).format(varRandom1, varRandom2, varRandom3, varRandom4, varRandom5, \ varRandom6, varRandom7, varRandom8, varRandom9, varRandom10) return scriptPadding1 # -- snippet 2 -- # elif rand == 2: scriptPadding2 = textwrap.dedent(""" {0} = '{2}' {1} = '{3}' if {0} != {1}: {0} = '{3}' {1} = {0} {0} = '{2}' """).format(varRandom1, varRandom2, varRandom3, varRandom4) return scriptPadding2 # -- snippet 3 -- # elif rand == 3: scriptPadding3 = textwrap.dedent(""" {0} = '{6}' {1} = '{7}' {2} = '{8}' {3} = '{9}' {4} = '{10}' {5} = '{11}' if {0} != {3}: {1} = {2} for {5} in {3}: if {5} != {2}: {1} = {1} else: {4} = {0} else: {2} = {0} {0} = {4} if {2} == {0}: for {5} in {0}: if {5} == {2}: {2} = {0} else: {2} = {4} """).format(varRandom1, varRandom2, varRandom3, varRandom4, varRandom5, \ varRandom6, varRandom7, varRandom8, varRandom9, varRandom10, \ varRandom11, varRandom12) return scriptPadding3 # -- snippet 4 -- # elif rand == 4: scriptPadding4 = textwrap.dedent(""" {0} = '{3}' {1} = '{4}' {2} = '{5}' if {0} == {1}: {2} = '{5}' {2} = {0} else: {2} = '{5}' {2} = '{3}' """).format(varRandom1, varRandom2, varRandom3, varRandom4, \ varRandom5, varRandom6,) return scriptPadding4 # -- snippet 5 -- # elif rand == 5: scriptPadding5 = textwrap.dedent(""" {0} = '{6}' {1} = '{7}' {2} = '{8}' {3} = '{9}' {4} = '{10}' {5} = '{11}' if {2} == {3}: for {5} in {4}: if {5} == {3}: {4} = {0} else: {3} = {1} """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8, varRandom9, \ varRandom10, varRandom11, varRandom12) return scriptPadding5 # -- snippet 6 -- # elif rand == 6: scriptPadding6 = textwrap.dedent(""" {0} = '{4}' {1} = '{5}' {2} = '{6}' {3} = '{7}' if {1} == {0}: for {0} in {1}: if {1} == {1}: {2} = '{3}' elif {2} == {3}: {3} = {0} else: {0} = {1} elif {2} == {2}: for {2} in {1}: if {3} == {1}: {2} = '{3}' elif {2} == {3}: {3} = {0} else: {0} = {1} for {2} in {1}: if {3} == {1}: {2} = '{3}' elif {2} == {3}: {3} = {0} else: {0} = {3} else: {0} = {1} """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8) return scriptPadding6 # -- snippet 7 -- # elif rand == 7: scriptPadding7 = textwrap.dedent(""" {0} = '{7}' {1} = '{8}' {2} = '{9}' {3} = '{10}' {4} = '{11}' {5} = '{12}' {6} = [ '{7}', '{9}', '{11}', '{13}' ] for {0} in {5}: for {1} in {2}: if {3} == {4}: {1} = {0} elif {4} == {1}: {1} = {5} else: {4} = {5} for {1} in {6}: {2} = {1} """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8, varRandom9, \ varRandom10, varRandom11, varRandom12, \ varRandom13, varRandom14) return scriptPadding7 # -- snippet 8 -- # elif rand == 8: scriptPadding8 = textwrap.dedent(""" {0} = '{1}' """).format(varRandom1, varRandom2) return scriptPadding8 # -- snippet 9 -- # elif rand == 9: scriptPadding9 = textwrap.dedent(""" {0} = '{1}' {2} = '{3} {1}' """).format(varRandom1, varRandom2, varRandom3, \ varRandom4,) return scriptPadding9 # ---------- Python random functions ---------- # # -- snippet 10 -- # elif rand == 10: scriptPadding10 = textwrap.dedent(""" def {0}(): {1} = '{2}' {3} = True """).format(varRandom1, varRandom2, varRandom3, \ varRandom4,) return scriptPadding10 # -- snippet 11 -- # elif rand == 11: scriptPadding11 = textwrap.dedent(""" def {0}({1}): {1} = '{2}' {3} = '{4}' if {1} == {3}: for {5} in {3}: if {5} == {3}: continue else: {6} = False """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7) return scriptPadding11 # -- snippet 12 -- # elif rand == 12: scriptPadding12 = textwrap.dedent(""" def {0}({1}, {2}): {1} = '{3}' {2} = '{4}' try: for {5} in {1}: if {5} == {2}: {6} = [] {7} = None else: pass except: pass """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8) return scriptPadding12 # -- snippet 13 -- # elif rand == 13: scriptPadding13 = textwrap.dedent(""" def {0}({1}, {2}, {3}): {1} = '{4}' {2} = '{5}' {6} = True while {6}: break for {6} in {2}: {3} = False """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7) return scriptPadding13 # ---------- Python random classes ---------- # # -- snippet 14 -- # elif rand == 14: scriptPadding14 = textwrap.dedent(""" class {0}: def __init__(self): self.{1} = '{2}' self.{3} = '{4}' def {5}(self): {6} = True while {6}: break """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7) return scriptPadding14 # -- snippet 15 -- # elif rand == 15: scriptPadding15 = textwrap.dedent(""" class {0}: def __init__(self): self.{1} = '{2}' self.{3} = '{4}' self.{5} = [] def {6}(self): self.{5} = True self.{3} = '{7}' """).format(varRandom1, varRandom2, varRandom3, \ varRandom4, varRandom5, varRandom6, \ varRandom7, varRandom8) return scriptPadding15 def AddRandomScripts(self, outputArg, mixerLengthArg, verboseArg): countScriptsAdded = 0 countLineAdded = 0 countLine = 0 checkLine = 0 checkQuotePassing = 0 checkCharPassing = 0 checkOtherCharPassing = 0 countRecursFiles = 0 basicIndentation = None addIndentScript = r".*\:{1}\s*$" quotesIntoVariable = r".*={1}\s*[\"|\']{3}" quotesEndMultipleLines = r"^\s*[\"|\']{3}\)?\.?" quotesInRegex = r"={1}\s*r{1}[\"|\']{3}" noAddScript = r"^\@|\s+\@|\s+return|\s*def\s+.+\s*\:{1}|^class\s+.+\s*\:{1}|.*[\{|\[|\(|\)|\]|\}|,|\\|^]\s*$|\s+yield.*|\s+raise.*" recursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False ) if verboseArg: print("\n[*] Check basic indentation...\n") basicIndentation = Padding.CheckBasicIndentation(self, recursFiles) if verboseArg: print("[*] Basic indentation : {} spaces".format(basicIndentation)) for number in recursFiles: countRecursFiles += 1 print("\n[+] Running add of random scripts in {} file(s)...\n".format(countRecursFiles)) # -- Count the number of lines that will be checked before filling -- # with Bar("Setting up ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with open(file , "r") as readFile: readF = readFile.readlines() for eachLine in readF: if not eachLine: continue countLine += 1 bar.next(1) bar.finish() # -- Padding script added -- # with Bar("Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: sys.stdout.write(eachLine) if eachLine == "\n": continue else: spaces = len(eachLine) - len(eachLine.lstrip()) # -- Detect code into 3 quotes excepted comments -- # if re.match(quotesIntoVariable, eachLine): if re.match(quotesInRegex, eachLine): pass else: checkQuotePassing += 1 continue elif re.match(quotesEndMultipleLines, eachLine): if re.match(quotesInRegex, eachLine): pass else: checkQuotePassing += 1 if checkQuotePassing == 2: checkQuotePassing = 0 continue if checkQuotePassing == 1: continue elif checkQuotePassing == 2: checkQuotePassing = 0 pass else: pass # -- Check dict, list, tuple in multiple lines -- # if checkCharPassing == 1: if re.match(r".*[\"|\'|\)|\]|\}|\w]\s*$", eachLine): checkCharPassing = 0 continue else: continue elif checkOtherCharPassing >= 1: if re.match(r".*[\"|\'|\)|\]|\}|\w]\s*$", eachLine): checkOtherCharPassing -= 1 continue else: if re.match(r".*[\(|\{|\[]\s*$", eachLine): checkOtherCharPassing += 1 continue else: pass if re.match(noAddScript, eachLine): if re.match(r".*[\\|,]\s*$", eachLine): if checkCharPassing == 1: continue else: checkCharPassing = 1 continue elif re.match(r".*[\(|\{|\[]\s*$", eachLine): checkOtherCharPassing += 1 continue else: continue # -- Adding scripts -- # elif re.match(addIndentScript, eachLine): spaces = spaces + basicIndentation # add indentation sys.stdout.write(textwrap.indent(Padding.ScriptsGenerator( self, randomClassesFunctions=False, mixerLengthArg=mixerLengthArg), self.simpleSpace * spaces) ) countScriptsAdded += 1 else: if spaces == 0: sys.stdout.write(textwrap.indent(Padding.ScriptsGenerator( self, randomClassesFunctions=True, mixerLengthArg=mixerLengthArg), self.simpleSpace * spaces) ) countScriptsAdded += 1 else: sys.stdout.write(textwrap.indent(Padding.ScriptsGenerator( self, randomClassesFunctions=False, mixerLengthArg=mixerLengthArg), self.simpleSpace * spaces) ) countScriptsAdded += 1 bar.next(1) bar.finish() # -- Check if padding has added in output script -- # with Bar("Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with open(file , "r") as readFile: readF = readFile.readlines() for eachLine in readF: if not eachLine: continue checkLine += 1 bar.next(1) bar.finish() countLineAdded = checkLine - countLine if checkLine > countLine: print("\n-> {} scripts added in {} file(s)\n".format(countScriptsAdded, countRecursFiles)) print("-> {} lines added in {} file(s)\n".format(countLineAdded, countRecursFiles)) return 0 else: return 1 def EmptyClasses(self, outputArg, mixerLengthArg, verboseArg): countRecursFiles = 0 counterToCheckIndent = 0 numberLine = 0 numberLineInFile = 0 emptyClassInfo = {} emptyClassInfoCheck = {} basicIndentation = None classDefined = r"\s*class\s+(\w+)" detectClass = r"\s*class\s+\w+" recursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False ) basicIndentation = Padding.CheckBasicIndentation(self, recursFiles) for number in recursFiles: countRecursFiles += 1 with Bar("Correction ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLineInFile = 0 numberLine = 0 # -- Count all line(s) in file -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLineInFile += 1 # -- Find empty class(es) in dict -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if counterToCheckIndent == 1: spacesAfterClass = len(eachLine) - len(eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterClass == spacesClass: if search: emptyClassInfo[search.group(1)] = file numberLineInFile += 1 # Adding one line because padding will be added numberLine += 1 # Adding one line because padding will be added if re.match(detectClass, eachLine): spacesClass = len(eachLine) - len(eachLine.lstrip()) if numberLine == numberLineInFile: # If empty class in last line search = re.search(classDefined, eachLine) if search: emptyClassInfo[search.group(1)] = file else: counterToCheckIndent += 1 search = re.search(classDefined, eachLine) # -- Add padding in empty class(es) -- # numberLine = 0 with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: numberLine += 1 if counterToCheckIndent == 1: spacesAfterClass = len(eachLine) - len(eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterClass == spacesClass: paddingVar1 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) paddingVar2 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) finalVarPadding = "{} = '{}'\n".format(paddingVar1, paddingVar2) spacesClass = spacesClass + basicIndentation sys.stdout.write(textwrap.indent(finalVarPadding, self.simpleSpace * spacesClass)) numberLine += 1 sys.stdout.write(eachLine) if re.match(detectClass, eachLine): spacesClass = len(eachLine) - len(eachLine.lstrip()) if numberLine == numberLineInFile: # If empty class in last line paddingVar1 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) paddingVar2 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) finalVarPadding = "{} = '{}'\n".format(paddingVar1, paddingVar2) spacesClass = spacesClass + basicIndentation sys.stdout.write(textwrap.indent(finalVarPadding, self.simpleSpace * spacesClass)) else: counterToCheckIndent += 1 bar.next(1) bar.finish() # -- Check if class(es) is still empty -- # if emptyClassInfo != {}: with Bar("Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLineInFile = 0 numberLine = 0 with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if counterToCheckIndent == 1: spacesAfterClass = len(eachLine) - len(eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterClass == spacesClass: if search: emptyClassInfo[search.group(1)] = file numberLineInFile += 1 numberLine += 1 if re.match(detectClass, eachLine): spacesClass = len(eachLine) - len(eachLine.lstrip()) if numberLine == numberLineInFile: # If empty class in last line search = re.search(classDefined, eachLine) if search: emptyClassInfo[search.group(1)] = file else: counterToCheckIndent += 1 search = re.search(classDefined, eachLine) bar.next(1) bar.finish() if emptyClassInfoCheck == {}: for key, value in emptyClassInfo.items(): print("\n-> File : {}".format(value)) print("-> Padding added in : {} ( empty class )".format(key)) return 0 else: if verboseArg: print("\n[!] No padding added to empty class(es)... :\n") for key, value in emptyClassInfoCheck.items(): print("\n-> File : {}".format(value)) print("-> Class : {}".format(key)) return 1 else: print("[!] No empty class found in {}".format(outputArg)) return 0 def EmptyFunctions(self, outputArg, mixerLengthArg, verboseArg): countRecursFiles = 0 counterToCheckIndent = 0 numberLine = 0 numberLineInFile = 0 emptyFuncInfo = {} emptyFuncInfoCheck = {} basicIndentation = None functionDefined = r"\s*def\s+(\w+)" detectFunction = r"\s*def\s+\w+" recursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False ) basicIndentation = Padding.CheckBasicIndentation(self, recursFiles) for number in recursFiles: countRecursFiles += 1 with Bar("Correction ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLineInFile = 0 numberLine = 0 # -- Count all line(s) in file -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLineInFile += 1 # -- Find empty function(s) in dict -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if counterToCheckIndent == 1: spacesAfterFunc = len(eachLine) - len(eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterFunc == spacesFunc: if search: emptyFuncInfo[search.group(1)] = file numberLineInFile += 1 # Adding one line because padding will be added numberLine += 1 # Adding one line because padding will be added if re.match(detectFunction, eachLine): spacesFunc = len(eachLine) - len(eachLine.lstrip()) if numberLine == numberLineInFile: # If empty function last line search = re.search(functionDefined, eachLine) if search: emptyFuncInfo[search.group(1)] = file else: counterToCheckIndent += 1 search = re.search(functionDefined, eachLine) # -- Add padding in empty function(s) -- # numberLine = 0 with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: numberLine += 1 if counterToCheckIndent == 1: spacesAfterFunc = len(eachLine) - len(eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterFunc == spacesFunc: paddingVar1 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) paddingVar2 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) finalVarPadding = "{} = '{}'\n".format(paddingVar1, paddingVar2) spacesFunc = spacesFunc + basicIndentation sys.stdout.write(textwrap.indent(finalVarPadding, self.simpleSpace * spacesFunc)) numberLine += 1 sys.stdout.write(eachLine) if re.match(detectFunction, eachLine): spacesFunc = len(eachLine) - len(eachLine.lstrip()) if numberLine == numberLineInFile: # IIf empty function last line paddingVar1 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) paddingVar2 = self.mixer.GetStringMixer(mixerLengthArgDefined=mixerLengthArg) finalVarPadding = "{} = '{}'\n".format(paddingVar1, paddingVar2) spacesFunc = spacesFunc + basicIndentation sys.stdout.write(textwrap.indent(finalVarPadding, self.simpleSpace * spacesFunc)) else: counterToCheckIndent += 1 bar.next(1) bar.finish() # -- Check if function(s) is still empty -- # if emptyFuncInfo != {}: with Bar("Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLineInFile = 0 numberLine = 0 with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if counterToCheckIndent == 1: spacesAfterFunc = len(eachLine) - len(eachLine.lstrip()) counterToCheckIndent = 0 if spacesAfterFunc == spacesFunc: if search: emptyFuncInfoCheck[search.group(1)] = file numberLineInFile += 1 numberLine += 1 if re.match(detectFunction, eachLine): spacesFunc = len(eachLine) - len(eachLine.lstrip()) if numberLine == numberLineInFile: # If empty function last line search = re.search(functionDefined, eachLine) if search: emptyFuncInfoCheck[search.group(1)] = file else: counterToCheckIndent += 1 search = re.search(functionDefined, eachLine) bar.next(1) bar.finish() if emptyFuncInfoCheck == {}: for key, value in emptyFuncInfo.items(): print("\n-> File : {}".format(value)) print("-> Padding added in : {} ( empty function )".format(key)) return 0 else: if verboseArg: print("\n[!] No padding added to empty function(s)... :\n") for key, value in emptyFuncInfoCheck.items(): print("\n-> File : {}".format(value)) print("-> Function : {}".format(key)) return 1 else: print("[!] No empty function found in {}".format(outputArg)) return 0
class Remove: def __init__(self): self.utils = Utils() def LinesSpaces(self, outputArg=None, verboseArg=None): checkLinesSpace = {} checkEmptyLine = 0 countRecursFiles = 0 numberLine = 0 recursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False) for file in recursFiles: countRecursFiles += 1 # -- Remove all empty lines -- # with Bar(PROGRESS_COLOUR + "Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with fileinput.FileInput(file, inplace=True) as inputFile: for eachLine in inputFile: if eachLine == "\n": pass else: sys.stdout.write(eachLine) bar.next(1) bar.finish() with Bar(PROGRESS_COLOUR + "Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLine = 0 with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: numberLine += 1 if eachLine == "\n": checkLinesSpace[numberLine] = file checkEmptyLine += 1 bar.next(1) bar.finish() if checkEmptyLine == 0: return EXIT_SUCCESS else: if verboseArg: print("\n[!] Empty line that not been removed... :\n") for key, value in checkLinesSpace.items(): print("\n-> File : {0}".format(value)) print("-> Line : {0}".format(key)) return EXIT_FAILURE def Comments(self, outputArg=None, verboseArg=None): checkRemoved = {} countLineOutput = 0 countLineInput = 0 noComments = 0 multipleLinesComments = 0 countRecursFiles = 0 numberLine = 0 commentsBeginLine = r"^\#.*" quoteOfCommentsEndLines = r".*[\'\|\"]{3}\s*$" quoteInRegex = r"={1}\s*r[\"|\']{3}" quoteOfCommentsMultipleLines = r"^\s+[\"|\']{3}|^[\"|\']{3}" quoteOfCommentsOneLine = r".*[\"]{3}.*[\"]{3}|.*[\']{3}.*[\']{3}\s*$" quoteIntoVariableOrPrint = r"s*print.*\(?[\"|\']{3}|.*\=\s*[\"|\']{3}" commentsAfterLine = r"\s*\#[^\"|^\'|^\.|^\?|^\*|^\!|^\]|^\[|^\\|^\)|^\(|^\{|^\}].*" recursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False) for number in recursFiles: countRecursFiles += 1 # -- Remove comments and Count comments will be removed -- # print("\n[+] Running remove comments in {0} file(s)...\n".format( countRecursFiles)) with Bar(PROGRESS_COLOUR + "Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: # -- Remove comments -- # with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: if "coding" in eachLine or "#!" in eachLine: sys.stdout.write(eachLine) continue if multipleLinesComments == 1 or noComments == 1: if re.match(quoteOfCommentsEndLines, eachLine): if re.match(quoteInRegex, eachLine): countLineInput += 1 continue elif re.match(quoteIntoVariableOrPrint, eachLine): countLineInput += 1 continue else: if multipleLinesComments == 1: multipleLinesComments = 0 countLineInput += 1 else: noComments = 0 sys.stdout.write(eachLine) continue else: if noComments == 1: sys.stdout.write(eachLine) else: countLineInput += 1 continue elif multipleLinesComments == 0: if re.match(quoteOfCommentsMultipleLines, eachLine): if re.match(quoteOfCommentsOneLine, eachLine): countLineInput += 1 continue else: countLineInput += 1 multipleLinesComments = 1 continue elif re.match(quoteIntoVariableOrPrint, eachLine): noComments = 1 sys.stdout.write(eachLine) continue else: sys.stdout.write(eachLine) else: sys.stdout.write(eachLine) bar.next(0.5) with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: if "coding" in eachLine or "#!" in eachLine: sys.stdout.write(eachLine) continue searchCommentsAfterLine = re.search( commentsAfterLine, eachLine) searchCommentsBeginLine = re.search( commentsBeginLine, eachLine) if searchCommentsBeginLine: countLineInput += 1 eachLine = eachLine.replace( searchCommentsBeginLine.group(0), "") sys.stdout.write(eachLine) elif searchCommentsAfterLine: eachLine = eachLine.replace( searchCommentsAfterLine.group(0), "") countLineInput += 1 sys.stdout.write(eachLine) else: sys.stdout.write(eachLine) bar.next(0.5) bar.finish() # -- Check if all comments are removed -- # noComments = 0 multipleLinesComments = 0 with Bar(PROGRESS_COLOUR + "Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with open(file, "r") as readFile: readF = readFile.readlines() numberLine = 0 for eachLine in readF: numberLine += 1 if "coding" in eachLine or "#!" in eachLine: continue else: if multipleLinesComments == 1 or noComments == 1: if re.match(quoteOfCommentsEndLines, eachLine): if re.match(quoteInRegex, eachLine): countLineOutput += 1 continue elif re.match(quoteIntoVariableOrPrint, eachLine): countLineOutput += 1 continue else: if multipleLinesComments == 1: multipleLinesComments = 0 checkRemoved[ numberLine] = "{0} ( multiple lines comments )".format( file) countLineOutput += 1 else: noComments = 0 continue else: if noComments == 1: pass else: countLineOutput += 1 continue elif multipleLinesComments == 0: if re.match(quoteOfCommentsMultipleLines, eachLine): if re.match(quoteOfCommentsOneLine, eachLine): checkRemoved[numberLine] = file countLineOutput += 1 continue else: countLineOutput += 1 multipleLinesComments = 1 continue elif re.match(quoteIntoVariableOrPrint, eachLine): noComments = 1 continue else: pass else: pass bar.next(0.5) with open(file, "r") as readFile: readF = readFile.readlines() numberLine = 0 for eachLine in readF: numberLine += 1 if "coding" in eachLine or "#!" in eachLine: continue else: searchCommentsAfterLine = re.search( commentsAfterLine, eachLine) searchCommentsBeginLine = re.search( commentsBeginLine, eachLine) if searchCommentsBeginLine: checkRemoved[numberLine] = file countLineOutput += 1 elif searchCommentsAfterLine: checkRemoved[numberLine] = file countLineOutput += 1 else: pass bar.next(0.5) bar.finish() if countLineOutput == 0: print( "\n-> {0} lines of comments removed\n".format(countLineInput)) return EXIT_SUCCESS else: if verboseArg: print("\n[!] Comments that not been removed... :\n") for key, value in checkRemoved.items(): print("\n-> File : {0}".format(value)) if "multiple lines comments" in value: print( "-> Line : {0} ( This is the end line of multiple lines comments )" .format(key)) else: print("-> Line : {0}".format(key)) print("\n-> {0} lines of comments no removed\n".format( countLineOutput)) return EXIT_FAILURE def TrashFiles(self, outputArg=None, verboseArg=None): countRecursFiles = 0 removeFiles = 0 checkPycFile = [] currentPosition = os.getcwd() detectPycFiles = r".*\.pyc$" removePycFiles = r"\w+\.pyc$" recursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="pyc", blockDir="__pycache__", blockFile=False, dirOnly=False) for number in recursFiles: countRecursFiles += 1 if countRecursFiles == 0: print("[!] No .pyc file(s) found in {0}".format(outputArg)) return EXIT_SUCCESS print("\n[+] Running remove {0} .pyc file(s)...\n".format( countRecursFiles)) # -- Check if .pyc file(s) exists and remove it -- # with Bar(PROGRESS_COLOUR + "Setting up ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: if re.match(detectPycFiles, file): removeFiles += 1 checkPycFile.append(file) bar.next(1) bar.finish() # -- Remove pyc file(s) -- # with Bar(PROGRESS_COLOUR + "Correction ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: if re.match(detectPycFiles, file): extractPycFiles = re.search(removePycFiles, file) moveFolder = re.sub(removePycFiles, "", file) os.chdir(moveFolder) os.remove(extractPycFiles.group(0)) os.chdir(currentPosition) bar.next(1) bar.finish() checkRecursFiles = self.utils.CheckFileDir(output=outputArg, detectFiles="pyc", blockDir="__pycache__", blockFile=False, dirOnly=False) if checkRecursFiles != []: if verboseArg: for pycFile in checkRecursFiles: print("-> .pyc file no removed : {0}".format(pycFile)) return EXIT_FAILURE else: if verboseArg: for pycFile in checkPycFile: print("-> .pyc file removed : {0}".format(pycFile)) print("\n-> {0} .pyc file(s) removed".format(removeFiles)) return EXIT_SUCCESS
class Replace: def __init__(self): self.mixer = Mixer() self.remove = Remove() self.utils = Utils() self.pythonExcludeDefaultString = "exclude/string_to_string_mixed/exclude_word_do_not_modify.txt" self.pythonExcludeUserString = "exclude/string_to_string_mixed/exclude_word_by_user.txt" def EachLine(self, line="", dictionary={}, fileNameImport=False, listModuleImport=False): getIndexLineList = [] returnLine = [] charValue = [] checkCharAfterWord = 1 wordSetter = 0 checkGetKey = "" checkGetWord = "" getLine = "" breakLine = "" if listModuleImport == True: detectChars = r"\.|\:|\)|\(|\=|\[|\]|\{|\}|\,|\+|\s|\*|\-|\%|\/|\^|\'|\"" else: detectChars = r"\.|\:|\)|\(|\=|\[|\]|\{|\}|\,|\+|\s|\*|\-|\%|\/|\^" # -- Get list of all letters in line -- # for indexLine, letterLine in enumerate(line): getIndexLineList.append(letterLine) # -- Loop in each letter of line -- # for indexLine, letterLine in enumerate(line): # -- Add in final line list all chars mixed -- # if charValue != []: for obfIndex, obfValue in enumerate(charValue): if obfIndex == 0: # First letter in string mixed are already add in the final line continue returnLine.append(obfValue) charValue = [] # -- If the variable is only a letter, check if the next character is specific so as not to replace it -- # if re.match(detectChars, letterLine): returnLine.append(letterLine) # -- Count indexes of word to move after it --# countDeleteIndex = 0 for i in getWord: countDeleteIndex += 1 wordSetter = countDeleteIndex - 2 # -2 Is to letter already append and the letter in progress else: # -- The index numbers of variable is decremented to add the mixed letters that be replaced -- # if wordSetter > 0: wordSetter -= 1 continue else: try: # -- Loop in the dictionary with already mixed values-- # for key, value in dictionary: for indexKey, letterKey in enumerate(key): for letterValue in value: # -- Check if letter of word is equal to letter of key -- # if letterKey == letterLine: # -- Begin process to check -- # if indexKey == 0: indexExplore = indexLine + len(key) # Place index position after the word # -- If indexError return to next loop -- # try: getIndexLineList[indexExplore] except IndexError: continue # -- Check the char after and before the word -- # if re.match(detectChars, getIndexLineList[indexExplore]): indexExploreBefore = indexLine - 1 # Index check if word finded is not into the other word indexExploreAfter = indexLine + 2 # Index check char after the end string found with 'detectChars' regex try: if not re.match(r"\w|\\|\%", getIndexLineList[indexExploreBefore]): # -- Check if it's 'from' and 'import' file in line to avoid replace name of file if variable is identic name to file -- # getLine = "".join(getIndexLineList) if fileNameImport == False: if "import" in getLine: if "from" in getLine: # -- Cut the line from the current index and check if it is not there is the keyword "import" in the line -- # breakLine = getIndexLineList[:indexLine] breakLine = "".join(breakLine) if not "import" in breakLine: # -- It's file because only 'from'key word -- # checkCharAfterWord = 1 else: checkCharAfterWord = 0 else: checkCharAfterWord = 1 # -- Check if after char find by 'detectChars' variable it's not ' or " -- # elif re.match(r"\"|\'", getIndexLineList[indexExploreAfter]): if re.match(r"\[|\(|\{", getIndexLineList[indexExploreAfter - 1]): checkCharAfterWord = 0 else: checkCharAfterWord = 1 else: checkCharAfterWord = 0 # -- Only for [-rfn, --replacefilsname] feature -- # else: # -- check if file name is imported - # breakLine = getIndexLineList[:indexLine] breakLine = "".join(breakLine) # -- if file name is imported after 'import', the file name is not replaced -- # if "import" in breakLine: checkCharAfterWord = 1 else: checkCharAfterWord = 0 else: checkCharAfterWord = 1 except IndexError: checkCharAfterWord = 0 pass else: checkCharAfterWord = 1 if checkCharAfterWord == 0: # -- Initialize vars -- # getCharAllInKey = [] getWord = [] indexExploreStart = indexLine indexExploreEnd = indexLine + len(key) - 1 # Remove -1, first letter is already increment # -- List contain all letters of key -- # for getLetterKey in key: getCharAllInKey.append(getLetterKey) # -- Check if all letters of key is equal to all letters of word -- # for indexCheckLetter, checkLetter in enumerate(getIndexLineList): if indexCheckLetter >= indexExploreStart and indexCheckLetter <= indexExploreEnd: getWord.append(checkLetter) # -- Check if number of chars in key equal number of chars in word -- # if list(set(getCharAllInKey) - set(getWord)) == []: checkGetWord = "".join(getWord) checkGetKey = "".join(getCharAllInKey) # -- Check if key == word -- # if checkGetWord == checkGetKey: for obfChar in value: charValue.append(obfChar) letterLine = letterValue raise BreakLoop else: continue else: continue else: continue else: continue else: continue raise BreakLoop except BreakLoop: returnLine.append(letterLine) # -- Rewrite the line -- # returnLine = "".join(returnLine) return returnLine[:] def StringToString(self, outputArg=None, mixerLevelArg=None, verboseArg=None): variablesDict = {} classesDict = {} functionsDict = {} allDict = {} classFuncDict = {} checkWordsMixed = [] wordsExcludedUser = [] wordsExcludedUserFound = [] wordsExcludedDefault = [] wordsExcludedDefaultFound = [] checkAllWords = [] checkWordsError = [] checkKeyWordsMixed = [] checkCountWordsMixed = 0 checkCountWordsValue = 0 checkQuotePassing = 0 countRecursFiles = 0 # -- Catch all variable(s)/class(es)/function(s) name -- # functionsDefined = r"def\s+(\w+)" classDefined = r"class\s+(\w+)" variablesErrorDefined = r"except(\s+\w+\s+as\s+)(\w)" variablesLoopDefined = r"for\s+([\w\s\,]+)(\s+in\s+)" variablesDefined = r"(^[\s|a-zA-Z_]+[\,\s\w]{0,})+(\s*=\s*[\[|\{\(|\w+|\"|\'])" quotesIntoVariable = r".*={1}\s*[\"|\']{3}" quotesEndMultipleLines = r"^\s*[\"|\']{3}\)?\.?" quotesInRegex = r"={1}\s*r{1}[\"|\']{3}" recursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False ) for file in recursFiles: countRecursFiles += 1 print("\n[+] Running replacement of variables/classes/functions in {0} file(s), he can be long... you have time to make a coffee :)\n".format(countRecursFiles)) # -- Replace variables/classes/functions to random strings with length defined -- # with Bar(PROGRESS_COLOUR + "Setting up ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: # -- Variables -- # search = re.search(variablesDefined, eachLine) if search: if "," in search.group(1): modifySearch = search.group(1).replace(",", " ") modifySearch = modifySearch.split() for i in modifySearch: if i not in variablesDict: mixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) i = i.strip() variablesDict[i] = mixer else: if search.group(1) not in variablesDict: mixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) modifySearch = search.group(1).strip() variablesDict[modifySearch] = mixer # -- Error variables -- # search = re.search(variablesErrorDefined, eachLine) if search: mixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) if search.group(2) not in variablesDict: variablesDict[search.group(2)] = mixer # -- Loop variables -- # search = re.search(variablesLoopDefined, eachLine) if search: if "," in search.group(1): modifySearch = search.group(1).replace(",", " ") modifySearch = modifySearch.split() for i in modifySearch: if i not in variablesDict: mixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) variablesDict[i] = mixer else: if search.group(1) not in variablesDict: mixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) variablesDict[search.group(1)] = mixer # -- Function(s) -- # search = re.search(functionsDefined, eachLine) if search: mixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) if search.group(1) not in functionsDict: if not "__init__" in search.group(1): functionsDict[search.group(1)] = mixer # -- Class(es) -- # search = re.search(classDefined, eachLine) if search: mixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) if search.group(1) not in classesDict: classesDict[search.group(1)] = mixer bar.next(1) # -- Remove excluded variables/classes/functions defined from 'exclude/string_to_string_mixed/exclude_word_do_not_modify.txt' -- # if os.path.exists(self.pythonExcludeDefaultString) == True: with open(self.pythonExcludeDefaultString, "r") as readFile: for word in readFile: if "#" in word or word == "\n": continue else: word = word.rstrip() wordsExcludedDefault.append(word) else: print(ERROR_COLOUR + "[-] '{0}' file not found\n".format(self.pythonExcludeDefaultString)) # -- Remove excluded variables/classes/functions defined from 'exclude/string_to_string_mixed/exclude_word_by_user.txt' -- # if os.path.exists(self.pythonExcludeUserString) == True: with open(self.pythonExcludeUserString, "r") as readFile: for word in readFile: if "#" in word or word == "\n": continue else: word = word.rstrip() wordsExcludedUser.append(word) else: print(ERROR_COLOUR + "[-] '{0}' file not found\n".format(self.pythonExcludeUserString)) for word in wordsExcludedUser: for key in variablesDict.keys(): if word in key: if re.match(r"^" + re.escape(word) + r"$", key): wordsExcludedUserFound.append(word) for key in classesDict.keys(): if word in key: if re.match(r"^" + re.escape(word) + r"$", key): wordsExcludedUserFound.append(word) for key in functionsDict.keys(): if word in key: if re.match(r"^" + re.escape(word) + r"$", key): wordsExcludedUserFound.append(word) for word in wordsExcludedDefault: for key in variablesDict.keys(): if word in key: if re.match(r"^" + re.escape(word) + r"$", key): wordsExcludedDefaultFound.append(word) for key in classesDict.keys(): if word in key: if re.match(r"^" + re.escape(word) + r"$", key): wordsExcludedDefaultFound.append(word) for key in functionsDict.keys(): if word in key: if re.match(r"^" + re.escape(word) + r"$", key): wordsExcludedDefaultFound.append(word) for word in wordsExcludedUserFound: if word in variablesDict.keys(): del variablesDict[word] if word in classesDict.keys(): del classesDict[word] if word in functionsDict.keys(): del functionsDict[word] for word in wordsExcludedDefaultFound: if word in variablesDict.keys(): del variablesDict[word] if word in classesDict.keys(): del classesDict[word] if word in functionsDict.keys(): del functionsDict[word] bar.finish() # -- Display variables/classes/functions found -- # if verboseArg: print("\n[+] Variable(s) found :\n") if variablesDict == {}: print("-> No result") else: for key, value in variablesDict.items(): print("-> {0} : {1}".format(key, value)) print("\n[+] Class(es) found :\n") if classesDict == {}: print("-> No result") else: for key, value in classesDict.items(): print("-> {0} : {1}".format(key, value)) print("\n[+] Function(s) found :\n") if functionsDict == {}: print("-> No result") else: for key, value in functionsDict.items(): print("-> {0} : {1}".format(key, value)) print("\n[+] String excluded found in '{0}' that have been matched from '{1}' :\n".format(self.pythonExcludeUserString, outputArg)) if wordsExcludedUserFound == []: print("-> No result") else: for word in wordsExcludedUserFound: print("-> {0} : excluded".format(word)) print("\n[+] String excluded found in '{0}' that have been matched from '{1}' :\n".format(self.pythonExcludeDefaultString, outputArg)) if wordsExcludedDefaultFound == []: print("-> No result") else: for word in wordsExcludedDefaultFound: print("-> {0} : excluded".format(word)) print("") # -- Merge all dicts -- # allDict = self.utils.DictMerge(dict1=allDict, dict2=variablesDict) allDict = self.utils.DictMerge(dict1=allDict, dict2=functionsDict) allDict = self.utils.DictMerge(dict1=allDict, dict2=classesDict) classFuncDict = self.utils.DictMerge(dict1=classFuncDict, dict2=classesDict) classFuncDict = self.utils.DictMerge(dict1=classFuncDict, dict2=functionsDict) # -- Change variables/classes/functions to mixed values -- # with Bar(PROGRESS_COLOUR + "Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: # -- Replace variable(s) only -- # with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: if not eachLine: continue else: if re.match(quotesIntoVariable, eachLine): if re.match(quotesInRegex, eachLine): sys.stdout.write(eachLine) else: checkQuotePassing += 1 eachLine = Replace.EachLine( self, line=eachLine, dictionary=allDict.items(), fileNameImport=False, listModuleImport=False ) sys.stdout.write(eachLine) continue elif re.match(quotesEndMultipleLines, eachLine): if re.match(quotesInRegex, eachLine): sys.stdout.write(eachLine) else: checkQuotePassing += 1 eachLine = Replace.EachLine( self, line=eachLine, dictionary=allDict.items(), fileNameImport=False, listModuleImport=False ) sys.stdout.write(eachLine) if checkQuotePassing == 2: checkQuotePassing = 0 continue if checkQuotePassing == 1: sys.stdout.write(eachLine) elif checkQuotePassing == 2: checkQuotePassing = 0 else: if re.match(r"\s*__all__\s*=\s*\[", eachLine): eachLine = Replace.EachLine( self, line=eachLine, dictionary=classFuncDict.items(), fileNameImport=False, listModuleImport=True ) sys.stdout.write(eachLine) else: eachLine = Replace.EachLine( self, line=eachLine, dictionary=allDict.items(), fileNameImport=False, listModuleImport=False ) sys.stdout.write(eachLine) bar.next(1) bar.finish() with Bar(PROGRESS_COLOUR + "Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: # -- Check if variables/classes/functions have been mixed -- # with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: for key, value in allDict.items(): if value in eachLine: if re.match(r"\w+" + re.escape(value) + r"\w+", eachLine): pass else: checkWordsMixed.append(value) checkKeyWordsMixed.append(key) bar.next(1) # -- Remove duplicated words -- # checkListWordsMixed = list(dict.fromkeys(checkWordsMixed)) checkKeyWordsMixed = list(dict.fromkeys(checkKeyWordsMixed)) for i in checkListWordsMixed: checkCountWordsMixed += 1 for i in allDict.values(): checkCountWordsValue += 1 bar.finish() if checkCountWordsMixed == checkCountWordsValue: print("\n-> {0} variable(s)/class(es)/function(s) replaced in {1} file(s)\n".format(checkCountWordsValue, countRecursFiles)) return EXIT_SUCCESS else: if verboseArg: for key in allDict.keys(): checkAllWords.append(key) checkWordsError = list(set(checkAllWords) - set(checkKeyWordsMixed)) print("\n[!] Word(s) that not been replaced, check if an error will appear when will launch your obfuscated code... :\n") if checkWordsError != []: for wordNoReplaced in checkWordsError: print("-> Word : {0}".format(wordNoReplaced)) return EXIT_FAILURE def StringsToHex(self, outputArg=None, mixerLevelArg=None, verboseArg=None): checkHexError = {} getLetterLineList = [] countRecursFiles = 0 checkPrint = 0 numberLine = 0 checkError = False hexLine = "" detectExecFunc = r"exec\(\w+\)" detectQuotes = r"\'{3}|\"{3}" recursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False ) for number in recursFiles: countRecursFiles += 1 print("\n[+] Running replace all strings to their hexadecimal value in {0} file(s)...\n".format(countRecursFiles)) # -- Replace all strings to their hexadecimal value -- # with Bar(PROGRESS_COLOUR + "Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: # -- Add a new first random line and move the old first line to the second line to avoid replacing it -- # checkPrint = 0 # initialize check print() func at the begining of each file with open(file, "r") as inputFile: stringRandomMixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) firstLine = "{0}\n".format(stringRandomMixer) line = inputFile.readlines() line.insert(0, firstLine) with open(file, "w") as inputFile: inputFile.writelines(line) # -- Replace all lines-- # with fileinput.input(file, inplace=True) as inputFile: for eachLine in inputFile: if checkPrint == 0: varMixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) sys.stdout.write(varMixer + "=\"\"\"") checkPrint = 1 else: getLetterLineList = [] for letterLine in eachLine: letterToHex = "\\x" + str(letterLine.encode().hex()) getLetterLineList.append(letterToHex) # Get list of all letters in line hexLine = "".join(getLetterLineList) sys.stdout.write(hexLine) # -- Add exec funtions to interpret hex code in strings -- # with open(file, "a") as inputFile: inputFile.write("\"\"\"") inputFile.write("\nexec({0})".format(varMixer)) bar.next(1) bar.finish() with Bar(PROGRESS_COLOUR + "Check ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: numberLine = 0 # -- Check if all lines are replaced of hexadecimal value -- # with open(file, "r") as inputFile: for eachLine in inputFile: numberLine += 1 if not eachLine: continue else: if not "\\x" in eachLine: if re.match(detectQuotes, eachLine): continue elif re.match(detectExecFunc, eachLine): continue else: checkHexError[numberLine] = file checkError = True else: continue bar.next(1) bar.finish() if checkError == False: return EXIT_SUCCESS else: if verboseArg: print("\n[!] Line(s) that have not been replaced by their hexadecimal values... :\n") for key, value in checkHexError.items(): print("\n-> File : {0}".format(value)) print("-> Line : {0}".format(key)) else: print(ERROR_COLOUR + "\n[-] Line(s) that have not been replaced by their hexadecimal values") return EXIT_FAILURE def FilesName(self, outputArg=None, mixerLevelArg=None, verboseArg=None): checkFilesFoundCompare = {} filesNameDict = {} filesNameDictNoExt = {} filesNameFound = [] filesNameFoundNoExt = [] filesNameMixed = [] wordsExcluded = [] wordsExcludedFound = [] importNoCompliantFound = [] badNameDir = [] numberLine = 0 countRecursFiles = 0 checkCountRecursFiles = 0 unix = False win = False currentPosition = os.getcwd() detectImport = r"\s*from\s+|\s*import\s+" recursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile="__init__", dirOnly=False ) recursFilesWithInit = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile=False, dirOnly=False ) recursDirs = self.utils.CheckFileDir( output=outputArg, detectFiles="", blockDir="__pycache__", blockFile=False, dirOnly=True ) for file in recursFiles: countRecursFiles += 1 print("\n[+] Running replace files name in {0} file(s)...\n".format(countRecursFiles)) with Bar(PROGRESS_COLOUR + "Setting up ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for file in recursFiles: if "\"" in file: parseFilePath = file.split("\"") win = True else: parseFilePath = file.split("/") unix = True mixer = self.mixer.GetStringMixer(lenght=mixerLevelArg) filesNameDict[parseFilePath[-1]] = mixer + ".py" filesNameDictNoExt[parseFilePath[-1].replace(".py", "")] = mixer removeExt = parseFilePath[-1].replace(".py","") filesNameFound.append(parseFilePath[-1]) filesNameFoundNoExt.append(removeExt) filesNameMixed.append(mixer + ".py") # -- Check if directory have the same name for directory in recursDirs: for fileName in filesNameFoundNoExt: if re.match(r".*/{1}" + re.escape(fileName) + r"/{1}.*", directory): wordsExcluded.append(fileName) badNameDir.append(fileName) for word in wordsExcluded: for fileNameNoExt in filesNameFoundNoExt: if fileNameNoExt == word: wordsExcludedFound.append(word) filesNameFoundNoExt.remove(word) for fileName in filesNameFound: if fileName == word: filesNameFound.remove(word) if word in filesNameDictNoExt.keys(): del filesNameDictNoExt[word] word = word + ".py" if word in filesNameDict.keys(): del filesNameDict[word] # -- Check if file name in code is after 'import' native python function --# for file in recursFiles: with open(file, "r") as readFile: readF = readFile.readlines() for eachLine in readF: if re.match(detectImport, eachLine): searchFileName = re.search(r"(import\s+)(.*)", eachLine) if searchFileName.group(2): searchFileName = searchFileName.group(2).replace(",", "") searchFileName = searchFileName.split() for i in searchFileName: i = i.strip() for fileNameNoExt in filesNameFoundNoExt: if fileNameNoExt == i: importNoCompliantFound.append(i) filesNameFoundNoExt.remove(i) for fileName in filesNameFound: fileName = fileName.replace(".py", "") if fileName == i: i = i + ".py" filesNameFound.remove(i) if i in filesNameDictNoExt.keys(): del filesNameDictNoExt[i] i = i + ".py" if i in filesNameDict.keys(): del filesNameDict[i] bar.next(1) bar.finish() # -- Diplay all files found with their mixed values if verbose arg is actived -- # if verboseArg: print("\n[+] File(s) name found with their mixed values :\n") if filesNameDict == {}: print("-> No result") else: for key, value in filesNameDict.items(): print("-> {0} : {1}".format(key, value)) if importNoCompliantFound != []: print("\n[+] File(s) name no compliant for 'replace file name' feature :\n") for i in importNoCompliantFound: print("-> {0} : no compliant ( file name excluded automatically )".format(i)) if badNameDir != []: print("\n[!] Directory that have same name of python file(s) :\n") for i in badNameDir: print("-> {0} : no compliant ( file name has been excluded automatically)".format(i)) i = i.rstrip() print("") # -- Replace all files name to random strings with length defined -- # with Bar(PROGRESS_COLOUR + "Obfuscation ", fill="=", max=countRecursFiles, suffix="%(percent)d%%") as bar: for fileInCode in recursFilesWithInit: # -- Rename all files in python code -- # with fileinput.input(fileInCode, inplace=True) as inputFile: for eachLine in inputFile: if re.match(detectImport, eachLine): eachLine = Replace.EachLine( self, line=eachLine, dictionary=filesNameDictNoExt.items(), fileNameImport=True, listModuleImport=False ) sys.stdout.write(eachLine) continue else: sys.stdout.write(eachLine) bar.next(0.5) for file in recursFiles: # -- Rename all files in their directories -- # if "\"" in file: parseFilePath = file.split("\"") win = True else: parseFilePath = file.split("/") unix = True for key, value in filesNameDict.items(): if key == parseFilePath[-1]: parseFilePath.remove(parseFilePath[-1]) if unix == True: parseFilePathToMove = "/".join(parseFilePath) else: parseFilePathToMove = "\"".join(parseFilePath) os.chdir(parseFilePathToMove) # Move in directory to rename python file os.rename(key, value) else: continue os.chdir(currentPosition) bar.next(0.5) bar.finish() checkRecursFiles = self.utils.CheckFileDir( output=outputArg, detectFiles="py", blockDir="__pycache__", blockFile="__init__", dirOnly=False ) for file in checkRecursFiles: checkCountRecursFiles += 1 # -- Check if all files name are been replaced to random strings -- # with Bar(PROGRESS_COLOUR + "Check ", fill="=", max=checkCountRecursFiles, suffix="%(percent)d%%") as bar: for file in checkRecursFiles: numberLine = 0 # -- Check for file name in directory -- # for key, value in filesNameDict.items(): if key in file: checkFilesFoundCompare[key] = value bar.next(1) bar.finish() if checkFilesFoundCompare != {} : if verboseArg: if checkFilesFoundCompare != {}: print("\n[!] File name that have not been replaced by their random string value... :\n") for key, value in checkFilesFoundCompare.items(): print("\n-> File : {0}".format(key)) print("-> Value mixed : {0}".format(value)) else: print(ERROR_COLOUR + "\n[-] File name that have not been replaced by their random string value") return EXIT_FAILURE else: return EXIT_SUCCESS