Python code coverage for Modules/_freeze_importlib.c

#	count	content
1	n/a	/* This is built as a stand-alone executable by the Makefile, and helps turn
2	n/a	Lib/importlib/_bootstrap.py into a frozen module in Python/importlib.h
3	n/a	*/
4	n/a
5	n/a	#include <Python.h>
6	n/a	#include <marshal.h>
7	n/a
8	n/a	#include <stdio.h>
9	n/a	#include <sys/types.h>
10	n/a	#include <sys/stat.h>
11	n/a	#ifndef MS_WINDOWS
12	n/a	#include <unistd.h>
13	n/a	#endif
14	n/a
15	n/a
16	n/a	/* To avoid a circular dependency on frozen.o, we create our own structure
17	n/a	of frozen modules instead, left deliberately blank so as to avoid
18	n/a	unintentional import of a stale version of _frozen_importlib. */
19	n/a
20	n/a	const static struct _frozen _PyImport_FrozenModules[] = {
21	n/a	{0, 0, 0} /* sentinel */
22	n/a	};
23	n/a
24	n/a	#ifndef MS_WINDOWS
25	n/a	/* On Windows, this links with the regular pythonXY.dll, so this variable comes
26	n/a	from frozen.obj. In the Makefile, frozen.o is not linked into this executable,
27	n/a	so we define the variable here. */
28	n/a	const struct _frozen *PyImport_FrozenModules;
29	n/a	#endif
30	n/a
31	n/a	const char header[] = "/* Auto-generated by Modules/_freeze_importlib.c */";
32	n/a
33	n/a	int
34	n/a	main(int argc, char *argv[])
35	n/a	{
36	n/a	char inpath, outpath;
37	n/a	FILE infile = NULL, outfile = NULL;
38	n/a	struct stat st;
39	n/a	size_t text_size, data_size, n;
40	n/a	char *text = NULL;
41	n/a	unsigned char *data;
42	n/a	PyObject code = NULL, marshalled = NULL;
43	n/a
44	n/a	PyImport_FrozenModules = _PyImport_FrozenModules;
45	n/a
46	n/a	if (argc != 3) {
47	n/a	fprintf(stderr, "need to specify input and output paths\n");
48	n/a	return 2;
49	n/a	}
50	n/a	inpath = argv[1];
51	n/a	outpath = argv[2];
52	n/a	infile = fopen(inpath, "rb");
53	n/a	if (infile == NULL) {
54	n/a	fprintf(stderr, "cannot open '%s' for reading\n", inpath);
55	n/a	goto error;
56	n/a	}
57	n/a	if (fstat(fileno(infile), &st)) {
58	n/a	fprintf(stderr, "cannot fstat '%s'\n", inpath);
59	n/a	goto error;
60	n/a	}
61	n/a	text_size = st.st_size;
62	n/a	text = (char *) malloc(text_size + 1);
63	n/a	if (text == NULL) {
64	n/a	fprintf(stderr, "could not allocate %ld bytes\n", (long) text_size);
65	n/a	goto error;
66	n/a	}
67	n/a	n = fread(text, 1, text_size, infile);
68	n/a	fclose(infile);
69	n/a	infile = NULL;
70	n/a	if (n < text_size) {
71	n/a	fprintf(stderr, "read too short: got %ld instead of %ld bytes\n",
72	n/a	(long) n, (long) text_size);
73	n/a	goto error;
74	n/a	}
75	n/a	text[text_size] = '\0';
76	n/a
77	n/a	Py_NoUserSiteDirectory++;
78	n/a	Py_NoSiteFlag++;
79	n/a	Py_IgnoreEnvironmentFlag++;
80	n/a
81	n/a	Py_SetProgramName(L"./_freeze_importlib");
82	n/a	/* Don't install importlib, since it could execute outdated bytecode. */
83	n/a	_Py_InitializeEx_Private(1, 0);
84	n/a
85	n/a	code = Py_CompileStringExFlags(text, "<frozen importlib._bootstrap>",
86	n/a	Py_file_input, NULL, 0);
87	n/a	if (code == NULL)
88	n/a	goto error;
89	n/a	free(text);
90	n/a	text = NULL;
91	n/a
92	n/a	marshalled = PyMarshal_WriteObjectToString(code, Py_MARSHAL_VERSION);
93	n/a	Py_CLEAR(code);
94	n/a	if (marshalled == NULL)
95	n/a	goto error;
96	n/a
97	n/a	assert(PyBytes_CheckExact(marshalled));
98	n/a	data = (unsigned char *) PyBytes_AS_STRING(marshalled);
99	n/a	data_size = PyBytes_GET_SIZE(marshalled);
100	n/a
101	n/a	/* Open the file in text mode. The hg checkout should be using the eol extension,
102	n/a	which in turn should cause the EOL style match the C library's text mode */
103	n/a	outfile = fopen(outpath, "w");
104	n/a	if (outfile == NULL) {
105	n/a	fprintf(stderr, "cannot open '%s' for writing\n", outpath);
106	n/a	goto error;
107	n/a	}
108	n/a	fprintf(outfile, "%s\n", header);
109	n/a	fprintf(outfile, "const unsigned char _Py_M__importlib[] = {\n");
110	n/a	for (n = 0; n < data_size; n += 16) {
111	n/a	size_t i, end = Py_MIN(n + 16, data_size);
112	n/a	fprintf(outfile, " ");
113	n/a	for (i = n; i < end; i++) {
114	n/a	fprintf(outfile, "%d,", (unsigned int) data[i]);
115	n/a	}
116	n/a	fprintf(outfile, "\n");
117	n/a	}
118	n/a	fprintf(outfile, "};\n");
119	n/a
120	n/a	Py_CLEAR(marshalled);
121	n/a
122	n/a	Py_Finalize();
123	n/a	if (outfile) {
124	n/a	if (ferror(outfile)) {
125	n/a	fprintf(stderr, "error when writing to '%s'\n", outpath);
126	n/a	goto error;
127	n/a	}
128	n/a	fclose(outfile);
129	n/a	}
130	n/a	return 0;
131	n/a
132	n/a	error:
133	n/a	PyErr_Print();
134	n/a	Py_Finalize();
135	n/a	if (infile)
136	n/a	fclose(infile);
137	n/a	if (outfile)
138	n/a	fclose(outfile);
139	n/a	if (text)
140	n/a	free(text);
141	n/a	if (marshalled)
142	n/a	Py_DECREF(marshalled);
143	n/a	return 1;
144	n/a	}