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