hlvm/hlvm/Runtime/Main.cpp

//===-- Runtime Main Implementation -----------------------------*- C++ -*-===//
//
//                      High Level Virtual Machine (HLVM)
//
// Copyright (C) 2006 Reid Spencer. All Rights Reserved.
//
// This software is free software; you can redistribute it and/or modify it 
// under the terms of the GNU Lesser General Public License as published by 
// the Free Software Foundation; either version 2.1 of the License, or (at 
// your option) any later version.
//
// This software 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.  See the GNU Lesser General Public License for 
// more details.
//
// You should have received a copy of the GNU Lesser General Public License 
// along with this library in the file named LICENSE.txt; if not, write to the 
// Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, 
// MA 02110-1301 USA
//
//===----------------------------------------------------------------------===//
/// @file hlvm/Runtime/Main.cpp
/// @author Reid Spencer <rspencer@reidspencer.org> (original author)
/// @date 2006/06/04
/// @since 0.1.0
/// @brief Implements the runtime main program.
//===----------------------------------------------------------------------===//

#include <apr-1/apr_getopt.h>
#include <apr-1/apr_file_io.h>
#include <stdlib.h>
#include <hlvm/Runtime/Internal.h>
#include <hlvm/Base/Configuration.h>

namespace {

apr_getopt_option_t hlvm_options[] = {
  { "help", 'h', 0, "Print help on using HLVM" },
  { "version", 'v', 0, "Print version information" },
  { 0, 0, 0, 0 }
};

void print_version( bool should_exit = true )
{
  apr_file_printf(_hlvm_stderr, 
    "HLVM Runtime Version %s; %s\n", HLVM_Version, HLVM_ConfigTime);
  apr_file_printf(_hlvm_stderr, "%s\n", HLVM_Copyright);
  apr_file_printf(_hlvm_stderr, "Contact %s For Details\n", HLVM_Maintainer);
  if (should_exit)
    exit(0);
}

void print_help()
{
  print_version(false);
  apr_file_printf(_hlvm_stderr,"\nOptions:\n");
  for (unsigned i = 0; i < sizeof(hlvm_options)/sizeof(hlvm_options[0])-1; ++i) {
    apr_file_printf(_hlvm_stderr, "  --%-15s  %-74s\n", 
      hlvm_options[i].name, hlvm_options[i].description);
  }
  exit(0);
}

void print_error(void*, const char* fmt, ...)
{
  va_list ap;
  va_start(ap,fmt);
  char* msg = apr_pvsprintf(_hlvm_pool, fmt, ap);
  hlvm_fatal(E_OPTION_ERROR,msg);
}

}

extern "C" {

#include <hlvm/Runtime/Main.h>
#include <hlvm/Runtime/Program.h>
#include <hlvm/Runtime/Memory.h>


/// This is the function called from the real main() in hlvm/tools/hlvm.  We 
/// do this because we don't want to expose the "Main" class to the outside
/// world. The interface to the HLVM Runtime is C even though the
/// implementation uses C++.
int hlvm_runtime_main(int argc, char**argv)
{
  int result = 0;
  // Initialize APR and HLVM
  _hlvm_initialize();

  // Process the options
  apr_getopt_t* options = 0;
  if (APR_SUCCESS != apr_getopt_init(&options, _hlvm_pool, argc, argv))
    hlvm_panic("Can't initialize apr_getopt");
  options->interleave = 0;
  options->errfn = print_error;
  options->errarg = _hlvm_stderr;
  int ch;
  const char* arg;
  hlvm_program_type entry_func = 0;
  apr_status_t stat = apr_getopt_long(options, hlvm_options, &ch, &arg); 
  while (stat != APR_EOF) {
    switch (stat) {
      case APR_SUCCESS:
      {
        switch (ch) {
          case 'h': print_help(); break;
          case 'v': print_version(); break;
          default:
            break;
        }
        break;
      }
      case APR_BADCH:
      {
        char option[3];
        option[0] = ch;
        option[1] = 0;
        hlvm_error(E_BAD_OPTION,option);
        return E_BAD_OPTION;
      }
      case APR_BADARG:
      {
        hlvm_error(E_MISSING_ARGUMENT);
        return E_MISSING_ARGUMENT;
      }
      case APR_EOF:
      default:
      {
        hlvm_panic("Unknown response from apr_getopt_long");
        break;
      }
    }
    stat = apr_getopt_long(options, hlvm_options, &ch, &arg); 
  }

  const char* prog_name = argv[options->ind];
  if (prog_name) {
    entry_func = hlvm_find_program(prog_name);
    // If we didn't get a start function ..
    if (!entry_func) {
      // Give an error
      hlvm_error(E_PROGRAM_NOT_FOUND,prog_name);
      return E_PROGRAM_NOT_FOUND;
    }
    result = (*entry_func)(options->argc, options->argv);
  } else {
    hlvm_error(E_NO_PROGRAM_NAME,0);
    return E_NO_PROGRAM_NAME;
  }
  return result;
}

} // end extern "C"