com.caucho.v5.http.websocket.WebSocketBaratineImpl Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of baratine Show documentation
Show all versions of baratine Show documentation
A reactive Java web server.
/*
* Copyright (c) 1998-2015 Caucho Technology -- all rights reserved
*
* This file is part of Baratine(TM)
*
* Each copy or derived work must preserve the copyright notice and this
* notice unmodified.
*
* Baratine is free software; you can redistribute it and/or modify
* it under the terms of the GNU General Public License as published by
* the Free Software Foundation; either version 2 of the License, or
* (at your option) any later version.
*
* Baratine is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
* of NON-INFRINGEMENT. See the GNU General Public License for more
* details.
*
* You should have received a copy of the GNU General Public License
* along with Baratine; if not, write to the
*
* Free Software Foundation, Inc.
* 59 Temple Place, Suite 330
* Boston, MA 02111-1307 USA
*
* @author Scott Ferguson
*/
package com.caucho.v5.http.websocket;
import static com.caucho.v5.websocket.io.WebSocketConstants.OP_BINARY;
import static com.caucho.v5.websocket.io.WebSocketConstants.OP_TEXT;
import static io.baratine.web.HttpStatus.METHOD_NOT_ALLOWED;
import java.io.IOException;
import java.security.MessageDigest;
import java.util.ArrayList;
import java.util.Objects;
import java.util.logging.Level;
import java.util.logging.Logger;
import com.caucho.v5.http.protocol.OutHttp;
import com.caucho.v5.http.protocol.OutHttpProxy;
import com.caucho.v5.http.protocol.OutResponseBase;
import com.caucho.v5.io.TempBuffer;
import com.caucho.v5.io.WriteBuffer;
import com.caucho.v5.network.port.StateConnection;
import com.caucho.v5.util.Base64Util;
import com.caucho.v5.util.Hex;
import com.caucho.v5.util.L10N;
import com.caucho.v5.util.ModulePrivate;
import com.caucho.v5.web.webapp.RequestBaratine;
import com.caucho.v5.web.webapp.RequestBaratineImpl;
import com.caucho.v5.websocket.io.FrameInputStream;
import com.caucho.v5.websocket.io.WebSocketBaratine;
import com.caucho.v5.websocket.io.WebSocketConstants;
import io.baratine.io.Buffer;
import io.baratine.pipe.Pipe;
import io.baratine.service.ServiceRef;
import io.baratine.web.HttpStatus;
import io.baratine.web.ServiceWebSocket;
/**
* websocket server container
*/
@ModulePrivate
public class WebSocketBaratineImpl
extends WebSocketBase
implements WebSocketBaratine
{
private static final L10N L = new L10N(WebSocketBaratineImpl.class);
private static final Logger log
= Logger.getLogger(WebSocketBaratineImpl.class.getName());
private String _uri;
//private FrameInputStream _fIs;
private InWebSocket _inBinary = new InWebSocketSkip();
private InWebSocket _inText = new InWebSocketSkip();
private char []_charBuf = new char[256];
private OutResponseBase _os;
private RequestBaratine _request;
private ServiceWebSocket _service;
private MessageState _state = MessageState.IDLE;
private TempBuffer _tBuf;
private int _headOffset;
private int _offset;
private int _frameLength;
private OutHttpProxy _outProxy;
private OutWebSocketWriter _outWriter;
private int _opMessage;
public WebSocketBaratineImpl(WebSocketManager manager,
ServiceWebSocket service,
Class type)
{
super(manager);
Objects.requireNonNull(service);
Objects.requireNonNull(type);
_service = service;
/*
TypeRef typeRef = TypeRef.of(service.getClass());
TypeRef typeRefService = typeRef.to(ServiceWebSocket.class);
//TypeRef type = typeRefService.param(0);
Class rawClass;
if (type != null) {
rawClass = type.rawClass();
}
else {
rawClass = String.class;
}
*/
if (Frame.class.equals(type)) {
readFrame((ServiceWebSocket) service);
}
else if (String.class.equals(type)) {
readString((ServiceWebSocket) service);
}
else if (Buffer.class.equals(type)) {
read((ServiceWebSocket) service);
}
else {
readString(manager.createSerializer(type, service));
}
//System.out.println("VALUE: " + type + " " + typeRefService + " " + typeRef);
}
public boolean handshake(RequestBaratine request)
throws Exception
{
Objects.requireNonNull(request);
_request = request;
RequestBaratineImpl req = (RequestBaratineImpl) request;
//ResponseBaratineImpl res = (ResponseBaratineImpl) response;
if (log.isLoggable(Level.FINE)) {
log.fine(this + " upgrade HTTP to WebSocket");
}
String method = req.method();
String uri = req.uri();
if (! "GET".equals(method)) {
req.sendError(METHOD_NOT_ALLOWED);
throw new IllegalStateException(L.l("HTTP Method must be 'GET', because the WebSocket protocol requires 'GET'.\n remote-IP: {0}",
req.addressRemote()));
}
String connection = req.header("Connection");
String upgrade = req.header("Upgrade");
if (! "websocket".equalsIgnoreCase(upgrade)) {
req.sendError(HttpStatus.BAD_REQUEST);
throw new IllegalStateException(L.l("HTTP Upgrade header '{0}' must be 'WebSocket', because the WebSocket protocol requires an Upgrade: WebSocket header.\n remote-IP: {1}",
upgrade,
req.addressRemote()));
}
if (connection == null
|| connection.toLowerCase().indexOf("upgrade") < 0) {
req.sendError(HttpStatus.BAD_REQUEST);
throw new IllegalStateException(L.l("HTTP Connection header '{0}' must be 'Upgrade', because the WebSocket protocol requires a Connection: Upgrade header.\n remote-IP: {1}",
connection,
req.addressRemote()));
}
String key = req.header("Sec-WebSocket-Key");
if (key == null) {
req.sendError(HttpStatus.BAD_REQUEST);
throw new IllegalStateException(L.l("HTTP Sec-WebSocket-Key header is required, because the WebSocket protocol requires an Origin header.\n remote-IP: {0}",
req.addressRemote()));
}
else if (key.length() != 24) {
req.sendError(HttpStatus.BAD_REQUEST);
throw new IllegalStateException(L.l("HTTP Sec-WebSocket-Key header is invalid '{0}' because it's not a 16-byte value.\n remote-IP: {1}",
key,
req.addressRemote()));
}
String version = req.header("Sec-WebSocket-Version");
String requiredVersion = WebSocketConstants.VERSION;
if (! requiredVersion.equals(version)) {
req.sendError(HttpStatus.BAD_REQUEST);
throw new IllegalStateException(L.l("HTTP Sec-WebSocket-Version header with value '{0}' is required, because the WebSocket protocol requires an Sec-WebSocket-Version header.\n remote-IP: {1}",
requiredVersion,
req.addressRemote()));
}
String extensions = req.header("Sec-WebSocket-Extensions");
boolean isMasked = true;
if (extensions != null && extensions.indexOf("x-unmasked") >= 0) {
isMasked = false;
}
ArrayList serverExtensionList = new ArrayList();
if (! isMasked) {
serverExtensionList.add("x-unmasked");
}
req.status(HttpStatus.SWITCHING_PROTOCOLS);//, "Switching Protocols");
req.header("Upgrade", "websocket");
req.header("Connection", "Upgrade");
String accept = calculateWebSocketAccept(key);
req.header("Sec-WebSocket-Accept", accept);
if (serverExtensionList.size() > 0) {
StringBuilder sb = new StringBuilder();
for (int i = 0; i < serverExtensionList.size(); i++) {
if (i > 0)
sb.append(", ");
sb.append(serverExtensionList.get(i));
}
req.header("Sec-WebSocket-Extensions", sb.toString());
}
req.contentLength(0);
FrameInputStream fIs;
fIs = new FrameInputStream();
// _fIs = fIs;
fIs.init(null, req.requestHttp().conn().readStream());
frameInput(fIs);
// Endpoint endpoint = _endpointSkeleton.newEndpoint(_factory, paths);
// Endpoint endpoint = wsCxt.getFactory().get();
//WebSocketBaratine wsConn = null;;
ConnectionWebSocketBaratine connWs
= new ConnectionWebSocketBaratine(this,
req.requestHttp());
// order for duplex
req.upgrade(connWs);
req.flush();
_outProxy = req.connHttp().outProxy();
_outWriter = new OutWebSocketWriter();
_service.open(this);
ServiceRef.flushOutbox();
return true;
}
/*
//@Override
public void init(ReadStream is, OutResponseBase os)
{
Objects.requireNonNull(is);
Objects.requireNonNull(os);
_fIs.init(this, is);
_os = os;
}
*/
//@Override
public void open()
{
if (_inBinary == null) {
_inBinary = new InWebSocketSkip();
}
if (_inText == null) {
_inText = new InWebSocketSkip();
}
}
@Override
public String uri()
{
return _request.uri();
}
@Override
public String path()
{
return _request.path();
}
@Override
public String pathInfo()
{
return _request.pathInfo();
}
/*
@Override
public void writePart(byte []buffer, int offset, int length)
{
// TODO Auto-generated method stub
}
*/
/*
@Override
public void writePart(String data)
{
// TODO Auto-generated method stub
}
*/
private void toTextFromIdle(int length)
{
TempBuffer tBuf = TempBuffer.allocate();
_tBuf = tBuf;
if (length >> 2 < 0x7d) {
_frameLength = 2;
}
else {
_frameLength = 4;
}
_headOffset = 0;
tBuf.length(_headOffset + _frameLength);
}
private void toBinaryFromIdle(int length)
{
TempBuffer tBuf = TempBuffer.allocate();
_tBuf = tBuf;
if (length >> 2 < 0x7d) {
_frameLength = 2;
}
else {
_frameLength = 4;
}
_headOffset = 0;
tBuf.length(_headOffset + _frameLength);
}
@Override
protected void send(TempBuffer tBuf)
{
_outProxy.writeFirst(_outWriter, tBuf, tBuf.length(), false);
}
@Override
protected void sendEnd(TempBuffer tBuf)
{
_outProxy.writeFirst(_outWriter, tBuf, tBuf.length(), true);
}
//
// impl
//
public StateConnection service()
{
if (readFrame()) {
return StateConnection.READ;
}
else {
return StateConnection.CLOSE;
}
}
private String calculateWebSocketAccept(String key)
{
try {
MessageDigest md = MessageDigest.getInstance("SHA1");
int length = key.length();
for (int i = 0; i < length; i++) {
md.update((byte) key.charAt(i));
}
String guid = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
length = guid.length();
for (int i = 0; i < length; i++) {
md.update((byte) guid.charAt(i));
}
byte []digest = md.digest();
return Base64Util.encode(digest);
} catch (Exception e) {
throw new RuntimeException(e);
}
}
@Override
public String toString()
{
return getClass().getSimpleName() + "[" + _request.uri() + "]";
}
//
// readers
//
private static interface InWebSocket
{
void read(FrameInputStream fIs)
throws IOException;
}
private static class InWebSocketSkip implements InWebSocket
{
@Override
public void read(FrameInputStream fIs)
throws IOException
{
fIs.skipToFrameEnd();
}
}
private static class InReadBinary implements InWebSocket
{
private Pipe _out;
private InReadBinary(Pipe out)
{
Objects.requireNonNull(out);
_out = out;
}
@Override
public void read(FrameInputStream fIs)
throws IOException
{
Buffer buffer = Buffer.create();
System.out.println("INR: " + _out);
fIs.readBuffer(buffer);
_out.next(buffer);
}
}
private static class InFrameBinary implements InWebSocket
{
private Pipe _out;
private InFrameBinary(Pipe out)
{
Objects.requireNonNull(out);
_out = out;
}
@Override
public void read(FrameInputStream fIs)
throws IOException
{
int op = fIs.getOpcode();
int len = (int) fIs.length();
boolean isPart = ! fIs.isFinal();
byte []buffer = new byte[(int) len];
int sublen = fIs.readBinary(buffer, 0, len);
FrameBinary frame = new FrameBinary(len, isPart, buffer);
_out.next(frame);
}
}
private static class FrameBinary implements Frame
{
private long _length;
private boolean _isPart;
private byte []_data;
FrameBinary(int length,
boolean isPart,
byte []buffer)
{
_length = length;
_isPart = isPart;
_data = buffer;
}
@Override
public boolean part()
{
return _isPart;
}
@Override
public FrameType type()
{
return FrameType.BINARY;
}
@Override
public String text()
{
return null;
}
@Override
public Buffer binary()
{
return Buffer.create(_data);
}
public String toString()
{
return (getClass().getSimpleName()
+ "[" + _length
+ "," + Hex.toShortHex(_data, 0, (int) _length)
+ (_isPart ? ",part" : "")
+ "]");
}
}
enum MessageState {
IDLE {
@Override
public MessageState toBinary() { return BINARY; }
@Override
public MessageState toText(WebSocketBaratineImpl ws, int length)
{
ws.toTextFromIdle(length);
return TEXT;
}
@Override
public MessageState toBinary(WebSocketBaratineImpl ws, int length)
{
ws.toBinaryFromIdle(length);
return TEXT;
}
},
BINARY
{
@Override
public boolean isActive() { return true; }
@Override
public int code() { return OP_BINARY; }
@Override
public MessageState toBinary() { return BINARY; }
@Override
public MessageState toCont() { return CONT; }
},
TEXT
{
@Override
public boolean isActive() { return true; }
@Override
public int code() { return OP_TEXT; }
},
CONT {
@Override
public boolean isActive() { return true; }
@Override
public MessageState toCont() { return CONT; }
},
DESTROYED {
};
public boolean isActive()
{
return false;
}
public MessageState toBinary()
{
throw new IllegalStateException(toString());
}
public MessageState toText(WebSocketBaratineImpl out, int length)
{
throw new IllegalStateException(toString());
}
public MessageState toBinary(WebSocketBaratineImpl out, int length)
{
throw new IllegalStateException(toString());
}
public MessageState toCont()
{
throw new IllegalStateException(toString());
}
public MessageState toIdle()
{
return IDLE;
}
public int code()
{
throw new IllegalStateException(toString());
}
}
/*
private class OutFlushImpl implements OutWebSocketFlush
{
@Override
public void flush(TempBuffer tBuf, int length)
{
byte []buffer = tBuf.buffer();
System.out.println("FLU: " + new String(buffer, 0, length) + " " + Hex.toHex(buffer, 0, length));
// TODO Auto-generated method stub
}
}
*/
private class OutWebSocketWriter implements OutHttp
{
@Override
public boolean writeFirst(WriteBuffer out, TempBuffer buffer, long length,
boolean isEnd)
{
if (out != null) {
try {
out.write(buffer.buffer(), 0, buffer.length());
out.flush(); // xxx:
} catch (Exception e) {
log.log(Level.WARNING, e.toString(), e);
}
}
return false;
}
@Override
public boolean writeNext(WriteBuffer out, TempBuffer buffer, boolean isEnd)
{
System.out.println("NEXT: " + out);
return false;
}
@Override
public void disconnect(WriteBuffer out)
{
}
@Override
public boolean canWrite(long sequence)
{
return true;
}
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy