/FileExtraction/Java_unseen/dlitz_resin/modules/resin/src/com/caucho/cloud/hmtp/HmtpRequest.java
/*
 * Copyright (c) 1998-2011 Caucho Technology -- all rights reserved
 *
 * This file is part of Resin(R) Open Source
 *
 * Each copy or derived work must preserve the copyright notice and this
 * notice unmodified.
 *
 * Resin Open Source is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *
 * Resin Open Source is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE, or any warranty
 * of NON-INFRINGEMENT.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Resin Open Source; if not, write to the
 *
 *   Free Software Foundation, Inc.
 *   59 Temple Place, Suite 330
 *   Boston, MA 02111-1307  USA
 *
 * @author Scott Ferguson
 */

package com.caucho.cloud.hmtp;

import java.io.IOException;
import java.io.InputStream;

import com.caucho.bam.broker.Broker;
import com.caucho.bam.broker.ManagedBroker;
import com.caucho.bam.broker.PassthroughBroker;
import com.caucho.bam.mailbox.Mailbox;
import com.caucho.bam.mailbox.MultiworkerMailbox;
import com.caucho.cloud.bam.BamSystem;
import com.caucho.hemp.servlet.ClientStubManager;
import com.caucho.hemp.servlet.ServerGatewayBroker;
import com.caucho.hemp.servlet.ServerProxyBroker;
import com.caucho.hmtp.HmtpWebSocketReader;
import com.caucho.hmtp.HmtpWebSocketWriter;
import com.caucho.network.listen.AbstractProtocolConnection;
import com.caucho.network.listen.SocketLink;
import com.caucho.util.L10N;
import com.caucho.vfs.ReadStream;
import com.caucho.vfs.WriteStream;

/**
 * Handles HMTP requests from a peer server.
 * 
 * If the request does not match one of the HMTP codes, it the HmtpRequest
 * will forward to the HMUX request for compatibility.
 */
public class HmtpRequest extends AbstractProtocolConnection
{
  private static final L10N L = new L10N(HmtpRequest.class);

  public static final int HMUX_TO_UNIDIR_HMTP = '7';
  public static final int HMUX_SWITCH_TO_HMTP = '8';
  public static final int HMUX_HMTP_OK        = '9';
  
  private SocketLink _conn;
  private BamSystem _bamService;
  
  private ReadStream _rawRead;
  private WriteStream _rawWrite;
  
  private boolean _isFirst;

  private HmtpWebSocketReader _hmtpReader;
  private HmtpWebSocketWriter _hmtpWriter;

  private ClientStubManager _clientManager;
  private Broker _toLinkBroker;
  private Broker _proxyBroker;
  
  private HmtpLinkActor _linkActor;

  public HmtpRequest(SocketLink conn,
                     BamSystem bamService)
  {
    _conn = conn;
    _bamService = bamService;
    
    if (conn == null)
      throw new NullPointerException();
    
    if (bamService == null)
      throw new NullPointerException();

    _rawRead = conn.getReadStream();
    _rawWrite = conn.getWriteStream();
  }

  @Override
  public boolean isWaitForRead()
  {
    return true;
  }
  
  @Override
  public void onStartConnection()
  {
    _isFirst = true;
  }

  @Override
  public boolean handleRequest()
    throws IOException
  {
    try {
      if (_isFirst) {
        return handleInitialRequest();
      }
      else {
        return dispatchHmtp();
      }
    } catch (RuntimeException e) {
      throw e;
    } catch (IOException e) {
      throw e;
    }
  }

  private boolean handleInitialRequest()
    throws IOException
  {
    _isFirst = false;
    
    ReadStream is = _rawRead;
    
    int ch = is.read();
    
    if (ch < 0)
      return false;
    
    boolean isUnidir = false;
    
    if (ch == HMUX_TO_UNIDIR_HMTP)
      isUnidir = true;
    else if (ch == HMUX_SWITCH_TO_HMTP)
      isUnidir = false;
    else
      throw new UnsupportedOperationException(L.l("0x{0} is an invalid HMUX code.",
                                                  Integer.toHexString(ch)));

    int len = (is.read() << 8) + is.read();
    int adminCode = is.read();
    boolean isAdmin = adminCode != 0;

    InputStream rawIs = is;
    
    is.skip(len - 1);

    _hmtpReader = new HmtpWebSocketReader(rawIs);

    _hmtpWriter = new HmtpWebSocketWriter(_rawWrite);
    // _hmtpWriter.setId(getRequestId());
    // _hmtpWriter.setAutoFlush(true);

    ManagedBroker broker = _bamService.getBroker();

    _hmtpWriter.setAddress("hmtp-server-" + _conn.getId() + "-hmtp");

    Mailbox toLinkMailbox = new MultiworkerMailbox(_hmtpWriter.getAddress(), _hmtpWriter, broker, 1);
    _toLinkBroker = new PassthroughBroker(toLinkMailbox);
    
    _clientManager = new ClientStubManager(broker, toLinkMailbox);

    _linkActor = new HmtpLinkActor(_toLinkBroker,
                                   _clientManager,
                                   _bamService.getLinkManager(),
                                   _conn.getRemoteHost());
    System.out.println("UNIDIR: " +isUnidir);
    if (isUnidir) {
      _proxyBroker = new ServerGatewayBroker(broker,
                                             _clientManager, 
                                             _linkActor.getActor());
    }
    else {
      _proxyBroker = new ServerProxyBroker(broker,
                                           _clientManager, 
                                           _linkActor.getActor());
    }

    return dispatchHmtp();
  }

  private boolean dispatchHmtp()
    throws IOException
  {
    HmtpWebSocketReader in = _hmtpReader;

    do {
      Broker broker = _proxyBroker;

      if (! in.readPacket(broker)) {
        return false;
      }
    } while (in.isDataAvailable());

    return true;
  }

  /**
   * Close when the socket closes.
   */
  @Override
  public void onCloseConnection()
  {
    HmtpLinkActor linkActor = _linkActor;
    _linkActor = null;

    Broker linkBroker = _toLinkBroker;
    _toLinkBroker = null;
    
    if (linkBroker != null)
      linkBroker.close();

    if (linkActor != null) {
      linkActor.onCloseConnection();
    }

    /*
    if (linkStream != null)
      linkStream.close();
      */

    HmtpWebSocketWriter writer = _hmtpWriter;

    if (writer != null)
      writer.close();
  }

  protected String getRequestId()
  {
    return "hmtp" + ":" + _conn.getId();
  }

  public final String dbgId()
  {
    return "Hmtp[" + _conn.getId() + "] ";
  }

  @Override
  public String toString()
  {
    return getClass().getSimpleName() + dbgId();
  }
}
