mirror of
				https://git.proxmox.com/git/mirror_frr
				synced 2025-11-04 01:43:38 +00:00 
			
		
		
		
	The old initialization/config API is deprecated in Python 3.11. Make clippy use the new one added in 3.8 if it's that version or newer. Signed-off-by: David Lamparter <equinox@opensourcerouting.org>
		
			
				
	
	
		
			158 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			158 lines
		
	
	
		
			3.4 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
// SPDX-License-Identifier: GPL-2.0-or-later
 | 
						|
/*
 | 
						|
 * clippy (CLI preparator in python) main executable
 | 
						|
 * Copyright (C) 2016-2017  David Lamparter for NetDEF, Inc.
 | 
						|
 */
 | 
						|
 | 
						|
#include "config.h"
 | 
						|
#include <Python.h>
 | 
						|
#include <string.h>
 | 
						|
#include <stdlib.h>
 | 
						|
#include <wchar.h>
 | 
						|
#include "getopt.h"
 | 
						|
 | 
						|
#include "command_graph.h"
 | 
						|
#include "clippy.h"
 | 
						|
 | 
						|
#if PY_VERSION_HEX >= 0x03080000
 | 
						|
/* new python init/config API added in Python 3.8 */
 | 
						|
int main(int argc, char **argv)
 | 
						|
{
 | 
						|
	PyStatus status;
 | 
						|
	PyPreConfig preconfig[1];
 | 
						|
	PyConfig config[1];
 | 
						|
 | 
						|
	PyPreConfig_InitPythonConfig(preconfig);
 | 
						|
	preconfig->configure_locale = 0;
 | 
						|
	preconfig->coerce_c_locale = 1;
 | 
						|
	preconfig->coerce_c_locale_warn = 0;
 | 
						|
	preconfig->isolated = 0;
 | 
						|
	preconfig->utf8_mode = 1;
 | 
						|
	preconfig->parse_argv = 0;
 | 
						|
 | 
						|
	status = Py_PreInitializeFromBytesArgs(preconfig, argc, argv);
 | 
						|
	if (PyStatus_Exception(status))
 | 
						|
		Py_ExitStatusException(status);
 | 
						|
 | 
						|
	PyConfig_InitPythonConfig(config);
 | 
						|
#if PY_VERSION_HEX >= 0x030b0000	/* 3.11 */
 | 
						|
	config->safe_path = 0;
 | 
						|
#endif
 | 
						|
 | 
						|
	status = PyConfig_SetBytesArgv(config, argc, argv);
 | 
						|
	if (PyStatus_Exception(status))
 | 
						|
		Py_ExitStatusException(status);
 | 
						|
 | 
						|
	PyConfig_SetBytesString(config, &config->program_name,
 | 
						|
				argc > 0 ? argv[0] : "clippy");
 | 
						|
	if (argc > 1)
 | 
						|
		PyConfig_SetBytesString(config, &config->run_filename, argv[1]);
 | 
						|
 | 
						|
	PyImport_AppendInittab("_clippy", command_py_init);
 | 
						|
 | 
						|
	status = Py_InitializeFromConfig(config);
 | 
						|
	if (PyStatus_Exception(status))
 | 
						|
		Py_ExitStatusException(status);
 | 
						|
 | 
						|
	PyConfig_Clear(config);
 | 
						|
 | 
						|
	return Py_RunMain();
 | 
						|
}
 | 
						|
 | 
						|
#else /* Python < 3.8 */
 | 
						|
/* old python init/config API, deprecated in Python 3.11 */
 | 
						|
#if PY_MAJOR_VERSION >= 3
 | 
						|
#define pychar wchar_t
 | 
						|
