HttpHandler.java

/*
 * Copyright 2014-2016 CyberVision, Inc.
 *
 * 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 org.kaaproject.kaa.server.transports.http.transport;

import static io.netty.handler.codec.http.HttpVersion.HTTP_1_1;

import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.handler.codec.http.DefaultFullHttpResponse;
import io.netty.handler.codec.http.HttpResponseStatus;

import org.kaaproject.kaa.server.common.server.NettyChannelContext;
import org.kaaproject.kaa.server.transport.EndpointRevocationException;
import org.kaaproject.kaa.server.transport.EndpointVerificationException;
import org.kaaproject.kaa.server.transport.InvalidSdkTokenException;
import org.kaaproject.kaa.server.transport.message.ErrorBuilder;
import org.kaaproject.kaa.server.transport.message.MessageBuilder;
import org.kaaproject.kaa.server.transport.message.MessageHandler;
import org.kaaproject.kaa.server.transports.http.transport.commands.AbstractHttpSyncCommand;
import org.kaaproject.kaa.server.transports.http.transport.messages.NettyHttpSyncMessage;
import org.kaaproject.kaa.server.transports.http.transport.netty.AbstractCommand;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.security.GeneralSecurityException;
import java.util.UUID;


public class HttpHandler extends SimpleChannelInboundHandler<AbstractCommand>
    implements MessageBuilder, ErrorBuilder {

  private static final Logger LOG = LoggerFactory.getLogger(HttpHandler.class);

  private final MessageHandler messageHandler;

  private final UUID uuid;

  private volatile AbstractHttpSyncCommand command;

  /**
   * Instantiates a new akka handler.
   *
   * @param uuid           - session uuid
   * @param messageHandler the message handler
   */
  public HttpHandler(UUID uuid, MessageHandler messageHandler) {
    super();
    this.messageHandler = messageHandler;
    this.uuid = uuid;
  }

  /*
   * (non-Javadoc)
   *
   * @see
   * org.kaaproject.kaa.server.common.http.server.DefaultHandler#channelRead0
   * (io.netty.channel.ChannelHandlerContext,
   * org.kaaproject.kaa.server.common.http.server.CommandProcessor)
   */
  @Override
  protected void channelRead0(final ChannelHandlerContext ctx, final AbstractCommand msg)
      throws Exception {
    this.command = (AbstractHttpSyncCommand) msg;
    NettyHttpSyncMessage message = new NettyHttpSyncMessage(uuid, msg.getNextProtocol(),
        new NettyChannelContext(ctx), command.getChannelType(),
        command, this, this);
    LOG.trace("Forwarding {} to handler", message);
    messageHandler.process(message);
  }

  @Override
  public Object[] build(Exception exception) {
    HttpResponseStatus status;
    if (exception instanceof EndpointVerificationException) {
      status = HttpResponseStatus.UNAUTHORIZED;
    } else if (exception instanceof EndpointRevocationException) {
      status = HttpResponseStatus.FORBIDDEN;
    } else if (exception instanceof GeneralSecurityException
        || exception instanceof IOException
        || exception instanceof IllegalArgumentException
        || exception instanceof InvalidSdkTokenException) {
      status = HttpResponseStatus.BAD_REQUEST;
    } else {
      status = HttpResponseStatus.INTERNAL_SERVER_ERROR;
    }

    command.setResponse(new DefaultFullHttpResponse(HTTP_1_1, status));
    return new Object[]{command};
  }

  @Override
  public Object[] build(byte[] responseData, byte[] responseSignatureData, boolean isEncrypted) {
    command.setResponseBody(responseData);
    if (responseSignatureData != null) {
      command.setResponseSignature(responseSignatureData);
    }
    return new Object[]{command};
  }

  @Override
  public Object[] build(byte[] messageData, boolean isEncrypted) {
    return build(messageData, null, isEncrypted);
  }
}