[fio.git] / filesetup.c

#include <unistd.h>
#include <fcntl.h>
#include <string.h>
#include <assert.h>
#include <dirent.h>
#include <sys/stat.h>
#include <sys/mman.h>
#include <sys/types.h>

#include "fio.h"
#include "os.h"

/*
 * Check if the file exists and it's large enough.
 */
static int file_ok(struct thread_data *td, struct fio_file *f)
{
	struct stat st;

	if (f->filetype != FIO_TYPE_FILE ||
	    (td->io_ops->flags & FIO_DISKLESSIO))
		return 0;

	if (lstat(f->file_name, &st) == -1)
		return 1;

	/*
	 * if it's a special file, size is always ok for now
	 */
	if (S_ISCHR(st.st_mode) || S_ISBLK(st.st_mode))
		return 0;
	if (st.st_size < (off_t) f->file_size)
		return 1;

	return 0;
}

static int create_file(struct thread_data *td, struct fio_file *f)
{
	unsigned long long left;
	unsigned int bs;
	char *b;
	int r;

	f->fd = open(f->file_name, O_WRONLY | O_CREAT | O_TRUNC, 0644);
	if (f->fd < 0) {
		td_verror(td, errno, "open");
		return 1;
	}

	if (ftruncate(f->fd, f->file_size) == -1) {
		td_verror(td, errno, "ftruncate");
		goto err;
	}

	if (posix_fallocate(f->fd, 0, f->file_size) < 0) {
		td_verror(td, errno, "posix_fallocate");
		goto err;
	}

	b = malloc(td->o.max_bs[DDIR_WRITE]);
	memset(b, 0, td->o.max_bs[DDIR_WRITE]);

	left = f->file_size;
	while (left && !td->terminate) {
		bs = td->o.max_bs[DDIR_WRITE];
		if (bs > left)
			bs = left;

		r = write(f->fd, b, bs);

		if (r == (int) bs) {
			left -= bs;
			continue;
		} else {
			if (r < 0)
				td_verror(td, errno, "write");
			else
				td_verror(td, EIO, "write");

			break;
		}
	}

	if (td->terminate)
		unlink(f->file_name);
	else if (td->o.create_fsync)
		fsync(f->fd);

	free(b);
	close(f->fd);
	f->fd = -1;
	return 0;
err:
	close(f->fd);
	f->fd = -1;
	return 1;
}

static unsigned long long set_rand_file_size(struct thread_data *td,
					     unsigned long long total_size)
{
	unsigned long long upper = total_size;
	unsigned long long ret;
	long r;

	if (upper > td->o.file_size_high)
		upper = td->o.file_size_high;
	else if (upper < td->o.file_size_low)
		return 0;
	else if (!upper)
		return 0;

	r = os_random_long(&td->file_size_state);
	ret = td->o.file_size_low + (unsigned long long) ((double) upper * (r / (RAND_MAX + 1.0)));
	ret -= (ret % td->o.rw_min_bs);
	if (ret > upper)
		ret = upper;
	return ret;
}

static int fill_file_size(struct thread_data *td, struct fio_file *f,
			  unsigned long long *file_size, int new_files)
{
	if (!td->o.file_size_low) {
		f->file_size = *file_size / new_files;
		f->real_file_size = f->file_size;
	} else {
		/*
		 * If we don't have enough space left for a file
		 * of the minimum size, bail.
		 */
		if (*file_size < td->o.file_size_low)
			return 1;

		f->file_size = set_rand_file_size(td, *file_size);
		f->real_file_size = f->file_size;
		*file_size -= f->file_size;
	}

	return 0;
}

static int create_files(struct thread_data *td)
{
	struct fio_file *f;
	int err, need_create, can_extend;
	unsigned long long total_file_size, local_file_size, create_size;
	unsigned int i, new_files;

	new_files = 0;
	total_file_size = td->o.size;
	for_each_file(td, f, i) {
		unsigned long long s;

		f->file_offset = td->o.start_offset;

		if (!total_file_size)
			continue;

		if (f->filetype != FIO_TYPE_FILE) {
			if (!f->file_size)
				f->file_size = total_file_size / td->o.nr_files;
			continue;
		}

		if (f->flags & FIO_FILE_EXISTS) {
			if ((f->file_size > td->o.size / td->o.nr_files) ||
			    !f->file_size)
				f->file_size = td->o.size / td->o.nr_files;

			s = f->file_size;
			if (s > total_file_size)
				s = total_file_size;

			total_file_size -= s;
		} else
			new_files++;
	}

	/*
	 * unless specifically asked for overwrite, let normal io extend it
	 */
	can_extend = !td->o.overwrite && !(td->io_ops->flags & FIO_NOEXTEND);
	if (can_extend && new_files) {
		for_each_file(td, f, i) {
			if (fill_file_size(td, f, &total_file_size, new_files)) {
				log_info("fio: limited to %d files\n", i);
				td->o.nr_files = i;
				break;
			}
		}

		return 0;
	}

	local_file_size = total_file_size;
	if (!local_file_size)
		local_file_size = -1;

	total_file_size = 0;
	need_create = 0;
	create_size = 0;
	for_each_file(td, f, i) {
		int file_there;

		if (f->filetype != FIO_TYPE_FILE)
			continue;
		if (f->flags & FIO_FILE_EXISTS) {
			total_file_size += f->file_size;
			continue;
		}

		if (fill_file_size(td, f, &local_file_size, new_files)) {
			log_info("fio: limited to %d files\n", i);
			new_files -= (td->o.nr_files - i);
			td->o.nr_files = i;
			break;
		}

		total_file_size += f->file_size;
		create_size += f->file_size;
		file_there = !file_ok(td, f);

		if (file_there && td_write(td) && !td->o.overwrite) {
			unlink(f->file_name);
			file_there = 0;
		}

		need_create += !file_there;
	}

	if (!need_create)
		return 0;

	if (!td->o.size && !total_file_size) {
		log_err("Need size for create\n");
		td_verror(td, EINVAL, "file_size");
		return 1;
	}

	temp_stall_ts = 1;
	log_info("%s: Laying out IO file(s) (%u files / %LuMiB)\n",
				td->o.name, new_files, create_size >> 20);

	err = 0;
	for_each_file(td, f, i) {
		/*
		 * Only unlink files that we created.
		 */
		f->flags &= ~FIO_FILE_UNLINK;
		if (file_ok(td, f)) {
			if (td->o.unlink)
				f->flags |= FIO_FILE_UNLINK;

			f->flags |= FIO_FILE_NOSORT;
			err = create_file(td, f);
			if (err)
				break;
		}
	}

	temp_stall_ts = 0;
	return err;
}

