overlord/snapshotstate/backend: introducing the snapshot backend

dirs/dirs.go

@@ -103,6 +103,7 @@ var (
 	SystemFontconfigCacheDir string
 
 	FreezerCgroupDir string
+	SnapshotsDir     string
 )
 
 const (
@@ -259,4 +260,5 @@ func SetRootDir(rootdir string) {
 	SystemFontconfigCacheDir = filepath.Join(rootdir, "/var/cache/fontconfig")
 
 	FreezerCgroupDir = filepath.Join(rootdir, "/sys/fs/cgroup/freezer/")
+	SnapshotsDir = filepath.Join(rootdir, "/var/spool/snapd/snapshots")
 }

overlord/snapshotstate/backend/backend.go

@@ -0,0 +1,256 @@
+// -*- Mode: Go; indent-tabs-mode: t -*-
+
+/*
+ * Copyright (C) 2018 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 3 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, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+package backend
+
+import (
+	"archive/zip"
+	"crypto"
+	"encoding/json"
+	"fmt"
+	"io"
+	"os"
+	"os/exec"
+	"path/filepath"
+	"regexp"
+	"sort"
+	"time"
+
+	"golang.org/x/net/context"
+
+	"github.com/snapcore/snapd/client"
+	"github.com/snapcore/snapd/dirs"
+	"github.com/snapcore/snapd/logger"
+	"github.com/snapcore/snapd/osutil"
+	"github.com/snapcore/snapd/snap"
+	"github.com/snapcore/snapd/strutil"
+)
+
+const (
+	archiveName  = "archive.tgz"
+	metadataName = "meta.json"
+	metaHashName = "meta.sha3_384"
+
+	userArchivePrefix = "user/"
+	userArchiveSuffix = ".tgz"
+)
+
+var (
+	dirOpen  = os.Open
+	dirNames = (*os.File).Readdirnames
+	open     = Open
+)
+
+// Iter loops over all snapshots in the snapshots directory, applying the
+// given function to each. The snapshot will be closed after the function
+// returns. If the function returns error, iteration is stopped (and if the
+// error isn't EOF, it's returned as the error of the iterator).
+func Iter(ctx context.Context, f func(*Reader) error) error {
+	if err := ctx.Err(); err != nil {
+		return err
+	}
+
+	dir, err := dirOpen(dirs.SnapshotsDir)
+	if err != nil {
+		if osutil.IsDirNotExist(err) {
+			// no dir -> no snapshots
+			return nil
+		}
+		return fmt.Errorf("cannot open snapshots directory: %v", err)
+	}
+	defer dir.Close()
+
+	var names []string
+	for err == nil {
+		names, err = dirNames(dir, 100)
+		for _, name := range names {
+			if err = ctx.Err(); err != nil {
+				break
+			}
+
+			filename := filepath.Join(dirs.SnapshotsDir, name)
+			rsh, openError := open(filename)
+			// rsh can be non-nil even when openError is not nil (in
+			// which case rsh.Broken will have a reason). f can
+			// check and either ignore or return an error when
+			// finding a broken snapshot.
+			if rsh != nil {
+				err = f(rsh)
+			} else {
+				// TODO: use warnings instead
+				logger.Noticef("cannot open snapshot %q: %v", name, openError)
+			}
+			if openError == nil {
+				// if openError was nil the snapshot was opened and needs closing
+				if closeError := rsh.Close(); err == nil {
+					err = closeError
+				}
+			}
+			if err != nil {
+				break
+			}
+		}
+	}
+
+	if err == io.EOF {
+		err = nil
+	}
+
+	return err
+}
+
+// List valid snapshots sets.
+func List(ctx context.Context, setID uint64, snapNames []string) ([]client.SnapshotSet, error) {
+	setshots := map[uint64][]*client.Snapshot{}
+	err := Iter(ctx, func(sh *Reader) error {
+		if setID == 0 || sh.SetID == setID {
+			if len(snapNames) == 0 || strutil.ListContains(snapNames, sh.Snap) {
+				setshots[sh.SetID] = append(setshots[sh.SetID], &sh.Snapshot)
+			}
+		}
+		return nil
+	})
+
+	sets := make([]client.SnapshotSet, 0, len(setshots))
+	for id, shots := range setshots {
+		sort.Sort(bySnap(shots))
+		sets = append(sets, client.SnapshotSet{ID: id, Snapshots: shots})
+	}
+
+	sort.Sort(byID(sets))
+
+	return sets, err
+}
+
+// Filename of the given client.Snapshot in this backend.
+func Filename(sh *client.Snapshot) string {
+	// this _needs_ the snap name and version to be valid
+	return filepath.Join(dirs.SnapshotsDir, fmt.Sprintf("%d_%s_%s.zip", sh.SetID, sh.Snap, sh.Version))
+}
+
+// Save a snapshot
+func Save(ctx context.Context, id uint64, si *snap.Info, cfg map[string]interface{}, usernames []string) (sh *client.Snapshot, e error) {
+	if err := os.MkdirAll(dirs.SnapshotsDir, 0700); err != nil {
+		return nil, err
+	}
+
+	sh = &client.Snapshot{
+		SetID:    id,
+		Snap:     si.Name(),
+		Revision: si.Revision,
+		Version:  si.Version,
+		Time:     time.Now(),
+		SHA3_384: make(map[string]string),
+		Size:     0,
+		Conf:     cfg,
+	}
+
+	aw, err := osutil.NewAtomicFile(Filename(sh), 0600, 0, osutil.NoChown, osutil.NoChown)
+	if err != nil {
+		return nil, err
+	}
+	// if things worked, we'll commit (and Cancel becomes a NOP)
+	defer aw.Cancel()
+
+	w := zip.NewWriter(aw)
+	// Note we don't “defer w.Close()” because Close on zip.Writers:
+	// 1. doesn't close the file descriptor (that's done by Cancel, above)
+	// 2. writes out the central directory of the zip
+	// (yes they're weird in this way: why is that called Close, even)
+	if err := addToZip(ctx, sh, w, archiveName, si.DataDir()); err != nil {
+		return nil, err
+	}
+
+	users, err := usersForUsernames(usernames)
+	if err != nil {
+		return nil, err
+	}
+
+	for _, usr := range users {
+		if err := addToZip(ctx, sh, w, userArchiveName(usr), si.UserDataDir(usr.HomeDir)); err != nil {
+			return nil, err
+		}
+	}
+
+	metaWriter, err := w.Create(metadataName)
+	if err != nil {
+		return nil, err
+	}
+
+	hasher := crypto.SHA3_384.New()
+	enc := json.NewEncoder(io.MultiWriter(metaWriter, hasher))
+	if err := enc.Encode(sh); err != nil {
+		return nil, err
+	}
+
+	hashWriter, err := w.Create(metaHashName)
+	if err != nil {
+		return nil, err
+	}
+	fmt.Fprintf(hashWriter, "%x\n", hasher.Sum(nil))
+	if err := w.Close(); err != nil {
+		return nil, err
+	}
+
+	if err := ctx.Err(); err != nil {
+		return nil, err
+	}
+
+	if err := aw.Commit(); err != nil {
+		return nil, err
+	}
+
+	return sh, nil
+}
+
+func addToZip(ctx context.Context, sh *client.Snapshot, w *zip.Writer, entry, dir string) error {
+	hasher := crypto.SHA3_384.New()
+	if exists, isDir, err := osutil.DirExists(dir); !exists || !isDir || err != nil {
+		return err
+	}
+	parent, dir := filepath.Split(dir)
+
+	archiveWriter, err := w.CreateHeader(&zip.FileHeader{Name: entry})
+	if err != nil {
+		return err
+	}
+
+	var sz sizer
+
+	cmd := exec.Command("tar",
+		"--create",
+		"--sparse", "--gzip",
+		"--directory", parent, dir, "common")
+	cmd.Env = []string{"GZIP=-9 -n"}
+	cmd.Stdout = io.MultiWriter(archiveWriter, hasher, &sz)
+	matchCounter := &strutil.MatchCounter{Regexp: regexp.MustCompile(".*"), N: 1}
+	cmd.Stderr = matchCounter
+	if err := osutil.RunWithContext(ctx, cmd); err != nil {
+		matches, count := matchCounter.Matches()
+		if count > 0 {
+			return fmt.Errorf("cannot create archive: %s (and %d more)", matches[0], count-1)
+		}
+		return fmt.Errorf("tar failed: %v", err)
+	}
+
+	sh.SHA3_384[entry] = fmt.Sprintf("%x", hasher.Sum(nil))
+	sh.Size += sz.size
+
+	return nil
+}