Projekt

Obecné

Profil

Stáhnout (33.3 KB) Statistiky
| Větev: | Tag: | Revize:
1 ed55c677 Jan Pašek
import json
2 5b57121e Captain_Trojan
from datetime import datetime
3
from itertools import chain
4 6422796d Stanislav Král
from json import JSONDecodeError
5
6 1d8ff0a3 Stanislav Král
from flask import request, Response
7 1fa243ca Jan Pašek
from injector import inject
8
9
from src.constants import CA_ID, \
10
    SSL_ID, SIGNATURE_ID, AUTHENTICATION_ID, \
11
    DATETIME_FORMAT, ROOT_CA_ID, INTERMEDIATE_CA_ID, CERTIFICATE_ID  # TODO DATABASE_FILE - not the Controller's
12 f8bd4850 Jan Pašek
from src.controllers.return_codes import *
13 2cecaf70 Jan Pašek
from src.exceptions.database_exception import DatabaseException
14 9cf9a19d Captain_Trojan
from src.exceptions.unknown_exception import UnknownException
15 5b57121e Captain_Trojan
from src.model.subject import Subject
16 2cecaf70 Jan Pašek
from src.services.certificate_service import CertificateService, RevocationReasonInvalidException, \
17 94f8d5cf Jan Pašek
    CertificateStatusInvalidException, CertificateNotFoundException, CertificateAlreadyRevokedException, \
18
    CertificateCannotBeSetToValid
19 1fa243ca Jan Pašek
#  responsibility.
20 b8f5752f Captain_Trojan
from src.services.cryptography import CryptographyException
21 5b57121e Captain_Trojan
from src.services.key_service import KeyService
22 5e31b492 David Friesecký
from src.utils.logger import Logger
23
from src.utils.util import dict_to_string
24 5b57121e Captain_Trojan
25 a2965e9a Captain_Trojan
EXTENSIONS = "extensions"
26
27 9247d70a Captain_Trojan
TREE_NODE_TYPE_COUNT = 3
28 b8f5752f Captain_Trojan
KEY_PEM = "key_pem"
29
PASSWORD = "password"
30
KEY = "key"
31 5b57121e Captain_Trojan
FILTERING = "filtering"
32 d0471de8 Captain_Trojan
PAGE = "page"
33
PER_PAGE = "per_page"
34 5b57121e Captain_Trojan
ISSUER = "issuer"
35
US = "usage"
36
NOT_AFTER = "notAfter"
37
NOT_BEFORE = "notBefore"
38
COMMON_NAME = "CN"
39
ID = "id"
40 a767aa70 Captain_Trojan
CA = "CA"
41 9247d70a Captain_Trojan
USAGE = "usage"
42
SUBJECT = "subject"
43
VALIDITY_DAYS = "validityDays"
44 d0471de8 Captain_Trojan
TYPE = "type"
45 afbdc84f Captain_Trojan
ISSUED_BY = "issuedby"
46 2cecaf70 Jan Pašek
STATUS = "status"
47
REASON = "reason"
48 9c704fb1 Jan Pašek
REASON_UNDEFINED = "unspecified"
49 1d8ff0a3 Stanislav Král
NAME = "name"
50
PASSWORD = "password"
51 9247d70a Captain_Trojan
52
E_NO_ISSUER_FOUND = {"success": False, "data": "No certificate authority with such unique ID exists."}
53 d53c2fdc Captain_Trojan
E_NO_CERTIFICATES_FOUND = {"success": False, "data": "No such certificate found."}
54 9e6f791a Jan Pašek
E_NO_CERTIFICATE_ALREADY_REVOKED = {"success": False, "data": "Certificate is already revoked."}
55 cfda1725 Stanislav Král
E_NO_CERT_PRIVATE_KEY_FOUND = {"success": False,
56
                               "data": "Internal server error (certificate's private key cannot be found)."}
57 5b57121e Captain_Trojan
E_NOT_JSON_FORMAT = {"success": False, "data": "The request must be JSON-formatted."}
58
E_CORRUPTED_DATABASE = {"success": False, "data": "Internal server error (corrupted database)."}
59
E_GENERAL_ERROR = {"success": False, "data": "Internal server error (unknown origin)."}
60
E_MISSING_PARAMETERS = {"success": False, "data": "Invalid request, missing parameters."}
61
E_WRONG_PARAMETERS = {"success": False, "data": "Invalid request, wrong parameters."}
62 b8f5752f Captain_Trojan
E_WRONG_PASSWORD = {"success": False, "data": "The provided passphrase does not match the provided key."}
63 1d8ff0a3 Stanislav Král
E_IDENTITY_NAME_NOT_SPECIFIED = {"success": False, "data": "Invalid request, missing identity name."}
64
E_IDENTITY_PASSWORD_NOT_SPECIFIED = {"success": False, "data": "Invalid request, missing identity password."}
65 5b57121e Captain_Trojan
66
67
class CertController:
68 d0471de8 Captain_Trojan
    USAGE_KEY_MAP = {'CA': CA_ID, 'SSL': SSL_ID, 'digitalSignature': SIGNATURE_ID, 'authentication': AUTHENTICATION_ID}
69
    INVERSE_USAGE_KEY_MAP = {k: v for v, k in USAGE_KEY_MAP.items()}
70
    FILTERING_TYPE_KEY_MAP = {'root': ROOT_CA_ID, 'inter': INTERMEDIATE_CA_ID, 'end': CERTIFICATE_ID}
71
    # INVERSE_FILTERING_TYPE_KEY_MAP = {k: v for v, k in FILTERING_TYPE_KEY_MAP.items()}
