2 * Copyright (c) 2015-2020 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 <db-crypto.h>
40 #include <file-system.h>
41 #include <key-provider.h>
43 #include <dpl/errno_string.h>
44 #include "smack-access.h"
50 const uid_t UID = 7654;
51 const gid_t GID = 7654;
52 const char *const DBPASS = "db-pass";
53 const char *const LABEL = "my-label";
54 const ClientId OWNER = "/" + string(LABEL);
55 const int ENC_SCHEME_OFFSET = 24;
56 const string TEST_DATA_STR = "test-data";
57 RawBuffer TEST_DATA(TEST_DATA_STR.begin(), TEST_DATA_STR.end());
58 const Password TEST_PASS = "custom user password";
59 const size_t IV_LEN = 16;
60 const size_t CHAIN_LEN = 3;
61 const mode_t MODE_0644 = S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH;
73 // [password][exportable]
74 Policy policy[2][2] = {
75 {{ Password(), false }, { Password(), true }},
76 {{ TEST_PASS, false }, { TEST_PASS, true }},
92 Item("data-alias1", DataType::BINARY_DATA, policy[NO_PASS][EXP])
97 Item("data-alias2", DataType::BINARY_DATA, policy[PASS][EXP])
103 Group::KEY_PAIR_RSA, {
104 Item("key-rsa-alias-prv1", DataType::KEY_RSA_PRIVATE, policy[NO_PASS][NO_EXP]),
105 Item("key-rsa-alias-pub1", DataType::KEY_RSA_PUBLIC, policy[NO_PASS][NO_EXP])
109 Group::KEY_PAIR_RSA, {
110 Item("key-rsa-alias-prv2", DataType::KEY_RSA_PRIVATE, policy[NO_PASS][EXP]),
111 Item("key-rsa-alias-pub2", DataType::KEY_RSA_PUBLIC, policy[NO_PASS][EXP]),
115 Group::KEY_PAIR_RSA, {
116 Item("key-rsa-alias-prv3", DataType::KEY_RSA_PRIVATE, policy[PASS][NO_EXP]),
117 Item("key-rsa-alias-pub3", DataType::KEY_RSA_PUBLIC, policy[PASS][NO_EXP]),
121 Group::KEY_PAIR_RSA, {
122 Item("key-rsa-alias-prv4", DataType::KEY_RSA_PRIVATE, policy[PASS][EXP]),
123 Item("key-rsa-alias-pub4", DataType::KEY_RSA_PUBLIC, policy[PASS][EXP]),
126 // different policies
128 Group::KEY_PAIR_RSA, {
129 Item("key-rsa-alias-prv5", DataType::KEY_RSA_PRIVATE, policy[PASS][NO_EXP]),
130 Item("key-rsa-alias-pub5", DataType::KEY_RSA_PUBLIC, policy[NO_PASS][EXP]),
136 Group::SINGLE_ITEM, {
137 Item("key-aes-alias1", DataType::KEY_AES, policy[NO_PASS][NO_EXP]),
141 Group::SINGLE_ITEM, {
142 Item("key-aes-alias2", DataType::KEY_AES, policy[NO_PASS][EXP]),
146 Group::SINGLE_ITEM, {
147 Item("key-aes-alias3", DataType::KEY_AES, policy[PASS][NO_EXP]),
151 Group::SINGLE_ITEM, {
152 Item("key-aes-alias4", DataType::KEY_AES, policy[PASS][EXP]),
159 Item("cert-root-alias1", DataType::CERTIFICATE, policy[NO_PASS][NO_EXP]),
160 Item("cert-im-ca-alias1", DataType::CERTIFICATE, policy[NO_PASS][NO_EXP]),
161 Item("cert-leaf-alias1", DataType::CERTIFICATE, policy[NO_PASS][NO_EXP]),
166 Item("cert-root-alias2", DataType::CERTIFICATE, policy[NO_PASS][EXP]),
167 Item("cert-im-ca-alias2", DataType::CERTIFICATE, policy[NO_PASS][EXP]),
168 Item("cert-leaf-alias2", DataType::CERTIFICATE, policy[NO_PASS][EXP]),
173 Item("cert-root-alias3", DataType::CERTIFICATE, policy[PASS][NO_EXP]),
174 Item("cert-im-ca-alias3", DataType::CERTIFICATE, policy[PASS][NO_EXP]),
175 Item("cert-leaf-alias3", DataType::CERTIFICATE, policy[PASS][NO_EXP]),
180 Item("cert-root-alias4", DataType::CERTIFICATE, policy[PASS][EXP]),
181 Item("cert-im-ca-alias4", DataType::CERTIFICATE, policy[PASS][EXP]),
182 Item("cert-leaf-alias4", DataType::CERTIFICATE, policy[PASS][EXP]),
188 Group::SINGLE_ITEM, {
189 Item("pkcs-alias1", DataType::CHAIN_CERT_0, policy[NO_PASS][NO_EXP]),
193 Group::SINGLE_ITEM, {
194 Item("pkcs-alias2", DataType::CHAIN_CERT_0, policy[NO_PASS][EXP]),
198 Group::SINGLE_ITEM, {
199 Item("pkcs-alias3", DataType::CHAIN_CERT_0, policy[PASS][NO_EXP]),
203 Group::SINGLE_ITEM, {
204 Item("pkcs-alias4", DataType::CHAIN_CERT_0, policy[PASS][EXP]),
209 const size_t CHAIN_SIZE = 3;
211 // TEST_ROOT_CA, expires 2035
212 std::string TEST_ROOT_CA =
213 "-----BEGIN CERTIFICATE-----\n"
214 "MIIDnzCCAoegAwIBAgIJAMH/ADkC5YSTMA0GCSqGSIb3DQEBBQUAMGYxCzAJBgNV\n"
215 "BAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMQ0wCwYDVQQKDARBQ01FMRAwDgYD\n"
216 "VQQLDAdUZXN0aW5nMSEwHwYDVQQDDBhUZXN0IHJvb3QgY2EgY2VydGlmaWNhdGUw\n"
217 "HhcNMTQxMjMwMTcyMTUyWhcNMjQxMjI3MTcyMTUyWjBmMQswCQYDVQQGEwJBVTET\n"
218 "MBEGA1UECAwKU29tZS1TdGF0ZTENMAsGA1UECgwEQUNNRTEQMA4GA1UECwwHVGVz\n"
219 "dGluZzEhMB8GA1UEAwwYVGVzdCByb290IGNhIGNlcnRpZmljYXRlMIIBIjANBgkq\n"
220 "hkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA0EJRdUtd2th0vTVF7QxvDKzyFCF3w9vC\n"
221 "9IDE/Yr12w+a9jd0s7/eG96qTHIYffS3B7x2MB+d4n+SR3W0qmYh7xk8qfEgH3da\n"
222 "eDoV59IZ9r543KM+g8jm6KffYGX1bIJVVY5OhBRbO9nY6byYpd5kbCIUB6dCf7/W\n"
223 "rQl1aIdLGFIegAzPGFPXDcU6F192686x54bxt/itMX4agHJ9ZC/rrTBIZghVsjJo\n"
224 "5/AH5WZpasv8sfrGiiohAxtieoYoJkv5MOYP4/2lPlOY+Cgw1Yoz+HHv31AllgFs\n"
225 "BquBb/kJVmCCNsAOcnvQzTZUsW/TXz9G2nwRdqI1nSy2JvVjZGsqGQIDAQABo1Aw\n"
226 "TjAdBgNVHQ4EFgQUt6pkzFt1PZlfYRL/HGnufF4frdwwHwYDVR0jBBgwFoAUt6pk\n"
227 "zFt1PZlfYRL/HGnufF4frdwwDAYDVR0TBAUwAwEB/zANBgkqhkiG9w0BAQUFAAOC\n"
228 "AQEAld7Qwq0cdzDQ51w1RVLwTR8Oy25PB3rzwEHcSGJmdqlMi3xOdaz80S1R1BBX\n"
229 "ldvGBG5Tn0vT7xSuhmSgI2/HnBpy9ocHVOmhtNB4473NieEpfTYrnGXrFxu46Wus\n"
230 "9m/ZnugcQ2G6C54A/NFtvgLmaC8uH8M7gKdS6uYUwJFQEofkjmd4UpOYSqmcRXhS\n"
231 "Jzd5FYFWkJhKJYp3nlENSOD8CUFFVGekm05nFN2gRVc/qaqQkEX77+XYvhodLRsV\n"
232 "qMn7nf7taidDKLO2T4bhujztnTYOhhaXKgPy7AtZ28N2wvX96VyAPB/vrchGmyBK\n"
233 "kOg11TpPdNDkhb1J4ZCh2gupDg==\n"
234 "-----END CERTIFICATE-----\n";
236 // TEST_IM_CA, signed by TEST_ROOT_CA, expires 2035
237 std::string TEST_IM_CA =
238 "-----BEGIN CERTIFICATE-----\n"
239 "MIIDljCCAn6gAwIBAgICEAAwDQYJKoZIhvcNAQEFBQAwZjELMAkGA1UEBhMCQVUx\n"
240 "EzARBgNVBAgMClNvbWUtU3RhdGUxDTALBgNVBAoMBEFDTUUxEDAOBgNVBAsMB1Rl\n"
241 "c3RpbmcxITAfBgNVBAMMGFRlc3Qgcm9vdCBjYSBjZXJ0aWZpY2F0ZTAeFw0xNTAx\n"
242 "MTYxNjQ1MzRaFw0zNTAxMTExNjQ1MzRaMGQxCzAJBgNVBAYTAkFVMRMwEQYDVQQI\n"
243 "DApTb21lLVN0YXRlMQ0wCwYDVQQKDARBQ01FMRAwDgYDVQQLDAdUZXN0aW5nMR8w\n"
244 "HQYDVQQDDBZUZXN0IElNIENBIGNlcnRpZmljYXRlMIIBIjANBgkqhkiG9w0BAQEF\n"
245 "AAOCAQ8AMIIBCgKCAQEAzmBF78qClgoKfnLAncMXZwZ14TW+5kags1+QCYeg3c7j\n"
246 "L9+RvDxIaX2tKf1sukJcwQfYqUlQkwt+58LMOb2ORtkpj8Or6WCWCZ0BzneT8ug7\n"
247 "nxJT4m9+bohMF0JoKjjB2H4KNMHamLIwUxRKt6nyfk81kVhJOi2vzzxd+UCPi6Pc\n"
248 "UAbJNH48eNgOIg55nyFovVzYj8GIo/9GvHJj83PPa/KlJZ+Z1qZASZZ/VYorplVT\n"
249 "thsHXKfejhFy5YJ9t7n/vyAQsyBsagZsvX19xnH41fbYXHKf8UbXG23rNaZlchs6\n"
250 "XJVLQdzOpj3WTj/lCocVHqLaZISLhNQ3aI7kUBUdiwIDAQABo1AwTjAdBgNVHQ4E\n"
251 "FgQUoCYNaCBP4jl/3SYQuK8Ka+6i3QEwHwYDVR0jBBgwFoAUt6pkzFt1PZlfYRL/\n"
252 "HGnufF4frdwwDAYDVR0TBAUwAwEB/zANBgkqhkiG9w0BAQUFAAOCAQEAjRzWiD97\n"
253 "Htv4Kxpm3P+C+xP9AEteCJfO+7p8MWgtWEJOknJyt55zeKS2JwZIq57KcbqD8U7v\n"
254 "vAUx1ymtUhlFPFd7J1mJ3pou+3aFYmGShYhGHpbrmUwjp7HVP588jrW1NoZVHdMc\n"
255 "4OgJWFrViXeu9+maIcekjMB/+9Y0dUgQuK5ZuT5H/Jwet7Th/o9uufTUZjBzRvrB\n"
256 "pbXgQpqgME2av4Q/6LuldPCTHLtWXgFUU2R+yCGmuGilvhFJnKoQryAbYnIQNWE8\n"
257 "SLoHQ9s1i7Zyb7HU6UAaqMOz15LBkyAqtNyJcO2p7Q/p5YK0xfD4xisI5qXucqVm\n"
259 "-----END CERTIFICATE-----\n";
261 // TEST_LEAF, signed by TEST_IM_CA, expires 2035
262 std::string TEST_LEAF =
263 "-----BEGIN CERTIFICATE-----\n"
264 "MIIDOzCCAiMCAQEwDQYJKoZIhvcNAQEFBQAwZDELMAkGA1UEBhMCQVUxEzARBgNV\n"
265 "BAgMClNvbWUtU3RhdGUxDTALBgNVBAoMBEFDTUUxEDAOBgNVBAsMB1Rlc3Rpbmcx\n"
266 "HzAdBgNVBAMMFlRlc3QgSU0gQ0EgY2VydGlmaWNhdGUwHhcNMTUwMTE2MTY0ODE0\n"
267 "WhcNMzUwMTExMTY0ODE0WjBjMQswCQYDVQQGEwJBVTETMBEGA1UECAwKU29tZS1T\n"
268 "dGF0ZTENMAsGA1UECgwEQUNNRTEQMA4GA1UECwwHVGVzdGluZzEeMBwGA1UEAwwV\n"
269 "VGVzdCBsZWFmIGNlcnRpZmljYXRlMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIB\n"
270 "CgKCAQEAzTdDIa2tDmRxFnIgiG+mBz8GoSVODs0ImNQGbqj+pLhBOFRH8fsah4Jl\n"
271 "z5YF9KwhMVLknnHGFLE/Nb7Ac35kEzhMQMpTRxohW83oxw3eZ8zN/FBoKqg4qHRq\n"
272 "QR8kS10YXTgrBR0ex/Vp+OUKEw6h7yL2r4Tpvrn9/qHwsxtLxqWbDIVf1O9b1Lfc\n"
273 "bllYMdmV5E62yN5tcwrDP8gvHjFnVeLzrG8wTpc9FR90/0Jkfp5jAJcArOBLrT0E\n"
274 "4VRqs+4HuwT8jAwFAmNnc7IYX5qSjtSWkmmHe73K/lzB+OiI0JEc/3eWUTWqwTSk\n"
275 "4tNCiQGBKJ39LXPTBBJdzmxVH7CUDQIDAQABMA0GCSqGSIb3DQEBBQUAA4IBAQAp\n"
276 "UdDOGu3hNiG+Vn10aQ6B1ZmOj3t+45gUV3sC+y8hB8EK1g4P5Ke9bVDts0T5eOnj\n"
277 "CSc+6VoND5O4adI0IFFRFljHNVnvjeosHfUZNnowsmA2ptQBtC1g5ZKRvKXlkC5/\n"
278 "i5BGgRqPFA7y9WB9Y05MrJHf3E+Oz/RBsLeeNiNN+rF5X1vYExvGHpo0M0zS0ze9\n"
279 "HtC0aOy8ocsTrQkf3ceHTAXx2i8ftoSSD4klojtWFpWMrNQa52F7wB9nU6FfKRuF\n"
280 "Zj/T1JkYXKkEwZU6nAR2jdZp3EP9xj3o15V/tyFcXHx6l8NTxn4cJb+Xe4VquQJz\n"
281 "6ON7PVe0ABN/AlwVQiFE\n"
282 "-----END CERTIFICATE-----\n";
287 void operator()(int *fd)
294 typedef std::unique_ptr<int, FdCloser> FdPtr;
296 uid_t getUid(const char *name)
299 struct passwd *result = nullptr;
300 int bufsize = sysconf(_SC_GETPW_R_SIZE_MAX);
303 bufsize = 16384; /* should be more than enough */
305 memset(&pwd, 0x00, sizeof(pwd));
307 std::vector<char> buf(bufsize, 0);
309 int ret = getpwnam_r(name, &pwd, buf.data(), bufsize, &result);
310 BOOST_REQUIRE_MESSAGE(ret == 0 && result, "getpwnam_r failed");
315 gid_t getGid(const char *name)
318 struct group *result = nullptr;
319 size_t bufsize = sysconf(_SC_GETGR_R_SIZE_MAX);
322 bufsize = 16384; /* should be more than enough */
324 memset(&grp, 0x00, sizeof(grp));
326 std::vector<char> buf(bufsize, 0);
328 int ret = getgrnam_r(name, &grp, buf.data(), bufsize, &result);
329 BOOST_REQUIRE_MESSAGE(ret == 0 && result, "getgrnam_r failed");
334 void restoreFile(const string &filename)
336 static uid_t CKM_UID = getUid(USER_NAME);
337 static gid_t CKM_GID = getGid(GROUP_NAME);
338 string sourcePath = string(MISC_DIR) + "/" + filename;
339 string targetPath = string(RW_DATA_DIR) + "/" + filename;
343 int sourceFd = TEMP_FAILURE_RETRY(open(sourcePath.c_str(), O_RDONLY));
345 BOOST_REQUIRE_MESSAGE(sourceFd > 0,
346 "Opening " << sourcePath << " failed: " << GetErrnoString(err));
348 FdPtr sourceFdPtr(&sourceFd);
350 int targetFd = TEMP_FAILURE_RETRY(creat(targetPath.c_str(), MODE_0644));
352 BOOST_REQUIRE_MESSAGE(targetFd > 0,
353 "Creating " << targetPath << " failed: " << GetErrnoString(err));
355 ret = fchown(targetFd, CKM_UID, CKM_GID);
357 BOOST_REQUIRE_MESSAGE(ret != -1, "fchown() failed: " << GetErrnoString(err));
359 FdPtr targetFdPtr(&targetFd);
361 struct stat sourceStat;
362 ret = fstat(sourceFd, &sourceStat);
364 BOOST_REQUIRE_MESSAGE(ret != -1, "fstat() failed: " << GetErrnoString(err));
366 ret = sendfile(targetFd, sourceFd, 0, sourceStat.st_size);
368 BOOST_REQUIRE_MESSAGE(ret != -1, "sendfile() failed: " << GetErrnoString(err));
370 ret = fsync(targetFd);
372 BOOST_REQUIRE_MESSAGE(ret != -1, "fsync() failed: " << GetErrnoString(err));
375 void generateRandom(size_t random_bytes, unsigned char *output)
377 if (random_bytes <= 0 || !output)
378 throw runtime_error("Invalid param");
380 std::ifstream is("/dev/urandom", std::ifstream::binary);
383 throw runtime_error("Failed to read /dev/urandom");
385 is.read(reinterpret_cast<char *>(output), random_bytes);
387 if (static_cast<std::streamsize>(random_bytes) != is.gcount())
388 throw runtime_error("Not enough bytes read from /dev/urandom");
391 RawBuffer createRandomBuffer(size_t random_bytes)
393 RawBuffer buffer(random_bytes);
394 generateRandom(buffer.size(), buffer.data());
397 } // namespace anonymous
400 SchemeTest::SchemeTest() : m_userChanged(false), m_directAccessEnabled(false)
402 m_control = Control::create();
403 m_mgr = Manager::create();
406 sa.add("System", LABEL, "rwx");
407 sa.add(LABEL, "System", "rwx");
408 sa.add(LABEL, "System::Run", "x");
412 SchemeTest::~SchemeTest()
419 void SchemeTest::RemoveUserData()
421 if (CKM_API_SUCCESS != m_control->lockUserKey(UID))
422 throw runtime_error("lockUserKey failed");
424 if (CKM_API_SUCCESS != m_control->removeUserData(UID))
425 throw runtime_error("removeUserData failed");
428 void SchemeTest::SwitchToUser()
433 if (CKM_API_SUCCESS != m_control->unlockUserKey(UID, DBPASS))
434 throw runtime_error("unlockUserKey failed");
439 if (smack_new_label_from_self(&label) <= 0)
440 throw runtime_error("smack_new_label_from_self failed");
442 m_origLabel = string(label);
445 if (0 > smack_set_label_for_self(LABEL))
446 throw runtime_error("smack_set_label_for_self failed");
448 if (0 > setegid(GID))
449 throw runtime_error("setegid failed");
451 if (0 > seteuid(UID))
452 throw runtime_error("seteuid failed");
454 m_userChanged = true;
457 void SchemeTest::SwitchToRoot()
463 throw runtime_error("seteuid failed");
466 throw runtime_error("setegid failed");
468 if (0 > smack_set_label_for_self(m_origLabel.c_str()))
469 throw runtime_error("smack_set_label_for_self failed");
471 if (m_control->lockUserKey(UID) != CKM_API_SUCCESS)
472 throw runtime_error("lockUserKey failed");
475 void SchemeTest::FillDb()
478 ifstream is(MISC_DIR "/encryption-scheme.p12");
481 throw runtime_error("Failed to read pkcs");
483 istreambuf_iterator<char> begin(is), end;
484 RawBuffer pkcsBuffer(begin, end);
485 auto pkcs = PKCS12::create(pkcsBuffer, Password());
488 throw runtime_error("Empty pkcs");
493 RawBuffer rootCaBuffer(TEST_ROOT_CA.begin(), TEST_ROOT_CA.end());
494 CertificateShPtr rootCa = CKM::Certificate::create(rootCaBuffer,
495 CKM::DataFormat::FORM_PEM);
496 RawBuffer imCaBuffer(TEST_IM_CA.begin(), TEST_IM_CA.end());
497 CertificateShPtr imCa = CKM::Certificate::create(imCaBuffer,
498 CKM::DataFormat::FORM_PEM);
499 RawBuffer leafBuffer(TEST_LEAF.begin(), TEST_LEAF.end());
500 CertificateShPtr leaf = CKM::Certificate::create(leafBuffer,
501 CKM::DataFormat::FORM_PEM);
503 for (const auto &g : GROUPS) {
505 case Group::KEY_PAIR_RSA:
506 if (g.items.size() != 2)
507 throw runtime_error("Wrong number of keys");
509 if (g.items[0].type != DataType::KEY_RSA_PRIVATE ||
510 g.items[1].type != DataType::KEY_RSA_PUBLIC)
511 throw runtime_error("Invalid item type");
513 if (CKM_API_SUCCESS != m_mgr->createKeyPairRSA(1024,
518 throw runtime_error("createKeyPair failed");
522 case Group::CERT_CHAIN:
523 if (g.items.size() != CHAIN_SIZE)
524 throw runtime_error("Wrong number of certificates");
526 if (g.items[0].type != DataType::CERTIFICATE ||
527 g.items[1].type != DataType::CERTIFICATE ||
528 g.items[2].type != DataType::CERTIFICATE)
529 throw runtime_error("Invalid item type");
531 if (CKM_API_SUCCESS != m_mgr->saveCertificate(g.items[0].alias, rootCa,
533 throw runtime_error("saveCertificate failed");
535 if (CKM_API_SUCCESS != m_mgr->saveCertificate(g.items[1].alias, imCa,
537 throw runtime_error("saveCertificate failed");
539 if (CKM_API_SUCCESS != m_mgr->saveCertificate(g.items[2].alias, leaf,
541 throw runtime_error("saveCertificate failed");
546 for (const auto &i : g.items) {
548 case DataType::BINARY_DATA:
549 if (CKM_API_SUCCESS != m_mgr->saveData(i.alias, TEST_DATA, i.policy))
550 throw runtime_error("saveData failed");
554 case DataType::KEY_AES:
555 if (CKM_API_SUCCESS != m_mgr->createKeyAES(256, i.alias, i.policy))
556 throw runtime_error("createKeyAES failed");
560 case DataType::CHAIN_CERT_0: // PKCS
561 if (CKM_API_SUCCESS != m_mgr->savePKCS12(i.alias, pkcs, i.policy, i.policy))
562 throw runtime_error("savePkcs12 failed");
567 throw runtime_error("unsupported data type");
576 void SchemeTest::CheckAliasInfo()
579 for (const auto &g : GROUPS) {
580 for (const auto &i : g.items) {
582 bool encStatus = false;
583 Password pass = i.policy.password;
591 case DataType::BINARY_DATA: {
592 ret = m_mgr->getDataEncryptionStatus(i.alias, encStatus);
593 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
594 "can not get data encryption status, ret: " +
595 std::to_string(ret));
599 case DataType::KEY_AES:
600 case DataType::KEY_RSA_PRIVATE:
601 case DataType::KEY_RSA_PUBLIC: {
602 ret = m_mgr->getKeyEncryptionStatus(i.alias, encStatus);
603 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
604 "can not get key encryption status, ret: " +
605 std::to_string(ret));
609 case DataType::CERTIFICATE: {
610 ret = m_mgr->getCertificateEncryptionStatus(i.alias, encStatus);
611 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
612 "can not get certificate encryption status, ret: " +
613 std::to_string(ret));
617 case DataType::CHAIN_CERT_0: {
618 ret = m_mgr->getCertificateEncryptionStatus(i.alias, encStatus);
619 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
620 "can not get certificate encryption status, ret: " +
621 std::to_string(ret));
622 ret = m_mgr->getKeyEncryptionStatus(i.alias, encStatus);
623 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
624 "can not get key encryption status, ret: " +
625 std::to_string(ret));
630 BOOST_FAIL("Unsupported data type " << i.type);
632 BOOST_REQUIRE_MESSAGE(encStatus == !i.policy.password.empty(), "item: " <<
633 i.alias << " has wrong encryption status: " << encStatus);
640 void SchemeTest::ReadAll(bool useWrongPass)
644 for (const auto &g : GROUPS) {
645 for (const auto &i : g.items) {
647 Password pass = i.policy.password;
657 case DataType::BINARY_DATA: {
658 RawBuffer receivedData;
659 ret = m_mgr->getData(i.alias, pass, receivedData);
660 BOOST_REQUIRE_MESSAGE(useWrongPass || receivedData == TEST_DATA,
661 "Received data is different for " << i.alias);
665 case DataType::KEY_AES:
666 case DataType::KEY_RSA_PRIVATE:
667 case DataType::KEY_RSA_PUBLIC: {
668 KeyShPtr receivedKey;
669 ret = m_mgr->getKey(i.alias, pass, receivedKey);
673 case DataType::CERTIFICATE: {
674 CertificateShPtr receivedCert;
675 ret = m_mgr->getCertificate(i.alias, pass, receivedCert);
679 case DataType::CHAIN_CERT_0: { // pkcs
681 ret = m_mgr->getPKCS12(i.alias, pass, pass, pkcs);
686 BOOST_FAIL("Unsupported data type " << i.type);
689 if (i.policy.extractable) {
691 BOOST_REQUIRE_MESSAGE(ret == CKM_API_ERROR_AUTHENTICATION_FAILED,
692 "Reading item " << i.alias << " should fail with " <<
693 CKM_API_ERROR_AUTHENTICATION_FAILED << " got: " << ret);
695 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS, "Reading item " << i.alias <<
696 " failed with " << ret);
698 BOOST_REQUIRE_MESSAGE(ret == CKM_API_ERROR_NOT_EXPORTABLE, "Item " << i.alias <<
699 " should not be exportable");
705 void SchemeTest::SignVerify()
709 for (const auto &g : GROUPS) {
710 if (g.type == Group::KEY_PAIR_RSA) {
711 BOOST_REQUIRE_MESSAGE(g.items.size() == 2, "Wrong number of keys");
712 BOOST_REQUIRE_MESSAGE(g.items[0].type == DataType::KEY_RSA_PRIVATE &&
713 g.items[1].type == DataType::KEY_RSA_PUBLIC, "Wrong key");
715 SignVerifyItem(g.items[0], g.items[1]);
717 for (const auto &i : g.items) {
719 case DataType::CHAIN_CERT_0:
720 SignVerifyItem(i, i);
731 void SchemeTest::EncryptDecrypt()
735 for (const auto &g : GROUPS) {
736 if (g.type == Group::KEY_PAIR_RSA) {
737 BOOST_REQUIRE_MESSAGE(g.items.size() == 2, "Wrong number of keys");
738 BOOST_REQUIRE_MESSAGE(g.items[0].type == DataType::KEY_RSA_PRIVATE &&
739 g.items[1].type == DataType::KEY_RSA_PUBLIC, "Wrong key");
741 EncryptDecryptItem(g.items[0], g.items[1]);
743 for (const auto &i : g.items) {
745 case DataType::KEY_AES:
746 EncryptDecryptItem(i);
749 case DataType::CHAIN_CERT_0:
750 EncryptDecryptItem(i, i);
761 void SchemeTest::CreateChain()
765 for (const auto &g : GROUPS) {
766 if (g.type == Group::CERT_CHAIN) {
767 BOOST_REQUIRE_MESSAGE(g.items.size() == CHAIN_SIZE, "Not enough certificates");
769 for (const auto &c : g.items)
770 BOOST_REQUIRE_MESSAGE(c.type == DataType::CERTIFICATE, "Wrong item type");
772 Items trusted(CHAIN_SIZE - 1);
773 std::copy(g.items.begin(), g.items.begin() + CHAIN_SIZE - 1, trusted.begin());
775 // last one is ee (leaf)
776 CreateChainItem(g.items.back(), trusted);
778 for (const auto &i : g.items) {
779 if (i.type == DataType::CHAIN_CERT_0) // PKCS
780 CreateChainItem(i, { i });
786 void SchemeTest::RemoveAll()
790 for (const auto &g : GROUPS) {
791 for (const auto &i : g.items) {
792 int ret = m_mgr->removeAlias(i.alias);
793 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
794 "removeAlias() failed with " << ret << " for " << i.alias);
798 size_t SchemeTest::CountObjects()
800 EnableDirectDbAccess();
804 for (const auto &g : GROUPS) {
805 for (const auto &i : g.items) {
807 // it is assumed that aliases are different
808 m_db->getRows(i.alias, OWNER, DataType::DB_FIRST, DataType::DB_LAST, rows);
816 void SchemeTest::RestoreDb()
819 restoreFile("key-7654");
820 restoreFile("db-key-7654");
821 restoreFile("db-7654");
822 m_directAccessEnabled = false;
825 void SchemeTest::CheckSchemeVersion(const ItemFilter &filter, int version)
827 EnableDirectDbAccess();
829 for (const auto &g : GROUPS) {
830 for (const auto &i : g.items) {
831 if (!filter.Matches(i))
835 m_db->getRows(i.alias, OWNER, filter.typeFrom, filter.typeTo, rows);
836 BOOST_REQUIRE_MESSAGE(rows.size() > 0, "No rows found for " << i.alias);
838 for (const auto &r : rows) {
839 BOOST_REQUIRE_MESSAGE(
840 (r.encryptionScheme >> ENC_SCHEME_OFFSET) == version,
841 "Wrong encryption scheme for " << i.alias << ". Expected " << version <<
842 " got: " << (r.encryptionScheme >> ENC_SCHEME_OFFSET));
848 void SchemeTest::EnableDirectDbAccess()
852 if (m_directAccessEnabled)
855 // direct access to db
857 auto wrappedDKEK = fs.getDKEK();
858 auto keyProvider = KeyProvider(wrappedDKEK, DBPASS);
859 if (!keyProvider.isInitialized()) {
860 keyProvider.migrateDomainKEK(wrappedDKEK, DBPASS);
861 fs.saveDKEK(keyProvider.getWrappedDomainKEK(DBPASS));
862 LogInfo("DKEK migrated");
865 auto wrappedDatabaseDEK = fs.getDBDEK();
866 RawBuffer key = keyProvider.getPureDEK(wrappedDatabaseDEK);
868 m_db.reset(new DB::Crypto(fs.getLegacyDBPath(), fs.getDBPath(), key));
869 m_directAccessEnabled = true;
871 // Legacy db files of the form db-$uid are incompatible with upstream sqlcipher.
872 // DB::Crypto(...) converts them to db0-$uid upstream-compatible, then deletes them.
873 // This function runs DB::Crypto(...) as root so db0-$uid are root-owned.
874 // However, database files need to be accessible to USER_NAME/GROUP_NAME (ex. ReadAll()).
875 // Thus the need to fix up ownership, much like restoreFile() does.
876 BOOST_REQUIRE(!chown(RW_DATA_DIR "/db0-7654", getUid(USER_NAME), getGid(GROUP_NAME)));
879 void SchemeTest::SignVerifyItem(const Item &itemPrv, const Item &itemPub)
882 KeyShPtr receivedKey;
884 // create/verify signature
885 ret = m_mgr->createSignature(itemPrv.alias,
886 itemPrv.policy.password,
888 HashAlgorithm::SHA512,
889 RSAPaddingAlgorithm::X931,
891 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
892 "createSignature() failed with " << ret <<
893 " for " << itemPrv.alias);
894 ret = m_mgr->verifySignature(itemPub.alias,
895 itemPub.policy.password,
898 HashAlgorithm::SHA512,
899 RSAPaddingAlgorithm::X931);
900 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
901 "verifySignature() failed with " << ret <<
902 " for " << itemPub.alias);
905 void SchemeTest::EncryptDecryptItem(const Item &item)
907 CryptoAlgorithm algo;
908 RawBuffer iv = createRandomBuffer(IV_LEN);
909 RawBuffer encrypted, decrypted;
912 algo.setParam(ParamName::ALGO_TYPE, AlgoType::AES_GCM);
913 algo.setParam(ParamName::ED_IV, iv);
915 ret = m_mgr->encrypt(algo, item.alias, item.policy.password, TEST_DATA,
917 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
918 "encrypt() failed iwth " << ret << " for " <<
921 ret = m_mgr->decrypt(algo, item.alias, item.policy.password, encrypted,
923 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
924 "decrypt() failed iwth " << ret << " for " <<
927 BOOST_REQUIRE_MESSAGE(decrypted == TEST_DATA,
928 "Decrypted data not equal to original");
931 void SchemeTest::EncryptDecryptItem(const Item &itemPrv, const Item &itemPub)
933 CryptoAlgorithm algo;
934 RawBuffer encrypted, decrypted;
937 algo.setParam(ParamName::ALGO_TYPE, AlgoType::RSA_OAEP);
939 ret = m_mgr->encrypt(algo, itemPub.alias, itemPub.policy.password, TEST_DATA,
941 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
942 "encrypt() failed iwth " << ret << " for " <<
945 ret = m_mgr->decrypt(algo, itemPrv.alias, itemPrv.policy.password, encrypted,
947 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
948 "decrypt() failed iwth " << ret << " for " <<
951 BOOST_REQUIRE_MESSAGE(decrypted == TEST_DATA,
952 "Decrypted data not equal to original");
955 void SchemeTest::CreateChainItem(const Item &leaf, const Items &certs)
957 CertificateShPtrVector chain;
960 if (!leaf.policy.extractable || !leaf.policy.password.empty())
963 for (const auto &i : certs) {
964 if (!i.policy.extractable || !i.policy.password.empty())
967 trusted.push_back(i.alias);
970 CertificateShPtr leafCrt;
971 int ret = m_mgr->getCertificate(leaf.alias, leaf.policy.password, leafCrt);
972 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
973 "getCertificate failed with " << ret << " for " <<
976 ret = m_mgr->getCertificateChain(leafCrt, AliasVector(), trusted, false, chain);
977 BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
978 "getCertificateChain() failed with " << ret);
979 BOOST_REQUIRE_MESSAGE(chain.size() == CHAIN_LEN,
980 "Wrong chain length: " << chain.size());