mixKeystore.py

#!/usr/bin/python

import subprocess
import logging, logging.config

try:
    from lamson.cron import getCronTab, CronEvent
except:
    def getCronTab():
        return None

from mixKey import *
from mixConfig import getRemailerConfig

_mixKeyStore = None
def getKeyStore():
    global _mixKeyStore
    if not _mixKeyStore:
        _mixKeyStore = loadFreshKeystore()
        
    return _mixKeyStore
    
def loadFreshKeystore():
    _mixKeyStore = MixKeyStore()
        
    # This assumes only one private key will be found in the file.
    try:
        f = open(getRemailerConfig()['filelocations']['secring.mix'], 'r')
    except IOError:
        f = open('../../' + getRemailerConfig()['filelocations']['secring.mix'], 'r')
    privateKeyLines = f.readlines()
    f.close()
    _mixKeyStore.addKey(privateKeyLines, getRemailerConfig('remailerkeypassword'))
    
    try:
        f = open(getRemailerConfig()['filelocations']['pubring.mix'], 'r')
    except IOError:
        f = open('../../' + getRemailerConfig()['filelocations']['pubring.mix'], 'r')
    lines = f.readlines()
    f.close()
    key = []
    for l in lines:
        l = l.strip()
        if l == "-----End Mix Key-----":
            key.append(l)
            _mixKeyStore.addKey(key)
            key = []
        else:
            key.append(l)
    return _mixKeyStore
    
def refreshStats():
    devnull = open("/dev/null", "w")
    subprocess.call(["wget", getRemailerConfig('statsdirectory') + "mlist.txt"], stderr=devnull)
    subprocess.call(["wget", getRemailerConfig('statsdirectory') + "rlist.txt"], stderr=devnull)
    subprocess.call(["wget", getRemailerConfig('statsdirectory') + "pubring.mix"], stderr=devnull)
    subprocess.call(["wget", getRemailerConfig('statsdirectory') + "pgp-all.asc"], stderr=devnull)
    subprocess.call(["mv", "mlist.txt", "rlist.txt", "pubring.mix", "pgp-all.asc", "app/data/"])
    devnull.close()

def refreshKeyStore():
    global _mixKeyStore
    _mixKeyStore = loadFreshKeystore()

class MixKeyStore:
    keystore = {}
    def __init__(self):
        cron = getCronTab()
        if cron:
            cron.add(CronEvent(refreshStats, 0, range(0,24)))
        else:
            logging.debug("No Cron Object found, cannot add refresh stats cron job")
    def addKey(self, keylines, passphrase=""):
        key = parseMixKey(keylines, passphrase)

        if key.KeyId in self.keystore:
            if isinstance(self.keystore[key.KeyId], PublicMixKey) and isinstance(key, PrivateMixKey):
                logging.info("Replacing Public Key " + key.KeyId + " in self.keystore with Private Key")
                key.loadPublicCapabilities(self.keystore[key.KeyId])
                self.keystore[key.KeyId] = key
            else:
                self.keystore[key.KeyId].loadPublicCapabilities(key)
                logging.info("Loading Public Key Properties of " + key.KeyId + " into Private Key in self.keystore ")
        else:
            self.keystore[key.KeyId] = key
            logging.info("Adding Key " + key.KeyId + " to self.keystore")
    def getKey(self, keyid):
        if keyid in self.keystore:
            return self.keystore[keyid]
        else:
            return None
    def getPublicKey(self, keyid):
        if keyid in self.keystore and isinstance(self.keystore[keyid], PrivateMixKey):
            return self.keystore[keyid].getPublicMixKey()
        elif keyid in self.keystore and isinstance(self.keystore[keyid], PublicMixKey):
            return self.keystore[keyid]
        else:
            return None
    def getPrivateKey(self, keyid):
        if keyid in self.keystore and isinstance(self.keystore[keyid], PrivateMixKey):
            return self.keystore[keyid]
        else:
            return None
    def listPrivateKeys(self):
        keys = []
        for k in self.keystore:
            if isinstance(self.keystore[k], PrivateMixKey):
                keys.append(self.keystore[k])
        return keys
    def listPublicKeys(self):
        keys = []
        for k in self.keystore:
            if isinstance(self.keystore[k], PublicMixKey):
                keys.append(self.keystore[k])
            else:
                keys.append(self.getPublicKey(k))
        return keys

if __name__ == "__main__":
    logging.config.fileConfig("../../config/test_logging.conf")
    keys = getKeyStore().listPublicKeys()
    keys.sort()
    for k in keys:
        print k.getMixKeyHeader()