def checkChallenge(self, originalChallenge, cipheredChallenge, crypto_v1=0):
if crypto_v1:
from agents.CryptoLib_v1 import crypto, CryptoLib, aggregatorKey, aes_decrypt
else:
from agents.CryptoLib import crypto, CryptoLib, aggregatorKey, aes_decrypt
# get cryptolib instance
crypt = CryptoLib()
return crypt.verifySignatureRSA(originalChallenge, cipheredChallenge, self.public_key(crypto_v1))
def checkChallenge(self,
originalChallenge,
cipheredChallenge,
crypto_v1=0):
if crypto_v1:
from agents.CryptoLib_v1 import crypto, CryptoLib, aggregatorKey, aes_decrypt
else:
from agents.CryptoLib import crypto, CryptoLib, aggregatorKey, aes_decrypt
# get cryptolib instance
crypt = CryptoLib()
return crypt.verifySignatureRSA(originalChallenge, cipheredChallenge,
self.public_key(crypto_v1))
def initLogin(self, ip, port, crypto_v1):
from geoip.models import IPRange
if crypto_v1:
from agents.CryptoLib_v1 import crypto, CryptoLib, aggregatorKey, aes_decrypt
else:
from agents.CryptoLib import crypto, CryptoLib, aggregatorKey, aes_decrypt
# get new challenge
if self.banned:
return False
from geoip.models import IPRange
# Check if network is banned
iprange = IPRange.ip_location(ip)
if iprange and iprange.banned:
return False
crypto = CryptoLib()
challenge = crypto.generateChallenge()
loginProcess = LoginProcess()
loginProcess.agent_id = self.id
loginProcess.ip = ip
loginProcess.port = port
loginProcess.challenge = challenge
loginProcess.save()
return loginProcess
def decodeMessage(self, encodedMessage):
# get cryptolib instance
crypt = CryptoLib()
message = crypt.decodeAES(encodedMessage, self.AESKey)
return message
def encodeMessageRSA(self, message):
# get cryptolib instance
crypt = CryptoLib()
encoded = crypt.encodeRSAPublicKey(message, self.public_key)
return encoded
def encodeMessage(self, message):
# get cryptolib instance
crypt = CryptoLib()
encoded = crypt.encodeAES(message, self.AESKey)
return encoded
def decodeMessage(self, encodedMessage):
# get cryptolib instance
crypt = CryptoLib()
message = crypt.decodeAES(encodedMessage, self.AESKey)
return message
def encodeMessageRSA(self, message):
# get cryptolib instance
crypt = CryptoLib()
encoded = crypt.encodeRSAPublicKey(message, self.public_key)
return encoded
def encodeMessage(self, message):
# get cryptolib instance
crypt = CryptoLib()
encoded = crypt.encodeAES(message, self.AESKey)
return encoded
