Compare commits
	
		
			17 Commits
		
	
	
		
			website/in
			...
			web/packag
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| 1dc4fbbb2b | |||
| 3332de267d | |||
| ab366d0ec2 | |||
| ac162582aa | |||
| 7d82e029d5 | |||
| 9b40ecb023 | |||
| 0cc0fdaae3 | |||
| b55b168718 | |||
| c46dc8f290 | |||
| e48da3520c | |||
| 1ec4652c60 | |||
| e375646705 | |||
| b84652d9d3 | |||
| 74b8da28ca | |||
| 9084c7c6b4 | |||
| 7a0b227b46 | |||
| cc9128fd46 | 
| @ -1,5 +1,5 @@ | ||||
| [bumpversion] | ||||
| current_version = 2024.10.4 | ||||
| current_version = 2024.6.3 | ||||
| tag = True | ||||
| commit = True | ||||
| parse = (?P<major>\d+)\.(?P<minor>\d+)\.(?P<patch>\d+)(?:-(?P<rc_t>[a-zA-Z-]+)(?P<rc_n>[1-9]\\d*))? | ||||
|  | ||||
| @ -29,9 +29,9 @@ outputs: | ||||
|   imageTags: | ||||
|     description: "Docker image tags" | ||||
|     value: ${{ steps.ev.outputs.imageTags }} | ||||
|   attestImageNames: | ||||
|     description: "Docker image names used for attestation" | ||||
|     value: ${{ steps.ev.outputs.attestImageNames }} | ||||
|   imageNames: | ||||
|     description: "Docker image names" | ||||
|     value: ${{ steps.ev.outputs.imageNames }} | ||||
|   imageMainTag: | ||||
|     description: "Docker image main tag" | ||||
|     value: ${{ steps.ev.outputs.imageMainTag }} | ||||
|  | ||||
| @ -51,24 +51,15 @@ else: | ||||
|         ] | ||||
|  | ||||
| image_main_tag = image_tags[0].split(":")[-1] | ||||
|  | ||||
|  | ||||
| def get_attest_image_names(image_with_tags: list[str]): | ||||
|     """Attestation only for GHCR""" | ||||
|     image_tags = [] | ||||
|     for image_name in set(name.split(":")[0] for name in image_with_tags): | ||||
|         if not image_name.startswith("ghcr.io"): | ||||
|             continue | ||||
|         image_tags.append(image_name) | ||||
|     return ",".join(set(image_tags)) | ||||
|  | ||||
| image_tags_rendered = ",".join(image_tags) | ||||
| image_names_rendered = ",".join(set(name.split(":")[0] for name in image_tags)) | ||||
|  | ||||
| with open(os.environ["GITHUB_OUTPUT"], "a+", encoding="utf-8") as _output: | ||||
|     print(f"shouldBuild={should_build}", file=_output) | ||||
|     print(f"sha={sha}", file=_output) | ||||
|     print(f"version={version}", file=_output) | ||||
|     print(f"prerelease={prerelease}", file=_output) | ||||
|     print(f"imageTags={','.join(image_tags)}", file=_output) | ||||
|     print(f"attestImageNames={get_attest_image_names(image_tags)}", file=_output) | ||||
|     print(f"imageTags={image_tags_rendered}", file=_output) | ||||
|     print(f"imageNames={image_names_rendered}", file=_output) | ||||
|     print(f"imageMainTag={image_main_tag}", file=_output) | ||||
|     print(f"imageMainName={image_tags[0]}", file=_output) | ||||
|  | ||||
							
								
								
									
										2
									
								
								.github/actions/setup/action.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/actions/setup/action.yml
									
									
									
									
										vendored
									
									
								
							| @ -14,7 +14,7 @@ runs: | ||||
|       run: | | ||||
|         pipx install poetry || true | ||||
|         sudo apt-get update | ||||
|         sudo apt-get install --no-install-recommends -y libpq-dev openssl libxmlsec1-dev pkg-config gettext libkrb5-dev krb5-kdc krb5-user krb5-admin-server | ||||
|         sudo apt-get install --no-install-recommends -y libpq-dev openssl libxmlsec1-dev pkg-config gettext | ||||
|     - name: Setup python and restore poetry | ||||
|       uses: actions/setup-python@v5 | ||||
|       with: | ||||
|  | ||||
							
								
								
									
										11
									
								
								.github/dependabot.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										11
									
								
								.github/dependabot.yml
									
									
									
									
										vendored
									
									
								
							| @ -23,6 +23,7 @@ updates: | ||||
|   - package-ecosystem: npm | ||||
|     directories: | ||||
|       - "/web" | ||||
|       - "/tests/wdio" | ||||
|       - "/web/sfe" | ||||
|     schedule: | ||||
|       interval: daily | ||||
| @ -43,11 +44,9 @@ updates: | ||||
|           - "babel-*" | ||||
|       eslint: | ||||
|         patterns: | ||||
|           - "@eslint/*" | ||||
|           - "@typescript-eslint/*" | ||||
|           - "eslint-*" | ||||
|           - "eslint" | ||||
|           - "typescript-eslint" | ||||
|           - "eslint-*" | ||||
|       storybook: | ||||
|         patterns: | ||||
|           - "@storybook/*" | ||||
| @ -55,16 +54,10 @@ updates: | ||||
|       esbuild: | ||||
|         patterns: | ||||
|           - "@esbuild/*" | ||||
|           - "esbuild*" | ||||
|       rollup: | ||||
|         patterns: | ||||
|           - "@rollup/*" | ||||
|           - "rollup-*" | ||||
|           - "rollup*" | ||||
|       swc: | ||||
|         patterns: | ||||
|           - "@swc/*" | ||||
|           - "swc-*" | ||||
|       wdio: | ||||
|         patterns: | ||||
|           - "@wdio/*" | ||||
|  | ||||
							
								
								
									
										2
									
								
								.github/pull_request_template.md
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/pull_request_template.md
									
									
									
									
										vendored
									
									
								
							| @ -1,7 +1,7 @@ | ||||
| <!-- | ||||
| 👋 Hi there! Welcome. | ||||
|  | ||||
| Please check the Contributing guidelines: https://docs.goauthentik.io/docs/developer-docs/#how-can-i-contribute | ||||
| Please check the Contributing guidelines: https://goauthentik.io/developer-docs/#how-can-i-contribute | ||||
| --> | ||||
|  | ||||
| ## Details | ||||
|  | ||||
							
								
								
									
										2
									
								
								.github/workflows/api-ts-publish.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/workflows/api-ts-publish.yml
									
									
									
									
										vendored
									
									
								
							| @ -40,7 +40,7 @@ jobs: | ||||
|         run: | | ||||
|           export VERSION=`node -e 'console.log(require("../gen-ts-api/package.json").version)'` | ||||
|           npm i @goauthentik/api@$VERSION | ||||
|       - uses: peter-evans/create-pull-request@v7 | ||||
|       - uses: peter-evans/create-pull-request@v6 | ||||
|         id: cpr | ||||
|         with: | ||||
|           token: ${{ steps.generate_token.outputs.token }} | ||||
|  | ||||
							
								
								
									
										28
									
								
								.github/workflows/ci-main.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										28
									
								
								.github/workflows/ci-main.yml
									
									
									
									
										vendored
									
									
								
							| @ -116,16 +116,10 @@ jobs: | ||||
|           poetry run make test | ||||
|           poetry run coverage xml | ||||
|       - if: ${{ always() }} | ||||
|         uses: codecov/codecov-action@v5 | ||||
|         uses: codecov/codecov-action@v4 | ||||
|         with: | ||||
|           flags: unit | ||||
|           token: ${{ secrets.CODECOV_TOKEN }} | ||||
|       - if: ${{ !cancelled() }} | ||||
|         uses: codecov/test-results-action@v1 | ||||
|         with: | ||||
|           flags: unit | ||||
|           file: unittest.xml | ||||
|           token: ${{ secrets.CODECOV_TOKEN }} | ||||
|   test-integration: | ||||
|     runs-on: ubuntu-latest | ||||
|     timeout-minutes: 30 | ||||
| @ -140,16 +134,10 @@ jobs: | ||||
|           poetry run coverage run manage.py test tests/integration | ||||
|           poetry run coverage xml | ||||
|       - if: ${{ always() }} | ||||
|         uses: codecov/codecov-action@v5 | ||||
|         uses: codecov/codecov-action@v4 | ||||
|         with: | ||||
|           flags: integration | ||||
|           token: ${{ secrets.CODECOV_TOKEN }} | ||||
|       - if: ${{ !cancelled() }} | ||||
|         uses: codecov/test-results-action@v1 | ||||
|         with: | ||||
|           flags: integration | ||||
|           file: unittest.xml | ||||
|           token: ${{ secrets.CODECOV_TOKEN }} | ||||
|   test-e2e: | ||||
|     name: test-e2e (${{ matrix.job.name }}) | ||||
|     runs-on: ubuntu-latest | ||||
| @ -180,7 +168,7 @@ jobs: | ||||
|         uses: ./.github/actions/setup | ||||
|       - name: Setup e2e env (chrome, etc) | ||||
|         run: | | ||||
|           docker compose -f tests/e2e/docker-compose.yml up -d --quiet-pull | ||||
|           docker compose -f tests/e2e/docker-compose.yml up -d | ||||
|       - id: cache-web | ||||
|         uses: actions/cache@v4 | ||||
|         with: | ||||
| @ -198,16 +186,10 @@ jobs: | ||||
|           poetry run coverage run manage.py test ${{ matrix.job.glob }} | ||||
|           poetry run coverage xml | ||||
|       - if: ${{ always() }} | ||||
|         uses: codecov/codecov-action@v5 | ||||
|         uses: codecov/codecov-action@v4 | ||||
|         with: | ||||
|           flags: e2e | ||||
|           token: ${{ secrets.CODECOV_TOKEN }} | ||||
|       - if: ${{ !cancelled() }} | ||||
|         uses: codecov/test-results-action@v1 | ||||
|         with: | ||||
|           flags: e2e | ||||
|           file: unittest.xml | ||||
|           token: ${{ secrets.CODECOV_TOKEN }} | ||||
|   ci-core-mark: | ||||
|     needs: | ||||
|       - lint | ||||
| @ -279,7 +261,7 @@ jobs: | ||||
|         id: attest | ||||
|         if: ${{ steps.ev.outputs.shouldBuild == 'true' }} | ||||
|         with: | ||||
|           subject-name: ${{ steps.ev.outputs.attestImageNames }} | ||||
|           subject-name: ${{ steps.ev.outputs.imageNames }} | ||||
|           subject-digest: ${{ steps.push.outputs.digest }} | ||||
|           push-to-registry: true | ||||
|   pr-comment: | ||||
|  | ||||
							
								
								
									
										4
									
								
								.github/workflows/ci-outpost.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										4
									
								
								.github/workflows/ci-outpost.yml
									
									
									
									
										vendored
									
									
								
							| @ -31,7 +31,7 @@ jobs: | ||||
|       - name: golangci-lint | ||||
|         uses: golangci/golangci-lint-action@v6 | ||||
|         with: | ||||
|           version: latest | ||||
|           version: v1.54.2 | ||||
|           args: --timeout 5000s --verbose | ||||
|           skip-cache: true | ||||
|   test-unittest: | ||||
| @ -115,7 +115,7 @@ jobs: | ||||
|         id: attest | ||||
|         if: ${{ steps.ev.outputs.shouldBuild == 'true' }} | ||||
|         with: | ||||
|           subject-name: ${{ steps.ev.outputs.attestImageNames }} | ||||
|           subject-name: ${{ steps.ev.outputs.imageNames }} | ||||
|           subject-digest: ${{ steps.push.outputs.digest }} | ||||
|           push-to-registry: true | ||||
|   build-binary: | ||||
|  | ||||
							
								
								
									
										24
									
								
								.github/workflows/ci-web.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										24
									
								
								.github/workflows/ci-web.yml
									
									
									
									
										vendored
									
									
								
							| @ -24,11 +24,17 @@ jobs: | ||||
|           - prettier-check | ||||
|         project: | ||||
|           - web | ||||
|           - tests/wdio | ||||
|         include: | ||||
|           - command: tsc | ||||
|             project: web | ||||
|           - command: lit-analyse | ||||
|             project: web | ||||
|         exclude: | ||||
|           - command: lint:lockfile | ||||
|             project: tests/wdio | ||||
|           - command: tsc | ||||
|             project: tests/wdio | ||||
|     steps: | ||||
|       - uses: actions/checkout@v4 | ||||
|       - uses: actions/setup-node@v4 | ||||
| @ -39,12 +45,21 @@ jobs: | ||||
|       - working-directory: ${{ matrix.project }}/ | ||||
|         run: | | ||||
|           npm ci | ||||
|           ${{ matrix.extra_setup }} | ||||
|       - name: Generate API | ||||
|         run: make gen-client-ts | ||||
|       - name: Lint | ||||
|         working-directory: ${{ matrix.project }}/ | ||||
|         run: npm run ${{ matrix.command }} | ||||
|   ci-web-mark: | ||||
|     needs: | ||||
|       - lint | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|       - run: echo mark | ||||
|   build: | ||||
|     needs: | ||||
|       - ci-web-mark | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|       - uses: actions/checkout@v4 | ||||
| @ -60,13 +75,6 @@ jobs: | ||||
|       - name: build | ||||
|         working-directory: web/ | ||||
|         run: npm run build | ||||
|   ci-web-mark: | ||||
|     needs: | ||||
|       - build | ||||
|       - lint | ||||
|     runs-on: ubuntu-latest | ||||
|     steps: | ||||
|       - run: echo mark | ||||
|   test: | ||||
|     needs: | ||||
|       - ci-web-mark | ||||
| @ -84,4 +92,4 @@ jobs: | ||||
|         run: make gen-client-ts | ||||
|       - name: test | ||||
|         working-directory: web/ | ||||
|         run: npm run test || exit 0 | ||||
|         run: npm run test | ||||
|  | ||||
| @ -24,7 +24,7 @@ jobs: | ||||
|       - name: Setup authentik env | ||||
|         uses: ./.github/actions/setup | ||||
|       - run: poetry run ak update_webauthn_mds | ||||
|       - uses: peter-evans/create-pull-request@v7 | ||||
|       - uses: peter-evans/create-pull-request@v6 | ||||
|         id: cpr | ||||
|         with: | ||||
|           token: ${{ steps.generate_token.outputs.token }} | ||||
|  | ||||
							
								
								
									
										2
									
								
								.github/workflows/image-compress.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										2
									
								
								.github/workflows/image-compress.yml
									
									
									
									
										vendored
									
									
								
							| @ -42,7 +42,7 @@ jobs: | ||||
|         with: | ||||
|           githubToken: ${{ steps.generate_token.outputs.token }} | ||||
|           compressOnly: ${{ github.event_name != 'pull_request' }} | ||||
|       - uses: peter-evans/create-pull-request@v7 | ||||
|       - uses: peter-evans/create-pull-request@v6 | ||||
|         if: "${{ github.event_name != 'pull_request' && steps.compress.outputs.markdown != '' }}" | ||||
|         id: cpr | ||||
|         with: | ||||
|  | ||||
							
								
								
									
										8
									
								
								.github/workflows/release-publish.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										8
									
								
								.github/workflows/release-publish.yml
									
									
									
									
										vendored
									
									
								
							| @ -51,14 +51,12 @@ jobs: | ||||
|           secrets: | | ||||
|             GEOIPUPDATE_ACCOUNT_ID=${{ secrets.GEOIPUPDATE_ACCOUNT_ID }} | ||||
|             GEOIPUPDATE_LICENSE_KEY=${{ secrets.GEOIPUPDATE_LICENSE_KEY }} | ||||
|           build-args: | | ||||
|             VERSION=${{ github.ref }} | ||||
|           tags: ${{ steps.ev.outputs.imageTags }} | ||||
|           platforms: linux/amd64,linux/arm64 | ||||
|       - uses: actions/attest-build-provenance@v1 | ||||
|         id: attest | ||||
|         with: | ||||
|           subject-name: ${{ steps.ev.outputs.attestImageNames }} | ||||
|           subject-name: ${{ steps.ev.outputs.imageNames }} | ||||
|           subject-digest: ${{ steps.push.outputs.digest }} | ||||
|           push-to-registry: true | ||||
|   build-outpost: | ||||
| @ -113,8 +111,6 @@ jobs: | ||||
|         id: push | ||||
|         with: | ||||
|           push: true | ||||
|           build-args: | | ||||
|             VERSION=${{ github.ref }} | ||||
|           tags: ${{ steps.ev.outputs.imageTags }} | ||||
|           file: ${{ matrix.type }}.Dockerfile | ||||
|           platforms: linux/amd64,linux/arm64 | ||||
| @ -122,7 +118,7 @@ jobs: | ||||
|       - uses: actions/attest-build-provenance@v1 | ||||
|         id: attest | ||||
|         with: | ||||
|           subject-name: ${{ steps.ev.outputs.attestImageNames }} | ||||
|           subject-name: ${{ steps.ev.outputs.imageNames }} | ||||
|           subject-digest: ${{ steps.push.outputs.digest }} | ||||
|           push-to-registry: true | ||||
|   build-outpost-binary: | ||||
|  | ||||
| @ -32,7 +32,7 @@ jobs: | ||||
|           poetry run ak compilemessages | ||||
|           make web-check-compile | ||||
|       - name: Create Pull Request | ||||
|         uses: peter-evans/create-pull-request@v7 | ||||
|         uses: peter-evans/create-pull-request@v6 | ||||
|         with: | ||||
|           token: ${{ steps.generate_token.outputs.token }} | ||||
|           branch: extract-compile-backend-translation | ||||
|  | ||||
							
								
								
									
										1
									
								
								.vscode/settings.json
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.vscode/settings.json
									
									
									
									
										vendored
									
									
								
							| @ -6,7 +6,6 @@ | ||||
|         "authn", | ||||
|         "entra", | ||||
|         "goauthentik", | ||||
|         "jwe", | ||||
|         "jwks", | ||||
|         "kubernetes", | ||||
|         "oidc", | ||||
|  | ||||
							
								
								
									
										32
									
								
								Dockerfile
									
									
									
									
									
								
							
							
						
						
									
										32
									
								
								Dockerfile
									
									
									
									
									
								
							| @ -1,7 +1,7 @@ | ||||
