1 /*
2  * Copyright (c) 2018 Stefan Sperling <stsp@openbsd.org>
3  *
4  * Permission to use, copy, modify, and distribute this software for any
5  * purpose with or without fee is hereby granted, provided that the above
6  * copyright notice and this permission notice appear in all copies.
7  *
8  * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
9  * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
10  * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
11  * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
12  * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
13  * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
14  * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
15  */
16 
17 #include <limits.h>
18 #include <stdlib.h>
19 #include <unistd.h>
20 #include <string.h>
21 #include <stdio.h>
22 
23 #include "got_opentemp.h"
24 #include "got_error.h"
25 
26 int
27 got_opentempfd(void)
28 {
29 	char name[PATH_MAX];
30 	int fd;
31 
32 	if (strlcpy(name, "/tmp/got.XXXXXXXX", sizeof(name)) >= sizeof(name))
33 		return -1;
34 
35 	fd = mkstemp(name);
36 	unlink(name);
37 	return fd;
38 }
39 
40 FILE *
41 got_opentemp(void)
42 {
43 	int fd;
44 	FILE *f;
45 
46 	fd = got_opentempfd();
47 	if (fd < 0)
48 		return NULL;
49 
50 	f = fdopen(fd, "w+");
51 	if (f == NULL) {
52 		close(fd);
53 		return NULL;
54 	}
55 
56 	return f;
57 }
58 
59 const struct got_error *
60 got_opentemp_named(char **path, FILE **outfile, const char *basepath)
61 {
62 	const struct got_error *err = NULL;
63 	int fd;
64 
65 	*outfile = NULL;
66 
67 	if (asprintf(path, "%s-XXXXXX", basepath) == -1) {
68 		*path = NULL;
69 		return got_error_from_errno("asprintf");
70 	}
71 
72 	fd = mkstemp(*path);
73 	if (fd == -1) {
74 		err = got_error_from_errno2("mkstemp", *path);
75 		free(*path);
76 		*path = NULL;
77 		return err;
78 	}
79 
80 	*outfile = fdopen(fd, "w+");
81 	if (*outfile == NULL) {
82 		err = got_error_from_errno2("fdopen", *path);
83 		free(*path);
84 		*path = NULL;
85 	}
86 
87 	return err;
88 }
89 
90 const struct got_error *
91 got_opentemp_named_fd(char **path, int *outfd, const char *basepath)
92 {
93 	const struct got_error *err = NULL;
94 	int fd;
95 
96 	*outfd = -1;
97 
98 	if (asprintf(path, "%s-XXXXXX", basepath) == -1) {
99 		*path = NULL;
100 		return got_error_from_errno("asprintf");
101 	}
102 
103 	fd = mkstemp(*path);
104 	if (fd == -1) {
105 		err = got_error_from_errno("mkstemp");
106 		free(*path);
107 		*path = NULL;
108 		return err;
109 	}
110 
111 	*outfd = fd;
112 	return err;
113 }