class Analyze:
def __init__(self):
self.utils = Utils()
def InputAvailable(self, inputArg, codeArg, verboseArg):
inputFileFound = []
inputFileEmpty = []
inputFileFoundCount = 0
inputFileEmptyCount = 0
if os.path.exists(inputArg) == True:
if os.path.isdir(inputArg) == True:
if codeArg == "python":
detectFile = "py"
blockdirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
inputArg, self.utils.Platform(), detectFile),
recursive=True)
]
if recursFiles == []:
print("[-] {0} directory empty".format(inputArg))
return ERROR_DIR_EMPTY
print("\n[+] Running analyze input...")
for file in recursFiles:
if re.match(blockdirs, file):
continue
else:
if os.path.getsize(file) > 0:
inputFileFound.append(file)
inputFileFoundCount += inputFileFoundCount + 1
else:
inputFileEmpty.append(file)
inputFileEmptyCount += inputFileEmptyCount + 1
if inputFileFoundCount >= 1 and inputFileEmptyCount < inputFileFoundCount:
if verboseArg:
print("\n[+] File input found :\n")
if inputFileFound == []:
print("-> no result")
else:
for file in inputFileFound:
print("-> {0}".format(file))
for file in inputFileEmpty:
print("-> {0} : empty".format(file))
print("")
return EXIT_SUCCESS
elif inputFileFoundCount == inputFileEmptyCount and inputFileFoundCount > 0:
print("[-] All files in directory specified are emtpy !.")
return ERROR_FILE_EMPTY
else:
print("[-] No file available in '{0}'.".format(inputArg))
return ERROR_FILE_NOT_FOUND
else:
print("[-] '{0}' is not a directory.".format(inputArg))
return ERROR_NOT_DIR
else:
print("'{0}' not exists.".format(inputArg))
return ERROR_PATH_NOT_FOUND
def OutputAvailable(self, inputArg, codeArg, outputArg, verboseArg):
outputFileFound = []
outputFileEmpty = []
outputFileFoundCount = 0
outputFileEmptyCount = 0
if os.path.exists(outputArg) == True:
removeRequest = input(
"[!] Output '{0}' already exists, do you want remove it ? (Y/N) : "
.format(outputArg))
removeRequest = removeRequest.upper()
if removeRequest == "Y":
try:
shutil.rmtree(outputArg)
if os.path.exists(outputArg) == False:
shutil.copytree(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isdir(outputArg) == True:
if codeArg == "python":
detectFile = "py"
blockdirs = r"__pycache__"
recursFiles = [
f for f in glob.glob(
"{0}{1}**{1}*.{2}".format(
inputArg, self.utils.Platform(),
detectFile),
recursive=True)
]
if recursFiles == []:
print("[-] {0} directory empty".format(
inputArg))
return ERROR_DIR_EMPTY
print("\n[+] Running analyze output...")
for file in recursFiles:
if re.match(blockdirs, file):
continue
else:
if os.path.getsize(file) > 0:
outputFileFound.append(file)
outputFileFoundCount += outputFileFoundCount + 1
else:
outputFileEmpty.append(file)
outputFileEmptyCount += outputFileEmptyCount + 1
if outputFileFoundCount >= 1 and outputFileFoundCount > outputFileEmptyCount:
if verboseArg:
print("\n[+] Ouput files copy :\n")
if outputFileFound == []:
print("-> no result")
else:
for file in outputFileFound:
print("-> {0}".format(file))
for file in outputFileEmpty:
print("-> {0} : empty".format(
file))
return EXIT_SUCCESS
else:
print("[-] No files available in '{0}'.".
format(outputArg))
return ERROR_FILE_NOT_FOUND
else:
print(
"[-] Copy '{0}' to '{1}' failed, this is not a output directory copied !."
.format(inputArg, outputArg))
return ERROR_NOT_DIR
else:
print("[-] Copy '{0}' to '{1}' failed.".format(
inputArg, outputArg))
return ERROR_CANNOT_COPY
else:
print("[-] Remove '{0}' failed.".format(outputArg))
return ERROR_CANNOT_REMOVE
except Exception as e:
print("[-] {0}".format(e))
return EXIT_FAILURE
else:
print("[-] Remove '{0}' failed, the user has refused.".format(
outputArg))
return ERROR_CANNOT_REMOVE
else:
try:
shutil.copytree(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isdir(outputArg) == True:
if codeArg == "python":
detectFile = "py"
blockdirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
inputArg, self.utils.Platform(), detectFile),
recursive=True)
]
if recursFiles == []:
print("[-] {0} directory empty".format(inputArg))
return ERROR_DIR_EMPTY
print("\n[+] Running analyze output...")
for file in recursFiles:
if re.match(blockdirs, file):
continue
else:
if os.path.getsize(file) > 0:
outputFileFound.append(file)
outputFileFoundCount += outputFileFoundCount + 1
else:
outputFileEmpty.append(file)
outputFileEmptyCount += outputFileEmptyCount + 1
if outputFileFoundCount >= 1 and outputFileFoundCount > outputFileEmptyCount:
if verboseArg:
print("\n[+] File output found :\n")
if outputFileFound == []:
print("-> no result")
else:
for file in outputFileFound:
print("-> {0} : copy".format(file))
for file in outputFileEmpty:
print(
"-> {0} : copy empty".format(file))
return EXIT_SUCCESS
else:
print("[-] No files available in '{0}'.".format(
outputArg))
return ERROR_FILE_NOT_FOUND
else:
print(
"[-] Copy '{0}' to '{1}' failed, this is not a output directory copied !."
.format(inputArg, outputArg))
return ERROR_NOT_DIR
else:
print("[-] Copy '{0}' to '{1}' failed.".format(
inputArg, outputArg))
return ERROR_CANNOT_COPY
except Exception as e:
print("[-] {0}".format(e))
return EXIT_FAILURE
class Padding:
def __init__(self):
self.mixer = Mixer()
self.remove = Remove()
self.utils = Utils()
def ScriptsGenerator(self, codeArg, mixerLevelArg):
if mixerLevelArg == "lower":
varRandom1 = self.mixer.GetStringMixer("lower")
varRandom2 = self.mixer.GetStringMixer("lower")
varRandom3 = self.mixer.GetStringMixer("lower")
varRandom4 = self.mixer.GetStringMixer("lower")
varRandom5 = self.mixer.GetStringMixer("lower")
varRandom6 = self.mixer.GetStringMixer("lower")
varRandom7 = self.mixer.GetStringMixer("lower")
varRandom8 = self.mixer.GetStringMixer("lower")
varRandom9 = self.mixer.GetStringMixer("lower")
varRandom10 = self.mixer.GetStringMixer("lower")
varRandom11 = self.mixer.GetStringMixer("lower")
varRandom12 = self.mixer.GetStringMixer("lower")
elif mixerLevelArg == "medium":
varRandom1 = self.mixer.GetStringMixer("medium")
varRandom2 = self.mixer.GetStringMixer("medium")
varRandom3 = self.mixer.GetStringMixer("medium")
varRandom4 = self.mixer.GetStringMixer("medium")
varRandom5 = self.mixer.GetStringMixer("medium")
varRandom6 = self.mixer.GetStringMixer("medium")
varRandom7 = self.mixer.GetStringMixer("medium")
varRandom8 = self.mixer.GetStringMixer("medium")
varRandom9 = self.mixer.GetStringMixer("medium")
varRandom10 = self.mixer.GetStringMixer("medium")
varRandom11 = self.mixer.GetStringMixer("medium")
varRandom12 = self.mixer.GetStringMixer("medium")
elif mixerLevelArg == "high":
varRandom1 = self.mixer.GetStringMixer("high")
varRandom2 = self.mixer.GetStringMixer("high")
varRandom3 = self.mixer.GetStringMixer("high")
varRandom4 = self.mixer.GetStringMixer("high")
varRandom5 = self.mixer.GetStringMixer("high")
varRandom6 = self.mixer.GetStringMixer("high")
varRandom7 = self.mixer.GetStringMixer("high")
varRandom8 = self.mixer.GetStringMixer("high")
varRandom9 = self.mixer.GetStringMixer("high")
varRandom10 = self.mixer.GetStringMixer("high")
varRandom11 = self.mixer.GetStringMixer("high")
varRandom12 = self.mixer.GetStringMixer("high")
# ---------- Python random scripts ---------- #
if codeArg == "python":
rand = random.randint(1, 5)
# -- script 1 -- #
if rand == 1:
scriptAssPadding1 = 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 scriptAssPadding1
# -- script 2 -- #
elif rand == 2:
scriptAssPadding2 = textwrap.dedent("""
{0} = '{4}'
{1} = '{5}'
if {0} != {1}:
{2} = '{6}'
{3} = '{7}'
{3} = {2}
""").format(varRandom1, varRandom2, varRandom3, varRandom4, varRandom5, \
varRandom6, varRandom7, varRandom8)
return scriptAssPadding2
# -- script 3 -- #
elif rand == 3:
scriptAssPadding3 = 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 scriptAssPadding3
# -- script 4 -- #
elif rand == 4:
scriptAssPadding4 = textwrap.dedent("""
{0} = '{4}'
{1} = '{5}'
{3} = '{7}'
if {0} == {1}:
{2} = '{6}'
{2} = {0}
else:
{2} = '{6}'
{2} = {3}
""").format(varRandom1, varRandom2, varRandom3, varRandom4, \
varRandom5, varRandom6, varRandom7, varRandom8)
return scriptAssPadding4
# -- script 5 -- #
elif rand == 5:
scriptAssPadding5 = 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 scriptAssPadding5
def AddScripts(self, codeArg, outputArg, mixerLevelArg):
listCheckLineWhitoutSpace = []
listCheckLine = []
countScriptsAdded = 0
countLineAdded = 0
countLine = 0
checkLine = 0
checkQuotePassing = 0
checkCharPassing = 0
countRecursFiles = 0
checkParenthesesCharPassing = 0
checkBracketsCharPassing = 0
checkBracesCharPassing = 0
if codeArg == "python":
detectFile = "py"
blockDirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFile),
recursive=True)
]
# -- Count the number of lines that will be checked before filling -- #
for file in recursFiles:
if re.match(blockDirs, file):
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if not eachLine:
continue
countLine += 1
for number in recursFiles:
countRecursFiles += 1
print("\n[+] Running adding of random scripts in {0} file(s)...\n".
format(countRecursFiles))
# -- Padding scripts added -- #
with tqdm(total=countRecursFiles) as pbar:
for file in recursFiles:
pbar.update(1)
if re.match(blockDirs, file):
continue
else:
with fileinput.input(file, inplace=True) as inputFile:
for eachLine in inputFile:
print(eachLine)
if eachLine == "\n":
continue
else:
if codeArg == "python":
spaces = len(eachLine) - len(
eachLine.lstrip()) # Check line indent
noAddScript = r"^\@|\s+\@|\s+return|\s*def\s+.+\s*\:{1}|^class\s+.+\s*\:{1}|.*[\[|\(|\{|\,|\\]$|\s+[\)|\]|\}]$"
addIndentScript = r".*\:{1}\s"
checkAddIndentScript = r".*\:{1}\s\w+"
quoteIntoVariable = r".*\={1}\s*\w*\.?\w*[\(|\.]{1}[\"|\']{3}|.*\={1}\s*[\"|\']{3}" # """ and ''' before an variables
quoteOfCommentariesMultipleLines = r"^\s*[\"|\']{3}$" # """ and ''' without before variables and if commentaries is over multiple lines
quoteOfEndCommentariesMultipleLines = r"^\s*[\"|\']{3}\)?\.?" # """ and ''' without before variables, if commentaries is over multiple lines and he finish by .format() funtion
listCheckLine = [] # Initialize var
for i in eachLine:
listCheckLine.append(i)
# -- Check if end char in line is "'" or '"' -- #
if re.match(r"\"|\'", listCheckLine[-2]):
try:
if re.match(
r"\'|\"", listCheckLine[-3]
) and re.match(
r"\'|\"",
listCheckLine[-4]):
pass
else:
continue
except IndexError:
continue
# -- Check code into """ or ''' -- #
if re.match(quoteIntoVariable, eachLine):
checkQuotePassing += 1
continue
elif re.match(
quoteOfCommentariesMultipleLines,
eachLine
) or re.match(
quoteOfEndCommentariesMultipleLines,
eachLine):
checkQuotePassing += 1
if checkQuotePassing == 2:
checkQuotePassing = 0
continue
if checkQuotePassing == 1:
continue
elif checkQuotePassing == 2:
checkQuotePassing = 0
continue
else:
checkQuotePassing = 0
# -- Add scripts -- #
if re.match(noAddScript, eachLine) is not None:
continue
elif re.match(addIndentScript,
eachLine) is not None:
if re.match(checkAddIndentScript,
eachLine) is not None:
continue
else:
if spaces == 0:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
elif spaces == 4:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
elif spaces == 8:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
elif spaces == 12:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
else:
continue
else:
if spaces == 0:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg), ""))
countScriptsAdded += 1
elif spaces == 4:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg), " "))
countScriptsAdded += 1
elif spaces == 8:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
elif spaces == 12:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
else:
continue
# -- Check padding has added in output script -- #
for file in recursFiles:
if re.match(blockDirs, file):
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if not eachLine:
continue
checkLine += 1
countLineAdded = checkLine - countLine
if (self.remove.LineBreaks(codeArg, outputArg) == 0):
if checkLine > countLine:
print("\n-> {0} scripts added in {1} file(s)\n".format(
countScriptsAdded, countRecursFiles))
print("-> {0} lines added in {1} file(s)\n".format(
countLineAdded, countRecursFiles))
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
class Replace:
def __init__(self):
self.mixer = Mixer()
self.remove = Remove()
self.utils = Utils()
self.pythonExcludeWords = "exclude_python_words.txt"
self.pythonIncludeWords = "include_python_words.txt"
def EachLine(self, codeArg, eachLine, Dict):
getIndexLineList = []
returnLine = []
charValue = []
checkCharAfterWord = 1
wordSetter = 0
checkGetKey = ""
checkGetWord = ""
getLine = ""
breakLine = ""
if codeArg == "python":
regReplace = r"(\.)|(:)|(\))|(\()|(=)|(\[)|(\])|({)|(})|(,)|(\+)|(\s)|(\*)|(\+)|(\-)"
# -- Get list of all letters in line -- #
for indexLine, letterLine in enumerate(eachLine):
getIndexLineList.append(letterLine)
# -- Loop in each letter of line -- #
for indexLine, letterLine in enumerate(eachLine):
# -- 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(regReplace, 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 Dict:
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:
# -- Place index position after the word -- #
indexExplore = indexLine + len(key)
# -- If indexError return to next loop -- #
try:
getIndexLineList[indexExplore]
except IndexError:
continue
# -- Check the char after the word -- #
if re.match(regReplace, getIndexLineList[indexExplore]):
# -- Check word finded is not into the other word -- #
indexExplore = indexLine - 1
if not re.match(r"(\w)", getIndexLineList[indexExplore]):
if codeArg == "python":
# -- Check if it's 'from' and 'import' word key in line to avoid replace name of file if variable is identic name to file -- #
getLine = "".join(getIndexLineList)
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 = 0
else:
checkCharAfterWord = 0
else:
checkCharAfterWord = 1
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 VarsDefinedByUser(self, oneFileArg, codeArg, outputArg, mixerLevelArg):
variablesDict = {}
checkVarsMixed = []
wordsExcluded = []
wordsExcludedFound = []
wordsInclude = []
checkCountVarsMixed = 0
checkCountVarsValue = 0
checkPassing = 0
numberReplaced = 0
isCommentary = 0
noCommentary = 0
if codeArg == "python":
variablesDefined = r"(^\w+|\w+)([\s|^\s]*=[\s|\w|\"|\'])" # Classic variables
variablesErrorDefined = r"except(\s+\w+\sas\s)(\w)" # Error variables
variablesLoopDefined = r"for\s([\w\s?\,?]{1,})(\sin)" # Loop variables
functionsDefined = r"def\s(\w+)" # Functions
classDefined = r"class\s(\w+)" # Classes
quoteOfCommentariesOneLine = r"[\"\']{3}.*[\"\']{3}" # """ and ''' without before variables and if commentary is over one line, (""" commentaries """)
noQuoteOfCommentaries = r"\s*\w+\s*\={1}\s*[\"\']{3}" # """ and ''' with before variables
print("######## [ Variables-Classes-Functions ] ########\n")
######################################### One file only #########################################
if oneFileArg:
# -- Find variables - classes - functions and mixed it -- #
with open(outputArg, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
# -- Classic variables -- #
search = re.search(variablesDefined, eachLine)
if search != None:
# -- Detect mixer -- #
mixer = self.mixer.GetStringMixer(mixerLevelArg)
# -- Add the mixer value of variables -- #
if search.group(1) not in variablesDict:
variablesDict[search.group(1)] = mixer
# -- Error variables -- #
search = re.search(variablesErrorDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(2) not in variablesDict:
variablesDict[search.group(2)] = mixer
# -- Loop variables -- #
search = re.search(variablesLoopDefined, eachLine)
if search != None:
if "," in search.group(1):
varsInFor = []
modifySearch = search.group(1).replace(",", " ")
modifySearch = modifySearch.split()
for i in modifySearch:
if i not in variablesDict:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
variablesDict[i] = mixer
else:
if search.group(1) not in variablesDict:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
variablesDict[search.group(1)] = mixer
# -- Functions -- #
search = re.search(functionsDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(1) not in variablesDict:
if codeArg == "python":
if not re.match(r"(^__init__$)", search.group(1)):
variablesDict[search.group(1)] = mixer
# -- Classes -- #
search = re.search(classDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(1) not in variablesDict:
variablesDict[search.group(1)] = mixer
# -- Delete excluded variables - classes - functions from excluded_python_varaibles.txt in dict -- #
if os.path.exists(self.pythonExcludeWords) == True:
with open(self.pythonExcludeWords, "r") as readFile:
for word in readFile:
if "#" in word or word == "\n":
continue
else:
word = word.rstrip()
wordsExcluded.append(word)
else:
print("[-] File '{0}' not found\n".format(self.pythonExcludeWords))
for word in wordsExcluded:
if word in variablesDict.keys():
wordsExcludedFound.append(word)
del variablesDict[word]
# -- Include variables - classes - functions defined from include_python_words.txt in dict -- #
if os.path.exists(self.pythonIncludeWords) == True:
with open(self.pythonIncludeWords, "r") as readFile:
for word in readFile:
if "#" in word or word == "\n":
continue
else:
word = word.rstrip()
wordsInclude.append(word)
else:
print("[-] File '{0}' not found\n".format(self.pythonIncludeWords))
for word in wordsInclude:
if word not in variablesDict.keys():
mixer = self.mixer.GetStringMixer(mixerLevelArg)
variablesDict[word] = mixer
print("\n[+] Variables - Classes - Functions found :\n")
for key, value in variablesDict.items():
print("-> {0} : {1}".format(key, value))
for word in wordsExcludedFound:
print("-> {0} : excluded".format(word))
print("\n[+] Running replacement of Variables - Classes - Functions...\n")
# -- Change variables - classes - functions to mixed values -- #
with fileinput.input(outputArg, inplace=True) as inputFile:
for eachLine in inputFile:
if not eachLine:
continue
else:
if codeArg == "python":
# -- Check code in """ or ''' -- #
if "\"\"\"" in eachLine or "\'\'\'" in eachLine:
if re.match(quoteOfCommentariesOneLine, eachLine):
print(eachLine)
continue
elif re.match(noQuoteOfCommentaries, eachLine):
eachLine = Replace.EachLine(self, codeArg, eachLine, variablesDict.items())
print(eachLine)
checkPassing += 1
continue
else:
checkPassing += 1
if checkPassing == 1:
print(eachLine)
continue
else:
checkPassing = 0
eachLine = Replace.EachLine(self, codeArg, eachLine, variablesDict.items())
print(eachLine)
# -- Check if variables - classes - functions have been mixed -- #
with open(outputArg, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
for value in variablesDict.values():
if value in eachLine:
checkVarsMixed.append(value)
# -- Remove duplicated key -- #
checkListVarsMixed = list(dict.fromkeys(checkVarsMixed))
for i in checkListVarsMixed:
checkCountVarsMixed += 1
for i in variablesDict.values():
checkCountVarsValue += 1
if checkCountVarsMixed == checkCountVarsValue:
print("-> {0} variables - classes - functions replaced\n".format(checkCountVarsValue))
if (self.remove.LineBreaks(oneFileArg, codeArg, outputArg) == 0):
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
######################################### Multiple files #########################################
else:
if codeArg == "python":
inputExt = "py"
blockdirs = r"__pycache__"
recursFiles = [f for f in glob.glob("{0}{1}**{1}*.{2}".format(outputArg, self.utils.Platform(), inputExt), recursive=True)]
# -- Find variables - classes - functions and mixed it -- #
for output in recursFiles:
if re.match(blockdirs, output):
continue
else:
with open(output, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
# -- Classic variables -- #
search = re.search(variablesDefined, eachLine)
if search != None:
# -- Detect mixer -- #
mixer = self.mixer.GetStringMixer(mixerLevelArg)
# -- Add the mixer value of variables -- #
if search.group(1) not in variablesDict:
variablesDict[search.group(1)] = mixer
# -- Error variables -- #
search = re.search(variablesErrorDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(2) not in variablesDict:
variablesDict[search.group(2)] = mixer
# -- Loop variables -- #
search = re.search(variablesLoopDefined, eachLine)
if search != None:
if "," in search.group(1):
varsInFor = []
modifySearch = search.group(1).replace(",", " ")
modifySearch = modifySearch.split()
for i in modifySearch:
if i not in variablesDict:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
variablesDict[i] = mixer
else:
if search.group(1) not in variablesDict:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
variablesDict[search.group(1)] = mixer
# -- Functions -- #
search = re.search(functionsDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(1) not in variablesDict:
if codeArg == "python":
if not re.match(r"(^__init__$)", search.group(1)):
variablesDict[search.group(1)] = mixer
# -- Classes -- #
search = re.search(classDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(1) not in variablesDict:
variablesDict[search.group(1)] = mixer
# -- Remove excluded variables - classes - functions defined from excluded_python_words.txt in dict -- #
if os.path.exists(self.pythonExcludeWords) == True:
with open(self.pythonExcludeWords, "r") as readFile:
for word in readFile:
if "#" in word or word == "\n":
continue
else:
word = word.rstrip()
wordsExcluded.append(word)
else:
print("[-] File '{0}' not found\n".format(self.pythonExcludeWords))
for word in wordsExcluded:
if word in variablesDict.keys():
wordsExcludedFound.append(word)
del variablesDict[word]
# -- Include variables - classes - functions defined from include_python_words.txt in dict -- #
if os.path.exists(self.pythonIncludeWords) == True:
with open(self.pythonIncludeWords, "r") as readFile:
for word in readFile:
if "#" in word or word == "\n":
continue
else:
word = word.rstrip()
wordsInclude.append(word)
else:
print("[-] File '{0}' not found\n".format(self.pythonIncludeWords))
for word in wordsInclude:
if word not in variablesDict.keys():
mixer = self.mixer.GetStringMixer(mixerLevelArg)
variablesDict[word] = mixer
print("\n[+] Variables - Classes - Functions found :\n")
for key, value in variablesDict.items():
print("-> {0} : {1}".format(key, value))
for word in wordsExcludedFound:
print("-> {0} : excluded".format(word))
print("\n[+] Running replacement of Variables - Classes - Functions...\n")
# -- Change variables - classes - functions to mixed values -- #
for output in recursFiles:
if re.match(blockdirs, output):
continue
else:
with fileinput.input(output, inplace=True) as inputFile:
for eachLine in inputFile:
if not eachLine:
continue
else:
if codeArg == "python":
# -- Check code in """ or ''' -- #
if "\"\"\"" in eachLine or "\'\'\'" in eachLine:
if re.match(quoteOfCommentariesOneLine, eachLine):
print(eachLine)
continue
elif re.match(noQuoteOfCommentaries, eachLine):
eachLine = Replace.EachLine(self, codeArg, eachLine, variablesDict.items())
print(eachLine)
checkPassing += 1
continue
else:
checkPassing += 1
if checkPassing == 1:
print(eachLine)
continue
else:
checkPassing = 0
eachLine = Replace.EachLine(self, codeArg, eachLine, variablesDict.items())
print(eachLine)
# -- Check if variables - classes - functions have been mixed -- #
for output in recursFiles:
if re.match(blockdirs, output):
continue
else:
with open(output, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
for value in variablesDict.values():
if value in eachLine:
checkVarsMixed.append(value)
# -- Remove duplicated key -- #
checkListVarsMixed = list(dict.fromkeys(checkVarsMixed))
for i in checkListVarsMixed:
checkCountVarsMixed += 1
for i in variablesDict.values():
checkCountVarsValue += 1
if checkCountVarsMixed == checkCountVarsValue:
print("-> {0} variables - classes - functions replaced\n".format(checkCountVarsValue))
if (self.remove.LineBreaks(oneFileArg, codeArg, outputArg) == 0):
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
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 Analyze:
def __init__(self):
self.utils = Utils()
def InputAvailable(self, inputArg=None, verboseArg=None):
inputFileFound = []
inputFileEmpty = []
inputFileFoundCount = 0
inputFileEmptyCount = 0
countRecursFiles = 0
numberLine = 0
if os.path.exists(inputArg) == True:
if os.path.isdir(inputArg) == True:
detectFiles = "py"
blockDir = "__pycache__"
recursFiles = [
file for file in glob.glob("{0}{1}**{1}*.{2}".format(
inputArg, self.utils.Platform(), detectFiles),
recursive=True)
]
if recursFiles == []:
print(ERROR_COLOUR +
"[-] {0} directory empty or python file not found".
format(inputArg))
return ERROR_DIR_EMPTY
for number in recursFiles:
countRecursFiles += 1
print("\n[+] Running analyze input of {0} file(s)...\n".format(
countRecursFiles))
with Bar(PROGRESS_COLOUR + "Analysis ",
fill="=",
max=countRecursFiles,
suffix="%(percent)d%%") as bar:
for file in recursFiles:
if blockDir in file:
continue
else:
if os.path.getsize(file) > 0:
inputFileFound.append(file)
inputFileFoundCount += inputFileFoundCount + 1
else:
inputFileEmpty.append(file)
inputFileEmptyCount += inputFileEmptyCount + 1
bar.next(1)
if inputFileFoundCount >= 1 and inputFileEmptyCount < inputFileFoundCount:
if verboseArg:
print("\n\n[+] File input found :\n")
if inputFileFound == []:
print("-> no result")
else:
for file in inputFileFound:
print("-> {0}".format(file))
for file in inputFileEmpty:
print("-> {0} : empty".format(file))
return EXIT_SUCCESS
elif inputFileFoundCount == inputFileEmptyCount and inputFileFoundCount > 0:
print(
ERROR_COLOUR +
"[-] All files in directory specified are emtpy !")
return ERROR_FILE_EMPTY
else:
print(
ERROR_COLOUR +
"[-] No file available in '{0}'.".format(inputArg))
return ERROR_FILE_NOT_FOUND
bar.finish()
else:
print(ERROR_COLOUR +
"[-] '{0}' is not a directory".format(inputArg))
return ERROR_DIR_NOT_FOUND
else:
print(ERROR_COLOUR + "[-] '{0}' not exists".format(inputArg))
return ERROR_PATH_NOT_FOUND
def OutputAvailable(self, inputArg=None, outputArg=None, verboseArg=None):
outputFileFound = []
outputFileEmpty = []
outputFileFoundCount = 0
outputFileEmptyCount = 0
countRecursFiles = 0
if os.path.exists(outputArg) == True:
removeRequest = input(
"[!] Output '{0}' already exists, do you want remove it ? (Y/N) : "
.format(outputArg))
removeRequest = removeRequest.upper()
if removeRequest == "Y":
try:
shutil.rmtree(outputArg)
if os.path.exists(outputArg) == False:
shutil.copytree(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isdir(outputArg) == True:
detectFiles = "py"
blockDir = "__pycache__"
recursFiles = [
file for file in glob.glob(
"{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(),
detectFiles),
recursive=True)
]
for number in recursFiles:
countRecursFiles += 1
print(
"\n[+] Running analyze output of {0} file(s)...\n"
.format(countRecursFiles))
if recursFiles == []:
print(
ERROR_COLOUR +
"[-] {0} directory empty, no copied file"
.format(inputArg))
return ERROR_DIR_EMPTY
with Bar(PROGRESS_COLOUR + "Analysis ",
fill="=",
max=countRecursFiles,
suffix="%(percent)d%%") as bar:
for file in recursFiles:
if blockDir in file:
continue
else:
if os.path.getsize(file) > 0:
outputFileFound.append(file)
outputFileFoundCount += outputFileFoundCount + 1
else:
outputFileEmpty.append(file)
outputFileEmptyCount += outputFileEmptyCount + 1
bar.next(1)
if outputFileFoundCount >= 1 and outputFileFoundCount > outputFileEmptyCount:
if verboseArg:
print(
"\n\n[+] Output files copy :\n"
)
if outputFileFound == []:
print("-> no result")
else:
for file in outputFileFound:
print(
"-> {0}".format(file))
for file in outputFileEmpty:
print("-> {0} : empty".
format(file))
return EXIT_SUCCESS
else:
print(ERROR_COLOUR +
"[-] No files available in '{0}'"
.format(outputArg))
return ERROR_FILE_NOT_FOUND
bar.finish()
else:
print(
ERROR_COLOUR +
"[-] Copy '{0}' to '{1}' failed, this is not a output directory copied"
.format(inputArg, outputArg))
return ERROR_DIR_NOT_FOUND
else:
print(ERROR_COLOUR +
"[-] Copy '{0}' to '{1}' failed".format(
inputArg, outputArg))
return ERROR_CAN_NOT_COPY
else:
print(ERROR_COLOUR +
"[-] Remove '{0}' failed".format(outputArg))
return ERROR_CAN_NOT_REMOVE
except Exception as e:
print(ERROR_COLOUR + "[-] {0}".format(e))
return EXIT_FAILURE
else:
print(ERROR_COLOUR +
"[-] Remove '{0}' failed, the user has refused".format(
outputArg))
return ERROR_CAN_NOT_REMOVE
else:
try:
shutil.copytree(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isdir(outputArg) == True:
detectFiles = "py"
blockDir = "__pycache__"
recursFiles = [
file
for file in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFiles),
recursive=True)
]
for number in recursFiles:
countRecursFiles += 1
print(
"\n[+] Running analyze output of {0} file(s)...\n".
format(countRecursFiles))
if recursFiles == []:
print(ERROR_COLOUR +
"[-] {0} directory empty, no copied file".
format(inputArg))
return ERROR_DIR_EMPTY
with Bar(PROGRESS_COLOUR + "Analysis ",
fill="=",
max=countRecursFiles,
suffix="%(percent)d%%") as bar:
for file in recursFiles:
if blockDir in file:
continue
else:
if os.path.getsize(file) > 0:
outputFileFound.append(file)
outputFileFoundCount += outputFileFoundCount + 1
else:
outputFileEmpty.append(file)
outputFileEmptyCount += outputFileEmptyCount + 1
bar.next(1)
if outputFileFoundCount >= 1 and outputFileFoundCount > outputFileEmptyCount:
if verboseArg:
print("\n\n[+] File output found :\n")
if outputFileFound == []:
print("-> no result")
else:
for file in outputFileFound:
print("-> {0} : copy".format(file))
for file in outputFileEmpty:
print("-> {0} : copy empty".format(
file))
return EXIT_SUCCESS
else:
print(ERROR_COLOUR +
"[-] No files available in '{0}'".format(
outputArg))
return ERROR_FILE_NOT_FOUND
bar.finish()
else:
print(
ERROR_COLOUR +
"[-] Copy '{0}' to '{1}' failed, this is not a output directory copied !"
.format(inputArg, outputArg))
return ERROR_DIR_NOT_FOUND
else:
print(ERROR_COLOUR +
"[-] Copy '{0}' to '{1}' failed".format(
inputArg, outputArg))
return ERROR_CAN_NOT_COPY
except Exception as e:
print(ERROR_COLOUR + "[-] {0}".format(e))
return EXIT_FAILURE
class Analyze:
def __init__(self):
self.utils = Utils()
def InputAvailable(self, oneFileArg, inputArg, codeArg):
inputFileFound = []
inputFileEmpty = []
inputFileFoundCount = 0
inputFileEmptyCount = 0
inputExt = ""
######################################### One file only #########################################
if oneFileArg == True:
if os.path.exists(inputArg) == True:
if os.path.isfile(inputArg) == True:
if codeArg == "python":
if ".py" in inputArg:
if os.path.getsize(inputArg) > 0:
return EXIT_SUCCESS
else:
print("[-] '{0}' is empty.".format(inputArg))
return ERROR_FILE_EMPTY
else:
print("[-] '{0}' is not python file.".format(
inputArg))
return ERROR_BAD_FILE_TYPE
else:
print("[-] '{0}' is not a file.".format(inputArg))
return ERROR_BAD_FILE_TYPE
else:
print("[-] '{0}' not exists.".format(inputArg))
return ERROR_PATH_NOT_FOUND
######################################### Multiple files #########################################
else:
if os.path.exists(inputArg) == True:
if os.path.isdir(inputArg) == True:
if codeArg == "python":
inputExt = "py"
blockdirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
inputArg, self.utils.Platform(), inputExt),
recursive=True)
]
for output in recursFiles:
if re.match(blockdirs, output):
continue
else:
if os.path.getsize(output) > 0:
inputFileFound.append(output)
inputFileFoundCount += inputFileFoundCount + 1
else:
inputFileEmpty.append(output)
inputFileEmptyCount += inputFileEmptyCount + 1
if inputFileFoundCount >= 2 and inputFileEmptyCount < 2:
return EXIT_SUCCESS
elif inputFileFoundCount == 1:
print(
"[-] Only one file available found in '{0}', select [-f --onefile] argument."
.format(inputArg))
return ERROR_BAD_ARGUMENTS
elif inputFileFoundCount == inputFileEmptyCount and inputFileEmptyCount > 0 and inputFileEmptyCount > 0:
print("[-] All files output specified are emtpy !.")
return ERROR_FILE_EMPTY
else:
print(
"[-] No file available in '{0}'.".format(inputArg))
return ERROR_FILE_NOT_FOUND
else:
print("[-] '{0}' is not a directory.".format(inputArg))
return ERROR_NOT_DIR
else:
print("'{0}' not exists.".format(inputArg))
return ERROR_PATH_NOT_FOUND
def OutputAvailable(self, oneFileArg, inputArg, codeArg, outputArg):
outputFileFound = []
outputFileEmpty = []
outputFileFoundCount = 0
outputFileEmptyCount = 0
outputExt = ""
if os.path.exists(outputArg) == True:
removeRequest = input(
"[!] Output '{0}' already exists, do you want remove it ? (Y/N) : "
.format(outputArg))
removeRequest = removeRequest.upper()
if removeRequest == "Y":
######################################### One file only #########################################
if oneFileArg == True:
try:
os.remove(outputArg)
if os.path.exists(outputArg) == False:
shutil.copy(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isfile(outputArg) == True:
if os.path.getsize(outputArg) > 0:
return EXIT_SUCCESS
else:
print(
"[-] Copy '{0}' to '{1}' failed, the files is empty !."
.format(inputArg, outputArg))
return ERROR_FILE_EMPTY
else:
print(
"[-] Copy '{0}' to '{1}' failed, this is not a output file copy !."
.format(inputArg, outputArg))
return ERROR_NOT_FILE
else:
print("[-] Copy '{0}' to '{1}' failed.".format(
inputArg, outputArg))
return ERROR_CANNOT_COPY
else:
print("[-] Remove '{0}' failed.".format(outputArg))
return ERROR_CANNOT_REMOVE
except Exception as e:
print("[-] {0}".format(e))
return EXIT_FAILURE
######################################### Multiple files #########################################
else:
try:
shutil.rmtree(outputArg)
if os.path.exists(outputArg) == False:
shutil.copytree(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isdir(outputArg) == True:
if codeArg == "python":
inputExt = "py"
blockdirs = r"__pycache__"
recursFiles = [
f for f in
glob.glob("{0}{1}**{1}*.{2}".format(
inputArg, self.utils.Platform(),
inputExt),
recursive=True)
]
for output in recursFiles:
if re.match(blockdirs, output):
continue
else:
if os.path.getsize(output) > 0:
outputFileFound.append(output)
outputFileFoundCount += outputFileFoundCount + 1
else:
outputFileEmpty.append(output)
outputFileEmptyCount += outputFileEmptyCount + 1
if outputFileFoundCount >= 2 and outputFileEmptyCount < 2:
return EXIT_SUCCESS
else:
print(
"[-] No files available in '{0}'.".
format(outputArg))
return ERROR_FILE_NOT_FOUND
else:
print(
"[-] Copy '{0}' to '{1}' failed, this is not a output directory copied !."
.format(inputArg, outputArg))
return ERROR_NOT_DIR
else:
print("[-] Copy '{0}' to '{1}' failed.".format(
inputArg, outputArg))
return ERROR_CANNOT_COPY
else:
print("[-] Remove '{0}' failed.".format(outputArg))
return ERROR_CANNOT_REMOVE
except Exception as e:
print("[-] {0}".format(e))
return EXIT_FAILURE
else:
print("[-] Remove '{0}' failed, the user has refused.".format(
outputArg))
return ERROR_CANNOT_REMOVE
else:
######################################### One file only #########################################
if oneFileArg == True:
try:
shutil.copy(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isfile(outputArg) == True:
if os.path.getsize(outputArg) > 0:
return EXIT_SUCCESS
else:
print(
"[-] Copy '{0}' to '{1}' failed, the file is empty !."
.format(inputArg, outputArg))
return ERROR_CANNOT_COPY
else:
print(
"[-] Copy '{0}' to '{1}' failed, this is not a output file copied !."
.format(inputArg, outputArg))
return ERROR_NOT_FILE
else:
print("[-] Copy '{0}' to '{1}' failed.".format(
inputArg, outputArg))
return ERROR_CANNOT_COPY
except Exception as e:
print("[-] {0}".format(e))
return EXIT_FAILURE
######################################### Multiple files #########################################
else:
try:
shutil.copytree(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isdir(outputArg) == True:
if codeArg == "python":
inputExt = "py"
blockdirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
inputArg, self.utils.Platform(), inputExt),
recursive=True)
]
for output in recursFiles:
if re.match(blockdirs, output):
continue
else:
if os.path.getsize(output) > 0:
outputFileFound.append(output)
outputFileFoundCount += outputFileFoundCount + 1
else:
outputFileEmpty.append(output)
outputFileEmptyCount += outputFileEmptyCount + 1
if outputFileFoundCount >= 2 and outputFileEmptyCount < 2:
return EXIT_SUCCESS
else:
print(
"[-] No files available in '{0}'.".format(
outputArg))
return ERROR_FILE_NOT_FOUND
else:
print(
"[-] Copy '{0}' to '{1}' failed, this is not a output directory copied !."
.format(inputArg, outputArg))
return ERROR_NOT_DIR
else:
print("[-] Copy '{0}' to '{1}' failed.".format(
inputArg, outputArg))
return ERROR_CANNOT_COPY
except Exception as e:
print("[-] {0}".format(e))
return EXIT_FAILURE
class Remove:
def __init__(self):
self.utils = Utils()
def Backslashes(self, codeArg, outputArg):
checkLine = 0
numberFiles = 0
numberGoodFile = 0
if codeArg == "python":
detectFiles = "py"
blockDir = "__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFiles),
recursive=True)
]
# -- Delete line breaks -- #
for file in recursFiles:
if blockDir in file:
continue
else:
with fileinput.FileInput(file, inplace=True) as inputFile:
for line in inputFile:
if line.strip():
print(line.rstrip())
# -- Check if all line breaks are deleted -- #
for file in recursFiles:
checkLine = 0 # Initialize check vars for the next file
if blockDir in file:
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if eachLine == "\n":
checkLine += 1
if checkLine == 0:
numberGoodFile += 1
numberFiles += 1
else:
numberGoodFile += 0
numberFiles += 1
if numberGoodFile == numberFiles:
return EXIT_SUCCESS
else:
return EXIT_FAILURE
def Commentaries(self, codeArg, outputArg):
countLineOutput = 0
countLineInput = 0
noCommentary = 0
isCommentary = 0
countRecursFiles = 0
if codeArg == "python":
detectFiles = "py"
blockDir = "__pycache__"
commentariesBeginLine = r"^\#.*" # Begin '#'
quoteOfCommentariesMultipleLines = r"^\s*[\"|\']{3}$" # """ and ''' without before variables and if commentaries is over multiple lines
quoteInRegex = r"\={1}\s*r[\"|\']{1}" # If quote in regex
quoteOfEndCommentariesMultipleLines = r"^\s*[\"|\']{3}\)?\.?" # """ and ''' without before variables, if commentaries is over multiple lines and he finish by .format() funtion
quoteOfCommentariesOneLine = r"[\"|\']{3}.*[\"|\']{3}$" # """ and ''' without before variables and if commentary is over one line, (""" commentaries """)
quoteIntoVariable = r".*\={1}\s*\w*\.?\w*[\(|\.]{1}[\"|\']{3}|.*\={1}\s*[\"|\']{3}" # """ and ''' with before variables
commentariesAfterLine = r"\s*\#[^\"|^\'|^\.|^\?|^\*|^\!|^\]|^\[|^\\|^\)|^\(|^\{|^\}].*" # '#' after line of code
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFiles),
recursive=True)
]
# -- Remove commentaries and Count commentaries will be removed -- #
for number in recursFiles:
countRecursFiles += 1
print("\n[+] Running remove commentaries in {0} file(s)...\n".format(
countRecursFiles))
with tqdm.tqdm(total=countRecursFiles) as pbar:
for file in recursFiles:
pbar.update(1)
if blockDir in file:
continue
else:
# -- Remove commentaries -- #
with fileinput.input(file, inplace=True) as inputFile:
for eachLine in inputFile:
searchCommentariesAfterLine = re.search(
commentariesAfterLine, eachLine)
searchCommentariesBeginLine = re.search(
commentariesBeginLine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
print(eachLine)
continue
if re.match(quoteInRegex, eachLine):
continue
elif re.match(quoteIntoVariable, eachLine):
noCommentary += 1
elif re.match(
quoteOfCommentariesMultipleLines,
eachLine
) or re.match(
quoteOfEndCommentariesMultipleLines,
eachLine):
isCommentary += 1
else:
pass
if re.match(quoteOfCommentariesOneLine,
eachLine):
countLineInput += 1
isCommentary = 0
continue
elif isCommentary == 1 and noCommentary == 0:
countLineInput += 1
continue
elif isCommentary == 0 and noCommentary == 1:
print(eachLine)
continue
elif isCommentary == 2:
countLineInput += 1
isCommentary = 0
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
print(eachLine)
continue
else:
pass
if searchCommentariesBeginLine is not None:
countLineInput += 1
eachLine = eachLine.replace(
searchCommentariesBeginLine.group(0), "")
print(eachLine)
elif searchCommentariesAfterLine is not None:
eachLine = eachLine.replace(
searchCommentariesAfterLine.group(0), "")
countLineInput += 1
print(eachLine)
else:
print(eachLine)
# -- Initialize vars -- #
isCommentary = 0
noCommentary = 0
# -- Check if all commentaries are removed -- #
for file in recursFiles:
countLineOutput = 0
if blockDir in file:
continue
else:
with open(file, "r") as readFile:
countLineOutput = 0
readF = readFile.readlines()
for eachLine in readF:
searchCommentariesAfterLine = re.search(
commentariesAfterLine, eachLine)
searchCommentariesBeginLine = re.search(
commentariesBeginLine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
continue
if re.match(quoteInRegex, eachLine):
continue
elif re.match(quoteIntoVariable, eachLine):
noCommentary += 1
elif re.match(
quoteOfCommentariesMultipleLines,
eachLine) or re.match(
quoteOfEndCommentariesMultipleLines,
eachLine):
isCommentary += 1
else:
pass
if re.match(quoteOfCommentariesOneLine, eachLine):
isCommentary = 0
countLineOutput += 1
continue
elif isCommentary == 1 and noCommentary == 0:
countLineOutput += 1
continue
elif isCommentary == 0 and noCommentary == 1:
continue
elif isCommentary == 2:
isCommentary = 0
countLineOutput += 1
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
continue
else:
pass
if searchCommentariesBeginLine is not None:
countLineOutput += 1
elif searchCommentariesAfterLine is not None:
countLineOutput += 1
else:
pass
if (Remove.Backslashes(self, codeArg, outputArg) == 0):
if countLineOutput == 0:
print("\n-> {0} lines of commentaries removed\n".format(
countLineInput))
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
def PrintFunctions(self, codeArg, outputArg):
countPrintLine = 0
countCheckPrintLine = 0
countRecursFiles = 0
checkPrintPy3MultipleLines = 0
checkPrintPy2MultipleLines = 0
if codeArg == "python":
detectFiles = "py"
blockDir = "__pycache__"
detectPrint = r"\s*print"
detectPythonPrint2 = r"\s*print\s*[\"|\']{1}"
detectPythonPrint3 = r"\s*print\s*\({1}"
detectPythonPrintMultipleLines = r"^\s+[\"\']{1}\s*\w+|^[\"\']{1}\s*\w+"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFiles),
recursive=True)
]
for number in recursFiles:
countRecursFiles += 1
print("\n[+] Running remove print function in {0} file(s)...\n".format(
countRecursFiles))
with tqdm.tqdm(total=countRecursFiles) as pbar:
for file in recursFiles:
pbar.update(1)
if blockDir in file:
continue
else:
# -- Remove all print functions -- #
with fileinput.input(file, inplace=True) as inputFile:
for eachLine in inputFile:
if re.match(detectPrint, eachLine):
countPrintLine += 1
# -- If print() python 3 is multiple lines -- #
if re.match(detectPythonPrint3, eachLine):
if "(" in eachLine and not ")" in eachLine:
checkPrintPy3MultipleLines += 1
continue
else:
continue
# -- If print python 2 is multiple lines -- #
elif re.match(detectPythonPrint2, eachLine):
checkPrintPy2MultipleLines += 1
continue
else:
if checkPrintPy3MultipleLines == 1:
if ")" in eachLine and not "(" in eachLine:
checkPrintPy3MultipleLines = 0
continue
else:
continue
elif checkPrintPy2MultipleLines > 0:
if re.match(detectPythonPrintMultipleLines,
eachLine):
checkPrintPy2MultipleLines += 1
continue
else:
checkPrintPy2MultipleLines = 0
print(eachLine)
continue
else:
print(eachLine)
# -- Check if all print functions are removed -- #
for file in recursFiles:
if blockDir in file:
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if re.match(detectPrint, eachLine):
countCheckPrintLine += 1
if (Remove.Backslashes(self, codeArg, outputArg) == 0):
if countCheckPrintLine == 0:
print("\n-> {0} print functions removed\n".format(
countPrintLine))
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
def TrashFiles(self, codeArg, outputArg):
removeFiles = 0
if codeArg == "python":
detectFiles = "pyc"
blockDir = "__pycache__"
detectFilesPycFile = r"\w+\.{1}pyc"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFiles),
recursive=True)
]
try:
# -- Remove pyc file(s)
for file in recursFiles:
os.remove(file)
removeFiles += removeFiles + 1
# -- Check if pyc file(s) are removed
for file in recursFiles:
if re.match(detectFilesPycFile, file):
return EXIT_FAILURE
return removeFiles
except Exception as e:
print(ERROR_COLOUR + "[-] {0}".format(e))
return EXIT_FAILURE
class Remove:
def __init__(self):
self.utils = Utils()
def LineBreaks(self, oneFileArg, codeArg, outputArg):
checkLine = 0
numberFiles = 0
numberFileGood = 0
######################################### One file only #########################################
if oneFileArg:
# -- Delete line breaks -- #
with fileinput.FileInput(outputArg, inplace=True) as inputFile:
for line in inputFile:
if line.strip():
print(line.rstrip())
# -- Check if all line breaks are deleted -- #
with open(outputArg, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if eachLine == "\n":
checkLine += 1
if checkLine == 0:
return EXIT_SUCCESS
else:
return EXIT_FAILURE
######################################### Multiple files #########################################
else:
if codeArg == "python":
inputExt = "py"
blockdirs = r"__pycache__"
recursFiles = [f for f in glob.glob("{0}{1}**{1}*.{2}".format(outputArg, self.utils.Platform(), inputExt), recursive=True)]
# -- Delete line breaks -- #
for output in recursFiles:
if re.match(blockdirs, output):
continue
else:
with fileinput.FileInput(output, inplace=True) as inputFile:
for line in inputFile:
if line.strip():
print(line.rstrip())
# -- Check if all line breaks are deleted -- #
for output in recursFiles:
checkLine = 0 # Initialize check vars for the next file
if re.match(blockdirs, output):
continue
else:
with open(output, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if eachLine == "\n":
checkLine += 1
if checkLine == 0:
numberFileGood += 1
numberFiles += 1
else:
numberFileGood += 0
numberFiles += 1
if numberFileGood == numberFiles:
return EXIT_SUCCESS
else:
return EXIT_FAILURE
def Commentaries(self, oneFileArg, codeArg, outputArg):
countLineOutput = 0
countLineInput = 0
noCommentary = 0
isCommentary = 0
if codeArg == "python":
classicCommentariesDefine = r"\#+.*" # begin '#' and begin code then '#'
quoteOfCommentariesMultipleLine = r"^[\"\']+" # """ and ''' without before variables and if commentaries is over multiple lines
quoteOfCommentariesOneLine = r"[\"\']{3}.*[\"\']{3}" # """ and ''' without before variables and if commentary is over one line, (""" commentaries """)
noQuoteOfCommentaries = r"\w+\s*\={1}\s*[\"\']{3}" # """ and ''' with before variables
print("############### [ Commentaries ] ###############\n")
print("\n[+] Running commentaries remove...\n")
######################################### One file only #########################################
if oneFileArg:
# -- Count commentaries will be removed -- #
with open(outputArg, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
search = re.search(classicCommentariesDefine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
continue
if re.match(noQuoteOfCommentaries, eachLine):
noCommentary += 1
elif re.match(quoteOfCommentariesMultipleLine, eachLine):
isCommentary += 1
else:
pass
if isCommentary == 1 and noCommentary == 0:
if re.match(quoteOfCommentariesOneLine, eachLine):
isCommentary = 0
countLineInput += 1
continue
countLineInput += 1
continue
elif isCommentary == 0 and noCommentary == 1:
continue
elif isCommentary == 2:
isCommentary = 0
countLineInput += 1
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
continue
else:
pass
if search is not None:
countLineInput += 1
# -- Remove commentaries -- #
with fileinput.input(outputArg, inplace=True) as inputFile:
for eachLine in inputFile:
search = re.search(classicCommentariesDefine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
print(eachLine)
continue
if re.match(noQuoteOfCommentaries, eachLine):
noCommentary += 1
elif re.match(quoteOfCommentariesMultipleLine, eachLine):
isCommentary += 1
else:
pass
if isCommentary == 1 and noCommentary == 0:
if re.match(quoteOfCommentariesOneLine, eachLine):
isCommentary = 0
continue
continue
elif isCommentary == 0 and noCommentary == 1:
print(eachLine)
continue
elif isCommentary == 2:
isCommentary = 0
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
print(eachLine)
continue
else:
pass
if search is not None:
eachLine = eachLine.replace(search.group(0), "")
print(eachLine)
else:
print(eachLine)
# -- Check if all commentaries are removed -- #
with open(outputArg, "r") as readFile:
countLineOutput = 0
readF = readFile.readlines()
for eachLine in readF:
search = re.search(classicCommentariesDefine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
continue
if re.match(noQuoteOfCommentaries, eachLine):
noCommentary += 1
elif re.match(quoteOfCommentariesMultipleLine, eachLine):
isCommentary += 1
else:
pass
if isCommentary == 1 and noCommentary == 0:
if re.match(quoteOfCommentariesOneLine, eachLine):
isCommentary = 0
countLineOutput += 1
continue
countLineOutput += 1
continue
elif isCommentary == 0 and noCommentary == 1:
continue
elif isCommentary == 2:
isCommentary = 0
countLineOutput += 1
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
continue
else:
pass
if search is not None:
countLineOutput += 1
if countLineOutput == 0:
print("-> {0} lines of commentaries removed\n".format(countLineInput))
if (Remove.LineBreaks(self, oneFileArg, codeArg, outputArg) == 0):
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
######################################### Multiple files #########################################
else:
if codeArg == "python":
inputExt = "py"
blockdirs = r"__pycache__"
recursFiles = [f for f in glob.glob("{0}{1}**{1}*.{2}".format(outputArg, self.utils.Platform(), inputExt), recursive=True)]
# -- Count commentaries will be removed -- #
for output in recursFiles:
if re.match(blockdirs, output):
continue
else:
with open(output, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
search = re.search(classicCommentariesDefine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
continue
if re.match(noQuoteOfCommentaries, eachLine):
noCommentary += 1
elif re.match(quoteOfCommentariesMultipleLine, eachLine):
isCommentary += 1
else:
pass
if isCommentary == 1 and noCommentary == 0:
if re.match(quoteOfCommentariesOneLine, eachLine):
isCommentary = 0
countLineInput += 1
continue
countLineInput += 1
continue
elif isCommentary == 0 and noCommentary == 1:
continue
elif isCommentary == 2:
isCommentary = 0
countLineInput += 1
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
continue
else:
pass
if search is not None:
countLineInput += 1
# -- Remove commentaries -- #
for output in recursFiles:
if re.match(blockdirs, output):
continue
else:
# -- Remove commentaries -- #
with fileinput.input(output, inplace=True) as inputFile:
for eachLine in inputFile:
search = re.search(classicCommentariesDefine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
print(eachLine.rstrip())
continue
if re.match(noQuoteOfCommentaries, eachLine):
noCommentary += 1
elif re.match(quoteOfCommentariesMultipleLine, eachLine):
isCommentary += 1
else:
pass
if isCommentary == 1 and noCommentary == 0:
if re.match(quoteOfCommentariesOneLine, eachLine):
isCommentary = 0
continue
continue
elif isCommentary == 0 and noCommentary == 1:
print(eachLine)
continue
elif isCommentary == 2:
isCommentary = 0
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
print(eachLine)
continue
else:
pass
if search is not None:
eachLine = eachLine.replace(search.group(0), "") # Util if line containt code and then '#'. (currently commentary)
print(eachLine)
else:
print(eachLine)
# -- Check if all commentaries are removed -- #
for output in recursFiles:
countLineOutput = 0
if re.match(blockdirs, output):
continue
else:
with open(output, "r") as readFile:
countLineOutput = 0
readF = readFile.readlines()
for eachLine in readF:
search = re.search(classicCommentariesDefine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
continue
if re.match(noQuoteOfCommentaries, eachLine):
noCommentary += 1
elif re.match(quoteOfCommentariesMultipleLine, eachLine):
isCommentary += 1
else:
pass
if isCommentary == 1 and noCommentary == 0:
if re.match(quoteOfCommentariesOneLine, eachLine):
isCommentary = 0
countLineOutput += 1
continue
countLineOutput += 1
continue
elif isCommentary == 0 and noCommentary == 1:
continue
elif isCommentary == 2:
isCommentary = 0
countLineOutput += 1
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
continue
else:
pass
if search is not None:
countLineOutput += 1
if countLineOutput == 0:
print("-> {0} lines of commentaries removed\n".format(countLineInput))
if (Remove.LineBreaks(self, oneFileArg, codeArg, outputArg) == 0):
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
def StringsToHex(self, codeArg, outputArg, mixerLevelArg):
getLetterLineList = []
countRecursFiles = 0
checkPrint = 0
checkHexError = False
checkPrintError = False
hexLine = ""
utils = Utils()
mixer = Mixer()
remove = Remove()
if codeArg == "python":
detectFiles = "py"
blockDir = "__pycache__"
detectExecFunc = r"exec\(\w+\)"
detectQuotes = r"[\'|\"]{3}"
recursFiles = [f for f in glob.glob("{0}{1}**{1}*.{2}".format(outputArg, utils.Platform(), detectFiles), recursive=True)]
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 tqdm.tqdm(total=countRecursFiles) as pbar:
for file in recursFiles:
checkPrint = 0 # initialize check print() func at the begining of each file
pbar.update(1)
if blockDir in file:
continue
else:
# -- Add a new first random line and move the old first line to the second line to avoid replacing it -- #
with open(file, "r") as inputFile:
stringRandomMixer = mixer.GetStringMixer(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 = mixer.GetStringMixer(mixerLevelArg)
print(varMixer + "=\"\"\"")
checkPrint = 1
else:
getLetterLineList = [] # initialize list
for letterLine in eachLine:
letterToHex = "\\x" + str(letterLine.encode().hex())
getLetterLineList.append(letterToHex) # Get list of all letters in line
hexLine = "".join(getLetterLineList)
print(hexLine)
# -- Add exec funtions to interpret hex code in strings -- #
with open(file, "a") as inputFile:
inputFile.write("\"\"\"")
inputFile.write("\nexec({0})".format(varMixer))
# -- Check if all lines are replaced of hexadecimal value -- #
for file in recursFiles:
if blockDir in file:
continue
else:
with open(file, "r") as inputFile:
for eachLine in inputFile:
if not eachLine:
continue
else:
if not "\\x" in eachLine:
if re.match(detectQuotes, eachLine):
continue
elif re.match(detectExecFunc, eachLine):
continue
else:
checkError = True
else:
continue
if (remove.Backslashes(codeArg, outputArg) == 0):
if checkHexError == False:
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
def FilesName(self, codeArg, outputArg, mixerLevelArg, verboseArg):
filesNameDict = {}
filesNameDictNoExt = {}
filesNameFound = []
filesNameFoundNoExt = []
filesNameMixed = []
checkFilesFound = []
countRecursFiles = 0
checkRenameInDir = 0
checkRenameInCode = 0
unix = False
win = False
currentPosition = os.getcwd()
utils = Utils()
mixer = Mixer()
remove = Remove()
if codeArg == "python":
detectFiles = "py"
blockDir = "__pycache__"
blockFile = "__init__"
detectImport = r"\s*from\s+|\s*import\s+"
recursFiles = [f for f in glob.glob("{0}{1}**{1}*.{2}".format(outputArg, utils.Platform(), detectFiles), recursive=True)]
for number in recursFiles:
countRecursFiles += 1
for file in recursFiles:
if blockDir in file or blockFile in file:
continue
else:
if "\"" in file:
parseFilePath = file.split("\"")
win = True
else:
parseFilePath = file.split("/")
unix = True
checkFilesFound.append(file)
mixer = self.mixer.GetStringMixer(mixerLevelArg)
filesNameDict[parseFilePath[-1]] = mixer + ".py"
filesNameDictNoExt[parseFilePath[-1].replace(".py", "")] = mixer
filesNameFound.append(parseFilePath[-1])
filesNameMixed.append(mixer + ".py")
removeExt = parseFilePath[-1].replace(".py","")
filesNameFoundNoExt.append(removeExt)
# -- Diplay all files found with their mixed values if verbose arg is actived -- #
if verboseArg:
print("\n[+] Files name found with their mixed values :\n")
for key, value in filesNameDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Running replace files name in {0} file(s)...\n".format(countRecursFiles))
# -- Replace all files name to random strings with length defined -- #
with tqdm.tqdm(total=countRecursFiles) as pbar:
for file in recursFiles:
pbar.update(1)
if blockDir in file:
continue
else:
# -- Rename all files in python code -- #
with fileinput.input(file, inplace=True) as inputFile:
for eachLine in inputFile:
try:
for fileName in filesNameFound:
for fileNameNoExt in filesNameFoundNoExt:
if fileName in eachLine or fileNameNoExt in eachLine:
if not ".py" in eachLine:
if re.match(detectImport, eachLine):
eachLine = Replace.EachLine(self, codeArg, eachLine, filesNameDictNoExt.items(), True)
print(eachLine)
raise BreakLoop
else:
continue
else:
eachLine = Replace.EachLine(self, codeArg, eachLine, filesNameDict.items(), True)
print(eachLine)
raise BreakLoop
else:
continue
print(eachLine)
except BreakLoop:
continue
# -- 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) # Return to old position
# -- Check if all files name are been replaced to random strings with length defined -- #
for i in checkFilesFound:
i.replace(".py", "")
# -- Check for file(s) name -- #
for file in recursFiles:
if blockDir in file or blockFile in file:
continue
else:
for i in checkFilesFound:
if i in file:
checkFilesFound.remove(i)
continue
# -- Check for code if file(s) name -- #
if checkFilesFound != []:
for file in recursFiles:
if blockDir in file or blockFile in file:
continue
else:
with open(file, "r") as inputFile:
for line in inputFile:
for i in checkFilesFound:
if i in line:
checkFilesFound.remove(i)
continue
if checkFilesFound == []:
if (remove.Backslashes(codeArg, outputArg) == 0):
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
class Replace:
def __init__(self):
self.mixer = Mixer()
self.remove = Remove()
self.utils = Utils()
self.pythonExcludeWords = "exclude/python/exclude_python_words.txt"
self.pythonIncludeWords = "include/python/include_python_words.txt"
def EachLine(self, codeArg, eachLine, Dict, forFilesName):
getIndexLineList = []
returnLine = []
charValue = []
checkCharAfterWord = 1
wordSetter = 0
checkGetKey = ""
checkGetWord = ""
getLine = ""
breakLine = ""
if codeArg == "python":
regReplace = r"\.|\:|\)|\(|\=|\[|\]|\{|\}|\,|\+|\s|\*|\-"
# -- Get list of all letters in line -- #
for indexLine, letterLine in enumerate(eachLine):
getIndexLineList.append(letterLine)
# -- Loop in each letter of line -- #
for indexLine, letterLine in enumerate(eachLine):
# -- 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(regReplace, 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 Dict:
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 -- #s
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(regReplace, getIndexLineList[indexExplore]):
indexExploreBefore = indexLine - 1 # Index check if word finded is not into the other word
indexExploreAfter = indexLine + 2 # Index check char after the the char finded with regReplace regex
if codeArg == "python":
try:
if not re.match(r"\w|\\", getIndexLineList[indexExploreBefore]):
# -- Check if it's 'from' and 'import' word key in line to avoid replace name of file if variable is identic name to file -- #
getLine = "".join(getIndexLineList)
if forFilesName == 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 = 0
# -- Check if after char find by 'regReplace' variable there no is ' or " -- #
elif re.match(r"\"|\'", getIndexLineList[indexExploreAfter]):
checkCharAfterWord = 1
else:
checkCharAfterWord = 0
# -- Only for -rfn, --replacefilesname feature -- #
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 StringsToStrings(self, codeArg, outputArg, mixerLevelArg, verboseArg):
variablesDict = {}
classesDict = {}
functionsDict = {}
includeDict = {}
allDict = {}
checkWordsMixed = []
wordsExcluded = []
wordsExcludedFound = []
wordsIncludedFound = []
wordsIncludedNotFound = []
checkCountWordsMixed = 0
checkCountWordsValue = 0
checkQuotePassing = 0
numberReplaced = 0
countRecursFiles = 0
if codeArg == "python":
detectFiles = "py"
blockDir = "__pycache__"
blockFile = "__init__"
functionsDefined = r"def\s(\w+)" # Functions
classDefined = r"class\s(\w+)" # Classes
variablesErrorDefined = r"except(\s+\w+\sas\s)(\w)" # Error variables
variablesLoopDefined = r"for\s+([\w\s\,]+)(\s+in\s+)" # Loop variables
variablesDefined = r"(^\w+|\w+)([\s|^\s]*=[\s|\w|\"|\'])" # Variables
quoteOfCommentariesMultipleLines = r"^\s*[\"|\']{3}$" # """ and ''' without before variables and if commentaries is over multiple lines
quoteInRegex = r"\={1}\s*r[\"|\']{1}" # If quote in regex
quoteOfEndCommentariesMultipleLines = r"^\s*[\"|\']{3}\)?\.?" # """ and ''' without before variables, if commentaries is over multiple lines and he finish by .format() funtion
quoteIntoVariable = r".*\={1}\s*\w*\.?\w*[\(|\.]{1}[\"|\']{3}|.*\={1}\s*[\"|\']{3}" # """ and ''' with before variables
recursFiles = [f for f in glob.glob("{0}{1}**{1}*.{2}".format(outputArg, self.utils.Platform(), detectFiles), recursive=True)]
# -- Replace variables/classes/functions to random strings with length defined -- #
for file in recursFiles:
if blockDir in file:
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
# -- Variables -- #
search = re.search(variablesDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(1) not in variablesDict:
variablesDict[search.group(1)] = mixer
# -- Error variables -- #
search = re.search(variablesErrorDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(2) not in variablesDict:
variablesDict[search.group(2)] = mixer
# -- Loop variables -- #
search = re.search(variablesLoopDefined, eachLine)
if search != None:
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(mixerLevelArg)
variablesDict[i] = mixer
else:
if search.group(1) not in variablesDict:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
variablesDict[search.group(1)] = mixer
# -- Functions -- #
search = re.search(functionsDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(1) not in functionsDict:
if codeArg == "python":
if not blockFile in search.group(1):
functionsDict[search.group(1)] = mixer
# -- Classes -- #
search = re.search(classDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(1) not in classesDict:
classesDict[search.group(1)] = mixer
# -- Remove excluded variables/classes/functions defined from 'exclude/python/exclude_python_words.txt' in dict -- #
if os.path.exists(self.pythonExcludeWords) == True:
with open(self.pythonExcludeWords, "r") as readFile:
for word in readFile:
if "#" in word or word == "\n":
continue
else:
word = word.rstrip()
wordsExcluded.append(word)
else:
print(ERROR_COLOUR + "[-] '{0}' file not found\n".format(self.pythonExcludeWords))
for word in wordsExcluded:
if word in variablesDict.keys():
wordsExcludedFound.append(word)
del variablesDict[word]
if word in classesDict.keys():
wordsExcludedFound.append(word)
del classesDict[word]
if word in functionsDict.keys():
wordsExcludedFound.append(word)
del functionsDict[word]
# -- Include variables/classes/functions defined from 'include/python/include_python_words.txt' in dict -- #
if os.path.exists(self.pythonIncludeWords) == True:
with open(self.pythonIncludeWords, "r") as readFile:
for word in readFile:
if "#" in word or word == "\n":
continue
else:
word = word.rstrip()
wordsIncludedFound.append(word)
else:
print(ERROR_COLOUR + "[-] '{0}' file not found\n".format(self.pythonIncludeWords))
for word in wordsIncludedFound:
if word not in variablesDict.keys() and word not in classesDict.keys() and word not in functionsDict.keys():
mixer = self.mixer.GetStringMixer(mixerLevelArg)
includeDict[word] = mixer
wordsIncludedNotFound.append(word)
for file in recursFiles:
if blockDir in file:
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
for word in wordsIncludedNotFound:
if word in eachLine:
wordsIncludedNotFound.remove(word)
for word in wordsIncludedNotFound:
if word in includeDict.keys():
del includeDict[word]
# -- Display variables/classes/functions found -- #
if verboseArg:
print("\n[+] Variables found :\n")
if variablesDict == {}:
print("-> No result")
else:
for key, value in variablesDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Classes found :\n")
if classesDict == {}:
print("-> No result")
else:
for key, value in classesDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Functions found :\n")
if functionsDict == {}:
print("-> No result")
else:
for key, value in functionsDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Include found :\n")
if includeDict == {}:
print("-> No result")
else:
for key, value in includeDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Exclude found :\n")
if wordsExcludedFound == []:
print("-> No result")
else:
for word in wordsExcludedFound:
print("-> {0} : excluded".format(word))
# -- Merge all dicts -- #
allDict = self.utils.DictMerge(allDict, variablesDict)
allDict = self.utils.DictMerge(allDict, classesDict)
allDict = self.utils.DictMerge(allDict, functionsDict)
allDict = self.utils.DictMerge(allDict, includeDict)
for number in recursFiles:
countRecursFiles += 1
# -- Change variables/classes/functions to mixed values -- #
print("\n[+] Running replacement of variables/classes/functions in {0} file(s)...\n".format(countRecursFiles))
with tqdm.tqdm(total=countRecursFiles) as pbar:
for file in recursFiles:
pbar.update(1)
if blockDir in file:
continue
else:
with fileinput.input(file, inplace=True) as inputFile:
for eachLine in inputFile:
if not eachLine:
continue
else:
if codeArg == "python":
# -- Check code into """ or ''' -- #
if re.match(quoteIntoVariable, eachLine):
checkQuotePassing += 1
eachLine = Replace.EachLine(self, codeArg, eachLine, allDict.items(), False)
print(eachLine)
continue
elif re.match(quoteOfCommentariesMultipleLines, eachLine) or re.match(quoteOfEndCommentariesMultipleLines, eachLine):
checkQuotePassing += 1
eachLine = Replace.EachLine(self, codeArg, eachLine, allDict.items(), False)
print(eachLine)
if checkQuotePassing == 2:
checkQuotePassing = 0
continue
if checkQuotePassing == 1:
print(eachLine)
continue
elif checkQuotePassing == 2:
checkQuotePassing = 0
continue
else:
eachLine = Replace.EachLine(self, codeArg, eachLine, allDict.items(), False)
print(eachLine)
continue
# -- Check if variables/classes/functions have been mixed -- #
for file in recursFiles:
if blockDir in file:
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
for value in allDict.values():
if value in eachLine:
checkWordsMixed.append(value)
# -- Remove duplicated key -- #
checkListWordsMixed = list(dict.fromkeys(checkWordsMixed))
for i in checkListWordsMixed:
checkCountWordsMixed += 1
for i in allDict.values():
checkCountWordsValue += 1
if (self.remove.Backslashes(codeArg, outputArg) == 0):
if checkCountWordsMixed == checkCountWordsValue:
print("\n-> {0} variables/classes/functions replaced in {1} file(s)\n".format(checkCountWordsValue, countRecursFiles))
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
def StringsToHex(self, codeArg, outputArg, mixerLevelArg):
getLetterLineList = []
countRecursFiles = 0
checkPrint = 0
checkHexError = False
checkPrintError = False
hexLine = ""
utils = Utils()
mixer = Mixer()
remove = Remove()
if codeArg == "python":
detectFiles = "py"
blockDir = "__pycache__"
detectExecFunc = r"exec\(\w+\)"
detectQuotes = r"[\'|\"]{3}"
recursFiles = [f for f in glob.glob("{0}{1}**{1}*.{2}".format(outputArg, utils.Platform(), detectFiles), recursive=True)]
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 tqdm.tqdm(total=countRecursFiles) as pbar:
for file in recursFiles:
checkPrint = 0 # initialize check print() func at the begining of each file
pbar.update(1)
if blockDir in file:
continue
else:
# -- Add a new first random line and move the old first line to the second line to avoid replacing it -- #
with open(file, "r") as inputFile:
stringRandomMixer = mixer.GetStringMixer(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 = mixer.GetStringMixer(mixerLevelArg)
print(varMixer + "=\"\"\"")
checkPrint = 1
else:
getLetterLineList = [] # initialize list
for letterLine in eachLine:
letterToHex = "\\x" + str(letterLine.encode().hex())
getLetterLineList.append(letterToHex) # Get list of all letters in line
hexLine = "".join(getLetterLineList)
print(hexLine)
# -- Add exec funtions to interpret hex code in strings -- #
with open(file, "a") as inputFile:
inputFile.write("\"\"\"")
inputFile.write("\nexec({0})".format(varMixer))
# -- Check if all lines are replaced of hexadecimal value -- #
for file in recursFiles:
if blockDir in file:
continue
else:
with open(file, "r") as inputFile:
for eachLine in inputFile:
if not eachLine:
continue
else:
if not "\\x" in eachLine:
if re.match(detectQuotes, eachLine):
continue
elif re.match(detectExecFunc, eachLine):
continue
else:
checkError = True
else:
continue
if (remove.Backslashes(codeArg, outputArg) == 0):
if checkHexError == False:
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
def FilesName(self, codeArg, outputArg, mixerLevelArg, verboseArg):
filesNameDict = {}
filesNameDictNoExt = {}
filesNameFound = []
filesNameFoundNoExt = []
filesNameMixed = []
checkFilesFound = []
countRecursFiles = 0
checkRenameInDir = 0
checkRenameInCode = 0
unix = False
win = False
currentPosition = os.getcwd()
utils = Utils()
mixer = Mixer()
remove = Remove()
if codeArg == "python":
detectFiles = "py"
blockDir = "__pycache__"
blockFile = "__init__"
detectImport = r"\s*from\s+|\s*import\s+"
recursFiles = [f for f in glob.glob("{0}{1}**{1}*.{2}".format(outputArg, utils.Platform(), detectFiles), recursive=True)]
for number in recursFiles:
countRecursFiles += 1
for file in recursFiles:
if blockDir in file or blockFile in file:
continue
else:
if "\"" in file:
parseFilePath = file.split("\"")
win = True
else:
parseFilePath = file.split("/")
unix = True
checkFilesFound.append(file)
mixer = self.mixer.GetStringMixer(mixerLevelArg)
filesNameDict[parseFilePath[-1]] = mixer + ".py"
filesNameDictNoExt[parseFilePath[-1].replace(".py", "")] = mixer
filesNameFound.append(parseFilePath[-1])
filesNameMixed.append(mixer + ".py")
removeExt = parseFilePath[-1].replace(".py","")
filesNameFoundNoExt.append(removeExt)
# -- Diplay all files found with their mixed values if verbose arg is actived -- #
if verboseArg:
print("\n[+] Files name found with their mixed values :\n")
for key, value in filesNameDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Running replace files name in {0} file(s)...\n".format(countRecursFiles))
# -- Replace all files name to random strings with length defined -- #
with tqdm.tqdm(total=countRecursFiles) as pbar:
for file in recursFiles:
pbar.update(1)
if blockDir in file:
continue
else:
# -- Rename all files in python code -- #
with fileinput.input(file, inplace=True) as inputFile:
for eachLine in inputFile:
try:
for fileName in filesNameFound:
for fileNameNoExt in filesNameFoundNoExt:
if fileName in eachLine or fileNameNoExt in eachLine:
if not ".py" in eachLine:
if re.match(detectImport, eachLine):
eachLine = Replace.EachLine(self, codeArg, eachLine, filesNameDictNoExt.items(), True)
print(eachLine)
raise BreakLoop
else:
continue
else:
eachLine = Replace.EachLine(self, codeArg, eachLine, filesNameDict.items(), True)
print(eachLine)
raise BreakLoop
else:
continue
print(eachLine)
except BreakLoop:
continue
# -- 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) # Return to old position
# -- Check if all files name are been replaced to random strings with length defined -- #
for i in checkFilesFound:
i.replace(".py", "")
# -- Check for file(s) name -- #
for file in recursFiles:
if blockDir in file or blockFile in file:
continue
else:
for i in checkFilesFound:
if i in file:
checkFilesFound.remove(i)
continue
# -- Check for code if file(s) name -- #
if checkFilesFound != []:
for file in recursFiles:
if blockDir in file or blockFile in file:
continue
else:
with open(file, "r") as inputFile:
for line in inputFile:
for i in checkFilesFound:
if i in line:
checkFilesFound.remove(i)
continue
if checkFilesFound == []:
if (remove.Backslashes(codeArg, outputArg) == 0):
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
class Padding:
def __init__(self):
self.mixer = Mixer()
self.remove = Remove()
self.utils = Utils()
def ScriptsGenerator(self, codeArg, mixerLevelArg):
if mixerLevelArg == "lower":
varRandom1 = self.mixer.GetStringMixer("lower")
varRandom2 = self.mixer.GetStringMixer("lower")
varRandom3 = self.mixer.GetStringMixer("lower")
varRandom4 = self.mixer.GetStringMixer("lower")
varRandom5 = self.mixer.GetStringMixer("lower")
varRandom6 = self.mixer.GetStringMixer("lower")
varRandom7 = self.mixer.GetStringMixer("lower")
varRandom8 = self.mixer.GetStringMixer("lower")
varRandom9 = self.mixer.GetStringMixer("lower")
varRandom10 = self.mixer.GetStringMixer("lower")
varRandom11 = self.mixer.GetStringMixer("lower")
varRandom12 = self.mixer.GetStringMixer("lower")
elif mixerLevelArg == "medium":
varRandom1 = self.mixer.GetStringMixer("medium")
varRandom2 = self.mixer.GetStringMixer("medium")
varRandom3 = self.mixer.GetStringMixer("medium")
varRandom4 = self.mixer.GetStringMixer("medium")
varRandom5 = self.mixer.GetStringMixer("medium")
varRandom6 = self.mixer.GetStringMixer("medium")
varRandom7 = self.mixer.GetStringMixer("medium")
varRandom8 = self.mixer.GetStringMixer("medium")
varRandom9 = self.mixer.GetStringMixer("medium")
varRandom10 = self.mixer.GetStringMixer("medium")
varRandom11 = self.mixer.GetStringMixer("medium")
varRandom12 = self.mixer.GetStringMixer("medium")
elif mixerLevelArg == "high":
varRandom1 = self.mixer.GetStringMixer("high")
varRandom2 = self.mixer.GetStringMixer("high")
varRandom3 = self.mixer.GetStringMixer("high")
varRandom4 = self.mixer.GetStringMixer("high")
varRandom5 = self.mixer.GetStringMixer("high")
varRandom6 = self.mixer.GetStringMixer("high")
varRandom7 = self.mixer.GetStringMixer("high")
varRandom8 = self.mixer.GetStringMixer("high")
varRandom9 = self.mixer.GetStringMixer("high")
varRandom10 = self.mixer.GetStringMixer("high")
varRandom11 = self.mixer.GetStringMixer("high")
varRandom12 = self.mixer.GetStringMixer("high")
# ---------- Python random scripts ---------- #
if codeArg == "python":
rand = random.randint(1, 5)
# -- script 1 -- #
if rand == 1:
scriptAssPadding1 = 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 scriptAssPadding1
# -- script 2 -- #
elif rand == 2:
scriptAssPadding2 = textwrap.dedent("""
{0} = '{4}'
{1} = '{5}'
if {0} != {1}:
{2} = '{6}'
{3} = '{7}'
{3} = {2}
""").format(varRandom1, varRandom2, varRandom3, varRandom4, varRandom5, \
varRandom6, varRandom7, varRandom8)
return scriptAssPadding2
# -- script 3 -- #
elif rand == 3:
scriptAssPadding3 = 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 scriptAssPadding3
# -- script 4 -- #
elif rand == 4:
scriptAssPadding4 = textwrap.dedent("""
{0} = '{4}'
{1} = '{5}'
{3} = '{7}'
if {0} == {1}:
{2} = '{6}'
{2} = {0}
else:
{2} = '{6}'
{2} = {3}
""").format(varRandom1, varRandom2, varRandom3, varRandom4, \
varRandom5, varRandom6, varRandom7, varRandom8)
return scriptAssPadding4
# -- script 5 -- #
elif rand == 5:
scriptAssPadding5 = 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 scriptAssPadding5
def AddScripts(self, codeArg, outputArg, mixerLevelArg):
countScriptsAdded = 0
countLineAdded = 0
countLine = 0
checkLine = 0
checkPassing = 0
countRecursFiles = 0
if codeArg == "python":
inputExt = "py"
blockDirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), inputExt),
recursive=True)
]
# -- Count the number of lines that will be checked before filling -- #
for output in recursFiles:
if re.match(blockDirs, output):
continue
else:
with open(output, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if not eachLine:
continue
countLine += 1
for number in recursFiles:
countRecursFiles += 1
print("\n[+] Running adding of random scripts in {0} file(s)...\n".
format(countRecursFiles))
# -- Padding scripts added -- #
with tqdm(total=countRecursFiles) as pbar:
for output in recursFiles:
pbar.update(1)
if re.match(blockDirs, output):
continue
else:
with fileinput.input(output, inplace=True) as inputFile:
for eachLine in inputFile:
print(eachLine)
if eachLine == "\n":
continue
else:
if codeArg == "python":
spaces = len(eachLine) - len(
eachLine.lstrip()) # Check line indent
noAddScript = r"(^[\#]+.*)|(\@|\s+\@)|(\s+return)|(\s+#\s{1,10}\w+)"
addIndentScript = r".*\:{1}\s"
checkAddIndentScript = r".*\:{1}\s\w+"
# -- Check if ',' char or '\' char,in end line -- #
listCheckEndLine = []
for i in eachLine:
listCheckEndLine.append(i)
if "," in listCheckEndLine[
-2] or "\\" in listCheckEndLine[-2]:
continue
# -- Check code between """ or ''' -- #
if '\"\"\"' in eachLine or "\'\'\'" in eachLine:
checkPassing += 1
if checkPassing == 1: # Loop until the next """ or '''
continue
else:
checkPassing = 0
if re.match(noAddScript, eachLine) is not None:
continue
elif re.match(addIndentScript,
eachLine) is not None:
if re.match(checkAddIndentScript,
eachLine) is not None:
continue
else:
if spaces == 0:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
elif spaces == 4:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
elif spaces == 8:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
elif spaces == 12:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
else:
continue
else:
if spaces == 0:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg), ""))
countScriptsAdded += 1
elif spaces == 4:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg), " "))
countScriptsAdded += 1
elif spaces == 8:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
elif spaces == 12:
print(
textwrap.indent(
Padding.ScriptsGenerator(
self, codeArg,
mixerLevelArg),
" "))
countScriptsAdded += 1
else:
continue
# -- Check padding has added in output script -- #
for output in recursFiles:
if re.match(blockDirs, output):
continue
else:
with open(output, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if not eachLine:
continue
checkLine += 1
countLineAdded = checkLine - countLine
if (self.remove.LineBreaks(codeArg, outputArg) == 0):
if checkLine > countLine:
print("\n-> {0} scripts added in {1} file(s)\n".format(
countScriptsAdded, countRecursFiles))
print("-> {0} lines added in {1} file(s)\n".format(
countLineAdded, countRecursFiles))
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
EXIT_FAILURE
class Remove:
def __init__(self):
self.utils = Utils()
def LineBreaks(self, codeArg, outputArg):
checkLine = 0
numberFiles = 0
numberFileGood = 0
if codeArg == "python":
detectFile = "py"
blockDirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFile),
recursive=True)
]
# -- Delete line breaks -- #
for output in recursFiles:
if re.match(blockDirs, output):
continue
else:
with fileinput.FileInput(output, inplace=True) as inputFile:
for line in inputFile:
if line.strip():
print(line.rstrip())
# -- Check if all line breaks are deleted -- #
for output in recursFiles:
checkLine = 0 # Initialize check vars for the next file
if re.match(blockDirs, output):
continue
else:
with open(output, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if eachLine == "\n":
checkLine += 1
if checkLine == 0:
numberFileGood += 1
numberFiles += 1
else:
numberFileGood += 0
numberFiles += 1
if numberFileGood == numberFiles:
return EXIT_SUCCESS
else:
return EXIT_FAILURE
def Commentaries(self, codeArg, outputArg):
countLineOutput = 0
countLineInput = 0
noCommentary = 0
isCommentary = 0
countRecursFiles = 0
if codeArg == "python":
classicCommentariesAfterLine = r"\s+\#.*" # '#' after line of code
classicCommentariesBeginLine = r"^\#.*" # begin '#'
quoteOfCommentariesMultipleLine = r"^[\"|\']{3}$" # """ and ''' without before variables and if commentaries is over multiple lines
quoteOfCommentariesOneLine = r"[\"|\']{3}.*[\"|\']{3}" # """ and ''' without before variables and if commentary is over one line, (""" commentaries """)
noQuoteOfCommentaries = r"\w+\s*[\=|\(]{1}\s*[\"|\']{3}" # """ and ''' with before variables
detectFile = "py"
blockDirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFile),
recursive=True)
]
# -- Count commentaries will be removed -- #
for output in recursFiles:
if re.match(blockDirs, output):
continue
else:
with open(output, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if codeArg == "python":
searchClassicCommentariesAfterLine = re.search(
classicCommentariesAfterLine, eachLine)
searchClassicCommentariesBeginLine = re.search(
classicCommentariesBeginLine, eachLine)
if "coding" in eachLine or "#!" in eachLine:
continue
if re.match(noQuoteOfCommentaries, eachLine):
noCommentary += 1
elif re.match(quoteOfCommentariesMultipleLine,
eachLine):
isCommentary += 1
else:
pass
if re.match(quoteOfCommentariesOneLine, eachLine):
isCommentary = 0
countLineInput += 1
continue
elif isCommentary == 1 and noCommentary == 0:
countLineInput += 1
continue
elif isCommentary == 0 and noCommentary == 1:
continue
elif isCommentary == 2:
isCommentary = 0
countLineInput += 1
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
continue
else:
pass
if searchClassicCommentariesBeginLine is not None:
countLineInput += 1
elif searchClassicCommentariesAfterLine is not None:
countLineInput += 1
else:
pass
# -- Initialize vars -- #
isCommentary = 0
noCommentary = 0
# -- Remove commentaries -- #
for number in recursFiles:
countRecursFiles += 1
print("\n[+] Running remove commentaries in {0} file(s)...\n".format(
countRecursFiles))
with tqdm(total=countRecursFiles) as pbar:
for output in recursFiles:
pbar.update(1)
if re.match(blockDirs, output):
continue
else:
# -- Remove commentaries -- #
with fileinput.input(output, inplace=True) as inputFile:
for eachLine in inputFile:
searchClassicCommentariesAfterLine = re.search(
classicCommentariesAfterLine, eachLine)
searchClassicCommentariesBeginLine = re.search(
classicCommentariesBeginLine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
print(eachLine)
continue
if re.match(noQuoteOfCommentaries, eachLine):
noCommentary += 1
elif re.match(quoteOfCommentariesMultipleLine,
eachLine):
isCommentary += 1
else:
pass
if re.match(quoteOfCommentariesOneLine,
eachLine):
isCommentary = 0
continue
elif isCommentary == 1 and noCommentary == 0:
continue
elif isCommentary == 0 and noCommentary == 1:
print(eachLine)
continue
elif isCommentary == 2:
isCommentary = 0
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
print(eachLine)
continue
else:
pass
if searchClassicCommentariesBeginLine is not None:
eachLine = eachLine.replace(
searchClassicCommentariesBeginLine.group(
0), "")
print(eachLine)
elif searchClassicCommentariesAfterLine is not None:
eachLine = eachLine.replace(
searchClassicCommentariesAfterLine.group(
0), "")
print(eachLine)
else:
print(eachLine)
# -- Initialize vars -- #
isCommentary = 0
noCommentary = 0
# -- Check if all commentaries are removed -- #
for output in recursFiles:
countLineOutput = 0
if re.match(blockDirs, output):
continue
else:
with open(output, "r") as readFile:
countLineOutput = 0
readF = readFile.readlines()
for eachLine in readF:
searchClassicCommentariesAfterLine = re.search(
classicCommentariesAfterLine, eachLine)
searchClassicCommentariesBeginLine = re.search(
classicCommentariesBeginLine, eachLine)
if codeArg == "python":
if "coding" in eachLine or "#!" in eachLine:
continue
if re.match(noQuoteOfCommentaries, eachLine):
noCommentary += 1
elif re.match(quoteOfCommentariesMultipleLine,
eachLine):
isCommentary += 1
else:
pass
if re.match(quoteOfCommentariesOneLine, eachLine):
isCommentary = 0
countLineOutput += 1
continue
elif isCommentary == 1 and noCommentary == 0:
countLineOutput += 1
continue
elif isCommentary == 0 and noCommentary == 1:
continue
elif isCommentary == 2:
isCommentary = 0
countLineOutput += 1
continue
elif isCommentary == 1 and noCommentary == 1:
isCommentary = 0
noCommentary = 0
continue
else:
pass
if searchClassicCommentariesBeginLine is not None:
countLineOutput += 1
elif searchClassicCommentariesAfterLine is not None:
countLineOutput += 1
else:
pass
if (Remove.LineBreaks(self, codeArg, outputArg) == 0):
if countLineOutput == 0:
print("\n-> {0} lines of commentaries removed\n".format(
countLineInput))
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
def PrintFunc(self, codeArg, outputArg):
countPrintLine = 0
countCheckPrintLine = 0
countRecursFiles = 0
if codeArg == "python":
detectPrint = r"\s{2,}print|^print"
detectFile = "py"
blockDirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFile),
recursive=True)
]
for number in recursFiles:
countRecursFiles += 1
print("\n[+] Running remove print function in {0} file(s)...\n".format(
countRecursFiles))
with tqdm(total=countRecursFiles) as pbar:
for output in recursFiles:
pbar.update(1)
if re.match(blockDirs, output):
continue
else:
# -- Remove all print functions -- #
with fileinput.input(output, inplace=True) as inputFile:
for eachLine in inputFile:
if re.match(detectPrint, eachLine):
countPrintLine += 1
continue
else:
print(eachLine)
# -- Check if all print functions are removed -- #
for output in recursFiles:
if re.match(blockDirs, output):
continue
else:
with open(output, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
if re.match(detectPrint, eachLine):
countCheckPrintLine += 1
if (Remove.LineBreaks(self, codeArg, outputArg) == 0):
if countCheckPrintLine == 0:
print("\n-> {0} lines of print functions removed\n".format(
countPrintLine))
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
class ReplaceWords:
def __init__(self):
self.mixer = Mixer()
self.remove = Remove()
self.utils = Utils()
self.pythonExcludeWords = "exclude/python/exclude_python_words.txt"
self.pythonIncludeWords = "include/python/include_python_words.txt"
def EachLine(self, codeArg, eachLine, Dict):
getIndexLineList = []
returnLine = []
charValue = []
checkCharAfterWord = 1
wordSetter = 0
checkGetKey = ""
checkGetWord = ""
getLine = ""
breakLine = ""
if codeArg == "python":
regReplace = r"(\.)|(:)|(\))|(\()|(=)|(\[)|(\])|({)|(})|(,)|(\+)|(\s)|(\*)|(\+)|(\-)"
# -- Get list of all letters in line -- #
for indexLine, letterLine in enumerate(eachLine):
getIndexLineList.append(letterLine)
# -- Loop in each letter of line -- #
for indexLine, letterLine in enumerate(eachLine):
# -- 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(regReplace, 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 Dict:
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:
# -- Place index position after the word -- #
indexExplore = indexLine + len(key)
# -- If indexError return to next loop -- #
try:
getIndexLineList[indexExplore]
except IndexError:
continue
# -- Check the char after the word -- #
if re.match(
regReplace,
getIndexLineList[
indexExplore]):
# -- Check word finded is not into the other word -- #
indexExploreBefore = indexLine - 1
if not re.match(
r"\w|\\",
getIndexLineList[
indexExploreBefore]
):
if codeArg == "python":
# -- Check if it's 'from' and 'import' word key in line to avoid replace name of file if variable is identic name to file -- #
getLine = "".join(
getIndexLineList)
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 = 0
else:
checkCharAfterWord = 0
else:
checkCharAfterWord = 1
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 VarsDefinedByUser(self, codeArg, outputArg, mixerLevelArg, verboseArg):
variablesDict = {}
classesDict = {}
functionsDict = {}
includeDict = {}
allDict = {}
checkWordsMixed = []
wordsExcluded = []
wordsExcludedFound = []
wordsIncludedFound = []
wordsIncludedNotFound = []
checkCountWordsMixed = 0
checkCountWordsValue = 0
checkQuotePassing = 0
numberReplaced = 0
countRecursFiles = 0
if codeArg == "python":
variablesDefined = r"(^\w+|\w+)([\s|^\s]*=[\s|\w|\"|\'])" # Variables
variablesErrorDefined = r"except(\s+\w+\sas\s)(\w)" # Error variables
variablesLoopDefined = r"for\s+([\w\s\,]+)(\s+in\s+)" # Loop variables
functionsDefined = r"def\s(\w+)" # Functions
classDefined = r"class\s(\w+)" # Classes
quoteOfEndCommentariesMultipleLines = r"^\s*[\"|\']{3}\)?\.?" # """ and ''' without before variables, if commentaries is over multiple lines and he finish by .format() funtion
quoteOfCommentariesMultipleLines = r"^\s*[\"|\']{3}$" # """ and ''' without before variables and if commentaries is over multiple lines
quoteInRegex = r"\={1}\s*r[\"|\']{1}" # If quote in regex
quoteIntoVariable = r".*\={1}\s*\w*\.?\w*[\(|\.]{1}[\"|\']{3}|.*\={1}\s*[\"|\']{3}" # """ and ''' with before variables
detectFile = "py"
blockDirs = r"__pycache__"
recursFiles = [
f for f in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg, self.utils.Platform(), detectFile),
recursive=True)
]
# -- Find variables/classes/functions and mixed it -- #
for file in recursFiles:
if re.match(blockDirs, file):
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
# -- Variables -- #
search = re.search(variablesDefined, eachLine)
if search != None:
# -- Detect mixer -- #
mixer = self.mixer.GetStringMixer(mixerLevelArg)
# -- Add the mixer value of variables -- #
if search.group(1) not in variablesDict:
variablesDict[search.group(1)] = mixer
# -- Error variables -- #
search = re.search(variablesErrorDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(2) not in variablesDict:
variablesDict[search.group(2)] = mixer
# -- Loop variables -- #
search = re.search(variablesLoopDefined, eachLine)
if search != None:
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(
mixerLevelArg)
variablesDict[i] = mixer
else:
if search.group(1) not in variablesDict:
mixer = self.mixer.GetStringMixer(
mixerLevelArg)
variablesDict[search.group(1)] = mixer
# -- Functions -- #
search = re.search(functionsDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(1) not in functionsDict:
if codeArg == "python":
if not re.match(r"__init__",
search.group(1)):
functionsDict[search.group(1)] = mixer
# -- Classes -- #
search = re.search(classDefined, eachLine)
if search != None:
mixer = self.mixer.GetStringMixer(mixerLevelArg)
if search.group(1) not in classesDict:
classesDict[search.group(1)] = mixer
# -- Remove excluded variables/classes/functions defined from 'exclude/python/exclude_python_words.txt' in dict -- #
if os.path.exists(self.pythonExcludeWords) == True:
with open(self.pythonExcludeWords, "r") as readFile:
for word in readFile:
if "#" in word or word == "\n":
continue
else:
word = word.rstrip()
wordsExcluded.append(word)
else:
print("[-] '{0}' file not found\n".format(self.pythonExcludeWords))
for word in wordsExcluded:
if word in variablesDict.keys():
wordsExcludedFound.append(word)
del variablesDict[word]
if word in classesDict.keys():
wordsExcludedFound.append(word)
del classesDict[word]
if word in functionsDict.keys():
wordsExcludedFound.append(word)
del functionsDict[word]
# -- Include variables/classes/functions defined from 'include/python/include_python_words.txt' in dict -- #
if os.path.exists(self.pythonIncludeWords) == True:
with open(self.pythonIncludeWords, "r") as readFile:
for word in readFile:
if "#" in word or word == "\n":
continue
else:
word = word.rstrip()
wordsIncludedFound.append(word)
else:
print("[-] '{0}' file not found\n".format(self.pythonIncludeWords))
for word in wordsIncludedFound:
if word not in variablesDict.keys(
) and word not in classesDict.keys(
) and word not in functionsDict.keys():
mixer = self.mixer.GetStringMixer(mixerLevelArg)
includeDict[word] = mixer
wordsIncludedNotFound.append(word)
for file in recursFiles:
if re.match(blockDirs, file):
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
for word in wordsIncludedNotFound:
if word in eachLine:
wordsIncludedNotFound.remove(word)
for word in wordsIncludedNotFound:
if word in includeDict.keys():
del includeDict[word]
# -- Display variables/classes/functions found -- #
if verboseArg:
print("\n[+] Variables found :\n")
if variablesDict == {}:
print("-> No result")
else:
for key, value in variablesDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Classes found :\n")
if classesDict == {}:
print("-> No result")
else:
for key, value in classesDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Functions found :\n")
if functionsDict == {}:
print("-> No result")
else:
for key, value in functionsDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Include found :\n")
if includeDict == {}:
print("-> No result")
else:
for key, value in includeDict.items():
print("-> {0} : {1}".format(key, value))
print("\n[+] Exclude found :\n")
if wordsExcludedFound == []:
print("-> No result")
else:
for word in wordsExcludedFound:
print("-> {0} : excluded".format(word))
# -- Merge all dicts -- #
allDict = self.utils.DictMerge(allDict, variablesDict)
allDict = self.utils.DictMerge(allDict, classesDict)
allDict = self.utils.DictMerge(allDict, functionsDict)
allDict = self.utils.DictMerge(allDict, includeDict)
for number in recursFiles:
countRecursFiles += 1
# -- Change variables/classes/functions to mixed values -- #
print(
"\n[+] Running replacement of variables/classes/functions in {0} file(s)...\n"
.format(countRecursFiles))
with tqdm(total=countRecursFiles) as pbar:
for file in recursFiles:
pbar.update(1)
if re.match(blockDirs, file):
continue
else:
with fileinput.input(file, inplace=True) as inputFile:
for eachLine in inputFile:
if not eachLine:
continue
else:
if codeArg == "python":
# -- Check code into """ or ''' -- #
if re.match(quoteIntoVariable, eachLine):
checkQuotePassing += 1
eachLine = ReplaceWords.EachLine(
self, codeArg, eachLine,
allDict.items())
print(eachLine)
continue
elif re.match(
quoteOfCommentariesMultipleLines,
eachLine
) or re.match(
quoteOfEndCommentariesMultipleLines,
eachLine):
checkQuotePassing += 1
eachLine = ReplaceWords.EachLine(
self, codeArg, eachLine,
allDict.items())
print(eachLine)
if checkQuotePassing == 2:
checkQuotePassing = 0
continue
if checkQuotePassing == 1:
print(eachLine)
continue
elif checkQuotePassing == 2:
checkQuotePassing = 0
continue
else:
eachLine = ReplaceWords.EachLine(
self, codeArg, eachLine,
allDict.items())
print(eachLine)
continue
# -- Check if variables/classes/functions have been mixed -- #
for file in recursFiles:
if re.match(blockDirs, file):
continue
else:
with open(file, "r") as readFile:
readF = readFile.readlines()
for eachLine in readF:
for value in allDict.values():
if value in eachLine:
checkWordsMixed.append(value)
# -- Remove duplicated key -- #
checkListWordsMixed = list(dict.fromkeys(checkWordsMixed))
for i in checkListWordsMixed:
checkCountWordsMixed += 1
for i in allDict.values():
checkCountWordsValue += 1
if (self.remove.LineBreaks(codeArg, outputArg) == 0):
if checkCountWordsMixed == checkCountWordsValue:
print(
"\n-> {0} variables/classes/functions replaced in {1} file(s)\n"
.format(checkCountWordsValue, countRecursFiles))
return EXIT_SUCCESS
else:
return EXIT_FAILURE
else:
return EXIT_FAILURE
class Analyze:
def __init__(self):
self.utils = Utils()
def InputAvailable(self, inputArg, verboseArg):
inputFileFound = []
inputFileEmpty = []
inputFileFoundCount = 0
inputFileEmptyCount = 0
countRecursFiles = 0
numberLine = 0
if os.path.exists(inputArg) == True:
if os.path.isdir(inputArg) == True:
detectFiles = "py"
blockDir = "__pycache__"
recursFiles = [
file for file in glob.glob("{0}{1}**{1}*.{2}".format(
inputArg,
self.utils.Platform(getOS=False, getPathType=True),
detectFiles),
recursive=True)
]
if recursFiles == []:
print(Colors.ERROR +
"[-] {} directory empty or python file not found".
format(inputArg) + Colors.DISABLE)
return 1
for number in recursFiles:
countRecursFiles += 1
print("\n[+] Running analyze input of {} file(s)...\n".format(
countRecursFiles))
with Bar("Analysis ",
fill="=",
max=countRecursFiles,
suffix="%(percent)d%%") as bar:
for file in recursFiles:
if blockDir in file:
continue
else:
if os.path.getsize(file) > 0:
inputFileFound.append(file)
inputFileFoundCount += inputFileFoundCount + 1
else:
inputFileEmpty.append(file)
inputFileEmptyCount += inputFileEmptyCount + 1
bar.next(1)
bar.finish()
if inputFileFoundCount >= 1 and inputFileEmptyCount < inputFileFoundCount:
if verboseArg:
print("\n\n[+] File input found :\n")
if inputFileFound == []:
print("-> no result")
else:
for file in inputFileFound:
print("-> {}".format(file))
for file in inputFileEmpty:
print("-> {} : empty".format(file))
return 0
elif inputFileFoundCount == inputFileEmptyCount and inputFileFoundCount > 0:
print(
Colors.ERROR +
"[-] All files in directory specified are emtpy !"
+ Colors.DISABLE)
return 1
else:
print(
Colors.ERROR +
"[-] No file available in '{}'.".format(inputArg) +
Colors.DISABLE)
return 1
else:
print(Colors.ERROR +
"[-] '{}' is not a directory".format(inputArg) +
Colors.DISABLE)
return 1
else:
print(Colors.ERROR + "[-] '{}' not exists".format(inputArg) +
Colors.DISABLE)
return 1
def OutputAvailable(self, inputArg, outputArg, verboseArg):
outputFileFound = []
outputFileEmpty = []
outputFileFoundCount = 0
outputFileEmptyCount = 0
countRecursFiles = 0
if os.path.exists(outputArg) == True:
deleteRequest = input(
"[!] Output '{}' already exists, do you want delete it ? (Y/N) : "
.format(outputArg))
deleteRequest = deleteRequest.upper()
if deleteRequest == "Y":
try:
shutil.rmtree(outputArg)
if os.path.exists(outputArg) == False:
shutil.copytree(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isdir(outputArg) == True:
detectFiles = "py"
blockDir = "__pycache__"
recursFiles = [
file for file in
glob.glob("{0}{1}**{1}*.{2}".format(
outputArg,
self.utils.Platform(getOS=False,
getPathType=True),
detectFiles),
recursive=True)
]
for number in recursFiles:
countRecursFiles += 1
print(
"\n[+] Running analyze output of {} file(s)...\n"
.format(countRecursFiles))
if recursFiles == []:
print(Colors.ERROR + "[-] {} directory empty, no copied file".format(inputArg) + \
Colors.DISABLE)
return 1
with Bar("Analysis ",
fill="=",
max=countRecursFiles,
suffix="%(percent)d%%") as bar:
for file in recursFiles:
if blockDir in file:
continue
else:
if os.path.getsize(file) > 0:
outputFileFound.append(file)
outputFileFoundCount += outputFileFoundCount + 1
else:
outputFileEmpty.append(file)
outputFileEmptyCount += outputFileEmptyCount + 1
bar.next(1)
bar.finish()
if outputFileFoundCount >= 1 and outputFileFoundCount > outputFileEmptyCount:
if verboseArg:
print(
"\n\n[+] Output files copy :\n"
)
if outputFileFound == []:
print("-> no result")
else:
for file in outputFileFound:
print("-> {}".format(file))
for file in outputFileEmpty:
print(
"-> {} : empty".format(
file))
return 0
else:
print(Colors.ERROR +
"[-] No files available in '{}'".
format(outputArg) +
Colors.DISABLE)
return 1
else:
print(Colors.ERROR + "[-] Copy '{}' to '{}' failed, this is not a output directory copied"\
.format(inputArg, outputArg) + Colors.DISABLE)
return 1
else:
print(Colors.ERROR +
"[-] Copy '{}' to '{}' failed".format(
inputArg, outputArg) + Colors.DISABLE)
return 1
else:
print(Colors.ERROR +
"[-] Delete '{}' failed".format(outputArg) +
Colors.DISABLE)
return 1
except Exception as e:
print(Colors.ERROR + "[-] {}".format(e) + Colors.DISABLE)
return 1
else:
print(Colors.ERROR +
"[-] Delete '{}' failed, the user has refused".format(
outputArg) + Colors.DISABLE)
return 1
else:
try:
shutil.copytree(inputArg, outputArg)
if os.path.exists(outputArg) == True:
if os.path.isdir(outputArg) == True:
detectFiles = "py"
blockDir = "__pycache__"
recursFiles = [
file
for file in glob.glob("{0}{1}**{1}*.{2}".format(
outputArg,
self.utils.Platform(getOS=False,
getPathType=True),
detectFiles),
recursive=True)
]
for number in recursFiles:
countRecursFiles += 1
print(
"\n[+] Running analyze output of {} file(s)...\n".
format(countRecursFiles))
if recursFiles == []:
print(Colors.ERROR +
"[-] {} directory empty, no copied file".
format(inputArg) + Colors.DISABLE)
return 1
with Bar("Analysis ",
fill="=",
max=countRecursFiles,
suffix="%(percent)d%%") as bar:
for file in recursFiles:
if blockDir in file:
continue
else:
if os.path.getsize(file) > 0:
outputFileFound.append(file)
outputFileFoundCount += outputFileFoundCount + 1
else:
outputFileEmpty.append(file)
outputFileEmptyCount += outputFileEmptyCount + 1
bar.next(1)
bar.finish()
if outputFileFoundCount >= 1 and outputFileFoundCount > outputFileEmptyCount:
if verboseArg:
print("\n\n[+] File output found :\n")
if outputFileFound == []:
print("-> no result")
else:
for file in outputFileFound:
print("-> {} : copy".format(file))
for file in outputFileEmpty:
print("-> {} : copy empty".format(
file))
return 0
else:
print(Colors.ERROR +
"[-] No files available in '{}'".format(
outputArg) + Colors.DISABLE)
return 1
else:
print(Colors.ERROR + "[-] Copy '{}' to '{}' failed, this is not a output directory copied !"\
.format(inputArg, outputArg) + Colors.DISABLE)
return 1
else:
print(Colors.ERROR + "[-] Copy '{}' to '{}' failed".format(
inputArg, outputArg) + Colors.DISABLE)
return 1
except Exception as e:
print(Colors.ERROR + "[-] {}".format(e) + Colors.DISABLE)
return 1