static int file_size(struct thread_data *td, struct fio_file *f)
{
	struct stat st;

	if (td->o.overwrite) {
		if (fstat(f->fd, &st) == -1) {
			td_verror(td, errno, "fstat");
			return 1;
		}

		f->real_file_size = st.st_size;

		if (!f->file_size || f->file_size > f->real_file_size)
			f->file_size = f->real_file_size;
	} else
		f->real_file_size = f->file_size;

	return 0;
}

static int bdev_size(struct thread_data *td, struct fio_file *f)
{
	unsigned long long bytes;
	int r;

	r = blockdev_size(f->fd, &bytes);
	if (r) {
		td_verror(td, r, "blockdev_size");
		return 1;
	}

	f->real_file_size = bytes;

	/*
	 * no extend possibilities, so limit size to device size if too large
	 */
	if (!f->file_size || f->file_size > f->real_file_size)
		f->file_size = f->real_file_size;

	f->file_size -= f->file_offset;
	return 0;
}

static int get_file_size(struct thread_data *td, struct fio_file *f)
{
	int ret = 0;

	if (f->filetype == FIO_TYPE_FILE) {
		if (!(f->flags & FIO_FILE_EXISTS))
			ret = file_size(td, f);
	} else if (f->filetype == FIO_TYPE_BD)
		ret = bdev_size(td, f);
	else
		f->real_file_size = -1;

	if (ret)
		return ret;

	if (f->file_offset > f->real_file_size) {
		log_err("%s: offset extends end (%Lu > %Lu)\n", td->o.name, f->file_offset, f->real_file_size);
		return 1;
	}

	return 0;
}

int file_invalidate_cache(struct thread_data *td, struct fio_file *f)
{
	int ret = 0;

	if (td->o.odirect)
		return 0;

	/*
	 * FIXME: add blockdev flushing too
	 */
	if (f->mmap)
		ret = madvise(f->mmap, f->file_size, MADV_DONTNEED);
	else if (f->filetype == FIO_TYPE_FILE)
		ret = fadvise(f->fd, f->file_offset, f->file_size, POSIX_FADV_DONTNEED);
	else if (f->filetype == FIO_TYPE_BD) {
		ret = blockdev_invalidate_cache(f->fd);
		if (ret < 0 && errno == EACCES && geteuid()) {
			log_err("fio: only root may flush block devices. Cache flush bypassed!\n");
			ret = 0;
		}
	} else if (f->filetype == FIO_TYPE_CHAR)
		ret = 0;

	if (ret < 0) {
		td_verror(td, errno, "invalidate_cache");
		return 1;
	}

	return ret;
}

void generic_close_file(struct thread_data fio_unused *td, struct fio_file *f)
{
	close(f->fd);
	f->fd = -1;
}

int generic_open_file(struct thread_data *td, struct fio_file *f)
{
	int flags = 0;

	if (td->o.odirect)
		flags |= OS_O_DIRECT;
	if (td->o.sync_io)
		flags |= O_SYNC;

	if (td_write(td) || td_rw(td)) {
		flags |= O_RDWR;

		if (f->filetype == FIO_TYPE_FILE)
			flags |= O_CREAT;

		f->fd = open(f->file_name, flags, 0600);
	} else {
		if (f->filetype == FIO_TYPE_CHAR)
			flags |= O_RDWR;
		else
			flags |= O_RDONLY;

		f->fd = open(f->file_name, flags);
	}

	if (f->fd == -1) {
		char buf[FIO_VERROR_SIZE];
		int __e = errno;

		snprintf(buf, sizeof(buf) - 1, "open(%s)", f->file_name);

		td_verror(td, __e, buf);
		if (__e == EINVAL && td->o.odirect)
			log_err("fio: destination does not support O_DIRECT\n");
		if (__e == EMFILE)
			log_err("fio: try reducing/setting openfiles (failed at %u of %u)\n", td->nr_open_files, td->o.nr_files);
		return 1;
	}

	if (get_file_size(td, f))
		goto err;

	if (td->o.invalidate_cache && file_invalidate_cache(td, f))
		goto err;

	if (td->o.fadvise_hint) {
		if (td_random(td))
			flags = POSIX_FADV_RANDOM;
		else
			flags = POSIX_FADV_SEQUENTIAL;

		if (fadvise(f->fd, f->file_offset, f->file_size, flags) < 0) {
			td_verror(td, errno, "fadvise");
			goto err;
		}
	}

	return 0;
err:
	close(f->fd);
	return 1;
}

