com.generallycloud.baseio.codec.http11.WebSocketProtocolEncoder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of baseio-all Show documentation
The all in one project of baseio
There is a newer version: 3.2.9-BETA-2
/*
 * Copyright 2015-2017 GenerallyCloud.com
 *  
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *  
 *      http://www.apache.org/licenses/LICENSE-2.0
 *  
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */ 
package com.generallycloud.baseio.codec.http11;

import java.io.IOException;

import com.generallycloud.baseio.buffer.ByteBuf;
import com.generallycloud.baseio.buffer.ByteBufAllocator;
import com.generallycloud.baseio.codec.http11.future.WebSocketReadFuture;
import com.generallycloud.baseio.common.MathUtil;
import com.generallycloud.baseio.component.ByteArrayBuffer;
import com.generallycloud.baseio.protocol.ChannelReadFuture;
import com.generallycloud.baseio.protocol.ChannelWriteFuture;
import com.generallycloud.baseio.protocol.ChannelWriteFutureImpl;
import com.generallycloud.baseio.protocol.ProtocolEncoder;

//WebSocket规定服务端不准向客户端发送mask过的数据
//A server MUST NOT mask any frames that it sends to the client.
public class WebSocketProtocolEncoder implements ProtocolEncoder {

	@Override
	public ChannelWriteFuture encode(ByteBufAllocator allocator, ChannelReadFuture future) throws IOException {
		
		WebSocketReadFuture f = (WebSocketReadFuture) future;

		ByteArrayBuffer buffer = future.getWriteBuffer();
		
		byte [] header;
		
		byte [] data = buffer.array();
		
		int size = buffer.size();
		
		byte header0 = (byte) (0x8f & (f.getType() | 0xf0));
		
		if (size < 126) {
			header = new byte[2];
			header[0] = header0;
			header[1] = (byte) size;
		}else if(size < ((1 << 16) -1)){
			header = new byte[4];
			header[0] = header0;
			header[1] = 126;
			header[3] = (byte)(size & 0xff);
			header[2] = (byte)((size >> 8) & 0x80);
		}else{
			header = new byte[6];
			header[0] = header0;
			header[1] = 127;
			MathUtil.int2Byte(header, size, 2);
		}
		
		ByteBuf buf = allocator.allocate(header.length + size);
		
		buf.put(header);
		
		buf.put(data,0,size);
		
		return new ChannelWriteFutureImpl(future, buf.flip());
	}
	
//	public IOWriteFuture encodeWithMask(BaseContext context, IOReadFuture readFuture) throws IOException {
//		
//		WebSocketReadFuture future = (WebSocketReadFuture) readFuture;
//
//		BufferedOutputStream o = future.getWriteBuffer();
//
//		byte [] header;
//		
//		int size = o.size();
//		
//		byte header0 = (byte) (0x8f & (future.getType() | 0xf0));
//		
//		if (size < 126) {
//			header = new byte[2];
//			header[0] = header0;
//			header[1] = (byte)(size | 0x80);
//		}else if(size < ((1 << 16) -1)){
//			header = new byte[4];
//			header[0] = header0;
//			header[1] = (byte) (126 | 0xff);
//			header[3] = (byte)(size & 0xff);
//			header[2] = (byte)((size >> 8) & 0x80);
//		}else{
//			header = new byte[6];
//			header[0] = header0;
//			header[1] = (byte) (127 | 0x80);
//			MathUtil.int2Byte(header, size, 2);
//		}
//		
//		ByteBuf buffer = context.getHeapByteBufferPool().allocate(header.length + size + 4);
//		
//		buffer.put(header);
//		
//		byte [] array = o.array();
//		
//		byte [] mask = MathUtil.int2Byte(size);
//		
//		for (int i = 0; i < size; i++) {
//			
//			array[i] = (byte)(array[i] ^ mask[i % 4]);
//		}
//		
//		buffer.put(mask);
//		
//		buffer.put(array,0,size);
//		
//		buffer.flip();
//
//		return new IOWriteFutureImpl(readFuture, buffer);
//	}

}