summaryrefslogtreecommitdiff
path: root/vmdb/plugins/virtualfs_plugin.py
blob: 9f440f827d38691da723d26478b4dda674979271 (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
# Copyright 2017  Lars Wirzenius
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# 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/>.
#
# =*= License: GPL-3+ =*=


import logging
import os

import vmdb


class VirtualFilesystemMountPlugin(vmdb.Plugin):
    def enable(self):
        self.app.step_runners.add(VirtualFilesystemMountStepRunner())


class VirtualFilesystemMountStepRunner(vmdb.StepRunnerInterface):

    virtuals = [
        ["none", "/proc", "proc"],
        ["none", "/dev", "devtmpfs"],
        ["none", "/dev/pts", "devpts"],
        ["none", "/dev/shm", "tmpfs"],
        ["none", "/run", "tmpfs"],
        ["none", "/run/lock", "tmpfs"],
        ["none", "/sys", "sysfs"],
    ]

    def get_key_spec(self):
        return {"virtual-filesystems": str}

    def run(self, values, settings, state):
        fstag = values["virtual-filesystems"]
        mount_point = state.tags.get_builder_mount_point(fstag)
        self.mount_virtuals(mount_point, state)

    def teardown(self, values, settings, state):
        self.unmount_virtuals(state)

    def mount_virtuals(self, rootfs, state):
        if not hasattr(state, "virtuals"):
            state.virtuals = []

        for device, mount_point, fstype in self.virtuals:
            path = os.path.join(rootfs, "./" + mount_point)
            if not os.path.exists(path):
                os.mkdir(path)
            vmdb.runcmd(["mount", "-t", fstype, device, path])
            state.virtuals.append(path)
        logging.debug("mounted virtuals: %r", state.virtuals)

    def unmount_virtuals(self, state):
        logging.debug("unmounting virtuals: %r", state.virtuals)
        for mount_point in reversed(state.virtuals):
            try:
                vmdb.unmount(mount_point)
            except vmdb.NotMounted as e:
                logging.warning(str(e))