int open_files(struct thread_data *td)
{
	struct fio_file *f;
	unsigned int i;
	int err = 0;

	for_each_file(td, f, i) {
		err = td_io_open_file(td, f);
		if (err)
			break;

		if (td->o.open_files == td->nr_open_files)
			break;
	}

	if (!err)
		return 0;

	for_each_file(td, f, i)
		td_io_close_file(td, f);

	return err;
}

int setup_files(struct thread_data *td)
{
	struct fio_file *f;
	unsigned int i;
	int err;

	/*
	 * if ioengine defines a setup() method, it's responsible for
	 * setting up everything in the td->files[] area.
	 */
	if (td->io_ops->setup)
		return td->io_ops->setup(td);

	if (create_files(td))
		return 1;

	err = open_files(td);
	if (err)
		return err;

	/*
	 * Recalculate the total file size now that files are set up.
	 */
	td->o.size = 0;
	for_each_file(td, f, i)
		td->o.size += f->file_size;

	td->io_size = td->o.size;
	if (td->io_size == 0) {
		log_err("%s: no io blocks\n", td->o.name);
		td_verror(td, EINVAL, "total_file_size");
		return 1;
	}

	if (!td->o.zone_size)
		td->o.zone_size = td->io_size;

	td->total_io_size = td->io_size * td->o.loops;

	for_each_file(td, f, i)
		td_io_close_file(td, f);

	return err;
}

int init_random_map(struct thread_data *td)
{
	int num_maps, blocks;
	struct fio_file *f;
	unsigned int i;

	if (td->o.norandommap)
		return 0;

	for_each_file(td, f, i) {
		blocks = (f->real_file_size + td->o.rw_min_bs - 1) / td->o.rw_min_bs;
		num_maps = (blocks + BLOCKS_PER_MAP-1)/ BLOCKS_PER_MAP;
		f->file_map = malloc(num_maps * sizeof(long));
		if (!f->file_map) {
			log_err("fio: failed allocating random map. If running a large number of jobs, try the 'norandommap' option\n");
			return 1;
		}
		f->num_maps = num_maps;
		memset(f->file_map, 0, num_maps * sizeof(long));
	}

	return 0;
}

void close_files(struct thread_data *td)
{
	struct fio_file *f;
	unsigned int i;

	for_each_file(td, f, i) {
		if ((f->flags & FIO_FILE_UNLINK) &&
		    f->filetype == FIO_TYPE_FILE)
			unlink(f->file_name);

		td_io_close_file(td, f);

		free(f->file_name);
		f->file_name = NULL;

		if (f->file_map) {
			free(f->file_map);
			f->file_map = NULL;
		}
	}

	td->o.filename = NULL;
	free(td->files);
	td->files = NULL;
	td->o.nr_files = 0;
}

static void get_file_type(struct fio_file *f)
{
	struct stat sb;

	f->filetype = FIO_TYPE_FILE;

	if (!lstat(f->file_name, &sb)) {
		f->flags |= FIO_FILE_EXISTS;

		if (S_ISBLK(sb.st_mode))
			f->filetype = FIO_TYPE_BD;
		else if (S_ISCHR(sb.st_mode))
			f->filetype = FIO_TYPE_CHAR;
		else {
			/*
			 * might as well do this here, and save a stat later on
			 */
			f->real_file_size = sb.st_size;
			f->file_size = f->real_file_size;
		}
	}
}

void add_file(struct thread_data *td, const char *fname)
{
	int cur_files = td->files_index;
	char file_name[PATH_MAX];
	struct fio_file *f;
	int len = 0;

	td->files = realloc(td->files, (cur_files + 1) * sizeof(*f));

	f = &td->files[cur_files];
	memset(f, 0, sizeof(*f));
	f->fd = -1;

	if (td->o.directory)
		len = sprintf(file_name, "%s/", td->o.directory);

	sprintf(file_name + len, "%s", fname);
	f->file_name = strdup(file_name);

	get_file_type(f);

	td->files_index++;
	if (f->filetype == FIO_TYPE_FILE)
		td->nr_normal_files++;
}

void get_file(struct fio_file *f)
{
	f->references++;
}

void put_file(struct thread_data *td, struct fio_file *f)
{
	if (!(f->flags & FIO_FILE_OPEN))
		return;

	assert(f->references);
	if (--f->references)
		return;

	if (should_fsync(td) && td->o.fsync_on_close)
		fsync(f->fd);

	if (td->io_ops->close_file)
		td->io_ops->close_file(td, f);
	td->nr_open_files--;
	f->flags &= ~FIO_FILE_OPEN;
}

static int recurse_dir(struct thread_data *td, const char *dirname)
{
	struct dirent *dir;
	int ret = 0;
	DIR *D;

	D = opendir(dirname);
	if (!D) {
		td_verror(td, errno, "opendir");
		return 1;
	}

	while ((dir = readdir(D)) != NULL) {
		char full_path[PATH_MAX];
		struct stat sb;

		if (!strcmp(dir->d_name, ".") || !strcmp(dir->d_name, ".."))
			continue;

		sprintf(full_path, "%s/%s", dirname, dir->d_name);

		if (lstat(full_path, &sb) == -1) {
			if (errno != ENOENT) {
				td_verror(td, errno, "stat");
				return 1;
			}
		}

		if (S_ISREG(sb.st_mode)) {
			add_file(td, full_path);
			td->o.nr_files++;
			continue;
		}

		if ((ret = recurse_dir(td, full_path)) != 0)
			break;
	}

	closedir(D);
	return ret;
}

