src/ckm/ckm-common.h

   1 /*
   2  *  Copyright (c) 2000 - 2019 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       ckm-common.h
  18  * @author     Krzysztof Jackiewicz (k.jackiewicz@samsung.com)
  19  * @version    1.0
  20  */
  21
  22 #pragma once
  23
  24 #include <string>
  25 #include <memory>
  26 #include <stdexcept>
  27 #include <ckm/ckm-type.h>
  28 #include <ckm/ckm-manager-async.h>
  29 #include <ckmc/ckmc-type.h>
  30 #include <ckmc/ckmc-error.h>
  31 #include <tests_common.h>
  32 #include <sys/types.h>
  33
  34 extern const std::string SMACK_USER_APP_PREFIX;
  35 extern const char *SYSTEM_LABEL;
  36 extern const char *TEST_LABEL;
  37 extern const char *TEST_LABEL_2;
  38 extern const char *TEST_LABEL_3;
  39 extern const char *TEST_LABEL_4;
  40 extern const char *TEST_LABEL_5;
  41
  42 // support for error printing
  43 const char * CKMCErrorToString(int error);
  44 std::string CKMCReadableError(int error);
  45
  46 // RUNNER_ASSERT wrappers
  47 template <typename F, typename... Args>
  48 void assert_result(int expected, F&& func, Args... args)
  49 {
  50     int ret = func(args...);
  51     RUNNER_ASSERT_MSG(ret == expected,
  52                       "Expected: " << CKMCErrorToString(expected) << "(" << expected << ")"
  53                       " got: " << CKMCErrorToString(ret) << "(" << ret << ")");
  54 }
  55
  56 template <typename F, typename... Args>
  57 void assert_positive(F&& func, Args... args)
  58 {
  59     assert_result(CKMC_ERROR_NONE, std::move(func), args...);
  60 }
  61
  62 template <typename F, typename... Args>
  63 void assert_invalid_param(F&& func, Args... args)
  64 {
  65     assert_result(CKMC_ERROR_INVALID_PARAMETER, std::move(func), args...);
  66 }
  67
  68
  69 // list operations
  70 template <typename T>
  71 size_t list_size(const T* list)
  72 {
  73     size_t size = 0;
  74     while(list) {
  75         list = list->next;
  76         size++;
  77     }
  78     return size;
  79 }
  80
  81 // scoped free
  82 typedef std::unique_ptr<char, void (*)(void *)> CharPtr;
  83
  84 std::string getLabel();
  85 // returns process owner id
  86 std::string getOwnerIdFromSelf();
  87
  88 std::string aliasWithLabel(const char *label, const char *alias);
  89
  90 std::string aliasWithLabelFromSelf(const char *alias);
  91
  92 void save_data(const char* alias, const char *data, size_t len, const char* password,
  93                int expected_err = CKMC_ERROR_NONE, bool exportable = true);
  94 void save_data(const char* alias, const char *data, int expected_err = CKMC_ERROR_NONE,
  95                bool exportable = true);
  96 void save_data(const char* alias, const char *data, size_t len,
  97                int expected_err = CKMC_ERROR_NONE, bool exportable = true);
  98 class ScopedSaveData
  99 {
 100 public:
 101     ScopedSaveData(const char* alias, const char *data, int expected_err = CKMC_ERROR_NONE);
 102     virtual ~ScopedSaveData();
 103
 104 private:
 105     std::string m_alias;
 106 };
 107
 108 class ScopedDBUnlock
 109 {
 110 public:
 111     ScopedDBUnlock(uid_t user_id, const char* passwd);
 112     virtual ~ScopedDBUnlock();
 113
 114 private:
 115     uid_t m_uid;
 116 };
 117
 118 void check_remove_allowed(const char* alias);
 119 void check_remove_denied(const char* alias);
 120 void check_remove_not_visible(const char* alias);
 121 void check_read(const char* alias,
 122                 const char *label,
 123                 const char *test_data,
 124                 size_t len,
 125                 int expected_code = CKMC_ERROR_NONE);
 126 void check_read(const char* alias,
 127                 const char *label,
 128                 const char *test_data,
 129                 int expected_code = CKMC_ERROR_NONE);
 130 void check_read_allowed(const char* alias, const char *data);
 131 void check_read_not_visible(const char* alias);
 132 void check_key(const char *alias,
 133                int expected_error = CKMC_ERROR_NONE,
 134                ckmc_key_type_e expected_type = CKMC_KEY_NONE);
 135 void check_key_allowed(const char *alias, ckmc_key_type_e expected_type = CKMC_KEY_NONE);
 136 void check_key_not_visible(const char *alias);
 137 void check_cert_allowed(const char *alias);
 138 void check_cert_not_visible(const char *alias);
 139 void allow_access(const char* alias, const char* accessor, int permissionMask);
 140 void allow_access_negative(const char* alias, const char* accessor, int permissionMask, int expectedCode);
 141 void deny_access(const char* alias, const char* accessor);
 142 void deny_access_negative(const char* alias, const char* accessor, int expectedCode);
 143
 144 void unlock_user_data(uid_t user_id, const char *passwd);
 145 void remove_user_data(uid_t user_id);
 146 void reset_user_data(uid_t user_id, const char *passwd);
 147
 148 ckmc_raw_buffer_s prepare_message_buffer(const char * input);
 149 void check_alias_list(const CKM::AliasVector& expected);
 150 void check_alias_info_list_helper(const CKM::AliasPwdVector& expected, const CKM::AliasPwdVector& actual,
 151                                   const std::string userSmackLabel = "");
 152 void check_alias_info_list(const CKM::AliasPwdVector& expected,
 153                            const std::string& userSmackLabel = "");
 154
 155 typedef enum {
 156     ALIAS_KEY,
 157     ALIAS_CERT,
 158     ALIAS_DATA
 159 } alias_type_;
 160 size_t count_aliases(alias_type_ type, size_t minimum_initial_element_count = 0);
 161 std::string sharedDatabase(const CKM::Alias & alias);
 162 CKM::RawBuffer createRandomBuffer(size_t random_bytes);
 163 ckmc_raw_buffer_s* createRandomBufferCAPI(size_t random_bytes);
 164
 165 ckmc_key_s *generate_AES_key(size_t lengthBits, const char *passwd);
 166 void validate_AES_key(ckmc_key_s *analyzed);
 167 void compare_AES_keys(ckmc_key_s *first, ckmc_key_s *second); // true if equal
 168 CKM::Policy generate_ckm_policy(int iterator_nr); // generates policy based on given number
 169
 170 // Test env class for database cleanup. Pass database uids to cleanup before and after test
 171 template <uid_t ...Args>
 172 class RemoveDataEnv;
 173
 174 template <>
 175 class RemoveDataEnv<>
 176 {
 177 public:
 178     void init(const std::string&)
 179     {}
 180     void finish()
 181     {}
 182 };
 183
 184 template <uid_t UID, uid_t ...Args>
 185 class RemoveDataEnv<UID, Args...> : public RemoveDataEnv<Args...>
 186 {
 187 public:
 188     void init(const std::string & str) {
 189         remove_user_data(UID);
 190         RemoveDataEnv<Args...>::init(str);
 191     }
 192     void finish() {
 193         RemoveDataEnv<Args...>::finish();
 194         remove_user_data(UID);
 195     }
 196 };
 197
 198 typedef std::shared_ptr<ckmc_raw_buffer_s> RawBufferPtr;
 199 typedef std::shared_ptr<struct __ckmc_param_list> ParamListPtr;
 200
 201 ParamListPtr createParamListPtr();
 202
 203 void assert_buffers_equal(const ckmc_raw_buffer_s b1, const ckmc_raw_buffer_s b2, bool equal=true);
 204
 205 RawBufferPtr create_raw_buffer(ckmc_raw_buffer_s* buffer);
 206
 207
 208 template <typename F, typename... Args>
 209 void test_no_observer(F&& func, Args... args)
 210 {
 211     CKM::ManagerAsync::ObserverPtr obs;
 212     CKM::ManagerAsync mgr;
 213
 214     try {
 215         (mgr.*func)(obs, args...);
 216         RUNNER_ASSERT_MSG(false, "function() should have thrown an exception");
 217     } catch (const std::invalid_argument& e) {
 218         RUNNER_ASSERT(true);
 219     } catch (...) {
 220         RUNNER_ASSERT_MSG(false, "Unexpected exception");
 221     }
 222 }