modules/security_origin_dao/dao/security_origin_dao.cpp

   1 /*
   2  * Copyright (c) 2011 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    security_origin_dao.cpp
  18  * @author  Jihoon Chung (jihoon.chung@samsung.com)
  19  * @version 1.0
  20  * @brief    This file contains the definition of security origin dao class.
  21  */
  22
  23 #include <wrt-commons/security-origin-dao/security_origin_database.h>
  24 #include <wrt-commons/security-origin-dao/security_origin_dao.h>
  25 #include <wrt-commons/security-origin-dao/security_origin_dao_types.h>
  26 #include <orm_generator_security_origin.h>
  27 #include <dpl/foreach.h>
  28 #include <dpl/wrt-dao-ro/widget_dao_read_only.h>
  29 #include <dpl/wrt-dao-ro/WrtDatabase.h>
  30 #include <dpl/wrt-dao-ro/widget_config.h>
  31 #include <dpl/wrt-dao-ro/global_config.h>
  32 #include <sys/stat.h>
  33 #include <fstream>
  34
  35 using namespace DPL::DB::ORM;
  36 using namespace DPL::DB::ORM::security_origin;
  37
  38 namespace SecurityOriginDB {
  39
  40 #define SQL_CONNECTION_EXCEPTION_HANDLER_BEGIN          Try
  41
  42 #define SQL_CONNECTION_EXCEPTION_HANDLER_END(message)   \
  43     Catch(DPL::DB::SqlConnection::Exception::Base) {       \
  44         LogError(message);                              \
  45         ReThrowMsg(SecurityOriginDAO::Exception::DatabaseError, \
  46                    message);                            \
  47     }
  48
  49 namespace {
  50 DPL::DB::SqlConnection::Flag::Option SECURITY_ORIGIN_DB_OPTION =
  51     DPL::DB::SqlConnection::Flag::RW;
  52 DPL::DB::SqlConnection::Flag::Type SECURITY_ORIGIN_DB_TYPE =
  53     DPL::DB::SqlConnection::Flag::UseLucene;
  54 const char* const SECURITY_ORIGIN_DB_NAME = ".security_origin.db";
  55 const char* const SECURITY_ORIGIN_DB_SQL_PATH =
  56     "/usr/share/wrt-engine/security_origin_db.sql";
  57
  58 std::string createDatabasePath(int widgetHandle)
  59 {
  60     using namespace DPL::DB::ORM;
  61     using namespace WrtDB::WidgetConfig;
  62     using namespace WrtDB::GlobalConfig;
  63
  64     SQL_CONNECTION_EXCEPTION_HANDLER_BEGIN
  65     {
  66         std::stringstream filename;
  67         WrtDB::WidgetDAOReadOnly widgetDAO(widgetHandle);
  68         DPL::Optional<DPL::String> pkgname = widgetDAO.getPkgname();
  69
  70         filename << GetWidgetPersistentStoragePath(*pkgname)
  71                  << "/"
  72                  << SECURITY_ORIGIN_DB_NAME;
  73         return filename.str();
  74     }
  75     SQL_CONNECTION_EXCEPTION_HANDLER_END("Fail to get database Path")
  76
  77 }
  78
  79 void checkDatabase(std::string databasePath)
  80 {
  81     SQL_CONNECTION_EXCEPTION_HANDLER_BEGIN
  82     {
  83         if (databasePath.empty()) {
  84             ThrowMsg(SecurityOriginDAO::Exception::DatabaseError,
  85                      "Wrong database Path is passed");
  86         }
  87
  88         struct stat buffer;
  89         if (stat(databasePath.c_str(), &buffer) != 0) {
  90
  91             //Create fresh database
  92             LogInfo("Creating database " << databasePath);
  93
  94             std::fstream file;
  95             file.open(SECURITY_ORIGIN_DB_SQL_PATH, std::ios_base::in);
  96             if (!file) {
  97                 ThrowMsg(SecurityOriginDAO::Exception::DatabaseError,
  98                          "Fail to get database Path");
  99             }
 100
 101             std::stringstream ssBuffer;
 102             ssBuffer << file.rdbuf();
 103
 104             file.close();
 105
 106             DPL::DB::SqlConnection con(databasePath,
 107                                        SECURITY_ORIGIN_DB_TYPE,
 108                                        SECURITY_ORIGIN_DB_OPTION);
 109             con.ExecCommand(ssBuffer.str().c_str());
 110         }
 111     }
 112     SQL_CONNECTION_EXCEPTION_HANDLER_END("Fail to get database Path")
 113 }
 114 }
 115
 116 SecurityOriginDAO::SecurityOriginDAO(int handle) :
 117     m_securityOriginDBPath(createDatabasePath(handle)),
 118     m_securityOriginDBInterface(m_securityOriginDBPath, SECURITY_ORIGIN_DB_TYPE)
 119 {
 120     checkDatabase(m_securityOriginDBPath);
 121     m_securityOriginDBInterface.AttachToThread(SECURITY_ORIGIN_DB_OPTION);
 122 }
 123
 124 SecurityOriginDAO::~SecurityOriginDAO()
 125 {
 126     m_securityOriginDBInterface.DetachFromThread();
 127 }
 128
 129 SecurityOriginDataList SecurityOriginDAO::getSecurityOriginDataList(void)
 130 {
 131     SQL_CONNECTION_EXCEPTION_HANDLER_BEGIN
 132     {
 133         SecurityOriginDataList list;
 134         SECURITY_ORIGIN_DB_SELECT(select, SecurityOriginInfo, &m_securityOriginDBInterface);
 135         typedef std::list<SecurityOriginInfo::Row> RowList;
 136         RowList rowList = select->GetRowList();
 137
 138         FOREACH(it, rowList) {
 139             Origin origin(it->Get_scheme(), it->Get_host(), it->Get_port());
 140             list.push_back(
 141                 SecurityOriginDataPtr(
 142                     new SecurityOriginData(
 143                         static_cast<Feature>(it->Get_feature()), origin)));
 144         }
 145         return list;
 146     }
 147     SQL_CONNECTION_EXCEPTION_HANDLER_END("Failed to get data  list")
 148 }
 149
 150 Result SecurityOriginDAO::getResult(const SecurityOriginData &securityOriginData)
 151 {
 152     SQL_CONNECTION_EXCEPTION_HANDLER_BEGIN
 153     {
 154         SECURITY_ORIGIN_DB_SELECT(select, SecurityOriginInfo, &m_securityOriginDBInterface);
 155         select->Where(
 156             And(And(And(Equals<SecurityOriginInfo::feature>(securityOriginData.feature),
 157                         Equals<SecurityOriginInfo::scheme>(securityOriginData.origin.scheme)),
 158                     Equals<SecurityOriginInfo::host>(securityOriginData.origin.host)),
 159                 Equals<SecurityOriginInfo::port>(securityOriginData.origin.port)));
 160         SecurityOriginInfo::Select::RowList rows = select->GetRowList();
 161
 162         if (rows.empty()) {
 163             return RESULT_UNKNOWN;
 164         }
 165         SecurityOriginInfo::Row row = rows.front();
 166         return static_cast<Result>(row.Get_result());
 167     }
 168     SQL_CONNECTION_EXCEPTION_HANDLER_END("Failed to get result for security origin")
 169 }
 170
 171 void SecurityOriginDAO::setSecurityOriginData(const SecurityOriginData &securityOriginData,
 172                                               const Result result)
 173 {
 174     SQL_CONNECTION_EXCEPTION_HANDLER_BEGIN
 175     {
 176         ScopedTransaction transaction(&m_securityOriginDBInterface);
 177         SecurityOriginInfo::Row row;
 178         row.Set_feature(securityOriginData.feature);
 179         row.Set_scheme(securityOriginData.origin.scheme);
 180         row.Set_host(securityOriginData.origin.host);
 181         row.Set_port(securityOriginData.origin.port);
 182         row.Set_result(result);
 183
 184         if (true == hasResult(securityOriginData)) {
 185             SECURITY_ORIGIN_DB_UPDATE(update,
 186                                SecurityOriginInfo,
 187                                &m_securityOriginDBInterface);
 188             update->Values(row);
 189             update->Execute();
 190         } else {
 191             SECURITY_ORIGIN_DB_INSERT(
 192                 insert,
 193                 SecurityOriginInfo,
 194                 &m_securityOriginDBInterface);
 195             insert->Values(row);
 196             insert->Execute();
 197         }
 198         transaction.Commit();
 199     }
 200     SQL_CONNECTION_EXCEPTION_HANDLER_END("Fail to set security origin data")
 201 }
 202
 203 void SecurityOriginDAO::removeSecurityOriginData(
 204     const SecurityOriginData &securityOriginData)
 205 {
 206     SQL_CONNECTION_EXCEPTION_HANDLER_BEGIN
 207     {
 208         ScopedTransaction transaction(&m_securityOriginDBInterface);
 209
 210         if (true == hasResult(securityOriginData)) {
 211             SECURITY_ORIGIN_DB_DELETE(del, SecurityOriginInfo, &m_securityOriginDBInterface)
 212             del->Where(
 213                 And(And(And(Equals<SecurityOriginInfo::feature>(securityOriginData.feature),
 214                             Equals<SecurityOriginInfo::scheme>(securityOriginData.origin.scheme)),
 215                         Equals<SecurityOriginInfo::host>(securityOriginData.origin.host)),
 216                     Equals<SecurityOriginInfo::port>(securityOriginData.origin.port)));
 217             del->Execute();
 218             transaction.Commit();
 219         }
 220     }
 221     SQL_CONNECTION_EXCEPTION_HANDLER_END("Fail to set security origin data")
 222 }
 223
 224 bool SecurityOriginDAO::hasResult(const SecurityOriginData &securityOriginData)
 225 {
 226     SQL_CONNECTION_EXCEPTION_HANDLER_BEGIN
 227     {
 228         SECURITY_ORIGIN_DB_SELECT(select, SecurityOriginInfo, &m_securityOriginDBInterface);
 229         select->Where(
 230             And(And(And(Equals<SecurityOriginInfo::feature>(securityOriginData.feature),
 231                         Equals<SecurityOriginInfo::scheme>(securityOriginData.origin.scheme)),
 232                     Equals<SecurityOriginInfo::host>(securityOriginData.origin.host)),
 233                 Equals<SecurityOriginInfo::port>(securityOriginData.origin.port)));
 234
 235         SecurityOriginInfo::Select::RowList rows = select->GetRowList();
 236
 237         return !rows.empty();
 238     }
 239     SQL_CONNECTION_EXCEPTION_HANDLER_END("Failed to get result for security origin")
 240 }
 241
 242 #undef SQL_CONNECTION_EXCEPTION_HANDLER_BEGIN
 243 #undef SQL_CONNECTION_EXCEPTION_HANDLER_END
 244
 245 } // namespace SecurityOriginDB