int add_dir_files(struct thread_data *td, const char *path)
{
	return recurse_dir(td, path);
}

void dup_files(struct thread_data *td, struct thread_data *org)
{
	struct fio_file *f;
	unsigned int i;
	size_t bytes;

	if (!org->files)
		return;

	bytes = org->files_index * sizeof(*f);
	td->files = malloc(bytes);
	memcpy(td->files, org->files, bytes);

	for_each_file(td, f, i) {
		if (f->file_name)
			f->file_name = strdup(f->file_name);
	}
}
Commit	Line	Data
53cdc686 JA	1	#include <unistd.h>
	2	#include <fcntl.h>
	3	#include <string.h>
	4	#include <assert.h>
bbf6b540	5	#include <dirent.h>
53cdc686 JA	6	#include <sys/stat.h>
53cdc686 JA	7	#include <sys/mman.h>
bbf6b540	8	#include <sys/types.h>
53cdc686 JA	9
	10	#include "fio.h"
	11	#include "os.h"
	12
25205e97 JA	13	/*
	14	* Check if the file exists and it's large enough.
	15	*/
	16	static int file_ok(struct thread_data td, struct fio_file f)
53cdc686	17	{
b2a15192	18	struct stat st;
53cdc686	19
af52b345	20	if (f->filetype != FIO_TYPE_FILE \|\|
b5af8293	21	(td->io_ops->flags & FIO_DISKLESSIO))
b2a15192 JA	22	return 0;
b2a15192 JA	23
fa01d139	24	if (lstat(f->file_name, &st) == -1)
25205e97	25	return 1;
fa01d139 JA	26
	27	/*
	28	* if it's a special file, size is always ok for now
	29	*/
	30	if (S_ISCHR(st.st_mode) \|\| S_ISBLK(st.st_mode))
	31	return 0;
	32	if (st.st_size < (off_t) f->file_size)
25205e97 JA	33	return 1;
	34
	35	return 0;
	36	}
	37
	38	static int create_file(struct thread_data td, struct fio_file f)
	39	{
	40	unsigned long long left;
	41	unsigned int bs;
	42	char *b;
	43	int r;
b2a15192	44
53cdc686 JA	45	f->fd = open(f->file_name, O_WRONLY \| O_CREAT \| O_TRUNC, 0644);
53cdc686 JA	46	if (f->fd < 0) {
e1161c32	47	td_verror(td, errno, "open");
53cdc686 JA	48	return 1;
	49	}
	50
	51	if (ftruncate(f->fd, f->file_size) == -1) {
e1161c32	52	td_verror(td, errno, "ftruncate");
53cdc686 JA	53	goto err;
	54	}
	55
40f8298c	56	if (posix_fallocate(f->fd, 0, f->file_size) < 0) {
e1161c32	57	td_verror(td, errno, "posix_fallocate");
40f8298c JA	58	goto err;
	59	}
	60
2dc1bbeb JA	61	b = malloc(td->o.max_bs[DDIR_WRITE]);
2dc1bbeb JA	62	memset(b, 0, td->o.max_bs[DDIR_WRITE]);
53cdc686 JA	63
	64	left = f->file_size;
	65	while (left && !td->terminate) {
2dc1bbeb	66	bs = td->o.max_bs[DDIR_WRITE];
53cdc686 JA	67	if (bs > left)
	68	bs = left;
	69
	70	r = write(f->fd, b, bs);
	71
	72	if (r == (int) bs) {
	73	left -= bs;
	74	continue;
	75	} else {
	76	if (r < 0)
e1161c32	77	td_verror(td, errno, "write");
53cdc686	78	else
e1161c32	79	td_verror(td, EIO, "write");
53cdc686 JA	80
	81	break;
	82	}
	83	}
	84
	85	if (td->terminate)
	86	unlink(f->file_name);
2dc1bbeb	87	else if (td->o.create_fsync)
53cdc686 JA	88	fsync(f->fd);
	89
	90	free(b);
	91	close(f->fd);
	92	f->fd = -1;
	93	return 0;
	94	err:
	95	close(f->fd);
	96	f->fd = -1;
	97	return 1;
	98	}
	99
9c60ce64 JA	100	static unsigned long long set_rand_file_size(struct thread_data *td,
	101	unsigned long long total_size)
	102	{
	103	unsigned long long upper = total_size;
	104	unsigned long long ret;
	105	long r;
	106
2dc1bbeb JA	107	if (upper > td->o.file_size_high)
	108	upper = td->o.file_size_high;
	109	else if (upper < td->o.file_size_low)
9c60ce64 JA	110	return 0;
	111	else if (!upper)
	112	return 0;
	113
	114	r = os_random_long(&td->file_size_state);
2dc1bbeb JA	115	ret = td->o.file_size_low + (unsigned long long) ((double) upper * (r / (RAND_MAX + 1.0)));
2dc1bbeb JA	116	ret -= (ret % td->o.rw_min_bs);
9c60ce64 JA	117	if (ret > upper)
	118	ret = upper;
	119	return ret;
	120	}
	121