72
73 5b57121e Captain_Trojan
74 1fa243ca Jan Pašek
    @inject
75
    def __init__(self, certificate_service: CertificateService, key_service: KeyService):
76
        self.certificate_service = certificate_service
77
        self.key_service = key_service
78
79
    def create_certificate(self):
80 5b57121e Captain_Trojan
        """create new certificate
81
82 9247d70a Captain_Trojan
        Create a new certificate based on given information
83 5b57121e Captain_Trojan
84
        :param body: Certificate data to be created
85
        :type body: dict | bytes
86
87
        :rtype: CreatedResponse
88
        """
89 5e31b492 David Friesecký
90
        Logger.info(f"\n\t{request.referrer}"
91
                    f"\n\t{request.method}   {request.path}   {request.scheme}")
92
93 9247d70a Captain_Trojan
        required_keys = {SUBJECT, USAGE, VALIDITY_DAYS}                             # required fields of the POST req
94 5b57121e Captain_Trojan
95 9247d70a Captain_Trojan
        if request.is_json:                                                         # accept JSON only
96 5b57121e Captain_Trojan
            body = request.get_json()
97 5e31b492 David Friesecký
98
            Logger.info(f"\n\tRequest body:"
99
                        f"\n{dict_to_string(body)}")
100
101 9247d70a Captain_Trojan
            if not all(k in body for k in required_keys):                           # verify that all keys are present
102 5e31b492 David Friesecký
                Logger.error(f"Invalid request, missing parameters")
103 9247d70a Captain_Trojan
                return E_MISSING_PARAMETERS, C_BAD_REQUEST
104 5b57121e Captain_Trojan
105 9247d70a Captain_Trojan
            if not isinstance(body[VALIDITY_DAYS], int):                            # type checking
106 5e31b492 David Friesecký
                Logger.error(f"Invalid request, wrong parameter '{VALIDITY_DAYS}'.")
107 9247d70a Captain_Trojan
                return E_WRONG_PARAMETERS, C_BAD_REQUEST
108 5b57121e Captain_Trojan
109 9247d70a Captain_Trojan
            subject = Subject.from_dict(body[SUBJECT])                              # generate Subject from passed dict
110 5b57121e Captain_Trojan
111 9247d70a Captain_Trojan
            if subject is None:                                                     # if the format is incorrect
112 5e31b492 David Friesecký
                Logger.error(f"Invalid request, wrong parameter '{SUBJECT}'.")
113 9247d70a Captain_Trojan
                return E_WRONG_PARAMETERS, C_BAD_REQUEST
114 5b57121e Captain_Trojan
115 5b6d9513 Captain_Trojan
            usages_dict = {}
116 5b57121e Captain_Trojan
117 b8f5752f Captain_Trojan
            if USAGE not in body or not isinstance(body[USAGE], list):              # type checking
118 5e31b492 David Friesecký
                Logger.error(f"Invalid request, wrong parameter '{USAGE}'.")
119 9247d70a Captain_Trojan
                return E_WRONG_PARAMETERS, C_BAD_REQUEST
120 5b57121e Captain_Trojan
121 b8f5752f Captain_Trojan
            for v in body[USAGE]:                                                   # for each usage
122 e9af380d Jan Pašek
                if v not in CertController.USAGE_KEY_MAP:                                 # check that it is a valid usage
123 b8f5752f Captain_Trojan
                    Logger.error(f"Invalid request, wrong parameter '{USAGE}'[{v}].")
124 9247d70a Captain_Trojan
                    return E_WRONG_PARAMETERS, C_BAD_REQUEST                        # and throw if it is not
125 3260320c Captain_Trojan
                usages_dict[CertController.USAGE_KEY_MAP[v]] = True                 # otherwise translate key and set
126 b8f5752f Captain_Trojan
127
            if KEY in body:
128
                if isinstance(body[KEY], dict):
129
                    if PASSWORD in body[KEY]:
130
                        passphrase = body[KEY][PASSWORD]
131
                        if KEY_PEM in body[KEY]:
132
                            key_pem = body[KEY][KEY_PEM]
133
                            if not self.key_service.verify_key(key_pem, passphrase=passphrase):
134
                                Logger.error(f"Passphrase specified but invalid.")
135
                                return E_WRONG_PASSWORD, C_BAD_REQUEST
136
                            key = self.key_service.wrap_custom_key(key_pem, passphrase=passphrase)
137
                        else:
138
                            key = self.key_service.create_new_key(passphrase)
139
                    else:
140
                        if KEY_PEM in body[KEY]:
141
                            key_pem = body[KEY][KEY_PEM]
142
                            if not self.key_service.verify_key(key_pem, passphrase=None):
143
                                Logger.error("Passphrase ommited but required.")
144
                                return E_WRONG_PASSWORD, C_BAD_REQUEST
145
                            key = self.key_service.wrap_custom_key(key_pem, passphrase=None)
146
                        else:
147 89aa967d Captain_Trojan
                            return E_WRONG_PARAMETERS, C_BAD_REQUEST                # if "key" exists but is empty
148 b8f5752f Captain_Trojan
                else:
149
                    return E_WRONG_PARAMETERS, C_BAD_REQUEST
150
            else:
151
                key = self.key_service.create_new_key()                             # if "key" does not exist
152 5b57121e Captain_Trojan
153 364a6830 Jan Pašek
            extensions = ""
154
            if EXTENSIONS in body:
155
                extensions = body[EXTENSIONS]
156
157 493022a0 Jan Pašek
            if CA not in body or body[CA] is None:                                  # if issuer omitted (legal) or none
