I'll write a 'lxc-snapshot' program using this api next, but here is
a first shot at the actual API functionality.

See the comments in lxccontainer.h and the example usage in
src/tests/snapshot.c for details.

Signed-off-by: Serge Hallyn <serge.hal...@ubuntu.com>
---
 src/lxc/lxccontainer.c | 281 +++++++++++++++++++++++++++++++++++++++++++++++++
 src/lxc/lxccontainer.h |  51 +++++++++
 src/tests/Makefile.am  |   7 +-
 src/tests/snapshot.c   | 130 +++++++++++++++++++++++
 4 files changed, 467 insertions(+), 2 deletions(-)
 create mode 100644 src/tests/snapshot.c

diff --git a/src/lxc/lxccontainer.c b/src/lxc/lxccontainer.c
index 3c657ca..989178a 100644
--- a/src/lxc/lxccontainer.c
+++ b/src/lxc/lxccontainer.c
@@ -27,6 +27,7 @@
 #include <errno.h>
 #include <fcntl.h>
 #include <sched.h>
+#include <dirent.h>
 #include "config.h"
 #include "lxc.h"
 #include "state.h"
@@ -68,6 +69,13 @@ static bool file_exists(char *f)
        return stat(f, &statbuf) == 0;
 }
 
+static void remove_trailing_slashes(char *p)
+{
+       int l = strlen(p);
+       while (--l >= 0 && (p[l] == '/' || p[l] == '\n'))
+               p[l] = '\0';
+}
+
 /*
  * A few functions to help detect when a container creation failed.
  * If a container creation was killed partway through, then trying
@@ -2192,6 +2200,275 @@ static int lxcapi_attach_run_wait(struct lxc_container 
*c, lxc_attach_options_t
        return lxc_wait_for_pid_status(pid);
 }
 
+int get_next_index(const char *lxcpath, char *cname)
+{
+       char *fname;
+       struct stat sb;
+       int i = 0, ret;
+
+       fname = alloca(strlen(lxcpath) + strlen(cname) + 20);
+       while (1) {
+               sprintf(fname, "%s/%s_%d", lxcpath, cname, i);
+               ret = stat(fname, &sb);
+               if (ret != 0)
+                       return i;
+               i++;
+       }
+}
+
+static int lxcapi_snapshot(struct lxc_container *c, char *commentfile)
+{
+       int i, flags, ret;
+       struct lxc_container *c2;
+       char snappath[MAXPATHLEN], newname[20];
+
+       // /var/lib/lxc -> /var/lib/lxcsnaps \0
+       ret = snprintf(snappath, MAXPATHLEN, "%ssnaps/%s", c->config_path, 
c->name);
+       if (ret < 0 || ret >= MAXPATHLEN)
+               return -1;
+       i = get_next_index(snappath, c->name);
+
+       if (mkdir_p(snappath, 0755) < 0) {
+               ERROR("Failed to create snapshot directory %s", snappath);
+               return -1;
+       }
+
+       ret = snprintf(newname, 20, "snap%d", i);
+       if (ret < 0 || ret >= 20)
+               return -1;
+
+       flags = LXC_CLONE_SNAPSHOT | LXC_CLONE_KEEPMACADDR | LXC_CLONE_KEEPNAME;
+       c2 = c->clone(c, newname, snappath, flags, NULL, NULL, 0, NULL);
+       if (!c2) {
+               ERROR("clone of %s:%s failed\n", c->config_path, c->name);
+               return -1;
+       }
+
+       lxc_container_put(c2);
+
+       // Now write down the creation time
+       time_t timer;
+       char buffer[25];
+       struct tm* tm_info;
+
+       time(&timer);
+       tm_info = localtime(&timer);
+
+       strftime(buffer, 25, "%Y:%m:%d %H:%M:%S", tm_info);
+
+       char *dfnam = alloca(strlen(snappath) + strlen(newname) + 5);
+       sprintf(dfnam, "%s/%s/ts", snappath, newname);
+       FILE *f = fopen(dfnam, "w");
+       if (!f) {
+               ERROR("Failed to open %s\n", dfnam);
+               return -1;
+       }
+       if (fprintf(f, "%s", buffer) < 0) {
+               SYSERROR("Writing timestamp");
+               fclose(f);
+               return -1;
+       }
+       if (fclose(f) != 0) {
+               SYSERROR("Writing timestamp");
+               return -1;
+       }
+
+       if (commentfile) {
+               // $p / $name / comment \0
+               int len = strlen(snappath) + strlen(newname) + 10;
+               char *path = alloca(len);
+               sprintf(path, "%s/%s/comment", snappath, newname);
+               return copy_file(commentfile, path) < 0 ? -1 : i;
+       }
+
+       return i;
+}
+
+static struct lxc_container *lxcsnap_open(struct lxc_snapshot *s)
+{
+       return NULL;
+}
+
+static void lxcsnap_free(struct lxc_snapshot *s)
+{
+       if (s->name)
+               free(s->name);
+       if (s->comment_pathname)
+               free(s->comment_pathname);
+       if (s->timestamp)
+               free(s->timestamp);
+       if (s->lxcpath)
+               free(s->lxcpath);
+}
+
+static char *get_snapcomment(char* snappath, char *name)
+{
+       // $snappath/$name/comment
+       int ret, len = strlen(snappath) + strlen(name) + 10;
+       char *s = malloc(len);
+
+       if (s) {
+               ret = snprintf(s, len, "%s/%s/comment", snappath, name);
+               if (ret < 0 || ret >= len) {
+                       free(s);
+                       s = NULL;
+               }
+       }
+       return s;
+}
+
+static char *get_timestamp(char* snappath, char *name)
+{
+       char path[MAXPATHLEN], *s = NULL;
+       int ret, len;
+       FILE *fin;
+
+       ret = snprintf(path, MAXPATHLEN, "%s/%s/ts", snappath, name);
+       if (ret < 0 || ret >= MAXPATHLEN)
+               return NULL;
+       if ((fin = fopen(path, "r")) == NULL)
+               return NULL;
+       (void) fseek(fin, 0, SEEK_END);
+       len = ftell(fin);
+       (void) fseek(fin, 0, SEEK_SET);
+       if (len > 0) {
+               s = malloc(len+1);
+               if (s) {
+                       s[len] = '\0';
+                       if (fread(s, 1, len, fin) != len) {
+                               SYSERROR("reading timestamp");
+                               free(s);
+                               s = NULL;
+                       }
+               }
+       }
+       fclose(fin);
+       return s;
+}
+
+static int lxcapi_snapshot_list(struct lxc_container *c, struct lxc_snapshot 
**ret_snaps)
+{
+       char snappath[MAXPATHLEN], path2[MAXPATHLEN];
+       int dirlen, count = 0, ret;
+       struct dirent dirent, *direntp;
+       struct lxc_snapshot *snaps =NULL, *nsnaps;
+       DIR *dir;
+
+       if (!c || !lxcapi_is_defined(c))
+               return -1;
+       // snappath is ${lxcpath}snaps/${lxcname}/
+       dirlen = snprintf(snappath, MAXPATHLEN, "%ssnaps/%s", c->config_path, 
c->name);
+       if (dirlen < 0 || dirlen >= MAXPATHLEN) {
+               ERROR("path name too long");
+               return -1;
+       }
+       if (!(dir = opendir(snappath))) {
+               INFO("failed to open %s - assuming no snapshots", snappath);
+               return 0;
+       }
+
+       while (!readdir_r(dir, &dirent, &direntp)) {
+               if (!direntp)
+                       break;
+
+               if (!strcmp(direntp->d_name, "."))
+                       continue;
+
+               if (!strcmp(direntp->d_name, ".."))
+                       continue;
+
+               ret = snprintf(path2, MAXPATHLEN, "%s/%s/config", snappath, 
direntp->d_name);
+               if (ret < 0 || ret >= MAXPATHLEN) {
+                       ERROR("pathname too long");
+                       goto out_free;
+               }
+               if (!file_exists(path2))
+                       continue;
+               nsnaps = realloc(snaps, (count + 1)*sizeof(*snaps));
+               if (!nsnaps) {
+                       SYSERROR("Out of memory");
+                       goto out_free;
+               }
+               snaps = nsnaps;
+               snaps[count].open = lxcsnap_open;
+               snaps[count].free = lxcsnap_free;
+               snaps[count].name = strdup(direntp->d_name);
+               if (!snaps[count].name)
+                       goto out_free;
+               snaps[count].lxcpath = strdup(snappath);
+               if (!snaps[count].lxcpath) {
+                       free(snaps[count].name);
+                       goto out_free;
+               }
+               snaps[count].comment_pathname = get_snapcomment(snappath, 
direntp->d_name);
+               snaps[count].timestamp = get_timestamp(snappath, 
direntp->d_name);
+               count++;
+       }
+
+       if (closedir(dir))
+               WARN("failed to close directory");
+
+       *ret_snaps = snaps;
+       return count;
+
+out_free:
+       if (snaps) {
+               int i;
+               for (i=0; i<count; i++)
+                       lxcsnap_free(&snaps[i]);
+               free(snaps);
+       }
+       return -1;
+}
+
+static bool lxcapi_snapshot_restore(struct lxc_container *c, char *snapname, 
char *newname)
+{
+       char clonelxcpath[MAXPATHLEN];
+       int ret;
+       struct lxc_container *snap, *rest;
+       struct bdev *bdev;
+       bool b = false;
+
+       if (!c || !c->name || !c->config_path)
+               return false;
+
+       bdev = bdev_init(c->lxc_conf->rootfs.path, c->lxc_conf->rootfs.mount, 
NULL);
+       if (!bdev) {
+               ERROR("Failed to find original backing store type");
+               return false;
+       }
+       if (strcmp(c->name, newname) == 0) {
+               if (!lxcapi_destroy(c)) {
+                       ERROR("Could not destroy existing container %s", 
newname);
+                       bdev_put(bdev);
+                       return false;
+               }
+       }
+       ret = snprintf(clonelxcpath, MAXPATHLEN, "%ssnaps/%s", c->config_path, 
c->name);
+       if (ret < 0 || ret >= MAXPATHLEN) {
+               bdev_put(bdev);
+               return false;
+       }
+       // how should we lock this?
+
+       snap = lxc_container_new(snapname, clonelxcpath);
+       if (!snap || !lxcapi_is_defined(snap)) {
+               ERROR("Could not open snapshot %s", snapname);
+               if (snap) lxc_container_put(snap);
+               bdev_put(bdev);
+               return false;
+       }
+
+       rest = lxcapi_clone(snap, newname, c->config_path, 0, bdev->type, NULL, 
0, NULL);
+       bdev_put(bdev);
+       if (rest && lxcapi_is_defined(rest))
+               b = true;
+       if (rest)
+               lxc_container_put(rest);
+       lxc_container_put(snap);
+       return b;
+}
+
 static int lxcapi_attach_run_waitl(struct lxc_container *c, 
lxc_attach_options_t *options, const char *program, const char *arg, ...)
 {
        va_list ap;
@@ -2237,6 +2514,7 @@ struct lxc_container *lxc_container_new(const char *name, 
const char *configpath
                goto err;
        }
 
+       remove_trailing_slashes(c->config_path);
        c->name = malloc(strlen(name)+1);
        if (!c->name) {
                fprintf(stderr, "Error allocating lxc_container name\n");
@@ -2305,6 +2583,9 @@ struct lxc_container *lxc_container_new(const char *name, 
const char *configpath
        c->attach = lxcapi_attach;
        c->attach_run_wait = lxcapi_attach_run_wait;
        c->attach_run_waitl = lxcapi_attach_run_waitl;
+       c->snapshot = lxcapi_snapshot;
+       c->snapshot_list = lxcapi_snapshot_list;
+       c->snapshot_restore = lxcapi_snapshot_restore;
 
        /* we'll allow the caller to update these later */
        if (lxc_log_init(NULL, "none", NULL, "lxc_container", 0, 
c->config_path)) {
diff --git a/src/lxc/lxccontainer.h b/src/lxc/lxccontainer.h
index ad6afa1..6cafc25 100644
--- a/src/lxc/lxccontainer.h
+++ b/src/lxc/lxccontainer.h
@@ -38,6 +38,8 @@
 
 struct bdev_specs;
 
+struct lxc_snapshot;
+
 struct lxc_container {
        // private fields
        char *name;
@@ -177,6 +179,55 @@ struct lxc_container {
        /* run program in container, wait for it to exit */
        int (*attach_run_wait)(struct lxc_container *c, lxc_attach_options_t 
*options, const char *program, const char * const argv[]);
        int (*attach_run_waitl)(struct lxc_container *c, lxc_attach_options_t 
*options, const char *program, const char *arg, ...);
+
+       /*
+       * snapshot:
+       * If you have /var/lib/lxc/c1 and call c->snapshot() the firs time, it
+       * will return 0, and the container will be /var/lib/lxcsnaps/c1/snap0.
+       * The second call will return 1, and the snapshot will be
+       * /var/lib/lxcsnaps/c1/snap1.
+       *
+       * On error, returns -1.
+       */
+       int (*snapshot)(struct lxc_container *c, char *commentfile);
+
+       /*
+        * snapshot_list() will return a description of all snapshots of c in
+        * a simple array.  See src/tests/snapshot.c for the proper way to
+        * free the allocated results.
+        *
+        * Returns the number of snapshots.
+        */
+       int (*snapshot_list)(struct lxc_container *, struct lxc_snapshot **);
+
+       /*
+        * snapshot_restore() will create a new container based on a snapshot.
+        * c is the container whose snapshot we look for, and snapname is the
+        * specific snapshot name (i.e. "snap0").  newname is the name to be
+        * used for the restored container.  If newname is the same as
+        * c->name, then c will first be destroyed.  That will fail if the
+        * snapshot is overlayfs-based, since the snapshots will pin the
+        * original container.
+        *
+        * The restored container will be a copy (not snapshot) of the snapshot,
+        * and restored in the lxcpath of the original container.
+        *
+        * As an example, c might be /var/lib/lxc/c1, snapname  might be 'snap0'
+        * which stands for /var/lib/lxcsnaps/c1/snap0.  If newname is c2,
+        * then snap0 will be copied to /var/lib/lxc/c2.
+        *
+        * Returns true on success, false on failure.
+        */
+       bool (*snapshot_restore)(struct lxc_container *c, char *snapname, char 
*newname);
+};
+
+struct lxc_snapshot {
+       char *name;
+       char *comment_pathname;
+       char *timestamp;
+       char *lxcpath;
+       struct lxc_container *(*open)(struct lxc_snapshot *);
+       void (*free)(struct lxc_snapshot *);
 };
 
 struct lxc_container *lxc_container_new(const char *name, const char 
*configpath);
diff --git a/src/tests/Makefile.am b/src/tests/Makefile.am
index 76b38f9..a6dacab 100644
--- a/src/tests/Makefile.am
+++ b/src/tests/Makefile.am
@@ -17,6 +17,7 @@ lxc_test_clonetest_SOURCES = clonetest.c
 lxc_test_console_SOURCES = console.c
 lxc_usernic_test_SOURCES = ../lxc/lxc_user_nic.c ../lxc/nl.c
 lxc_usernic_test_CFLAGS = -DISTEST
+lxc_test_snapshot_SOURCES = snapshot.c
 
 AM_CFLAGS=-I$(top_srcdir)/src \
        -DLXCROOTFSMOUNT=\"$(LXCROOTFSMOUNT)\" \
@@ -28,7 +29,8 @@ AM_CFLAGS=-I$(top_srcdir)/src \
 bin_PROGRAMS = lxc-test-containertests lxc-test-locktests lxc-test-startone \
        lxc-test-destroytest lxc-test-saveconfig lxc-test-createtest \
        lxc-test-shutdowntest lxc-test-get_item lxc-test-getkeys 
lxc-test-lxcpath \
-       lxc-test-cgpath lxc-test-clonetest lxc-test-console lxc-usernic-test
+       lxc-test-cgpath lxc-test-clonetest lxc-test-console lxc-usernic-test \
+       lxc-test-snapshot
 
 bin_SCRIPTS = lxc-test-usernic
 
@@ -48,4 +50,5 @@ EXTRA_DIST = \
        clonetest.c \
        startone.c \
        console.c \
-       lxc-test-usernic
+       lxc-test-usernic \
+       snapshot.c
diff --git a/src/tests/snapshot.c b/src/tests/snapshot.c
new file mode 100644
index 0000000..7298e53
--- /dev/null
+++ b/src/tests/snapshot.c
@@ -0,0 +1,130 @@
+/* liblxcapi
+ *
+ * Copyright © 2013 Serge Hallyn <serge.hal...@ubuntu.com>.
+ * Copyright © 2013 Canonical Ltd.
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2, as
+ * published by the Free Software Foundation.
+ *
+ * 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.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+ */
+#include "../lxc/lxccontainer.h"
+
+#include <errno.h>
+#include <stdlib.h>
+#include "../lxc/lxc.h"
+
+#define MYNAME "snapxxx1"
+#define RESTNAME "snapxxx2"
+
+void try_to_remove()
+{
+       struct lxc_container *c;
+       char snappath[1024];
+       c = lxc_container_new(RESTNAME, NULL);
+       if (c) {
+               if (c->is_defined(c))
+                       c->destroy(c);
+               lxc_container_put(c);
+       }
+       snprintf(snappath, 1024, "%ssnaps/%s", lxc_get_default_config_path(), 
MYNAME);
+       c = lxc_container_new("snap0", snappath);
+       if (c) {
+               if (c->is_defined(c))
+                       c->destroy(c);
+               lxc_container_put(c);
+       }
+       c = lxc_container_new(MYNAME, NULL);
+       if (c) {
+               if (c->is_defined(c))
+                       c->destroy(c);
+               lxc_container_put(c);
+       }
+}
+
+int main(int argc, char *argv[])
+{
+       struct lxc_container *c;
+       char *template = "busybox";
+
+       if (argc > 1)
+               template = argv[1];
+
+       try_to_remove();
+       c = lxc_container_new(MYNAME, NULL);
+       if (!c) {
+               fprintf(stderr, "%s: %d: failed to load first container\n", 
__FILE__, __LINE__);
+               exit(1);
+       }
+
+       if (c->is_defined(c)) {
+               fprintf(stderr, "%d: %s thought it was defined\n", __LINE__, 
MYNAME);
+               (void) c->destroy(c);
+       }
+       if (!c->set_config_item(c, "lxc.network.type", "empty")) {
+               fprintf(stderr, "%s: %d: failed to set network type\n", 
__FILE__, __LINE__);
+               goto err;
+       }
+       c->save_config(c, NULL);
+       if (!c->createl(c, template, NULL, NULL, 0, NULL)) {
+               fprintf(stderr, "%s: %d: failed to create %s container\n", 
__FILE__, __LINE__, template);
+               goto err;
+       }
+       c->load_config(c, NULL);
+
+       if (c->snapshot(c, NULL) != 0) {
+               fprintf(stderr, "%s: %d: failed to create snapsot\n", __FILE__, 
__LINE__);
+               goto err;
+       }
+
+       // rootfs should be ${lxcpath}snaps/${lxcname}/snap0/rootfs
+       struct stat sb;
+       int ret;
+       char path[1024];
+       snprintf(path, 1024, "%ssnaps/%s/snap0/rootfs", 
lxc_get_default_config_path(), MYNAME);
+       ret = stat(path, &sb);
+       if (ret != 0) {
+               fprintf(stderr, "%s: %d: snapshot was not actually created\n", 
__FILE__, __LINE__);
+               goto err;
+       }
+
+       struct lxc_snapshot *s;
+       int i, n;
+
+       n = c->snapshot_list(c, &s);
+       if (n < 1) {
+               fprintf(stderr, "%s: %d: failed listing containers\n", 
__FILE__, __LINE__);
+               goto err;
+       }
+       if (strcmp(s->name, "snap0") != 0) {
+               fprintf(stderr, "%s: %d: snapshot had bad name\n", __FILE__, 
__LINE__);
+               goto err;
+       }
+       for (i=0; i<n; i++) {
+               s[i].free(&s[i]);
+       }
+       free(s);
+
+       if (!c->snapshot_restore(c, "snap0", RESTNAME)) {
+               fprintf(stderr, "%s: %d: failed to restore snapshot\n", 
__FILE__, __LINE__);
+               goto err;
+       }
+
+       printf("All tests passed\n");
+       lxc_container_put(c);
+       exit(0);
+
+err:
+       lxc_container_put(c);
+       fprintf(stderr, "Exiting on error\n");
+       try_to_remove();
+       exit(1);
+}
-- 
1.8.3.2


------------------------------------------------------------------------------
Learn the latest--Visual Studio 2012, SharePoint 2013, SQL 2012, more!
Discover the easy way to master current and previous Microsoft technologies
and advance your career. Get an incredible 1,500+ hours of step-by-step
tutorial videos with LearnDevNow. Subscribe today and save!
http://pubads.g.doubleclick.net/gampad/clk?id=58041391&iu=/4140/ostg.clktrk
_______________________________________________
Lxc-devel mailing list
Lxc-devel@lists.sourceforge.net
https://lists.sourceforge.net/lists/listinfo/lxc-devel

Reply via email to