68727076 JA	122	static int fill_file_size(struct thread_data td, struct fio_file f,
	123	unsigned long long *file_size, int new_files)
	124	{
	125	if (!td->o.file_size_low) {
	126	f->file_size = *file_size / new_files;
	127	f->real_file_size = f->file_size;
	128	} else {
	129	/*
	130	* If we don't have enough space left for a file
	131	* of the minimum size, bail.
	132	*/
	133	if (*file_size < td->o.file_size_low)
	134	return 1;
	135
	136	f->file_size = set_rand_file_size(td, *file_size);
	137	f->real_file_size = f->file_size;
	138	*file_size -= f->file_size;
	139	}
	140
	141	return 0;
	142	}
	143
53cdc686 JA	144	static int create_files(struct thread_data *td)
	145	{
	146	struct fio_file *f;
af52b345	147	int err, need_create, can_extend;
e8be395b	148	unsigned long long total_file_size, local_file_size, create_size;
e407bec2	149	unsigned int i, new_files;
53cdc686	150
e407bec2	151	new_files = 0;
2dc1bbeb	152	total_file_size = td->o.size;
1549441c	153	for_each_file(td, f, i) {
96d32d51 JA	154	unsigned long long s;
96d32d51 JA	155
2dc1bbeb	156	f->file_offset = td->o.start_offset;
96d32d51	157
ee3dcd95 JA	158	if (!total_file_size)
ee3dcd95 JA	159	continue;
1549441c	160
1d56dfee JA	161	if (f->filetype != FIO_TYPE_FILE) {
	162	if (!f->file_size)
	163	f->file_size = total_file_size / td->o.nr_files;
	164	continue;
	165	}
	166
96d32d51	167	if (f->flags & FIO_FILE_EXISTS) {
ee3dcd95 JA	168	if ((f->file_size > td->o.size / td->o.nr_files) \|\|
	169	!f->file_size)
	170	f->file_size = td->o.size / td->o.nr_files;
	171
96d32d51 JA	172	s = f->file_size;
	173	if (s > total_file_size)
	174	s = total_file_size;
	175
	176	total_file_size -= s;
e407bec2 JA	177	} else
e407bec2 JA	178	new_files++;
1549441c	179	}
f697125a	180
53cdc686 JA	181	/*
	182	* unless specifically asked for overwrite, let normal io extend it
	183	*/
2dc1bbeb	184	can_extend = !td->o.overwrite && !(td->io_ops->flags & FIO_NOEXTEND);
3f344202	185	if (can_extend && new_files) {
68727076 JA	186	for_each_file(td, f, i) {
	187	if (fill_file_size(td, f, &total_file_size, new_files)) {
	188	log_info("fio: limited to %d files\n", i);
	189	td->o.nr_files = i;
	190	break;
	191	}
	192	}
	193
53cdc686	194	return 0;
68727076	195	}
53cdc686	196
9c60ce64 JA	197	local_file_size = total_file_size;
	198	if (!local_file_size)
	199	local_file_size = -1;
	200
f4c4f4d7 JA	201	total_file_size = 0;
f4c4f4d7 JA	202	need_create = 0;
e8be395b	203	create_size = 0;
af52b345 JA	204	for_each_file(td, f, i) {
	205	int file_there;
	206
	207	if (f->filetype != FIO_TYPE_FILE)
	208	continue;
f4c4f4d7 JA	209	if (f->flags & FIO_FILE_EXISTS) {
f4c4f4d7 JA	210	total_file_size += f->file_size;
96d32d51	211	continue;
f4c4f4d7	212	}
96d32d51	213
68727076 JA	214	if (fill_file_size(td, f, &local_file_size, new_files)) {
	215	log_info("fio: limited to %d files\n", i);
	216	new_files -= (td->o.nr_files - i);
	217	td->o.nr_files = i;
	218	break;
9c60ce64	219	}
eff6861d	220
f4c4f4d7	221	total_file_size += f->file_size;
e8be395b	222	create_size += f->file_size;
af52b345	223	file_there = !file_ok(td, f);
eff6861d	224
2dc1bbeb	225	if (file_there && td_write(td) && !td->o.overwrite) {
af52b345 JA	226	unlink(f->file_name);
af52b345 JA	227	file_there = 0;
eff6861d	228	}
af52b345 JA	229
af52b345 JA	230	need_create += !file_there;
eff6861d	231	}
25205e97	232
25205e97 JA	233	if (!need_create)
	234	return 0;
	235
2dc1bbeb	236	if (!td->o.size && !total_file_size) {
f1027063	237	log_err("Need size for create\n");
e1161c32	238	td_verror(td, EINVAL, "file_size");
f1027063 JA	239	return 1;
	240	}
	241
53cdc686	242	temp_stall_ts = 1;
6d86144d	243	log_info("%s: Laying out IO file(s) (%u files / %LuMiB)\n",
2dc1bbeb	244	td->o.name, new_files, create_size >> 20);
53cdc686 JA	245
	246	err = 0;
	247	for_each_file(td, f, i) {
9b031dc8 JA	248	/*
	249	* Only unlink files that we created.
	250	*/
f11bd94d	251	f->flags &= ~FIO_FILE_UNLINK;
25205e97	252	if (file_ok(td, f)) {
2dc1bbeb	253	if (td->o.unlink)
f11bd94d JA	254	f->flags \|= FIO_FILE_UNLINK;
f11bd94d JA	255
160b966d	256	f->flags \|= FIO_FILE_NOSORT;
25205e97 JA	257	err = create_file(td, f);
	258	if (err)
	259	break;
	260	}
53cdc686 JA	261	}
	262
	263	temp_stall_ts = 0;
	264	return err;
	265	}
	266
	267	static int file_size(struct thread_data td, struct fio_file f)
	268	{
	269	struct stat st;
	270
2dc1bbeb	271	if (td->o.overwrite) {
53cdc686	272	if (fstat(f->fd, &st) == -1) {
e1161c32	273	td_verror(td, errno, "fstat");
53cdc686 JA	274	return 1;
	275	}
	276
	277	f->real_file_size = st.st_size;
	278
	279	if (!f->file_size \|\| f->file_size > f->real_file_size)
	280	f->file_size = f->real_file_size;
745508dd JA	281	} else
745508dd JA	282	f->real_file_size = f->file_size;
53cdc686	283
53cdc686 JA	284	return 0;
	285	}
	286
	287	static int bdev_size(struct thread_data td, struct fio_file f)
	288	{
	289	unsigned long long bytes;
	290	int r;
	291
	292	r = blockdev_size(f->fd, &bytes);
	293	if (r) {
e1161c32	294	td_verror(td, r, "blockdev_size");
53cdc686 JA	295	return 1;
	296	}
	297
	298	f->real_file_size = bytes;
	299
	300	/*
	301	* no extend possibilities, so limit size to device size if too large
	302	*/
	303	if (!f->file_size \|\| f->file_size > f->real_file_size)
	304	f->file_size = f->real_file_size;
	305
	306	f->file_size -= f->file_offset;
	307	return 0;
	308	}
	309
	310	static int get_file_size(struct thread_data td, struct fio_file f)
	311	{
	312	int ret = 0;
	313
96d32d51 JA	314	if (f->filetype == FIO_TYPE_FILE) {
	315	if (!(f->flags & FIO_FILE_EXISTS))
	316	ret = file_size(td, f);
	317	} else if (f->filetype == FIO_TYPE_BD)
53cdc686 JA	318	ret = bdev_size(td, f);
	319	else
	320	f->real_file_size = -1;
	321
	322	if (ret)
	323	return ret;
	324
	325	if (f->file_offset > f->real_file_size) {
2dc1bbeb	326	log_err("%s: offset extends end (%Lu > %Lu)\n", td->o.name, f->file_offset, f->real_file_size);
53cdc686 JA	327	return 1;
	328	}
	329
53cdc686 JA	330	return 0;
	331	}
	332
