Annotation of embedaddon/rsync/support/file-attr-restore, revision 1.1

1.1     ! misho       1: #!/usr/bin/perl
        !             2: # This script will parse the output of "find ARG [ARG...] -ls" and
        !             3: # apply (at your discretion) the permissions, owner, and group info
        !             4: # it reads onto any existing files and dirs (it doesn't try to affect
        !             5: # symlinks).  Run this with --help (-h) for a usage summary.
        !             6: 
        !             7: use strict;
        !             8: use Getopt::Long;
        !             9: 
        !            10: our($p_opt, $o_opt, $g_opt, $map_file, $dry_run, $verbosity, $help_opt);
        !            11: 
        !            12: &Getopt::Long::Configure('bundling');
        !            13: &usage if !&GetOptions(
        !            14:     'all|a' => sub { $p_opt = $o_opt = $g_opt = 1 },
        !            15:     'perms|p' => \$p_opt,
        !            16:     'owner|o' => \$o_opt,
        !            17:     'groups|g' => \$g_opt,
        !            18:     'map|m=s' => \$map_file,
        !            19:     'dry-run|n' => \$dry_run,
        !            20:     'help|h' => \$help_opt,
        !            21:     'verbose|v+' => \$verbosity,
        !            22: ) || $help_opt;
        !            23: 
        !            24: our(%uid_hash, %gid_hash);
        !            25: 
        !            26: $" = ', '; # How to join arrays referenced in double-quotes.
        !            27: 
        !            28: &parse_map_file($map_file) if defined $map_file;
        !            29: 
        !            30: my $detail_line = qr{
        !            31:     ^ \s* \d+ \s+             # ignore inode
        !            32:     \d+ \s+                   # ignore size
        !            33:     ([-bcdlps])               # 1. File type
        !            34:     ( [-r][-w][-xsS]          # 2. user-permissions
        !            35:       [-r][-w][-xsS]          #    group-permissions
        !            36:       [-r][-w][-xtT] ) \s+    #    other-permissions
        !            37:     \d+ \s+                   # ignore number of links
        !            38:     (\S+) \s+                 # 3. owner
        !            39:     (\S+) \s+                 # 4. group
        !            40:     (?: \d+ \s+ )?            # ignore size (when present)
        !            41:     \w+ \s+ \d+ \s+           # ignore month and date
        !            42:     \d+ (?: : \d+ )? \s+      # ignore time or year
        !            43:     ([^\r\n]+) $              # 5. name
        !            44: }x;
        !            45: 
        !            46: while (<>) {
        !            47:     my($type, $perms, $owner, $group, $name) = /$detail_line/;
        !            48:     die "Invalid input line $.:\n$_" unless defined $name;
        !            49:     die "A filename is not properly escaped:\n$_" unless $name =~ /^[^"\\]*(\\(\d\d\d|\D)[^"\\]*)*$/;
        !            50:     my $fn = $name;
        !            51:     $fn =~ s/\\(\d+|.)/ eval "\"\\$1\"" /eg;
        !            52:     if ($type eq '-') {
        !            53:        undef $type unless -f $fn;
        !            54:     } elsif ($type eq 'd') {
        !            55:        undef $type unless -d $fn;
        !            56:     } elsif ($type eq 'b') {
        !            57:        undef $type unless -b $fn;
        !            58:     } elsif ($type eq 'c') {
        !            59:        undef $type unless -c $fn;
        !            60:     } elsif ($type eq 'p') {
        !            61:        undef $type unless -p $fn;
        !            62:     } elsif ($type eq 's') {
        !            63:        undef $type unless -S $fn;
        !            64:     } else {
        !            65:        if ($verbosity) {
        !            66:            if ($type eq 'l') {
        !            67:                $name =~ s/ -> .*//;
        !            68:                $type = 'symlink';
        !            69:            } else {
        !            70:                $type = "type '$type'";
        !            71:            }
        !            72:            print "Skipping $name ($type ignored)\n";
        !            73:        }
        !            74:        next;
        !            75:     }
        !            76:     if (!defined $type) {
        !            77:        my $reason = -e _ ? "types don't match" : 'missing';
        !            78:        print "Skipping $name ($reason)\n";
        !            79:        next;
        !            80:     }
        !            81:     my($cur_mode, $cur_uid, $cur_gid) = (stat(_))[2,4,5];
        !            82:     $cur_mode &= 07777;
        !            83:     my $highs = join('', $perms =~ /..(.)..(.)..(.)/);
        !            84:     $highs =~ tr/-rwxSTst/00001111/;
        !            85:     $perms =~ tr/-STrwxst/00011111/;
        !            86:     my $mode = $p_opt ? oct('0b' . $highs . $perms) : $cur_mode;
        !            87:     my $uid = $o_opt ? $uid_hash{$owner} : $cur_uid;
        !            88:     if (!defined $uid) {
        !            89:        if ($owner =~ /^\d+$/) {
        !            90:            $uid = $owner;
        !            91:        } else {
        !            92:            $uid = getpwnam($owner);
        !            93:        }
        !            94:        $uid_hash{$owner} = $uid;
        !            95:     }
        !            96:     my $gid = $g_opt ? $gid_hash{$group} : $cur_gid;
        !            97:     if (!defined $gid) {
        !            98:        if ($group =~ /^\d+$/) {
        !            99:            $gid = $group;
        !           100:        } else {
        !           101:            $gid = getgrnam($group);
        !           102:        }
        !           103:        $gid_hash{$group} = $gid;
        !           104:     }
        !           105: 
        !           106:     my @changes;
        !           107:     if ($mode != $cur_mode) {
        !           108:        push(@changes, 'permissions');
        !           109:        if (!$dry_run && !chmod($mode, $fn)) {
        !           110:            warn "chmod($mode, \"$name\") failed: $!\n";
        !           111:        }
        !           112:     }
        !           113:     if ($uid != $cur_uid || $gid != $cur_gid) {
        !           114:        push(@changes, 'owner') if $uid != $cur_uid;
        !           115:        push(@changes, 'group') if $gid != $cur_gid;
        !           116:        if (!$dry_run) {
        !           117:            if (!chown($uid, $gid, $fn)) {
        !           118:                warn "chown($uid, $gid, \"$name\") failed: $!\n";
        !           119:            }
        !           120:            if (($mode & 06000) && !chmod($mode, $fn)) {
        !           121:                warn "post-chown chmod($mode, \"$name\") failed: $!\n";
        !           122:            }
        !           123:        }
        !           124:     }
        !           125:     if (@changes) {
        !           126:        print "$name: changed @changes\n";
        !           127:     } elsif ($verbosity) {
        !           128:        print "$name: OK\n";
        !           129:     }
        !           130: }
        !           131: exit;
        !           132: 
        !           133: sub parse_map_file
        !           134: {
        !           135:     my($fn) = @_;
        !           136:     open(IN, $fn) or die "Unable to open $fn: $!\n";
        !           137:     while (<IN>) {
        !           138:        if (/^user\s+(\S+)\s+(\S+)/) {
        !           139:            $uid_hash{$1} = $2;
        !           140:        } elsif (/^group\s+(\S+)\s+(\S+)/) {
        !           141:            $gid_hash{$1} = $2;
        !           142:        } else {
        !           143:            die "Invalid line #$. in mapfile `$fn':\n$_";
        !           144:        }
        !           145:     }
        !           146:     close IN;
        !           147: }
        !           148: 
        !           149: sub usage
        !           150: {
        !           151:     die <<EOT;
        !           152: Usage: file-attr-restore [OPTIONS] FILE [FILE...]
        !           153:  -a, --all       Restore all the attributes (-pog)
        !           154:  -p, --perms     Restore the permissions
        !           155:  -o, --owner     Restore the ownership
        !           156:  -g, --groups    Restore the group
        !           157:  -m, --map=FILE  Read user/group mappings from FILE
        !           158:  -n, --dry-run   Don't actually make the changes
        !           159:  -v, --verbose   Increase verbosity
        !           160:  -h, --help      Show this help text
        !           161: 
        !           162: The FILE arg(s) should have been created by running the "find"
        !           163: program with "-ls" as the output specifier.
        !           164: 
        !           165: The input file for the --map option must be in this format:
        !           166: 
        !           167:     user FROM TO
        !           168:     group FROM TO
        !           169: 
        !           170: The "FROM" should be an user/group mentioned in the input, and the TO
        !           171: should be either a uid/gid number, or a local user/group name.
        !           172: EOT
        !           173: }

FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>