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
144
145
146
147
148
149
|
/* 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 != 1) {
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);
try(in->fd < 0, err, errno, "failed opening input file: %s", strerror(ret));
err:
return ret;
}
int direct_get_array(struct stream * const restrict in, ssize_t idx, struct entry_l * const data);
int direct_get_list(struct stream * const restrict in, ssize_t idx, struct entry_l * const data);
int direct_put_array(struct stream * const restrict in, ssize_t idx, const struct entry_l * const data);
int direct_put_list(struct stream * const restrict in, ssize_t idx, const struct entry_l * const data);
|
