aboutsummaryrefslogtreecommitdiff
path: root/nix
diff options
context:
space:
mode:
Diffstat (limited to 'nix')
-rw-r--r--nix/guix-register/guix-register.cc168
1 files changed, 168 insertions, 0 deletions
diff --git a/nix/guix-register/guix-register.cc b/nix/guix-register/guix-register.cc
new file mode 100644
index 0000000000..0a028f0cfe
--- /dev/null
+++ b/nix/guix-register/guix-register.cc
@@ -0,0 +1,168 @@
+/* GNU Guix --- Functional package management for GNU
+ Copyright (C) 2013 Ludovic Courtès <ludo@gnu.org>
+ Copyright (C) 2007, 2008, 2009, 2010, 2011, 2012,
+ 2013 Eelco Dolstra <eelco.dolstra@logicblox.com>
+
+ This file is part of GNU Guix.
+
+ GNU Guix is free software; you can redistribute it and/or modify it
+ under the terms of the GNU General Public License as published by
+ the Free Software Foundation; either version 3 of the License, or (at
+ your option) any later version.
+
+ GNU Guix is distributed in the hope that it will be useful, but
+ WITHOUT ANY WARRANTY; without even the implied warranty of
+ MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ GNU General Public License for more details.
+
+ You should have received a copy of the GNU General Public License
+ along with GNU Guix. If not, see <http://www.gnu.org/licenses/>. */
+
+/* This file derives from the implementation of 'nix-store
+ --register-validity', by Eelco Dolstra, as found in the Nix package
+ manager's src/nix-store/nix-store.cc. */
+
+#include <config.h>
+
+#include <globals.hh>
+#include <local-store.hh>
+
+#include <iostream>
+#include <fstream>
+#include <cstdlib>
+#include <cstdio>
+
+#include <argp.h>
+
+using namespace nix;
+
+/* Input stream where we read closure descriptions. */
+static std::istream *input = &std::cin;
+
+
+
+/* Command-line options. */
+
+const char *argp_program_version =
+ "guix-register (" PACKAGE_NAME ") " PACKAGE_VERSION;
+const char *argp_program_bug_address = PACKAGE_BUGREPORT;
+
+static char doc[] =
+"guix-register -- register a closure as valid in a store\
+\v\
+This program is used internally when populating a store with data \
+from an existing store. It updates the new store's database with \
+information about which store files are valid, and what their \
+references are.";
+
+static const struct argp_option options[] =
+ {
+ { "prefix", 'p', "DIRECTORY", 0,
+ "Open the store that lies under DIRECTORY" },
+ { 0, 0, 0, 0, 0 }
+ };
+
+/* Parse a single option. */
+static error_t
+parse_opt (int key, char *arg, struct argp_state *state)
+{
+ switch (key)
+ {
+ case 'p':
+ {
+ string prefix = canonPath (arg);
+ settings.nixStore = prefix + NIX_STORE_DIR;
+ settings.nixDataDir = prefix + NIX_DATA_DIR;
+ settings.nixLogDir = prefix + NIX_LOG_DIR;
+ settings.nixStateDir = prefix + NIX_STATE_DIR;
+ settings.nixDBPath = settings.nixStateDir + "/db";
+ break;
+ }
+
+ case ARGP_KEY_ARG:
+ {
+ std::ifstream *file;
+
+ if (state->arg_num >= 2)
+ /* Too many arguments. */
+ argp_usage (state);
+
+ file = new std::ifstream ();
+ file->open (arg);
+
+ input = file;
+ }
+ break;
+
+ default:
+ return (error_t) ARGP_ERR_UNKNOWN;
+ }
+
+ return (error_t) 0;
+}
+
+/* Argument parsing. */
+static struct argp argp = { options, parse_opt, 0, doc };
+
+
+/* Read from INPUT the description of a closure, and register it as valid in
+ STORE. The expected format on INPUT is that used by #:references-graphs:
+
+ FILE
+ DERIVER
+ NUMBER-OF-REFERENCES
+ REF1
+ ...
+ REFN
+
+ This is really meant as an internal format. */
+static void
+register_validity (LocalStore *store, std::istream &input,
+ bool reregister = true, bool hashGiven = false,
+ bool canonicalise = true)
+{
+ ValidPathInfos infos;
+
+ while (1)
+ {
+ ValidPathInfo info = decodeValidPathInfo (input, hashGiven);
+ if (info.path == "")
+ break;
+ if (!store->isValidPath (info.path) || reregister)
+ {
+ /* !!! races */
+ if (canonicalise)
+ canonicalisePathMetaData (info.path, -1);
+
+ if (!hashGiven)
+ {
+ HashResult hash = hashPath (htSHA256, info.path);
+ info.hash = hash.first;
+ info.narSize = hash.second;
+ }
+ infos.push_back (info);
+ }
+ }
+
+ store->registerValidPaths (infos);
+}
+
+
+int
+main (int argc, char *argv[])
+{
+ try
+ {
+ argp_parse (&argp, argc, argv, 0, 0, 0);
+
+ LocalStore store;
+ register_validity (&store, *input);
+ }
+ catch (std::exception &e)
+ {
+ fprintf (stderr, "error: %s\n", e.what ());
+ return EXIT_FAILURE;
+ }
+
+ return EXIT_SUCCESS;
+}