+++ /dev/null
-/*
- * Copyright © 2014 Intel Corporation
- *
- * Permission is hereby granted, free of charge, to any person obtaining a
- * copy of this software and associated documentation files (the "Software"),
- * to deal in the Software without restriction, including without limitation
- * the rights to use, copy, modify, merge, publish, distribute, sublicense,
- * and/or sell copies of the Software, and to permit persons to whom the
- * Software is furnished to do so, subject to the following conditions:
- *
- * The above copyright notice and this permission notice (including the next
- * paragraph) shall be included in all copies or substantial portions of the
- * Software.
- *
- * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
- * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
- * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
- * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
- * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
- * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
- * IN THE SOFTWARE.
- */
-
-/* A collection of unit tests for blob.c */
-
-#include <inttypes.h>
-#include <stdio.h>
-#include <stdlib.h>
-#include <stdbool.h>
-#include <string.h>
-#ifdef _MSC_VER
-#include <BaseTsd.h>
-typedef SSIZE_T ssize_t;
-#endif
-
-#include "util/ralloc.h"
-#include "blob.h"
-
-#define bytes_test_str "bytes_test"
-#define reserve_test_str "reserve_test"
-
-/* This placeholder must be the same length as the next overwrite_test_str */
-#define placeholder_str "XXXXXXXXXXXXXX"
-#define overwrite_test_str "overwrite_test"
-#define uint32_test 0x12345678
-#define uint32_placeholder 0xDEADBEEF
-#define uint32_overwrite 0xA1B2C3D4
-#define uint64_test 0x1234567890ABCDEF
-#define string_test_str "string_test"
-
-bool error = false;
-
-static void
-expect_equal(uint64_t expected, uint64_t actual, const char *test)
-{
- if (actual != expected) {
- fprintf(stderr,
- "Error: Test '%s' failed: "
- "Expected=%" PRIu64 ", "
- "Actual=%" PRIu64 "\n",
- test, expected, actual);
- error = true;
- }
-}
-
-static void
-expect_unequal(uint64_t expected, uint64_t actual, const char *test)
-{
- if (actual == expected) {
- fprintf(stderr,
- "Error: Test '%s' failed: Result=%" PRIu64 ", "
- "but expected something different.\n",
- test, actual);
- error = true;
- }
-}
-
-static void
-expect_equal_str(const char *expected, const char *actual, const char *test)
-{
- if (strcmp(expected, actual)) {
- fprintf (stderr, "Error: Test '%s' failed:\n\t"
- "Expected=\"%s\", Actual=\"%s\"\n",
- test, expected, actual);
- error = true;
- }
-}
-
-static void
-expect_equal_bytes(uint8_t *expected, const uint8_t *actual,
- size_t num_bytes, const char *test)
-{
- size_t i;
-
- if (memcmp(expected, actual, num_bytes)) {
- fprintf (stderr, "Error: Test '%s' failed:\n\t", test);
-
- fprintf (stderr, "Expected=[");
- for (i = 0; i < num_bytes; i++) {
- if (i != 0)
- fprintf(stderr, ", ");
- fprintf(stderr, "0x%02x", expected[i]);
- }
- fprintf (stderr, "]");
-
- fprintf (stderr, "Actual=[");
- for (i = 0; i < num_bytes; i++) {
- if (i != 0)
- fprintf(stderr, ", ");
- fprintf(stderr, "0x%02x", actual[i]);
- }
- fprintf (stderr, "]\n");
-
- error = true;
- }
-}
-
-/* Test at least one call of each blob_write_foo and blob_read_foo function,
- * verifying that we read out everything we wrote, that every bytes is
- * consumed, and that the overrun bit is not set.
- */
-static void
-test_write_and_read_functions (void)
-{
- struct blob blob;
- struct blob_reader reader;
- ssize_t reserved;
- size_t str_offset, uint_offset;
- uint8_t reserve_buf[sizeof(reserve_test_str)];
-
- blob_init(&blob);
-
- /*** Test blob by writing one of every possible kind of value. */
-
- blob_write_bytes(&blob, bytes_test_str, sizeof(bytes_test_str));
-
- reserved = blob_reserve_bytes(&blob, sizeof(reserve_test_str));
- blob_overwrite_bytes(&blob, reserved, reserve_test_str, sizeof(reserve_test_str));
-
- /* Write a placeholder, (to be replaced later via overwrite_bytes) */
- str_offset = blob.size;
- blob_write_bytes(&blob, placeholder_str, sizeof(placeholder_str));
-
- blob_write_uint32(&blob, uint32_test);
-
- /* Write a placeholder, (to be replaced later via overwrite_uint32) */
- uint_offset = blob.size;
- blob_write_uint32(&blob, uint32_placeholder);
-
- blob_write_uint64(&blob, uint64_test);
-
- blob_write_intptr(&blob, (intptr_t) &blob);
-
- blob_write_string(&blob, string_test_str);
-
- /* Finally, overwrite our placeholders. */
- blob_overwrite_bytes(&blob, str_offset, overwrite_test_str,
- sizeof(overwrite_test_str));
- blob_overwrite_uint32(&blob, uint_offset, uint32_overwrite);
-
- /*** Now read each value and verify. */
- blob_reader_init(&reader, blob.data, blob.size);
-
- expect_equal_str(bytes_test_str,
- blob_read_bytes(&reader, sizeof(bytes_test_str)),
- "blob_write/read_bytes");
-
- blob_copy_bytes(&reader, reserve_buf, sizeof(reserve_buf));
- expect_equal_str(reserve_test_str, (char *) reserve_buf,
- "blob_reserve_bytes/blob_copy_bytes");
-
- expect_equal_str(overwrite_test_str,
- blob_read_bytes(&reader, sizeof(overwrite_test_str)),
- "blob_overwrite_bytes");
-
- expect_equal(uint32_test, blob_read_uint32(&reader),
- "blob_write/read_uint32");
- expect_equal(uint32_overwrite, blob_read_uint32(&reader),
- "blob_overwrite_uint32");
- expect_equal(uint64_test, blob_read_uint64(&reader),
- "blob_write/read_uint64");
- expect_equal((intptr_t) &blob, blob_read_intptr(&reader),
- "blob_write/read_intptr");
- expect_equal_str(string_test_str, blob_read_string(&reader),
- "blob_write/read_string");
-
- expect_equal(reader.end - reader.data, reader.current - reader.data,
- "read_consumes_all_bytes");
- expect_equal(false, reader.overrun, "read_does_not_overrun");
-
- blob_finish(&blob);
-}
-
-/* Test that data values are written and read with proper alignment. */
-static void
-test_alignment(void)
-{
- struct blob blob;
- struct blob_reader reader;
- uint8_t bytes[] = "ABCDEFGHIJKLMNOP";
- size_t delta, last, num_bytes;
-
- blob_init(&blob);
-
- /* First, write an intptr value to the blob and capture that size. This is
- * the expected offset between any pair of intptr values (if written with
- * alignment).
- */
- blob_write_intptr(&blob, (intptr_t) &blob);
-
- delta = blob.size;
- last = blob.size;
-
- /* Then loop doing the following:
- *
- * 1. Write an unaligned number of bytes
- * 2. Verify that write results in an unaligned size
- * 3. Write an intptr_t value
- * 2. Verify that that write results in an aligned size
- */
- for (num_bytes = 1; num_bytes < sizeof(intptr_t); num_bytes++) {
- blob_write_bytes(&blob, bytes, num_bytes);
-
- expect_unequal(delta, blob.size - last, "unaligned write of bytes");
-
- blob_write_intptr(&blob, (intptr_t) &blob);
-
- expect_equal(2 * delta, blob.size - last, "aligned write of intptr");
-
- last = blob.size;
- }
-
- /* Finally, test that reading also does proper alignment. Since we know
- * that values were written with all the right alignment, all we have to do
- * here is verify that correct values are read.
- */
- blob_reader_init(&reader, blob.data, blob.size);
-
- expect_equal((intptr_t) &blob, blob_read_intptr(&reader),
- "read of initial, aligned intptr_t");
-
- for (num_bytes = 1; num_bytes < sizeof(intptr_t); num_bytes++) {
- expect_equal_bytes(bytes, blob_read_bytes(&reader, num_bytes),
- num_bytes, "unaligned read of bytes");
- expect_equal((intptr_t) &blob, blob_read_intptr(&reader),
- "aligned read of intptr_t");
- }
-
- blob_finish(&blob);
-}
-
-/* Test that we detect overrun. */
-static void
-test_overrun(void)
-{
- struct blob blob;
- struct blob_reader reader;
- uint32_t value = 0xdeadbeef;
-
- blob_init(&blob);
-
- blob_write_uint32(&blob, value);
-
- blob_reader_init(&reader, blob.data, blob.size);
-
- expect_equal(value, blob_read_uint32(&reader), "read before overrun");
- expect_equal(false, reader.overrun, "overrun flag not set");
- expect_equal(0, blob_read_uint32(&reader), "read at overrun");
- expect_equal(true, reader.overrun, "overrun flag set");
-
- blob_finish(&blob);
-}
-
-/* Test that we can read and write some large objects, (exercising the code in
- * the blob_write functions to realloc blob->data.
- */
-static void
-test_big_objects(void)
-{
- void *ctx = ralloc_context(NULL);
- struct blob blob;
- struct blob_reader reader;
- int size = 1000;
- int count = 1000;
- size_t i;
- char *buf;
-
- blob_init(&blob);
-
- /* Initialize our buffer. */
- buf = ralloc_size(ctx, size);
- for (i = 0; i < size; i++) {
- buf[i] = i % 256;
- }
-
- /* Write it many times. */
- for (i = 0; i < count; i++) {
- blob_write_bytes(&blob, buf, size);
- }
-
- blob_reader_init(&reader, blob.data, blob.size);
-
- /* Read and verify it many times. */
- for (i = 0; i < count; i++) {
- expect_equal_bytes((uint8_t *) buf, blob_read_bytes(&reader, size), size,
- "read of large objects");
- }
-
- expect_equal(reader.end - reader.data, reader.current - reader.data,
- "number of bytes read reading large objects");
-
- expect_equal(false, reader.overrun,
- "overrun flag not set reading large objects");
-
- blob_finish(&blob);
- ralloc_free(ctx);
-}
-
-int
-main (void)
-{
- test_write_and_read_functions ();
- test_alignment ();
- test_overrun ();
- test_big_objects ();
-
- return error ? 1 : 0;
-}
--- /dev/null
+/*
+ * Copyright © 2014 Intel Corporation
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a
+ * copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice (including the next
+ * paragraph) shall be included in all copies or substantial portions of the
+ * Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+ * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
+ * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+ * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
+ * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
+ * IN THE SOFTWARE.
+ */
+
+#include <inttypes.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <stdbool.h>
+#include <string.h>
+#ifdef _MSC_VER
+#include <BaseTsd.h>
+typedef SSIZE_T ssize_t;
+#endif
+
+#include "util/ralloc.h"
+#include "blob.h"
+
+#include <gtest/gtest.h>
+
+static testing::AssertionResult
+bytes_equal_pred(const char *a_expr,
+ const char *b_expr,
+ const char *c_expr,
+ const uint8_t *expected,
+ const uint8_t *actual,
+ size_t num_bytes)
+{
+ if (memcmp(expected, actual, num_bytes)) {
+ std::stringstream result;
+
+ result << a_expr << " and " << b_expr << " are different "
+ << "when comparing " << num_bytes << " bytes):\n";
+
+ for (size_t i = 0; i < num_bytes; i++) {
+ if (actual[i] != expected[i]) {
+ result << "Mismatch at index " << i << ": "
+ << "0x" << std::setfill('0') << std::setw(2) << std::right << std::hex << unsigned(actual[i])
+ << " != "
+ << "0x" << std::setfill('0') << std::setw(2) << std::right << std::hex << unsigned(expected[i])
+ << "\n";
+ }
+ }
+ result << "\n";
+
+ result << " Actual: [";
+ for (size_t i = 0; i < num_bytes; i++) {
+ if (i % 16 == 0)
+ result << "\n";
+ unsigned v = actual[i];
+ result << " 0x" << std::setfill('0') << std::setw(2)
+ << std::right << std::hex << v;
+ }
+ result << "]\n";
+
+ result << "Expected: [";
+ for (size_t i = 0; i < num_bytes; i++) {
+ if (i % 16 == 0)
+ result << "\n";
+ unsigned v = expected[i];
+ result << " 0x" << std::setfill('0') << std::setw(2)
+ << std::right << std::hex << v;
+ }
+ result << "]\n";
+
+ return testing::AssertionFailure() << result.str();
+ } else {
+ return testing::AssertionSuccess();
+ }
+}
+
+#define EXPECT_BYTES_EQUAL(expected, actual, num_bytes) \
+ EXPECT_PRED_FORMAT3(bytes_equal_pred, expected, actual, num_bytes)
+
+#define bytes_test_str "bytes_test"
+#define reserve_test_str "reserve_test"
+
+// This placeholder must be the same length as the next overwrite_test_str.
+#define placeholder_str "XXXXXXXXXXXXXX"
+#define overwrite_test_str "overwrite_test"
+#define uint32_test 0x12345678
+#define uint32_placeholder 0xDEADBEEF
+#define uint32_overwrite 0xA1B2C3D4
+#define uint64_test 0x1234567890ABCDEF
+#define string_test_str "string_test"
+
+
+// Test at least one call of each blob_write_foo and blob_read_foo function,
+// verifying that we read out everything we wrote, that every bytes is
+// consumed, and that the overrun bit is not set.
+TEST(BlobTest, WriteAndReadFunctions)
+{
+ struct blob blob;
+ struct blob_reader reader;
+ ssize_t reserved;
+ size_t str_offset, uint_offset;
+ uint8_t reserve_buf[sizeof(reserve_test_str)];
+
+ blob_init(&blob);
+
+ // Test blob by writing one of every possible kind of value.
+
+ blob_write_bytes(&blob, bytes_test_str, sizeof(bytes_test_str));
+
+ reserved = blob_reserve_bytes(&blob, sizeof(reserve_test_str));
+ blob_overwrite_bytes(&blob, reserved, reserve_test_str, sizeof(reserve_test_str));
+
+ // Write a placeholder, (to be replaced later via overwrite_bytes).
+ str_offset = blob.size;
+ blob_write_bytes(&blob, placeholder_str, sizeof(placeholder_str));
+
+ blob_write_uint32(&blob, uint32_test);
+
+ // Write a placeholder, (to be replaced later via overwrite_uint32).
+ uint_offset = blob.size;
+ blob_write_uint32(&blob, uint32_placeholder);
+
+ blob_write_uint64(&blob, uint64_test);
+
+ blob_write_intptr(&blob, (intptr_t) &blob);
+
+ blob_write_string(&blob, string_test_str);
+
+ // Finally, overwrite our placeholders.
+ blob_overwrite_bytes(&blob, str_offset, overwrite_test_str,
+ sizeof(overwrite_test_str));
+ blob_overwrite_uint32(&blob, uint_offset, uint32_overwrite);
+
+ // Now read each value and verify.
+
+ blob_reader_init(&reader, blob.data, blob.size);
+
+ EXPECT_STREQ(bytes_test_str,
+ (const char *)blob_read_bytes(&reader, sizeof(bytes_test_str))) <<
+ "blob_write/read_bytes";
+
+ blob_copy_bytes(&reader, reserve_buf, sizeof(reserve_buf));
+
+ EXPECT_STREQ(reserve_test_str, (char *) reserve_buf)
+ << "blob_reserve_bytes/blob_copy_bytes";
+
+ EXPECT_STREQ(overwrite_test_str,
+ (const char *) blob_read_bytes(&reader, sizeof(overwrite_test_str)))
+ << "blob_overwrite_bytes";
+
+ EXPECT_EQ(uint32_test, blob_read_uint32(&reader)) << "blob_write/read_uint32";
+ EXPECT_EQ(uint32_overwrite, blob_read_uint32(&reader)) << "blob_overwrite_uint32";
+ EXPECT_EQ(uint64_test, blob_read_uint64(&reader)) << "blob_write/read_uint64";
+ EXPECT_EQ((intptr_t) &blob, blob_read_intptr(&reader)) << "blob_write/read_intptr";
+
+ EXPECT_STREQ(string_test_str, blob_read_string(&reader)) << "blob_write/read_string";
+
+ EXPECT_EQ(reader.end - reader.data, reader.current - reader.data) << "read_consumes_all_bytes";
+ EXPECT_FALSE(reader.overrun) << "read_does_not_overrun";
+
+ blob_finish(&blob);
+}
+
+// Test that data values are written and read with proper alignment.
+TEST(BlobTest, Alignment)
+{
+ struct blob blob;
+ struct blob_reader reader;
+ uint8_t bytes[] = "ABCDEFGHIJKLMNOP";
+ size_t delta, last, num_bytes;
+
+ blob_init(&blob);
+
+ // First, write an intptr value to the blob and capture that size. This is
+ // the expected offset between any pair of intptr values (if written with
+ // alignment).
+ blob_write_intptr(&blob, (intptr_t) &blob);
+
+ delta = blob.size;
+ last = blob.size;
+
+ // Then loop doing the following:
+ //
+ // 1. Write an unaligned number of bytes
+ // 2. Verify that write results in an unaligned size
+ // 3. Write an intptr_t value
+ // 2. Verify that that write results in an aligned size
+ //
+ for (num_bytes = 1; num_bytes < sizeof(intptr_t); num_bytes++) {
+ blob_write_bytes(&blob, bytes, num_bytes);
+
+ EXPECT_NE(delta, blob.size - last) << "unaligned write of bytes";
+
+ blob_write_intptr(&blob, (intptr_t) &blob);
+
+ EXPECT_EQ(2 * delta, blob.size - last) << "aligned write of intptr";
+
+ last = blob.size;
+ }
+
+ // Finally, test that reading also does proper alignment. Since we know
+ // that values were written with all the right alignment, all we have to do
+ // here is verify that correct values are read.
+ blob_reader_init(&reader, blob.data, blob.size);
+
+ EXPECT_EQ((intptr_t) &blob, blob_read_intptr(&reader))
+ << "read of initial, aligned intptr_t";
+
+ for (num_bytes = 1; num_bytes < sizeof(intptr_t); num_bytes++) {
+ EXPECT_BYTES_EQUAL(bytes, (const uint8_t *) blob_read_bytes(&reader, num_bytes),
+ num_bytes) << "unaligned read of bytes";
+ EXPECT_EQ((intptr_t) &blob, blob_read_intptr(&reader)) << "aligned read of intptr_t";
+ }
+
+ blob_finish(&blob);
+}
+
+// Test that we detect overrun.
+TEST(BlobTest, DetectOverrun)
+{
+ struct blob blob;
+ struct blob_reader reader;
+ uint32_t value = 0xdeadbeef;
+
+ blob_init(&blob);
+
+ blob_write_uint32(&blob, value);
+
+ blob_reader_init(&reader, blob.data, blob.size);
+
+ EXPECT_EQ(value, blob_read_uint32(&reader)) << "read before overrun";
+ EXPECT_FALSE(reader.overrun);
+ EXPECT_EQ(0, blob_read_uint32(&reader)) << "read at overrun";
+ EXPECT_TRUE(reader.overrun);
+
+ blob_finish(&blob);
+}
+
+// Test that we can read and write some large objects, (exercising the code in
+// the blob_write functions to realloc blob->data.
+TEST(BlobTest, BigObjects)
+{
+ void *ctx = ralloc_context(NULL);
+ struct blob blob;
+ struct blob_reader reader;
+ int size = 1000;
+ int count = 1000;
+ char *buf;
+
+ blob_init(&blob);
+
+ // Initialize our buffer.
+ buf = (char *) ralloc_size(ctx, size);
+ for (int i = 0; i < size; i++) {
+ buf[i] = i % 256;
+ }
+
+ // Write it many times.
+ for (int i = 0; i < count; i++) {
+ blob_write_bytes(&blob, buf, size);
+ }
+
+ blob_reader_init(&reader, blob.data, blob.size);
+
+ // Read and verify it many times.
+ for (int i = 0; i < count; i++) {
+ EXPECT_BYTES_EQUAL((uint8_t *) buf,
+ (const uint8_t *) blob_read_bytes(&reader, size), size)
+ << "read of large objects, iteration " << i;
+ }
+
+ EXPECT_EQ(reader.end - reader.data, reader.current - reader.data)
+ << "number of bytes read reading large objects";
+
+ EXPECT_FALSE(reader.overrun) << "overrun flag not set reading large objects";
+
+ blob_finish(&blob);
+ ralloc_free(ctx);
+}