215 lines
		
	
	
		
			6.8 KiB
		
	
	
	
		
			Perl
		
	
	
		
			Executable File
		
	
			
		
		
	
	
			215 lines
		
	
	
		
			6.8 KiB
		
	
	
	
		
			Perl
		
	
	
		
			Executable File
		
	
| #!/usr/bin/perl
 | |
| #
 | |
| # Copyright (c) 2006 Josh England
 | |
| #
 | |
| # This script can be used to save/restore full permissions and ownership data
 | |
| # within a git working tree.
 | |
| #
 | |
| # To save permissions/ownership data, place this script in your .git/hooks
 | |
| # directory and enable a `pre-commit` hook with the following lines:
 | |
| #      #!/bin/sh
 | |
| #     SUBDIRECTORY_OK=1 . git-sh-setup
 | |
| #     $GIT_DIR/hooks/setgitperms.perl -r
 | |
| #
 | |
| # To restore permissions/ownership data, place this script in your .git/hooks
 | |
| # directory and enable a `post-merge` and `post-checkout` hook with the
 | |
| # following lines:
 | |
| #      #!/bin/sh
 | |
| #     SUBDIRECTORY_OK=1 . git-sh-setup
 | |
| #     $GIT_DIR/hooks/setgitperms.perl -w
 | |
| #
 | |
| use strict;
 | |
| use Getopt::Long;
 | |
| use File::Find;
 | |
| use File::Basename;
 | |
| 
 | |
| my $usage =
 | |
| "usage: setgitperms.perl [OPTION]... <--read|--write>
 | |
| This program uses a file `.gitmeta` to store/restore permissions and uid/gid
 | |
| info for all files/dirs tracked by git in the repository.
 | |
| 
 | |
| ---------------------------------Read Mode-------------------------------------
 | |
| -r,  --read         Reads perms/etc from working dir into a .gitmeta file
 | |
| -s,  --stdout       Output to stdout instead of .gitmeta
 | |
| -d,  --diff         Show unified diff of perms file (XOR with --stdout)
 | |
| 
 | |
| ---------------------------------Write Mode------------------------------------
 | |
| -w,  --write        Modify perms/etc in working dir to match the .gitmeta file
 | |
| -v,  --verbose      Be verbose
 | |
| 
 | |
| \n";
 | |
| 
 | |
| my ($stdout, $showdiff, $verbose, $read_mode, $write_mode);
 | |
| 
 | |
| if ((@ARGV < 0) || !GetOptions(
 | |
| 			       "stdout",         \$stdout,
 | |
| 			       "diff",           \$showdiff,
 | |
| 			       "read",           \$read_mode,
 | |
| 			       "write",          \$write_mode,
 | |
| 			       "verbose",        \$verbose,
 | |
| 			      )) { die $usage; }
 | |
| die $usage unless ($read_mode xor $write_mode);
 | |
| 
 | |
| my $topdir = `git rev-parse --show-cdup` or die "\n"; chomp $topdir;
 | |
| my $gitdir = $topdir . '.git';
 | |
| my $gitmeta = $topdir . '.gitmeta';
 | |
| 
 | |
