extlibs/tinycbor/tinycbor/src/cborencoder_close_container_checked.c

   1 /****************************************************************************
   2 **
   3 ** Copyright (C) 2015 Intel Corporation
   4 **
   5 ** Permission is hereby granted, free of charge, to any person obtaining a copy
   6 ** of this software and associated documentation files (the "Software"), to deal
   7 ** in the Software without restriction, including without limitation the rights
   8 ** to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
   9 ** copies of the Software, and to permit persons to whom the Software is
  10 ** furnished to do so, subject to the following conditions:
  11 **
  12 ** The above copyright notice and this permission notice shall be included in
  13 ** all copies or substantial portions of the Software.
  14 **
  15 ** THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16 ** IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17 ** FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
  18 ** AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19 ** LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
  20 ** OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
  21 ** THE SOFTWARE.
  22 **
  23 ****************************************************************************/
  24
  25 #define _BSD_SOURCE 1
  26 #define _DEFAULT_SOURCE 1
  27 #ifndef __STDC_LIMIT_MACROS
  28 #  define __STDC_LIMIT_MACROS 1
  29 #endif
  30
  31 #include "cbor.h"
  32 #include "cborconstants_p.h"
  33 #include "compilersupport_p.h"
  34 #include "extract_number_p.h"
  35
  36 #include <assert.h>
  37
  38 #include "assert_p.h"       /* Always include last */
  39
  40 /**
  41  * \addtogroup CborEncoding
  42  * @{
  43  */
  44
  45 /**
  46  *
  47  * Closes the CBOR container (array or map) provided by \a containerEncoder and
  48  * updates the CBOR stream provided by \a encoder. Both parameters must be the
  49  * same as were passed to cbor_encoder_create_array() or
  50  * cbor_encoder_create_map().
  51  *
  52  * Unlike cbor_encoder_close_container(), this function checks that the number
  53  * of items (or pair of items, in the case of a map) was correct. If the number
  54  * of items inserted does not match the length originally passed to
  55  * cbor_encoder_create_array() or cbor_encoder_create_map(), this function
  56  * returns either CborErrorTooFewItems or CborErrorTooManyItems.
  57  *
  58  * \sa cbor_encoder_create_array(), cbor_encoder_create_map()
  59  */
  60 CborError cbor_encoder_close_container_checked(CborEncoder *encoder, const CborEncoder *containerEncoder)
  61 {
  62     const uint8_t *ptr = encoder->ptr;
  63     CborError err = cbor_encoder_close_container(encoder, containerEncoder);
  64     if (containerEncoder->flags & CborIteratorFlag_UnknownLength || encoder->end == NULL)
  65         return err;
  66
  67     /* check what the original length was */
  68     uint64_t actually_added;
  69     err = extract_number(&ptr, encoder->ptr, &actually_added);
  70     if (err)
  71         return err;
  72
  73     if (containerEncoder->flags & CborIteratorFlag_ContainerIsMap) {
  74         if (actually_added > SIZE_MAX / 2)
  75             return CborErrorDataTooLarge;
  76         actually_added *= 2;
  77     }
  78     return actually_added == containerEncoder->added ? CborNoError :
  79            actually_added < containerEncoder->added ? CborErrorTooManyItems : CborErrorTooFewItems;
  80 }
  81
  82 /** @} */