/*
 * Copyright 2009-present MongoDB, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */


#include <bson/bson.h>
#include <fcntl.h>

#include "TestSuite.h"


static void
test_reader_from_data (void)
{
   bson_reader_t *reader;
   uint8_t *buffer;
   const bson_t *b;
   uint32_t i;
   bool eof = false;

   buffer = bson_malloc0 (4095);
   for (i = 0; i < 4095; i += 5) {
      buffer[i] = 5;
   }

   reader = bson_reader_new_from_data (buffer, 4095);

   for (i = 0; (b = bson_reader_read (reader, &eof)); i++) {
      const uint8_t *buf = bson_get_data (b);

      /* do nothing */
      BSON_ASSERT (b->len == 5);
      BSON_ASSERT (buf[0] == 5);
      BSON_ASSERT (buf[1] == 0);
      BSON_ASSERT (buf[2] == 0);
      BSON_ASSERT (buf[3] == 0);
      BSON_ASSERT (buf[4] == 0);
   }

   BSON_ASSERT (i == (4095 / 5));

   ASSERT_CMPINT (eof, ==, true);

   bson_free (buffer);

   bson_reader_destroy (reader);
}


static void
test_reader_from_data_overflow (void)
{
   bson_reader_t *reader;
   uint8_t *buffer;
   const bson_t *b;
   uint32_t i;
   bool eof = false;

   buffer = bson_malloc0 (4096);
   for (i = 0; i < 4095; i += 5) {
      buffer[i] = 5;
   }

   buffer[4095] = 5;

   reader = bson_reader_new_from_data (buffer, 4096);

   for (i = 0; (b = bson_reader_read (reader, &eof)); i++) {
      const uint8_t *buf = bson_get_data (b);
      BSON_ASSERT (b->len == 5);
      BSON_ASSERT (buf[0] == 5);
      BSON_ASSERT (buf[1] == 0);
      BSON_ASSERT (buf[2] == 0);
      BSON_ASSERT (buf[3] == 0);
      BSON_ASSERT (buf[4] == 0);
      eof = false;
   }

   BSON_ASSERT (i == (4095 / 5));

   ASSERT_CMPINT (eof, ==, false);

   bson_free (buffer);

   bson_reader_destroy (reader);
}

static void
test_reader_from_data_document_length_too_large (void)
{
   bson_reader_t *reader;
   uint8_t *buffer;
   bool eof = false;

   buffer = bson_malloc0 (5);
   buffer[0] = 6;

   reader = bson_reader_new_from_data (buffer, 5);
   BSON_ASSERT (!bson_reader_read (reader, &eof));
   ASSERT_CMPINT (eof, ==, false);

   bson_free (buffer);

   bson_reader_destroy (reader);
}

static void
test_reader_from_data_document_length_too_small (void)
{
   bson_reader_t *reader;
   uint8_t *buffer;
   bool eof = false;

   buffer = bson_malloc0 (5);
   buffer[0] = 4;

   reader = bson_reader_new_from_data (buffer, 5);
   BSON_ASSERT (!bson_reader_read (reader, &eof));
   ASSERT_CMPINT (eof, ==, false);

   bson_free (buffer);

   bson_reader_destroy (reader);
}

static ssize_t
test_reader_from_handle_read (void *handle, void *buf, size_t len)
{
   return bson_read (*(int *) handle, buf, len);
}

static void
test_reader_from_handle_destroy (void *handle)
{
   bson_close (*(int *) handle);
}

static void
test_reader_from_handle (void)
{
   bson_reader_t *reader;
   const bson_t *b;
   bson_iter_t iter;
   uint32_t i;
   bool eof = true;
   int fd;

   fd = bson_open (BSON_BINARY_DIR "/stream.bson", O_RDONLY);
   BSON_ASSERT (-1 != fd);

   reader = bson_reader_new_from_handle ((void *) &fd, &test_reader_from_handle_read, &test_reader_from_handle_destroy);

   for (i = 0; i < 1000; i++) {
      eof = false;
      b = bson_reader_read (reader, &eof);
      BSON_ASSERT (b);
      BSON_ASSERT (bson_iter_init (&iter, b));
      BSON_ASSERT (!bson_iter_next (&iter));
   }

   ASSERT_CMPINT (eof, ==, false);
   b = bson_reader_read (reader, &eof);
   BSON_ASSERT (!b);
   ASSERT_CMPINT (eof, ==, true);
   bson_reader_destroy (reader);
}


