133 lines
3.6 KiB
C
133 lines
3.6 KiB
C
/* -*- mode: C; c-file-style: "gnu"; indent-tabs-mode: nil; -*-
|
|
*
|
|
* Copyright (C) 2011 Colin Walters <walters@verbum.org>
|
|
*
|
|
* This program 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 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program 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 this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
|
|
*
|
|
* Author: Colin Walters <walters@verbum.org>
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include "ot-builtins.h"
|
|
#include "ostree.h"
|
|
|
|
#include <glib/gi18n.h>
|
|
|
|
static char *repo_path;
|
|
static gboolean quiet;
|
|
|
|
static GOptionEntry options[] = {
|
|
{ "repo", 0, 0, G_OPTION_ARG_FILENAME, &repo_path, "Repository path", NULL },
|
|
{ "quiet", 'q', 0, G_OPTION_ARG_NONE, &quiet, "Don't display informational messages", NULL },
|
|
{ NULL }
|
|
};
|
|
|
|
typedef struct {
|
|
guint n_objects;
|
|
} HtFsckData;
|
|
|
|
static void
|
|
object_iter_callback (OstreeRepo *repo,
|
|
const char *path,
|
|
GFileInfo *file_info,
|
|
gpointer user_data)
|
|
{
|
|
HtFsckData *data = user_data;
|
|
struct stat stbuf;
|
|
GChecksum *checksum = NULL;
|
|
GError *error = NULL;
|
|
char *dirname = NULL;
|
|
char *checksum_prefix = NULL;
|
|
char *checksum_string = NULL;
|
|
char *filename_checksum = NULL;
|
|
char *dot;
|
|
|
|
dirname = g_path_get_dirname (path);
|
|
checksum_prefix = g_path_get_basename (dirname);
|
|
|
|
/* nlinks = g_file_info_get_attribute_uint32 (file_info, "unix::nlink");
|
|
if (nlinks < 2 && !quiet)
|
|
g_printerr ("note: floating object: %s\n", path); */
|
|
|
|
if (!ostree_stat_and_checksum_file (-1, path, &checksum, &stbuf, &error))
|
|
goto out;
|
|
|
|
filename_checksum = g_strdup (g_file_info_get_name (file_info));
|
|
dot = strrchr (filename_checksum, '.');
|
|
g_assert (dot != NULL);
|
|
*dot = '\0';
|
|
|
|
checksum_string = g_strconcat (checksum_prefix, filename_checksum, NULL);
|
|
|
|
if (strcmp (checksum_string, g_checksum_get_string (checksum)) != 0)
|
|
{
|
|
g_printerr ("ERROR: corrupted object '%s' expected checksum: %s\n",
|
|
path, g_checksum_get_string (checksum));
|
|
}
|
|
|
|
data->n_objects++;
|
|
|
|
out:
|
|
if (checksum != NULL)
|
|
g_checksum_free (checksum);
|
|
g_free (dirname);
|
|
g_free (checksum_prefix);
|
|
g_free (checksum_string);
|
|
g_free (filename_checksum);
|
|
if (error != NULL)
|
|
{
|
|
g_printerr ("%s\n", error->message);
|
|
g_clear_error (&error);
|
|
}
|
|
}
|
|
|
|
gboolean
|
|
ostree_builtin_fsck (int argc, char **argv, const char *prefix, GError **error)
|
|
{
|
|
GOptionContext *context;
|
|
HtFsckData data;
|
|
gboolean ret = FALSE;
|
|
OstreeRepo *repo = NULL;
|
|
|
|
context = g_option_context_new ("- Check the repository for consistency");
|
|
g_option_context_add_main_entries (context, options, NULL);
|
|
|
|
if (!g_option_context_parse (context, &argc, &argv, error))
|
|
goto out;
|
|
|
|
if (repo_path == NULL)
|
|
repo_path = ".";
|
|
|
|
data.n_objects = 0;
|
|
|
|
repo = ostree_repo_new (repo_path);
|
|
if (!ostree_repo_check (repo, error))
|
|
goto out;
|
|
|
|
if (!ostree_repo_iter_objects (repo, object_iter_callback, &data, error))
|
|
goto out;
|
|
|
|
if (!quiet)
|
|
g_printerr ("Total Objects: %u\n", data.n_objects);
|
|
|
|
ret = TRUE;
|
|
out:
|
|
if (context)
|
|
g_option_context_free (context);
|
|
g_clear_object (&repo);
|
|
return ret;
|
|
}
|