snapshot: ensure that snapshots cannot be created unless with create_snapshot

diff --git a/src/snapshot.c b/src/snapshot.c
index ea3894bb91b33b1f52c3a280ab30be986648ad80..399aaea597a824a065f9e64a06033bd10cc8a8ff 100644 (file)
--- a/src/snapshot.c
+++ b/src/snapshot.c
@@ -47,6 +47,21 @@ static void snapshot_set_state(Snapshot *s, SnapshotState state) {
         unit_notify(UNIT(s), state_translation_table[old_state], state_translation_table[state]);
 }
 
+static int snapshot_load(Unit *u) {
+        Snapshot *s = SNAPSHOT(u);
+
+        assert(u);
+        assert(u->meta.load_state == UNIT_STUB);
+
+        /* Make sure that only snapshots created via snapshot_create()
+         * can be loaded */
+        if (!s->by_snapshot_create)
+                return -ENOENT;
+
+        u->meta.load_state = UNIT_LOADED;
+        return 0;
+}
+
 static int snapshot_coldplug(Unit *u) {
         Snapshot *s = SNAPSHOT(u);
 
@@ -197,12 +212,16 @@ int snapshot_create(Manager *m, const char *name, bool cleanup, Snapshot **_s) {
                 name = n;
         }
 
-        r = manager_load_unit(m, name, NULL, &u);
+        r = manager_load_unit_prepare(m, name, NULL, &u);
         free(n);
 
         if (r < 0)
                 goto fail;
 
+        SNAPSHOT(u)->by_snapshot_create = true;
+        manager_dispatch_load_queue(m);
+        assert(u->meta.load_state == UNIT_LOADED);
+
         HASHMAP_FOREACH_KEY(other, k, m->units, i) {
 
                 if (UNIT_VTABLE(other)->no_snapshots)
@@ -258,7 +277,7 @@ const UnitVTable snapshot_vtable = {
         .no_snapshots = true,
         .no_gc = true,
 
-        .load = unit_load_nop,
+        .load = snapshot_load,
         .coldplug = snapshot_coldplug,
 
         .dump = snapshot_dump,

diff --git a/src/snapshot.h b/src/snapshot.h
index 959a5090ecf318d0242191cf954b61788d55a660..25617906392c2d3acc518af9d793d7e78ecb3f3a 100644 (file)
--- a/src/snapshot.h
+++ b/src/snapshot.h
@@ -39,6 +39,7 @@ struct Snapshot {
         SnapshotState state, deserialized_state;
 
         bool cleanup;
+        bool by_snapshot_create:1;
 };
 
 extern const UnitVTable snapshot_vtable;