#coding:utf8
#register.py
#功能說(shuō)明：用戶運(yùn)行程序后，自動(dòng)檢測(cè)認(rèn)證狀態(tài)，如果未經(jīng)認(rèn)證，就需要注冊(cè)。注冊(cè)過(guò)程是用戶將程序運(yùn)行后顯示的機(jī)器碼（卷序號(hào)）發(fā)回給管理員，管理員通過(guò)加密后生成加密文件或字符串給回用戶。
#每次登錄，在有注冊(cè)文件或者注冊(cè)碼的情況下，軟件就會(huì)通過(guò)DES和base64解碼，如果解碼后和重新獲取的機(jī)器碼一致，則通過(guò)認(rèn)證，進(jìn)入主程序。
import base64
import win32api
from pyDes import *
#from binascii import a2b_hex #如果需要用二進(jìn)制編碼保存注冊(cè)碼和注冊(cè)文件可以使用binascii轉(zhuǎn)換
class register:
def __init__(self):
self.Des_Key = "BHC#@*UM" # Key
self.Des_IV = "\x22\x33\x35\x81\xBC\x38\x5A\xE7" # 自定IV向量
#獲取C盤(pán)卷序列號(hào)
#使用C盤(pán)卷序列號(hào)的優(yōu)點(diǎn)是長(zhǎng)度短，方便操作，比如1513085707，但是對(duì)C盤(pán)進(jìn)行格式化或重裝電腦等操作會(huì)影響C盤(pán)卷序列號(hào)。
#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 CVolumeSerialNumber
if CVolumeSerialNumber:
return str(CVolumeSerialNumber) #number is long type，has to be changed to str for comparing to content after.
else:
return 0
#使用DES加base64的形式加密
#考慮過(guò)使用M2Crypto和rsa，但是都因?yàn)樵趙indows環(huán)境中糟糕的安裝配置過(guò)程而放棄
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) #轉(zhuǎn)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 DecryptStr
return DecryptStr
#獲取注冊(cè)碼，驗(yàn)證成功后生成注冊(cè)文件
def regist(self):
key = raw_input('please input your register code: ')
#由于輸入類(lèi)似“12”這種不符合base64規(guī)則的字符串會(huì)引起異常，所以需要增加輸入判斷
#while key
if 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 str
key_decrypted=str(self.DesDecrypt(base64.b64decode(key)))
#print chardet.detect(key_decrypted)
#print key_decrypted
#type(key_decrypted) is str
if 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."
#讀寫(xiě)文件要加判斷
with open('./register','w') as f:
f.write(key)
f.close()
return True
else:
return False
else:
return False
else:
self.regist()
return False
def checkAuthored(self):
content=self.getCVolumeSerialNumber()
checkAuthoredResult = 0
#讀寫(xiě)文件要加判斷
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 = 1
else:
checkAuthoredResult = -1
else:
checkAuthoredResult = -2
else:
checkAuthoredResult = -3
else:
self.regist()
except IOError:
print IOError
print checkAuthoredResult
return checkAuthoredResult
if __name__ == '__main__':
reg=register()
reg.regist()

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

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