git-cvsimport.perl 26.5 KB
Newer Older
1
#!/usr/bin/perl -w
2

3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18
# This tool is copyright (c) 2005, Matthias Urlichs.
# It is released under the Gnu Public License, version 2.
#
# The basic idea is to aggregate CVS check-ins into related changes.
# Fortunately, "cvsps" does that for us; all we have to do is to parse
# its output.
#
# Checking out the files is done by a single long-running CVS connection
# / server process.
#
# The head revision is on branch "origin" by default.
# You can change that with the '-o' option.

use strict;
use warnings;
use Getopt::Std;
19
use File::Spec;
20
use File::Temp qw(tempfile tmpnam);
21 22 23
use File::Path qw(mkpath);
use File::Basename qw(basename dirname);
use Time::Local;
M
Matthias Urlichs 已提交
24 25
use IO::Socket;
use IO::Pipe;
J
Jeff King 已提交
26
use POSIX qw(strftime dup2 ENOENT);
27
use IPC::Open2;
28 29 30 31

$SIG{'PIPE'}="IGNORE";
$ENV{'TZ'}="UTC";

32
our ($opt_h,$opt_o,$opt_v,$opt_k,$opt_u,$opt_d,$opt_p,$opt_C,$opt_z,$opt_i,$opt_P, $opt_s,$opt_m,$opt_M,$opt_A,$opt_S,$opt_L, $opt_a);
33
my (%conv_author_name, %conv_author_email);
34

35 36 37
sub usage(;$) {
	my $msg = shift;
	print(STDERR "Error: $msg\n") if $msg;
38
	print STDERR <<END;
M
Matthias Urlichs 已提交
39
Usage: ${\basename $0}     # fetch/update GIT from CVS
40
       [-o branch-for-HEAD] [-h] [-v] [-d CVSROOT] [-A author-conv-file]
41 42 43
       [-p opts-for-cvsps] [-P file] [-C GIT_repository] [-z fuzz] [-i] [-k]
       [-u] [-s subst] [-a] [-m] [-M regex] [-S regex] [-L commitlimit]
       [CVS_module]
44 45 46 47
END
	exit(1);
}

