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
|
/* $OpenBSD: util.c,v 1.1 2018/11/06 18:11:11 anton Exp $ */
/*
* Copyright (c) 2018 Anton Lindqvist <anton@openbsd.org>
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <sys/wait.h>
#include <err.h>
#include <errno.h>
#include <limits.h>
#include <paths.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "util.h"
static const char *tmpdir(void);
int
make_file(off_t size)
{
char template[PATH_MAX];
const char *dir;
int fd, n;
dir = tmpdir();
n = snprintf(template, sizeof(template), "%sflock.XXXXXX", dir);
if (n == -1 || n >= (int)sizeof(template))
errc(1, ENAMETOOLONG, "%s", __func__);
fd = mkstemp(template);
if (fd == -1)
err(1, "mkstemp");
if (ftruncate(fd, size) == -1)
err(1, "ftruncate");
if (unlink(template) == -1)
err(1, "unlink");
return (fd);
}
int
safe_waitpid(pid_t pid)
{
int save_errno;
int status;
save_errno = errno;
errno = 0;
while (waitpid(pid, &status, 0) != pid) {
if (errno == EINTR)
continue;
err(1, "waitpid");
}
errno = save_errno;
return (status);
}
__dead void
usage(void)
{
fprintf(stderr, "usage: %s [-v] testnum\n", getprogname());
exit(1);
}
static const char *
tmpdir(void)
{
static char path[PATH_MAX];
const char *p;
size_t len;
int n;
int nosuffix = 1;
p = getenv("TMPDIR");
if (p == NULL || *p == '\0')
p = _PATH_TMP;
len = strlen(p);
if (len == 0)
errx(1, "%s: empty path", __func__);
if (p[len - 1] == '/')
nosuffix = 0;
n = snprintf(path, sizeof(path), "%s%s", p, nosuffix ? "/" : "");
if (n == -1 || n >= (int)sizeof(path))
errc(1, ENAMETOOLONG, "%s", __func__);
return (path);
}
|
