155 lines
5.4 KiB
Python
155 lines
5.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Based on the buildah connection plugin
|
|
# Copyright (c) 2017 Ansible Project
|
|
# 2018 Kushal Das
|
|
# GNU General Public License v3.0+ (see COPYING or https://www.gnu.org/licenses/gpl-3.0.txt)
|
|
#
|
|
#
|
|
# Written by: Kushal Das (https://github.com/kushaldas)
|
|
|
|
from __future__ import (absolute_import, division, print_function)
|
|
__metaclass__ = type
|
|
|
|
|
|
DOCUMENTATION = '''
|
|
name: qubes
|
|
short_description: Interact with an existing QubesOS AppVM
|
|
|
|
description:
|
|
- Run commands or put/fetch files to an existing Qubes AppVM using qubes tools.
|
|
|
|
author: Kushal Das (@kushaldas)
|
|
|
|
|
|
options:
|
|
remote_addr:
|
|
description:
|
|
- vm name
|
|
default: inventory_hostname
|
|
vars:
|
|
- name: ansible_host
|
|
remote_user:
|
|
description:
|
|
- The user to execute as inside the vm.
|
|
default: The *user* account as default in Qubes OS.
|
|
vars:
|
|
- name: ansible_user
|
|
# keyword:
|
|
# - name: hosts
|
|
'''
|
|
|
|
import subprocess
|
|
|
|
from ansible.module_utils.common.text.converters import to_bytes
|
|
from ansible.plugins.connection import ConnectionBase, ensure_connect
|
|
from ansible.errors import AnsibleConnectionFailure
|
|
from ansible.utils.display import Display
|
|
|
|
display = Display()
|
|
|
|
|
|
# this _has to be_ named Connection
|
|
class Connection(ConnectionBase):
|
|
"""This is a connection plugin for qubes: it uses qubes-run-vm binary to interact with the containers."""
|
|
|
|
# String used to identify this Connection class from other classes
|
|
transport = 'community.general.qubes'
|
|
has_pipelining = True
|
|
|
|
def __init__(self, play_context, new_stdin, *args, **kwargs):
|
|
super(Connection, self).__init__(play_context, new_stdin, *args, **kwargs)
|
|
|
|
self._remote_vmname = self._play_context.remote_addr
|
|
self._connected = False
|
|
# Default username in Qubes
|
|
self.user = "user"
|
|
if self._play_context.remote_user:
|
|
self.user = self._play_context.remote_user
|
|
|
|
def _qubes(self, cmd=None, in_data=None, shell="qubes.VMShell"):
|
|
"""run qvm-run executable
|
|
|
|
:param cmd: cmd string for remote system
|
|
:param in_data: data passed to qvm-run-vm's stdin
|
|
:return: return code, stdout, stderr
|
|
"""
|
|
display.vvvv("CMD: ", cmd)
|
|
if not cmd.endswith("\n"):
|
|
cmd = cmd + "\n"
|
|
local_cmd = []
|
|
|
|
# For dom0
|
|
local_cmd.extend(["qvm-run", "--pass-io", "--service"])
|
|
if self.user != "user":
|
|
# Means we have a remote_user value
|
|
local_cmd.extend(["-u", self.user])
|
|
|
|
local_cmd.append(self._remote_vmname)
|
|
|
|
local_cmd.append(shell)
|
|
|
|
local_cmd = [to_bytes(i, errors='surrogate_or_strict') for i in local_cmd]
|
|
|
|
display.vvvv("Local cmd: ", local_cmd)
|
|
|
|
display.vvv("RUN %s" % (local_cmd,), host=self._remote_vmname)
|
|
p = subprocess.Popen(local_cmd, shell=False, stdin=subprocess.PIPE,
|
|
stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
|
|
|
# Here we are writing the actual command to the remote bash
|
|
p.stdin.write(to_bytes(cmd, errors='surrogate_or_strict'))
|
|
stdout, stderr = p.communicate(input=in_data)
|
|
return p.returncode, stdout, stderr
|
|
|
|
def _connect(self):
|
|
"""No persistent connection is being maintained."""
|
|
super(Connection, self)._connect()
|
|
self._connected = True
|
|
|
|
@ensure_connect
|
|
def exec_command(self, cmd, in_data=None, sudoable=False):
|
|
"""Run specified command in a running QubesVM """
|
|
super(Connection, self).exec_command(cmd, in_data=in_data, sudoable=sudoable)
|
|
|
|
display.vvvv("CMD IS: %s" % cmd)
|
|
|
|
rc, stdout, stderr = self._qubes(cmd)
|
|
|
|
display.vvvvv("STDOUT %r STDERR %r" % (stderr, stderr))
|
|
return rc, stdout, stderr
|
|
|
|
def put_file(self, in_path, out_path):
|
|
""" Place a local file located in 'in_path' inside VM at 'out_path' """
|
|
super(Connection, self).put_file(in_path, out_path)
|
|
display.vvv("PUT %s TO %s" % (in_path, out_path), host=self._remote_vmname)
|
|
|
|
with open(in_path, "rb") as fobj:
|
|
source_data = fobj.read()
|
|
|
|
retcode, dummy, dummy = self._qubes('cat > "{0}"\n'.format(out_path), source_data, "qubes.VMRootShell")
|
|
# if qubes.VMRootShell service not supported, fallback to qubes.VMShell and
|
|
# hope it will have appropriate permissions
|
|
if retcode == 127:
|
|
retcode, dummy, dummy = self._qubes('cat > "{0}"\n'.format(out_path), source_data)
|
|
|
|
if retcode != 0:
|
|
raise AnsibleConnectionFailure('Failed to put_file to {0}'.format(out_path))
|
|
|
|
def fetch_file(self, in_path, out_path):
|
|
"""Obtain file specified via 'in_path' from the container and place it at 'out_path' """
|
|
super(Connection, self).fetch_file(in_path, out_path)
|
|
display.vvv("FETCH %s TO %s" % (in_path, out_path), host=self._remote_vmname)
|
|
|
|
# We are running in dom0
|
|
cmd_args_list = ["qvm-run", "--pass-io", self._remote_vmname, "cat {0}".format(in_path)]
|
|
with open(out_path, "wb") as fobj:
|
|
p = subprocess.Popen(cmd_args_list, shell=False, stdout=fobj)
|
|
p.communicate()
|
|
if p.returncode != 0:
|
|
raise AnsibleConnectionFailure('Failed to fetch file to {0}'.format(out_path))
|
|
|
|
def close(self):
|
|
""" Closing the connection """
|
|
super(Connection, self).close()
|
|
self._connected = False
|