Skip to content

Commit

Permalink
transactions: move all tmpobject-%s files under a per-boot subdir
Browse files Browse the repository at this point in the history 
This prevents to use files after a kernel crash or power failure and
that can be not completely synced to disk.

Signed-off-by: Giuseppe Scrivano <[email protected]>
  • Loading branch information
@giuseppe
giuseppe committed Jan 29, 2015
1 parent 35821c9 commit c88ea80
Show file tree
Hide file tree
Showing 3 changed files with 137 additions and 45 deletions.
168 changes: 127 additions & 41 deletions src/libostree/ostree-repo-commit.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -60,15 +60,42 @@ _ostree_repo_ensure_loose_objdir_at (int dfd,
return TRUE;
}

void
static const gchar *
ostree_repo_get_tmpobject_bootid (GError **error)
{
static gchar *contents;
static gsize bootid_initialized;
if (g_once_init_enter (&bootid_initialized))
{
if (g_file_get_contents ("/proc/sys/kernel/random/boot_id",
&contents,
NULL,
error))
g_strdelimit (contents, "\n", '\0');

g_once_init_leave (&bootid_initialized, 1);
}

return contents;
}

gboolean
_ostree_repo_get_tmpobject_path (char *output,
const char *checksum,
OstreeObjectType objtype)
OstreeObjectType objtype,
GCancellable *cancellable,
GError **error)
{
const char *boot_id;
if ((boot_id = ostree_repo_get_tmpobject_bootid (error)) == NULL)
return FALSE;

g_sprintf (output,
"tmpobject-%s.%s",
"%s/tmpobject-%s.%s",
boot_id,
checksum,
ostree_object_type_to_string (objtype));
return TRUE;
}

