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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
import gi
import os
import subprocess
from chrysacase import ChrysalideTestCase
from gi.repository import Gio, GLib
from pychrysalide.glibext import SecretStorage
class TestSecretStorage(ChrysalideTestCase):
"""TestCase for secret storage features."""
@classmethod
def setUpClass(cls):
super(TestSecretStorage, cls).setUpClass()
cls.log('Creating GSettings schema...')
path = os.path.dirname(os.path.realpath(__file__))
subprocess.run([ 'glib-compile-schemas', path ])
os.environ['GSETTINGS_SCHEMA_DIR'] = path + ':' + os.environ['GSETTINGS_SCHEMA_DIR']
@classmethod
def tearDownClass(cls):
super(TestSecretStorage, cls).tearDownClass()
cls.log('Removing compiled GSettings schema...')
os.environ['GSETTINGS_SCHEMA_DIR'] = ':'.join(os.environ['GSETTINGS_SCHEMA_DIR'].split(':')[1:])
path = os.path.dirname(os.path.realpath(__file__))
filename = os.path.join(path, 'gschemas.compiled')
if os.path.exists(filename):
os.remove(filename)
def testMasterKeyDefinition(self):
"""Check for cryptographic parameters for secret storage."""
settings = Gio.Settings.new('re.chrysalide.tests.secstorage')
storage = SecretStorage(settings)
settings.reset('master')
self.assertEqual(len(settings.get_value('master').unpack()), 0)
self.assertFalse(storage.has_key)
settings.set_value('master', GLib.Variant('ay', b'ABC'))
self.assertFalse(storage.has_key)
settings.set_value('master', GLib.Variant('ay', b'A' * 23))
self.assertTrue(storage.has_key)
def testMasterKeyCreation(self):
"""Create and update cryptographic parameters for secret storage."""
settings = Gio.Settings.new('re.chrysalide.tests.secstorage')
storage = SecretStorage(settings)
settings.reset('salt')
settings.reset('master')
self.assertFalse(storage.has_key)
status = storage.set_password('')
self.assertTrue(status);
self.assertTrue(storage.has_key)
self.assertTrue(storage.is_locked)
status = storage.unlock('')
self.assertTrue(status)
self.assertFalse(storage.is_locked)
storage.lock()
self.assertTrue(storage.is_locked)
status = storage.unlock('XXX')
self.assertFalse(status)
self.assertTrue(storage.is_locked)
def testDataEncryption(self):
"""Create and update cryptographic parameters for secret storage."""
settings = Gio.Settings.new('re.chrysalide.tests.secstorage')
storage = SecretStorage(settings)
settings.reset('salt')
settings.reset('master')
status = storage.set_password('<s3cUre>')
self.assertTrue(status);
status = storage.unlock('<s3cUre>')
self.assertTrue(status)
original = b'ABC'
encrypted = storage.encrypt_data(original)
self.assertIsNotNone(encrypted)
plain = storage.decrypt_data(encrypted)
self.assertIsNotNone(plain)
self.assertEqual(original, plain)
original = b'A' * 136
encrypted = storage.encrypt_data(original)
self.assertIsNotNone(encrypted)
plain = storage.decrypt_data(encrypted)
self.assertIsNotNone(plain)
self.assertEqual(original, plain)
|
