#include <Python.h>

#include "gcc-python.h"

#include "gcc-python-closure.h"

#include "gcc-python-wrappers.h"

#include "gcc-c-api/gcc-location.h"

#include "gcc-c-api/gcc-variable.h"

#include "gcc-c-api/gcc-declaration.h"

#include "gcc-c-api/gcc-diagnostics.h"

#include "gcc-c-api/gcc-option.h"

int plugin_is_GPL_compatible;

#include "plugin-version.h"

#if 1

#include "cp/name-lookup.h" /* for global_namespace */

#include "tree.h"

#include "function.h"

#include "cgraph.h"

#if (GCC_VERSION >= 4009)

#include "stringpool.h"

#include "c-pragma.h" /* for parse_in */

#if 0

#define LOG(msg) \

(void)fprintf(stderr, "%s:%i:%s\n", __FILE__, __LINE__, (msg))

#define LOG(msg) ((void)0);

#define GCC_PYTHON_TRACE_ALL_EVENTS 0

#if GCC_PYTHON_TRACE_ALL_EVENTS

static const char* event_name[] = {

#define DEFEVENT(NAME) \

#NAME,

# include "plugin.def"

# undef DEFEVENT

};

static void

trace_callback(enum plugin_event event, void *gcc_data, void *user_data)

{

fprintf(stderr,

"%s:%i:trace_callback(%s, %p, %p)\n",

__FILE__, __LINE__, event_name[event], gcc_data, user_data);

fprintf(stderr, " cfun:%p\n", cfun);

}

#define DEFEVENT(NAME) \

static void trace_callback_for_##NAME(void *gcc_data, void *user_data) \

{ \

trace_callback(NAME, gcc_data, user_data); \

}

# include "plugin.def"

# undef DEFEVENT

#endif /* GCC_PYTHON_TRACE_ALL_EVENTS */

__typeof__ (parse_in) parse_in __attribute__ ((weak));

static PyObject*

PyGcc_define_macro(PyObject *self,

PyObject *args, PyObject *kwargs)

{

const char *macro;

const char *keywords[] = {"macro",

NULL};

if (!PyArg_ParseTupleAndKeywords(args, kwargs,

"s:define_preprocessor_name", (char**)keywords,

&macro)) {

return NULL;

}

if (0) {

fprintf(stderr, "gcc.define_macro(\"%s\")\n", macro);

}

if (!parse_in) {

return PyErr_Format(PyExc_ValueError,

"gcc.define_macro(\"%s\") called without a compilation unit",

macro);

}

if (!PyGcc_IsWithinEvent(NULL)) {

return PyErr_Format(PyExc_ValueError,

"gcc.define_macro(\"%s\") called from outside an event callback",

macro);

}

cpp_define (parse_in, macro);

Py_RETURN_NONE;

}

static PyObject *

PyGcc_set_location(PyObject *self, PyObject *args)

{

PyGccLocation *loc_obj;

if (!PyArg_ParseTuple(args,

"O!:set_location",

&PyGccLocation_TypeObj, &loc_obj)) {

return NULL;

}

gcc_set_input_location(loc_obj->loc);

Py_RETURN_NONE;

}

static bool add_option_to_list(gcc_option opt, void *user_data)

{

PyObject *result = (PyObject*)user_data;

PyObject *obj_opt;

obj_opt = PyGccOption_New(opt);

if (!obj_opt) {

return true;

}

if (-1 == PyList_Append(result, obj_opt)) {

Py_DECREF(obj_opt);

return true;

}

/* Success: */

Py_DECREF(obj_opt);

return false;

}

static PyObject *

PyGcc_get_option_list(PyObject *self, PyObject *args)

{

PyObject *result;

result = PyList_New(0);

if (!result) {

return NULL;

}

if (gcc_for_each_option(add_option_to_list, result)) {

Py_DECREF(result);

return NULL;

}

/* Success: */

return result;

}

static bool add_option_to_dict(gcc_option opt, void *user_data)

