forked from ungleich-public/cdist
[type/__postgres_role] Handle password changes
This commit is contained in:
parent
c36df82882
commit
7b7ca4d385
2 changed files with 88 additions and 11 deletions
|
@ -34,9 +34,23 @@ esac
|
|||
|
||||
rolename=${__object_id:?}
|
||||
|
||||
|
||||
psql_query() {
|
||||
su -l "${postgres_user}" -c "$(
|
||||
printf "psql -F '\034' -R '\036' -wAc '%s'" \
|
||||
"$(printf %s "$*" | sed "s/'/'\\\\''/g")"
|
||||
)"
|
||||
}
|
||||
|
||||
password_check_login() (
|
||||
PGPASSWORD=$(cat "${__object:?}/parameter/password"; printf .)
|
||||
PGPASSWORD=${PGPASSWORD%?.}
|
||||
export PGPASSWORD
|
||||
psql -w -h localhost -U "${rolename}" template1 -c '\q' >/dev/null 2>&1
|
||||
)
|
||||
|
||||
role_properties=$(
|
||||
cmd=$(printf "psql -F '\034' -R '\036' -wAc \"SELECT * FROM pg_roles WHERE rolname='%s'\"" "${rolename}")
|
||||
su -l "${postgres_user}" -c "${cmd}" \
|
||||
psql_query "SELECT * FROM pg_roles WHERE rolname = '${rolename}'" \
|
||||
| awk '
|
||||
BEGIN { RS = "\036"; FS = "\034" }
|
||||
/^\([0-9]+ rows?\)/ { exit }
|
||||
|
@ -69,12 +83,62 @@ then
|
|||
)
|
||||
|
||||
test "${bool_is}" = "${bool_should}" || {
|
||||
echo 'different'
|
||||
exit 0
|
||||
state='different properties'
|
||||
}
|
||||
done
|
||||
|
||||
echo 'present'
|
||||
# Check password
|
||||
passwd_stored=$(
|
||||
psql_query "SELECT rolpassword FROM pg_authid WHERE rolname = '${rolename}'" \
|
||||
| awk 'BEGIN { RS = "\036" } NR == 2'
|
||||
printf .
|
||||
)
|
||||
passwd_stored=${passwd_stored%?.}
|
||||
|
||||
passwd_should=$(cat "${__object}/parameter/password"; printf .)
|
||||
passwd_should=${passwd_should%?.}
|
||||
|
||||
if expr "${passwd_stored}" : 'SCRAM-SHA-256\$.*$' >/dev/null
|
||||
then
|
||||
# SCRAM-SHA-256 "encrypted" password
|
||||
# NOTE: There is currently no easy way to check SCRAM passwords
|
||||
password_check_login || state="${state:-different} password"
|
||||
elif expr "${passwd_stored}" : 'md5[0-9a-f]\{32\}$' >/dev/null
|
||||
then
|
||||
# MD5 "encrypted" password
|
||||
if command -v md5sum >/dev/null 2>&1
|
||||
then
|
||||
should_md5=$(
|
||||
printf '%s%s' "${passwd_should}" "${rolename}" \
|
||||
| md5sum - | sed -e 's/[^0-9a-f]*$//')
|
||||
elif command -v gmd5sum >/dev/null 2>&1
|
||||
then
|
||||
should_md5=$(
|
||||
printf '%s%s' "${passwd_should}" "${rolename}" \
|
||||
| gmd5sum - | sed -e 's/[^0-9a-f]*$//')
|
||||
elif command -v openssl >/dev/null 2>&1
|
||||
then
|
||||
should_md5=$(
|
||||
printf '%s%s' "${passwd_should}" "${rolename}" \
|
||||
| openssl dgst -md5 | sed 's/^.* //')
|
||||
fi
|
||||
|
||||
if test -n "${should_md5}"
|
||||
then
|
||||
test "${passwd_stored}" = "md5${should_md5}" \
|
||||
|| state="${state:-different} password"
|
||||
else
|
||||
password_check_login || state="${state:-different} password"
|
||||
fi
|
||||
else
|
||||
# unencrypted password (unsupported since PostgreSQL 10)
|
||||
test "${passwd_stored}" = "${passwd_should}" \
|
||||
|| state="${state:-different} password"
|
||||
fi
|
||||
|
||||
test -n "${state}" || state='present'
|
||||
else
|
||||
echo 'absent'
|
||||
state='absent'
|
||||
fi
|
||||
|
||||
echo "${state}"
|
||||
|
|
|
@ -58,7 +58,9 @@ in
|
|||
then
|
||||
quoted_password=$(
|
||||
delim='$$'
|
||||
while grep -q -F "${delim}" "${__object:?}/parameter/password"
|
||||
# NOTE: Strip away trailing $ because with it the check breaks
|
||||
# if the password ends with $ + random value.
|
||||
while grep -q -F "${delim%$}" "${__object:?}/parameter/password"
|
||||
do
|
||||
delim="\$$(LC_ALL=C tr -cd '[:alpha:]' </dev/urandom | dd bs=1 count=4 2>/dev/null)$"
|
||||
done
|
||||
|
@ -88,12 +90,23 @@ in
|
|||
query=$(printf 'CREATE ROLE "%s" WITH %s PASSWORD %s;' \
|
||||
"${rolename}" "${booleans}" "${quoted_password:-NULL}")
|
||||
;;
|
||||
(different)
|
||||
query=$(printf 'ALTER ROLE "%s" WITH %s PASSWORD %s;' \
|
||||
"${rolename}" "${booleans}" "${quoted_password:-NULL}")
|
||||
(different*)
|
||||
query="ALTER ROLE \"${rolename}\" WITH"
|
||||
|
||||
if expr "${state_is}" : 'different.*properties' >/dev/null
|
||||
then
|
||||
query="${query} ${booleans}"
|
||||
fi
|
||||
if expr "${state_is}" : 'different.*password' >/dev/null
|
||||
then
|
||||
query="${query} PASSWORD ${quoted_password:-NULL}"
|
||||
fi
|
||||
|
||||
query="${query};"
|
||||
;;
|
||||
(*)
|
||||
exit 1 # TODO: error msg
|
||||
printf 'Invalid state reported by state explorer: %s\n' "${state_is}" >&2
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
|
||||
|
|
Loading…
Reference in a new issue