f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn/* liblxcapi
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn *
8cd80b50efe2107ac351bfd0285050dd183398e7Stéphane Graber * Copyright © 2013 Serge Hallyn <serge.hallyn@ubuntu.com>.
8cd80b50efe2107ac351bfd0285050dd183398e7Stéphane Graber * Copyright © 2013 Canonical Ltd.
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn *
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * This program is free software; you can redistribute it and/or modify
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * it under the terms of the GNU General Public License version 2, as
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * published by the Free Software Foundation.
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn *
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * This program is distributed in the hope that it will be useful,
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * but WITHOUT ANY WARRANTY; without even the implied warranty of
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * GNU General Public License for more details.
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn *
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * You should have received a copy of the GNU General Public License along
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * with this program; if not, write to the Free Software Foundation, Inc.,
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn */
948955a2d6f8e4e28bceada9666c5831de4a6bb8Stéphane Graber#include <lxc/lxccontainer.h>
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn#include <errno.h>
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn#include <stdlib.h>
4012c89148c42dc7c15809bec348746681f96eddStéphane Graber#include <stdio.h>
95ee490bbdb97ab2b4f1dfa63a0a26e0dd1c2f17Stéphane Graber#include <string.h>
95ee490bbdb97ab2b4f1dfa63a0a26e0dd1c2f17Stéphane Graber#include <sys/stat.h>
f2363e38d07db2ad4796b5580b95e3665ed34a88S.Çağlar Onur#include "lxc/lxc.h"
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn#define MYNAME "snapxxx1"
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn#define MYNAME2 "snapxxx3"
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn#define RESTNAME "snapxxx2"
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
0b98289ef95e83fcd493ba6688e57aa8b4fb53f7Andrey Mazostatic void try_to_remove(void)
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn{
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn struct lxc_container *c;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn c = lxc_container_new(RESTNAME, NULL);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (c) {
18aa217bb187b551206fe66075229ba382a1dbe4Serge Hallyn c->snapshot_destroy_all(c);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (c->is_defined(c))
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn c->destroy(c);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn lxc_container_put(c);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn c = lxc_container_new(MYNAME2, NULL);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn if (c) {
18aa217bb187b551206fe66075229ba382a1dbe4Serge Hallyn c->destroy_with_snapshots(c);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn lxc_container_put(c);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn c = lxc_container_new(MYNAME, NULL);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (c) {
18aa217bb187b551206fe66075229ba382a1dbe4Serge Hallyn c->snapshot_destroy_all(c);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (c->is_defined(c))
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn c->destroy(c);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn lxc_container_put(c);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn}
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallynint main(int argc, char *argv[])
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn{
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn struct lxc_container *c, *c2 = NULL;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn char *template = "busybox";
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (argc > 1)
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn template = argv[1];
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn try_to_remove();
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn c = lxc_container_new(MYNAME, NULL);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (!c) {
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn fprintf(stderr, "%s: %d: failed to load first container\n", __FILE__, __LINE__);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn exit(1);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (c->is_defined(c)) {
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn fprintf(stderr, "%d: %s thought it was defined\n", __LINE__, MYNAME);
18aa217bb187b551206fe66075229ba382a1dbe4Serge Hallyn (void) c->destroy_with_snapshots(c);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (!c->set_config_item(c, "lxc.network.type", "empty")) {
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn fprintf(stderr, "%s: %d: failed to set network type\n", __FILE__, __LINE__);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn goto err;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn c->save_config(c, NULL);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (!c->createl(c, template, NULL, NULL, 0, NULL)) {
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn fprintf(stderr, "%s: %d: failed to create %s container\n", __FILE__, __LINE__, template);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn goto err;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn c->load_config(c, NULL);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (c->snapshot(c, NULL) != 0) {
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn fprintf(stderr, "%s: %d: failed to create snapshot\n", __FILE__, __LINE__);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn goto err;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
18aa217bb187b551206fe66075229ba382a1dbe4Serge Hallyn // rootfs should be ${lxcpath}${lxcname}/snaps/snap0/rootfs
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn struct stat sb;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn int ret;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn char path[1024];
18aa217bb187b551206fe66075229ba382a1dbe4Serge Hallyn snprintf(path, 1024, "%s/%s/snaps/snap0/rootfs", lxc_get_global_config_item("lxc.lxcpath"), MYNAME);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn ret = stat(path, &sb);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (ret != 0) {
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn fprintf(stderr, "%s: %d: snapshot was not actually created\n", __FILE__, __LINE__);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn goto err;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn struct lxc_snapshot *s;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn int i, n;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn n = c->snapshot_list(c, &s);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (n < 1) {
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn fprintf(stderr, "%s: %d: failed listing containers\n", __FILE__, __LINE__);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn goto err;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (strcmp(s->name, "snap0") != 0) {
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn fprintf(stderr, "%s: %d: snapshot had bad name\n", __FILE__, __LINE__);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn goto err;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn for (i=0; i<n; i++) {
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn s[i].free(&s[i]);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn free(s);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn if (!c->snapshot_restore(c, "snap0", RESTNAME)) {
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn fprintf(stderr, "%s: %d: failed to restore snapshot\n", __FILE__, __LINE__);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn goto err;
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn }
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur if (!c->snapshot_destroy(c, "snap0")) {
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur fprintf(stderr, "%s: %d: failed to destroy snapshot\n", __FILE__, __LINE__);
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur goto err;
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur }
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur
e60e630c4a1e7179a37d3edd7925b59affb45785Serge Hallyn c2 = lxc_container_new(RESTNAME, NULL);
e60e630c4a1e7179a37d3edd7925b59affb45785Serge Hallyn if (!c2 || !c2->is_defined(c2)) {
e60e630c4a1e7179a37d3edd7925b59affb45785Serge Hallyn fprintf(stderr, "%s: %d: external snapshot restore failed\n", __FILE__, __LINE__);
e60e630c4a1e7179a37d3edd7925b59affb45785Serge Hallyn goto err;
e60e630c4a1e7179a37d3edd7925b59affb45785Serge Hallyn }
e60e630c4a1e7179a37d3edd7925b59affb45785Serge Hallyn lxc_container_put(c2);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn c2 = c->clone(c, MYNAME2, NULL, LXC_CLONE_SNAPSHOT, "overlayfs", NULL, 0, NULL);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn if (!c2) {
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn fprintf(stderr, "%d: %s overlayfs clone failed\n", __LINE__, MYNAME2);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn goto good;
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn }
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn if (c2->snapshot(c2, NULL) != 0) {
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn fprintf(stderr, "%s: %d: failed to create snapshot\n", __FILE__, __LINE__);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn goto err;
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn }
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn n = c2->snapshot_list(c2, &s);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn if (n < 1) {
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn fprintf(stderr, "%s: %d: failed listing containers\n", __FILE__, __LINE__);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn goto err;
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn }
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn if (strcmp(s->name, "snap0") != 0) {
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn fprintf(stderr, "%s: %d: snapshot had bad name\n", __FILE__, __LINE__);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn goto err;
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn }
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn for (i=0; i<n; i++) {
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn s[i].free(&s[i]);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn }
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn free(s);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn if (!c2->snapshot_restore(c2, "snap0", NULL)) {
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn fprintf(stderr, "%s: %d: failed to restore overlayfs snapshot\n", __FILE__, __LINE__);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn goto err;
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn }
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn if (!c2->snapshot_destroy(c2, "snap0")) {
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn fprintf(stderr, "%s: %d: failed to destroy overlayfs snapshot\n", __FILE__, __LINE__);
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn goto err;
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn }
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyn
de269ee8dec1cf2d882de385f8dd6591e97121e2Serge Hallyngood:
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn lxc_container_put(c);
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur try_to_remove();
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur printf("All tests passed\n");
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur exit(0);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallynerr:
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn lxc_container_put(c);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn try_to_remove();
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur
58b61f6d3442e984a88c093dbdfd20f1d0a82587S.Çağlar Onur fprintf(stderr, "Exiting on error\n");
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn exit(1);
f5dd1d532a76a1b57cf341db821eae85ea1118c5Serge Hallyn}