logo

utils-std

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

mv.c (11924B)

    

  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. #define _POSIX_C_SOURCE 200809L
  6. #define _GNU_SOURCE // copy_file_range
  7. #define _FILE_OFFSET_BITS 64
  9. // NetBSD <10 hides fdopendir behind _NETBSD_SOURCE
  10. #if __NetBSD_Version__ < 1000000000
  11. #define _NETBSD_SOURCE
  12. #endif
  14. #include "../lib/consent.h"
  15. #include "../lib/fs.h"
  16. #include "../lib/getopt_nolong.h"
  18. #include <dirent.h> // fdopendir
  19. #include <errno.h>
  20. #include <fcntl.h>  // open
  21. #include <libgen.h> // basename
  22. #include <limits.h> // PATH_MAX
  23. #include <locale.h> // setlocale
  24. #include <stdbool.h>
  25. #include <stdint.h>   // SIZE_MAX
  26. #include <stdio.h>    // fprintf, rename
  27. #include <string.h>   // strcmp
  28. #include <sys/stat.h> // stat, S_ISDIR
  29. #include <unistd.h>   // getopt
  31. // Workaround against GNU glibc
  32. // https://sourceware.org/bugzilla/show_bug.cgi?id=18228
  33. #if defined(__linux__) && !defined(O_SEARCH)
  34. // As defined in musl
  35. #define O_SEARCH O_PATH
  36. #endif
  38. const char *argv0 = "mv";
  40. bool no_clob = false, force = false, interact = false, verbose = false;
  42. static int stdin_tty = 0;
  44. struct named_fd
  45. {
  46. 	int fd;
  47. 	const char *name;
  48. 	const char *sep;
  49. };
  51. static int do_renameat(struct named_fd srcdir,
  52.                        const char *restrict src,
  53.                        struct named_fd destdir,
  54.                        const char *restrict dest);
  56. static int
  57. copy_file_unlink(struct named_fd srcdir,
  58.                  const char *restrict src,
  59.                  struct stat src_status,
  60.                  struct named_fd destdir,
  61.                  const char *restrict dest)
  62. {
  63. 	int in = openat(srcdir.fd, src, O_RDONLY | O_NOCTTY);
  64. 	if(in < 0)
  65. 	{
  66. 		fprintf(stderr,
  67. 		        "mv: error: Failed opening source '%s%s%s': %s\n",
  68. 		        srcdir.name,
  69. 		        srcdir.sep,
  70. 		        src,
  71. 		        strerror(errno));
  72. 		errno = 0;
  73. 		return -1;
  74. 	}
  76. 	int out = openat(destdir.fd, dest, O_WRONLY | O_CREAT | O_NOCTTY, src_status.st_mode);
  77. 	if(out < 0)
  78. 	{
  79. 		fprintf(stderr,
  80. 		        "mv: error: Failed opening destination '%s%s%s': %s\n",
  81. 		        destdir.name,
  82. 		        destdir.sep,
  83. 		        dest,
  84. 		        strerror(errno));
  85. 		errno = 0;
  86. 		return -1;
  87. 	}
  89. 	const struct timespec times[2] = {src_status.st_atim, src_status.st_mtim};
  90. 	if(futimens(out, times) != 0)
  91. 	{
  92. 		fprintf(stderr,
  93. 		        "mv: warning: Failed copying access & modification times to '%s%s%s': %s\n",
  94. 		        destdir.name,
  95. 		        destdir.sep,
  96. 		        dest,
  97. 		        strerror(errno));
  98. 		errno = 0;
  99. 	}
  101. 	if(fchown(out, src_status.st_uid, src_status.st_gid) != 0)
  102. 	{
  103. 		fprintf(stderr,
  104. 		        "mv: warning: Failed copying owner & group to '%s%s%s': %s\n",
  105. 		        destdir.name,
  106. 		        destdir.sep,
  107. 		        dest,
  108. 		        strerror(errno));
  109. 		errno = 0;
  110. 	}
  112. 	if(auto_file_copy(in, out, src_status.st_size, 0) < 0) return -1;
  114. 	return unlinkat(srcdir.fd, src, 0);
  115. }
  117. static int
  118. rename_dir_entries(struct named_fd srcdir, struct named_fd destdir)
  119. {
  120. 	DIR *dirsrc = fdopendir(srcdir.fd);
  121. 	if(dirsrc == NULL)
  122. 	{
  123. 		fprintf(stderr,
  124. 		        "mv: error: Failed fd-opening source directory '%s': %s\n",
  125. 		        srcdir.name ? srcdir.name : ".",
  126. 		        strerror(errno));
  127. 		return -1;
  128. 	}
  130. 	while(true)
  131. 	{
  132. 		errno = 0;
  133. 		struct dirent *dirsrc_ent = readdir(dirsrc);
  134. 		if(dirsrc_ent == NULL)
  135. 		{
  136. 			if(errno == 0) break;
  138. 			fprintf(stderr,
  139. 			        "mv: error: Failed reading source directory '%s': %s\n",
  140. 			        srcdir.name ? srcdir.name : ".",
  141. 			        strerror(errno));
  142. 			closedir(dirsrc);
  143. 			errno = 0;
  144. 			return -1;
  145. 		}
  147. 		if(strcmp(dirsrc_ent->d_name, ".") == 0) continue;
  148. 		if(strcmp(dirsrc_ent->d_name, "..") == 0) continue;
  150. 		if(do_renameat(srcdir, dirsrc_ent->d_name, destdir, dirsrc_ent->d_name) < 0)
  151. 		{
  152. 			closedir(dirsrc);
  153. 			return -1;
  154. 		}
  155. 	}
  157. 	return 0;
  158. }
  160. static int
  161. do_renameat(struct named_fd srcdir,
  162.             const char *restrict src,
  163.             struct named_fd destdir,
  164.             const char *restrict dest)
  165. {
  166. 	if(destdir.fd == srcdir.fd && strcmp(src, dest) == 0)
  167. 	{
  168. 		fprintf(stderr, "mv: error: Passed to both source and destination: '%s'\n", src);
  169. 		return -1;
  170. 	}
  172. 	struct stat src_status;
  173. 	if(fstatat(srcdir.fd, src, &src_status, AT_SYMLINK_NOFOLLOW) < 0)
  174. 	{
  175. 		fprintf(stderr,
  176. 		        "mv: error: Failed getting status for source file '%s%s%s': %s\n",
  177. 		        srcdir.name,
  178. 		        srcdir.sep,
  179. 		        src,
  180. 		        strerror(errno));
  181. 		return -1;
  182. 	}
  183. 	if(S_ISLNK(src_status.st_mode))
  184. 	{
  185. 		struct stat src_link_status;
  186. 		if(fstatat(srcdir.fd, src, &src_link_status, 0) == 0)
  187. 		{
  188. 			src_status = src_link_status;
  189. 		}
  190. 	}
  192. 	errno = 0;
  194. 	struct stat dest_status;
  195. 	int ret = fstatat(destdir.fd, dest, &dest_status, 0);
  196. 	if(ret < 0 && errno != ENOENT)
  197. 	{
  198. 		fprintf(stderr,
  199. 		        "mv: error: Failed getting status for destination file '%s%s%s': %s\n",
  200. 		        destdir.name,
  201. 		        destdir.sep,
  202. 		        dest,
  203. 		        strerror(errno));
  204. 		return -1;
  205. 	}
  207. 	errno = 0;
  209. 	if(ret == 0)
  210. 	{
  211. 		if(dest_status.st_ino == src_status.st_ino && dest_status.st_dev == src_status.st_dev) return 0;
  213. 		if(no_clob)
  214. 		{
  215. 			fprintf(stderr,
  216. 			        "mv: error: Destination file '%s%s%s' already exists\n",
  217. 			        destdir.name,
  218. 			        destdir.sep,
  219. 			        dest);
  220. 			return -1;
  221. 		}
  223. 		if(!force)
  224. 		{
  225. 			if(interact)
  226. 			{
  227. 				if(!consentf("mv: Destination file '%s%s%s' already exists, overwrite? [yN] ",
  228. 				             destdir.name,
  229. 				             destdir.sep,
  230. 				             dest))
  231. 					return 0;
  232. 			}
  233. 			else if(stdin_tty)
  234. 			{
  235. 				if(faccessat(destdir.fd, dest, W_OK, 0) == 0)
  236. 				{
  237. 					if(!consentf(
  238. 					       "mv: error: No write permissions for destination file '%s%s%s', overwrite? [yN] ",
  239. 					       destdir.name,
  240. 					       destdir.sep,
  241. 					       dest))
  242. 						return 0;
  243. 				}
  244. 				else
  245. 				{
  246. 					errno = 0;
  247. 				}
  248. 			}
  249. 		}
  250. 	}
  252. 	if(renameat(srcdir.fd, src, destdir.fd, dest) < 0)
  253. 	{
  254. 		switch(errno)
  255. 		{
  256. 		case EXDEV:
  257. 			errno = 0;
  259. 			if(S_ISDIR(src_status.st_mode))
  260. 			{
  261. 				char child_srcdir_name[PATH_MAX] = "";
  262. 				snprintf(child_srcdir_name, PATH_MAX, "%s%s%s", srcdir.name, srcdir.sep, src);
  264. 				struct named_fd child_srcdir = {
  265. 				    .fd = openat(srcdir.fd, src, O_RDONLY | O_DIRECTORY | O_CLOEXEC),
  266. 				    .name = child_srcdir_name,
  267. 				    .sep = "/",
  268. 				};
  270. 				if(child_srcdir.fd < 0)
  271. 				{
  272. 					fprintf(stderr,
  273. 					        "mv: error: Failed opening source directory '%s%s%s': %s\n",
  274. 					        srcdir.name,
  275. 					        srcdir.sep,
  276. 					        src,
  277. 					        strerror(errno));
  278. 					return -1;
  279. 				}
  281. 				if(mkdirat(destdir.fd, dest, src_status.st_mode) < 0)
  282. 				{
  283. 					fprintf(stderr,
  284. 					        "mv: error: Failed creating destination directory '%s%s%s': %s\n",
  285. 					        destdir.name,
  286. 					        destdir.sep,
  287. 					        dest,
  288. 					        strerror(errno));
  289. 					return -1;
  290. 				}
  292. 				char child_destdir_name[PATH_MAX] = "";
  293. 				snprintf(child_destdir_name, PATH_MAX, "%s%s%s", destdir.name, destdir.sep, dest);
  294. 				struct named_fd child_destdir = {
  295. 				    .fd = openat(destdir.fd, dest, O_RDONLY | O_DIRECTORY | O_CLOEXEC),
  296. 				    .name = child_destdir_name,
  297. 				    .sep = "/",
  298. 				};
  300. 				if(rename_dir_entries(child_srcdir, child_destdir) < 0) return -1;
  302. 				close(child_srcdir.fd);
  303. 				close(child_destdir.fd);
  305. 				if(unlinkat(srcdir.fd, src, AT_REMOVEDIR) < 0)
  306. 				{
  307. 					fprintf(stderr,
  308. 					        "mv: error: Failed removing source directory '%s%s%s': %s\n",
  309. 					        srcdir.name,
  310. 					        srcdir.sep,
  311. 					        src,
  312. 					        strerror(errno));
  313. 					return -1;
  314. 				}
  315. 			}
  316. 			else
  317. 			{
  318. 				if(copy_file_unlink(srcdir, src, src_status, destdir, dest) < 0) return -1;
  319. 			}
  320. 			break;
  321. 		case EISDIR:
  322. 		case ENOTDIR:
  323. 			if(destdir.fd != AT_FDCWD)
  324. 			{
  325. 				fprintf(
  326. 				    stderr, "mv: error: Failed moving '%s' into '%s': %s\n", src, dest, strerror(errno));
  327. 				return -1;
  328. 			}
  330. 			int tmp_destdir = openat(destdir.fd, dest, O_RDONLY | O_DIRECTORY | O_CLOEXEC);
  331. 			if(tmp_destdir < 0)
  332. 			{
  333. 				fprintf(stderr,
  334. 				        "mv: error: Failed opening destination directory '%s': %s\n",
  335. 				        dest,
  336. 				        strerror(errno));
  337. 				return -1;
  338. 			}
  340. 			if(renameat(srcdir.fd, src, tmp_destdir, src) < 0)
  341. 			{
  342. 				fprintf(stderr,
  343. 				        "mv: error: Failed moving '%s' into directory '%s': %s\n",
  344. 				        src,
  345. 				        dest,
  346. 				        strerror(errno));
  347. 				return -1;
  348. 			}
  350. 			if(close(tmp_destdir) < 0)
  351. 			{
  352. 				fprintf(stderr, "mv: error: Failed closing directory '%s': %s\n", dest, strerror(errno));
  353. 				return -1;
  354. 			}
  355. 			break;
  356. 		default:
  357. 			fprintf(stderr,
  358. 			        "mv: error: Failed moving '%s' to '%s%s%s': %s\n",
  359. 			        src,
  360. 			        destdir.name,
  361. 			        destdir.sep,
  362. 			        dest,
  363. 			        strerror(errno));
  364. 			return -1;
  365. 		}
  366. 	}
  368. 	if(verbose)
  369. 		fprintf(stderr, "mv: renamed '%s' -> '%s%s%s'\n", src, destdir.name, destdir.sep, dest);
  371. 	return 0;
  372. }
  374. static void
  375. usage(void)
  376. {
  377. 	fprintf(stderr, "Usage: mv [-f|-i|-n] [-v] source dest\n");
  378. 	fprintf(stderr, "       mv [-f|-i|-n] [-v] source... destdir\n");
  379. 	fprintf(stderr, "       mv [-f|-i|-n] [-v] -t destdir source...\n");
  380. }
  382. int
  383. main(int argc, char *argv[])
  384. {
  385. 	struct named_fd destdir = {
  386. 	    .fd = AT_FDCWD,
  387. 	    .name = "",
  388. 	    .sep = "",
  389. 	};
  390. 	struct named_fd srcdir = {
  391. 	    .fd = AT_FDCWD,
  392. 	    .name = "",
  393. 	    .sep = "",
  394. 	};
  396. 	for(int c = -1; (c = getopt_nolong(argc, argv, ":fint:v")) != -1;)
  397. 	{
  398. 		switch(c)
  399. 		{
  400. 		case 'f':
  401. 			force = true;
  402. 			interact = false;
  403. 			no_clob = false;
  404. 			break;
  405. 		case 'i':
  406. 			force = false;
  407. 			interact = true;
  408. 			no_clob = false;
  409. 			break;
  410. 		case 'n':
  411. 			force = false;
  412. 			interact = false;
  413. 			no_clob = true;
  414. 			break;
  415. 		case 't':
  416. 			destdir.name = optarg;
  417. 			destdir.sep = "/";
  418. 			destdir.fd = open(optarg, O_RDONLY | O_DIRECTORY | O_CLOEXEC);
  420. 			if(destdir.fd < 0)
  421. 			{
  422. 				fprintf(stderr,
  423. 				        "mv: error: Failed opening destination directory '%s': %s\n",
  424. 				        optarg,
  425. 				        strerror(errno));
  426. 				return 1;
  427. 			}
  428. 			break;
  429. 		case 'v':
  430. 			verbose = true;
  431. 			break;
  432. 		case ':':
  433. 			fprintf(stderr, "mv: error: Missing operand for option: '-%c'\n", optopt);
  434. 			usage();
  435. 			return 1;
  436. 		case '?':
  437. 			if(!got_long_opt) fprintf(stderr, "mv: error: Unrecognised option: '-%c'\n", optopt);
  438. 			usage();
  439. 			return 1;
  440. 		}
  441. 	}
  443. 	argc -= optind;
  444. 	argv += optind;
  446. 	char *lc_all = setlocale(LC_ALL, "");
  447. 	if(lc_all == NULL)
  448. 	{
  449. 		fprintf(stderr,
  450. 		        "%s: warning: Failed loading locales. setlocale(LC_ALL, \"\"): %s\n",
  451. 		        argv0,
  452. 		        strerror(errno));
  453. 	}
  454. 	errno = 0;
  456. 	consent_init();
  458. 	stdin_tty = isatty(STDIN_FILENO);
  459. 	if(!stdin_tty) errno = 0;
  461. 	if(destdir.fd == AT_FDCWD)
  462. 	{
  463. 		if(argc <= 1)
  464. 		{
  465. 			fprintf(stderr, "mv: error: Not enough operands, %d given, expect >= 2\n", argc);
  466. 			return 1;
  467. 		}
  469. 		struct stat dest_status;
  470. 		int ret_stat = fstatat(destdir.fd, argv[1], &dest_status, 0);
  471. 		if(
  472. 		    // clang-format off
  473. 		    argc == 2 && (
  474. 		        (ret_stat != 0 && errno == ENOENT) ||
  475. 		        (ret_stat == 0 && !S_ISDIR(dest_status.st_mode))
  476. 		    )
  477. 		    // clang-format on
  478. 		)
  479. 		{
  480. 			int ret = do_renameat(srcdir, argv[0], destdir, argv[1]);
  482. 			consent_finish();
  483. 			return ret < 0 ? 1 : 0;
  484. 		}
  485. 		errno = 0;
  487. 		argc--;
  488. 		destdir.name = argv[argc];
  489. 		destdir.sep = "/";
  490. 		destdir.fd = open(destdir.name, O_RDONLY | O_DIRECTORY | O_CLOEXEC);
  491. 		if(destdir.fd < 0)
  492. 		{
  493. 			fprintf(stderr,
  494. 			        "mv: error: Failed opening destination directory '%s': %s\n",
  495. 			        destdir.name,
  496. 			        strerror(errno));
  498. 			consent_finish();
  499. 			return 1;
  500. 		}
  501. 	}
  503. 	for(int i = 0; i < argc; i++)
  504. 	{
  505. 		char arg[PATH_MAX] = "";
  506. 		strcpy(arg, argv[i]);
  508. 		char *filename = basename(arg);
  510. 		if(do_renameat(srcdir, argv[i], destdir, filename) < 0)
  511. 		{
  512. 			consent_finish();
  513. 			return 1;
  514. 		}
  515. 	}
  517. 	consent_finish();
  519. 	if(close(destdir.fd) < 0)
  520. 	{
  521. 		fprintf(
  522. 		    stderr, "mv: error: Failed closing directory '%s': %s\n", destdir.name, strerror(errno));
  523. 		return 1;
  524. 	}
  526. 	return 0;
  527. }