1. 1
2. 2
3. 3
4. 4
5. 5
6. 6
7. 7
8. 8
9. 9
10. 10
11. 11
12. 12
13. 13
14. 14
15. 15
16. 16
17. 17
18. 18
19. 19
20. 20
21. 21
22. 22
23. 23
24. 24
25. 25
26. 26
27. 27
28. 28
29. 29
30. 30
31. 31
32. 32
33. 33
34. 34
35. 35
36. 36

package secctx

import (
	"net/http"

	"github.com/gorilla/securecookie"
	"github.com/gorilla/sessions"
)

// CookieStore can store and retrieve SSPI context handles to/from an encrypted Cookie.
type CookieStore struct {
	store *sessions.CookieStore
}

// NewCookieStore creates a new CookieStore for storing and retrieving of SSPI context handles
// to/from encrypted Cookies
func NewCookieStore() *CookieStore {
	s := &CookieStore{}
	s.store = sessions.NewCookieStore([]byte(securecookie.GenerateRandomKey(32)))
	return s
}

// GetHandle retrieves a *websspi.CtxtHandle value from the store
func (s *CookieStore) GetHandle(r *http.Request) (interface{}, error) {
	session, _ := s.store.Get(r, "websspi")
	contextHandle := session.Values["contextHandle"]
	return contextHandle, nil
}

// SetHandle saves a *websspi.CtxtHandle value to the store
func (s *CookieStore) SetHandle(r *http.Request, w http.ResponseWriter, contextHandle interface{}) error {
	session, _ := s.store.Get(r, "websspi")
	session.Values["contextHandle"] = contextHandle
	err := session.Save(r, w)
	return err
}