static wchar_t *wconv(const char *s)
 | 
						|
{
 | 
						|
	size_t outlen = s ? mbstowcs(NULL, s, 0) : 0;
 | 
						|
	wchar_t *out = malloc((outlen + 1) * sizeof(wchar_t));
 | 
						|
 | 
						|
	if (outlen > 0)
 | 
						|
		mbstowcs(out, s, outlen);
 | 
						|
	out[outlen] = 0;
 | 
						|
	return out;
 | 
						|
}
 | 
						|
#else
 | 
						|
#define pychar char
 | 
						|
#define wconv(x) x
 | 
						|
#endif
 | 
						|
 | 
						|
int main(int argc, char **argv)
 | 
						|
{
 | 
						|
	pychar **wargv;
 | 
						|
 | 
						|
#if PY_VERSION_HEX >= 0x03040000 /* 3.4 */
 | 
						|
	Py_SetStandardStreamEncoding("UTF-8", NULL);
 | 
						|
#endif
 | 
						|
	wchar_t *name = wconv(argv[0]);
 | 
						|
	Py_SetProgramName(name);
 | 
						|
	PyImport_AppendInittab("_clippy", command_py_init);
 | 
						|
 | 
						|
	Py_Initialize();
 | 
						|
 | 
						|
	wargv = malloc(argc * sizeof(pychar *));
 | 
						|
	for (int i = 1; i < argc; i++)
 | 
						|
		wargv[i - 1] = wconv(argv[i]);
 | 
						|
	PySys_SetArgv(argc - 1, wargv);
 | 
						|
 | 
						|
	const char *pyfile = argc > 1 ? argv[1] : NULL;
 | 
						|
	FILE *fp;
 | 
						|
	if (pyfile) {
 | 
						|
		fp = fopen(pyfile, "r");
 | 
						|
		if (!fp) {
 | 
						|
			fprintf(stderr, "%s: %s\n", pyfile, strerror(errno));
 | 
						|
 | 
						|
			free(name);
 | 
						|
			return 1;
 | 
						|
		}
 | 
						|
	} else {
 | 
						|
		fp = stdin;
 | 
						|
		char *ver = strdup(Py_GetVersion());
 | 
						|
		char *cr = strchr(ver, '\n');
 | 
						|
		if (cr)
 | 
						|
			*cr = ' ';
 | 
						|
		fprintf(stderr, "clippy interactive shell\n(Python %s)\n", ver);
 | 
						|
		free(ver);
 | 
						|
		PyRun_SimpleString(
 | 
						|
			"import rlcompleter, readline\n"
 | 
						|
			"readline.parse_and_bind('tab: complete')");
 | 
						|
	}
 | 
						|
 | 
						|
	if (PyRun_AnyFile(fp, pyfile)) {
 | 
						|
		if (PyErr_Occurred())
 | 
						|
			PyErr_Print();
 | 
						|
 | 
						|
		free(name);
 | 
						|
		return 1;
 | 
						|
	}
 | 
						|
	Py_Finalize();
 | 
						|
 | 
						|
#if PY_MAJOR_VERSION >= 3
 | 
						|
	for (int i = 1; i < argc; i++)
 | 
						|
		free(wargv[i - 1]);
 | 
						|
#endif
 | 
						|
	free(name);
 | 
						|
	free(wargv);
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
#endif /* Python < 3.8 */
 | 
						|
 | 
						|
/* and now for the ugly part... provide simplified logging functions so we
 | 
						|
 * don't need to link libzebra (which would be a circular build dep) */
 | 
						|
 | 
						|
#include "log.h"
 | 
						|
 | 
						|
PRINTFRR(3, 0)
 | 
						|
void vzlogx(const struct xref_logmsg *xref, int prio,
 | 
						|
	    const char *format, va_list args)
 | 
						|
{
 | 
						|
	vfprintf(stderr, format, args);
 | 
						|
	fputs("\n", stderr);
 | 
						|
}
 | 
						|
 | 
						|
void memory_oom(size_t size, const char *name)
 | 
						|
{
 | 
						|
	abort();
 | 
						|
}
 |