158 b8f5752f Captain_Trojan
                cert = self.certificate_service.create_root_ca(                     # create a root CA
159 5b57121e Captain_Trojan
                    key,
160
                    subject,
161 9247d70a Captain_Trojan
                    usages=usages_dict,                                             # TODO ignoring usages -> discussion
162 e9af380d Jan Pašek
                    days=body[VALIDITY_DAYS],
163
                    extensions=extensions
164 5b57121e Captain_Trojan
                )
165
            else:
166 b8f5752f Captain_Trojan
                issuer = self.certificate_service.get_certificate(body[CA])         # get base issuer info
167 5b57121e Captain_Trojan
168 e9af380d Jan Pašek
                if issuer is None:                                              # if such issuer does not exist
169 5e31b492 David Friesecký
                    Logger.error(f"No certificate authority with such unique ID exists 'ID = {key.private_key_id}'.")
170 e9af380d Jan Pašek
                    self.key_service.delete_key(key.private_key_id)             # free
171
                    return E_NO_ISSUER_FOUND, C_BAD_REQUEST                     # and throw
172 5b57121e Captain_Trojan
173 e9af380d Jan Pašek
                issuer_key = self.key_service.get_key(issuer.private_key_id)    # get issuer's key, which must exist
174 5b57121e Captain_Trojan
175 e9af380d Jan Pašek
                if issuer_key is None:                                          # if it does not
176 5e31b492 David Friesecký
                    Logger.error(f"Internal server error (corrupted database).")
177 e9af380d Jan Pašek
                    self.key_service.delete_key(key.private_key_id)             # free
178
                    return E_CORRUPTED_DATABASE, C_INTERNAL_SERVER_ERROR        # and throw
179 5b57121e Captain_Trojan
180 1fa243ca Jan Pašek
                f = self.certificate_service.create_ca if CA_ID in usages_dict and usages_dict[CA_ID] else \
181
                    self.certificate_service.create_end_cert
182 5b57121e Captain_Trojan
183 e9af380d Jan Pašek
                # noinspection PyArgumentList
184
                cert = f(                                                       # create inter CA or end cert
185
                    key,                                                        # according to whether 'CA' is among
186
                    subject,                                                    # the usages' fields
187 89aa967d Captain_Trojan
                    issuer,
188
                    issuer_key,
189
                    usages=usages_dict,
190 e9af380d Jan Pašek
                    days=body[VALIDITY_DAYS],
191
                    extensions=extensions
192 89aa967d Captain_Trojan
                )
193 5b57121e Captain_Trojan
194
            if cert is not None:
195
                return {"success": True,
196 9247d70a Captain_Trojan
                        "data": cert.certificate_id}, C_CREATED_SUCCESSFULLY
197
            else:                                                                   # if this fails, then
198 5e31b492 David Friesecký
                Logger.error(f"Internal error: The certificate could not have been created.")
199 b8f5752f Captain_Trojan
                self.key_service.delete_key(key.private_key_id)                     # free
200 9247d70a Captain_Trojan
                return {"success": False,                                           # and wonder what the cause is,
201
                        "data": "Internal error: The certificate could not have been created."}, C_BAD_REQUEST
202
                                                                                    # as obj/None carries only one bit
203
                                                                                    # of error information
204 5b57121e Captain_Trojan
        else:
205 5e31b492 David Friesecký
            Logger.error(f"The request must be JSON-formatted.")
206 9247d70a Captain_Trojan
            return E_NOT_JSON_FORMAT, C_BAD_REQUEST                                 # throw in case of non-JSON format
207 5b57121e Captain_Trojan
208 1fa243ca Jan Pašek
    def get_certificate_by_id(self, id):
209 5b57121e Captain_Trojan
        """get certificate by ID
210
211 9247d70a Captain_Trojan
        Get certificate in PEM format by ID
212 5b57121e Captain_Trojan
213
        :param id: ID of a certificate to be queried
214
        :type id: dict | bytes
215
216
        :rtype: PemResponse
217
        """
218 5e31b492 David Friesecký
219
        Logger.info(f"\n\t{request.referrer}"
220
                    f"\n\t{request.method}   {request.path}   {request.scheme}"
221
                    f"\n\tCertificate ID = {id}")
222 fb987403 Captain_Trojan
        try:
223
            v = int(id)
224
        except ValueError:
225 5e31b492 David Friesecký
            Logger.error(f"Invalid request, wrong parameters 'id'[{id}].")
226 9247d70a Captain_Trojan
            return E_WRONG_PARAMETERS, C_BAD_REQUEST
227 fb987403 Captain_Trojan
228 1fa243ca Jan Pašek
        cert = self.certificate_service.get_certificate(v)
229 fb987403 Captain_Trojan
230
        if cert is None:
231 5e31b492 David Friesecký
            Logger.error(f"No such certificate found 'ID = {v}'.")
232 9cf9a19d Captain_Trojan
            return E_NO_CERTIFICATES_FOUND, C_NO_DATA
233 fb987403 Captain_Trojan
        else:
234 9247d70a Captain_Trojan
            return {"success": True, "data": cert.pem_data}, C_SUCCESS
235 5b57121e Captain_Trojan
236 1fa243ca Jan Pašek
    def get_certificate_details_by_id(self, id):
237 5b57121e Captain_Trojan
        """get certificate's details by ID
238
239 9247d70a Captain_Trojan
        Get certificate details by ID
240 5b57121e Captain_Trojan
241
        :param id: ID of a certificate whose details are to be queried
242
        :type id: dict | bytes
243
244
        :rtype: CertificateResponse
245
        """
246 5e31b492 David Friesecký
247
        Logger.info(f"\n\t{request.referrer}"
248
                    f"\n\t{request.method}   {request.path}   {request.scheme}"
249
                    f"\n\tCertificate ID = {id}")
