summaryrefslogtreecommitdiff
path: root/scripts/profiling/mail/mx.py
blob: 55084a40f1ce6c710eb58111cfdf477f101cf98b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import datetime
import uuid
import json
import timeit


from leap.keymanager import openpgp
from leap.soledad.common.document import ServerDocument
from leap.soledad.common.crypto import (
    EncryptionSchemes,
    ENC_JSON_KEY,
    ENC_SCHEME_KEY,
)


from util import log


message = """To: Ed Snowden <snowden@bitmask.net>
Date: %s
From: Glenn Greenwald <greenwald@bitmask.net>

hi!

"""


def get_message():
    return message % datetime.datetime.now().strftime("%a %b %d %H:%M:%S:%f %Y")


def get_enc_json(pubkey, message):
    with openpgp.TempGPGWrapper(gpgbinary='/usr/bin/gpg') as gpg:
        gpg.import_keys(pubkey)
        key = gpg.list_keys().pop()
        # We don't care about the actual address, so we use a
        # dummy one, we just care about the import of the pubkey
        openpgp_key = openpgp._build_key_from_gpg("dummy@mail.com",
                                                  key, pubkey)
        enc_json = str(gpg.encrypt(
            json.dumps(
                {'incoming': True, 'content': message},
                ensure_ascii=False),
            openpgp_key.fingerprint,
            symmetric=False))
    return enc_json


def get_new_doc(enc_json):
    doc = ServerDocument(doc_id=str(uuid.uuid4()))
    doc.content = {
        'incoming': True,
        ENC_SCHEME_KEY: EncryptionSchemes.PUBKEY,
        ENC_JSON_KEY: enc_json
    }
    return doc


def get_pubkey():
    with open('./keys/5447A9AD50E3075ECCE432711B450E665FE63573.pub') as f:
        return f.read()


def put_one_message(pubkey, db):
    enc_json = get_enc_json(pubkey, get_message())
    doc = get_new_doc(enc_json)
    db.put_doc(doc)


def put_lots_of_messages(db, number):
    log("Populating database with %d encrypted messages... "
        % number, line_break=False)
    pubkey = get_pubkey()

    def _put_one_message():
        put_one_message(pubkey, db)
    time = timeit.timeit(_put_one_message, number=number)
    log("done.")
    average_time = time / number
    log("put_one_message average time: %f" % average_time)
    return average_time