{

PyObject *dict = (PyObject*)user_data;

PyObject *opt_obj;

opt_obj = PyGccOption_New(opt);

if (!opt_obj) {

return true;

}

if (-1 == PyDict_SetItemString(dict,

gcc_option_get_text(opt),

opt_obj)) {

Py_DECREF(opt_obj);

return true;

}

Py_DECREF(opt_obj);

return false;

}

static PyObject *

PyGcc_get_option_dict(PyObject *self, PyObject *args)

{

PyObject *dict;

dict = PyDict_New();

if (!dict) {

return NULL;

}

if (gcc_for_each_option(add_option_to_dict, dict)) {

Py_DECREF(dict);

return NULL;

}

/* Success: */

return dict;

}

static PyObject *

PyGcc_get_parameters(PyObject *self, PyObject *args)

{

PyObject *dict;

size_t i;

dict = PyDict_New();

if (!dict) {

goto error;

}

for (i = 0; i < get_num_compiler_params(); i++) {

PyObject *param_obj = PyGccParameter_New((compiler_param)i);

if (!param_obj) {

goto error;

}

if (-1 == PyDict_SetItemString(dict,

compiler_params[i].option,

param_obj)) {

Py_DECREF(param_obj);

goto error;

}

Py_DECREF(param_obj);

}

return dict;

error:

Py_XDECREF(dict);

return NULL;

}

IMPL_APPENDER(add_var_to_list,

gcc_variable,

PyGccVariable_New)

static PyObject *

PyGcc_get_variables(PyObject *self, PyObject *args)

{

IMPL_GLOBAL_LIST_MAKER(gcc_for_each_variable,

add_var_to_list)

}

static PyObject *

PyGcc_maybe_get_identifier(PyObject *self, PyObject *args)

{

const char *str;

tree t;

if (!PyArg_ParseTuple(args,

"s:maybe_get_identifier",

&str)) {

return NULL;

}

t = maybe_get_identifier(str);

return PyGccTree_New(gcc_private_make_tree(t));

}

static PyObject *PyGcc_make_translation_unit_decl(gcc_translation_unit_decl decl)

{

gcc_tree tree = gcc_translation_unit_decl_as_gcc_tree(decl);

return PyGccTree_New(tree);

}

IMPL_APPENDER(add_translation_unit_decl_to_list,

gcc_translation_unit_decl,

PyGcc_make_translation_unit_decl)

static PyObject *

PyGcc_get_translation_units(PyObject *self, PyObject *args)

{

IMPL_GLOBAL_LIST_MAKER(gcc_for_each_translation_unit_decl,

add_translation_unit_decl_to_list)

}

__typeof__ (global_namespace) global_namespace __attribute__ ((weak));

static PyObject *

PyGcc_get_global_namespace(PyObject *self, PyObject *args)

{

/* (global_namespace will be NULL outside the C++ frontend, giving a

return PyGccTree_New(gcc_private_make_tree(global_namespace));

}

static PyObject *

PyGcc_dump(PyObject *self, PyObject *arg)

{

PyObject *str_obj;

/*

if (!dump_file) {

/* The most common case; make it fast */

Py_RETURN_NONE;

}

str_obj = PyObject_Str(arg);

if (!str_obj) {

return NULL;

}

/* FIXME: encoding issues */

/* FIXME: GIL */

if (!fwrite(PyGccString_AsString(str_obj),

strlen(PyGccString_AsString(str_obj)),

1,

dump_file)) {

Py_DECREF(str_obj);

return PyErr_SetFromErrnoWithFilename(PyExc_IOError, dump_file_name);

}

Py_DECREF(str_obj);

Py_RETURN_NONE;

}

static PyObject *

PyGcc_get_dump_file_name(PyObject *self, PyObject *noargs)