| if ($write_mode) {
 | |
|     # Update the working dir permissions/ownership based on data from .gitmeta
 | |
|     open (IN, "<$gitmeta") or die "Could not open $gitmeta for reading: $!\n";
 | |
|     while (defined ($_ = <IN>)) {
 | |
| 	chomp;
 | |
| 	if (/^(.*)  mode=(\S+)\s+uid=(\d+)\s+gid=(\d+)/) {
 | |
| 	    # Compare recorded perms to actual perms in the working dir
 | |
| 	    my ($path, $mode, $uid, $gid) = ($1, $2, $3, $4);
 | |
| 	    my $fullpath = $topdir . $path;
 | |
| 	    my (undef,undef,$wmode,undef,$wuid,$wgid) = lstat($fullpath);
 | |
| 	    $wmode = sprintf "%04o", $wmode & 07777;
 | |
| 	    if ($mode ne $wmode) {
 | |
| 		$verbose && print "Updating permissions on $path: old=$wmode, new=$mode\n";
 | |
| 		chmod oct($mode), $fullpath;
 | |
| 	    }
 | |
| 	    if ($uid != $wuid || $gid != $wgid) {
 | |
| 		if ($verbose) {
 | |
| 		    # Print out user/group names instead of uid/gid
 | |
| 		    my $pwname  = getpwuid($uid);
 | |
| 		    my $grpname  = getgrgid($gid);
 | |
| 		    my $wpwname  = getpwuid($wuid);
 | |
| 		    my $wgrpname  = getgrgid($wgid);
 | |
| 		    $pwname = $uid if !defined $pwname;
 | |
| 		    $grpname = $gid if !defined $grpname;
 | |
| 		    $wpwname = $wuid if !defined $wpwname;
 | |
| 		    $wgrpname = $wgid if !defined $wgrpname;
 | |
| 
 | |
| 		    print "Updating uid/gid on $path: old=$wpwname/$wgrpname, new=$pwname/$grpname\n";
 | |
| 		}
 | |
| 		chown $uid, $gid, $fullpath;
 | |
| 	    }
 | |
| 	}
 | |
| 	else {
 | |
| 	    warn "Invalid input format in $gitmeta:\n\t$_\n";
 | |
| 	}
 | |
|     }
 | |
|     close IN;
 | |
| }
 | |