e5b401d4 JA	333	int file_invalidate_cache(struct thread_data td, struct fio_file f)
	334	{
	335	int ret = 0;
	336
2dc1bbeb	337	if (td->o.odirect)
b5af8293 JA	338	return 0;
b5af8293 JA	339
e5b401d4 JA	340	/*
	341	* FIXME: add blockdev flushing too
	342	*/
b5af8293	343	if (f->mmap)
e5b401d4	344	ret = madvise(f->mmap, f->file_size, MADV_DONTNEED);
467d1b6b	345	else if (f->filetype == FIO_TYPE_FILE)
b5af8293	346	ret = fadvise(f->fd, f->file_offset, f->file_size, POSIX_FADV_DONTNEED);
7e0e25c9	347	else if (f->filetype == FIO_TYPE_BD) {
b5af8293	348	ret = blockdev_invalidate_cache(f->fd);
7e0e25c9 JA	349	if (ret < 0 && errno == EACCES && geteuid()) {
	350	log_err("fio: only root may flush block devices. Cache flush bypassed!\n");
	351	ret = 0;
	352	}
	353	} else if (f->filetype == FIO_TYPE_CHAR)
e5b401d4 JA	354	ret = 0;
	355
	356	if (ret < 0) {
e1161c32	357	td_verror(td, errno, "invalidate_cache");
e5b401d4 JA	358	return 1;
	359	}
	360
ad2da605	361	return ret;
e5b401d4 JA	362	}
e5b401d4 JA	363
b5af8293	364	void generic_close_file(struct thread_data fio_unused td, struct fio_file f)
53cdc686	365	{
b5af8293 JA	366	close(f->fd);
b5af8293 JA	367	f->fd = -1;
53cdc686 JA	368	}
53cdc686 JA	369
b5af8293	370	int generic_open_file(struct thread_data td, struct fio_file f)
53cdc686	371	{
53cdc686 JA	372	int flags = 0;
53cdc686 JA	373
2dc1bbeb	374	if (td->o.odirect)
2fd233b7	375	flags \|= OS_O_DIRECT;
2dc1bbeb	376	if (td->o.sync_io)
2fd233b7	377	flags \|= O_SYNC;
53cdc686	378
2fd233b7 JA	379	if (td_write(td) \|\| td_rw(td)) {
2fd233b7 JA	380	flags \|= O_RDWR;
53cdc686	381
af52b345	382	if (f->filetype == FIO_TYPE_FILE)
2fd233b7	383	flags \|= O_CREAT;
2fd233b7	384
b5af8293	385	f->fd = open(f->file_name, flags, 0600);
2fd233b7	386	} else {
af52b345	387	if (f->filetype == FIO_TYPE_CHAR)
2fd233b7 JA	388	flags \|= O_RDWR;
	389	else
	390	flags \|= O_RDONLY;
	391
b5af8293	392	f->fd = open(f->file_name, flags);
53cdc686 JA	393	}
	394
	395	if (f->fd == -1) {
e4e33258	396	char buf[FIO_VERROR_SIZE];
e1161c32 JA	397	int __e = errno;
e1161c32 JA	398
e4e33258 JA	399	snprintf(buf, sizeof(buf) - 1, "open(%s)", f->file_name);
	400
	401	td_verror(td, __e, buf);
2dc1bbeb	402	if (__e == EINVAL && td->o.odirect)
9d80e114	403	log_err("fio: destination does not support O_DIRECT\n");
8ab53873	404	if (__e == EMFILE)
2dc1bbeb	405	log_err("fio: try reducing/setting openfiles (failed at %u of %u)\n", td->nr_open_files, td->o.nr_files);
53cdc686 JA	406	return 1;
	407	}
	408
b5af8293 JA	409	if (get_file_size(td, f))
	410	goto err;
	411
2dc1bbeb	412	if (td->o.invalidate_cache && file_invalidate_cache(td, f))
b5af8293 JA	413	goto err;
b5af8293 JA	414
148cf6a9 JA	415	if (td->o.fadvise_hint) {
	416	if (td_random(td))
	417	flags = POSIX_FADV_RANDOM;
	418	else
	419	flags = POSIX_FADV_SEQUENTIAL;
d2f3ac35	420
148cf6a9	421	if (fadvise(f->fd, f->file_offset, f->file_size, flags) < 0) {
b5af8293 JA	422	td_verror(td, errno, "fadvise");
	423	goto err;
	424	}
bdb4e2e9	425	}
53cdc686 JA	426
53cdc686 JA	427	return 0;
b5af8293 JA	428	err:
	429	close(f->fd);
	430	return 1;
	431	}
	432
