1 files changed, 59 insertions, 0 deletions
diff --git a/contrib/libcbor/test/memory_utils_test.c b/contrib/libcbor/test/memory_utils_test.c
new file mode 100644
index 000000000000..6cf07c7da934
--- /dev/null
+++ b/contrib/libcbor/test/memory_utils_test.c
@@ -0,0 +1,59 @@
+/*
+ * Copyright (c) 2014-2020 Pavel Kalvoda <me@pavelkalvoda.com>
+ *
+ * libcbor is free software; you can redistribute it and/or modify
+ * it under the terms of the MIT license. See LICENSE for details.
+ */
+
+#include "cbor/internal/memory_utils.h"
+#include <string.h>
+#include "assertions.h"
+
+static void test_safe_multiply(void **_CBOR_UNUSED(_state)) {
+  assert_true(_cbor_safe_to_multiply(1, 1));
+  assert_true(_cbor_safe_to_multiply(SIZE_MAX, 0));
+  assert_true(_cbor_safe_to_multiply(SIZE_MAX, 1));
+  assert_false(_cbor_safe_to_multiply(SIZE_MAX, 2));
+  assert_false(_cbor_safe_to_multiply(SIZE_MAX, SIZE_MAX));
+}
+
+static void test_safe_add(void **_CBOR_UNUSED(_state)) {
+  assert_true(_cbor_safe_to_add(1, 1));
+  assert_true(_cbor_safe_to_add(SIZE_MAX - 1, 1));
+  assert_true(_cbor_safe_to_add(SIZE_MAX, 0));
+  assert_false(_cbor_safe_to_add(SIZE_MAX, 1));
+  assert_false(_cbor_safe_to_add(SIZE_MAX, 2));
+  assert_false(_cbor_safe_to_add(SIZE_MAX, SIZE_MAX));
+  assert_false(_cbor_safe_to_add(SIZE_MAX - 1, SIZE_MAX));
+  assert_false(_cbor_safe_to_add(SIZE_MAX - 1, SIZE_MAX - 1));
+}
+
+static void test_safe_signalling_add(void **_CBOR_UNUSED(_state)) {
+  assert_size_equal(_cbor_safe_signaling_add(1, 2), 3);
+  assert_size_equal(_cbor_safe_signaling_add(0, 1), 0);
+  assert_size_equal(_cbor_safe_signaling_add(0, SIZE_MAX), 0);
+  assert_size_equal(_cbor_safe_signaling_add(1, SIZE_MAX), 0);
+  assert_size_equal(_cbor_safe_signaling_add(1, SIZE_MAX - 1), SIZE_MAX);
+}
+
+static void test_realloc_multiple(void **_CBOR_UNUSED(_state)) {
+  unsigned char *data = malloc(1);
+  data[0] = 0x2a;
+
+  data = _cbor_realloc_multiple(data, /*item_size=*/1, /*item_count=*/10);
+  assert_size_equal(data[0], 0x2a);
+  data[9] = 0x2b;  // Sanitizer will stop us if not ok
+  free(data);
+
+  assert_null(_cbor_realloc_multiple(NULL, SIZE_MAX, 2));
+}
+
+int main(void) {
+  const struct CMUnitTest tests[] = {
+      cmocka_unit_test(test_safe_multiply),
+      cmocka_unit_test(test_safe_add),
+      cmocka_unit_test(test_safe_signalling_add),
+      cmocka_unit_test(test_realloc_multiple),
+  };
+  return cmocka_run_group_tests(tests, NULL, NULL);
+}