{

/* gcc/tree-pass.h declares:

return PyGccStringOrNone(dump_file_name);

}

static PyObject *

PyGcc_get_dump_base_name(PyObject *self, PyObject *noargs)

{

/*

return PyGccStringOrNone(dump_base_name);

}

static PyObject *

PyGcc_get_is_lto(PyObject *self, PyObject *noargs)

{

/*

return PyBool_FromLong(in_lto_p);

}

static PyMethodDef GccMethods[] = {

{"register_attribute",

(PyCFunction)PyGcc_RegisterAttribute,

(METH_VARARGS | METH_KEYWORDS),

"Register an attribute."},

{"register_callback",

(PyCFunction)PyGcc_RegisterCallback,

(METH_VARARGS | METH_KEYWORDS),

"Register a callback, to be called when various GCC events occur."},

{"define_macro",

(PyCFunction)PyGcc_define_macro,

(METH_VARARGS | METH_KEYWORDS),

"Pre-define a named value in the preprocessor."},

/* Diagnostics: */

{"permerror", PyGcc_permerror, METH_VARARGS,

NULL},

{"error",

(PyCFunction)PyGcc_error,

(METH_VARARGS | METH_KEYWORDS),

("Report an error\n"

"FIXME\n")},

{"warning",

(PyCFunction)PyGcc_warning,

(METH_VARARGS | METH_KEYWORDS),

("Report a warning\n"

"FIXME\n")},

{"inform",

(PyCFunction)PyGcc_inform,

(METH_VARARGS | METH_KEYWORDS),

("Report an information message\n"

"FIXME\n")},

{"set_location",

(PyCFunction)PyGcc_set_location,

METH_VARARGS,

("Temporarily set the default location for error reports\n")},

/* Options: */

{"get_option_list",

PyGcc_get_option_list,

METH_NOARGS,

"Get all command-line options, as a list of gcc.Option instances"},

{"get_option_dict",

PyGcc_get_option_dict,

METH_NOARGS,

("Get all command-line options, as a dict from command-line text strings "

"to gcc.Option instances")},

{"get_parameters", PyGcc_get_parameters, METH_NOARGS,

"Get all tunable GCC parameters. Returns a dictionary, mapping from"

"option name -> gcc.Parameter instance"},

{"get_variables", PyGcc_get_variables, METH_NOARGS,

"Get all variables in this compilation unit as a list of gcc.Variable"},

{"maybe_get_identifier", PyGcc_maybe_get_identifier, METH_VARARGS,

"Get the gcc.IdentifierNode with this name, if it exists, otherwise None"},

{"get_translation_units", PyGcc_get_translation_units, METH_NOARGS,

"Get a list of all gcc.TranslationUnitDecl"},

{"get_global_namespace", PyGcc_get_global_namespace, METH_NOARGS,

"C++: get the global namespace (aka '::') as a gcc.NamespaceDecl"},

/* Version handling: */

{"get_plugin_gcc_version", PyGcc_get_plugin_gcc_version, METH_NOARGS,

"Get the gcc.Version that this plugin was compiled with"},

{"get_gcc_version", PyGcc_get_gcc_version, METH_NOARGS,

"Get the gcc.Version for this version of GCC"},

{"get_callgraph_nodes", PyGcc_get_callgraph_nodes, METH_VARARGS,

"Get a list of all gcc.CallgraphNode instances"},

/* Dump files */

{"dump", PyGcc_dump, METH_O,

"Dump str() of the argument to the current dump file (or silently discard it when no dump file is open)"},

{"get_dump_file_name", PyGcc_get_dump_file_name, METH_NOARGS,

"Get the name of the current dump file (or None)"},

{"get_dump_base_name", PyGcc_get_dump_base_name, METH_NOARGS,

"Get the base name used when writing dump files"},

{"is_lto", PyGcc_get_is_lto, METH_NOARGS,

"Determine whether or not we're being invoked during link-time optimization"},

/* Garbage collection */

{"_force_garbage_collection", PyGcc__force_garbage_collection, METH_NOARGS,

"Forcibly trigger a single run of GCC's garbage collector"},

{"_gc_selftest", PyGcc__gc_selftest, METH_NOARGS,

"Run a garbage-collection selftest"},

/* Sentinel: */

{NULL, NULL, 0, NULL}

};

static struct

{

PyObject *module;

PyObject *argument_dict;

PyObject *argument_tuple;

} PyGcc_globals;

#if PY_MAJOR_VERSION == 3

