30 lines
547 B
Go
30 lines
547 B
Go
// Copyright 2014 The DST Authors. All rights reserved.
|
|
// Use of this source code is governed by an MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package dst
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"crypto/sha256"
|
|
"encoding/binary"
|
|
"net"
|
|
)
|
|
|
|
var cookieKey = make([]byte, 16)
|
|
|
|
func init() {
|
|
_, err := rand.Reader.Read(cookieKey)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
}
|
|
|
|
func cookie(remote net.Addr) uint32 {
|
|
hash := sha256.New()
|
|
hash.Write([]byte(remote.String()))
|
|
hash.Write(cookieKey)
|
|
bs := hash.Sum(nil)
|
|
return binary.BigEndian.Uint32(bs)
|
|
}
|