Projekt

Obecné

Profil

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