r00t2/vaultpass
Archived
1
0
Fork 0
Code Issues Pull Requests Projects Releases Activity

more testing stuff

Browse Source
This commit is contained in:
brent s. 2020-04-02 07:27:04 -04:00
parent 616374754b
commit 7928a45ef5
Signed by: bts
GPG Key ID: 8C004C2F93481F6B
2 changed files with 159 additions and 52 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
testing/serverconf.py
View File

@ -8,10 +8,10 @@ conf_file = './testserver.json'
log_file = './vault.log' log_file = './vault.log'


default_conf = {'listener': [ default_conf = {'listener': [
{'tcp': '127.0.0.1:8200', {'tcp': {'address': '127.0.0.1:8200',
'tls_disable': True} 'tls_disable': True}}
], ],
'storage': {'file': './data'}, 'storage': {'file': {'path': './data'}},
'log_level': 'Debug', # highest is 'Trace' 'log_level': 'Debug', # highest is 'Trace'
'pid_file': './test.pid', 'pid_file': './test.pid',
'raw_storage_endpoint': True, 'raw_storage_endpoint': True,
@ -29,7 +29,7 @@ def genConf(confdict = None):
storage = confdict.get('storage') storage = confdict.get('storage')
if storage: if storage:
if 'file' in storage.keys(): if 'file' in storage.keys():
storage['file'] = os.path.abspath(os.path.expanduser(storage['file'])) storage['file']['path'] = os.path.abspath(os.path.expanduser(storage['file']['path']))
confdict['storage'] = storage confdict['storage'] = storage
if 'pid_file' in confdict: if 'pid_file' in confdict:
confdict['pid_file'] = os.path.abspath(os.path.expanduser(confdict['pid_file'])) confdict['pid_file'] = os.path.abspath(os.path.expanduser(confdict['pid_file']))

203
testing/spawn.py
View File

@ -3,74 +3,164 @@
import json import json
import os import os
import re import re
import shutil
import socket import socket
import subprocess import subprocess
import time
## ##
import hvac import hvac
import psutil import psutil
from lxml import etree
## ##
from . import serverconf from . import serverconf
from . import vauptpassconf from . import vauptpassconf




clientconf_file = './test.config.xml'


_url_re = re.compile(r'^(?P<proto>https?)://(?P<addr>[^:/]+)(:(?P<port>[0-9]+)?)?(?P<path>/.*)?$') _url_re = re.compile(r'^(?P<proto>https?)://(?P<addr>[^:/]+)(:(?P<port>[0-9]+)?)?(?P<path>/.*)?$')




class VaultSpawner(object): class VaultSpawner(object):
client = hvac.Client() client = hvac.Client()
binary_name = 'vault' binary_name = 'vault' # The name of the process, not a path!
binary_path = binary_name # Can be an absolute path; the binary Vault should be started via.
client_conf = None
cmd = None
unseal = None
is_running = None
local = True
pid = None


def __init__(self, conf, genconf = True, is_dev = False): def __init__(self, conf, genconf = True):
self.conf = conf self.conf = conf
self.genconf = genconf self.genconf = genconf
self.pid = None
self.process = None
self._parseConf() self._parseConf()
self._getCreds()
self._connCheck()


def _getProcess(self): def _connCheck(self, bind = True):
processes = [p for p in psutil.process_iter() if p.name() == self.binary_name] listener = self.conf['listener'][0]['tcp']
if not processes: addr = listener['address']
self.process = None is_not_tls = listener.get('tls_disable', False)
self.pid = None url = '{0}://{1}'.format(('http' if is_not_tls else 'https'), addr)
return(None) if not _url_re.search(url):
raise ValueError('Invalid server address')
self.client.url = url
r = _url_re.search(self.client.url) r = _url_re.search(self.client.url)
if not r: if not r:
raise ValueError('Invalid server URL') raise ValueError('Invalid server URL')
try: try:
port = int(r.groupdict().get('port', 8200)) self.port = int(r.groupdict().get('port', 80))
except ValueError: except ValueError:
port = 8200 self.port = 80
ip = socket.gethostbyname(r.groupdict('addr')) self.ip = socket.gethostbyname(r.groupdict()['addr'])
pids = [] sock = socket.socket()
# First we try the easy way, but requires root privs even if you ran vault as your current user. try:
has_priv = True if bind:
for p in processes: try:
pids.append(p) sock.bind((self.ip, self.port))
self.pid = None
self.is_running = False
except OSError as e:
if e.errno == 98:
self.is_running = True
elif e.errno == 99:
# The address isn't on this box.
self.local = False
self.pid = None
sock.connect((self.ip, self.port))
sock.close()
except (ConnectionRefusedError, ConnectionAbortedError, ConnectionResetError):
self.is_running = False
finally:
try: try:
p_port = p.connections() sock.close()
except (psutil.AccessDenied, psutil.AccessDenied): except NameError:
has_priv = False # No idea how we got here, but...
break pass
if not has_priv: return(None)
conns = [c for c in psutil.net_connections() if c.laddr.port == port and c.laddr.ip == ip]
if not conns: def _getClientConf(self, fpath = clientconf_file):
self.process = None clientconf = os.path.abspath(os.path.expanduser(fpath))
self.pid = None self.client_conf = vauptpassconf.getConfig(clientconf)
return(None) return(None)
for c in conns:
if not c.pid: def _getCreds(self, new_unseal = None, new_auth = None, write_conf = clientconf_file):
continue self._getClientConf()
if c.pid in pids: rewrite_xml = False
self.pid = c.pid # TODO: finish regen of client conf and re-parse so new elements get added to both,
self.process = psutil.Process(pid = self.pid) # and write out namespaced xml
if not all((self.process, self.pid)): unseal_xml = self.client_conf.namespaced_xml.find('.//{0}unseal'.format(self.client_conf.xml.nsmap[None]))
if len(conns) == 1 and len(pids) == 1: self.unseal = unseal_xml.text
self.process = pids[0] auth_xml = self.client_conf.xml.find('.//{0}auth'.format(self.client_conf.xml.nsmap[None]))
self.pid = self.process.pid token_xml = auth_xml.find('.//token')
if unseal_xml is not None and not new_unseal:
self.unseal = unseal_xml.text
if token_xml is not None and not new_auth:
self.client.token = token_xml.text
if new_unseal:
unseal_xml.getparent().replace(unseal_xml, new_unseal)
rewrite_xml = True
if new_auth:
auth_xml.getparent().replace(auth_xml, new_auth)
rewrite_xml = True
if rewrite_xml:
write_conf = os.path.abspath(os.path.expanduser(write_conf))
with open(write_conf, 'w') as fh:
fh.write() # TODO: which object?
return(None)

def _getProcess(self):
def clear():
self.pid = None
return(None)
self._getCreds()
if not self.local:
clear()
return(None)
processes = [p for p in psutil.process_iter() if p.name() == self.binary_name]
if not processes:
clear()
return(None)
if self.is_running:
if len(processes) != 1 and os.geteuid() != 0:
# Vault hides its PID in the network connections/hides its FDs,
# so we have *no* way to get the PID as a regular user.
raise RuntimeError('Cannot determine Vault instance to manage')
elif len(processes) == 1:
self.pid = self.processes[0].pid
else:
# We're running as root.
conns = [c for c in psutil.net_connections() if c.laddr.ip == ip and c.laddr.port == port]
if not len(conns) == 1:
# This, theoretically, should never happen.
raise RuntimeError('Cannot determine Vault instance to manage')
else: else:
raise RuntimeError('Could not reliably determine which Vault instance to manage') self.pid = conns[0].pid

return(None)
return(None)


def _initChk(self):
self._getClientConf()
self._connCheck()
if not self.is_running:
return(False)
if not self.client.sys.is_initialized():
init_rslt = self.client.sys.initialize(secret_shares = 1, secret_threshold = 1)
self.unseal = init_rslt['keys_base64'][0]
self.client.token = init_rslt['root_token']
newauth = etree.Element('auth')
newtoken = etree.Element('token')
newtoken.text = self.client.token
newauth.append(newtoken)
newunseal = etree.Element('unseal')
newunseal.text = self.unseal
self._getCreds(new_auth = newauth, new_unseal = newunseal)
if self.client.sys.is_sealed():
self.client.sys.submit_unseal_key(self.unseal)
return(True)


def _parseConf(self): def _parseConf(self):
is_hcl = False is_hcl = False
@ -95,16 +185,33 @@ class VaultSpawner(object):
self.conf = serverconf.parseHCL(rawconf) self.conf = serverconf.parseHCL(rawconf)
if self.genconf: if self.genconf:
serverconf.genConf(confdict = self.conf) serverconf.genConf(confdict = self.conf)
listener = self.conf['listener'][0]['tcp']
addr = listener['address']
is_tls = listener.get('tls_disable', False)
url = '{0}://{1}'.format(('https' if is_tls else 'http'), addr)
if not _url_re.search(url):
raise ValueError('Invalid server address')
self.client.url = url
return(None) return(None)


def start(self): def start(self):
if any((self.pid, self.process)): self._getProcess()
if self.is_running:
# Already started. # Already started.
self._initChk()
return(None) return(None)
if not self.local:
# It's a remote address
self._initChk()
return(None)
cmd_str = [self.binary_path, 'server']
cmd_str.extend(['-config', serverconf.conf_file])
# We have to use Popen because even vault server doesn't daemonize.
# Gorram it, HashiCorp.
self.cmd = subprocess.Popen(cmd_str, stdout = subprocess.PIPE, stderr = subprocess.PIPE)
if self.cmd.returncode != 0:
print('STDERR:\n{0}'.format(self.cmd.stderr.decode('utf-8')))
raise RuntimeError('Vault did not start correctly')
attempts = 5
seconds = 3
while not self.is_running and attempts != 0:
self._connCheck()
time.sleep(seconds)
if not self.is_running:
raise TimeoutError('Could not start Vault')
self._initChk()
return(None)