2 files changed, 122 insertions, 0 deletions
diff --git a/code/environments/production/modules/ca_extend/files/common.sh b/code/environments/production/modules/ca_extend/files/common.sh
new file mode 100644
index 0000000..be052f7
--- /dev/null
+++ b/code/environments/production/modules/ca_extend/files/common.sh
@@ -0,0 +1,38 @@
+#!/bin/bash
+
+# TODO: helper task?
+
+# Exit with an error message and error code, defaulting to 1
+fail() {
+  # Print a stderr: entry if there were anything printed to stderr
+  if [[ -s $_tmp ]]; then
+    # Hack to try and output valid json by replacing newlines with spaces.
+    echo "{ \"status\": \"error\", \"message\": \"$1\", \"stderr\": \"$(tr '\n' ' ' <"$_tmp")\" }"
+  else
+    echo "{ \"status\": \"error\", \"message\": \"$1\" }"
+  fi
+
+  exit "${2:-1}"
+}
+
+success() {
+  echo "$1"
+  exit 0
+}
+
+# Test for colors. If unavailable, unset variables are ok
+# shellcheck disable=SC2034
+if tput colors &>/dev/null; then
+  green="$(tput setaf 2)"
+  red="$(tput setaf 1)"
+  reset="$(tput sgr0)"
+fi
+
+_tmp="$(mktemp)"
+exec 2>>"$_tmp"
+
+# Use indirection to munge PT_ environment variables
+# e.g. "$PT_version" becomes "$version"
+for v in ${!PT_*}; do
+  declare "${v#*PT_}"="${!v}"
+done
diff --git a/code/environments/production/modules/ca_extend/files/extend.sh b/code/environments/production/modules/ca_extend/files/extend.sh
new file mode 100644
index 0000000..10fb837
--- /dev/null
+++ b/code/environments/production/modules/ca_extend/files/extend.sh
@@ -0,0 +1,84 @@
+#!/bin/bash
+
+# Credit: https://github.com/Sharpie
+
+set -e
+
+PUPPET_BIN='/opt/puppetlabs/puppet/bin'
+
+ca_cert=$("${PUPPET_BIN}/puppet" config print --section master cacert)
+ca_key=$("${PUPPET_BIN}/puppet" config print --section master cakey)
+ca_dir=$(dirname "${ca_cert}")
+
+printf 'CA certificate file: %s\n' "${ca_cert}" >&2
+printf 'CA private key file: %s\n' "${ca_key}" >&2
+
+printf '\nChecking CA chain length...\n' >&2
+chain_length=$(grep -cF 'BEGIN CERTIFICATE' "${ca_cert}")
+
+if (( chain_length > 1 )); then
+  printf '%s certificates were found in: %s\n' "${chain_length}" "${ca_cert}" >&2
+  printf 'This script only works on CA files that contain a single certificate.\n' >&2
+  exit 1
+elif (( chain_length != 1 )); then
+  printf 'No certificates found in: %s\n' "${ca_cert}" >&2
+  exit 1
+else
+  printf '%s certificate found in: %s\n' "${chain_length}" "${ca_cert}" >&2
+fi
+
+# Compute start and end dates for new certificate.
+# Formats the year as YY instead of YYYY because the latter isn't supported
+# until OpenSSL 1.1.1.
+start_date=$(date -u --date='-24 hours' '+%y%m%d%H%M%SZ')
+end_date=$(date -u --date='+15 years' '+%y%m%d%H%M%SZ')
+ca_serial_num=$("${PUPPET_BIN}/openssl" x509 -in "${ca_cert}" -noout -serial|cut -d= -f2)
+
+# Build a temporary directory with files required to renew the CA cert.
+workdir=$(mktemp -d -t renew_ca_cert.XXX)
+printf '%s' "${ca_serial_num}" > "${workdir}/serial"
+touch "${workdir}/inventory"
+touch "${workdir}/inventory.attr"
+
+cat <<EOT > "${workdir}/openssl.cnf"
+[ca]
+default_ca=ca_settings
+
+[ca_settings]
+serial=${workdir}/serial
+new_certs_dir=${workdir}
+database=${workdir}/inventory
+default_md=sha256
+policy=ca_policy
+x509_extensions=cert_extensions
+
+[ca_policy]
+commonName=supplied
+
+[cert_extensions]
+basicConstraints=critical,CA:TRUE
+keyUsage=keyCertSign,cRLSign
+subjectKeyIdentifier=hash
+authorityKeyIdentifier=issuer:always
+EOT
+
+# Generate a signing request from the existing certificate
+"${PUPPET_BIN}/openssl" x509 -x509toreq \
+  -in "${ca_cert}" \
+  -signkey "${ca_key}" \
+  -out "${workdir}/ca_csr.pem"
+
+# Sign the request
+new_ca_cert="${ca_dir}/ca_crt-expires-${end_date}.pem"
+
+yes | "${PUPPET_BIN}/openssl" ca \
+  -in "${workdir}/ca_csr.pem" \
+  -keyfile "${ca_key}" \
+  -config "${workdir}/openssl.cnf" \
+  -selfsign \
+  -startdate "${start_date}" \
+  -enddate "${end_date}" \
+  -out "${new_ca_cert}" >&2
+
+printf '\nRenewed CA certificate\n' >&2
+printf '%s\n' "${new_ca_cert}"