lib/controller/localdb/login_ldap_docker_test.sh

   1 #!/bin/bash
   2
   3 # Copyright (C) The Arvados Authors. All rights reserved.
   4 #
   5 # SPDX-License-Identifier: AGPL-3.0
   6
   7 # This script demonstrates using LDAP for Arvados user authentication.
   8 #
   9 # It configures arvados controller in a docker container, optionally
  10 # with pam_ldap(5) configured to authenticate against an OpenLDAP
  11 # server in a second docker container.
  12 #
  13 # After adding a "foo" user entry, it uses curl to check that the
  14 # Arvados controller's login endpoint accepts the "foo" account
  15 # username/password and rejects invalid credentials.
  16 #
  17 # It is intended to be run inside .../build/run-tests.sh (in
  18 # interactive mode: "test lib/controller/localdb -tags=docker
  19 # -check.f=LDAP -check.vv"). It assumes ARVADOS_TEST_API_HOST points
  20 # to a RailsAPI server and the desired version of arvados-server is
  21 # installed in $GOPATH/bin.
  22
  23 set -e -o pipefail
  24
  25 debug=/dev/null
  26 if [[ -n ${ARVADOS_DEBUG} ]]; then
  27     debug=/dev/stderr
  28     set -x
  29 fi
  30
  31 case "${config_method}" in
  32     pam | ldap)
  33         ;;
  34     *)
  35         echo >&2 "\$config_method env var must be 'pam' or 'ldap'"
  36         exit 1
  37         ;;
  38 esac
  39
  40 hostname="$(hostname)"
  41 tmpdir="$(mktemp -d)"
  42 cleanup() {
  43     trap - ERR
  44     rm -r ${tmpdir}
  45     for h in ${ldapctr} ${ctrlctr}; do
  46         if [[ -n ${h} ]]; then
  47             docker kill ${h}
  48         fi
  49     done
  50 }
  51 trap cleanup ERR
  52
  53 if [[ -z "$(docker image ls -q osixia/openldap:1.3.0)" ]]; then
  54     echo >&2 "Pulling docker image for ldap server"
  55     docker pull osixia/openldap:1.3.0
  56 fi
  57
  58 ldapctr=ldap-${RANDOM}
  59 echo >&2 "Starting ldap server in docker container ${ldapctr}"
  60 docker run --rm --detach \
  61        -p 389 -p 636 \
  62        --name=${ldapctr} \
  63        osixia/openldap:1.3.0
  64 docker logs --follow ${ldapctr} 2>$debug >$debug &
  65 ldaphostport=$(docker port ${ldapctr} 389/tcp)
  66 ldapport=${ldaphostport##*:}
  67 ldapurl="ldap://${hostname}:${ldapport}"
  68 passwordhash="$(docker exec -i ${ldapctr} slappasswd -s "secret")"
  69
  70 # These are the default admin credentials for osixia/openldap:1.3.0
  71 adminuser=admin
  72 adminpassword=admin
  73
  74 cat >"${tmpdir}/zzzzz.yml" <<EOF
  75 Clusters:
  76   zzzzz:
  77     PostgreSQL:
  78       Connection:
  79         client_encoding: utf8
  80         host: ${hostname}
  81         port: "${pgport}"
  82         dbname: arvados_test
  83         user: arvados
  84         password: insecure_arvados_test
  85     ManagementToken: e687950a23c3a9bceec28c6223a06c79
  86     SystemRootToken: systemusertesttoken1234567890aoeuidhtnsqjkxbmwvzpy
  87     API:
  88       RequestTimeout: 30s
  89     TLS:
  90       Insecure: true
  91     Collections:
  92       BlobSigningKey: zfhgfenhffzltr9dixws36j1yhksjoll2grmku38mi7yxd66h5j4q9w4jzanezacp8s6q0ro3hxakfye02152hncy6zml2ed0uc
  93       TrustAllContent: true
  94       ForwardSlashNameSubstitution: /
  95     Services:
  96       RailsAPI:
  97         InternalURLs:
  98           "https://${hostname}:${ARVADOS_TEST_API_HOST##*:}/": {}
  99       Controller:
 100         ExternalURL: http://0.0.0.0:9999/
 101         InternalURLs:
 102           "http://0.0.0.0:9999/": {}
 103     SystemLogs:
 104       LogLevel: debug
 105 EOF
 106 case "${config_method}" in
 107     pam)
 108         setup_pam_ldap="apt update && DEBIAN_FRONTEND=noninteractive apt install -y ldap-utils libpam-ldap && pam-auth-update --package /usr/share/pam-configs/ldap"
 109         cat >>"${tmpdir}/zzzzz.yml" <<EOF
 110     Login:
 111       PAM:
 112         Enable: true
 113         # Without this specific DefaultEmailDomain, inserted users
 114         # would prevent subsequent database/reset from working (see
 115         # database_controller.rb).
 116         DefaultEmailDomain: example.com
 117 EOF
 118         ;;
 119     ldap)
 120         setup_pam_ldap=""
 121         cat >>"${tmpdir}/zzzzz.yml" <<EOF
 122     Login:
 123       LDAP:
 124         Enable: true
 125         URL: ${ldapurl}
 126         StartTLS: false
 127         SearchBase: dc=example,dc=org
 128         SearchBindUser: cn=admin,dc=example,dc=org
 129         SearchBindPassword: admin
 130 EOF
 131             ;;
 132 esac
 133
 134 cat >&2 "${tmpdir}/zzzzz.yml"
 135
 136 cat >"${tmpdir}/pam_ldap.conf" <<EOF
 137 base dc=example,dc=org
 138 ldap_version 3
 139 uri ${ldapurl}
 140 pam_password crypt
 141 binddn cn=${adminuser},dc=example,dc=org
 142 bindpw ${adminpassword}
 143 EOF
 144
 145 cat >"${tmpdir}/add_example_user.ldif" <<EOF
 146 dn: cn=bar,dc=example,dc=org
 147 objectClass: posixGroup
 148 objectClass: top
 149 cn: bar
 150 gidNumber: 11111
 151 description: "Example group 'bar'"
 152
 153 dn: uid=foo-bar,dc=example,dc=org
 154 uid: foo-bar
 155 cn: "Foo Bar"
 156 givenName: Foo
 157 sn: Bar
 158 mail: foo-bar-baz@example.com
 159 objectClass: inetOrgPerson
 160 objectClass: posixAccount
 161 objectClass: top
 162 objectClass: shadowAccount
 163 shadowMax: 180
 164 shadowMin: 1
 165 shadowWarning: 7
 166 shadowLastChange: 10701
 167 loginShell: /bin/bash
 168 uidNumber: 11111
 169 gidNumber: 11111
 170 homeDirectory: /home/foo-bar
 171 userPassword: ${passwordhash}
 172 EOF
 173
 174 echo >&2 "Adding example user entry user=foo-bar pass=secret (retrying until server comes up)"
 175 docker run --rm --entrypoint= \
 176        -v "${tmpdir}/add_example_user.ldif":/add_example_user.ldif:ro \
 177        osixia/openldap:1.3.0 \
 178        bash -c "for f in \$(seq 1 5); do if ldapadd -H '${ldapurl}' -D 'cn=${adminuser},dc=example,dc=org' -w '${adminpassword}' -f /add_example_user.ldif; then exit 0; else sleep 2; fi; done; echo 'failed to add user entry'; exit 1"
 179
 180 echo >&2 "Building arvados controller binary to run in container"
 181 go build -o "${tmpdir}" ../../../cmd/arvados-server
 182
 183 ctrlctr=ctrl-${RANDOM}
 184 echo >&2 "Starting arvados controller in docker container ${ctrlctr}"
 185 docker run --detach --rm --name=${ctrlctr} \
 186        -p 9999 \
 187        -v "${tmpdir}/pam_ldap.conf":/etc/pam_ldap.conf:ro \
 188        -v "${tmpdir}/arvados-server":/bin/arvados-server:ro \
 189        -v "${tmpdir}/zzzzz.yml":/etc/arvados/config.yml:ro \
 190        -v $(realpath "${PWD}/../../.."):/arvados:ro \
 191        debian:10 \
 192        bash -c "${setup_pam_ldap:-true} && arvados-server controller"
 193 docker logs --follow ${ctrlctr} 2>$debug >$debug &
 194 ctrlhostport=$(docker port ${ctrlctr} 9999/tcp)
 195
 196 echo >&2 "Waiting for arvados controller to come up..."
 197 for f in $(seq 1 20); do
 198     if curl -s "http://${ctrlhostport}/arvados/v1/config" >/dev/null; then
 199         break
 200     else
 201         sleep 1
 202     fi
 203     echo -n >&2 .
 204 done
 205 echo >&2
 206 echo >&2 "Arvados controller is up at http://${ctrlhostport}"
 207
 208 check_contains() {
 209     resp="${1}"
 210     str="${2}"
 211     if ! echo "${resp}" | fgrep -q "${str}"; then
 212         echo >&2 "${resp}"
 213         echo >&2 "FAIL: expected in response, but not found: ${str@Q}"
 214         return 1
 215     fi
 216 }
 217
 218 set +x
 219
 220 echo >&2 "Testing authentication failure"
 221 resp="$(set -x; curl -s --include -d username=foo-bar -d password=nosecret "http://${ctrlhostport}/arvados/v1/users/authenticate" | tee $debug)"
 222 check_contains "${resp}" "HTTP/1.1 401"
 223 if [[ "${config_method}" = ldap ]]; then
 224     check_contains "${resp}" '{"errors":["LDAP: Authentication failure (with username \"foo-bar\" and password)"]}'
 225 else
 226     check_contains "${resp}" '{"errors":["PAM: Authentication failure (with username \"foo-bar\" and password)"]}'
 227 fi
 228
 229 echo >&2 "Testing authentication success"
 230 resp="$(set -x; curl -s --include -d username=foo-bar -d password=secret "http://${ctrlhostport}/arvados/v1/users/authenticate" | tee $debug)"
 231 check_contains "${resp}" "HTTP/1.1 200"
 232 check_contains "${resp}" '"api_token":"'
 233 check_contains "${resp}" '"scopes":["all"]'
 234 check_contains "${resp}" '"uuid":"zzzzz-gj3su-'
 235
 236 secret="${resp##*api_token\":\"}"
 237 secret="${secret%%\"*}"
 238 uuid="${resp##*uuid\":\"}"
 239 uuid="${uuid%%\"*}"
 240 token="v2/$uuid/$secret"
 241 echo >&2 "New token is ${token}"
 242
 243 resp="$(set -x; curl -s --include -H "Authorization: Bearer ${token}" "http://${ctrlhostport}/arvados/v1/users/current" | tee $debug)"
 244 check_contains "${resp}" "HTTP/1.1 200"
 245 if [[ "${config_method}" = ldap ]]; then
 246     # user fields come from LDAP attributes
 247     check_contains "${resp}" '"first_name":"Foo"'
 248     check_contains "${resp}" '"last_name":"Bar"'
 249     check_contains "${resp}" '"username":"foobar"' # "-" removed by rails api
 250     check_contains "${resp}" '"email":"foo-bar-baz@example.com"'
 251 else
 252     # PAMDefaultEmailDomain
 253     check_contains "${resp}" '"email":"foo-bar@example.com"'
 254 fi
 255
 256 cleanup