94 lines
2.5 KiB
Go
94 lines
2.5 KiB
Go
package xmpp
|
||
|
||
import (
|
||
"git.kingpenguin.tk/chteufleur/go-xmpp.git/src/xmpp"
|
||
|
||
"crypto/rand"
|
||
"log"
|
||
"strconv"
|
||
)
|
||
|
||
const (
|
||
dictionary = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"
|
||
|
||
REPLY_UNREACHABLE = "reply_unreachable"
|
||
REPLY_DENY = "reply_deny"
|
||
REPLY_OK = "reply_ok"
|
||
|
||
TYPE_SEND_MESSAGE = "type_send_message"
|
||
TYPE_SEND_IQ = "type_send_iq"
|
||
)
|
||
|
||
type Client struct {
|
||
JID string
|
||
Method string
|
||
Domain string
|
||
Transaction string
|
||
|
||
TypeSend string
|
||
IdMap string
|
||
|
||
ChanReply chan string
|
||
}
|
||
|
||
func (client *Client) QueryClient() {
|
||
log.Printf("%sQuery JID %s", LogInfo, client.JID)
|
||
isAutoGeneratedTranctionID := false
|
||
if client.Transaction == "" {
|
||
// Random transaction ID generation
|
||
client.Transaction = TransactionID()
|
||
isAutoGeneratedTranctionID = true
|
||
}
|
||
clientJID, _ := xmpp.ParseJID(client.JID)
|
||
if clientJID.Resource == "" {
|
||
client.askViaMessage(isAutoGeneratedTranctionID)
|
||
} else {
|
||
client.askViaIQ()
|
||
}
|
||
}
|
||
|
||
func (client *Client) askViaIQ() {
|
||
stanzaID++
|
||
stanzaIDstr := strconv.Itoa(stanzaID)
|
||
m := xmpp.Iq{Type: xmpp.IQTypeGet, To: client.JID, From: jid.Domain, Id: stanzaIDstr}
|
||
confirm := &xmpp.Confirm{Id: client.Transaction, Method: client.Method, URL: client.Domain}
|
||
m.PayloadEncode(confirm)
|
||
WaitIqMessages[stanzaIDstr] = client
|
||
comp.Out <- m
|
||
|
||
client.TypeSend = TYPE_SEND_IQ
|
||
client.IdMap = stanzaIDstr
|
||
}
|
||
|
||
func (client *Client) askViaMessage(isAutoGeneratedTranctionID bool) {
|
||
m := xmpp.Message{From: jid.Domain, To: client.JID, Type: xmpp.MessageTypeNormal}
|
||
|
||
m.Thread = xmpp.SessionID()
|
||
m.Body = client.Domain + " (with method " + client.Method + ") need to validate your identity, do you agree ?"
|
||
m.Body += "\nValidation code : " + client.Transaction
|
||
if !isAutoGeneratedTranctionID {
|
||
// Send only if the transaction ID is not autogenerated
|
||
m.Body += "\nPlease check that this code is the same as on " + client.Domain
|
||
}
|
||
m.Body += "\n\nIf your client doesn't support that functionnality, please send back the validation code to confirm the request."
|
||
m.Confir = &xmpp.Confirm{Id: client.Transaction, Method: client.Method, URL: client.Domain}
|
||
|
||
log.Printf("%sSend message %v", LogInfo, m)
|
||
WaitMessageAnswers[client.Transaction] = client
|
||
comp.Out <- m
|
||
|
||
client.TypeSend = TYPE_SEND_MESSAGE
|
||
client.IdMap = client.Transaction
|
||
}
|
||
|
||
func TransactionID() string {
|
||
var bytes = make([]byte, 8)
|
||
if _, err := rand.Read(bytes); err != nil {
|
||
panic(err)
|
||
}
|
||
for k, v := range bytes {
|
||
bytes[k] = dictionary[v%byte(len(dictionary))]
|
||
}
|
||
return string(bytes)
|
||
}
|