python.c 1.99 KB
Newer Older
1 2
/* Minimal main program -- everything is loaded from the library */

Guido van Rossum's avatar
Guido van Rossum committed
3
#include "Python.h"
4
#include <locale.h>
Guido van Rossum's avatar
Guido van Rossum committed
5

6 7 8 9
#ifdef __FreeBSD__
#include <floatingpoint.h>
#endif

10 11 12 13 14 15 16
#ifdef MS_WINDOWS
int
wmain(int argc, wchar_t **argv)
{
	return Py_Main(argc, argv);
}
#else
Guido van Rossum's avatar
Guido van Rossum committed
17
int
Fredrik Lundh's avatar
Fredrik Lundh committed
18
main(int argc, char **argv)
19
{
20
	wchar_t **argv_copy = (wchar_t **)PyMem_Malloc(sizeof(wchar_t*)*argc);
21
	/* We need a second copies, as Python might modify the first one. */
22
	wchar_t **argv_copy2 = (wchar_t **)PyMem_Malloc(sizeof(wchar_t*)*argc);
23 24
	int i, res;
	char *oldloc;
25 26 27 28 29 30 31 32 33 34 35
	/* 754 requires that FP exceptions run in "no stop" mode by default,
	 * and until C vendors implement C99's ways to control FP exceptions,
	 * Python requires non-stop mode.  Alas, some platforms enable FP
	 * exceptions by default.  Here we disable them.
	 */
#ifdef __FreeBSD__
	fp_except_t m;

	m = fpgetmask();
	fpsetmask(m & ~FP_X_OFL);
#endif
36
	if (!argv_copy || !argv_copy2) {
37
		fprintf(stderr, "out of memory\n");
38 39 40 41 42
		return 1;
	}
	oldloc = setlocale(LC_ALL, NULL);
	setlocale(LC_ALL, "");
	for (i = 0; i < argc; i++) {
43 44 45 46 47 48 49
#ifdef HAVE_BROKEN_MBSTOWCS
		/* Some platforms have a broken implementation of
		 * mbstowcs which does not count the characters that
		 * would result from conversion.  Use an upper bound.
		 */
		size_t argsize = strlen(argv[i]);
#else
50
		size_t argsize = mbstowcs(NULL, argv[i], 0);
51 52
#endif
		size_t count;
53
		if (argsize == (size_t)-1) {
54
			fprintf(stderr, "Could not convert argument %d to string\n", i);
55 56
			return 1;
		}
57
		argv_copy[i] = (wchar_t *)PyMem_Malloc((argsize+1)*sizeof(wchar_t));
58 59
		argv_copy2[i] = argv_copy[i];
		if (!argv_copy[i]) {
60
			fprintf(stderr, "out of memory\n");
61 62
			return 1;
		}
63 64
		count = mbstowcs(argv_copy[i], argv[i], argsize+1);
		if (count == (size_t)-1) {
65
			fprintf(stderr, "Could not convert argument %d to string\n", i);
66 67
			return 1;
		}
68 69 70 71 72 73 74 75 76
	}
	setlocale(LC_ALL, oldloc);
	res = Py_Main(argc, argv_copy);
	for (i = 0; i < argc; i++) {
		PyMem_Free(argv_copy2[i]);
	}
	PyMem_Free(argv_copy);
	PyMem_Free(argv_copy2);
	return res;
77
}
78
#endif