src/sim/main.cc

   1 /*
   2  * Copyright (c) 2008 The Hewlett-Packard Development Company
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions are
   7  * met: redistributions of source code must retain the above copyright
   8  * notice, this list of conditions and the following disclaimer;
   9  * redistributions in binary form must reproduce the above copyright
  10  * notice, this list of conditions and the following disclaimer in the
  11  * documentation and/or other materials provided with the distribution;
  12  * neither the name of the copyright holders nor the names of its
  13  * contributors may be used to endorse or promote products derived from
  14  * this software without specific prior written permission.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  17  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  18  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  19  * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  20  * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  21  * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  22  * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  23  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  24  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  25  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  26  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #include <Python.h>
  30
  31 #include "sim/init.hh"
  32 #include "sim/init_signals.hh"
  33
  34 // main() is now pretty stripped down and just sets up python and then
  35 // calls initM5Python which loads the various embedded python modules
  36 // into the python environment and then starts things running by
  37 // calling m5Main.
  38 int
  39 main(int argc, char **argv)
  40 {
  41     int ret;
  42
  43     // Initialize m5 special signal handling.
  44     initSignals();
  45
  46 #if PY_MAJOR_VERSION >= 3
  47     std::unique_ptr<wchar_t[], decltype(&PyMem_RawFree)> program(
  48         Py_DecodeLocale(argv[0], NULL),
  49         &PyMem_RawFree);
  50     Py_SetProgramName(program.get());
  51 #else
  52     Py_SetProgramName(argv[0]);
  53 #endif
  54
  55     // Register native modules with Python's init system before
  56     // initializing the interpreter.
  57     registerNativeModules();
  58
  59     // initialize embedded Python interpreter
  60     Py_Initialize();
  61
  62     // Initialize the embedded m5 python library
  63     ret = EmbeddedPython::initAll();
  64
  65     if (ret == 0) {
  66         // start m5
  67         ret = m5Main(argc, argv);
  68     }
  69
  70     // clean up Python intepreter.
  71     Py_Finalize();
  72
  73     return ret;
  74 }