1 #!/usr/bin/env python2.4
2 from basetest
import BaseTest
3 import sys
, tempfile
, os
, shutil
6 sys
.path
.insert(0, '..')
7 from zeroinstall
.injector
import gpg
, model
, trust
8 from zeroinstall
.support
import basedir
10 err_sig
= """-----BEGIN PGP MESSAGE-----
11 Version: GnuPG v1.4.0 (GNU/Linux)
13 owGbwMvMwCTYk9R5Infvsj7G01xJDE513j1OiSlcHfbMrCDOBJisINP6XQwLGjzn
14 tMxedXc3y75I7r1hQZFTb/ewMcx3yefZ8zb/vZd10I7LEYdDj4fnKsYAAA==
16 -----END PGP MESSAGE-----
19 bad_sig
= """-----BEGIN PGP SIGNED MESSAGE-----
23 -----BEGIN PGP SIGNATURE-----
24 Version: GnuPG v1.4.0 (GNU/Linux)
26 iD8DBQFCfk3grgeCgFmlPMERAhl8AKC0aktrLzz646zTY0TRzdnxPdbLBgCeJWbk
27 GRVbJusevCKvtoSn7RAW2mg=
29 -----END PGP SIGNATURE-----
32 good_sig
= """-----BEGIN PGP SIGNED MESSAGE-----
36 -----BEGIN PGP SIGNATURE-----
37 Version: GnuPG v1.4.0 (GNU/Linux)
39 iD8DBQFCfk3grgeCgFmlPMERAhl8AKC0aktrLzz646zTY0TRzdnxPdbLBgCeJWbk
40 GRVbJusevCKvtoSn7RAW2mg=
42 -----END PGP SIGNATURE-----
45 bad_xml_main
= """<?xml version='1.0'?>
49 ('last line is not end-of-comment',
50 """<!-- Base64 Signature
52 ('No signature block in XML',
54 iD8DBQBDtpK9rgeCgFmlPMERAg0gAKCaJhXFnk
57 ('extra data on comment line',
58 """<!-- Base64 Signature data
59 iD8DBQBDtpK9rgeCgFmlPMERAg0gAKCaJhXFnk
62 ('last line is not end-of-comment',
63 """<!-- Base64 Signature
64 iD8DBQBDtpK9rgeCgFmlPMERAg0gAKCaJhXFnk
65 WZRBLT0an56WYaBODukSsf4=
68 ('Invalid base 64 encoded signature:',
69 """<!-- Base64 Signature
70 iD8DBQBDtpK9rgeCgFmlPMERAg0gAKCaJhXFnk
71 WZRBLT0an56WYaBODukSsf4=
75 ('Invalid characters found',
76 """<!-- Base64 Signature
77 iD8DBQBDtpK9rge<CgFmlPMERAg0gAKCaJhXFnk
78 WZRBLT0an56WYaBODukSsf4=
82 good_xml_sig
= """<?xml version='1.0'?>
85 iD8DBQBDuChIrgeCgFmlPMERAnGEAJ0ZS1PeyWonx6xS/mgpYTKNgSXa5QCeMSYPHhNcvxu3f84y
90 bad_xml_sig
= """<?xml version='1.0'?>
93 iD8DBQBDuChIrgeCgFmlPMERAnGEAJ0ZS1PeyWonx6xS/mgpYTKNgSXa5QCeMSYPHhNcvxu3f84y
98 from data
import thomas_key
100 class TestGPG(BaseTest
):
104 stream
= tempfile
.TemporaryFile()
105 stream
.write(thomas_key
)
107 gpg
.import_key(stream
)
108 trust
.trust_db
.trust_key(
109 '92429807C9853C0744A68B9AAE07828059A53CC1')
111 def testImportBad(self
):
112 stream
= tempfile
.TemporaryFile()
113 stream
.write("Bad key")
116 gpg
.import_key(stream
)
118 except model
.SafeException
:
121 def testErrSig(self
):
122 stream
= tempfile
.TemporaryFile()
123 stream
.write(err_sig
)
125 data
, sigs
= gpg
.check_stream(stream
)
126 self
.assertEquals("Bad\n", data
.read())
127 assert len(sigs
) == 1
128 assert isinstance(sigs
[0], gpg
.ErrSig
)
129 assert sigs
[0].need_key() == "8C6289C86DBDA68E"
130 self
.assertEquals("17", sigs
[0].status
[gpg
.ErrSig
.ALG
])
131 assert sigs
[0].is_trusted() is False
132 assert str(sigs
[0]).startswith('ERROR')
134 def testBadSig(self
):
135 self
.assertEquals("Hell0\n", self
.check_bad(bad_sig
))
137 def testBadXMLSig(self
):
138 self
.assertEquals(bad_xml_sig
, self
.check_bad(bad_xml_sig
))
140 def testInvalidXMLSig(self
):
141 for error
, sig
in invalid_xmls_sigs
:
143 self
.check_bad(bad_xml_main
+ '\n' + sig
)
144 except model
.SafeException
, ex
:
145 if error
not in str(ex
):
146 raise model
.SafeException(str(ex
) + '\nSig:\n' + sig
)
148 def check_bad(self
, sig
):
149 stream
= tempfile
.TemporaryFile()
152 data
, sigs
= gpg
.check_stream(stream
)
153 assert len(sigs
) == 1
154 assert isinstance(sigs
[0], gpg
.BadSig
)
155 self
.assertEquals("AE07828059A53CC1",
156 sigs
[0].status
[gpg
.BadSig
.KEYID
])
157 assert sigs
[0].is_trusted() is False
158 assert sigs
[0].need_key() is None
159 assert str(sigs
[0]).startswith('BAD')
162 def testGoodSig(self
):
163 self
.assertEquals("Hello\n", self
.check_good(good_sig
))
165 def testGoodXMLSig(self
):
166 self
.assertEquals(good_xml_sig
, self
.check_good(good_xml_sig
))
168 def check_good(self
, sig
):
169 stream
= tempfile
.TemporaryFile()
172 data
, sigs
= gpg
.check_stream(stream
)
173 assert len(sigs
) == 1
174 assert isinstance(sigs
[0], gpg
.ValidSig
)
175 self
.assertEquals("92429807C9853C0744A68B9AAE07828059A53CC1",
177 assert sigs
[0].is_trusted() is True
178 assert sigs
[0].need_key() is None
179 assert str(sigs
[0]).startswith('Valid')
180 for item
in sigs
[0].get_details():
181 if item
[0] in ('pub', 'uid') and len(item
) > 9:
183 "Thomas Leonard <tal197@users.sourceforge.net>",
187 self
.fail("Missing name")
191 stream
= tempfile
.TemporaryFile()
192 stream
.write("Hello")
195 gpg
.check_stream(stream
)
197 except model
.SafeException
:
200 def testLoadKeys(self
):
201 self
.assertEquals({}, gpg
.load_keys([]))
203 suite
= unittest
.makeSuite(TestGPG
)
204 if __name__
== '__main__':
205 sys
.argv
.append('-v')