38 lines
746 B
Go
38 lines
746 B
Go
|
package ntlmssp
|
||
|
|
|
||
|
|
import (
|
||
|
|
"encoding/base64"
|
||
|
|
"strings"
|
||
|
|
)
|
||
|
|
|
||
|
|
type authheader string
|
||
|
|
|
||
|
|
func (h authheader) IsBasic() bool {
|
||
|
|
return strings.HasPrefix(string(h), "Basic ")
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h authheader) IsNegotiate() bool {
|
||
|
|
return strings.HasPrefix(string(h), "Negotiate")
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h authheader) IsNTLM() bool {
|
||
|
|
return strings.HasPrefix(string(h), "NTLM")
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h authheader) GetData() ([]byte, error) {
|
||
|
|
p := strings.Split(string(h), " ")
|
||
|
|
if len(p) < 2 {
|
||
|
|
return nil, nil
|
||
|
|
}
|
||
|
|
return base64.StdEncoding.DecodeString(string(p[1]))
|
||
|
|
}
|
||
|
|
|
||
|
|
func (h authheader) GetBasicCreds() (username, password string, err error) {
|
||
|
|
d, err := h.GetData()
|
||
|
|
if err != nil {
|
||
|
|
return "", "", err
|
||
|
|
}
|
||
|
|
parts := strings.SplitN(string(d), ":", 2)
|
||
|
|
return parts[0], parts[1], nil
|
||
|
|
}
|