Compare commits
	
		
			297 Commits
		
	
	
		
			fix/issue_
			...
			template-f
		
	
	| Author | SHA1 | Date | |
|---|---|---|---|
| b0f8c29d69 | |||
| f20d7a200e | |||
| 3eb2c9bcc0 | |||
| 037ef3a2a0 | |||
| 61bc89959b | |||
| 2797d26b93 | |||
| 51609d696d | |||
| c0d08df161 | |||
| 643a97f0a5 | |||
| 155a31fd70 | |||
| c6f9d5df7b | |||
| ea85331a7e | |||
| 4f4c5253dd | |||
| 83b2fc36df | |||
| d99d2b8bdc | |||
| 9b96d04b3a | |||
| ca5b99eb16 | |||
| 4c1676e97c | |||
| 81855cf2fe | |||
| bd904027be | |||
| 0ffc97db15 | |||
| 2c515b1e17 | |||
| f8900fbaf3 | |||
| 0f4a98d9c6 | |||
| 8853f25b45 | |||
| 1c40f7b95a | |||
| 9b5d6ec1af | |||
| 36d29a9ae1 | |||
| 0606b1aba4 | |||
| 03d5dad867 | |||
| 38a9e46af3 | |||
| 5eb848e376 | |||
| 61a293daad | |||
| edf3300944 | |||
| 5d9c40eac8 | |||
| 6ebfbcb66e | |||
| bf0235c113 | |||
| 895cd23b57 | |||
| c908d9e95e | |||
| a07fd8d54b | |||
| 39a46a6dc4 | |||
| ad71960d77 | |||
| 2a384511f5 | |||
| 4dcc104947 | |||
| 71fe526e47 | |||
| 03e3f516ac | |||
| 3b59333246 | |||
| 4e800c14cb | |||
| 789b29a3e7 | |||
| 857b6e63a0 | |||
| edc937dd78 | |||
| d98b6f29d4 | |||
| 53ba2a0ca8 | |||
| ae364292e6 | |||
| f15bc2df97 | |||
| b27d49e55f | |||
| e0d2beb225 | |||
| 2313b4755b | |||
| 1cffadecb0 | |||
| 5e163d6da1 | |||
| 0626e18674 | |||
| e986a62a12 | |||
| e25afcb84a | |||
| bb95613104 | |||
| 89dfac2f57 | |||
| 31462b55e6 | |||
| 60337c1cf0 | |||
| 343d3bb1fb | |||
| 11fe86c4f6 | |||
| 963ce085e4 | |||
| 3642b89ab0 | |||
| 8cfb371ed3 | |||
| 6e74edb9f2 | |||
| 397905f8f0 | |||
| 7fd35b1dfc | |||
| 9ba03f5439 | |||
| 1139d6d27c | |||
| 077fd966c2 | |||
| bd41822a57 | |||
| dfd3d76434 | |||
| 397e98906d | |||
| 65d8da8c64 | |||
| 5b435297c5 | |||
| f792fd42f6 | |||
| 70c0fdd5fa | |||
| 9b636eba01 | |||
| a982224502 | |||
| 6a16cccb40 | |||
| 6dac91e2b4 | |||
| 3e2d0532d1 | |||
| 4e1300650b | |||
| 06b3ed0c9c | |||
| 395ad722b7 | |||
| 9917d81246 | |||
| 2a87687d34 | |||
| a726c2260a | |||
| 44e0bfd4ef | |||
| 8d0b362c9c | |||
| e5e53f034e | |||
| 71b87127d1 | |||
| d5d67fe22d | |||
| 5d2685341d | |||
| f1ac4ff9c9 | |||
| 79f4c66286 | |||
| 1f82094c0b | |||
| 35440acba3 | |||
| eca9901704 | |||
| 6ddd5a3d5f | |||
| 5664e62eca | |||
| 1403f17d62 | |||
| 1ac8989e81 | |||
| b0a1db77e3 | |||
| 46da4cb59e | |||
| 154df5cdf7 | |||
| 5b889456f6 | |||
| 3eaed82c48 | |||
| feaf9d8bc9 | |||
| 2899668ae2 | |||
| 4c25e1bb24 | |||
| 464ff3f5b1 | |||
| 22eb5f56f1 | |||
| 7e48e87f49 | |||
| 8ce12f7850 | |||
| 2514baabeb | |||
| 945930a507 | |||
| 537a80ad97 | |||
| 5c993e23fe | |||
| eb2db18494 | |||
| 12a46a8426 | |||
| 4a1213310a | |||
| 84c2097148 | |||
| c05dedc573 | |||
| 18c197e75b | |||
| 0c26a0bce2 | |||
| 5fd6a4cead | |||
| 51fb1bd8e7 | |||
| 4a30f87a42 | |||
| 8e6b6ede30 | |||
| af30c2a68e | |||
| 9b65627a3e | |||
| 4bad91c901 | |||
| f3c479d077 | |||
| b024df9903 | |||
| f6a6458088 | |||
| f0dc0e8900 | |||
| 79e89b0376 | |||
| 4cc7d91379 | |||
| 245909e31a | |||
| 997a1ddb3d | |||
| 42335a60bf | |||
| fc539332e1 | |||
| d9efb02078 | |||
| 6212250e19 | |||
| c18beefc8f | |||
| f23da6e402 | |||
| e934b246c8 | |||
| ead684a410 | |||
| d782aadab7 | |||
| 4ac6f83aea | |||
| 6281d36a69 | |||
| 8129ad4ec0 | |||
| 24eea415b2 | |||
| a615ce8e95 | |||
| 5b275cf7fb | |||
| d6e91c119f | |||
| 7841e47e74 | |||
| ad2a4bea3e | |||
| a554c085c1 | |||
| ff0d978754 | |||
| de48e62819 | |||
| e50e995d2f | |||
| 3bf4156cb3 | |||
| 89990facf5 | |||
| 48545950ed | |||
| 0544aa5fae | |||
| 5d69455b87 | |||
| 3d291cf4da | |||
| 44d7c42dc7 | |||
| 4ea4e925e3 | |||
| 169172c85f | |||
| adea637fa4 | |||
| 0231277d9c | |||
| 45643ed1f6 | |||
| 3823d56dbd | |||
| 43cfd59ac0 | |||
| c8555bbf59 | |||
| a4251a3410 | |||
| 50985f9b0b | |||
| 9ec24528d4 | |||
| 5eac38c0cc | |||
| 010df0c31c | |||
| 7ba858eff3 | |||
| 817d2d5ff8 | |||
| 70e34e03b4 | |||
| d61f9f6d57 | |||
| bdf81706b8 | |||
| 7b56602fc9 | |||
| 7c6e25a996 | |||
| 0eeaeaf1ff | |||
| 9ce4337b11 | |||
| c6a3c7371c | |||
| 42a7cf10f2 | |||
| bb4f7b1193 | |||
| 3eecfb835b | |||
| 92ab856bd3 | |||
| 178549a756 | |||
| 67d178aa11 | |||
| ef53abace9 | |||
| 5effb3a0f6 | |||
| 3a37916a8f | |||
| 428d5ac9cf | |||
| 7b4037fdda | |||
| 2c7bbcc27b | |||
| 19fb24de99 | |||
| 2709702896 | |||
| 7d0d5a7dc2 | |||
| 6a04a2ca69 | |||
| ea561c9da6 | |||
| 9b9c55f17c | |||
| bd5e78bd44 | |||
| ab98028022 | |||
| 813ff64ba1 | |||
| c99e742214 | |||
| dac6ad3cd6 | |||
| e4d2a53ccc | |||
| 3b6775fd9c | |||
| 5882e0b2cb | |||
| 65f0b471d8 | |||
| 7d054db1a5 | |||
| cb75ba2e5e | |||
| 36cecc1391 | |||
| 81b91d8777 | |||
| 41dc23b3c2 | |||
| 370eff1494 | |||
| 0ff8def03b | |||
| b01cafd9fe | |||
| 90aa8abb80 | |||
| fd21aae4f9 | |||
| 360223a2ff | |||
| 0e83de2697 | |||
| a23bac9d9b | |||
| 220378b3f2 | |||
| 363d655378 | |||
| e93b2a1a75 | |||
| 76665cf65e | |||
| 3ad7f4dc24 | |||
| c5045e8792 | |||
| a8c9b3a8ba | |||
| 148506639a | |||
| 53814d9919 | |||
| 08b04c32f5 | |||
| 1c1d97339d | |||
| cafa9c1737 | |||
| 5f64347ba1 | |||
| 45ef54480a | |||
| a3dc8af4c6 | |||
| 36933a0aca | |||
| 8f689890df | |||
| ec49b2e0e0 | |||
| 22ebe05706 | |||
| f0e58a6f49 | |||
| a3d642c08e | |||
| 5d42cb9185 | |||
| 1fd0cc5bb5 | |||
| deef365ff5 | |||
| d1ae6287f2 | |||
| 2e152cd264 | |||
| f5941e403b | |||
| ff3cf8c10e | |||
| bfa6328172 | |||
| 4c9691c932 | |||
| a0f1566b4c | |||
| 46261a4f42 | |||
| 8b42ff1e97 | |||
| ca4cb0d251 | |||
| a5a0fa79dd | |||
| c06a871f61 | |||
| 4a3df67134 | |||
| 422ccf61fa | |||
| d989f23907 | |||
| 059180edef | |||
| 22f30634a8 | |||
| 35ff418c42 | |||
| 7826e7a605 | |||
| 64f1b8207d | |||
| b2c13f0614 | |||
| 6965628020 | |||
| 608f63e9a2 | |||
| 22fa3a7fba | |||
| bcfd6fefa7 | |||
| eae18d0016 | |||
| 4a12a57c5f | |||
| 71294b7deb | |||
| 5af907db0c | |||
| 63a118a2ba | |||
| d9a3c34a44 | |||
| 23bdad7574 | 
@ -1,5 +1,5 @@
 | 
			
		||||
[bumpversion]
 | 
			
		||||
current_version = 2025.2.2
 | 
			
		||||
current_version = 2025.2.4
 | 
			
		||||
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*))?
 | 
			
		||||
@ -17,6 +17,8 @@ optional_value = final
 | 
			
		||||
 | 
			
		||||
[bumpversion:file:pyproject.toml]
 | 
			
		||||
 | 
			
		||||
[bumpversion:file:uv.lock]
 | 
			
		||||
 | 
			
		||||
[bumpversion:file:package.json]
 | 
			
		||||
 | 
			
		||||
[bumpversion:file:docker-compose.yml]
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										1
									
								
								.github/workflows/api-py-publish.yml
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										1
									
								
								.github/workflows/api-py-publish.yml
									
									
									
									
										vendored
									
									
								
							@ -30,7 +30,6 @@ jobs:
 | 
			
		||||
        uses: actions/setup-python@v5
 | 
			
		||||
        with:
 | 
			
		||||
          python-version-file: "pyproject.toml"
 | 
			
		||||
          cache: "poetry"
 | 
			
		||||
      - name: Generate API Client
 | 
			
		||||
        run: make gen-client-py
 | 
			
		||||
      - name: Publish package
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										45
									
								
								.github/workflows/packages-npm-publish.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							
							
						
						
									
										45
									
								
								.github/workflows/packages-npm-publish.yml
									
									
									
									
										vendored
									
									
										Normal file
									
								
							@ -0,0 +1,45 @@
 | 
			
		||||
name: authentik-packages-npm-publish
 | 
			
		||||
on:
 | 
			
		||||
  push:
 | 
			
		||||
    branches: [main]
 | 
			
		||||
    paths:
 | 
			
		||||
      - packages/docusaurus-config
 | 
			
		||||
      - packages/eslint-config
 | 
			
		||||
      - packages/prettier-config
 | 
			
		||||
      - packages/tsconfig
 | 
			
		||||
  workflow_dispatch:
 | 
			
		||||
jobs:
 | 
			
		||||
  publish:
 | 
			
		||||
    if: ${{ github.repository != 'goauthentik/authentik-internal' }}
 | 
			
		||||
    runs-on: ubuntu-latest
 | 
			
		||||
    strategy:
 | 
			
		||||
      fail-fast: false
 | 
			
		||||
      matrix:
 | 
			
		||||
        package:
 | 
			
		||||
          - docusaurus-config
 | 
			
		||||
          - eslint-config
 | 
			
		||||
          - prettier-config
 | 
			
		||||
          - tsconfig
 | 
			
		||||
    steps:
 | 
			
		||||
      - uses: actions/checkout@v4
 | 
			
		||||
        with:
 | 
			
		||||
          fetch-depth: 2
 | 
			
		||||
      - uses: actions/setup-node@v4
 | 
			
		||||
        with:
 | 
			
		||||
          node-version-file: packages/${{ matrix.package }}/package.json
 | 
			
		||||
          registry-url: "https://registry.npmjs.org"
 | 
			
		||||
      - name: Get changed files
 | 
			
		||||
        id: changed-files
 | 
			
		||||
        uses: tj-actions/changed-files@ed68ef82c095e0d48ec87eccea555d944a631a4c
 | 
			
		||||
        with:
 | 
			
		||||
          files: |
 | 
			
		||||
            packages/${{ matrix.package }}/package.json
 | 
			
		||||
      - name: Publish package
 | 
			
		||||
        if: steps.changed-files.outputs.any_changed == 'true'
 | 
			
		||||
        working-directory: packages/${{ matrix.package}}
 | 
			
		||||
        run: |
 | 
			
		||||
          npm ci
 | 
			
		||||
          npm run build
 | 
			
		||||
          npm publish
 | 
			
		||||
        env:
 | 
			
		||||
          NODE_AUTH_TOKEN: ${{ secrets.NPM_PUBLISH_TOKEN }}
 | 
			
		||||
							
								
								
									
										5
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							
							
						
						
									
										5
									
								
								.gitignore
									
									
									
									
										vendored
									
									
								
							@ -11,6 +11,10 @@ local_settings.py
 | 
			
		||||
db.sqlite3
 | 
			
		||||
media
 | 
			
		||||
 | 
			
		||||
# Node
 | 
			
		||||
 | 
			
		||||
node_modules
 | 
			
		||||
 | 
			
		||||
# If your build process includes running collectstatic, then you probably don't need or want to include staticfiles/
 | 
			
		||||
# in your Git repository. Update and uncomment the following line accordingly.
 | 
			
		||||
# <django-project-name>/staticfiles/
 | 
			
		||||
@ -33,6 +37,7 @@ eggs/
 | 
			
		||||
lib64/
 | 
			
		||||
parts/
 | 
			
		||||
dist/
 | 
			
		||||
out/
 | 
			
		||||
sdist/
 | 
			
		||||
var/
 | 
			
		||||
wheels/
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										47
									
								
								.prettierignore
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										47
									
								
								.prettierignore
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,47 @@
 | 
			
		||||
# Prettier Ignorefile
 | 
			
		||||
 | 
			
		||||
## Static Files
 | 
			
		||||
**/LICENSE
 | 
			
		||||
 | 
			
		||||
authentik/stages/**/*
 | 
			
		||||
 | 
			
		||||
## Build asset directories
 | 
			
		||||
coverage
 | 
			
		||||
dist
 | 
			
		||||
out
 | 
			
		||||
.docusaurus
 | 
			
		||||
website/docs/developer-docs/api/**/*
 | 
			
		||||
 | 
			
		||||
## Environment
 | 
			
		||||
*.env
 | 
			
		||||
 | 
			
		||||
## Secrets
 | 
			
		||||
*.secrets
 | 
			
		||||
 | 
			
		||||
## Yarn
 | 
			
		||||
.yarn/**/*
 | 
			
		||||
 | 
			
		||||
## Node
 | 
			
		||||
node_modules
 | 
			
		||||
coverage
 | 
			
		||||
 | 
			
		||||
## Configs
 | 
			
		||||
*.log
 | 
			
		||||
*.yaml
 | 
			
		||||
*.yml
 | 
			
		||||
 | 
			
		||||
# Templates
 | 
			
		||||
# TODO: Rename affected files to *.template.* or similar.
 | 
			
		||||
*.html
 | 
			
		||||
*.mdx
 | 
			
		||||
*.md
 | 
			
		||||
 | 
			
		||||
## Import order matters
 | 
			
		||||
poly.ts
 | 
			
		||||
src/locale-codes.ts
 | 
			
		||||
src/locales/
 | 
			
		||||
 | 
			
		||||
# Storybook
 | 
			
		||||
storybook-static/
 | 
			
		||||
.storybook/css-import-maps*
 | 
			
		||||
 | 
			
		||||
@ -23,6 +23,8 @@ docker-compose.yml              @goauthentik/infrastructure
 | 
			
		||||
Makefile                        @goauthentik/infrastructure
 | 
			
		||||
.editorconfig                   @goauthentik/infrastructure
 | 
			
		||||
CODEOWNERS                      @goauthentik/infrastructure
 | 
			
		||||
# Web packages
 | 
			
		||||
packages/                       @goauthentik/frontend
 | 
			
		||||
# Web
 | 
			
		||||
web/                            @goauthentik/frontend
 | 
			
		||||
tests/wdio/                     @goauthentik/frontend
 | 
			
		||||
 | 
			
		||||
@ -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} docker.io/library/golang:1.24-bookworm AS go-builder
 | 
			
		||||
 | 
			
		||||
ARG TARGETOS
 | 
			
		||||
ARG TARGETARCH
 | 
			
		||||
@ -76,7 +76,7 @@ COPY ./go.sum /go/src/goauthentik.io/go.sum
 | 
			
		||||
RUN --mount=type=cache,sharing=locked,target=/go/pkg/mod \
 | 
			
		||||
    --mount=type=cache,id=go-build-$TARGETARCH$TARGETVARIANT,sharing=locked,target=/root/.cache/go-build \
 | 
			
		||||
    if [ "$TARGETARCH" = "arm64" ]; then export CC=aarch64-linux-gnu-gcc && export CC_FOR_TARGET=gcc-aarch64-linux-gnu; fi && \
 | 
			
		||||
    CGO_ENABLED=1 GOEXPERIMENT="systemcrypto" GOFLAGS="-tags=requirefips" GOARM="${TARGETVARIANT#v}" \
 | 
			
		||||
    CGO_ENABLED=1 GOFIPS140=latest GOARM="${TARGETVARIANT#v}" \
 | 
			
		||||
    go build -o /go/authentik ./cmd/server
 | 
			
		||||
 | 
			
		||||
# Stage 4: MaxMind GeoIP
 | 
			
		||||
@ -94,9 +94,9 @@ 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: Download uv
 | 
			
		||||
FROM ghcr.io/astral-sh/uv:0.6.10 AS uv
 | 
			
		||||
FROM ghcr.io/astral-sh/uv:0.6.14 AS uv
 | 
			
		||||
# Stage 6: Base python image
 | 
			
		||||
FROM ghcr.io/goauthentik/fips-python:3.12.8-slim-bookworm-fips AS python-base
 | 
			
		||||
FROM ghcr.io/goauthentik/fips-python:3.12.10-slim-bookworm-fips AS python-base
 | 
			
		||||
 | 
			
		||||
ENV VENV_PATH="/ak-root/.venv" \
 | 
			
		||||
    PATH="/lifecycle:/ak-root/.venv/bin:$PATH" \
 | 
			
		||||
 | 
			
		||||
@ -2,7 +2,7 @@
 | 
			
		||||
 | 
			
		||||
from os import environ
 | 
			
		||||
 | 
			
		||||
__version__ = "2025.2.2"
 | 
			
		||||
__version__ = "2025.2.4"
 | 
			
		||||
ENV_GIT_HASH_KEY = "GIT_BUILD_HASH"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -3,15 +3,20 @@
 | 
			
		||||
{% load authentik_core %}
 | 
			
		||||
 | 
			
		||||
{% block title %}
 | 
			
		||||
API Browser - {{ brand.branding_title }}
 | 
			
		||||
  API Browser - {{ brand.branding_title }}
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block head %}
 | 
			
		||||
<script src="{% versioned_script 'dist/standalone/api-browser/index-%v.js' %}" type="module"></script>
 | 
			
		||||
<meta name="theme-color" content="#151515" media="(prefers-color-scheme: light)">
 | 
			
		||||
<meta name="theme-color" content="#151515" media="(prefers-color-scheme: dark)">
 | 
			
		||||
  <meta name="theme-color" content="#18191a" media="(prefers-color-scheme: dark)">
 | 
			
		||||
  <meta name="theme-color" content="#ffffff" media="(prefers-color-scheme: light)">
 | 
			
		||||
 | 
			
		||||
  <script
 | 
			
		||||
    data-test-id="entrypoint"
 | 
			
		||||
    src="{% versioned_script 'dist/standalone/api-browser/index-%v.js' %}"
 | 
			
		||||
    type="module">
 | 
			
		||||
  </script>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block body %}
 | 
			
		||||
<ak-api-browser schemaPath="{{ path }}"></ak-api-browser>
 | 
			
		||||
  <ak-api-browser schemaPath="{{ path }}"></ak-api-browser>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
@ -7,7 +7,7 @@ from rest_framework.exceptions import ValidationError
 | 
			
		||||
from rest_framework.fields import CharField, DateTimeField
 | 
			
		||||
from rest_framework.request import Request
 | 
			
		||||
from rest_framework.response import Response
 | 
			
		||||
from rest_framework.serializers import ListSerializer, ModelSerializer
 | 
			
		||||
from rest_framework.serializers import ListSerializer
 | 
			
		||||
from rest_framework.viewsets import ModelViewSet
 | 
			
		||||
 | 
			
		||||
from authentik.blueprints.models import BlueprintInstance
 | 
			
		||||
@ -15,7 +15,7 @@ from authentik.blueprints.v1.importer import Importer
 | 
			
		||||
from authentik.blueprints.v1.oci import OCI_PREFIX
 | 
			
		||||
from authentik.blueprints.v1.tasks import apply_blueprint, blueprints_find_dict
 | 
			
		||||
from authentik.core.api.used_by import UsedByMixin
 | 
			
		||||
from authentik.core.api.utils import JSONDictField, PassiveSerializer
 | 
			
		||||
from authentik.core.api.utils import JSONDictField, ModelSerializer, PassiveSerializer
 | 
			
		||||