250
251 5b6d9513 Captain_Trojan
        try:
252
            v = int(id)
253
        except ValueError:
254 5e31b492 David Friesecký
            Logger.error(f"Invalid request, wrong parameters 'id'[{id}].")
255 9247d70a Captain_Trojan
            return E_WRONG_PARAMETERS, C_BAD_REQUEST
256 5b6d9513 Captain_Trojan
257 1fa243ca Jan Pašek
        cert = self.certificate_service.get_certificate(v)
258 5b6d9513 Captain_Trojan
259
        if cert is None:
260 5e31b492 David Friesecký
            Logger.error(f"No such certificate found 'ID = {v}'.")
261 a53e5aef Jan Pašek
            return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND
262
263
        data = self.cert_to_dict_full(cert)
264
        if data is None:
265
            return E_CORRUPTED_DATABASE, C_INTERNAL_SERVER_ERROR
266
267
        try:
268
            state = self.certificate_service.get_certificate_state(v)
269
            data["status"] = state
270
        except CertificateNotFoundException:
271 53b0c8bc Jan Pašek
            Logger.error(f"No such certificate found 'ID = {id}'.")
272 a53e5aef Jan Pašek
273
        return {"success": True, "data": data}, C_SUCCESS
274 5b57121e Captain_Trojan
275 1fa243ca Jan Pašek
    def get_certificate_list(self):
276 5b57121e Captain_Trojan
        """get list of certificates
277
278 9247d70a Captain_Trojan
        Lists certificates based on provided filtering options
279 5b57121e Captain_Trojan
280
        :param filtering: Filter certificate type to be queried
281
        :type filtering: dict | bytes
282
283
        :rtype: CertificateListResponse
284
        """
285 5e31b492 David Friesecký
286
        Logger.info(f"\n\t{request.referrer}"
287
                    f"\n\t{request.method}   {request.path}   {request.scheme}")
288
289 ed55c677 Jan Pašek
290
        # the filtering parameter can be read as URL argument or as a request body
291 d0471de8 Captain_Trojan
        if request.is_json:
292
            data = request.get_json()
293
        else:
294
            data = {}
295 6422796d Stanislav Král
296 d0471de8 Captain_Trojan
        if FILTERING in request.args.keys():
297
            try:
298
                data[FILTERING] = json.loads(request.args[FILTERING])
299
            except JSONDecodeError:
300
                Logger.error(f"The request must be JSON-formatted.")
301
                return E_NOT_JSON_FORMAT, C_BAD_REQUEST
302
303
        if PAGE in request.args.keys():
304
            try:
305
                data[PAGE] = json.loads(request.args[PAGE])
306
            except JSONDecodeError:
307
                Logger.error(f"The request must be JSON-formatted.")
308
                return E_NOT_JSON_FORMAT, C_BAD_REQUEST
309
310
        if PER_PAGE in request.args.keys():
311
            try:
312
                data[PER_PAGE] = json.loads(request.args[PER_PAGE])
313
            except JSONDecodeError:
314
                Logger.error(f"The request must be JSON-formatted.")
315
                return E_NOT_JSON_FORMAT, C_BAD_REQUEST
316
317
        Logger.info(f"\n\tRequest body:"
318
                    f"\n{dict_to_string(data)}")
319
320
        target_types = {ROOT_CA_ID, INTERMEDIATE_CA_ID, CERTIFICATE_ID}
321
        target_usages = {v for v in CertController.INVERSE_USAGE_KEY_MAP.keys()}
322
        target_cn_substring = None
323
        issuer_id = -1
324 a767aa70 Captain_Trojan
325
        unfiltered = True
326
327 d0471de8 Captain_Trojan
        if PER_PAGE in data:
328 a767aa70 Captain_Trojan
            unfiltered = False
329 d0471de8 Captain_Trojan
            page = data.get(PAGE, 0)
330
            per_page = data[PER_PAGE]
331
        else:
332
            page = None
333
            per_page = None
334
335
        if FILTERING in data:                                                   # if the 'filtering' field exists
336 a767aa70 Captain_Trojan
            unfiltered = False
337 d0471de8 Captain_Trojan
            if isinstance(data[FILTERING], dict):                               # and it is also a 'dict'
338
339
                # noinspection DuplicatedCode
340
                if TYPE in data[FILTERING]:                                     # containing 'type'
341
                    if isinstance(data[FILTERING][TYPE], list):                 # which is a 'list',
342
                                                                                # map every field to id
343
                        try:
344
                            target_types = {CertController.FILTERING_TYPE_KEY_MAP[v] for v in data[FILTERING][TYPE]}
345
                        except KeyError as e:
346
                            Logger.error(f"Invalid request, wrong parameters '{FILTERING}.{TYPE}' - '{e}'.")
347 9247d70a Captain_Trojan
                            return E_WRONG_PARAMETERS, C_BAD_REQUEST
348 d0471de8 Captain_Trojan
                    else:
349
                        Logger.error(f"Invalid request, wrong parameters '{FILTERING}.{TYPE}'.")
350
                        return E_WRONG_PARAMETERS, C_BAD_REQUEST
351
352
                # noinspection DuplicatedCode
353
                if USAGE in data[FILTERING]:                                    # containing 'usage'
354
                    if isinstance(data[FILTERING][USAGE], list):                # which is a 'list',
355
                                                                                # map every field to id
356
                        try:
357
                            target_usages = {CertController.USAGE_KEY_MAP[v] for v in data[FILTERING][USAGE]}
358
                        except KeyError as e:
359 a767aa70 Captain_Trojan
                            Logger.error(f"Invalid request, wrong parameters '{FILTERING}.{USAGE}' - '{e}'.")
