| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110 |
- // +build examples
- package main
- import (
- "encoding/hex"
- "fmt"
- "gopkg.in/jcmturner/goidentity.v3"
- "gopkg.in/jcmturner/gokrb5.v6/client"
- "gopkg.in/jcmturner/gokrb5.v6/config"
- "gopkg.in/jcmturner/gokrb5.v6/credentials"
- "gopkg.in/jcmturner/gokrb5.v6/keytab"
- "gopkg.in/jcmturner/gokrb5.v6/service"
- "gopkg.in/jcmturner/gokrb5.v6/testdata"
- "io/ioutil"
- "log"
- "net/http"
- "net/http/httptest"
- "os"
- )
- func main() {
- s := httpServer()
- defer s.Close()
- b, _ := hex.DecodeString(testdata.TESTUSER1_KEYTAB)
- kt, _ := keytab.Parse(b)
- c, _ := config.NewConfigFromString(testdata.TEST_KRB5CONF)
- cl := client.NewClientWithKeytab("testuser1", "USER.GOKRB5", kt)
- cl.WithConfig(c)
- httpRequest(s.URL, cl)
- b, _ = hex.DecodeString(testdata.TESTUSER2_KEYTAB)
- kt, _ = keytab.Parse(b)
- c, _ = config.NewConfigFromString(testdata.TEST_KRB5CONF)
- cl = client.NewClientWithKeytab("testuser2", "USER.GOKRB5", kt)
- cl.WithConfig(c)
- httpRequest(s.URL, cl)
- //httpRequest("http://host.test.gokrb5/index.html")
- }
- 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.Printf("Error on AS_REQ: %v\n", err)
- }
- r, _ := http.NewRequest("GET", url, nil)
- err = cl.SetSPNEGOHeader(r, "HTTP/host.res.gokrb5")
- if err != nil {
- l.Printf("Error setting client SPNEGO header: %v", err)
- }
- httpResp, err := http.DefaultClient.Do(r)
- if err != nil {
- l.Printf("Request error: %v\n", err)
- }
- fmt.Fprintf(os.Stdout, "Response Code: %v\n", httpResp.StatusCode)
- content, _ := ioutil.ReadAll(httpResp.Body)
- fmt.Fprintf(os.Stdout, "Response Body:\n%s\n", content)
- }
- func httpServer() *httptest.Server {
- l := log.New(os.Stderr, "GOKRB5 Service: ", log.Ldate|log.Ltime|log.Lshortfile)
- b, _ := hex.DecodeString(testdata.SYSHTTP_KEYTAB)
- kt, _ := keytab.Parse(b)
- th := http.HandlerFunc(testAppHandler)
- c := service.NewSPNEGOConfig(kt)
- c.ServicePrincipal = "sysHTTP"
- s := httptest.NewServer(service.SPNEGOKRB5Authenticate(th, c, l))
- return s
- }
- func testAppHandler(w http.ResponseWriter, r *http.Request) {
- ctx := r.Context()
- fmt.Fprint(w, "<html>\n<p><h1>TEST.GOKRB5 Handler</h1></p>\n")
- if validuser, ok := ctx.Value(service.CTXKeyAuthenticated).(bool); ok && validuser {
- if creds, ok := ctx.Value(service.CTXKeyCredentials).(goidentity.Identity); ok {
- fmt.Fprintf(w, "<ul><li>Authenticed user: %s</li>\n", creds.UserName())
- fmt.Fprintf(w, "<li>User's realm: %s</li>\n", creds.Domain())
- fmt.Fprint(w, "<li>Authz Attributes (Group Memberships):</li><ul>\n")
- for _, s := range creds.AuthzAttributes() {
- fmt.Fprintf(w, "<li>%v</li>\n", s)
- }
- fmt.Fprint(w, "</ul>\n")
- if ADCreds, ok := creds.Attributes()[credentials.AttributeKeyADCredentials].(credentials.ADCredentials); ok {
- // Now access the fields of the ADCredentials struct. For example:
- fmt.Fprintf(w, "<li>EffectiveName: %v</li>\n", ADCreds.EffectiveName)
- fmt.Fprintf(w, "<li>FullName: %v</li>\n", ADCreds.FullName)
- fmt.Fprintf(w, "<li>UserID: %v</li>\n", ADCreds.UserID)
- fmt.Fprintf(w, "<li>PrimaryGroupID: %v</li>\n", ADCreds.PrimaryGroupID)
- fmt.Fprintf(w, "<li>Group SIDs: %v</li>\n", ADCreds.GroupMembershipSIDs)
- fmt.Fprintf(w, "<li>LogOnTime: %v</li>\n", ADCreds.LogOnTime)
- fmt.Fprintf(w, "<li>LogOffTime: %v</li>\n", ADCreds.LogOffTime)
- fmt.Fprintf(w, "<li>PasswordLastSet: %v</li>\n", ADCreds.PasswordLastSet)
- fmt.Fprintf(w, "<li>LogonServer: %v</li>\n", ADCreds.LogonServer)
- fmt.Fprintf(w, "<li>LogonDomainName: %v</li>\n", ADCreds.LogonDomainName)
- fmt.Fprintf(w, "<li>LogonDomainID: %v</li>\n", ADCreds.LogonDomainID)
- }
- fmt.Fprint(w, "</ul>")
- }
- } else {
- w.WriteHeader(http.StatusUnauthorized)
- fmt.Fprint(w, "Authentication failed")
- }
- fmt.Fprint(w, "</html>")
- return
- }
|
