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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
# -*- coding: utf-8 -*-
# walktree.py
# Copyright (C) 2013 LEAP
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
"""
Tests for the walktree module.
"""
import os
import sys
import pprint
from email import parser
from leap.mail import walk as W
DEBUG = os.environ.get("BITMASK_MAIL_DEBUG")
p = parser.Parser()
# TODO pass an argument of the type of message
##################################################
# Input from hell
if len(sys.argv) > 1:
FILENAME = sys.argv[1]
else:
FILENAME = "rfc822.multi-signed.message"
"""
FILENAME = "rfc822.plain.message"
FILENAME = "rfc822.multi-minimal.message"
"""
msg = p.parse(open(FILENAME))
DO_CHECK = False
#################################################
parts = W.get_parts(msg)
if DEBUG:
def trim(item):
item = item[:10]
[trim(part["phash"]) for part in parts if part.get('phash', None)]
raw_docs = list(W.get_raw_docs(msg, parts))
body_phash_fun = [W.get_body_phash_simple,
W.get_body_phash_multi][int(msg.is_multipart())]
body_phash = body_phash_fun(W.get_payloads(msg))
parts_map = W.walk_msg_tree(parts, body_phash=body_phash)
# TODO add missing headers!
expected = {
'body': '1ddfa80485',
'multi': True,
'part_map': {
1: {
'headers': {'Content-Disposition': 'inline',
'Content-Type': 'multipart/mixed; '
'boundary="z0eOaCaDLjvTGF2l"'},
'multi': True,
'part_map': {1: {'ctype': 'text/plain',
'headers': [
('Content-Type',
'text/plain; charset=utf-8'),
('Content-Disposition',
'inline'),
('Content-Transfer-Encoding',
'quoted-printable')],
'multi': False,
'parts': 1,
'phash': '1ddfa80485',
'size': 206},
2: {'ctype': 'text/plain',
'headers': [('Content-Type',
'text/plain; charset=us-ascii'),
('Content-Disposition',
'attachment; '
'filename="attach.txt"')],
'multi': False,
'parts': 1,
'phash': '7a94e4d769',
'size': 133},
3: {'ctype': 'application/octet-stream',
'headers': [('Content-Type',
'application/octet-stream'),
('Content-Disposition',
'attachment; filename="hack.ico"'),
('Content-Transfer-Encoding',
'base64')],
'multi': False,
'parts': 1,
'phash': 'c42cccebbd',
'size': 12736}}},
2: {'ctype': 'application/pgp-signature',
'headers': [('Content-Type', 'application/pgp-signature')],
'multi': False,
'parts': 1,
'phash': '8f49fbf749',
'size': 877}}}
if DEBUG and DO_CHECK:
# TODO turn this into a proper unittest
assert(parts_map == expected)
print "Structure: OK"
print
print "RAW DOCS"
pprint.pprint(raw_docs)
print
print "PARTS MAP"
pprint.pprint(parts_map)
|
