summaryrefslogtreecommitdiff
path: root/sd_login/type_machine.go
blob: 5892017febe70dd039ff133a6dbd996ab6b5f3ec (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
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
}