/*
+ * Copyright (c) 2012 ARM Limited
+ * All rights reserved
+ *
+ * The license below extends only to copyright in the software and shall
+ * not be construed as granting a license to any other intellectual
+ * property including but not limited to intellectual property relating
+ * to a hardware implementation of the functionality of the software
+ * licensed hereunder. You may use the software subject to the license
+ * terms below provided that you ensure that this notice is replicated
+ * unmodified and in its entirety in all distributions of the software,
+ * modified or unmodified, in source code or in binary form.
+ *
* Copyright (c) 2000-2005 The Regents of The University of Michigan
* Copyright (c) 2008 The Hewlett-Packard Development Company
* All rights reserved.
*/
#include <Python.h>
+
#include <marshal.h>
-#include <signal.h>
+#include <zlib.h>
#include <iostream>
+#include <list>
#include <string>
-#include <zlib.h>
#include "base/cprintf.hh"
#include "base/misc.hh"
+#include "base/types.hh"
+#include "config/have_protobuf.hh"
#include "sim/async.hh"
#include "sim/core.hh"
-#include "base/types.hh"
#include "sim/init.hh"
-using namespace std;
-
-/// Stats signal handler.
-void
-dumpStatsHandler(int sigtype)
-{
- async_event = true;
- async_statdump = true;
-}
+#if HAVE_PROTOBUF
+#include <google/protobuf/stubs/common.h>
+#endif
-void
-dumprstStatsHandler(int sigtype)
-{
- async_event = true;
- async_statdump = true;
- async_statreset = true;
-}
+using namespace std;
-/// Exit signal handler.
-void
-exitNowHandler(int sigtype)
-{
- async_event = true;
- async_exit = true;
-}
+// The python library is totally messed up with respect to constness,
+// so make a simple macro to make life a little easier
+#define PyCC(x) (const_cast<char *>(x))
-/// Abort signal handler.
-void
-abortHandler(int sigtype)
+EmbeddedPython *EmbeddedPython::importer = NULL;
+PyObject *EmbeddedPython::importerModule = NULL;
+EmbeddedPython::EmbeddedPython(const char *filename, const char *abspath,
+ const char *modpath, const unsigned char *code, int zlen, int len)
+ : filename(filename), abspath(abspath), modpath(modpath), code(code),
+ zlen(zlen), len(len)
{
- ccprintf(cerr, "Program aborted at cycle %d\n", curTick);
+ // if we've added the importer keep track of it because we need it
+ // to bootstrap.
+ if (string(modpath) == string("importer"))
+ importer = this;
+ else
+ getList().push_back(this);
}
-/*
- * M5 can do several special things when various signals are sent.
- * None are mandatory.
- */
-void
-initSignals()
+list<EmbeddedPython *> &
+EmbeddedPython::getList()
{
- // Floating point exceptions may happen on misspeculated paths, so
- // ignore them
- signal(SIGFPE, SIG_IGN);
-
- // We use SIGTRAP sometimes for debugging
- signal(SIGTRAP, SIG_IGN);
-
- // Dump intermediate stats
- signal(SIGUSR1, dumpStatsHandler);
-
- // Dump intermediate stats and reset them
- signal(SIGUSR2, dumprstStatsHandler);
-
- // Exit cleanly on Interrupt (Ctrl-C)
- signal(SIGINT, exitNowHandler);
-
- // Print out cycle number on abort
- signal(SIGABRT, abortHandler);
+ static list<EmbeddedPython *> the_list;
+ return the_list;
}
/*
* Uncompress and unmarshal the code object stored in the
- * EmbeddedPyModule
+ * EmbeddedPython
*/
PyObject *
-getCode(const EmbeddedPyModule *pymod)
+EmbeddedPython::getCode() const
{
- assert(pymod->zlen == pymod->code_end - pymod->code);
- Bytef *marshalled = new Bytef[pymod->mlen];
- uLongf unzlen = pymod->mlen;
- int ret = uncompress(marshalled, &unzlen, (const Bytef *)pymod->code,
- pymod->zlen);
+ Bytef marshalled[len];
+ uLongf unzlen = len;
+ int ret = uncompress(marshalled, &unzlen, (const Bytef *)code, zlen);
if (ret != Z_OK)
panic("Could not uncompress code: %s\n", zError(ret));
- assert(unzlen == pymod->mlen);
+ assert(unzlen == (uLongf)len);
- return PyMarshal_ReadObjectFromString((char *)marshalled, pymod->mlen);
+ return PyMarshal_ReadObjectFromString((char *)marshalled, len);
}
-// The python library is totally messed up with respect to constness,
-// so make a simple macro to make life a little easier
-#define PyCC(x) (const_cast<char *>(x))
+bool
+EmbeddedPython::addModule() const
+{
+ PyObject *code = getCode();
+ PyObject *result = PyObject_CallMethod(importerModule, PyCC("add_module"),
+ PyCC("sssO"), filename, abspath, modpath, code);
+ if (!result) {
+ PyErr_Print();
+ return false;
+ }
+
+ Py_DECREF(result);
+ return true;
+}
/*
* Load and initialize all of the python parts of M5, including Swig
* and the embedded module importer.
*/
int
-initM5Python()
+EmbeddedPython::initAll()
{
- extern void initSwig();
-
- // initialize SWIG modules. initSwig() is autogenerated and calls
- // all of the individual swig initialization functions.
- initSwig();
-
// Load the importer module
- PyObject *code = getCode(&embeddedPyImporter);
- PyObject *module = PyImport_ExecCodeModule(PyCC("importer"), code);
- if (!module) {
+ PyObject *code = importer->getCode();
+ importerModule = PyImport_ExecCodeModule(PyCC("importer"), code);
+ if (!importerModule) {
PyErr_Print();
return 1;
}
// Load the rest of the embedded python files into the embedded
// python importer
- const EmbeddedPyModule *pymod = &embeddedPyModules[0];
- while (pymod->filename) {
- PyObject *code = getCode(pymod);
- PyObject *result = PyObject_CallMethod(module, PyCC("add_module"),
- PyCC("ssO"), pymod->filename, pymod->modpath, code);
- if (!result) {
- PyErr_Print();
+ list<EmbeddedPython *>::iterator i = getList().begin();
+ list<EmbeddedPython *>::iterator end = getList().end();
+ for (; i != end; ++i)
+ if (!(*i)->addModule())
return 1;
- }
- Py_DECREF(result);
- ++pymod;
- }
return 0;
}
+EmbeddedSwig::EmbeddedSwig(void (*init_func)())
+ : initFunc(init_func)
+{
+ getList().push_back(this);
+}
+
+list<EmbeddedSwig *> &
+EmbeddedSwig::getList()
+{
+ static list<EmbeddedSwig *> the_list;
+ return the_list;
+}
+
+void
+EmbeddedSwig::initAll()
+{
+ // initialize SWIG modules. initSwig() is autogenerated and calls
+ // all of the individual swig initialization functions.
+ list<EmbeddedSwig *>::iterator i = getList().begin();
+ list<EmbeddedSwig *>::iterator end = getList().end();
+ for (; i != end; ++i)
+ (*i)->initFunc();
+}
+
+int
+initM5Python()
+{
+ EmbeddedSwig::initAll();
+ return EmbeddedPython::initAll();
+}
+
+/*
+ * Make the commands array weak so that they can be overridden (used
+ * by unit tests to specify a different python main function.
+ */
+const char * __attribute__((weak)) m5MainCommands[] = {
+ "import m5",
+ "m5.main()",
+ 0 // sentinel is required
+};
+
/*
* Start up the M5 simulator. This mostly vectors into the python
* main function.
int
m5Main(int argc, char **argv)
{
+#if HAVE_PROTOBUF
+ // Verify that the version of the protobuf library that we linked
+ // against is compatible with the version of the headers we
+ // compiled against.
+ GOOGLE_PROTOBUF_VERIFY_VERSION;
+#endif
+
PySys_SetArgv(argc, argv);
// We have to set things up in the special __main__ module
// import the main m5 module
PyObject *result;
- result = PyRun_String("import m5", Py_file_input, dict, dict);
- if (!result) {
- PyErr_Print();
- return 1;
- }
- Py_DECREF(result);
+ const char **command = m5MainCommands;
- // Start m5
- result = PyRun_String("m5.main()", Py_file_input, dict, dict);
- if (!result) {
- PyErr_Print();
- return 1;
+ // evaluate each command in the m5MainCommands array (basically a
+ // bunch of python statements.
+ while (*command) {
+ result = PyRun_String(*command, Py_file_input, dict, dict);
+ if (!result) {
+ PyErr_Print();
+ return 1;
+ }
+ Py_DECREF(result);
+
+ command++;
}
- Py_DECREF(result);
+
+#if HAVE_PROTOBUF
+ google::protobuf::ShutdownProtobufLibrary();
+#endif
return 0;
}