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_path_lib.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 	int n = 0;
70 	char *s = strdup(path), *p;
71 
72 	*count = 0;
73 
74 	if (s == NULL)
75 		return got_error(GOT_ERR_NO_MEM);
76 
77 	do {
78 		p = strsep(&s, "/");
79 		if (s && *s != '/')
80 			(*count)++;
81 	} while (p);
82 
83 	return NULL;
84 }
85 
86 FILE *
87 got_opentemp(void)
88 {
89 	char name[PATH_MAX];
90 	int fd;
91 	FILE *f;
92 
93 	if (strlcpy(name, "/tmp/got.XXXXXXXX", sizeof(name)) >= sizeof(name))
94 		return NULL;
95 
96 	fd = mkstemp(name);
97 	if (fd < 0)
98 		return NULL;
99 
100 	unlink(name);
101 	f = fdopen(fd, "w+");
102 	if (f == NULL) {
103 		close(fd);
104 		return NULL;
105 	}
106 
107 	return f;
108 }
109 
110 const struct got_error *
111 got_opentemp_named(char **path, FILE **outfile, const char *basepath)
112 {
113 	const struct got_error *err = NULL;
114 	int fd, ret;
115 
116 	if (asprintf(path, "%s-XXXXXX", basepath) == -1) {
117 		*path = NULL;
118 		return got_error(GOT_ERR_NO_MEM);
119 	}
120 
121 	fd = mkstemp(*path);
122 	if (fd == -1) {
123 		err = got_error_from_errno();
124 		free(*path);
125 		*path = NULL;
126 		return err;
127 	}
128 
129 	*outfile = fdopen(fd, "w+");
130 	if (*outfile == NULL) {
131 		err = got_error_from_errno();
132 		free(*path);
133 		*path = NULL;
134 	}
135 
136 	return err;
137 }