com.davfx.ninio.http.websocket.WebsocketHttpServerHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ninio Show documentation
A Java NIO HTTP client/server as light as possible
There is a newer version: 1.0.99
package com.davfx.ninio.http.websocket;

import java.io.IOException;
import java.nio.ByteBuffer;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.davfx.ninio.common.Address;
import com.davfx.ninio.common.FailableCloseableByteBufferHandler;
import com.davfx.ninio.common.ReadyConnection;
import com.davfx.ninio.http.HttpRequest;
import com.davfx.ninio.http.HttpResponse;
import com.davfx.ninio.http.HttpServerHandler;
import com.google.common.base.Charsets;
import com.google.common.hash.Hashing;
import com.google.common.io.BaseEncoding;

public final class WebsocketHttpServerHandler implements HttpServerHandler {
	private static final Logger LOGGER = LoggerFactory.getLogger(WebsocketHttpServerHandler.class);

	private final ReadyConnection wrappee;
	private HttpRequest request;
	private HttpServerHandler.Write write;
	private boolean textResponses = false;
	
	public WebsocketHttpServerHandler(ReadyConnection wrappee) {
		this.wrappee = wrappee;
	}
	
	public WebsocketHttpServerHandler withTextResponses(boolean textResponses) {
		this.textResponses = textResponses;
		return this;
	}
	
	@Override
	public void ready(final Write write) {
		this.write = write;

		String wsKey = request.getHeaders().get("Sec-WebSocket-Key");
		String wsVersion = request.getHeaders().get("Sec-WebSocket-Version");

		if (!"13".equals(wsVersion)) {
			write.failed(new IOException("Current implementation does not handle this version: " + wsVersion));
			return;
		}

		HttpResponse response = new HttpResponse(101, "Switching Protocols");
		response.getHeaders().put("Connection", "Upgrade");
		response.getHeaders().put("Upgrade", "websocket");
		response.getHeaders().put("Sec-WebSocket-Accept", BaseEncoding.base64().encode(Hashing.sha1().hashBytes((wsKey + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11").getBytes(Charsets.UTF_8)).asBytes()));
		// We don't need this hack anymore // response.getHeaders().put(Http.TRANSFER_ENCODING, "none");
		
		write.write(response);

		LOGGER.debug("Client connected");
		
		wrappee.connected(new FailableCloseableByteBufferHandler() {
			@Override
			public void close() {
				write.handle(null, WebsocketFrameReader.headerOf(true, 0x08, null));
			}
			@Override
			public void failed(IOException e) {
				write.failed(e);
			}
			@Override
			public void handle(Address address, ByteBuffer buffer) {
				write.handle(null, WebsocketFrameReader.headerOf(true, textResponses ? 0x01 : 0x02, buffer));
				write.handle(null, buffer);
			}
		});
	}

	private static final int PING_MAX_BUFFER_SIZE = 1024;
	private WebsocketFrameReader websocketFrameReader = null;
	
	@Override
	public void handle(Address address, ByteBuffer buffer) {
		websocketFrameReader.handle(address, buffer);
	}

	@Override
	public void handle(HttpRequest request) {
		this.request = request;
		websocketFrameReader = new WebsocketFrameReader(new WebsocketFrameReader.Handler() {
			private ByteBuffer pingBuffer = null;
			@Override
			public void failed(IOException e) {
				write.failed(e);
			}
			@Override
			public void handle(int opcode, long frameLength, ByteBuffer partialBuffer) {
				if (opcode == 0x09) {
					if (pingBuffer == null) {
						if (frameLength > PING_MAX_BUFFER_SIZE) {
							write.failed(new IOException("Ping frame is too big: " + frameLength));
							return;
						}
						pingBuffer = ByteBuffer.allocate((int) frameLength);
					}
					
					pingBuffer.put(partialBuffer);
					
					if (pingBuffer.position() == frameLength) {
						pingBuffer.flip();
						write.handle(null, WebsocketFrameReader.headerOf(true, 0x0A, pingBuffer));
						write.handle(null, pingBuffer);
						pingBuffer = null;
					}
					
					return;
				}
				
				if ((opcode == 0x01) || (opcode == 0x02)) {
					wrappee.handle(null, partialBuffer);
					return;
				}

				if (opcode == 0x08) {
					LOGGER.debug("Connection closed");
					write.close();
					wrappee.close();
					return;
				}
				
				write.failed(new IOException("Current implementation does not handle this opcode: " + opcode));
			}
		});
	}
	
	@Override
	public void close() {
		wrappee.close();
	}
	
	@Override
	public void failed(IOException e) {
		wrappee.failed(e);
	}
}