1*cdea53deSanton /* $OpenBSD: util.c,v 1.1 2018/11/06 18:11:11 anton Exp $ */
2*cdea53deSanton
3*cdea53deSanton /*
4*cdea53deSanton * Copyright (c) 2018 Anton Lindqvist <anton@openbsd.org>
5*cdea53deSanton *
6*cdea53deSanton * Permission to use, copy, modify, and distribute this software for any
7*cdea53deSanton * purpose with or without fee is hereby granted, provided that the above
8*cdea53deSanton * copyright notice and this permission notice appear in all copies.
9*cdea53deSanton *
10*cdea53deSanton * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
11*cdea53deSanton * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
12*cdea53deSanton * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
13*cdea53deSanton * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
14*cdea53deSanton * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
15*cdea53deSanton * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
16*cdea53deSanton * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
17*cdea53deSanton */
18*cdea53deSanton
19*cdea53deSanton #include <sys/wait.h>
20*cdea53deSanton
21*cdea53deSanton #include <err.h>
22*cdea53deSanton #include <errno.h>
23*cdea53deSanton #include <limits.h>
24*cdea53deSanton #include <paths.h>
25*cdea53deSanton #include <stdio.h>
26*cdea53deSanton #include <stdlib.h>
27*cdea53deSanton #include <string.h>
28*cdea53deSanton #include <unistd.h>
29*cdea53deSanton
30*cdea53deSanton #include "util.h"
31*cdea53deSanton
32*cdea53deSanton static const char *tmpdir(void);
33*cdea53deSanton
34*cdea53deSanton int
make_file(off_t size)35*cdea53deSanton make_file(off_t size)
36*cdea53deSanton {
37*cdea53deSanton char template[PATH_MAX];
38*cdea53deSanton const char *dir;
39*cdea53deSanton int fd, n;
40*cdea53deSanton
41*cdea53deSanton dir = tmpdir();
42*cdea53deSanton n = snprintf(template, sizeof(template), "%sflock.XXXXXX", dir);
43*cdea53deSanton if (n == -1 || n >= (int)sizeof(template))
44*cdea53deSanton errc(1, ENAMETOOLONG, "%s", __func__);
45*cdea53deSanton
46*cdea53deSanton fd = mkstemp(template);
47*cdea53deSanton if (fd == -1)
48*cdea53deSanton err(1, "mkstemp");
49*cdea53deSanton if (ftruncate(fd, size) == -1)
50*cdea53deSanton err(1, "ftruncate");
51*cdea53deSanton if (unlink(template) == -1)
52*cdea53deSanton err(1, "unlink");
53*cdea53deSanton
54*cdea53deSanton return (fd);
55*cdea53deSanton }
56*cdea53deSanton
57*cdea53deSanton int
safe_waitpid(pid_t pid)58*cdea53deSanton safe_waitpid(pid_t pid)
59*cdea53deSanton {
60*cdea53deSanton int save_errno;
61*cdea53deSanton int status;
62*cdea53deSanton
63*cdea53deSanton save_errno = errno;
64*cdea53deSanton errno = 0;
65*cdea53deSanton while (waitpid(pid, &status, 0) != pid) {
66*cdea53deSanton if (errno == EINTR)
67*cdea53deSanton continue;
68*cdea53deSanton err(1, "waitpid");
69*cdea53deSanton }
70*cdea53deSanton errno = save_errno;
71*cdea53deSanton
72*cdea53deSanton return (status);
73*cdea53deSanton }
74*cdea53deSanton
75*cdea53deSanton __dead void
usage(void)76*cdea53deSanton usage(void)
77*cdea53deSanton {
78*cdea53deSanton fprintf(stderr, "usage: %s [-v] testnum\n", getprogname());
79*cdea53deSanton exit(1);
80*cdea53deSanton }
81*cdea53deSanton
82*cdea53deSanton static const char *
tmpdir(void)83*cdea53deSanton tmpdir(void)
84*cdea53deSanton {
85*cdea53deSanton static char path[PATH_MAX];
86*cdea53deSanton const char *p;
87*cdea53deSanton size_t len;
88*cdea53deSanton int n;
89*cdea53deSanton int nosuffix = 1;
90*cdea53deSanton
91*cdea53deSanton p = getenv("TMPDIR");
92*cdea53deSanton if (p == NULL || *p == '\0')
93*cdea53deSanton p = _PATH_TMP;
94*cdea53deSanton len = strlen(p);
95*cdea53deSanton if (len == 0)
96*cdea53deSanton errx(1, "%s: empty path", __func__);
97*cdea53deSanton if (p[len - 1] == '/')
98*cdea53deSanton nosuffix = 0;
99*cdea53deSanton n = snprintf(path, sizeof(path), "%s%s", p, nosuffix ? "/" : "");
100*cdea53deSanton if (n == -1 || n >= (int)sizeof(path))
101*cdea53deSanton errc(1, ENAMETOOLONG, "%s", __func__);
102*cdea53deSanton
103*cdea53deSanton return (path);
104*cdea53deSanton }
105