static struct PyModuleDef module_def = {

PyModuleDef_HEAD_INIT,

"gcc", /* name of module */

NULL,

-1,

GccMethods

};

PyMODINIT_FUNC PyInit_gcc(void)

{

#if PY_MAJOR_VERSION == 3

PyObject *m;

m = PyModule_Create(&module_def);

Py_InitModule("gcc", GccMethods);

#if PY_MAJOR_VERSION == 3

return m;

}

static int

PyGcc_init_gcc_module(struct plugin_name_args *plugin_info)

{

int i;

if (!PyGcc_globals.module) {

return 0;

}

/* Set up int constants for each of the enum plugin_event values: */

#define DEFEVENT(NAME) \

PyModule_AddIntMacro(PyGcc_globals.module, NAME);

# include "plugin.def"

# undef DEFEVENT

PyGcc_globals.argument_dict = PyDict_New();

if (!PyGcc_globals.argument_dict) {

return 0;

}

PyGcc_globals.argument_tuple = PyTuple_New(plugin_info->argc);

if (!PyGcc_globals.argument_tuple) {

return 0;

}

/* PySys_SetArgvEx(plugin_info->argc, plugin_info->argv, 0); */

for (i=0; i<plugin_info->argc; i++) {

struct plugin_argument *arg = &plugin_info->argv[i];

PyObject *key;

PyObject *value;

PyObject *pair;

key = PyGccString_FromString(arg->key);

if (arg->value) {

value = PyGccString_FromString(plugin_info->argv[i].value);

} else {

value = Py_None;

}

PyDict_SetItem(PyGcc_globals.argument_dict, key, value);

// FIXME: ref counts?

pair = Py_BuildValue("(s, s)", arg->key, arg->value);

if (!pair) {

return 1;

}

PyTuple_SetItem(PyGcc_globals.argument_tuple, i, pair);

}

PyModule_AddObject(PyGcc_globals.module, "argument_dict", PyGcc_globals.argument_dict);

PyModule_AddObject(PyGcc_globals.module, "argument_tuple", PyGcc_globals.argument_tuple);

/* Pass properties: */

PyModule_AddIntMacro(PyGcc_globals.module, PROP_gimple_any);

PyModule_AddIntMacro(PyGcc_globals.module, PROP_gimple_lcf);

PyModule_AddIntMacro(PyGcc_globals.module, PROP_gimple_leh);

PyModule_AddIntMacro(PyGcc_globals.module, PROP_cfg);

#if (GCC_VERSION >= 4008)

/* PROP_referenced_vars went away in GCC 4.8 (in r190067) */

PyModule_AddIntMacro(PyGcc_globals.module, PROP_referenced_vars);

PyModule_AddIntMacro(PyGcc_globals.module, PROP_ssa);

PyModule_AddIntMacro(PyGcc_globals.module, PROP_no_crit_edges);

PyModule_AddIntMacro(PyGcc_globals.module, PROP_rtl);

PyModule_AddIntMacro(PyGcc_globals.module, PROP_gimple_lomp);

PyModule_AddIntMacro(PyGcc_globals.module, PROP_cfglayout);

PyModule_AddIntMacro(PyGcc_globals.module, PROP_gimple_lcx);

PyModule_AddIntMacro(PyGcc_globals.module, GCC_VERSION);

/* Success: */

return 1;

}

static void PyGcc_run_any_command(void)

{

PyObject* command_obj; /* borrowed ref */

int result;

const char *command_str;

command_obj = PyDict_GetItemString(PyGcc_globals.argument_dict, "command");

if (!command_obj) {

return;

}

command_str = PyGccString_AsString(command_obj);

if (0) {

fprintf(stderr, "Running: %s\n", command_str);

}

result = PyRun_SimpleString(command_str);

if (-1 == result) {

/* Error running the python command */

Py_Finalize();

exit(1);

}

}

static void PyGcc_run_any_script(void)

