Elia Battiston | 4750d3c | 2022-07-14 13:24:56 +0000 | [diff] [blame] | 1 | package types |
| 2 | |
| 3 | import ( |
| 4 | "strings" |
| 5 | |
| 6 | "github.com/jcmturner/gokrb5/v8/iana/nametype" |
| 7 | ) |
| 8 | |
| 9 | // Reference: https://www.ietf.org/rfc/rfc4120.txt |
| 10 | // Section: 5.2.2 |
| 11 | |
| 12 | // PrincipalName implements RFC 4120 type: https://tools.ietf.org/html/rfc4120#section-5.2.2 |
| 13 | type PrincipalName struct { |
| 14 | NameType int32 `asn1:"explicit,tag:0"` |
| 15 | NameString []string `asn1:"generalstring,explicit,tag:1"` |
| 16 | } |
| 17 | |
| 18 | // NewPrincipalName creates a new PrincipalName from the name type int32 and name string provided. |
| 19 | func NewPrincipalName(ntype int32, spn string) PrincipalName { |
| 20 | return PrincipalName{ |
| 21 | NameType: ntype, |
| 22 | NameString: strings.Split(spn, "/"), |
| 23 | } |
| 24 | } |
| 25 | |
| 26 | // GetSalt returns a salt derived from the PrincipalName. |
| 27 | func (pn PrincipalName) GetSalt(realm string) string { |
| 28 | var sb []byte |
| 29 | sb = append(sb, realm...) |
| 30 | for _, n := range pn.NameString { |
| 31 | sb = append(sb, n...) |
| 32 | } |
| 33 | return string(sb) |
| 34 | } |
| 35 | |
| 36 | // Equal tests if the PrincipalName is equal to the one provided. |
| 37 | func (pn PrincipalName) Equal(n PrincipalName) bool { |
| 38 | if len(pn.NameString) != len(n.NameString) { |
| 39 | return false |
| 40 | } |
| 41 | //https://tools.ietf.org/html/rfc4120#section-6.2 - the name type is not significant when checking for equivalence |
| 42 | for i, s := range pn.NameString { |
| 43 | if n.NameString[i] != s { |
| 44 | return false |
| 45 | } |
| 46 | } |
| 47 | return true |
| 48 | } |
| 49 | |
| 50 | // PrincipalNameString returns the PrincipalName in string form. |
| 51 | func (pn PrincipalName) PrincipalNameString() string { |
| 52 | return strings.Join(pn.NameString, "/") |
| 53 | } |
| 54 | |
| 55 | // ParseSPNString will parse a string in the format <service>/<name>@<realm> |
| 56 | // a PrincipalName type will be returned with the name type set to KRB_NT_PRINCIPAL(1) |
| 57 | // and the realm will be returned as a string. If the "@<realm>" suffix |
| 58 | // is not included in the SPN then the value of realm string returned will be "" |
| 59 | func ParseSPNString(spn string) (pn PrincipalName, realm string) { |
| 60 | if strings.Contains(spn, "@") { |
| 61 | s := strings.Split(spn, "@") |
| 62 | realm = s[len(s)-1] |
| 63 | spn = strings.TrimSuffix(spn, "@"+realm) |
| 64 | } |
| 65 | pn = NewPrincipalName(nametype.KRB_NT_PRINCIPAL, spn) |
| 66 | return |
| 67 | } |