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
|
# pylint: disable=missing-module-docstring
# pylint: disable=missing-class-docstring
# pylint: disable=missing-function-docstring
import unittest
from PyKCS11 import PyKCS11
# use OpenSC PKCS#11 Spy to check what PIN value is sent to the PKCS#11
# library
class TestUtil(unittest.TestCase):
def setUp(self):
self.pkcs11 = PyKCS11.PyKCS11Lib()
self.pkcs11.load()
self.slot = self.pkcs11.getSlotList(tokenPresent=True)[0]
self.session = self.pkcs11.openSession(
self.slot, PyKCS11.CKF_SERIAL_SESSION | PyKCS11.CKF_RW_SESSION
)
def tearDown(self):
self.pkcs11.closeAllSessions(self.slot)
del self.pkcs11
def test_login(self):
self.session.login("1234")
self.session.logout()
def test_wrong(self):
with self.assertRaises(PyKCS11.PyKCS11Error) as cm:
self.session.login("wrong PIN")
the_exception = cm.exception
self.assertEqual(the_exception.value, PyKCS11.CKR_PIN_INCORRECT)
self.assertEqual(str(the_exception), "CKR_PIN_INCORRECT (0x000000A0)")
def test_ckbytelist(self):
pin = PyKCS11.ckbytelist("1234")
self.session.login(pin)
self.session.logout()
def test_binary(self):
with self.assertRaises(PyKCS11.PyKCS11Error) as cm:
pin = PyKCS11.ckbytelist([1, 2, 3, 4])
self.session.login(pin)
the_exception = cm.exception
self.assertEqual(the_exception.value, PyKCS11.CKR_PIN_INCORRECT)
self.assertEqual(str(the_exception), "CKR_PIN_INCORRECT (0x000000A0)")
def test_null(self):
# SoftHSM2 does not support pinpad (pin = NULL)
with self.assertRaises(PyKCS11.PyKCS11Error) as cm:
self.session.login(None)
the_exception = cm.exception
self.assertEqual(the_exception.value, PyKCS11.CKR_ARGUMENTS_BAD)
self.assertEqual(str(the_exception), "CKR_ARGUMENTS_BAD (0x00000007)")
|
