aboutsummaryrefslogtreecommitdiff
path: root/rpkid/test-pow-cms.py
blob: 2dbc0d6ad59418cbe9d7ea0486abcd2223e71819 (plain) (blame)
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
# $Id$

# Copyright (C) 2008  American Registry for Internet Numbers ("ARIN")
#
# Permission to use, copy, modify, and distribute this software for any
# purpose with or without fee is hereby granted, provided that the above
# copyright notice and this permission notice appear in all copies.
#
# THE SOFTWARE IS PROVIDED "AS IS" AND ARIN DISCLAIMS ALL WARRANTIES WITH
# REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
# AND FITNESS.  IN NO EVENT SHALL ARIN BE LIABLE FOR ANY SPECIAL, DIRECT,
# INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
# LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
# OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
# PERFORMANCE OF THIS SOFTWARE.

import POW, rpki.x509, os

key = rpki.x509.RSA(Auto_file = "biz-certs/Alice-EE.key").get_POW()
ee = rpki.x509.X509(Auto_file = "biz-certs/Alice-EE.cer").get_POW()

ca = rpki.x509.X509(Auto_file = "biz-certs/Alice-CA.cer").get_POW()
ta = rpki.x509.X509(Auto_file = "biz-certs/Alice-Root.cer").get_POW()

store = POW.X509Store()
store.addTrust(ta)

if store.verify(ca):
  print "Verified CA"
  store.addTrust(ca)
else:
  print "Couldn't verify CA"

if store.verify(ee):
  print "Verified EE"
  store.addTrust(ee)
else:
  print "Couldn't verify EE"

oid = "1.2.840.113549.1.9.16.1.24"

plaintext = "Wombats Are Us"

for args in ((ee, key, [ca], plaintext, oid),
             (ee, key, [ca], plaintext, oid, True),
             (ee, key, [ca], plaintext, oid, False),
             (ee, key, [], plaintext, oid),
             (ee, key, [], plaintext, oid, True),
             (ee, key, [], plaintext, oid, False)):

  print "Testing", repr(args)

  cms = POW.CMS()
  cms.sign(*args)

  if False:
    f = open("test-pow-cms.der", "w")
    f.write(cms.derWrite())
    f.close()
    if False:
      f = os.popen("dumpasn1 2>&1 -a test-pow-cms.der")
      print "\n".join(x for x in f.read().splitlines() if x.startswith(" "))
      f.close()

  cms.verify(store, [ee])