static GVariant *
Expand Down Expand Up @@ -289,7 +316,9 @@ commit_loose_object_trusted (OstreeRepo *self,
if (self->in_transaction)
{
char tmpbuf[_OSTREE_LOOSE_PATH_MAX];
_ostree_repo_get_tmpobject_path (tmpbuf, checksum, objtype);
if (! _ostree_repo_get_tmpobject_path (tmpbuf, checksum, objtype,
cancellable, error))
goto out;
tmp_dest = g_strdup (tmpbuf);
dir = self->tmp_dir_fd;
dest = tmp_dest;
Expand Down Expand Up @@ -944,6 +973,7 @@ ostree_repo_prepare_transaction (OstreeRepo *self,
gboolean ret = FALSE;
gboolean ret_transaction_resume = FALSE;
gs_free char *transaction_str = NULL;
const char *boot_id;

g_return_val_if_fail (self->in_transaction == FALSE, FALSE);

Expand All @@ -963,6 +993,20 @@ ostree_repo_prepare_transaction (OstreeRepo *self,
if (!ot_gfile_ensure_unlinked (self->transaction_lock_path, cancellable, error))
goto out;
}

if ((boot_id = ostree_repo_get_tmpobject_bootid (error)) == NULL)
goto out;

if (mkdirat (self->tmp_dir_fd, boot_id, 0777) == -1)
{
int errsv = errno;
if (G_UNLIKELY (errsv != EEXIST))
{
gs_set_error_from_errno (error, errsv);
goto out;
}
}

transaction_str = g_strdup_printf ("pid=%llu", (unsigned long long) getpid ());
if (!g_file_make_symbolic_link (self->transaction_lock_path, transaction_str,
cancellable, error))
Expand All @@ -975,9 +1019,82 @@ ostree_repo_prepare_transaction (OstreeRepo *self,
return ret;
}

static gboolean
rename_pending_loose_objects (OstreeRepo *self,
GCancellable *cancellable,
GError **error)
{
gboolean ret = FALSE;
const char *boot_id;
gs_unref_object GFile *tmpobjectsdir = NULL;
int dir_fd = -1;
gs_dirfd_iterator_cleanup GSDirFdIterator child_dfd_iter = { 0, };

if ((boot_id = ostree_repo_get_tmpobject_bootid (error)) == NULL)
goto out;

tmpobjectsdir = g_file_get_child (self->tmp_dir, boot_id);
if (! tmpobjectsdir)
goto out;

if (! gs_file_open_dir_fd (tmpobjectsdir, &dir_fd, cancellable, error))
goto out;

if (!gs_dirfd_iterator_init_at (self->tmp_dir_fd, boot_id, FALSE, &child_dfd_iter, error))
goto out;

while (TRUE)
{
struct dirent *out_dent;

if (!gs_dirfd_iterator_next_dent (&child_dfd_iter, &out_dent, cancellable, error))
goto out;

if (out_dent == NULL)
break;

if (strncmp (out_dent->d_name, "tmpobject-", 10) == 0)
{
char loose_path[_OSTREE_LOOSE_PATH_MAX];
gs_free gchar *checksum = NULL;
OstreeObjectType type;
ostree_object_from_string (out_dent->d_name + 10,
&checksum,
&type);

_ostree_loose_path (loose_path, checksum, type, self->mode);

if (!_ostree_repo_ensure_loose_objdir_at (self->objects_dir_fd, loose_path,
cancellable, error))
goto out;

if (G_UNLIKELY (renameat (dir_fd, out_dent->d_name,
self->objects_dir_fd, loose_path) < 0))
{
(void) unlinkat (self->tmp_dir_fd, out_dent->d_name, 0);
if (errno != EEXIST)
{
gs_set_error_from_errno (error, errno);
g_prefix_error (error, "Storing file '%s': ", loose_path);
goto out;
}
}
continue;
}
}

if (!gs_shutil_rm_rf_at (self->tmp_dir_fd, boot_id, cancellable, error))
goto out;

ret = TRUE;
out:
if (dir_fd >= 0)
close (dir_fd);
return ret;
}

static gboolean
cleanup_tmpdir (OstreeRepo *self,
gboolean move_tmpobject,
GCancellable *cancellable,
GError **error)
{
Expand All @@ -1000,47 +1117,13 @@ cleanup_tmpdir (OstreeRepo *self,
GFile *path;
guint64 mtime;
guint64 delta;
gs_free char *basename = NULL;

if (!gs_file_enumerator_iterate (enumerator, &file_info, &path,
cancellable, error))
goto out;
if (file_info == NULL)
break;

if (move_tmpobject)
{
basename = g_file_get_basename (path);
if (strncmp (basename, "tmpobject-", 10) == 0)
{
char loose_path[_OSTREE_LOOSE_PATH_MAX];
gs_free gchar *checksum = NULL;
OstreeObjectType type;
ostree_object_from_string (basename + 10,
&checksum,
&type);

_ostree_loose_path (loose_path, checksum, type, self->mode);

if (!_ostree_repo_ensure_loose_objdir_at (self->objects_dir_fd, loose_path,
cancellable, error))
goto out;

if (G_UNLIKELY (renameat (self->tmp_dir_fd, basename,
self->objects_dir_fd, loose_path) < 0))
{
(void) unlinkat (self->tmp_dir_fd, basename, 0);
if (errno != EEXIST)
{
gs_set_error_from_errno (error, errno);
g_prefix_error (error, "Storing file '%s': ", loose_path);
goto out;
}
}
continue;
}
}

mtime = g_file_info_get_attribute_uint64 (file_info, "time::modified");
if (mtime > curtime_secs)
continue;
Expand Down Expand Up @@ -1183,7 +1266,10 @@ ostree_repo_commit_transaction (OstreeRepo *self,
goto out;
}

if (!cleanup_tmpdir (self, TRUE, cancellable, error))
if (! rename_pending_loose_objects (self, cancellable, error))
goto out;

if (!cleanup_tmpdir (self, cancellable, error))
goto out;

if (self->loose_object_devino_hash)
Expand Down Expand Up @@ -1217,7 +1303,7 @@ ostree_repo_abort_transaction (OstreeRepo *self,
if (!self->in_transaction)
return TRUE;

if (!cleanup_tmpdir (self, FALSE, cancellable, error))
if (!cleanup_tmpdir (self, cancellable, error))
goto out;

if (self->loose_object_devino_hash)
Expand Down
6 changes: 4 additions & 2 deletions src/libostree/ostree-repo-private.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -82,10 +82,12 @@ _ostree_repo_ensure_loose_objdir_at (int dfd,
const char *loose_path,
GCancellable *cancellable,
GError **error);
void
gboolean
_ostree_repo_get_tmpobject_path (char *output,
const char *checksum,
OstreeObjectType objtype);
OstreeObjectType objtype,
GCancellable *cancellable,
GError **error);

gboolean
_ostree_repo_find_object (OstreeRepo *self,
Expand Down
8 changes: 6 additions & 2 deletions src/libostree/ostree-repo.c
View file
Original file line number Diff line number Diff line change
Expand Up @@ -1717,7 +1717,9 @@ load_metadata_internal (OstreeRepo *self,

if (self->in_transaction && fd < 0)
{
_ostree_repo_get_tmpobject_path (loose_path_buf, sha256, objtype);
if (! _ostree_repo_get_tmpobject_path (loose_path_buf, sha256, objtype,
cancellable, error))
goto out;
if (!openat_allow_noent (self->tmp_dir_fd, loose_path_buf, &fd, cancellable, error))
goto out;
}
Expand Down Expand Up @@ -2129,7 +2131,9 @@ _ostree_repo_has_loose_object (OstreeRepo *self,

if (self->in_transaction)
{
_ostree_repo_get_tmpobject_path (loose_path_buf, checksum, objtype);
if (! _ostree_repo_get_tmpobject_path (loose_path_buf, checksum, objtype,
cancellable, error))
goto out;
do
res = fstatat (self->tmp_dir_fd, loose_path_buf, &stbuf, AT_SYMLINK_NOFOLLOW);
while (G_UNLIKELY (res == -1 && errno == EINTR));
Expand Down

0 comments on commit c88ea80

Please sign in to comment.