File: pgp.py

package info (click to toggle)
dpkg-scriptlib 0.1-2hamm1
  • links: PTS
  • area: main
  • in suites: hamm, slink
  • size: 188 kB
  • ctags: 230
  • sloc: python: 1,685; perl: 534; makefile: 41; sh: 18
file content (127 lines) | stat: -rw-r--r-- 4,134 bytes parent folder | download | duplicates (2)
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
# $Id$
#
# Copyright (C) 1997  Klee Dienes <klee@mit.edu>
#
# 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 2 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, write to the Free Software
# Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

import subfile, string, os, sys, strfile, filter, outstr, regex, regex_syntax, strfile

pgpargs = '+tmp=/tmp +language=en +myname="" +batchmode=on +clearsig=on +textmode=on +charset=noconv +armor=on +armorlines=0 +compress=off +completes_needed=1 +marginals_needed=0 +cert_depth=1 +pubring=/usr/lib/dpkgcert/pubring.pgp +secring=/usr/lib/dpkgcert/secring.pgp +verbose=1 +interactive=off +nomanual=on'

def sign (body, uid, passphrase = ''):
	f = strfile.strfile (passphrase + '\n' + body)
	nf = filter.filter (f, "PGPPASSFD=0 pgp -fs %s -u '%s'" % (pgpargs, uid))
	s = nf.read ()
	bstr = '-----BEGIN PGP SIGNED MESSAGE-----\n'
	estr = '-----END PGP SIGNATURE-----\n'
	errtxt = nf.stderr().read()
	if ((s[:len (bstr)] != bstr) or (s[len(s)-len(estr):] != estr)):
		raise ValueError, 'error signing message: ' + string.translate (errtxt, string.maketrans ('\n', ' '))
	nf.close ()
	return s

regex.set_syntax (regex_syntax.RE_SYNTAX_EMACS)
verexp = regex.compile ('^Good signature from user "\([^"]+\)"\.$')
sigexp = regex.compile ('^Signature made \(.*\) using \([0-9]+\)-bit key, key ID \([A-Z0-9]+\)$')
brexp = regex.compile ('[Bb]ad signature')

def parse_err (l):

	if (brexp.search (l[7][:-1]) >= 0):
		raise ValueError, 'bad signature'
	
	if (verexp.match (l[7][:-1]) < 0):
		raise ValueError, 'unable to parse PGP output "%s"' % l[7][:-1]
	if (sigexp.match (l[8][:-1]) < 0):
		raise ValueError, 'unable to parse PGP output "%s"' % l[8][:-1]

	r = {}
	r['keyid'] = sigexp.group (3)
	r['bits'] = string.atoi (sigexp.group (2))
	r['date'] = sigexp.group (1)
	r['keyname'] = verexp.group (1)

	return r

def verify (body, signature, keyname, keyid):
	r = checksig (body, signature)
	if (string.lower (r['keyid']) != string.lower (keyid)):
		raise ValueError, 'invalid keyid %s (should be %s)' % (r['keyid'], keyid)
	if (r['keyname'] != keyname):
		raise ValueError, 'invalid key name "%s" (should be "%s")' % (r['keyname'], keyname)

def checksig (body, signature):
	
	of = outstr.outstr ()
	write_message (of, body, signature)
	f = strfile.strfile (of.getstr ())
	of.close ()
	nf = filter.filter (f, 'pgp -f %s > /dev/null' % pgpargs)
	errlines = nf.stderr().readlines()
	f.close ()
	nf.close ()
	r = parse_err (errlines)
	return r

def write_message (f, body, sig):

	f.write ('-----BEGIN PGP SIGNED MESSAGE-----\n\n')
	f.write (body)
	f.write ('\n-----BEGIN PGP SIGNATURE-----\n')
	f.write (sig)
	f.write ('-----END PGP SIGNATURE-----\n')

def parse_message (s):
	f = strfile.strfile (s)
	r = read_message (f)
	f.close ()
	return r

def read_message (f):

	body = ''
	signature = ''

	while 1:
		s = f.readline ()
		if (s == ''):
			return None
		if (s == '\n'):
			continue
		if (s == '-----BEGIN PGP SIGNED MESSAGE-----\n'):
			break
		raise ValueError, 'parse error looking for certificate'
	if (f.readline () != '\n'):
		raise ValueError, 'missing ASCII armor before message body'
	while 1:
		s = f.readline ()
		if (s == ''):
			raise ValueError, 'end-of-file while reading certificate'
		if (s == '-----BEGIN PGP SIGNATURE-----\n'):
			break
		body = body + s
	if (body[-1] != '\n'):
		raise ValueError, 'missing ASCII armor after message body'
	body = body[:-1]
	while 1:
		s = f.readline ()
		if (s == ''):
			raise ValueError, 'end-of-file while reading certificate'
		if (s == '-----END PGP SIGNATURE-----\n'):
			break
		signature = signature + s

	return body, signature