tests/encryption-scheme/scheme-test.cpp

   1 /*
   2  *  Copyright (c) 2015 Samsung Electronics Co., Ltd All Rights Reserved
   3  *
   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
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  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
  15  */
  16 /*
  17  * @file       scheme-test.cpp
  18  * @author     Krzysztof Jackiewicz (k.jackiewicz@samsung.com)
  19  * @version    1.0
  20  */
  21 #include <scheme-test.h>
  22
  23 #include <sys/smack.h>
  24 #include <sys/types.h>
  25 #include <sys/stat.h>
  26 #include <sys/sendfile.h>
  27 #include <fcntl.h>
  28 #include <unistd.h>
  29 #include <string.h>
  30
  31 #include <fstream>
  32 #include <stdexcept>
  33
  34 #include <boost/test/unit_test.hpp>
  35
  36 #include <smack-access.h>
  37
  38 #include <db-crypto.h>
  39 #include <file-system.h>
  40 #include <key-provider.h>
  41 #include <db-row.h>
  42 #include <crypto-init.h>
  43
  44 using namespace CKM;
  45 using namespace std;
  46
  47 namespace {
  48 const uid_t UID = 7654;
  49 const gid_t GID = 7654;
  50 const char* const DBPASS = "db-pass";
  51 const char* const LABEL = "my-label";
  52 const Label DB_LABEL = "/" + string(LABEL);
  53 const int ENC_SCHEME_OFFSET = 24;
  54 const string TEST_DATA_STR = "test-data";
  55 RawBuffer TEST_DATA(TEST_DATA_STR.begin(), TEST_DATA_STR.end());
  56 const Password TEST_PASS = "custom user password";
  57 const size_t IV_LEN = 16;
  58 const size_t CHAIN_LEN = 3;
  59 const uid_t CKM_UID = 0;
  60 const gid_t CKM_GID = 0;
  61
  62 enum {
  63     NO_PASS = 0,
  64     PASS = 1
  65 };
  66
  67 enum {
  68     NO_EXP = 0,
  69     EXP = 1
  70 };
  71
  72 // [password][exportable]
  73 Policy policy[2][2] = {
  74         {{ Password(), false }, { Password(), true }},
  75         {{ TEST_PASS,  false }, { TEST_PASS,  true }},
  76 };
  77
  78 struct Group {
  79     enum {
  80         SINGLE_ITEM,
  81         KEY_PAIR_RSA,
  82         CERT_CHAIN
  83     } type;
  84     Items items;
  85 };
  86
  87 Group GROUPS[] = {
  88     // Data
  89     { Group::SINGLE_ITEM, {
  90             Item("data-alias1", DataType::BINARY_DATA,  policy[NO_PASS][EXP])
  91     }},
  92     { Group::SINGLE_ITEM, {
  93             Item("data-alias2", DataType::BINARY_DATA,  policy[PASS][EXP])
  94     }},
  95
  96     // RSA keys
  97     { Group::KEY_PAIR_RSA, {
  98             Item("key-rsa-alias-prv1", DataType::KEY_RSA_PRIVATE,  policy[NO_PASS][NO_EXP]),
  99             Item("key-rsa-alias-pub1", DataType::KEY_RSA_PUBLIC,   policy[NO_PASS][NO_EXP])
 100     }},
 101     { Group::KEY_PAIR_RSA, {
 102             Item("key-rsa-alias-prv2", DataType::KEY_RSA_PRIVATE,  policy[NO_PASS][EXP]),
 103             Item("key-rsa-alias-pub2", DataType::KEY_RSA_PUBLIC,   policy[NO_PASS][EXP]),
 104     }},
 105     { Group::KEY_PAIR_RSA, {
 106             Item("key-rsa-alias-prv3", DataType::KEY_RSA_PRIVATE,  policy[PASS][NO_EXP]),
 107             Item("key-rsa-alias-pub3", DataType::KEY_RSA_PUBLIC,   policy[PASS][NO_EXP]),
 108     }},
 109     { Group::KEY_PAIR_RSA, {
 110             Item("key-rsa-alias-prv4", DataType::KEY_RSA_PRIVATE,  policy[PASS][EXP]),
 111             Item("key-rsa-alias-pub4", DataType::KEY_RSA_PUBLIC,   policy[PASS][EXP]),
 112     }},
 113     // different policies
 114     { Group::KEY_PAIR_RSA, {
 115             Item("key-rsa-alias-prv5", DataType::KEY_RSA_PRIVATE,  policy[PASS][NO_EXP]),
 116             Item("key-rsa-alias-pub5", DataType::KEY_RSA_PUBLIC,   policy[NO_PASS][EXP]),
 117     }},
 118
 119     // AES
 120     { Group::SINGLE_ITEM, {
 121             Item("key-aes-alias1",     DataType::KEY_AES,          policy[NO_PASS][NO_EXP]),
 122     }},
 123     { Group::SINGLE_ITEM, {
 124             Item("key-aes-alias2",     DataType::KEY_AES,          policy[NO_PASS][EXP]),
 125     }},
 126     { Group::SINGLE_ITEM, {
 127             Item("key-aes-alias3",     DataType::KEY_AES,          policy[PASS][NO_EXP]),
 128     }},
 129     { Group::SINGLE_ITEM, {
 130             Item("key-aes-alias4",     DataType::KEY_AES,          policy[PASS][EXP]),
 131     }},
 132
 133     // Certificates
 134     { Group::CERT_CHAIN, {
 135             Item("cert-root-alias1",   DataType::CERTIFICATE,      policy[NO_PASS][NO_EXP]),
 136             Item("cert-im-ca-alias1",  DataType::CERTIFICATE,      policy[NO_PASS][NO_EXP]),
 137             Item("cert-leaf-alias1",   DataType::CERTIFICATE,      policy[NO_PASS][NO_EXP]),
 138     }},
 139     { Group::CERT_CHAIN, {
 140             Item("cert-root-alias2",   DataType::CERTIFICATE,      policy[NO_PASS][EXP]),
 141             Item("cert-im-ca-alias2",  DataType::CERTIFICATE,      policy[NO_PASS][EXP]),
 142             Item("cert-leaf-alias2",   DataType::CERTIFICATE,      policy[NO_PASS][EXP]),
 143     }},
 144     { Group::CERT_CHAIN, {
 145             Item("cert-root-alias3",   DataType::CERTIFICATE,      policy[PASS][NO_EXP]),
 146             Item("cert-im-ca-alias3",  DataType::CERTIFICATE,      policy[PASS][NO_EXP]),
 147             Item("cert-leaf-alias3",   DataType::CERTIFICATE,      policy[PASS][NO_EXP]),
 148     }},
 149     { Group::CERT_CHAIN, {
 150             Item("cert-root-alias4",   DataType::CERTIFICATE,      policy[PASS][EXP]),
 151             Item("cert-im-ca-alias4",  DataType::CERTIFICATE,      policy[PASS][EXP]),
 152             Item("cert-leaf-alias4",   DataType::CERTIFICATE,      policy[PASS][EXP]),
 153     }},
 154
 155     // PKCS
 156     { Group::SINGLE_ITEM, {
 157             Item("pkcs-alias1",        DataType::CHAIN_CERT_0,     policy[NO_PASS][NO_EXP]),
 158     }},
 159     { Group::SINGLE_ITEM, {
 160             Item("pkcs-alias2",        DataType::CHAIN_CERT_0,     policy[NO_PASS][EXP]),
 161     }},
 162     { Group::SINGLE_ITEM, {
 163             Item("pkcs-alias3",        DataType::CHAIN_CERT_0,     policy[PASS][NO_EXP]),
 164     }},
 165     { Group::SINGLE_ITEM, {
 166             Item("pkcs-alias4",        DataType::CHAIN_CERT_0,     policy[PASS][EXP]),
 167     }},
 168 };
 169
 170 const size_t CHAIN_SIZE = 3;
 171
 172 // TEST_ROOT_CA, expires 2035
 173 std::string TEST_ROOT_CA =
 174     "-----BEGIN CERTIFICATE-----\n"
 175     "MIIDnzCCAoegAwIBAgIJAMH/ADkC5YSTMA0GCSqGSIb3DQEBBQUAMGYxCzAJBgNV\n"
 176     "BAYTAkFVMRMwEQYDVQQIDApTb21lLVN0YXRlMQ0wCwYDVQQKDARBQ01FMRAwDgYD\n"
 177     "VQQLDAdUZXN0aW5nMSEwHwYDVQQDDBhUZXN0IHJvb3QgY2EgY2VydGlmaWNhdGUw\n"
 178     "HhcNMTQxMjMwMTcyMTUyWhcNMjQxMjI3MTcyMTUyWjBmMQswCQYDVQQGEwJBVTET\n"
 179     "MBEGA1UECAwKU29tZS1TdGF0ZTENMAsGA1UECgwEQUNNRTEQMA4GA1UECwwHVGVz\n"
 180     "dGluZzEhMB8GA1UEAwwYVGVzdCByb290IGNhIGNlcnRpZmljYXRlMIIBIjANBgkq\n"
 181     "hkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA0EJRdUtd2th0vTVF7QxvDKzyFCF3w9vC\n"
 182     "9IDE/Yr12w+a9jd0s7/eG96qTHIYffS3B7x2MB+d4n+SR3W0qmYh7xk8qfEgH3da\n"
 183     "eDoV59IZ9r543KM+g8jm6KffYGX1bIJVVY5OhBRbO9nY6byYpd5kbCIUB6dCf7/W\n"
 184     "rQl1aIdLGFIegAzPGFPXDcU6F192686x54bxt/itMX4agHJ9ZC/rrTBIZghVsjJo\n"
 185     "5/AH5WZpasv8sfrGiiohAxtieoYoJkv5MOYP4/2lPlOY+Cgw1Yoz+HHv31AllgFs\n"
 186     "BquBb/kJVmCCNsAOcnvQzTZUsW/TXz9G2nwRdqI1nSy2JvVjZGsqGQIDAQABo1Aw\n"
 187     "TjAdBgNVHQ4EFgQUt6pkzFt1PZlfYRL/HGnufF4frdwwHwYDVR0jBBgwFoAUt6pk\n"
 188     "zFt1PZlfYRL/HGnufF4frdwwDAYDVR0TBAUwAwEB/zANBgkqhkiG9w0BAQUFAAOC\n"
 189     "AQEAld7Qwq0cdzDQ51w1RVLwTR8Oy25PB3rzwEHcSGJmdqlMi3xOdaz80S1R1BBX\n"
 190     "ldvGBG5Tn0vT7xSuhmSgI2/HnBpy9ocHVOmhtNB4473NieEpfTYrnGXrFxu46Wus\n"
 191     "9m/ZnugcQ2G6C54A/NFtvgLmaC8uH8M7gKdS6uYUwJFQEofkjmd4UpOYSqmcRXhS\n"
 192     "Jzd5FYFWkJhKJYp3nlENSOD8CUFFVGekm05nFN2gRVc/qaqQkEX77+XYvhodLRsV\n"
 193     "qMn7nf7taidDKLO2T4bhujztnTYOhhaXKgPy7AtZ28N2wvX96VyAPB/vrchGmyBK\n"
 194     "kOg11TpPdNDkhb1J4ZCh2gupDg==\n"
 195     "-----END CERTIFICATE-----\n";
 196
 197 // TEST_IM_CA, signed by TEST_ROOT_CA, expires 2035
 198 std::string TEST_IM_CA =
 199     "-----BEGIN CERTIFICATE-----\n"
 200     "MIIDljCCAn6gAwIBAgICEAAwDQYJKoZIhvcNAQEFBQAwZjELMAkGA1UEBhMCQVUx\n"
 201     "EzARBgNVBAgMClNvbWUtU3RhdGUxDTALBgNVBAoMBEFDTUUxEDAOBgNVBAsMB1Rl\n"
 202     "c3RpbmcxITAfBgNVBAMMGFRlc3Qgcm9vdCBjYSBjZXJ0aWZpY2F0ZTAeFw0xNTAx\n"
 203     "MTYxNjQ1MzRaFw0zNTAxMTExNjQ1MzRaMGQxCzAJBgNVBAYTAkFVMRMwEQYDVQQI\n"
 204     "DApTb21lLVN0YXRlMQ0wCwYDVQQKDARBQ01FMRAwDgYDVQQLDAdUZXN0aW5nMR8w\n"
 205     "HQYDVQQDDBZUZXN0IElNIENBIGNlcnRpZmljYXRlMIIBIjANBgkqhkiG9w0BAQEF\n"
 206     "AAOCAQ8AMIIBCgKCAQEAzmBF78qClgoKfnLAncMXZwZ14TW+5kags1+QCYeg3c7j\n"
 207     "L9+RvDxIaX2tKf1sukJcwQfYqUlQkwt+58LMOb2ORtkpj8Or6WCWCZ0BzneT8ug7\n"
 208     "nxJT4m9+bohMF0JoKjjB2H4KNMHamLIwUxRKt6nyfk81kVhJOi2vzzxd+UCPi6Pc\n"
 209     "UAbJNH48eNgOIg55nyFovVzYj8GIo/9GvHJj83PPa/KlJZ+Z1qZASZZ/VYorplVT\n"
 210     "thsHXKfejhFy5YJ9t7n/vyAQsyBsagZsvX19xnH41fbYXHKf8UbXG23rNaZlchs6\n"
 211     "XJVLQdzOpj3WTj/lCocVHqLaZISLhNQ3aI7kUBUdiwIDAQABo1AwTjAdBgNVHQ4E\n"
 212     "FgQUoCYNaCBP4jl/3SYQuK8Ka+6i3QEwHwYDVR0jBBgwFoAUt6pkzFt1PZlfYRL/\n"
 213     "HGnufF4frdwwDAYDVR0TBAUwAwEB/zANBgkqhkiG9w0BAQUFAAOCAQEAjRzWiD97\n"
 214     "Htv4Kxpm3P+C+xP9AEteCJfO+7p8MWgtWEJOknJyt55zeKS2JwZIq57KcbqD8U7v\n"
 215     "vAUx1ymtUhlFPFd7J1mJ3pou+3aFYmGShYhGHpbrmUwjp7HVP588jrW1NoZVHdMc\n"
 216     "4OgJWFrViXeu9+maIcekjMB/+9Y0dUgQuK5ZuT5H/Jwet7Th/o9uufTUZjBzRvrB\n"
 217     "pbXgQpqgME2av4Q/6LuldPCTHLtWXgFUU2R+yCGmuGilvhFJnKoQryAbYnIQNWE8\n"
 218     "SLoHQ9s1i7Zyb7HU6UAaqMOz15LBkyAqtNyJcO2p7Q/p5YK0xfD4xisI5qXucqVm\n"
 219     "F2obL5qJSTN/RQ==\n"
 220     "-----END CERTIFICATE-----\n";
 221
 222 // TEST_LEAF, signed by TEST_IM_CA, expires 2035
 223 std::string TEST_LEAF =
 224     "-----BEGIN CERTIFICATE-----\n"
 225     "MIIDOzCCAiMCAQEwDQYJKoZIhvcNAQEFBQAwZDELMAkGA1UEBhMCQVUxEzARBgNV\n"
 226     "BAgMClNvbWUtU3RhdGUxDTALBgNVBAoMBEFDTUUxEDAOBgNVBAsMB1Rlc3Rpbmcx\n"
 227     "HzAdBgNVBAMMFlRlc3QgSU0gQ0EgY2VydGlmaWNhdGUwHhcNMTUwMTE2MTY0ODE0\n"
 228     "WhcNMzUwMTExMTY0ODE0WjBjMQswCQYDVQQGEwJBVTETMBEGA1UECAwKU29tZS1T\n"
 229     "dGF0ZTENMAsGA1UECgwEQUNNRTEQMA4GA1UECwwHVGVzdGluZzEeMBwGA1UEAwwV\n"
 230     "VGVzdCBsZWFmIGNlcnRpZmljYXRlMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIB\n"
 231     "CgKCAQEAzTdDIa2tDmRxFnIgiG+mBz8GoSVODs0ImNQGbqj+pLhBOFRH8fsah4Jl\n"
 232     "z5YF9KwhMVLknnHGFLE/Nb7Ac35kEzhMQMpTRxohW83oxw3eZ8zN/FBoKqg4qHRq\n"
 233     "QR8kS10YXTgrBR0ex/Vp+OUKEw6h7yL2r4Tpvrn9/qHwsxtLxqWbDIVf1O9b1Lfc\n"
 234     "bllYMdmV5E62yN5tcwrDP8gvHjFnVeLzrG8wTpc9FR90/0Jkfp5jAJcArOBLrT0E\n"
 235     "4VRqs+4HuwT8jAwFAmNnc7IYX5qSjtSWkmmHe73K/lzB+OiI0JEc/3eWUTWqwTSk\n"
 236     "4tNCiQGBKJ39LXPTBBJdzmxVH7CUDQIDAQABMA0GCSqGSIb3DQEBBQUAA4IBAQAp\n"
 237     "UdDOGu3hNiG+Vn10aQ6B1ZmOj3t+45gUV3sC+y8hB8EK1g4P5Ke9bVDts0T5eOnj\n"
 238     "CSc+6VoND5O4adI0IFFRFljHNVnvjeosHfUZNnowsmA2ptQBtC1g5ZKRvKXlkC5/\n"
 239     "i5BGgRqPFA7y9WB9Y05MrJHf3E+Oz/RBsLeeNiNN+rF5X1vYExvGHpo0M0zS0ze9\n"
 240     "HtC0aOy8ocsTrQkf3ceHTAXx2i8ftoSSD4klojtWFpWMrNQa52F7wB9nU6FfKRuF\n"
 241     "Zj/T1JkYXKkEwZU6nAR2jdZp3EP9xj3o15V/tyFcXHx6l8NTxn4cJb+Xe4VquQJz\n"
 242     "6ON7PVe0ABN/AlwVQiFE\n"
 243     "-----END CERTIFICATE-----\n";
 244
 245
 246
 247 struct FdCloser {
 248     void operator()(int* fd) {
 249         if(fd)
 250             close(*fd);
 251     }
 252 };
 253
 254 typedef std::unique_ptr<int, FdCloser> FdPtr;
 255
 256 void restoreFile(const string& filename) {
 257     string sourcePath = "/usr/share/ckm-db-test/" + filename;
 258     string targetPath = "/opt/data/ckm/" + filename;
 259
 260     int err, ret;
 261
 262     int sourceFd = TEMP_FAILURE_RETRY(open(sourcePath.c_str(), O_RDONLY));
 263     err = errno;
 264     BOOST_REQUIRE_MESSAGE(sourceFd > 0, "Opening " << sourcePath << " failed: " << strerror(err));
 265
 266     FdPtr sourceFdPtr(&sourceFd);
 267
 268     int targetFd = TEMP_FAILURE_RETRY(creat(targetPath.c_str(), 0644));
 269     err = errno;
 270     BOOST_REQUIRE_MESSAGE(targetFd > 0, "Creating " << targetPath << " failed: " << strerror(err));
 271
 272     ret = fchown(targetFd, CKM_UID, CKM_GID);
 273     err = errno;
 274     BOOST_REQUIRE_MESSAGE(ret != -1, "fchown() failed: " << strerror(err));
 275
 276     FdPtr targetFdPtr(&targetFd);
 277
 278     struct stat sourceStat;
 279     ret = fstat(sourceFd, &sourceStat);
 280     err = errno;
 281     BOOST_REQUIRE_MESSAGE(ret != -1, "fstat() failed: " << strerror(err));
 282
 283     ret = sendfile(targetFd, sourceFd, 0, sourceStat.st_size);
 284     err = errno;
 285     BOOST_REQUIRE_MESSAGE(ret != -1, "sendfile() failed: " << strerror(err));
 286
 287     ret = fsync(targetFd);
 288     err = errno;
 289     BOOST_REQUIRE_MESSAGE(ret != -1, "fsync() failed: " << strerror(err));
 290
 291     // TODO scoped close
 292     close(targetFd);
 293     close(sourceFd);
 294 }
 295
 296 void generateRandom(size_t random_bytes, unsigned char *output)
 297 {
 298     if(random_bytes<=0 || !output)
 299         throw runtime_error("Invalid param");
 300
 301     std::ifstream is("/dev/urandom", std::ifstream::binary);
 302     if(!is)
 303         throw runtime_error("Failed to read /dev/urandom");
 304     is.read(reinterpret_cast<char*>(output), random_bytes);
 305     if(static_cast<std::streamsize>(random_bytes) != is.gcount())
 306         throw runtime_error("Not enough bytes read from /dev/urandom");
 307 }
 308
 309 RawBuffer createRandomBuffer(size_t random_bytes)
 310 {
 311     RawBuffer buffer(random_bytes);
 312     generateRandom(buffer.size(), buffer.data());
 313     return buffer;
 314 }
 315 } // namespace anonymous
 316
 317
 318 SchemeTest::SchemeTest() : m_userChanged(false), m_directAccessEnabled(false) {
 319     m_control = Control::create();
 320     m_mgr = Manager::create();
 321     initOpenSsl();
 322
 323     SmackAccess sa;
 324     sa.add("System", LABEL, "rwx");
 325     sa.add(LABEL, "System", "rwx");
 326     sa.add(LABEL, "System::Run", "x");
 327     sa.apply();
 328 }
 329
 330 SchemeTest::~SchemeTest() {
 331     try {
 332         SwitchToRoot();
 333     } catch (...) {}
 334 }
 335
 336 void SchemeTest::RemoveUserData() {
 337     if(CKM_API_SUCCESS != m_control->lockUserKey(UID))
 338         throw runtime_error("lockUserKey failed");
 339
 340     if(CKM_API_SUCCESS != m_control->removeUserData(UID))
 341         throw runtime_error("removeUserData failed");
 342 }
 343
 344 void SchemeTest::SwitchToUser() {
 345     if (m_userChanged)
 346         return;
 347
 348     if(CKM_API_SUCCESS != m_control->unlockUserKey(UID, DBPASS))
 349         throw runtime_error("unlockUserKey failed");
 350
 351     // get calling label
 352     char* label = NULL;
 353     if (smack_new_label_from_self(&label) <= 0)
 354         throw runtime_error("smack_new_label_from_self failed");
 355
 356     m_origLabel = string(label);
 357     free(label);
 358
 359     if(0 > smack_set_label_for_self(LABEL))
 360         throw runtime_error("smack_set_label_for_self failed");
 361
 362     if(0 > setegid(GID))
 363         throw runtime_error("setegid failed");
 364
 365     if(0 > seteuid(UID))
 366         throw runtime_error("seteuid failed");
 367
 368     m_userChanged = true;
 369 }
 370
 371 void SchemeTest::SwitchToRoot() {
 372     if (!m_userChanged)
 373         return;
 374
 375     if(0 > seteuid(0))
 376         throw runtime_error("seteuid failed");
 377     if(0 > setegid(0))
 378         throw runtime_error("setegid failed");
 379
 380     if(0 > smack_set_label_for_self(m_origLabel.c_str()))
 381         throw runtime_error("smack_set_label_for_self failed");
 382
 383     if(m_control->lockUserKey(UID) != CKM_API_SUCCESS)
 384         throw runtime_error("lockUserKey failed");
 385 }
 386
 387 void SchemeTest::FillDb() {
 388     // pkcs
 389     ifstream is("/usr/share/ckm-db-test/encryption-scheme.p12");
 390     if(!is)
 391         throw runtime_error("Failed to read pkcs");
 392     istreambuf_iterator<char> begin(is), end;
 393     RawBuffer pkcsBuffer(begin, end);
 394     auto pkcs = PKCS12::create(pkcsBuffer, Password());
 395     if(pkcs->empty())
 396         throw runtime_error("Empty pkcs");
 397
 398     SwitchToUser();
 399
 400     // certificates
 401     RawBuffer rootCaBuffer(TEST_ROOT_CA.begin(), TEST_ROOT_CA.end());
 402     CertificateShPtr rootCa = CKM::Certificate::create(rootCaBuffer, CKM::DataFormat::FORM_PEM);
 403     RawBuffer imCaBuffer(TEST_IM_CA.begin(), TEST_IM_CA.end());
 404     CertificateShPtr imCa = CKM::Certificate::create(imCaBuffer, CKM::DataFormat::FORM_PEM);
 405     RawBuffer leafBuffer(TEST_LEAF.begin(), TEST_LEAF.end());
 406     CertificateShPtr leaf = CKM::Certificate::create(leafBuffer, CKM::DataFormat::FORM_PEM);
 407
 408     for(const auto& g:GROUPS) {
 409         switch (g.type) {
 410         case Group::KEY_PAIR_RSA:
 411             if(g.items.size() != 2)
 412                 throw runtime_error("Wrong number of keys");
 413             if( g.items[0].type != DataType::KEY_RSA_PRIVATE ||
 414                 g.items[1].type != DataType::KEY_RSA_PUBLIC)
 415                 throw runtime_error("Invalid item type");
 416
 417             if(CKM_API_SUCCESS != m_mgr->createKeyPairRSA(1024,
 418                                                           g.items[0].alias,
 419                                                           g.items[1].alias,
 420                                                           g.items[0].policy,
 421                                                           g.items[1].policy))
 422                 throw runtime_error("createKeyPair failed");
 423             break;
 424
 425         case Group::CERT_CHAIN:
 426             if(g.items.size() != CHAIN_SIZE)
 427                 throw runtime_error("Wrong number of certificates");
 428             if( g.items[0].type != DataType::CERTIFICATE ||
 429                 g.items[1].type != DataType::CERTIFICATE ||
 430                 g.items[2].type != DataType::CERTIFICATE)
 431                 throw runtime_error("Invalid item type");
 432
 433             if(CKM_API_SUCCESS != m_mgr->saveCertificate(g.items[0].alias, rootCa, g.items[0].policy))
 434                 throw runtime_error("saveCertificate failed");
 435             if(CKM_API_SUCCESS != m_mgr->saveCertificate(g.items[1].alias, imCa, g.items[1].policy))
 436                 throw runtime_error("saveCertificate failed");
 437             if(CKM_API_SUCCESS != m_mgr->saveCertificate(g.items[2].alias, leaf, g.items[2].policy))
 438                 throw runtime_error("saveCertificate failed");
 439             break;
 440
 441         default:
 442             for(const auto& i:g.items) {
 443                 switch (i.type) {
 444                 case DataType::BINARY_DATA:
 445                     if(CKM_API_SUCCESS != m_mgr->saveData(i.alias, TEST_DATA, i.policy))
 446                         throw runtime_error("saveData failed");
 447                     break;
 448
 449                 case DataType::KEY_AES:
 450                     if(CKM_API_SUCCESS != m_mgr->createKeyAES(256, i.alias, i.policy))
 451                         throw runtime_error("createKeyAES failed");
 452                     break;
 453
 454                 case DataType::CHAIN_CERT_0:    // PKCS
 455                     if(CKM_API_SUCCESS != m_mgr->savePKCS12(i.alias, pkcs, i.policy, i.policy))
 456                         throw runtime_error("savePkcs12 failed");
 457                     break;
 458
 459                 default:
 460                     throw runtime_error("unsupported data type");
 461                 }
 462             }
 463             break;
 464         }
 465     }
 466 }
 467
 468 void SchemeTest::ReadAll(bool useWrongPass) {
 469     SwitchToUser();
 470
 471     for(const auto& g:GROUPS) {
 472         for(const auto& i:g.items) {
 473             int ret;
 474             Password pass = i.policy.password;
 475             if(useWrongPass) {
 476                 if(pass.empty())
 477                     pass = TEST_PASS;
 478                 else
 479                     pass = Password();
 480             }
 481
 482             switch (i.type) {
 483             case DataType::BINARY_DATA:
 484             {
 485                 RawBuffer receivedData;
 486                 ret = m_mgr->getData(i.alias, pass, receivedData);
 487                 BOOST_REQUIRE_MESSAGE(useWrongPass || receivedData == TEST_DATA,
 488                                       "Received data is different for " << i.alias);
 489                 break;
 490             }
 491
 492             case DataType::KEY_AES:
 493             case DataType::KEY_RSA_PRIVATE:
 494             case DataType::KEY_RSA_PUBLIC:
 495             {
 496                 KeyShPtr receivedKey;
 497                 ret = m_mgr->getKey(i.alias, pass, receivedKey);
 498                 break;
 499             }
 500
 501             case DataType::CERTIFICATE:
 502             {
 503                 CertificateShPtr receivedCert;
 504                 ret = m_mgr->getCertificate(i.alias, pass, receivedCert);
 505                 break;
 506             }
 507
 508             case DataType::CHAIN_CERT_0: // pkcs
 509             {
 510                 PKCS12ShPtr pkcs;
 511                 ret = m_mgr->getPKCS12(i.alias, pass, pass, pkcs);
 512                 break;
 513             }
 514
 515             default:
 516                 BOOST_FAIL("Unsupported data type " << i.type);
 517             }
 518
 519             if(i.policy.extractable) {
 520                 if(useWrongPass)
 521                     BOOST_REQUIRE_MESSAGE(ret == CKM_API_ERROR_AUTHENTICATION_FAILED,
 522                                           "Reading item " << i.alias << " should fail with " <<
 523                                           CKM_API_ERROR_AUTHENTICATION_FAILED << " got: " << ret);
 524                 else
 525                     BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS, "Reading item " << i.alias <<
 526                                           " failed with " << ret);
 527             }
 528             else
 529                 BOOST_REQUIRE_MESSAGE(ret == CKM_API_ERROR_NOT_EXPORTABLE, "Item " << i.alias <<
 530                                       " should not be exportable");
 531         }
 532     }
 533 }
 534
 535 void SchemeTest::SignVerify() {
 536     SwitchToUser();
 537
 538     for(const auto& g:GROUPS) {
 539         if(g.type == Group::KEY_PAIR_RSA) {
 540             BOOST_REQUIRE_MESSAGE(g.items.size() == 2, "Wrong number of keys");
 541             BOOST_REQUIRE_MESSAGE(g.items[0].type == DataType::KEY_RSA_PRIVATE &&
 542                                   g.items[1].type == DataType::KEY_RSA_PUBLIC, "Wrong key");
 543
 544             SignVerifyItem(g.items[0], g.items[1]);
 545         } else {
 546             for(const auto& i:g.items) {
 547                 switch (i.type) {
 548                 case DataType::CHAIN_CERT_0:
 549                     SignVerifyItem(i, i);
 550                     break;
 551
 552                 default:
 553                     break;
 554                 }
 555             }
 556         }
 557     }
 558 }
 559
 560 void SchemeTest::EncryptDecrypt() {
 561     SwitchToUser();
 562
 563     for(const auto& g:GROUPS) {
 564         if(g.type == Group::KEY_PAIR_RSA) {
 565             BOOST_REQUIRE_MESSAGE(g.items.size() == 2, "Wrong number of keys");
 566             BOOST_REQUIRE_MESSAGE(g.items[0].type == DataType::KEY_RSA_PRIVATE &&
 567                                   g.items[1].type == DataType::KEY_RSA_PUBLIC, "Wrong key");
 568
 569             EncryptDecryptItem(g.items[0], g.items[1]);
 570         } else {
 571             for(const auto& i:g.items) {
 572                 switch (i.type) {
 573                 case DataType::KEY_AES:
 574                     EncryptDecryptItem(i);
 575                     break;
 576
 577                 case DataType::CHAIN_CERT_0:
 578                     EncryptDecryptItem(i, i);
 579                     break;
 580
 581                 default:
 582                     break;
 583                 }
 584             }
 585         }
 586     }
 587 }
 588
 589 void SchemeTest::CreateChain() {
 590     SwitchToUser();
 591
 592     for(const auto& g:GROUPS) {
 593         if(g.type == Group::CERT_CHAIN) {
 594             BOOST_REQUIRE_MESSAGE(g.items.size() == CHAIN_SIZE, "Not enough certificates");
 595             for(const auto& c:g.items)
 596                 BOOST_REQUIRE_MESSAGE(c.type == DataType::CERTIFICATE, "Wrong item type");
 597             Items trusted(CHAIN_SIZE-1);
 598             std::copy(g.items.begin(), g.items.begin() + CHAIN_SIZE-1, trusted.begin());
 599
 600             // last one is ee (leaf)
 601             CreateChainItem(g.items.back(), trusted);
 602         } else {
 603             for(const auto& i:g.items) {
 604                 if(i.type == DataType::CHAIN_CERT_0) // PKCS
 605                     CreateChainItem(i, { i });
 606             }
 607         }
 608     }
 609 }
 610
 611 void SchemeTest::RemoveAll() {
 612     SwitchToUser();
 613
 614     for(const auto& g:GROUPS) {
 615         for(const auto& i:g.items) {
 616             int ret = m_mgr->removeAlias(i.alias);
 617             BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
 618                                   "removeAlias() failed with " << ret << " for " << i.alias);
 619         }
 620     }
 621 }
 622 size_t SchemeTest::CountObjects() {
 623     EnableDirectDbAccess();
 624
 625     size_t ret = 0;
 626     for(const auto& g:GROUPS) {
 627         for(const auto& i:g.items) {
 628             DB::RowVector rows;
 629             // it is assumed that aliases are different
 630             m_db->getRows(i.alias, DB_LABEL, DataType::DB_FIRST, DataType::DB_LAST, rows);
 631             ret += rows.size();
 632         }
 633     }
 634     return ret;
 635 }
 636
 637 void SchemeTest::RestoreDb() {
 638     restoreFile("key-7654");
 639     restoreFile("db-key-7654");
 640     restoreFile("db-7654");
 641     m_db.reset();
 642     m_directAccessEnabled = false;
 643 }
 644
 645 void SchemeTest::CheckSchemeVersion(const ItemFilter& filter, int version) {
 646     EnableDirectDbAccess();
 647
 648     for(const auto& g:GROUPS) {
 649         for(const auto& i:g.items) {
 650             if(!filter.Matches(i))
 651                 continue;
 652
 653             DB::RowVector rows;
 654             m_db->getRows(i.alias, DB_LABEL, filter.typeFrom, filter.typeTo, rows);
 655             BOOST_REQUIRE_MESSAGE(rows.size() > 0, "No rows found for " << i.alias);
 656             for(const auto& r : rows) {
 657                 BOOST_REQUIRE_MESSAGE(
 658                         (r.encryptionScheme >> ENC_SCHEME_OFFSET) == version,
 659                         "Wrong encryption scheme for " << i.alias << ". Expected " << version <<
 660                         " got: " << (r.encryptionScheme >> ENC_SCHEME_OFFSET));
 661             }
 662         }
 663     }
 664 }
 665
 666 void SchemeTest::EnableDirectDbAccess() {
 667     SwitchToRoot();
 668
 669     if(m_directAccessEnabled)
 670         return;
 671
 672     // direct access to db
 673     FileSystem fs(UID);
 674     auto wrappedDKEK = fs.getDKEK();
 675     auto keyProvider = KeyProvider(wrappedDKEK, DBPASS);
 676
 677     auto wrappedDatabaseDEK = fs.getDBDEK();
 678     RawBuffer key = keyProvider.getPureDEK(wrappedDatabaseDEK);
 679
 680     m_db.reset(new DB::Crypto(fs.getDBPath(), key));
 681     m_directAccessEnabled = true;
 682 }
 683
 684 void SchemeTest::SignVerifyItem(const Item& itemPrv, const Item& itemPub) {
 685     int ret;
 686     KeyShPtr receivedKey;
 687     RawBuffer signature;
 688     // create/verify signature
 689     ret = m_mgr->createSignature(itemPrv.alias,
 690                                  itemPrv.policy.password,
 691                                  TEST_DATA,
 692                                  HashAlgorithm::SHA512,
 693                                  RSAPaddingAlgorithm::X931,
 694                                  signature);
 695     BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS, "createSignature() failed with " << ret <<
 696                           " for " << itemPrv.alias);
 697     ret = m_mgr->verifySignature(itemPub.alias,
 698                                  itemPub.policy.password,
 699                                  TEST_DATA,
 700                                  signature,
 701                                  HashAlgorithm::SHA512,
 702                                  RSAPaddingAlgorithm::X931);
 703     BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS, "verifySignature() failed with " << ret <<
 704                           " for " << itemPub.alias);
 705
 706 }
 707
 708 void SchemeTest::EncryptDecryptItem(const Item& item) {
 709     CryptoAlgorithm algo;
 710     RawBuffer iv = createRandomBuffer(IV_LEN);
 711     RawBuffer encrypted, decrypted;
 712     int ret;
 713
 714     algo.setParam(ParamName::ALGO_TYPE, AlgoType::AES_GCM);
 715     algo.setParam(ParamName::ED_IV, iv);
 716
 717     ret = m_mgr->encrypt(algo, item.alias, item.policy.password, TEST_DATA, encrypted);
 718     BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS, "encrypt() failed iwth " << ret << " for " <<
 719                           item.alias);
 720
 721     ret = m_mgr->decrypt(algo, item.alias, item.policy.password, encrypted, decrypted);
 722     BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS, "decrypt() failed iwth " << ret << " for " <<
 723                           item.alias);
 724
 725     BOOST_REQUIRE_MESSAGE(decrypted == TEST_DATA, "Decrypted data not equal to original");
 726 }
 727
 728 void SchemeTest::EncryptDecryptItem(const Item& itemPrv, const Item& itemPub) {
 729     CryptoAlgorithm algo;
 730     RawBuffer encrypted, decrypted;
 731     int ret;
 732
 733     algo.setParam(ParamName::ALGO_TYPE, AlgoType::RSA_OAEP);
 734
 735     ret = m_mgr->encrypt(algo, itemPub.alias, itemPub.policy.password, TEST_DATA, encrypted);
 736     BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS, "encrypt() failed iwth " << ret << " for " <<
 737                           itemPub.alias);
 738
 739     ret = m_mgr->decrypt(algo, itemPrv.alias, itemPrv.policy.password, encrypted, decrypted);
 740     BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS, "decrypt() failed iwth " << ret << " for " <<
 741                           itemPrv.alias);
 742
 743     BOOST_REQUIRE_MESSAGE(decrypted == TEST_DATA, "Decrypted data not equal to original");
 744 }
 745
 746 void SchemeTest::CreateChainItem(const Item& leaf, const Items& certs) {
 747     CertificateShPtrVector chain;
 748     AliasVector trusted;
 749
 750     if(!leaf.policy.extractable || !leaf.policy.password.empty())
 751         return;
 752
 753     for(const auto& i : certs) {
 754         if(!i.policy.extractable || !i.policy.password.empty())
 755             return;
 756         trusted.push_back(i.alias);
 757     }
 758
 759     CertificateShPtr leafCrt;
 760     int ret = m_mgr->getCertificate(leaf.alias, leaf.policy.password, leafCrt);
 761     BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
 762                           "getCertificate failed with " << ret << " for " <<
 763                           leaf.alias);
 764
 765     ret = m_mgr->getCertificateChain(leafCrt, AliasVector(), trusted, false, chain);
 766     BOOST_REQUIRE_MESSAGE(ret == CKM_API_SUCCESS,
 767                           "getCertificateChain() failed with " << ret);
 768     BOOST_REQUIRE_MESSAGE(chain.size() == CHAIN_LEN, "Wrong chain length: " << chain.size());
 769 }