Py_learn

有时间整理下python吧.

0x1.FileRead2Write

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = 'ahoo'

import sys
import io
import os
import codecs
import re
import shutil
import random
import logging
import logging.config

#--------------------文件配置----------
logfilePath = os.path.join(os.path.dirname(__file__), 'logging.conf')
logging.config.fileConfig('logging.conf')
logging.getLogger()


sys.stdout = io.TextIOWrapper(
                sys.stdout.buffer,
                encoding='utf-8') #改变标准输出的默认编码

class FileRead2Write(object):

    def __init__(self):
        self.AuthorSign = True

    def ReadInputFile(
            self,
            InPath,
            ReadTye = 'r'):

        logall = []

        #这个判断应该放到参数类里. if not os.path.isfile(InPath):
        fpR = codecs.open(InPath,ReadTye,'utf-8')
        for line in fpR:
            if None == line:
                pass
            else:
                logall.append(line)
        fpR.close()
        return logall

    def WriteOutputFile(self,OutPath,OutList= [],WriteTye = 'a+'):  #后面可能改成词典
        fpW = codecs.open(OutPath,WriteTye,'utf-8')
        if self.AuthorSign == True:
            fpW.write('\n*****************************************************\r\n')
            fpW.write('*                ahoo EngObfuscate                       ')
            fpW.write('\n***************************************************\r\n\n')
            self.AuthorSign = False
        for i in OutList:
            fpW.write(i)
        fpW.close()
        return True

    def WriteOutputFileEx_ListShuffle(self,OutPath,OutList= [],WriteTye = 'a+'):    #后面可能改成词典
        fpW = codecs.open(OutPath,WriteTye,'utf-8')
        if self.AuthorSign == True:
            fpW.write('\n*****************************************************\r\n')
            fpW.write('*                ahoo EngObfuscate                       ')
            fpW.write('\n***************************************************\r\n\n')
            self.AuthorSign = False
        if len(OutList) == 0:
            fpW.write('\n')
            return True
        random.shuffle(OutList)
        for i in OutList:
            fpW.write(i)
        fpW.close()
        return True

    def WriteOutputFileEx_LineStr(self,OutPath,LineStr,WriteTye = 'a+'):
        fpW = codecs.open(OutPath,WriteTye,'utf-8')
        if self.AuthorSign == True:
            fpW.write('\n***************************************************\n')
            fpW.write('*                ahoo EngObfuscate                       ')
            fpW.write('\n***************************************************\n\n')
            self.AuthorSign = False
        fpW.write('\n' + LineStr + '\n')
        fpW.close()
        return True

    def OpenOutPath(self,OutPath,program = '"D:\\Program Files (x86)\\Notepad++\\notepad++.exe" '):
        return os.system(program + OutPath)         # program = 'notepad.exe' 

    #将list写入另一个中.
    def WriteList2List(self,list1 = [],list2 = []):
        if len(list1) == 0:
            pass
        else:
            for i in list1:
                list2.append(i)

        return True

    ''' 1.list排序
        ransomFamilyList = list(set(ransomFamilyList))
        2.list最后一行去掉\n
        ransomFamilyList[-1] = ransomFamilyList[-1].strip('\n')
        3.去空格.
        ransomFamilyList.append(i.strip(' ') + '\n')    

        4.遍历目录
        for parent,dirnames,filenames in os.walk(InputDir):
            for filename in filenames:
                fpRan.ReadInputFile(os.path.join(parent, filename)) 

        5.for循环 #a =1
        #[i for i in range(0,10,1)  a = a+i ]
    '''

0x2.RandomSequence

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = 'ahoo'

import sys
import io
import os
import codecs
import re
import shutil
import random
import logging
import logging.config

#--------------------文件配置----------
logfilePath = os.path.join(os.path.dirname(__file__), 'logging.conf')
logging.config.fileConfig('logging.conf')
logging.getLogger()

#sys.stdout = io.TextIOWrapper(
#               sys.stdout.buffer,
#               encoding='utf-8') #改变标准输出的默认编码



