2 * Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License
17 * @file scheme-test.cpp
18 * @author Krzysztof Jackiewicz (k.jackiewicz@samsung.com)
21 #include <scheme-test.h>
23 #include <sys/smack.h>
24 #include <sys/types.h>
26 #include <sys/sendfile.h>
37 #include <boost/test/unit_test.hpp>
39 #include <smack-access.h>
41 #include <db-crypto.h>
42 #include <file-system.h>
43 #include <key-provider.h>
45 #include <crypto-init.h>
46 #include <dpl/errno_string.h>
52 const uid_t UID = 7654;
53 const gid_t GID = 7654;
54 const char *const DBPASS = "db-pass";
55 const char *const LABEL = "my-label";
56 const ClientId OWNER = "/" + string(LABEL);
57 const int ENC_SCHEME_OFFSET = 24;
58 const string TEST_DATA_STR = "test-data";
59 RawBuffer TEST_DATA(TEST_DATA_STR.begin(), TEST_DATA_STR.end());
60 const Password TEST_PASS = "custom user password";
61 const size_t IV_LEN = 16;
62 const size_t CHAIN_LEN = 3;
74 // [password][exportable]
75 Policy policy[2][2] = {
76 {{ Password(), false }, { Password(), true }},
77 {{ TEST_PASS, false }, { TEST_PASS, true }},
93 Item("data-alias1", DataType::BINARY_DATA, policy[NO_PASS][EXP])
98 Item("data-alias2", DataType::BINARY_DATA, policy[PASS][EXP])
104 Group::KEY_PAIR_RSA, {
105 Item("key-rsa-alias-prv1", DataType::KEY_RSA_PRIVATE, policy[NO_PASS][NO_EXP]),
106 Item("key-rsa-alias-pub1", DataType::KEY_RSA_PUBLIC, policy[NO_PASS][NO_EXP])
110 Group::KEY_PAIR_RSA, {
111 Item("key-rsa-alias-prv2", DataType::KEY_RSA_PRIVATE, policy[NO_PASS][EXP]),
112 Item("key-rsa-alias-pub2", DataType::KEY_RSA_PUBLIC, policy[NO_PASS][EXP]),
116 Group::KEY_PAIR_RSA, {
117 Item("key-rsa-alias-prv3", DataType::KEY_RSA_PRIVATE, policy[PASS][NO_EXP]),
118 Item("key-rsa-alias-pub3", DataType::KEY_RSA_PUBLIC, policy[PASS][NO_EXP]),
122 Group::KEY_PAIR_RSA, {
123 Item("key-rsa-alias-prv4", DataType::KEY_RSA_PRIVATE, policy[PASS][EXP]),
124 Item("key-rsa-alias-pub4", DataType::KEY_RSA_PUBLIC, policy[PASS][EXP]),
127 // different policies
129 Group::KEY_PAIR_RSA, {
130 Item("key-rsa-alias-prv5", DataType::KEY_RSA_PRIVATE, policy[PASS][NO_EXP]),
131 Item("key-rsa-alias-pub5", DataType::KEY_RSA_PUBLIC, policy[NO_PASS][EXP]),
137 Group::SINGLE_ITEM, {
138 Item("key-aes-alias1", DataType::KEY_AES, policy[NO_PASS][NO_EXP]),
142 Group::SINGLE_ITEM, {
143 Item("key-aes-alias2", DataType::KEY_AES, policy[NO_PASS][EXP]),
147 Group::SINGLE_ITEM, {
148 Item("key-aes-alias3", DataType::KEY_AES, policy[PASS][NO_EXP]),
152 Group::SINGLE_ITEM, {
153 Item("key-aes-alias4", DataType::KEY_AES, policy[PASS][EXP]),
160 Item("cert-root-alias1", DataType::CERTIFICATE, policy[NO_PASS][NO_EXP]),
161 Item("cert-im-ca-alias1", DataType::CERTIFICATE, policy[NO_PASS][NO_EXP]),
162 Item("cert-leaf-alias1", DataType::CERTIFICATE, policy[NO_PASS][NO_EXP]),
167 Item("cert-root-alias2", DataType::CERTIFICATE, policy[NO_PASS][EXP]),
168 Item("cert-im-ca-alias2", DataType::CERTIFICATE, policy[NO_PASS][EXP]),
169 Item("cert-leaf-alias2", DataType::CERTIFICATE, policy[NO_PASS][EXP]),
174 Item("cert-root-alias3", DataType::CERTIFICATE, policy[PASS][NO_EXP]),
175 Item("cert-im-ca-alias3", DataType::CERTIFICATE, policy[PASS][NO_EXP]),
176 Item("cert-leaf-alias3", DataType::CERTIFICATE, policy[PASS][NO_EXP]),
181 Item("cert-root-alias4", DataType::CERTIFICATE, policy[PASS][EXP]),
182 Item("cert-im-ca-alias4", DataType::CERTIFICATE, policy[PASS][EXP]),
183 Item("cert-leaf-alias4", DataType::CERTIFICATE, policy[PASS][EXP]),
189 Group::SINGLE_ITEM, {
190 Item("pkcs-alias1", DataType::CHAIN_CERT_0, policy[NO_PASS][NO_EXP]),
194 Group::SINGLE_ITEM, {
195 Item("pkcs-alias2", DataType::CHAIN_CERT_0, policy[NO_PASS][EXP]),
199 Group::SINGLE_ITEM, {
200 Item("pkcs-alias3", DataType::CHAIN_CERT_0, policy[PASS][NO_EXP]),
204 Group::SINGLE_ITEM, {
205 Item("pkcs-alias4", DataType::CHAIN_CERT_0, policy[PASS][EXP]),
210 const size_t CHAIN_SIZE = 3;
212 // TEST_ROOT_CA, expires 2035
213 std::string TEST_ROOT_CA =
214 "-----BEGIN CERTIFICATE-----\n"
215 "MIIDnzCCAoegAwIBAgIJAMH/ADkC5YSTMA0GCSqGSIb3DQEBBQUAMGYxCzAJBgNV\n"
216 "BAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMQ0wCwYDVQQKDARBQ01FMRAwDgYD\n"
217 "VQQLDAdUZXN0aW5nMSEwHwYDVQQDDBhUZXN0IHJvb3QgY2EgY2VydGlmaWNhdGUw\n"
218 "HhcNMTQxMjMwMTcyMTUyWhcNMjQxMjI3MTcyMTUyWjBmMQswCQYDVQQGEwJBVTET\n"
219 "MBEGA1UECAwKU29tZS1TdGF0ZTENMAsGA1UECgwEQUNNRTEQMA4GA1UECwwHVGVz\n"
220 "dGluZzEhMB8GA1UEAwwYVGVzdCByb290IGNhIGNlcnRpZmljYXRlMIIBIjANBgkq\n"
221 "hkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA0EJRdUtd2th0vTVF7QxvDKzyFCF3w9vC\n"
222 "9IDE/Yr12w+a9jd0s7/eG96qTHIYffS3B7x2MB+d4n+SR3W0qmYh7xk8qfEgH3da\n"
223 "eDoV59IZ9r543KM+g8jm6KffYGX1bIJVVY5OhBRbO9nY6byYpd5kbCIUB6dCf7/W\n"
224 "rQl1aIdLGFIegAzPGFPXDcU6F192686x54bxt/itMX4agHJ9ZC/rrTBIZghVsjJo\n"
225 "5/AH5WZpasv8sfrGiiohAxtieoYoJkv5MOYP4/2lPlOY+Cgw1Yoz+HHv31AllgFs\n"
226 "BquBb/kJVmCCNsAOcnvQzTZUsW/TXz9G2nwRdqI1nSy2JvVjZGsqGQIDAQABo1Aw\n"
227 "TjAdBgNVHQ4EFgQUt6pkzFt1PZlfYRL/HGnufF4frdwwHwYDVR0jBBgwFoAUt6pk\n"
228 "zFt1PZlfYRL/HGnufF4frdwwDAYDVR0TBAUwAwEB/zANBgkqhkiG9w0BAQUFAAOC\n"
229 "AQEAld7Qwq0cdzDQ51w1RVLwTR8Oy25PB3rzwEHcSGJmdqlMi3xOdaz80S1R1BBX\n"
230 "ldvGBG5Tn0vT7xSuhmSgI2/HnBpy9ocHVOmhtNB4473NieEpfTYrnGXrFxu46Wus\n"
231 "9m/ZnugcQ2G6C54A/NFtvgLmaC8uH8M7gKdS6uYUwJFQEofkjmd4UpOYSqmcRXhS\n"
232 "Jzd5FYFWkJhKJYp3nlENSOD8CUFFVGekm05nFN2gRVc/qaqQkEX77+XYvhodLRsV\n"
233 "qMn7nf7taidDKLO2T4bhujztnTYOhhaXKgPy7AtZ28N2wvX96VyAPB/vrchGmyBK\n"
234 "kOg11TpPdNDkhb1J4ZCh2gupDg==\n"
235 "-----END CERTIFICATE-----\n";
237 // TEST_IM_CA, signed by TEST_ROOT_CA, expires 2035
238 std::string TEST_IM_CA =
239 "-----BEGIN CERTIFICATE-----\n"
240 "MIIDljCCAn6gAwIBAgICEAAwDQYJKoZIhvcNAQEFBQAwZjELMAkGA1UEBhMCQVUx\n"
241 "EzARBgNVBAgMClNvbWUtU3RhdGUxDTALBgNVBAoMBEFDTUUxEDAOBgNVBAsMB1Rl\n"
242 "c3RpbmcxITAfBgNVBAMMGFRlc3Qgcm9vdCBjYSBjZXJ0aWZpY2F0ZTAeFw0xNTAx\n"
243 "MTYxNjQ1MzRaFw0zNTAxMTExNjQ1MzRaMGQxCzAJBgNVBAYTAkFVMRMwEQYDVQQI\n"
244 "DApTb21lLVN0YXRlMQ0wCwYDVQQKDARBQ01FMRAwDgYDVQQLDAdUZXN0aW5nMR8w\n"
245 "HQYDVQQDDBZUZXN0IElNIENBIGNlcnRpZmljYXRlMIIBIjANBgkqhkiG9w0BAQEF\n"
246 "AAOCAQ8AMIIBCgKCAQEAzmBF78qClgoKfnLAncMXZwZ14TW+5kags1+QCYeg3c7j\n"
247 "L9+RvDxIaX2tKf1sukJcwQfYqUlQkwt+58LMOb2ORtkpj8Or6WCWCZ0BzneT8ug7\n"
248 "nxJT4m9+bohMF0JoKjjB2H4KNMHamLIwUxRKt6nyfk81kVhJOi2vzzxd+UCPi6Pc\n"
249 "UAbJNH48eNgOIg55nyFovVzYj8GIo/9GvHJj83PPa/KlJZ+Z1qZASZZ/VYorplVT\n"
250 "thsHXKfejhFy5YJ9t7n/vyAQsyBsagZsvX19xnH41fbYXHKf8UbXG23rNaZlchs6\n"
251 "XJVLQdzOpj3WTj/lCocVHqLaZISLhNQ3aI7kUBUdiwIDAQABo1AwTjAdBgNVHQ4E\n"
252 "FgQUoCYNaCBP4jl/3SYQuK8Ka+6i3QEwHwYDVR0jBBgwFoAUt6pkzFt1PZlfYRL/\n"
253 "HGnufF4frdwwDAYDVR0TBAUwAwEB/zANBgkqhkiG9w0BAQUFAAOCAQEAjRzWiD97\n"
254 "Htv4Kxpm3P+C+xP9AEteCJfO+7p8MWgtWEJOknJyt55zeKS2JwZIq57KcbqD8U7v\n"
255 "vAUx1ymtUhlFPFd7J1mJ3pou+3aFYmGShYhGHpbrmUwjp7HVP588jrW1NoZVHdMc\n"
256 "4OgJWFrViXeu9+maIcekjMB/+9Y0dUgQuK5ZuT5H/Jwet7Th/o9uufTUZjBzRvrB\n"
257 "pbXgQpqgME2av4Q/6LuldPCTHLtWXgFUU2R+yCGmuGilvhFJnKoQryAbYnIQNWE8\n"
258 "SLoHQ9s1i7Zyb7HU6UAaqMOz15LBkyAqtNyJcO2p7Q/p5YK0xfD4xisI5qXucqVm\n"
260 "-----END CERTIFICATE-----\n";
262 // TEST_LEAF, signed by TEST_IM_CA, expires 2035
263 std::string TEST_LEAF =
264 "-----BEGIN CERTIFICATE-----\n"
265 "MIIDOzCCAiMCAQEwDQYJKoZIhvcNAQEFBQAwZDELMAkGA1UEBhMCQVUxEzARBgNV\n"
266 "BAgMClNvbWUtU3RhdGUxDTALBgNVBAoMBEFDTUUxEDAOBgNVBAsMB1Rlc3Rpbmcx\n"
267 "HzAdBgNVBAMMFlRlc3QgSU0gQ0EgY2VydGlmaWNhdGUwHhcNMTUwMTE2MTY0ODE0\n"
268 "WhcNMzUwMTExMTY0ODE0WjBjMQswCQYDVQQGEwJBVTETMBEGA1UECAwKU29tZS1T\n"
269 "dGF0ZTENMAsGA1UECgwEQUNNRTEQMA4GA1UECwwHVGVzdGluZzEeMBwGA1UEAwwV\n"
270 "VGVzdCBsZWFmIGNlcnRpZmljYXRlMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIB\n"
271 "CgKCAQEAzTdDIa2tDmRxFnIgiG+mBz8GoSVODs0ImNQGbqj+pLhBOFRH8fsah4Jl\n"
272 "z5YF9KwhMVLknnHGFLE/Nb7Ac35kEzhMQMpTRxohW83oxw3eZ8zN/FBoKqg4qHRq\n"
273 "QR8kS10YXTgrBR0ex/Vp+OUKEw6h7yL2r4Tpvrn9/qHwsxtLxqWbDIVf1O9b1Lfc\n"
274 "bllYMdmV5E62yN5tcwrDP8gvHjFnVeLzrG8wTpc9FR90/0Jkfp5jAJcArOBLrT0E\n"
275 "4VRqs+4HuwT8jAwFAmNnc7IYX5qSjtSWkmmHe73K/lzB+OiI0JEc/3eWUTWqwTSk\n"
276 "4tNCiQGBKJ39LXPTBBJdzmxVH7CUDQIDAQABMA0GCSqGSIb3DQEBBQUAA4IBAQAp\n"
277 "UdDOGu3hNiG+Vn10aQ6B1ZmOj3t+45gUV3sC+y8hB8EK1g4P5Ke9bVDts0T5eOnj\n"
278 "CSc+6VoND5O4adI0IFFRFljHNVnvjeosHfUZNnowsmA2ptQBtC1g5ZKRvKXlkC5/\n"
279 "i5BGgRqPFA7y9WB9Y05MrJHf3E+Oz/RBsLeeNiNN+rF5X1vYExvGHpo0M0zS0ze9\n"
280 "HtC0aOy8ocsTrQkf3ceHTAXx2i8ftoSSD4klojtWFpWMrNQa52F7wB9nU6FfKRuF\n"
281 "Zj/T1JkYXKkEwZU6nAR2jdZp3EP9xj3o15V/tyFcXHx6l8NTxn4cJb+Xe4VquQJz\n"
282 "6ON7PVe0ABN/AlwVQiFE\n"
283 "-----END CERTIFICATE-----\n";
288 void operator()(int *fd)
295 typedef std::unique_ptr<int, FdCloser> FdPtr;
297 uid_t getUid(const char *name)
300 struct passwd *result = nullptr;
301 int bufsize = sysconf(_SC_GETPW_R_SIZE_MAX);
304 bufsize = 16384; /* should be more than enough */
306 memset(&pwd, 0x00, sizeof(pwd));
308 std::vector<char> buf(bufsize, 0);
310 int ret = getpwnam_r(name, &pwd, buf.data(), bufsize, &result);
311 BOOST_REQUIRE_MESSAGE(ret == 0 && result, "getpwnam_r failed");
316 gid_t getGid(const char *name)
319 struct group *result = nullptr;
320 size_t bufsize = sysconf(_SC_GETGR_R_SIZE_MAX);
323 bufsize = 16384; /* should be more than enough */
325 memset(&grp, 0x00, sizeof(grp));
327 std::vector<char> buf(bufsize, 0);
329 int ret = getgrnam_r(name, &grp, buf.data(), bufsize, &result);
330 BOOST_REQUIRE_MESSAGE(ret == 0 && result, "getgrnam_r failed");
335 void restoreFile(const string &filename)
337 static uid_t CKM_UID = getUid(USER_NAME);
338 static gid_t CKM_GID = getGid(GROUP_NAME);
339 string sourcePath = string(DB_TEST_DIR) + "/" + filename;
340 string targetPath = string(RW_DATA_DIR) + "/" + filename;
344 int sourceFd = TEMP_FAILURE_RETRY(open(sourcePath.c_str(), O_RDONLY));
346 BOOST_REQUIRE_MESSAGE(sourceFd > 0,
347 "Opening " << sourcePath << " failed: " << GetErrnoString(err));
349 FdPtr sourceFdPtr(&sourceFd);
351 int targetFd = TEMP_FAILURE_RETRY(creat(targetPath.c_str(), 0644));
353 BOOST_REQUIRE_MESSAGE(targetFd > 0,
354 "Creating " << targetPath << " failed: " << GetErrnoString(err));
356 ret = fchown(targetFd, CKM_UID, CKM_GID);
358 BOOST_REQUIRE_MESSAGE(ret != -1, "fchown() failed: " << GetErrnoString(err));
360 FdPtr targetFdPtr(&targetFd);
362 struct stat sourceStat;
363 ret = fstat(sourceFd, &sourceStat);
365 BOOST_REQUIRE_MESSAGE(ret != -1, "fstat() failed: " << GetErrnoString(err));
367 ret = sendfile(targetFd, sourceFd, 0, sourceStat.st_size);
369 BOOST_REQUIRE_MESSAGE(ret != -1, "sendfile() failed: " << GetErrnoString(err));
371 ret = fsync(targetFd);
373 BOOST_REQUIRE_MESSAGE(ret != -1, "fsync() failed: " << GetErrnoString(err));
376 void generateRandom(size_t random_bytes, unsigned char *output)
378 if (random_bytes <= 0 || !output)
379 throw runtime_error("Invalid param");
381 std::ifstream is("/dev/urandom", std::ifstream::binary);
384 throw runtime_error("Failed to read /dev/urandom");
386 is.read(reinterpret_cast<char *>(output), random_bytes);
388 if (static_cast<std::streamsize>(random_bytes) != is.gcount())
389 throw runtime_error("Not enough bytes read from /dev/urandom");
392 RawBuffer createRandomBuffer(size_t random_bytes)
394 RawBuffer buffer(random_bytes);
395 generateRandom(buffer.size(), buffer.data());
398 } // namespace anonymous
401 SchemeTest::SchemeTest() : m_userChanged(false), m_directAccessEnabled(false)
403 m_control = Control::create();
404 m_mgr = Manager::create();
408 sa.add("System", LABEL, "rwx");
409 sa.add(LABEL, "System", "rwx");
410 sa.add(LABEL, "System::Run", "x");
414 SchemeTest::~SchemeTest()
421 void SchemeTest::RemoveUserData()
423 if (CKM_API_SUCCESS != m_control->lockUserKey(UID))
424 throw runtime_error("lockUserKey failed");
426 if (CKM_API_SUCCESS != m_control->removeUserData(UID))
427 throw runtime_error("removeUserData failed");
430 void SchemeTest::SwitchToUser()
435 if (CKM_API_SUCCESS != m_control->unlockUserKey(UID, DBPASS))
436 throw runtime_error("unlockUserKey failed");
441 if (smack_new_label_from_self(&label) <= 0)
442 throw runtime_error("smack_new_label_from_self failed");
444 m_origLabel = string(label);
447 if (0 > smack_set_label_for_self(LABEL))
448 throw runtime_error("smack_set_label_for_self failed");
450 if (0 > setegid(GID))
451 throw runtime_error("setegid failed");
453 if (0 > seteuid(UID))
454 throw runtime_error("seteuid failed");
456 m_userChanged = true;
459 void SchemeTest::SwitchToRoot()
465 throw runtime_error("seteuid failed");
468 throw runtime_error("setegid failed");
470 if (0 > smack_set_label_for_self(m_origLabel.c_str()))
471 throw runtime_error("smack_set_label_for_self failed");
473 if (m_control->lockUserKey(UID) != CKM_API_SUCCESS)
474 throw runtime_error("lockUserKey failed");
477 void SchemeTest::FillDb()
480 ifstream is(DB_TEST_DIR "/encryption-scheme.p12");
483 throw runtime_error("Failed to read pkcs");
485 istreambuf_iterator<char> begin(is), end;
486 RawBuffer pkcsBuffer(begin, end);
487 auto pkcs = PKCS12::create(pkcsBuffer, Password());
490 throw runtime_error("Empty pkcs");
495 RawBuffer rootCaBuffer(TEST_ROOT_CA.begin(), TEST_ROOT_CA.end());
496 CertificateShPtr rootCa = CKM::Certificate::create(rootCaBuffer,
497 CKM::DataFormat::FORM_PEM);
498 RawBuffer imCaBuffer(TEST_IM_CA.begin(), TEST_IM_CA.end());
499 CertificateShPtr imCa = CKM::Certificate::create(imCaBuffer,
500 CKM::DataFormat::FORM_PEM);
501 RawBuffer leafBuffer(TEST_LEAF.begin(), TEST_LEAF.end());
502 CertificateShPtr leaf = CKM::Certificate::create(leafBuffer,
503 CKM::DataFormat::FORM_PEM);
505 for (const auto &g : GROUPS) {
507 case Group::KEY_PAIR_RSA:
508 if (g.items.size() != 2)
509 throw runtime_error("Wrong number of keys");
511 if (g.items[0].type != DataType::KEY_RSA_PRIVATE ||
512 g.items[1].type != DataType::KEY_RSA_PUBLIC)
513 throw runtime_error("Invalid item type");
515 if (CKM_API_SUCCESS != m_mgr->createKeyPairRSA(1024,
520 throw runtime_error("createKeyPair failed");
524 case Group::CERT_CHAIN:
525 if (g.items.size() != CHAIN_SIZE)
526 throw runtime_error("Wrong number of certificates");
528 if (g.items[0].type != DataType::CERTIFICATE ||
529 g.items[1].type != DataType::CERTIFICATE ||
530 g.items[2].type != DataType::CERTIFICATE)
531 throw runtime_error("Invalid item type");
533 if (CKM_API_SUCCESS != m_mgr->saveCertificate(g.items[0].alias, rootCa,
535 throw runtime_error("saveCertificate failed");
537 if (CKM_API_SUCCESS != m_mgr->saveCertificate(g.items[1].alias, imCa,
539 throw runtime_error("saveCertificate failed");
541 if (CKM_API_SUCCESS != m_mgr->saveCertificate(g.items[2].alias, leaf,
543 throw runtime_error("saveCertificate failed");
548 for (const auto &i : g.items) {
550 case DataType::BINARY_DATA:
551 if (CKM_API_SUCCESS != m_mgr->saveData(i.alias, TEST_DATA, i.policy))
552 throw runtime_error("saveData failed");
556 case DataType::KEY_AES:
557 if (CKM_API_SUCCESS != m_mgr->createKeyAES(256, i.alias, i.policy))
558 throw runtime_error("createKeyAES failed");
562 case DataType::CHAIN_CERT_0: // PKCS
563 if (CKM_API_SUCCESS != m_mgr->savePKCS12(i.alias, pkcs, i.policy, i.policy))
564 throw runtime_error("savePkcs12 failed");
569 throw runtime_error("unsupported data type");
578 void SchemeTest::ReadAll(bool useWrongPass)
582 for (const auto &g : GROUPS) {
583 for (const auto &i : g.items) {
585 Password pass = i.policy.password;
595 case DataType::BINARY_DATA: {
596 RawBuffer receivedData;
597 ret = m_mgr->getData(i.alias, pass, receivedData);
598 BOOST_REQUIRE_MESSAGE(useWrongPass || receivedData == TEST_DATA,
599 "Received data is different for " << i.alias);
603 case DataType::KEY_AES:
604 case DataType::KEY_RSA_PRIVATE:
605 case DataType::KEY_RSA_PUBLIC: {
606 KeyShPtr receivedKey;
607 ret = m_mgr->getKey(i.alias, pass, receivedKey);
611 case DataType::CERTIFICATE: {
612 CertificateShPtr receivedCert;
613 ret = m_mgr->getCertificate(i.alias, pass, receivedCert);
617 case DataType::CHAIN_CERT_0: { // pkcs
619 ret = m_mgr->getPKCS12(i.alias, pass, pass, pkcs);
624 BOOST_FAIL("Unsupported data type " << i.type);
627 if (i.policy.extractable) {
629 BOOST_REQUIRE_MESSAGE(ret == CKM_API_ERROR_AUTHENTICATION_FAILED,
630 "Reading item " << i.alias << " should fail with " <<
631 CKM_API_ERROR_AUTHENTICATION_FAILED << " got: " << ret);
633 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS, "Reading item " << i.alias <<
634 " failed with " << ret);
636 BOOST_REQUIRE_MESSAGE(ret == CKM_API_ERROR_NOT_EXPORTABLE, "Item " << i.alias <<
637 " should not be exportable");
643 void SchemeTest::SignVerify()
647 for (const auto &g : GROUPS) {
648 if (g.type == Group::KEY_PAIR_RSA) {
649 BOOST_REQUIRE_MESSAGE(g.items.size() == 2, "Wrong number of keys");
650 BOOST_REQUIRE_MESSAGE(g.items[0].type == DataType::KEY_RSA_PRIVATE &&
651 g.items[1].type == DataType::KEY_RSA_PUBLIC, "Wrong key");
653 SignVerifyItem(g.items[0], g.items[1]);
655 for (const auto &i : g.items) {
657 case DataType::CHAIN_CERT_0:
658 SignVerifyItem(i, i);
669 void SchemeTest::EncryptDecrypt()
673 for (const auto &g : GROUPS) {
674 if (g.type == Group::KEY_PAIR_RSA) {
675 BOOST_REQUIRE_MESSAGE(g.items.size() == 2, "Wrong number of keys");
676 BOOST_REQUIRE_MESSAGE(g.items[0].type == DataType::KEY_RSA_PRIVATE &&
677 g.items[1].type == DataType::KEY_RSA_PUBLIC, "Wrong key");
679 EncryptDecryptItem(g.items[0], g.items[1]);
681 for (const auto &i : g.items) {
683 case DataType::KEY_AES:
684 EncryptDecryptItem(i);
687 case DataType::CHAIN_CERT_0:
688 EncryptDecryptItem(i, i);
699 void SchemeTest::CreateChain()
703 for (const auto &g : GROUPS) {
704 if (g.type == Group::CERT_CHAIN) {
705 BOOST_REQUIRE_MESSAGE(g.items.size() == CHAIN_SIZE, "Not enough certificates");
707 for (const auto &c : g.items)
708 BOOST_REQUIRE_MESSAGE(c.type == DataType::CERTIFICATE, "Wrong item type");
710 Items trusted(CHAIN_SIZE - 1);
711 std::copy(g.items.begin(), g.items.begin() + CHAIN_SIZE - 1, trusted.begin());
713 // last one is ee (leaf)
714 CreateChainItem(g.items.back(), trusted);
716 for (const auto &i : g.items) {
717 if (i.type == DataType::CHAIN_CERT_0) // PKCS
718 CreateChainItem(i, { i });
724 void SchemeTest::RemoveAll()
728 for (const auto &g : GROUPS) {
729 for (const auto &i : g.items) {
730 int ret = m_mgr->removeAlias(i.alias);
731 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
732 "removeAlias() failed with " << ret << " for " << i.alias);
736 size_t SchemeTest::CountObjects()
738 EnableDirectDbAccess();
742 for (const auto &g : GROUPS) {
743 for (const auto &i : g.items) {
745 // it is assumed that aliases are different
746 m_db->getRows(i.alias, OWNER, DataType::DB_FIRST, DataType::DB_LAST, rows);
754 void SchemeTest::RestoreDb()
756 restoreFile("key-7654");
757 restoreFile("db-key-7654");
758 restoreFile("db-7654");
760 m_directAccessEnabled = false;
763 void SchemeTest::CheckSchemeVersion(const ItemFilter &filter, int version)
765 EnableDirectDbAccess();
767 for (const auto &g : GROUPS) {
768 for (const auto &i : g.items) {
769 if (!filter.Matches(i))
773 m_db->getRows(i.alias, OWNER, filter.typeFrom, filter.typeTo, rows);
774 BOOST_REQUIRE_MESSAGE(rows.size() > 0, "No rows found for " << i.alias);
776 for (const auto &r : rows) {
777 BOOST_REQUIRE_MESSAGE(
778 (r.encryptionScheme >> ENC_SCHEME_OFFSET) == version,
779 "Wrong encryption scheme for " << i.alias << ". Expected " << version <<
780 " got: " << (r.encryptionScheme >> ENC_SCHEME_OFFSET));
786 void SchemeTest::EnableDirectDbAccess()
790 if (m_directAccessEnabled)
793 // direct access to db
795 auto wrappedDKEK = fs.getDKEK();
796 auto keyProvider = KeyProvider(wrappedDKEK, DBPASS);
798 auto wrappedDatabaseDEK = fs.getDBDEK();
799 RawBuffer key = keyProvider.getPureDEK(wrappedDatabaseDEK);
801 m_db.reset(new DB::Crypto(fs.getDBPath(), key));
802 m_directAccessEnabled = true;
805 void SchemeTest::SignVerifyItem(const Item &itemPrv, const Item &itemPub)
808 KeyShPtr receivedKey;
810 // create/verify signature
811 ret = m_mgr->createSignature(itemPrv.alias,
812 itemPrv.policy.password,
814 HashAlgorithm::SHA512,
815 RSAPaddingAlgorithm::X931,
817 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
818 "createSignature() failed with " << ret <<
819 " for " << itemPrv.alias);
820 ret = m_mgr->verifySignature(itemPub.alias,
821 itemPub.policy.password,
824 HashAlgorithm::SHA512,
825 RSAPaddingAlgorithm::X931);
826 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
827 "verifySignature() failed with " << ret <<
828 " for " << itemPub.alias);
831 void SchemeTest::EncryptDecryptItem(const Item &item)
833 CryptoAlgorithm algo;
834 RawBuffer iv = createRandomBuffer(IV_LEN);
835 RawBuffer encrypted, decrypted;
838 algo.setParam(ParamName::ALGO_TYPE, AlgoType::AES_GCM);
839 algo.setParam(ParamName::ED_IV, iv);
841 ret = m_mgr->encrypt(algo, item.alias, item.policy.password, TEST_DATA,
843 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
844 "encrypt() failed iwth " << ret << " for " <<
847 ret = m_mgr->decrypt(algo, item.alias, item.policy.password, encrypted,
849 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
850 "decrypt() failed iwth " << ret << " for " <<
853 BOOST_REQUIRE_MESSAGE(decrypted == TEST_DATA,
854 "Decrypted data not equal to original");
857 void SchemeTest::EncryptDecryptItem(const Item &itemPrv, const Item &itemPub)
859 CryptoAlgorithm algo;
860 RawBuffer encrypted, decrypted;
863 algo.setParam(ParamName::ALGO_TYPE, AlgoType::RSA_OAEP);
865 ret = m_mgr->encrypt(algo, itemPub.alias, itemPub.policy.password, TEST_DATA,
867 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
868 "encrypt() failed iwth " << ret << " for " <<
871 ret = m_mgr->decrypt(algo, itemPrv.alias, itemPrv.policy.password, encrypted,
873 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
874 "decrypt() failed iwth " << ret << " for " <<
877 BOOST_REQUIRE_MESSAGE(decrypted == TEST_DATA,
878 "Decrypted data not equal to original");
881 void SchemeTest::CreateChainItem(const Item &leaf, const Items &certs)
883 CertificateShPtrVector chain;
886 if (!leaf.policy.extractable || !leaf.policy.password.empty())
889 for (const auto &i : certs) {
890 if (!i.policy.extractable || !i.policy.password.empty())
893 trusted.push_back(i.alias);
896 CertificateShPtr leafCrt;
897 int ret = m_mgr->getCertificate(leaf.alias, leaf.policy.password, leafCrt);
898 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
899 "getCertificate failed with " << ret << " for " <<
902 ret = m_mgr->getCertificateChain(leafCrt, AliasVector(), trusted, false, chain);
903 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
904 "getCertificateChain() failed with " << ret);
905 BOOST_REQUIRE_MESSAGE(chain.size() == CHAIN_LEN,
906 "Wrong chain length: " << chain.size());