| # syntax=docker/dockerfile:1 | ||||
|  | ||||
| # Stage 1: Build website | ||||
| FROM --platform=${BUILDPLATFORM} docker.io/library/node:22 AS website-builder | ||||
| FROM --platform=${BUILDPLATFORM} docker.io/library/node:22 as website-builder | ||||
|  | ||||
| ENV NODE_ENV=production | ||||
|  | ||||
| @ -20,7 +20,7 @@ COPY ./SECURITY.md /work/ | ||||
| RUN npm run build-bundled | ||||
|  | ||||
| # Stage 2: Build webui | ||||
| FROM --platform=${BUILDPLATFORM} docker.io/library/node:22 AS web-builder | ||||
| FROM --platform=${BUILDPLATFORM} docker.io/library/node:22 as web-builder | ||||
|  | ||||
| ARG GIT_BUILD_HASH | ||||
| ENV GIT_BUILD_HASH=$GIT_BUILD_HASH | ||||
| @ -43,7 +43,7 @@ COPY ./gen-ts-api /work/web/node_modules/@goauthentik/api | ||||
| RUN npm run build | ||||
|  | ||||
| # Stage 3: Build go proxy | ||||
| FROM --platform=${BUILDPLATFORM} mcr.microsoft.com/oss/go/microsoft/golang:1.23-fips-bookworm AS go-builder | ||||
| FROM --platform=${BUILDPLATFORM} mcr.microsoft.com/oss/go/microsoft/golang:1.22-fips-bookworm AS go-builder | ||||
|  | ||||
| ARG TARGETOS | ||||
| ARG TARGETARCH | ||||
| @ -80,7 +80,7 @@ RUN --mount=type=cache,sharing=locked,target=/go/pkg/mod \ | ||||
|     go build -o /go/authentik ./cmd/server | ||||
|  | ||||
| # Stage 4: MaxMind GeoIP | ||||
| FROM --platform=${BUILDPLATFORM} ghcr.io/maxmind/geoipupdate:v7.1.0 AS geoip | ||||
| FROM --platform=${BUILDPLATFORM} ghcr.io/maxmind/geoipupdate:v7.0.1 as geoip | ||||
|  | ||||
| ENV GEOIPUPDATE_EDITION_IDS="GeoLite2-City GeoLite2-ASN" | ||||
| ENV GEOIPUPDATE_VERBOSE="1" | ||||
| @ -94,10 +94,7 @@ RUN --mount=type=secret,id=GEOIPUPDATE_ACCOUNT_ID \ | ||||
|     /bin/sh -c "/usr/bin/entry.sh || echo 'Failed to get GeoIP database, disabling'; exit 0" | ||||
|  | ||||
| # Stage 5: Python dependencies | ||||
| FROM ghcr.io/goauthentik/fips-python:3.12.7-slim-bookworm-fips-full AS python-deps | ||||
|  | ||||
| ARG TARGETARCH | ||||
| ARG TARGETVARIANT | ||||
| FROM ghcr.io/goauthentik/fips-python:3.12.3-slim-bookworm-fips-full AS python-deps | ||||
|  | ||||
| WORKDIR /ak-root/poetry | ||||
|  | ||||
| @ -110,7 +107,7 @@ RUN rm -f /etc/apt/apt.conf.d/docker-clean; echo 'Binary::apt::APT::Keep-Downloa | ||||
| RUN --mount=type=cache,id=apt-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/var/cache/apt \ | ||||
|     apt-get update && \ | ||||
|     # Required for installing pip packages | ||||
|     apt-get install -y --no-install-recommends build-essential pkg-config libpq-dev libkrb5-dev | ||||
|     apt-get install -y --no-install-recommends build-essential pkg-config libpq-dev | ||||
|  | ||||
| RUN --mount=type=bind,target=./pyproject.toml,src=./pyproject.toml \ | ||||
|     --mount=type=bind,target=./poetry.lock,src=./poetry.lock \ | ||||
| @ -124,24 +121,24 @@ RUN --mount=type=bind,target=./pyproject.toml,src=./pyproject.toml \ | ||||
|     pip install --force-reinstall /wheels/*" | ||||
|  | ||||
| # Stage 6: Run | ||||
| FROM ghcr.io/goauthentik/fips-python:3.12.7-slim-bookworm-fips-full AS final-image | ||||
| FROM ghcr.io/goauthentik/fips-python:3.12.3-slim-bookworm-fips-full AS final-image | ||||
|  | ||||
| ARG VERSION | ||||
| ARG GIT_BUILD_HASH | ||||
| ARG VERSION | ||||
| ENV GIT_BUILD_HASH=$GIT_BUILD_HASH | ||||
|  | ||||
| LABEL org.opencontainers.image.url=https://goauthentik.io | ||||
| LABEL org.opencontainers.image.description="goauthentik.io Main server image, see https://goauthentik.io for more info." | ||||
| LABEL org.opencontainers.image.source=https://github.com/goauthentik/authentik | ||||
| LABEL org.opencontainers.image.version=${VERSION} | ||||
| LABEL org.opencontainers.image.revision=${GIT_BUILD_HASH} | ||||
| LABEL org.opencontainers.image.url https://goauthentik.io | ||||
| LABEL org.opencontainers.image.description goauthentik.io Main server image, see https://goauthentik.io for more info. | ||||
| LABEL org.opencontainers.image.source https://github.com/goauthentik/authentik | ||||
| LABEL org.opencontainers.image.version ${VERSION} | ||||
| LABEL org.opencontainers.image.revision ${GIT_BUILD_HASH} | ||||
|  | ||||
| WORKDIR / | ||||
|  | ||||
| # We cannot cache this layer otherwise we'll end up with a bigger image | ||||
| RUN apt-get update && \ | ||||
|     # Required for runtime | ||||
|     apt-get install -y --no-install-recommends libpq5 libmaxminddb0 ca-certificates libkrb5-3 libkadm5clnt-mit12 libkdb5-10 && \ | ||||
|     apt-get install -y --no-install-recommends libpq5 libmaxminddb0 ca-certificates && \ | ||||
|     # Required for bootstrap & healtcheck | ||||
|     apt-get install -y --no-install-recommends runit && \ | ||||
|     apt-get clean && \ | ||||
| @ -161,7 +158,6 @@ COPY ./tests /tests | ||||
| COPY ./manage.py / | ||||
| COPY ./blueprints /blueprints | ||||
| COPY ./lifecycle/ /lifecycle | ||||
| COPY ./authentik/sources/kerberos/krb5.conf /etc/krb5.conf | ||||
| COPY --from=go-builder /go/authentik /bin/authentik | ||||
| COPY --from=python-deps /ak-root/venv /ak-root/venv | ||||
| COPY --from=web-builder /work/web/dist/ /web/dist/ | ||||
|  | ||||
							
								
								
									
										8
									
								
								Makefile
									
									
									
									
									
								
							
							
						
						
									
										8
									
								
								Makefile
									
									
									
									
									
								
							| @ -19,13 +19,14 @@ pg_name := $(shell python -m authentik.lib.config postgresql.name 2>/dev/null) | ||||
| CODESPELL_ARGS = -D - -D .github/codespell-dictionary.txt \ | ||||
| 		-I .github/codespell-words.txt \ | ||||
| 		-S 'web/src/locales/**' \ | ||||
| 		-S 'website/docs/developer-docs/api/reference/**' \ | ||||
| 		-S 'website/developer-docs/api/reference/**' \ | ||||
| 		authentik \ | ||||
| 		internal \ | ||||
| 		cmd \ | ||||
| 		web/src \ | ||||
| 		website/src \ | ||||
| 		website/blog \ | ||||
| 		website/developer-docs \ | ||||
| 		website/docs \ | ||||
| 		website/integrations \ | ||||
| 		website/src | ||||
| @ -42,7 +43,7 @@ help:  ## Show this help | ||||
| 		sort | ||||
| 	@echo "" | ||||
|  | ||||
| go-test: | ||||
| test-go: | ||||
| 	go test -timeout 0 -v -race -cover ./... | ||||
|  | ||||
| test-docker:  ## Run all tests in a docker-compose | ||||
| @ -209,9 +210,6 @@ web: web-lint-fix web-lint web-check-compile  ## Automatically fix formatting is | ||||
| web-install:  ## Install the necessary libraries to build the Authentik UI | ||||
| 	cd web && npm ci | ||||
|  | ||||
| web-test: ## Run tests for the Authentik UI | ||||
| 	cd web && npm run test | ||||
|  | ||||
| web-watch:  ## Build and watch the Authentik UI for changes, updating automatically | ||||
| 	rm -rf web/dist/ | ||||
| 	mkdir web/dist/ | ||||
|  | ||||
| @ -15,9 +15,7 @@ | ||||
|  | ||||
| ## What is authentik? | ||||
|  | ||||
| authentik is an open-source Identity Provider that emphasizes flexibility and versatility, with support for a wide set of protocols. | ||||
|  | ||||
| Our [enterprise offer](https://goauthentik.io/pricing) can also be used as a self-hosted replacement for large-scale deployments of Okta/Auth0, Entra ID, Ping Identity, or other legacy IdPs for employees and B2B2C use. | ||||
| authentik is an open-source Identity Provider that emphasizes flexibility and versatility. It can be seamlessly integrated into existing environments to support new protocols. authentik is also a great solution for implementing sign-up, recovery, and other similar features in your application, saving you the hassle of dealing with them. | ||||
|  | ||||
| ## Installation | ||||
|  | ||||
| @ -34,7 +32,7 @@ For bigger setups, there is a Helm Chart [here](https://github.com/goauthentik/h | ||||
|  | ||||
| ## Development | ||||
|  | ||||
| See [Developer Documentation](https://docs.goauthentik.io/docs/developer-docs/?utm_source=github) | ||||
| See [Developer Documentation](https://goauthentik.io/developer-docs/?utm_source=github) | ||||
|  | ||||
| ## Security | ||||
|  | ||||
|  | ||||
| @ -18,10 +18,10 @@ Even if the issue is not a CVE, we still greatly appreciate your help in hardeni | ||||
|  | ||||
| (.x being the latest patch release for each version) | ||||
|  | ||||
| | Version   | Supported | | ||||
| | --------- | --------- | | ||||
| | 2024.8.x  | ✅        | | ||||
| | 2024.10.x | ✅        | | ||||
| | Version  | Supported | | ||||
| | -------- | --------- | | ||||
| | 2024.4.x | ✅        | | ||||
| | 2024.6.x | ✅        | | ||||
|  | ||||
| ## Reporting a Vulnerability | ||||
|  | ||||
|  | ||||
| @ -2,7 +2,7 @@ | ||||
|  | ||||
| from os import environ | ||||
|  | ||||
| __version__ = "2024.10.4" | ||||
| __version__ = "2024.6.3" | ||||
| ENV_GIT_HASH_KEY = "GIT_BUILD_HASH" | ||||
|  | ||||
|  | ||||
|  | ||||
| @ -73,7 +73,7 @@ class SystemInfoSerializer(PassiveSerializer): | ||||
|             "authentik_version": get_full_version(), | ||||
|             "environment": get_env(), | ||||
|             "openssl_fips_enabled": ( | ||||
|                 backend._fips_enabled if LicenseKey.get_total().status().is_valid else None | ||||
|                 backend._fips_enabled if LicenseKey.get_total().is_valid() else None | ||||
|             ), | ||||
|             "openssl_version": OPENSSL_VERSION, | ||||
|             "platform": platform.platform(), | ||||
|  | ||||
| @ -12,7 +12,6 @@ from rest_framework.views import APIView | ||||
| from authentik import __version__, get_build_hash | ||||
| from authentik.admin.tasks import VERSION_CACHE_KEY, VERSION_NULL, update_latest_version | ||||
| from authentik.core.api.utils import PassiveSerializer | ||||
| from authentik.outposts.models import Outpost | ||||
|  | ||||
|  | ||||
| class VersionSerializer(PassiveSerializer): | ||||
| @ -23,7 +22,6 @@ class VersionSerializer(PassiveSerializer): | ||||
|     version_latest_valid = SerializerMethodField() | ||||
|     build_hash = SerializerMethodField() | ||||
|     outdated = SerializerMethodField() | ||||
|     outpost_outdated = SerializerMethodField() | ||||
|  | ||||
|     def get_build_hash(self, _) -> str: | ||||
|         """Get build hash, if version is not latest or released""" | ||||
| @ -49,15 +47,6 @@ class VersionSerializer(PassiveSerializer): | ||||
|         """Check if we're running the latest version""" | ||||
|         return parse(self.get_version_current(instance)) < parse(self.get_version_latest(instance)) | ||||
|  | ||||
|     def get_outpost_outdated(self, _) -> bool: | ||||
|         """Check if any outpost is outdated/has a version mismatch""" | ||||
|         any_outdated = False | ||||
|         for outpost in Outpost.objects.all(): | ||||
|             for state in outpost.state: | ||||
|                 if state.version_outdated: | ||||
|                     any_outdated = True | ||||
|         return any_outdated | ||||
|  | ||||
|  | ||||
| class VersionView(APIView): | ||||
|     """Get running and latest version.""" | ||||
|  | ||||
| @ -1,33 +0,0 @@ | ||||
| from rest_framework.permissions import IsAdminUser | ||||
| from rest_framework.viewsets import ReadOnlyModelViewSet | ||||
|  | ||||
| from authentik.admin.models import VersionHistory | ||||
| from authentik.core.api.utils import ModelSerializer | ||||
|  | ||||
|  | ||||
| class VersionHistorySerializer(ModelSerializer): | ||||
|     """VersionHistory Serializer""" | ||||
|  | ||||
|     class Meta: | ||||
|         model = VersionHistory | ||||
|         fields = [ | ||||
|             "id", | ||||
|             "timestamp", | ||||
|             "version", | ||||
|             "build", | ||||
|         ] | ||||
|  | ||||
|  | ||||
| class VersionHistoryViewSet(ReadOnlyModelViewSet): | ||||
|     """VersionHistory Viewset""" | ||||
|  | ||||
|     queryset = VersionHistory.objects.all() | ||||
|     serializer_class = VersionHistorySerializer | ||||
|     permission_classes = [IsAdminUser] | ||||
|     filterset_fields = [ | ||||
|         "version", | ||||
|         "build", | ||||
|     ] | ||||
|     search_fields = ["version", "build"] | ||||
|     ordering = ["-timestamp"] | ||||
|     pagination_class = None | ||||
| @ -1,22 +0,0 @@ | ||||
| """authentik admin models""" | ||||
|  | ||||
| from django.db import models | ||||
| from django.utils.translation import gettext_lazy as _ | ||||
|  | ||||
|  | ||||
| class VersionHistory(models.Model): | ||||
|     id = models.BigAutoField(primary_key=True) | ||||
|     timestamp = models.DateTimeField() | ||||
|     version = models.TextField() | ||||
|     build = models.TextField() | ||||
|  | ||||
|     class Meta: | ||||
|         managed = False | ||||
|         db_table = "authentik_version_history" | ||||
|         ordering = ("-timestamp",) | ||||
|         verbose_name = _("Version history") | ||||
|         verbose_name_plural = _("Version history") | ||||
|         default_permissions = [] | ||||
|  | ||||
|     def __str__(self): | ||||
|         return f"{self.version}.{self.build} ({self.timestamp})" | ||||
| @ -1,8 +1,10 @@ | ||||
| """authentik admin tasks""" | ||||
|  | ||||
| import re | ||||
|  | ||||
| from django.core.cache import cache | ||||
| from django.core.validators import URLValidator | ||||
| from django.db import DatabaseError, InternalError, ProgrammingError | ||||
| from django.utils.translation import gettext_lazy as _ | ||||
| from packaging.version import parse | ||||
| from requests import RequestException | ||||
| from structlog.stdlib import get_logger | ||||
| @ -19,6 +21,8 @@ LOGGER = get_logger() | ||||
| VERSION_NULL = "0.0.0" | ||||
| VERSION_CACHE_KEY = "authentik_latest_version" | ||||
| VERSION_CACHE_TIMEOUT = 8 * 60 * 60  # 8 hours | ||||
| # Chop of the first ^ because we want to search the entire string | ||||
| URL_FINDER = URLValidator.regex.pattern[1:] | ||||
| LOCAL_VERSION = parse(__version__) | ||||
|  | ||||
|  | ||||
| @ -74,16 +78,10 @@ def update_latest_version(self: SystemTask): | ||||
|                 context__new_version=upstream_version, | ||||
|             ).exists(): | ||||
|                 return | ||||
|             Event.new( | ||||
|                 EventAction.UPDATE_AVAILABLE, | ||||
|                 message=_( | ||||
|                     "New version {version} available!".format( | ||||
|                         version=upstream_version, | ||||
|                     ) | ||||
|                 ), | ||||
|                 new_version=upstream_version, | ||||
|                 changelog=data.get("stable", {}).get("changelog_url"), | ||||
|             ).save() | ||||
|             event_dict = {"new_version": upstream_version} | ||||
|             if match := re.search(URL_FINDER, data.get("stable", {}).get("changelog", "")): | ||||
|                 event_dict["message"] = f"Changelog: {match.group()}" | ||||
|             Event.new(EventAction.UPDATE_AVAILABLE, **event_dict).save() | ||||
|     except (RequestException, IndexError) as exc: | ||||
|         cache.set(VERSION_CACHE_KEY, VERSION_NULL, VERSION_CACHE_TIMEOUT) | ||||
|         self.set_error(exc) | ||||
|  | ||||
| @ -17,7 +17,6 @@ RESPONSE_VALID = { | ||||
|     "stable": { | ||||
|         "version": "99999999.9999999", | ||||
|         "changelog": "See https://goauthentik.io/test", | ||||
|         "changelog_url": "https://goauthentik.io/test", | ||||
|         "reason": "bugfix", | ||||
|     }, | ||||
| } | ||||
| @ -36,7 +35,7 @@ class TestAdminTasks(TestCase): | ||||
|                 Event.objects.filter( | ||||
|                     action=EventAction.UPDATE_AVAILABLE, | ||||
|                     context__new_version="99999999.9999999", | ||||
|                     context__message="New version 99999999.9999999 available!", | ||||
|                     context__message="Changelog: https://goauthentik.io/test", | ||||
|                 ).exists() | ||||
|             ) | ||||
|             # test that a consecutive check doesn't create a duplicate event | ||||
| @ -46,7 +45,7 @@ class TestAdminTasks(TestCase): | ||||
|                     Event.objects.filter( | ||||
|                         action=EventAction.UPDATE_AVAILABLE, | ||||
|                         context__new_version="99999999.9999999", | ||||
|                         context__message="New version 99999999.9999999 available!", | ||||
|                         context__message="Changelog: https://goauthentik.io/test", | ||||
|                     ) | ||||
|                 ), | ||||
|                 1, | ||||
|  | ||||
| @ -6,7 +6,6 @@ from authentik.admin.api.meta import AppsViewSet, ModelViewSet | ||||
| from authentik.admin.api.metrics import AdministrationMetricsViewSet | ||||
| from authentik.admin.api.system import SystemView | ||||
| from authentik.admin.api.version import VersionView | ||||
| from authentik.admin.api.version_history import VersionHistoryViewSet | ||||
| from authentik.admin.api.workers import WorkerView | ||||
|  | ||||
| api_urlpatterns = [ | ||||
| @ -18,7 +17,6 @@ api_urlpatterns = [ | ||||
|         name="admin_metrics", | ||||
|     ), | ||||
|     path("admin/version/", VersionView.as_view(), name="admin_version"), | ||||
|     ("admin/version/history", VersionHistoryViewSet, "version_history"), | ||||
|     path("admin/workers/", WorkerView.as_view(), name="admin_workers"), | ||||
|     path("admin/system/", SystemView.as_view(), name="admin_system"), | ||||
| ] | ||||
|  | ||||
| @ -7,7 +7,7 @@ API Browser - {{ brand.branding_title }} | ||||
| {% endblock %} | ||||
|  | ||||
| {% block head %} | ||||
| <script src="{% versioned_script 'dist/standalone/api-browser/index-%v.js' %}" type="module"></script> | ||||
| {% versioned_script "dist/standalone/api-browser/index-%v.js" %} | ||||
| <meta name="theme-color" content="#151515" media="(prefers-color-scheme: light)"> | ||||
| <meta name="theme-color" content="#151515" media="(prefers-color-scheme: dark)"> | ||||
| {% endblock %} | ||||
|  | ||||
| @ -51,11 +51,9 @@ class BlueprintInstanceSerializer(ModelSerializer): | ||||
|         context = self.instance.context if self.instance else {} | ||||
|         valid, logs = Importer.from_string(content, context).validate() | ||||
|         if not valid: | ||||
|             text_logs = "\n".join([x["event"] for x in logs]) | ||||
|             raise ValidationError( | ||||
|                 [ | ||||
|                     _("Failed to validate blueprint"), | ||||
|                     *[f"- {x.event}" for x in logs], | ||||
|                 ] | ||||
|                 _("Failed to validate blueprint: {logs}".format_map({"logs": text_logs})) | ||||
|             ) | ||||
|         return content | ||||
|  | ||||
|  | ||||
| @ -29,7 +29,9 @@ def check_blueprint_v1_file(BlueprintInstance: type, db_alias, path: Path): | ||||
|         if version != 1: | ||||
|             return | ||||
|         blueprint_file.seek(0) | ||||
|     instance = BlueprintInstance.objects.using(db_alias).filter(path=path).first() | ||||
|     instance: BlueprintInstance = ( | ||||
|         BlueprintInstance.objects.using(db_alias).filter(path=path).first() | ||||
|     ) | ||||
|     rel_path = path.relative_to(Path(CONFIG.get("blueprints_dir"))) | ||||
|     meta = None | ||||
|     if metadata: | ||||
|  | ||||
| @ -27,8 +27,7 @@ def blueprint_tester(file_name: Path) -> Callable: | ||||
|         base = Path("blueprints/") | ||||
|         rel_path = Path(file_name).relative_to(base) | ||||
|         importer = Importer.from_string(BlueprintInstance(path=str(rel_path)).retrieve()) | ||||
|         validation, logs = importer.validate() | ||||
|         self.assertTrue(validation, logs) | ||||
|         self.assertTrue(importer.validate()[0]) | ||||
|         self.assertTrue(importer.apply()) | ||||
|  | ||||
|     return tester | ||||
|  | ||||
| @ -78,5 +78,5 @@ class TestBlueprintsV1API(APITestCase): | ||||
|         self.assertEqual(res.status_code, 400) | ||||
|         self.assertJSONEqual( | ||||
|             res.content.decode(), | ||||
|             {"content": ["Failed to validate blueprint", "- Invalid blueprint version"]}, | ||||
|             {"content": ["Failed to validate blueprint: Invalid blueprint version"]}, | ||||
|         ) | ||||
|  | ||||
| @ -51,10 +51,6 @@ from authentik.enterprise.providers.microsoft_entra.models import ( | ||||
|     MicrosoftEntraProviderUser, | ||||
| ) | ||||
| from authentik.enterprise.providers.rac.models import ConnectionToken | ||||
| from authentik.enterprise.stages.authenticator_endpoint_gdtc.models import ( | ||||
|     EndpointDevice, | ||||
|     EndpointDeviceConnection, | ||||
| ) | ||||
| from authentik.events.logs import LogEvent, capture_logs | ||||
| from authentik.events.models import SystemTask | ||||
| from authentik.events.utils import cleanse_dict | ||||
| @ -73,7 +69,7 @@ from authentik.stages.authenticator_webauthn.models import WebAuthnDeviceType | ||||
| from authentik.tenants.models import Tenant | ||||
|  | ||||
| # Context set when the serializer is created in a blueprint context | ||||
| # Update website/docs/customize/blueprints/v1/models.md when used | ||||
| # Update website/developer-docs/blueprints/v1/models.md when used | ||||
| SERIALIZER_CONTEXT_BLUEPRINT = "blueprint_entry" | ||||
|  | ||||
|  | ||||
| @ -123,8 +119,6 @@ def excluded_models() -> list[type[Model]]: | ||||
|         GoogleWorkspaceProviderGroup, | ||||
|         MicrosoftEntraProviderUser, | ||||
|         MicrosoftEntraProviderGroup, | ||||
|         EndpointDevice, | ||||
|         EndpointDeviceConnection, | ||||
|     ) | ||||
|  | ||||
|  | ||||
| @ -177,7 +171,7 @@ class Importer: | ||||
|     def default_context(self): | ||||
|         """Default context""" | ||||
|         return { | ||||
|             "goauthentik.io/enterprise/licensed": LicenseKey.get_total().status().is_valid, | ||||
|             "goauthentik.io/enterprise/licensed": LicenseKey.get_total().is_valid(), | ||||
|             "goauthentik.io/rbac/models": rbac_models(), | ||||
|         } | ||||
|  | ||||
| @ -293,11 +287,7 @@ class Importer: | ||||
|  | ||||
|         serializer_kwargs = {} | ||||
|         model_instance = existing_models.first() | ||||
|         if ( | ||||
|             not isinstance(model(), BaseMetaModel) | ||||
|             and model_instance | ||||
|             and entry.state != BlueprintEntryDesiredState.MUST_CREATED | ||||
|         ): | ||||
|         if not isinstance(model(), BaseMetaModel) and model_instance: | ||||
|             self.logger.debug( | ||||
|                 "Initialise serializer with instance", | ||||
|                 model=model, | ||||
| @ -307,12 +297,11 @@ class Importer: | ||||
|             serializer_kwargs["instance"] = model_instance | ||||
|             serializer_kwargs["partial"] = True | ||||
|         elif model_instance and entry.state == BlueprintEntryDesiredState.MUST_CREATED: | ||||
|             msg = ( | ||||
|                 f"State is set to {BlueprintEntryDesiredState.MUST_CREATED.value} " | ||||
|                 "and object exists already", | ||||
|             ) | ||||
|             raise EntryInvalidError.from_entry( | ||||
|                 ValidationError({k: msg for k in entry.identifiers.keys()}, "unique"), | ||||
|                 ( | ||||
|                     f"State is set to {BlueprintEntryDesiredState.MUST_CREATED} " | ||||
|                     "and object exists already", | ||||
|                 ), | ||||
|                 entry, | ||||
|             ) | ||||
|         else: | ||||
| @ -440,7 +429,7 @@ class Importer: | ||||
|         orig_import = deepcopy(self._import) | ||||
|         if self._import.version != 1: | ||||
|             self.logger.warning("Invalid blueprint version") | ||||
|             return False, [LogEvent("Invalid blueprint version", log_level="warning", logger=None)] | ||||
|             return False, [{"event": "Invalid blueprint version"}] | ||||
|         with ( | ||||
|             transaction_rollback(), | ||||
|             capture_logs() as logs, | ||||
|  | ||||
| @ -4,7 +4,7 @@ from collections.abc import Callable | ||||
|  | ||||
| from django.http.request import HttpRequest | ||||
| from django.http.response import HttpResponse | ||||
| from django.utils.translation import override | ||||
| from django.utils.translation import activate | ||||
|  | ||||
| from authentik.brands.utils import get_brand_for_request | ||||
|  | ||||
| @ -18,12 +18,10 @@ class BrandMiddleware: | ||||
|         self.get_response = get_response | ||||
|  | ||||
|     def __call__(self, request: HttpRequest) -> HttpResponse: | ||||
|         locale_to_set = None | ||||
|         if not hasattr(request, "brand"): | ||||
|             brand = get_brand_for_request(request) | ||||
|             request.brand = brand | ||||
|             locale = brand.default_locale | ||||
|             if locale != "": | ||||
|                 locale_to_set = locale | ||||
|         with override(locale_to_set): | ||||
|             return self.get_response(request) | ||||
|                 activate(locale) | ||||
|         return self.get_response(request) | ||||
|  | ||||
| @ -1,55 +1,39 @@ | ||||
| """Authenticator Devices API Views""" | ||||
|  | ||||
| from django.utils.translation import gettext_lazy as _ | ||||
| from drf_spectacular.types import OpenApiTypes | ||||
| from drf_spectacular.utils import OpenApiParameter, extend_schema | ||||
| from rest_framework.fields import ( | ||||
|     BooleanField, | ||||
|     CharField, | ||||
|     DateTimeField, | ||||
|     IntegerField, | ||||
|     SerializerMethodField, | ||||
| ) | ||||
| from rest_framework.permissions import IsAuthenticated | ||||
| from rest_framework.permissions import IsAdminUser, IsAuthenticated | ||||
| from rest_framework.request import Request | ||||
| from rest_framework.response import Response | ||||
| from rest_framework.viewsets import ViewSet | ||||
|  | ||||
| from authentik.core.api.utils import MetaNameSerializer | ||||
| from authentik.enterprise.stages.authenticator_endpoint_gdtc.models import EndpointDevice | ||||
| from authentik.rbac.decorators import permission_required | ||||
| from authentik.stages.authenticator import device_classes, devices_for_user | ||||
| from authentik.stages.authenticator.models import Device | ||||
| from authentik.stages.authenticator_webauthn.models import WebAuthnDevice | ||||
|  | ||||
|  | ||||
| class DeviceSerializer(MetaNameSerializer): | ||||
|     """Serializer for Duo authenticator devices""" | ||||
|  | ||||
|     pk = CharField() | ||||
|     pk = IntegerField() | ||||
|     name = CharField() | ||||
|     type = SerializerMethodField() | ||||
|     confirmed = BooleanField() | ||||
|     created = DateTimeField(read_only=True) | ||||
|     last_updated = DateTimeField(read_only=True) | ||||
|     last_used = DateTimeField(read_only=True, allow_null=True) | ||||
|     extra_description = SerializerMethodField() | ||||
|  | ||||
|     def get_type(self, instance: Device) -> str: | ||||
|         """Get type of device""" | ||||
|         return instance._meta.label | ||||
|  | ||||
|     def get_extra_description(self, instance: Device) -> str: | ||||
|         """Get extra description""" | ||||
|         if isinstance(instance, WebAuthnDevice): | ||||
|             return ( | ||||
|                 instance.device_type.description | ||||
|                 if instance.device_type | ||||
|                 else _("Extra description not available") | ||||
|             ) | ||||
|         if isinstance(instance, EndpointDevice): | ||||
|             return instance.data.get("deviceSignals", {}).get("deviceModel") | ||||
|         return "" | ||||
|  | ||||
|  | ||||
| class DeviceViewSet(ViewSet): | ||||
|     """Viewset for authenticator devices""" | ||||
| @ -68,7 +52,7 @@ class AdminDeviceViewSet(ViewSet): | ||||
|     """Viewset for authenticator devices""" | ||||
|  | ||||
|     serializer_class = DeviceSerializer | ||||
|     permission_classes = [] | ||||
|     permission_classes = [IsAdminUser] | ||||
|  | ||||
|     def get_devices(self, **kwargs): | ||||
|         """Get all devices in all child classes""" | ||||
| @ -86,10 +70,6 @@ class AdminDeviceViewSet(ViewSet): | ||||
|         ], | ||||
|         responses={200: DeviceSerializer(many=True)}, | ||||
|     ) | ||||
|     @permission_required( | ||||
|         None, | ||||
|         [f"{model._meta.app_label}.view_{model._meta.model_name}" for model in device_classes()], | ||||
|     ) | ||||
|     def list(self, request: Request) -> Response: | ||||
|         """Get all devices for current user""" | ||||
|         kwargs = {} | ||||
|  | ||||
| @ -30,10 +30,8 @@ from authentik.core.api.utils import ( | ||||
|     PassiveSerializer, | ||||
| ) | ||||
| from authentik.core.expression.evaluator import PropertyMappingEvaluator | ||||
| from authentik.core.expression.exceptions import PropertyMappingExpressionException | ||||
| from authentik.core.models import Group, PropertyMapping, User | ||||
| from authentik.events.utils import sanitize_item | ||||
| from authentik.lib.utils.errors import exception_to_string | ||||
| from authentik.policies.api.exec import PolicyTestSerializer | ||||
| from authentik.rbac.decorators import permission_required | ||||
|  | ||||
| @ -164,15 +162,12 @@ class PropertyMappingViewSet( | ||||
|  | ||||
|         response_data = {"successful": True, "result": ""} | ||||
|         try: | ||||
|             result = mapping.evaluate(dry_run=True, **context) | ||||
|             result = mapping.evaluate(**context) | ||||
|             response_data["result"] = dumps( | ||||
|                 sanitize_item(result), indent=(4 if format_result else None) | ||||
|             ) | ||||
|         except PropertyMappingExpressionException as exc: | ||||
|             response_data["result"] = exception_to_string(exc.exc) | ||||
|             response_data["successful"] = False | ||||
|         except Exception as exc: | ||||
|             response_data["result"] = exception_to_string(exc) | ||||
|             response_data["result"] = str(exc) | ||||
|             response_data["successful"] = False | ||||
|         response = PropertyMappingTestResultSerializer(response_data) | ||||
|         return Response(response.data) | ||||
|  | ||||
| @ -38,7 +38,6 @@ class ProviderSerializer(ModelSerializer, MetaNameSerializer): | ||||
|             "name", | ||||
|             "authentication_flow", | ||||
|             "authorization_flow", | ||||
|             "invalidation_flow", | ||||
|             "property_mappings", | ||||
|             "component", | ||||
|             "assigned_application_slug", | ||||
| @ -51,7 +50,6 @@ class ProviderSerializer(ModelSerializer, MetaNameSerializer): | ||||
|         ] | ||||
|         extra_kwargs = { | ||||
|             "authorization_flow": {"required": True, "allow_null": False}, | ||||
|             "invalidation_flow": {"required": True, "allow_null": False}, | ||||
|         } | ||||
|  | ||||
|  | ||||
|  | ||||
| @ -1,12 +1,10 @@ | ||||
| """transactional application and provider creation""" | ||||
|  | ||||
| from django.apps import apps | ||||
| from django.db.models import Model | ||||
| from django.utils.translation import gettext as _ | ||||
| from drf_spectacular.utils import PolymorphicProxySerializer, extend_schema, extend_schema_field | ||||
| from rest_framework.exceptions import PermissionDenied, ValidationError | ||||
| from rest_framework.exceptions import ValidationError | ||||
| from rest_framework.fields import BooleanField, CharField, ChoiceField, DictField, ListField | ||||
| from rest_framework.permissions import IsAuthenticated | ||||
| from rest_framework.permissions import IsAdminUser | ||||
| from rest_framework.request import Request | ||||
| from rest_framework.response import Response | ||||
| from rest_framework.views import APIView | ||||
| @ -24,7 +22,6 @@ from authentik.core.api.applications import ApplicationSerializer | ||||
| from authentik.core.api.utils import PassiveSerializer | ||||
| from authentik.core.models import Provider | ||||
| from authentik.lib.utils.reflection import all_subclasses | ||||
| from authentik.policies.api.bindings import PolicyBindingSerializer | ||||
|  | ||||
|  | ||||
| def get_provider_serializer_mapping(): | ||||
| @ -48,13 +45,6 @@ class TransactionProviderField(DictField): | ||||
|     """Dictionary field which can hold provider creation data""" | ||||
|  | ||||
|  | ||||
| class TransactionPolicyBindingSerializer(PolicyBindingSerializer): | ||||
|     """PolicyBindingSerializer which does not require target as target is set implicitly""" | ||||
|  | ||||
|     class Meta(PolicyBindingSerializer.Meta): | ||||
|         fields = [x for x in PolicyBindingSerializer.Meta.fields if x != "target"] | ||||
|  | ||||
|  | ||||
| class TransactionApplicationSerializer(PassiveSerializer): | ||||
|     """Serializer for creating a provider and an application in one transaction""" | ||||
|  | ||||
| @ -62,8 +52,6 @@ class TransactionApplicationSerializer(PassiveSerializer): | ||||
|     provider_model = ChoiceField(choices=list(get_provider_serializer_mapping().keys())) | ||||
|     provider = TransactionProviderField() | ||||
|  | ||||
|     policy_bindings = TransactionPolicyBindingSerializer(many=True, required=False) | ||||
|  | ||||
|     _provider_model: type[Provider] = None | ||||
|  | ||||
|     def validate_provider_model(self, fq_model_name: str) -> str: | ||||
| @ -108,19 +96,6 @@ class TransactionApplicationSerializer(PassiveSerializer): | ||||
|                 id="app", | ||||
|             ) | ||||
|         ) | ||||
|         for binding in attrs.get("policy_bindings", []): | ||||
|             binding["target"] = KeyOf(None, ScalarNode(tag="", value="app")) | ||||
|             for key, value in binding.items(): | ||||
|                 if not isinstance(value, Model): | ||||
|                     continue | ||||
|                 binding[key] = value.pk | ||||
|             blueprint.entries.append( | ||||
|                 BlueprintEntry( | ||||
|                     model="authentik_policies.policybinding", | ||||
|                     state=BlueprintEntryDesiredState.MUST_CREATED, | ||||
|                     identifiers=binding, | ||||
|                 ) | ||||
|             ) | ||||
|         importer = Importer(blueprint, {}) | ||||
|         try: | ||||
|             valid, _ = importer.validate(raise_validation_errors=True) | ||||
| @ -145,7 +120,8 @@ class TransactionApplicationResponseSerializer(PassiveSerializer): | ||||
| class TransactionalApplicationView(APIView): | ||||
|     """Create provider and application and attach them in a single transaction""" | ||||
|  | ||||
|     permission_classes = [IsAuthenticated] | ||||
|     # TODO: Migrate to a more specific permission | ||||
|     permission_classes = [IsAdminUser] | ||||
|  | ||||
|     @extend_schema( | ||||
|         request=TransactionApplicationSerializer(), | ||||
| @ -157,23 +133,8 @@ class TransactionalApplicationView(APIView): | ||||
|         """Convert data into a blueprint, validate it and apply it""" | ||||
|         data = TransactionApplicationSerializer(data=request.data) | ||||
|         data.is_valid(raise_exception=True) | ||||
|         blueprint: Blueprint = data.validated_data | ||||
|         for entry in blueprint.entries: | ||||
|             full_model = entry.get_model(blueprint) | ||||
|             app, __, model = full_model.partition(".") | ||||
|             if not request.user.has_perm(f"{app}.add_{model}"): | ||||
|                 raise PermissionDenied( | ||||
|                     { | ||||
|                         entry.id: _( | ||||
|                             "User lacks permission to create {model}".format_map( | ||||
|                                 { | ||||
|                                     "model": full_model, | ||||
|                                 } | ||||
|                             ) | ||||
|                         ) | ||||
|                     } | ||||
|                 ) | ||||
|         importer = Importer(blueprint, {}) | ||||
|  | ||||
|         importer = Importer(data.validated_data, {}) | ||||
|         applied = importer.apply() | ||||
|         response = {"applied": False, "logs": []} | ||||
|         response["applied"] = applied | ||||
|  | ||||
| @ -14,7 +14,6 @@ from rest_framework.request import Request | ||||
| from rest_framework.response import Response | ||||
|  | ||||
| from authentik.core.api.utils import PassiveSerializer | ||||
| from authentik.rbac.filters import ObjectFilter | ||||
|  | ||||
|  | ||||
| class DeleteAction(Enum): | ||||
| @ -54,7 +53,7 @@ class UsedByMixin: | ||||
|     @extend_schema( | ||||
|         responses={200: UsedBySerializer(many=True)}, | ||||
|     ) | ||||
|     @action(detail=True, pagination_class=None, filter_backends=[ObjectFilter]) | ||||
|     @action(detail=True, pagination_class=None, filter_backends=[]) | ||||
|     def used_by(self, request: Request, *args, **kwargs) -> Response: | ||||
|         """Get a list of all objects that use this object""" | ||||
|         model: Model = self.get_object() | ||||
|  | ||||
| @ -666,12 +666,7 @@ class UserViewSet(UsedByMixin, ModelViewSet): | ||||
|  | ||||
|     @permission_required("authentik_core.impersonate") | ||||
|     @extend_schema( | ||||
|         request=inline_serializer( | ||||
|             "ImpersonationSerializer", | ||||
|             { | ||||
|                 "reason": CharField(required=True), | ||||
|             }, | ||||
|         ), | ||||
|         request=OpenApiTypes.NONE, | ||||
|         responses={ | ||||
|             "204": OpenApiResponse(description="Successfully started impersonation"), | ||||
|             "401": OpenApiResponse(description="Access denied"), | ||||
| @ -683,27 +678,18 @@ class UserViewSet(UsedByMixin, ModelViewSet): | ||||
|         if not request.tenant.impersonation: | ||||
|             LOGGER.debug("User attempted to impersonate", user=request.user) | ||||
|             return Response(status=401) | ||||
|         user_to_be = self.get_object() | ||||
|         reason = request.data.get("reason", "") | ||||
|         # Check both object-level perms and global perms | ||||
|         if not request.user.has_perm( | ||||
|             "authentik_core.impersonate", user_to_be | ||||
|         ) and not request.user.has_perm("authentik_core.impersonate"): | ||||
|         if not request.user.has_perm("impersonate"): | ||||
|             LOGGER.debug("User attempted to impersonate without permissions", user=request.user) | ||||
|             return Response(status=401) | ||||
|         user_to_be = self.get_object() | ||||
|         if user_to_be.pk == self.request.user.pk: | ||||
|             LOGGER.debug("User attempted to impersonate themselves", user=request.user) | ||||
|             return Response(status=401) | ||||
|         if not reason and request.tenant.impersonation_require_reason: | ||||
|             LOGGER.debug( | ||||
|                 "User attempted to impersonate without providing a reason", user=request.user | ||||
|             ) | ||||
|             return Response(status=401) | ||||
|  | ||||
|         request.session[SESSION_KEY_IMPERSONATE_ORIGINAL_USER] = request.user | ||||
|         request.session[SESSION_KEY_IMPERSONATE_USER] = user_to_be | ||||
|  | ||||
|         Event.new(EventAction.IMPERSONATION_STARTED, reason=reason).from_http(request, user_to_be) | ||||
|         Event.new(EventAction.IMPERSONATION_STARTED).from_http(request, user_to_be) | ||||
|  | ||||
|         return Response(status=201) | ||||
|  | ||||
|  | ||||
| @ -9,11 +9,10 @@ class Command(TenantCommand): | ||||
|  | ||||
|     def add_arguments(self, parser): | ||||
|         parser.add_argument("--type", type=str, required=True) | ||||
|         parser.add_argument("--all", action="store_true", default=False) | ||||
|         parser.add_argument("usernames", nargs="*", type=str) | ||||
|         parser.add_argument("--all", action="store_true") | ||||
|         parser.add_argument("usernames", nargs="+", type=str) | ||||
|  | ||||
|     def handle_per_tenant(self, **options): | ||||
|         print(options) | ||||
|         new_type = UserTypes(options["type"]) | ||||
|         qs = ( | ||||
|             User.objects.exclude_anonymous() | ||||
| @ -23,9 +22,6 @@ class Command(TenantCommand): | ||||
|         if options["usernames"] and options["all"]: | ||||
|             self.stderr.write("--all and usernames specified, only one can be specified") | ||||
|             return | ||||
|         if not options["usernames"] and not options["all"]: | ||||
|             self.stderr.write("--all or usernames must be specified") | ||||
|             return | ||||
|         if options["usernames"] and not options["all"]: | ||||
|             qs = qs.filter(username__in=options["usernames"]) | ||||
|         updated = qs.update(type=new_type) | ||||
|  | ||||
| @ -4,7 +4,6 @@ import code | ||||
| import platform | ||||
| import sys | ||||
| import traceback | ||||
| from pprint import pprint | ||||
|  | ||||
| from django.apps import apps | ||||
| from django.core.management.base import BaseCommand | ||||
| @ -35,9 +34,7 @@ class Command(BaseCommand): | ||||
|  | ||||
|     def get_namespace(self): | ||||
|         """Prepare namespace with all models""" | ||||
|         namespace = { | ||||
|             "pprint": pprint, | ||||
|         } | ||||
|         namespace = {} | ||||
|  | ||||
|         # Gather Django models and constants from each app | ||||
|         for app in apps.get_app_configs(): | ||||
|  | ||||
| @ -5,7 +5,7 @@ from contextvars import ContextVar | ||||
| from uuid import uuid4 | ||||
|  | ||||
| from django.http import HttpRequest, HttpResponse | ||||
| from django.utils.translation import override | ||||
| from django.utils.translation import activate | ||||
| from sentry_sdk.api import set_tag | ||||
| from structlog.contextvars import STRUCTLOG_KEY_PREFIX | ||||
|  | ||||
| @ -31,19 +31,17 @@ class ImpersonateMiddleware: | ||||
|     def __call__(self, request: HttpRequest) -> HttpResponse: | ||||
|         # No permission checks are done here, they need to be checked before | ||||
|         # SESSION_KEY_IMPERSONATE_USER is set. | ||||
|         locale_to_set = None | ||||
|         if request.user.is_authenticated: | ||||
|             locale = request.user.locale(request) | ||||
|             if locale != "": | ||||
|                 locale_to_set = locale | ||||
|                 activate(locale) | ||||
|  | ||||
|         if SESSION_KEY_IMPERSONATE_USER in request.session: | ||||
|             request.user = request.session[SESSION_KEY_IMPERSONATE_USER] | ||||
|             # Ensure that the user is active, otherwise nothing will work | ||||
|             request.user.is_active = True | ||||
|  | ||||
|         with override(locale_to_set): | ||||
|             return self.get_response(request) | ||||
|         return self.get_response(request) | ||||
|  | ||||
|  | ||||
| class RequestIDMiddleware: | ||||
|  | ||||
| @ -1,55 +0,0 @@ | ||||
| # Generated by Django 5.0.9 on 2024-10-02 11:35 | ||||
|  | ||||
| import django.db.models.deletion | ||||
| from django.db import migrations, models | ||||
|  | ||||
| from django.apps.registry import Apps | ||||
| from django.db import migrations, models | ||||
| from django.db.backends.base.schema import BaseDatabaseSchemaEditor | ||||
|  | ||||
|  | ||||
| def migrate_invalidation_flow_default(apps: Apps, schema_editor: BaseDatabaseSchemaEditor): | ||||
|     from authentik.flows.models import FlowDesignation, FlowAuthenticationRequirement | ||||
|  | ||||
|     db_alias = schema_editor.connection.alias | ||||
|  | ||||
|     Flow = apps.get_model("authentik_flows", "Flow") | ||||
|     Provider = apps.get_model("authentik_core", "Provider") | ||||
|  | ||||
|     # So this flow is managed via a blueprint, bue we're in a migration so we don't want to rely on that | ||||
|     # since the blueprint is just an empty flow we can just create it here | ||||
|     # and let it be managed by the blueprint later | ||||
|     flow, _ = Flow.objects.using(db_alias).update_or_create( | ||||
|         slug="default-provider-invalidation-flow", | ||||
|         defaults={ | ||||
|             "name": "Logged out of application", | ||||
|             "title": "You've logged out of %(app)s.", | ||||
|             "authentication": FlowAuthenticationRequirement.NONE, | ||||
|             "designation": FlowDesignation.INVALIDATION, | ||||
|         }, | ||||
|     ) | ||||
|     Provider.objects.using(db_alias).filter(invalidation_flow=None).update(invalidation_flow=flow) | ||||
|  | ||||
|  | ||||
| class Migration(migrations.Migration): | ||||
|  | ||||
|     dependencies = [ | ||||
|         ("authentik_core", "0039_source_group_matching_mode_alter_group_name_and_more"), | ||||
|         ("authentik_flows", "0027_auto_20231028_1424"), | ||||
|     ] | ||||
|  | ||||
|     operations = [ | ||||
|         migrations.AddField( | ||||
|             model_name="provider", | ||||
|             name="invalidation_flow", | ||||
|             field=models.ForeignKey( | ||||
|                 default=None, | ||||
|                 help_text="Flow used ending the session from a provider.", | ||||
|                 null=True, | ||||
|                 on_delete=django.db.models.deletion.SET_DEFAULT, | ||||
|                 related_name="provider_invalidation", | ||||
|                 to="authentik_flows.flow", | ||||
|             ), | ||||
|         ), | ||||
|         migrations.RunPython(migrate_invalidation_flow_default), | ||||
|     ] | ||||
| @ -330,13 +330,11 @@ class User(SerializerModel, GuardianUserMixin, AttributesMixin, AbstractUser): | ||||
|         """superuser == staff user""" | ||||
|         return self.is_superuser  # type: ignore | ||||
|  | ||||
|     def set_password(self, raw_password, signal=True, sender=None): | ||||
|     def set_password(self, raw_password, signal=True): | ||||
|         if self.pk and signal: | ||||
|             from authentik.core.signals import password_changed | ||||
|  | ||||
|             if not sender: | ||||
|                 sender = self | ||||
|             password_changed.send(sender=sender, user=self, password=raw_password) | ||||
|             password_changed.send(sender=self, user=self, password=raw_password) | ||||
|         self.password_change_date = now() | ||||
|         return super().set_password(raw_password) | ||||
|  | ||||
| @ -393,23 +391,14 @@ class Provider(SerializerModel): | ||||
|         ), | ||||
|         related_name="provider_authentication", | ||||
|     ) | ||||
|  | ||||
|     authorization_flow = models.ForeignKey( | ||||
|         "authentik_flows.Flow", | ||||
|         # Set to cascade even though null is allowed, since most providers | ||||
|         # still require an authorization flow set | ||||
|         on_delete=models.CASCADE, | ||||
|         null=True, | ||||
|         help_text=_("Flow used when authorizing this provider."), | ||||
|         related_name="provider_authorization", | ||||
|     ) | ||||
|     invalidation_flow = models.ForeignKey( | ||||
|         "authentik_flows.Flow", | ||||
|         on_delete=models.SET_DEFAULT, | ||||
|         default=None, | ||||
|         null=True, | ||||
|         help_text=_("Flow used ending the session from a provider."), | ||||
|         related_name="provider_invalidation", | ||||
|     ) | ||||
|  | ||||
|     property_mappings = models.ManyToManyField("PropertyMapping", default=None, blank=True) | ||||
|  | ||||
| @ -477,6 +466,8 @@ class ApplicationQuerySet(QuerySet): | ||||
|     def with_provider(self) -> "QuerySet[Application]": | ||||
|         qs = self.select_related("provider") | ||||
|         for subclass in Provider.objects.get_queryset()._get_subclasses_recurse(Provider): | ||||
|             if LOOKUP_SEP in subclass: | ||||
|                 continue | ||||
|             qs = qs.select_related(f"provider__{subclass}") | ||||
|         return qs | ||||
|  | ||||
| @ -554,24 +545,15 @@ class Application(SerializerModel, PolicyBindingModel): | ||||
|         if not self.provider: | ||||
|             return None | ||||
|  | ||||
|         candidates = [] | ||||
|         base_class = Provider | ||||
|         for subclass in base_class.objects.get_queryset()._get_subclasses_recurse(base_class): | ||||
|             parent = self.provider | ||||
|             for level in subclass.split(LOOKUP_SEP): | ||||
|                 try: | ||||
|                     parent = getattr(parent, level) | ||||
|                 except AttributeError: | ||||
|                     break | ||||
|             if parent in candidates: | ||||
|         for subclass in Provider.objects.get_queryset()._get_subclasses_recurse(Provider): | ||||
|             # We don't care about recursion, skip nested models | ||||
|             if LOOKUP_SEP in subclass: | ||||
|                 continue | ||||
|             idx = subclass.count(LOOKUP_SEP) | ||||
|             if type(parent) is not base_class: | ||||
|                 idx += 1 | ||||
|             candidates.insert(idx, parent) | ||||
|         if not candidates: | ||||
|             return None | ||||
|         return candidates[-1] | ||||
|             try: | ||||
|                 return getattr(self.provider, subclass) | ||||
|             except AttributeError: | ||||
|                 pass | ||||
|         return None | ||||
|  | ||||
|     def __str__(self): | ||||
|         return str(self.name) | ||||
| @ -919,7 +901,7 @@ class PropertyMapping(SerializerModel, ManagedModel): | ||||
|         except ControlFlowException as exc: | ||||
|             raise exc | ||||
|         except Exception as exc: | ||||
|             raise PropertyMappingExpressionException(exc, self) from exc | ||||
|             raise PropertyMappingExpressionException(self, exc) from exc | ||||
|  | ||||
|     def __str__(self): | ||||
|         return f"Property Mapping {self.name}" | ||||
|  | ||||
| @ -1,9 +1,11 @@ | ||||
| """Source decision helper""" | ||||
|  | ||||
| from enum import Enum | ||||
| from typing import Any | ||||
|  | ||||
| from django.contrib import messages | ||||
| from django.db import IntegrityError, transaction | ||||
| from django.db.models.query_utils import Q | ||||
| from django.http import HttpRequest, HttpResponse | ||||
| from django.shortcuts import redirect | ||||
| from django.urls import reverse | ||||
| @ -14,11 +16,12 @@ from authentik.core.models import ( | ||||
|     Group, | ||||
|     GroupSourceConnection, | ||||
|     Source, | ||||
|     SourceGroupMatchingModes, | ||||
|     SourceUserMatchingModes, | ||||
|     User, | ||||
|     UserSourceConnection, | ||||
| ) | ||||
| from authentik.core.sources.mapper import SourceMapper | ||||
| from authentik.core.sources.matcher import Action, SourceMatcher | ||||
| from authentik.core.sources.stage import ( | ||||
|     PLAN_CONTEXT_SOURCES_CONNECTION, | ||||
|     PostSourceStage, | ||||
| @ -51,6 +54,16 @@ SESSION_KEY_OVERRIDE_FLOW_TOKEN = "authentik/flows/source_override_flow_token" | ||||
| PLAN_CONTEXT_SOURCE_GROUPS = "source_groups" | ||||
|  | ||||
|  | ||||
| class Action(Enum): | ||||
|     """Actions that can be decided based on the request | ||||
|     and source settings""" | ||||
|  | ||||
|     LINK = "link" | ||||
|     AUTH = "auth" | ||||
|     ENROLL = "enroll" | ||||
|     DENY = "deny" | ||||
|  | ||||
|  | ||||
| class MessageStage(StageView): | ||||
|     """Show a pre-configured message after the flow is done""" | ||||
|  | ||||
| @ -73,7 +86,6 @@ class SourceFlowManager: | ||||
|  | ||||
|     source: Source | ||||
|     mapper: SourceMapper | ||||
|     matcher: SourceMatcher | ||||
|     request: HttpRequest | ||||
|  | ||||
|     identifier: str | ||||
| @ -96,9 +108,6 @@ class SourceFlowManager: | ||||
|     ) -> None: | ||||
|         self.source = source | ||||
|         self.mapper = SourceMapper(self.source) | ||||
|         self.matcher = SourceMatcher( | ||||
|             self.source, self.user_connection_type, self.group_connection_type | ||||
|         ) | ||||
|         self.request = request | ||||
|         self.identifier = identifier | ||||
|         self.user_info = user_info | ||||
| @ -122,24 +131,66 @@ class SourceFlowManager: | ||||
|  | ||||
|     def get_action(self, **kwargs) -> tuple[Action, UserSourceConnection | None]:  # noqa: PLR0911 | ||||
|         """decide which action should be taken""" | ||||
|         new_connection = self.user_connection_type(source=self.source, identifier=self.identifier) | ||||
|         # When request is authenticated, always link | ||||
|         if self.request.user.is_authenticated: | ||||
|             new_connection = self.user_connection_type( | ||||
|                 source=self.source, identifier=self.identifier | ||||
|             ) | ||||
|             new_connection.user = self.request.user | ||||
|             new_connection = self.update_user_connection(new_connection, **kwargs) | ||||
|             if existing := self.user_connection_type.objects.filter( | ||||
|                 source=self.source, identifier=self.identifier | ||||
|             ).first(): | ||||
|                 existing = self.update_user_connection(existing) | ||||
|                 return Action.AUTH, existing | ||||
|             return Action.LINK, new_connection | ||||
|  | ||||
|         action, connection = self.matcher.get_user_action(self.identifier, self.user_properties) | ||||
|         if connection: | ||||
|             connection = self.update_user_connection(connection, **kwargs) | ||||
|         return action, connection | ||||
|         existing_connections = self.user_connection_type.objects.filter( | ||||
|             source=self.source, identifier=self.identifier | ||||
|         ) | ||||
|         if existing_connections.exists(): | ||||
|             connection = existing_connections.first() | ||||
|             return Action.AUTH, self.update_user_connection(connection, **kwargs) | ||||
|         # No connection exists, but we match on identifier, so enroll | ||||
|         if self.source.user_matching_mode == SourceUserMatchingModes.IDENTIFIER: | ||||
|             # We don't save the connection here cause it doesn't have a user assigned yet | ||||
|             return Action.ENROLL, self.update_user_connection(new_connection, **kwargs) | ||||
|  | ||||
|         # Check for existing users with matching attributes | ||||
|         query = Q() | ||||
|         # Either query existing user based on email or username | ||||
|         if self.source.user_matching_mode in [ | ||||
|             SourceUserMatchingModes.EMAIL_LINK, | ||||
|             SourceUserMatchingModes.EMAIL_DENY, | ||||
|         ]: | ||||
|             if not self.user_properties.get("email", None): | ||||
|                 self._logger.warning("Refusing to use none email") | ||||
|                 return Action.DENY, None | ||||
|             query = Q(email__exact=self.user_properties.get("email", None)) | ||||
|         if self.source.user_matching_mode in [ | ||||
|             SourceUserMatchingModes.USERNAME_LINK, | ||||
|             SourceUserMatchingModes.USERNAME_DENY, | ||||
|         ]: | ||||
|             if not self.user_properties.get("username", None): | ||||
|                 self._logger.warning("Refusing to use none username") | ||||
|                 return Action.DENY, None | ||||
|             query = Q(username__exact=self.user_properties.get("username", None)) | ||||
|         self._logger.debug("trying to link with existing user", query=query) | ||||
|         matching_users = User.objects.filter(query) | ||||
|         # No matching users, always enroll | ||||
|         if not matching_users.exists(): | ||||
|             self._logger.debug("no matching users found, enrolling") | ||||
|             return Action.ENROLL, self.update_user_connection(new_connection, **kwargs) | ||||
|  | ||||
|         user = matching_users.first() | ||||
|         if self.source.user_matching_mode in [ | ||||
|             SourceUserMatchingModes.EMAIL_LINK, | ||||
|             SourceUserMatchingModes.USERNAME_LINK, | ||||
|         ]: | ||||
|             new_connection.user = user | ||||
|             new_connection = self.update_user_connection(new_connection, **kwargs) | ||||
|             return Action.LINK, new_connection | ||||
|         if self.source.user_matching_mode in [ | ||||
|             SourceUserMatchingModes.EMAIL_DENY, | ||||
|             SourceUserMatchingModes.USERNAME_DENY, | ||||
|         ]: | ||||
|             self._logger.info("denying source because user exists", user=user) | ||||
|             return Action.DENY, None | ||||
|         # Should never get here as default enroll case is returned above. | ||||
|         return Action.DENY, None  # pragma: no cover | ||||
|  | ||||
|     def update_user_connection( | ||||
|         self, connection: UserSourceConnection, **kwargs | ||||
| @ -277,6 +328,7 @@ class SourceFlowManager: | ||||
|         connection: UserSourceConnection, | ||||
|     ) -> HttpResponse: | ||||
|         """Login user and redirect.""" | ||||
|         flow_kwargs = {PLAN_CONTEXT_PENDING_USER: connection.user} | ||||
|         return self._prepare_flow( | ||||
|             self.source.authentication_flow, | ||||
|             connection, | ||||
| @ -290,11 +342,7 @@ class SourceFlowManager: | ||||
|                     ), | ||||
|                 ) | ||||
|             ], | ||||
|             **{ | ||||
|                 PLAN_CONTEXT_PENDING_USER: connection.user, | ||||
|                 PLAN_CONTEXT_PROMPT: delete_none_values(self.user_properties), | ||||
|                 PLAN_CONTEXT_USER_PATH: self.source.get_user_path(), | ||||
|             }, | ||||
|             **flow_kwargs, | ||||
|         ) | ||||
|  | ||||
|     def handle_existing_link( | ||||
| @ -360,16 +408,74 @@ class SourceFlowManager: | ||||
| class GroupUpdateStage(StageView): | ||||
|     """Dynamically injected stage which updates the user after enrollment/authentication.""" | ||||
|  | ||||
|     def get_action( | ||||
|         self, group_id: str, group_properties: dict[str, Any | dict[str, Any]] | ||||
|     ) -> tuple[Action, GroupSourceConnection | None]: | ||||
|         """decide which action should be taken""" | ||||
|         new_connection = self.group_connection_type(source=self.source, identifier=group_id) | ||||
|  | ||||
|         existing_connections = self.group_connection_type.objects.filter( | ||||
|             source=self.source, identifier=group_id | ||||
|         ) | ||||
|         if existing_connections.exists(): | ||||
|             return Action.LINK, existing_connections.first() | ||||
|         # No connection exists, but we match on identifier, so enroll | ||||
|         if self.source.group_matching_mode == SourceGroupMatchingModes.IDENTIFIER: | ||||
|             # We don't save the connection here cause it doesn't have a user assigned yet | ||||
|             return Action.ENROLL, new_connection | ||||
|  | ||||
|         # Check for existing groups with matching attributes | ||||
|         query = Q() | ||||
|         if self.source.group_matching_mode in [ | ||||
|             SourceGroupMatchingModes.NAME_LINK, | ||||
|             SourceGroupMatchingModes.NAME_DENY, | ||||
|         ]: | ||||
|             if not group_properties.get("name", None): | ||||
|                 LOGGER.warning( | ||||
|                     "Refusing to use none group name", source=self.source, group_id=group_id | ||||
|                 ) | ||||
|                 return Action.DENY, None | ||||
|             query = Q(name__exact=group_properties.get("name")) | ||||
|         LOGGER.debug( | ||||
|             "trying to link with existing group", source=self.source, query=query, group_id=group_id | ||||
|         ) | ||||
|         matching_groups = Group.objects.filter(query) | ||||
|         # No matching groups, always enroll | ||||
|         if not matching_groups.exists(): | ||||
|             LOGGER.debug( | ||||
|                 "no matching groups found, enrolling", source=self.source, group_id=group_id | ||||
|             ) | ||||
|             return Action.ENROLL, new_connection | ||||
|  | ||||
|         group = matching_groups.first() | ||||
|         if self.source.group_matching_mode in [ | ||||
|             SourceGroupMatchingModes.NAME_LINK, | ||||
|         ]: | ||||
|             new_connection.group = group | ||||
|             return Action.LINK, new_connection | ||||
|         if self.source.group_matching_mode in [ | ||||
|             SourceGroupMatchingModes.NAME_DENY, | ||||
|         ]: | ||||
|             LOGGER.info( | ||||
|                 "denying source because group exists", | ||||
|                 source=self.source, | ||||
|                 group=group, | ||||
|                 group_id=group_id, | ||||
|             ) | ||||
|             return Action.DENY, None | ||||
|         # Should never get here as default enroll case is returned above. | ||||
|         return Action.DENY, None  # pragma: no cover | ||||
|  | ||||
|     def handle_group( | ||||
|         self, group_id: str, group_properties: dict[str, Any | dict[str, Any]] | ||||
|     ) -> Group | None: | ||||
|         action, connection = self.matcher.get_group_action(group_id, group_properties) | ||||
|         action, connection = self.get_action(group_id, group_properties) | ||||
|         if action == Action.ENROLL: | ||||
|             group = Group.objects.create(**group_properties) | ||||
|             connection.group = group | ||||
|             connection.save() | ||||
|             return group | ||||
|         elif action in (Action.LINK, Action.AUTH): | ||||
|         elif action == Action.LINK: | ||||
|             group = connection.group | ||||
|             group.update_attributes(group_properties) | ||||
|             connection.save() | ||||
| @ -383,7 +489,6 @@ class GroupUpdateStage(StageView): | ||||
|         self.group_connection_type: GroupSourceConnection = ( | ||||
|             self.executor.current_stage.group_connection_type | ||||
|         ) | ||||
|         self.matcher = SourceMatcher(self.source, None, self.group_connection_type) | ||||
|  | ||||
|         raw_groups: dict[str, dict[str, Any | dict[str, Any]]] = self.executor.plan.context[ | ||||
|             PLAN_CONTEXT_SOURCE_GROUPS | ||||
|  | ||||
| @ -1,152 +0,0 @@ | ||||
| """Source user and group matching""" | ||||
|  | ||||
| from dataclasses import dataclass | ||||
| from enum import Enum | ||||
| from typing import Any | ||||
|  | ||||
| from django.db.models import Q | ||||
| from structlog import get_logger | ||||
|  | ||||
| from authentik.core.models import ( | ||||
|     Group, | ||||
|     GroupSourceConnection, | ||||
|     Source, | ||||
|     SourceGroupMatchingModes, | ||||
|     SourceUserMatchingModes, | ||||
|     User, | ||||
|     UserSourceConnection, | ||||
| ) | ||||
|  | ||||
|  | ||||
| class Action(Enum): | ||||
|     """Actions that can be decided based on the request and source settings""" | ||||
|  | ||||
|     LINK = "link" | ||||
|     AUTH = "auth" | ||||
|     ENROLL = "enroll" | ||||
|     DENY = "deny" | ||||
|  | ||||
|  | ||||
| @dataclass | ||||
| class MatchableProperty: | ||||
|     property: str | ||||
|     link_mode: SourceUserMatchingModes | SourceGroupMatchingModes | ||||
|     deny_mode: SourceUserMatchingModes | SourceGroupMatchingModes | ||||
|  | ||||
|  | ||||
| class SourceMatcher: | ||||
|     def __init__( | ||||
|         self, | ||||
|         source: Source, | ||||
|         user_connection_type: type[UserSourceConnection], | ||||
|         group_connection_type: type[GroupSourceConnection], | ||||
|     ): | ||||
|         self.source = source | ||||
|         self.user_connection_type = user_connection_type | ||||
|         self.group_connection_type = group_connection_type | ||||
|         self._logger = get_logger().bind(source=self.source) | ||||
|  | ||||
|     def get_action( | ||||
|         self, | ||||
|         object_type: type[User | Group], | ||||
|         matchable_properties: list[MatchableProperty], | ||||
|         identifier: str, | ||||
|         properties: dict[str, Any | dict[str, Any]], | ||||
|     ) -> tuple[Action, UserSourceConnection | GroupSourceConnection | None]: | ||||
|         connection_type = None | ||||
|         matching_mode = None | ||||
|         identifier_matching_mode = None | ||||
|         if object_type == User: | ||||
|             connection_type = self.user_connection_type | ||||
|             matching_mode = self.source.user_matching_mode | ||||
|             identifier_matching_mode = SourceUserMatchingModes.IDENTIFIER | ||||
|         if object_type == Group: | ||||
|             connection_type = self.group_connection_type | ||||
|             matching_mode = self.source.group_matching_mode | ||||
|             identifier_matching_mode = SourceGroupMatchingModes.IDENTIFIER | ||||
|         if not connection_type or not matching_mode or not identifier_matching_mode: | ||||
|             return Action.DENY, None | ||||
|  | ||||
|         new_connection = connection_type(source=self.source, identifier=identifier) | ||||
|  | ||||
|         existing_connections = connection_type.objects.filter( | ||||
|             source=self.source, identifier=identifier | ||||
|         ) | ||||
|         if existing_connections.exists(): | ||||
|             return Action.AUTH, existing_connections.first() | ||||
|         # No connection exists, but we match on identifier, so enroll | ||||
|         if matching_mode == identifier_matching_mode: | ||||
|             # We don't save the connection here cause it doesn't have a user/group assigned yet | ||||
|             return Action.ENROLL, new_connection | ||||
|  | ||||
|         # Check for existing users with matching attributes | ||||
|         query = Q() | ||||
|         for matchable_property in matchable_properties: | ||||
|             property = matchable_property.property | ||||
|             if matching_mode in [matchable_property.link_mode, matchable_property.deny_mode]: | ||||
|                 if not properties.get(property, None): | ||||
|                     self._logger.warning( | ||||
|                         "Refusing to use none property", identifier=identifier, property=property | ||||
|                     ) | ||||
|                     return Action.DENY, None | ||||
|                 query_args = { | ||||
|                     f"{property}__exact": properties[property], | ||||
|                 } | ||||
|                 query = Q(**query_args) | ||||
|         self._logger.debug( | ||||
|             "Trying to link with existing object", query=query, identifier=identifier | ||||
|         ) | ||||
|         matching_objects = object_type.objects.filter(query) | ||||
|         # Not matching objects, always enroll | ||||
|         if not matching_objects.exists(): | ||||
|             self._logger.debug("No matching objects found, enrolling") | ||||
|             return Action.ENROLL, new_connection | ||||
|  | ||||
|         obj = matching_objects.first() | ||||
|         if matching_mode in [mp.link_mode for mp in matchable_properties]: | ||||
|             attr = None | ||||
|             if object_type == User: | ||||
|                 attr = "user" | ||||
|             if object_type == Group: | ||||
|                 attr = "group" | ||||
|             setattr(new_connection, attr, obj) | ||||
|             return Action.LINK, new_connection | ||||
|         if matching_mode in [mp.deny_mode for mp in matchable_properties]: | ||||
|             self._logger.info("Denying source because object exists", obj=obj) | ||||
|             return Action.DENY, None | ||||
|  | ||||
|         # Should never get here as default enroll case is returned above. | ||||
|         return Action.DENY, None  # pragma: no cover | ||||
|  | ||||
|     def get_user_action( | ||||
|         self, identifier: str, properties: dict[str, Any | dict[str, Any]] | ||||
|     ) -> tuple[Action, UserSourceConnection | None]: | ||||
|         return self.get_action( | ||||
|             User, | ||||
|             [ | ||||
|                 MatchableProperty( | ||||
|                     "username", | ||||
|                     SourceUserMatchingModes.USERNAME_LINK, | ||||
|                     SourceUserMatchingModes.USERNAME_DENY, | ||||
|                 ), | ||||
|                 MatchableProperty( | ||||
|                     "email", SourceUserMatchingModes.EMAIL_LINK, SourceUserMatchingModes.EMAIL_DENY | ||||
|                 ), | ||||
|             ], | ||||
|             identifier, | ||||
|             properties, | ||||
|         ) | ||||
|  | ||||
|     def get_group_action( | ||||
|         self, identifier: str, properties: dict[str, Any | dict[str, Any]] | ||||
|     ) -> tuple[Action, GroupSourceConnection | None]: | ||||
|         return self.get_action( | ||||
|             Group, | ||||
|             [ | ||||
|                 MatchableProperty( | ||||
|                     "name", SourceGroupMatchingModes.NAME_LINK, SourceGroupMatchingModes.NAME_DENY | ||||
|                 ), | ||||
|             ], | ||||
|             identifier, | ||||
|             properties, | ||||
|         ) | ||||
| @ -15,8 +15,8 @@ | ||||
|         {% endblock %} | ||||
|         <link rel="stylesheet" type="text/css" href="{% static 'dist/authentik.css' %}"> | ||||
|         <link rel="stylesheet" type="text/css" href="{% static 'dist/custom.css' %}" data-inject> | ||||
|         <script src="{% versioned_script 'dist/poly-%v.js' %}" type="module"></script> | ||||
|         <script src="{% versioned_script 'dist/standalone/loading/index-%v.js' %}" type="module"></script> | ||||
|         {% versioned_script "dist/poly-%v.js" %} | ||||
|         {% versioned_script "dist/standalone/loading/index-%v.js" %} | ||||
|         {% block head %} | ||||
|         {% endblock %} | ||||
|         <meta name="sentry-trace" content="{{ sentry_trace }}" /> | ||||
|  | ||||
| @ -3,7 +3,7 @@ | ||||
| {% load authentik_core %} | ||||
|  | ||||
| {% block head %} | ||||
| <script src="{% versioned_script 'dist/admin/AdminInterface-%v.js' %}" type="module"></script> | ||||
| {% versioned_script "dist/admin/AdminInterface-%v.js" %} | ||||
| <meta name="theme-color" content="#18191a" media="(prefers-color-scheme: dark)"> | ||||
| <meta name="theme-color" content="#ffffff" media="(prefers-color-scheme: light)"> | ||||
| {% include "base/header_js.html" %} | ||||
|  | ||||
							
								
								
									
										43
									
								
								authentik/core/templates/if/end_session.html
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										43
									
								
								authentik/core/templates/if/end_session.html
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,43 @@ | ||||
| {% extends 'login/base_full.html' %} | ||||
|  | ||||
| {% load static %} | ||||
| {% load i18n %} | ||||
|  | ||||
| {% block title %} | ||||
| {% trans 'End session' %} - {{ brand.branding_title }} | ||||
| {% endblock %} | ||||
|  | ||||
| {% block card_title %} | ||||
| {% blocktrans with application=application.name %} | ||||
| You've logged out of {{ application }}. | ||||
| {% endblocktrans %} | ||||
| {% endblock %} | ||||
|  | ||||
| {% block card %} | ||||
| <form method="POST" class="pf-c-form"> | ||||
|     <p> | ||||
|         {% blocktrans with application=application.name branding_title=brand.branding_title %} | ||||
|             You've logged out of {{ application }}. You can go back to the overview to launch another application, or log out of your {{ branding_title }} account. | ||||
|         {% endblocktrans %} | ||||
|     </p> | ||||
|  | ||||
|     <a id="ak-back-home" href="{% url 'authentik_core:root-redirect' %}" class="pf-c-button pf-m-primary"> | ||||
|         {% trans 'Go back to overview' %} | ||||
|     </a> | ||||
|  | ||||
|     <a id="logout" href="{% url 'authentik_flows:default-invalidation' %}" class="pf-c-button pf-m-secondary"> | ||||
|         {% blocktrans with branding_title=brand.branding_title %} | ||||
|             Log out of {{ branding_title }} | ||||
|         {% endblocktrans %} | ||||
|     </a> | ||||
|  | ||||
|     {% if application.get_launch_url %} | ||||
|     <a href="{{ application.get_launch_url }}" class="pf-c-button pf-m-secondary"> | ||||
|         {% blocktrans with application=application.name %} | ||||
|             Log back into {{ application }} | ||||
|         {% endblocktrans %} | ||||
|     </a> | ||||
|     {% endif %} | ||||
|  | ||||
| </form> | ||||
| {% endblock %} | ||||
| @ -3,7 +3,7 @@ | ||||
| {% load authentik_core %} | ||||
|  | ||||
| {% block head %} | ||||
| <script src="{% versioned_script 'dist/user/UserInterface-%v.js' %}" type="module"></script> | ||||
| {% versioned_script "dist/user/UserInterface-%v.js" %} | ||||
| <meta name="theme-color" content="#1c1e21" media="(prefers-color-scheme: light)"> | ||||
| <meta name="theme-color" content="#1c1e21" media="(prefers-color-scheme: dark)"> | ||||
| {% include "base/header_js.html" %} | ||||
|  | ||||
| @ -2,6 +2,7 @@ | ||||
|  | ||||
| from django import template | ||||
| from django.templatetags.static import static as static_loader | ||||
| from django.utils.safestring import mark_safe | ||||
|  | ||||
| from authentik import get_full_version | ||||
|  | ||||
| @ -11,4 +12,10 @@ register = template.Library() | ||||
| @register.simple_tag() | ||||
| def versioned_script(path: str) -> str: | ||||
|     """Wrapper around {% static %} tag that supports setting the version""" | ||||
|     return static_loader(path.replace("%v", get_full_version())) | ||||
|     returned_lines = [ | ||||
|         ( | ||||
|             f'<script src="{static_loader(path.replace("%v", get_full_version()))}' | ||||
|             '" type="module"></script>' | ||||
|         ), | ||||
|     ] | ||||
|     return mark_safe("".join(returned_lines))  # nosec | ||||
|  | ||||
| @ -9,12 +9,9 @@ from rest_framework.test import APITestCase | ||||
|  | ||||
| from authentik.core.models import Application | ||||
| from authentik.core.tests.utils import create_test_admin_user, create_test_flow | ||||
| from authentik.lib.generators import generate_id | ||||
| from authentik.policies.dummy.models import DummyPolicy | ||||
| from authentik.policies.models import PolicyBinding | ||||
| from authentik.providers.oauth2.models import OAuth2Provider, RedirectURI, RedirectURIMatchingMode | ||||
| from authentik.providers.proxy.models import ProxyProvider | ||||
| from authentik.providers.saml.models import SAMLProvider | ||||
| from authentik.providers.oauth2.models import OAuth2Provider | ||||
|  | ||||
|  | ||||
| class TestApplicationsAPI(APITestCase): | ||||
| @ -24,7 +21,7 @@ class TestApplicationsAPI(APITestCase): | ||||
|         self.user = create_test_admin_user() | ||||
|         self.provider = OAuth2Provider.objects.create( | ||||
|             name="test", | ||||
|             redirect_uris=[RedirectURI(RedirectURIMatchingMode.STRICT, "http://some-other-domain")], | ||||
|             redirect_uris="http://some-other-domain", | ||||
|             authorization_flow=create_test_flow(), | ||||
|         ) | ||||
|         self.allowed: Application = Application.objects.create( | ||||
| @ -134,7 +131,6 @@ class TestApplicationsAPI(APITestCase): | ||||
|                             "assigned_application_name": "allowed", | ||||
|                             "assigned_application_slug": "allowed", | ||||
|                             "authentication_flow": None, | ||||
|                             "invalidation_flow": None, | ||||
|                             "authorization_flow": str(self.provider.authorization_flow.pk), | ||||
|                             "component": "ak-provider-oauth2-form", | ||||
|                             "meta_model_name": "authentik_providers_oauth2.oauth2provider", | ||||
| @ -187,7 +183,6 @@ class TestApplicationsAPI(APITestCase): | ||||
|                             "assigned_application_name": "allowed", | ||||
|                             "assigned_application_slug": "allowed", | ||||
|                             "authentication_flow": None, | ||||
|                             "invalidation_flow": None, | ||||
|                             "authorization_flow": str(self.provider.authorization_flow.pk), | ||||
|                             "component": "ak-provider-oauth2-form", | ||||
|                             "meta_model_name": "authentik_providers_oauth2.oauth2provider", | ||||
| @ -227,31 +222,3 @@ class TestApplicationsAPI(APITestCase): | ||||
|                 ], | ||||
|             }, | ||||
|         ) | ||||
|  | ||||
|     def test_get_provider(self): | ||||
|         """Ensure that proxy providers (at the time of writing that is the only provider | ||||
|         that inherits from another proxy type (OAuth) instead of inheriting from the root | ||||
|         provider class) is correctly looked up and selected from the database""" | ||||
|         slug = generate_id() | ||||
|         provider = ProxyProvider.objects.create(name=generate_id()) | ||||
|         Application.objects.create( | ||||
|             name=generate_id(), | ||||
|             slug=slug, | ||||
|             provider=provider, | ||||
|         ) | ||||
|         self.assertEqual(Application.objects.get(slug=slug).get_provider(), provider) | ||||
|         self.assertEqual( | ||||
|             Application.objects.with_provider().get(slug=slug).get_provider(), provider | ||||
|         ) | ||||
|  | ||||
|         slug = generate_id() | ||||
|         provider = SAMLProvider.objects.create(name=generate_id()) | ||||
|         Application.objects.create( | ||||
|             name=generate_id(), | ||||
|             slug=slug, | ||||
|             provider=provider, | ||||
|         ) | ||||
|         self.assertEqual(Application.objects.get(slug=slug).get_provider(), provider) | ||||
|         self.assertEqual( | ||||
|             Application.objects.with_provider().get(slug=slug).get_provider(), provider | ||||
|         ) | ||||
|  | ||||
| @ -1,59 +0,0 @@ | ||||
| """Test Devices API""" | ||||
|  | ||||
| from json import loads | ||||
|  | ||||
| from django.urls import reverse | ||||
| from rest_framework.test import APITestCase | ||||
|  | ||||
| from authentik.core.tests.utils import create_test_admin_user, create_test_user | ||||
|  | ||||
|  | ||||
| class TestDevicesAPI(APITestCase): | ||||
|     """Test applications API""" | ||||
|  | ||||
|     def setUp(self) -> None: | ||||
|         self.admin = create_test_admin_user() | ||||
|         self.user1 = create_test_user() | ||||
|         self.device1 = self.user1.staticdevice_set.create() | ||||
|         self.user2 = create_test_user() | ||||
|         self.device2 = self.user2.staticdevice_set.create() | ||||
|  | ||||
|     def test_user_api(self): | ||||
|         """Test user API""" | ||||
|         self.client.force_login(self.user1) | ||||
|         response = self.client.get( | ||||
|             reverse( | ||||
|                 "authentik_api:device-list", | ||||
|             ) | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|         body = loads(response.content.decode()) | ||||
|         self.assertEqual(len(body), 1) | ||||
|         self.assertEqual(body[0]["pk"], str(self.device1.pk)) | ||||
|  | ||||
|     def test_user_api_as_admin(self): | ||||
|         """Test user API""" | ||||
|         self.client.force_login(self.admin) | ||||
|         response = self.client.get( | ||||
|             reverse( | ||||
|                 "authentik_api:device-list", | ||||
|             ) | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|         body = loads(response.content.decode()) | ||||
|         self.assertEqual(len(body), 0) | ||||
|  | ||||
|     def test_admin_api(self): | ||||
|         """Test admin API""" | ||||
|         self.client.force_login(self.admin) | ||||
|         response = self.client.get( | ||||
|             reverse( | ||||
|                 "authentik_api:admin-device-list", | ||||
|             ) | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|         body = loads(response.content.decode()) | ||||
|         self.assertEqual(len(body), 2) | ||||
|         self.assertEqual( | ||||
|             {body[0]["pk"], body[1]["pk"]}, {str(self.device1.pk), str(self.device2.pk)} | ||||
|         ) | ||||
| @ -3,10 +3,10 @@ | ||||
| from json import loads | ||||
|  | ||||
| from django.urls import reverse | ||||
| from guardian.shortcuts import assign_perm | ||||
| from rest_framework.test import APITestCase | ||||
|  | ||||
| from authentik.core.tests.utils import create_test_admin_user, create_test_user | ||||
| from authentik.core.models import User | ||||
| from authentik.core.tests.utils import create_test_admin_user | ||||
| from authentik.tenants.utils import get_current_tenant | ||||
|  | ||||
|  | ||||
| @ -15,7 +15,7 @@ class TestImpersonation(APITestCase): | ||||
|  | ||||
|     def setUp(self) -> None: | ||||
|         super().setUp() | ||||
|         self.other_user = create_test_user() | ||||
|         self.other_user = User.objects.create(username="to-impersonate") | ||||
|         self.user = create_test_admin_user() | ||||
|  | ||||
|     def test_impersonate_simple(self): | ||||
| @ -29,8 +29,7 @@ class TestImpersonation(APITestCase): | ||||
|             reverse( | ||||
|                 "authentik_api:user-impersonate", | ||||
|                 kwargs={"pk": self.other_user.pk}, | ||||
|             ), | ||||
|             data={"reason": "some reason"}, | ||||
|             ) | ||||
|         ) | ||||
|  | ||||
|         response = self.client.get(reverse("authentik_api:user-me")) | ||||
| @ -45,55 +44,12 @@ class TestImpersonation(APITestCase): | ||||
|         self.assertEqual(response_body["user"]["username"], self.user.username) | ||||
|         self.assertNotIn("original", response_body) | ||||
|  | ||||
|     def test_impersonate_global(self): | ||||
|         """Test impersonation with global permissions""" | ||||
|         new_user = create_test_user() | ||||
|         assign_perm("authentik_core.impersonate", new_user) | ||||
|         assign_perm("authentik_core.view_user", new_user) | ||||
|         self.client.force_login(new_user) | ||||
|  | ||||
|         response = self.client.post( | ||||
|             reverse( | ||||
|                 "authentik_api:user-impersonate", | ||||
|                 kwargs={"pk": self.other_user.pk}, | ||||
|             ), | ||||
|             data={"reason": "some reason"}, | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 201) | ||||
|  | ||||
|         response = self.client.get(reverse("authentik_api:user-me")) | ||||
|         response_body = loads(response.content.decode()) | ||||
|         self.assertEqual(response_body["user"]["username"], self.other_user.username) | ||||
|         self.assertEqual(response_body["original"]["username"], new_user.username) | ||||
|  | ||||
|     def test_impersonate_scoped(self): | ||||
|         """Test impersonation with scoped permissions""" | ||||
|         new_user = create_test_user() | ||||
|         assign_perm("authentik_core.impersonate", new_user, self.other_user) | ||||
|         assign_perm("authentik_core.view_user", new_user, self.other_user) | ||||
|         self.client.force_login(new_user) | ||||
|  | ||||
|         response = self.client.post( | ||||
|             reverse( | ||||
|                 "authentik_api:user-impersonate", | ||||
|                 kwargs={"pk": self.other_user.pk}, | ||||
|             ), | ||||
|             data={"reason": "some reason"}, | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 201) | ||||
|  | ||||
|         response = self.client.get(reverse("authentik_api:user-me")) | ||||
|         response_body = loads(response.content.decode()) | ||||
|         self.assertEqual(response_body["user"]["username"], self.other_user.username) | ||||
|         self.assertEqual(response_body["original"]["username"], new_user.username) | ||||
|  | ||||
|     def test_impersonate_denied(self): | ||||
|         """test impersonation without permissions""" | ||||
|         self.client.force_login(self.other_user) | ||||
|  | ||||
|         response = self.client.post( | ||||
|             reverse("authentik_api:user-impersonate", kwargs={"pk": self.user.pk}), | ||||
|             data={"reason": "some reason"}, | ||||
|             reverse("authentik_api:user-impersonate", kwargs={"pk": self.user.pk}) | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 403) | ||||
|  | ||||
| @ -109,8 +65,7 @@ class TestImpersonation(APITestCase): | ||||
|         self.client.force_login(self.user) | ||||
|  | ||||
|         response = self.client.post( | ||||
|             reverse("authentik_api:user-impersonate", kwargs={"pk": self.other_user.pk}), | ||||
|             data={"reason": "some reason"}, | ||||
|             reverse("authentik_api:user-impersonate", kwargs={"pk": self.other_user.pk}) | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 401) | ||||
|  | ||||
| @ -123,22 +78,7 @@ class TestImpersonation(APITestCase): | ||||
|         self.client.force_login(self.user) | ||||
|  | ||||
|         response = self.client.post( | ||||
|             reverse("authentik_api:user-impersonate", kwargs={"pk": self.user.pk}), | ||||
|             data={"reason": "some reason"}, | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 401) | ||||
|  | ||||
|         response = self.client.get(reverse("authentik_api:user-me")) | ||||
|         response_body = loads(response.content.decode()) | ||||
|         self.assertEqual(response_body["user"]["username"], self.user.username) | ||||
|  | ||||
|     def test_impersonate_reason_required(self): | ||||
|         """test impersonation that user must provide reason""" | ||||
|         self.client.force_login(self.user) | ||||
|  | ||||
|         response = self.client.post( | ||||
|             reverse("authentik_api:user-impersonate", kwargs={"pk": self.user.pk}), | ||||
|             data={"reason": ""}, | ||||
|             reverse("authentik_api:user-impersonate", kwargs={"pk": self.user.pk}) | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 401) | ||||
|  | ||||
|  | ||||
| @ -81,22 +81,6 @@ class TestSourceFlowManager(TestCase): | ||||
|             reverse("authentik_core:if-user") + "#/settings;page-sources", | ||||
|         ) | ||||
|  | ||||
|     def test_authenticated_auth(self): | ||||
|         """Test authenticated user linking""" | ||||
|         user = User.objects.create(username="foo", email="foo@bar.baz") | ||||
|         UserOAuthSourceConnection.objects.create( | ||||
|             user=user, source=self.source, identifier=self.identifier | ||||
|         ) | ||||
|         request = get_request("/", user=user) | ||||
|         flow_manager = OAuthSourceFlowManager( | ||||
|             self.source, request, self.identifier, {"info": {}}, {} | ||||
|         ) | ||||
|         action, connection = flow_manager.get_action() | ||||
|         self.assertEqual(action, Action.AUTH) | ||||
|         self.assertIsNotNone(connection.pk) | ||||
|         response = flow_manager.get_flow() | ||||
|         self.assertEqual(response.status_code, 302) | ||||
|  | ||||
|     def test_unauthenticated_link(self): | ||||
|         """Test un-authenticated user linking""" | ||||
|         flow_manager = OAuthSourceFlowManager( | ||||
|  | ||||
| @ -1,13 +1,11 @@ | ||||
| """Test Transactional API""" | ||||
|  | ||||
| from django.urls import reverse | ||||
| from guardian.shortcuts import assign_perm | ||||
| from rest_framework.test import APITestCase | ||||
|  | ||||
| from authentik.core.models import Application, Group | ||||
| from authentik.core.tests.utils import create_test_flow, create_test_user | ||||
| from authentik.core.models import Application | ||||
| from authentik.core.tests.utils import create_test_admin_user, create_test_flow | ||||
| from authentik.lib.generators import generate_id | ||||
| from authentik.policies.models import PolicyBinding | ||||
| from authentik.providers.oauth2.models import OAuth2Provider | ||||
|  | ||||
|  | ||||
| @ -15,68 +13,12 @@ class TestTransactionalApplicationsAPI(APITestCase): | ||||
|     """Test Transactional API""" | ||||
|  | ||||
|     def setUp(self) -> None: | ||||
|         self.user = create_test_user() | ||||
|         assign_perm("authentik_core.add_application", self.user) | ||||
|         assign_perm("authentik_providers_oauth2.add_oauth2provider", self.user) | ||||
|         self.user = create_test_admin_user() | ||||
|  | ||||
|     def test_create_transactional(self): | ||||
|         """Test transactional Application + provider creation""" | ||||
|         self.client.force_login(self.user) | ||||
|         uid = generate_id() | ||||
|         response = self.client.put( | ||||
|             reverse("authentik_api:core-transactional-application"), | ||||
|             data={ | ||||
|                 "app": { | ||||
|                     "name": uid, | ||||
|                     "slug": uid, | ||||
|                 }, | ||||
|                 "provider_model": "authentik_providers_oauth2.oauth2provider", | ||||
|                 "provider": { | ||||
|                     "name": uid, | ||||
|                     "authorization_flow": str(create_test_flow().pk), | ||||
|                     "invalidation_flow": str(create_test_flow().pk), | ||||
|                     "redirect_uris": [], | ||||
|                 }, | ||||
|             }, | ||||
|         ) | ||||
|         self.assertJSONEqual(response.content.decode(), {"applied": True, "logs": []}) | ||||
|         provider = OAuth2Provider.objects.filter(name=uid).first() | ||||
|         self.assertIsNotNone(provider) | ||||
|         app = Application.objects.filter(slug=uid).first() | ||||
|         self.assertIsNotNone(app) | ||||
|         self.assertEqual(app.provider.pk, provider.pk) | ||||
|  | ||||
|     def test_create_transactional_permission_denied(self): | ||||
|         """Test transactional Application + provider creation (missing permissions)""" | ||||
|         self.client.force_login(self.user) | ||||
|         uid = generate_id() | ||||
|         response = self.client.put( | ||||
|             reverse("authentik_api:core-transactional-application"), | ||||
|             data={ | ||||
|                 "app": { | ||||
|                     "name": uid, | ||||
|                     "slug": uid, | ||||
|                 }, | ||||
|                 "provider_model": "authentik_providers_saml.samlprovider", | ||||
|                 "provider": { | ||||
|                     "name": uid, | ||||
|                     "authorization_flow": str(create_test_flow().pk), | ||||
|                     "invalidation_flow": str(create_test_flow().pk), | ||||
|                     "acs_url": "https://goauthentik.io", | ||||
|                 }, | ||||
|             }, | ||||
|         ) | ||||
|         self.assertJSONEqual( | ||||
|             response.content.decode(), | ||||
|             {"provider": "User lacks permission to create authentik_providers_saml.samlprovider"}, | ||||
|         ) | ||||
|  | ||||
|     def test_create_transactional_bindings(self): | ||||
|         """Test transactional Application + provider creation""" | ||||
|         assign_perm("authentik_policies.add_policybinding", self.user) | ||||
|         self.client.force_login(self.user) | ||||
|         uid = generate_id() | ||||
|         group = Group.objects.create(name=generate_id()) | ||||
|         authorization_flow = create_test_flow() | ||||
|         response = self.client.put( | ||||
|             reverse("authentik_api:core-transactional-application"), | ||||
| @ -89,10 +31,7 @@ class TestTransactionalApplicationsAPI(APITestCase): | ||||
|                 "provider": { | ||||
|                     "name": uid, | ||||
|                     "authorization_flow": str(authorization_flow.pk), | ||||
|                     "invalidation_flow": str(authorization_flow.pk), | ||||
|                     "redirect_uris": [], | ||||
|                 }, | ||||
|                 "policy_bindings": [{"group": group.pk, "order": 0}], | ||||
|             }, | ||||
|         ) | ||||
|         self.assertJSONEqual(response.content.decode(), {"applied": True, "logs": []}) | ||||
| @ -101,10 +40,6 @@ class TestTransactionalApplicationsAPI(APITestCase): | ||||
|         app = Application.objects.filter(slug=uid).first() | ||||
|         self.assertIsNotNone(app) | ||||
|         self.assertEqual(app.provider.pk, provider.pk) | ||||
|         binding = PolicyBinding.objects.filter(target=app).first() | ||||
|         self.assertIsNotNone(binding) | ||||
|         self.assertEqual(binding.target, app) | ||||
|         self.assertEqual(binding.group, group) | ||||
|  | ||||
|     def test_create_transactional_invalid(self): | ||||
|         """Test transactional Application + provider creation""" | ||||
| @ -121,46 +56,10 @@ class TestTransactionalApplicationsAPI(APITestCase): | ||||
|                 "provider": { | ||||
|                     "name": uid, | ||||
|                     "authorization_flow": "", | ||||
|                     "invalidation_flow": "", | ||||
|                     "redirect_uris": [], | ||||
|                 }, | ||||
|             }, | ||||
|         ) | ||||
|         self.assertJSONEqual( | ||||
|             response.content.decode(), | ||||
|             { | ||||
|                 "provider": { | ||||
|                     "authorization_flow": ["This field may not be null."], | ||||
|                     "invalidation_flow": ["This field may not be null."], | ||||
|                 } | ||||
|             }, | ||||
|         ) | ||||
|  | ||||
|     def test_create_transactional_duplicate_name_provider(self): | ||||
|         """Test transactional Application + provider creation""" | ||||
|         self.client.force_login(self.user) | ||||
|         uid = generate_id() | ||||
|         OAuth2Provider.objects.create( | ||||
|             name=uid, | ||||
|             authorization_flow=create_test_flow(), | ||||
|             invalidation_flow=create_test_flow(), | ||||
|         ) | ||||
|         response = self.client.put( | ||||
|             reverse("authentik_api:core-transactional-application"), | ||||
|             data={ | ||||
|                 "app": { | ||||
|                     "name": uid, | ||||
|                     "slug": uid, | ||||
|                 }, | ||||
|                 "provider_model": "authentik_providers_oauth2.oauth2provider", | ||||
|                 "provider": { | ||||
|                     "name": uid, | ||||
|                     "authorization_flow": str(create_test_flow().pk), | ||||
|                     "invalidation_flow": str(create_test_flow().pk), | ||||
|                 }, | ||||
|             }, | ||||
|         ) | ||||
|         self.assertJSONEqual( | ||||
|             response.content.decode(), | ||||
|             {"provider": {"name": ["State is set to must_created and object exists already"]}}, | ||||
|             {"provider": {"authorization_flow": ["This field may not be null."]}}, | ||||
|         ) | ||||
|  | ||||
| @ -5,6 +5,7 @@ from channels.sessions import CookieMiddleware | ||||
| from django.conf import settings | ||||
| from django.contrib.auth.decorators import login_required | ||||
| from django.urls import path | ||||
| from django.views.decorators.csrf import ensure_csrf_cookie | ||||
|  | ||||
| from authentik.core.api.applications import ApplicationViewSet | ||||
| from authentik.core.api.authenticated_sessions import AuthenticatedSessionViewSet | ||||
| @ -23,6 +24,7 @@ from authentik.core.views.interface import ( | ||||
|     InterfaceView, | ||||
|     RootRedirectView, | ||||
| ) | ||||
| from authentik.core.views.session import EndSessionView | ||||
| from authentik.flows.views.interface import FlowInterfaceView | ||||
| from authentik.root.asgi_middleware import SessionMiddleware | ||||
| from authentik.root.messages.consumer import MessageConsumer | ||||
| @ -43,21 +45,26 @@ urlpatterns = [ | ||||
|     # Interfaces | ||||
|     path( | ||||
|         "if/admin/", | ||||
|         BrandDefaultRedirectView.as_view(template_name="if/admin.html"), | ||||
|         ensure_csrf_cookie(BrandDefaultRedirectView.as_view(template_name="if/admin.html")), | ||||
|         name="if-admin", | ||||
|     ), | ||||
|     path( | ||||
|         "if/user/", | ||||
|         BrandDefaultRedirectView.as_view(template_name="if/user.html"), | ||||
|         ensure_csrf_cookie(BrandDefaultRedirectView.as_view(template_name="if/user.html")), | ||||
|         name="if-user", | ||||
|     ), | ||||
|     path( | ||||
|         "if/flow/<slug:flow_slug>/", | ||||
|         # FIXME: move this url to the flows app...also will cause all | ||||
|         # of the reverse calls to be adjusted | ||||
|         FlowInterfaceView.as_view(), | ||||
|         ensure_csrf_cookie(FlowInterfaceView.as_view()), | ||||
|         name="if-flow", | ||||
|     ), | ||||
|     path( | ||||
|         "if/session-end/<slug:application_slug>/", | ||||
|         ensure_csrf_cookie(EndSessionView.as_view()), | ||||
|         name="if-session-end", | ||||
|     ), | ||||
|     # Fallback for WS | ||||
|     path("ws/outpost/<uuid:pk>/", InterfaceView.as_view(template_name="if/admin.html")), | ||||
|     path( | ||||
|  | ||||
							
								
								
									
										23
									
								
								authentik/core/views/session.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										23
									
								
								authentik/core/views/session.py
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,23 @@ | ||||
| """authentik Session Views""" | ||||
|  | ||||
| from typing import Any | ||||
|  | ||||
| from django.shortcuts import get_object_or_404 | ||||
| from django.views.generic.base import TemplateView | ||||
|  | ||||
| from authentik.core.models import Application | ||||
| from authentik.policies.views import PolicyAccessView | ||||
|  | ||||
|  | ||||
| class EndSessionView(TemplateView, PolicyAccessView): | ||||
|     """Allow the client to end the Session""" | ||||
|  | ||||
|     template_name = "if/end_session.html" | ||||
|  | ||||
|     def resolve_provider_application(self): | ||||
|         self.application = get_object_or_404(Application, slug=self.kwargs["application_slug"]) | ||||
|  | ||||
|     def get_context_data(self, **kwargs: Any) -> dict[str, Any]: | ||||
|         context = super().get_context_data(**kwargs) | ||||
|         context["application"] = self.application | ||||
|         return context | ||||
| @ -24,7 +24,6 @@ from rest_framework.fields import ( | ||||
| from rest_framework.filters import OrderingFilter, SearchFilter | ||||
| from rest_framework.request import Request | ||||
| from rest_framework.response import Response | ||||
| from rest_framework.validators import UniqueValidator | ||||
| from rest_framework.viewsets import ModelViewSet | ||||
| from structlog.stdlib import get_logger | ||||
|  | ||||
| @ -36,7 +35,6 @@ from authentik.crypto.builder import CertificateBuilder, PrivateKeyAlg | ||||
| from authentik.crypto.models import CertificateKeyPair | ||||
| from authentik.events.models import Event, EventAction | ||||
| from authentik.rbac.decorators import permission_required | ||||
| from authentik.rbac.filters import ObjectFilter | ||||
|  | ||||
| LOGGER = get_logger() | ||||
|  | ||||
| @ -182,10 +180,7 @@ class CertificateDataSerializer(PassiveSerializer): | ||||
| class CertificateGenerationSerializer(PassiveSerializer): | ||||
|     """Certificate generation parameters""" | ||||
|  | ||||
|     common_name = CharField( | ||||
|         validators=[UniqueValidator(queryset=CertificateKeyPair.objects.all())], | ||||
|         source="name", | ||||
|     ) | ||||
|     common_name = CharField() | ||||
|     subject_alt_name = CharField(required=False, allow_blank=True, label=_("Subject-alt name")) | ||||
|     validity_days = IntegerField(initial=365) | ||||
|     alg = ChoiceField(default=PrivateKeyAlg.RSA, choices=PrivateKeyAlg.choices) | ||||
| @ -246,10 +241,11 @@ class CertificateKeyPairViewSet(UsedByMixin, ModelViewSet): | ||||
|     def generate(self, request: Request) -> Response: | ||||
|         """Generate a new, self-signed certificate-key pair""" | ||||
|         data = CertificateGenerationSerializer(data=request.data) | ||||
|         data.is_valid(raise_exception=True) | ||||
|         if not data.is_valid(): | ||||
|             return Response(data.errors, status=400) | ||||
|         raw_san = data.validated_data.get("subject_alt_name", "") | ||||
|         sans = raw_san.split(",") if raw_san != "" else [] | ||||
|         builder = CertificateBuilder(data.validated_data["name"]) | ||||
|         builder = CertificateBuilder(data.validated_data["common_name"]) | ||||
|         builder.alg = data.validated_data["alg"] | ||||
|         builder.build( | ||||
|             subject_alt_names=sans, | ||||
| @ -269,7 +265,7 @@ class CertificateKeyPairViewSet(UsedByMixin, ModelViewSet): | ||||
|         ], | ||||
|         responses={200: CertificateDataSerializer(many=False)}, | ||||
|     ) | ||||
|     @action(detail=True, pagination_class=None, filter_backends=[ObjectFilter]) | ||||
|     @action(detail=True, pagination_class=None, filter_backends=[]) | ||||
|     def view_certificate(self, request: Request, pk: str) -> Response: | ||||
|         """Return certificate-key pairs certificate and log access""" | ||||
|         certificate: CertificateKeyPair = self.get_object() | ||||
| @ -299,7 +295,7 @@ class CertificateKeyPairViewSet(UsedByMixin, ModelViewSet): | ||||
|         ], | ||||
|         responses={200: CertificateDataSerializer(many=False)}, | ||||
|     ) | ||||
|     @action(detail=True, pagination_class=None, filter_backends=[ObjectFilter]) | ||||
|     @action(detail=True, pagination_class=None, filter_backends=[]) | ||||
|     def view_private_key(self, request: Request, pk: str) -> Response: | ||||
|         """Return certificate-key pairs private key and log access""" | ||||
|         certificate: CertificateKeyPair = self.get_object() | ||||
|  | ||||
| @ -18,7 +18,7 @@ from authentik.crypto.models import CertificateKeyPair | ||||
| from authentik.crypto.tasks import MANAGED_DISCOVERED, certificate_discovery | ||||
| from authentik.lib.config import CONFIG | ||||
| from authentik.lib.generators import generate_id, generate_key | ||||
| from authentik.providers.oauth2.models import OAuth2Provider, RedirectURI, RedirectURIMatchingMode | ||||
| from authentik.providers.oauth2.models import OAuth2Provider | ||||
|  | ||||
|  | ||||
| class TestCrypto(APITestCase): | ||||
| @ -89,17 +89,6 @@ class TestCrypto(APITestCase): | ||||
|         self.assertIsInstance(ext[1], DNSName) | ||||
|         self.assertEqual(ext[1].value, "baz") | ||||
|  | ||||
|     def test_builder_api_duplicate(self): | ||||
|         """Test Builder (via API)""" | ||||
|         cert = create_test_cert() | ||||
|         self.client.force_login(create_test_admin_user()) | ||||
|         res = self.client.post( | ||||
|             reverse("authentik_api:certificatekeypair-generate"), | ||||
|             data={"common_name": cert.name, "subject_alt_name": "bar,baz", "validity_days": 3}, | ||||
|         ) | ||||
|         self.assertEqual(res.status_code, 400) | ||||
|         self.assertJSONEqual(res.content, {"common_name": ["This field must be unique."]}) | ||||
|  | ||||
|     def test_builder_api_empty_san(self): | ||||
|         """Test Builder (via API)""" | ||||
|         self.client.force_login(create_test_admin_user()) | ||||
| @ -225,46 +214,6 @@ class TestCrypto(APITestCase): | ||||
|         self.assertEqual(200, response.status_code) | ||||
|         self.assertIn("Content-Disposition", response) | ||||
|  | ||||
|     def test_certificate_download_denied(self): | ||||
|         """Test certificate export (download)""" | ||||
|         self.client.logout() | ||||
|         keypair = create_test_cert() | ||||
|         response = self.client.get( | ||||
|             reverse( | ||||
|                 "authentik_api:certificatekeypair-view-certificate", | ||||
|                 kwargs={"pk": keypair.pk}, | ||||
|             ) | ||||
|         ) | ||||
|         self.assertEqual(403, response.status_code) | ||||
|         response = self.client.get( | ||||
|             reverse( | ||||
|                 "authentik_api:certificatekeypair-view-certificate", | ||||
|                 kwargs={"pk": keypair.pk}, | ||||
|             ), | ||||
|             data={"download": True}, | ||||
|         ) | ||||
|         self.assertEqual(403, response.status_code) | ||||
|  | ||||
|     def test_private_key_download_denied(self): | ||||
|         """Test private_key export (download)""" | ||||
|         self.client.logout() | ||||
|         keypair = create_test_cert() | ||||
|         response = self.client.get( | ||||
|             reverse( | ||||
|                 "authentik_api:certificatekeypair-view-private-key", | ||||
|                 kwargs={"pk": keypair.pk}, | ||||
|             ) | ||||
|         ) | ||||
|         self.assertEqual(403, response.status_code) | ||||
|         response = self.client.get( | ||||
|             reverse( | ||||
|                 "authentik_api:certificatekeypair-view-private-key", | ||||
|                 kwargs={"pk": keypair.pk}, | ||||
|             ), | ||||
|             data={"download": True}, | ||||
|         ) | ||||
|         self.assertEqual(403, response.status_code) | ||||
|  | ||||
|     def test_used_by(self): | ||||
|         """Test used_by endpoint""" | ||||
|         self.client.force_login(create_test_admin_user()) | ||||
| @ -274,7 +223,7 @@ class TestCrypto(APITestCase): | ||||
|             client_id="test", | ||||
|             client_secret=generate_key(), | ||||
|             authorization_flow=create_test_flow(), | ||||
|             redirect_uris=[RedirectURI(RedirectURIMatchingMode.STRICT, "http://localhost")], | ||||
|             redirect_uris="http://localhost", | ||||
|             signing_key=keypair, | ||||
|         ) | ||||
|         response = self.client.get( | ||||
| @ -297,26 +246,6 @@ class TestCrypto(APITestCase): | ||||
|             ], | ||||
|         ) | ||||
|  | ||||
|     def test_used_by_denied(self): | ||||
|         """Test used_by endpoint""" | ||||
|         self.client.logout() | ||||
|         keypair = create_test_cert() | ||||
|         OAuth2Provider.objects.create( | ||||
|             name=generate_id(), | ||||
|             client_id="test", | ||||
|             client_secret=generate_key(), | ||||
|             authorization_flow=create_test_flow(), | ||||
|             redirect_uris=[RedirectURI(RedirectURIMatchingMode.STRICT, "http://localhost")], | ||||
|             signing_key=keypair, | ||||
|         ) | ||||
|         response = self.client.get( | ||||
|             reverse( | ||||
|                 "authentik_api:certificatekeypair-used-by", | ||||
|                 kwargs={"pk": keypair.pk}, | ||||
|             ) | ||||
|         ) | ||||
|         self.assertEqual(403, response.status_code) | ||||
|  | ||||
|     def test_discovery(self): | ||||
|         """Test certificate discovery""" | ||||
|         name = generate_id() | ||||
|  | ||||
| @ -1,11 +1,12 @@ | ||||
| """Enterprise API Views""" | ||||
|  | ||||
| from dataclasses import asdict | ||||
| from datetime import timedelta | ||||
|  | ||||
| from django.utils.timezone import now | ||||
| from django.utils.translation import gettext as _ | ||||
| from drf_spectacular.types import OpenApiTypes | ||||
| from drf_spectacular.utils import OpenApiParameter, extend_schema, inline_serializer | ||||
| from drf_spectacular.utils import extend_schema, inline_serializer | ||||
| from rest_framework.decorators import action | ||||
| from rest_framework.exceptions import ValidationError | ||||
| from rest_framework.fields import CharField, IntegerField | ||||
| @ -29,7 +30,7 @@ class EnterpriseRequiredMixin: | ||||
|  | ||||
|     def validate(self, attrs: dict) -> dict: | ||||
|         """Check that a valid license exists""" | ||||
|         if not LicenseKey.cached_summary().status.is_valid: | ||||
|         if not LicenseKey.cached_summary().has_license: | ||||
|             raise ValidationError(_("Enterprise is required to create/update this object.")) | ||||
|         return super().validate(attrs) | ||||
|  | ||||
| @ -86,7 +87,7 @@ class LicenseViewSet(UsedByMixin, ModelViewSet): | ||||
|         }, | ||||
|     ) | ||||
|     @action(detail=False, methods=["GET"]) | ||||
|     def install_id(self, request: Request) -> Response: | ||||
|     def get_install_id(self, request: Request) -> Response: | ||||
|         """Get install_id""" | ||||
|         return Response( | ||||
|             data={ | ||||
| @ -99,22 +100,12 @@ class LicenseViewSet(UsedByMixin, ModelViewSet): | ||||
|         responses={ | ||||
|             200: LicenseSummarySerializer(), | ||||
|         }, | ||||
|         parameters=[ | ||||
|             OpenApiParameter( | ||||
|                 name="cached", | ||||
|                 location=OpenApiParameter.QUERY, | ||||
|                 type=OpenApiTypes.BOOL, | ||||
|                 default=True, | ||||
|             ) | ||||
|         ], | ||||
|     ) | ||||
|     @action(detail=False, methods=["GET"], permission_classes=[IsAuthenticated]) | ||||
|     def summary(self, request: Request) -> Response: | ||||
|         """Get the total license status""" | ||||
|         summary = LicenseKey.cached_summary() | ||||
|         if request.query_params.get("cached", "true").lower() == "false": | ||||
|             summary = LicenseKey.get_total().summary() | ||||
|         response = LicenseSummarySerializer(instance=summary) | ||||
|         response = LicenseSummarySerializer(data=asdict(LicenseKey.cached_summary())) | ||||
|         response.is_valid(raise_exception=True) | ||||
|         return Response(response.data) | ||||
|  | ||||
|     @permission_required(None, ["authentik_enterprise.view_license"]) | ||||
| @ -137,7 +128,7 @@ class LicenseViewSet(UsedByMixin, ModelViewSet): | ||||
|         forecast_for_months = 12 | ||||
|         response = LicenseForecastSerializer( | ||||
|             data={ | ||||
|                 "internal_users": LicenseKey.get_internal_user_count(), | ||||
|                 "internal_users": LicenseKey.get_default_user_count(), | ||||
|                 "external_users": LicenseKey.get_external_user_count(), | ||||
|                 "forecasted_internal_users": (internal_in_last_month * forecast_for_months), | ||||
|                 "forecasted_external_users": (external_in_last_month * forecast_for_months), | ||||
|  | ||||
| @ -25,4 +25,4 @@ class AuthentikEnterpriseConfig(EnterpriseConfig): | ||||
|         """Actual enterprise check, cached""" | ||||
|         from authentik.enterprise.license import LicenseKey | ||||
|  | ||||
|         return LicenseKey.cached_summary().status.is_valid | ||||
|         return LicenseKey.cached_summary().valid | ||||
|  | ||||
| @ -3,37 +3,24 @@ | ||||
| from base64 import b64decode | ||||
| from binascii import Error | ||||
| from dataclasses import asdict, dataclass, field | ||||
| from datetime import UTC, datetime, timedelta | ||||
| from datetime import datetime, timedelta | ||||
| from enum import Enum | ||||
| from functools import lru_cache | ||||
| from time import mktime | ||||
|  | ||||
| from cryptography.exceptions import InvalidSignature | ||||
| from cryptography.x509 import Certificate, load_der_x509_certificate, load_pem_x509_certificate | ||||
| from dacite import DaciteError, from_dict | ||||
| from dacite import from_dict | ||||
| from django.core.cache import cache | ||||
| from django.db.models.query import QuerySet | ||||
| from django.utils.timezone import now | ||||
| from jwt import PyJWTError, decode, get_unverified_header | ||||
| from rest_framework.exceptions import ValidationError | ||||
| from rest_framework.fields import ( | ||||
|     ChoiceField, | ||||
|     DateTimeField, | ||||
|     IntegerField, | ||||
|     ListField, | ||||
| ) | ||||
| from rest_framework.fields import BooleanField, DateTimeField, IntegerField | ||||
|  | ||||
| from authentik.core.api.utils import PassiveSerializer | ||||
| from authentik.core.models import User, UserTypes | ||||
| from authentik.enterprise.models import ( | ||||
|     THRESHOLD_READ_ONLY_WEEKS, | ||||
|     THRESHOLD_WARNING_ADMIN_WEEKS, | ||||
|     THRESHOLD_WARNING_EXPIRY_WEEKS, | ||||
|     THRESHOLD_WARNING_USER_WEEKS, | ||||
|     License, | ||||
|     LicenseUsage, | ||||
|     LicenseUsageStatus, | ||||
| ) | ||||
| from authentik.enterprise.models import License, LicenseUsage | ||||
| from authentik.tenants.utils import get_unique_identifier | ||||
|  | ||||
| CACHE_KEY_ENTERPRISE_LICENSE = "goauthentik.io/enterprise/license" | ||||
| @ -55,9 +42,6 @@ def get_license_aud() -> str: | ||||
| class LicenseFlags(Enum): | ||||
|     """License flags""" | ||||
|  | ||||
|     TRIAL = "trial" | ||||
|     NON_PRODUCTION = "non_production" | ||||
|  | ||||
|  | ||||
| @dataclass | ||||
| class LicenseSummary: | ||||
| @ -65,9 +49,12 @@ class LicenseSummary: | ||||
|  | ||||
|     internal_users: int | ||||
|     external_users: int | ||||
|     status: LicenseUsageStatus | ||||
|     valid: bool | ||||
|     show_admin_warning: bool | ||||
|     show_user_warning: bool | ||||
|     read_only: bool | ||||
|     latest_valid: datetime | ||||
|     license_flags: list[LicenseFlags] | ||||
|     has_license: bool | ||||
|  | ||||
|  | ||||
| class LicenseSummarySerializer(PassiveSerializer): | ||||
| @ -75,9 +62,12 @@ class LicenseSummarySerializer(PassiveSerializer): | ||||
|  | ||||
|     internal_users = IntegerField(required=True) | ||||
|     external_users = IntegerField(required=True) | ||||
|     status = ChoiceField(choices=LicenseUsageStatus.choices) | ||||
|     valid = BooleanField() | ||||
|     show_admin_warning = BooleanField() | ||||
|     show_user_warning = BooleanField() | ||||
|     read_only = BooleanField() | ||||
|     latest_valid = DateTimeField() | ||||
|     license_flags = ListField(child=ChoiceField(choices=tuple(x.value for x in LicenseFlags))) | ||||
|     has_license = BooleanField() | ||||
|  | ||||
|  | ||||
| @dataclass | ||||
| @ -90,10 +80,10 @@ class LicenseKey: | ||||
|     name: str | ||||
|     internal_users: int = 0 | ||||
|     external_users: int = 0 | ||||
|     license_flags: list[LicenseFlags] = field(default_factory=list) | ||||
|     flags: list[LicenseFlags] = field(default_factory=list) | ||||
|  | ||||
|     @staticmethod | ||||
|     def validate(jwt: str, check_expiry=True) -> "LicenseKey": | ||||
|     def validate(jwt: str) -> "LicenseKey": | ||||
|         """Validate the license from a given JWT""" | ||||
|         try: | ||||
|             headers = get_unverified_header(jwt) | ||||
| @ -117,28 +107,26 @@ class LicenseKey: | ||||
|                     our_cert.public_key(), | ||||
|                     algorithms=["ES512"], | ||||
|                     audience=get_license_aud(), | ||||
|                     options={"verify_exp": check_expiry, "verify_signature": check_expiry}, | ||||
|                 ), | ||||
|             ) | ||||
|         except PyJWTError: | ||||
|             unverified = decode(jwt, options={"verify_signature": False}) | ||||
|             if unverified["aud"] != get_license_aud(): | ||||
|                 raise ValidationError("Invalid Install ID in license") from None | ||||
|             raise ValidationError("Unable to verify license") from None | ||||
|         return body | ||||
|  | ||||
|     @staticmethod | ||||
|     def get_total() -> "LicenseKey": | ||||
|         """Get a summarized version of all (not expired) licenses""" | ||||
|         active_licenses = License.objects.filter(expiry__gte=now()) | ||||
|         total = LicenseKey(get_license_aud(), 0, "Summarized license", 0, 0) | ||||
|         for lic in License.objects.all(): | ||||
|         for lic in active_licenses: | ||||
|             total.internal_users += lic.internal_users | ||||
|             total.external_users += lic.external_users | ||||
|             exp_ts = int(mktime(lic.expiry.timetuple())) | ||||
|             if total.exp == 0: | ||||
|                 total.exp = exp_ts | ||||
|             total.exp = max(total.exp, exp_ts) | ||||
|             total.license_flags.extend(lic.status.license_flags) | ||||
|             if exp_ts <= total.exp: | ||||
|                 total.exp = exp_ts | ||||
|             total.flags.extend(lic.status.flags) | ||||
|         return total | ||||
|  | ||||
|     @staticmethod | ||||
| @ -147,7 +135,7 @@ class LicenseKey: | ||||
|         return User.objects.all().exclude_anonymous().exclude(is_active=False) | ||||
|  | ||||
|     @staticmethod | ||||
|     def get_internal_user_count(): | ||||
|     def get_default_user_count(): | ||||
|         """Get current default user count""" | ||||
|         return LicenseKey.base_user_qs().filter(type=UserTypes.INTERNAL).count() | ||||
|  | ||||
| @ -156,73 +144,59 @@ class LicenseKey: | ||||
|         """Get current external user count""" | ||||
|         return LicenseKey.base_user_qs().filter(type=UserTypes.EXTERNAL).count() | ||||
|  | ||||
|     def _last_valid_date(self): | ||||
|         last_valid_date = ( | ||||
|             LicenseUsage.objects.order_by("-record_date") | ||||
|             .filter(status=LicenseUsageStatus.VALID) | ||||
|             .first() | ||||
|         ) | ||||
|         if not last_valid_date: | ||||
|             return datetime.fromtimestamp(0, UTC) | ||||
|         return last_valid_date.record_date | ||||
|     def is_valid(self) -> bool: | ||||
|         """Check if the given license body covers all users | ||||
|  | ||||
|     def status(self) -> LicenseUsageStatus: | ||||
|         """Check if the given license body covers all users, and is valid.""" | ||||
|         last_valid = self._last_valid_date() | ||||
|         if self.exp == 0 and not License.objects.exists(): | ||||
|             return LicenseUsageStatus.UNLICENSED | ||||
|         _now = now() | ||||
|         # Check limit-exceeded based status | ||||
|         internal_users = self.get_internal_user_count() | ||||
|         external_users = self.get_external_user_count() | ||||
|         if internal_users > self.internal_users or external_users > self.external_users: | ||||
|             if last_valid < _now - timedelta(weeks=THRESHOLD_READ_ONLY_WEEKS): | ||||
|                 return LicenseUsageStatus.READ_ONLY | ||||
|             if last_valid < _now - timedelta(weeks=THRESHOLD_WARNING_USER_WEEKS): | ||||
|                 return LicenseUsageStatus.LIMIT_EXCEEDED_USER | ||||
|             if last_valid < _now - timedelta(weeks=THRESHOLD_WARNING_ADMIN_WEEKS): | ||||
|                 return LicenseUsageStatus.LIMIT_EXCEEDED_ADMIN | ||||
|         # Check expiry based status | ||||
|         if datetime.fromtimestamp(self.exp, UTC) < _now: | ||||
|             if datetime.fromtimestamp(self.exp, UTC) < _now - timedelta( | ||||
|                 weeks=THRESHOLD_READ_ONLY_WEEKS | ||||
|             ): | ||||
|                 return LicenseUsageStatus.READ_ONLY | ||||
|             return LicenseUsageStatus.EXPIRED | ||||
|         # Expiry warning | ||||
|         if datetime.fromtimestamp(self.exp, UTC) <= _now + timedelta( | ||||
|             weeks=THRESHOLD_WARNING_EXPIRY_WEEKS | ||||
|         ): | ||||
|             return LicenseUsageStatus.EXPIRY_SOON | ||||
|         return LicenseUsageStatus.VALID | ||||
|         Only checks the current count, no historical data is checked""" | ||||
|         default_users = self.get_default_user_count() | ||||
|         if default_users > self.internal_users: | ||||
|             return False | ||||
|         active_users = self.get_external_user_count() | ||||
|         if active_users > self.external_users: | ||||
|             return False | ||||
|         return True | ||||
|  | ||||
|     def record_usage(self): | ||||
|         """Capture the current validity status and metrics and save them""" | ||||
|         threshold = now() - timedelta(hours=8) | ||||
|         usage = ( | ||||
|             LicenseUsage.objects.order_by("-record_date").filter(record_date__gte=threshold).first() | ||||
|         ) | ||||
|         if not usage: | ||||
|             usage = LicenseUsage.objects.create( | ||||
|                 internal_user_count=self.get_internal_user_count(), | ||||
|         if not LicenseUsage.objects.filter(record_date__gte=threshold).exists(): | ||||
|             LicenseUsage.objects.create( | ||||
|                 user_count=self.get_default_user_count(), | ||||
|                 external_user_count=self.get_external_user_count(), | ||||
|                 status=self.status(), | ||||
|                 within_limits=self.is_valid(), | ||||
|             ) | ||||
|         summary = asdict(self.summary()) | ||||
|         # Also cache the latest summary for the middleware | ||||
|         cache.set(CACHE_KEY_ENTERPRISE_LICENSE, summary, timeout=CACHE_EXPIRY_ENTERPRISE_LICENSE) | ||||
|         return usage | ||||
|         return summary | ||||
|  | ||||
|     @staticmethod | ||||
|     def last_valid_date() -> datetime: | ||||
|         """Get the last date the license was valid""" | ||||
|         usage: LicenseUsage = ( | ||||
|             LicenseUsage.filter_not_expired(within_limits=True).order_by("-record_date").first() | ||||
|         ) | ||||
|         if not usage: | ||||
|             return now() | ||||
|         return usage.record_date | ||||
|  | ||||
|     def summary(self) -> LicenseSummary: | ||||
|         """Summary of license status""" | ||||
|         status = self.status() | ||||
|         has_license = License.objects.all().count() > 0 | ||||
|         last_valid = LicenseKey.last_valid_date() | ||||
|         show_admin_warning = last_valid < now() - timedelta(weeks=2) | ||||
|         show_user_warning = last_valid < now() - timedelta(weeks=4) | ||||
|         read_only = last_valid < now() - timedelta(weeks=6) | ||||
|         latest_valid = datetime.fromtimestamp(self.exp) | ||||
|         return LicenseSummary( | ||||
|             show_admin_warning=show_admin_warning and has_license, | ||||
|             show_user_warning=show_user_warning and has_license, | ||||
|             read_only=read_only and has_license, | ||||
|             latest_valid=latest_valid, | ||||
|             internal_users=self.internal_users, | ||||
|             external_users=self.external_users, | ||||
|             status=status, | ||||
|             license_flags=self.license_flags, | ||||
|             valid=self.is_valid(), | ||||
|             has_license=has_license, | ||||
|         ) | ||||
|  | ||||
|     @staticmethod | ||||
| @ -231,8 +205,4 @@ class LicenseKey: | ||||
|         summary = cache.get(CACHE_KEY_ENTERPRISE_LICENSE) | ||||
|         if not summary: | ||||
|             return LicenseKey.get_total().summary() | ||||
|         try: | ||||
|             return from_dict(LicenseSummary, summary) | ||||
|         except DaciteError: | ||||
|             cache.delete(CACHE_KEY_ENTERPRISE_LICENSE) | ||||
|             return LicenseKey.get_total().summary() | ||||
|         return from_dict(LicenseSummary, summary) | ||||
|  | ||||
| @ -8,7 +8,6 @@ from structlog.stdlib import BoundLogger, get_logger | ||||
|  | ||||
| from authentik.enterprise.api import LicenseViewSet | ||||
| from authentik.enterprise.license import LicenseKey | ||||
| from authentik.enterprise.models import LicenseUsageStatus | ||||
| from authentik.flows.views.executor import FlowExecutorView | ||||
| from authentik.lib.utils.reflection import class_to_path | ||||
|  | ||||
| @ -44,7 +43,7 @@ class EnterpriseMiddleware: | ||||
|         cached_status = LicenseKey.cached_summary() | ||||
|         if not cached_status: | ||||
|             return True | ||||
|         if cached_status.status == LicenseUsageStatus.READ_ONLY: | ||||
|         if cached_status.read_only: | ||||
|             return False | ||||
|         return True | ||||
|  | ||||
| @ -54,10 +53,10 @@ class EnterpriseMiddleware: | ||||
|         if request.method.lower() in ["get", "head", "options", "trace"]: | ||||
|             return True | ||||
|         # Always allow requests to manage licenses | ||||
|         if request.resolver_match._func_path == class_to_path(LicenseViewSet): | ||||
|         if class_to_path(request.resolver_match.func) == class_to_path(LicenseViewSet): | ||||
|             return True | ||||
|         # Flow executor is mounted as an API path but explicitly allowed | ||||
|         if request.resolver_match._func_path == class_to_path(FlowExecutorView): | ||||
|         if class_to_path(request.resolver_match.func) == class_to_path(FlowExecutorView): | ||||
|             return True | ||||
|         # Only apply these restrictions to the API | ||||
|         if "authentik_api" not in request.resolver_match.app_names: | ||||
|  | ||||
| @ -1,68 +0,0 @@ | ||||
| # Generated by Django 5.0.8 on 2024-08-08 14:15 | ||||
|  | ||||
| from django.db import migrations, models | ||||
| from django.apps.registry import Apps | ||||
| from django.db.backends.base.schema import BaseDatabaseSchemaEditor | ||||
|  | ||||
|  | ||||
| def migrate_license_usage(apps: Apps, schema_editor: BaseDatabaseSchemaEditor): | ||||
|     LicenseUsage = apps.get_model("authentik_enterprise", "licenseusage") | ||||
|     db_alias = schema_editor.connection.alias | ||||
|  | ||||
|     for usage in LicenseUsage.objects.using(db_alias).all(): | ||||
|         usage.status = "valid" if usage.within_limits else "limit_exceeded_admin" | ||||
|         usage.save() | ||||
|  | ||||
|  | ||||
| class Migration(migrations.Migration): | ||||
|  | ||||
|     dependencies = [ | ||||
|         ("authentik_enterprise", "0002_rename_users_license_internal_users_and_more"), | ||||
|     ] | ||||
|  | ||||
|     operations = [ | ||||
|         migrations.AddField( | ||||
|             model_name="licenseusage", | ||||
|             name="status", | ||||
|             field=models.TextField( | ||||
|                 choices=[ | ||||
|                     ("unlicensed", "Unlicensed"), | ||||
|                     ("valid", "Valid"), | ||||
|                     ("expired", "Expired"), | ||||
|                     ("expiry_soon", "Expiry Soon"), | ||||
|                     ("limit_exceeded_admin", "Limit Exceeded Admin"), | ||||
|                     ("limit_exceeded_user", "Limit Exceeded User"), | ||||
|                     ("read_only", "Read Only"), | ||||
|                 ], | ||||
|                 default=None, | ||||
|                 null=True, | ||||
|             ), | ||||
|             preserve_default=False, | ||||
|         ), | ||||
|         migrations.RunPython(migrate_license_usage), | ||||
|         migrations.RemoveField( | ||||
|             model_name="licenseusage", | ||||
|             name="within_limits", | ||||
|         ), | ||||
|         migrations.AlterField( | ||||
|             model_name="licenseusage", | ||||
|             name="status", | ||||
|             field=models.TextField( | ||||
|                 choices=[ | ||||
|                     ("unlicensed", "Unlicensed"), | ||||
|                     ("valid", "Valid"), | ||||
|                     ("expired", "Expired"), | ||||
|                     ("expiry_soon", "Expiry Soon"), | ||||
|                     ("limit_exceeded_admin", "Limit Exceeded Admin"), | ||||
|                     ("limit_exceeded_user", "Limit Exceeded User"), | ||||
|                     ("read_only", "Read Only"), | ||||
|                 ], | ||||
|             ), | ||||
|             preserve_default=False, | ||||
|         ), | ||||
|         migrations.RenameField( | ||||
|             model_name="licenseusage", | ||||
|             old_name="user_count", | ||||
|             new_name="internal_user_count", | ||||
|         ), | ||||
|     ] | ||||
| @ -17,17 +17,6 @@ if TYPE_CHECKING: | ||||
|     from authentik.enterprise.license import LicenseKey | ||||
|  | ||||
|  | ||||
| def usage_expiry(): | ||||
|     """Keep license usage records for 3 months""" | ||||
|     return now() + timedelta(days=30 * 3) | ||||
|  | ||||
|  | ||||
| THRESHOLD_WARNING_ADMIN_WEEKS = 2 | ||||
| THRESHOLD_WARNING_USER_WEEKS = 4 | ||||
| THRESHOLD_WARNING_EXPIRY_WEEKS = 2 | ||||
| THRESHOLD_READ_ONLY_WEEKS = 6 | ||||
|  | ||||
|  | ||||
| class License(SerializerModel): | ||||
|     """An authentik enterprise license""" | ||||
|  | ||||
| @ -50,7 +39,7 @@ class License(SerializerModel): | ||||
|         """Get parsed license status""" | ||||
|         from authentik.enterprise.license import LicenseKey | ||||
|  | ||||
|         return LicenseKey.validate(self.key, check_expiry=False) | ||||
|         return LicenseKey.validate(self.key) | ||||
|  | ||||
|     class Meta: | ||||
|         indexes = (HashIndex(fields=("key",)),) | ||||
| @ -58,23 +47,9 @@ class License(SerializerModel): | ||||
|         verbose_name_plural = _("Licenses") | ||||
|  | ||||
|  | ||||
| class LicenseUsageStatus(models.TextChoices): | ||||
|     """License states an instance/tenant can be in""" | ||||
|  | ||||
|     UNLICENSED = "unlicensed" | ||||
|     VALID = "valid" | ||||
|     EXPIRED = "expired" | ||||
|     EXPIRY_SOON = "expiry_soon" | ||||
|     # User limit exceeded, 2 week threshold, show message in admin interface | ||||
|     LIMIT_EXCEEDED_ADMIN = "limit_exceeded_admin" | ||||
|     # User limit exceeded, 4 week threshold, show message in user interface | ||||
|     LIMIT_EXCEEDED_USER = "limit_exceeded_user" | ||||
|     READ_ONLY = "read_only" | ||||
|  | ||||
|     @property | ||||
|     def is_valid(self) -> bool: | ||||
|         """Quickly check if a license is valid""" | ||||
|         return self in [LicenseUsageStatus.VALID, LicenseUsageStatus.EXPIRY_SOON] | ||||
| def usage_expiry(): | ||||
|     """Keep license usage records for 3 months""" | ||||
|     return now() + timedelta(days=30 * 3) | ||||
|  | ||||
|  | ||||
| class LicenseUsage(ExpiringModel): | ||||
| @ -84,9 +59,9 @@ class LicenseUsage(ExpiringModel): | ||||
|  | ||||
|     usage_uuid = models.UUIDField(primary_key=True, editable=False, default=uuid4) | ||||
|  | ||||
|     internal_user_count = models.BigIntegerField() | ||||
|     user_count = models.BigIntegerField() | ||||
|     external_user_count = models.BigIntegerField() | ||||
|     status = models.TextField(choices=LicenseUsageStatus.choices) | ||||
|     within_limits = models.BooleanField() | ||||
|  | ||||
|     record_date = models.DateTimeField(auto_now_add=True) | ||||
|  | ||||
|  | ||||
| @ -13,7 +13,7 @@ class EnterprisePolicyAccessView(PolicyAccessView): | ||||
|  | ||||
|     def check_license(self): | ||||
|         """Check license""" | ||||
|         if not LicenseKey.get_total().status().is_valid: | ||||
|         if not LicenseKey.get_total().is_valid(): | ||||
|             return PolicyResult(False, _("Enterprise required to access this feature.")) | ||||
|         if self.request.user.type != UserTypes.INTERNAL: | ||||
|             return PolicyResult(False, _("Feature only accessible for internal users.")) | ||||
|  | ||||
| @ -6,10 +6,7 @@ from authentik.core.api.providers import ProviderSerializer | ||||
| from authentik.core.api.used_by import UsedByMixin | ||||
| from authentik.enterprise.api import EnterpriseRequiredMixin | ||||
| from authentik.enterprise.providers.google_workspace.models import GoogleWorkspaceProvider | ||||
| from authentik.enterprise.providers.google_workspace.tasks import ( | ||||
|     google_workspace_sync, | ||||
|     google_workspace_sync_objects, | ||||
| ) | ||||
| from authentik.enterprise.providers.google_workspace.tasks import google_workspace_sync | ||||
| from authentik.lib.sync.outgoing.api import OutgoingSyncProviderStatusMixin | ||||
|  | ||||
|  | ||||
| @ -55,4 +52,3 @@ class GoogleWorkspaceProviderViewSet(OutgoingSyncProviderStatusMixin, UsedByMixi | ||||
|     search_fields = ["name"] | ||||
|     ordering = ["name"] | ||||
|     sync_single_task = google_workspace_sync | ||||
|     sync_objects_task = google_workspace_sync_objects | ||||
|  | ||||
| @ -181,7 +181,7 @@ class GoogleWorkspaceProviderMapping(PropertyMapping): | ||||
|  | ||||
|     @property | ||||
|     def component(self) -> str: | ||||
|         return "ak-property-mapping-provider-google-workspace-form" | ||||
|         return "ak-property-mapping-google-workspace-form" | ||||
|  | ||||
|     @property | ||||
|     def serializer(self) -> type[Serializer]: | ||||
|  | ||||
| @ -6,10 +6,7 @@ from authentik.core.api.providers import ProviderSerializer | ||||
| from authentik.core.api.used_by import UsedByMixin | ||||
| from authentik.enterprise.api import EnterpriseRequiredMixin | ||||
| from authentik.enterprise.providers.microsoft_entra.models import MicrosoftEntraProvider | ||||
| from authentik.enterprise.providers.microsoft_entra.tasks import ( | ||||
|     microsoft_entra_sync, | ||||
|     microsoft_entra_sync_objects, | ||||
| ) | ||||
| from authentik.enterprise.providers.microsoft_entra.tasks import microsoft_entra_sync | ||||
| from authentik.lib.sync.outgoing.api import OutgoingSyncProviderStatusMixin | ||||
|  | ||||
|  | ||||
| @ -53,4 +50,3 @@ class MicrosoftEntraProviderViewSet(OutgoingSyncProviderStatusMixin, UsedByMixin | ||||
|     search_fields = ["name"] | ||||
|     ordering = ["name"] | ||||
|     sync_single_task = microsoft_entra_sync | ||||
|     sync_objects_task = microsoft_entra_sync_objects | ||||
|  | ||||
| @ -170,7 +170,7 @@ class MicrosoftEntraProviderMapping(PropertyMapping): | ||||
|  | ||||
|     @property | ||||
|     def component(self) -> str: | ||||
|         return "ak-property-mapping-provider-microsoft-entra-form" | ||||
|         return "ak-property-mapping-microsoft-entra-form" | ||||
|  | ||||
|     @property | ||||
|     def serializer(self) -> type[Serializer]: | ||||
|  | ||||
| @ -16,28 +16,13 @@ class RACProviderSerializer(EnterpriseRequiredMixin, ProviderSerializer): | ||||
|  | ||||
|     class Meta: | ||||
|         model = RACProvider | ||||
|         fields = [ | ||||
|             "pk", | ||||
|             "name", | ||||
|             "authentication_flow", | ||||
|             "authorization_flow", | ||||
|             "property_mappings", | ||||
|             "component", | ||||
|             "assigned_application_slug", | ||||
|             "assigned_application_name", | ||||
|             "assigned_backchannel_application_slug", | ||||
|             "assigned_backchannel_application_name", | ||||
|             "verbose_name", | ||||
|             "verbose_name_plural", | ||||
|             "meta_model_name", | ||||
|         fields = ProviderSerializer.Meta.fields + [ | ||||
|             "settings", | ||||
|             "outpost_set", | ||||
|             "connection_expiry", | ||||
|             "delete_token_on_disconnect", | ||||
|         ] | ||||
|         extra_kwargs = { | ||||
|             "authorization_flow": {"required": True, "allow_null": False}, | ||||
|         } | ||||
|         extra_kwargs = ProviderSerializer.Meta.extra_kwargs | ||||
|  | ||||
|  | ||||
| class RACProviderViewSet(UsedByMixin, ModelViewSet): | ||||
|  | ||||
| @ -1,20 +0,0 @@ | ||||
| # Generated by Django 5.0.8 on 2024-08-12 12:54 | ||||
|  | ||||
| from django.db import migrations | ||||
|  | ||||
|  | ||||
| class Migration(migrations.Migration): | ||||
|  | ||||
|     dependencies = [ | ||||
|         ("authentik_providers_rac", "0004_alter_connectiontoken_expires"), | ||||
|     ] | ||||
|  | ||||
|     operations = [ | ||||
|         migrations.AlterModelOptions( | ||||
|             name="racpropertymapping", | ||||
|             options={ | ||||
|                 "verbose_name": "RAC Provider Property Mapping", | ||||
|                 "verbose_name_plural": "RAC Provider Property Mappings", | ||||
|             }, | ||||
|         ), | ||||
|     ] | ||||
| @ -125,7 +125,7 @@ class RACPropertyMapping(PropertyMapping): | ||||
|  | ||||
|     @property | ||||
|     def component(self) -> str: | ||||
|         return "ak-property-mapping-provider-rac-form" | ||||
|         return "ak-property-mapping-rac-form" | ||||
|  | ||||
|     @property | ||||
|     def serializer(self) -> type[Serializer]: | ||||
| @ -136,8 +136,8 @@ class RACPropertyMapping(PropertyMapping): | ||||
|         return RACPropertyMappingSerializer | ||||
|  | ||||
|     class Meta: | ||||
|         verbose_name = _("RAC Provider Property Mapping") | ||||
|         verbose_name_plural = _("RAC Provider Property Mappings") | ||||
|         verbose_name = _("RAC Property Mapping") | ||||
|         verbose_name_plural = _("RAC Property Mappings") | ||||
|  | ||||
|  | ||||
| class ConnectionToken(ExpiringModel): | ||||
|  | ||||
| @ -3,7 +3,7 @@ | ||||
| {% load authentik_core %} | ||||
|  | ||||
| {% block head %} | ||||
| <script src="{% versioned_script 'dist/enterprise/rac/index-%v.js' %}" type="module"></script> | ||||
| {% versioned_script "dist/enterprise/rac/index-%v.js" %} | ||||
| <meta name="theme-color" content="#18191a" media="(prefers-color-scheme: dark)"> | ||||
| <meta name="theme-color" content="#ffffff" media="(prefers-color-scheme: light)"> | ||||
| <link rel="icon" href="{{ tenant.branding_favicon }}"> | ||||
|  | ||||
| @ -1,46 +0,0 @@ | ||||
| """Test RAC Provider""" | ||||
|  | ||||
| from datetime import timedelta | ||||
| from time import mktime | ||||
| from unittest.mock import MagicMock, patch | ||||
|  | ||||
| from django.urls import reverse | ||||
| from django.utils.timezone import now | ||||
| from rest_framework.test import APITestCase | ||||
|  | ||||
| from authentik.core.tests.utils import create_test_admin_user, create_test_flow | ||||
| from authentik.enterprise.license import LicenseKey | ||||
| from authentik.enterprise.models import License | ||||
| from authentik.lib.generators import generate_id | ||||
|  | ||||
|  | ||||
| class TestAPI(APITestCase): | ||||
|     """Test Provider API""" | ||||
|  | ||||
|     def setUp(self) -> None: | ||||
|         self.user = create_test_admin_user() | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=int(mktime((now() + timedelta(days=3000)).timetuple())), | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     def test_create(self): | ||||
|         """Test creation of RAC Provider""" | ||||
|         License.objects.create(key=generate_id()) | ||||
|         self.client.force_login(self.user) | ||||
|         response = self.client.post( | ||||
|             reverse("authentik_api:racprovider-list"), | ||||
|             data={ | ||||
|                 "name": generate_id(), | ||||
|                 "authorization_flow": create_test_flow().pk, | ||||
|             }, | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 201) | ||||
| @ -3,6 +3,7 @@ | ||||
| from channels.auth import AuthMiddleware | ||||
| from channels.sessions import CookieMiddleware | ||||
| from django.urls import path | ||||
| from django.views.decorators.csrf import ensure_csrf_cookie | ||||
|  | ||||
| from authentik.enterprise.providers.rac.api.connection_tokens import ConnectionTokenViewSet | ||||
| from authentik.enterprise.providers.rac.api.endpoints import EndpointViewSet | ||||
| @ -18,12 +19,12 @@ from authentik.root.middleware import ChannelsLoggingMiddleware | ||||
| urlpatterns = [ | ||||
|     path( | ||||
|         "application/rac/<slug:app>/<uuid:endpoint>/", | ||||
|         RACStartView.as_view(), | ||||
|         ensure_csrf_cookie(RACStartView.as_view()), | ||||
|         name="start", | ||||
|     ), | ||||
|     path( | ||||
|         "if/rac/<str:token>/", | ||||
|         RACInterface.as_view(), | ||||
|         ensure_csrf_cookie(RACInterface.as_view()), | ||||
|         name="if-rac", | ||||
|     ), | ||||
| ] | ||||
| @ -43,7 +44,7 @@ websocket_urlpatterns = [ | ||||
|  | ||||
| api_urlpatterns = [ | ||||
|     ("providers/rac", RACProviderViewSet), | ||||
|     ("propertymappings/provider/rac", RACPropertyMappingViewSet), | ||||
|     ("propertymappings/rac", RACPropertyMappingViewSet), | ||||
|     ("rac/endpoints", EndpointViewSet), | ||||
|     ("rac/connection_tokens", ConnectionTokenViewSet), | ||||
| ] | ||||
|  | ||||
| @ -17,7 +17,6 @@ TENANT_APPS = [ | ||||
|     "authentik.enterprise.providers.google_workspace", | ||||
|     "authentik.enterprise.providers.microsoft_entra", | ||||
|     "authentik.enterprise.providers.rac", | ||||
|     "authentik.enterprise.stages.authenticator_endpoint_gdtc", | ||||
|     "authentik.enterprise.stages.source", | ||||
| ] | ||||
|  | ||||
|  | ||||
| @ -3,7 +3,7 @@ | ||||
| from datetime import datetime | ||||
|  | ||||
| from django.core.cache import cache | ||||
| from django.db.models.signals import post_delete, post_save, pre_save | ||||
| from django.db.models.signals import post_save, pre_save | ||||
| from django.dispatch import receiver | ||||
| from django.utils.timezone import get_current_timezone | ||||
|  | ||||
| @ -27,9 +27,3 @@ def post_save_license(sender: type[License], instance: License, **_): | ||||
|     """Trigger license usage calculation when license is saved""" | ||||
|     cache.delete(CACHE_KEY_ENTERPRISE_LICENSE) | ||||
|     enterprise_update_usage.delay() | ||||
|  | ||||
|  | ||||
| @receiver(post_delete, sender=License) | ||||
| def post_delete_license(sender: type[License], instance: License, **_): | ||||
|     """Clear license cache when license is deleted""" | ||||
|     cache.delete(CACHE_KEY_ENTERPRISE_LICENSE) | ||||
|  | ||||
| @ -1,82 +0,0 @@ | ||||
| """AuthenticatorEndpointGDTCStage API Views""" | ||||
|  | ||||
| from django_filters.rest_framework.backends import DjangoFilterBackend | ||||
| from rest_framework import mixins | ||||
| from rest_framework.filters import OrderingFilter, SearchFilter | ||||
| from rest_framework.permissions import IsAdminUser | ||||
| from rest_framework.serializers import ModelSerializer | ||||
| from rest_framework.viewsets import GenericViewSet, ModelViewSet | ||||
| from structlog.stdlib import get_logger | ||||
|  | ||||
| from authentik.api.authorization import OwnerFilter, OwnerPermissions | ||||
| from authentik.core.api.used_by import UsedByMixin | ||||
| from authentik.enterprise.api import EnterpriseRequiredMixin | ||||
| from authentik.enterprise.stages.authenticator_endpoint_gdtc.models import ( | ||||
|     AuthenticatorEndpointGDTCStage, | ||||
|     EndpointDevice, | ||||
| ) | ||||
| from authentik.flows.api.stages import StageSerializer | ||||
|  | ||||
| LOGGER = get_logger() | ||||
|  | ||||
|  | ||||
| class AuthenticatorEndpointGDTCStageSerializer(EnterpriseRequiredMixin, StageSerializer): | ||||
|     """AuthenticatorEndpointGDTCStage Serializer""" | ||||
|  | ||||
|     class Meta: | ||||
|         model = AuthenticatorEndpointGDTCStage | ||||
|         fields = StageSerializer.Meta.fields + [ | ||||
|             "configure_flow", | ||||
|             "friendly_name", | ||||
|             "credentials", | ||||
|         ] | ||||
|  | ||||
|  | ||||
| class AuthenticatorEndpointGDTCStageViewSet(UsedByMixin, ModelViewSet): | ||||
|     """AuthenticatorEndpointGDTCStage Viewset""" | ||||
|  | ||||
|     queryset = AuthenticatorEndpointGDTCStage.objects.all() | ||||
|     serializer_class = AuthenticatorEndpointGDTCStageSerializer | ||||
|     filterset_fields = [ | ||||
|         "name", | ||||
|         "configure_flow", | ||||
|     ] | ||||
|     search_fields = ["name"] | ||||
|     ordering = ["name"] | ||||
|  | ||||
|  | ||||
| class EndpointDeviceSerializer(ModelSerializer): | ||||
|     """Serializer for Endpoint authenticator devices""" | ||||
|  | ||||
|     class Meta: | ||||
|         model = EndpointDevice | ||||
|         fields = ["pk", "name"] | ||||
|         depth = 2 | ||||
|  | ||||
|  | ||||
| class EndpointDeviceViewSet( | ||||
|     mixins.RetrieveModelMixin, | ||||
|     mixins.ListModelMixin, | ||||
|     UsedByMixin, | ||||
|     GenericViewSet, | ||||
| ): | ||||
|     """Viewset for Endpoint authenticator devices""" | ||||
|  | ||||
|     queryset = EndpointDevice.objects.all() | ||||
|     serializer_class = EndpointDeviceSerializer | ||||
|     search_fields = ["name"] | ||||
|     filterset_fields = ["name"] | ||||
|     ordering = ["name"] | ||||
|     permission_classes = [OwnerPermissions] | ||||
|     filter_backends = [OwnerFilter, DjangoFilterBackend, OrderingFilter, SearchFilter] | ||||
|  | ||||
|  | ||||
| class EndpointAdminDeviceViewSet(ModelViewSet): | ||||
|     """Viewset for Endpoint authenticator devices (for admins)""" | ||||
|  | ||||
|     permission_classes = [IsAdminUser] | ||||
|     queryset = EndpointDevice.objects.all() | ||||
|     serializer_class = EndpointDeviceSerializer | ||||
|     search_fields = ["name"] | ||||
|     filterset_fields = ["name"] | ||||
|     ordering = ["name"] | ||||
| @ -1,13 +0,0 @@ | ||||
| """authentik Endpoint app config""" | ||||
|  | ||||
| from authentik.enterprise.apps import EnterpriseConfig | ||||
|  | ||||
|  | ||||
| class AuthentikStageAuthenticatorEndpointConfig(EnterpriseConfig): | ||||
|     """authentik endpoint config""" | ||||
|  | ||||
|     name = "authentik.enterprise.stages.authenticator_endpoint_gdtc" | ||||
|     label = "authentik_stages_authenticator_endpoint_gdtc" | ||||
|     verbose_name = "authentik Enterprise.Stages.Authenticator.Endpoint GDTC" | ||||
|     default = True | ||||
|     mountpoint = "endpoint/gdtc/" | ||||
| @ -1,115 +0,0 @@ | ||||
| # Generated by Django 5.0.9 on 2024-10-22 11:40 | ||||
|  | ||||
| import django.db.models.deletion | ||||
| import uuid | ||||
| from django.conf import settings | ||||
| from django.db import migrations, models | ||||
|  | ||||
|  | ||||
| class Migration(migrations.Migration): | ||||
|  | ||||
|     initial = True | ||||
|  | ||||
|     dependencies = [ | ||||
|         ("authentik_flows", "0027_auto_20231028_1424"), | ||||
|         migrations.swappable_dependency(settings.AUTH_USER_MODEL), | ||||
|     ] | ||||
|  | ||||
|     operations = [ | ||||
|         migrations.CreateModel( | ||||
|             name="AuthenticatorEndpointGDTCStage", | ||||
|             fields=[ | ||||
|                 ( | ||||
|                     "stage_ptr", | ||||
|                     models.OneToOneField( | ||||
|                         auto_created=True, | ||||
|                         on_delete=django.db.models.deletion.CASCADE, | ||||
|                         parent_link=True, | ||||
|                         primary_key=True, | ||||
|                         serialize=False, | ||||
|                         to="authentik_flows.stage", | ||||
|                     ), | ||||
|                 ), | ||||
|                 ("friendly_name", models.TextField(null=True)), | ||||
|                 ("credentials", models.JSONField()), | ||||
|                 ( | ||||
|                     "configure_flow", | ||||
|                     models.ForeignKey( | ||||
|                         blank=True, | ||||
|                         help_text="Flow used by an authenticated user to configure this Stage. If empty, user will not be able to configure this stage.", | ||||
|                         null=True, | ||||
|                         on_delete=django.db.models.deletion.SET_NULL, | ||||
|                         to="authentik_flows.flow", | ||||
|                     ), | ||||
|                 ), | ||||
|             ], | ||||
|             options={ | ||||
|                 "verbose_name": "Endpoint Authenticator Google Device Trust Connector Stage", | ||||
|                 "verbose_name_plural": "Endpoint Authenticator Google Device Trust Connector Stages", | ||||
|             }, | ||||
|             bases=("authentik_flows.stage", models.Model), | ||||
|         ), | ||||
|         migrations.CreateModel( | ||||
|             name="EndpointDevice", | ||||
|             fields=[ | ||||
|                 ("created", models.DateTimeField(auto_now_add=True)), | ||||
|                 ("last_updated", models.DateTimeField(auto_now=True)), | ||||
|                 ( | ||||
|                     "name", | ||||
|                     models.CharField( | ||||
|                         help_text="The human-readable name of this device.", max_length=64 | ||||
|                     ), | ||||
|                 ), | ||||
|                 ( | ||||
|                     "confirmed", | ||||
|                     models.BooleanField(default=True, help_text="Is this device ready for use?"), | ||||
|                 ), | ||||
|                 ("last_used", models.DateTimeField(null=True)), | ||||
|                 ("uuid", models.UUIDField(default=uuid.uuid4, primary_key=True, serialize=False)), | ||||
|                 ( | ||||
|                     "host_identifier", | ||||
|                     models.TextField( | ||||
|                         help_text="A unique identifier for the endpoint device, usually the device serial number", | ||||
|                         unique=True, | ||||
|                     ), | ||||
|                 ), | ||||
|                 ("data", models.JSONField()), | ||||
|                 ( | ||||
|                     "user", | ||||
|                     models.ForeignKey( | ||||
|                         on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL | ||||
|                     ), | ||||
|                 ), | ||||
|             ], | ||||
|             options={ | ||||
|                 "verbose_name": "Endpoint Device", | ||||
|                 "verbose_name_plural": "Endpoint Devices", | ||||
|             }, | ||||
|         ), | ||||
|         migrations.CreateModel( | ||||
|             name="EndpointDeviceConnection", | ||||
|             fields=[ | ||||
|                 ( | ||||
|                     "id", | ||||
|                     models.AutoField( | ||||
|                         auto_created=True, primary_key=True, serialize=False, verbose_name="ID" | ||||
|                     ), | ||||
|                 ), | ||||
|                 ("attributes", models.JSONField()), | ||||
|                 ( | ||||
|                     "device", | ||||
|                     models.ForeignKey( | ||||
|                         on_delete=django.db.models.deletion.CASCADE, | ||||
|                         to="authentik_stages_authenticator_endpoint_gdtc.endpointdevice", | ||||
|                     ), | ||||
|                 ), | ||||
|                 ( | ||||
|                     "stage", | ||||
|                     models.ForeignKey( | ||||
|                         on_delete=django.db.models.deletion.CASCADE, | ||||
|                         to="authentik_stages_authenticator_endpoint_gdtc.authenticatorendpointgdtcstage", | ||||
|                     ), | ||||
|                 ), | ||||
|             ], | ||||
|         ), | ||||
|     ] | ||||
| @ -1,101 +0,0 @@ | ||||
| """Endpoint stage""" | ||||
|  | ||||
| from uuid import uuid4 | ||||
|  | ||||
| from django.contrib.auth import get_user_model | ||||
| from django.db import models | ||||
| from django.utils.translation import gettext_lazy as _ | ||||
| from google.oauth2.service_account import Credentials | ||||
| from rest_framework.serializers import BaseSerializer, Serializer | ||||
|  | ||||
| from authentik.core.types import UserSettingSerializer | ||||
| from authentik.flows.models import ConfigurableStage, FriendlyNamedStage, Stage | ||||
| from authentik.flows.stage import StageView | ||||
| from authentik.lib.models import SerializerModel | ||||
| from authentik.stages.authenticator.models import Device | ||||
|  | ||||
|  | ||||
| class AuthenticatorEndpointGDTCStage(ConfigurableStage, FriendlyNamedStage, Stage): | ||||
|     """Setup Google Chrome Device-trust connection""" | ||||
|  | ||||
|     credentials = models.JSONField() | ||||
|  | ||||
|     def google_credentials(self): | ||||
|         return { | ||||
|             "credentials": Credentials.from_service_account_info( | ||||
|                 self.credentials, scopes=["https://www.googleapis.com/auth/verifiedaccess"] | ||||
|             ), | ||||
|         } | ||||
|  | ||||
|     @property | ||||
|     def serializer(self) -> type[BaseSerializer]: | ||||
|         from authentik.enterprise.stages.authenticator_endpoint_gdtc.api import ( | ||||
|             AuthenticatorEndpointGDTCStageSerializer, | ||||
|         ) | ||||
|  | ||||
|         return AuthenticatorEndpointGDTCStageSerializer | ||||
|  | ||||
|     @property | ||||
|     def view(self) -> type[StageView]: | ||||
|         from authentik.enterprise.stages.authenticator_endpoint_gdtc.stage import ( | ||||
|             AuthenticatorEndpointStageView, | ||||
|         ) | ||||
|  | ||||
|         return AuthenticatorEndpointStageView | ||||
|  | ||||
|     @property | ||||
|     def component(self) -> str: | ||||
|         return "ak-stage-authenticator-endpoint-gdtc-form" | ||||
|  | ||||
|     def ui_user_settings(self) -> UserSettingSerializer | None: | ||||
|         return UserSettingSerializer( | ||||
|             data={ | ||||
|                 "title": self.friendly_name or str(self._meta.verbose_name), | ||||
|                 "component": "ak-user-settings-authenticator-endpoint", | ||||
|             } | ||||
|         ) | ||||
|  | ||||
|     def __str__(self) -> str: | ||||
|         return f"Endpoint Authenticator Google Device Trust Connector Stage {self.name}" | ||||
|  | ||||
|     class Meta: | ||||
|         verbose_name = _("Endpoint Authenticator Google Device Trust Connector Stage") | ||||
|         verbose_name_plural = _("Endpoint Authenticator Google Device Trust Connector Stages") | ||||
|  | ||||
|  | ||||
| class EndpointDevice(SerializerModel, Device): | ||||
|     """Endpoint Device for a single user""" | ||||
|  | ||||
|     uuid = models.UUIDField(primary_key=True, default=uuid4) | ||||
|     host_identifier = models.TextField( | ||||
|         unique=True, | ||||
|         help_text="A unique identifier for the endpoint device, usually the device serial number", | ||||
|     ) | ||||
|  | ||||
|     user = models.ForeignKey(get_user_model(), on_delete=models.CASCADE) | ||||
|     data = models.JSONField() | ||||
|  | ||||
|     @property | ||||
|     def serializer(self) -> Serializer: | ||||
|         from authentik.enterprise.stages.authenticator_endpoint_gdtc.api import ( | ||||
|             EndpointDeviceSerializer, | ||||
|         ) | ||||
|  | ||||
|         return EndpointDeviceSerializer | ||||
|  | ||||
|     def __str__(self): | ||||
|         return str(self.name) or str(self.user_id) | ||||
|  | ||||
|     class Meta: | ||||
|         verbose_name = _("Endpoint Device") | ||||
|         verbose_name_plural = _("Endpoint Devices") | ||||
|  | ||||
|  | ||||
| class EndpointDeviceConnection(models.Model): | ||||
|     device = models.ForeignKey(EndpointDevice, on_delete=models.CASCADE) | ||||
|     stage = models.ForeignKey(AuthenticatorEndpointGDTCStage, on_delete=models.CASCADE) | ||||
|  | ||||
|     attributes = models.JSONField() | ||||
|  | ||||
|     def __str__(self) -> str: | ||||
|         return f"Endpoint device connection {self.device_id} to {self.stage_id}" | ||||
| @ -1,32 +0,0 @@ | ||||
| from django.http import HttpResponse | ||||
| from django.urls import reverse | ||||
| from django.utils.translation import gettext_lazy as _ | ||||
|  | ||||
| from authentik.flows.challenge import ( | ||||
|     Challenge, | ||||
|     ChallengeResponse, | ||||
|     FrameChallenge, | ||||
|     FrameChallengeResponse, | ||||
| ) | ||||
| from authentik.flows.stage import ChallengeStageView | ||||
|  | ||||
|  | ||||
| class AuthenticatorEndpointStageView(ChallengeStageView): | ||||
|     """Endpoint stage""" | ||||
|  | ||||
|     response_class = FrameChallengeResponse | ||||
|  | ||||
|     def get_challenge(self, *args, **kwargs) -> Challenge: | ||||
|         return FrameChallenge( | ||||
|             data={ | ||||
|                 "component": "xak-flow-frame", | ||||
|                 "url": self.request.build_absolute_uri( | ||||
|                     reverse("authentik_stages_authenticator_endpoint_gdtc:chrome") | ||||
|                 ), | ||||
|                 "loading_overlay": True, | ||||
|                 "loading_text": _("Verifying your browser..."), | ||||
|             } | ||||
|         ) | ||||
|  | ||||
|     def challenge_valid(self, response: ChallengeResponse) -> HttpResponse: | ||||
|         return self.executor.stage_ok() | ||||
| @ -1,9 +0,0 @@ | ||||
| <html> | ||||
| <script> | ||||
|   window.parent.postMessage({ | ||||
|     message: "submit", | ||||
|     source: "goauthentik.io", | ||||
|     context: "flow-executor" | ||||
|   }); | ||||
| </script> | ||||
| </html> | ||||
| @ -1,26 +0,0 @@ | ||||
| """API URLs""" | ||||
|  | ||||
| from django.urls import path | ||||
|  | ||||
| from authentik.enterprise.stages.authenticator_endpoint_gdtc.api import ( | ||||
|     AuthenticatorEndpointGDTCStageViewSet, | ||||
|     EndpointAdminDeviceViewSet, | ||||
|     EndpointDeviceViewSet, | ||||
| ) | ||||
| from authentik.enterprise.stages.authenticator_endpoint_gdtc.views.dtc import ( | ||||
|     GoogleChromeDeviceTrustConnector, | ||||
| ) | ||||
|  | ||||
| urlpatterns = [ | ||||
|     path("chrome/", GoogleChromeDeviceTrustConnector.as_view(), name="chrome"), | ||||
| ] | ||||
|  | ||||
| api_urlpatterns = [ | ||||
|     ("authenticators/endpoint", EndpointDeviceViewSet), | ||||
|     ( | ||||
|         "authenticators/admin/endpoint", | ||||
|         EndpointAdminDeviceViewSet, | ||||
|         "admin-endpointdevice", | ||||
|     ), | ||||
|     ("stages/authenticator/endpoint_gdtc", AuthenticatorEndpointGDTCStageViewSet), | ||||
| ] | ||||
| @ -1,84 +0,0 @@ | ||||
| from json import dumps, loads | ||||
| from typing import Any | ||||
|  | ||||
| from django.http import HttpRequest, HttpResponse, HttpResponseRedirect | ||||
| from django.template.response import TemplateResponse | ||||
| from django.urls import reverse | ||||
| from django.views import View | ||||
| from googleapiclient.discovery import build | ||||
|  | ||||
| from authentik.enterprise.stages.authenticator_endpoint_gdtc.models import ( | ||||
|     AuthenticatorEndpointGDTCStage, | ||||
|     EndpointDevice, | ||||
|     EndpointDeviceConnection, | ||||
| ) | ||||
| from authentik.flows.planner import PLAN_CONTEXT_PENDING_USER, FlowPlan | ||||
| from authentik.flows.views.executor import SESSION_KEY_PLAN | ||||
| from authentik.stages.password.stage import PLAN_CONTEXT_METHOD, PLAN_CONTEXT_METHOD_ARGS | ||||
|  | ||||
| # Header we get from chrome that initiates verified access | ||||
| HEADER_DEVICE_TRUST = "X-Device-Trust" | ||||
| # Header we send to the client with the challenge | ||||
| HEADER_ACCESS_CHALLENGE = "X-Verified-Access-Challenge" | ||||
| # Header we get back from the client that we verify with google | ||||
| HEADER_ACCESS_CHALLENGE_RESPONSE = "X-Verified-Access-Challenge-Response" | ||||
| # Header value for x-device-trust that initiates the flow | ||||
| DEVICE_TRUST_VERIFIED_ACCESS = "VerifiedAccess" | ||||
|  | ||||
|  | ||||
| class GoogleChromeDeviceTrustConnector(View): | ||||
|     """Google Chrome Device-trust connector based endpoint authenticator""" | ||||
|  | ||||
|     def get_flow_plan(self) -> FlowPlan: | ||||
|         flow_plan: FlowPlan = self.request.session[SESSION_KEY_PLAN] | ||||
|         return flow_plan | ||||
|  | ||||
|     def setup(self, request: HttpRequest, *args: Any, **kwargs: Any) -> None: | ||||
|         super().setup(request, *args, **kwargs) | ||||
|         stage: AuthenticatorEndpointGDTCStage = self.get_flow_plan().bindings[0].stage | ||||
|         self.google_client = build( | ||||
|             "verifiedaccess", | ||||
|             "v2", | ||||
|             cache_discovery=False, | ||||
|             **stage.google_credentials(), | ||||
|         ) | ||||
|  | ||||
|     def get(self, request: HttpRequest) -> HttpResponse: | ||||
|         x_device_trust = request.headers.get(HEADER_DEVICE_TRUST) | ||||
|         x_access_challenge_response = request.headers.get(HEADER_ACCESS_CHALLENGE_RESPONSE) | ||||
|         if x_device_trust == "VerifiedAccess" and x_access_challenge_response is None: | ||||
|             challenge = self.google_client.challenge().generate().execute() | ||||
|             res = HttpResponseRedirect( | ||||
|                 self.request.build_absolute_uri( | ||||
|                     reverse("authentik_stages_authenticator_endpoint_gdtc:chrome") | ||||
|                 ) | ||||
|             ) | ||||
|             res[HEADER_ACCESS_CHALLENGE] = dumps(challenge) | ||||
|             return res | ||||
|         if x_access_challenge_response: | ||||
|             response = ( | ||||
|                 self.google_client.challenge() | ||||
|                 .verify(body=loads(x_access_challenge_response)) | ||||
|                 .execute() | ||||
|             ) | ||||
|             # Remove deprecated string representation of deviceSignals | ||||
|             response.pop("deviceSignal", None) | ||||
|             flow_plan: FlowPlan = self.get_flow_plan() | ||||
|             device, _ = EndpointDevice.objects.update_or_create( | ||||
|                 host_identifier=response["deviceSignals"]["serialNumber"], | ||||
|                 user=flow_plan.context.get(PLAN_CONTEXT_PENDING_USER), | ||||
|                 defaults={"name": response["deviceSignals"]["hostname"], "data": response}, | ||||
|             ) | ||||
|             EndpointDeviceConnection.objects.update_or_create( | ||||
|                 device=device, | ||||
|                 stage=flow_plan.bindings[0].stage, | ||||
|                 defaults={ | ||||
|                     "attributes": response, | ||||
|                 }, | ||||
|             ) | ||||
|             flow_plan.context.setdefault(PLAN_CONTEXT_METHOD, "trusted_endpoint") | ||||
|             flow_plan.context.setdefault(PLAN_CONTEXT_METHOD_ARGS, {}) | ||||
|             flow_plan.context[PLAN_CONTEXT_METHOD_ARGS].setdefault("endpoints", []) | ||||
|             flow_plan.context[PLAN_CONTEXT_METHOD_ARGS]["endpoints"].append(response) | ||||
|             request.session[SESSION_KEY_PLAN] = flow_plan | ||||
|         return TemplateResponse(request, "stages/authenticator_endpoint/google_chrome_dtc.html") | ||||
| @ -9,26 +9,10 @@ from django.utils.timezone import now | ||||
| from rest_framework.exceptions import ValidationError | ||||
|  | ||||
| from authentik.enterprise.license import LicenseKey | ||||
| from authentik.enterprise.models import ( | ||||
|     THRESHOLD_READ_ONLY_WEEKS, | ||||
|     THRESHOLD_WARNING_ADMIN_WEEKS, | ||||
|     THRESHOLD_WARNING_USER_WEEKS, | ||||
|     License, | ||||
|     LicenseUsage, | ||||
|     LicenseUsageStatus, | ||||
| ) | ||||
| from authentik.enterprise.models import License | ||||
| from authentik.lib.generators import generate_id | ||||
|  | ||||
| # Valid license expiry | ||||
| expiry_valid = int(mktime((now() + timedelta(days=3000)).timetuple())) | ||||
| # Valid license expiry, expires soon | ||||
| expiry_soon = int(mktime((now() + timedelta(hours=10)).timetuple())) | ||||
| # Invalid license expiry, recently expired | ||||
| expiry_expired = int(mktime((now() - timedelta(hours=10)).timetuple())) | ||||
| # Invalid license expiry, expired longer ago | ||||
| expiry_expired_read_only = int( | ||||
|     mktime((now() - timedelta(weeks=THRESHOLD_READ_ONLY_WEEKS + 1)).timetuple()) | ||||
| ) | ||||
| _exp = int(mktime((now() + timedelta(days=3000)).timetuple())) | ||||
|  | ||||
|  | ||||
| class TestEnterpriseLicense(TestCase): | ||||
| @ -39,7 +23,7 @@ class TestEnterpriseLicense(TestCase): | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_valid, | ||||
|                 exp=_exp, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
| @ -49,7 +33,7 @@ class TestEnterpriseLicense(TestCase): | ||||
|     def test_valid(self): | ||||
|         """Check license verification""" | ||||
|         lic = License.objects.create(key=generate_id()) | ||||
|         self.assertTrue(lic.status.status().is_valid) | ||||
|         self.assertTrue(lic.status.is_valid()) | ||||
|         self.assertEqual(lic.internal_users, 100) | ||||
|  | ||||
|     def test_invalid(self): | ||||
| @ -62,7 +46,7 @@ class TestEnterpriseLicense(TestCase): | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_valid, | ||||
|                 exp=_exp, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
| @ -72,186 +56,11 @@ class TestEnterpriseLicense(TestCase): | ||||
|     def test_valid_multiple(self): | ||||
|         """Check license verification""" | ||||
|         lic = License.objects.create(key=generate_id()) | ||||
|         self.assertTrue(lic.status.status().is_valid) | ||||
|         self.assertTrue(lic.status.is_valid()) | ||||
|         lic2 = License.objects.create(key=generate_id()) | ||||
|         self.assertTrue(lic2.status.status().is_valid) | ||||
|         self.assertTrue(lic2.status.is_valid()) | ||||
|         total = LicenseKey.get_total() | ||||
|         self.assertEqual(total.internal_users, 200) | ||||
|         self.assertEqual(total.external_users, 200) | ||||
|         self.assertEqual(total.exp, expiry_valid) | ||||
|         self.assertTrue(total.status().is_valid) | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_valid, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_internal_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_external_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.record_usage", | ||||
|         MagicMock(), | ||||
|     ) | ||||
|     def test_limit_exceeded_read_only(self): | ||||
|         """Check license verification""" | ||||
|         License.objects.create(key=generate_id()) | ||||
|         usage = LicenseUsage.objects.create( | ||||
|             internal_user_count=100, | ||||
|             external_user_count=100, | ||||
|             status=LicenseUsageStatus.VALID, | ||||
|         ) | ||||
|         usage.record_date = now() - timedelta(weeks=THRESHOLD_READ_ONLY_WEEKS + 1) | ||||
|         usage.save(update_fields=["record_date"]) | ||||
|         self.assertEqual(LicenseKey.get_total().summary().status, LicenseUsageStatus.READ_ONLY) | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_valid, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_internal_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_external_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.record_usage", | ||||
|         MagicMock(), | ||||
|     ) | ||||
|     def test_limit_exceeded_user_warning(self): | ||||
|         """Check license verification""" | ||||
|         License.objects.create(key=generate_id()) | ||||
|         usage = LicenseUsage.objects.create( | ||||
|             internal_user_count=100, | ||||
|             external_user_count=100, | ||||
|             status=LicenseUsageStatus.VALID, | ||||
|         ) | ||||
|         usage.record_date = now() - timedelta(weeks=THRESHOLD_WARNING_USER_WEEKS + 1) | ||||
|         usage.save(update_fields=["record_date"]) | ||||
|         self.assertEqual( | ||||
|             LicenseKey.get_total().summary().status, LicenseUsageStatus.LIMIT_EXCEEDED_USER | ||||
|         ) | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_valid, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_internal_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_external_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.record_usage", | ||||
|         MagicMock(), | ||||
|     ) | ||||
|     def test_limit_exceeded_admin_warning(self): | ||||
|         """Check license verification""" | ||||
|         License.objects.create(key=generate_id()) | ||||
|         usage = LicenseUsage.objects.create( | ||||
|             internal_user_count=100, | ||||
|             external_user_count=100, | ||||
|             status=LicenseUsageStatus.VALID, | ||||
|         ) | ||||
|         usage.record_date = now() - timedelta(weeks=THRESHOLD_WARNING_ADMIN_WEEKS + 1) | ||||
|         usage.save(update_fields=["record_date"]) | ||||
|         self.assertEqual( | ||||
|             LicenseKey.get_total().summary().status, LicenseUsageStatus.LIMIT_EXCEEDED_ADMIN | ||||
|         ) | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_expired_read_only, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.record_usage", | ||||
|         MagicMock(), | ||||
|     ) | ||||
|     def test_expiry_read_only(self): | ||||
|         """Check license verification""" | ||||
|         License.objects.create(key=generate_id()) | ||||
|         self.assertEqual(LicenseKey.get_total().summary().status, LicenseUsageStatus.READ_ONLY) | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_expired, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.record_usage", | ||||
|         MagicMock(), | ||||
|     ) | ||||
|     def test_expiry_expired(self): | ||||
|         """Check license verification""" | ||||
|         License.objects.create(key=generate_id()) | ||||
|         self.assertEqual(LicenseKey.get_total().summary().status, LicenseUsageStatus.EXPIRED) | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_soon, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.record_usage", | ||||
|         MagicMock(), | ||||
|     ) | ||||
|     def test_expiry_soon(self): | ||||
|         """Check license verification""" | ||||
|         License.objects.create(key=generate_id()) | ||||
|         self.assertEqual(LicenseKey.get_total().summary().status, LicenseUsageStatus.EXPIRY_SOON) | ||||
|         self.assertEqual(total.exp, _exp) | ||||
|         self.assertTrue(total.is_valid()) | ||||
|  | ||||
| @ -1,217 +0,0 @@ | ||||
| """read only tests""" | ||||
|  | ||||
| from datetime import timedelta | ||||
| from unittest.mock import MagicMock, patch | ||||
|  | ||||
| from django.urls import reverse | ||||
| from django.utils.timezone import now | ||||
|  | ||||
| from authentik.core.tests.utils import create_test_admin_user, create_test_flow, create_test_user | ||||
| from authentik.enterprise.license import LicenseKey | ||||
| from authentik.enterprise.models import ( | ||||
|     THRESHOLD_READ_ONLY_WEEKS, | ||||
|     License, | ||||
|     LicenseUsage, | ||||
|     LicenseUsageStatus, | ||||
| ) | ||||
| from authentik.enterprise.tests.test_license import expiry_valid | ||||
| from authentik.flows.models import ( | ||||
|     FlowDesignation, | ||||
|     FlowStageBinding, | ||||
| ) | ||||
| from authentik.flows.tests import FlowTestCase | ||||
| from authentik.lib.generators import generate_id | ||||
| from authentik.stages.identification.models import IdentificationStage, UserFields | ||||
| from authentik.stages.password import BACKEND_INBUILT | ||||
| from authentik.stages.password.models import PasswordStage | ||||
| from authentik.stages.user_login.models import UserLoginStage | ||||
|  | ||||
|  | ||||
| class TestReadOnly(FlowTestCase): | ||||
|     """Test read_only""" | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_valid, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_internal_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_external_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.record_usage", | ||||
|         MagicMock(), | ||||
|     ) | ||||
|     def test_login(self): | ||||
|         """Test flow, ensure login is still possible with read only mode""" | ||||
|         License.objects.create(key=generate_id()) | ||||
|         usage = LicenseUsage.objects.create( | ||||
|             internal_user_count=100, | ||||
|             external_user_count=100, | ||||
|             status=LicenseUsageStatus.VALID, | ||||
|         ) | ||||
|         usage.record_date = now() - timedelta(weeks=THRESHOLD_READ_ONLY_WEEKS + 1) | ||||
|         usage.save(update_fields=["record_date"]) | ||||
|  | ||||
|         flow = create_test_flow( | ||||
|             FlowDesignation.AUTHENTICATION, | ||||
|         ) | ||||
|  | ||||
|         ident_stage = IdentificationStage.objects.create( | ||||
|             name=generate_id(), | ||||
|             user_fields=[UserFields.E_MAIL], | ||||
|             pretend_user_exists=False, | ||||
|         ) | ||||
|         FlowStageBinding.objects.create( | ||||
|             target=flow, | ||||
|             stage=ident_stage, | ||||
|             order=0, | ||||
|         ) | ||||
|         password_stage = PasswordStage.objects.create( | ||||
|             name=generate_id(), backends=[BACKEND_INBUILT] | ||||
|         ) | ||||
|         FlowStageBinding.objects.create( | ||||
|             target=flow, | ||||
|             stage=password_stage, | ||||
|             order=1, | ||||
|         ) | ||||
|         login_stage = UserLoginStage.objects.create( | ||||
|             name=generate_id(), | ||||
|         ) | ||||
|         FlowStageBinding.objects.create( | ||||
|             target=flow, | ||||
|             stage=login_stage, | ||||
|             order=2, | ||||
|         ) | ||||
|  | ||||
|         user = create_test_user() | ||||
|  | ||||
|         exec_url = reverse("authentik_api:flow-executor", kwargs={"flow_slug": flow.slug}) | ||||
|         response = self.client.get(exec_url) | ||||
|         self.assertStageResponse( | ||||
|             response, | ||||
|             flow, | ||||
|             component="ak-stage-identification", | ||||
|             password_fields=False, | ||||
|             primary_action="Log in", | ||||
|             sources=[], | ||||
|             show_source_labels=False, | ||||
|             user_fields=[UserFields.E_MAIL], | ||||
|         ) | ||||
|         response = self.client.post(exec_url, {"uid_field": user.email}, follow=True) | ||||
|         self.assertStageResponse(response, flow, component="ak-stage-password") | ||||
|         response = self.client.post(exec_url, {"password": user.username}, follow=True) | ||||
|         self.assertStageRedirects(response, reverse("authentik_core:root-redirect")) | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_valid, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_internal_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_external_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.record_usage", | ||||
|         MagicMock(), | ||||
|     ) | ||||
|     def test_manage_licenses(self): | ||||
|         """Test that managing licenses is still possible""" | ||||
|         license = License.objects.create(key=generate_id()) | ||||
|         usage = LicenseUsage.objects.create( | ||||
|             internal_user_count=100, | ||||
|             external_user_count=100, | ||||
|             status=LicenseUsageStatus.VALID, | ||||
|         ) | ||||
|         usage.record_date = now() - timedelta(weeks=THRESHOLD_READ_ONLY_WEEKS + 1) | ||||
|         usage.save(update_fields=["record_date"]) | ||||
|  | ||||
|         admin = create_test_admin_user() | ||||
|         self.client.force_login(admin) | ||||
|  | ||||
|         # Reading is always allowed | ||||
|         response = self.client.get(reverse("authentik_api:license-list")) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|  | ||||
|         # Writing should also be allowed | ||||
|         response = self.client.patch( | ||||
|             reverse("authentik_api:license-detail", kwargs={"pk": license.pk}) | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|  | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.validate", | ||||
|         MagicMock( | ||||
|             return_value=LicenseKey( | ||||
|                 aud="", | ||||
|                 exp=expiry_valid, | ||||
|                 name=generate_id(), | ||||
|                 internal_users=100, | ||||
|                 external_users=100, | ||||
|             ) | ||||
|         ), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_internal_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.get_external_user_count", | ||||
|         MagicMock(return_value=1000), | ||||
|     ) | ||||
|     @patch( | ||||
|         "authentik.enterprise.license.LicenseKey.record_usage", | ||||
|         MagicMock(), | ||||
|     ) | ||||
|     def test_manage_flows(self): | ||||
|         """Test flow""" | ||||
|         License.objects.create(key=generate_id()) | ||||
|         usage = LicenseUsage.objects.create( | ||||
|             internal_user_count=100, | ||||
|             external_user_count=100, | ||||
|             status=LicenseUsageStatus.VALID, | ||||
|         ) | ||||
|         usage.record_date = now() - timedelta(weeks=THRESHOLD_READ_ONLY_WEEKS + 1) | ||||
|         usage.save(update_fields=["record_date"]) | ||||
|  | ||||
|         admin = create_test_admin_user() | ||||
|         self.client.force_login(admin) | ||||
|  | ||||
|         # Read only is still allowed | ||||
|         response = self.client.get(reverse("authentik_api:flow-list")) | ||||
|         self.assertEqual(response.status_code, 200) | ||||
|  | ||||
|         flow = create_test_flow() | ||||
|         # Writing is not | ||||
|         response = self.client.patch( | ||||
|             reverse("authentik_api:flow-detail", kwargs={"slug": flow.slug}) | ||||
|         ) | ||||
|         self.assertJSONEqual( | ||||
|             response.content, | ||||
|             {"detail": "Request denied due to expired/invalid license.", "code": "denied_license"}, | ||||
|         ) | ||||
|         self.assertEqual(response.status_code, 400) | ||||
| @ -69,5 +69,8 @@ class NotificationViewSet( | ||||
|     @action(detail=False, methods=["post"]) | ||||
|     def mark_all_seen(self, request: Request) -> Response: | ||||
|         """Mark all the user's notifications as seen""" | ||||
|         Notification.objects.filter(user=request.user, seen=False).update(seen=True) | ||||
|         notifications = Notification.objects.filter(user=request.user) | ||||
|         for notification in notifications: | ||||
|             notification.seen = True | ||||
|         Notification.objects.bulk_update(notifications, ["seen"]) | ||||
|         return Response({}, status=204) | ||||
|  | ||||
| @ -50,7 +50,7 @@ class ASNContextProcessor(MMDBContextProcessor): | ||||
|         """Wrapper for Reader.asn""" | ||||
|         with start_span( | ||||
|             op="authentik.events.asn.asn", | ||||
|             name=ip_address, | ||||
|             description=ip_address, | ||||
|         ): | ||||
|             if not self.configured(): | ||||
|                 return None | ||||
|  | ||||
| @ -51,7 +51,7 @@ class GeoIPContextProcessor(MMDBContextProcessor): | ||||
|         """Wrapper for Reader.city""" | ||||
|         with start_span( | ||||
|             op="authentik.events.geo.city", | ||||
|             name=ip_address, | ||||
|             description=ip_address, | ||||
|         ): | ||||
|             if not self.configured(): | ||||
|                 return None | ||||
|  | ||||
| @ -49,7 +49,6 @@ from authentik.policies.models import PolicyBindingModel | ||||
| from authentik.root.middleware import ClientIPMiddleware | ||||
| from authentik.stages.email.utils import TemplateEmailMessage | ||||
| from authentik.tenants.models import Tenant | ||||
| from authentik.tenants.utils import get_current_tenant | ||||
|  | ||||
| LOGGER = get_logger() | ||||
| DISCORD_FIELD_LIMIT = 25 | ||||
| @ -59,11 +58,7 @@ NOTIFICATION_SUMMARY_LENGTH = 75 | ||||
| def default_event_duration(): | ||||
|     """Default duration an Event is saved. | ||||
|     This is used as a fallback when no brand is available""" | ||||
|     try: | ||||
|         tenant = get_current_tenant(only=["event_retention"]) | ||||
|         return now() + timedelta_from_string(tenant.event_retention) | ||||
|     except Tenant.DoesNotExist: | ||||
|         return now() + timedelta(days=365) | ||||
|     return now() + timedelta(days=365) | ||||
|  | ||||
|  | ||||
| def default_brand(): | ||||
| @ -250,6 +245,12 @@ class Event(SerializerModel, ExpiringModel): | ||||
|             if QS_QUERY in self.context["http_request"]["args"]: | ||||
|                 wrapped = self.context["http_request"]["args"][QS_QUERY] | ||||
|                 self.context["http_request"]["args"] = cleanse_dict(QueryDict(wrapped)) | ||||
|         if hasattr(request, "tenant"): | ||||
|             tenant: Tenant = request.tenant | ||||
|             # Because self.created only gets set on save, we can't use it's value here | ||||
|             # hence we set self.created to now and then use it | ||||
|             self.created = now() | ||||
|             self.expires = self.created + timedelta_from_string(tenant.event_retention) | ||||
|         if hasattr(request, "brand"): | ||||
|             brand: Brand = request.brand | ||||
|             self.brand = sanitize_dict(model_to_dict(brand)) | ||||
|  | ||||
| @ -1,22 +1,19 @@ | ||||
| """authentik events signal listener""" | ||||
|  | ||||
| from importlib import import_module | ||||
| from typing import Any | ||||
|  | ||||
| from django.conf import settings | ||||
| from django.contrib.auth.signals import user_logged_in, user_logged_out | ||||
| from django.db.models.signals import post_save, pre_delete | ||||
| from django.dispatch import receiver | ||||
| from django.http import HttpRequest | ||||
| from rest_framework.request import Request | ||||
|  | ||||
| from authentik.core.models import AuthenticatedSession, User | ||||
| from authentik.core.models import User | ||||
| from authentik.core.signals import login_failed, password_changed | ||||
| from authentik.events.apps import SYSTEM_TASK_STATUS | ||||
| from authentik.events.models import Event, EventAction, SystemTask | ||||
| from authentik.events.tasks import event_notification_handler, gdpr_cleanup | ||||
| from authentik.flows.models import Stage | ||||
| from authentik.flows.planner import PLAN_CONTEXT_OUTPOST, PLAN_CONTEXT_SOURCE, FlowPlan | ||||
| from authentik.flows.planner import PLAN_CONTEXT_SOURCE, FlowPlan | ||||
| from authentik.flows.views.executor import SESSION_KEY_PLAN | ||||
| from authentik.root.monitoring import monitoring_set | ||||
| from authentik.stages.invitation.models import Invitation | ||||
| @ -26,7 +23,6 @@ from authentik.stages.user_write.signals import user_write | ||||
| from authentik.tenants.utils import get_current_tenant | ||||
|  | ||||
| SESSION_LOGIN_EVENT = "login_event" | ||||
| _session_engine = import_module(settings.SESSION_ENGINE) | ||||
|  | ||||
|  | ||||
| @receiver(user_logged_in) | ||||
| @ -42,25 +38,13 @@ def on_user_logged_in(sender, request: HttpRequest, user: User, **_): | ||||
|             # Save the login method used | ||||
|             kwargs[PLAN_CONTEXT_METHOD] = flow_plan.context[PLAN_CONTEXT_METHOD] | ||||
|             kwargs[PLAN_CONTEXT_METHOD_ARGS] = flow_plan.context.get(PLAN_CONTEXT_METHOD_ARGS, {}) | ||||
|         if PLAN_CONTEXT_OUTPOST in flow_plan.context: | ||||
|             # Save outpost context | ||||
|             kwargs[PLAN_CONTEXT_OUTPOST] = flow_plan.context[PLAN_CONTEXT_OUTPOST] | ||||
|     event = Event.new(EventAction.LOGIN, **kwargs).from_http(request, user=user) | ||||
|     request.session[SESSION_LOGIN_EVENT] = event | ||||
|     request.session.save() | ||||
|  | ||||
|  | ||||
| def get_login_event(request_or_session: HttpRequest | AuthenticatedSession | None) -> Event | None: | ||||
| def get_login_event(request: HttpRequest) -> Event | None: | ||||
|     """Wrapper to get login event that can be mocked in tests""" | ||||
|     session = None | ||||
|     if not request_or_session: | ||||
|         return None | ||||
|     if isinstance(request_or_session, HttpRequest | Request): | ||||
|         session = request_or_session.session | ||||
|     if isinstance(request_or_session, AuthenticatedSession): | ||||
|         SessionStore = _session_engine.SessionStore | ||||
|         session = SessionStore(request_or_session.session_key) | ||||
|     return session.get(SESSION_LOGIN_EVENT, None) | ||||
|     return request.session.get(SESSION_LOGIN_EVENT, None) | ||||
|  | ||||
|  | ||||
| @receiver(user_logged_out) | ||||
|  | ||||
| @ -6,7 +6,6 @@ from django.db.models import Model | ||||
| from django.test import TestCase | ||||
|  | ||||
| from authentik.core.models import default_token_key | ||||
| from authentik.events.models import default_event_duration | ||||
| from authentik.lib.utils.reflection import get_apps | ||||
|  | ||||
|  | ||||
| @ -21,7 +20,7 @@ def model_tester_factory(test_model: type[Model]) -> Callable: | ||||
|         allowed = 0 | ||||
|         # Token-like objects need to lookup the current tenant to get the default token length | ||||
|         for field in test_model._meta.fields: | ||||
|             if field.default in [default_token_key, default_event_duration]: | ||||
|             if field.default == default_token_key: | ||||
|                 allowed += 1 | ||||
|         with self.assertNumQueries(allowed): | ||||
|             str(test_model()) | ||||
|  | ||||
| @ -2,8 +2,7 @@ | ||||
|  | ||||
| from unittest.mock import MagicMock, patch | ||||
|  | ||||
| from django.urls import reverse | ||||
| from rest_framework.test import APITestCase | ||||
| from django.test import TestCase | ||||
|  | ||||
| from authentik.core.models import Group, User | ||||
| from authentik.events.models import ( | ||||
| @ -11,7 +10,6 @@ from authentik.events.models import ( | ||||
|     EventAction, | ||||
|     Notification, | ||||
|     NotificationRule, | ||||
|     NotificationSeverity, | ||||
|     NotificationTransport, | ||||
|     NotificationWebhookMapping, | ||||
|     TransportMode, | ||||
| @ -22,7 +20,7 @@ from authentik.policies.exceptions import PolicyException | ||||
| from authentik.policies.models import PolicyBinding | ||||
|  | ||||
|  | ||||
| class TestEventsNotifications(APITestCase): | ||||
| class TestEventsNotifications(TestCase): | ||||
|     """Test Event Notifications""" | ||||
|  | ||||
|     def setUp(self) -> None: | ||||
| @ -133,15 +131,3 @@ class TestEventsNotifications(APITestCase): | ||||
|         Notification.objects.all().delete() | ||||
|         Event.new(EventAction.CUSTOM_PREFIX).save() | ||||
|         self.assertEqual(Notification.objects.first().body, "foo") | ||||
|  | ||||
|     def test_api_mark_all_seen(self): | ||||
|         """Test mark_all_seen""" | ||||
|         self.client.force_login(self.user) | ||||
|  | ||||
|         Notification.objects.create( | ||||
|             severity=NotificationSeverity.NOTICE, body="foo", user=self.user, seen=False | ||||
|         ) | ||||
|  | ||||
|         response = self.client.post(reverse("authentik_api:notification-mark-all-seen")) | ||||
|         self.assertEqual(response.status_code, 204) | ||||
|         self.assertFalse(Notification.objects.filter(body="foo", seen=False).exists()) | ||||
|  | ||||
| @ -37,7 +37,6 @@ from authentik.lib.utils.file import ( | ||||
| ) | ||||
| from authentik.lib.views import bad_request_message | ||||
| from authentik.rbac.decorators import permission_required | ||||
| from authentik.rbac.filters import ObjectFilter | ||||
|  | ||||
| LOGGER = get_logger() | ||||
|  | ||||
| @ -282,7 +281,7 @@ class FlowViewSet(UsedByMixin, ModelViewSet): | ||||
|             400: OpenApiResponse(description="Flow not applicable"), | ||||
|         }, | ||||
|     ) | ||||
|     @action(detail=True, pagination_class=None, filter_backends=[ObjectFilter]) | ||||
|     @action(detail=True, pagination_class=None, filter_backends=[]) | ||||
|     def execute(self, request: Request, slug: str): | ||||
|         """Execute flow for current user""" | ||||
|         # Because we pre-plan the flow here, and not in the planner, we need to manually clear | ||||
|  | ||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user
	