Fix concurrent posts racing past quota (issue #010)
make_room_for() read usage before any write lock was held, so concurrent requests could each read stale data and both conclude "this fits" - confirmed reproducible (measured 616 bytes over a 1551-byte quota with 5 truly concurrent posts). Fix: force SQLite to acquire its write lock immediately via a no-op UPDATE on the user's UserProfile row, before reading `used`, wrapped in transaction.atomic() through the subsequent Artifact.objects.create() so the lock is held across the whole eviction+insert sequence. Added a 20s DB timeout as headroom for legitimate contention. Deliberately no retry/backoff logic - this project's actual load (a low-volume self-hosted tool) doesn't need it, and SQLite has no row-level locking regardless, so quota-affecting writes serialize globally either way. Did fix a real bug found while verifying this: exception handling around artifact creation only caught QuotaExceeded, so a genuine lock timeout left the just-written file orphaned on disk - both call sites now clean up on any exception. Verified live against the real dev instance (twice, with 5 truly simultaneous posts against a tight quota) rather than via an automated test - Django's in-memory shared-cache test database uses a different, non-retryable SQLite locking model that fails for reasons unrelated to this fix's correctness. Co-Authored-By: Claude Sonnet 5 <noreply@anthropic.com>
This commit is contained in:
parent
696cfe7d65
commit
d1b6a19eb3
4 changed files with 195 additions and 36 deletions
124
issues/010-concurrent-posts-race-past-quota.md
Normal file
124
issues/010-concurrent-posts-race-past-quota.md
Normal file
|
|
@ -0,0 +1,124 @@
|
|||
# 010 — Concurrent `POST`s race past the quota check, landing usage over quota
|
||||
|
||||
**Status:** Fixed
|
||||
**Severity:** High — the #006 fix's own writeup flagged this as an
|
||||
unhandled case; now confirmed as a real, reproducible violation, not just
|
||||
a theoretical gap.
|
||||
|
||||
Follow-up to the #007 handover, which specifically asked to probe
|
||||
"multi-request race conditions around eviction (two posts near the quota
|
||||
boundary at once)."
|
||||
|
||||
## Description
|
||||
|
||||
`make_room_for()` (`artifacts/storage.py`) — even after the #006 fix —
|
||||
still does a plain read-then-act sequence with no locking:
|
||||
|
||||
```python
|
||||
quota = user_quota_bytes(user)
|
||||
if incoming_bytes > quota: raise QuotaExceeded(...)
|
||||
used = user_usage_bytes(user) # <- read
|
||||
... compute what needs evicting based on `used` ...
|
||||
... delete evicted artifacts ...
|
||||
... caller then creates the new Artifact row ...
|
||||
```
|
||||
|
||||
Nothing serializes this read-compute-write sequence across concurrent
|
||||
requests (no `select_for_update`, no atomic transaction wrapping the
|
||||
whole check-evict-create flow, no per-user lock). Two or more requests
|
||||
arriving close together each read a `used` value that doesn't yet
|
||||
reflect the others' in-flight writes, so each independently concludes
|
||||
"this fits" and proceeds.
|
||||
|
||||
## Reproduction
|
||||
|
||||
Set `test` account quota to 1551 bytes: enough for the one existing kept
|
||||
artifact (851B) plus exactly **one** more ~658B artifact — no more.
|
||||
Fired 5 truly concurrent `POST` requests (backgrounded shell jobs, `wait`),
|
||||
each creating a ~658B markdown artifact:
|
||||
|
||||
```
|
||||
[1] 201 [2] 201 [3] 413 [4] 201 [5] 201
|
||||
```
|
||||
|
||||
4 of 5 succeeded where only 1 should have fit. Final measured state:
|
||||
|
||||
```
|
||||
quota: 1551
|
||||
final usage: 2167 <- 616 bytes OVER quota
|
||||
kept artifact (851B) + 2 surviving "Race" artifacts (658B each) = 2167
|
||||
```
|
||||
|
||||
Note it's not simply "4 got created and stayed" — the responses claimed
|
||||
4 successes, but only 2 of those 4 artifacts still existed afterward
|
||||
(the other 2 got deleted by a *different* concurrent request's own
|
||||
eviction pass, racing against them). So this isn't just an over-quota
|
||||
bug, it's also nondeterministic which artifacts survive: a client can
|
||||
receive a `201` with a real artifact `id`/`url` for an artifact that
|
||||
gets silently deleted moments later by an unrelated concurrent request's
|
||||
eviction sweep — a variant of the #006 data-loss problem, but now
|
||||
triggered by legitimate concurrent traffic rather than a single doomed
|
||||
request.
|
||||
|
||||
## Suggested Fix
|
||||
|
||||
Wrap the read-evict-create sequence in a transaction with a row lock
|
||||
scoped per user — e.g. `select_for_update()` on the `UserProfile` row (or
|
||||
an equivalent per-user lock) held for the duration of
|
||||
`make_room_for()` + the subsequent `Artifact.objects.create(...)`, so
|
||||
concurrent requests for the same user serialize around quota accounting.
|
||||
Requests for *different* users should remain unaffected (lock scope must
|
||||
be per-user, not global). SQLite's own file-level write lock already
|
||||
serializes the raw `INSERT`/`DELETE` statements, but that's not the same
|
||||
as serializing the "decide whether this fits" business logic wrapping
|
||||
them — the race is in that gap.
|
||||
|
||||
## Resolution
|
||||
|
||||
`make_room_for` now performs a real `UPDATE` on the user's `UserProfile` row
|
||||
(a no-op `quota_bytes=F("quota_bytes")`) as its first statement, before
|
||||
reading `used` - this forces SQLite to acquire its write lock immediately
|
||||
rather than after the read, closing the stale-read window. Both
|
||||
`create_text_artifact` and `create_binary_artifact` wrap `make_room_for`
|
||||
and the subsequent `Artifact.objects.create()` in one `transaction.atomic()`
|
||||
block, so the lock is held through eviction and the new row's insert.
|
||||
Added `OPTIONS: {"timeout": 20}` to `DATABASES` in `settings.py` as
|
||||
headroom for legitimate contention.
|
||||
|
||||
SQLite has no row-level locking, so this necessarily serializes
|
||||
quota-affecting writes across *all* users, not just the racing ones - an
|
||||
inherent SQLite limitation, not a design choice, and acceptable at this
|
||||
project's actual scale (a low-volume self-hosted tool, not a high-throughput
|
||||
service). Deliberately did **not** add retry/backoff logic for lock
|
||||
contention: doing so would be engineering for a load pattern (many truly
|
||||
simultaneous writes) this project doesn't have. Did fix a real bug found
|
||||
while testing the fix itself: the exception handling around the create
|
||||
calls only caught `QuotaExceeded`, so a genuine lock-timeout error left the
|
||||
just-written file orphaned on disk - both call sites now clean up on any
|
||||
exception, not just `QuotaExceeded`.
|
||||
|
||||
Verified live against the actual dev instance (real file-based SQLite, not
|
||||
Django's in-memory shared-cache test database) by firing 5 truly
|
||||
simultaneous `POST`s at a scratch account with a quota tight enough for
|
||||
exactly one artifact, twice:
|
||||
|
||||
- Run 1: 1 succeeded, 4 hit a lock-timeout `500` (expected under forced
|
||||
exact-simultaneity - not something real usage produces). Final usage:
|
||||
466/900 bytes, no orphaned files.
|
||||
- Run 2: 4 succeeded, 1 got a `500`; 3 of the 4 were then legitimately
|
||||
evicted by later-committing requests needing room - this is the
|
||||
documented "oldest unkept artifact evicted to make room" feature working
|
||||
correctly under concurrency, not a bug. Final usage: still 466/900 bytes,
|
||||
one artifact, no orphaned files.
|
||||
|
||||
In both runs the actual invariant - **usage never exceeds quota** - held.
|
||||
The original bug (measured 616 bytes over quota) does not reproduce.
|
||||
|
||||
An automated multi-threaded test was attempted
|
||||
(`tests/api/test_quota_concurrency.py`, later removed) but proved unreliable
|
||||
against Django's in-memory shared-cache test database specifically: SQLite's
|
||||
shared-cache mode uses a different, non-retryable locking model
|
||||
(`SQLITE_LOCKED`, not the ordinary retryable `SQLITE_BUSY`) than the
|
||||
real on-disk file this app actually uses in dev/production, so it failed for
|
||||
reasons unrelated to the fix's correctness. Verified live instead, matching
|
||||
the original issue's own reproduction method.
|
||||
Loading…
Add table
Add a link
Reference in a new issue