mirror of
https://github.com/Buriburizaem0n/nezha_domains.git
synced 2026-11-04 16:40:21 +00:00
fix: harden dashboard security boundaries
This commit is contained in:
1 parent
c726930833
commit
36b742a7e9
12 files changed
+357
-46
No files matched your search
@@ -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)
|
||||
|
||||
Reference in new issue
Block a user