diff --git a/src/tsung/ts_jabber_common.erl b/src/tsung/ts_jabber_common.erl index 8ca25a3b..7e1ff81e 100644 --- a/src/tsung/ts_jabber_common.erl +++ b/src/tsung/ts_jabber_common.erl @@ -54,9 +54,9 @@ get_message(Jabber=#jabber{type = 'connect'}) -> connect(Jabber); get_message(#jabber{type = 'starttls'}) -> starttls(); -get_message(#jabber{type = 'close', id=Id,username=User,passwd=Pwd,user_server=UserServer}) -> +get_message(#jabber{type = 'close', id=Id,username=User,passwd=Pwd,user_server=UserServer,version=Version}) -> ts_user_server:remove_connected(UserServer,set_id(Id,User,Pwd)), - close(); + close(Version); get_message(#jabber{type = 'presence'}) -> presence(); get_message(#jabber{type = 'presence:initial', id=Id,username=User,passwd=Pwd,user_server=UserServer}) -> @@ -311,6 +311,10 @@ get_message2(Jabber=#jabber{type = 'auth_sasl_session'}) -> %%---------------------------------------------------------------------- %% Func: connect/1 %%---------------------------------------------------------------------- +connect(#jabber{domain=Domain, version="websocket"}) -> + list_to_binary([ + ""]); connect(#jabber{domain=Domain, version = Version}) -> VersionStr = case Version of "legacy" -> @@ -326,10 +330,13 @@ connect(#jabber{domain=Domain, version = Version}) -> "' xmlns='jabber:client' ",VersionStr,"xmlns:stream='http://etherx.jabber.org/streams'>"]). %%---------------------------------------------------------------------- -%% Func: close/0 +%% Func: close/1 %% Purpose: close jabber session %%---------------------------------------------------------------------- -close () -> list_to_binary(""). +close("websocket") -> + <<"">>; +close(_Version) -> + <<"">>. %%---------------------------------------------------------------------- %% Func: starttls/0