* in cases involving special characters. Here a C version.
*/
#include <stdio.h>
+#ifdef HAVE_STDIO_EXT_H
+# include <stdio_ext.h>
+#endif
+#ifndef HAVE___FPURGE
+# ifdef HAVE_FPURGE
+# define HAVE___FPURGE 1
+# define __fpurge fpurge
+# endif
+#endif
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <getopt.h>
+#include <fcntl.h>
+#include <libgen.h>
#include <unistd.h>
+#include <termios.h>
#include <sys/types.h>
#include <sys/stat.h>
#include "xalloc.h"
#include "c.h"
#include "closestream.h"
+#include "optutils.h"
+#include "rpmatch.h"
-static int do_rename(char *from, char *to, char *s, int verbose, int symtarget)
-{
- char *newname, *where, *p, *q, *target;
- int flen, tlen, slen;
- struct stat sb;
+#define RENAME_EXIT_SOMEOK 2
+#define RENAME_EXIT_NOTHING 4
+#define RENAME_EXIT_UNEXPLAINED 64
- if (symtarget) {
- if (lstat(s, &sb) == -1)
- err(EXIT_FAILURE, _("%s: lstat failed"), s);
+static int tty_cbreak = 0;
+static int all = 0;
+static int last = 0;
- if (!S_ISLNK(sb.st_mode))
- errx(EXIT_FAILURE, _("%s: not a symbolic link"), s);
+/* Find the first place in `orig` where we'll perform a replacement. NULL if
+ there are no replacements to do. */
+static char *find_initial_replace(char *from, char *to, char *orig)
+{
+ char *search_start = orig;
- target = xmalloc(sb.st_size + 1);
- if (readlink(s, target, sb.st_size + 1) < 0)
- err(EXIT_FAILURE, _("%s: readlink failed"), s);
+ if (strchr(from, '/') == NULL && strchr(to, '/') == NULL) {
+ /* We only want to search in the final path component. Don't
+ include the final '/' in that component; if `from` is empty,
+ we want it to first match after the '/', not before. */
+ search_start = basename(orig);
+ }
- target[sb.st_size] = '\0';
- where = strstr(target, from);
- } else
- where = strstr(s, from);
+ return strstr(search_start, from);
+}
- if (where == NULL)
- return 0;
+static int string_replace(char *from, char *to, char *orig, char **newname)
+{
+ char *p, *q, *where;
+ size_t count = 0, fromlen = strlen(from);
- flen = strlen(from);
- tlen = strlen(to);
- if (symtarget) {
- slen = strlen(target);
- p = target;
- } else {
- slen = strlen(s);
- p = s;
+ p = where = find_initial_replace(from, to, orig);
+ if (where == NULL)
+ return 1;
+ count++;
+ while ((all || last) && p && *p) {
+ p = strstr(p + (last ? 1 : max(fromlen, (size_t) 1)), from);
+ if (p) {
+ if (all)
+ count++;
+ if (last)
+ where = p;
+ }
+ }
+ p = orig;
+ *newname = xmalloc(strlen(orig) - count * fromlen + count * strlen(to) + 1);
+ q = *newname;
+ while (count--) {
+ while (p < where)
+ *q++ = *p++;
+ p = to;
+ while (*p)
+ *q++ = *p++;
+ if (fromlen > 0) {
+ p = where + fromlen;
+ where = strstr(p, from);
+ } else {
+ p = where;
+ where += 1;
+ }
}
- newname = xmalloc(tlen + slen + 1);
-
- q = newname;
- while (p < where)
- *q++ = *p++;
- p = to;
- while (*p)
- *q++ = *p++;
- p = where + flen;
while (*p)
*q++ = *p++;
*q = 0;
+ return 0;
+}
- if (symtarget) {
- if (0 > unlink(s))
- err(EXIT_FAILURE, _("%s: unlink failed"), s);
- if (symlink(newname, s) != 0)
- err(EXIT_FAILURE, _("%s: symlinking to %s failed"), s, newname);
- if (verbose)
- printf("%s: `%s' -> `%s'\n", s, target, newname);
- } else {
- if (rename(s, newname) != 0)
- err(EXIT_FAILURE, _("%s: rename to %s failed"), s, newname);
+static int ask(char *name)
+{
+ int c;
+ char buf[2];
+ printf(_("%s: overwrite `%s'? "), program_invocation_short_name, name);
+ fflush(stdout);
+ if ((c = fgetc(stdin)) == EOF) {
+ buf[0] = 'n';
+ printf("n\n");
+ }
+ else {
+ buf[0] = c;
+ if (c != '\n' && tty_cbreak) {
+#ifdef HAVE___FPURGE
+ /* Possibly purge a multi-byte character; or do a
+ required purge of the rest of the line (including
+ the newline) if the tty has been put back in
+ canonical mode (for example by a shell after a
+ SIGTSTP signal). */
+ __fpurge(stdin);
+#endif
+ printf("\n");
+ }
+ else if (c != '\n')
+ while ((c = fgetc(stdin)) != '\n' && c != EOF);
+ }
+ buf[1] = '\0';
+ if (rpmatch(buf) == RPMATCH_YES)
+ return 0;
+
+ return 1;
+}
+
+static int do_symlink(char *from, char *to, char *s, int verbose, int noact,
+ int nooverwrite, int interactive)
+{
+ char *newname = NULL, *target = NULL;
+ int ret = 1;
+ ssize_t ssz;
+ struct stat sb;
+
+ if ( faccessat(AT_FDCWD, s, F_OK, AT_SYMLINK_NOFOLLOW) != 0 &&
+ errno != EINVAL )
+ /* Skip if AT_SYMLINK_NOFOLLOW is not supported; lstat() below will
+ detect the access error */
+ {
+ warn(_("%s: not accessible"), s);
+ return 2;
+ }
+
+ if (lstat(s, &sb) == -1) {
+ warn(_("stat of %s failed"), s);
+ return 2;
+ }
+ if (!S_ISLNK(sb.st_mode)) {
+ warnx(_("%s: not a symbolic link"), s);
+ return 2;
+ }
+ target = xmalloc(sb.st_size + 1);
+
+ ssz = readlink(s, target, sb.st_size + 1);
+ if (ssz < 0) {
+ warn(_("%s: readlink failed"), s);
+ free(target);
+ return 2;
+ }
+ target[ssz] = '\0';
+
+ if (string_replace(from, to, target, &newname) != 0)
+ ret = 0;
+
+ if (ret == 1 && (nooverwrite || interactive) && lstat(newname, &sb) != 0)
+ nooverwrite = interactive = 0;
+
+ if ( ret == 1 &&
+ (nooverwrite || (interactive && (noact || ask(newname) != 0))) )
+ {
if (verbose)
- printf("`%s' -> `%s'\n", s, newname);
+ printf(_("Skipping existing link: `%s' -> `%s'\n"), s, target);
+ ret = 0;
+ }
+
+ if (ret == 1) {
+ if (!noact && 0 > unlink(s)) {
+ warn(_("%s: unlink failed"), s);
+ ret = 2;
+ }
+ else if (!noact && symlink(newname, s) != 0) {
+ warn(_("%s: symlinking to %s failed"), s, newname);
+ ret = 2;
+ }
+ }
+ if (verbose && (noact || ret == 1))
+ printf("%s: `%s' -> `%s'\n", s, target, newname);
+ free(newname);
+ free(target);
+ return ret;
+}
+
+static int do_file(char *from, char *to, char *s, int verbose, int noact,
+ int nooverwrite, int interactive)
+{
+ char *newname = NULL;
+ int ret = 1;
+ struct stat sb;
+
+ if ( faccessat(AT_FDCWD, s, F_OK, AT_SYMLINK_NOFOLLOW) != 0 &&
+ errno != EINVAL )
+ /* Skip if AT_SYMLINK_NOFOLLOW is not supported; lstat() below will
+ detect the access error */
+ {
+ warn(_("%s: not accessible"), s);
+ return 2;
+ }
+
+ if (lstat(s, &sb) == -1) {
+ warn(_("stat of %s failed"), s);
+ return 2;
}
+ if (string_replace(from, to, s, &newname) != 0)
+ return 0;
+ if ((nooverwrite || interactive) && access(newname, F_OK) != 0)
+ nooverwrite = interactive = 0;
+ if (nooverwrite || (interactive && (noact || ask(newname) != 0))) {
+ if (verbose)
+ printf(_("Skipping existing file: `%s'\n"), newname);
+ ret = 0;
+ }
+ else if (!noact && rename(s, newname) != 0) {
+ warn(_("%s: rename to %s failed"), s, newname);
+ ret = 2;
+ }
+ if (verbose && (noact || ret == 1))
+ printf("`%s' -> `%s'\n", s, newname);
free(newname);
- return 1;
+ return ret;
}
-static void __attribute__ ((__noreturn__)) usage(FILE * out)
+static void __attribute__((__noreturn__)) usage(void)
{
- fputs(_("\nUsage:\n"), out);
+ FILE *out = stdout;
+ fputs(USAGE_HEADER, out);
fprintf(out,
- _(" %s [options] expression replacement file...\n"),
+ _(" %s [options] <expression> <replacement> <file>...\n"),
program_invocation_short_name);
- fputs(_("\nOptions:\n"), out);
- fputs(_(" -v, --verbose explain what is being done\n"
- " -V, --version output version information and exit\n"
- " -s, --symlink act on symlink target\n"
- " -h, --help display this help and exit\n\n"), out);
+ fputs(USAGE_SEPARATOR, out);
+ fputs(_("Rename files.\n"), out);
- exit(out == stderr ? EXIT_FAILURE : EXIT_SUCCESS);
+ fputs(USAGE_OPTIONS, out);
+ fputs(_(" -v, --verbose explain what is being done\n"), out);
+ fputs(_(" -s, --symlink act on the target of symlinks\n"), out);
+ fputs(_(" -n, --no-act do not make any changes\n"), out);
+ fputs(_(" -a, --all replace all occurrences\n"), out);
+ fputs(_(" -l, --last replace only the last occurrence\n"), out);
+ fputs(_(" -o, --no-overwrite don't overwrite existing files\n"), out);
+ fputs(_(" -i, --interactive prompt before overwrite\n"), out);
+ fputs(USAGE_SEPARATOR, out);
+ fprintf(out, USAGE_HELP_OPTIONS(21));
+ fprintf(out, USAGE_MAN_TAIL("rename(1)"));
+ exit(EXIT_SUCCESS);
}
int main(int argc, char **argv)
{
char *from, *to;
- int i, c, symtarget=0, verbose = 0;
+ int i, c, ret = 0, verbose = 0, noact = 0, nooverwrite = 0, interactive = 0;
+ struct termios tio;
+ int (*do_rename)(char *from, char *to, char *s, int verbose, int noact,
+ int nooverwrite, int interactive) = do_file;
static const struct option longopts[] = {
{"verbose", no_argument, NULL, 'v'},
{"version", no_argument, NULL, 'V'},
{"help", no_argument, NULL, 'h'},
+ {"all", no_argument, NULL, 'a'},
+ {"last", no_argument, NULL, 'l'},
+ {"no-act", no_argument, NULL, 'n'},
+ {"no-overwrite", no_argument, NULL, 'o'},
+ {"interactive", no_argument, NULL, 'i'},
{"symlink", no_argument, NULL, 's'},
{NULL, 0, NULL, 0}
};
+ static const ul_excl_t excl[] = { /* rows and cols in ASCII order */
+ { 'a','l' },
+ { 'i','o' },
+ { 0 }
+ };
+ int excl_st[ARRAY_SIZE(excl)] = UL_EXCL_STATUS_INIT;
setlocale(LC_ALL, "");
bindtextdomain(PACKAGE, LOCALEDIR);
textdomain(PACKAGE);
- atexit(close_stdout);
+ close_stdout_atexit();
- while ((c = getopt_long(argc, argv, "vsVh", longopts, NULL)) != -1)
+ while ((c = getopt_long(argc, argv, "vsVhnaloi", longopts, NULL)) != -1) {
+ err_exclusive_options(c, longopts, excl, excl_st);
switch (c) {
+ case 'n':
+ noact = 1;
+ break;
+ case 'a':
+ all = 1;
+ break;
+ case 'l':
+ last = 1;
+ break;
case 'v':
verbose = 1;
break;
+ case 'o':
+ nooverwrite = 1;
+ break;
+ case 'i':
+ interactive = 1;
+ break;
case 's':
- symtarget = 1;
+ do_rename = do_symlink;
break;
+
case 'V':
- printf(UTIL_LINUX_VERSION);
- return EXIT_SUCCESS;
+ print_version(EXIT_SUCCESS);
case 'h':
- usage(stdout);
+ usage();
default:
- usage(stderr);
+ errtryhelp(EXIT_FAILURE);
}
+ }
argc -= optind;
argv += optind;
if (argc < 3) {
- warnx("not enough arguments");
- usage(stderr);
+ warnx(_("not enough arguments"));
+ errtryhelp(EXIT_FAILURE);
}
from = argv[0];
to = argv[1];
+ if (!strcmp(from, to))
+ return RENAME_EXIT_NOTHING;
+
+ tty_cbreak = 0;
+ if (interactive && isatty(STDIN_FILENO) != 0) {
+ if (tcgetattr(STDIN_FILENO, &tio) != 0)
+ warn(_("failed to get terminal attributes"));
+ else if (!(tio.c_lflag & ICANON) && tio.c_cc[VMIN] == 1)
+ tty_cbreak = 1;
+ }
+
for (i = 2; i < argc; i++)
- do_rename(from, to, argv[i], verbose, symtarget);
+ ret |= do_rename(from, to, argv[i], verbose, noact, nooverwrite, interactive);
- return EXIT_SUCCESS;
+ switch (ret) {
+ case 0:
+ return RENAME_EXIT_NOTHING;
+ case 1:
+ return EXIT_SUCCESS;
+ case 2:
+ return EXIT_FAILURE;
+ case 3:
+ return RENAME_EXIT_SOMEOK;
+ default:
+ return RENAME_EXIT_UNEXPLAINED;
+ }
}
projects / thirdparty / util-linux.git / blobdiff