{

PyObject* script_name;

FILE *fp;

int result;

script_name = PyDict_GetItemString(PyGcc_globals.argument_dict, "script");

if (!script_name) {

return;

}

fp = fopen(PyGccString_AsString(script_name), "r");

if (!fp) {

fprintf(stderr,

"Unable to read python script: %s\n",

PyGccString_AsString(script_name));

exit(1);

}

result = PyRun_SimpleFile(fp, PyGccString_AsString(script_name));

fclose(fp);

if (-1 == result) {

/* Error running the python script */

Py_Finalize();

exit(1);

}

}

setup_sys(struct plugin_name_args *plugin_info)

{

/*

int result = 0; /* failure */

PyObject *full_name = NULL;

PyObject *base_name = NULL;

const char *program =

"import sys;\n"

"import os;\n"

"sys.path.append(os.path.abspath(os.path.dirname(sys.plugin_full_name)))\n";

/* Setup "sys.plugin_full_name" */

full_name = PyGccString_FromString(plugin_info->full_name);

if (!full_name) {

goto error;

}

if (-1 == PySys_SetObject((char*)"plugin_full_name", full_name)) {

goto error;

}

/* Setup "sys.plugin_base_name" */

base_name = PyGccString_FromString(plugin_info->base_name);

if (!base_name) {

goto error;

}

if (-1 == PySys_SetObject((char*)"plugin_base_name", base_name)) {

goto error;

}

/* Add the plugin's path to sys.path */

if (-1 == PyRun_SimpleString(program)) {

goto error;

}

#ifdef PLUGIN_PYTHONPATH

{

/*

const char *program2 =

"import sys;\n"

"import os;\n"

"sys.path.append('" PLUGIN_PYTHONPATH "')\n";

if (-1 == PyRun_SimpleString(program2)) {

goto error;

}

}

/* Success: */

result = 1;

error:

Py_XDECREF(full_name);

Py_XDECREF(base_name);

return result;

}

on_plugin_finish(void *gcc_data, void *user_data)

{

/*

Py_Finalize();

}

extern int

plugin_init (struct plugin_name_args *plugin_info,

struct plugin_gcc_version *version) __attribute__((nonnull));

plugin_init (struct plugin_name_args *plugin_info,

struct plugin_gcc_version *version)

{

LOG("plugin_init started");

if (!plugin_default_version_check (version, &gcc_version)) {

return 1;

}

#if PY_MAJOR_VERSION >= 3

/*

Py_UnbufferedStdioFlag = 1;

PyImport_AppendInittab("gcc", PyInit_gcc);

LOG("calling Py_Initialize...");

Py_Initialize();

LOG("Py_Initialize finished");

PyGcc_globals.module = PyImport_ImportModule("gcc");

PyEval_InitThreads();

if (!PyGcc_init_gcc_module(plugin_info)) {

return 1;

}

if (!setup_sys(plugin_info)) {

return 1;

}

/* Init other modules */

PyGcc_wrapper_init();

/* FIXME: properly integrate them within the module hierarchy */

PyGcc_version_init(version);

autogenerated_callgraph_init_types(); /* FIXME: error checking! */

autogenerated_cfg_init_types(); /* FIXME: error checking! */

autogenerated_function_init_types(); /* FIXME: error checking! */

autogenerated_gimple_init_types(); /* FIXME: error checking! */

autogenerated_location_init_types(); /* FIXME: error checking! */

autogenerated_option_init_types(); /* FIXME: error checking! */

autogenerated_parameter_init_types(); /* FIXME: error checking! */

autogenerated_pass_init_types(); /* FIXME: error checking! */

autogenerated_pretty_printer_init_types(); /* FIXME: error checking! */

autogenerated_rtl_init_types(); /* FIXME: error checking! */

autogenerated_tree_init_types(); /* FIXME: error checking! */

autogenerated_variable_init_types(); /* FIXME: error checking! */

autogenerated_callgraph_add_types(PyGcc_globals.module);

autogenerated_cfg_add_types(PyGcc_globals.module);

autogenerated_function_add_types(PyGcc_globals.module);

autogenerated_gimple_add_types(PyGcc_globals.module);

autogenerated_location_add_types(PyGcc_globals.module);

autogenerated_option_add_types(PyGcc_globals.module);

autogenerated_parameter_add_types(PyGcc_globals.module);

autogenerated_pass_add_types(PyGcc_globals.module);

autogenerated_pretty_printer_add_types(PyGcc_globals.module);

autogenerated_rtl_add_types(PyGcc_globals.module);

autogenerated_tree_add_types(PyGcc_globals.module);

autogenerated_variable_add_types(PyGcc_globals.module);

/* Register at-exit finalization for the plugin: */

register_callback(plugin_info->base_name, PLUGIN_FINISH,

on_plugin_finish, NULL);

PyGcc_run_any_command();

PyGcc_run_any_script();

//printf("%s:%i:got here\n", __FILE__, __LINE__);

#if GCC_PYTHON_TRACE_ALL_EVENTS

#define DEFEVENT(NAME) \

if (NAME != PLUGIN_PASS_MANAGER_SETUP && \

NAME != PLUGIN_INFO && \

NAME != PLUGIN_REGISTER_GGC_ROOTS && \

NAME != PLUGIN_REGISTER_GGC_CACHES) { \

register_callback(plugin_info->base_name, NAME, \

trace_callback_for_##NAME, NULL); \

}

# include "plugin.def"

# undef DEFEVENT

#endif /* GCC_PYTHON_TRACE_ALL_EVENTS */

LOG("init_plugin finished");

return 0;

}

PyObject *

PyGccStringOrNone(const char *str_or_null)

{

if (str_or_null) {

return PyGccString_FromString(str_or_null);

} else {

Py_RETURN_NONE;

}

}

PyObject *

PyGcc_int_from_decimal_string_buffer(const char *buf)

{

PyObject *long_obj;

#if PY_MAJOR_VERSION < 3

long long_val;

int overflow;

long_obj = PyLong_FromString((char *)buf, NULL, 10);

if (!long_obj) {

return NULL;

}

#if PY_MAJOR_VERSION >= 3

return long_obj;

long_val = PyLong_AsLongAndOverflow(long_obj, &overflow);

if (overflow) {

/* Doesn't fit in a PyIntObject; use the PyLongObject: */

return long_obj;

} else {

/* Fits in a PyIntObject: use that */

PyObject *int_obj = PyInt_FromLong(long_val);

if (!int_obj) {

return long_obj;

}

Py_DECREF(long_obj);

return int_obj;

}

}

#if (GCC_VERSION < 5000)

PyGcc_DoubleIntAsText(double_int di, bool is_unsigned,

char *out, int bufsize)

{

FILE *f;

assert(out);

assert(bufsize > 256); /* FIXME */

out[0] = '\0';

f = fmemopen(out, bufsize, "w");

dump_double_int (f, di, is_unsigned);

fclose(f);

}

PyObject *

PyGcc_int_from_double_int(double_int di, bool is_unsigned)

{

char buf[512]; /* FIXME */

PyGcc_DoubleIntAsText(di, is_unsigned, buf, sizeof(buf));

return PyGcc_int_from_decimal_string_buffer(buf);

}

#endif /* #if (GCC_VERSION < 5000) */

char *

PyGcc_strdup(const char *str)

{

char *result;

char *dst;

result = (char*)PyMem_Malloc(strlen(str) + 1);

if (!result) {

return NULL;

}

dst = result;

while (*str) {

*(dst++) = *(str++);

}

*dst = '\0';

return result;

}

void PyGcc_PrintException(const char *msg)

{

/* Handler for Python exceptions */

assert(msg);

/*

gcc_error_at(gcc_get_input_location(), msg);

/* Print the traceback: */

PyErr_PrintEx(1);

}

PyObject *

PyGcc_GetReprOfAttribute(PyObject *obj, const char *attrname)

{

PyObject *attr_obj;

PyObject *attr_repr;

attr_obj = PyObject_GetAttrString(obj, attrname);

if (!attr_obj) {

return NULL;

}

attr_repr = PyObject_Repr(attr_obj);

if (!attr_repr) {

Py_DECREF(attr_obj);

return NULL;

}

return attr_repr;

}