/*
 * Embedded Linux library
 * Copyright (C) 2015  Intel Corporation
 *
 * SPDX-License-Identifier: LGPL-2.1-or-later
 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif

#define _GNU_SOURCE
#include <stdio.h>
#include <string.h>
#include <stdint.h>
#include <assert.h>

#include <ell/ell.h>

#include "ell/useful.h"

static unsigned int nlpo2(unsigned int x)
{
	x--;
	x |= (x >> 1);
	x |= (x >> 2);
	x |= (x >> 4);
	x |= (x >> 8);
	x |= (x >> 16);
	return x + 1;
}

static unsigned int fls(unsigned int x)
{
	return x ? sizeof(x) * 8 - __builtin_clz(x) : 0;
}

static unsigned int align_power2(unsigned int u)
{
	return 1 << fls(u - 1);
}

static void test_power2(const void *data)
{
	size_t i;

	for (i = 1; i < 1000000; i++) {
		size_t size1, size2, size3 = 1;

		size1 = nlpo2(i);
		size2 = align_power2(i);

		/* Find the next power of two */
		while (size3 < i && size3 < SIZE_MAX)
			size3 <<= 1;

		assert(size1 == size2);
		assert(size2 == size3);
		assert(size3 == size1);
	}
}

static void test_alloc(const void *data)
{
	int i;

	for (i = 2; i < 10000; i++) {
		struct l_ringbuf *rb;

		rb = l_ringbuf_new(i);
		assert(rb != NULL);

		assert(l_ringbuf_capacity(rb) == l_ringbuf_avail(rb));

		l_ringbuf_free(rb);
	}
}

static void test_printf(const void *data)
{
	static size_t rb_size = 500;
	static size_t rb_capa = 512;
	struct l_ringbuf *rb;
	int i;

	rb = l_ringbuf_new(rb_size);
	assert(rb != NULL);
	assert(l_ringbuf_capacity(rb) == rb_capa);

	for (i = 0; i < 10000; i++) {
		size_t len, count = i % rb_capa;
		char *str, *ptr;

		if (!count)
			continue;

		len = asprintf(&str, "%*c", (int) count, 'x');
		assert(len == count);

		len = l_ringbuf_printf(rb, "%s", str);
		assert(len == count);
		assert(l_ringbuf_len(rb) == count);
		assert(l_ringbuf_avail(rb) == rb_capa - len);

		ptr = l_ringbuf_peek(rb, 0, &len);
		assert(ptr != NULL);
		assert(len == count);
		assert(strncmp(str, ptr, len) == 0);

		len = l_ringbuf_drain(rb, count);
		assert(len == count);
		assert(l_ringbuf_len(rb) == 0);
		assert(l_ringbuf_avail(rb) == rb_capa);

		free(str);
	}

	l_ringbuf_free(rb);
}

static void test_append(const void *unused)
{
	static const uint8_t data[6] = { 1, 2, 3, 4, 5, 6 };
	static const size_t rb_size = 12;
	static const size_t rb_capa = 16;
	size_t len_no_wrap;
	ssize_t appended;
	ssize_t space_left;
	void *rb_data;
	struct l_ringbuf *rb;

	rb = l_ringbuf_new(rb_size);
	assert(rb != NULL);
	assert(l_ringbuf_capacity(rb) == rb_capa);

	appended = l_ringbuf_append(rb, data, sizeof(data));
	assert(appended == sizeof(data));

	appended = l_ringbuf_append(rb, data, sizeof(data));
	assert(appended == sizeof(data));

	space_left = minsize(l_ringbuf_avail(rb), sizeof(data));
	appended = l_ringbuf_append(rb, data, sizeof(data));
	assert(appended == space_left);

	rb_data = l_ringbuf_peek(rb, 0, &len_no_wrap);
	assert(memcmp(rb_data, data, sizeof(data)) == 0);
	l_ringbuf_drain(rb, sizeof(data));

	rb_data = l_ringbuf_peek(rb, 0, &len_no_wrap);
	assert(memcmp(rb_data, data, sizeof(data)) == 0);
	l_ringbuf_drain(rb, sizeof(data));

	rb_data = l_ringbuf_peek(rb, 0, &len_no_wrap);
	assert(memcmp(rb_data, data, len_no_wrap) == 0);
	l_ringbuf_drain(rb, len_no_wrap);

	l_ringbuf_free(rb);
}

static void test_append2(const void *unused)
{
	static const uint8_t expected_data1[14] =  {3, 4, 5, 6, 1, 2, 3, 4, 5,
							6, 1, 2, 3, 4 };
	static const uint8_t expected_data2[2] = { 5, 6 };
	static const uint8_t data[6] = { 1, 2, 3, 4, 5, 6 };
	static const size_t rb_size = 12;
	static const size_t rb_capa = 16;
	size_t len_no_wrap;
	ssize_t appended;
	ssize_t space_left;
	void *rb_data;
	struct l_ringbuf *rb;

	rb = l_ringbuf_new(rb_size);
	assert(rb != NULL);
	assert(l_ringbuf_capacity(rb) == rb_capa);

	appended = l_ringbuf_append(rb, data, sizeof(data));
	assert(appended == sizeof(data));

	appended = l_ringbuf_append(rb, data, sizeof(data));
	assert(appended == sizeof(data));

	space_left = minsize(l_ringbuf_avail(rb), sizeof(data));
	l_ringbuf_drain(rb, sizeof(data) - space_left);

	appended = l_ringbuf_append(rb, data, sizeof(data));
	assert(appended == sizeof(data));

	rb_data = l_ringbuf_peek(rb, 0, &len_no_wrap);
	assert(len_no_wrap == sizeof(expected_data1));
	assert(memcmp(expected_data1, rb_data, len_no_wrap) == 0);

	l_ringbuf_drain(rb, len_no_wrap);

	rb_data = l_ringbuf_peek(rb, 0, &len_no_wrap);
	assert(len_no_wrap == sizeof(expected_data2));
	assert(memcmp(expected_data2, rb_data, len_no_wrap) == 0);

	l_ringbuf_free(rb);
}

int main(int argc, char *argv[])
{
	l_test_init(&argc, &argv);

	l_test_add("/ringbuf/power2", test_power2, NULL);
	l_test_add("/ringbuf/alloc", test_alloc, NULL);
	l_test_add("/ringbuf/printf", test_printf, NULL);
	l_test_add("/ringbuf/append", test_append, NULL);
	l_test_add("/ringbuf/append2", test_append2, NULL);

	return l_test_run();
}