360 d0471de8 Captain_Trojan
                            return E_WRONG_PARAMETERS, C_BAD_REQUEST
361
                    else:
362
                        Logger.error(f"Invalid request, wrong parameters '{FILTERING}.{USAGE}'.")
363
                        return E_WRONG_PARAMETERS, C_BAD_REQUEST
364 5b57121e Captain_Trojan
365 d0471de8 Captain_Trojan
                if COMMON_NAME in data[FILTERING]:                              # containing 'CN'
366
                    if isinstance(data[FILTERING][COMMON_NAME], str):           # which is a 'str'
367
                        target_cn_substring = data[FILTERING][COMMON_NAME]
368
                    else:
369
                        Logger.error(f"Invalid request, wrong parameters '{FILTERING}.{COMMON_NAME}'.")
370
                        return E_WRONG_PARAMETERS, C_BAD_REQUEST
371 5b57121e Captain_Trojan
372 d0471de8 Captain_Trojan
                if ISSUED_BY in data[FILTERING]:                                # containing 'issuedby'
373
                    if isinstance(data[FILTERING][ISSUED_BY], int):             # which is an 'int'
374
                        issuer_id = data[FILTERING][ISSUED_BY]                  # then get its children only
375 9247d70a Captain_Trojan
376 d0471de8 Captain_Trojan
            else:
377
                Logger.error(f"Invalid request, wrong parameters '{FILTERING}'.")
378
                return E_WRONG_PARAMETERS, C_BAD_REQUEST
379
380 a767aa70 Captain_Trojan
        if unfiltered:                                                      # if not filtering
381
            certs = self.certificate_service.get_certificates()
382
        elif issuer_id >= 0:                                                # if filtering by an issuer
383
            try:
384
                                                                            # get his children, filtered
385
                certs = self.certificate_service.get_certificates_issued_by_filter(
386
                    issuer_id=issuer_id,
387 d0471de8 Captain_Trojan
                    target_types=target_types,
388
                    target_usages=target_usages,
389
                    target_cn_substring=target_cn_substring,
390
                    page=page,
391
                    per_page=per_page
392
                )
393 a767aa70 Captain_Trojan
            except CertificateNotFoundException:                            # if id does not exist
394
                Logger.error(f"No such certificate found 'ID = {issuer_id}'.")
395
                return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND                 # throw
396 afbdc84f Captain_Trojan
        else:
397 a767aa70 Captain_Trojan
            certs = self.certificate_service.get_certificates_filter(
398
                target_types=target_types,
399
                target_usages=target_usages,
400
                target_cn_substring=target_cn_substring,
401
                page=page,
402
                per_page=per_page
403
            )
404 5b57121e Captain_Trojan
405
        if certs is None:
406 5e31b492 David Friesecký
            Logger.error(f"Internal server error (unknown origin).")
407 9247d70a Captain_Trojan
            return E_GENERAL_ERROR, C_INTERNAL_SERVER_ERROR
408 5b57121e Captain_Trojan
        elif len(certs) == 0:
409 5e31b492 David Friesecký
            # TODO check log level
410
            Logger.warning(f"No such certificate found (empty list).")
411 a767aa70 Captain_Trojan
            return {"success": True, "data": []}, C_SUCCESS
412 5b57121e Captain_Trojan
        else:
413
            ret = []
414
            for c in certs:
415 1fa243ca Jan Pašek
                data = self.cert_to_dict_partial(c)
416 5b6d9513 Captain_Trojan
                if data is None:
417 5e31b492 David Friesecký
                    Logger.error(f"Internal server error (corrupted database).")
418 9247d70a Captain_Trojan
                    return E_CORRUPTED_DATABASE, C_INTERNAL_SERVER_ERROR
419 5b57121e Captain_Trojan
                ret.append(
420 5b6d9513 Captain_Trojan
                    data
421 5b57121e Captain_Trojan
                )
422 9247d70a Captain_Trojan
            return {"success": True, "data": ret}, C_SUCCESS
423 5b57121e Captain_Trojan
424 1fa243ca Jan Pašek
    def get_certificate_root_by_id(self, id):
425 5b57121e Captain_Trojan
        """get certificate's root of trust chain by ID
426
427 9247d70a Captain_Trojan
        Get certificate's root of trust chain in PEM format by ID
428 5b57121e Captain_Trojan
429
        :param id: ID of a child certificate whose root is to be queried
430
        :type id: dict | bytes
431
432
        :rtype: PemResponse
433
        """
434 5e31b492 David Friesecký
435
        Logger.info(f"\n\t{request.referrer}"
436
                    f"\n\t{request.method}   {request.path}   {request.scheme}"
437
                    f"\n\tCertificate ID = {id}")
438
439 d53c2fdc Captain_Trojan
        try:
440
            v = int(id)
441
        except ValueError:
442 5e31b492 David Friesecký
            Logger.error(f"Invalid request, wrong parameters 'id'[{id}].")
443 9247d70a Captain_Trojan
            return E_WRONG_PARAMETERS, C_BAD_REQUEST
444 d53c2fdc Captain_Trojan
445 1fa243ca Jan Pašek
        cert = self.certificate_service.get_certificate(v)
446 d53c2fdc Captain_Trojan
447
        if cert is None:
448 5e31b492 David Friesecký
            Logger.error(f"No such certificate found 'ID = {v}'.")
449 11a90594 Jan Pašek
            return E_NO_CERTIFICATES_FOUND, C_NO_DATA
450 d53c2fdc Captain_Trojan
451 1fa243ca Jan Pašek
        trust_chain = self.certificate_service.get_chain_of_trust(cert.parent_id, exclude_root=False)
