1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
/*
* Copyright 2018 Google, Inc.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are
* met: redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer;
* redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution;
* neither the name of the copyright holders nor the names of its
* contributors may be used to endorse or promote products derived from
* this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*
* Authors: Gabe Black
*/
#include "systemc/sc_main.hh"
#include <cstring>
#include "base/logging.hh"
#include "python/pybind11/pybind.hh"
#include "sim/init.hh"
// A default version of this function in case one isn't otherwise defined.
// This ensures everything will link properly whether or not the user defined
// a custom sc_main function. If they didn't but still try to call it, throw
// an error and die.
[[gnu::weak]] int
sc_main(int argc, char *argv[])
{
// If python attempts to call sc_main but no sc_main was defined...
fatal("sc_main called but not defined.\n");
}
namespace sc_core
{
namespace
{
bool scMainCalled = false;
int _argc = 0;
char **_argv = NULL;
// This wrapper adapts the python version of sc_main to the c++ version.
void
sc_main(pybind11::args args)
{
panic_if(scMainCalled, "sc_main called more than once.");
_argc = args.size();
_argv = new char *[_argc];
// Initialize all the _argvs to NULL so we can delete [] them
// unconditionally.
for (int idx = 0; idx < _argc; idx++)
_argv[idx] = NULL;
// Attempt to convert all the arguments to strings. If that fails, clean
// up after ourselves. Also don't count this as a call to sc_main since
// we never got to the c++ version of that function.
try {
for (int idx = 0; idx < _argc; idx++) {
std::string arg = args[idx].cast<std::string>();
_argv[idx] = new char[arg.length() + 1];
strcpy(_argv[idx], arg.c_str());
}
} catch (...) {
// If that didn't work for some reason (probably a conversion error)
// blow away _argv and _argc and pass on the exception.
for (int idx = 0; idx < _argc; idx++)
delete [] _argv[idx];
delete [] _argv;
_argc = 0;
throw;
}
// At this point we're going to call the c++ sc_main, so we can't try
// again later.
scMainCalled = true;
//TODO Start a new fiber to call sc_main from.
::sc_main(_argc, _argv);
}
// Make our sc_main wrapper available in the internal _m5 python module under
// the systemc submodule.
void
systemc_pybind(pybind11::module &m_internal)
{
pybind11::module m = m_internal.def_submodule("systemc");
m.def("sc_main", &sc_main);
}
EmbeddedPyBind embed_("systemc", &systemc_pybind);
} // anonymous namespace
int
sc_argc()
{
return _argc;
}
const char *const *
sc_argv()
{
return _argv;
}
} // namespace sc_core
|
