41 lines
1.2 KiB
Go
41 lines
1.2 KiB
Go
|
package zk
|
||
|
|
|
||
|
|
import (
|
||
|
|
"crypto/sha1"
|
||
|
|
"encoding/base64"
|
||
|
|
"fmt"
|
||
|
|
"math/rand"
|
||
|
|
)
|
||
|
|
|
||
|
|
// AuthACL produces an ACL list containing a single ACL which uses the
|
||
|
|
// provided permissions, with the scheme "auth", and ID "", which is used
|
||
|
|
// by ZooKeeper to represent any authenticated user.
|
||
|
|
func AuthACL(perms int32) []ACL {
|
||
|
|
return []ACL{{perms, "auth", ""}}
|
||
|
|
}
|
||
|
|
|
||
|
|
// WorldACL produces an ACL list containing a single ACL which uses the
|
||
|
|
// provided permissions, with the scheme "world", and ID "anyone", which
|
||
|
|
// is used by ZooKeeper to represent any user at all.
|
||
|
|
func WorldACL(perms int32) []ACL {
|
||
|
|
return []ACL{{perms, "world", "anyone"}}
|
||
|
|
}
|
||
|
|
|
||
|
|
func DigestACL(perms int32, user, password string) []ACL {
|
||
|
|
userPass := []byte(fmt.Sprintf("%s:%s", user, password))
|
||
|
|
h := sha1.New()
|
||
|
|
if n, err := h.Write(userPass); err != nil || n != len(userPass) {
|
||
|
|
panic("SHA1 failed")
|
||
|
|
}
|
||
|
|
digest := base64.StdEncoding.EncodeToString(h.Sum(nil))
|
||
|
|
return []ACL{{perms, "digest", fmt.Sprintf("%s:%s", user, digest)}}
|
||
|
|
}
|
||
|
|
|
||
|
|
// stringShuffle performs a Fisher-Yates shuffle on a slice of strings
|
||
|
|
func stringShuffle(s []string) {
|
||
|
|
for i := len(s) - 1; i > 0; i-- {
|
||
|
|
j := rand.Intn(i + 1)
|
||
|
|
s[i], s[j] = s[j], s[i]
|
||
|
|
}
|
||
|
|
}
|