dgit-badcommit-fixup

   1 #!/usr/bin/perl -w
   2 #
   3 # Script to help with fallout from #849041.
   4 #
   5 # usage:
   6 #   dgit-badcommit-fixup --test
   7 #   dgit-badcommit-fixup --real
   8
   9 use strict;
  10
  11 use POSIX;
  12 use IPC::Open2;
  13 use Data::Dumper;
  14
  15 my $real;
  16
  17 foreach my $a (@ARGV) {
  18     if ($a eq '--test') {
  19         $real = 0;
  20     } elsif ($a eq '--real') {
  21         $real = 1;
  22     } else {
  23         die "$a ?";
  24     }
  25 }
  26
  27 die unless defined $real;
  28
  29 my $gcfpid = open2 \*GCFO, \*GCFI, 'git cat-file --batch' or die $!;
  30
  31 our %count;
  32
  33 no warnings qw(recursion);
  34
  35 sub runcmd {
  36     system @_ and die "@_ $! $?";
  37 }
  38
  39 $!=0; $?=0;
  40 my $bare = `git rev-parse --is-bare-repository`;
  41 die "$? $!" if $?;
  42 chomp $bare or die;
  43
  44 sub getobj ($$) {
  45     my ($obj, $type) = @_;
  46     print GCFI $obj, "\n" or die $!;
  47     my $x = <GCFO>;
  48     my ($gtype, $gsize) = $x =~ m/^\w+ (\w+) (\d+)\n/ or die "$obj ?";
  49     $gtype eq $type or die "$obj $gtype != $type ?";
  50     my $gdata;
  51     (read GCFO, $gdata, $gsize) == $gsize or die "$obj $!";
  52     $x = <GCFO>;
  53     $x eq "\n" or die "$obj ($_) $!";
  54     $count{inspected}++;
  55     return $gdata;
  56 }
  57
  58 sub hashobj ($$) {
  59     my ($data,$type) = @_;
  60     my $gwopid = open2 \*GWO, \*GWI,
  61         "git hash-object -w -t $type --stdin"
  62         or die $!;
  63     print GWI $data or die $!;
  64     close GWI or die $!;
  65     $_ = <GWO>;
  66     close GWO or die $!;
  67     waitpid $gwopid,0 == $gwopid or die $!;
  68     die $? if $?;
  69     m/^(\w+)\n/ or die "$_ ?";
  70     $count{"rewritten $type"}++;
  71     return $1;
  72 }
  73
  74 our %memo;
  75
  76 sub rewrite_commit ($);
  77 sub rewrite_commit ($) {
  78     my ($obj) = @_;
  79     my $m = \ $memo{$obj};
  80     return $$m if defined $$m;
  81     my $olddata = getobj $obj, 'commit';
  82     $olddata =~ m/(?<=\n)(?=\n)/ or die "$obj ?";
  83     my $msg = $';
  84     local $_ = $`;
  85     s{^(parent )(\w+)$}{ $1 . rewrite_commit($2) }gme;
  86     $count{'fix overwrite'} += s{^commiter }{committer }gm;
  87     if (!m{^author }m && !m{^committer }m) {
  88         m{^parent (\w+)}m or die "$obj ?";
  89         my $parent = getobj $1, 'commit';
  90         $parent =~ m/^(?:.+\n)+(author .*\ncommitter .*\n)/;
  91         m/\n$/ or die "$obj ?";
  92         $_ .= $1;
  93         $count{'fix import'}++;
  94     }
  95     my $newdata = $_.$msg;
  96     my $newobj;
  97     if ($newdata eq $olddata) {
  98         $newobj = $obj;
  99         $count{unchanged}++;
 100 #print STDERR "UNCHANGED $obj\n";
 101     } else {
 102         $newobj = hashobj $newdata, 'commit';
 103 #print STDERR "REWRITTEN $obj $newobj\n";
 104     }
 105     $$m= $newobj;
 106     return $newobj;
 107 }
 108
 109 our @updates;
 110
 111 sub filter_updates () {
 112     @updates = grep { $_->[1] ne $_->[2] } @updates;
 113 }
 114
 115 sub rewrite_tag ($) {
 116     my ($obj) = @_;
 117     $_ = getobj $obj, 'tag';
 118     m/^type (\w+)\n/m or die "$obj ?";
 119     if ($1 ne 'commit') {
 120         $count{"oddtags $1"}++;
 121         return $obj;
 122     }
 123     m/^object (\w+)\n/m or die "$obj ?";
 124     my $oldref = $1;
 125     my $newref = rewrite_commit $oldref;
 126     if ($oldref eq $newref) {
 127         return $obj;
 128     }
 129     s/^(object )\w+$/ $1.$newref /me or die "$obj ($_) ?";
 130     s/^-----BEGIN PGP SIGNATURE-----\n.*^-----END PGP SIGNATURE-----\n$//sm;
 131     return hashobj $_, 'tag';
 132 }
 133
 134 sub edit_rewrite_map ($) {
 135     my ($old) = @_;
 136
 137     filter_updates();
 138     return $old unless @updates;
 139
 140     my $td = 'dgit-broken-fixup.tmp';
 141     runcmd qw(rm -rf), $td;
 142     mkdir $td, 0700 or die "$td $!";
 143     chdir $td or die $!;
 144     runcmd qw(git init -q);
 145     runcmd qw(git config gc.auto 0);
 146     runcmd qw(rm -rf .git/objects);
 147     symlink "../../objects", ".git/objects" or die $!;
 148
 149     my %map;
 150
 151     if ($old) {
 152         runcmd qw(git checkout -q), $old;
 153         open M, "map" or die $!;
 154         while (<M>) {
 155             m/^(\w+)(?:\s+(\w+))?$/ or die;
 156             $map{$1} = $2;
 157             $count{rewrite_map_previous}++;
 158         }
 159         M->error and die $!;
 160         close M or die $!;
 161     }
 162
 163     foreach my $oldc (keys %memo) {
 164         my $newc = $memo{$oldc};
 165         next if $oldc eq $newc;
 166         $map{$oldc} = $newc;
 167     }
 168     foreach my $up (@updates) { # catches tags
 169         $map{ $up->[1] } = $up->[2];
 170     }
 171
 172     open M, ">", "map" or die $!;
 173     printf M "%s%s\n",
 174         $_, (defined $map{$_} ? " $map{$_}" : "")
 175         or die $!
 176         foreach keys %map;
 177     close M or die $!;
 178
 179     if (!$old) {
 180         runcmd qw(git add map);
 181     }
 182
 183     runcmd qw(git commit -q),
 184         qw(-m), 'dgit-badcommit-fixup',
 185         qw(map);
 186
 187     $!=0; $?=0;
 188     my $new = `git rev-parse HEAD`;
 189     die "$? $!" if $?;
 190     chomp $new or die;
 191
 192     chdir '..' or die $!;
 193     runcmd qw(rm -rf), $td;
 194
 195     $count{rewrite_map_updated}++;
 196
 197     return $new;
 198 }
 199
 200 $!=0; $?=0;
 201 my $refs=`git for-each-ref`;
 202 die "$? $!" if $?;
 203
 204 chomp $refs;
 205
 206 our $org_rewrite_map;
 207
 208 foreach my $rline (split /\n/, $refs) {
 209     my ($obj, $type, $refname) =
 210         $rline =~ m/^(\w+)\s+(\w+)\s+(\S.*)/
 211         or die "$_ ?";
 212     if ($refname eq 'refs/dgit-rewrite/map') {
 213         $org_rewrite_map = $obj;
 214         continue;
 215     }
 216     my $rewrite;
 217     if ($type eq 'commit') {
 218         $rewrite = rewrite_commit($obj);
 219     } elsif ($type eq 'tag') {
 220         $rewrite = rewrite_tag($obj);
 221     } else {
 222         warn "ref $refname refers to $type\n";
 223         next;
 224     }
 225     push @updates, [ $refname, $obj, $rewrite ];
 226 }
 227
 228 if ($bare eq 'true') {
 229     my $new_rewrite_map = edit_rewrite_map($org_rewrite_map);
 230     push @updates, [ 'refs/dgit-rewrite/map',
 231                      ($org_rewrite_map // '0'x40),
 232                      ($new_rewrite_map // '0'x40),
 233                      1 ];
 234 }
 235
 236 filter_updates();
 237
 238 if (!@updates) {
 239     print Dumper(\%count), "nothing to do\n";
 240     exit 0;
 241 }
 242
 243 #print Dumper(\@updates);
 244
 245 open U, "|git update-ref -m 'dgit bad commit fixup' --stdin" or die $!;
 246
 247 for my $up (@updates) {
 248     my ($ref, $old, $new, $nobackup) = @$up;
 249     my $otherref = $ref;
 250     $otherref =~ s{^refs/}{};
 251     if ($real) {
 252         print U <<END or die $! unless $nobackup;
 253 create refs/dgit-badcommit/$otherref $old
 254 END
 255         print U <<END or die $!;
 256 update $ref $new $old
 257 END
 258     } else {
 259         print U <<END or die $!;
 260 update refs/dgit-badfixuptest/$otherref $new
 261 END
 262     }
 263 }
 264
 265 $?=0; $!=0;
 266 close U or die "$? $!";
 267 die $? if $?;
 268
 269 print Dumper(\%count);
 270
 271 if ($real) {
 272     print "old values saved in refs/dgit-badcommit/\n" or die $!;
 273 } else {
 274     print "testing output saved in refs/dgit-badfixuptest/\n" or die $!;
 275 }