333ee4e87d876d053d069272627999a0dc911a88
[gpgme.git] / lang / python / tests / t-verify.py
1 #!/usr/bin/env python3
2
3 # Copyright (C) 2016 g10 Code GmbH
4 #
5 # This file is part of GPGME.
6 #
7 # GPGME is free software; you can redistribute it and/or modify it
8 # under the terms of the GNU General Public License as published by
9 # the Free Software Foundation; either version 2 of the License, or
10 # (at your option) any later version.
11 #
12 # GPGME is distributed in the hope that it will be useful, but WITHOUT
13 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
14 # or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General
15 # Public License for more details.
16 #
17 # You should have received a copy of the GNU Lesser General Public
18 # License along with this program; if not, see <http://www.gnu.org/licenses/>.
19
20 import os
21 from pyme import core, constants, errors
22 import support
23
24 test_text1 = "Just GNU it!\n"
25 test_text1f= "Just GNU it?\n"
26 test_sig1 = """-----BEGIN PGP SIGNATURE-----
27
28 iN0EABECAJ0FAjoS+i9FFIAAAAAAAwA5YmFyw7bDpMO8w58gZGFzIHdhcmVuIFVt
29 bGF1dGUgdW5kIGpldHp0IGVpbiBwcm96ZW50JS1aZWljaGVuNRSAAAAAAAgAJGZv
30 b2Jhci4xdGhpcyBpcyBhIG5vdGF0aW9uIGRhdGEgd2l0aCAyIGxpbmVzGhpodHRw
31 Oi8vd3d3Lmd1Lm9yZy9wb2xpY3kvAAoJEC1yfMdoaXc0JBIAoIiLlUsvpMDOyGEc
32 dADGKXF/Hcb+AKCJWPphZCphduxSvrzH0hgzHdeQaA==
33 =nts1
34 -----END PGP SIGNATURE-----
35 """
36
37 test_sig2 = """-----BEGIN PGP MESSAGE-----
38
39 owGbwMvMwCSoW1RzPCOz3IRxjXQSR0lqcYleSUWJTZOvjVdpcYmCu1+oQmaJIleH
40 GwuDIBMDGysTSIqBi1MApi+nlGGuwDeHao53HBr+FoVGP3xX+kvuu9fCMJvl6IOf
41 y1kvP4y+8D5a11ang0udywsA
42 =Crq6
43 -----END PGP MESSAGE-----
44 """
45
46 # A message with a prepended but unsigned plaintext packet.
47 double_plaintext_sig = """-----BEGIN PGP MESSAGE-----
48
49 rDRiCmZvb2Jhci50eHRF4pxNVGhpcyBpcyBteSBzbmVha3kgcGxhaW50ZXh0IG1l
50 c3NhZ2UKowGbwMvMwCSoW1RzPCOz3IRxTWISa6JebnG666MFD1wzSzJSixQ81XMV
51 UlITUxTyixRyKxXKE0uSMxQyEosVikvyCwpSU/S4FNCArq6Ce1F+aXJGvoJvYlGF
52 erFCTmJxiUJ5flFKMVeHGwuDIBMDGysTyA4GLk4BmO036xgWzMgzt9V85jCtfDFn
53 UqVooWlGXHwNw/xg/fVzt9VNbtjtJ/fhUqYo0/LyCGEA
54 =6+AK
55 -----END PGP MESSAGE-----
56 """
57
58 def check_result(result, summary, fpr, status, notation):
59     assert len(result.signatures) == 1, "Unexpected number of signatures"
60     sig = result.signatures[0]
61     assert sig.summary == summary, "Unexpected signature summary"
62     assert sig.fpr == fpr
63     assert errors.GPGMEError(sig.status).getcode() == status
64
65     if notation:
66         expected_notations = {
67             "bar": b"\xc3\xb6\xc3\xa4\xc3\xbc\xc3\x9f".decode() +
68             " das waren Umlaute und jetzt ein prozent%-Zeichen",
69             "foobar.1":  "this is a notation data with 2 lines",
70             None: "http://www.gu.org/policy/",
71         }
72         assert len(sig.notations) == len(expected_notations)
73
74         for r in sig.notations:
75             assert not 'name_len' in dir(r)
76             assert not 'value_len' in dir(r)
77             assert r.name in expected_notations
78             assert r.value == expected_notations[r.name], \
79                 "Expected {!r}, got {!r}".format(expected_notations[r.name],
80                                                  r.value)
81             expected_notations.pop(r.name)
82
83         assert len(expected_notations) == 0
84
85     assert not sig.wrong_key_usage
86     assert sig.validity == constants.VALIDITY_UNKNOWN
87     assert errors.GPGMEError(sig.validity_reason).getcode() == errors.NO_ERROR
88
89
90 support.init_gpgme(constants.PROTOCOL_OpenPGP)
91 c = core.Context()
92 c.set_armor(True)
93
94 # Checking a valid message.
95 text = core.Data(test_text1)
96 sig = core.Data(test_sig1)
97 c.op_verify(sig, text, None)
98 result = c.op_verify_result()
99 check_result(result, 0, "A0FF4590BB6122EDEF6E3C542D727CC768697734",
100              errors.NO_ERROR, True)
101
102
103 # Checking a manipulated message.
104 text = core.Data(test_text1f)
105 sig.seek(0, os.SEEK_SET)
106 c.op_verify(sig, text, None)
107 result = c.op_verify_result()
108 check_result(result, constants.SIGSUM_RED, "2D727CC768697734",
109              errors.BAD_SIGNATURE, False)
110
111 # Checking a normal signature.
112 text = core.Data()
113 sig = core.Data(test_sig2)
114 c.op_verify(sig, None, text)
115 result = c.op_verify_result()
116 check_result(result, 0, "A0FF4590BB6122EDEF6E3C542D727CC768697734",
117              errors.NO_ERROR, False)
118
119 # Checking an invalid message.
120 text = core.Data()
121 sig = core.Data(double_plaintext_sig)
122 try:
123     c.op_verify(sig, None, text)
124 except Exception as e:
125     assert type(e) == errors.GPGMEError
126     assert e.getcode() == errors.BAD_DATA
127 else:
128     assert False, "Expected an error but got none."