io.scalecube.socketio.pipeline.JsonpPollingHandler.java Source code

Introduction

Here is the source code for io.scalecube.socketio.pipeline.JsonpPollingHandler.java

Source

/**
 * 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.getMethod();
            final QueryStringDecoder queryDecoder = new QueryStringDecoder(req.getUri());
            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.getHeaderClientIPParamValue(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 = preprocessJsonpContent(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 preprocessJsonpContent(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;
    }

}