2 * Copyright (c) 2011 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 task_ecnrypt_resource.cpp
18 * @author Soyoung Kim (sy037.kim@samsung.com)
20 * @brief Implementation file for installer task encrypt resource
22 #include "task_encrypt_resource.h"
24 #undef __USE_FILE_OFFSET64
31 #include <dpl/log/log.h>
32 #include <dpl/errno_string.h>
33 #include <dpl/foreach.h>
34 #include <dpl/wrt-dao-ro/global_config.h>
36 #include <widget_install/job_widget_install.h>
37 #include <widget_install/widget_install_context.h>
38 #include <widget_install/widget_install_errors.h>
40 using namespace WrtDB;
41 using namespace WRTEncryptor;
44 std::set<std::string>& getSupportedForEncryption()
46 static std::set<std::string> encryptSet;
47 if (encryptSet.empty()) {
48 encryptSet.insert(".html");
49 encryptSet.insert(".css");
50 encryptSet.insert(".js");
55 bool isSupportedForEncryption(const std::string &file)
57 size_t foundKey = file.rfind(".");
58 if (std::string::npos != foundKey) {
59 std::string mimeType = file.substr(foundKey);
60 return getSupportedForEncryption().count(mimeType) > 0;
67 namespace WidgetInstall {
68 TaskEncryptResource::TaskEncryptResource(InstallerContext& context) :
69 DPL::TaskDecl<TaskEncryptResource>(this),
72 AddStep(&TaskEncryptResource::StepEncryptResource);
75 void TaskEncryptResource::StepEncryptResource()
77 LogDebug("Step Encrypt resource");
78 m_resEnc = new ResourceEncryptor;
79 m_resEnc->CreateEncryptionKey(DPL::ToUTF8String(*m_context.
80 widgetConfig.pkgname));
82 EncryptDirectory(m_context.locations->getTemporaryRootDir());
85 void TaskEncryptResource::EncryptDirectory(std::string path)
89 char * const paths[] = {const_cast<char * const>(path.c_str()), NULL};
91 if ((fts = fts_open(paths, FTS_PHYSICAL|FTS_NOCHDIR, NULL)) == NULL) {
94 LogWarning(__PRETTY_FUNCTION__ << ": fts_open failed with error: "
96 ThrowMsg(Exceptions::InternalError, "Error reading directory: "
100 while ((ftsent = fts_read(fts)) != NULL) {
101 switch (ftsent->fts_info) {
107 //directories, non-regular files, dangling symbolic links
112 //regular files and other objects that can be counted
113 if (isSupportedForEncryption(ftsent->fts_path)) {
114 EncryptFile(ftsent->fts_path);
122 LogWarning(__PRETTY_FUNCTION__
123 << ": traversal failed on file: "
126 << strerror(ftsent->fts_errno));
127 ThrowMsg(Exceptions::InternalError, "Error reading file");
131 if (fts_close(fts) == -1) {
133 LogWarning(__PRETTY_FUNCTION__ << ": fts_close failed with error: "
138 void TaskEncryptResource::EncryptFile(const std::string &fileName)
142 LogDebug("Need to ecnrypt file Name " << fileName);
143 std::string encFile = fileName + ".enc";
146 int ret = stat(fileName.c_str(), &buf);
148 size_t fileSize = buf.st_size;
150 FILE* resFp = fopen(fileName.c_str(), "r");
151 if ( NULL == resFp) {
152 LogError("Couldnot open file : " << fileName);
156 int blockSize = m_resEnc->GetBlockSize(fileSize);
157 LogDebug("Get block size : " << blockSize);
159 unsigned char readBuf[fileSize];
160 unsigned char outEncBuf[blockSize];
161 memset(readBuf, 0, fileSize);
162 memset(outEncBuf, 0, blockSize);
164 fread(readBuf, sizeof(unsigned char), fileSize, resFp);
167 m_resEnc->EncryptChunk(readBuf, outEncBuf, fileSize);
169 FILE* encFp = fopen(encFile.c_str(), "w");
171 LogError("Failed to open ecryption file");
174 fwrite(outEncBuf, sizeof(unsigned char), blockSize, encFp);
177 LogDebug("Success to encrypt file");
178 LogDebug("Remove unecrypted file : " << fileName);
180 unlink(fileName.c_str());
181 if ((rename(encFile.c_str(), fileName.c_str())) != 0) {
182 ThrowMsg(Exceptions::ExtractFileFailed, fileName);
185 std::string realPath = fileName;
186 realPath.replace(0, m_context.locations->getTemporaryRootDir().length(),
187 m_context.locations->getSourceDir());
189 WrtDB::EncryptedFileInfo info;
190 info.fileName = DPL::FromUTF8String(realPath);
191 info.fileSize = fileSize;
193 m_context.widgetConfig.encryptedFiles.insert(info);
196 Catch(ResourceEncryptor::Exception::Base)
198 ReThrowMsg(Exceptions::ExtractFileFailed, fileName);
201 } //namespace WidgetInstall