48 49 50 51 52 53
sub read_author_info($) {
	my ($file) = @_;
	my $user;
	open my $f, '<', "$file" or die("Failed to open $file: $!\n");

	while (<$f>) {
54
		# Expected format is this:
55
		#   exon=Andreas Ericsson <ae@op5.se>
56
		if (m/^(\S+?)\s*=\s*(.+?)\s*<(.+)>\s*$/) {
57
			$user = $1;
58 59
			$conv_author_name{$user} = $2;
			$conv_author_email{$user} = $3;
60
		}
61 62 63 64 65 66 67 68 69 70 71 72 73 74 75
		# However, we also read from CVSROOT/users format
		# to ease migration.
		elsif (/^(\w+):(['"]?)(.+?)\2\s*$/) {
			my $mapped;
			($user, $mapped) = ($1, $3);
			if ($mapped =~ /^\s*(.*?)\s*<(.*)>\s*$/) {
				$conv_author_name{$user} = $1;
				$conv_author_email{$user} = $2;
			}
			elsif ($mapped =~ /^<?(.*)>?$/) {
				$conv_author_name{$user} = $user;
				$conv_author_email{$user} = $1;
			}
		}
		# NEEDSWORK: Maybe warn on unrecognized lines?
76 77 78 79 80 81 82 83 84 85
	}
	close ($f);
}

sub write_author_info($) {
	my ($file) = @_;
	open my $f, '>', $file or
	  die("Failed to open $file for writing: $!");

	foreach (keys %conv_author_name) {
86
		print $f "$_=$conv_author_name{$_} <$conv_author_email{$_}>\n";
87 88 89 90
	}
	close ($f);
}

91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
# convert getopts specs for use by git-repo-config
sub read_repo_config {
    # Split the string between characters, unless there is a ':'
    # So "abc:de" becomes ["a", "b", "c:", "d", "e"]
	my @opts = split(/ *(?!:)/, shift);
	foreach my $o (@opts) {
		my $key = $o;
		$key =~ s/://g;
		my $arg = 'git-repo-config';
		$arg .= ' --bool' if ($o !~ /:$/);

        chomp(my $tmp = `$arg --get cvsimport.$key`);
		if ($tmp && !($arg =~ /--bool/ && $tmp eq 'false')) {
            no strict 'refs';
            my $opt_name = "opt_" . $key;
            if (!$$opt_name) {
                $$opt_name = $tmp;
            }
		}
	}
    if (@ARGV == 0) {
        chomp(my $module = `git-repo-config --get cvsimport.module`);
        push(@ARGV, $module);
    }
}

my $opts = "haivmkuo:d:p:C:z:s:M:P:A:S:L:";
read_repo_config($opts);
getopts($opts) or usage();
120 121
usage if $opt_h;

122
@ARGV <= 1 or usage("You can't specify more than one CVS module");
123

J
Junio C Hamano 已提交
124
if ($opt_d) {
M
Matthias Urlichs 已提交
125
	$ENV{"CVSROOT"} = $opt_d;
J
Junio C Hamano 已提交
126
} elsif (-f 'CVS/Root') {
127 128 129 130 131
	open my $f, '<', 'CVS/Root' or die 'Failed to open CVS/Root';
	$opt_d = <$f>;
	chomp $opt_d;
	close $f;
	$ENV{"CVSROOT"} = $opt_d;
J
Junio C Hamano 已提交
132
} elsif ($ENV{"CVSROOT"}) {
M
Matthias Urlichs 已提交
133 134
	$opt_d = $ENV{"CVSROOT"};
} else {
135
	usage("CVSROOT needs to be set");
M
Matthias Urlichs 已提交
136 137
}
$opt_o ||= "origin";
138
$opt_s ||= "-";
139 140
$opt_a ||= 0;

141
my $git_tree = $opt_C;
M
Matthias Urlichs 已提交
142 143
$git_tree ||= ".";

144 145 146 147 148 149 150 151
my $cvs_tree;
if ($#ARGV == 0) {
	$cvs_tree = $ARGV[0];
} elsif (-f 'CVS/Repository') {
	open my $f, '<', 'CVS/Repository' or 
	    die 'Failed to open CVS/Repository';
	$cvs_tree = <$f>;
	chomp $cvs_tree;
152
	close $f;
153
} else {
154
	usage("CVS module has to be specified");
155 156
}

157 158 159 160 161 162 163 164
our @mergerx = ();
if ($opt_m) {
	@mergerx = ( qr/\W(?:from|of|merge|merging|merged) (\w+)/i );
}
if ($opt_M) {
	push (@mergerx, qr/$opt_M/);
}

165 166 167 168 169
# Remember UTC of our starting time
# we'll want to avoid importing commits
# that are too recent
our $starttime = time();

170 171 172 173 174
select(STDERR); $|=1; select(STDOUT);


package CVSconn;
# Basic CVS dialog.
M
Matthias Urlichs 已提交
175
# We're only interested in connecting and downloading, so ...
176

177 178
use File::Spec;
use File::Temp qw(tempfile);
M
Matthias Urlichs 已提交
179 180
use POSIX qw(strftime dup2);

181
sub new {
J
Junio C Hamano 已提交
182
	my ($what,$repo,$subdir) = @_;
183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201
	$what=ref($what) if ref($what);

	my $self = {};
	$self->{'buffer'} = "";
	bless($self,$what);

	$repo =~ s#/+$##;
	$self->{'fullrep'} = $repo;
	$self->conn();

	$self->{'subdir'} = $subdir;
	$self->{'lines'} = undef;

	return $self;
}

sub conn {
	my $self = shift;
	my $repo = $self->{'fullrep'};
J
Junio C Hamano 已提交
202 203
	if ($repo =~ s/^:pserver(?:([^:]*)):(?:(.*?)(?::(.*?))?@)?([^:\/]*)(?::(\d*))?//) {
		my ($param,$user,$pass,$serv,$port) = ($1,$2,$3,$4,$5);
204

J
Junio C Hamano 已提交
205 206
		my ($proxyhost,$proxyport);
		if ($param && ($param =~ m/proxy=([^;]+)/)) {
207 208 209
			$proxyhost = $1;
			# Default proxyport, if not specified, is 8080.
			$proxyport = 8080;
J
Junio C Hamano 已提交
210
			if ($ENV{"CVS_PROXY_PORT"}) {
211 212
				$proxyport = $ENV{"CVS_PROXY_PORT"};
			}
J
Junio C Hamano 已提交
213
			if ($param =~ m/proxyport=([^;]+)/) {
214 215 216 217
				$proxyport = $1;
			}
		}

218
		$user="anonymous" unless defined $user;
M
Matthias Urlichs 已提交
219
		my $rr2 = "-";
J
Junio C Hamano 已提交
220
		unless ($port) {
221 222 223 224 225
			$rr2 = ":pserver:$user\@$serv:$repo";
			$port=2401;
		}
		my $rr = ":pserver:$user\@$serv:$port$repo";

J
Junio C Hamano 已提交
226
		unless ($pass) {
227 228
			open(H,$ENV{'HOME'}."/.cvspass") and do {
				# :pserver:cvs@mea.tmt.tele.fi:/cvsroot/zmailer Ah<Z
J
Junio C Hamano 已提交
229
				while (<H>) {
230 231 232
					chomp;
					s/^\/\d+\s+//;
					my ($w,$p) = split(/\s/,$_,2);
J
Junio C Hamano 已提交
233
					if ($w eq $rr or $w eq $rr2) {
234 235 236 237 238 239 240 241
						$pass = $p;
						last;
					}
				}
			};
		}
		$pass="A" unless $pass;

242
		my ($s, $rep);
J
Junio C Hamano 已提交
243
		if ($proxyhost) {
244 245 246 247 248 249 250 251 252 253 254 255 256 257 258

			# Use a HTTP Proxy. Only works for HTTP proxies that
			# don't require user authentication
			#
			# See: http://www.ietf.org/rfc/rfc2817.txt

			$s = IO::Socket::INET->new(PeerHost => $proxyhost, PeerPort => $proxyport);
			die "Socket to $proxyhost: $!\n" unless defined $s;
			$s->write("CONNECT $serv:$port HTTP/1.1\r\nHost: $serv:$port\r\n\r\n")
	                        or die "Write to $proxyhost: $!\n";
	                $s->flush();

			$rep = <$s>;

			# The answer should look like 'HTTP/1.x 2yy ....'
J
Junio C Hamano 已提交
259
			if (!($rep =~ m#^HTTP/1\.. 2[0-9][0-9]#)) {
260 261 262 263 264 265 266 267 268 269 270 271 272 273
				die "Proxy connect: $rep\n";
			}
			# Skip up to the empty line of the proxy server output
			# including the response headers.
			while ($rep = <$s>) {
				last if (!defined $rep ||
					 $rep eq "\n" ||
					 $rep eq "\r\n");
			}
		} else {
			$s = IO::Socket::INET->new(PeerHost => $serv, PeerPort => $port);
			die "Socket to $serv: $!\n" unless defined $s;
		}

274 275 276 277
		$s->write("BEGIN AUTH REQUEST\n$repo\n$user\n$pass\nEND AUTH REQUEST\n")
			or die "Write to $serv: $!\n";
		$s->flush();

278
		$rep = <$s>;
279

J
Junio C Hamano 已提交
280
		if ($rep ne "I LOVE YOU\n") {
281 282 283 284 285
			$rep="<unknown>" unless $rep;
			die "AuthReply: $rep\n";
		}
		$self->{'socketo'} = $s;
		$self->{'socketi'} = $s;
S
Sven Verdoolaege 已提交
286
	} else { # local or ext: Fork off our own cvs server.
287 288 289 290
		my $pr = IO::Pipe->new();
		my $pw = IO::Pipe->new();
		my $pid = fork();
		die "Fork: $!\n" unless defined $pid;
S
Sven Verdoolaege 已提交
291 292
		my $cvs = 'cvs';
		$cvs = $ENV{CVS_SERVER} if exists $ENV{CVS_SERVER};
S
Sven Verdoolaege 已提交
293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310 311
		my $rsh = 'rsh';
		$rsh = $ENV{CVS_RSH} if exists $ENV{CVS_RSH};

		my @cvs = ($cvs, 'server');
		my ($local, $user, $host);
		$local = $repo =~ s/:local://;
		if (!$local) {
		    $repo =~ s/:ext://;
		    $local = !($repo =~ s/^(?:([^\@:]+)\@)?([^:]+)://);
		    ($user, $host) = ($1, $2);
		}
		if (!$local) {
		    if ($user) {
			unshift @cvs, $rsh, '-l', $user, $host;
		    } else {
			unshift @cvs, $rsh, $host;
		    }
		}

J
Junio C Hamano 已提交
312
		unless ($pid) {
313 314 315 316 317 318
			$pr->writer();
			$pw->reader();
			dup2($pw->fileno(),0);
			dup2($pr->fileno(),1);
			$pr->close();
			$pw->close();
S
Sven Verdoolaege 已提交
319
			exec(@cvs);
320 321 322 323 324 325 326 327 328
		}
		$pw->writer();
		$pr->reader();
		$self->{'socketo'} = $pw;
		$self->{'socketi'} = $pr;
	}
	$self->{'socketo'}->write("Root $repo\n");

	# Trial and error says that this probably is the minimum set
329
	$self->{'socketo'}->write("Valid-responses ok error Valid-requests Mode M Mbinary E Checked-in Created Updated Merged Removed\n");
330 331 332 333 334

	$self->{'socketo'}->write("valid-requests\n");
	$self->{'socketo'}->flush();

	chomp(my $rep=$self->readline());
J
Junio C Hamano 已提交
335
	if ($rep !~ s/^Valid-requests\s*//) {
336 337 338 339 340 341 342 343 344 345 346
		$rep="<unknown>" unless $rep;
		die "Expected Valid-requests from server, but got: $rep\n";
	}
	chomp(my $res=$self->readline());
	die "validReply: $res\n" if $res ne "ok";

	$self->{'socketo'}->write("UseUnchanged\n") if $rep =~ /\bUseUnchanged\b/;
	$self->{'repo'} = $repo;
}

sub readline {
J
Junio C Hamano 已提交
347
	my ($self) = @_;
348 349 350 351 352 353
	return $self->{'socketi'}->getline();
}

sub _file {
	# Request a file with a given revision.
	# Trial and error says this is a good way to do it. :-/
J
Junio C Hamano 已提交
354
	my ($self,$fn,$rev) = @_;
355 356
	$self->{'socketo'}->write("Argument -N\n") or return undef;
	$self->{'socketo'}->write("Argument -P\n") or return undef;
357 358 359 360
	# -kk: Linus' version doesn't use it - defaults to off
	if ($opt_k) {
	    $self->{'socketo'}->write("Argument -kk\n") or return undef;
	}
361 362 363 364 365 366
	$self->{'socketo'}->write("Argument -r\n") or return undef;
	$self->{'socketo'}->write("Argument $rev\n") or return undef;
	$self->{'socketo'}->write("Argument --\n") or return undef;
	$self->{'socketo'}->write("Argument $self->{'subdir'}/$fn\n") or return undef;
	$self->{'socketo'}->write("Directory .\n") or return undef;
	$self->{'socketo'}->write("$self->{'repo'}\n") or return undef;
367
	# $self->{'socketo'}->write("Sticky T1.0\n") or return undef;
368 369 370 371 372 373 374 375
	$self->{'socketo'}->write("co\n") or return undef;
	$self->{'socketo'}->flush() or return undef;
	$self->{'lines'} = 0;
	return 1;
}
sub _line {
	# Read a line from the server.
	# ... except that 'line' may be an entire file. ;-)
J
Junio C Hamano 已提交
376
	my ($self, $fh) = @_;
377 378 379
	die "Not in lines" unless defined $self->{'lines'};

	my $line;
380
	my $res=0;
J
Junio C Hamano 已提交
381
	while (defined($line = $self->readline())) {
382 383 384 385 386 387 388 389
		# M U gnupg-cvs-rep/AUTHORS
		# Updated gnupg-cvs-rep/
		# /daten/src/rsync/gnupg-cvs-rep/AUTHORS
		# /AUTHORS/1.1///T1.1
		# u=rw,g=rw,o=rw
		# 0
		# ok

J
Junio C Hamano 已提交
390
		if ($line =~ s/^(?:Created|Updated) //) {
391 392 393 394 395 396 397 398 399
			$line = $self->readline(); # path
			$line = $self->readline(); # Entries line
			my $mode = $self->readline(); chomp $mode;
			$self->{'mode'} = $mode;
			defined (my $cnt = $self->readline())
				or die "EOF from server after 'Changed'\n";
			chomp $cnt;
			die "Duh: Filesize $cnt" if $cnt !~ /^\d+$/;
			$line="";
400
			$res = $self->_fetchfile($fh, $cnt);
J
Junio C Hamano 已提交
401
		} elsif ($line =~ s/^ //) {
402 403
			print $fh $line;
			$res += length($line);
J
Junio C Hamano 已提交
404
		} elsif ($line =~ /^M\b/) {
405
			# output, do nothing
J
Junio C Hamano 已提交
406
		} elsif ($line =~ /^Mbinary\b/) {
407 408 409 410 411
			my $cnt;
			die "EOF from server after 'Mbinary'" unless defined ($cnt = $self->readline());
			chomp $cnt;
			die "Duh: Mbinary $cnt" if $cnt !~ /^\d+$/ or $cnt<1;
			$line="";
412
			$res += $self->_fetchfile($fh, $cnt);
413 414
		} else {
			chomp $line;
J
Junio C Hamano 已提交
415
			if ($line eq "ok") {
416 417
				# print STDERR "S: ok (".length($res).")\n";
				return $res;
J
Junio C Hamano 已提交
418
			} elsif ($line =~ s/^E //) {
419
				# print STDERR "S: $line\n";
J
Junio C Hamano 已提交
420
			} elsif ($line =~ /^(Remove-entry|Removed) /i) {
421 422 423 424 425
				$line = $self->readline(); # filename
				$line = $self->readline(); # OK
				chomp $line;
				die "Unknown: $line" if $line ne "ok";
				return -1;
426 427 428 429 430
			} else {
				die "Unknown: $line\n";
			}
		}
	}
M
Martin Mares 已提交
431
	return undef;
432 433
}
sub file {
J
Junio C Hamano 已提交
434
	my ($self,$fn,$rev) = @_;
435 436
	my $res;

437 438 439 440 441 442
	my ($fh, $name) = tempfile('gitcvs.XXXXXX', 
		    DIR => File::Spec->tmpdir(), UNLINK => 1);

	$self->_file($fn,$rev) and $res = $self->_line($fh);

	if (!defined $res) {
M
Martin Mares 已提交
443 444
	    print STDERR "Server has gone away while fetching $fn $rev, retrying...\n";
	    truncate $fh, 0;
445
	    $self->conn();
M
Martin Mares 已提交
446
	    $self->_file($fn,$rev) or die "No file command send";
447
	    $res = $self->_line($fh);
M
Martin Mares 已提交
448
	    die "Retry failed" unless defined $res;
449
	}
450
	close ($fh);
451

452
	return ($name, $res);
453
}
454 455
sub _fetchfile {
	my ($self, $fh, $cnt) = @_;
456
	my $res = 0;
457
	my $bufsize = 1024 * 1024;
J
Junio C Hamano 已提交
458
	while ($cnt) {
459 460 461 462 463 464 465 466 467 468 469 470
	    if ($bufsize > $cnt) {
		$bufsize = $cnt;
	    }
	    my $buf;
	    my $num = $self->{'socketi'}->read($buf,$bufsize);
	    die "Server: Filesize $cnt: $num: $!\n" if not defined $num or $num<=0;
	    print $fh $buf;
	    $res += $num;
	    $cnt -= $num;
	}
	return $res;
}
471 472 473 474


package main;

M
Matthias Urlichs 已提交
475
my $cvs = CVSconn->new($opt_d, $cvs_tree);
476 477 478


sub pdate($) {
J
Junio C Hamano 已提交
479
	my ($d) = @_;
480 481 482 483
	m#(\d{2,4})/(\d\d)/(\d\d)\s(\d\d):(\d\d)(?::(\d\d))?#
		or die "Unparseable date: $d\n";
	my $y=$1; $y-=1900 if $y>1900;
	return timegm($6||0,$5,$4,$3,$2-1,$y);
484 485
}

486
sub pmode($) {
J
Junio C Hamano 已提交
487
	my ($mode) = @_;
488 489 490 491
	my $m = 0;
	my $mm = 0;
	my $um = 0;
	for my $x(split(//,$mode)) {
J
Junio C Hamano 已提交
492
		if ($x eq ",") {
493 494 495
			$m |= $mm&$um;
			$mm = 0;
			$um = 0;
J
Junio C Hamano 已提交
496 497 498 499 500 501 502
		} elsif ($x eq "u") { $um |= 0700;
		} elsif ($x eq "g") { $um |= 0070;
		} elsif ($x eq "o") { $um |= 0007;
		} elsif ($x eq "r") { $mm |= 0444;
		} elsif ($x eq "w") { $mm |= 0222;
		} elsif ($x eq "x") { $mm |= 0111;
		} elsif ($x eq "=") { # do nothing
503 504 505 506 507 508
		} else { die "Unknown mode: $mode\n";
		}
	}
	$m |= $mm&$um;
	return $m;
}
509

510 511 512 513
sub getwd() {
	my $pwd = `pwd`;
	chomp $pwd;
	return $pwd;
514 515
}

J
Jeff King 已提交
516 517 518 519
sub is_sha1 {
	my $s = shift;
	return $s =~ /^[a-f0-9]{40}$/;
}
520

J
Jeff King 已提交
521
sub get_headref ($$) {
522 523 524
    my $name    = shift;
    my $git_dir = shift; 
    
J
Jeff King 已提交
525
    my $f = "$git_dir/refs/heads/$name";
J
Junio C Hamano 已提交
526
    if (open(my $fh, $f)) {
J
Jeff King 已提交
527 528 529
	    chomp(my $r = <$fh>);
	    is_sha1($r) or die "Cannot get head id for $name ($r): $!";
	    return $r;
530
    }
J
Jeff King 已提交
531 532
    die "unable to open $f: $!" unless $! == POSIX::ENOENT;
    return undef;
533 534
}

535 536 537 538
-d $git_tree
	or mkdir($git_tree,0777)
	or die "Could not create $git_tree: $!";
chdir($git_tree);
539

540
my $last_branch = "";
541
my $orig_branch = "";
542
my %branch_date;
543
my $tip_at_start = undef;
544 545 546 547

my $git_dir = $ENV{"GIT_DIR"} || ".git";
$git_dir = getwd()."/".$git_dir unless $git_dir =~ m#^/#;
$ENV{"GIT_DIR"} = $git_dir;
548 549
my $orig_git_index;
$orig_git_index = $ENV{GIT_INDEX_FILE} if exists $ENV{GIT_INDEX_FILE};
550 551

my %index; # holds filenames of one index per branch
552

J
Junio C Hamano 已提交
553
unless (-d $git_dir) {
554
	system("git-init");
555 556 557 558 559
	die "Cannot init the GIT db at $git_tree: $?\n" if $?;
	system("git-read-tree");
	die "Cannot init an empty tree: $?\n" if $?;

	$last_branch = $opt_o;
560
	$orig_branch = "";
561
} else {
P
Pavel Roskin 已提交
562 563 564 565 566
	open(F, "git-symbolic-ref HEAD |") or
		die "Cannot run git-symbolic-ref: $!\n";
	chomp ($last_branch = <F>);
	$last_branch = basename($last_branch);
	close(F);
J
Junio C Hamano 已提交
567
	unless ($last_branch) {
568 569 570 571
		warn "Cannot read the last branch name: $! -- assuming 'master'\n";
		$last_branch = "master";
	}
	$orig_branch = $last_branch;
572
	$tip_at_start = `git-rev-parse --verify HEAD`;
573 574

	# Get the last import timestamps
575 576 577
	my $fmt = '($ref, $author) = (%(refname), %(author));';
	open(H, "git-for-each-ref --perl --format='$fmt' refs/heads |") or
		die "Cannot run git-for-each-ref: $!\n";
J
Junio C Hamano 已提交
578
	while (defined(my $entry = <H>)) {
579 580 581 582 583
		my ($ref, $author);
		eval($entry) || die "cannot eval refs list: $@";
		my ($head) = ($ref =~ m|^refs/heads/(.*)|);
		$author =~ /^.*\s(\d+)\s[-+]\d{4}$/;
		$branch_date{$head} = $1;
584
	}
585
	close(H);
586 587 588 589 590
        if (!exists $branch_date{$opt_o}) {
		die "Branch '$opt_o' does not exist.\n".
		       "Either use the correct '-o branch' option,\n".
		       "or import to a new repository.\n";
        }
591 592 593 594 595
}

-d $git_dir
	or die "Could not create git subdir ($git_dir).\n";

596 597 598 599 600 601 602 603
# now we read (and possibly save) author-info as well
-f "$git_dir/cvs-authors" and
  read_author_info("$git_dir/cvs-authors");
if ($opt_A) {
	read_author_info($opt_A);
	write_author_info("$git_dir/cvs-authors");
}

604 605 606 607 608

#
# run cvsps into a file unless we are getting
# it passed as a file via $opt_P
#
609
my $cvspsfile;
610 611 612
unless ($opt_P) {
	print "Running cvsps...\n" if $opt_v;
	my $pid = open(CVSPS,"-|");
613
	my $cvspsfh;
614
	die "Cannot fork: $!\n" unless defined $pid;
J
Junio C Hamano 已提交
615
	unless ($pid) {
616 617 618 619 620 621 622 623 624
		my @opt;
		@opt = split(/,/,$opt_p) if defined $opt_p;
		unshift @opt, '-z', $opt_z if defined $opt_z;
		unshift @opt, '-q'         unless defined $opt_v;
		unless (defined($opt_p) && $opt_p =~ m/--no-cvs-direct/) {
			push @opt, '--cvs-direct';
		}
		exec("cvsps","--norc",@opt,"-u","-A",'--root',$opt_d,$cvs_tree);
		die "Could not start cvsps: $!\n";
625
	}
626 627
	($cvspsfh, $cvspsfile) = tempfile('gitXXXXXX', SUFFIX => '.cvsps',
					  DIR => File::Spec->tmpdir());
628 629
	while (<CVSPS>) {
	    print $cvspsfh $_;
630
	}
631 632
	close CVSPS;
	close $cvspsfh;
633 634
} else {
	$cvspsfile = $opt_P;
635 636
}

637
open(CVS, "<$cvspsfile") or die $!;
638

639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656
## cvsps output:
#---------------------
#PatchSet 314
#Date: 1999/09/18 13:03:59
#Author: wkoch
#Branch: STABLE-BRANCH-1-0
#Ancestor branch: HEAD
#Tag: (none)
#Log:
#    See ChangeLog: Sat Sep 18 13:03:28 CEST 1999  Werner Koch
#Members:
#	README:1.57->1.57.2.1
#	VERSION:1.96->1.96.2.1
#
#---------------------

my $state = 0;

J
Jeff King 已提交
657 658 659
sub update_index (\@\@) {
	my $old = shift;
	my $new = shift;
660 661 662 663
	open(my $fh, '|-', qw(git-update-index -z --index-info))
		or die "unable to open git-update-index: $!";
	print $fh
		(map { "0 0000000000000000000000000000000000000000\t$_\0" }
J
Jeff King 已提交
664
			@$old),
665
		(map { '100' . sprintf('%o', $_->[0]) . " $_->[1]\t$_->[2]\0" }
J
Jeff King 已提交
666
			@$new)
667 668 669 670
		or die "unable to write to git-update-index: $!";
	close $fh
		or die "unable to write to git-update-index: $!";
	$? and die "git-update-index reported error: $?";
J
Jeff King 已提交
671
}
672

J
Jeff King 已提交
673 674 675 676 677 678 679
sub write_tree () {
	open(my $fh, '-|', qw(git-write-tree))
		or die "unable to open git-write-tree: $!";
	chomp(my $tree = <$fh>);
	is_sha1($tree)
		or die "Cannot get tree id ($tree): $!";
	close($fh)
680 681
		or die "Error running git-write-tree: $?\n";
	print "Tree ID $tree\n" if $opt_v;
J
Jeff King 已提交
682 683
	return $tree;
}
684

J
Junio C Hamano 已提交
685 686
my ($patchset,$date,$author_name,$author_email,$branch,$ancestor,$tag,$logmsg);
my (@old,@new,@skipped,%ignorebranch);
687 688 689 690

# commits that cvsps cannot place anywhere...
$ignorebranch{'#CVSPS_NO_BRANCH'} = 1;

J
Jeff King 已提交
691
sub commit {
692 693
	if ($branch eq $opt_o && !$index{branch} && !get_headref($branch, $git_dir)) {
	    # looks like an initial commit
694
	    # use the index primed by git-init
695 696 697 698 699 700 701 702 703 704 705 706 707 708 709 710 711 712
	    $ENV{GIT_INDEX_FILE} = '.git/index';
	    $index{$branch} = '.git/index';
	} else {
	    # use an index per branch to speed up
	    # imports of projects with many branches
	    unless ($index{$branch}) {
		$index{$branch} = tmpnam();
		$ENV{GIT_INDEX_FILE} = $index{$branch};
		if ($ancestor) {
		    system("git-read-tree", $ancestor);
		} else {
		    system("git-read-tree", $branch);
		}
		die "read-tree failed: $?\n" if $?;
	    }
	}
        $ENV{GIT_INDEX_FILE} = $index{$branch};

J
Jeff King 已提交
713 714 715 716 717 718 719 720 721 722 723 724 725 726
	update_index(@old, @new);
	@old = @new = ();
	my $tree = write_tree();
	my $parent = get_headref($last_branch, $git_dir);
	print "Parent ID " . ($parent ? $parent : "(empty)") . "\n" if $opt_v;

	my @commit_args;
	push @commit_args, ("-p", $parent) if $parent;

	# loose detection of merges
	# based on the commit msg
	foreach my $rx (@mergerx) {
		next unless $logmsg =~ $rx && $1;
		my $mparent = $1 eq 'HEAD' ? $opt_o : $1;
J
Junio C Hamano 已提交
727
		if (my $sha1 = get_headref($mparent, $git_dir)) {
J
Jeff King 已提交
728 729
			push @commit_args, '-p', $mparent;
			print "Merge parent branch: $mparent\n" if $opt_v;
730
		}
731
	}
J
Jeff King 已提交
732 733

	my $commit_date = strftime("+0000 %Y-%m-%d %H:%M:%S",gmtime($date));
734 735 736 737 738 739
	$ENV{GIT_AUTHOR_NAME} = $author_name;
	$ENV{GIT_AUTHOR_EMAIL} = $author_email;
	$ENV{GIT_AUTHOR_DATE} = $commit_date;
	$ENV{GIT_COMMITTER_NAME} = $author_name;
	$ENV{GIT_COMMITTER_EMAIL} = $author_email;
	$ENV{GIT_COMMITTER_DATE} = $commit_date;
J
Jeff King 已提交
740 741
	my $pid = open2(my $commit_read, my $commit_write,
		'git-commit-tree', $tree, @commit_args);
742 743 744 745 746

	# compatibility with git2cvs
	substr($logmsg,32767) = "" if length($logmsg) > 32767;
	$logmsg =~ s/[\s\n]+\z//;

747 748 749
	if (@skipped) {
	    $logmsg .= "\n\n\nSKIPPED:\n\t";
	    $logmsg .= join("\n\t", @skipped) . "\n";
M
Martin Langhoff 已提交
750
	    @skipped = ();
751 752
	}

J
Jeff King 已提交
753
	print($commit_write "$logmsg\n") && close($commit_write)
754
		or die "Error writing to git-commit-tree: $!\n";
M
Matthias Urlichs 已提交
755

J
Jeff King 已提交
756 757 758
	print "Committed patch $patchset ($branch $commit_date)\n" if $opt_v;
	chomp(my $cid = <$commit_read>);
	is_sha1($cid) or die "Cannot get commit id ($cid): $!\n";
759
	print "Commit ID $cid\n" if $opt_v;
J
Jeff King 已提交
760
	close($commit_read);
M
Matthias Urlichs 已提交
761 762 763

	waitpid($pid,0);
	die "Error running git-commit-tree: $?\n" if $?;
764

765
	system("git-update-ref refs/heads/$branch $cid") == 0
766 767
		or die "Cannot write branch $branch for update: $!\n";

J
Junio C Hamano 已提交
768 769 770
	if ($tag) {
		my ($in, $out) = ('','');
	        my ($xtag) = $tag;
771 772
		$xtag =~ s/\s+\*\*.*$//; # Remove stuff like ** INVALID ** and ** FUNKY **
		$xtag =~ tr/_/\./ if ( $opt_u );
773
		$xtag =~ s/[\/]/$opt_s/g;
774 775 776 777 778
		
		my $pid = open2($in, $out, 'git-mktag');
		print $out "object $cid\n".
		    "type commit\n".
		    "tag $xtag\n".
779
		    "tagger $author_name <$author_email>\n"
780 781 782 783 784 785 786 787 788 789 790 791 792 793 794 795 796
		    or die "Cannot create tag object $xtag: $!\n";
		close($out)
		    or die "Cannot create tag object $xtag: $!\n";

		my $tagobj = <$in>;
		chomp $tagobj;

		if ( !close($in) or waitpid($pid, 0) != $pid or
		     $? != 0 or $tagobj !~ /^[0123456789abcdef]{40}$/ ) {
		    die "Cannot create tag object $xtag: $!\n";
	        }
		

		open(C,">$git_dir/refs/tags/$xtag")
			or die "Cannot create tag $xtag: $!\n";
		print C "$tagobj\n"
			or die "Cannot write tag $xtag: $!\n";
797
		close(C)
798 799 800
			or die "Cannot write tag $xtag: $!\n";

		print "Created tag '$xtag' on '$branch'\n" if $opt_v;
801 802 803
	}
};

804
my $commitcount = 1;
J
Junio C Hamano 已提交
805
while (<CVS>) {
806
	chomp;
J
Junio C Hamano 已提交
807
	if ($state == 0 and /^-+$/) {
808
		$state = 1;
J
Junio C Hamano 已提交
809
	} elsif ($state == 0) {
810 811
		$state = 1;
		redo;
J
Junio C Hamano 已提交
812
	} elsif (($state==0 or $state==1) and s/^PatchSet\s+//) {
813 814
		$patchset = 0+$_;
		$state=2;
J
Junio C Hamano 已提交
815
	} elsif ($state == 2 and s/^Date:\s+//) {
816
		$date = pdate($_);
J
Junio C Hamano 已提交
817
		unless ($date) {
818 819 820 821 822
			print STDERR "Could not parse date: $_\n";
			$state=0;
			next;
		}
		$state=3;
J
Junio C Hamano 已提交
823
	} elsif ($state == 3 and s/^Author:\s+//) {
824
		s/\s+$//;
825 826
		if (/^(.*?)\s+<(.*)>/) {
		    ($author_name, $author_email) = ($1, $2);
827 828 829
		} elsif ($conv_author_name{$_}) {
			$author_name = $conv_author_name{$_};
			$author_email = $conv_author_email{$_};
830 831 832
		} else {
		    $author_name = $author_email = $_;
		}
833
		$state = 4;
J
Junio C Hamano 已提交
834
	} elsif ($state == 4 and s/^Branch:\s+//) {
835
		s/\s+$//;
836
		s/[\/]/$opt_s/g;
837 838
		$branch = $_;
		$state = 5;
J
Junio C Hamano 已提交
839
	} elsif ($state == 5 and s/^Ancestor branch:\s+//) {
840 841
		s/\s+$//;
		$ancestor = $_;
842
		$ancestor = $opt_o if $ancestor eq "HEAD";
843
		$state = 6;
J
Junio C Hamano 已提交
844
	} elsif ($state == 5) {
845 846 847
		$ancestor = undef;
		$state = 6;
		redo;
J
Junio C Hamano 已提交
848
	} elsif ($state == 6 and s/^Tag:\s+//) {
849
		s/\s+$//;
J
Junio C Hamano 已提交
850
		if ($_ eq "(none)") {
851 852 853 854 855
			$tag = undef;
		} else {
			$tag = $_;
		}
		$state = 7;
J
Junio C Hamano 已提交
856
	} elsif ($state == 7 and /^Log:/) {
857 858
		$logmsg = "";
		$state = 8;
J
Junio C Hamano 已提交
859
	} elsif ($state == 8 and /^Members:/) {
860
		$branch = $opt_o if $branch eq "HEAD";
J
Junio C Hamano 已提交
861
		if (defined $branch_date{$branch} and $branch_date{$branch} >= $date) {
862
			# skip
863
			print "skip patchset $patchset: $date before $branch_date{$branch}\n" if $opt_v;
864 865 866
			$state = 11;
			next;
		}
867
		if (!$opt_a && $starttime - 300 - (defined $opt_z ? $opt_z : 300) <= $date) {
868 869 870 871 872 873 874 875
			# skip if the commit is too recent
			# that the cvsps default fuzz is 300s, we give ourselves another
			# 300s just in case -- this also prevents skipping commits
			# due to server clock drift
			print "skip patchset $patchset: $date too recent\n" if $opt_v;
			$state = 11;
			next;
		}
876 877 878 879 880
		if (exists $ignorebranch{$branch}) {
			print STDERR "Skipping $branch\n";
			$state = 11;
			next;
		}
J
Junio C Hamano 已提交
881 882
		if ($ancestor) {
			if ($ancestor eq $branch) {
883 884 885
				print STDERR "Branch $branch erroneously stems from itself -- changed ancestor to $opt_o\n";
				$ancestor = $opt_o;
			}
J
Junio C Hamano 已提交
886
			if (-f "$git_dir/refs/heads/$branch") {
887 888 889 890
				print STDERR "Branch $branch already exists!\n";
				$state=11;
				next;
			}
J
Junio C Hamano 已提交
891
			unless (open(H,"$git_dir/refs/heads/$ancestor")) {
892
				print STDERR "Branch $ancestor does not exist!\n";
893
				$ignorebranch{$branch} = 1;
894 895 896 897 898
				$state=11;
				next;
			}
			chomp(my $id = <H>);
			close(H);
J
Junio C Hamano 已提交
899
			unless (open(H,"> $git_dir/refs/heads/$branch")) {
900
				print STDERR "Could not create branch $branch: $!\n";
901
				$ignorebranch{$branch} = 1;
902 903 904 905 906 907 908 909
				$state=11;
				next;
			}
			print H "$id\n"
				or die "Could not write branch $branch: $!";
			close(H)
				or die "Could not write branch $branch: $!";
		}
910
		$last_branch = $branch if $branch ne $last_branch;
911
		$state = 9;
J
Junio C Hamano 已提交
912
	} elsif ($state == 8) {
913
		$logmsg .= "$_\n";
J
Junio C Hamano 已提交
914
	} elsif ($state == 9 and /^\s+(.+?):(INITIAL|\d+(?:\.\d+)+)->(\d+(?:\.\d+)+)\s*$/) {
915
#	VERSION:1.96->1.96.2.1
M
Matthias Urlichs 已提交
916
		my $init = ($2 eq "INITIAL");
917
		my $fn = $1;
M
Matthias Urlichs 已提交
918 919
		my $rev = $3;
		$fn =~ s#^/+##;
920 921 922 923 924 925
		if ($opt_S && $fn =~ m/$opt_S/) {
		    print "SKIPPING $fn v $rev\n";
		    push(@skipped, $fn);
		    next;
		}
		print "Fetching $fn   v $rev\n" if $opt_v;
926
		my ($tmpname, $size) = $cvs->file($fn,$rev);
J
Junio C Hamano 已提交
927
		if ($size == -1) {
928 929 930 931
			push(@old,$fn);
			print "Drop $fn\n" if $opt_v;
		} else {
			print "".($init ? "New" : "Update")." $fn: $size bytes\n" if $opt_v;
932 933 934 935
			my $pid = open(my $F, '-|');
			die $! unless defined $pid;
			if (!$pid) {
			    exec("git-hash-object", "-w", $tmpname)
936
				or die "Cannot create object: $!\n";
937
			}
938 939 940 941 942 943
			my $sha = <$F>;
			chomp $sha;
			close $F;
			my $mode = pmode($cvs->{'mode'});
			push(@new,[$mode, $sha, $fn]); # may be resurrected!
		}
944
		unlink($tmpname);
J
Junio C Hamano 已提交
945
	} elsif ($state == 9 and /^\s+(.+?):\d+(?:\.\d+)+->(\d+(?:\.\d+)+)\(DEAD\)\s*$/) {
M
Matthias Urlichs 已提交
946 947 948
		my $fn = $1;
		$fn =~ s#^/+##;
		push(@old,$fn);
949
		print "Delete $fn\n" if $opt_v;
J
Junio C Hamano 已提交
950
	} elsif ($state == 9 and /^\s*$/) {
951
		$state = 10;
J
Junio C Hamano 已提交
952
	} elsif (($state == 9 or $state == 10) and /^-+$/) {
953 954
		$commitcount++;
		if ($opt_L && $commitcount > $opt_L) {
955 956
			last;
		}
957
		commit();
958 959 960
		if (($commitcount & 1023) == 0) {
			system("git repack -a -d");
		}
961
		$state = 1;
J
Junio C Hamano 已提交
962
	} elsif ($state == 11 and /^-+$/) {
963
		$state = 1;
J
Junio C Hamano 已提交
964
	} elsif (/^-+$/) { # end of unknown-line processing
965
		$state = 1;
J
Junio C Hamano 已提交
966
	} elsif ($state != 11) { # ignore stuff when skipping
967 968 969
		print "* UNKNOWN LINE * $_\n";
	}
}
970
commit() if $branch and $state != 11;
971

972 973 974 975
unless ($opt_P) {
	unlink($cvspsfile);
}

976 977 978 979 980 981 982 983 984 985
# The heuristic of repacking every 1024 commits can leave a
# lot of unpacked data.  If there is more than 1MB worth of
# not-packed objects, repack once more.
my $line = `git-count-objects`;
if ($line =~ /^(\d+) objects, (\d+) kilobytes$/) {
  my ($n_objects, $kb) = ($1, $2);
  1024 < $kb
    and system("git repack -a -d");
}

986
foreach my $git_index (values %index) {
987 988 989
    if ($git_index ne '.git/index') {
	unlink($git_index);
    }
990
}
991

992 993 994 995 996 997
if (defined $orig_git_index) {
	$ENV{GIT_INDEX_FILE} = $orig_git_index;
} else {
	delete $ENV{GIT_INDEX_FILE};
}

998
# Now switch back to the branch we were in before all of this happened
J
Junio C Hamano 已提交
999
if ($orig_branch) {
1000 1001 1002 1003 1004 1005
	print "DONE.\n" if $opt_v;
	if ($opt_i) {
		exit 0;
	}
	my $tip_at_end = `git-rev-parse --verify HEAD`;
	if ($tip_at_start ne $tip_at_end) {
1006
		for ($tip_at_start, $tip_at_end) { chomp; }
1007 1008 1009 1010 1011 1012 1013 1014 1015
		print "Fetched into the current branch.\n" if $opt_v;
		system(qw(git-read-tree -u -m),
		       $tip_at_start, $tip_at_end);
		die "Fast-forward update failed: $?\n" if $?;
	}
	else {
		system(qw(git-merge cvsimport HEAD), "refs/heads/$opt_o");
		die "Could not merge $opt_o into the current branch.\n" if $?;
	}
1016 1017 1018
} else {
	$orig_branch = "master";
	print "DONE; creating $orig_branch branch\n" if $opt_v;
1019
	system("git-update-ref", "refs/heads/master", "refs/heads/$opt_o")
1020
		unless -f "$git_dir/refs/heads/master";
P
Pavel Roskin 已提交
1021
	system('git-update-ref', 'HEAD', "$orig_branch");
1022 1023 1024 1025
	unless ($opt_i) {
		system('git checkout');
		die "checkout failed: $?\n" if $?;
	}
1026
}