//go:build examples
// +build examples
package main
import (
"encoding/hex"
"errors"
"fmt"
"io"
"log"
"net/http"
"net/http/httptest"
"os"
"github.com/gorilla/sessions"
"github.com/jcmturner/goidentity/v6"
"github.com/oiweiwei/gokrb5.fork/v9/client"
"github.com/oiweiwei/gokrb5.fork/v9/config"
"github.com/oiweiwei/gokrb5.fork/v9/credentials"
"github.com/oiweiwei/gokrb5.fork/v9/keytab"
"github.com/oiweiwei/gokrb5.fork/v9/service"
"github.com/oiweiwei/gokrb5.fork/v9/spnego"
"github.com/oiweiwei/gokrb5.fork/v9/test/testdata"
)
func main() {
s := httpServer()
defer s.Close()
fmt.Printf("Listening on %s\n", s.URL)
l := log.New(os.Stderr, "GOKRB5 Client: ", log.Ldate|log.Ltime|log.Lshortfile)
b, _ := hex.DecodeString(testdata.KEYTAB_TESTUSER1_USER_GOKRB5)
kt := keytab.New()
kt.Unmarshal(b)
c, _ := config.NewFromString(testdata.KRB5_CONF)
cl := client.NewWithKeytab("testuser1", "USER.GOKRB5", kt, c, client.DisablePAFXFAST(true), client.Logger(l))
httpRequest(s.URL, cl)
b, _ = hex.DecodeString(testdata.KEYTAB_TESTUSER2_USER_GOKRB5)
kt = keytab.New()
kt.Unmarshal(b)
c, _ = config.NewFromString(testdata.KRB5_CONF)
cl = client.NewWithKeytab("testuser2", "USER.GOKRB5", kt, c, client.DisablePAFXFAST(true), client.Logger(l))
httpRequest(s.URL, cl)
}
func httpRequest(url string, cl *client.Client) {
l := log.New(os.Stderr, "GOKRB5 Client: ", log.Ldate|log.Ltime|log.Lshortfile)
err := cl.Login()
if err != nil {
l.Fatalf("Error on AS_REQ: %v\n", err)
}
spnegoCl := spnego.NewClient(cl, nil, "HTTP/host.res.gokrb5")
// Make the request for the first time with no session
r, _ := http.NewRequest("GET", url, nil)
httpResp, err := spnegoCl.Do(r)
if err != nil {
l.Fatalf("error making request: %v", err)
}
fmt.Fprintf(os.Stdout, "Response Code: %v\n", httpResp.StatusCode)
content, _ := io.ReadAll(httpResp.Body)
fmt.Fprintf(os.Stdout, "Response Body:\n%s\n", content)
// Make the request again which should use the session
httpResp, err = spnegoCl.Do(r)
if err != nil {
l.Fatalf("error making request: %v", err)
}
fmt.Fprintf(os.Stdout, "Response Code: %v\n", httpResp.StatusCode)
content, _ = io.ReadAll(httpResp.Body)
fmt.Fprintf(os.Stdout, "Response Body:\n%s\n", content)
}
func httpServer() *httptest.Server {
l := log.New(os.Stderr, "GOKRB5 Service Tests: ", log.Ldate|log.Ltime|log.Lshortfile)
b, _ := hex.DecodeString(testdata.KEYTAB_SYSHTTP_RES_GOKRB5)
kt := keytab.New()
kt.Unmarshal(b)
th := http.HandlerFunc(testAppHandler)
s := httptest.NewServer(spnego.SPNEGOKRB5Authenticate(th, kt, service.Logger(l), service.KeytabPrincipal("sysHTTP"), service.SessionManager(NewSessionMgr("gokrb5"))))
return s
}
func testAppHandler(w http.ResponseWriter, r *http.Request) {
creds := goidentity.FromHTTPRequestContext(r)
fmt.Fprint(w, "\nGOKRB5 Handler
\n")
if creds != nil && creds.Authenticated() {
fmt.Fprintf(w, "- Authenticed user: %s
\n", creds.UserName())
fmt.Fprintf(w, "- User's realm: %s
\n", creds.Domain())
fmt.Fprint(w, "- Authz Attributes (Group Memberships):
\n")
for _, s := range creds.AuthzAttributes() {
fmt.Fprintf(w, "- %v
\n", s)
}
fmt.Fprint(w, "
\n")
if ADCredsJSON, ok := creds.Attributes()[credentials.AttributeKeyADCredentials]; ok {
//ADCreds := new(credentials.ADCredentials)
ADCreds := ADCredsJSON.(credentials.ADCredentials)
//err := json.Unmarshal(aj, ADCreds)
//if err == nil {
// Now access the fields of the ADCredentials struct. For example:
fmt.Fprintf(w, "- EffectiveName: %v
\n", ADCreds.EffectiveName)
fmt.Fprintf(w, "- FullName: %v
\n", ADCreds.FullName)
fmt.Fprintf(w, "- UserID: %v
\n", ADCreds.UserID)
fmt.Fprintf(w, "- PrimaryGroupID: %v
\n", ADCreds.PrimaryGroupID)
fmt.Fprintf(w, "- Group SIDs: %v
\n", ADCreds.GroupMembershipSIDs)
fmt.Fprintf(w, "- LogOnTime: %v
\n", ADCreds.LogOnTime)
fmt.Fprintf(w, "- LogOffTime: %v
\n", ADCreds.LogOffTime)
fmt.Fprintf(w, "- PasswordLastSet: %v
\n", ADCreds.PasswordLastSet)
fmt.Fprintf(w, "- LogonServer: %v
\n", ADCreds.LogonServer)
fmt.Fprintf(w, "- LogonDomainName: %v
\n", ADCreds.LogonDomainName)
fmt.Fprintf(w, "- LogonDomainID: %v
\n", ADCreds.LogonDomainID)
//}
}
fmt.Fprint(w, "
")
} else {
w.WriteHeader(http.StatusUnauthorized)
fmt.Fprint(w, "Authentication failed")
}
fmt.Fprint(w, "")
return
}
type SessionMgr struct {
skey []byte
store sessions.Store
cookieName string
}
func NewSessionMgr(cookieName string) SessionMgr {
skey := []byte("thisistestsecret") // Best practice is to load this key from a secure location.
return SessionMgr{
skey: skey,
store: sessions.NewCookieStore(skey),
cookieName: cookieName,
}
}
func (smgr SessionMgr) Get(r *http.Request, k string) ([]byte, error) {
s, err := smgr.store.Get(r, smgr.cookieName)
if err != nil {
return nil, err
}
if s == nil {
return nil, errors.New("nil session")
}
b, ok := s.Values[k].([]byte)
if !ok {
return nil, fmt.Errorf("could not get bytes held in session at %s", k)
}
return b, nil
}
func (smgr SessionMgr) New(w http.ResponseWriter, r *http.Request, k string, v []byte) error {
s, err := smgr.store.New(r, smgr.cookieName)
if err != nil {
return fmt.Errorf("could not get new session from session manager: %v", err)
}
s.Values[k] = v
return s.Save(r, w)
}