from authentik.rbac.decorators import permission_required
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -36,6 +36,7 @@ from authentik.core.models import (
 | 
			
		||||
    GroupSourceConnection,
 | 
			
		||||
    PropertyMapping,
 | 
			
		||||
    Provider,
 | 
			
		||||
    Session,
 | 
			
		||||
    Source,
 | 
			
		||||
    User,
 | 
			
		||||
    UserSourceConnection,
 | 
			
		||||
@ -108,6 +109,7 @@ def excluded_models() -> list[type[Model]]:
 | 
			
		||||
        Policy,
 | 
			
		||||
        PolicyBindingModel,
 | 
			
		||||
        # Classes that have other dependencies
 | 
			
		||||
        Session,
 | 
			
		||||
        AuthenticatedSession,
 | 
			
		||||
        # Classes which are only internally managed
 | 
			
		||||
        # FIXME: these shouldn't need to be explicitly listed, but rather based off of a mixin
 | 
			
		||||
 | 
			
		||||
@ -46,7 +46,7 @@ LOGGER = get_logger()
 | 
			
		||||
 | 
			
		||||
def user_app_cache_key(user_pk: str, page_number: int | None = None) -> str:
 | 
			
		||||
    """Cache key where application list for user is saved"""
 | 
			
		||||
    key = f"{CACHE_PREFIX}/app_access/{user_pk}"
 | 
			
		||||
    key = f"{CACHE_PREFIX}app_access/{user_pk}"
 | 
			
		||||
    if page_number:
 | 
			
		||||
        key += f"/{page_number}"
 | 
			
		||||
    return key
 | 
			
		||||
 | 
			
		||||
@ -5,6 +5,7 @@ from typing import TypedDict
 | 
			
		||||
from rest_framework import mixins
 | 
			
		||||
from rest_framework.fields import SerializerMethodField
 | 
			
		||||
from rest_framework.request import Request
 | 
			
		||||
from rest_framework.serializers import CharField, DateTimeField, IPAddressField
 | 
			
		||||
from rest_framework.viewsets import GenericViewSet
 | 
			
		||||
from ua_parser import user_agent_parser
 | 
			
		||||
 | 
			
		||||
@ -54,6 +55,11 @@ class UserAgentDict(TypedDict):
 | 
			
		||||
class AuthenticatedSessionSerializer(ModelSerializer):
 | 
			
		||||
    """AuthenticatedSession Serializer"""
 | 
			
		||||
 | 
			
		||||
    expires = DateTimeField(source="session.expires", read_only=True)
 | 
			
		||||
    last_ip = IPAddressField(source="session.last_ip", read_only=True)
 | 
			
		||||
    last_user_agent = CharField(source="session.last_user_agent", read_only=True)
 | 
			
		||||
    last_used = DateTimeField(source="session.last_used", read_only=True)
 | 
			
		||||
 | 
			
		||||
    current = SerializerMethodField()
 | 
			
		||||
    user_agent = SerializerMethodField()
 | 
			
		||||
    geo_ip = SerializerMethodField()
 | 
			
		||||
@ -62,19 +68,19 @@ class AuthenticatedSessionSerializer(ModelSerializer):
 | 
			
		||||
    def get_current(self, instance: AuthenticatedSession) -> bool:
 | 
			
		||||
        """Check if session is currently active session"""
 | 
			
		||||
        request: Request = self.context["request"]
 | 
			
		||||
        return request._request.session.session_key == instance.session_key
 | 
			
		||||
        return request._request.session.session_key == instance.session.session_key
 | 
			
		||||
 | 
			
		||||
    def get_user_agent(self, instance: AuthenticatedSession) -> UserAgentDict:
 | 
			
		||||
        """Get parsed user agent"""
 | 
			
		||||
        return user_agent_parser.Parse(instance.last_user_agent)
 | 
			
		||||
        return user_agent_parser.Parse(instance.session.last_user_agent)
 | 
			
		||||
 | 
			
		||||
    def get_geo_ip(self, instance: AuthenticatedSession) -> GeoIPDict | None:  # pragma: no cover
 | 
			
		||||
        """Get GeoIP Data"""
 | 
			
		||||
        return GEOIP_CONTEXT_PROCESSOR.city_dict(instance.last_ip)
 | 
			
		||||
        return GEOIP_CONTEXT_PROCESSOR.city_dict(instance.session.last_ip)
 | 
			
		||||
 | 
			
		||||
    def get_asn(self, instance: AuthenticatedSession) -> ASNDict | None:  # pragma: no cover
 | 
			
		||||
        """Get ASN Data"""
 | 
			
		||||
        return ASN_CONTEXT_PROCESSOR.asn_dict(instance.last_ip)
 | 
			
		||||
        return ASN_CONTEXT_PROCESSOR.asn_dict(instance.session.last_ip)
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
        model = AuthenticatedSession
 | 
			
		||||
@ -90,6 +96,7 @@ class AuthenticatedSessionSerializer(ModelSerializer):
 | 
			
		||||
            "last_used",
 | 
			
		||||
            "expires",
 | 
			
		||||
        ]
 | 
			
		||||
        extra_args = {"uuid": {"read_only": True}}
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class AuthenticatedSessionViewSet(
 | 
			
		||||
@ -101,9 +108,10 @@ class AuthenticatedSessionViewSet(
 | 
			
		||||
):
 | 
			
		||||
    """AuthenticatedSession Viewset"""
 | 
			
		||||
 | 
			
		||||
    queryset = AuthenticatedSession.objects.all()
 | 
			
		||||
    lookup_field = "uuid"
 | 
			
		||||
    queryset = AuthenticatedSession.objects.select_related("session").all()
 | 
			
		||||
    serializer_class = AuthenticatedSessionSerializer
 | 
			
		||||
    search_fields = ["user__username", "last_ip", "last_user_agent"]
 | 
			
		||||
    filterset_fields = ["user__username", "last_ip", "last_user_agent"]
 | 
			
		||||
    search_fields = ["user__username", "session__last_ip", "session__last_user_agent"]
 | 
			
		||||
    filterset_fields = ["user__username", "session__last_ip", "session__last_user_agent"]
 | 
			
		||||
    ordering = ["user__username"]
 | 
			
		||||
    owner_field = "user"
 | 
			
		||||
 | 
			
		||||
@ -6,7 +6,7 @@ from django.utils.translation import gettext_lazy as _
 | 
			
		||||
from django_filters.filters import BooleanFilter
 | 
			
		||||
from django_filters.filterset import FilterSet
 | 
			
		||||
from rest_framework import mixins
 | 
			
		||||
from rest_framework.fields import SerializerMethodField
 | 
			
		||||
from rest_framework.fields import ReadOnlyField, SerializerMethodField
 | 
			
		||||
from rest_framework.viewsets import GenericViewSet
 | 
			
		||||
 | 
			
		||||
from authentik.core.api.object_types import TypesMixin
 | 
			
		||||
@ -18,10 +18,10 @@ from authentik.core.models import Provider
 | 
			
		||||
class ProviderSerializer(ModelSerializer, MetaNameSerializer):
 | 
			
		||||
    """Provider Serializer"""
 | 
			
		||||
 | 
			
		||||
    assigned_application_slug = SerializerMethodField()
 | 
			
		||||
    assigned_application_name = SerializerMethodField()
 | 
			
		||||
    assigned_backchannel_application_slug = SerializerMethodField()
 | 
			
		||||
    assigned_backchannel_application_name = SerializerMethodField()
 | 
			
		||||
    assigned_application_slug = ReadOnlyField(source="application.slug")
 | 
			
		||||
    assigned_application_name = ReadOnlyField(source="application.name")
 | 
			
		||||
    assigned_backchannel_application_slug = ReadOnlyField(source="backchannel_application.slug")
 | 
			
		||||
    assigned_backchannel_application_name = ReadOnlyField(source="backchannel_application.name")
 | 
			
		||||
 | 
			
		||||
    component = SerializerMethodField()
 | 
			
		||||
 | 
			
		||||
@ -31,38 +31,6 @@ class ProviderSerializer(ModelSerializer, MetaNameSerializer):
 | 
			
		||||
            return ""
 | 
			
		||||
        return obj.component
 | 
			
		||||
 | 
			
		||||
    def get_assigned_application_slug(self, obj: Provider) -> str:
 | 
			
		||||
        """Get application slug, return empty string if no application exists"""
 | 
			
		||||
        try:
 | 
			
		||||
            return obj.application.slug
 | 
			
		||||
        except Provider.application.RelatedObjectDoesNotExist:
 | 
			
		||||
            return ""
 | 
			
		||||
 | 
			
		||||
    def get_assigned_application_name(self, obj: Provider) -> str:
 | 
			
		||||
        """Get application name, return empty string if no application exists"""
 | 
			
		||||
        try:
 | 
			
		||||
            return obj.application.name
 | 
			
		||||
        except Provider.application.RelatedObjectDoesNotExist:
 | 
			
		||||
            return ""
 | 
			
		||||
 | 
			
		||||
    def get_assigned_backchannel_application_slug(self, obj: Provider) -> str:
 | 
			
		||||
        """Get backchannel application slug.
 | 
			
		||||
 | 
			
		||||
        Returns an empty string if no backchannel application exists.
 | 
			
		||||
        """
 | 
			
		||||
        if not obj.backchannel_application:
 | 
			
		||||
            return ""
 | 
			
		||||
        return obj.backchannel_application.slug or ""
 | 
			
		||||
 | 
			
		||||
    def get_assigned_backchannel_application_name(self, obj: Provider) -> str:
 | 
			
		||||
        """Get backchannel application name.
 | 
			
		||||
 | 
			
		||||
        Returns an empty string if no backchannel application exists.
 | 
			
		||||
        """
 | 
			
		||||
        if not obj.backchannel_application:
 | 
			
		||||
            return ""
 | 
			
		||||
        return obj.backchannel_application.name or ""
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
        model = Provider
 | 
			
		||||
        fields = [
 | 
			
		||||
 | 
			
		||||
@ -179,10 +179,13 @@ class UserSourceConnectionSerializer(SourceSerializer):
 | 
			
		||||
            "user",
 | 
			
		||||
            "source",
 | 
			
		||||
            "source_obj",
 | 
			
		||||
            "identifier",
 | 
			
		||||
            "created",
 | 
			
		||||
            "last_updated",
 | 
			
		||||
        ]
 | 
			
		||||
        extra_kwargs = {
 | 
			
		||||
            "created": {"read_only": True},
 | 
			
		||||
            "last_updated": {"read_only": True},
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -199,7 +202,7 @@ class UserSourceConnectionViewSet(
 | 
			
		||||
    queryset = UserSourceConnection.objects.all()
 | 
			
		||||
    serializer_class = UserSourceConnectionSerializer
 | 
			
		||||
    filterset_fields = ["user", "source__slug"]
 | 
			
		||||
    search_fields = ["source__slug"]
 | 
			
		||||
    search_fields = ["user__username", "source__slug", "identifier"]
 | 
			
		||||
    ordering = ["source__slug", "pk"]
 | 
			
		||||
    owner_field = "user"
 | 
			
		||||
 | 
			
		||||
@ -218,9 +221,11 @@ class GroupSourceConnectionSerializer(SourceSerializer):
 | 
			
		||||
            "source_obj",
 | 
			
		||||
            "identifier",
 | 
			
		||||
            "created",
 | 
			
		||||
            "last_updated",
 | 
			
		||||
        ]
 | 
			
		||||
        extra_kwargs = {
 | 
			
		||||
            "created": {"read_only": True},
 | 
			
		||||
            "last_updated": {"read_only": True},
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -237,6 +242,5 @@ class GroupSourceConnectionViewSet(
 | 
			
		||||
    queryset = GroupSourceConnection.objects.all()
 | 
			
		||||
    serializer_class = GroupSourceConnectionSerializer
 | 
			
		||||
    filterset_fields = ["group", "source__slug"]
 | 
			
		||||
    search_fields = ["source__slug"]
 | 
			
		||||
    search_fields = ["group__name", "source__slug", "identifier"]
 | 
			
		||||
    ordering = ["source__slug", "pk"]
 | 
			
		||||
    owner_field = "user"
 | 
			
		||||
 | 
			
		||||
@ -6,8 +6,6 @@ from typing import Any
 | 
			
		||||
 | 
			
		||||
from django.contrib.auth import update_session_auth_hash
 | 
			
		||||
from django.contrib.auth.models import Permission
 | 
			
		||||
from django.contrib.sessions.backends.cache import KEY_PREFIX
 | 
			
		||||
from django.core.cache import cache
 | 
			
		||||
from django.db.models.functions import ExtractHour
 | 
			
		||||
from django.db.transaction import atomic
 | 
			
		||||
from django.db.utils import IntegrityError
 | 
			
		||||
@ -71,8 +69,8 @@ from authentik.core.middleware import (
 | 
			
		||||
from authentik.core.models import (
 | 
			
		||||
    USER_ATTRIBUTE_TOKEN_EXPIRING,
 | 
			
		||||
    USER_PATH_SERVICE_ACCOUNT,
 | 
			
		||||
    AuthenticatedSession,
 | 
			
		||||
    Group,
 | 
			
		||||
    Session,
 | 
			
		||||
    Token,
 | 
			
		||||
    TokenIntents,
 | 
			
		||||
    User,
 | 
			
		||||
@ -226,6 +224,7 @@ class UserSerializer(ModelSerializer):
 | 
			
		||||
            "name",
 | 
			
		||||
            "is_active",
 | 
			
		||||
            "last_login",
 | 
			
		||||
            "date_joined",
 | 
			
		||||
            "is_superuser",
 | 
			
		||||
            "groups",
 | 
			
		||||
            "groups_obj",
 | 
			
		||||
@ -240,6 +239,7 @@ class UserSerializer(ModelSerializer):
 | 
			
		||||
        ]
 | 
			
		||||
        extra_kwargs = {
 | 
			
		||||
            "name": {"allow_blank": True},
 | 
			
		||||
            "date_joined": {"read_only": True},
 | 
			
		||||
            "password_change_date": {"read_only": True},
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
@ -373,7 +373,7 @@ class UsersFilter(FilterSet):
 | 
			
		||||
        method="filter_attributes",
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
    is_superuser = BooleanFilter(field_name="ak_groups", lookup_expr="is_superuser")
 | 
			
		||||
    is_superuser = BooleanFilter(field_name="ak_groups", method="filter_is_superuser")
 | 
			
		||||
    uuid = UUIDFilter(field_name="uuid")
 | 
			
		||||
 | 
			
		||||
    path = CharFilter(field_name="path")
 | 
			
		||||
@ -391,6 +391,11 @@ class UsersFilter(FilterSet):
 | 
			
		||||
        queryset=Group.objects.all().order_by("name"),
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
    def filter_is_superuser(self, queryset, name, value):
 | 
			
		||||
        if value:
 | 
			
		||||
            return queryset.filter(ak_groups__is_superuser=True).distinct()
 | 
			
		||||
        return queryset.exclude(ak_groups__is_superuser=True).distinct()
 | 
			
		||||
 | 
			
		||||
    def filter_attributes(self, queryset, name, value):
 | 
			
		||||
        """Filter attributes by query args"""
 | 
			
		||||
        try:
 | 
			
		||||
@ -767,9 +772,6 @@ class UserViewSet(UsedByMixin, ModelViewSet):
 | 
			
		||||
        response = super().partial_update(request, *args, **kwargs)
 | 
			
		||||
        instance: User = self.get_object()
 | 
			
		||||
        if not instance.is_active:
 | 
			
		||||
            sessions = AuthenticatedSession.objects.filter(user=instance)
 | 
			
		||||
            session_ids = sessions.values_list("session_key", flat=True)
 | 
			
		||||
            cache.delete_many(f"{KEY_PREFIX}{session}" for session in session_ids)
 | 
			
		||||
            sessions.delete()
 | 
			
		||||
            Session.objects.filter(authenticatedsession__user=instance).delete()
 | 
			
		||||
            LOGGER.debug("Deleted user's sessions", user=instance.username)
 | 
			
		||||
        return response
 | 
			
		||||
 | 
			
		||||
@ -20,6 +20,8 @@ from rest_framework.serializers import (
 | 
			
		||||
    raise_errors_on_nested_writes,
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
from authentik.rbac.permissions import assign_initial_permissions
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def is_dict(value: Any):
 | 
			
		||||
    """Ensure a value is a dictionary, useful for JSONFields"""
 | 
			
		||||
@ -29,6 +31,14 @@ def is_dict(value: Any):
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class ModelSerializer(BaseModelSerializer):
 | 
			
		||||
    def create(self, validated_data):
 | 
			
		||||
        instance = super().create(validated_data)
 | 
			
		||||
 | 
			
		||||
        request = self.context.get("request")
 | 
			
		||||
        if request and hasattr(request, "user") and not request.user.is_anonymous:
 | 
			
		||||
            assign_initial_permissions(request.user, instance)
 | 
			
		||||
 | 
			
		||||
        return instance
 | 
			
		||||
 | 
			
		||||
    def update(self, instance: Model, validated_data):
 | 
			
		||||
        raise_errors_on_nested_writes("update", self, validated_data)
 | 
			
		||||
 | 
			
		||||
@ -24,6 +24,15 @@ class InbuiltBackend(ModelBackend):
 | 
			
		||||
        self.set_method("password", request)
 | 
			
		||||
        return user
 | 
			
		||||
 | 
			
		||||
    async def aauthenticate(
 | 
			
		||||
        self, request: HttpRequest, username: str | None, password: str | None, **kwargs: Any
 | 
			
		||||
    ) -> User | None:
 | 
			
		||||
        user = await super().aauthenticate(request, username=username, password=password, **kwargs)
 | 
			
		||||
        if not user:
 | 
			
		||||
            return None
 | 
			
		||||
        self.set_method("password", request)
 | 
			
		||||
        return user
 | 
			
		||||
 | 
			
		||||
    def set_method(self, method: str, request: HttpRequest | None, **kwargs):
 | 
			
		||||
        """Set method data on current flow, if possbiel"""
 | 
			
		||||
        if not request:
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										15
									
								
								authentik/core/management/commands/clearsessions.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										15
									
								
								authentik/core/management/commands/clearsessions.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,15 @@
 | 
			
		||||
"""Change user type"""
 | 
			
		||||
 | 
			
		||||
from importlib import import_module
 | 
			
		||||
 | 
			
		||||
from django.conf import settings
 | 
			
		||||
 | 
			
		||||
from authentik.tenants.management import TenantCommand
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Command(TenantCommand):
 | 
			
		||||
    """Delete all sessions"""
 | 
			
		||||
 | 
			
		||||
    def handle_per_tenant(self, **options):
 | 
			
		||||
        engine = import_module(settings.SESSION_ENGINE)
 | 
			
		||||
        engine.SessionStore.clear_expired()
 | 
			
		||||
@ -2,9 +2,14 @@
 | 
			
		||||
 | 
			
		||||
from collections.abc import Callable
 | 
			
		||||
from contextvars import ContextVar
 | 
			
		||||
from functools import partial
 | 
			
		||||
from uuid import uuid4
 | 
			
		||||
 | 
			
		||||
from django.contrib.auth.models import AnonymousUser
 | 
			
		||||
from django.core.exceptions import ImproperlyConfigured
 | 
			
		||||
from django.http import HttpRequest, HttpResponse
 | 
			
		||||
from django.utils.deprecation import MiddlewareMixin
 | 
			
		||||
from django.utils.functional import SimpleLazyObject
 | 
			
		||||
from django.utils.translation import override
 | 
			
		||||
from sentry_sdk.api import set_tag
 | 
			
		||||
from structlog.contextvars import STRUCTLOG_KEY_PREFIX
 | 
			
		||||
@ -20,6 +25,40 @@ CTX_HOST = ContextVar[str | None](STRUCTLOG_KEY_PREFIX + "host", default=None)
 | 
			
		||||
CTX_AUTH_VIA = ContextVar[str | None](STRUCTLOG_KEY_PREFIX + KEY_AUTH_VIA, default=None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def get_user(request):
 | 
			
		||||
    if not hasattr(request, "_cached_user"):
 | 
			
		||||
        user = None
 | 
			
		||||
        if (authenticated_session := request.session.get("authenticatedsession", None)) is not None:
 | 
			
		||||
            user = authenticated_session.user
 | 
			
		||||
        request._cached_user = user or AnonymousUser()
 | 
			
		||||
    return request._cached_user
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
async def aget_user(request):
 | 
			
		||||
    if not hasattr(request, "_cached_user"):
 | 
			
		||||
        user = None
 | 
			
		||||
        if (
 | 
			
		||||
            authenticated_session := await request.session.aget("authenticatedsession", None)
 | 
			
		||||
        ) is not None:
 | 
			
		||||
            user = authenticated_session.user
 | 
			
		||||
        request._cached_user = user or AnonymousUser()
 | 
			
		||||
    return request._cached_user
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class AuthenticationMiddleware(MiddlewareMixin):
 | 
			
		||||
    def process_request(self, request):
 | 
			
		||||
        if not hasattr(request, "session"):
 | 
			
		||||
            raise ImproperlyConfigured(
 | 
			
		||||
                "The Django authentication middleware requires session "
 | 
			
		||||
                "middleware to be installed. Edit your MIDDLEWARE setting to "
 | 
			
		||||
                "insert "
 | 
			
		||||
                "'authentik.root.middleware.SessionMiddleware' before "
 | 
			
		||||
                "'authentik.core.middleware.AuthenticationMiddleware'."
 | 
			
		||||
            )
 | 
			
		||||
        request.user = SimpleLazyObject(lambda: get_user(request))
 | 
			
		||||
        request.auser = partial(aget_user, request)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class ImpersonateMiddleware:
 | 
			
		||||
    """Middleware to impersonate users"""
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -0,0 +1,19 @@
 | 
			
		||||
# Generated by Django 5.0.13 on 2025-04-07 14:04
 | 
			
		||||
 | 
			
		||||
from django.db import migrations, models
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("authentik_core", "0043_alter_group_options"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.AddField(
 | 
			
		||||
            model_name="usersourceconnection",
 | 
			
		||||
            name="new_identifier",
 | 
			
		||||
            field=models.TextField(default=""),
 | 
			
		||||
            preserve_default=False,
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -0,0 +1,30 @@
 | 
			
		||||
from django.db import migrations, models
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("authentik_core", "0044_usersourceconnection_new_identifier"),
 | 
			
		||||
        ("authentik_sources_kerberos", "0003_migrate_userkerberossourceconnection_identifier"),
 | 
			
		||||
        ("authentik_sources_oauth", "0009_migrate_useroauthsourceconnection_identifier"),
 | 
			
		||||
        ("authentik_sources_plex", "0005_migrate_userplexsourceconnection_identifier"),
 | 
			
		||||
        ("authentik_sources_saml", "0019_migrate_usersamlsourceconnection_identifier"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.RenameField(
 | 
			
		||||
            model_name="usersourceconnection",
 | 
			
		||||
            old_name="new_identifier",
 | 
			
		||||
            new_name="identifier",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddIndex(
 | 
			
		||||
            model_name="usersourceconnection",
 | 
			
		||||
            index=models.Index(fields=["identifier"], name="authentik_c_identif_59226f_idx"),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddIndex(
 | 
			
		||||
            model_name="usersourceconnection",
 | 
			
		||||
            index=models.Index(
 | 
			
		||||
                fields=["source", "identifier"], name="authentik_c_source__649e04_idx"
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
							
								
								
									
										238
									
								
								authentik/core/migrations/0046_session_and_more.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										238
									
								
								authentik/core/migrations/0046_session_and_more.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,238 @@
 | 
			
		||||
# Generated by Django 5.0.11 on 2025-01-27 12:58
 | 
			
		||||
 | 
			
		||||
import uuid
 | 
			
		||||
import pickle  # nosec
 | 
			
		||||
from django.core import signing
 | 
			
		||||
from django.contrib.auth import BACKEND_SESSION_KEY, HASH_SESSION_KEY, SESSION_KEY
 | 
			
		||||
from django.db import migrations, models
 | 
			
		||||
import django.db.models.deletion
 | 
			
		||||
from django.conf import settings
 | 
			
		||||
from django.contrib.sessions.backends.cache import KEY_PREFIX
 | 
			
		||||
from django.utils.timezone import now, timedelta
 | 
			
		||||
from authentik.lib.migrations import progress_bar
 | 
			
		||||
from authentik.root.middleware import ClientIPMiddleware
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
SESSION_CACHE_ALIAS = "default"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class PickleSerializer:
 | 
			
		||||
    """
 | 
			
		||||
    Simple wrapper around pickle to be used in signing.dumps()/loads() and
 | 
			
		||||
    cache backends.
 | 
			
		||||
    """
 | 
			
		||||
 | 
			
		||||
    def __init__(self, protocol=None):
 | 
			
		||||
        self.protocol = pickle.HIGHEST_PROTOCOL if protocol is None else protocol
 | 
			
		||||
 | 
			
		||||
    def dumps(self, obj):
 | 
			
		||||
        """Pickle data to be stored in redis"""
 | 
			
		||||
        return pickle.dumps(obj, self.protocol)
 | 
			
		||||
 | 
			
		||||
    def loads(self, data):
 | 
			
		||||
        """Unpickle data to be loaded from redis"""
 | 
			
		||||
        return pickle.loads(data)  # nosec
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def _migrate_session(
 | 
			
		||||
    apps,
 | 
			
		||||
    db_alias,
 | 
			
		||||
    session_key,
 | 
			
		||||
    session_data,
 | 
			
		||||
    expires,
 | 
			
		||||
):
 | 
			
		||||
    Session = apps.get_model("authentik_core", "Session")
 | 
			
		||||
    OldAuthenticatedSession = apps.get_model("authentik_core", "OldAuthenticatedSession")
 | 
			
		||||
    AuthenticatedSession = apps.get_model("authentik_core", "AuthenticatedSession")
 | 
			
		||||
 | 
			
		||||
    old_auth_session = (
 | 
			
		||||
        OldAuthenticatedSession.objects.using(db_alias).filter(session_key=session_key).first()
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
    args = {
 | 
			
		||||
        "session_key": session_key,
 | 
			
		||||
        "expires": expires,
 | 
			
		||||
        "last_ip": ClientIPMiddleware.default_ip,
 | 
			
		||||
        "last_user_agent": "",
 | 
			
		||||
        "session_data": {},
 | 
			
		||||
    }
 | 
			
		||||
    for k, v in session_data.items():
 | 
			
		||||
        if k == "authentik/stages/user_login/last_ip":
 | 
			
		||||
            args["last_ip"] = v
 | 
			
		||||
        elif k in ["last_user_agent", "last_used"]:
 | 
			
		||||
            args[k] = v
 | 
			
		||||
        elif args in [SESSION_KEY, BACKEND_SESSION_KEY, HASH_SESSION_KEY]:
 | 
			
		||||
            pass
 | 
			
		||||
        else:
 | 
			
		||||
            args["session_data"][k] = v
 | 
			
		||||
    if old_auth_session:
 | 
			
		||||
        args["last_user_agent"] = old_auth_session.last_user_agent
 | 
			
		||||
        args["last_used"] = old_auth_session.last_used
 | 
			
		||||
 | 
			
		||||
    args["session_data"] = pickle.dumps(args["session_data"])
 | 
			
		||||
    session = Session.objects.using(db_alias).create(**args)
 | 
			
		||||
 | 
			
		||||
    if old_auth_session:
 | 
			
		||||
        AuthenticatedSession.objects.using(db_alias).create(
 | 
			
		||||
            session=session,
 | 
			
		||||
            user=old_auth_session.user,
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def migrate_redis_sessions(apps, schema_editor):
 | 
			
		||||
    from django.core.cache import caches
 | 
			
		||||
 | 
			
		||||
    db_alias = schema_editor.connection.alias
 | 
			
		||||
    cache = caches[SESSION_CACHE_ALIAS]
 | 
			
		||||
 | 
			
		||||
    # Not a redis cache, skipping
 | 
			
		||||
    if not hasattr(cache, "keys"):
 | 
			
		||||
        return
 | 
			
		||||
 | 
			
		||||
    print("\nMigrating Redis sessions to database, this might take a couple of minutes...")
 | 
			
		||||
    for key, session_data in progress_bar(cache.get_many(cache.keys(f"{KEY_PREFIX}*")).items()):
 | 
			
		||||
        _migrate_session(
 | 
			
		||||
            apps=apps,
 | 
			
		||||
            db_alias=db_alias,
 | 
			
		||||
            session_key=key.removeprefix(KEY_PREFIX),
 | 
			
		||||
            session_data=session_data,
 | 
			
		||||
            expires=now() + timedelta(seconds=cache.ttl(key)),
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def migrate_database_sessions(apps, schema_editor):
 | 
			
		||||
    DjangoSession = apps.get_model("sessions", "Session")
 | 
			
		||||
    db_alias = schema_editor.connection.alias
 | 
			
		||||
 | 
			
		||||
    print("\nMigration database sessions, this might take a couple of minutes...")
 | 
			
		||||
    for django_session in progress_bar(DjangoSession.objects.using(db_alias).all()):
 | 
			
		||||
        session_data = signing.loads(
 | 
			
		||||
            django_session.session_data,
 | 
			
		||||
            salt="django.contrib.sessions.SessionStore",
 | 
			
		||||
            serializer=PickleSerializer,
 | 
			
		||||
        )
 | 
			
		||||
        _migrate_session(
 | 
			
		||||
            apps=apps,
 | 
			
		||||
            db_alias=db_alias,
 | 
			
		||||
            session_key=django_session.session_key,
 | 
			
		||||
            session_data=session_data,
 | 
			
		||||
            expires=django_session.expire_date,
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("sessions", "0001_initial"),
 | 
			
		||||
        ("authentik_core", "0045_rename_new_identifier_usersourceconnection_identifier_and_more"),
 | 
			
		||||
        ("authentik_providers_oauth2", "0027_accesstoken_authentik_p_expires_9f24a5_idx_and_more"),
 | 
			
		||||
        ("authentik_providers_rac", "0006_connectiontoken_authentik_p_expires_91f148_idx_and_more"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        # Rename AuthenticatedSession to OldAuthenticatedSession
 | 
			
		||||
        migrations.RenameModel(
 | 
			
		||||
            old_name="AuthenticatedSession",
 | 
			
		||||
            new_name="OldAuthenticatedSession",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RenameIndex(
 | 
			
		||||
            model_name="oldauthenticatedsession",
 | 
			
		||||
            new_name="authentik_c_expires_cf4f72_idx",
 | 
			
		||||
            old_name="authentik_c_expires_08251d_idx",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RenameIndex(
 | 
			
		||||
            model_name="oldauthenticatedsession",
 | 
			
		||||
            new_name="authentik_c_expirin_c1f17f_idx",
 | 
			
		||||
            old_name="authentik_c_expirin_9cd839_idx",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RenameIndex(
 | 
			
		||||
            model_name="oldauthenticatedsession",
 | 
			
		||||
            new_name="authentik_c_expirin_e04f5d_idx",
 | 
			
		||||
            old_name="authentik_c_expirin_195a84_idx",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RenameIndex(
 | 
			
		||||
            model_name="oldauthenticatedsession",
 | 
			
		||||
            new_name="authentik_c_session_a44819_idx",
 | 
			
		||||
            old_name="authentik_c_session_d0f005_idx",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RunSQL(
 | 
			
		||||
            sql="ALTER INDEX authentik_core_authenticatedsession_user_id_5055b6cf RENAME TO authentik_core_oldauthenticatedsession_user_id_5055b6cf",
 | 
			
		||||
            reverse_sql="ALTER INDEX authentik_core_oldauthenticatedsession_user_id_5055b6cf RENAME TO authentik_core_authenticatedsession_user_id_5055b6cf",
 | 
			
		||||
        ),
 | 
			
		||||
        # Create new Session and AuthenticatedSession models
 | 
			
		||||
        migrations.CreateModel(
 | 
			
		||||
            name="Session",
 | 
			
		||||
            fields=[
 | 
			
		||||
                (
 | 
			
		||||
                    "session_key",
 | 
			
		||||
                    models.CharField(
 | 
			
		||||
                        max_length=40, primary_key=True, serialize=False, verbose_name="session key"
 | 
			
		||||
                    ),
 | 
			
		||||
                ),
 | 
			
		||||
                ("expires", models.DateTimeField(default=None, null=True)),
 | 
			
		||||
                ("expiring", models.BooleanField(default=True)),
 | 
			
		||||
                ("session_data", models.BinaryField(verbose_name="session data")),
 | 
			
		||||
                ("last_ip", models.GenericIPAddressField()),
 | 
			
		||||
                ("last_user_agent", models.TextField(blank=True)),
 | 
			
		||||
                ("last_used", models.DateTimeField(auto_now=True)),
 | 
			
		||||
            ],
 | 
			
		||||
            options={
 | 
			
		||||
                "default_permissions": [],
 | 
			
		||||
                "verbose_name": "Session",
 | 
			
		||||
                "verbose_name_plural": "Sessions",
 | 
			
		||||
            },
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddIndex(
 | 
			
		||||
            model_name="session",
 | 
			
		||||
            index=models.Index(fields=["expires"], name="authentik_c_expires_d2f607_idx"),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddIndex(
 | 
			
		||||
            model_name="session",
 | 
			
		||||
            index=models.Index(fields=["expiring"], name="authentik_c_expirin_7c2cfb_idx"),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddIndex(
 | 
			
		||||
            model_name="session",
 | 
			
		||||
            index=models.Index(
 | 
			
		||||
                fields=["expiring", "expires"], name="authentik_c_expirin_1ab2e4_idx"
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddIndex(
 | 
			
		||||
            model_name="session",
 | 
			
		||||
            index=models.Index(
 | 
			
		||||
                fields=["expires", "session_key"], name="authentik_c_expires_c49143_idx"
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.CreateModel(
 | 
			
		||||
            name="AuthenticatedSession",
 | 
			
		||||
            fields=[
 | 
			
		||||
                (
 | 
			
		||||
                    "session",
 | 
			
		||||
                    models.OneToOneField(
 | 
			
		||||
                        on_delete=django.db.models.deletion.CASCADE,
 | 
			
		||||
                        primary_key=True,
 | 
			
		||||
                        serialize=False,
 | 
			
		||||
                        to="authentik_core.session",
 | 
			
		||||
                    ),
 | 
			
		||||
                ),
 | 
			
		||||
                ("uuid", models.UUIDField(default=uuid.uuid4, unique=True)),
 | 
			
		||||
                (
 | 
			
		||||
                    "user",
 | 
			
		||||
                    models.ForeignKey(
 | 
			
		||||
                        on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL
 | 
			
		||||
                    ),
 | 
			
		||||
                ),
 | 
			
		||||
            ],
 | 
			
		||||
            options={
 | 
			
		||||
                "verbose_name": "Authenticated Session",
 | 
			
		||||
                "verbose_name_plural": "Authenticated Sessions",
 | 
			
		||||
            },
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RunPython(
 | 
			
		||||
            code=migrate_redis_sessions,
 | 
			
		||||
            reverse_code=migrations.RunPython.noop,
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RunPython(
 | 
			
		||||
            code=migrate_database_sessions,
 | 
			
		||||
            reverse_code=migrations.RunPython.noop,
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -0,0 +1,18 @@
 | 
			
		||||
# Generated by Django 5.0.11 on 2025-01-27 13:02
 | 
			
		||||
 | 
			
		||||
from django.db import migrations
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("authentik_core", "0046_session_and_more"),
 | 
			
		||||
        ("authentik_providers_rac", "0007_migrate_session"),
 | 
			
		||||
        ("authentik_providers_oauth2", "0028_migrate_session"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.DeleteModel(
 | 
			
		||||
            name="OldAuthenticatedSession",
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -1,6 +1,7 @@
 | 
			
		||||
"""authentik core models"""
 | 
			
		||||
 | 
			
		||||
from datetime import datetime
 | 
			
		||||
from enum import StrEnum
 | 
			
		||||
from hashlib import sha256
 | 
			
		||||
from typing import Any, Optional, Self
 | 
			
		||||
from uuid import uuid4
 | 
			
		||||
@ -9,6 +10,7 @@ from deepmerge import always_merger
 | 
			
		||||
from django.contrib.auth.hashers import check_password
 | 
			
		||||
from django.contrib.auth.models import AbstractUser
 | 
			
		||||
from django.contrib.auth.models import UserManager as DjangoUserManager
 | 
			
		||||
from django.contrib.sessions.base_session import AbstractBaseSession
 | 
			
		||||
from django.db import models
 | 
			
		||||
from django.db.models import Q, QuerySet, options
 | 
			
		||||
from django.db.models.constants import LOOKUP_SEP
 | 
			
		||||
@ -646,19 +648,30 @@ class SourceUserMatchingModes(models.TextChoices):
 | 
			
		||||
    """Different modes a source can handle new/returning users"""
 | 
			
		||||
 | 
			
		||||
    IDENTIFIER = "identifier", _("Use the source-specific identifier")
 | 
			
		||||
    EMAIL_LINK = "email_link", _(
 | 
			
		||||
        "Link to a user with identical email address. Can have security implications "
 | 
			
		||||
        "when a source doesn't validate email addresses."
 | 
			
		||||
    EMAIL_LINK = (
 | 
			
		||||
        "email_link",
 | 
			
		||||
        _(
 | 
			
		||||
            "Link to a user with identical email address. Can have security implications "
 | 
			
		||||
            "when a source doesn't validate email addresses."
 | 
			
		||||
        ),
 | 
			
		||||
    )
 | 
			
		||||
    EMAIL_DENY = "email_deny", _(
 | 
			
		||||
        "Use the user's email address, but deny enrollment when the email address already exists."
 | 
			
		||||
    EMAIL_DENY = (
 | 
			
		||||
        "email_deny",
 | 
			
		||||
        _(
 | 
			
		||||
            "Use the user's email address, but deny enrollment when the email address already "
 | 
			
		||||
            "exists."
 | 
			
		||||
        ),
 | 
			
		||||
    )
 | 
			
		||||
    USERNAME_LINK = "username_link", _(
 | 
			
		||||
        "Link to a user with identical username. Can have security implications "
 | 
			
		||||
        "when a username is used with another source."
 | 
			
		||||
    USERNAME_LINK = (
 | 
			
		||||
        "username_link",
 | 
			
		||||
        _(
 | 
			
		||||
            "Link to a user with identical username. Can have security implications "
 | 
			
		||||
            "when a username is used with another source."
 | 
			
		||||
        ),
 | 
			
		||||
    )
 | 
			
		||||
    USERNAME_DENY = "username_deny", _(
 | 
			
		||||
        "Use the user's username, but deny enrollment when the username already exists."
 | 
			
		||||
    USERNAME_DENY = (
 | 
			
		||||
        "username_deny",
 | 
			
		||||
        _("Use the user's username, but deny enrollment when the username already exists."),
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -666,12 +679,16 @@ class SourceGroupMatchingModes(models.TextChoices):
 | 
			
		||||
    """Different modes a source can handle new/returning groups"""
 | 
			
		||||
 | 
			
		||||
    IDENTIFIER = "identifier", _("Use the source-specific identifier")
 | 
			
		||||
    NAME_LINK = "name_link", _(
 | 
			
		||||
        "Link to a group with identical name. Can have security implications "
 | 
			
		||||
        "when a group name is used with another source."
 | 
			
		||||
    NAME_LINK = (
 | 
			
		||||
        "name_link",
 | 
			
		||||
        _(
 | 
			
		||||
            "Link to a group with identical name. Can have security implications "
 | 
			
		||||
            "when a group name is used with another source."
 | 
			
		||||
        ),
 | 
			
		||||
    )
 | 
			
		||||
    NAME_DENY = "name_deny", _(
 | 
			
		||||
        "Use the group name, but deny enrollment when the name already exists."
 | 
			
		||||
    NAME_DENY = (
 | 
			
		||||
        "name_deny",
 | 
			
		||||
        _("Use the group name, but deny enrollment when the name already exists."),
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -730,8 +747,7 @@ class Source(ManagedModel, SerializerModel, PolicyBindingModel):
 | 
			
		||||
        choices=SourceGroupMatchingModes.choices,
 | 
			
		||||
        default=SourceGroupMatchingModes.IDENTIFIER,
 | 
			
		||||
        help_text=_(
 | 
			
		||||
            "How the source determines if an existing group should be used or "
 | 
			
		||||
            "a new group created."
 | 
			
		||||
            "How the source determines if an existing group should be used or a new group created."
 | 
			
		||||
        ),
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
@ -761,11 +777,17 @@ class Source(ManagedModel, SerializerModel, PolicyBindingModel):
 | 
			
		||||
    @property
 | 
			
		||||
    def component(self) -> str:
 | 
			
		||||
        """Return component used to edit this object"""
 | 
			
		||||
        if self.managed == self.MANAGED_INBUILT:
 | 
			
		||||
            return ""
 | 
			
		||||
        raise NotImplementedError
 | 
			
		||||
 | 
			
		||||
    @property
 | 
			
		||||
    def property_mapping_type(self) -> "type[PropertyMapping]":
 | 
			
		||||
        """Return property mapping type used by this object"""
 | 
			
		||||
        if self.managed == self.MANAGED_INBUILT:
 | 
			
		||||
            from authentik.core.models import PropertyMapping
 | 
			
		||||
 | 
			
		||||
            return PropertyMapping
 | 
			
		||||
        raise NotImplementedError
 | 
			
		||||
 | 
			
		||||
    def ui_login_button(self, request: HttpRequest) -> UILoginButton | None:
 | 
			
		||||
@ -780,10 +802,14 @@ class Source(ManagedModel, SerializerModel, PolicyBindingModel):
 | 
			
		||||
 | 
			
		||||
    def get_base_user_properties(self, **kwargs) -> dict[str, Any | dict[str, Any]]:
 | 
			
		||||
        """Get base properties for a user to build final properties upon."""
 | 
			
		||||
        if self.managed == self.MANAGED_INBUILT:
 | 
			
		||||
            return {}
 | 
			
		||||
        raise NotImplementedError
 | 
			
		||||
 | 
			
		||||
    def get_base_group_properties(self, **kwargs) -> dict[str, Any | dict[str, Any]]:
 | 
			
		||||
        """Get base properties for a group to build final properties upon."""
 | 
			
		||||
        if self.managed == self.MANAGED_INBUILT:
 | 
			
		||||
            return {}
 | 
			
		||||
        raise NotImplementedError
 | 
			
		||||
 | 
			
		||||
    def __str__(self):
 | 
			
		||||
@ -814,6 +840,7 @@ class UserSourceConnection(SerializerModel, CreatedUpdatedModel):
 | 
			
		||||
 | 
			
		||||
    user = models.ForeignKey(User, on_delete=models.CASCADE)
 | 
			
		||||
    source = models.ForeignKey(Source, on_delete=models.CASCADE)
 | 
			
		||||
    identifier = models.TextField()
 | 
			
		||||
 | 
			
		||||
    objects = InheritanceManager()
 | 
			
		||||
 | 
			
		||||
@ -827,6 +854,10 @@ class UserSourceConnection(SerializerModel, CreatedUpdatedModel):
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
        unique_together = (("user", "source"),)
 | 
			
		||||
        indexes = (
 | 
			
		||||
            models.Index(fields=("identifier",)),
 | 
			
		||||
            models.Index(fields=("source", "identifier")),
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class GroupSourceConnection(SerializerModel, CreatedUpdatedModel):
 | 
			
		||||
@ -997,45 +1028,75 @@ class PropertyMapping(SerializerModel, ManagedModel):
 | 
			
		||||
        verbose_name_plural = _("Property Mappings")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class AuthenticatedSession(ExpiringModel):
 | 
			
		||||
    """Additional session class for authenticated users. Augments the standard django session
 | 
			
		||||
    to achieve the following:
 | 
			
		||||
        - Make it queryable by user
 | 
			
		||||
        - Have a direct connection to user objects
 | 
			
		||||
        - Allow users to view their own sessions and terminate them
 | 
			
		||||
        - Save structured and well-defined information.
 | 
			
		||||
    """
 | 
			
		||||
class Session(ExpiringModel, AbstractBaseSession):
 | 
			
		||||
    """User session with extra fields for fast access"""
 | 
			
		||||
 | 
			
		||||
    uuid = models.UUIDField(default=uuid4, primary_key=True)
 | 
			
		||||
    # Remove upstream field because we're using our own ExpiringModel
 | 
			
		||||
    expire_date = None
 | 
			
		||||
    session_data = models.BinaryField(_("session data"))
 | 
			
		||||
 | 
			
		||||
    session_key = models.CharField(max_length=40)
 | 
			
		||||
    user = models.ForeignKey(User, on_delete=models.CASCADE)
 | 
			
		||||
 | 
			
		||||
    last_ip = models.TextField()
 | 
			
		||||
    # Keep in sync with Session.Keys
 | 
			
		||||
    last_ip = models.GenericIPAddressField()
 | 
			
		||||
    last_user_agent = models.TextField(blank=True)
 | 
			
		||||
    last_used = models.DateTimeField(auto_now=True)
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
        verbose_name = _("Session")
 | 
			
		||||
        verbose_name_plural = _("Sessions")
 | 
			
		||||
        indexes = ExpiringModel.Meta.indexes + [
 | 
			
		||||
            models.Index(fields=["expires", "session_key"]),
 | 
			
		||||
        ]
 | 
			
		||||
        default_permissions = []
 | 
			
		||||
 | 
			
		||||
    def __str__(self):
 | 
			
		||||
        return self.session_key
 | 
			
		||||
 | 
			
		||||
    class Keys(StrEnum):
 | 
			
		||||
        """
 | 
			
		||||
        Keys to be set with the session interface for the fields above to be updated.
 | 
			
		||||
 | 
			
		||||
        If a field is added here that needs to be initialized when the session is initialized,
 | 
			
		||||
        it must also be reflected in authentik.root.middleware.SessionMiddleware.process_request
 | 
			
		||||
        and in authentik.core.sessions.SessionStore.__init__
 | 
			
		||||
        """
 | 
			
		||||
 | 
			
		||||
        LAST_IP = "last_ip"
 | 
			
		||||
        LAST_USER_AGENT = "last_user_agent"
 | 
			
		||||
        LAST_USED = "last_used"
 | 
			
		||||
 | 
			
		||||
    @classmethod
 | 
			
		||||
    def get_session_store_class(cls):
 | 
			
		||||
        from authentik.core.sessions import SessionStore
 | 
			
		||||
 | 
			
		||||
        return SessionStore
 | 
			
		||||
 | 
			
		||||
    def get_decoded(self):
 | 
			
		||||
        raise NotImplementedError
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class AuthenticatedSession(SerializerModel):
 | 
			
		||||
    session = models.OneToOneField(Session, on_delete=models.CASCADE, primary_key=True)
 | 
			
		||||
    # We use the session as primary key, but we need the API to be able to reference
 | 
			
		||||
    # this object uniquely without exposing the session key
 | 
			
		||||
    uuid = models.UUIDField(default=uuid4, unique=True)
 | 
			
		||||
 | 
			
		||||
    user = models.ForeignKey(User, on_delete=models.CASCADE)
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
        verbose_name = _("Authenticated Session")
 | 
			
		||||
        verbose_name_plural = _("Authenticated Sessions")
 | 
			
		||||
        indexes = ExpiringModel.Meta.indexes + [
 | 
			
		||||
            models.Index(fields=["session_key"]),
 | 
			
		||||
        ]
 | 
			
		||||
 | 
			
		||||
    def __str__(self) -> str:
 | 
			
		||||
        return f"Authenticated Session {self.session_key[:10]}"
 | 
			
		||||
        return f"Authenticated Session {str(self.pk)[:10]}"
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def from_request(request: HttpRequest, user: User) -> Optional["AuthenticatedSession"]:
 | 
			
		||||
        """Create a new session from a http request"""
 | 
			
		||||
        from authentik.root.middleware import ClientIPMiddleware
 | 
			
		||||
 | 
			
		||||
        if not hasattr(request, "session") or not request.session.session_key:
 | 
			
		||||
        if not hasattr(request, "session") or not request.session.exists(
 | 
			
		||||
            request.session.session_key
 | 
			
		||||
        ):
 | 
			
		||||
            return None
 | 
			
		||||
        return AuthenticatedSession(
 | 
			
		||||
            session_key=request.session.session_key,
 | 
			
		||||
            session=Session.objects.filter(session_key=request.session.session_key).first(),
 | 
			
		||||
            user=user,
 | 
			
		||||
            last_ip=ClientIPMiddleware.get_client_ip(request),
 | 
			
		||||
            last_user_agent=request.META.get("HTTP_USER_AGENT", ""),
 | 
			
		||||
            expires=request.session.get_expiry_date(),
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										168
									
								
								authentik/core/sessions.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										168
									
								
								authentik/core/sessions.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,168 @@
 | 
			
		||||
"""authentik sessions engine"""
 | 
			
		||||
 | 
			
		||||
import pickle  # nosec
 | 
			
		||||
 | 
			
		||||
from django.contrib.auth import BACKEND_SESSION_KEY, HASH_SESSION_KEY, SESSION_KEY
 | 
			
		||||
from django.contrib.sessions.backends.db import SessionStore as SessionBase
 | 
			
		||||
from django.core.exceptions import SuspiciousOperation
 | 
			
		||||
from django.utils import timezone
 | 
			
		||||
from django.utils.functional import cached_property
 | 
			
		||||
from structlog.stdlib import get_logger
 | 
			
		||||
 | 
			
		||||
from authentik.root.middleware import ClientIPMiddleware
 | 
			
		||||
 | 
			
		||||
LOGGER = get_logger()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class SessionStore(SessionBase):
 | 
			
		||||
    def __init__(self, session_key=None, last_ip=None, last_user_agent=""):
 | 
			
		||||
        super().__init__(session_key)
 | 
			
		||||
        self._create_kwargs = {
 | 
			
		||||
            "last_ip": last_ip or ClientIPMiddleware.default_ip,
 | 
			
		||||
            "last_user_agent": last_user_agent,
 | 
			
		||||
        }
 | 
			
		||||
 | 
			
		||||
    @classmethod
 | 
			
		||||
    def get_model_class(cls):
 | 
			
		||||
        from authentik.core.models import Session
 | 
			
		||||
 | 
			
		||||
        return Session
 | 
			
		||||
 | 
			
		||||
    @cached_property
 | 
			
		||||
    def model_fields(self):
 | 
			
		||||
        return [k.value for k in self.model.Keys]
 | 
			
		||||
 | 
			
		||||
    def _get_session_from_db(self):
 | 
			
		||||
        try:
 | 
			
		||||
            return (
 | 
			
		||||
                self.model.objects.select_related(
 | 
			
		||||
                    "authenticatedsession",
 | 
			
		||||
                    "authenticatedsession__user",
 | 
			
		||||
                )
 | 
			
		||||
                .prefetch_related(
 | 
			
		||||
                    "authenticatedsession__user__groups",
 | 
			
		||||
                    "authenticatedsession__user__user_permissions",
 | 
			
		||||
                )
 | 
			
		||||
                .get(
 | 
			
		||||
                    session_key=self.session_key,
 | 
			
		||||
                    expires__gt=timezone.now(),
 | 
			
		||||
                )
 | 
			
		||||
            )
 | 
			
		||||
        except (self.model.DoesNotExist, SuspiciousOperation) as exc:
 | 
			
		||||
            if isinstance(exc, SuspiciousOperation):
 | 
			
		||||
                LOGGER.warning(str(exc))
 | 
			
		||||
            self._session_key = None
 | 
			
		||||
 | 
			
		||||
    async def _aget_session_from_db(self):
 | 
			
		||||
        try:
 | 
			
		||||
            return (
 | 
			
		||||
                await self.model.objects.select_related(
 | 
			
		||||
                    "authenticatedsession",
 | 
			
		||||
                    "authenticatedsession__user",
 | 
			
		||||
                )
 | 
			
		||||
                .prefetch_related(
 | 
			
		||||
                    "authenticatedsession__user__groups",
 | 
			
		||||
                    "authenticatedsession__user__user_permissions",
 | 
			
		||||
                )
 | 
			
		||||
                .aget(
 | 
			
		||||
                    session_key=self.session_key,
 | 
			
		||||
                    expires__gt=timezone.now(),
 | 
			
		||||
                )
 | 
			
		||||
            )
 | 
			
		||||
        except (self.model.DoesNotExist, SuspiciousOperation) as exc:
 | 
			
		||||
            if isinstance(exc, SuspiciousOperation):
 | 
			
		||||
                LOGGER.warning(str(exc))
 | 
			
		||||
            self._session_key = None
 | 
			
		||||
 | 
			
		||||
    def encode(self, session_dict):
 | 
			
		||||
        return pickle.dumps(session_dict, protocol=pickle.HIGHEST_PROTOCOL)
 | 
			
		||||
 | 
			
		||||
    def decode(self, session_data):
 | 
			
		||||
        try:
 | 
			
		||||
            return pickle.loads(session_data)  # nosec
 | 
			
		||||
        except pickle.PickleError:
 | 
			
		||||
            # ValueError, unpickling exceptions. If any of these happen, just return an empty
 | 
			
		||||
            # dictionary (an empty session)
 | 
			
		||||
            pass
 | 
			
		||||
        return {}
 | 
			
		||||
 | 
			
		||||
    def load(self):
 | 
			
		||||
        s = self._get_session_from_db()
 | 
			
		||||
        if s:
 | 
			
		||||
            return {
 | 
			
		||||
                "authenticatedsession": getattr(s, "authenticatedsession", None),
 | 
			
		||||
                **{k: getattr(s, k) for k in self.model_fields},
 | 
			
		||||
                **self.decode(s.session_data),
 | 
			
		||||
            }
 | 
			
		||||
        else:
 | 
			
		||||
            return {}
 | 
			
		||||
 | 
			
		||||
    async def aload(self):
 | 
			
		||||
        s = await self._aget_session_from_db()
 | 
			
		||||
        if s:
 | 
			
		||||
            return {
 | 
			
		||||
                "authenticatedsession": getattr(s, "authenticatedsession", None),
 | 
			
		||||
                **{k: getattr(s, k) for k in self.model_fields},
 | 
			
		||||
                **self.decode(s.session_data),
 | 
			
		||||
            }
 | 
			
		||||
        else:
 | 
			
		||||
            return {}
 | 
			
		||||
 | 
			
		||||
    def create_model_instance(self, data):
 | 
			
		||||
        args = {
 | 
			
		||||
            "session_key": self._get_or_create_session_key(),
 | 
			
		||||
            "expires": self.get_expiry_date(),
 | 
			
		||||
            "session_data": {},
 | 
			
		||||
            **self._create_kwargs,
 | 
			
		||||
        }
 | 
			
		||||
        for k, v in data.items():
 | 
			
		||||
            # Don't save:
 | 
			
		||||
            # - unused auth data
 | 
			
		||||
            # - related models
 | 
			
		||||
            if k in [SESSION_KEY, BACKEND_SESSION_KEY, HASH_SESSION_KEY, "authenticatedsession"]:
 | 
			
		||||
                pass
 | 
			
		||||
            elif k in self.model_fields:
 | 
			
		||||
                args[k] = v
 | 
			
		||||
            else:
 | 
			
		||||
                args["session_data"][k] = v
 | 
			
		||||
        args["session_data"] = self.encode(args["session_data"])
 | 
			
		||||
        return self.model(**args)
 | 
			
		||||
 | 
			
		||||
    async def acreate_model_instance(self, data):
 | 
			
		||||
        args = {
 | 
			
		||||
            "session_key": await self._aget_or_create_session_key(),
 | 
			
		||||
            "expires": await self.aget_expiry_date(),
 | 
			
		||||
            "session_data": {},
 | 
			
		||||
            **self._create_kwargs,
 | 
			
		||||
        }
 | 
			
		||||
        for k, v in data.items():
 | 
			
		||||
            # Don't save:
 | 
			
		||||
            # - unused auth data
 | 
			
		||||
            # - related models
 | 
			
		||||
            if k in [SESSION_KEY, BACKEND_SESSION_KEY, HASH_SESSION_KEY, "authenticatedsession"]:
 | 
			
		||||
                pass
 | 
			
		||||
            elif k in self.model_fields:
 | 
			
		||||
                args[k] = v
 | 
			
		||||
            else:
 | 
			
		||||
                args["session_data"][k] = v
 | 
			
		||||
        args["session_data"] = self.encode(args["session_data"])
 | 
			
		||||
        return self.model(**args)
 | 
			
		||||
 | 
			
		||||
    @classmethod
 | 
			
		||||
    def clear_expired(cls):
 | 
			
		||||
        cls.get_model_class().objects.filter(expires__lt=timezone.now()).delete()
 | 
			
		||||
 | 
			
		||||
    @classmethod
 | 
			
		||||
    async def aclear_expired(cls):
 | 
			
		||||
        await cls.get_model_class().objects.filter(expires__lt=timezone.now()).adelete()
 | 
			
		||||
 | 
			
		||||
    def cycle_key(self):
 | 
			
		||||
        data = self._session
 | 
			
		||||
        key = self.session_key
 | 
			
		||||
        self.create()
 | 
			
		||||
        self._session_cache = data
 | 
			
		||||
        if key:
 | 
			
		||||
            self.delete(key)
 | 
			
		||||
        if (authenticated_session := data.get("authenticatedsession")) is not None:
 | 
			
		||||
            authenticated_session.session_id = self.session_key
 | 
			
		||||
            authenticated_session.save(force_insert=True)
 | 
			
		||||
@ -1,11 +1,10 @@
 | 
			
		||||
"""authentik core signals"""
 | 
			
		||||
 | 
			
		||||
from django.contrib.auth.signals import user_logged_in, user_logged_out
 | 
			
		||||
from django.contrib.sessions.backends.cache import KEY_PREFIX
 | 
			
		||||
from django.contrib.auth.signals import user_logged_in
 | 
			
		||||
from django.core.cache import cache
 | 
			
		||||
from django.core.signals import Signal
 | 
			
		||||
from django.db.models import Model
 | 
			
		||||
from django.db.models.signals import post_save, pre_delete, pre_save
 | 
			
		||||
from django.db.models.signals import post_delete, post_save, pre_save
 | 
			
		||||
from django.dispatch import receiver
 | 
			
		||||
from django.http.request import HttpRequest
 | 
			
		||||
from structlog.stdlib import get_logger
 | 
			
		||||
@ -15,6 +14,7 @@ from authentik.core.models import (
 | 
			
		||||
    AuthenticatedSession,
 | 
			
		||||
    BackchannelProvider,
 | 
			
		||||
    ExpiringModel,
 | 
			
		||||
    Session,
 | 
			
		||||
    User,
 | 
			
		||||
    default_token_duration,
 | 
			
		||||
)
 | 
			
		||||
@ -49,19 +49,10 @@ def user_logged_in_session(sender, request: HttpRequest, user: User, **_):
 | 
			
		||||
        session.save()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@receiver(user_logged_out)
 | 
			
		||||
def user_logged_out_session(sender, request: HttpRequest, user: User, **_):
 | 
			
		||||
    """Delete AuthenticatedSession if it exists"""
 | 
			
		||||
    if not request.session or not request.session.session_key:
 | 
			
		||||
        return
 | 
			
		||||
    AuthenticatedSession.objects.filter(session_key=request.session.session_key).delete()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@receiver(pre_delete, sender=AuthenticatedSession)
 | 
			
		||||
@receiver(post_delete, sender=AuthenticatedSession)
 | 
			
		||||
def authenticated_session_delete(sender: type[Model], instance: "AuthenticatedSession", **_):
 | 
			
		||||
    """Delete session when authenticated session is deleted"""
 | 
			
		||||
    cache_key = f"{KEY_PREFIX}{instance.session_key}"
 | 
			
		||||
    cache.delete(cache_key)
 | 
			
		||||
    Session.objects.filter(session_key=instance.pk).delete()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@receiver(pre_save)
 | 
			
		||||
 | 
			
		||||
@ -2,22 +2,16 @@
 | 
			
		||||
 | 
			
		||||
from datetime import datetime, timedelta
 | 
			
		||||
 | 
			
		||||
from django.conf import ImproperlyConfigured
 | 
			
		||||
from django.contrib.sessions.backends.cache import KEY_PREFIX
 | 
			
		||||
from django.contrib.sessions.backends.db import SessionStore as DBSessionStore
 | 
			
		||||
from django.core.cache import cache
 | 
			
		||||
from django.utils.timezone import now
 | 
			
		||||
from structlog.stdlib import get_logger
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import (
 | 
			
		||||
    USER_ATTRIBUTE_EXPIRES,
 | 
			
		||||
    USER_ATTRIBUTE_GENERATED,
 | 
			
		||||
    AuthenticatedSession,
 | 
			
		||||
    ExpiringModel,
 | 
			
		||||
    User,
 | 
			
		||||
)
 | 
			
		||||
from authentik.events.system_tasks import SystemTask, TaskStatus, prefill_task
 | 
			
		||||
from authentik.lib.config import CONFIG
 | 
			
		||||
from authentik.root.celery import CELERY_APP
 | 
			
		||||
 | 
			
		||||
LOGGER = get_logger()
 | 
			
		||||
@ -38,40 +32,6 @@ def clean_expired_models(self: SystemTask):
 | 
			
		||||
            obj.expire_action()
 | 
			
		||||
        LOGGER.debug("Expired models", model=cls, amount=amount)
 | 
			
		||||
        messages.append(f"Expired {amount} {cls._meta.verbose_name_plural}")
 | 
			
		||||
    # Special case
 | 
			
		||||
    amount = 0
 | 
			
		||||
 | 
			
		||||
    for session in AuthenticatedSession.objects.all():
 | 
			
		||||
        match CONFIG.get("session_storage", "cache"):
 | 
			
		||||
            case "cache":
 | 
			
		||||
                cache_key = f"{KEY_PREFIX}{session.session_key}"
 | 
			
		||||
                value = None
 | 
			
		||||
                try:
 | 
			
		||||
                    value = cache.get(cache_key)
 | 
			
		||||
 | 
			
		||||
                except Exception as exc:
 | 
			
		||||
                    LOGGER.debug("Failed to get session from cache", exc=exc)
 | 
			
		||||
                if not value:
 | 
			
		||||
                    session.delete()
 | 
			
		||||
                    amount += 1
 | 
			
		||||
            case "db":
 | 
			
		||||
                if not (
 | 
			
		||||
                    DBSessionStore.get_model_class()
 | 
			
		||||
                    .objects.filter(session_key=session.session_key, expire_date__gt=now())
 | 
			
		||||
                    .exists()
 | 
			
		||||
                ):
 | 
			
		||||
                    session.delete()
 | 
			
		||||
                    amount += 1
 | 
			
		||||
            case _:
 | 
			
		||||
                # Should never happen, as we check for other values in authentik/root/settings.py
 | 
			
		||||
                raise ImproperlyConfigured(
 | 
			
		||||
                    "Invalid session_storage setting, allowed values are db and cache"
 | 
			
		||||
                )
 | 
			
		||||
    if CONFIG.get("session_storage", "cache") == "db":
 | 
			
		||||
        DBSessionStore.clear_expired()
 | 
			
		||||
    LOGGER.debug("Expired sessions", model=AuthenticatedSession, amount=amount)
 | 
			
		||||
 | 
			
		||||
    messages.append(f"Expired {amount} {AuthenticatedSession._meta.verbose_name_plural}")
 | 
			
		||||
    self.set_status(TaskStatus.SUCCESSFUL, *messages)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -2,30 +2,35 @@
 | 
			
		||||
{% get_current_language as LANGUAGE_CODE %}
 | 
			
		||||
 | 
			
		||||
<script>
 | 
			
		||||
    window.authentik = {
 | 
			
		||||
        locale: "{{ LANGUAGE_CODE }}",
 | 
			
		||||
        config: JSON.parse('{{ config_json|escapejs }}'),
 | 
			
		||||
        brand: JSON.parse('{{ brand_json|escapejs }}'),
 | 
			
		||||
        versionFamily: "{{ version_family }}",
 | 
			
		||||
        versionSubdomain: "{{ version_subdomain }}",
 | 
			
		||||
        build: "{{ build }}",
 | 
			
		||||
        api: {
 | 
			
		||||
            base: "{{ base_url }}",
 | 
			
		||||
            relBase: "{{ base_url_rel }}",
 | 
			
		||||
        },
 | 
			
		||||
    };
 | 
			
		||||
    window.addEventListener("DOMContentLoaded", function () {
 | 
			
		||||
        {% for message in messages %}
 | 
			
		||||
        window.dispatchEvent(
 | 
			
		||||
            new CustomEvent("ak-message", {
 | 
			
		||||
                bubbles: true,
 | 
			
		||||
                composed: true,
 | 
			
		||||
                detail: {
 | 
			
		||||
                    level: "{{ message.tags|escapejs }}",
 | 
			
		||||
                    message: "{{ message.message|escapejs }}",
 | 
			
		||||
                },
 | 
			
		||||
            }),
 | 
			
		||||
        );
 | 
			
		||||
        {% endfor %}
 | 
			
		||||
    });
 | 
			
		||||
  window.authentik = {
 | 
			
		||||
    locale: "{{ LANGUAGE_CODE }}",
 | 
			
		||||
    config: JSON.parse('{{ config_json|escapejs }}' || '{}'),
 | 
			
		||||
    brand: JSON.parse('{{ brand_json|escapejs }}' || '{}'),
 | 
			
		||||
    versionFamily: '{{ version_family }}',
 | 
			
		||||
    versionSubdomain: '{{ version_subdomain }}',
 | 
			
		||||
    build: '{{ build }}',
 | 
			
		||||
    api: {
 | 
			
		||||
      base: '{{ base_url }}',
 | 
			
		||||
      relBase: '{{ base_url_rel }}',
 | 
			
		||||
    },
 | 
			
		||||
  };
 | 
			
		||||
</script>
 | 
			
		||||
 | 
			
		||||
{% if messages %}
 | 
			
		||||
  <script>
 | 
			
		||||
    window.addEventListener("DOMContentLoaded", function () {
 | 
			
		||||
      {% for message in messages %}
 | 
			
		||||
        window.dispatchEvent(
 | 
			
		||||
          new CustomEvent("ak-message", {
 | 
			
		||||
              bubbles: true,
 | 
			
		||||
              composed: true,
 | 
			
		||||
              detail: {
 | 
			
		||||
                  level: "{{ message.tags|escapejs }}",
 | 
			
		||||
                  message: "{{ message.message|escapejs }}",
 | 
			
		||||
              },
 | 
			
		||||
          }),
 | 
			
		||||
        );
 | 
			
		||||
      {% endfor %}
 | 
			
		||||
    });
 | 
			
		||||
  </script>
 | 
			
		||||
{% endif %}
 | 
			
		||||
 | 
			
		||||
@ -2,31 +2,85 @@
 | 
			
		||||
{% load i18n %}
 | 
			
		||||
{% load authentik_core %}
 | 
			
		||||
 | 
			
		||||
<!DOCTYPE html>
 | 
			
		||||
<!doctype html>
 | 
			
		||||
<html lang="{{ get_current_language }}">
 | 
			
		||||
  <head>
 | 
			
		||||
    <meta charset="UTF-8">
 | 
			
		||||
    <meta name="viewport" content="width=device-width, initial-scale=1, maximum-scale=1">
 | 
			
		||||
 | 
			
		||||
<html>
 | 
			
		||||
    <head>
 | 
			
		||||
        <meta charset="UTF-8">
 | 
			
		||||
        <meta name="viewport" content="width=device-width, initial-scale=1, maximum-scale=1">
 | 
			
		||||
        {# Darkreader breaks the site regardless of theme as its not compatible with webcomponents, and we default to a dark theme based on preferred colour-scheme #}
 | 
			
		||||
        <meta name="darkreader-lock">
 | 
			
		||||
        <title>{% block title %}{% trans title|default:brand.branding_title %}{% endblock %}</title>
 | 
			
		||||
        <link rel="icon" href="{{ brand.branding_favicon_url }}">
 | 
			
		||||
        <link rel="shortcut icon" href="{{ brand.branding_favicon_url }}">
 | 
			
		||||
        {% block head_before %}
 | 
			
		||||
        {% endblock %}
 | 
			
		||||
        <link rel="stylesheet" type="text/css" href="{% static 'dist/authentik.css' %}">
 | 
			
		||||
        <style>{{ brand.branding_custom_css }}</style>
 | 
			
		||||
        <script src="{% versioned_script 'dist/poly-%v.js' %}" type="module"></script>
 | 
			
		||||
        <script src="{% versioned_script 'dist/standalone/loading/index-%v.js' %}" type="module"></script>
 | 
			
		||||
        {% block head %}
 | 
			
		||||
        {% endblock %}
 | 
			
		||||
        <meta name="sentry-trace" content="{{ sentry_trace }}" />
 | 
			
		||||
    </head>
 | 
			
		||||
    <body>
 | 
			
		||||
        {% block body %}
 | 
			
		||||
        {% endblock %}
 | 
			
		||||
        {% block scripts %}
 | 
			
		||||
        {% endblock %}
 | 
			
		||||
    </body>
 | 
			
		||||
    {% comment %}
 | 
			
		||||
      Darkreader breaks the site regardless of theme as its not compatible with webcomponents,
 | 
			
		||||
      and we default to a dark theme based on preferred colour-scheme.
 | 
			
		||||
    {% endcomment %}
 | 
			
		||||
 | 
			
		||||
    <meta name="darkreader-lock">
 | 
			
		||||
 | 
			
		||||
    <title>{% block title %}{% trans title|default:brand.branding_title %}{% endblock %}</title>
 | 
			
		||||
 | 
			
		||||
    <link rel="icon" href="{{ brand.branding_favicon_url }}">
 | 
			
		||||
    <link rel="shortcut icon" href="{{ brand.branding_favicon_url }}">
 | 
			
		||||
 | 
			
		||||
    {% block head_before %}
 | 
			
		||||
    {% endblock %}
 | 
			
		||||
 | 
			
		||||
    <link rel="stylesheet" type="text/css" href="{% static 'dist/authentik.css' %}">
 | 
			
		||||
 | 
			
		||||
    <style data-test-id="color-scheme">
 | 
			
		||||
      @media (prefers-color-scheme: dark) {
 | 
			
		||||
        :root {
 | 
			
		||||
          color-scheme: dark light;
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      @media (prefers-color-scheme: light) {
 | 
			
		||||
        :root {
 | 
			
		||||
          color-scheme: light dark;
 | 
			
		||||
        }
 | 
			
		||||
      }
 | 
			
		||||
    </style>
 | 
			
		||||
 | 
			
		||||
    <style data-test-id="custom-branding-css">
 | 
			
		||||
      {{ brand.branding_custom_css }}
 | 
			
		||||
    </style>
 | 
			
		||||
 | 
			
		||||
    <script
 | 
			
		||||
      data-test-id="entrypoint-polyfill"
 | 
			
		||||
      src="{% versioned_script 'dist/poly-%v.js' %}"
 | 
			
		||||
      type="module">
 | 
			
		||||
    </script>
 | 
			
		||||
 | 
			
		||||
    <script
 | 
			
		||||
      data-test-id="entrypoint-loading"
 | 
			
		||||
      src="{% versioned_script 'dist/standalone/loading/index-%v.js' %}"
 | 
			
		||||
      type="module">
 | 
			
		||||
    </script>
 | 
			
		||||
 | 
			
		||||
    {% block head %}
 | 
			
		||||
    {% endblock %}
 | 
			
		||||
 | 
			
		||||
    <meta name="sentry-trace" content="{{ sentry_trace }}">
 | 
			
		||||
  </head>
 | 
			
		||||
 | 
			
		||||
  <body>
 | 
			
		||||
    {% block body %}{% endblock %}
 | 
			
		||||
    {% block scripts %}{% endblock %}
 | 
			
		||||
 | 
			
		||||
    <noscript>
 | 
			
		||||
      <style>
 | 
			
		||||
        body {
 | 
			
		||||
          font-family: var(--ak-font-family-base), sans-serif;
 | 
			
		||||
        }
 | 
			
		||||
      </style>
 | 
			
		||||
 | 
			
		||||
      <h1>
 | 
			
		||||
        {% blocktrans with brand_title=title|default:brand.branding_title %}
 | 
			
		||||
          JavaScript is required to use {{ brand_title }}
 | 
			
		||||
        {% endblocktrans %}
 | 
			
		||||
      </h1>
 | 
			
		||||
 | 
			
		||||
      <p>
 | 
			
		||||
        {% trans 'Please enable JavaScript in your browser settings and reload the page. If you are using a browser extension that blocks JavaScript, please disable it for this site.' %}
 | 
			
		||||
      </p>
 | 
			
		||||
    </noscript>
 | 
			
		||||
  </body>
 | 
			
		||||
</html>
 | 
			
		||||
 | 
			
		||||
@ -3,15 +3,22 @@
 | 
			
		||||
{% load authentik_core %}
 | 
			
		||||
 | 
			
		||||
{% block head %}
 | 
			
		||||
<script src="{% versioned_script 'dist/admin/AdminInterface-%v.js' %}" type="module"></script>
 | 
			
		||||
<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" %}
 | 
			
		||||
  <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" %}
 | 
			
		||||
 | 
			
		||||
  <script
 | 
			
		||||
    data-test-id="entrypoint"
 | 
			
		||||
    src="{% versioned_script 'dist/admin/AdminInterface-%v.js' %}"
 | 
			
		||||
    type="module">
 | 
			
		||||
  </script>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block body %}
 | 
			
		||||
<ak-message-container></ak-message-container>
 | 
			
		||||
<ak-interface-admin>
 | 
			
		||||
  <ak-message-container></ak-message-container>
 | 
			
		||||
 | 
			
		||||
  <ak-interface-admin>
 | 
			
		||||
    <ak-loading></ak-loading>
 | 
			
		||||
</ak-interface-admin>
 | 
			
		||||
  </ak-interface-admin>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
@ -4,18 +4,22 @@
 | 
			
		||||
{% load i18n %}
 | 
			
		||||
 | 
			
		||||
{% block title %}
 | 
			
		||||
{{ brand.branding_title }}
 | 
			
		||||
  {{ brand.branding_title }}
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block card_title %}
 | 
			
		||||
{% trans title %}
 | 
			
		||||
  {% trans title %}
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block card %}
 | 
			
		||||
<form method="POST" class="pf-c-form">
 | 
			
		||||
  <form method="POST" class="pf-c-form">
 | 
			
		||||
    <p>{% trans message %}</p>
 | 
			
		||||
    <a id="ak-back-home" href="{% url 'authentik_core:root-redirect' %}" class="pf-c-button pf-m-primary">
 | 
			
		||||
        {% trans 'Go home' %}
 | 
			
		||||
 | 
			
		||||
    <a
 | 
			
		||||
      id="ak-back-home"
 | 
			
		||||
      href="{% url 'authentik_core:root-redirect' %}"
 | 
			
		||||
      class="pf-c-button pf-m-primary">
 | 
			
		||||
      {% trans 'Go home' %}
 | 
			
		||||
    </a>
 | 
			
		||||
</form>
 | 
			
		||||
  </form>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
@ -3,15 +3,22 @@
 | 
			
		||||
{% load authentik_core %}
 | 
			
		||||
 | 
			
		||||
{% block head %}
 | 
			
		||||
<script src="{% versioned_script 'dist/user/UserInterface-%v.js' %}" type="module"></script>
 | 
			
		||||
<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" %}
 | 
			
		||||
  <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" %}
 | 
			
		||||
 | 
			
		||||
  <script
 | 
			
		||||
    data-test-id="entrypoint"
 | 
			
		||||
    src="{% versioned_script 'dist/user/UserInterface-%v.js' %}"
 | 
			
		||||
    type="module">
 | 
			
		||||
  </script>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block body %}
 | 
			
		||||
<ak-message-container></ak-message-container>
 | 
			
		||||
<ak-interface-user>
 | 
			
		||||
  <ak-message-container></ak-message-container>
 | 
			
		||||
 | 
			
		||||
  <ak-interface-user>
 | 
			
		||||
    <ak-loading></ak-loading>
 | 
			
		||||
</ak-interface-user>
 | 
			
		||||
  </ak-interface-user>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
@ -4,79 +4,93 @@
 | 
			
		||||
{% load i18n %}
 | 
			
		||||
 | 
			
		||||
{% block head_before %}
 | 
			
		||||
<link rel="prefetch" href="{{ request.brand.branding_default_flow_background_url }}" />
 | 
			
		||||
<link rel="stylesheet" type="text/css" href="{% static 'dist/patternfly.min.css' %}">
 | 
			
		||||
<link rel="stylesheet" type="text/css" href="{% static 'dist/theme-dark.css' %}" media="(prefers-color-scheme: dark)">
 | 
			
		||||
{% include "base/header_js.html" %}
 | 
			
		||||
  <link rel="prefetch" href="{{ request.brand.branding_default_flow_background_url }}">
 | 
			
		||||
 | 
			
		||||
  <link rel="stylesheet" type="text/css" href="{% static 'dist/patternfly.min.css' %}">
 | 
			
		||||
  <link
 | 
			
		||||
    rel="stylesheet"
 | 
			
		||||
    type="text/css"
 | 
			
		||||
    href="{% static 'dist/theme-dark.css' %}"
 | 
			
		||||
    media="(prefers-color-scheme: dark)"
 | 
			
		||||
 >
 | 
			
		||||
 | 
			
		||||
  {% include "base/header_js.html" %}
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block head %}
 | 
			
		||||
<style>
 | 
			
		||||
:root {
 | 
			
		||||
    --ak-flow-background: url("{{ request.brand.branding_default_flow_background_url }}");
 | 
			
		||||
    --pf-c-background-image--BackgroundImage: var(--ak-flow-background);
 | 
			
		||||
    --pf-c-background-image--BackgroundImage-2x: var(--ak-flow-background);
 | 
			
		||||
    --pf-c-background-image--BackgroundImage--sm: var(--ak-flow-background);
 | 
			
		||||
    --pf-c-background-image--BackgroundImage--sm-2x: var(--ak-flow-background);
 | 
			
		||||
    --pf-c-background-image--BackgroundImage--lg: var(--ak-flow-background);
 | 
			
		||||
}
 | 
			
		||||
/* Form with user */
 | 
			
		||||
.form-control-static {
 | 
			
		||||
    margin-top: var(--pf-global--spacer--sm);
 | 
			
		||||
    display: flex;
 | 
			
		||||
    align-items: center;
 | 
			
		||||
    justify-content: space-between;
 | 
			
		||||
}
 | 
			
		||||
.form-control-static .avatar {
 | 
			
		||||
    display: flex;
 | 
			
		||||
    align-items: center;
 | 
			
		||||
}
 | 
			
		||||
.form-control-static img {
 | 
			
		||||
    margin-right: var(--pf-global--spacer--xs);
 | 
			
		||||
}
 | 
			
		||||
.form-control-static a {
 | 
			
		||||
    padding-top: var(--pf-global--spacer--xs);
 | 
			
		||||
    padding-bottom: var(--pf-global--spacer--xs);
 | 
			
		||||
    line-height: var(--pf-global--spacer--xl);
 | 
			
		||||
}
 | 
			
		||||
</style>
 | 
			
		||||
  <style data-test-id="base-full-root-styles">
 | 
			
		||||
    :root {
 | 
			
		||||
      --ak-flow-background: url("{{ request.brand.branding_default_flow_background_url }}");
 | 
			
		||||
      --pf-c-background-image--BackgroundImage: var(--ak-flow-background);
 | 
			
		||||
      --pf-c-background-image--BackgroundImage-2x: var(--ak-flow-background);
 | 
			
		||||
      --pf-c-background-image--BackgroundImage--sm: var(--ak-flow-background);
 | 
			
		||||
      --pf-c-background-image--BackgroundImage--sm-2x: var(--ak-flow-background);
 | 
			
		||||
      --pf-c-background-image--BackgroundImage--lg: var(--ak-flow-background);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    /* Form with user */
 | 
			
		||||
    .form-control-static {
 | 
			
		||||
      margin-top: var(--pf-global--spacer--sm);
 | 
			
		||||
      display: flex;
 | 
			
		||||
      align-items: center;
 | 
			
		||||
      justify-content: space-between;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    .form-control-static .avatar {
 | 
			
		||||
      display: flex;
 | 
			
		||||
      align-items: center;
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    .form-control-static img {
 | 
			
		||||
      margin-right: var(--pf-global--spacer--xs);
 | 
			
		||||
    }
 | 
			
		||||
 | 
			
		||||
    .form-control-static a {
 | 
			
		||||
      padding-top: var(--pf-global--spacer--xs);
 | 
			
		||||
      padding-bottom: var(--pf-global--spacer--xs);
 | 
			
		||||
      line-height: var(--pf-global--spacer--xl);
 | 
			
		||||
    }
 | 
			
		||||
  </style>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block body %}
 | 
			
		||||
<div class="pf-c-background-image">
 | 
			
		||||
</div>
 | 
			
		||||
<ak-message-container></ak-message-container>
 | 
			
		||||
<div class="pf-c-login stacked">
 | 
			
		||||
  <div class="pf-c-background-image"></div>
 | 
			
		||||
 | 
			
		||||
  <ak-message-container></ak-message-container>
 | 
			
		||||
 | 
			
		||||
  <div class="pf-c-login stacked">
 | 
			
		||||
    <div class="ak-login-container">
 | 
			
		||||
        <main class="pf-c-login__main">
 | 
			
		||||
            <div class="pf-c-login__main-header pf-c-brand ak-brand">
 | 
			
		||||
                <img src="{{ brand.branding_logo_url }}" alt="authentik Logo" />
 | 
			
		||||
            </div>
 | 
			
		||||
            <header class="pf-c-login__main-header">
 | 
			
		||||
                <h1 class="pf-c-title pf-m-3xl">
 | 
			
		||||
                    {% block card_title %}
 | 
			
		||||
                    {% endblock %}
 | 
			
		||||
                </h1>
 | 
			
		||||
            </header>
 | 
			
		||||
            <div class="pf-c-login__main-body">
 | 
			
		||||
                {% block card %}
 | 
			
		||||
                {% endblock %}
 | 
			
		||||
            </div>
 | 
			
		||||
        </main>
 | 
			
		||||
        <footer class="pf-c-login__footer">
 | 
			
		||||
            <ul class="pf-c-list pf-m-inline">
 | 
			
		||||
                {% for link in footer_links %}
 | 
			
		||||
                <li>
 | 
			
		||||
                    <a href="{{ link.href }}">{{ link.name }}</a>
 | 
			
		||||
                </li>
 | 
			
		||||
                {% endfor %}
 | 
			
		||||
                <li>
 | 
			
		||||
                    <span>
 | 
			
		||||
                        {% trans 'Powered by authentik' %}
 | 
			
		||||
                    </span>
 | 
			
		||||
                </li>
 | 
			
		||||
            </ul>
 | 
			
		||||
        </footer>
 | 
			
		||||
      <main class="pf-c-login__main">
 | 
			
		||||
        <div class="pf-c-login__main-header pf-c-brand ak-brand">
 | 
			
		||||
          <img src="{{ brand.branding_logo_url }}" alt="authentik Logo">
 | 
			
		||||
        </div>
 | 
			
		||||
 | 
			
		||||
        <header class="pf-c-login__main-header">
 | 
			
		||||
          <h1 class="pf-c-title pf-m-3xl">
 | 
			
		||||
            {% block card_title %}
 | 
			
		||||
            {% endblock %}
 | 
			
		||||
          </h1>
 | 
			
		||||
        </header>
 | 
			
		||||
 | 
			
		||||
        <div class="pf-c-login__main-body">
 | 
			
		||||
          {% block card %}
 | 
			
		||||
          {% endblock %}
 | 
			
		||||
        </div>
 | 
			
		||||
      </main>
 | 
			
		||||
 | 
			
		||||
      <footer class="pf-c-login__footer">
 | 
			
		||||
        <ul class="pf-c-list pf-m-inline">
 | 
			
		||||
          {% for link in footer_links %}
 | 
			
		||||
            <li>
 | 
			
		||||
              <a href="{{ link.href }}">{{ link.name }}</a>
 | 
			
		||||
            </li>
 | 
			
		||||
          {% endfor %}
 | 
			
		||||
 | 
			
		||||
          <li>
 | 
			
		||||
            <span>{% trans 'Powered by authentik' %}</span>
 | 
			
		||||
          </li>
 | 
			
		||||
        </ul>
 | 
			
		||||
      </footer>
 | 
			
		||||
    </div>
 | 
			
		||||
</div>
 | 
			
		||||
  </div>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
@ -1,9 +1,17 @@
 | 
			
		||||
"""Test API Utils"""
 | 
			
		||||
 | 
			
		||||
from rest_framework.exceptions import ValidationError
 | 
			
		||||
from rest_framework.serializers import (
 | 
			
		||||
    HyperlinkedModelSerializer,
 | 
			
		||||
)
 | 
			
		||||
from rest_framework.serializers import (
 | 
			
		||||
    ModelSerializer as BaseModelSerializer,
 | 
			
		||||
)
 | 
			
		||||
from rest_framework.test import APITestCase
 | 
			
		||||
 | 
			
		||||
from authentik.core.api.utils import ModelSerializer as CustomModelSerializer
 | 
			
		||||
from authentik.core.api.utils import is_dict
 | 
			
		||||
from authentik.lib.utils.reflection import all_subclasses
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class TestAPIUtils(APITestCase):
 | 
			
		||||
@ -14,3 +22,14 @@ class TestAPIUtils(APITestCase):
 | 
			
		||||
        self.assertIsNone(is_dict({}))
 | 
			
		||||
        with self.assertRaises(ValidationError):
 | 
			
		||||
            is_dict("foo")
 | 
			
		||||
 | 
			
		||||
    def test_all_serializers_descend_from_custom(self):
 | 
			
		||||
        """Test that every serializer we define descends from our own ModelSerializer"""
 | 
			
		||||
        # Weirdly, there's only one serializer in `rest_framework` which descends from
 | 
			
		||||
        # ModelSerializer: HyperlinkedModelSerializer
 | 
			
		||||
        expected = {CustomModelSerializer, HyperlinkedModelSerializer}
 | 
			
		||||
        actual = set(all_subclasses(BaseModelSerializer)) - set(
 | 
			
		||||
            all_subclasses(CustomModelSerializer)
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        self.assertEqual(expected, actual)
 | 
			
		||||
 | 
			
		||||
@ -133,8 +133,6 @@ class TestApplicationsAPI(APITestCase):
 | 
			
		||||
                        "provider_obj": {
 | 
			
		||||
                            "assigned_application_name": "allowed",
 | 
			
		||||
                            "assigned_application_slug": "allowed",
 | 
			
		||||
                            "assigned_backchannel_application_name": "",
 | 
			
		||||
                            "assigned_backchannel_application_slug": "",
 | 
			
		||||
                            "authentication_flow": None,
 | 
			
		||||
                            "invalidation_flow": None,
 | 
			
		||||
                            "authorization_flow": str(self.provider.authorization_flow.pk),
 | 
			
		||||
@ -188,8 +186,6 @@ class TestApplicationsAPI(APITestCase):
 | 
			
		||||
                        "provider_obj": {
 | 
			
		||||
                            "assigned_application_name": "allowed",
 | 
			
		||||
                            "assigned_application_slug": "allowed",
 | 
			
		||||
                            "assigned_backchannel_application_name": "",
 | 
			
		||||
                            "assigned_backchannel_application_slug": "",
 | 
			
		||||
                            "authentication_flow": None,
 | 
			
		||||
                            "invalidation_flow": None,
 | 
			
		||||
                            "authorization_flow": str(self.provider.authorization_flow.pk),
 | 
			
		||||
 | 
			
		||||
@ -5,7 +5,7 @@ from json import loads
 | 
			
		||||
from django.urls.base import reverse
 | 
			
		||||
from rest_framework.test import APITestCase
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import User
 | 
			
		||||
from authentik.core.models import AuthenticatedSession, Session, User
 | 
			
		||||
from authentik.core.tests.utils import create_test_admin_user
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -30,3 +30,18 @@ class TestAuthenticatedSessionsAPI(APITestCase):
 | 
			
		||||
        self.assertEqual(response.status_code, 200)
 | 
			
		||||
        body = loads(response.content.decode())
 | 
			
		||||
        self.assertEqual(body["pagination"]["count"], 1)
 | 
			
		||||
 | 
			
		||||
    def test_delete(self):
 | 
			
		||||
        """Test deletion"""
 | 
			
		||||
        self.client.force_login(self.user)
 | 
			
		||||
        self.assertEqual(AuthenticatedSession.objects.all().count(), 1)
 | 
			
		||||
        self.assertEqual(Session.objects.all().count(), 1)
 | 
			
		||||
        response = self.client.delete(
 | 
			
		||||
            reverse(
 | 
			
		||||
                "authentik_api:authenticatedsession-detail",
 | 
			
		||||
                kwargs={"uuid": AuthenticatedSession.objects.first().uuid},
 | 
			
		||||
            )
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(response.status_code, 204)
 | 
			
		||||
        self.assertEqual(AuthenticatedSession.objects.all().count(), 0)
 | 
			
		||||
        self.assertEqual(Session.objects.all().count(), 0)
 | 
			
		||||
 | 
			
		||||
@ -3,8 +3,7 @@
 | 
			
		||||
from django.urls import reverse
 | 
			
		||||
from rest_framework.test import APITestCase
 | 
			
		||||
 | 
			
		||||
from authentik.core.api.providers import ProviderSerializer
 | 
			
		||||
from authentik.core.models import Application, PropertyMapping, Provider
 | 
			
		||||
from authentik.core.models import PropertyMapping
 | 
			
		||||
from authentik.core.tests.utils import create_test_admin_user
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -25,51 +24,3 @@ class TestProvidersAPI(APITestCase):
 | 
			
		||||
            reverse("authentik_api:provider-types"),
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(response.status_code, 200)
 | 
			
		||||
 | 
			
		||||
    def test_provider_serializer_without_application(self):
 | 
			
		||||
        """Test that Provider serializer handles missing application gracefully"""
 | 
			
		||||
        # Create a provider without an application
 | 
			
		||||
        provider = Provider.objects.create(name="test-provider")
 | 
			
		||||
 | 
			
		||||
        serializer = ProviderSerializer(instance=provider)
 | 
			
		||||
        serialized_data = serializer.data
 | 
			
		||||
 | 
			
		||||
        # Check that fields return empty strings when no application exists
 | 
			
		||||
        self.assertEqual(serialized_data["assigned_application_slug"], "")
 | 
			
		||||
        self.assertEqual(serialized_data["assigned_application_name"], "")
 | 
			
		||||
        self.assertEqual(serialized_data["assigned_backchannel_application_slug"], "")
 | 
			
		||||
        self.assertEqual(serialized_data["assigned_backchannel_application_name"], "")
 | 
			
		||||
 | 
			
		||||
    def test_provider_serializer_with_application(self):
 | 
			
		||||
        """Test that Provider serializer correctly includes application data"""
 | 
			
		||||
        # Create an application
 | 
			
		||||
        app = Application.objects.create(name="Test App", slug="test-app")
 | 
			
		||||
 | 
			
		||||
        # Create a provider with an application
 | 
			
		||||
        provider = Provider.objects.create(name="test-provider-with-app")
 | 
			
		||||
        app.provider = provider
 | 
			
		||||
        app.save()
 | 
			
		||||
 | 
			
		||||
        serializer = ProviderSerializer(instance=provider)
 | 
			
		||||
        serialized_data = serializer.data
 | 
			
		||||
 | 
			
		||||
        # Check that fields return correct values when application exists
 | 
			
		||||
        self.assertEqual(serialized_data["assigned_application_slug"], "test-app")
 | 
			
		||||
        self.assertEqual(serialized_data["assigned_application_name"], "Test App")
 | 
			
		||||
        self.assertEqual(serialized_data["assigned_backchannel_application_slug"], "")
 | 
			
		||||
        self.assertEqual(serialized_data["assigned_backchannel_application_name"], "")
 | 
			
		||||
 | 
			
		||||
    def test_provider_api_response(self):
 | 
			
		||||
        """Test that the API response includes empty strings for missing applications"""
 | 
			
		||||
        # Create a provider without an application
 | 
			
		||||
        provider = Provider.objects.create(name="test-provider-api")
 | 
			
		||||
 | 
			
		||||
        response = self.client.get(
 | 
			
		||||
            reverse("authentik_api:provider-detail", kwargs={"pk": provider.pk}),
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        self.assertEqual(response.status_code, 200)
 | 
			
		||||
        self.assertEqual(response.data["assigned_application_slug"], "")
 | 
			
		||||
        self.assertEqual(response.data["assigned_application_name"], "")
 | 
			
		||||
        self.assertEqual(response.data["assigned_backchannel_application_slug"], "")
 | 
			
		||||
        self.assertEqual(response.data["assigned_backchannel_application_name"], "")
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										19
									
								
								authentik/core/tests/test_source_api.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										19
									
								
								authentik/core/tests/test_source_api.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,19 @@
 | 
			
		||||
from django.apps import apps
 | 
			
		||||
from django.urls import reverse
 | 
			
		||||
from rest_framework.test import APITestCase
 | 
			
		||||
 | 
			
		||||
from authentik.core.tests.utils import create_test_admin_user
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class TestSourceAPI(APITestCase):
 | 
			
		||||
    def setUp(self) -> None:
 | 
			
		||||
        self.user = create_test_admin_user()
 | 
			
		||||
        self.client.force_login(self.user)
 | 
			
		||||
 | 
			
		||||
    def test_builtin_source_used_by(self):
 | 
			
		||||
        """Test Providers's types endpoint"""
 | 
			
		||||
        apps.get_app_config("authentik_core").source_inbuilt()
 | 
			
		||||
        response = self.client.get(
 | 
			
		||||
            reverse("authentik_api:source-used-by", kwargs={"slug": "authentik-built-in"}),
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(response.status_code, 200)
 | 
			
		||||
@ -1,9 +1,8 @@
 | 
			
		||||
"""Test Users API"""
 | 
			
		||||
 | 
			
		||||
from datetime import datetime
 | 
			
		||||
from json import loads
 | 
			
		||||
 | 
			
		||||
from django.contrib.sessions.backends.cache import KEY_PREFIX
 | 
			
		||||
from django.core.cache import cache
 | 
			
		||||
from django.urls.base import reverse
 | 
			
		||||
from rest_framework.test import APITestCase
 | 
			
		||||
 | 
			
		||||
@ -11,11 +10,17 @@ from authentik.brands.models import Brand
 | 
			
		||||
from authentik.core.models import (
 | 
			
		||||
    USER_ATTRIBUTE_TOKEN_EXPIRING,
 | 
			
		||||
    AuthenticatedSession,
 | 
			
		||||
    Session,
 | 
			
		||||
    Token,
 | 
			
		||||
    User,
 | 
			
		||||
    UserTypes,
 | 
			
		||||
)
 | 
			
		||||
from authentik.core.tests.utils import create_test_admin_user, create_test_brand, create_test_flow
 | 
			
		||||
from authentik.core.tests.utils import (
 | 
			
		||||
    create_test_admin_user,
 | 
			
		||||
    create_test_brand,
 | 
			
		||||
    create_test_flow,
 | 
			
		||||
    create_test_user,
 | 
			
		||||
)
 | 
			
		||||
from authentik.flows.models import FlowDesignation
 | 
			
		||||
from authentik.lib.generators import generate_id, generate_key
 | 
			
		||||
from authentik.stages.email.models import EmailStage
 | 
			
		||||
@ -26,7 +31,7 @@ class TestUsersAPI(APITestCase):
 | 
			
		||||
 | 
			
		||||
    def setUp(self) -> None:
 | 
			
		||||
        self.admin = create_test_admin_user()
 | 
			
		||||
        self.user = User.objects.create(username="test-user")
 | 
			
		||||
        self.user = create_test_user()
 | 
			
		||||
 | 
			
		||||
    def test_filter_type(self):
 | 
			
		||||
        """Test API filtering by type"""
 | 
			
		||||
@ -41,6 +46,35 @@ class TestUsersAPI(APITestCase):
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(response.status_code, 200)
 | 
			
		||||
 | 
			
		||||
    def test_filter_is_superuser(self):
 | 
			
		||||
        """Test API filtering by superuser status"""
 | 
			
		||||
        User.objects.all().delete()
 | 
			
		||||
        admin = create_test_admin_user()
 | 
			
		||||
        self.client.force_login(admin)
 | 
			
		||||
        # Test superuser
 | 
			
		||||
        response = self.client.get(
 | 
			
		||||
            reverse("authentik_api:user-list"),
 | 
			
		||||
            data={
 | 
			
		||||
                "is_superuser": True,
 | 
			
		||||
            },
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(response.status_code, 200)
 | 
			
		||||
        body = loads(response.content)
 | 
			
		||||
        self.assertEqual(len(body["results"]), 1)
 | 
			
		||||
        self.assertEqual(body["results"][0]["username"], admin.username)
 | 
			
		||||
        # Test non-superuser
 | 
			
		||||
        user = create_test_user()
 | 
			
		||||
        response = self.client.get(
 | 
			
		||||
            reverse("authentik_api:user-list"),
 | 
			
		||||
            data={
 | 
			
		||||
                "is_superuser": False,
 | 
			
		||||
            },
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(response.status_code, 200)
 | 
			
		||||
        body = loads(response.content)
 | 
			
		||||
        self.assertEqual(len(body["results"]), 1, body)
 | 
			
		||||
        self.assertEqual(body["results"][0]["username"], user.username)
 | 
			
		||||
 | 
			
		||||
    def test_list_with_groups(self):
 | 
			
		||||
        """Test listing with groups"""
 | 
			
		||||
        self.client.force_login(self.admin)
 | 
			
		||||
@ -99,6 +133,8 @@ class TestUsersAPI(APITestCase):
 | 
			
		||||
    def test_recovery_email_no_flow(self):
 | 
			
		||||
        """Test user recovery link (no recovery flow set)"""
 | 
			
		||||
        self.client.force_login(self.admin)
 | 
			
		||||
        self.user.email = ""
 | 
			
		||||
        self.user.save()
 | 
			
		||||
        response = self.client.post(
 | 
			
		||||
            reverse("authentik_api:user-recovery-email", kwargs={"pk": self.user.pk})
 | 
			
		||||
        )
 | 
			
		||||
@ -344,12 +380,15 @@ class TestUsersAPI(APITestCase):
 | 
			
		||||
        """Ensure sessions are deleted when a user is deactivated"""
 | 
			
		||||
        user = create_test_admin_user()
 | 
			
		||||
        session_id = generate_id()
 | 
			
		||||
        AuthenticatedSession.objects.create(
 | 
			
		||||
            user=user,
 | 
			
		||||
        session = Session.objects.create(
 | 
			
		||||
            session_key=session_id,
 | 
			
		||||
            last_ip="",
 | 
			
		||||
            last_ip="255.255.255.255",
 | 
			
		||||
            last_user_agent="",
 | 
			
		||||
        )
 | 
			
		||||
        AuthenticatedSession.objects.create(
 | 
			
		||||
            session=session,
 | 
			
		||||
            user=user,
 | 
			
		||||
        )
 | 
			
		||||
        cache.set(KEY_PREFIX + session_id, "foo")
 | 
			
		||||
 | 
			
		||||
        self.client.force_login(self.admin)
 | 
			
		||||
        response = self.client.patch(
 | 
			
		||||
@ -360,5 +399,7 @@ class TestUsersAPI(APITestCase):
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(response.status_code, 200)
 | 
			
		||||
 | 
			
		||||
        self.assertIsNone(cache.get(KEY_PREFIX + session_id))
 | 
			
		||||
        self.assertFalse(AuthenticatedSession.objects.filter(session_key=session_id).exists())
 | 
			
		||||
        self.assertFalse(Session.objects.filter(session_key=session_id).exists())
 | 
			
		||||
        self.assertFalse(
 | 
			
		||||
            AuthenticatedSession.objects.filter(session__session_key=session_id).exists()
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
@ -1,7 +1,5 @@
 | 
			
		||||
"""authentik URL Configuration"""
 | 
			
		||||
 | 
			
		||||
from channels.auth import AuthMiddleware
 | 
			
		||||
from channels.sessions import CookieMiddleware
 | 
			
		||||
from django.conf import settings
 | 
			
		||||
from django.contrib.auth.decorators import login_required
 | 
			
		||||
from django.urls import path
 | 
			
		||||
@ -13,7 +11,11 @@ from authentik.core.api.devices import AdminDeviceViewSet, DeviceViewSet
 | 
			
		||||
from authentik.core.api.groups import GroupViewSet
 | 
			
		||||
from authentik.core.api.property_mappings import PropertyMappingViewSet
 | 
			
		||||
from authentik.core.api.providers import ProviderViewSet
 | 
			
		||||
from authentik.core.api.sources import SourceViewSet, UserSourceConnectionViewSet
 | 
			
		||||
from authentik.core.api.sources import (
 | 
			
		||||
    GroupSourceConnectionViewSet,
 | 
			
		||||
    SourceViewSet,
 | 
			
		||||
    UserSourceConnectionViewSet,
 | 
			
		||||
)
 | 
			
		||||
from authentik.core.api.tokens import TokenViewSet
 | 
			
		||||
from authentik.core.api.transactional_applications import TransactionalApplicationView
 | 
			
		||||
from authentik.core.api.users import UserViewSet
 | 
			
		||||
@ -25,7 +27,7 @@ from authentik.core.views.interface import (
 | 
			
		||||
    RootRedirectView,
 | 
			
		||||
)
 | 
			
		||||
from authentik.flows.views.interface import FlowInterfaceView
 | 
			
		||||
from authentik.root.asgi_middleware import SessionMiddleware
 | 
			
		||||
from authentik.root.asgi_middleware import AuthMiddlewareStack
 | 
			
		||||
from authentik.root.messages.consumer import MessageConsumer
 | 
			
		||||
from authentik.root.middleware import ChannelsLoggingMiddleware
 | 
			
		||||
 | 
			
		||||
@ -81,6 +83,7 @@ api_urlpatterns = [
 | 
			
		||||
    ("core/tokens", TokenViewSet),
 | 
			
		||||
    ("sources/all", SourceViewSet),
 | 
			
		||||
    ("sources/user_connections/all", UserSourceConnectionViewSet),
 | 
			
		||||
    ("sources/group_connections/all", GroupSourceConnectionViewSet),
 | 
			
		||||
    ("providers/all", ProviderViewSet),
 | 
			
		||||
    ("propertymappings/all", PropertyMappingViewSet),
 | 
			
		||||
    ("authenticators/all", DeviceViewSet, "device"),
 | 
			
		||||
@ -94,9 +97,7 @@ api_urlpatterns = [
 | 
			
		||||
websocket_urlpatterns = [
 | 
			
		||||
    path(
 | 
			
		||||
        "ws/client/",
 | 
			
		||||
        ChannelsLoggingMiddleware(
 | 
			
		||||
            CookieMiddleware(SessionMiddleware(AuthMiddleware(MessageConsumer.as_asgi())))
 | 
			
		||||
        ),
 | 
			
		||||
        ChannelsLoggingMiddleware(AuthMiddlewareStack(MessageConsumer.as_asgi())),
 | 
			
		||||
    ),
 | 
			
		||||
]
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -102,7 +102,7 @@ def ssf_user_session_delete_session_revoked(sender, instance: AuthenticatedSessi
 | 
			
		||||
            "format": "complex",
 | 
			
		||||
            "session": {
 | 
			
		||||
                "format": "opaque",
 | 
			
		||||
                "id": sha256(instance.session_key.encode("ascii")).hexdigest(),
 | 
			
		||||
                "id": sha256(instance.session.session_key.encode("ascii")).hexdigest(),
 | 
			
		||||
            },
 | 
			
		||||
            "user": {
 | 
			
		||||
                "format": "email",
 | 
			
		||||
 | 
			
		||||
@ -4,10 +4,9 @@ from rest_framework.exceptions import PermissionDenied, ValidationError
 | 
			
		||||
from rest_framework.fields import CharField, ChoiceField, ListField, SerializerMethodField
 | 
			
		||||
from rest_framework.request import Request
 | 
			
		||||
from rest_framework.response import Response
 | 
			
		||||
from rest_framework.serializers import ModelSerializer
 | 
			
		||||
from structlog.stdlib import get_logger
 | 
			
		||||
 | 
			
		||||
from authentik.core.api.utils import PassiveSerializer
 | 
			
		||||
from authentik.core.api.utils import ModelSerializer, PassiveSerializer
 | 
			
		||||
from authentik.enterprise.providers.ssf.models import (
 | 
			
		||||
    DeliveryMethods,
 | 
			
		||||
    EventTypes,
 | 
			
		||||
 | 
			
		||||
@ -2,11 +2,11 @@
 | 
			
		||||
 | 
			
		||||
from rest_framework import mixins
 | 
			
		||||
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.core.api.used_by import UsedByMixin
 | 
			
		||||
from authentik.core.api.utils import ModelSerializer
 | 
			
		||||
from authentik.enterprise.api import EnterpriseRequiredMixin
 | 
			
		||||
from authentik.enterprise.stages.authenticator_endpoint_gdtc.models import (
 | 
			
		||||
    AuthenticatorEndpointGDTCStage,
 | 
			
		||||
 | 
			
		||||
@ -59,7 +59,7 @@ def get_login_event(request_or_session: HttpRequest | AuthenticatedSession | Non
 | 
			
		||||
        session = request_or_session.session
 | 
			
		||||
    if isinstance(request_or_session, AuthenticatedSession):
 | 
			
		||||
        SessionStore = _session_engine.SessionStore
 | 
			
		||||
        session = SessionStore(request_or_session.session_key)
 | 
			
		||||
        session = SessionStore(request_or_session.session.session_key)
 | 
			
		||||
    return session.get(SESSION_LOGIN_EVENT, None)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -2,53 +2,63 @@
 | 
			
		||||
{% load i18n %}
 | 
			
		||||
{% load authentik_core %}
 | 
			
		||||
 | 
			
		||||
<!DOCTYPE html>
 | 
			
		||||
<!doctype html>
 | 
			
		||||
 | 
			
		||||
<html lang="en">
 | 
			
		||||
    <head>
 | 
			
		||||
        <meta charset="UTF-8">
 | 
			
		||||
        <meta name="viewport" content="width=device-width, initial-scale=1, maximum-scale=1">
 | 
			
		||||
        <title>{% block title %}{% trans title|default:brand.branding_title %}{% endblock %}</title>
 | 
			
		||||
        <link rel="icon" href="{{ brand.branding_favicon_url }}">
 | 
			
		||||
        <link rel="shortcut icon" href="{{ brand.branding_favicon_url }}">
 | 
			
		||||
        {% block head_before %}
 | 
			
		||||
        {% endblock %}
 | 
			
		||||
        <link rel="stylesheet" type="text/css" href="{% static 'dist/sfe/bootstrap.min.css' %}">
 | 
			
		||||
        <meta name="sentry-trace" content="{{ sentry_trace }}" />
 | 
			
		||||
        {% include "base/header_js.html" %}
 | 
			
		||||
        <style>
 | 
			
		||||
          html,
 | 
			
		||||
          body {
 | 
			
		||||
            height: 100%;
 | 
			
		||||
          }
 | 
			
		||||
          body {
 | 
			
		||||
            background-image: url("{{ flow.background_url }}");
 | 
			
		||||
            background-repeat: no-repeat;
 | 
			
		||||
            background-size: cover;
 | 
			
		||||
          }
 | 
			
		||||
          .card {
 | 
			
		||||
            padding: 3rem;
 | 
			
		||||
          }
 | 
			
		||||
  <head>
 | 
			
		||||
    <meta charset="UTF-8">
 | 
			
		||||
    <meta name="viewport" content="width=device-width, initial-scale=1, maximum-scale=1">
 | 
			
		||||
 | 
			
		||||
          .form-signin {
 | 
			
		||||
            max-width: 330px;
 | 
			
		||||
            padding: 1rem;
 | 
			
		||||
          }
 | 
			
		||||
    <title>{% block title %}{% trans title|default:brand.branding_title %}{% endblock %}</title>
 | 
			
		||||
 | 
			
		||||
          .form-signin .form-floating:focus-within {
 | 
			
		||||
            z-index: 2;
 | 
			
		||||
          }
 | 
			
		||||
          .brand-icon {
 | 
			
		||||
            max-width: 100%;
 | 
			
		||||
          }
 | 
			
		||||
        </style>
 | 
			
		||||
    </head>
 | 
			
		||||
    <body class="d-flex align-items-center py-4 bg-body-tertiary">
 | 
			
		||||
      <div class="card m-auto">
 | 
			
		||||
        <main class="form-signin w-100 m-auto" id="flow-sfe-container">
 | 
			
		||||
        </main>
 | 
			
		||||
        <span class="mt-3 mb-0 text-muted text-center">{% trans 'Powered by authentik' %}</span>
 | 
			
		||||
      </div>
 | 
			
		||||
      <script src="{% static 'dist/sfe/index.js' %}"></script>
 | 
			
		||||
    </body>
 | 
			
		||||
    <link rel="icon" href="{{ brand.branding_favicon_url }}">
 | 
			
		||||
    <link rel="shortcut icon" href="{{ brand.branding_favicon_url }}">
 | 
			
		||||
 | 
			
		||||
    {% block head_before %}
 | 
			
		||||
    {% endblock %}
 | 
			
		||||
 | 
			
		||||
    <link rel="stylesheet" type="text/css" href="{% static 'dist/sfe/bootstrap.min.css' %}">
 | 
			
		||||
    <meta name="sentry-trace" content="{{ sentry_trace }}">
 | 
			
		||||
 | 
			
		||||
    {% include "base/header_js.html" %}
 | 
			
		||||
 | 
			
		||||
    <style>
 | 
			
		||||
      html,
 | 
			
		||||
      body {
 | 
			
		||||
        height: 100%;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      body {
 | 
			
		||||
        background-image: url("{{ flow.background_url }}");
 | 
			
		||||
        background-repeat: no-repeat;
 | 
			
		||||
        background-size: cover;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      .card {
 | 
			
		||||
        padding: 3rem;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      .form-signin {
 | 
			
		||||
        max-width: 330px;
 | 
			
		||||
        padding: 1rem;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      .form-signin .form-floating:focus-within {
 | 
			
		||||
        z-index: 2;
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      .brand-icon {
 | 
			
		||||
        max-width: 100%;
 | 
			
		||||
      }
 | 
			
		||||
    </style>
 | 
			
		||||
  </head>
 | 
			
		||||
 | 
			
		||||
  <body class="d-flex align-items-center py-4 bg-body-tertiary">
 | 
			
		||||
    <div class="card m-auto">
 | 
			
		||||
      <main class="form-signin w-100 m-auto" id="flow-sfe-container"></main>
 | 
			
		||||
      <span class="mt-3 mb-0 text-muted text-center">{% trans 'Powered by authentik' %}</span>
 | 
			
		||||
    </div>
 | 
			
		||||
 | 
			
		||||
    <script src="{% static 'dist/sfe/index.js' %}"></script>
 | 
			
		||||
  </body>
 | 
			
		||||
</html>
 | 
			
		||||
 | 
			
		||||
@ -4,31 +4,45 @@
 | 
			
		||||
{% load authentik_core %}
 | 
			
		||||
 | 
			
		||||
{% block head_before %}
 | 
			
		||||
{{ block.super }}
 | 
			
		||||
<link rel="prefetch" href="{{ flow.background_url }}" />
 | 
			
		||||
{% if flow.compatibility_mode and not inspector %}
 | 
			
		||||
<script>ShadyDOM = { force: !navigator.webdriver };</script>
 | 
			
		||||
{% endif %}
 | 
			
		||||
{% include "base/header_js.html" %}
 | 
			
		||||
<script>
 | 
			
		||||
window.authentik.flow = {
 | 
			
		||||
    "layout": "{{ flow.layout }}",
 | 
			
		||||
};
 | 
			
		||||
</script>
 | 
			
		||||
  {{ block.super }}
 | 
			
		||||
 | 
			
		||||
  <link rel="prefetch" href="{{ flow.background_url }}">
 | 
			
		||||
 | 
			
		||||
  {% if flow.compatibility_mode and not inspector %}
 | 
			
		||||
    <script>
 | 
			
		||||
      window.ShadyDOM = { force: !navigator.webdriver };
 | 
			
		||||
    </script>
 | 
			
		||||
  {% endif %}
 | 
			
		||||
 | 
			
		||||
  {% include "base/header_js.html" %}
 | 
			
		||||
 | 
			
		||||
  <script>
 | 
			
		||||
    window.authentik.flow = {
 | 
			
		||||
      layout: "{{ flow.layout }}",
 | 
			
		||||
    };
 | 
			
		||||
  </script>
 | 
			
		||||
 | 
			
		||||
  <meta name="ak-flow-layout" content="{{ flow.layout }}">
 | 
			
		||||
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block head %}
 | 
			
		||||
<script src="{% versioned_script 'dist/flow/FlowInterface-%v.js' %}" type="module"></script>
 | 
			
		||||
<style>
 | 
			
		||||
:root {
 | 
			
		||||
    --ak-flow-background: url("{{ flow.background_url }}");
 | 
			
		||||
}
 | 
			
		||||
</style>
 | 
			
		||||
  <script
 | 
			
		||||
    data-test-id="entrypoint"
 | 
			
		||||
    src="{% versioned_script 'dist/flow/FlowInterface-%v.js' %}"
 | 
			
		||||
    type="module">
 | 
			
		||||
  </script>
 | 
			
		||||
 | 
			
		||||
  <style data-test-id="flow-root-styles">
 | 
			
		||||
    :root {
 | 
			
		||||
      --ak-flow-background: url("{{ flow.background_url }}");
 | 
			
		||||
    }
 | 
			
		||||
  </style>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block body %}
 | 
			
		||||
<ak-message-container></ak-message-container>
 | 
			
		||||
<ak-flow-executor flowSlug="{{ flow.slug }}">
 | 
			
		||||
  <ak-message-container></ak-message-container>
 | 
			
		||||
  <ak-flow-executor flowSlug="{{ flow.slug }}">
 | 
			
		||||
    <ak-loading></ak-loading>
 | 
			
		||||
</ak-flow-executor>
 | 
			
		||||
  </ak-flow-executor>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
@ -69,6 +69,7 @@ SESSION_KEY_APPLICATION_PRE = "authentik/flows/application_pre"
 | 
			
		||||
SESSION_KEY_GET = "authentik/flows/get"
 | 
			
		||||
SESSION_KEY_POST = "authentik/flows/post"
 | 
			
		||||
SESSION_KEY_HISTORY = "authentik/flows/history"
 | 
			
		||||
SESSION_KEY_AUTH_STARTED = "authentik/flows/auth_started"
 | 
			
		||||
QS_KEY_TOKEN = "flow_token"  # nosec
 | 
			
		||||
QS_QUERY = "query"
 | 
			
		||||
 | 
			
		||||
@ -453,6 +454,7 @@ class FlowExecutorView(APIView):
 | 
			
		||||
            SESSION_KEY_APPLICATION_PRE,
 | 
			
		||||
            SESSION_KEY_PLAN,
 | 
			
		||||
            SESSION_KEY_GET,
 | 
			
		||||
            SESSION_KEY_AUTH_STARTED,
 | 
			
		||||
            # We might need the initial POST payloads for later requests
 | 
			
		||||
            # SESSION_KEY_POST,
 | 
			
		||||
            # We don't delete the history on purpose, as a user might
 | 
			
		||||
 | 
			
		||||
@ -6,14 +6,22 @@ from django.shortcuts import get_object_or_404
 | 
			
		||||
from ua_parser.user_agent_parser import Parse
 | 
			
		||||
 | 
			
		||||
from authentik.core.views.interface import InterfaceView
 | 
			
		||||
from authentik.flows.models import Flow
 | 
			
		||||
from authentik.flows.models import Flow, FlowDesignation
 | 
			
		||||
from authentik.flows.views.executor import SESSION_KEY_AUTH_STARTED
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class FlowInterfaceView(InterfaceView):
 | 
			
		||||
    """Flow interface"""
 | 
			
		||||
 | 
			
		||||
    def get_context_data(self, **kwargs: Any) -> dict[str, Any]:
 | 
			
		||||
        kwargs["flow"] = get_object_or_404(Flow, slug=self.kwargs.get("flow_slug"))
 | 
			
		||||
        flow = get_object_or_404(Flow, slug=self.kwargs.get("flow_slug"))
 | 
			
		||||
        kwargs["flow"] = flow
 | 
			
		||||
        if (
 | 
			
		||||
            not self.request.user.is_authenticated
 | 
			
		||||
            and flow.designation == FlowDesignation.AUTHENTICATION
 | 
			
		||||
        ):
 | 
			
		||||
            self.request.session[SESSION_KEY_AUTH_STARTED] = True
 | 
			
		||||
            self.request.session.save()
 | 
			
		||||
        kwargs["inspector"] = "inspector" in self.request.GET
 | 
			
		||||
        return super().get_context_data(**kwargs)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -356,6 +356,14 @@ def redis_url(db: int) -> str:
 | 
			
		||||
def django_db_config(config: ConfigLoader | None = None) -> dict:
 | 
			
		||||
    if not config:
 | 
			
		||||
        config = CONFIG
 | 
			
		||||
 | 
			
		||||
    pool_options = False
 | 
			
		||||
    use_pool = config.get_bool("postgresql.use_pool", False)
 | 
			
		||||
    if use_pool:
 | 
			
		||||
        pool_options = config.get_dict_from_b64_json("postgresql.pool_options", True)
 | 
			
		||||
        if not pool_options:
 | 
			
		||||
            pool_options = True
 | 
			
		||||
 | 
			
		||||
    db = {
 | 
			
		||||
        "default": {
 | 
			
		||||
            "ENGINE": "authentik.root.db",
 | 
			
		||||
@ -369,6 +377,7 @@ def django_db_config(config: ConfigLoader | None = None) -> dict:
 | 
			
		||||
                "sslrootcert": config.get("postgresql.sslrootcert"),
 | 
			
		||||
                "sslcert": config.get("postgresql.sslcert"),
 | 
			
		||||
                "sslkey": config.get("postgresql.sslkey"),
 | 
			
		||||
                "pool": pool_options,
 | 
			
		||||
            },
 | 
			
		||||
            "CONN_MAX_AGE": config.get_optional_int("postgresql.conn_max_age", 0),
 | 
			
		||||
            "CONN_HEALTH_CHECKS": config.get_bool("postgresql.conn_health_checks", False),
 | 
			
		||||
 | 
			
		||||
@ -21,6 +21,7 @@ postgresql:
 | 
			
		||||
  user: authentik
 | 
			
		||||
  port: 5432
 | 
			
		||||
  password: "env://POSTGRES_PASSWORD"
 | 
			
		||||
  use_pool: False
 | 
			
		||||
  test:
 | 
			
		||||
    name: test_authentik
 | 
			
		||||
  default_schema: public
 | 
			
		||||
 | 
			
		||||
@ -18,7 +18,7 @@ from sentry_sdk import start_span
 | 
			
		||||
from sentry_sdk.tracing import Span
 | 
			
		||||
from structlog.stdlib import get_logger
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import AuthenticatedSession, User
 | 
			
		||||
from authentik.core.models import User
 | 
			
		||||
from authentik.events.models import Event
 | 
			
		||||
from authentik.lib.expression.exceptions import ControlFlowException
 | 
			
		||||
from authentik.lib.utils.http import get_http_session
 | 
			
		||||
@ -203,9 +203,7 @@ class BaseEvaluator:
 | 
			
		||||
            provider = OAuth2Provider.objects.get(name=provider)
 | 
			
		||||
        session = None
 | 
			
		||||
        if hasattr(request, "session") and request.session.session_key:
 | 
			
		||||
            session = AuthenticatedSession.objects.filter(
 | 
			
		||||
                session_key=request.session.session_key
 | 
			
		||||
            ).first()
 | 
			
		||||
            session = request.session["authenticatedsession"]
 | 
			
		||||
        access_token = AccessToken(
 | 
			
		||||
            provider=provider,
 | 
			
		||||
            user=user,
 | 
			
		||||
 | 
			
		||||
@ -18,6 +18,15 @@ class SerializerModel(models.Model):
 | 
			
		||||
    @property
 | 
			
		||||
    def serializer(self) -> type[BaseSerializer]:
 | 
			
		||||
        """Get serializer for this model"""
 | 
			
		||||
        # Special handling for built-in source
 | 
			
		||||
        if (
 | 
			
		||||
            hasattr(self, "managed")
 | 
			
		||||
            and hasattr(self, "MANAGED_INBUILT")
 | 
			
		||||
            and self.managed == self.MANAGED_INBUILT
 | 
			
		||||
        ):
 | 
			
		||||
            from authentik.core.api.sources import SourceSerializer
 | 
			
		||||
 | 
			
		||||
            return SourceSerializer
 | 
			
		||||
        raise NotImplementedError
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -217,6 +217,7 @@ class TestConfig(TestCase):
 | 
			
		||||
                    "HOST": "foo",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": False,
 | 
			
		||||
                        "sslcert": "foo",
 | 
			
		||||
                        "sslkey": "foo",
 | 
			
		||||
                        "sslmode": "foo",
 | 
			
		||||
@ -267,6 +268,7 @@ class TestConfig(TestCase):
 | 
			
		||||
                    "HOST": "foo",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": False,
 | 
			
		||||
                        "sslcert": "foo",
 | 
			
		||||
                        "sslkey": "foo",
 | 
			
		||||
                        "sslmode": "foo",
 | 
			
		||||
@ -285,6 +287,7 @@ class TestConfig(TestCase):
 | 
			
		||||
                    "HOST": "bar",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": False,
 | 
			
		||||
                        "sslcert": "foo",
 | 
			
		||||
                        "sslkey": "foo",
 | 
			
		||||
                        "sslmode": "foo",
 | 
			
		||||
@ -333,6 +336,7 @@ class TestConfig(TestCase):
 | 
			
		||||
                    "HOST": "foo",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": False,
 | 
			
		||||
                        "sslcert": "foo",
 | 
			
		||||
                        "sslkey": "foo",
 | 
			
		||||
                        "sslmode": "foo",
 | 
			
		||||
@ -351,6 +355,7 @@ class TestConfig(TestCase):
 | 
			
		||||
                    "HOST": "bar",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": False,
 | 
			
		||||
                        "sslcert": "foo",
 | 
			
		||||
                        "sslkey": "foo",
 | 
			
		||||
                        "sslmode": "foo",
 | 
			
		||||
@ -394,6 +399,7 @@ class TestConfig(TestCase):
 | 
			
		||||
                    "HOST": "foo",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": False,
 | 
			
		||||
                        "sslcert": "foo",
 | 
			
		||||
                        "sslkey": "foo",
 | 
			
		||||
                        "sslmode": "foo",
 | 
			
		||||
@ -412,6 +418,7 @@ class TestConfig(TestCase):
 | 
			
		||||
                    "HOST": "bar",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": False,
 | 
			
		||||
                        "sslcert": "foo",
 | 
			
		||||
                        "sslkey": "foo",
 | 
			
		||||
                        "sslmode": "foo",
 | 
			
		||||
@ -451,6 +458,7 @@ class TestConfig(TestCase):
 | 
			
		||||
                    "HOST": "foo",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": False,
 | 
			
		||||
                        "sslcert": "foo",
 | 
			
		||||
                        "sslkey": "foo",
 | 
			
		||||
                        "sslmode": "foo",
 | 
			
		||||
@ -469,6 +477,7 @@ class TestConfig(TestCase):
 | 
			
		||||
                    "HOST": "bar",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": False,
 | 
			
		||||
                        "sslcert": "bar",
 | 
			
		||||
                        "sslkey": "foo",
 | 
			
		||||
                        "sslmode": "foo",
 | 
			
		||||
@ -484,3 +493,87 @@ class TestConfig(TestCase):
 | 
			
		||||
                },
 | 
			
		||||
            },
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    def test_db_pool(self):
 | 
			
		||||
        """Test DB Config with pool"""
 | 
			
		||||
        config = ConfigLoader()
 | 
			
		||||
        config.set("postgresql.host", "foo")
 | 
			
		||||
        config.set("postgresql.name", "foo")
 | 
			
		||||
        config.set("postgresql.user", "foo")
 | 
			
		||||
        config.set("postgresql.password", "foo")
 | 
			
		||||
        config.set("postgresql.port", "foo")
 | 
			
		||||
        config.set("postgresql.test.name", "foo")
 | 
			
		||||
        config.set("postgresql.use_pool", True)
 | 
			
		||||
        conf = django_db_config(config)
 | 
			
		||||
        self.assertEqual(
 | 
			
		||||
            conf,
 | 
			
		||||
            {
 | 
			
		||||
                "default": {
 | 
			
		||||
                    "ENGINE": "authentik.root.db",
 | 
			
		||||
                    "HOST": "foo",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": True,
 | 
			
		||||
                        "sslcert": None,
 | 
			
		||||
                        "sslkey": None,
 | 
			
		||||
                        "sslmode": None,
 | 
			
		||||
                        "sslrootcert": None,
 | 
			
		||||
                    },
 | 
			
		||||
                    "PASSWORD": "foo",
 | 
			
		||||
                    "PORT": "foo",
 | 
			
		||||
                    "TEST": {"NAME": "foo"},
 | 
			
		||||
                    "USER": "foo",
 | 
			
		||||
                    "CONN_MAX_AGE": 0,
 | 
			
		||||
                    "CONN_HEALTH_CHECKS": False,
 | 
			
		||||
                    "DISABLE_SERVER_SIDE_CURSORS": False,
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    def test_db_pool_options(self):
 | 
			
		||||
        """Test DB Config with pool"""
 | 
			
		||||
        config = ConfigLoader()
 | 
			
		||||
        config.set("postgresql.host", "foo")
 | 
			
		||||
        config.set("postgresql.name", "foo")
 | 
			
		||||
        config.set("postgresql.user", "foo")
 | 
			
		||||
        config.set("postgresql.password", "foo")
 | 
			
		||||
        config.set("postgresql.port", "foo")
 | 
			
		||||
        config.set("postgresql.test.name", "foo")
 | 
			
		||||
        config.set("postgresql.use_pool", True)
 | 
			
		||||
        config.set(
 | 
			
		||||
            "postgresql.pool_options",
 | 
			
		||||
            base64.b64encode(
 | 
			
		||||
                dumps(
 | 
			
		||||
                    {
 | 
			
		||||
                        "max_size": 15,
 | 
			
		||||
                    }
 | 
			
		||||
                ).encode()
 | 
			
		||||
            ).decode(),
 | 
			
		||||
        )
 | 
			
		||||
        conf = django_db_config(config)
 | 
			
		||||
        self.assertEqual(
 | 
			
		||||
            conf,
 | 
			
		||||
            {
 | 
			
		||||
                "default": {
 | 
			
		||||
                    "ENGINE": "authentik.root.db",
 | 
			
		||||
                    "HOST": "foo",
 | 
			
		||||
                    "NAME": "foo",
 | 
			
		||||
                    "OPTIONS": {
 | 
			
		||||
                        "pool": {
 | 
			
		||||
                            "max_size": 15,
 | 
			
		||||
                        },
 | 
			
		||||
                        "sslcert": None,
 | 
			
		||||
                        "sslkey": None,
 | 
			
		||||
                        "sslmode": None,
 | 
			
		||||
                        "sslrootcert": None,
 | 
			
		||||
                    },
 | 
			
		||||
                    "PASSWORD": "foo",
 | 
			
		||||
                    "PORT": "foo",
 | 
			
		||||
                    "TEST": {"NAME": "foo"},
 | 
			
		||||
                    "USER": "foo",
 | 
			
		||||
                    "CONN_MAX_AGE": 0,
 | 
			
		||||
                    "CONN_HEALTH_CHECKS": False,
 | 
			
		||||
                    "DISABLE_SERVER_SIDE_CURSORS": False,
 | 
			
		||||
                }
 | 
			
		||||
            },
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
@ -35,3 +35,4 @@ class AuthentikPoliciesConfig(ManagedAppConfig):
 | 
			
		||||
    label = "authentik_policies"
 | 
			
		||||
    verbose_name = "authentik Policies"
 | 
			
		||||
    default = True
 | 
			
		||||
    mountpoint = "policy/"
 | 
			
		||||
 | 
			
		||||
@ -66,7 +66,9 @@ class GeoIPPolicy(Policy):
 | 
			
		||||
        if not static_results and not dynamic_results:
 | 
			
		||||
            return PolicyResult(True)
 | 
			
		||||
 | 
			
		||||
        passing = any(r.passing for r in static_results) and all(r.passing for r in dynamic_results)
 | 
			
		||||
        static_passing = any(r.passing for r in static_results) if static_results else True
 | 
			
		||||
        dynamic_passing = all(r.passing for r in dynamic_results)
 | 
			
		||||
        passing = static_passing and dynamic_passing
 | 
			
		||||
        messages = chain(
 | 
			
		||||
            *[r.messages for r in static_results], *[r.messages for r in dynamic_results]
 | 
			
		||||
        )
 | 
			
		||||
@ -113,13 +115,19 @@ class GeoIPPolicy(Policy):
 | 
			
		||||
        to previous authentication requests"""
 | 
			
		||||
        # Get previous login event and GeoIP data
 | 
			
		||||
        previous_logins = Event.objects.filter(
 | 
			
		||||
            action=EventAction.LOGIN, user__pk=request.user.pk, context__geo__isnull=False
 | 
			
		||||
            action=EventAction.LOGIN,
 | 
			
		||||
            user__pk=request.user.pk,  # context__geo__isnull=False
 | 
			
		||||
        ).order_by("-created")[: self.history_login_count]
 | 
			
		||||
        _now = now()
 | 
			
		||||
        geoip_data: GeoIPDict | None = request.context.get("geoip")
 | 
			
		||||
        if not geoip_data:
 | 
			
		||||
            return PolicyResult(False)
 | 
			
		||||
        if not previous_logins.exists():
 | 
			
		||||
            return PolicyResult(True)
 | 
			
		||||
        result = False
 | 
			
		||||
        for previous_login in previous_logins:
 | 
			
		||||
            if "geo" not in previous_login.context:
 | 
			
		||||
                continue
 | 
			
		||||
            previous_login_geoip: GeoIPDict = previous_login.context["geo"]
 | 
			
		||||
 | 
			
		||||
            # Figure out distance
 | 
			
		||||
@ -142,7 +150,8 @@ class GeoIPPolicy(Policy):
 | 
			
		||||
                (MAX_DISTANCE_HOUR_KM * rel_time_hours) + self.distance_tolerance_km
 | 
			
		||||
            ):
 | 
			
		||||
                return PolicyResult(False, _("Distance is further than possible."))
 | 
			
		||||
        return PolicyResult(True)
 | 
			
		||||
            result = True
 | 
			
		||||
        return PolicyResult(result)
 | 
			
		||||
 | 
			
		||||
    class Meta(Policy.PolicyMeta):
 | 
			
		||||
        verbose_name = _("GeoIP Policy")
 | 
			
		||||
 | 
			
		||||
@ -163,7 +163,7 @@ class TestGeoIPPolicy(TestCase):
 | 
			
		||||
        result: PolicyResult = policy.passes(self.request)
 | 
			
		||||
        self.assertFalse(result.passing)
 | 
			
		||||
 | 
			
		||||
    def test_history_impossible_travel(self):
 | 
			
		||||
    def test_history_impossible_travel_failing(self):
 | 
			
		||||
        """Test history checks"""
 | 
			
		||||
        Event.objects.create(
 | 
			
		||||
            action=EventAction.LOGIN,
 | 
			
		||||
@ -181,6 +181,24 @@ class TestGeoIPPolicy(TestCase):
 | 
			
		||||
        result: PolicyResult = policy.passes(self.request)
 | 
			
		||||
        self.assertFalse(result.passing)
 | 
			
		||||
 | 
			
		||||
    def test_history_impossible_travel_passing(self):
 | 
			
		||||
        """Test history checks"""
 | 
			
		||||
        Event.objects.create(
 | 
			
		||||
            action=EventAction.LOGIN,
 | 
			
		||||
            user=get_user(self.user),
 | 
			
		||||
            context={
 | 
			
		||||
                # Random location in Canada
 | 
			
		||||
                "geo": {"lat": 55.868351, "long": -104.441011},
 | 
			
		||||
            },
 | 
			
		||||
        )
 | 
			
		||||
        # Same location
 | 
			
		||||
        self.request.context["geoip"] = {"lat": 55.868351, "long": -104.441011}
 | 
			
		||||
 | 
			
		||||
        policy = GeoIPPolicy.objects.create(check_impossible_travel=True)
 | 
			
		||||
 | 
			
		||||
        result: PolicyResult = policy.passes(self.request)
 | 
			
		||||
        self.assertTrue(result.passing)
 | 
			
		||||
 | 
			
		||||
    def test_history_no_geoip(self):
 | 
			
		||||
        """Test history checks (previous login with no geoip data)"""
 | 
			
		||||
        Event.objects.create(
 | 
			
		||||
@ -195,3 +213,18 @@ class TestGeoIPPolicy(TestCase):
 | 
			
		||||
 | 
			
		||||
        result: PolicyResult = policy.passes(self.request)
 | 
			
		||||
        self.assertFalse(result.passing)
 | 
			
		||||
 | 
			
		||||
    def test_impossible_travel_no_geoip(self):
 | 
			
		||||
        """Test impossible travel checks (previous login with no geoip data)"""
 | 
			
		||||
        Event.objects.create(
 | 
			
		||||
            action=EventAction.LOGIN,
 | 
			
		||||
            user=get_user(self.user),
 | 
			
		||||
            context={},
 | 
			
		||||
        )
 | 
			
		||||
        # Random location in Poland
 | 
			
		||||
        self.request.context["geoip"] = {"lat": 50.950613, "long": 20.363679}
 | 
			
		||||
 | 
			
		||||
        policy = GeoIPPolicy.objects.create(check_impossible_travel=True)
 | 
			
		||||
 | 
			
		||||
        result: PolicyResult = policy.passes(self.request)
 | 
			
		||||
        self.assertFalse(result.passing)
 | 
			
		||||
 | 
			
		||||
@ -2,7 +2,6 @@
 | 
			
		||||
 | 
			
		||||
from django.contrib.auth.signals import user_logged_in
 | 
			
		||||
from django.db import transaction
 | 
			
		||||
from django.db.models import F
 | 
			
		||||
from django.dispatch import receiver
 | 
			
		||||
from django.http import HttpRequest
 | 
			
		||||
from structlog.stdlib import get_logger
 | 
			
		||||
@ -13,20 +12,29 @@ from authentik.events.context_processors.geoip import GEOIP_CONTEXT_PROCESSOR
 | 
			
		||||
from authentik.policies.reputation.models import Reputation, reputation_expiry
 | 
			
		||||
from authentik.root.middleware import ClientIPMiddleware
 | 
			
		||||
from authentik.stages.identification.signals import identification_failed
 | 
			
		||||
from authentik.tenants.utils import get_current_tenant
 | 
			
		||||
 | 
			
		||||
LOGGER = get_logger()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def clamp(value, min, max):
 | 
			
		||||
    return sorted([min, value, max])[1]
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def update_score(request: HttpRequest, identifier: str, amount: int):
 | 
			
		||||
    """Update score for IP and User"""
 | 
			
		||||
    remote_ip = ClientIPMiddleware.get_client_ip(request)
 | 
			
		||||
    tenant = get_current_tenant()
 | 
			
		||||
    new_score = clamp(amount, tenant.reputation_lower_limit, tenant.reputation_upper_limit)
 | 
			
		||||
 | 
			
		||||
    with transaction.atomic():
 | 
			
		||||
        reputation, created = Reputation.objects.select_for_update().get_or_create(
 | 
			
		||||
            ip=remote_ip,
 | 
			
		||||
            identifier=identifier,
 | 
			
		||||
            defaults={
 | 
			
		||||
                "score": amount,
 | 
			
		||||
                "score": clamp(
 | 
			
		||||
                    amount, tenant.reputation_lower_limit, tenant.reputation_upper_limit
 | 
			
		||||
                ),
 | 
			
		||||
                "ip_geo_data": GEOIP_CONTEXT_PROCESSOR.city_dict(remote_ip) or {},
 | 
			
		||||
                "ip_asn_data": ASN_CONTEXT_PROCESSOR.asn_dict(remote_ip) or {},
 | 
			
		||||
                "expires": reputation_expiry(),
 | 
			
		||||
@ -34,9 +42,15 @@ def update_score(request: HttpRequest, identifier: str, amount: int):
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        if not created:
 | 
			
		||||
            reputation.score = F("score") + amount
 | 
			
		||||
            new_score = clamp(
 | 
			
		||||
                reputation.score + amount,
 | 
			
		||||
                tenant.reputation_lower_limit,
 | 
			
		||||
                tenant.reputation_upper_limit,
 | 
			
		||||
            )
 | 
			
		||||
            reputation.score = new_score
 | 
			
		||||
            reputation.save()
 | 
			
		||||
    LOGGER.info("Updated score", amount=amount, for_user=identifier, for_ip=remote_ip)
 | 
			
		||||
 | 
			
		||||
    LOGGER.info("Updated score", amount=new_score, for_user=identifier, for_ip=remote_ip)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@receiver(login_failed)
 | 
			
		||||
 | 
			
		||||
@ -6,9 +6,11 @@ from authentik.core.models import User
 | 
			
		||||
from authentik.lib.generators import generate_id
 | 
			
		||||
from authentik.policies.reputation.api import ReputationPolicySerializer
 | 
			
		||||
from authentik.policies.reputation.models import Reputation, ReputationPolicy
 | 
			
		||||
from authentik.policies.reputation.signals import update_score
 | 
			
		||||
from authentik.policies.types import PolicyRequest
 | 
			
		||||
from authentik.stages.password import BACKEND_INBUILT
 | 
			
		||||
from authentik.stages.password.stage import authenticate
 | 
			
		||||
from authentik.tenants.models import DEFAULT_REPUTATION_LOWER_LIMIT, DEFAULT_REPUTATION_UPPER_LIMIT
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class TestReputationPolicy(TestCase):
 | 
			
		||||
@ -17,36 +19,48 @@ class TestReputationPolicy(TestCase):
 | 
			
		||||
    def setUp(self):
 | 
			
		||||
        self.request_factory = RequestFactory()
 | 
			
		||||
        self.request = self.request_factory.get("/")
 | 
			
		||||
        self.test_ip = "127.0.0.1"
 | 
			
		||||
        self.test_username = "test"
 | 
			
		||||
        self.ip = "127.0.0.1"
 | 
			
		||||
        self.username = "username"
 | 
			
		||||
        self.password = generate_id()
 | 
			
		||||
        # We need a user for the one-to-one in userreputation
 | 
			
		||||
        self.user = User.objects.create(username=self.test_username)
 | 
			
		||||
        self.user = User.objects.create(username=self.username)
 | 
			
		||||
        self.user.set_password(self.password)
 | 
			
		||||
        self.backends = [BACKEND_INBUILT]
 | 
			
		||||
 | 
			
		||||
    def test_ip_reputation(self):
 | 
			
		||||
        """test IP reputation"""
 | 
			
		||||
        # Trigger negative reputation
 | 
			
		||||
        authenticate(
 | 
			
		||||
            self.request, self.backends, username=self.test_username, password=self.test_username
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(Reputation.objects.get(ip=self.test_ip).score, -1)
 | 
			
		||||
        authenticate(self.request, self.backends, username=self.username, password=self.username)
 | 
			
		||||
        self.assertEqual(Reputation.objects.get(ip=self.ip).score, -1)
 | 
			
		||||
 | 
			
		||||
    def test_user_reputation(self):
 | 
			
		||||
        """test User reputation"""
 | 
			
		||||
        # Trigger negative reputation
 | 
			
		||||
        authenticate(
 | 
			
		||||
            self.request, self.backends, username=self.test_username, password=self.test_username
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(Reputation.objects.get(identifier=self.test_username).score, -1)
 | 
			
		||||
        authenticate(self.request, self.backends, username=self.username, password=self.username)
 | 
			
		||||
        self.assertEqual(Reputation.objects.get(identifier=self.username).score, -1)
 | 
			
		||||
 | 
			
		||||
    def test_update_reputation(self):
 | 
			
		||||
        """test reputation update"""
 | 
			
		||||
        Reputation.objects.create(identifier=self.test_username, ip=self.test_ip, score=43)
 | 
			
		||||
        Reputation.objects.create(identifier=self.username, ip=self.ip, score=4)
 | 
			
		||||
        # Trigger negative reputation
 | 
			
		||||
        authenticate(
 | 
			
		||||
            self.request, self.backends, username=self.test_username, password=self.test_username
 | 
			
		||||
        authenticate(self.request, self.backends, username=self.username, password=self.username)
 | 
			
		||||
        self.assertEqual(Reputation.objects.get(identifier=self.username).score, 3)
 | 
			
		||||
 | 
			
		||||
    def test_reputation_lower_limit(self):
 | 
			
		||||
        """test reputation lower limit"""
 | 
			
		||||
        Reputation.objects.create(identifier=self.username, ip=self.ip)
 | 
			
		||||
        update_score(self.request, identifier=self.username, amount=-1000)
 | 
			
		||||
        self.assertEqual(
 | 
			
		||||
            Reputation.objects.get(identifier=self.username).score, DEFAULT_REPUTATION_LOWER_LIMIT
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    def test_reputation_upper_limit(self):
 | 
			
		||||
        """test reputation upper limit"""
 | 
			
		||||
        Reputation.objects.create(identifier=self.username, ip=self.ip)
 | 
			
		||||
        update_score(self.request, identifier=self.username, amount=1000)
 | 
			
		||||
        self.assertEqual(
 | 
			
		||||
            Reputation.objects.get(identifier=self.username).score, DEFAULT_REPUTATION_UPPER_LIMIT
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(Reputation.objects.get(identifier=self.test_username).score, 42)
 | 
			
		||||
 | 
			
		||||
    def test_policy(self):
 | 
			
		||||
        """Test Policy"""
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										110
									
								
								authentik/policies/templates/policies/buffer.html
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										110
									
								
								authentik/policies/templates/policies/buffer.html
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,110 @@
 | 
			
		||||
{% extends 'login/base_full.html' %}
 | 
			
		||||
 | 
			
		||||
{% load static %}
 | 
			
		||||
{% load i18n %}
 | 
			
		||||
 | 
			
		||||
{% block head %}
 | 
			
		||||
{{ block.super }}
 | 
			
		||||
 | 
			
		||||
<script>
 | 
			
		||||
  let redirecting = false;
 | 
			
		||||
 | 
			
		||||
  const checkAuth = async () => {
 | 
			
		||||
    if (redirecting) return true;
 | 
			
		||||
 | 
			
		||||
    const url = "{{ check_auth_url }}";
 | 
			
		||||
    console.debug("authentik/policies/buffer: Checking authentication...");
 | 
			
		||||
 | 
			
		||||
    try {
 | 
			
		||||
      const result = await fetch(url, {
 | 
			
		||||
        method: "HEAD",
 | 
			
		||||
      });
 | 
			
		||||
 | 
			
		||||
      if (result.status >= 400) {
 | 
			
		||||
        return false
 | 
			
		||||
      }
 | 
			
		||||
 | 
			
		||||
      console.debug("authentik/policies/buffer: Continuing");
 | 
			
		||||
      redirecting = true;
 | 
			
		||||
 | 
			
		||||
      if ("{{ auth_req_method }}" === "post") {
 | 
			
		||||
        document.querySelector("form").submit();
 | 
			
		||||
      } else {
 | 
			
		||||
        window.location.assign("{{ continue_url|escapejs }}");
 | 
			
		||||
      }
 | 
			
		||||
    } catch {
 | 
			
		||||
      return false;
 | 
			
		||||
    }
 | 
			
		||||
  };
 | 
			
		||||
 | 
			
		||||
  let timeout = 100;
 | 
			
		||||
  let offset = 20;
 | 
			
		||||
  let attempt = 0;
 | 
			
		||||
 | 
			
		||||
  const main = async () => {
 | 
			
		||||
    attempt += 1;
 | 
			
		||||
 | 
			
		||||
    await checkAuth();
 | 
			
		||||
 | 
			
		||||
    console.debug(`authentik/policies/buffer: Waiting ${timeout}ms...`);
 | 
			
		||||
 | 
			
		||||
    setTimeout(main, timeout);
 | 
			
		||||
 | 
			
		||||
    timeout += (offset * attempt);
 | 
			
		||||
 | 
			
		||||
    if (timeout >= 2000) {
 | 
			
		||||
      timeout = 2000;
 | 
			
		||||
    }
 | 
			
		||||
  }
 | 
			
		||||
 | 
			
		||||
  document.addEventListener("visibilitychange", async () => {
 | 
			
		||||
    if (document.hidden) return;
 | 
			
		||||
 | 
			
		||||
    console.debug("authentik/policies/buffer: Checking authentication on tab activate...");
 | 
			
		||||
 | 
			
		||||
    await checkAuth();
 | 
			
		||||
  });
 | 
			
		||||
 | 
			
		||||
  main();
 | 
			
		||||
</script>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block title %}
 | 
			
		||||
  {% trans 'Waiting for authentication...' %} - {{ brand.branding_title }}
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block card_title %}
 | 
			
		||||
  {% trans 'Waiting for authentication...' %}
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block card %}
 | 
			
		||||
  <form class="pf-c-form" method="{{ auth_req_method }}" action="{{ continue_url }}">
 | 
			
		||||
    {% if auth_req_method == "post" %}
 | 
			
		||||
      {% for key, value in auth_req_body.items %}
 | 
			
		||||
        <input type="hidden" name="{{ key }}" value="{{ value }}">
 | 
			
		||||
      {% endfor %}
 | 
			
		||||
    {% endif %}
 | 
			
		||||
 | 
			
		||||
    <div class="pf-c-empty-state">
 | 
			
		||||
      <div class="pf-c-empty-state__content">
 | 
			
		||||
        <div class="pf-c-empty-state__icon">
 | 
			
		||||
          <span class="pf-c-spinner pf-m-xl" role="progressbar">
 | 
			
		||||
            <span class="pf-c-spinner__clipper"></span>
 | 
			
		||||
            <span class="pf-c-spinner__lead-ball"></span>
 | 
			
		||||
            <span class="pf-c-spinner__tail-ball"></span>
 | 
			
		||||
          </span>
 | 
			
		||||
        </div>
 | 
			
		||||
 | 
			
		||||
        <h1 class="pf-c-title pf-m-lg">
 | 
			
		||||
          {% trans "You're already authenticating in another tab. This page will refresh once authentication is completed." %}
 | 
			
		||||
        </h1>
 | 
			
		||||
      </div>
 | 
			
		||||
    </div>
 | 
			
		||||
 | 
			
		||||
    <div class="pf-c-form__group pf-m-action">
 | 
			
		||||
      <a href="{{ auth_req_url }}" class="pf-c-button pf-m-primary pf-m-block">
 | 
			
		||||
        {% trans "Authenticate in this tab" %}
 | 
			
		||||
      </a>
 | 
			
		||||
    </div>
 | 
			
		||||
  </form>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
@ -13,63 +13,69 @@
 | 
			
		||||
 | 
			
		||||
{% block card %}
 | 
			
		||||
<form class="pf-c-form">
 | 
			
		||||
    {% csrf_token %}
 | 
			
		||||
    {% if user.is_authenticated %}
 | 
			
		||||
        <div class="pf-c-form__group">
 | 
			
		||||
            <div class="form-control-static">
 | 
			
		||||
                <div class="avatar">
 | 
			
		||||
                    <img class="pf-c-avatar" src="{{ user.avatar }}" alt="{% trans "User's avatar" %}" />
 | 
			
		||||
                    {{ user.username }}
 | 
			
		||||
                </div>
 | 
			
		||||
                <div slot="link">
 | 
			
		||||
                    <a href="{{ cancel }}">{% trans "Not you?" %}</a>
 | 
			
		||||
                </div>
 | 
			
		||||
            </div>
 | 
			
		||||
        </div>
 | 
			
		||||
    {% endif %}
 | 
			
		||||
  {% csrf_token %}
 | 
			
		||||
 | 
			
		||||
  {% if user.is_authenticated %}
 | 
			
		||||
    <div class="pf-c-form__group">
 | 
			
		||||
        <p>
 | 
			
		||||
            <i class="pf-icon pf-icon-error-circle-o"></i>
 | 
			
		||||
            {% trans 'Request has been denied.' %}
 | 
			
		||||
        </p>
 | 
			
		||||
        {% if error %}
 | 
			
		||||
        <hr>
 | 
			
		||||
        <p>
 | 
			
		||||
            {{ error }}
 | 
			
		||||
        </p>
 | 
			
		||||
        {% endif %}
 | 
			
		||||
        {% if policy_result %}
 | 
			
		||||
            <hr>
 | 
			
		||||
            {% if policy_result.messages %}
 | 
			
		||||
            <em>{% trans 'Messages:' %}</em>
 | 
			
		||||
            <ul class="pf-c-list">
 | 
			
		||||
                {% for message in policy_result.messages %}
 | 
			
		||||
                <li>
 | 
			
		||||
                    {{ message }}
 | 
			
		||||
                </li>
 | 
			
		||||
                {% endfor %}
 | 
			
		||||
            </ul>
 | 
			
		||||
            {% endif %}
 | 
			
		||||
            {% if policy_result.source_results %}
 | 
			
		||||
            <em>{% trans 'Explanation:' %}</em>
 | 
			
		||||
            <ul class="pf-c-list">
 | 
			
		||||
                {% for source_result in policy_result.source_results %}
 | 
			
		||||
                <li>
 | 
			
		||||
                    {% blocktrans with name=source_result.source_binding result=source_result.passing %}
 | 
			
		||||
                    Policy binding '{{ name }}' returned result '{{ result }}'
 | 
			
		||||
                    {% endblocktrans %}
 | 
			
		||||
                    {% if source_result.messages %}
 | 
			
		||||
                    <ul class="pf-c-list">
 | 
			
		||||
                        {% for message in source_result.messages %}
 | 
			
		||||
                            <li>{{ message }}</li>
 | 
			
		||||
                        {% endfor %}
 | 
			
		||||
                    </ul>
 | 
			
		||||
                    {% endif %}
 | 
			
		||||
                </li>
 | 
			
		||||
                {% endfor %}
 | 
			
		||||
            </ul>
 | 
			
		||||
            {% endif %}
 | 
			
		||||
        {% endif %}
 | 
			
		||||
      <div class="form-control-static">
 | 
			
		||||
        <div class="avatar">
 | 
			
		||||
          <img class="pf-c-avatar" src="{{ user.avatar }}" alt="{% trans "User's avatar" %}">
 | 
			
		||||
          {{ user.username }}
 | 
			
		||||
        </div>
 | 
			
		||||
        <div slot="link">
 | 
			
		||||
          <a href="{{ cancel }}">{% trans "Not you?" %}</a>
 | 
			
		||||
        </div>
 | 
			
		||||
      </div>
 | 
			
		||||
    </div>
 | 
			
		||||
  {% endif %}
 | 
			
		||||
 | 
			
		||||
  <div class="pf-c-form__group">
 | 
			
		||||
    <p>
 | 
			
		||||
      <i class="pf-icon pf-icon-error-circle-o"></i>
 | 
			
		||||
      {% trans 'Request has been denied.' %}
 | 
			
		||||
    </p>
 | 
			
		||||
 | 
			
		||||
    {% if error %}
 | 
			
		||||
      <hr>
 | 
			
		||||
 | 
			
		||||
      <p>{{ error }}</p>
 | 
			
		||||
    {% endif %}
 | 
			
		||||
 | 
			
		||||
    {% if policy_result %}
 | 
			
		||||
      <hr>
 | 
			
		||||
 | 
			
		||||
      {% if policy_result.messages %}
 | 
			
		||||
        <em>{% trans 'Messages:' %}</em>
 | 
			
		||||
 | 
			
		||||
        <ul class="pf-c-list">
 | 
			
		||||
          {% for message in policy_result.messages %}
 | 
			
		||||
            <li>{{ message }}</li>
 | 
			
		||||
          {% endfor %}
 | 
			
		||||
        </ul>
 | 
			
		||||
      {% endif %}
 | 
			
		||||
 | 
			
		||||
      {% if policy_result.source_results %}
 | 
			
		||||
        <em>{% trans 'Explanation:' %}</em>
 | 
			
		||||
 | 
			
		||||
        <ul class="pf-c-list">
 | 
			
		||||
          {% for source_result in policy_result.source_results %}
 | 
			
		||||
            <li>
 | 
			
		||||
              {% blocktrans with name=source_result.source_binding result=source_result.passing %}
 | 
			
		||||
              Policy binding '{{ name }}' returned result '{{ result }}'
 | 
			
		||||
              {% endblocktrans %}
 | 
			
		||||
 | 
			
		||||
              {% if source_result.messages %}
 | 
			
		||||
                <ul class="pf-c-list">
 | 
			
		||||
                  {% for message in source_result.messages %}
 | 
			
		||||
                    <li>{{ message }}</li>
 | 
			
		||||
                  {% endfor %}
 | 
			
		||||
                </ul>
 | 
			
		||||
              {% endif %}
 | 
			
		||||
            </li>
 | 
			
		||||
          {% endfor %}
 | 
			
		||||
        </ul>
 | 
			
		||||
      {% endif %}
 | 
			
		||||
    {% endif %}
 | 
			
		||||
  </div>
 | 
			
		||||
</form>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										121
									
								
								authentik/policies/tests/test_views.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										121
									
								
								authentik/policies/tests/test_views.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,121 @@
 | 
			
		||||
from django.contrib.auth.models import AnonymousUser
 | 
			
		||||
from django.contrib.sessions.middleware import SessionMiddleware
 | 
			
		||||
from django.http import HttpResponse
 | 
			
		||||
from django.test import RequestFactory, TestCase
 | 
			
		||||
from django.urls import reverse
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import Application, Provider
 | 
			
		||||
from authentik.core.tests.utils import create_test_flow, create_test_user
 | 
			
		||||
from authentik.flows.models import FlowDesignation
 | 
			
		||||
from authentik.flows.planner import FlowPlan
 | 
			
		||||
from authentik.flows.views.executor import SESSION_KEY_PLAN
 | 
			
		||||
from authentik.lib.generators import generate_id
 | 
			
		||||
from authentik.lib.tests.utils import dummy_get_response
 | 
			
		||||
from authentik.policies.views import (
 | 
			
		||||
    QS_BUFFER_ID,
 | 
			
		||||
    SESSION_KEY_BUFFER,
 | 
			
		||||
    BufferedPolicyAccessView,
 | 
			
		||||
    BufferView,
 | 
			
		||||
    PolicyAccessView,
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class TestPolicyViews(TestCase):
 | 
			
		||||
    """Test PolicyAccessView"""
 | 
			
		||||
 | 
			
		||||
    def setUp(self):
 | 
			
		||||
        super().setUp()
 | 
			
		||||
        self.factory = RequestFactory()
 | 
			
		||||
        self.user = create_test_user()
 | 
			
		||||
 | 
			
		||||
    def test_pav(self):
 | 
			
		||||
        """Test simple policy access view"""
 | 
			
		||||
        provider = Provider.objects.create(
 | 
			
		||||
            name=generate_id(),
 | 
			
		||||
        )
 | 
			
		||||
        app = Application.objects.create(name=generate_id(), slug=generate_id(), provider=provider)
 | 
			
		||||
 | 
			
		||||
        class TestView(PolicyAccessView):
 | 
			
		||||
            def resolve_provider_application(self):
 | 
			
		||||
                self.provider = provider
 | 
			
		||||
                self.application = app
 | 
			
		||||
 | 
			
		||||
            def get(self, *args, **kwargs):
 | 
			
		||||
                return HttpResponse("foo")
 | 
			
		||||
 | 
			
		||||
        req = self.factory.get("/")
 | 
			
		||||
        req.user = self.user
 | 
			
		||||
        res = TestView.as_view()(req)
 | 
			
		||||
        self.assertEqual(res.status_code, 200)
 | 
			
		||||
        self.assertEqual(res.content, b"foo")
 | 
			
		||||
 | 
			
		||||
    def test_pav_buffer(self):
 | 
			
		||||
        """Test simple policy access view"""
 | 
			
		||||
        provider = Provider.objects.create(
 | 
			
		||||
            name=generate_id(),
 | 
			
		||||
        )
 | 
			
		||||
        app = Application.objects.create(name=generate_id(), slug=generate_id(), provider=provider)
 | 
			
		||||
        flow = create_test_flow(FlowDesignation.AUTHENTICATION)
 | 
			
		||||
 | 
			
		||||
        class TestView(BufferedPolicyAccessView):
 | 
			
		||||
            def resolve_provider_application(self):
 | 
			
		||||
                self.provider = provider
 | 
			
		||||
                self.application = app
 | 
			
		||||
 | 
			
		||||
            def get(self, *args, **kwargs):
 | 
			
		||||
                return HttpResponse("foo")
 | 
			
		||||
 | 
			
		||||
        req = self.factory.get("/")
 | 
			
		||||
        req.user = AnonymousUser()
 | 
			
		||||
        middleware = SessionMiddleware(dummy_get_response)
 | 
			
		||||
        middleware.process_request(req)
 | 
			
		||||
        req.session[SESSION_KEY_PLAN] = FlowPlan(flow.pk)
 | 
			
		||||
        req.session.save()
 | 
			
		||||
        res = TestView.as_view()(req)
 | 
			
		||||
        self.assertEqual(res.status_code, 302)
 | 
			
		||||
        self.assertTrue(res.url.startswith(reverse("authentik_policies:buffer")))
 | 
			
		||||
 | 
			
		||||
    def test_pav_buffer_skip(self):
 | 
			
		||||
        """Test simple policy access view (skip buffer)"""
 | 
			
		||||
        provider = Provider.objects.create(
 | 
			
		||||
            name=generate_id(),
 | 
			
		||||
        )
 | 
			
		||||
        app = Application.objects.create(name=generate_id(), slug=generate_id(), provider=provider)
 | 
			
		||||
        flow = create_test_flow(FlowDesignation.AUTHENTICATION)
 | 
			
		||||
 | 
			
		||||
        class TestView(BufferedPolicyAccessView):
 | 
			
		||||
            def resolve_provider_application(self):
 | 
			
		||||
                self.provider = provider
 | 
			
		||||
                self.application = app
 | 
			
		||||
 | 
			
		||||
            def get(self, *args, **kwargs):
 | 
			
		||||
                return HttpResponse("foo")
 | 
			
		||||
 | 
			
		||||
        req = self.factory.get("/?skip_buffer=true")
 | 
			
		||||
        req.user = AnonymousUser()
 | 
			
		||||
        middleware = SessionMiddleware(dummy_get_response)
 | 
			
		||||
        middleware.process_request(req)
 | 
			
		||||
        req.session[SESSION_KEY_PLAN] = FlowPlan(flow.pk)
 | 
			
		||||
        req.session.save()
 | 
			
		||||
        res = TestView.as_view()(req)
 | 
			
		||||
        self.assertEqual(res.status_code, 302)
 | 
			
		||||
        self.assertTrue(res.url.startswith(reverse("authentik_flows:default-authentication")))
 | 
			
		||||
 | 
			
		||||
    def test_buffer(self):
 | 
			
		||||
        """Test buffer view"""
 | 
			
		||||
        uid = generate_id()
 | 
			
		||||
        req = self.factory.get(f"/?{QS_BUFFER_ID}={uid}")
 | 
			
		||||
        req.user = AnonymousUser()
 | 
			
		||||
        middleware = SessionMiddleware(dummy_get_response)
 | 
			
		||||
        middleware.process_request(req)
 | 
			
		||||
        ts = generate_id()
 | 
			
		||||
        req.session[SESSION_KEY_BUFFER % uid] = {
 | 
			
		||||
            "method": "get",
 | 
			
		||||
            "body": {},
 | 
			
		||||
            "url": f"/{ts}",
 | 
			
		||||
        }
 | 
			
		||||
        req.session.save()
 | 
			
		||||
 | 
			
		||||
        res = BufferView.as_view()(req)
 | 
			
		||||
        self.assertEqual(res.status_code, 200)
 | 
			
		||||
        self.assertIn(ts, res.render().content.decode())
 | 
			
		||||
@ -1,7 +1,14 @@
 | 
			
		||||
"""API URLs"""
 | 
			
		||||
 | 
			
		||||
from django.urls import path
 | 
			
		||||
 | 
			
		||||
from authentik.policies.api.bindings import PolicyBindingViewSet
 | 
			
		||||
from authentik.policies.api.policies import PolicyViewSet
 | 
			
		||||
from authentik.policies.views import BufferView
 | 
			
		||||
 | 
			
		||||
urlpatterns = [
 | 
			
		||||
    path("buffer", BufferView.as_view(), name="buffer"),
 | 
			
		||||
]
 | 
			
		||||
 | 
			
		||||
api_urlpatterns = [
 | 
			
		||||
    ("policies/all", PolicyViewSet),
 | 
			
		||||
 | 
			
		||||
@ -1,23 +1,37 @@
 | 
			
		||||
"""authentik access helper classes"""
 | 
			
		||||
 | 
			
		||||
from typing import Any
 | 
			
		||||
from uuid import uuid4
 | 
			
		||||
 | 
			
		||||
from django.contrib import messages
 | 
			
		||||
from django.contrib.auth.mixins import AccessMixin
 | 
			
		||||
from django.contrib.auth.views import redirect_to_login
 | 
			
		||||
from django.http import HttpRequest, HttpResponse
 | 
			
		||||
from django.http import HttpRequest, HttpResponse, QueryDict
 | 
			
		||||
from django.shortcuts import redirect
 | 
			
		||||
from django.urls import reverse
 | 
			
		||||
from django.utils.http import urlencode
 | 
			
		||||
from django.utils.translation import gettext as _
 | 
			
		||||
from django.views.generic.base import View
 | 
			
		||||
from django.views.generic.base import TemplateView, View
 | 
			
		||||
from structlog.stdlib import get_logger
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import Application, Provider, User
 | 
			
		||||
from authentik.flows.views.executor import SESSION_KEY_APPLICATION_PRE, SESSION_KEY_POST
 | 
			
		||||
from authentik.flows.models import Flow, FlowDesignation
 | 
			
		||||
from authentik.flows.planner import FlowPlan
 | 
			
		||||
from authentik.flows.views.executor import (
 | 
			
		||||
    SESSION_KEY_APPLICATION_PRE,
 | 
			
		||||
    SESSION_KEY_AUTH_STARTED,
 | 
			
		||||
    SESSION_KEY_PLAN,
 | 
			
		||||
    SESSION_KEY_POST,
 | 
			
		||||
)
 | 
			
		||||
from authentik.lib.sentry import SentryIgnoredException
 | 
			
		||||
from authentik.policies.denied import AccessDeniedResponse
 | 
			
		||||
from authentik.policies.engine import PolicyEngine
 | 
			
		||||
from authentik.policies.types import PolicyRequest, PolicyResult
 | 
			
		||||
 | 
			
		||||
LOGGER = get_logger()
 | 
			
		||||
QS_BUFFER_ID = "af_bf_id"
 | 
			
		||||
QS_SKIP_BUFFER = "skip_buffer"
 | 
			
		||||
SESSION_KEY_BUFFER = "authentik/policies/pav_buffer/%s"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class RequestValidationError(SentryIgnoredException):
 | 
			
		||||
@ -125,3 +139,65 @@ class PolicyAccessView(AccessMixin, View):
 | 
			
		||||
            for message in result.messages:
 | 
			
		||||
                messages.error(self.request, _(message))
 | 
			
		||||
        return result
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def url_with_qs(url: str, **kwargs):
 | 
			
		||||
    """Update/set querystring of `url` with the parameters in `kwargs`. Original query string
 | 
			
		||||
    parameters are retained"""
 | 
			
		||||
    if "?" not in url:
 | 
			
		||||
        return url + f"?{urlencode(kwargs)}"
 | 
			
		||||
    url, _, qs = url.partition("?")
 | 
			
		||||
    qs = QueryDict(qs, mutable=True)
 | 
			
		||||
    qs.update(kwargs)
 | 
			
		||||
    return url + f"?{urlencode(qs.items())}"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class BufferView(TemplateView):
 | 
			
		||||
    """Buffer view"""
 | 
			
		||||
 | 
			
		||||
    template_name = "policies/buffer.html"
 | 
			
		||||
 | 
			
		||||
    def get_context_data(self, **kwargs):
 | 
			
		||||
        buf_id = self.request.GET.get(QS_BUFFER_ID)
 | 
			
		||||
        buffer: dict = self.request.session.get(SESSION_KEY_BUFFER % buf_id)
 | 
			
		||||
        kwargs["auth_req_method"] = buffer["method"]
 | 
			
		||||
        kwargs["auth_req_body"] = buffer["body"]
 | 
			
		||||
        kwargs["auth_req_url"] = url_with_qs(buffer["url"], **{QS_SKIP_BUFFER: True})
 | 
			
		||||
        kwargs["check_auth_url"] = reverse("authentik_api:user-me")
 | 
			
		||||
        kwargs["continue_url"] = url_with_qs(buffer["url"], **{QS_BUFFER_ID: buf_id})
 | 
			
		||||
        return super().get_context_data(**kwargs)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class BufferedPolicyAccessView(PolicyAccessView):
 | 
			
		||||
    """PolicyAccessView which buffers access requests in case the user is not logged in"""
 | 
			
		||||
 | 
			
		||||
    def handle_no_permission(self):
 | 
			
		||||
        plan: FlowPlan | None = self.request.session.get(SESSION_KEY_PLAN)
 | 
			
		||||
        authenticating = self.request.session.get(SESSION_KEY_AUTH_STARTED)
 | 
			
		||||
        if plan:
 | 
			
		||||
            flow = Flow.objects.filter(pk=plan.flow_pk).first()
 | 
			
		||||
            if not flow or flow.designation != FlowDesignation.AUTHENTICATION:
 | 
			
		||||
                LOGGER.debug("Not buffering request, no flow or flow not for authentication")
 | 
			
		||||
                return super().handle_no_permission()
 | 
			
		||||
        if not plan and authenticating is None:
 | 
			
		||||
            LOGGER.debug("Not buffering request, no flow plan active")
 | 
			
		||||
            return super().handle_no_permission()
 | 
			
		||||
        if self.request.GET.get(QS_SKIP_BUFFER):
 | 
			
		||||
            LOGGER.debug("Not buffering request, explicit skip")
 | 
			
		||||
            return super().handle_no_permission()
 | 
			
		||||
        buffer_id = str(uuid4())
 | 
			
		||||
        LOGGER.debug("Buffering access request", bf_id=buffer_id)
 | 
			
		||||
        self.request.session[SESSION_KEY_BUFFER % buffer_id] = {
 | 
			
		||||
            "body": self.request.POST,
 | 
			
		||||
            "url": self.request.build_absolute_uri(self.request.get_full_path()),
 | 
			
		||||
            "method": self.request.method.lower(),
 | 
			
		||||
        }
 | 
			
		||||
        return redirect(
 | 
			
		||||
            url_with_qs(reverse("authentik_policies:buffer"), **{QS_BUFFER_ID: buffer_id})
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    def dispatch(self, request, *args, **kwargs):
 | 
			
		||||
        response = super().dispatch(request, *args, **kwargs)
 | 
			
		||||
        if QS_BUFFER_ID in self.request.GET:
 | 
			
		||||
            self.request.session.pop(SESSION_KEY_BUFFER % self.request.GET[QS_BUFFER_ID], None)
 | 
			
		||||
        return response
 | 
			
		||||
 | 
			
		||||
@ -126,7 +126,7 @@ class IDToken:
 | 
			
		||||
        id_token.iat = int(now.timestamp())
 | 
			
		||||
        id_token.auth_time = int(token.auth_time.timestamp())
 | 
			
		||||
        if token.session:
 | 
			
		||||
            id_token.sid = hash_session_key(token.session.session_key)
 | 
			
		||||
            id_token.sid = hash_session_key(token.session.session.session_key)
 | 
			
		||||
 | 
			
		||||
        # We use the timestamp of the user's last successful login (EventAction.LOGIN) for auth_time
 | 
			
		||||
        auth_event = get_login_event(token.session)
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										116
									
								
								authentik/providers/oauth2/migrations/0028_migrate_session.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										116
									
								
								authentik/providers/oauth2/migrations/0028_migrate_session.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,116 @@
 | 
			
		||||
# Generated by Django 5.0.11 on 2025-01-27 13:00
 | 
			
		||||
 | 
			
		||||
from django.db import migrations
 | 
			
		||||
import django.db.models.deletion
 | 
			
		||||
from django.db import migrations, models
 | 
			
		||||
from functools import partial
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def migrate_sessions(apps, schema_editor, model):
 | 
			
		||||
    Model = apps.get_model("authentik_providers_oauth2", model)
 | 
			
		||||
    AuthenticatedSession = apps.get_model("authentik_core", "AuthenticatedSession")
 | 
			
		||||
    db_alias = schema_editor.connection.alias
 | 
			
		||||
 | 
			
		||||
    for obj in Model.objects.using(db_alias).all():
 | 
			
		||||
        if not obj.old_session:
 | 
			
		||||
            continue
 | 
			
		||||
        obj.session = (
 | 
			
		||||
            AuthenticatedSession.objects.using(db_alias)
 | 
			
		||||
            .filter(session__session_key=obj.old_session.session_key)
 | 
			
		||||
            .first()
 | 
			
		||||
        )
 | 
			
		||||
        if obj.session:
 | 
			
		||||
            obj.save()
 | 
			
		||||
        else:
 | 
			
		||||
            obj.delete()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("authentik_providers_oauth2", "0027_accesstoken_authentik_p_expires_9f24a5_idx_and_more"),
 | 
			
		||||
        ("authentik_core", "0046_session_and_more"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.RenameField(
 | 
			
		||||
            model_name="accesstoken",
 | 
			
		||||
            old_name="session",
 | 
			
		||||
            new_name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RenameField(
 | 
			
		||||
            model_name="authorizationcode",
 | 
			
		||||
            old_name="session",
 | 
			
		||||
            new_name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RenameField(
 | 
			
		||||
            model_name="devicetoken",
 | 
			
		||||
            old_name="session",
 | 
			
		||||
            new_name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RenameField(
 | 
			
		||||
            model_name="refreshtoken",
 | 
			
		||||
            old_name="session",
 | 
			
		||||
            new_name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddField(
 | 
			
		||||
            model_name="accesstoken",
 | 
			
		||||
            name="session",
 | 
			
		||||
            field=models.ForeignKey(
 | 
			
		||||
                default=None,
 | 
			
		||||
                null=True,
 | 
			
		||||
                on_delete=django.db.models.deletion.CASCADE,
 | 
			
		||||
                to="authentik_core.authenticatedsession",
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddField(
 | 
			
		||||
            model_name="authorizationcode",
 | 
			
		||||
            name="session",
 | 
			
		||||
            field=models.ForeignKey(
 | 
			
		||||
                default=None,
 | 
			
		||||
                null=True,
 | 
			
		||||
                on_delete=django.db.models.deletion.CASCADE,
 | 
			
		||||
                to="authentik_core.authenticatedsession",
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddField(
 | 
			
		||||
            model_name="devicetoken",
 | 
			
		||||
            name="session",
 | 
			
		||||
            field=models.ForeignKey(
 | 
			
		||||
                default=None,
 | 
			
		||||
                null=True,
 | 
			
		||||
                on_delete=django.db.models.deletion.SET_DEFAULT,
 | 
			
		||||
                to="authentik_core.authenticatedsession",
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddField(
 | 
			
		||||
            model_name="refreshtoken",
 | 
			
		||||
            name="session",
 | 
			
		||||
            field=models.ForeignKey(
 | 
			
		||||
                default=None,
 | 
			
		||||
                null=True,
 | 
			
		||||
                on_delete=django.db.models.deletion.SET_DEFAULT,
 | 
			
		||||
                to="authentik_core.authenticatedsession",
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RunPython(code=partial(migrate_sessions, model="AccessToken")),
 | 
			
		||||
        migrations.RunPython(code=partial(migrate_sessions, model="AuthorizationCode")),
 | 
			
		||||
        migrations.RunPython(code=partial(migrate_sessions, model="DeviceToken")),
 | 
			
		||||
        migrations.RunPython(code=partial(migrate_sessions, model="RefreshToken")),
 | 
			
		||||
        migrations.RemoveField(
 | 
			
		||||
            model_name="accesstoken",
 | 
			
		||||
            name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RemoveField(
 | 
			
		||||
            model_name="authorizationcode",
 | 
			
		||||
            name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RemoveField(
 | 
			
		||||
            model_name="devicetoken",
 | 
			
		||||
            name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RemoveField(
 | 
			
		||||
            model_name="refreshtoken",
 | 
			
		||||
            name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -1,18 +1,30 @@
 | 
			
		||||
from django.contrib.auth.signals import user_logged_out
 | 
			
		||||
from django.db.models.signals import post_save
 | 
			
		||||
from django.db.models.signals import post_save, pre_delete
 | 
			
		||||
from django.dispatch import receiver
 | 
			
		||||
from django.http import HttpRequest
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import User
 | 
			
		||||
from authentik.core.models import AuthenticatedSession, User
 | 
			
		||||
from authentik.providers.oauth2.models import AccessToken, DeviceToken, RefreshToken
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@receiver(user_logged_out)
 | 
			
		||||
def user_logged_out_oauth_access_token(sender, request: HttpRequest, user: User, **_):
 | 
			
		||||
    """Revoke access tokens upon user logout"""
 | 
			
		||||
def user_logged_out_oauth_tokens_removal(sender, request: HttpRequest, user: User, **_):
 | 
			
		||||
    """Revoke tokens upon user logout"""
 | 
			
		||||
    if not request.session or not request.session.session_key:
 | 
			
		||||
        return
 | 
			
		||||
    AccessToken.objects.filter(user=user, session__session_key=request.session.session_key).delete()
 | 
			
		||||
    AccessToken.objects.filter(
 | 
			
		||||
        user=user,
 | 
			
		||||
        session__session__session_key=request.session.session_key,
 | 
			
		||||
    ).delete()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@receiver(pre_delete, sender=AuthenticatedSession)
 | 
			
		||||
def user_session_deleted_oauth_tokens_removal(sender, instance: AuthenticatedSession, **_):
 | 
			
		||||
    """Revoke tokens upon user logout"""
 | 
			
		||||
    AccessToken.objects.filter(
 | 
			
		||||
        user=instance.user,
 | 
			
		||||
        session__session__session_key=instance.session.session_key,
 | 
			
		||||
    ).delete()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@receiver(post_save, sender=User)
 | 
			
		||||
@ -20,6 +32,6 @@ def user_deactivated(sender, instance: User, **_):
 | 
			
		||||
    """Remove user tokens when deactivated"""
 | 
			
		||||
    if instance.is_active:
 | 
			
		||||
        return
 | 
			
		||||
    AccessToken.objects.filter(session__user=instance).delete()
 | 
			
		||||
    RefreshToken.objects.filter(session__user=instance).delete()
 | 
			
		||||
    DeviceToken.objects.filter(session__user=instance).delete()
 | 
			
		||||
    AccessToken.objects.filter(user=instance).delete()
 | 
			
		||||
    RefreshToken.objects.filter(user=instance).delete()
 | 
			
		||||
    DeviceToken.objects.filter(user=instance).delete()
 | 
			
		||||
 | 
			
		||||
@ -7,12 +7,13 @@ from dataclasses import asdict
 | 
			
		||||
from django.urls import reverse
 | 
			
		||||
from django.utils import timezone
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import Application
 | 
			
		||||
from authentik.core.models import Application, AuthenticatedSession, Session
 | 
			
		||||
from authentik.core.tests.utils import create_test_admin_user, create_test_cert, create_test_flow
 | 
			
		||||
from authentik.lib.generators import generate_id
 | 
			
		||||
from authentik.providers.oauth2.models import (
 | 
			
		||||
    AccessToken,
 | 
			
		||||
    ClientTypes,
 | 
			
		||||
    DeviceToken,
 | 
			
		||||
    IDToken,
 | 
			
		||||
    OAuth2Provider,
 | 
			
		||||
    RedirectURI,
 | 
			
		||||
@ -20,6 +21,7 @@ from authentik.providers.oauth2.models import (
 | 
			
		||||
    RefreshToken,
 | 
			
		||||
)
 | 
			
		||||
from authentik.providers.oauth2.tests.utils import OAuthTestCase
 | 
			
		||||
from authentik.root.middleware import ClientIPMiddleware
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class TesOAuth2Revoke(OAuthTestCase):
 | 
			
		||||
@ -135,3 +137,86 @@ class TesOAuth2Revoke(OAuthTestCase):
 | 
			
		||||
            },
 | 
			
		||||
        )
 | 
			
		||||
        self.assertEqual(res.status_code, 200)
 | 
			
		||||
 | 
			
		||||
    def test_revoke_logout(self):
 | 
			
		||||
        """Test revoke on logout"""
 | 
			
		||||
        self.client.force_login(self.user)
 | 
			
		||||
        AccessToken.objects.create(
 | 
			
		||||
            provider=self.provider,
 | 
			
		||||
            user=self.user,
 | 
			
		||||
            session=self.client.session["authenticatedsession"],
 | 
			
		||||
            token=generate_id(),
 | 
			
		||||
            auth_time=timezone.now(),
 | 
			
		||||
            _scope="openid user profile",
 | 
			
		||||
            _id_token=json.dumps(
 | 
			
		||||
                asdict(
 | 
			
		||||
                    IDToken("foo", "bar"),
 | 
			
		||||
                )
 | 
			
		||||
            ),
 | 
			
		||||
        )
 | 
			
		||||
        self.client.logout()
 | 
			
		||||
        self.assertEqual(AccessToken.objects.all().count(), 0)
 | 
			
		||||
 | 
			
		||||
    def test_revoke_session_delete(self):
 | 
			
		||||
        """Test revoke on logout"""
 | 
			
		||||
        session = AuthenticatedSession.objects.create(
 | 
			
		||||
            session=Session.objects.create(
 | 
			
		||||
                session_key=generate_id(),
 | 
			
		||||
                last_ip=ClientIPMiddleware.default_ip,
 | 
			
		||||
            ),
 | 
			
		||||
            user=self.user,
 | 
			
		||||
        )
 | 
			
		||||
        AccessToken.objects.create(
 | 
			
		||||
            provider=self.provider,
 | 
			
		||||
            user=self.user,
 | 
			
		||||
            session=session,
 | 
			
		||||
            token=generate_id(),
 | 
			
		||||
            auth_time=timezone.now(),
 | 
			
		||||
            _scope="openid user profile",
 | 
			
		||||
            _id_token=json.dumps(
 | 
			
		||||
                asdict(
 | 
			
		||||
                    IDToken("foo", "bar"),
 | 
			
		||||
                )
 | 
			
		||||
            ),
 | 
			
		||||
        )
 | 
			
		||||
        session.delete()
 | 
			
		||||
        self.assertEqual(AccessToken.objects.all().count(), 0)
 | 
			
		||||
 | 
			
		||||
    def test_revoke_user_deactivated(self):
 | 
			
		||||
        """Test revoke on logout"""
 | 
			
		||||
        AccessToken.objects.create(
 | 
			
		||||
            provider=self.provider,
 | 
			
		||||
            user=self.user,
 | 
			
		||||
            token=generate_id(),
 | 
			
		||||
            auth_time=timezone.now(),
 | 
			
		||||
            _scope="openid user profile",
 | 
			
		||||
            _id_token=json.dumps(
 | 
			
		||||
                asdict(
 | 
			
		||||
                    IDToken("foo", "bar"),
 | 
			
		||||
                )
 | 
			
		||||
            ),
 | 
			
		||||
        )
 | 
			
		||||
        RefreshToken.objects.create(
 | 
			
		||||
            provider=self.provider,
 | 
			
		||||
            user=self.user,
 | 
			
		||||
            token=generate_id(),
 | 
			
		||||
            auth_time=timezone.now(),
 | 
			
		||||
            _scope="openid user profile",
 | 
			
		||||
            _id_token=json.dumps(
 | 
			
		||||
                asdict(
 | 
			
		||||
                    IDToken("foo", "bar"),
 | 
			
		||||
                )
 | 
			
		||||
            ),
 | 
			
		||||
        )
 | 
			
		||||
        DeviceToken.objects.create(
 | 
			
		||||
            provider=self.provider,
 | 
			
		||||
            user=self.user,
 | 
			
		||||
            _scope="openid user profile",
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        self.user.is_active = False
 | 
			
		||||
        self.user.save()
 | 
			
		||||
 | 
			
		||||
        self.assertEqual(AccessToken.objects.all().count(), 0)
 | 
			
		||||
        self.assertEqual(RefreshToken.objects.all().count(), 0)
 | 
			
		||||
        self.assertEqual(DeviceToken.objects.all().count(), 0)
 | 
			
		||||
 | 
			
		||||
@ -15,7 +15,7 @@ from django.utils import timezone
 | 
			
		||||
from django.utils.translation import gettext as _
 | 
			
		||||
from structlog.stdlib import get_logger
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import Application, AuthenticatedSession
 | 
			
		||||
from authentik.core.models import Application
 | 
			
		||||
from authentik.events.models import Event, EventAction
 | 
			
		||||
from authentik.events.signals import get_login_event
 | 
			
		||||
from authentik.flows.challenge import (
 | 
			
		||||
@ -30,7 +30,7 @@ from authentik.flows.stage import StageView
 | 
			
		||||
from authentik.lib.utils.time import timedelta_from_string
 | 
			
		||||
from authentik.lib.views import bad_request_message
 | 
			
		||||
from authentik.policies.types import PolicyRequest
 | 
			
		||||
from authentik.policies.views import PolicyAccessView, RequestValidationError
 | 
			
		||||
from authentik.policies.views import BufferedPolicyAccessView, RequestValidationError
 | 
			
		||||
from authentik.providers.oauth2.constants import (
 | 
			
		||||
    PKCE_METHOD_PLAIN,
 | 
			
		||||
    PKCE_METHOD_S256,
 | 
			
		||||
@ -316,9 +316,7 @@ class OAuthAuthorizationParams:
 | 
			
		||||
            expires=now + timedelta_from_string(self.provider.access_code_validity),
 | 
			
		||||
            scope=self.scope,
 | 
			
		||||
            nonce=self.nonce,
 | 
			
		||||
            session=AuthenticatedSession.objects.filter(
 | 
			
		||||
                session_key=request.session.session_key
 | 
			
		||||
            ).first(),
 | 
			
		||||
            session=request.session["authenticatedsession"],
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        if self.code_challenge and self.code_challenge_method:
 | 
			
		||||
@ -328,7 +326,7 @@ class OAuthAuthorizationParams:
 | 
			
		||||
        return code
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class AuthorizationFlowInitView(PolicyAccessView):
 | 
			
		||||
class AuthorizationFlowInitView(BufferedPolicyAccessView):
 | 
			
		||||
    """OAuth2 Flow initializer, checks access to application and starts flow"""
 | 
			
		||||
 | 
			
		||||
    params: OAuthAuthorizationParams
 | 
			
		||||
@ -615,9 +613,7 @@ class OAuthFulfillmentStage(StageView):
 | 
			
		||||
            expires=access_token_expiry,
 | 
			
		||||
            provider=self.provider,
 | 
			
		||||
            auth_time=auth_event.created if auth_event else now,
 | 
			
		||||
            session=AuthenticatedSession.objects.filter(
 | 
			
		||||
                session_key=self.request.session.session_key
 | 
			
		||||
            ).first(),
 | 
			
		||||
            session=self.request.session["authenticatedsession"],
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        id_token = IDToken.new(self.provider, token, self.request)
 | 
			
		||||
 | 
			
		||||
@ -20,4 +20,4 @@ def logout_proxy_revoke_direct(sender: type[User], request: HttpRequest, **_):
 | 
			
		||||
@receiver(pre_delete, sender=AuthenticatedSession)
 | 
			
		||||
def logout_proxy_revoke(sender: type[AuthenticatedSession], instance: AuthenticatedSession, **_):
 | 
			
		||||
    """Catch logout by expiring sessions being deleted"""
 | 
			
		||||
    proxy_on_logout.delay(instance.session_key)
 | 
			
		||||
    proxy_on_logout.delay(instance.session.session_key)
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										60
									
								
								authentik/providers/rac/migrations/0007_migrate_session.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										60
									
								
								authentik/providers/rac/migrations/0007_migrate_session.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,60 @@
 | 
			
		||||
# Generated by Django 5.0.11 on 2025-01-27 12:59
 | 
			
		||||
 | 
			
		||||
from django.db import migrations
 | 
			
		||||
import django.db.models.deletion
 | 
			
		||||
from django.db import migrations, models
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def migrate_sessions(apps, schema_editor):
 | 
			
		||||
    ConnectionToken = apps.get_model("authentik_providers_rac", "ConnectionToken")
 | 
			
		||||
    AuthenticatedSession = apps.get_model("authentik_core", "AuthenticatedSession")
 | 
			
		||||
    db_alias = schema_editor.connection.alias
 | 
			
		||||
 | 
			
		||||
    for token in ConnectionToken.objects.using(db_alias).all():
 | 
			
		||||
        token.session = (
 | 
			
		||||
            AuthenticatedSession.objects.using(db_alias)
 | 
			
		||||
            .filter(session_key=token.old_session.session_key)
 | 
			
		||||
            .first()
 | 
			
		||||
        )
 | 
			
		||||
        if token.session:
 | 
			
		||||
            token.save()
 | 
			
		||||
        else:
 | 
			
		||||
            token.delete()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("authentik_providers_rac", "0006_connectiontoken_authentik_p_expires_91f148_idx_and_more"),
 | 
			
		||||
        ("authentik_core", "0046_session_and_more"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.RenameField(
 | 
			
		||||
            model_name="connectiontoken",
 | 
			
		||||
            old_name="session",
 | 
			
		||||
            new_name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.AddField(
 | 
			
		||||
            model_name="connectiontoken",
 | 
			
		||||
            name="session",
 | 
			
		||||
            field=models.ForeignKey(
 | 
			
		||||
                null=True,
 | 
			
		||||
                on_delete=django.db.models.deletion.CASCADE,
 | 
			
		||||
                to="authentik_core.authenticatedsession",
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RunPython(code=migrate_sessions),
 | 
			
		||||
        migrations.AlterField(
 | 
			
		||||
            model_name="connectiontoken",
 | 
			
		||||
            name="session",
 | 
			
		||||
            field=models.ForeignKey(
 | 
			
		||||
                on_delete=django.db.models.deletion.CASCADE,
 | 
			
		||||
                to="authentik_core.authenticatedsession",
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.RemoveField(
 | 
			
		||||
            model_name="connectiontoken",
 | 
			
		||||
            name="old_session",
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -8,7 +8,7 @@ from django.db.models.signals import post_delete, post_save, pre_delete
 | 
			
		||||
from django.dispatch import receiver
 | 
			
		||||
from django.http import HttpRequest
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import User
 | 
			
		||||
from authentik.core.models import AuthenticatedSession, User
 | 
			
		||||
from authentik.providers.rac.api.endpoints import user_endpoint_cache_key
 | 
			
		||||
from authentik.providers.rac.consumer_client import (
 | 
			
		||||
    RAC_CLIENT_GROUP_SESSION,
 | 
			
		||||
@ -32,6 +32,18 @@ def user_logged_out_session(sender, request: HttpRequest, user: User, **_):
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@receiver(pre_delete, sender=AuthenticatedSession)
 | 
			
		||||
def user_session_deleted(sender, instance: AuthenticatedSession, **_):
 | 
			
		||||
    layer = get_channel_layer()
 | 
			
		||||
    async_to_sync(layer.group_send)(
 | 
			
		||||
        RAC_CLIENT_GROUP_SESSION
 | 
			
		||||
        % {
 | 
			
		||||
            "session": instance.session.session_key,
 | 
			
		||||
        },
 | 
			
		||||
        {"type": "event.disconnect", "reason": "session_logout"},
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@receiver(pre_delete, sender=ConnectionToken)
 | 
			
		||||
def pre_delete_connection_token_disconnect(sender, instance: ConnectionToken, **_):
 | 
			
		||||
    """Disconnect session when connection token is deleted"""
 | 
			
		||||
 | 
			
		||||
@ -3,16 +3,22 @@
 | 
			
		||||
{% load authentik_core %}
 | 
			
		||||
 | 
			
		||||
{% block head %}
 | 
			
		||||
<script src="{% versioned_script 'dist/rac/index-%v.js' %}" type="module"></script>
 | 
			
		||||
<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_url }}">
 | 
			
		||||
<link rel="shortcut icon" href="{{ tenant.branding_favicon_url }}">
 | 
			
		||||
{% include "base/header_js.html" %}
 | 
			
		||||
  <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_url }}">
 | 
			
		||||
  <link rel="shortcut icon" href="{{ tenant.branding_favicon_url }}">
 | 
			
		||||
 | 
			
		||||
  {% include "base/header_js.html" %}
 | 
			
		||||
 | 
			
		||||
  <script
 | 
			
		||||
    src="{% versioned_script 'dist/rac/index-%v.js' %}"
 | 
			
		||||
    type="module">
 | 
			
		||||
  </script>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
{% block body %}
 | 
			
		||||
<ak-rac token="{{ url_kwargs.token }}" endpointName="{{ token.endpoint.name }}">
 | 
			
		||||
  <ak-loading></ak-loading>
 | 
			
		||||
</ak-rac>
 | 
			
		||||
  <ak-rac token="{{ url_kwargs.token }}" endpointName="{{ token.endpoint.name }}">
 | 
			
		||||
    <ak-loading></ak-loading>
 | 
			
		||||
  </ak-rac>
 | 
			
		||||
{% endblock %}
 | 
			
		||||
 | 
			
		||||
@ -74,8 +74,6 @@ class TestEndpointsAPI(APITestCase):
 | 
			
		||||
                            "component": "ak-provider-rac-form",
 | 
			
		||||
                            "assigned_application_slug": self.app.slug,
 | 
			
		||||
                            "assigned_application_name": self.app.name,
 | 
			
		||||
                            "assigned_backchannel_application_slug": "",
 | 
			
		||||
                            "assigned_backchannel_application_name": "",
 | 
			
		||||
                            "verbose_name": "RAC Provider",
 | 
			
		||||
                            "verbose_name_plural": "RAC Providers",
 | 
			
		||||
                            "meta_model_name": "authentik_providers_rac.racprovider",
 | 
			
		||||
@ -126,8 +124,6 @@ class TestEndpointsAPI(APITestCase):
 | 
			
		||||
                            "component": "ak-provider-rac-form",
 | 
			
		||||
                            "assigned_application_slug": self.app.slug,
 | 
			
		||||
                            "assigned_application_name": self.app.name,
 | 
			
		||||
                            "assigned_backchannel_application_slug": "",
 | 
			
		||||
                            "assigned_backchannel_application_name": "",
 | 
			
		||||
                            "connection_expiry": "hours=8",
 | 
			
		||||
                            "delete_token_on_disconnect": False,
 | 
			
		||||
                            "verbose_name": "RAC Provider",
 | 
			
		||||
@ -157,8 +153,6 @@ class TestEndpointsAPI(APITestCase):
 | 
			
		||||
                            "component": "ak-provider-rac-form",
 | 
			
		||||
                            "assigned_application_slug": self.app.slug,
 | 
			
		||||
                            "assigned_application_name": self.app.name,
 | 
			
		||||
                            "assigned_backchannel_application_slug": "",
 | 
			
		||||
                            "assigned_backchannel_application_name": "",
 | 
			
		||||
                            "connection_expiry": "hours=8",
 | 
			
		||||
                            "delete_token_on_disconnect": False,
 | 
			
		||||
                            "verbose_name": "RAC Provider",
 | 
			
		||||
 | 
			
		||||
@ -2,7 +2,7 @@
 | 
			
		||||
 | 
			
		||||
from django.test import TransactionTestCase
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import Application, AuthenticatedSession
 | 
			
		||||
from authentik.core.models import Application, AuthenticatedSession, Session
 | 
			
		||||
from authentik.core.tests.utils import create_test_admin_user
 | 
			
		||||
from authentik.lib.generators import generate_id
 | 
			
		||||
from authentik.providers.rac.models import (
 | 
			
		||||
@ -36,13 +36,15 @@ class TestModels(TransactionTestCase):
 | 
			
		||||
 | 
			
		||||
    def test_settings_merge(self):
 | 
			
		||||
        """Test settings merge"""
 | 
			
		||||
        session = Session.objects.create(
 | 
			
		||||
            session_key=generate_id(),
 | 
			
		||||
            last_ip="255.255.255.255",
 | 
			
		||||
        )
 | 
			
		||||
        auth_session = AuthenticatedSession.objects.create(session=session, user=self.user)
 | 
			
		||||
        token = ConnectionToken.objects.create(
 | 
			
		||||
            provider=self.provider,
 | 
			
		||||
            endpoint=self.endpoint,
 | 
			
		||||
            session=AuthenticatedSession.objects.create(
 | 
			
		||||
                user=self.user,
 | 
			
		||||
                session_key=generate_id(),
 | 
			
		||||
            ),
 | 
			
		||||
            session=auth_session,
 | 
			
		||||
        )
 | 
			
		||||
        path = f"/tmp/connection/{token.token}"  # nosec
 | 
			
		||||
        self.assertEqual(
 | 
			
		||||
 | 
			
		||||
@ -1,7 +1,5 @@
 | 
			
		||||
"""rac urls"""
 | 
			
		||||
 | 
			
		||||
from channels.auth import AuthMiddleware
 | 
			
		||||
from channels.sessions import CookieMiddleware
 | 
			
		||||
from django.urls import path
 | 
			
		||||
 | 
			
		||||
from authentik.outposts.channels import TokenOutpostMiddleware
 | 
			
		||||
@ -12,7 +10,7 @@ from authentik.providers.rac.api.providers import RACProviderViewSet
 | 
			
		||||
from authentik.providers.rac.consumer_client import RACClientConsumer
 | 
			
		||||
from authentik.providers.rac.consumer_outpost import RACOutpostConsumer
 | 
			
		||||
from authentik.providers.rac.views import RACInterface, RACStartView
 | 
			
		||||
from authentik.root.asgi_middleware import SessionMiddleware
 | 
			
		||||
from authentik.root.asgi_middleware import AuthMiddlewareStack
 | 
			
		||||
from authentik.root.middleware import ChannelsLoggingMiddleware
 | 
			
		||||
 | 
			
		||||
urlpatterns = [
 | 
			
		||||
@ -31,9 +29,7 @@ urlpatterns = [
 | 
			
		||||
websocket_urlpatterns = [
 | 
			
		||||
    path(
 | 
			
		||||
        "ws/rac/<str:token>/",
 | 
			
		||||
        ChannelsLoggingMiddleware(
 | 
			
		||||
            CookieMiddleware(SessionMiddleware(AuthMiddleware(RACClientConsumer.as_asgi())))
 | 
			
		||||
        ),
 | 
			
		||||
        ChannelsLoggingMiddleware(AuthMiddlewareStack(RACClientConsumer.as_asgi())),
 | 
			
		||||
    ),
 | 
			
		||||
    path(
 | 
			
		||||
        "ws/outpost_rac/<str:channel>/",
 | 
			
		||||
 | 
			
		||||
@ -8,7 +8,7 @@ from django.urls import reverse
 | 
			
		||||
from django.utils.timezone import now
 | 
			
		||||
from django.utils.translation import gettext as _
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import Application, AuthenticatedSession
 | 
			
		||||
from authentik.core.models import Application
 | 
			
		||||
from authentik.core.views.interface import InterfaceView
 | 
			
		||||
from authentik.events.models import Event, EventAction
 | 
			
		||||
from authentik.flows.challenge import RedirectChallenge
 | 
			
		||||
@ -18,11 +18,11 @@ from authentik.flows.planner import PLAN_CONTEXT_APPLICATION, FlowPlanner
 | 
			
		||||
from authentik.flows.stage import RedirectStage
 | 
			
		||||
from authentik.lib.utils.time import timedelta_from_string
 | 
			
		||||
from authentik.policies.engine import PolicyEngine
 | 
			
		||||
from authentik.policies.views import PolicyAccessView
 | 
			
		||||
from authentik.policies.views import BufferedPolicyAccessView
 | 
			
		||||
from authentik.providers.rac.models import ConnectionToken, Endpoint, RACProvider
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class RACStartView(PolicyAccessView):
 | 
			
		||||
class RACStartView(BufferedPolicyAccessView):
 | 
			
		||||
    """Start a RAC connection by checking access and creating a connection token"""
 | 
			
		||||
 | 
			
		||||
    endpoint: Endpoint
 | 
			
		||||
@ -113,9 +113,7 @@ class RACFinalStage(RedirectStage):
 | 
			
		||||
            provider=self.provider,
 | 
			
		||||
            endpoint=self.endpoint,
 | 
			
		||||
            settings=self.executor.plan.context.get("connection_settings", {}),
 | 
			
		||||
            session=AuthenticatedSession.objects.filter(
 | 
			
		||||
                session_key=self.request.session.session_key
 | 
			
		||||
            ).first(),
 | 
			
		||||
            session=self.request.session["authenticatedsession"],
 | 
			
		||||
            expires=now() + timedelta_from_string(self.provider.connection_expiry),
 | 
			
		||||
            expiring=True,
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
@ -0,0 +1,22 @@
 | 
			
		||||
# Generated by Django 5.0.13 on 2025-03-31 13:50
 | 
			
		||||
 | 
			
		||||
import authentik.lib.models
 | 
			
		||||
from django.db import migrations, models
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("authentik_providers_saml", "0017_samlprovider_authn_context_class_ref_mapping"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.AlterField(
 | 
			
		||||
            model_name="samlprovider",
 | 
			
		||||
            name="acs_url",
 | 
			
		||||
            field=models.TextField(
 | 
			
		||||
                validators=[authentik.lib.models.DomainlessURLValidator(schemes=("http", "https"))],
 | 
			
		||||
                verbose_name="ACS URL",
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -10,6 +10,7 @@ from structlog.stdlib import get_logger
 | 
			
		||||
from authentik.core.api.object_types import CreatableType
 | 
			
		||||
from authentik.core.models import PropertyMapping, Provider
 | 
			
		||||
from authentik.crypto.models import CertificateKeyPair
 | 
			
		||||
from authentik.lib.models import DomainlessURLValidator
 | 
			
		||||
from authentik.lib.utils.time import timedelta_string_validator
 | 
			
		||||
from authentik.sources.saml.processors.constants import (
 | 
			
		||||
    DSA_SHA1,
 | 
			
		||||
@ -40,7 +41,9 @@ class SAMLBindings(models.TextChoices):
 | 
			
		||||
class SAMLProvider(Provider):
 | 
			
		||||
    """SAML 2.0 Endpoint for applications which support SAML."""
 | 
			
		||||
 | 
			
		||||
    acs_url = models.URLField(verbose_name=_("ACS URL"))
 | 
			
		||||
    acs_url = models.TextField(
 | 
			
		||||
        validators=[DomainlessURLValidator(schemes=("http", "https"))], verbose_name=_("ACS URL")
 | 
			
		||||
    )
 | 
			
		||||
    audience = models.TextField(
 | 
			
		||||
        default="",
 | 
			
		||||
        blank=True,
 | 
			
		||||
 | 
			
		||||
@ -15,7 +15,7 @@ from authentik.flows.models import in_memory_stage
 | 
			
		||||
from authentik.flows.planner import PLAN_CONTEXT_APPLICATION, PLAN_CONTEXT_SSO, FlowPlanner
 | 
			
		||||
from authentik.flows.views.executor import SESSION_KEY_POST
 | 
			
		||||
from authentik.lib.views import bad_request_message
 | 
			
		||||
from authentik.policies.views import PolicyAccessView
 | 
			
		||||
from authentik.policies.views import BufferedPolicyAccessView
 | 
			
		||||
from authentik.providers.saml.exceptions import CannotHandleAssertion
 | 
			
		||||
from authentik.providers.saml.models import SAMLBindings, SAMLProvider
 | 
			
		||||
from authentik.providers.saml.processors.authn_request_parser import AuthNRequestParser
 | 
			
		||||
@ -35,7 +35,7 @@ from authentik.stages.consent.stage import (
 | 
			
		||||
LOGGER = get_logger()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class SAMLSSOView(PolicyAccessView):
 | 
			
		||||
class SAMLSSOView(BufferedPolicyAccessView):
 | 
			
		||||
    """SAML SSO Base View, which plans a flow and injects our final stage.
 | 
			
		||||
    Calls get/post handler."""
 | 
			
		||||
 | 
			
		||||
@ -83,7 +83,7 @@ class SAMLSSOView(PolicyAccessView):
 | 
			
		||||
 | 
			
		||||
    def post(self, request: HttpRequest, application_slug: str) -> HttpResponse:
 | 
			
		||||
        """GET and POST use the same handler, but we can't
 | 
			
		||||
        override .dispatch easily because PolicyAccessView's dispatch"""
 | 
			
		||||
        override .dispatch easily because BufferedPolicyAccessView's dispatch"""
 | 
			
		||||
        return self.get(request, application_slug)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										41
									
								
								authentik/rbac/api/initial_permissions.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										41
									
								
								authentik/rbac/api/initial_permissions.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,41 @@
 | 
			
		||||
"""RBAC Initial Permissions"""
 | 
			
		||||
 | 
			
		||||
from rest_framework.serializers import ListSerializer
 | 
			
		||||
from rest_framework.viewsets import ModelViewSet
 | 
			
		||||
 | 
			
		||||
from authentik.core.api.used_by import UsedByMixin
 | 
			
		||||
from authentik.core.api.utils import ModelSerializer
 | 
			
		||||
from authentik.rbac.api.rbac import PermissionSerializer
 | 
			
		||||
from authentik.rbac.models import InitialPermissions
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class InitialPermissionsSerializer(ModelSerializer):
 | 
			
		||||
    """InitialPermissions serializer"""
 | 
			
		||||
 | 
			
		||||
    permissions_obj = ListSerializer(
 | 
			
		||||
        child=PermissionSerializer(),
 | 
			
		||||
        read_only=True,
 | 
			
		||||
        source="permissions",
 | 
			
		||||
        required=False,
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
        model = InitialPermissions
 | 
			
		||||
        fields = [
 | 
			
		||||
            "pk",
 | 
			
		||||
            "name",
 | 
			
		||||
            "mode",
 | 
			
		||||
            "role",
 | 
			
		||||
            "permissions",
 | 
			
		||||
            "permissions_obj",
 | 
			
		||||
        ]
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class InitialPermissionsViewSet(UsedByMixin, ModelViewSet):
 | 
			
		||||
    """InitialPermissions viewset"""
 | 
			
		||||
 | 
			
		||||
    queryset = InitialPermissions.objects.all()
 | 
			
		||||
    serializer_class = InitialPermissionsSerializer
 | 
			
		||||
    search_fields = ["name"]
 | 
			
		||||
    ordering = ["name"]
 | 
			
		||||
    filterset_fields = ["name"]
 | 
			
		||||
							
								
								
									
										39
									
								
								authentik/rbac/migrations/0005_initialpermissions.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										39
									
								
								authentik/rbac/migrations/0005_initialpermissions.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,39 @@
 | 
			
		||||
# Generated by Django 5.0.13 on 2025-04-07 13:05
 | 
			
		||||
 | 
			
		||||
import django.db.models.deletion
 | 
			
		||||
from django.db import migrations, models
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("auth", "0012_alter_user_first_name_max_length"),
 | 
			
		||||
        ("authentik_rbac", "0004_alter_systempermission_options"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.CreateModel(
 | 
			
		||||
            name="InitialPermissions",
 | 
			
		||||
            fields=[
 | 
			
		||||
                (
 | 
			
		||||
                    "id",
 | 
			
		||||
                    models.AutoField(
 | 
			
		||||
                        auto_created=True, primary_key=True, serialize=False, verbose_name="ID"
 | 
			
		||||
                    ),
 | 
			
		||||
                ),
 | 
			
		||||
                ("name", models.TextField(max_length=150, unique=True)),
 | 
			
		||||
                ("mode", models.CharField(choices=[("user", "User"), ("role", "Role")])),
 | 
			
		||||
                ("permissions", models.ManyToManyField(blank=True, to="auth.permission")),
 | 
			
		||||
                (
 | 
			
		||||
                    "role",
 | 
			
		||||
                    models.ForeignKey(
 | 
			
		||||
                        on_delete=django.db.models.deletion.CASCADE, to="authentik_rbac.role"
 | 
			
		||||
                    ),
 | 
			
		||||
                ),
 | 
			
		||||
            ],
 | 
			
		||||
            options={
 | 
			
		||||
                "verbose_name": "Initial Permissions",
 | 
			
		||||
                "verbose_name_plural": "Initial Permissions",
 | 
			
		||||
            },
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -3,6 +3,7 @@
 | 
			
		||||
from uuid import uuid4
 | 
			
		||||
 | 
			
		||||
from django.contrib.auth.management import _get_all_permissions
 | 
			
		||||
from django.contrib.auth.models import Permission
 | 
			
		||||
from django.db import models
 | 
			
		||||
from django.db.transaction import atomic
 | 
			
		||||
from django.utils.translation import gettext_lazy as _
 | 
			
		||||
@ -75,6 +76,35 @@ class Role(SerializerModel):
 | 
			
		||||
        ]
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class InitialPermissionsMode(models.TextChoices):
 | 
			
		||||
    """Determines which entity the initial permissions are assigned to."""
 | 
			
		||||
 | 
			
		||||
    USER = "user", _("User")
 | 
			
		||||
    ROLE = "role", _("Role")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class InitialPermissions(SerializerModel):
 | 
			
		||||
    """Assigns permissions for newly created objects."""
 | 
			
		||||
 | 
			
		||||
    name = models.TextField(max_length=150, unique=True)
 | 
			
		||||
    mode = models.CharField(choices=InitialPermissionsMode.choices)
 | 
			
		||||
    role = models.ForeignKey(Role, on_delete=models.CASCADE)
 | 
			
		||||
    permissions = models.ManyToManyField(Permission, blank=True)
 | 
			
		||||
 | 
			
		||||
    @property
 | 
			
		||||
    def serializer(self) -> type[BaseSerializer]:
 | 
			
		||||
        from authentik.rbac.api.initial_permissions import InitialPermissionsSerializer
 | 
			
		||||
 | 
			
		||||
        return InitialPermissionsSerializer
 | 
			
		||||
 | 
			
		||||
    def __str__(self) -> str:
 | 
			
		||||
        return f"Initial Permissions for Role #{self.role_id}, applying to #{self.mode}"
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
        verbose_name = _("Initial Permissions")
 | 
			
		||||
        verbose_name_plural = _("Initial Permissions")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class SystemPermission(models.Model):
 | 
			
		||||
    """System-wide permissions that are not related to any direct
 | 
			
		||||
    database model"""
 | 
			
		||||
 | 
			
		||||
@ -1,9 +1,13 @@
 | 
			
		||||
"""RBAC Permissions"""
 | 
			
		||||
 | 
			
		||||
from django.contrib.contenttypes.models import ContentType
 | 
			
		||||
from django.db.models import Model
 | 
			
		||||
from guardian.shortcuts import assign_perm
 | 
			
		||||
from rest_framework.permissions import BasePermission, DjangoObjectPermissions
 | 
			
		||||
from rest_framework.request import Request
 | 
			
		||||
 | 
			
		||||
from authentik.rbac.models import InitialPermissions, InitialPermissionsMode
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class ObjectPermissions(DjangoObjectPermissions):
 | 
			
		||||
    """RBAC Permissions"""
 | 
			
		||||
@ -51,3 +55,20 @@ def HasPermission(*perm: str) -> type[BasePermission]:
 | 
			
		||||
            return bool(request.user and request.user.has_perms(perm))
 | 
			
		||||
 | 
			
		||||
    return checker
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
# TODO: add `user: User` type annotation without circular dependencies.
 | 
			
		||||
# The author of this function isn't proficient/patient enough to do it.
 | 
			
		||||
def assign_initial_permissions(user, instance: Model):
 | 
			
		||||
    # Performance here should not be an issue, but if needed, there are many optimization routes
 | 
			
		||||
    initial_permissions_list = InitialPermissions.objects.filter(role__group__in=user.groups.all())
 | 
			
		||||
    for initial_permissions in initial_permissions_list:
 | 
			
		||||
        for permission in initial_permissions.permissions.all():
 | 
			
		||||
            if permission.content_type != ContentType.objects.get_for_model(instance):
 | 
			
		||||
                continue
 | 
			
		||||
            assign_to = (
 | 
			
		||||
                user
 | 
			
		||||
                if initial_permissions.mode == InitialPermissionsMode.USER
 | 
			
		||||
                else initial_permissions.role.group
 | 
			
		||||
            )
 | 
			
		||||
            assign_perm(permission, assign_to, instance)
 | 
			
		||||
 | 
			
		||||
							
								
								
									
										116
									
								
								authentik/rbac/tests/test_initial_permissions.py
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										116
									
								
								authentik/rbac/tests/test_initial_permissions.py
									
									
									
									
									
										Normal file
									
								
							@ -0,0 +1,116 @@
 | 
			
		||||
"""Test InitialPermissions"""
 | 
			
		||||
 | 
			
		||||
from django.contrib.auth.models import Permission
 | 
			
		||||
from guardian.shortcuts import assign_perm
 | 
			
		||||
from rest_framework.reverse import reverse
 | 
			
		||||
from rest_framework.test import APITestCase
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import Group
 | 
			
		||||
from authentik.core.tests.utils import create_test_user
 | 
			
		||||
from authentik.lib.generators import generate_id
 | 
			
		||||
from authentik.rbac.models import InitialPermissions, InitialPermissionsMode, Role
 | 
			
		||||
from authentik.stages.dummy.models import DummyStage
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class TestInitialPermissions(APITestCase):
 | 
			
		||||
    """Test InitialPermissions"""
 | 
			
		||||
 | 
			
		||||
    def setUp(self) -> None:
 | 
			
		||||
        self.user = create_test_user()
 | 
			
		||||
        self.same_role_user = create_test_user()
 | 
			
		||||
        self.different_role_user = create_test_user()
 | 
			
		||||
 | 
			
		||||
        self.role = Role.objects.create(name=generate_id())
 | 
			
		||||
        self.different_role = Role.objects.create(name=generate_id())
 | 
			
		||||
 | 
			
		||||
        self.group = Group.objects.create(name=generate_id())
 | 
			
		||||
        self.different_group = Group.objects.create(name=generate_id())
 | 
			
		||||
 | 
			
		||||
        self.group.roles.add(self.role)
 | 
			
		||||
        self.group.users.add(self.user, self.same_role_user)
 | 
			
		||||
        self.different_group.roles.add(self.different_role)
 | 
			
		||||
        self.different_group.users.add(self.different_role_user)
 | 
			
		||||
 | 
			
		||||
        self.ip = InitialPermissions.objects.create(
 | 
			
		||||
            name=generate_id(), mode=InitialPermissionsMode.USER, role=self.role
 | 
			
		||||
        )
 | 
			
		||||
        self.view_role = Permission.objects.filter(codename="view_role").first()
 | 
			
		||||
        self.ip.permissions.add(self.view_role)
 | 
			
		||||
 | 
			
		||||
        assign_perm("authentik_rbac.add_role", self.user)
 | 
			
		||||
        self.client.force_login(self.user)
 | 
			
		||||
 | 
			
		||||
    def test_different_role(self):
 | 
			
		||||
        """InitialPermissions for different role does nothing"""
 | 
			
		||||
        self.ip.role = self.different_role
 | 
			
		||||
        self.ip.save()
 | 
			
		||||
 | 
			
		||||
        self.client.post(reverse("authentik_api:roles-list"), {"name": "test-role"})
 | 
			
		||||
 | 
			
		||||
        role = Role.objects.filter(name="test-role").first()
 | 
			
		||||
        self.assertFalse(self.user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
 | 
			
		||||
    def test_different_model(self):
 | 
			
		||||
        """InitialPermissions for different model does nothing"""
 | 
			
		||||
        assign_perm("authentik_stages_dummy.add_dummystage", self.user)
 | 
			
		||||
 | 
			
		||||
        self.client.post(
 | 
			
		||||
            reverse("authentik_api:stages-dummy-list"), {"name": "test-stage", "throw-error": False}
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        role = Role.objects.filter(name="test-role").first()
 | 
			
		||||
        self.assertFalse(self.user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
        stage = DummyStage.objects.filter(name="test-stage").first()
 | 
			
		||||
        self.assertFalse(self.user.has_perm("authentik_stages_dummy.view_dummystage", stage))
 | 
			
		||||
 | 
			
		||||
    def test_mode_user(self):
 | 
			
		||||
        """InitialPermissions adds user permission in user mode"""
 | 
			
		||||
        self.client.post(reverse("authentik_api:roles-list"), {"name": "test-role"})
 | 
			
		||||
 | 
			
		||||
        role = Role.objects.filter(name="test-role").first()
 | 
			
		||||
        self.assertTrue(self.user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
        self.assertFalse(self.same_role_user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
 | 
			
		||||
    def test_mode_role(self):
 | 
			
		||||
        """InitialPermissions adds role permission in role mode"""
 | 
			
		||||
        self.ip.mode = InitialPermissionsMode.ROLE
 | 
			
		||||
        self.ip.save()
 | 
			
		||||
 | 
			
		||||
        self.client.post(reverse("authentik_api:roles-list"), {"name": "test-role"})
 | 
			
		||||
 | 
			
		||||
        role = Role.objects.filter(name="test-role").first()
 | 
			
		||||
        self.assertTrue(self.user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
        self.assertTrue(self.same_role_user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
 | 
			
		||||
    def test_many_permissions(self):
 | 
			
		||||
        """InitialPermissions can add multiple permissions"""
 | 
			
		||||
        change_role = Permission.objects.filter(codename="change_role").first()
 | 
			
		||||
        self.ip.permissions.add(change_role)
 | 
			
		||||
 | 
			
		||||
        self.client.post(reverse("authentik_api:roles-list"), {"name": "test-role"})
 | 
			
		||||
 | 
			
		||||
        role = Role.objects.filter(name="test-role").first()
 | 
			
		||||
        self.assertTrue(self.user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
        self.assertTrue(self.user.has_perm("authentik_rbac.change_role", role))
 | 
			
		||||
 | 
			
		||||
    def test_permissions_separated_by_role(self):
 | 
			
		||||
        """When the triggering user is part of two different roles with InitialPermissions in role
 | 
			
		||||
        mode, it only adds permissions to the relevant role."""
 | 
			
		||||
        self.ip.mode = InitialPermissionsMode.ROLE
 | 
			
		||||
        self.ip.save()
 | 
			
		||||
        different_ip = InitialPermissions.objects.create(
 | 
			
		||||
            name=generate_id(), mode=InitialPermissionsMode.ROLE, role=self.different_role
 | 
			
		||||
        )
 | 
			
		||||
        change_role = Permission.objects.filter(codename="change_role").first()
 | 
			
		||||
        different_ip.permissions.add(change_role)
 | 
			
		||||
        self.different_group.users.add(self.user)
 | 
			
		||||
 | 
			
		||||
        self.client.post(reverse("authentik_api:roles-list"), {"name": "test-role"})
 | 
			
		||||
 | 
			
		||||
        role = Role.objects.filter(name="test-role").first()
 | 
			
		||||
        self.assertTrue(self.user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
        self.assertTrue(self.same_role_user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
        self.assertFalse(self.different_role_user.has_perm("authentik_rbac.view_role", role))
 | 
			
		||||
        self.assertTrue(self.user.has_perm("authentik_rbac.change_role", role))
 | 
			
		||||
        self.assertFalse(self.same_role_user.has_perm("authentik_rbac.change_role", role))
 | 
			
		||||
        self.assertTrue(self.different_role_user.has_perm("authentik_rbac.change_role", role))
 | 
			
		||||
@ -1,5 +1,6 @@
 | 
			
		||||
"""RBAC API urls"""
 | 
			
		||||
 | 
			
		||||
from authentik.rbac.api.initial_permissions import InitialPermissionsViewSet
 | 
			
		||||
from authentik.rbac.api.rbac import RBACPermissionViewSet
 | 
			
		||||
from authentik.rbac.api.rbac_assigned_by_roles import RoleAssignedPermissionViewSet
 | 
			
		||||
from authentik.rbac.api.rbac_assigned_by_users import UserAssignedPermissionViewSet
 | 
			
		||||
@ -21,5 +22,6 @@ api_urlpatterns = [
 | 
			
		||||
    ("rbac/permissions/users", UserPermissionViewSet, "permissions-users"),
 | 
			
		||||
    ("rbac/permissions/roles", RolePermissionViewSet, "permissions-roles"),
 | 
			
		||||
    ("rbac/permissions", RBACPermissionViewSet),
 | 
			
		||||
    ("rbac/roles", RoleViewSet),
 | 
			
		||||
    ("rbac/roles", RoleViewSet, "roles"),
 | 
			
		||||
    ("rbac/initial_permissions", InitialPermissionsViewSet, "initial-permissions"),
 | 
			
		||||
]
 | 
			
		||||
 | 
			
		||||
@ -50,7 +50,7 @@ class TestRecovery(TestCase):
 | 
			
		||||
        )
 | 
			
		||||
        token = Token.objects.get(intent=TokenIntents.INTENT_RECOVERY, user=self.user)
 | 
			
		||||
        self.client.get(reverse("authentik_recovery:use-token", kwargs={"key": token.key}))
 | 
			
		||||
        self.assertEqual(int(self.client.session["_auth_user_id"]), token.user.pk)
 | 
			
		||||
        self.assertEqual(self.client.session["authenticatedsession"].user.pk, token.user.pk)
 | 
			
		||||
 | 
			
		||||
    def test_recovery_view_invalid(self):
 | 
			
		||||
        """Test recovery view with invalid token"""
 | 
			
		||||
 | 
			
		||||
@ -1,8 +1,12 @@
 | 
			
		||||
"""ASGI middleware"""
 | 
			
		||||
 | 
			
		||||
from channels.auth import UserLazyObject
 | 
			
		||||
from channels.db import database_sync_to_async
 | 
			
		||||
from channels.middleware import BaseMiddleware
 | 
			
		||||
from channels.sessions import CookieMiddleware
 | 
			
		||||
from channels.sessions import InstanceSessionWrapper as UpstreamInstanceSessionWrapper
 | 
			
		||||
from channels.sessions import SessionMiddleware as UpstreamSessionMiddleware
 | 
			
		||||
from django.contrib.auth.models import AnonymousUser
 | 
			
		||||
 | 
			
		||||
from authentik.root.middleware import SessionMiddleware as HTTPSessionMiddleware
 | 
			
		||||
 | 
			
		||||
@ -33,3 +37,48 @@ class SessionMiddleware(UpstreamSessionMiddleware):
 | 
			
		||||
        await wrapper.resolve_session()
 | 
			
		||||
 | 
			
		||||
        return await self.inner(wrapper.scope, receive, wrapper.send)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@database_sync_to_async
 | 
			
		||||
def get_user(scope):
 | 
			
		||||
    """
 | 
			
		||||
    Return the user model instance associated with the given scope.
 | 
			
		||||
    If no user is retrieved, return an instance of `AnonymousUser`.
 | 
			
		||||
    """
 | 
			
		||||
    if "session" not in scope:
 | 
			
		||||
        raise ValueError(
 | 
			
		||||
            "Cannot find session in scope. You should wrap your consumer in SessionMiddleware."
 | 
			
		||||
        )
 | 
			
		||||
    user = None
 | 
			
		||||
    if (authenticated_session := scope["session"].get("authenticated_session", None)) is not None:
 | 
			
		||||
        user = authenticated_session.user
 | 
			
		||||
    return user or AnonymousUser()
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class AuthMiddleware(BaseMiddleware):
 | 
			
		||||
    def populate_scope(self, scope):
 | 
			
		||||
        # Make sure we have a session
 | 
			
		||||
        if "session" not in scope:
 | 
			
		||||
            raise ValueError(
 | 
			
		||||
                "AuthMiddleware cannot find session in scope. SessionMiddleware must be above it."
 | 
			
		||||
            )
 | 
			
		||||
        # Add it to the scope if it's not there already
 | 
			
		||||
        if "user" not in scope:
 | 
			
		||||
            scope["user"] = UserLazyObject()
 | 
			
		||||
 | 
			
		||||
    async def resolve_scope(self, scope):
 | 
			
		||||
        scope["user"]._wrapped = await get_user(scope)
 | 
			
		||||
 | 
			
		||||
    async def __call__(self, scope, receive, send):
 | 
			
		||||
        scope = dict(scope)
 | 
			
		||||
        # Scope injection/mutation per this middleware's needs.
 | 
			
		||||
        self.populate_scope(scope)
 | 
			
		||||
        # Grab the finalized/resolved scope
 | 
			
		||||
        await self.resolve_scope(scope)
 | 
			
		||||
 | 
			
		||||
        return await super().__call__(scope, receive, send)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
# Handy shortcut for applying all three layers at once
 | 
			
		||||
def AuthMiddlewareStack(inner):
 | 
			
		||||
    return CookieMiddleware(SessionMiddleware(AuthMiddleware(inner)))
 | 
			
		||||
 | 
			
		||||
@ -49,7 +49,7 @@ class SessionMiddleware(UpstreamSessionMiddleware):
 | 
			
		||||
        return False
 | 
			
		||||
 | 
			
		||||
    @staticmethod
 | 
			
		||||
    def decode_session_key(key: str) -> str:
 | 
			
		||||
    def decode_session_key(key: str | None) -> str | None:
 | 
			
		||||
        """Decode raw session cookie, and parse JWT"""
 | 
			
		||||
        # We need to support the standard django format of just a session key
 | 
			
		||||
        # for testing setups, where the session is directly set
 | 
			
		||||
@ -64,7 +64,11 @@ class SessionMiddleware(UpstreamSessionMiddleware):
 | 
			
		||||
    def process_request(self, request: HttpRequest):
 | 
			
		||||
        raw_session = request.COOKIES.get(settings.SESSION_COOKIE_NAME)
 | 
			
		||||
        session_key = SessionMiddleware.decode_session_key(raw_session)
 | 
			
		||||
        request.session = self.SessionStore(session_key)
 | 
			
		||||
        request.session = self.SessionStore(
 | 
			
		||||
            session_key,
 | 
			
		||||
            last_ip=ClientIPMiddleware.get_client_ip(request),
 | 
			
		||||
            last_user_agent=request.META.get("HTTP_USER_AGENT", ""),
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
    def process_response(self, request: HttpRequest, response: HttpResponse) -> HttpResponse:
 | 
			
		||||
        """
 | 
			
		||||
 | 
			
		||||
@ -1,23 +0,0 @@
 | 
			
		||||
"""
 | 
			
		||||
Module for abstract serializer/unserializer base classes.
 | 
			
		||||
"""
 | 
			
		||||
 | 
			
		||||
import pickle  # nosec
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class PickleSerializer:
 | 
			
		||||
    """
 | 
			
		||||
    Simple wrapper around pickle to be used in signing.dumps()/loads() and
 | 
			
		||||
    cache backends.
 | 
			
		||||
    """
 | 
			
		||||
 | 
			
		||||
    def __init__(self, protocol=None):
 | 
			
		||||
        self.protocol = pickle.HIGHEST_PROTOCOL if protocol is None else protocol
 | 
			
		||||
 | 
			
		||||
    def dumps(self, obj):
 | 
			
		||||
        """Pickle data to be stored in redis"""
 | 
			
		||||
        return pickle.dumps(obj, self.protocol)
 | 
			
		||||
 | 
			
		||||
    def loads(self, data):
 | 
			
		||||
        """Unpickle data to be loaded from redis"""
 | 
			
		||||
        return pickle.loads(data)  # nosec
 | 
			
		||||
@ -7,7 +7,6 @@ from pathlib import Path
 | 
			
		||||
 | 
			
		||||
import orjson
 | 
			
		||||
from celery.schedules import crontab
 | 
			
		||||
from django.conf import ImproperlyConfigured
 | 
			
		||||
from sentry_sdk import set_tag
 | 
			
		||||
from xmlsec import enable_debug_trace
 | 
			
		||||
 | 
			
		||||
@ -43,7 +42,6 @@ SESSION_COOKIE_DOMAIN = CONFIG.get("cookie_domain", None)
 | 
			
		||||
APPEND_SLASH = False
 | 
			
		||||
 | 
			
		||||
AUTHENTICATION_BACKENDS = [
 | 
			
		||||
    "django.contrib.auth.backends.ModelBackend",
 | 
			
		||||
    BACKEND_INBUILT,
 | 
			
		||||
    BACKEND_APP_PASSWORD,
 | 
			
		||||
    BACKEND_LDAP,
 | 
			
		||||
@ -229,17 +227,7 @@ CACHES = {
 | 
			
		||||
DJANGO_REDIS_SCAN_ITERSIZE = 1000
 | 
			
		||||
DJANGO_REDIS_IGNORE_EXCEPTIONS = True
 | 
			
		||||
DJANGO_REDIS_LOG_IGNORED_EXCEPTIONS = True
 | 
			
		||||
match CONFIG.get("session_storage", "cache"):
 | 
			
		||||
    case "cache":
 | 
			
		||||
        SESSION_ENGINE = "django.contrib.sessions.backends.cache"
 | 
			
		||||
    case "db":
 | 
			
		||||
        SESSION_ENGINE = "django.contrib.sessions.backends.db"
 | 
			
		||||
    case _:
 | 
			
		||||
        raise ImproperlyConfigured(
 | 
			
		||||
            "Invalid session_storage setting, allowed values are db and cache"
 | 
			
		||||
        )
 | 
			
		||||
SESSION_SERIALIZER = "authentik.root.sessions.pickle.PickleSerializer"
 | 
			
		||||
SESSION_CACHE_ALIAS = "default"
 | 
			
		||||
SESSION_ENGINE = "authentik.core.sessions"
 | 
			
		||||
# Configured via custom SessionMiddleware
 | 
			
		||||
# SESSION_COOKIE_SAMESITE = "None"
 | 
			
		||||
# SESSION_COOKIE_SECURE = True
 | 
			
		||||
@ -256,7 +244,7 @@ MIDDLEWARE = [
 | 
			
		||||
    "django_prometheus.middleware.PrometheusBeforeMiddleware",
 | 
			
		||||
    "authentik.root.middleware.ClientIPMiddleware",
 | 
			
		||||
    "authentik.stages.user_login.middleware.BoundSessionMiddleware",
 | 
			
		||||
    "django.contrib.auth.middleware.AuthenticationMiddleware",
 | 
			
		||||
    "authentik.core.middleware.AuthenticationMiddleware",
 | 
			
		||||
    "authentik.core.middleware.RequestIDMiddleware",
 | 
			
		||||
    "authentik.brands.middleware.BrandMiddleware",
 | 
			
		||||
    "authentik.events.middleware.AuditMiddleware",
 | 
			
		||||
 | 
			
		||||
@ -28,8 +28,8 @@ def pytest_report_header(*_, **__):
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def pytest_collection_modifyitems(config: pytest.Config, items: list[pytest.Item]) -> None:
 | 
			
		||||
    current_id = int(environ.get("CI_RUN_ID", 0)) - 1
 | 
			
		||||
    total_ids = int(environ.get("CI_TOTAL_RUNS", 0))
 | 
			
		||||
    current_id = int(environ.get("CI_RUN_ID", "0")) - 1
 | 
			
		||||
    total_ids = int(environ.get("CI_TOTAL_RUNS", "0"))
 | 
			
		||||
 | 
			
		||||
    if total_ids:
 | 
			
		||||
        num_tests = len(items)
 | 
			
		||||
 | 
			
		||||
@ -1,13 +1,11 @@
 | 
			
		||||
"""Kerberos Source Serializer"""
 | 
			
		||||
 | 
			
		||||
from rest_framework.viewsets import ModelViewSet
 | 
			
		||||
 | 
			
		||||
from authentik.core.api.sources import (
 | 
			
		||||
    GroupSourceConnectionSerializer,
 | 
			
		||||
    GroupSourceConnectionViewSet,
 | 
			
		||||
    UserSourceConnectionSerializer,
 | 
			
		||||
    UserSourceConnectionViewSet,
 | 
			
		||||
)
 | 
			
		||||
from authentik.core.api.used_by import UsedByMixin
 | 
			
		||||
from authentik.sources.kerberos.models import (
 | 
			
		||||
    GroupKerberosSourceConnection,
 | 
			
		||||
    UserKerberosSourceConnection,
 | 
			
		||||
@ -15,33 +13,20 @@ from authentik.sources.kerberos.models import (
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class UserKerberosSourceConnectionSerializer(UserSourceConnectionSerializer):
 | 
			
		||||
    """Kerberos Source Serializer"""
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
    class Meta(UserSourceConnectionSerializer.Meta):
 | 
			
		||||
        model = UserKerberosSourceConnection
 | 
			
		||||
        fields = UserSourceConnectionSerializer.Meta.fields + ["identifier"]
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class UserKerberosSourceConnectionViewSet(UsedByMixin, ModelViewSet):
 | 
			
		||||
    """Source Viewset"""
 | 
			
		||||
 | 
			
		||||
class UserKerberosSourceConnectionViewSet(UserSourceConnectionViewSet, ModelViewSet):
 | 
			
		||||
    queryset = UserKerberosSourceConnection.objects.all()
 | 
			
		||||
    serializer_class = UserKerberosSourceConnectionSerializer
 | 
			
		||||
    filterset_fields = ["source__slug"]
 | 
			
		||||
    search_fields = ["source__slug"]
 | 
			
		||||
    ordering = ["source__slug"]
 | 
			
		||||
    owner_field = "user"
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class GroupKerberosSourceConnectionSerializer(GroupSourceConnectionSerializer):
 | 
			
		||||
    """OAuth Group-Source connection Serializer"""
 | 
			
		||||
 | 
			
		||||
    class Meta(GroupSourceConnectionSerializer.Meta):
 | 
			
		||||
        model = GroupKerberosSourceConnection
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class GroupKerberosSourceConnectionViewSet(GroupSourceConnectionViewSet):
 | 
			
		||||
    """Group-source connection Viewset"""
 | 
			
		||||
 | 
			
		||||
class GroupKerberosSourceConnectionViewSet(GroupSourceConnectionViewSet, ModelViewSet):
 | 
			
		||||
    queryset = GroupKerberosSourceConnection.objects.all()
 | 
			
		||||
    serializer_class = GroupKerberosSourceConnectionSerializer
 | 
			
		||||
 | 
			
		||||
@ -0,0 +1,28 @@
 | 
			
		||||
from django.db import migrations
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
def migrate_identifier(apps, schema_editor):
 | 
			
		||||
    db_alias = schema_editor.connection.alias
 | 
			
		||||
    UserKerberosSourceConnection = apps.get_model(
 | 
			
		||||
        "authentik_sources_kerberos", "UserKerberosSourceConnection"
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
    for connection in UserKerberosSourceConnection.objects.using(db_alias).all():
 | 
			
		||||
        connection.new_identifier = connection.identifier
 | 
			
		||||
        connection.save(using=db_alias)
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("authentik_sources_kerberos", "0002_kerberossource_kadmin_type"),
 | 
			
		||||
        ("authentik_core", "0044_usersourceconnection_new_identifier"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.RunPython(code=migrate_identifier, reverse_code=migrations.RunPython.noop),
 | 
			
		||||
        migrations.RemoveField(
 | 
			
		||||
            model_name="userkerberossourceconnection",
 | 
			
		||||
            name="identifier",
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -372,8 +372,6 @@ class KerberosSourcePropertyMapping(PropertyMapping):
 | 
			
		||||
class UserKerberosSourceConnection(UserSourceConnection):
 | 
			
		||||
    """Connection to configured Kerberos Sources."""
 | 
			
		||||
 | 
			
		||||
    identifier = models.TextField()
 | 
			
		||||
 | 
			
		||||
    @property
 | 
			
		||||
    def serializer(self) -> type[Serializer]:
 | 
			
		||||
        from authentik.sources.kerberos.api.source_connection import (
 | 
			
		||||
 | 
			
		||||
@ -15,11 +15,22 @@ from rest_framework.response import Response
 | 
			
		||||
from rest_framework.viewsets import ModelViewSet
 | 
			
		||||
 | 
			
		||||
from authentik.core.api.property_mappings import PropertyMappingFilterSet, PropertyMappingSerializer
 | 
			
		||||
from authentik.core.api.sources import SourceSerializer
 | 
			
		||||
from authentik.core.api.sources import (
 | 
			
		||||
    GroupSourceConnectionSerializer,
 | 
			
		||||
    GroupSourceConnectionViewSet,
 | 
			
		||||
    SourceSerializer,
 | 
			
		||||
    UserSourceConnectionSerializer,
 | 
			
		||||
    UserSourceConnectionViewSet,
 | 
			
		||||
)
 | 
			
		||||
from authentik.core.api.used_by import UsedByMixin
 | 
			
		||||
from authentik.crypto.models import CertificateKeyPair
 | 
			
		||||
from authentik.lib.sync.outgoing.api import SyncStatusSerializer
 | 
			
		||||
from authentik.sources.ldap.models import LDAPSource, LDAPSourcePropertyMapping
 | 
			
		||||
from authentik.sources.ldap.models import (
 | 
			
		||||
    GroupLDAPSourceConnection,
 | 
			
		||||
    LDAPSource,
 | 
			
		||||
    LDAPSourcePropertyMapping,
 | 
			
		||||
    UserLDAPSourceConnection,
 | 
			
		||||
)
 | 
			
		||||
from authentik.sources.ldap.tasks import CACHE_KEY_STATUS, SYNC_CLASSES
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -99,6 +110,7 @@ class LDAPSourceSerializer(SourceSerializer):
 | 
			
		||||
            "sync_groups",
 | 
			
		||||
            "sync_parent_group",
 | 
			
		||||
            "connectivity",
 | 
			
		||||
            "lookup_groups_from_user",
 | 
			
		||||
        ]
 | 
			
		||||
        extra_kwargs = {"bind_password": {"write_only": True}}
 | 
			
		||||
 | 
			
		||||
@ -134,6 +146,7 @@ class LDAPSourceViewSet(UsedByMixin, ModelViewSet):
 | 
			
		||||
        "sync_parent_group",
 | 
			
		||||
        "user_property_mappings",
 | 
			
		||||
        "group_property_mappings",
 | 
			
		||||
        "lookup_groups_from_user",
 | 
			
		||||
    ]
 | 
			
		||||
    search_fields = ["name", "slug"]
 | 
			
		||||
    ordering = ["name"]
 | 
			
		||||
@ -219,3 +232,23 @@ class LDAPSourcePropertyMappingViewSet(UsedByMixin, ModelViewSet):
 | 
			
		||||
    filterset_class = LDAPSourcePropertyMappingFilter
 | 
			
		||||
    search_fields = ["name"]
 | 
			
		||||
    ordering = ["name"]
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class UserLDAPSourceConnectionSerializer(UserSourceConnectionSerializer):
 | 
			
		||||
    class Meta(UserSourceConnectionSerializer.Meta):
 | 
			
		||||
        model = UserLDAPSourceConnection
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class UserLDAPSourceConnectionViewSet(UserSourceConnectionViewSet, ModelViewSet):
 | 
			
		||||
    queryset = UserLDAPSourceConnection.objects.all()
 | 
			
		||||
    serializer_class = UserLDAPSourceConnectionSerializer
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class GroupLDAPSourceConnectionSerializer(GroupSourceConnectionSerializer):
 | 
			
		||||
    class Meta(GroupSourceConnectionSerializer.Meta):
 | 
			
		||||
        model = GroupLDAPSourceConnection
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class GroupLDAPSourceConnectionViewSet(GroupSourceConnectionViewSet, ModelViewSet):
 | 
			
		||||
    queryset = GroupLDAPSourceConnection.objects.all()
 | 
			
		||||
    serializer_class = GroupLDAPSourceConnectionSerializer
 | 
			
		||||
 | 
			
		||||
@ -0,0 +1,24 @@
 | 
			
		||||
# Generated by Django 5.0.13 on 2025-03-26 17:06
 | 
			
		||||
 | 
			
		||||
from django.db import migrations, models
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        (
 | 
			
		||||
            "authentik_sources_ldap",
 | 
			
		||||
            "0006_rename_ldappropertymapping_ldapsourcepropertymapping_and_more",
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.AddField(
 | 
			
		||||
            model_name="ldapsource",
 | 
			
		||||
            name="lookup_groups_from_user",
 | 
			
		||||
            field=models.BooleanField(
 | 
			
		||||
                default=False,
 | 
			
		||||
                help_text="Lookup group membership based on a user attribute instead of a group attribute. This allows nested group resolution on systems like FreeIPA and Active Directory",
 | 
			
		||||
            ),
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -0,0 +1,57 @@
 | 
			
		||||
# Generated by Django 5.0.14 on 2025-04-11 11:46
 | 
			
		||||
 | 
			
		||||
import django.db.models.deletion
 | 
			
		||||
from django.db import migrations, models
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class Migration(migrations.Migration):
 | 
			
		||||
 | 
			
		||||
    dependencies = [
 | 
			
		||||
        ("authentik_core", "0047_delete_oldauthenticatedsession"),
 | 
			
		||||
        ("authentik_sources_ldap", "0007_ldapsource_lookup_groups_from_user"),
 | 
			
		||||
    ]
 | 
			
		||||
 | 
			
		||||
    operations = [
 | 
			
		||||
        migrations.CreateModel(
 | 
			
		||||
            name="GroupLDAPSourceConnection",
 | 
			
		||||
            fields=[
 | 
			
		||||
                (
 | 
			
		||||
                    "groupsourceconnection_ptr",
 | 
			
		||||
                    models.OneToOneField(
 | 
			
		||||
                        auto_created=True,
 | 
			
		||||
                        on_delete=django.db.models.deletion.CASCADE,
 | 
			
		||||
                        parent_link=True,
 | 
			
		||||
                        primary_key=True,
 | 
			
		||||
                        serialize=False,
 | 
			
		||||
                        to="authentik_core.groupsourceconnection",
 | 
			
		||||
                    ),
 | 
			
		||||
                ),
 | 
			
		||||
            ],
 | 
			
		||||
            options={
 | 
			
		||||
                "verbose_name": "Group LDAP Source Connection",
 | 
			
		||||
                "verbose_name_plural": "Group LDAP Source Connections",
 | 
			
		||||
            },
 | 
			
		||||
            bases=("authentik_core.groupsourceconnection",),
 | 
			
		||||
        ),
 | 
			
		||||
        migrations.CreateModel(
 | 
			
		||||
            name="UserLDAPSourceConnection",
 | 
			
		||||
            fields=[
 | 
			
		||||
                (
 | 
			
		||||
                    "usersourceconnection_ptr",
 | 
			
		||||
                    models.OneToOneField(
 | 
			
		||||
                        auto_created=True,
 | 
			
		||||
                        on_delete=django.db.models.deletion.CASCADE,
 | 
			
		||||
                        parent_link=True,
 | 
			
		||||
                        primary_key=True,
 | 
			
		||||
                        serialize=False,
 | 
			
		||||
                        to="authentik_core.usersourceconnection",
 | 
			
		||||
                    ),
 | 
			
		||||
                ),
 | 
			
		||||
            ],
 | 
			
		||||
            options={
 | 
			
		||||
                "verbose_name": "User LDAP Source Connection",
 | 
			
		||||
                "verbose_name_plural": "User LDAP Source Connections",
 | 
			
		||||
            },
 | 
			
		||||
            bases=("authentik_core.usersourceconnection",),
 | 
			
		||||
        ),
 | 
			
		||||
    ]
 | 
			
		||||
@ -15,7 +15,13 @@ from ldap3 import ALL, NONE, RANDOM, Connection, Server, ServerPool, Tls
 | 
			
		||||
from ldap3.core.exceptions import LDAPException, LDAPInsufficientAccessRightsResult, LDAPSchemaError
 | 
			
		||||
from rest_framework.serializers import Serializer
 | 
			
		||||
 | 
			
		||||
from authentik.core.models import Group, PropertyMapping, Source
 | 
			
		||||
from authentik.core.models import (
 | 
			
		||||
    Group,
 | 
			
		||||
    GroupSourceConnection,
 | 
			
		||||
    PropertyMapping,
 | 
			
		||||
    Source,
 | 
			
		||||
    UserSourceConnection,
 | 
			
		||||
)
 | 
			
		||||
from authentik.crypto.models import CertificateKeyPair
 | 
			
		||||
from authentik.lib.config import CONFIG
 | 
			
		||||
from authentik.lib.models import DomainlessURLValidator
 | 
			
		||||
@ -123,6 +129,14 @@ class LDAPSource(Source):
 | 
			
		||||
        Group, blank=True, null=True, default=None, on_delete=models.SET_DEFAULT
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
    lookup_groups_from_user = models.BooleanField(
 | 
			
		||||
        default=False,
 | 
			
		||||
        help_text=_(
 | 
			
		||||
            "Lookup group membership based on a user attribute instead of a group attribute. "
 | 
			
		||||
            "This allows nested group resolution on systems like FreeIPA and Active Directory"
 | 
			
		||||
        ),
 | 
			
		||||
    )
 | 
			
		||||
 | 
			
		||||
    @property
 | 
			
		||||
    def component(self) -> str:
 | 
			
		||||
        return "ak-source-ldap-form"
 | 
			
		||||
@ -304,3 +318,31 @@ class LDAPSourcePropertyMapping(PropertyMapping):
 | 
			
		||||
    class Meta:
 | 
			
		||||
        verbose_name = _("LDAP Source Property Mapping")
 | 
			
		||||
        verbose_name_plural = _("LDAP Source Property Mappings")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class UserLDAPSourceConnection(UserSourceConnection):
 | 
			
		||||
    @property
 | 
			
		||||
    def serializer(self) -> type[Serializer]:
 | 
			
		||||
        from authentik.sources.ldap.api import (
 | 
			
		||||
            UserLDAPSourceConnectionSerializer,
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        return UserLDAPSourceConnectionSerializer
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
        verbose_name = _("User LDAP Source Connection")
 | 
			
		||||
        verbose_name_plural = _("User LDAP Source Connections")
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
class GroupLDAPSourceConnection(GroupSourceConnection):
 | 
			
		||||
    @property
 | 
			
		||||
    def serializer(self) -> type[Serializer]:
 | 
			
		||||
        from authentik.sources.ldap.api import (
 | 
			
		||||
            GroupLDAPSourceConnectionSerializer,
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        return GroupLDAPSourceConnectionSerializer
 | 
			
		||||
 | 
			
		||||
    class Meta:
 | 
			
		||||
        verbose_name = _("Group LDAP Source Connection")
 | 
			
		||||
        verbose_name_plural = _("Group LDAP Source Connections")
 | 
			
		||||
 | 
			
		||||
@ -14,7 +14,12 @@ from authentik.core.models import Group
 | 
			
		||||
from authentik.core.sources.mapper import SourceMapper
 | 
			
		||||
from authentik.events.models import Event, EventAction
 | 
			
		||||
from authentik.lib.sync.outgoing.exceptions import StopSync
 | 
			
		||||
from authentik.sources.ldap.models import LDAP_UNIQUENESS, LDAPSource, flatten
 | 
			
		||||
from authentik.sources.ldap.models import (
 | 
			
		||||
    LDAP_UNIQUENESS,
 | 
			
		||||
    GroupLDAPSourceConnection,
 | 
			
		||||
    LDAPSource,
 | 
			
		||||
    flatten,
 | 
			
		||||
)
 | 
			
		||||
from authentik.sources.ldap.sync.base import BaseLDAPSynchronizer
 | 
			
		||||
 | 
			
		||||
 | 
			
		||||
@ -89,6 +94,12 @@ class GroupLDAPSynchronizer(BaseLDAPSynchronizer):
 | 
			
		||||
                    defaults,
 | 
			
		||||
                )
 | 
			
		||||
                self._logger.debug("Created group with attributes", **defaults)
 | 
			
		||||
                if not GroupLDAPSourceConnection.objects.filter(
 | 
			
		||||
                    source=self._source, identifier=uniq
 | 
			
		||||
                ):
 | 
			
		||||
                    GroupLDAPSourceConnection.objects.create(
 | 
			
		||||
                        source=self._source, group=ak_group, identifier=uniq
 | 
			
		||||
                    )
 | 
			
		||||
            except SkipObjectException:
 | 
			
		||||
                continue
 | 
			
		||||
            except PropertyMappingExpressionException as exc:
 | 
			
		||||
 | 
			
		||||
Some files were not shown because too many files have changed in this diff Show More
		Reference in New Issue
	
	Block a user