452
        if trust_chain is None or len(trust_chain) == 0:
453 5e31b492 David Friesecký
            Logger.error(f"No such certificate found (empty list).")
454 11a90594 Jan Pašek
            return E_NO_CERTIFICATES_FOUND, C_NO_DATA
455 d53c2fdc Captain_Trojan
456 11a90594 Jan Pašek
        return {"success": True, "data": trust_chain[-1].pem_data}, C_SUCCESS
457 5b57121e Captain_Trojan
458 1fa243ca Jan Pašek
    def get_certificate_trust_chain_by_id(self, id):
459 61fb67b4 Jan Pašek
        """get certificate's trust chain by ID (including root certificate)
460 5b57121e Captain_Trojan
461 9247d70a Captain_Trojan
        Get certificate trust chain in PEM format by ID
462 5b57121e Captain_Trojan
463
        :param id: ID of a child certificate whose chain is to be queried
464
        :type id: dict | bytes
465
466
        :rtype: PemResponse
467
        """
468
469 5e31b492 David Friesecký
        Logger.info(f"\n\t{request.referrer}"
470
                    f"\n\t{request.method}   {request.path}   {request.scheme}"
471
                    f"\n\tCertificate ID = {id}")
472
473 aa740737 Captain_Trojan
        try:
474
            v = int(id)
475
        except ValueError:
476 5e31b492 David Friesecký
            Logger.error(f"Invalid request, wrong parameters 'id'[{id}].")
477 9247d70a Captain_Trojan
            return E_WRONG_PARAMETERS, C_BAD_REQUEST
478 aa740737 Captain_Trojan
479 1fa243ca Jan Pašek
        cert = self.certificate_service.get_certificate(v)
480 aa740737 Captain_Trojan
481
        if cert is None:
482 5e31b492 David Friesecký
            Logger.error(f"No such certificate found 'ID = {v}'.")
483 9cf9a19d Captain_Trojan
            return E_NO_CERTIFICATES_FOUND, C_NO_DATA
484 aa740737 Captain_Trojan
485 11a90594 Jan Pašek
        if cert.parent_id is None:
486 5e31b492 David Friesecký
            Logger.error(f"Parent ID is empty in certificate 'ID = {v}'.")
487 11a90594 Jan Pašek
            return E_NO_CERTIFICATES_FOUND, C_NO_DATA
488 aa740737 Captain_Trojan
489 61fb67b4 Jan Pašek
        trust_chain = self.certificate_service.get_chain_of_trust(cert.parent_id, exclude_root=False)
490 11a90594 Jan Pašek
491
        ret = []
492
        for intermediate in trust_chain:
493
            ret.append(intermediate.pem_data)
494 5b6d9513 Captain_Trojan
495 9247d70a Captain_Trojan
        return {"success": True, "data": "".join(ret)}, C_SUCCESS
496 5b6d9513 Captain_Trojan
497 2cecaf70 Jan Pašek
    def set_certificate_status(self, id):
498
        """
499
        Revoke a certificate given by ID
500
            - revocation request may contain revocation reason
501
            - revocation reason is verified based on the possible predefined values
502
            - if revocation reason is not specified 'undefined' value is used
503
        :param id: Identifier of the certificate to be revoked
504
        :type id: int
505
506
        :rtype: SuccessResponse | ErrorResponse (see OpenAPI definition)
507
        """
508
509 5e31b492 David Friesecký
        Logger.info(f"\n\t{request.referrer}"
510
                    f"\n\t{request.method}   {request.path}   {request.scheme}"
511
                    f"\n\tCertificate ID = {id}")
512
513
        required_keys = {STATUS}  # required keys
514 2cecaf70 Jan Pašek
515
        # check if the request contains a JSON body
516
        if request.is_json:
517
            request_body = request.get_json()
518 5e31b492 David Friesecký
519
            Logger.info(f"\n\tRequest body:"
520
                        f"\n{dict_to_string(request_body)}")
521
522
            # try to parse certificate identifier -> if it is not int return error 400
523
            try:
524
                identifier = int(id)
525
            except ValueError:
526
                Logger.error(f"Invalid request, wrong parameters 'id'[{id}].")
527
                return E_WRONG_PARAMETERS, C_BAD_REQUEST
528
529 2cecaf70 Jan Pašek
            # verify that all required keys are present
530
            if not all(k in request_body for k in required_keys):
531 5e31b492 David Friesecký
                Logger.error(f"Invalid request, missing parameters.")
532 2cecaf70 Jan Pašek
                return E_MISSING_PARAMETERS, C_BAD_REQUEST
533
534
            # get status and reason from the request
535
            status = request_body[STATUS]
536
            reason = request_body.get(REASON, REASON_UNDEFINED)
537
            try:
538
                # set certificate status using certificate_service
539 9c704fb1 Jan Pašek
                self.certificate_service.set_certificate_revocation_status(identifier, status, reason)
540 9e6f791a Jan Pašek
            except (RevocationReasonInvalidException, CertificateStatusInvalidException):
541 2cecaf70 Jan Pašek
                # these exceptions are thrown in case invalid status or revocation reason is passed to the controller
542 5e31b492 David Friesecký
                Logger.error(f"Invalid request, wrong parameters.")
543 2cecaf70 Jan Pašek
                return E_WRONG_PARAMETERS, C_BAD_REQUEST
544 9e6f791a Jan Pašek
            except CertificateAlreadyRevokedException:
545 5e31b492 David Friesecký
                Logger.error(f"Certificate is already revoked 'ID = {identifier}'.")
