/
spnego.go
61 lines (49 loc) · 1.19 KB
/
spnego.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
package httpauth
import (
"encoding/asn1"
"encoding/base64"
"fmt"
"log"
"net/url"
"reflect"
"strings"
"github.com/jcmturner/gokrb5/v8/gssapi"
)
const (
NTLMSSP_NAME string = "NTLMSSP"
SPNEGO_NAME string = string(gssapi.OIDSPNEGO)
)
type SpnegoProvider interface {
GetToken(url *url.URL, responseToken string) (string, bool, error)
Close() error
SetLogger(logger *log.Logger)
}
func IsNTLMToken(token string) bool {
isNtlm := strings.Contains(token, "TlRMTVNTU")
return isNtlm
}
func GetMechanismsFromHttpFieldValue(token string) ([]string, error) {
var result []string
var err error
if strings.Contains(token, " ") {
temp := strings.Split(token, " ")
token = temp[1]
}
if IsNTLMToken(token) {
result = append(result, NTLMSSP_NAME)
} else {
var decodedToken []byte
decodedToken, err = base64.StdEncoding.DecodeString(token)
if err == nil {
var oid asn1.ObjectIdentifier
_, err = asn1.UnmarshalWithParams(decodedToken, &oid, fmt.Sprintf("application,explicit,tag:%v", 0))
if reflect.DeepEqual(oid, asn1.ObjectIdentifier(gssapi.OIDSPNEGO.OID())) {
result = append(result, SPNEGO_NAME)
}
}
if err != nil {
fmt.Println(err)
}
}
return result, err
}