/*
* Only unlink files that we created.
*/
- f->unlink = 0;
+ f->flags &= ~FIO_FILE_UNLINK;
if (file_ok(td, f)) {
- f->unlink = td->unlink;
+ if (td->unlink)
+ f->flags |= FIO_FILE_UNLINK;
+
err = create_file(td, f);
if (err)
break;
unsigned int i;
for_each_file(td, f, i) {
- if (!td->filename && f->unlink &&
+ if (!f->file_name && (f->flags & FIO_FILE_UNLINK) &&
f->filetype == FIO_TYPE_FILE) {
unlink(f->file_name);
+ free(f->file_name);
f->file_name = NULL;
}
td->nr_files = 0;
}
-static void get_file_type(struct thread_data *td, struct fio_file *f)
+static void get_file_type(struct fio_file *f)
{
struct stat sb;
f->filetype = FIO_TYPE_FILE;
- if (!lstat(td->filename, &sb)) {
+ if (!lstat(f->file_name, &sb)) {
if (S_ISBLK(sb.st_mode))
f->filetype = FIO_TYPE_BD;
else if (S_ISCHR(sb.st_mode))
f = &td->files[cur_files];
memset(f, 0, sizeof(*f));
f->fd = -1;
- f->file_name = fname;
+ f->file_name = strdup(fname);
- get_file_type(td, f);
+ get_file_type(f);
td->open_files++;
td->nr_uniq_files = td->open_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 (td->io_ops->close_file)
+ td->io_ops->close_file(td, f);
+ td->nr_open_files--;
+ f->flags &= ~FIO_FILE_OPEN;
+}