summaryrefslogtreecommitdiffstats
path: root/contrib/tools/python3/src/Programs
diff options
context:
space:
mode:
authorDevtools Arcadia <[email protected]>2022-02-07 18:08:42 +0300
committerDevtools Arcadia <[email protected]>2022-02-07 18:08:42 +0300
commit1110808a9d39d4b808aef724c861a2e1a38d2a69 (patch)
treee26c9fed0de5d9873cce7e00bc214573dc2195b7 /contrib/tools/python3/src/Programs
intermediate changes
ref:cde9a383711a11544ce7e107a78147fb96cc4029
Diffstat (limited to 'contrib/tools/python3/src/Programs')
-rw-r--r--contrib/tools/python3/src/Programs/_freeze_importlib.c167
-rw-r--r--contrib/tools/python3/src/Programs/python.c17
2 files changed, 184 insertions, 0 deletions
diff --git a/contrib/tools/python3/src/Programs/_freeze_importlib.c b/contrib/tools/python3/src/Programs/_freeze_importlib.c
new file mode 100644
index 00000000000..2e4ccbb154a
--- /dev/null
+++ b/contrib/tools/python3/src/Programs/_freeze_importlib.c
@@ -0,0 +1,167 @@
+/* This is built as a stand-alone executable by the Makefile, and helps turn
+ Lib/importlib/_bootstrap.py into a frozen module in Python/importlib.h
+*/
+
+#include <Python.h>
+#include <marshal.h>
+
+#include <stdio.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#ifndef MS_WINDOWS
+#include <unistd.h>
+#endif
+
+/* To avoid a circular dependency on frozen.o, we create our own structure
+ of frozen modules instead, left deliberately blank so as to avoid
+ unintentional import of a stale version of _frozen_importlib. */
+
+static const struct _frozen _PyImport_FrozenModules[] = {
+ {0, 0, 0} /* sentinel */
+};
+
+#ifndef MS_WINDOWS
+/* On Windows, this links with the regular pythonXY.dll, so this variable comes
+ from frozen.obj. In the Makefile, frozen.o is not linked into this executable,
+ so we define the variable here. */
+const struct _frozen *PyImport_FrozenModules;
+#endif
+
+static const char header[] =
+ "/* Auto-generated by Programs/_freeze_importlib.c */";
+
+int
+main(int argc, char *argv[])
+{
+ const char *name, *inpath, *outpath;
+ char buf[100];
+ FILE *infile = NULL, *outfile = NULL;
+ struct _Py_stat_struct stat;
+ size_t text_size, data_size, i, n;
+ char *text = NULL;
+ unsigned char *data;
+ PyObject *code = NULL, *marshalled = NULL;
+
+ PyImport_FrozenModules = _PyImport_FrozenModules;
+
+ if (argc != 4) {
+ fprintf(stderr, "need to specify the name, input and output paths\n");
+ return 2;
+ }
+ name = argv[1];
+ inpath = argv[2];
+ outpath = argv[3];
+ infile = fopen(inpath, "rb");
+ if (infile == NULL) {
+ fprintf(stderr, "cannot open '%s' for reading\n", inpath);
+ goto error;
+ }
+ if (_Py_fstat_noraise(fileno(infile), &stat)) {
+ fprintf(stderr, "cannot fstat '%s'\n", inpath);
+ goto error;
+ }
+ text_size = (size_t)stat.st_size;
+ text = (char *) malloc(text_size + 1);
+ if (text == NULL) {
+ fprintf(stderr, "could not allocate %ld bytes\n", (long) text_size);
+ goto error;
+ }
+ n = fread(text, 1, text_size, infile);
+ fclose(infile);
+ infile = NULL;
+ if (n < text_size) {
+ fprintf(stderr, "read too short: got %ld instead of %ld bytes\n",
+ (long) n, (long) text_size);
+ goto error;
+ }
+ text[text_size] = '\0';
+
+ PyConfig config;
+ PyConfig_InitIsolatedConfig(&config);
+
+ config.site_import = 0;
+
+ PyStatus status;
+ status = PyConfig_SetString(&config, &config.program_name,
+ L"./_freeze_importlib");
+ if (PyStatus_Exception(status)) {
+ PyConfig_Clear(&config);
+ Py_ExitStatusException(status);
+ }
+
+ /* Don't install importlib, since it could execute outdated bytecode. */
+ config._install_importlib = 0;
+ config._init_main = 0;
+
+ status = Py_InitializeFromConfig(&config);
+ PyConfig_Clear(&config);
+ if (PyStatus_Exception(status)) {
+ Py_ExitStatusException(status);
+ }
+
+ sprintf(buf, "<frozen %s>", name);
+ code = Py_CompileStringExFlags(text, buf, Py_file_input, NULL, 0);
+ if (code == NULL)
+ goto error;
+ free(text);
+ text = NULL;
+
+ marshalled = PyMarshal_WriteObjectToString(code, Py_MARSHAL_VERSION);
+ Py_CLEAR(code);
+ if (marshalled == NULL)
+ goto error;
+
+ assert(PyBytes_CheckExact(marshalled));
+ data = (unsigned char *) PyBytes_AS_STRING(marshalled);
+ data_size = PyBytes_GET_SIZE(marshalled);
+
+ /* Open the file in text mode. The hg checkout should be using the eol extension,
+ which in turn should cause the EOL style match the C library's text mode */
+ outfile = fopen(outpath, "w");
+ if (outfile == NULL) {
+ fprintf(stderr, "cannot open '%s' for writing\n", outpath);
+ goto error;
+ }
+ fprintf(outfile, "%s\n", header);
+ for (i = n = 0; name[i] != '\0'; i++) {
+ if (name[i] != '.') {
+ buf[n++] = name[i];
+ }
+ }
+ buf[n] = '\0';
+ fprintf(outfile, "const unsigned char _Py_M__%s[] = {\n", buf);
+ for (n = 0; n < data_size; n += 16) {
+ size_t i, end = Py_MIN(n + 16, data_size);
+ fprintf(outfile, " ");
+ for (i = n; i < end; i++) {
+ fprintf(outfile, "%u,", (unsigned int) data[i]);
+ }
+ fprintf(outfile, "\n");
+ }
+ fprintf(outfile, "};\n");
+
+ Py_CLEAR(marshalled);
+
+ Py_Finalize();
+ if (outfile) {
+ if (ferror(outfile)) {
+ fprintf(stderr, "error when writing to '%s'\n", outpath);
+ goto error;
+ }
+ fclose(outfile);
+ }
+ return 0;
+
+error:
+ PyErr_Print();
+ Py_Finalize();
+ if (infile)
+ fclose(infile);
+ if (outfile)
+ fclose(outfile);
+ if (text)
+ free(text);
+ if (marshalled)
+ Py_DECREF(marshalled);
+ return 1;
+}
diff --git a/contrib/tools/python3/src/Programs/python.c b/contrib/tools/python3/src/Programs/python.c
new file mode 100644
index 00000000000..84148f7767a
--- /dev/null
+++ b/contrib/tools/python3/src/Programs/python.c
@@ -0,0 +1,17 @@
+/* Minimal main program -- everything is loaded from the library */
+
+#include "Python.h"
+
+#ifdef MS_WINDOWS
+int
+wmain(int argc, wchar_t **argv)
+{
+ return Py_Main(argc, argv);
+}
+#else
+int
+main(int argc, char **argv)
+{
+ return Py_BytesMain(argc, argv);
+}
+#endif