Drop all encryption requests upon disconnection
[platform/core/security/key-manager.git] / src / manager / service / encryption-logic.cpp
1 /*
2  *  Copyright (c) 2000 - 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       encryption-logic.cpp
18  * @author     Krzysztof Jackiewicz (k.jackiewicz@samsung.com)
19  * @version    1.0
20  */
21
22 #include <encryption-logic.h>
23 #include <ckm/ckm-error.h>
24 #include <dpl/log/log.h>
25
26 namespace CKM {
27
28 void EncryptionLogic::Crypt(const CryptoRequest &request)
29 {
30         // check arguments
31         if (request.input.empty()) {
32                 LogError("No input data");
33                 m_service.RespondToClient(request, CKM_API_ERROR_INPUT_PARAM);
34                 return;
35         }
36         m_counter++;
37
38         // store request in the map
39         auto ret = m_requestsMap.insert(std::make_pair(m_counter, request));
40
41         if (!ret.second) {
42                 LogError("Request with id = " << m_counter << " already exists");
43                 m_service.RespondToClient(request, CKM_API_ERROR_SERVER_ERROR);
44                 return;
45         }
46
47         // request key
48         try {
49                 m_service.RequestKey(m_counter, request);
50         } catch (...) {
51                 LogError("Key request failed");
52                 m_requestsMap.erase(m_counter);
53                 m_service.RespondToClient(request, CKM_API_ERROR_SERVER_ERROR);
54         }
55 }
56
57 void EncryptionLogic::DropRequests(const ConnectionID& connectionID)
58 {
59         for (auto it = m_requestsMap.begin(); it != m_requestsMap.end();) {
60                 if (it->second.conn.counter == connectionID.counter)
61                         it = m_requestsMap.erase(it);
62                 else
63                         it++;
64         }
65 }
66
67 void EncryptionLogic::KeyRetrieved(MsgKeyResponse response)
68 {
69         auto it = m_requestsMap.find(response.id);
70
71         if (it == m_requestsMap.end()) {
72                 LogError("No matching request found"); // nothing we can do
73                 return;
74         }
75
76         CryptoRequest req = std::move(it->second);
77         m_requestsMap.erase(it);
78
79         if (response.error != CKM_API_SUCCESS) {
80                 LogError("Attempt to retrieve key failed with error: " << response.error);
81                 m_service.RespondToClient(req, response.error);
82                 return;
83         }
84
85         if (!response.key) {
86                 LogError("Retrieved key is empty");
87                 m_service.RespondToClient(req, CKM_API_ERROR_SERVER_ERROR);
88                 return;
89         }
90
91         // encrypt/decrypt
92         try {
93                 RawBuffer output;
94
95                 if (req.command == EncryptionCommand::ENCRYPT)
96                         output = response.key->encrypt(req.cas, req.input);
97                 else
98                         output = response.key->decrypt(req.cas, req.input);
99
100                 m_service.RespondToClient(req, CKM_API_SUCCESS, output);
101         } catch (const Exc::Exception &ex) {
102                 m_service.RespondToClient(req, ex.error());
103         } catch (...) {
104                 LogError("Uncaught exception from encrypt/decrypt.");
105                 m_service.RespondToClient(req, CKM_API_ERROR_SERVER_ERROR);
106         }
107 }
108
109 } /* namespace CKM */