1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
|
/* SPDX-License-Identifier: LGPL-2.1-only */
/* Copyright (C) 2020 Gediminas Jakutis */
#include <sys/types.h>
#include <sys/stat.h>
#include <linux/limits.h>
#include <unistd.h>
#include <fcntl.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <libgen.h>
#include "io.h"
#include "defs.h"
#include "datagen.h"
#include "cache.h"
static int stream_open_in(struct stream * const in, const struct settings * const s);
static int stream_open_out(struct stream * const in, const struct settings * const s);
static int stream_open_special(struct stream * const in);
int stream_open(struct stream * const in, const struct settings * const s)
{
int ret = 0;
try(!in || in->fd > 0 || !in->name || !s, err, EINVAL, "invalid argunent");
if (in->out == 1) {
ret = stream_open_out(in, s);
} else if (!in->out) {
ret = stream_open_in(in, s);
} else {
ret = stream_open_special(in);
}
err:
return ret;
}
int stream_close(struct stream * const in)
{
int ret = 0;
try(!in || in->fd < 0, early_err, EINVAL, "invalid argunent");
if (!in->out) {
goto out;
}
if (in->name) {
char path[PATH_MAX];
struct stat st;
snprintf(path, PATH_MAX, "/proc/self/fd/%i", in->fd);
if (!stat(in->name, &st)) {
if (st.st_mode & S_IFREG) {
unlink(in->name);
} else {
ret = EINVAL;
rin_err("the given output file already exists and is not a regular file");
goto err;
}
}
try(linkat(AT_FDCWD, path, AT_FDCWD, in->name, AT_SYMLINK_FOLLOW), err, errno, "error linking output file to filesystem: %s", strerror(ret));
} else {
rin_err("no output filename given");
ret = EINVAL;
goto err;
}
out:
err:
close(in->fd);
in->fd = -1;
early_err:
return ret;
}
static int stream_open_out(struct stream * const in, const struct settings * const s)
{
struct stat st;
mode_t mode = S_IRUSR | S_IWUSR | S_IRGRP;
char *dname = NULL;
char *tmp[2];
int ret = 0;
tmp[0] = strdup(in->name);
tmp[1] = dirname(tmp[0]);
dname = strdup(tmp[1]);
free(tmp[0]);
try(stat(dname, &st), err, errno, "stat failed: %s", strerror(ret));
try(!(st.st_mode & S_IFDIR), err, EINVAL, "invalid output path");
if (!stat(in->name, &st)) {
try(!(st.st_mode & S_IFREG), err, EINVAL, "the given output file already exists and is not a regular file");
mode = st.st_mode;
}
in->fd = open(dname, O_TMPFILE | O_WRONLY, mode);
try(in->fd < 0, err, errno, "failed creating temporary output file: %s", strerror(ret));
try(ftruncate(in->fd, s->stride * in->n), err, errno, "failed setting output file size: %s", strerror(ret));
err:
free(dname);
return ret;
}
static int stream_open_in(struct stream * const in, const struct settings * const s)
{
struct stat st;
int ret = 0;
try(stat(in->name, &st), err, errno, "stat failed: %s", strerror(ret));
try(!(st.st_mode & S_IFREG) || !st.st_size || (st.st_size % s->stride), err, EINVAL, "invalid input file");
in->n = st.st_size / s->stride;
in->fd = open(in->name, O_RDONLY | O_NOATIME);
try(in->fd < 0, err, errno, "failed opening input file: %s", strerror(ret));
if (in->cached) {
cache_create(in, s);
cache_populate(in);
}
err:
return ret;
}
static int stream_open_special(struct stream * const in)
{
struct stat st;
int ret = 0;
try(stat(in->name, &st), err, errno, "stat failed: %s", strerror(ret));
in->fd = open(in->name, O_RDONLY | O_NOATIME);
try(in->fd < 0, err, errno, "failed opening input file: %s", strerror(ret));
err:
return ret;
}
|