| elsif ($read_mode) {
 | |
|     # Handle merge conflicts in the .gitperms file
 | |
|     if (-e "$gitdir/MERGE_MSG") {
 | |
| 	if (`grep ====== $gitmeta`) {
 | |
| 	    # Conflict not resolved -- abort the commit
 | |
| 	    print "PERMISSIONS/OWNERSHIP CONFLICT\n";
 | |
| 	    print "    Resolve the conflict in the $gitmeta file and then run\n";
 | |
| 	    print "    `.git/hooks/setgitperms.perl --write` to reconcile.\n";
 | |
| 	    exit 1;
 | |
| 	}
 | |
| 	elsif (`grep $gitmeta $gitdir/MERGE_MSG`) {
 | |
| 	    # A conflict in .gitmeta has been manually resolved. Verify that
 | |
| 	    # the working dir perms matches the current .gitmeta perms for
 | |
| 	    # each file/dir that conflicted.
 | |
| 	    # This is here because a `setgitperms.perl --write` was not
 | |
| 	    # performed due to a merge conflict, so permissions/ownership
 | |
| 	    # may not be consistent with the manually merged .gitmeta file.
 | |
| 	    my @conflict_diff = `git show \$(cat $gitdir/MERGE_HEAD)`;
 | |
| 	    my @conflict_files;
 | |
| 	    my $metadiff = 0;
 | |
| 
 | |
| 	    # Build a list of files that conflicted from the .gitmeta diff
 | |
| 	    foreach my $line (@conflict_diff) {
 | |
| 		if ($line =~ m|^diff --git a/$gitmeta b/$gitmeta|) {
 | |
| 		    $metadiff = 1;
 | |
| 		}
 | |
| 		elsif ($line =~ /^diff --git/) {
 | |
| 		    $metadiff = 0;
 | |
| 		}
 | |
| 		elsif ($metadiff && $line =~ /^\+(.*)  mode=/) {
 | |
| 		    push @conflict_files, $1;
 | |
| 		}
 | |
| 	    }
 | |
| 
 | |
| 	    # Verify that each conflict file now has permissions consistent
 | |
| 	    # with the .gitmeta file
 | |
| 	    foreach my $file (@conflict_files) {
 | |
| 		my $absfile = $topdir . $file;
 | |
| 		my $gm_entry = `grep "^$file  mode=" $gitmeta`;
 | |
| 		if ($gm_entry =~ /mode=(\d+)  uid=(\d+)  gid=(\d+)/) {
 | |
| 		    my ($gm_mode, $gm_uid, $gm_gid) = ($1, $2, $3);
 | |
| 		    my (undef,undef,$mode,undef,$uid,$gid) = lstat("$absfile");
 | |
| 		    $mode = sprintf("%04o", $mode & 07777);
 | |
| 		    if (($gm_mode ne $mode) || ($gm_uid != $uid)
 | |
| 			|| ($gm_gid != $gid)) {
 | |
| 			print "PERMISSIONS/OWNERSHIP CONFLICT\n";
 | |
| 			print "    Mismatch found for file: $file\n";
 | |
| 			print "    Run `.git/hooks/setgitperms.perl --write` to reconcile.\n";
 | |
| 			exit 1;
 | |
| 		    }
 | |
| 		}
 | |
| 		else {
 | |
| 		    print "Warning! Permissions/ownership no longer being tracked for file: $file\n";
 | |
| 		}
 | |
| 	    }
 | |
| 	}
 | |
|     }
 | |
| 
 | |
|     # No merge conflicts -- write out perms/ownership data to .gitmeta file
 | |
|     unless ($stdout) {
 | |
| 	open (OUT, ">$gitmeta.tmp") or die "Could not open $gitmeta.tmp for writing: $!\n";
 | |
|     }
 | |
| 
 | |
|     my @files = `git ls-files`;
 | |
|     my %dirs;
 | |
| 
 | |
|     foreach my $path (@files) {
 | |
| 	chomp $path;
 | |
| 	# We have to manually add stats for parent directories
 | |
| 	my $parent = dirname($path);
 | |
| 	while (!exists $dirs{$parent}) {
 | |
| 	    $dirs{$parent} = 1;
 | |
| 	    next if $parent eq '.';
 | |
| 	    printstats($parent);
 | |
| 	    $parent = dirname($parent);
 | |
| 	}
 | |
| 	# Now the git-tracked file
 | |
| 	printstats($path);
 | |
|     }
 | |
| 
 | |
|     # diff the temporary metadata file to see if anything has changed
 | |
|     # If no metadata has changed, don't overwrite the real file
 | |
|     # This is just so `git commit -a` doesn't try to commit a bogus update
 | |
|     unless ($stdout) {
 | |
| 	if (! -e $gitmeta) {
 | |
| 	    rename "$gitmeta.tmp", $gitmeta;
 | |
| 	}
 | |
| 	else {
 | |
| 	    my $diff = `diff -U 0 $gitmeta $gitmeta.tmp`;
 | |
| 	    if ($diff ne '') {
 | |
| 		rename "$gitmeta.tmp", $gitmeta;
 | |
| 	    }
 | |
| 	    else {
 | |
| 		unlink "$gitmeta.tmp";
 | |
| 	    }
 | |
| 	    if ($showdiff) {
 | |
| 		print $diff;
 | |
| 	    }
 | |
| 	}
 | |
| 	close OUT;
 | |
|     }
 | |
|     # Make sure the .gitmeta file is tracked
 | |
|     system("git add $gitmeta");
 | |
| }
 | |
| 
 | |
| 
 | |
| sub printstats {
 | |
|     my $path = $_[0];
 | |
|     $path =~ s/@/\@/g;
 | |
|     my (undef,undef,$mode,undef,$uid,$gid) = lstat($path);
 | |
|     $path =~ s/%/\%/g;
 | |
|     if ($stdout) {
 | |
| 	print $path;
 | |
| 	printf "  mode=%04o  uid=$uid  gid=$gid\n", $mode & 07777;
 | |
|     }
 | |
|     else {
 | |
| 	print OUT $path;
 | |
| 	printf OUT "  mode=%04o  uid=$uid  gid=$gid\n", $mode & 07777;
 | |
|     }
 | |
| }
 |