#include <string.h>
#include <stdlib.h>
#include <stdint.h>
#include "test.h"

#define N 400
static char buf[N];
static char buf2[N];

static void *(*volatile pmemset)(void *, int, size_t);

static char *aligned(void *p)
{
	return (char*)(((uintptr_t)p + 63) & -64U);
}

static void test_align(int align, int len)
{
	char *s = aligned(buf+64) + align;
	char *want = aligned(buf2+64) + align;
	char *p;
	int i;

	if (len + 64 > buf + N - s || len + 64 > buf2 + N - want)
		abort();
	for (i = 0; i < N; i++)
		buf[i] = buf2[i] = ' ';
	for (i = 0; i < len; i++)
		want[i] = '#';
	p = pmemset(s, '#', len);
	if (p != s)
		t_error("memset(%p,...) returned %p\n", s, p);
	for (i = -64; i < len+64; i++)
		if (s[i] != want[i]) {
			t_error("memset(align %d, '#', %d) failed at pos %d\n", align, len, i);
			t_printf("got : '%.*s'\n", len+128, s-64);
			t_printf("want: '%.*s'\n", len+128, want-64);
			break;
		}
}

static void test_value(int c)
{
	int i;

	pmemset(buf, c, 10);
	for (i = 0; i < 10; i++)
		if ((unsigned char)buf[i] != (unsigned char)c) {
			t_error("memset(%d) failed: got %d\n", c, buf[i]);
			break;
		}
}

int main(void)
{
	int i,j,k;

	pmemset = memset;

	for (i = 0; i < 16; i++)
		for (j = 0; j < 200; j++)
			test_align(i,j);

	test_value('c');
	test_value(0);
	test_value(-1);
	test_value(-5);
	test_value(0xab);
	return t_status;
}