1 /*
2  *  PSA crypto client code
3  */
4 /*
5  *  Copyright The Mbed TLS Contributors
6  *  SPDX-License-Identifier: Apache-2.0
7  *
8  *  Licensed under the Apache License, Version 2.0 (the "License"); you may
9  *  not use this file except in compliance with the License.
10  *  You may obtain a copy of the License at
11  *
12  *  http://www.apache.org/licenses/LICENSE-2.0
13  *
14  *  Unless required by applicable law or agreed to in writing, software
15  *  distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
16  *  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  *  See the License for the specific language governing permissions and
18  *  limitations under the License.
19  */
20 
21 #include "common.h"
22 #include "psa/crypto.h"
23 
24 #if defined(MBEDTLS_PSA_CRYPTO_CLIENT)
25 
26 #include <string.h>
27 #include "mbedtls/platform.h"
28 
psa_reset_key_attributes(psa_key_attributes_t * attributes)29 void psa_reset_key_attributes(psa_key_attributes_t *attributes)
30 {
31     mbedtls_free(attributes->domain_parameters);
32     memset(attributes, 0, sizeof(*attributes));
33 }
34 
psa_set_key_domain_parameters(psa_key_attributes_t * attributes,psa_key_type_t type,const uint8_t * data,size_t data_length)35 psa_status_t psa_set_key_domain_parameters(psa_key_attributes_t *attributes,
36                                            psa_key_type_t type,
37                                            const uint8_t *data,
38                                            size_t data_length)
39 {
40     uint8_t *copy = NULL;
41 
42     if (data_length != 0) {
43         copy = mbedtls_calloc(1, data_length);
44         if (copy == NULL) {
45             return PSA_ERROR_INSUFFICIENT_MEMORY;
46         }
47         memcpy(copy, data, data_length);
48     }
49     /* After this point, this function is guaranteed to succeed, so it
50      * can start modifying `*attributes`. */
51 
52     if (attributes->domain_parameters != NULL) {
53         mbedtls_free(attributes->domain_parameters);
54         attributes->domain_parameters = NULL;
55         attributes->domain_parameters_size = 0;
56     }
57 
58     attributes->domain_parameters = copy;
59     attributes->domain_parameters_size = data_length;
60     attributes->core.type = type;
61     return PSA_SUCCESS;
62 }
63 
psa_get_key_domain_parameters(const psa_key_attributes_t * attributes,uint8_t * data,size_t data_size,size_t * data_length)64 psa_status_t psa_get_key_domain_parameters(
65     const psa_key_attributes_t *attributes,
66     uint8_t *data, size_t data_size, size_t *data_length)
67 {
68     if (attributes->domain_parameters_size > data_size) {
69         return PSA_ERROR_BUFFER_TOO_SMALL;
70     }
71     *data_length = attributes->domain_parameters_size;
72     if (attributes->domain_parameters_size != 0) {
73         memcpy(data, attributes->domain_parameters,
74                attributes->domain_parameters_size);
75     }
76     return PSA_SUCCESS;
77 }
78 
79 #endif /* MBEDTLS_PSA_CRYPTO_CLIENT */
80