Mercurial > gemma
view auth/session.go @ 200:8426a92fda00
Persistent session store: Implmented Renew.
author | Sascha L. Teichmann <teichmann@intevation.de> |
---|---|
date | Sun, 22 Jul 2018 09:25:25 +0200 |
parents | e85413e5befa |
children | f345edb409b2 |
line wrap: on
line source
package auth import ( "crypto/rand" "encoding/base64" "encoding/binary" "io" "time" ) type Session struct { ExpiresAt int64 `json:"expires"` User string `json:"user"` Password string `json:"password"` Roles []string `json:"roles"` } const ( sessionKeyLength = 20 maxTokenValid = time.Hour * 3 ) func NewSession(user, password string, roles []string) *Session { // Create the Claims return &Session{ ExpiresAt: time.Now().Add(maxTokenValid).Unix(), User: user, Password: password, Roles: roles, } } func (s *Session) serialize(w io.Writer) error { var err error write := func(data interface{}) { if err == nil { err = binary.Write(w, binary.BigEndian, data) } } write(s.ExpiresAt) write(s.User) write(s.Password) write(uint32(len(s.Roles))) for _, role := range s.Roles { write(role) } return err } func (s *Session) deserialize(r io.Reader) error { var err error read := func(data interface{}) { if err == nil { err = binary.Read(r, binary.BigEndian, data) } } var x Session var n uint32 read(&x.ExpiresAt) read(&x.User) read(&x.Password) read(&n) x.Roles = make([]string, n) for i := uint32(0); n > 0 && i < n; i++ { read(&x.Roles[i]) } if err == nil { *s = x } return err } func GenerateSessionKey() string { return base64.URLEncoding.EncodeToString(GenerateRandomKey(sessionKeyLength)) } func GenerateRandomKey(length int) []byte { k := make([]byte, length) if _, err := io.ReadFull(rand.Reader, k); err != nil { return nil } return k } func GenerateSession(user, password string) (string, *Session, error) { roles, err := AllOtherRoles(user, password) if err != nil { return "", nil, err } token := GenerateSessionKey() session := NewSession(user, password, roles) ConnPool.Add(token, session) return token, session, nil }