org.voovan.network.messagesplitter.HttpMessageSplitter Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of voovan-web Show documentation
Show all versions of voovan-web Show documentation
Voovan is a java framwork and it not depends on any third-party framework.
package org.voovan.network.messagesplitter;
import org.voovan.Global;
import org.voovan.http.HttpSessionParam;
import org.voovan.http.HttpRequestType;
import org.voovan.http.message.HttpStatic;
import org.voovan.network.IoSession;
import org.voovan.network.MessageSplitter;
import org.voovan.tools.TByteBuffer;
import java.math.BigInteger;
import java.nio.ByteBuffer;
/**
* Http 消息分割类
*
* @author helyho
*
* Voovan Framework.
* WebSite: https://github.com/helyho/Voovan
* Licence: Apache v2 License
*/
public class HttpMessageSplitter implements MessageSplitter {
@Override
public int canSplite(IoSession session, ByteBuffer byteBuffer) {
int result = -1;
//返回 -1: 报文未完整接受, -2: 报文异常, 管理连接
if(byteBuffer.limit()==0){
return -1;
}
if(HttpRequestType.WEBSOCKET.equals(session.getAttribute(HttpSessionParam.TYPE)) ){
result = isWebSocketFrame(byteBuffer);
} else {
if (!session.containAttribute(HttpSessionParam.TYPE)) {
session.setAttribute(HttpSessionParam.TYPE, HttpRequestType.HTTP);
}
return 0;
}
return result;
}
/**
* 判断缓冲区中的数据是否是一个 WebSocket 帧
* @param buffer 缓冲区对象
* @return WebSocket 帧报文长度,-1不是WebSocket 帧, 大于0 返回的 WebSocket 的长度
*/
public static int isWebSocketFrame(ByteBuffer buffer) {
// 接受数据的大小
int maxpacketsize = buffer.remaining();
// 期望数据包的实际大小
int expectPackagesize = 2;
if (maxpacketsize < expectPackagesize) {
return -2;
}
byte finByte = buffer.get();
boolean fin = finByte >> 8 != 0;
byte rsv = (byte) ((finByte & ~(byte) 128) >> 4);
if (rsv != 0) {
return -2;
}
byte maskByte = buffer.get();
boolean mask = (maskByte & -128) != 0;
int payloadlength = (byte) (maskByte & ~(byte) 128);
int optcode = (byte) (finByte & 15);
if(optcode < 0 && optcode > 10){
return -2;
}
if (!fin) {
if (optcode == 9 || optcode == 10 || optcode == 8) {
return -2;
}
}
if (payloadlength >= 0 && payloadlength <= 125) {
} else {
if (optcode == 9 || optcode == 10 || optcode == 8) {
return -2;
}
if (payloadlength == 126) {
expectPackagesize += 2;
byte[] sizebytes = new byte[3];
sizebytes[1] = buffer.get();
sizebytes[2] = buffer.get();
payloadlength = new BigInteger(sizebytes).intValue();
} else {
expectPackagesize += 8;
byte[] bytes = new byte[8];
for (int i = 0; i < 8; i++) {
bytes[i] = buffer.get();
}
long length = new BigInteger(bytes).longValue();
if (length <= Integer.MAX_VALUE) {
payloadlength = (int) length;
}
}
}
expectPackagesize += (mask ? 4 : 0);
expectPackagesize += payloadlength;
// 如果实际接受的数据小于数据包的大小则报错
if (maxpacketsize < expectPackagesize) {
buffer.position(0);
return -2;
} else {
buffer.position(0);
return expectPackagesize;
}
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy