Skip to content

Commit

Permalink
Added test for acheck_password() to ensure make_password is called fo…
Browse files Browse the repository at this point in the history
…r unusable passwords.

This is a follow up for the fix of CVE-2024-39329
(5d86458) where the timing of
verify_password() was standardized when checking unusable passwords.
  • Loading branch information
nessita committed Aug 8, 2024
1 parent f8ef457 commit e1606d2
Showing 1 changed file with 41 additions and 20 deletions.
61 changes: 41 additions & 20 deletions tests/auth_tests/test_hashers.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
from contextlib import contextmanager
from unittest import mock, skipUnless

from django.conf.global_settings import PASSWORD_HASHERS
Expand Down Expand Up @@ -452,8 +453,33 @@ def test_check_password_calls_harden_runtime(self):
check_password("wrong_password", encoded)
self.assertEqual(hasher.harden_runtime.call_count, 1)

def test_check_password_calls_make_password_to_fake_runtime(self):
@contextmanager
def assertMakePasswordCalled(self, password, encoded, hasher_side_effect):
hasher = get_hasher("default")
with (
mock.patch(
"django.contrib.auth.hashers.identify_hasher",
side_effect=hasher_side_effect,
) as mock_identify_hasher,
mock.patch(
"django.contrib.auth.hashers.make_password"
) as mock_make_password,
mock.patch(
"django.contrib.auth.hashers.get_random_string",
side_effect=lambda size: "x" * size,
),
mock.patch.object(hasher, "verify"),
):
# Ensure make_password is called to standardize timing.
yield
self.assertEqual(hasher.verify.call_count, 0)
self.assertEqual(mock_identify_hasher.mock_calls, [mock.call(encoded)])
self.assertEqual(
mock_make_password.mock_calls,
[mock.call("x" * UNUSABLE_PASSWORD_SUFFIX_LENGTH)],
)

def test_check_password_calls_make_password_to_fake_runtime(self):
cases = [
(None, None, None), # no plain text password provided
("foo", make_password(password=None), None), # unusable encoded
Expand All @@ -462,27 +488,22 @@ def test_check_password_calls_make_password_to_fake_runtime(self):
for password, encoded, hasher_side_effect in cases:
with (
self.subTest(encoded=encoded),
mock.patch(
"django.contrib.auth.hashers.identify_hasher",
side_effect=hasher_side_effect,
) as mock_identify_hasher,
mock.patch(
"django.contrib.auth.hashers.make_password"
) as mock_make_password,
mock.patch(
"django.contrib.auth.hashers.get_random_string",
side_effect=lambda size: "x" * size,
),
mock.patch.object(hasher, "verify"),
self.assertMakePasswordCalled(password, encoded, hasher_side_effect),
):
# Ensure make_password is called to standardize timing.
check_password(password, encoded)
self.assertEqual(hasher.verify.call_count, 0)
self.assertEqual(mock_identify_hasher.mock_calls, [mock.call(encoded)])
self.assertEqual(
mock_make_password.mock_calls,
[mock.call("x" * UNUSABLE_PASSWORD_SUFFIX_LENGTH)],
)

async def test_acheck_password_calls_make_password_to_fake_runtime(self):
cases = [
(None, None, None), # no plain text password provided
("foo", make_password(password=None), None), # unusable encoded
("letmein", make_password(password="letmein"), ValueError), # valid encoded
]
for password, encoded, hasher_side_effect in cases:
with (
self.subTest(encoded=encoded),
self.assertMakePasswordCalled(password, encoded, hasher_side_effect),
):
await acheck_password(password, encoded)

def test_encode_invalid_salt(self):
hasher_classes = [
Expand Down

0 comments on commit e1606d2

Please sign in to comment.