hhvm/hphp/compiler/statement/try_statement.cpp

/*
   +----------------------------------------------------------------------+
   | HipHop for PHP                                                       |
   +----------------------------------------------------------------------+
   | Copyright (c) 2010- Facebook, Inc. (http://www.facebook.com)         |
   +----------------------------------------------------------------------+
   | This source file is subject to version 3.01 of the PHP license,      |
   | that is bundled with this package in the file LICENSE, and is        |
   | available through the world-wide-web at the following url:           |
   | http://www.php.net/license/3_01.txt                                  |
   | If you did not receive a copy of the PHP license and are unable to   |
   | obtain it through the world-wide-web, please send a note to          |
   | license@php.net so we can mail you a copy immediately.               |
   +----------------------------------------------------------------------+
*/

#include <compiler/statement/try_statement.h>
#include <compiler/statement/statement_list.h>
#include <compiler/analysis/function_scope.h>

using namespace HPHP;

///////////////////////////////////////////////////////////////////////////////
// constructors/destructors

TryStatement::TryStatement
(STATEMENT_CONSTRUCTOR_PARAMETERS,
 StatementPtr tryStmt, StatementListPtr catches)
  : Statement(STATEMENT_CONSTRUCTOR_PARAMETER_VALUES(TryStatement)),
    m_tryStmt(tryStmt), m_catches(catches) {
}

TryStatement::TryStatement
(STATEMENT_CONSTRUCTOR_PARAMETERS,
 StatementPtr tryStmt, StatementListPtr catches, StatementPtr finallyStmt)
  : Statement(STATEMENT_CONSTRUCTOR_PARAMETER_VALUES(TryStatement)),
    m_tryStmt(tryStmt), m_catches(catches), m_finallyStmt(finallyStmt) {
}

StatementPtr TryStatement::clone() {
  TryStatementPtr stmt(new TryStatement(*this));
  stmt->m_tryStmt = Clone(m_tryStmt);
  if (m_catches)
    stmt->m_catches = Clone(m_catches);
  if (m_finallyStmt)
    stmt->m_finallyStmt = Clone(m_finallyStmt);
  return stmt;
}

int TryStatement::getRecursiveCount() const {
  return (m_tryStmt ? m_tryStmt->getRecursiveCount() : 0) +
    (m_catches->getCount() ? m_catches->getRecursiveCount() : 0) + 
    (m_finallyStmt ? m_finallyStmt->getRecursiveCount() : 0);
}

///////////////////////////////////////////////////////////////////////////////
// parser functions

///////////////////////////////////////////////////////////////////////////////
// static analysis functions

void TryStatement::analyzeProgram(AnalysisResultPtr ar) {
  if (m_tryStmt) m_tryStmt->analyzeProgram(ar);
  if (m_catches)
    m_catches->analyzeProgram(ar);
  if (m_finallyStmt)
    m_finallyStmt->analyzeProgram(ar);
  if (ar->getPhase() == AnalysisResult::AnalyzeAll) {
    FunctionScopeRawPtr fs = getFunctionScope();
    if (fs) fs->setHasTry();
  }
}

bool TryStatement::hasDecl() const {
  if (m_tryStmt && m_tryStmt->hasDecl()) return true;
  if (m_catches && m_catches->hasDecl()) return true;
  if (m_finallyStmt && m_finallyStmt->hasDecl()) return true;
  return false;
}

bool TryStatement::hasRetExp() const {
  if (m_tryStmt && m_tryStmt->hasRetExp()) return true;
  if (m_catches && m_catches->hasRetExp()) return true;
  if (m_finallyStmt && m_finallyStmt->hasRetExp()) return true;
  return false;
}

ConstructPtr TryStatement::getNthKid(int n) const {
  switch (n) {
    case 0:
      return m_tryStmt;
    case 1:
      return m_catches;
    case 2:
      return m_finallyStmt;
    default:
      assert(false);
      break;
  }
  return ConstructPtr();
}

int TryStatement::getKidCount() const {
  return 3;
}

void TryStatement::setNthKid(int n, ConstructPtr cp) {
  switch (n) {
    case 0:
      m_tryStmt = boost::dynamic_pointer_cast<Statement>(cp);
      break;
    case 1:
      m_catches = boost::dynamic_pointer_cast<StatementList>(cp);
      break;
    case 2:
      m_finallyStmt = boost::dynamic_pointer_cast<Statement>(cp);
    default:
      assert(false);
      break;
  }
}

void TryStatement::inferTypes(AnalysisResultPtr ar) {
  if (m_tryStmt) m_tryStmt->inferTypes(ar);
  if (m_catches) m_catches->inferTypes(ar);
  if (m_finallyStmt) m_finallyStmt->inferTypes(ar);
}

///////////////////////////////////////////////////////////////////////////////
// code generation functions

void TryStatement::outputPHP(CodeGenerator &cg, AnalysisResultPtr ar) {
  cg_indentBegin("try {\n");
  if (m_tryStmt) m_tryStmt->outputPHP(cg, ar);
  cg_indentEnd("}");
  if (m_catches->getCount())
    m_catches->outputPHP(cg, ar);
  cg_printf("\n");
}

void TryStatement::outputCPPImpl(CodeGenerator &cg, AnalysisResultPtr ar) {
  int c_count = m_catches->getCount();
  bool has_finally = m_finallyStmt ? true : false;
  
  assert(c_count || has_finally);

  if (has_finally) {
    cg_indentBegin("{\n");
    cg_printf("Object exception;\n");
    cg_printf("bool has_exception = false;\n");
    cg_indentBegin("try {\n");
  }
  if (c_count) cg_indentBegin("try {\n");
  if (m_tryStmt) m_tryStmt->outputCPP(cg, ar);
  if (c_count) cg_indentEnd("}");
  if (c_count) {
    cg_indentBegin(" catch (Object e) {\n");
    for (int i = 0; i < c_count; i++) {
      if (i > 0) cg_printf(" else ");
      (*m_catches)[i]->outputCPP(cg, ar);
    }
    cg_printf(" else {\n");
    cg_printf("  throw;\n");
    cg_printf("}\n");
    cg_indentEnd("}\n");
  }
  if (has_finally) {
    cg_indentEnd("}");
    cg_indentBegin(" catch (Object e) {\n");
    cg_printf("exception = e;\n");
    cg_printf("has_exception = true;\n");
    cg_indentEnd("}\n");
    m_finallyStmt->outputCPP(cg, ar);
    cg_printf("if (has_exception) throw exception;\n");
    cg_indentEnd("}\n");
  }
}