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 <stdio.h>
21 #include <string.h>
22 
23 #include "got_error.h"
24 
25 #include "got_lib_path.h"
26 
27 int
28 got_path_is_absolute(const char *path)
29 {
30 	return path[0] == '/';
31 }
32 
33 char *
34 got_path_get_absolute(const char *relpath) 
35 {
36 	char cwd[PATH_MAX];
37 	char *abspath;
38 
39 	if (getcwd(cwd, sizeof(cwd)) == NULL)
40 		return NULL;
41 
42 	if (asprintf(&abspath, "%s/%s/", cwd, relpath) == -1)
43 		return NULL;
44 
45 	return abspath;
46 }
47 
48 char *
49 got_path_normalize(const char *path)
50 {
51 	char *resolved;
52 
53 	resolved = realpath(path, NULL);
54 	if (resolved == NULL)
55 		return NULL;
56 	
57 	if (!got_path_is_absolute(resolved)) {
58 		char *abspath = got_path_get_absolute(resolved);
59 		free(resolved);
60 		resolved = abspath;
61 	}
62 
63 	return resolved;
64 }
65 
66 const struct got_error *
67 got_path_segment_count(int *count, const char *path)
68 {
69 	char *s = strdup(path), *p;
70 
71 	*count = 0;
72 
73 	if (s == NULL)
74 		return got_error_from_errno();
75 
76 	do {
77 		p = strsep(&s, "/");
78 		if (s && *s != '/')
79 			(*count)++;
80 	} while (p);
81 
82 	return NULL;
83 }
84 
85 int
86 got_opentempfd(void)
87 {
88 	char name[PATH_MAX];
89 	int fd;
90 
91 	if (strlcpy(name, "/tmp/got.XXXXXXXX", sizeof(name)) >= sizeof(name))
92 		return -1;
93 
94 	fd = mkstemp(name);
95 	unlink(name);
96 	return fd;
97 }
98 
99 FILE *
100 got_opentemp(void)
101 {
102 	int fd;
103 	FILE *f;
104 
105 	fd = got_opentempfd();
106 	if (fd < 0)
107 		return NULL;
108 
109 	f = fdopen(fd, "w+");
110 	if (f == NULL) {
111 		close(fd);
112 		return NULL;
113 	}
114 
115 	return f;
116 }
117 
118 const struct got_error *
119 got_opentemp_named(char **path, FILE **outfile, const char *basepath)
120 {
121 	const struct got_error *err = NULL;
122 	int fd;
123 
124 	if (asprintf(path, "%s-XXXXXX", basepath) == -1) {
125 		*path = NULL;
126 		return got_error_from_errno();
127 	}
128 
129 	fd = mkstemp(*path);
130 	if (fd == -1) {
131 		err = got_error_from_errno();
132 		free(*path);
133 		*path = NULL;
134 		return err;
135 	}
136 
137 	*outfile = fdopen(fd, "w+");
138 	if (*outfile == NULL) {
139 		err = got_error_from_errno();
140 		free(*path);
141 		*path = NULL;
142 	}
143 
144 	return err;
145 }