class RandomSequence(object):

    varStrRemoveRepeatL = []
    varStrRemoveRepeatL.append('ahoo')

    def __init__(self):
        self.author = 'ahoo'
        pass


    #随机一个长度2-5的字符串,一般用作变量名  
    def randStr_az(
            self,
            minNum=2,
            maxNum=5):          #判断一下大小传反了的情况

        randstr = ''
        if minNum > maxNum:
            minNum = min(minNum,maxNum)
            maxNum = max(minNum,maxNum)
        strLengt = random.randint(minNum,maxNum)
        for i in range(1,strLengt,1):
            chrTem = chr(random.randint(97,122))
            randstr = randstr + chrTem
        return randstr



    def randStr_AZUp(self,minNum=2,maxNum=5):
        randstr = ''
        if minNum > maxNum:
            minNum = min(minNum,maxNum)
            maxNum = max(minNum,maxNum)
        strLengt = random.randint(minNum,maxNum)
        for i in range(1,strLengt,1):
            chrTem = chr(random.randint(65,92))
            randstr = randstr + chrTem
        return randstr


    #从正常的代码列表中随机一句,当废话用,混淆效果更好.
    '''吧代码生成一个pprint py库在这调用,下版.'''
    JsCodeList = [
        'new Function("a", "b", "return a+b;");',
        'var ybdetof5 = new ActiveXObject("Scripting.FileSystemObject");'
        ]
    def randCodeLine(self,CodeList = []):
        if len(CodeList) == 0:
            CodeList.append('Life is short,U need the Eng')
        return random.choice(CodeList)


    #随机字符:'m'
    def randChr(self):
        return random.choice('abcdefghijklmnopqrstuvwxyz!@#$%^&*()')


    #随机字符list:['n','f','y','b']
    def randChrEx_List(self):
        return random.sample('zyxwvutsrqponmlkjihgfedcba', 5)


    #随机生成一个List:['nihao','ahoo','a']
    def randStrList(self,minNum=3,maxNum = 5):
        if minNum > maxNum:
            minNum = min(minNum,maxNum)
            maxNum = max(minNum,maxNum)

        arrLengt= random.randint(minNum,maxNum)
        arrList =[]
        for i in range(0,arrLengt,1):
            arrList.append(self.randStr_az())
        return arrList


    #生成数组模式1:['xu', 'm', 'l', 'ahoo', 'milh'][3]
    def randStrArrary(self,itemstr):
        arrList     = self.randStrList()
        index       = random.randint(0,len(arrList)-1)
        arrList[index] = itemstr
        return str(arrList) + '[' + str(index) + ']'


    #生成数组模式2: ('var ab = "ahoo"', "['df', ab, 'puu', 'chx', 'avu'][1]")
    def randStrArryEx_var(self,itemstr):
        arrList     = self.randStrList()
        index       = random.randint(0,len(arrList)-1)

        #随机一个varName,确保不要重复.
        varName = self.randStr_az(3,7)
        while varName in self.varStrRemoveRepeatL:
            varName = self.randStr_az(4,8)
        varStrItem = 'var '+ varName + ' = "' + itemstr + '"\n'
        self.varStrRemoveRepeatL.append(varName)

        #生成数组
        arrList[index] = varName
        replaceTemp = str(arrList) + '[' + str(index) + ']'

        #替换为变量形式.
        replaceTemp_pattern = re.compile('\''+varName+'\'')
        replaceTemp = replaceTemp_pattern.sub(varName,replaceTemp)

        return varStrItem , replaceTemp


    #生成数组模式3: 将一句话(自定义特殊格式)分割为数组加变量: 
    #'open#@process' ---> {'var ax = "open"' : '['nihao',ax,'a'][1]',,}
    def randSelfTypeStr2ArraryTypeStr(self,SelfTypeStr):
        replacestr = ''
        varStrList = []
        for i_list_split_line in SelfTypeStr.split('#@'):
            varStr,arrStr = self.randStrArryEx_var(i_list_split_line)
            replacestr = replacestr + arrStr + ' + '
            varStrList.append(varStr)
        return varStrList,replacestr[:-3]


    #随机一个function vbs的
    def randFun_Vb(self):
        return 'waitfortest\r\n coming~\r\n'
        pass


    def randFunList_Vb(self,MaxLen):
        funList=[]
        for i in range(0,MaxLen,1):
            funList.append(self.randFun_Vb())
        return funList

