fix: harden dashboard security boundaries

This commit is contained in:
naiba committed 2026-10-01 04:28:34 +00:00
1 parent c726930833
commit 36b742a7e9
12 files changed
+357 -46

No files matched your search

+56 -3
View File
@@ -455,7 +455,8 @@ func TestServerTransferBroadcastReachesSubscribers(t *testing.T) {
defer cleanup()
seedServerForTransfer(t, 1, 100)
id, ch := c.Subscribe()
id, ch, err := c.Subscribe(100)
require.NoError(t, err)
defer c.Unsubscribe(id)
tr := initiateAndRegister(t, c, 1, 100, 200, 1)
@@ -469,7 +470,7 @@ func TestServerTransferBroadcastReachesSubscribers(t *testing.T) {
t.Fatal("expected Pending broadcast within 1s")
}
_, _, err := markPendingVerified(t, c, 1)
_, _, err = markPendingVerified(t, c, 1)
require.NoError(t, err)
select {
case ev := <-ch:
@@ -479,6 +480,57 @@ func TestServerTransferBroadcastReachesSubscribers(t *testing.T) {
}
}
func TestServerTransferSubscribeEnforcesPerUserLimitAndReleasesQuota(t *testing.T) {
c, cleanup := setupTransferFixture(t)
defer cleanup()
ids := make([]uint64, 0, maxServerTransferSubscribersPerUser)
for range maxServerTransferSubscribersPerUser {
id, ch, err := c.Subscribe(100)
require.NoError(t, err)
require.NotNil(t, ch)
ids = append(ids, id)
}
_, _, err := c.Subscribe(100)
require.ErrorIs(t, err, errTooManyServerTransferSubscribersForUser)
otherID, _, err := c.Subscribe(200)
require.NoError(t, err, "one user's quota must not block another user")
c.Unsubscribe(otherID)
c.Unsubscribe(ids[0])
replacementID, _, err := c.Subscribe(100)
require.NoError(t, err, "disconnecting must release the user's quota")
c.Unsubscribe(replacementID)
for _, id := range ids[1:] {
c.Unsubscribe(id)
}
}
func TestServerTransferSubscribeEnforcesGlobalLimit(t *testing.T) {
c, cleanup := setupTransferFixture(t)
defer cleanup()
ids := make([]uint64, 0, maxServerTransferSubscribersGlobal)
for userID := uint64(1); len(ids) < maxServerTransferSubscribersGlobal; userID++ {
for range maxServerTransferSubscribersPerUser {
if len(ids) == maxServerTransferSubscribersGlobal {
break
}
id, _, err := c.Subscribe(userID)
require.NoError(t, err)
ids = append(ids, id)
}
}
_, _, err := c.Subscribe(999999)
require.ErrorIs(t, err, errTooManyServerTransferSubscribersGlobal)
for _, id := range ids {
c.Unsubscribe(id)
}
}
// The "one active transfer per server" invariant must hold under concurrent
// callers (two operators batch-moving the same server, or batch-move racing
// retry). The old flow had a TOCTOU between HasPending() and Initiate() that
@@ -1626,7 +1678,8 @@ func TestOnServersDeletedTerminatesPendingTransfersAndClearsIndexes(t *testing.T
tr := initiateAndRegister(t, c, 1, 100, 200, 1)
require.True(t, c.HasPending(1))
subID, ch := c.Subscribe()
subID, ch, err := c.Subscribe(100)
require.NoError(t, err)
defer c.Unsubscribe(subID)
require.NoError(t, DB.Unscoped().Delete(&model.Server{}, tr.ServerID).Error)