1 files changed, 73 insertions, 0 deletions
diff --git a/sd_login/type_machine.go b/sd_login/type_machine.go
new file mode 100644
index 0000000..5892017
--- /dev/null
+++ b/sd_login/type_machine.go
@@ -0,0 +1,73 @@
+// Copyright (C) 2016 Luke Shumaker <lukeshu@sbcglobal.net>
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package sd_login
+
+import (
+	"strconv"
+	"strings"
+
+	"golang.org/x/sys/unix"
+)
+
+type Machine string
+
+// running VMs/containers
+func GetMachines() ([]Machine, error) {
+	strs, err := get_files_in_directory("/run/systemd/machines/")
+	if err != nil {
+		return nil, err
+	}
+	var machines []Machine
+	for _, str := range strs {
+		if strings.HasPrefix(str, "unit:") || !valid_machine_name(str) {
+			continue
+		}
+		machines = append(machines, Machine(str))
+	}
+	return machines, nil
+}
+
+func (m Machine) GetClass() (string, error) {
+	env, err := parse_env_file("/run/systemd/machines/" + string(m))
+	if err != nil {
+		return "", err
+	}
+	class, ok := env["CLASS"]
+	if !ok {
+		return "", unix.ENXIO // or EIO?
+	}
+	return class, nil
+}
+
+func (m Machine) GetIfIndices() ([]int, error) {
+	env, err := parse_env_file("/run/systemd/machines/" + string(m))
+	if err != nil {
+		return nil, err
+	}
+	netif, ok := env["NETIF"]
+	if !ok {
+		return nil, unix.ENXIO // or EIO?
+	}
+
+	var ifindices []int
+	for _, word := range strings.FieldsFunc(netif, func(c rune) bool { return strings.ContainsRune(whitespace, c) }) {
+		ifindex, err := strconv.Atoi(word)
+		if err != nil {
+			return nil, err
+		}
+		ifindices = append(ifindices, ifindex)
+	}
+	return ifindices, nil
+}