2017-10-10 09:10:02 +00:00
|
|
|
package cookie
|
|
|
|
|
|
|
|
import (
|
|
|
|
"crypto/sha1"
|
|
|
|
"fmt"
|
|
|
|
"strings"
|
|
|
|
|
2019-08-03 01:58:23 +00:00
|
|
|
"github.com/containous/traefik/v2/pkg/log"
|
2017-10-10 09:10:02 +00:00
|
|
|
)
|
|
|
|
|
|
|
|
const cookieNameLength = 6
|
|
|
|
|
2020-05-11 10:06:07 +00:00
|
|
|
// GetName of a cookie.
|
2020-07-07 12:42:03 +00:00
|
|
|
func GetName(cookieName, backendName string) string {
|
2017-10-10 09:10:02 +00:00
|
|
|
if len(cookieName) != 0 {
|
|
|
|
return sanitizeName(cookieName)
|
|
|
|
}
|
|
|
|
|
|
|
|
return GenerateName(backendName)
|
|
|
|
}
|
|
|
|
|
2020-05-11 10:06:07 +00:00
|
|
|
// GenerateName Generate a hashed name.
|
2017-10-10 09:10:02 +00:00
|
|
|
func GenerateName(backendName string) string {
|
|
|
|
data := []byte("_TRAEFIK_BACKEND_" + backendName)
|
|
|
|
|
|
|
|
hash := sha1.New()
|
|
|
|
_, err := hash.Write(data)
|
|
|
|
if err != nil {
|
|
|
|
// Impossible case
|
2019-09-13 17:28:04 +00:00
|
|
|
log.WithoutContext().Errorf("Fail to create cookie name: %v", err)
|
2017-10-10 09:10:02 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
return fmt.Sprintf("_%x", hash.Sum(nil))[:cookieNameLength]
|
|
|
|
}
|
|
|
|
|
2020-05-11 10:06:07 +00:00
|
|
|
// sanitizeName According to [RFC 2616](https://www.ietf.org/rfc/rfc2616.txt) section 2.2.
|
2017-10-10 09:10:02 +00:00
|
|
|
func sanitizeName(backend string) string {
|
|
|
|
sanitizer := func(r rune) rune {
|
|
|
|
switch r {
|
|
|
|
case '!', '#', '$', '%', '&', '\'', '*', '+', '-', '.', '^', '`', '|', '~':
|
|
|
|
return r
|
|
|
|
}
|
|
|
|
|
|
|
|
switch {
|
|
|
|
case 'a' <= r && r <= 'z':
|
|
|
|
fallthrough
|
|
|
|
case 'A' <= r && r <= 'Z':
|
|
|
|
fallthrough
|
|
|
|
case '0' <= r && r <= '9':
|
|
|
|
return r
|
|
|
|
default:
|
|
|
|
return '_'
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
return strings.Map(sanitizer, backend)
|
|
|
|
}
|