#coding:utf8#register.py#功能說明：用戶運行程序后，自動檢測認證狀態，如果未經認證，就需要注冊。注冊過程是用戶將程序運行后顯示的機器碼（卷序號）發回給管理員，管理員通過加密后生成加密文件或字符串給回用戶。#每次登錄，在有注冊文件或者注冊碼的情況下，軟件就會通過DES和base64解碼，如果解碼后和重新獲取的機器碼一致，則通過認證，進入主程序。import base64import win32apifrom pyDes import *#from binascii import a2b_hex #如果需要用二進制編碼保存注冊碼和注冊文件可以使用binascii轉換class register:def __init__(self):self.Des_Key = "BHC#@*UM" # Keyself.Des_IV = "/x22/x33/x35/x81/xBC/x38/x5A/xE7" # 自定IV向量#獲取C盤卷序列號#使用C盤卷序列號的優點是長度短，方便操作，比如1513085707，但是對C盤進行格式化或重裝電腦等操作會影響C盤卷序列號。#win32api.GetVolumeInformation(Volume Name, Volume Serial Number, Maximum Component Length of a file name, Sys Flags, File System Name)#return('', 1513085707, 255, 65470719, 'NTFS'),volume serial number is 1513085707.def getCVolumeSerialNumber(self):CVolumeSerialNumber=win32api.GetVolumeInformation("C://")[1]#print chardet.detect(str(CVolumeSerialNumber))#print CVolumeSerialNumberif CVolumeSerialNumber:return str(CVolumeSerialNumber) #number is long type，has to be changed to str for comparing to content after.else:return 0#使用DES加base64的形式加密#考慮過使用M2Crypto和rsa，但是都因為在windows環境中糟糕的安裝配置過程而放棄def DesEncrypt(self,str):k = des(self.Des_Key, CBC, self.Des_IV, pad=None, padmode=PAD_PKCS5)EncryptStr = k.encrypt(str)#EncryptStr = binascii.unhexlify(k.encrypt(str))return base64.b64encode(EncryptStr) #轉base64編碼返回#des解碼def DesDecrypt(self,str):k = des(self.Des_Key, CBC, self.Des_IV, pad=None, padmode=PAD_PKCS5)DecryptStr = k.decrypt(str)#DecryptStr = a2b_hex(k.decrypt(str))print DecryptStrreturn DecryptStr#獲取注冊碼，驗證成功后生成注冊文件def regist(self):key = raw_input('please input your register code: ')#由于輸入類似“12”這種不符合base64規則的字符串會引起異常，所以需要增加輸入判斷#while keyif key:content = self.getCVolumeSerialNumber() //number has been changed to str type after use str() #print chardet.detect(content)#print type(content)#print content#type(key_decrypted) is strkey_decrypted=str(self.DesDecrypt(base64.b64decode(key)))#print chardet.detect(key_decrypted)#print key_decrypted#type(key_decrypted) is strif content!=0 and key_decrypted!=0:if content != key_decrypted:print "wrong register code, please check and input your register code again:"self.regist()elif content==key_decrypted:print "register succeed."#讀寫文件要加判斷with open('./register','w') as f:f.write(key)f.close()return Trueelse:return Falseelse:return Falseelse:self.regist()return Falsedef checkAuthored(self):content=self.getCVolumeSerialNumber()checkAuthoredResult = 0#讀寫文件要加判斷try:f=open('./register','r')if f:key=f.read()if key:key_decrypted=self.DesDecrypt(base64.b64decode(key))if key_decrypted:if key_decrypted == content:checkAuthoredResult = 1else:checkAuthoredResult = -1else:checkAuthoredResult = -2else:checkAuthoredResult = -3else:self.regist()except IOError:print IOErrorprint checkAuthoredResultreturn checkAuthoredResultif __name__ == '__main__':reg=register()reg.regist()

#CVolumeSerialNumber: 1513085707#after encryption: ro5RVXZoP0KmnogYDeepUg==#the HardDiskNumber: 32535332584e4741343536393237204620202020#after encryption: MzI1MzUzMzI1ODRlNDc0MTM0MzUzNjM5MzIzNzIwNDYyMDIwMjAyMA==

#雖然使用wmi可以獲取磁盤序列號，可是磁盤序列號是3253533258**************3237204620202020，加密后太長，不方便操作，所以棄置import wmidef getHardDiskNumber(self):c = wmi.WMI()for physical_disk in c.Win32_DiskDrive():return physical_disk.SerialNumber