mirror of
https://github.com/juanfont/headscale.git
synced 2026-10-31 18:54:11 +09:00
Kristoffer Dalby
044d8f5a01
Bounds the DERP HTTP client's error body read. DERP region IDs are now
tailcfg.DERPRegionID.
(cherry picked from commit 3aece6501, adapted)
29 lines
868 B
Go
29 lines
868 B
Go
package derp
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"tailscale.com/tailcfg"
|
|
)
|
|
|
|
// TestMergeDERPMapsClonesRegions ensures merged DERP maps own their regions
|
|
// rather than aliasing the source pointers, so a later in-place node shuffle
|
|
// cannot mutate a shared or previously served map.
|
|
func TestMergeDERPMapsClonesRegions(t *testing.T) {
|
|
src := &tailcfg.DERPMap{
|
|
Regions: map[tailcfg.DERPRegionID]*tailcfg.DERPRegion{
|
|
1: {RegionID: 1, Nodes: []*tailcfg.DERPNode{{Name: "a"}, {Name: "b"}}},
|
|
},
|
|
}
|
|
|
|
merged := mergeDERPMaps([]*tailcfg.DERPMap{src})
|
|
|
|
assert.NotSame(t, src.Regions[1], merged.Regions[1],
|
|
"merged region must not alias the source region pointer")
|
|
|
|
merged.Regions[1].Nodes[0] = &tailcfg.DERPNode{Name: "mutated"}
|
|
assert.Equal(t, "a", src.Regions[1].Nodes[0].Name,
|
|
"source region was mutated through a shared pointer")
|
|
}
|