Branch data     Line data    Source code

       1                 :            : /*
       2                 :            :    Copyright (c) 2021 Fraunhofer AISEC. See the COPYRIGHT
       3                 :            :    file at the top-level directory of this distribution.
       4                 :            : 
       5                 :            :    Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
       6                 :            :    http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
       7                 :            :    <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
       8                 :            :    option. This file may not be copied, modified, or distributed
       9                 :            :    except according to those terms.
      10                 :            : */
      11                 :            : 
      12                 :            : #include <stdint.h>
      13                 :            : 
      14                 :            : #include "edhoc/edhoc_cose.h"
      15                 :            : 
      16                 :            : #include "common/oscore_edhoc_error.h"
      17                 :            : 
      18                 :            : #include "cbor/edhoc_encode_enc_structure.h"
      19                 :            : #include "cbor/edhoc_encode_sig_structure.h"
      20                 :            : 
      21                 :          9 : enum err cose_enc_structure_encode(const struct byte_array *context,
      22                 :            :                                    const struct byte_array *protected,
      23                 :            :                                    const struct byte_array *external_aad,
      24                 :            :                                    struct byte_array *out)
      25                 :            : {
      26                 :            :         struct edhoc_enc_structure enc_structure;
      27                 :            : 
      28                 :          9 :         enc_structure.edhoc_enc_structure_context.value = context->ptr;
      29                 :          9 :         enc_structure.edhoc_enc_structure_context.len = context->len;
      30                 :          9 :         enc_structure.edhoc_enc_structure_external_aad.value =
      31                 :          9 :                 external_aad->ptr;
      32                 :          9 :         enc_structure.edhoc_enc_structure_external_aad.len = external_aad->len;
      33                 :            : 
      34                 :            :         /* NULL protected with zero size is acceptable from EDHOC point of view,
      35                 :            :          * but CBOR encoder does not accept NULL as input argument.
      36                 :            :          * Internally it calls memmove that generates runtime error when input
      37                 :            :          * is NULL even if length is set to 0.
      38                 :            :          * Workaround is to provide dummy buffer to avoid passing NULL. It does not
      39                 :            :          * impact the EDHOC process, since protected length is set to 0 and no value
      40                 :            :          * is copied to the EDHOC message. */
      41                 :            :         const char dummy_buffer;
      42                 :            : 
      43         [ +  - ]:          9 :         if (NULL == protected->ptr) {
      44         [ -  + ]:          9 :                 if (0 != protected->len) {
      45                 :          0 :                         return wrong_parameter;
      46                 :            :                 } else {
      47                 :          9 :                         enc_structure.edhoc_enc_structure_protected.value =
      48                 :            :                                 (const uint8_t *)&dummy_buffer;
      49                 :            :                 }
      50                 :            :         } else {
      51                 :          0 :                 enc_structure.edhoc_enc_structure_protected.value =
      52                 :          0 :                         protected->ptr;
      53                 :            :         }
      54                 :            : 
      55                 :          9 :         enc_structure.edhoc_enc_structure_protected.len = protected->len;
      56                 :            : 
      57                 :            :         size_t payload_len_out;
      58         [ -  + ]:          9 :         TRY_EXPECT(cbor_encode_edhoc_enc_structure(out->ptr, out->len,
      59                 :            :                                                    &enc_structure,
      60                 :            :                                                    &payload_len_out),
      61                 :            :                    0);
      62                 :          9 :         out->len = (uint32_t)payload_len_out;
      63                 :          9 :         return ok;
      64                 :            : }
      65                 :            : 
      66                 :         12 : enum err cose_sig_structure_encode(const struct byte_array *context,
      67                 :            :                                    const struct byte_array *protected,
      68                 :            :                                    const struct byte_array *external_aad,
      69                 :            :                                    const struct byte_array *payload,
      70                 :            :                                    struct byte_array *out)
      71                 :            : {
      72                 :            :         struct sig_structure sig_structure;
      73                 :            : 
      74                 :         12 :         sig_structure.sig_structure_context.value = context->ptr;
      75                 :         12 :         sig_structure.sig_structure_context.len = context->len;
      76                 :         12 :         sig_structure.sig_structure_protected.value = protected->ptr;
      77                 :         12 :         sig_structure.sig_structure_protected.len = protected->len;
      78                 :         12 :         sig_structure.sig_structure_external_aad.value = external_aad->ptr;
      79                 :         12 :         sig_structure.sig_structure_external_aad.len = external_aad->len;
      80                 :         12 :         sig_structure.sig_structure_payload.value = payload->ptr;
      81                 :         12 :         sig_structure.sig_structure_payload.len = payload->len;
      82                 :            : 
      83                 :            :         size_t payload_len_out;
      84         [ -  + ]:         12 :         TRY_EXPECT(cbor_encode_sig_structure(out->ptr, out->len, &sig_structure,
      85                 :            :                                              &payload_len_out),
      86                 :            :                    0);
      87                 :            : 
      88                 :         12 :         out->len = (uint32_t)payload_len_out;
      89                 :         12 :         return ok;
      90                 :            : }