From b31c29a03da97a11436d6e076a99ca9598c28f9d Mon Sep 17 00:00:00 2001
From: Mickael Remond <mremond@process-one.net>
Date: Thu, 11 Jan 2018 23:00:59 +0100
Subject: [PATCH] Implements dummy auth + stream error

---
 cmd/xmpp_component/xmpp_component.go |  2 +-
 component.go                         | 45 ++++++++++++++++++++++++++--
 parser.go                            |  7 +++--
 stream.go                            |  5 ++++
 4 files changed, 53 insertions(+), 6 deletions(-)

diff --git a/cmd/xmpp_component/xmpp_component.go b/cmd/xmpp_component/xmpp_component.go
index 65c0e4b..ea97c1f 100644
--- a/cmd/xmpp_component/xmpp_component.go
+++ b/cmd/xmpp_component/xmpp_component.go
@@ -3,5 +3,5 @@ package main
 import "fluux.io/xmpp"
 
 func main() {
-	xmpp.Open("test")
+	xmpp.Open("mqtt.localhost")
 }
diff --git a/component.go b/component.go
index d457d88..d5555d2 100644
--- a/component.go
+++ b/component.go
@@ -1,7 +1,10 @@
 package xmpp
 
 import (
+	"encoding/xml"
+	"errors"
 	"fmt"
+	"io"
 	"net"
 	"time"
 )
@@ -14,24 +17,60 @@ const componentStreamOpen = "<?xml version='1.0'?><stream:stream to='%s' xmlns='
 type Component struct {
 	// TCP level connection
 	conn net.Conn
+
+	// read / write
+	socketProxy io.ReadWriter
+	decoder     *xml.Decoder
 }
 
 // TODO Helper to prepare connection string
 func Open(connStr string) error {
+	c := Component{}
+
 	var conn net.Conn
 	var err error
-
 	if conn, err = net.DialTimeout("tcp", "localhost:8888", time.Duration(5)*time.Second); err != nil {
 		return err
 	}
+	c.conn = conn
 
 	// TODO send stream open and check for reply
 	// Send stream open tag
-	componentHost := "mqtt.localhost"
+	componentHost := connStr // TODO Fix me
 	if _, err := fmt.Fprintf(conn, componentStreamOpen, componentHost, NSComponent, NSStream); err != nil {
-		fmt.Println("Cannot send stream open.")
+		fmt.Println("cannot send stream open.")
 		return err
 	}
+	c.decoder = xml.NewDecoder(conn)
+
+	// Initialize xml decoder and extract streamID from reply
+	streamId, err := initDecoder(c.decoder)
+	if err != nil {
+		fmt.Println("cannot init decoder")
+		return err
+	}
+
+	fmt.Println("StreamID = ", streamId)
+
+	// Authentication
+	if _, err := fmt.Fprint(conn, "<handshake>aaee83c26aeeafcbabeabfcbcd50df997e0a2a1e</handshake>"); err != nil {
+		fmt.Println("cannot send stream open.")
+		return err
+	}
+
+	// Next message should be either success or failure.
+	name, val, err := next(c.decoder)
+	if err != nil {
+		fmt.Println(err)
+		return err
+	}
+
+	switch v := val.(type) {
+	case *StreamError:
+		fmt.Printf("error: %s", v.Error.Local)
+	default:
+		return errors.New("unexpected packet, got " + name.Local + " in " + name.Space)
+	}
 
 	return nil
 }
diff --git a/parser.go b/parser.go
index 39165cb..f2bfc41 100644
--- a/parser.go
+++ b/parser.go
@@ -8,11 +8,12 @@ import (
 )
 
 // Reads and checks the opening XMPP stream element.
-// It returns a stream structure containing:
+// TODO It returns a stream structure containing:
 // - Host: You can check the host against the host you were expecting to connect to
 // - Id: the Stream ID is a temporary shared secret used for some hash calculation. It is also used by ProcessOne
 //       reattach features (allowing to resume an existing stream at the point the connection was interrupted, without
 //       getting through the authentication process.
+// TODO We should handle stream error from XEP-0114 ( <conflict/> or <host-unknown/> )
 func initDecoder(p *xml.Decoder) (sessionID string, err error) {
 	for {
 		var t xml.Token
@@ -72,7 +73,9 @@ func next(p *xml.Decoder) (xml.Name, interface{}, error) {
 	// Put it in an interface and allocate one.
 	var nv interface{}
 	switch se.Name.Space + " " + se.Name.Local {
-	// TODO: general case = Parse IQ / presence / message => split SASL case
+	case NSStream + " error":
+		nv = &StreamError{}
+		// TODO: general case = Parse IQ / presence / message => split SASL case
 	case nsSASL + " success":
 		nv = &saslSuccess{}
 	case nsSASL + " failure":
diff --git a/stream.go b/stream.go
index 9138c2f..510169c 100644
--- a/stream.go
+++ b/stream.go
@@ -13,6 +13,11 @@ type streamFeatures struct {
 	Any        []xml.Name `xml:",any"`
 }
 
+type StreamError struct {
+	XMLName xml.Name `xml:"http://etherx.jabber.org/streams error"`
+	Error   xml.Name `xml:",any"`
+}
+
 type Caps struct {
 	XMLName xml.Name `xml:"http://jabber.org/protocol/caps c"`
 	Hash    string   `xml:"hash,attr"`