From ce7fce49f5c1dbd23ab834b7739ded47a4c7a4e9 Mon Sep 17 00:00:00 2001
From: Mikhail Klementev <jollheef@riseup.net>
Date: Sat, 17 Nov 2018 20:20:59 +0000
Subject: [PATCH] Add config submodule

---
 config/config.go | 174 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 174 insertions(+)
 create mode 100644 config/config.go

diff --git a/config/config.go b/config/config.go
new file mode 100644
index 0000000..6d33728
--- /dev/null
+++ b/config/config.go
@@ -0,0 +1,174 @@
+// Copyright 2018 Mikhail Klementev. All rights reserved.
+// Use of this source code is governed by a AGPLv3 license
+// (or later) that can be found in the LICENSE file.
+
+package config
+
+import (
+	"errors"
+	"fmt"
+	"io/ioutil"
+	"os"
+	"regexp"
+	"strings"
+
+	"github.com/naoina/toml"
+)
+
+type KernelMask struct {
+	DistroType    DistroType
+	DistroRelease string // 18.04/7.4.1708/9.1
+	ReleaseMask   string
+}
+
+type ArtifactType int
+
+const (
+	KernelModule ArtifactType = iota
+	KernelExploit
+)
+
+func (at ArtifactType) String() string {
+	return [...]string{"module", "exploit"}[at]
+}
+
+func (at *ArtifactType) UnmarshalTOML(data []byte) (err error) {
+	stype := strings.Trim(string(data), `"`)
+	stypelower := strings.ToLower(stype)
+	if strings.Contains(stypelower, "module") {
+		*at = KernelModule
+	} else if strings.Contains(stypelower, "exploit") {
+		*at = KernelExploit
+	} else {
+		err = errors.New(fmt.Sprintf("Type %s is unsupported", stype))
+	}
+	return
+}
+
+type Artifact struct {
+	Name             string
+	Type             ArtifactType
+	SourcePath       string
+	SupportedKernels []KernelMask
+}
+
+func (ka Artifact) checkSupport(ki KernelInfo, km KernelMask) (
+	supported bool, err error) {
+
+	if ki.DistroType != km.DistroType {
+		supported = false
+		return
+	}
+
+	// DistroRelease is optional
+	if km.DistroRelease != "" && ki.DistroRelease != km.DistroRelease {
+		supported = false
+		return
+	}
+
+	supported, err = regexp.MatchString(km.ReleaseMask, ki.KernelRelease)
+	return
+}
+
+func (ka Artifact) Supported(ki KernelInfo) (supported bool, err error) {
+	for _, km := range ka.SupportedKernels {
+		supported, err = ka.checkSupport(ki, km)
+		if supported {
+			break
+		}
+
+	}
+	return
+}
+
+type DistroType int
+
+const (
+	Ubuntu DistroType = iota
+	CentOS
+	Debian
+)
+
+var DistroTypeStrings = [...]string{"Ubuntu", "CentOS", "Debian"}
+
+func NewDistroType(dType string) (dt DistroType, err error) {
+	err = dt.UnmarshalTOML([]byte(dType))
+	return
+}
+
+func (dt DistroType) String() string {
+	return DistroTypeStrings[dt]
+}
+
+func (dt *DistroType) UnmarshalTOML(data []byte) (err error) {
+	sDistro := strings.Trim(string(data), `"`)
+	if strings.EqualFold(sDistro, "Ubuntu") {
+		*dt = Ubuntu
+	} else if strings.EqualFold(sDistro, "CentOS") {
+		*dt = CentOS
+	} else if strings.EqualFold(sDistro, "Debian") {
+		*dt = Debian
+	} else {
+		err = errors.New(fmt.Sprintf("Distro %s is unsupported", sDistro))
+	}
+	return
+}
+
+type KernelInfo struct {
+	DistroType    DistroType
+	DistroRelease string // 18.04/7.4.1708/9.1
+
+	// Must be *exactly* same as in `uname -r`
+	KernelRelease string
+
+	// Build-time information
+	ContainerName string
+
+	// Runtime information
+	KernelPath string
+	InitrdPath string
+	RootFS     string
+}
+
+type KernelConfig struct {
+	Kernels []KernelInfo
+}
+
+func readFileAll(path string) (buf []byte, err error) {
+	f, err := os.Open(path)
+	if err != nil {
+		return
+	}
+	defer f.Close()
+
+	buf, err = ioutil.ReadAll(f)
+	return
+}
+
+func ReadKernelConfig(path string) (kernelCfg KernelConfig, err error) {
+	buf, err := readFileAll(path)
+	if err != nil {
+		return
+	}
+
+	err = toml.Unmarshal(buf, &kernelCfg)
+	if err != nil {
+		return
+	}
+
+	return
+}
+
+func ReadArtifactConfig(path string) (artifactCfg Artifact, err error) {
+	buf, err := readFileAll(path)
+	if err != nil {
+		return
+	}
+
+	err = toml.Unmarshal(buf, &artifactCfg)
+	if err != nil {
+		return
+	}
+
+	return
+}