0x3 StrSplitEng

#!/usr/bin/env python3
# -*- coding: utf-8 -*-

__author__ = 'ahoo'

import sys
import io
import os
import codecs
import re
import shutil
import random
import logging
import logging.config

#--------------------文件配置----------
logfilePath = os.path.join(os.path.dirname(__file__), 'logging.conf')
logging.config.fileConfig('logging.conf')
logging.getLogger()


#sys.stdout = io.TextIOWrapper(
#               sys.stdout.buffer,
#               encoding='utf-8') #改变标准输出的默认编码


PutPath = '24_analysis.txt'             #JsVirus文件(卡饭精睿包2016.12.16.24).
OutPath = '24_EngRefactorObfuscate.vir' #提取到的文件.

class StrSplitEng(object):

    def __init__(self):
        self.author = 'ahoo'
        pass

    #分割line返回list:'Scripting.FileSystemObject'-->['Sc', 'ri', 'pt', 'ing', '.Fil', 'eSys', 'temO', 'bj', 'ect']
    def StrSplitLine2List(self,strForSplit):
        result = []
        strleng = len(strForSplit)
        if len(strForSplit) == 0:
            pass
        elif strleng <= 4:
            result.append(strForSplit)
        else:
            randlen = random.randint(2,4)
            result.append(strForSplit[:randlen])
            tempList = self.StrSplitLine2List(strForSplit[randlen:])

            for j in tempList:
                result.append(j)
        return result


    #分割一个line中的元素返回以'#@'分割的字符串.
    # 'Scripting.FileSystemObject'-->Scri#@pti#@ng.F#@ileS#@yst#@em#@Ob#@ject
    def StrSplitLine2SelfTypeStr(self,strForSplit1):
        resultStr = ''
        tempList = self.StrSplitLine2List(strForSplit1)
        if len(tempList) == 0:
            return resultStr    
        else:
            for i in tempList:
                resultStr  = resultStr + i + '#@'

        return resultStr[:-2]


    #分割一个list中的元素返回list.    
    def StrSplit(self,strForSplit = []):
        strSplitList = []
        result = []
        if len(strForSplit) == 0:
            #print(strForSplit)
            return result

        for i in strForSplit:
            strleng = len(i)
            if strleng <= 4:
                result.append(i)
            else:
                #randlen = random.randint(2,int(strleng/2))
                randlen = random.randint(2,4)
                #print(randlen)
                #print(i[:randlen])
                #print(i[randlen:])
                strSplitList.append(i[:randlen])
                strSplitList.append(i[randlen:])
                #print(strSplitList)
                tempList = StrSplit(strSplitList)

                for j in tempList:
                    result.append(j)
        #print('result\n')
        #print(result)
        return result

0x4.ObfuscateEng

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
'''
    v1.0 --发布.
'''
__author__ = 'ahoo'

import sys
import io
import os
import codecs
import re
import shutil
import random
import logging
import logging.config
from Obfuscateer.FileRead2Write import FileRead2Write 
from Obfuscateer.RandomSequence import RandomSequence
from Obfuscateer.StrSplitEng    import StrSplitEng

#--------------------文件配置----------
logfilePath = os.path.join(os.path.dirname(__file__), 'logging.conf')
logging.config.fileConfig('logging.conf')
logging.getLogger()


#sys.stdout = io.TextIOWrapper(
#               sys.stdout.buffer,
#               encoding='utf-8') #改变标准输出的默认编码


