about summary refs log tree commit diff
path: root/scripts/optimise-store.pl
blob: 556174c2253f3c2e13d5a2367c4e3aa15d782fe0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
#! /usr/bin/perl -w

use strict;
use File::Basename;


#my @paths = ("/nix/store");
my @paths = ("test");


print "hashing...\n";

my $hashList = "/tmp/nix-optimise-hash-list";

system("find @paths -type f -print0 | xargs -0 md5sum -- > $hashList") == 0
    or die "cannot hash store files";


print "sorting by hash...\n";

system("sort $hashList > $hashList.sorted") == 0
    or die "cannot sort list";


sub atomicLink {
    my $target = shift;
    my $new = shift;
    my $tmpNew = "${new}_optimise.$$";

    # Make the directory writable temporarily.
    my $dir = dirname $new;
    my @st = stat $dir or die;

    chmod ($st[2] | 0200, $dir) or die "cannot make `$dir' writable: $!";
    
    link $target, $tmpNew or die "cannot create hard link `$tmpNew': $!";

    rename $tmpNew, $new or die "cannot rename `$tmpNew' to `$new': $!";

    chmod ($st[2], $dir) or die "cannot restore permission on `$dir': $!";
    utime ($st[8], $st[9], $dir) or die "cannot restore timestamp on `$dir': $!";
}


print "hard-linking...\n";

open LIST, "<$hashList.sorted" or die;

my $prevFile;
my $prevHash;
my $prevInode;
my $prevExec;

my $totalSpace = 0;
my $savedSpace = 0;

while (<LIST>) {
    /^([0-9a-f]*)\s+(.*)$/ or die;
    my $curFile = $2;
    my $curHash = $1;

    my @st = stat $curFile or die;
    next if ($st[2] & 0222) != 0; # skip writable files

    my $fileSize = $st[7];
    $totalSpace += $fileSize;
    my $isExec = ($st[2] & 0111) == 0111;

    if (defined $prevHash && $curHash eq $prevHash
        && $prevExec == $isExec)
    {
        
        if ($st[1] != $prevInode) {
            print "$curFile = $prevFile\n";
            atomicLink $prevFile, $curFile;
            $savedSpace += $fileSize;
        }
        
    } else {
        $prevFile = $curFile;
        $prevHash = $curHash;
        $prevInode = $st[1];
        $prevExec = ($st[2] & 0111) == 0111;
    }
}

print "total space = $totalSpace\n";
print "saved space = $savedSpace\n";
my $savings = ($savedSpace / $totalSpace) * 100.0;
print "savings = $savings %\n";

close LIST;