Projekt

Obecné

Profil

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