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 #print STDERR ">$obj|$olddata<\n";
  83     $olddata =~ m/(?<=\n)(?=\n)/ or die "$obj ?";
  84     my $msg = $';
  85     local $_ = $`;
  86     s{^(parent )(\w+)$}{ $1 . rewrite_commit($2) }gme;
  87     $count{'fix overwrite'} += s{^commiter }{committer }gm;
  88     if (!m{^author }m && !m{^committer }m) {
  89         m{^parent (\w+)}m or die "$obj ?";
  90         my $parent = getobj $1, 'commit';
  91         $parent =~ m/^(?:.+\n)+(author .*\ncommitter .*\n)/;
  92         m/\n$/ or die "$obj ?";
  93         $_ .= $1;
  94         $count{'fix import'}++;
  95     }
  96     my $newdata = $_.$msg;
  97     my $newobj;
  98     if ($newdata eq $olddata) {
  99         $newobj = $obj;
 100         $count{unchanged}++;
 101     } else {
 102         $newobj = hashobj $newdata, 'commit';
 103     }
 104     $$m= $newobj;
 105     return $newobj;
 106 }
 107
 108 our @updates;
 109
 110 sub filter_updates () {
 111     @updates = grep { $_->[1] ne $_->[2] } @updates;
 112 }
 113
 114 sub rewrite_tag ($) {
 115     my ($obj) = @_;
 116     $_ = getobj $obj, 'tag';
 117     m/^type (\w+)\n/m or die "$obj ?";
 118     if ($1 ne 'commit') {
 119         $count{"oddtags $1"}++;
 120         return $obj;
 121     }
 122     m/^object (\w+)\n/m or die "$obj ?";
 123     my $oldref = $1;
 124     my $newref = rewrite_commit $oldref;
 125     if ($oldref eq $newref) {
 126         return $obj;
 127     }
 128     s/^(object )\w+$/ $1.$newref /me or die "$obj ($_) ?";
 129     s/^-----BEGIN PGP SIGNATURE-----\n.*^-----END PGP SIGNATURE-----\n$//sm;
 130     return hashobj $_, 'tag';
 131 }
 132
 133 sub edit_rewrite_map ($) {
 134     my ($old) = @_;
 135
 136     filter_updates();
 137     return $old unless @updates;
 138
 139     my $td = 'dgit-broken-fixup.tmp';
 140     runcmd qw(rm -rf), $td;
 141     mkdir $td, 0700 or die "$td $!";
 142     chdir $td or die $!;
 143     runcmd qw(git init -q);
 144     runcmd qw(git config gc.auto 0);
 145     runcmd qw(rm -rf .git/objects);
 146     symlink "../../objects", ".git/objects" or die $!;
 147
 148     my %map;
 149
 150     if ($old) {
 151         runcmd qw(git checkout -q), $old;
 152         open M, "map" or die $!;
 153         while (<M>) {
 154             m/^(\w+)(?:\s+(\w+))?$/ or die;
 155             $map{$1} = $2;
 156             $count{rewrite_map_previous}++;
 157         }
 158         M->error and die $!;
 159         close M or die $!;
 160     }
 161
 162     foreach my $up (@updates) {
 163         $map{ $up->[1] } = $up->[2];
 164     }
 165
 166     open M, ">", "map" or die $!;
 167     printf M "%s%s\n",
 168         $_, (defined $map{$_} ? " $map{$_}" : "")
 169         or die $!
 170         foreach keys %map;
 171     close M or die $!;
 172
 173     if (!$old) {
 174         runcmd qw(git add map);
 175     }
 176
 177     runcmd qw(git commit -q),
 178         qw(-m), 'dgit-badcommit-fixup',
 179         qw(map);
 180
 181     $!=0; $?=0;
 182     my $new = `git rev-parse HEAD`;
 183     die "$? $!" if $?;
 184     chomp $new or die;
 185
 186     chdir '..' or die $!;
 187     runcmd qw(rm -rf), $td;
 188
 189     $count{rewrite_map_updated}++;
 190
 191     return $new;
 192 }
 193
 194 $!=0; $?=0;
 195 my $refs=`git for-each-ref`;
 196 die "$? $!" if $?;
 197
 198 chomp $refs;
 199
 200 our $org_rewrite_map;
 201
 202 foreach my $rline (split /\n/, $refs) {
 203     my ($obj, $type, $refname) =
 204         $rline =~ m/^(\w+)\s+(\w+)\s+(\S.*)/
 205         or die "$_ ?";
 206     if ($refname eq 'refs/dgit-rewrite/map') {
 207         $org_rewrite_map = $obj;
 208         continue;
 209     }
 210     my $rewrite;
 211     if ($type eq 'commit') {
 212         $rewrite = rewrite_commit($obj);
 213     } elsif ($type eq 'tag') {
 214         $rewrite = rewrite_tag($obj);
 215     } else {
 216         warn "ref $refname refers to $type\n";
 217         next;
 218     }
 219     push @updates, [ $refname, $obj, $rewrite ];
 220 }
 221
 222 if ($bare eq 'true') {
 223     my $new_rewrite_map = edit_rewrite_map($org_rewrite_map);
 224     push @updates, [ 'refs/dgit-rewrite/map',
 225                      ($org_rewrite_map // '0'x40),
 226                      ($new_rewrite_map // '0'x40),
 227                      1 ];
 228 }
 229
 230 filter_updates();
 231
 232 if (!@updates) {
 233     print Dumper(\%count), "nothing to do\n";
 234     exit 0;
 235 }
 236
 237 #print Dumper(\@updates);
 238
 239 open U, "|git update-ref -m 'dgit bad commit fixup' --stdin" or die $!;
 240
 241 if ($real && $bare eq 'false') {
 242     print "detaching your HEAD\n" or die $!;
 243     runcmd 'git checkout --detach';
 244 }
 245
 246 for my $up (@updates) {
 247     my ($ref, $old, $new, $nobackup) = @$up;
 248     my $otherref = $ref;
 249     $otherref =~ s{^refs/}{};
 250     if ($real) {
 251         print U <<END or die $! unless $nobackup;
 252 create refs/dgit-badcommit/$otherref $old
 253 END
 254         print U <<END or die $!;
 255 update $ref $new $old
 256 END
 257     } else {
 258         print U <<END or die $!;
 259 update refs/dgit-badfixuptest/$otherref $new
 260 END
 261     }
 262 }
 263
 264 $?=0; $!=0;
 265 close U or die "$? $!";
 266 die $? if $?;
 267
 268 print Dumper(\%count);
 269
 270 if ($real) {
 271     print "old values saved in refs/dgit-badcommit/\n" or die $!;
 272 } else {
 273     print "testing output saved in refs/dgit-badfixuptest/\n" or die $!;
 274 }