com.firenio.codec.http11.HttpFrame Maven / Gradle / Ivy
/*
* Copyright 2015 The FireNio Project
*
* 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.firenio.codec.http11;
import static com.firenio.codec.http11.HttpHeader.Content_Type;
import static com.firenio.codec.http11.HttpHeader.Sec_WebSocket_Accept;
import static com.firenio.codec.http11.HttpHeader.Sec_WebSocket_Key;
import static com.firenio.codec.http11.HttpHeader.Upgrade;
import java.util.HashMap;
import java.util.Map;
import com.firenio.buffer.ByteBuf;
import com.firenio.collection.IntMap;
import com.firenio.common.Cryptos;
import com.firenio.common.Util;
import com.firenio.component.Channel;
import com.firenio.component.Frame;
/**
* Content-Type: application/x-www-form-urlencoded Content-Type:
* multipart/form-data; boundary=----WebKitFormBoundaryKA6dsRskWA4CdJek
*/
public class HttpFrame extends Frame {
static final KMPUtil KMP_BOUNDARY = new KMPUtil("boundary=");
private int connection = HttpConnection.KEEP_ALIVE.getId();
private int contentLength;
private int contentType = HttpContentType.text_plain_utf8.getId();
private byte[] date;
private int decodeState;
private int headerLength;
private boolean isForm;
private int method;
private Map params = new HashMap<>();
private IntMap request_headers = new IntMap<>(16);
private String requestURL;
private IntMap response_headers;
private int status = HttpStatus.C200.getStatus();
public String getBoundary() {
if (isForm) {
String contentType = getRequestHeader(Content_Type.getId());
int index = KMP_BOUNDARY.match(contentType);
if (index != -1) {
return contentType.substring(index + 9);
}
}
return null;
}
public HttpConnection getConnection() {
return HttpConnection.get(connection);
}
public int getConnectionId() {
return connection;
}
public int getContentLength() {
return contentLength;
}
public HttpContentType getContentType() {
return HttpContentType.get(contentType);
}
public int getContentTypeId() {
return contentType;
}
public byte[] getDate() {
return date;
}
protected int getDecodeState() {
return decodeState;
}
@Override
public String getFrameName() {
return getRequestURL();
}
HttpHeader getHeader(String name) {
HttpHeader header = HttpHeader.ALL.get(name);
if (header == null) {
return HttpHeader.ALL.get(name.toLowerCase());
}
return header;
}
protected int getHeaderLength() {
return headerLength;
}
public String getHost() {
return getRequestHeader(HttpHeader.Host);
}
public HttpMethod getMethod() {
return HttpMethod.get(method);
}
public int getMethodId() {
return method;
}
public String getRequestHeader(HttpHeader name) {
return request_headers.get(name.getId());
}
public String getRequestHeader(int name) {
return request_headers.get(name);
}
public IntMap getRequestHeaders() {
return request_headers;
}
public String getRequestParam(String key) {
return params.get(key);
}
public Map getRequestParams() {
return params;
}
/**
*
*
* First line of HTTP request
* Returned Value
*
* POST /some/path.html HTTP/1.1
*
* /some/path.html
*
* GET http://foo.bar/a.html HTTP/1.0
*
* /a.html
*
* GET /xyz?a=b HTTP/1.1
*
* /xyz
*
*/
public String getRequestURL() {
return requestURL;
}
public IntMap getResponseHeaders() {
return response_headers;
}
public HttpStatus getStatus() {
return HttpStatus.get(status);
}
public int getStatusId() {
return status;
}
public boolean hasContent() {
return getContent() != null;
}
@Override
public int headerLength() {
return 0;
}
public boolean isForm() {
return isForm;
}
public boolean isGet() {
return method == HttpMethod.GET.getId();
}
public HttpFrame reset() {
this.requestURL = null;
this.method = 0;
this.contentLength = 0;
this.headerLength = 0;
this.isForm = false;
this.contentType = HttpContentType.text_plain_utf8.getId();
this.connection = HttpConnection.KEEP_ALIVE.getId();
this.status = HttpStatus.C200.getStatus();
this.decodeState = HttpCodec.decode_state_line_one;
this.params.clear();
this.request_headers.clear();
if (response_headers != null) {
response_headers.clear();
}
super.reset();
return this;
}
public void setConnection(HttpConnection connection) {
this.connection = connection.getId();
}
protected void setContentLength(int contentLength) {
this.contentLength = contentLength;
}
public void setContentType(HttpContentType contentType) {
this.contentType = contentType.getId();
}
public void setDate(byte[] date) {
this.date = date;
}
protected void setDecodeState(int decodeState) {
this.decodeState = decodeState;
}
public void setForm(boolean isForm) {
this.isForm = isForm;
}
protected void setHeaderLength(int headerLength) {
this.headerLength = headerLength;
}
public void setMethod(HttpMethod method) {
this.method = method.getId();
}
void setReadHeader(String name, String value) {
HttpHeader header = getHeader(name);
if (header != null) {
request_headers.put(header.getId(), value);
}
}
public void setRequestHeader(HttpHeader header, String value) {
this.request_headers.put(header.getId(), value);
}
public void setRequestHeaders(IntMap requestHeaders) {
this.request_headers = requestHeaders;
}
public void setRequestParams(Map params) {
this.params = params;
}
public void setRequestURL(String url) {
this.requestURL = url;
}
public void setResponseHeader(HttpHeader name, byte[] value) {
setResponseHeader0(name.getId(), value, false);
}
public void setResponseHeader(int name, byte[] value) {
setResponseHeader0(name, value, false);
}
private void setResponseHeader0(int name, byte[] value, boolean absent) {
if (response_headers == null) {
response_headers = new IntMap<>(8);
}
response_headers.put(name, value);
}
public void setResponseHeaderIfAbsent(HttpHeader name, byte[] value) {
setResponseHeader0(name.getId(), value, true);
}
public void setResponseHeaderIfAbsent(int name, byte[] value) {
setResponseHeader0(name, value, true);
}
public void setReuestParam(String key, String value) {
if (params == null) {
params = new HashMap<>();
}
this.params.put(key, value);
}
public void setStatus(HttpStatus status) {
this.status = status.getStatus();
}
public String toString() {
return getRequestURL();
}
public boolean updateWebSocketProtocol(final Channel ch) throws Exception {
String Sec_WebSocket_Key_Value = getRequestHeader(Sec_WebSocket_Key);
if (!Util.isNullOrBlank(Sec_WebSocket_Key_Value)) {
//FIXME 258EAFA5-E914-47DA-95CA-C5AB0DC85B11 必须这个值？
String Sec_WebSocket_Key_Magic = Sec_WebSocket_Key_Value + "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
byte[] key_array = Cryptos.SHA1(Sec_WebSocket_Key_Magic);
String acceptKey = Cryptos.base64_en(key_array);
setStatus(HttpStatus.C101);
setConnection(HttpConnection.UPGRADE);
setResponseHeader(Upgrade, HttpStatic.websocket_bytes);
setResponseHeader(Sec_WebSocket_Accept, acceptKey.getBytes());
((HttpAttachment) ch.getAttachment()).setWebsocketFrameName(getFrameName());
ByteBuf buf = ch.encode(this);
ch.setCodec(WebSocketCodec.PROTOCOL_ID);
ch.writeAndFlush(buf);
return true;
}
return false;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy