6bf74aa20d
This commit adds two new cli commands to allow an operator to read and write passwords into a configured Hashicorp Vault KV. Change-Id: Icf0eaf7544fcbdf7b83f697cc711446f47118a4d
121 lines
4.1 KiB
Python
Executable File
121 lines
4.1 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
# 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.
|
|
|
|
import argparse
|
|
import os
|
|
import sys
|
|
|
|
import hvac
|
|
import yaml
|
|
|
|
from kolla_ansible.hashi_vault import hashicorp_vault_client
|
|
|
|
|
|
def writepwd(passwords_file, vault_kv_path, vault_mount_point, vault_namespace,
|
|
vault_addr, vault_role_id, vault_secret_id, vault_token,
|
|
vault_cacert):
|
|
with open(passwords_file, 'r') as f:
|
|
passwords = yaml.safe_load(f.read())
|
|
|
|
if not isinstance(passwords, dict):
|
|
print("ERROR: Passwords file not in expected key/value format")
|
|
sys.exit(1)
|
|
|
|
client = hashicorp_vault_client(vault_namespace, vault_addr, vault_role_id,
|
|
vault_secret_id, vault_token, vault_cacert)
|
|
|
|
for key, value in passwords.items():
|
|
# Ignore empty values
|
|
if not value:
|
|
continue
|
|
|
|
if isinstance(value, str):
|
|
value = dict(password=value)
|
|
|
|
try:
|
|
remote_value = client.secrets.kv.v2.read_secret_version(
|
|
mount_point=vault_mount_point,
|
|
path="{}/{}".format(vault_kv_path, key))
|
|
except hvac.exceptions.InvalidPath:
|
|
# Add to KV if value does not exists
|
|
remote_value = None
|
|
|
|
# Update KV is value has changed or it does not exist
|
|
if not remote_value or remote_value['data']['data'] != value:
|
|
client.secrets.kv.v2.create_or_update_secret(
|
|
mount_point=vault_mount_point,
|
|
path="{}/{}".format(vault_kv_path, key),
|
|
secret=value)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
'-p', '--passwords', type=str,
|
|
default=os.path.abspath('/etc/kolla/passwords.yml'),
|
|
help='Path to the passwords.yml file')
|
|
parser.add_argument(
|
|
'-kv', '--vault-mount-point', type=str,
|
|
default='kv',
|
|
help='Path to the KV mount point')
|
|
parser.add_argument(
|
|
'-kvp', '--vault-kv-path', type=str,
|
|
default='kolla_passwords',
|
|
help='Path to store passwords within your configured KV mount point')
|
|
parser.add_argument(
|
|
'-n', '--vault-namespace', type=str,
|
|
default='',
|
|
help='Vault namespace (enterprise only)')
|
|
parser.add_argument(
|
|
'-v', '--vault-addr', type=str,
|
|
required=True,
|
|
help='Address to connect to an existing Hashicorp Vault')
|
|
parser.add_argument(
|
|
'-r', '--vault-role-id', type=str,
|
|
default='',
|
|
help='Role-ID to authenticate to Vault. This must be used in '
|
|
'conjunction with --secret-id')
|
|
parser.add_argument(
|
|
'-s', '--vault-secret-id', type=str,
|
|
default='',
|
|
help='Secret-ID to authenticate to Vault. This must be used in '
|
|
'conjunction with --role-id')
|
|
parser.add_argument(
|
|
'-t', '--vault-token', type=str,
|
|
default='',
|
|
help='Vault token to authenticate to Vault')
|
|
parser.add_argument(
|
|
'-c', '--vault-cacert', type=str,
|
|
default='',
|
|
help='Path to CA certificate file')
|
|
|
|
args = parser.parse_args()
|
|
passwords_file = os.path.expanduser(args.passwords)
|
|
vault_kv_path = args.vault_kv_path
|
|
vault_mount_point = args.vault_mount_point
|
|
vault_namespace = args.vault_namespace
|
|
vault_addr = args.vault_addr
|
|
vault_role_id = args.vault_role_id
|
|
vault_secret_id = args.vault_secret_id
|
|
vault_token = args.vault_token
|
|
vault_cacert = os.path.expanduser(args.vault_cacert)
|
|
|
|
writepwd(passwords_file, vault_kv_path, vault_mount_point, vault_namespace,
|
|
vault_addr, vault_role_id, vault_secret_id, vault_token,
|
|
vault_cacert)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|