21972cde JA	433	int open_files(struct thread_data *td)
	434	{
	435	struct fio_file *f;
af52b345 JA	436	unsigned int i;
af52b345 JA	437	int err = 0;
21972cde JA	438
21972cde JA	439	for_each_file(td, f, i) {
b5af8293	440	err = td_io_open_file(td, f);
21972cde JA	441	if (err)
21972cde JA	442	break;
b5af8293	443
2dc1bbeb	444	if (td->o.open_files == td->nr_open_files)
b5af8293	445	break;
21972cde JA	446	}
21972cde JA	447
7abf833d JA	448	if (!err)
	449	return 0;
	450
bdb4e2e9	451	for_each_file(td, f, i)
b5af8293	452	td_io_close_file(td, f);
7abf833d	453
21972cde JA	454	return err;
	455	}
	456
53cdc686 JA	457	int setup_files(struct thread_data *td)
	458	{
	459	struct fio_file *f;
af52b345 JA	460	unsigned int i;
af52b345 JA	461	int err;
53cdc686 JA	462
	463	/*
	464	* if ioengine defines a setup() method, it's responsible for
	465	* setting up everything in the td->files[] area.
	466	*/
	467	if (td->io_ops->setup)
	468	return td->io_ops->setup(td);
	469
	470	if (create_files(td))
	471	return 1;
	472
21972cde	473	err = open_files(td);
f1027063 JA	474	if (err)
	475	return err;
	476
0a7eb121 JA	477	/*
	478	* Recalculate the total file size now that files are set up.
	479	*/
2dc1bbeb	480	td->o.size = 0;
0a7eb121	481	for_each_file(td, f, i)
2dc1bbeb	482	td->o.size += f->file_size;
0a7eb121	483
2dc1bbeb	484	td->io_size = td->o.size;
53cdc686	485	if (td->io_size == 0) {
2dc1bbeb	486	log_err("%s: no io blocks\n", td->o.name);
e1161c32	487	td_verror(td, EINVAL, "total_file_size");
53cdc686 JA	488	return 1;
	489	}
	490
2dc1bbeb JA	491	if (!td->o.zone_size)
2dc1bbeb JA	492	td->o.zone_size = td->io_size;
53cdc686	493
2dc1bbeb	494	td->total_io_size = td->io_size * td->o.loops;
53cdc686	495
bdb4e2e9	496	for_each_file(td, f, i)
b5af8293	497	td_io_close_file(td, f);
21972cde JA	498
21972cde JA	499	return err;
53cdc686 JA	500	}
53cdc686 JA	501
68727076 JA	502	int init_random_map(struct thread_data *td)
	503	{
	504	int num_maps, blocks;
	505	struct fio_file *f;
	506	unsigned int i;
	507
	508	if (td->o.norandommap)
	509	return 0;
	510
	511	for_each_file(td, f, i) {
	512	blocks = (f->real_file_size + td->o.rw_min_bs - 1) / td->o.rw_min_bs;
	513	num_maps = (blocks + BLOCKS_PER_MAP-1)/ BLOCKS_PER_MAP;
	514	f->file_map = malloc(num_maps * sizeof(long));
	515	if (!f->file_map) {
	516	log_err("fio: failed allocating random map. If running a large number of jobs, try the 'norandommap' option\n");
	517	return 1;
	518	}
	519	f->num_maps = num_maps;
	520	memset(f->file_map, 0, num_maps * sizeof(long));
	521	}
	522
	523	return 0;
	524	}
	525