class ObfuscateMethod(object):

    rdClass = RandomSequence()
    spClass = StrSplitEng()

    def __init__(self):
        self.author = 'ahoo'


    #1.替换""之间的. 
    def ObfuscateQuotes(self,InputList=[]):
        writeListTemp = []
        varStrTemp = []
        if len(InputList) == 0:
            return varStrTemp,writeListTemp

        for lin1 in InputList:
            if lin1 == "":
                #writeListTemp.append(lin1)
                pass
            else:
                for m in re.findall('"\s*[^"\,\+]+\s*"',lin1):
                    if len(m) >= 2:
                        pattern_quotes = re.compile(m[1:-1])
                        strtemp = self.spClass.StrSplitLine2SelfTypeStr(m[1:-1])
                        varlsit,replaceTempstr =  self.rdClass.randSelfTypeStr2ArraryTypeStr(strtemp)
                        #print(replaceTempstr1)
                        lin1 = pattern_quotes.sub(replaceTempstr,lin1,count=1)
                        #print(lin1)
                        for varItem in varlsit:
                            varStrTemp.append(varItem)
                    else:
                        lin1 = m
                writeListTemp.append(lin1)
                pass

        return varStrTemp,writeListTemp

    #2.替换[]和()之间的.
    def OufuscateBracket(self,InputList1=[]):
        writeListTemp1 = []
        varStrTemp1 = []

        if len(InputList1) == 0:
            return varStrTemp1,writeListTemp1

        for line in InputList1:
            if line == "":
                #writeListTemp1.append(line)
                pass
            else:
                for i in re.findall('\[(\s*"[^\[\]\(\)]+"\s*)\]',line):
                    pattern_bracket = re.compile(i)
                    strtemp = self.spClass.StrSplitLine2SelfTypeStr(i)
                    varlsit,replaceTempstr = self.rdClass.randSelfTypeStr2ArraryTypeStr(strtemp)
                    line = pattern_bracket.sub(replaceTempstr,line,count=1)
                    for varItem in varlsit:
                            varStrTemp1.append(varItem)

                for j in re.findall('\((\s*"[^\[\]\(\)]+"\s*)\)',line):
                    pattern_bracket = re.compile(j)
                    strtemp = self.spClass.StrSplitLine2SelfTypeStr(j)
                    varlsit,replaceTempstr = self.rdClass.randSelfTypeStr2ArraryTypeStr(strtemp)
                    line = pattern_bracket.sub(replaceTempstr,line,count=1)
                    for varItem in varlsit:
                            varStrTemp1.append(varItem)
                writeListTemp1.append(line) 
                pass

        return varStrTemp1,writeListTemp1

class EngCla(object):

    varStr = []

    def __init__(self,PutPath,OutPath):
        self.author = 'ahoo'
        self.PutPath = PutPath
        self.OutPath = OutPath
    def Eng(self):
        try:

            fpClass = FileRead2Write()
            obfuCla = ObfuscateMethod()

            #1.读取文件到LineList
            myInputList = fpClass.ReadInputFile(self.PutPath)


            #2.替换.
            varTem,writeTem  = obfuCla.ObfuscateQuotes(myInputList)
            #varTem1,__  = obfuCla.OufuscateBracket(myInputList)

            fpClass.WriteList2List(varTem,self.varStr)
            #fpClass.WriteList2List(varTem1,self.varStr)

            #logging.debug(varTem)

            #3.输出
            fpClass.WriteOutputFileEx_ListShuffle(self.OutPath,self.varStr)
            fpClass.WriteOutputFile(self.OutPath,writeTem)
            #fpClass.OpenOutPath(self.OutPath)

            logging.info('The Code has been Splited,there is my advice! Thanks!')
            print('The Code has been Splited,there is my advice! Thanks!')

        except :                                            #except Exception as e:  logging.debug(e)
            logging.exception('Eng has a exception info.')

        return True

0x5 调用例子

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
'''
'''
__author__ = 'ahoo'

import os
import logging
import logging.config
from Obfuscateer.ObfuscateEngM  import EngCla
#1.NeedReplaceFile--".C:\Users\tech\AppData\Local\Programs\Python\Python35\Lib\site-packages\Obfuscateer-0.1-py3.5.egg\Obfuscateer\ObfuscateEngM.py" --Fix---:from Obfuscateer.FileRead2Write import FileRead2Write 

#--------------------文件配置----------
logfilePath = os.path.join(os.path.dirname(__file__), 'logging.conf')
logging.config.fileConfig('logging.conf')
logging.getLogger()


PutPath = 'Sample\\24_analysis.txt'             #JsVirus文件(卡饭精睿包2016.12.16.24).
OutPath = 'Sample\\24_EngRefactorObfuscate.vir' #提取到的文件.



if __name__ == '__main__':
    eng = EngCla(PutPath,OutPath)
    eng.Eng()
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值