static void
test_reader_tell (void)
{
   bson_reader_t *reader;
   const bson_t *b;
   uint32_t i;
   bson_iter_t iter;
   bool eof = true;
   int fd;

   fd = bson_open (BSON_BINARY_DIR "/stream.bson", O_RDONLY);
   BSON_ASSERT (-1 != fd);

   reader = bson_reader_new_from_handle ((void *) &fd, &test_reader_from_handle_read, &test_reader_from_handle_destroy);

   for (i = 0; i < 1000; i++) {
      if (i) {
         ASSERT_CMPINT (5 * i, ==, (int) bson_reader_tell (reader));
      } else {
         ASSERT_CMPINT (0, ==, (int) bson_reader_tell (reader));
      }
      eof = false;
      b = bson_reader_read (reader, &eof);
      BSON_ASSERT (b);
      BSON_ASSERT (bson_iter_init (&iter, b));
      BSON_ASSERT (!bson_iter_next (&iter));
   }

   ASSERT_CMPINT (5000, ==, (int) bson_reader_tell (reader));
   ASSERT (!eof);
   b = bson_reader_read (reader, &eof);
   BSON_ASSERT (!b);
   BSON_ASSERT (eof);
   bson_reader_destroy (reader);
}


static void
test_reader_from_handle_corrupt (void)
{
   bson_reader_t *reader;
   const bson_t *b;
   uint32_t i;
   bson_iter_t iter;
   bool eof;
   int fd;

   fd = bson_open (BSON_BINARY_DIR "/stream_corrupt.bson", O_RDONLY);
   BSON_ASSERT (-1 != fd);

   reader = bson_reader_new_from_handle ((void *) &fd, &test_reader_from_handle_read, &test_reader_from_handle_destroy);

   for (i = 0; i < 1000; i++) {
      b = bson_reader_read (reader, &eof);
      BSON_ASSERT (b);
      BSON_ASSERT (bson_iter_init (&iter, b));
      BSON_ASSERT (!bson_iter_next (&iter));
   }

   b = bson_reader_read (reader, &eof);
   BSON_ASSERT (!b);
   bson_reader_destroy (reader);
}


static void
test_reader_grow_buffer (void)
{
   bson_reader_t *reader;
   const bson_t *b;
   bool eof = false;
   int fd;

   fd = bson_open (BSON_BINARY_DIR "/readergrow.bson", O_RDONLY);
   BSON_ASSERT (-1 != fd);

   reader = bson_reader_new_from_handle ((void *) &fd, &test_reader_from_handle_read, &test_reader_from_handle_destroy);

   b = bson_reader_read (reader, &eof);
   BSON_ASSERT (b);
   BSON_ASSERT (!eof);

   b = bson_reader_read (reader, &eof);
   BSON_ASSERT (!b);
   BSON_ASSERT (eof);

   bson_reader_destroy (reader);
}


static void
test_reader_reset (void)
{
   uint8_t buffer[10];
   bson_reader_t *reader;
   bool eof;

   memset (buffer, 0, sizeof buffer);

   /* two empty bson documents, length prefix indicates 5 bytes */
   buffer[0] = buffer[5] = 5;

   reader = bson_reader_new_from_data (buffer, sizeof buffer);

   BSON_ASSERT (bson_reader_read (reader, &eof)->len == 5 && !eof);
   BSON_ASSERT (bson_reader_read (reader, &eof)->len == 5 && !eof);
   BSON_ASSERT (!bson_reader_read (reader, &eof) && eof);
   BSON_ASSERT (bson_reader_tell (reader) == 10);

   bson_reader_reset (reader);

   BSON_ASSERT (bson_reader_tell (reader) == 0);
   BSON_ASSERT (bson_reader_read (reader, &eof)->len == 5 && !eof);
   BSON_ASSERT (bson_reader_read (reader, &eof)->len == 5 && !eof);
   BSON_ASSERT (!bson_reader_read (reader, &eof) && eof);
   BSON_ASSERT (bson_reader_tell (reader) == 10);

   bson_reader_destroy (reader);
}


void
test_reader_install (TestSuite *suite)
{
   TestSuite_Add (suite, "/bson/reader/new_from_data", test_reader_from_data);
   TestSuite_Add (suite, "/bson/reader/new_from_data_overflow", test_reader_from_data_overflow);
   TestSuite_Add (
      suite, "/bson/reader/new_from_data_document_length_too_large", test_reader_from_data_document_length_too_large);
   TestSuite_Add (
      suite, "/bson/reader/new_from_data_document_length_too_small", test_reader_from_data_document_length_too_small);
   TestSuite_Add (suite, "/bson/reader/new_from_handle", test_reader_from_handle);
   TestSuite_Add (suite, "/bson/reader/tell", test_reader_tell);
   TestSuite_Add (suite, "/bson/reader/new_from_handle_corrupt", test_reader_from_handle_corrupt);
   TestSuite_Add (suite, "/bson/reader/grow_buffer", test_reader_grow_buffer);
   TestSuite_Add (suite, "/bson/reader/reset", test_reader_reset);
}