546 9e6f791a Jan Pašek
                return E_NO_CERTIFICATE_ALREADY_REVOKED, C_BAD_REQUEST
547
            except CertificateNotFoundException:
548 5e31b492 David Friesecký
                Logger.error(f"No such certificate found 'ID = {identifier}'.")
549 9e6f791a Jan Pašek
                return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND
550 94f8d5cf Jan Pašek
            except CertificateCannotBeSetToValid as e:
551
                return {"success": False, "data": str(e)}, C_BAD_REQUEST
552 2cecaf70 Jan Pašek
            return {"success": True,
553 9c704fb1 Jan Pašek
                    "data": "Certificate status updated successfully."}, C_SUCCESS
554 2cecaf70 Jan Pašek
        # throw an error in case the request does not contain a json body
555
        else:
556 5e31b492 David Friesecký
            Logger.error(f"The request must be JSON-formatted.")
557 2cecaf70 Jan Pašek
            return E_NOT_JSON_FORMAT, C_BAD_REQUEST
558
559 1fa243ca Jan Pašek
    def cert_to_dict_partial(self, c):
560 9247d70a Captain_Trojan
        """
561
        Dictionarizes a certificate directly fetched from the database. Contains partial information.
562
        :param c: target cert
563
        :return: certificate dict (compliant with some parts of the REST API)
564
        """
565 5e31b492 David Friesecký
566
        # TODO check log
567
        Logger.debug(f"Function launched.")
568
569 1fa243ca Jan Pašek
        c_issuer = self.certificate_service.get_certificate(c.parent_id)
570 5b6d9513 Captain_Trojan
        if c_issuer is None:
571
            return None
572
573
        return {
574
            ID: c.certificate_id,
575
            COMMON_NAME: c.common_name,
576
            NOT_BEFORE: datetime.strptime(c.valid_from, DATETIME_FORMAT).date(),
577
            NOT_AFTER: datetime.strptime(c.valid_to, DATETIME_FORMAT).date(),
578 d0471de8 Captain_Trojan
            USAGE: {CertController.INVERSE_USAGE_KEY_MAP[k]: v for k, v in c.usages.items()},
579 5b6d9513 Captain_Trojan
            ISSUER: {
580
                ID: c_issuer.certificate_id,
581
                COMMON_NAME: c_issuer.common_name
582
            }
583
        }
584
585 1fa243ca Jan Pašek
    def cert_to_dict_full(self, c):
586 9247d70a Captain_Trojan
        """
587
        Dictionarizes a certificate directly fetched from the database, but adds subject info.
588
        Contains full information.
589
        :param c: target cert
590
        :return: certificate dict (compliant with some parts of the REST API)
591
        """
592 5e31b492 David Friesecký
593
        Logger.info(f"Function launched.")
594
595 1fa243ca Jan Pašek
        subj = self.certificate_service.get_subject_from_certificate(c)
596
        c_issuer = self.certificate_service.get_certificate(c.parent_id)
597 5b6d9513 Captain_Trojan
        if c_issuer is None:
598
            return None
599
600
        return {
601
            SUBJECT: subj.to_dict(),
602
            NOT_BEFORE: datetime.strptime(c.valid_from, DATETIME_FORMAT).date(),
603
            NOT_AFTER: datetime.strptime(c.valid_to, DATETIME_FORMAT).date(),
604 d0471de8 Captain_Trojan
            USAGE: {CertController.INVERSE_USAGE_KEY_MAP[k]: v for k, v in c.usages.items()},
605 5b6d9513 Captain_Trojan
            CA: c_issuer.certificate_id
606
        }
607 cfda1725 Stanislav Král
608 ce8b9aaf Stanislav Král
    def get_private_key_of_a_certificate(self, id):
609 cfda1725 Stanislav Král
        """
610
        Get a private key used to sign a certificate in PEM format specified by certificate's ID
611
612 ce8b9aaf Stanislav Král
        :param id: ID of a certificate whose private key is to be queried
613
        :type id: dict | bytes
614
615
        :rtype: PemResponse
616
        """
617
618 5e31b492 David Friesecký
        Logger.info(f"\n\t{request.referrer}"
619
                    f"\n\t{request.method}   {request.path}   {request.scheme}"
620
                    f"\n\tCertificate ID = {id}")
621
622 ce8b9aaf Stanislav Král
        # try to parse the supplied ID
623
        try:
624
            v = int(id)
625
        except ValueError:
626 5e31b492 David Friesecký
            Logger.error(f"Invalid request, wrong parameters 'id'[{id}].")
627 ce8b9aaf Stanislav Král
            return E_WRONG_PARAMETERS, C_BAD_REQUEST
628
629 eade7427 Stanislav Král
        # find a certificate using the given ID
630 ce8b9aaf Stanislav Král
        cert = self.certificate_service.get_certificate(v)
631
632
        if cert is None:
633 5e31b492 David Friesecký
            Logger.error(f"No such certificate found 'ID = {v}'.")
634 ce8b9aaf Stanislav Král
            return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND
635
        else:
636
            # certificate exists, fetch it's private key
637
            private_key = self.key_service.get_key(cert.private_key_id)
638
            if cert is None:
639 5e31b492 David Friesecký
                Logger.error(f"Internal server error (certificate's private key cannot be found).")
640 ce8b9aaf Stanislav Král
                return E_NO_CERT_PRIVATE_KEY_FOUND, C_INTERNAL_SERVER_ERROR
641
            else:
642
                return {"success": True, "data": private_key.private_key}, C_SUCCESS
643
644
    def get_public_key_of_a_certificate(self, id):
