diff --git a/Makefile.am b/Makefile.am index d25ccc5c28..70857e62ad 100644 --- a/Makefile.am +++ b/Makefile.am @@ -19,8 +19,12 @@ init-state: $(INSTALL) -d $(DESTDIR)$(localstatedir)/nix/db $(INSTALL) -d $(DESTDIR)$(localstatedir)/log/nix $(INSTALL) -d $(DESTDIR)$(localstatedir)/nix/profiles + $(INSTALL) -d $(DESTDIR)$(localstatedir)/nix/gcroots + $(INSTALL) -d $(DESTDIR)$(localstatedir)/nix/gcroots/tmp + rm -f $(DESTDIR)$(localstatedir)/nix/gcroots/profiles + ln -s $(localstatedir)/nix/profiles $(DESTDIR)$(localstatedir)/nix/gcroots/profiles $(INSTALL) -d $(DESTDIR)$(prefix)/store # $(bindir)/nix-store --init else init-state: -endif \ No newline at end of file +endif diff --git a/scripts/nix-collect-garbage.in b/scripts/nix-collect-garbage.in index fc67f1d55d..8b571536d8 100755 --- a/scripts/nix-collect-garbage.in +++ b/scripts/nix-collect-garbage.in @@ -3,41 +3,74 @@ use strict; use IPC::Open2; -my $linkdir = "@localstatedir@/nix/profiles"; -my $storedir = "@storedir@"; +my $rootsDir = "@localstatedir@/nix/gcroots"; +my $storeDir = "@storedir@"; my %alive; -my $keepsuccessors = 1; +my $keepSuccessors = 1; my $invert = 0; +my @roots = (); + + +# Parse the command line. foreach my $arg (@ARGV) { - if ($arg eq "--no-successors") { $keepsuccessors = 0; } + if ($arg eq "--no-successors") { $keepSuccessors = 0; } elsif ($arg eq "--invert") { $invert = 1; } else { die "unknown argument `$arg'" }; } -opendir(DIR, $linkdir) or die "cannot open directory $linkdir: $!"; -my @links = readdir DIR or die "cannot read directory $linkdir: $!"; -closedir DIR; -my @roots; -foreach my $link (@links) { - $link = $linkdir . "/" . $link; - next if (!($link =~ /.gcroot$/)); - open ROOT, "<$link" or die "cannot open $link: $!"; +# Read all GC roots from the given file. +sub readRoots { + my $fileName = shift; + open ROOT, "<$fileName" or die "cannot open `$fileName': $!"; while () { chomp; foreach my $root (split ' ') { - die "bad root `$root' in file `$link'" unless $root =~ /^\S+$/; + die "bad root `$root' in file `$fileName'" + unless $root =~ /^\S+$/; push @roots, $root; } } close ROOT; } + +# Recursively finds all *.gcroot files in the given directory. +sub findRoots; +sub findRoots { + my $followSymlinks = shift; + my $dir = shift; + + opendir(DIR, $dir) or die "cannot open directory `$dir': $!"; + my @names = readdir DIR or die "cannot read directory `$dir': $!"; + closedir DIR; + + foreach my $name (@names) { + next if $name eq "." || $name eq ".."; + $name = $dir . "/" . $name; + if ($name =~ /.gcroot$/ && -f $name) { + readRoots $name; + } + elsif (-d $name) { + if ($followSymlinks || !-l $name) { + findRoots 0, $name; + } + } + } + +} + + +# Find GC roots, starting at $rootsDir. +findRoots 1, $rootsDir; + + +# Determine all store paths reachable from the roots. my $extraarg = ""; -if ($keepsuccessors) { $extraarg = "--include-successors"; }; +if ($keepSuccessors) { $extraarg = "--include-successors"; }; my $pid = open2(\*READ, \*WRITE, "@bindir@/nix-store --query --requisites $extraarg @roots") or die "determining live paths"; close WRITE; @@ -53,14 +86,15 @@ $? == 0 or die "determining live paths"; exit 0 if ($invert); -opendir(DIR, $storedir) or die "cannot open directory $storedir: $!"; -my @names = readdir DIR; -closedir DIR; -foreach my $name (@names) { +# Using that information, find all store paths *not* reachable from +# the roots. +opendir(DIR, $storeDir) or die "cannot open directory $storeDir: $!"; +foreach my $name (readdir DIR) { next if ($name eq "." || $name eq ".."); - $name = "$storedir/$name"; + $name = "$storeDir/$name"; if (!$alive{$name}) { print "$name\n"; } } +closedir DIR;