go-xmpp/jid.go

81 lines
1.6 KiB
Go
Raw Normal View History

package xmpp // import "gosrc.io/xmpp"
import (
2019-06-07 09:40:10 +00:00
"fmt"
"strings"
"unicode"
)
type Jid struct {
username string
domain string
resource string
}
func NewJid(sjid string) (*Jid, error) {
jid := new(Jid)
2019-06-07 09:40:10 +00:00
if sjid == "" {
return jid, fmt.Errorf("jid cannot be empty")
}
2019-06-07 09:40:10 +00:00
s1 := strings.SplitN(sjid, "@", 2)
if len(s1) == 1 { // This is a server or component JID
jid.domain = s1[0]
} else { // JID has a local username part
if s1[0] == "" {
return jid, fmt.Errorf("invalid jid '%s", sjid)
}
jid.username = s1[0]
if s1[1] == "" {
return jid, fmt.Errorf("domain cannot be empty")
}
jid.domain = s1[1]
}
2019-06-07 09:40:10 +00:00
// Extract resource from domain field
s2 := strings.SplitN(jid.domain, "/", 2)
if len(s2) == 2 { // If len = 1, domain is already correct, and resource is already empty
jid.domain = s2[0]
jid.resource = s2[1]
}
2019-06-07 09:40:10 +00:00
if !isUsernameValid(jid.username) {
return jid, fmt.Errorf("invalid username in JID '%s'", sjid)
}
if !isDomainValid(jid.domain) {
return jid, fmt.Errorf("invalid domain in JID '%s'", sjid)
}
return jid, nil
}
func isUsernameValid(username string) bool {
invalidRunes := []rune{'@', '/', '\'', '"', ':', '<', '>'}
return strings.IndexFunc(username, isInvalid(invalidRunes)) < 0
}
func isDomainValid(domain string) bool {
2019-06-07 09:40:10 +00:00
if len(domain) == 0 {
return false
}
invalidRunes := []rune{'@', '/'}
return strings.IndexFunc(domain, isInvalid(invalidRunes)) < 0
}
func isInvalid(invalidRunes []rune) func(c rune) bool {
isInvalid := func(c rune) bool {
if unicode.IsSpace(c) {
return true
}
for _, r := range invalidRunes {
if c == r {
return true
}
}
return false
}
return isInvalid
}