/**
* Copyright 2012 Ronen Hamias, Anton Kharenko
*
* 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 io.scalecube.socketio.pipeline;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.net.SocketAddress;
import java.util.List;
import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandler;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.handler.codec.http.FullHttpRequest;
import io.netty.handler.codec.http.HttpMethod;
import io.netty.handler.codec.http.QueryStringDecoder;
import io.netty.util.CharsetUtil;
import io.netty.util.ReferenceCountUtil;
import io.scalecube.socketio.TransportType;
import io.scalecube.socketio.packets.ConnectPacket;
import io.scalecube.socketio.packets.Packet;
import io.scalecube.socketio.serialization.PacketFramer;
@ChannelHandler.Sharable
public class JsonpPollingHandler extends ChannelInboundHandlerAdapter {
private final Logger log = LoggerFactory.getLogger(getClass());
private final String connectPath;
private final String remoteAddressHeader;
public JsonpPollingHandler(final String handshakePath, final String remoteAddressHeader) {
this.connectPath = handshakePath + TransportType.JSONP_POLLING.getName();
this.remoteAddressHeader = remoteAddressHeader;
}
@Override
public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
if (msg instanceof FullHttpRequest) {
final FullHttpRequest req = (FullHttpRequest) msg;
final HttpMethod requestMethod = req.method();
final QueryStringDecoder queryDecoder = new QueryStringDecoder(req.uri());
final String requestPath = queryDecoder.path();
if (requestPath.startsWith(connectPath)) {
if (log.isDebugEnabled())
log.debug("Received HTTP JSONP-Polling request: {} {} from channel: {}", requestMethod, requestPath, ctx.channel());
final String sessionId = PipelineUtils.getSessionId(requestPath);
final String origin = PipelineUtils.getOrigin(req);
if (HttpMethod.GET.equals(requestMethod)) {
// Process polling request from client
SocketAddress clientIp = PipelineUtils.resolveClientIpByRemoteAddressHeader(req, remoteAddressHeader);
String jsonpIndexParam = PipelineUtils.extractParameter(queryDecoder, "i");
final ConnectPacket packet = new ConnectPacket(sessionId, origin);
packet.setTransportType(TransportType.JSONP_POLLING);
packet.setJsonpIndexParam(jsonpIndexParam);
packet.setRemoteAddress(clientIp);
ctx.fireChannelRead(packet);
} else if (HttpMethod.POST.equals(requestMethod)) {
// Process message request from client
ByteBuf buffer = req.content();
String content = buffer.toString(CharsetUtil.UTF_8);
if (content.startsWith("d=")) {
QueryStringDecoder queryStringDecoder = new QueryStringDecoder(content, CharsetUtil.UTF_8, false);
content = PipelineUtils.extractParameter(queryStringDecoder, "d");
content = prepareJsonpContent(content);
ByteBuf buf = PipelineUtils.copiedBuffer(ctx.alloc(), content);
List<Packet> packets = PacketFramer.decodePacketsFrame(buf);
buf.release();
for (Packet packet : packets) {
packet.setSessionId(sessionId);
packet.setOrigin(origin);
ctx.fireChannelRead(packet);
}
} else {
log.warn("Can't process HTTP JSONP-Polling message. Incorrect content format: {} from channel: {}", content,
ctx.channel());
}
} else {
log.warn("Can't process HTTP JSONP-Polling request. Unknown request method: {} from channel: {}", requestMethod,
ctx.channel());
}
ReferenceCountUtil.release(msg);
return;
}
}
super.channelRead(ctx, msg);
}
private String prepareJsonpContent(String content) {
if (content.startsWith("\"")) {
content = content.substring(1);
}
if (content.endsWith("\"")) {
content = content.substring(0, content.length() - 1);
}
// RemoveFix extra slashes
content = content.replace("\\\\", "\\");
content = content.replace("\\\"", "\"");
return content;
}
}