From e40d4a5604a75540d94782d405dfff2000143f61 Mon Sep 17 00:00:00 2001
From: Eelco Dolstra <e.dolstra@tudelft.nl>
Date: Mon, 13 Nov 2006 16:48:27 +0000
Subject: [PATCH] * Option `--reregister' in `nix-store --register-validity'. 
 We need   this in the NixOS installer (or in the buildfarm) to ensure that
 the   cryptographic hash of the path contents still matches the actual  
 contents.

---
 doc/manual/release-notes.xml |  4 ++++
 src/nix-store/main.cc        | 26 ++++++++++++++++++--------
 2 files changed, 22 insertions(+), 8 deletions(-)

diff --git a/doc/manual/release-notes.xml b/doc/manual/release-notes.xml
index c4fd11f44..098e2a449 100644
--- a/doc/manual/release-notes.xml
+++ b/doc/manual/release-notes.xml
@@ -28,6 +28,10 @@
 
 
   <listitem><para>TODO: now using Berkeley DB 4.5.</para></listitem>
+
+
+  <listitem><para>Option <option>--reregister</option> in
+  <command>nix-store --register-validity</command>.</para></listitem>
     
 
 </itemizedlist>
diff --git a/src/nix-store/main.cc b/src/nix-store/main.cc
index ec35d430d..5a87806d0 100644
--- a/src/nix-store/main.cc
+++ b/src/nix-store/main.cc
@@ -138,6 +138,14 @@ static void opAddFixed(Strings opFlags, Strings opArgs)
 }
 
 
+static Hash parseHash16or32(HashType ht, const string & s)
+{
+    return s.size() == Hash(ht).hashSize * 2
+        ? parseHash(ht, s)
+        : parseHash32(ht, s);
+}
+
+
 /* Hack to support caching in `nix-prefetch-url'. */
 static void opPrintFixedPath(Strings opFlags, Strings opArgs)
 {
@@ -153,13 +161,9 @@ static void opPrintFixedPath(Strings opFlags, Strings opArgs)
     string hash = *i++;
     string name = *i++;
 
-    HashType ht(parseHashType(hashAlgo));
-    Hash h = hash.size() == Hash(ht).hashSize * 2
-        ? parseHash(ht, hash)
-        : parseHash32(ht, hash);
-    
     cout << format("%1%\n") %
-        makeFixedOutputPath(recursive, hashAlgo, h, name);
+        makeFixedOutputPath(recursive, hashAlgo,
+            parseHash16or32(parseHashType(hashAlgo), hash), name);
 }
 
 
@@ -487,7 +491,13 @@ static void opClearSubstitutes(Strings opFlags, Strings opArgs)
 
 static void opRegisterValidity(Strings opFlags, Strings opArgs)
 {
-    if (!opFlags.empty()) throw UsageError("unknown flag");
+    bool reregister = false; // !!! maybe this should be the default
+        
+    for (Strings::iterator i = opFlags.begin();
+         i != opFlags.end(); ++i)
+        if (*i == "--reregister") reregister = true;
+        else throw UsageError(format("unknown flag `%1%'") % *i);
+
     if (!opArgs.empty()) throw UsageError("no arguments expected");
 
     ValidPathInfos infos;
@@ -505,7 +515,7 @@ static void opRegisterValidity(Strings opFlags, Strings opArgs)
             info.references.insert(s);
         }
         if (!cin || cin.eof()) throw Error("missing input");
-        if (!isValidPath(info.path)) {
+        if (!isValidPath(info.path) || reregister) {
             /* !!! races */
             canonicalisePathMetaData(info.path);
             info.hash = hashPath(htSHA256, info.path);