TheArchive/tailscale
1
0
Fork 0
mirror of https://github.com/tailscale/tailscale.git synced 2025-10-25 10:09:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

stun: check high bits in Is, add tests

Browse Source 
Also use new stun.TxID type in stunner.

Signed-off-by: Brad Fitzpatrick <bradfitz@tailscale.com>
This commit is contained in:
Brad Fitzpatrick
2020-02-26 11:34:01 -08:00
parent 2489ea4268
commit 14abc82033
3 changed files with 32 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
stunner/stunner.go
View File

@@ -40,7 +40,7 @@ type Stunner struct {
type session struct {
replied chan struct{} // closed when server responds
tIDs [][12]byte // transaction IDs sent to a server
tIDs []stun.TxID // transaction IDs sent to a server
}
// Receive delivers a STUN packet to the stunner.
@@ -90,7 +90,7 @@ func (s *Stunner) Run(ctx context.Context) error {
}
for _, server := range s.Servers {
// Generate the transaction IDs for this session.
tIDs := make([][12]byte, len(retryDurations))
tIDs := make([]stun.TxID, len(retryDurations))
for i := range tIDs {
if _, err := rand.Read(tIDs[i][:]); err != nil {
return fmt.Errorf("stunner: rand failed: %v", err)
@@ -147,7 +147,7 @@ func (s *Stunner) runServer(ctx context.Context, server string) {
}
}
func (s *Stunner) sendSTUN(ctx context.Context, tID [12]byte, server string) error {
func (s *Stunner) sendSTUN(ctx context.Context, tID stun.TxID, server string) error {
host, port, err := net.SplitHostPort(server)
if err != nil {
return err