/****************************************************************************
 *
 *   Copyright (C) 2012 PX4 Development Team. All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 * 3. Neither the name PX4 nor the names of its contributors may be
 *    used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
 * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
 * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
 * OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
 * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 ****************************************************************************/

/**
 * @file tests_file.c
 *
 * File write test.
 */

#include <sys/stat.h>
#include <dirent.h>
#include <stdio.h>
#include <stddef.h>
#include <unistd.h>
#include <fcntl.h>
#include <systemlib/err.h>
#include <systemlib/perf_counter.h>
#include <string.h>

#include <drivers/drv_hrt.h>

#include "tests.h"

int
test_file(int argc, char *argv[])
{
	const iterations = 200;

	/* check if microSD card is mounted */
	struct stat buffer;
	if (stat("/fs/microsd/", &buffer)) {
		warnx("no microSD card mounted, aborting file test");
		return 1;
	}

	uint8_t buf[512];
	hrt_abstime start, end;
	perf_counter_t wperf = perf_alloc(PC_ELAPSED, "SD writes (aligned)");

	int fd = open("/fs/microsd/testfile", O_TRUNC | O_WRONLY | O_CREAT);
	memset(buf, 0, sizeof(buf));

	warnx("aligned write - please wait..");

	if ((0x3 & (uintptr_t)buf))
		warnx("memory is unaligned!");

	start = hrt_absolute_time();
	for (unsigned i = 0; i < iterations; i++) {
		perf_begin(wperf);
		write(fd, buf, sizeof(buf));
		fsync(fd);
		perf_end(wperf);
	}
	end = hrt_absolute_time();

	warnx("%dKiB in %llu microseconds", iterations / 2, end - start);

	perf_print_counter(wperf);
	perf_free(wperf);

	int ret = unlink("/fs/microsd/testfile");

	if (ret)
		err(1, "UNLINKING FILE FAILED");

	warnx("unaligned write - please wait..");

	struct {
		uint8_t byte;
		uint8_t unaligned[512];
	} unaligned_buf;

	if ((0x3 & (uintptr_t)unaligned_buf.unaligned) == 0)
		warnx("creating unaligned memory failed.");

	wperf = perf_alloc(PC_ELAPSED, "SD writes (unaligned)");

	start = hrt_absolute_time();
	for (unsigned i = 0; i < iterations; i++) {
		perf_begin(wperf);
		write(fd, unaligned_buf.unaligned, sizeof(unaligned_buf.unaligned));
		fsync(fd);
		perf_end(wperf);
	}
	end = hrt_absolute_time();

	close(fd);

	warnx("%dKiB in %llu microseconds", iterations / 2, end - start);

	perf_print_counter(wperf);
	perf_free(wperf);

	/* list directory */
	DIR           *d;
	struct dirent *dir;
	d = opendir("/fs/microsd");
	if (d) {

		while ((dir = readdir(d)) != NULL) {
			//printf("%s\n", dir->d_name);
		}

		closedir(d);

		warnx("directory listing ok (FS mounted and readable)");

	} else {
		/* failed opening dir */
		err(1, "FAILED LISTING MICROSD ROOT DIRECTORY");
	}

	return 0;
}
#if 0
int
test_file(int argc, char *argv[])
{
	const iterations = 1024;

	/* check if microSD card is mounted */
	struct stat buffer;
	if (stat("/fs/microsd/", &buffer)) {
		warnx("no microSD card mounted, aborting file test");
		return 1;
	}

	uint8_t buf[512];
	hrt_abstime start, end;
	perf_counter_t wperf = perf_alloc(PC_ELAPSED, "SD writes");

	int fd = open("/fs/microsd/testfile", O_TRUNC | O_WRONLY | O_CREAT);
	memset(buf, 0, sizeof(buf));

	start = hrt_absolute_time();
	for (unsigned i = 0; i < iterations; i++) {
		perf_begin(wperf);
		write(fd, buf, sizeof(buf));
		perf_end(wperf);
	}
	end = hrt_absolute_time();

	close(fd);

	unlink("/fs/microsd/testfile");

	warnx("%dKiB in %llu microseconds", iterations / 2, end - start);
	perf_print_counter(wperf);
	perf_free(wperf);

	warnx("running unlink test");

	/* ensure that common commands do not run against file count limits */
	for (unsigned i = 0; i < 64; i++) {

		warnx("unlink iteration #%u", i);

		int fd = open("/fs/microsd/testfile", O_TRUNC | O_WRONLY | O_CREAT);
		if (fd < 0)
			errx(1, "failed opening test file before unlink()");
		int ret = write(fd, buf, sizeof(buf));
		if (ret < 0)
			errx(1, "failed writing test file before unlink()");
		close(fd);

		ret = unlink("/fs/microsd/testfile");
		if (ret != OK)
			errx(1, "failed unlinking test file");

		fd = open("/fs/microsd/testfile", O_TRUNC | O_WRONLY | O_CREAT);
		if (fd < 0)
			errx(1, "failed opening test file after unlink()");
		ret = write(fd, buf, sizeof(buf));
		if (ret < 0)
			errx(1, "failed writing test file after unlink()");
		close(fd);
	}

	return 0;
}
#endif