def __init__(self,
directory,
cipherKey=None,
macKey=None,
counter=None,
knockPort=None):
self.counterFile = None
self.directory = directory
self.name = directory.rstrip('/').split('/')[-1]
if (cipherKey == None):
self.deserialize()
else:
self.cipherKey = cipherKey
self.macKey = macKey
self.counter = counter
self.knockPort = knockPort
self.cryptoEngine = CryptoEngine(self, self.cipherKey, self.macKey,
self.counter)
def __init__(self, directory, cipherKey=None, macKey=None, counter=None, knockPort=None):
self.counterFile = None
self.directory = directory
self.name = directory.rstrip('/').split('/')[-1]
if cipherKey is None:
self.deserialize()
else:
self.cipherKey = cipherKey
self.macKey = macKey
self.counter = counter
self.knockPort = knockPort
self.cryptoEngine = CryptoEngine(self, self.cipherKey, self.macKey, self.counter)
class Profile:
def __init__(self, directory, cipherKey=None, macKey=None, counter=None, knockPort=None):
self.counterFile = None
self.directory = directory
self.name = directory.rstrip('/').split('/')[-1]
if cipherKey is None:
self.deserialize()
else:
self.cipherKey = cipherKey
self.macKey = macKey
self.counter = counter
self.knockPort = knockPort
self.cryptoEngine = CryptoEngine(self, self.cipherKey, self.macKey, self.counter)
def deserialize(self):
self.cipherKey = self.loadCipherKey()
self.macKey = self.loadMacKey()
self.counter = self.loadCounter()
self.knockPort = self.loadConfig()
def serialize(self):
self.storeCipherKey()
self.storeMacKey()
self.storeCounter()
self.storeConfig()
# Getters And Setters
def getIPAddrs(self):
return self.ipAddressList
def setIPAddrs(self, ipAddressList):
self.ipAddressList = ipAddressList
def getName(self):
return self.name
def getDirectory(self):
return self.directory
def getKnockPort(self):
return self.knockPort
def setCounter(self, counter):
self.counter = counter
# Encrypt And Decrypt
def decrypt(self, ciphertext, windowSize):
return self.cryptoEngine.decrypt(ciphertext, windowSize)
def encrypt(self, plaintext):
return self.cryptoEngine.encrypt(plaintext)
# Serialization Methods
def loadCipherKey(self):
return self.loadKey(self.directory + "/cipher.key")
def loadMacKey(self):
return self.loadKey(self.directory + "/mac.key")
def loadCounter(self):
# Privsep bullshit...
if self.counterFile is None:
self.counterFile = open(self.directory + "/counter", 'r+')
counter = self.counterFile.readline()
counter = counter.rstrip("\n")
return int(counter)
def loadConfig(self):
config = ConfigParser.SafeConfigParser()
config.read(self.directory + "/config")
return config.get('main', 'knock_port')
def loadKey(self, keyFile):
file = open(keyFile, 'r')
key = binascii.a2b_base64(file.readline())
file.close()
return key
def storeCipherKey(self):
self.storeKey(self.cipherKey, self.directory + "/cipher.key")
def storeMacKey(self):
self.storeKey(self.macKey, self.directory + "/mac.key")
def storeCounter(self):
# Privsep bullshit...
if self.counterFile is None:
self.counterFile = open(self.directory + '/counter', 'w')
self.setPermissions(self.directory + '/counter')
self.counterFile.seek(0)
self.counterFile.write(str(self.counter) + "\n")
self.counterFile.flush()
def storeConfig(self):
config = ConfigParser.SafeConfigParser()
config.add_section('main')
config.set('main', 'knock_port', str(self.knockPort))
configFile = open(self.directory + "/config", 'w')
config.write(configFile)
configFile.close()
self.setPermissions(self.directory + "/config")
def storeKey(self, key, path):
file = open(path, 'w')
file.write(binascii.b2a_base64(key))
file.close()
self.setPermissions(path)
# Permissions
def setPermissions(self, path):
os.chmod(path, stat.S_IRUSR | stat.S_IWUSR)
# Debug
def printHex(self, val):
for c in val:
print "%#x" % ord(c),
print ""
class Profile:
def __init__(self,
directory,
cipherKey=None,
macKey=None,
counter=None,
knockPort=None):
self.counterFile = None
self.directory = directory
self.name = directory.rstrip('/').split('/')[-1]
if (cipherKey == None):
self.deserialize()
else:
self.cipherKey = cipherKey
self.macKey = macKey
self.counter = counter
self.knockPort = knockPort
self.cryptoEngine = CryptoEngine(self, self.cipherKey, self.macKey,
self.counter)
def deserialize(self):
self.cipherKey = self.loadCipherKey()
self.macKey = self.loadMacKey()
self.counter = self.loadCounter()
self.knockPort = self.loadConfig()
def serialize(self):
self.storeCipherKey()
self.storeMacKey()
self.storeCounter()
self.storeConfig()
# Getters And Setters
def getIPAddrs(self):
return self.ipAddressList
def setIPAddrs(self, ipAddressList):
self.ipAddressList = ipAddressList
def getName(self):
return self.name
def getDirectory(self):
return self.directory
def getKnockPort(self):
return self.knockPort
def setCounter(self, counter):
self.counter = counter
# Encrypt And Decrypt
def decrypt(self, ciphertext, windowSize):
return self.cryptoEngine.decrypt(ciphertext, windowSize)
def encrypt(self, plaintext):
return self.cryptoEngine.encrypt(plaintext)
# Serialization Methods
def loadCipherKey(self):
return self.loadKey(self.directory + "/cipher.key")
def loadMacKey(self):
return self.loadKey(self.directory + "/mac.key")
def loadCounter(self):
# Privsep bullshit...
if (self.counterFile == None):
self.counterFile = open(self.directory + "/counter", 'r+')
counter = self.counterFile.readline()
counter = counter.rstrip("\n")
return int(counter)
def loadConfig(self):
config = ConfigParser.SafeConfigParser()
config.read(self.directory + "/config")
return config.get('main', 'knock_port')
def loadKey(self, keyFile):
file = open(keyFile, 'r')
key = binascii.a2b_base64(file.readline())
file.close()
return key
def storeCipherKey(self):
self.storeKey(self.cipherKey, self.directory + "/cipher.key")
def storeMacKey(self):
self.storeKey(self.macKey, self.directory + "/mac.key")
def storeCounter(self):
# Privsep bullshit...
if (self.counterFile == None):
self.counterFile = open(self.directory + '/counter', 'w')
self.setPermissions(self.directory + '/counter')
self.counterFile.seek(0)
self.counterFile.write(str(self.counter) + "\n")
self.counterFile.flush()
def storeConfig(self):
config = ConfigParser.SafeConfigParser()
config.add_section('main')
config.set('main', 'knock_port', str(self.knockPort))
configFile = open(self.directory + "/config", 'w')
config.write(configFile)
configFile.close()
self.setPermissions(self.directory + "/config")
def storeKey(self, key, path):
file = open(path, 'w')
file.write(binascii.b2a_base64(key))
file.close()
self.setPermissions(path)
# Permissions
def setPermissions(self, path):
os.chmod(path, stat.S_IRUSR | stat.S_IWUSR)
# Debug
def printHex(self, val):
for c in val:
print "%#x" % ord(c),
print ""
