logo

utils-std

Collection of commonly available Unix tools git clone https://anongit.hacktivis.me/git/utils-std.git/

install.c (7902B)

    

  1. // utils-std: Collection of commonly available Unix tools
  2. // SPDX-FileCopyrightText: 2017 Haelwenn (lanodan) Monnier <contact+utils@hacktivis.me>
  3. // SPDX-License-Identifier: MPL-2.0
  5. // For copy_file_range
  6. #define _GNU_SOURCE     // musl, glibc
  7. #define _DEFAULT_SOURCE // FreeBSD
  9. #include "../config.h" // HAS_*
  10. #include "../libutils/fs.h"
  11. #include "../libutils/getopt_nolong.h"
  12. #include "../libutils/lib_mkdir.h"
  13. #include "../libutils/lib_string.h"
  14. #include "../libutils/mode.h"
  15. #include "../libutils/user_group_parse.h"
  17. #include <assert.h>
  18. #include <errno.h>
  19. #include <fcntl.h>  // linkat, AT_SYMLINK_FOLLOW
  20. #include <libgen.h> // dirname
  21. #include <limits.h> // PATH_MAX
  22. #include <stdbool.h>
  23. #include <stdio.h>  // fprintf
  24. #include <string.h> // strerror
  25. #include <sys/stat.h>
  26. #include <unistd.h> // getopt
  27. #ifdef HAS_GETOPT_LONG
  28. #include <getopt.h>
  29. #endif
  31. bool preserve_times = false, create_directories = false;
  33. // See lib/lib_mkdir.c
  34. bool mkdir_parents_verbose = false;
  35. mode_t mkdir_parents_filemask;
  37. mode_t mode = 00755;
  38. uid_t user = (uid_t)-1;
  39. gid_t group = (gid_t)-1;
  40. const char *argv0 = "install";
  42. bool opt_v = false;
  44. static int
  45. do_install(char *src, char *dest, bool is_dir)
  46. {
  47. 	int src_fd = open(src, O_RDONLY);
  48. 	if(src_fd < 0)
  49. 	{
  50. 		fprintf(stderr,
  51. 		        "%s: error: Failed opening file '%s' for reading: %s\n",
  52. 		        argv0,
  53. 		        src,
  54. 		        strerror(errno));
  55. 		return -1;
  56. 	}
  58. 	struct stat src_stat;
  59. 	if(fstat(src_fd, &src_stat) < 0)
  60. 	{
  61. 		fprintf(stderr,
  62. 		        "%s: error: Failed getting status for source '%s': %s\n",
  63. 		        argv0,
  64. 		        dest,
  65. 		        strerror(errno));
  66. 		return -1;
  67. 	}
  69. 	if(!is_dir)
  70. 	{
  71. 		struct stat dest_stat;
  72. 		if(stat(dest, &dest_stat) < 0)
  73. 		{
  74. 			if(errno != ENOENT)
  75. 			{
  76. 				fprintf(stderr,
  77. 				        "%s: error: Failed getting status for destination '%s': %s\n",
  78. 				        argv0,
  79. 				        dest,
  80. 				        strerror(errno));
  81. 				return -1;
  82. 			}
  83. 			else
  84. 				errno = 0;
  85. 		}
  86. 		else
  87. 		{
  88. 			if(S_ISDIR(dest_stat.st_mode))
  89. 			{
  90. 				is_dir = true;
  91. 			}
  92. 			else
  93. 			{
  94. 				if(unlink(dest) < 0)
  95. 				{
  96. 					fprintf(stderr,
  97. 					        "%s: error: Failed removing existing file at destination '%s': %s\n",
  98. 					        argv0,
  99. 					        dest,
  100. 					        strerror(errno));
  101. 					return -1;
  102. 				}
  103. 			}
  104. 		}
  105. 	}
  107. 	char *dest_path = dest;
  109. 	if(is_dir)
  110. 	{
  111. 		static char target[PATH_MAX] = "";
  112. 		char *src_basename = static_basename(src);
  114. 		if(snprintf(target, PATH_MAX, "%s/%s", dest, src_basename) < 0)
  115. 		{
  116. 			fprintf(stderr,
  117. 			        "%s: error: Failed joining destination '%s' and source '%s'\n",
  118. 			        argv0,
  119. 			        dest,
  120. 			        src_basename);
  121. 			return -1;
  122. 		}
  124. 		dest_path = target;
  125. 	}
  127. 	int dest_fd = open(dest_path, O_CREAT | O_WRONLY | O_TRUNC, mode);
  128. 	if(dest_fd < 0)
  129. 	{
  130. 		fprintf(stderr,
  131. 		        "%s: error: Failed create-opening file '%s' for writing: %s\n",
  132. 		        argv0,
  133. 		        dest_path,
  134. 		        strerror(errno));
  135. 		return -1;
  136. 	}
  138. 	if(opt_v) fprintf(stderr, "%s: Made file: %s\n", argv0, dest);
  140. 	posix_fadvise(src_fd, 0, 0, POSIX_FADV_SEQUENTIAL);
  141. 	posix_fadvise(dest_fd, 0, 0, POSIX_FADV_SEQUENTIAL);
  142. 	errno = 0;
  144. 	if(auto_file_copy(src_fd, dest_fd, src_stat.st_size, 0) < 0)
  145. 	{
  146. 		fprintf(stderr,
  147. 		        "%s: error: Failed copying data from '%s' to '%s': %s\n",
  148. 		        argv0,
  149. 		        src,
  150. 		        dest_path,
  151. 		        strerror(errno));
  152. 		return -1;
  153. 	}
  155. 	if(user != (uid_t)-1 || group != (gid_t)-1)
  156. 		if(fchown(dest_fd, user, group) < 0)
  157. 		{
  158. 			fprintf(stderr,
  159. 			        "%s: error: Failed changing ownership of '%s': %s\n",
  160. 			        argv0,
  161. 			        dest_path,
  162. 			        strerror(errno));
  163. 			return -1;
  164. 		}
  166. 	if(preserve_times)
  167. 	{
  168. 		struct timespec src_times[2] = {src_stat.st_atim, src_stat.st_mtim};
  169. 		if(futimens(dest_fd, src_times) != 0)
  170. 		{
  171. 			fprintf(stderr,
  172. 			        "%s: error: Failed setting access/modification times on '%s': %s\n",
  173. 			        argv0,
  174. 			        dest_path,
  175. 			        strerror(errno));
  176. 			return -1;
  177. 		}
  178. 	}
  180. 	if(close(src_fd) < 0)
  181. 	{
  182. 		fprintf(stderr, "%s: error: Failed closing '%s'\n", argv0, src);
  183. 		return -1;
  184. 	}
  186. 	if(close(dest_fd) < 0)
  187. 	{
  188. 		fprintf(stderr, "%s: error: Failed closing '%s'\n", argv0, dest_path);
  189. 		return -1;
  190. 	}
  192. 	return 0;
  193. }
  195. static void
  196. usage(void)
  197. {
  198. 	fprintf(stderr, "\
  199. Usage: install [-CcDpTv] [-g group] [-m mode] [-o owner] source... destination\n\
  200.        install [-CcDpTv] [-g group] [-m mode] [-o owner] -t destination source...\n\
  201.        install -d [-cv] [-g group] [-m mode] [-o owner] directory...\n\
  202. ");
  203. }
  205. int
  206. main(int argc, char *argv[])
  207. {
  208. 	const char *errstr = NULL;
  209. 	bool create_parents = false;
  210. 	bool opt_T = false;
  211. 	char *dest = NULL;
  213. 	mkdir_parents_filemask = umask(0);
  214. 	umask(mkdir_parents_filemask);
  216. #ifdef HAS_GETOPT_LONG
  217. 	// Strictly for GNUisms compatibility so no long-only options
  218. 	// clang-format off
  219. 	static struct option opts[] = {
  220. 		{"compare", no_argument, NULL, 'c'},
  221. 		{"directory", no_argument, NULL, 'd'},
  222. 		{"group", required_argument, NULL, 'g'},
  223. 		{"mode", required_argument, NULL, 'm'},
  224. 		{"owner", required_argument, NULL, 'o'},
  225. 		{"preserve-timestamps", no_argument, NULL, 'p'},
  226. 		{"target-directory", required_argument, NULL, 't'},
  227. 		{"no-target-directory", no_argument, NULL, 'T'},
  228. 		{"verbose", no_argument, NULL, 'v'},
  229. 		{0, 0, 0, 0},
  230. 	};
  231. 	// clang-format on
  233. 	// Need + as first character to get POSIX-style option parsing
  234. 	for(int c = -1; (c = getopt_long(argc, argv, "+:CcDdpTt:g:m:o:v", opts, NULL)) != -1;)
  235. #else
  236. 	for(int c = -1; (c = getopt_nolong(argc, argv, ":CcDdpTt:g:m:o:v")) != -1;)
  237. #endif
  238. 	{
  239. 		switch(c)
  240. 		{
  241. 		case 'C':
  242. 			// ignore, for compatibility
  243. 			break;
  244. 		case 'c':
  245. 			// ignore, modern default behavior
  246. 			break;
  247. 		case 'D':
  248. 			create_parents = true;
  249. 			break;
  250. 		case 'd':
  251. 			create_directories = true;
  252. 			break;
  253. 		case 'p':
  254. 			preserve_times = true;
  255. 			break;
  256. 		case 'g':
  257. 			if(parse_group(optarg, &group) != 0) return 1;
  258. 			break;
  259. 		case 'o':
  260. 			if(parse_user(optarg, &user) != 0) return 1;
  261. 			break;
  262. 		case 'T':
  263. 			opt_T = true;
  264. 			break;
  265. 		case 't':
  266. 			dest = optarg;
  267. 			break;
  268. 		case 'm':
  269. 			mode = new_mode(optarg, 0755, &errstr);
  270. 			if(errstr != NULL)
  271. 			{
  272. 				fprintf(stderr, "install: error: Failed parsing mode '%s': %s\n", optarg, errstr);
  273. 				return 1;
  274. 			}
  275. 			break;
  276. 		case 'v':
  277. 			opt_v = true;
  278. 			mkdir_parents_verbose = true;
  279. 			break;
  280. 		case ':':
  281. 			fprintf(stderr, "install: error: Missing operand for option: '-%c'\n", optopt);
  282. 			usage();
  283. 			return 1;
  284. 		case '?':
  285. 			GETOPT_UNKNOWN_OPT
  286. 			usage();
  287. 			return 1;
  288. 		}
  289. 	}
  291. 	argc -= optind;
  292. 	argv += optind;
  294. 	if(create_directories)
  295. 	{
  296. 		for(int i = 0; i < argc; i++)
  297. 		{
  298. 			char *dest = argv[i];
  300. 			if(mkdir_parents(dest, mode) != 0) return -1;
  302. 			if(opt_v) fprintf(stderr, "%s: Made directory: %s\n", argv0, dest);
  304. 			if(user != (uid_t)-1 || group != (gid_t)-1)
  305. 			{
  306. 				if(chown(dest, user, group) < 0)
  307. 				{
  308. 					fprintf(stderr,
  309. 					        "%s: error: Failed changing ownership of '%s': %s\n",
  310. 					        argv0,
  311. 					        dest,
  312. 					        strerror(errno));
  313. 					return -1;
  314. 				}
  315. 			}
  316. 		}
  318. 		return 0;
  319. 	}
  321. 	if(dest == NULL) dest = argv[--argc];
  322. 	bool multi_src = argc > 1;
  324. 	if(opt_T)
  325. 	{
  326. 		if(argc != 1)
  327. 		{
  328. 			fprintf(stderr,
  329. 			        "%s: error: Option -T passed, expected exactly 1 source operand, got %d\n",
  330. 			        argv0,
  331. 			        argc);
  332. 			return 1;
  333. 		}
  335. 		assert(!multi_src);
  336. 	}
  337. 	else if(argc < 1)
  338. 	{
  339. 		fprintf(stderr, "%s: error: Expected at least 1 source operand\n", argv0);
  340. 		return 1;
  341. 	}
  343. 	if(create_parents)
  344. 	{
  345. 		char *destdir = dest;
  346. 		// Same as in mkdir_parents
  347. 		mode_t parent_mode = (S_IWUSR | S_IXUSR | ~mkdir_parents_filemask) & 0777;
  349. 		char path_dup[PATH_MAX] = "";
  350. 		if(!multi_src)
  351. 		{
  352. 			lib_strlcpy(path_dup, dest, PATH_MAX);
  354. 			destdir = dirname(path_dup);
  355. 		}
  357. 		if(mkdir_parents(destdir, parent_mode) != 0) return 1;
  358. 	}
  360. 	for(int i = 0; i < argc; i++)
  361. 	{
  362. 		char *src = argv[i];
  363. 		if(do_install(src, dest, multi_src) < 0) return 1;
  364. 	}
  366. 	return 0;
  367. }