diff options
author | 2018-03-30 17:52:53 +0300 | |
---|---|---|
committer | 2018-03-30 17:52:53 +0300 | |
commit | bda2d764a9c187da5aa8cd74fb8c4a9574d3996e (patch) | |
tree | 2d5bc98a5744f8f6a0ec4e7f9893c2355235267c /juick-www/src/test/java/com/juick/server/StreamComponentServer.java | |
parent | 7eb13edc1720b13924d7cbe715be67a535f0e02e (diff) |
xmpp integration tests
Diffstat (limited to 'juick-www/src/test/java/com/juick/server/StreamComponentServer.java')
-rw-r--r-- | juick-www/src/test/java/com/juick/server/StreamComponentServer.java | 61 |
1 files changed, 61 insertions, 0 deletions
diff --git a/juick-www/src/test/java/com/juick/server/StreamComponentServer.java b/juick-www/src/test/java/com/juick/server/StreamComponentServer.java new file mode 100644 index 00000000..8c66c2e8 --- /dev/null +++ b/juick-www/src/test/java/com/juick/server/StreamComponentServer.java @@ -0,0 +1,61 @@ +package com.juick.server; + +import com.juick.xmpp.extensions.Handshake; +import org.apache.commons.codec.digest.DigestUtils; +import org.xmlpull.v1.XmlPullParserException; +import rocks.xmpp.addr.Jid; + +import java.io.IOException; +import java.io.InputStream; +import java.io.OutputStream; +import java.util.UUID; + +import static com.juick.server.StreamNamespaces.NS_COMPONENT_ACCEPT; +import static com.juick.server.StreamNamespaces.NS_STREAM; + +/** + * Created by vitalyster on 30.01.2017. + */ +public class StreamComponentServer extends Stream { + + private String streamId, secret; + + public String getStreamId() { + return streamId; + } + + + public StreamComponentServer(InputStream is, OutputStream os, String password) throws XmlPullParserException { + super(null, null, is, os); + secret = password; + streamId = UUID.randomUUID().toString(); + } + @Override + public void handshake() throws XmlPullParserException, IOException { + parser.next(); + if (!parser.getName().equals("stream") + || !parser.getNamespace(null).equals(NS_COMPONENT_ACCEPT) + || !parser.getNamespace("stream").equals(NS_STREAM)) { + throw new IOException("invalid stream"); + } + Jid domain = Jid.of(parser.getAttributeValue(null, "to")); + if (streamHandler.filter(null, domain)) { + send(new XMPPError(XMPPError.Type.cancel, "forbidden").toString()); + throw new IOException("invalid domain"); + } + from = domain; + to = domain; + send(String.format("<stream:stream xmlns:stream='%s' " + + "xmlns='%s' from='%s' id='%s'>", NS_STREAM, NS_COMPONENT_ACCEPT, from.asBareJid().toEscapedString(), streamId)); + Handshake handshake = Handshake.parse(parser); + boolean authenticated = handshake.getValue().equals(DigestUtils.sha1Hex(streamId + secret)); + setLoggedIn(authenticated); + if (!authenticated) { + send(new XMPPError(XMPPError.Type.cancel, "not-authorized").toString()); + streamHandler.fail(new IOException("stream:stream, failed authentication")); + return; + } + send(new Handshake().toString()); + streamHandler.ready(); + } +} |