Beispiel #1
0
 def generate_crypto(self):
     self.crypto = Crypto()
     self.keyloader = KeyLoader()
     self.load_keys()
Beispiel #2
0
class Settings(ConfigurationConsumer):
    '''Singleton-ish object which holds information about this user'''

    def __init__(self, alias):
        self.alias = alias
        self.get_uid()
        self.generate_crypto()

    def generate_crypto(self):
        self.crypto = Crypto()
        self.keyloader = KeyLoader()
        self.load_keys()

    def get_uid(self):
        '''Load UID from users.json or create it'''
        users_filename = self.config_path('users.json')
        users = json.load(open(users_filename,'r'))
        try:
            self.uid = users[self.alias]
        except Exception as msg:
            self.uid = self.generate_uid(32)
            Display.warn('Generated New UID: {0}'.format(self.uid))
            users[self.alias] = self.uid
            with open(users_filename,'w') as out:
                json.dump(users, out)

    def generate_uid(self, size=6, chars=string.ascii_uppercase+string.digits):
        '''Randomly generate a UID for this user'''
        return ''.join(random.choice(chars) for x in range(size))

    def generate_keys(self):
        '''Generate new RSA-2048 keys for this client (accurate off)'''
        Display.log('Generating new keys...')
        self.crypto.generate()
        self.save_keys()
        Display.log('Done.\n')

    def load_keys(self):
        '''Load public and private key from key.json; generates if empty'''
        filename = self.config_path('.data/{0}.pyna'.format(self.uid))
        if not os.path.isfile(filename):
            self.generate_keys()
            return
        self.password_loop(filename)

    def password_loop(self, filename):
        '''Loops until the correct password is entered'''
        attempts = 0
        while attempts < 3:
            password = input('Please enter password to unlock ({0} attempts remaining):  '.format(3-attempts))
            try:
                with open(filename,'rb') as enc_data:
                    encrypted = pickle.load(enc_data)
                    data = self.keyloader.unlock_with_password(password, encrypted)
                    self.crypto.load(data['privateKey'])
                    Display.log('User settings loaded.')
                    return
            except:
                Display.warn('Wrong password')
                attempts += 1

        Display.error('You have exceeded the number of attempts to log in as this user.')
        sys.exit(1)


    def save_keys(self):
        '''Save public and private key to key.json'''
        password = input('Please enter password to lock:  ')
        privout = self.crypto.private.exportKey('PEM').decode('utf-8')
        data = json.dumps({"privateKey":privout})
        locked = self.keyloader.lock_with_password(password, data)
        keyfile = self.config_path('.data/{0}.pyna'.format(self.uid))
        with open(keyfile,'wb') as auth:
            pickle.dump(locked, auth)
Beispiel #3
0
 def createCrypto(self):
     self.crypto = Crypto()
     self.keyloader = KeyLoader()
     self.loadKeys()