645
        """
646
        Get a public key of a certificate in PEM format specified by certificate's ID
647
648 cfda1725 Stanislav Král
        :param id: ID of a certificate whose public key is to be queried
649
        :type id: dict | bytes
650
651
        :rtype: PemResponse
652
        """
653
654 5e31b492 David Friesecký
        Logger.info(f"\n\t{request.referrer}"
655
                    f"\n\t{request.method}   {request.path}   {request.scheme}"
656
                    f"\n\tCertificate ID = {id}")
657
658 cfda1725 Stanislav Král
        # try to parse the supplied ID
659
        try:
660
            v = int(id)
661
        except ValueError:
662 5e31b492 David Friesecký
            Logger.error(f"Invalid request, wrong parameters 'id'[{id}].")
663 cfda1725 Stanislav Král
            return E_WRONG_PARAMETERS, C_BAD_REQUEST
664
665 eade7427 Stanislav Král
        # find a certificate using the given ID
666 cfda1725 Stanislav Král
        cert = self.certificate_service.get_certificate(v)
667
668
        if cert is None:
669 5e31b492 David Friesecký
            Logger.error(f"No such certificate found 'ID = {v}'.")
670 cfda1725 Stanislav Král
            return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND
671
        else:
672 d3bfacfc Stanislav Král
            return {"success": True, "data": self.certificate_service.get_public_key_from_certificate(cert)}, C_SUCCESS
673 9cf9a19d Captain_Trojan
674
    def delete_certificate(self, id):
675
        """
676
        Deletes a certificate identified by ID, including its corresponding subtree (all descendants).
677
        :param id: target certificate ID
678
        :rtype: DeleteResponse
679
        """
680 5e31b492 David Friesecký
681
        Logger.info(f"\n\t{request.referrer}"
682
                    f"\n\t{request.method}   {request.path}   {request.scheme}"
683
                    f"\n\tCertificate ID = {id}")
684
685 9cf9a19d Captain_Trojan
        try:
686
            v = int(id)
687
        except ValueError:
688 5e31b492 David Friesecký
            Logger.error(f"Invalid request, wrong parameters 'id'[{id}].")
689 9cf9a19d Captain_Trojan
            return E_WRONG_PARAMETERS, C_BAD_REQUEST
690
691
        try:
692
            self.certificate_service.delete_certificate(v)
693
        except CertificateNotFoundException:
694 5e31b492 David Friesecký
            Logger.error(f"No such certificate found 'ID = {v}'.")
695 9cf9a19d Captain_Trojan
            return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND
696
        except DatabaseException:
697 5e31b492 David Friesecký
            Logger.error(f"Internal server error (corrupted database).")
698 9cf9a19d Captain_Trojan
            return E_CORRUPTED_DATABASE, C_INTERNAL_SERVER_ERROR
699
        except CertificateStatusInvalidException or RevocationReasonInvalidException or UnknownException:
700 5e31b492 David Friesecký
            Logger.error(f"Internal server error (unknown origin).")
701 9cf9a19d Captain_Trojan
            return E_GENERAL_ERROR, C_INTERNAL_SERVER_ERROR
702
703 1d8ff0a3 Stanislav Král
        return {"success": True, "data": "The certificate and its descendants have been successfully deleted."}
704
705
    def generate_certificate_pkcs_identity(self, id):
706
        """
707
        Generates a PKCS12 identity (including the chain of trust) of the certificate given by the specified ID.
708
        Response is of application/x-pkcs12 type.
709
710
        :param id: ID of a certificate whose PKCS12 identity should be generated
711
        :type id: int
712
713
        :rtype: Response
714
        """
715
716
        Logger.info(f"\n\t{request.referrer}"
717
                    f"\n\t{request.method}   {request.path}   {request.scheme}"
718
                    f"\n\tCertificate ID = {id}")
719
720
        # try to parse the supplied ID
721
        try:
722
            v = int(id)
723
        except ValueError:
724
            Logger.error(f"Invalid request, wrong parameters 'id'[{id}] (expected integer).")
725
            return E_WRONG_PARAMETERS, C_BAD_REQUEST
726
727
        # find a certificate using the given ID
728
        cert = self.certificate_service.get_certificate(v)
729
730
        if request.is_json:                                                         # accept JSON only
731
            body = request.get_json()
732
733
            # check whether the request is well formed meaning that it contains all required fields
734
            if NAME not in body.keys():
735
                return E_IDENTITY_NAME_NOT_SPECIFIED, C_BAD_REQUEST
736
737
            if PASSWORD not in body.keys():
738
                return E_IDENTITY_PASSWORD_NOT_SPECIFIED, C_BAD_REQUEST
739
740
            # parse required fields from the request
741
            identity_name = body[NAME]
742
            identity_password = body[PASSWORD]
743
744
            # check whether a certificated specified by the given ID exists
745
            if cert is None:
746
                Logger.error(f"No such certificate found 'ID = {v}'.")
747
                return E_NO_CERTIFICATES_FOUND, C_NOT_FOUND
748
            else:
749
                # try to load it's private key
750
                key = self.key_service.get_key(cert.private_key_id)
751
                if key is None:
752
                    Logger.error(
753
                        f"The private key 'ID = {cert.private_key_id}'of the certificate 'ID = {cert.certificate_id}' does not exist.")
754
                    return E_NO_CERTIFICATES_FOUND, C_INTERNAL_SERVER_ERROR
755
                else:
756
                    # generate PKCS12 identity
757 04805a41 Stanislav Král
                    identity_byte_array = self.certificate_service.generate_pkcs_identity(cert, key,
758 1d8ff0a3 Stanislav Král
                                                                                          identity_name,
759
                                                                                          identity_password)
760
                    return Response(identity_byte_array, mimetype='application/x-pkcs12')