53cdc686 JA	526	void close_files(struct thread_data *td)
53cdc686 JA	527	{
0ab8db89	528	struct fio_file *f;
af52b345	529	unsigned int i;
53cdc686	530
0ab8db89	531	for_each_file(td, f, i) {
fa1da865 JA	532	if ((f->flags & FIO_FILE_UNLINK) &&
fa1da865 JA	533	f->filetype == FIO_TYPE_FILE)
132ad46d	534	unlink(f->file_name);
bdb4e2e9	535
b5af8293	536	td_io_close_file(td, f);
b3dc7f07	537
fa1da865 JA	538	free(f->file_name);
	539	f->file_name = NULL;
	540
c343981b	541	if (f->file_map) {
b3dc7f07	542	free(f->file_map);
c343981b JA	543	f->file_map = NULL;
c343981b JA	544	}
53cdc686	545	}
b4a6a59a	546
2dc1bbeb	547	td->o.filename = NULL;
cade3ef4	548	free(td->files);
b4a6a59a	549	td->files = NULL;
2dc1bbeb	550	td->o.nr_files = 0;
53cdc686	551	}
af52b345	552
e3bab463	553	static void get_file_type(struct fio_file *f)
af52b345 JA	554	{
	555	struct stat sb;
	556
	557	f->filetype = FIO_TYPE_FILE;
	558
e3bab463	559	if (!lstat(f->file_name, &sb)) {
96d32d51 JA	560	f->flags \|= FIO_FILE_EXISTS;
96d32d51 JA	561
af52b345 JA	562	if (S_ISBLK(sb.st_mode))
	563	f->filetype = FIO_TYPE_BD;
	564	else if (S_ISCHR(sb.st_mode))
	565	f->filetype = FIO_TYPE_CHAR;
96d32d51 JA	566	else {
	567	/*
	568	* might as well do this here, and save a stat later on
	569	*/
	570	f->real_file_size = sb.st_size;
	571	f->file_size = f->real_file_size;
	572	}
af52b345 JA	573	}
	574	}
	575
	576	void add_file(struct thread_data td, const char fname)
	577	{
7b4e4fe5	578	int cur_files = td->files_index;
bd0ee748	579	char file_name[PATH_MAX];
af52b345	580	struct fio_file *f;
bd0ee748	581	int len = 0;
af52b345 JA	582
	583	td->files = realloc(td->files, (cur_files + 1) * sizeof(*f));
	584
	585	f = &td->files[cur_files];
	586	memset(f, 0, sizeof(*f));
	587	f->fd = -1;
bd0ee748 JA	588
	589	if (td->o.directory)
	590	len = sprintf(file_name, "%s/", td->o.directory);
	591
	592	sprintf(file_name + len, "%s", fname);
	593	f->file_name = strdup(file_name);
af52b345	594
e3bab463	595	get_file_type(f);
af52b345	596
7b4e4fe5	597	td->files_index++;
1549441c JA	598	if (f->filetype == FIO_TYPE_FILE)
1549441c JA	599	td->nr_normal_files++;
af52b345	600	}
0ad920e7 JA	601
	602	void get_file(struct fio_file *f)
	603	{
	604	f->references++;
	605	}
	606
	607	void put_file(struct thread_data td, struct fio_file f)
	608	{
	609	if (!(f->flags & FIO_FILE_OPEN))
	610	return;
	611
	612	assert(f->references);
	613	if (--f->references)
	614	return;
	615
2dc1bbeb	616	if (should_fsync(td) && td->o.fsync_on_close)
ebb1415f JA	617	fsync(f->fd);
ebb1415f JA	618
0ad920e7 JA	619	if (td->io_ops->close_file)
	620	td->io_ops->close_file(td, f);
	621	td->nr_open_files--;
	622	f->flags &= ~FIO_FILE_OPEN;
	623	}
bbf6b540 JA	624
	625	static int recurse_dir(struct thread_data td, const char dirname)
	626	{
	627	struct dirent *dir;
	628	int ret = 0;
	629	DIR *D;
	630
	631	D = opendir(dirname);
	632	if (!D) {
	633	td_verror(td, errno, "opendir");
	634	return 1;
	635	}
	636
	637	while ((dir = readdir(D)) != NULL) {
	638	char full_path[PATH_MAX];
	639	struct stat sb;
	640
e85b2b83 JA	641	if (!strcmp(dir->d_name, ".") \|\| !strcmp(dir->d_name, ".."))
e85b2b83 JA	642	continue;
96d32d51	643
bbf6b540 JA	644	sprintf(full_path, "%s/%s", dirname, dir->d_name);
	645
	646	if (lstat(full_path, &sb) == -1) {
	647	if (errno != ENOENT) {
	648	td_verror(td, errno, "stat");
	649	return 1;
	650	}
	651	}
	652
	653	if (S_ISREG(sb.st_mode)) {
	654	add_file(td, full_path);
2dc1bbeb	655	td->o.nr_files++;
bbf6b540 JA	656	continue;
	657	}
	658
bbf6b540 JA	659	if ((ret = recurse_dir(td, full_path)) != 0)
	660	break;
	661	}
	662
	663	closedir(D);
	664	return ret;
	665	}
	666
	667	int add_dir_files(struct thread_data td, const char path)
	668	{
	669	return recurse_dir(td, path);
	670	}
cade3ef4 JA	671
	672	void dup_files(struct thread_data td, struct thread_data org)
	673	{
	674	struct fio_file *f;
	675	unsigned int i;
	676	size_t bytes;
	677
	678	if (!org->files)
	679	return;
	680
	681	bytes = org->files_index * sizeof(*f);
	682	td->files = malloc(bytes);
	683	memcpy(td->files, org->files, bytes);
	684
	685	for_each_file(td, f, i) {
	686	if (f->file_name)
	687	f->file_name = strdup(f->file_name);
	688	}
	689	}