353073 Add websocket origin setter

This commit is contained in:
Greg Wilkins 2011-08-22 12:02:00 +10:00
parent 74b4d6bc2c
commit 6867dae4ee
1 changed files with 20 additions and 1 deletions

View File

@ -83,6 +83,7 @@ public class WebSocketClient extends AggregateLifeCycle
private final List<String> _extensions=new CopyOnWriteArrayList<String>(); private final List<String> _extensions=new CopyOnWriteArrayList<String>();
private int _bufferSize=64*1024; private int _bufferSize=64*1024;
private String _origin;
private String _protocol; private String _protocol;
private int _maxIdleTime=-1; private int _maxIdleTime=-1;
@ -202,6 +203,24 @@ public class WebSocketClient extends AggregateLifeCycle
_protocol = protocol; _protocol = protocol;
} }
/* ------------------------------------------------------------ */
/** Get the origin of the client
* @return The clients Origin
*/
public String getOrigin()
{
return _origin;
}
/* ------------------------------------------------------------ */
/** Set the origin of the client
* @param origin the origin of the client (eg "http://example.com")
*/
public void setOrigin(String origin)
{
_origin = origin;
}
/* ------------------------------------------------------------ */ /* ------------------------------------------------------------ */
public Map<String,String> getCookies() public Map<String,String> getCookies()
{ {
@ -455,7 +474,7 @@ public class WebSocketClient extends AggregateLifeCycle
"Upgrade: websocket\r\n"+ "Upgrade: websocket\r\n"+
"Connection: Upgrade\r\n"+ "Connection: Upgrade\r\n"+
"Sec-WebSocket-Key: "+_key+"\r\n"+ "Sec-WebSocket-Key: "+_key+"\r\n"+
"Sec-WebSocket-Origin: http://example.com\r\n"+ (_origin==null?"":"Sec-WebSocket-Origin: "+_origin+"\r\n")+
"Sec-WebSocket-Version: "+WebSocketConnectionD10.VERSION+"\r\n"; "Sec-WebSocket-Version: "+WebSocketConnectionD10.VERSION+"\r\n";
if (holder.getProtocol()!=null) if (holder.getProtocol()!=null)