class CommandHandler(object):
def __init__(self):
self.gpg = None
self.gmail = None
self.gpgmail = None
self.initialized = False
self.logger = logging.getLogger('CommandHandler')
def parse(self, bundle):
if not 'command' in bundle:
self.logger.error("no command in bundle {}".format(bundle))
return None
result = None
command = bundle["command"]
if command == 'init':
result = self.init(bundle)
else:
# do nothing if not intialized
if not self.initialized:
return False
if command == 'verify':
result = self.verify(bundle)
elif command == 'sign':
result = self.sign(bundle)
elif command == 'import':
result = self.import_key(bundle)
return result
def init(self, bundle):
if not 'username' in bundle:
return False
self.gpg = GPG(use_agent=True)
self.gpgmail = GPGMail(self.gpg)
self.gmail = Gmail(bundle['username'])
self.initialized = True
return {'version': VERSION}
def verify(self, bundle):
if not 'id' in bundle:
return False
id = bundle['id']
def _verify():
mail = self.gmail.get(id)
return self.gpgmail.verify(mail)
# verify the message only if bundle['force'] = true
if 'force' in bundle and bundle['force']:
return _verify()
# or content_type of message is 'multipart/signed'
headers = self.gmail.get_headers(id, ['Content-Type', 'Message-ID'])
if 'Content-Type' in headers:
content_type = headers['Content-Type']
if content_type.find('multipart/signed') >= 0:
return _verify()
# or if message is multipart and it may contain a pgp-signature
is_multipart = content_type.find('multipart/') >= 0
if (is_multipart and 'Message-ID' in headers):
rfc822msgid = headers['Message-ID']
query = '("BEGIN PGP SIGNATURE" "END PGP SIGNATURE")'
match = self.gmail.message_matches(id, query, rfc822msgid)
if match:
return _verify()
# else
return None
def sign(self, bundle):
if not 'id' in bundle:
return False
result = False
id = bundle['id']
try:
draft = self.gmail.get(id)
new_message = self.gpgmail.sign(draft)
if new_message:
self.gmail.send(id, new_message)
result = True
except Exception, e:
self.logger.exception(e)
finally:
class CommandHandler(object):
def __init__(self):
self.gpg = None
self.gmail = None
self.gpgmail = None
self.initialized = False
self.logger = logging.getLogger('CommandHandler')
def parse(self, bundle):
if not 'command' in bundle:
self.logger.error("no command in bundle {}".format(bundle))
return None
result = None
command = bundle["command"]
if command == 'init':
result = self.init(bundle)
else:
# do nothing if not intialized
if not self.initialized:
return False
if command == 'verify':
result = self.verify(bundle)
elif command == 'sign':
result = self.sign(bundle)
elif command == 'import':
result = self.import_key(bundle)
return result
def init(self, bundle):
if not 'username' in bundle:
return False
self.gpg = GPG(use_agent=True)
self.gpgmail = GPGMail(self.gpg)
self.gmail = Gmail(bundle['username'])
self.initialized = True
return {'version': VERSION}
def verify(self, bundle):
if not 'id' in bundle:
return False
id = bundle['id']
def _verify():
mail = self.gmail.get(id)
return self.gpgmail.verify(mail)
# verify the message only if bundle['force'] = true
if 'force' in bundle and bundle['force']:
return _verify()
# or content_type of message is 'multipart/signed'
headers = self.gmail.get_headers(id, ['Content-Type', 'Message-ID'])
if 'Content-Type' in headers:
content_type = headers['Content-Type']
if content_type.find('multipart/signed') >= 0:
return _verify()
# or if message is multipart and it may contain a pgp-signature
is_multipart = content_type.find('multipart/') >= 0
if (is_multipart and 'Message-ID' in headers):
rfc822msgid = headers['Message-ID']
query = 'filename:asc || filename:gpg || filename:signature || ("BEGIN PGP SIGNATURE" "END PGP SIGNATURE")'
match = self.gmail.message_matches(id, query, rfc822msgid)
if match:
return _verify()
# else
return None
def sign(self, bundle):
if not 'id' in bundle:
return False
result = False
id = bundle['id']
try:
draft = self.gmail.get(id)
new_message = self.gpgmail.sign(draft)
if new_message:
self.gmail.send(id, new_message)
result = True
except Exception as e:
self.logger.exception(e)
finally:
return result
def import_key(self, bundle):
if not 'id' in bundle:
return False
imp = self.gpg.recv_keys('keyserver.ubuntu.com', bundle['id'])
result = imp.imported > 0
complete_result = imp.results[0]
self.logger.info("import key {} - {}: {}".format(
complete_result['fingerprint'], result, complete_result['text']))
return result
