tests/ckm/ckm-common.cpp

   1 /*
   2  *  Copyright (c) 2000 - 2014 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.cpp
  18  * @author     Krzysztof Jackiewicz (k.jackiewicz@samsung.com)
  19  * @version    1.0
  20  */
  21
  22 #include <sys/smack.h>
  23 #include <ckm-common.h>
  24 #include <tests_common.h>
  25 #include <access_provider2.h>
  26 #include <ckm/ckm-manager.h>
  27
  28 void switch_to_storage_user(const char* label)
  29 {
  30     AccessProvider ap(label);
  31     ap.allowAPI("key-manager::api-storage", "rw");
  32     ap.applyAndSwithToUser(APP_UID, APP_GID);
  33 }
  34
  35 void switch_to_storage_ocsp_user(const char* label)
  36 {
  37     AccessProvider ap(label);
  38     ap.allowAPI("key-manager::api-storage", "rw");
  39     ap.allowAPI("key-manager::api-ocsp", "rw");
  40     ap.applyAndSwithToUser(APP_UID, APP_GID);
  41 }
  42
  43 DBCleanup::~DBCleanup()
  44 {
  45     // Let it throw. If db can't be cleared further tests are unreliable
  46     CKM::ManagerShPtr mgr = CKM::Manager::create();
  47     for(const auto& it:m_keys)
  48         mgr->removeKey(it);
  49     m_keys.clear();
  50     for(const auto& it:m_certs)
  51         mgr->removeCertificate(it);
  52     m_certs.clear();
  53     for(const auto& it:m_data)
  54         mgr->removeData(it);
  55     m_data.clear();
  56 }
  57
  58 // returns process label
  59 CharPtr get_label()
  60 {
  61     int ret;
  62     char* my_label = NULL;
  63     RUNNER_ASSERT_MSG(0 <= (ret = smack_new_label_from_self(&my_label)),
  64                          "Failed to get smack label for self. Error: " << ret);
  65
  66     return CharPtr(my_label, free);
  67 }
  68
  69 // changes process label
  70 void change_label(const char* label)
  71 {
  72     int ret = smack_set_label_for_self(label);
  73     RUNNER_ASSERT_MSG(0 == ret, "Error in smack_set_label_for_self. Error: " << ret);
  74 }
  75
  76 ScopedLabel::ScopedLabel(const char* label) : m_original_label(get_label())
  77 {
  78     change_label(label);
  79 }
  80
  81 ScopedLabel::~ScopedLabel()
  82 {
  83     /*
  84      * Let it throw. If we can't restore label then remaining tests results will be
  85      * unreliable anyway.
  86      */
  87     change_label(m_original_label.get());
  88 }