xref: /vim-8.2.3635/src/if_python3.c (revision fda3729a) 
1 /* vi:set ts=8 sts=4 sw=4:
2  *
3  * VIM - Vi IMproved    by Bram Moolenaar
4  *
5  * Do ":help uganda"  in Vim to read copying and usage conditions.
6  * Do ":help credits" in Vim to see a list of people who contributed.
7  * See README.txt for an overview of the Vim source code.
8  */
9 /*
10  * Python extensions by Paul Moore.
11  * Changes for Unix by David Leonard.
12  *
13  * This consists of four parts:
14  * 1. Python interpreter main program
15  * 2. Python output stream: writes output via [e]msg().
16  * 3. Implementation of the Vim module for Python
17  * 4. Utility functions for handling the interface between Vim and Python.
18  */
19 
20 /*
21  * Roland Puntaier 2009/sept/16:
22  * Adaptations to support both python3.x and python2.x
23  */
24 
25 /* uncomment this if used with the debug version of python */
26 /* #define Py_DEBUG */
27 /* Note: most of time you can add -DPy_DEBUG to CFLAGS in place of uncommenting
28  */
29 /* uncomment this if used with the debug version of python, but without its
30  * allocator */
31 /* #define Py_DEBUG_NO_PYMALLOC */
32 
33 #include "vim.h"
34 
35 #include <limits.h>
36 
37 /* Python.h defines _POSIX_THREADS itself (if needed) */
38 #ifdef _POSIX_THREADS
39 # undef _POSIX_THREADS
40 #endif
41 
42 #if defined(_WIN32) && defined(HAVE_FCNTL_H)
43 # undef HAVE_FCNTL_H
44 #endif
45 
46 #ifdef _DEBUG
47 # undef _DEBUG
48 #endif
49 
50 #ifdef F_BLANK
51 # undef F_BLANK
52 #endif
53 
54 #ifdef HAVE_STDARG_H
55 # undef HAVE_STDARG_H   /* Python's config.h defines it as well. */
56 #endif
57 #ifdef _POSIX_C_SOURCE  /* defined in feature.h */
58 # undef _POSIX_C_SOURCE
59 #endif
60 #ifdef _XOPEN_SOURCE
61 # undef _XOPEN_SOURCE	/* pyconfig.h defines it as well. */
62 #endif
63 
64 #define PY_SSIZE_T_CLEAN
65 
66 #include <Python.h>
67 
68 #if defined(MACOS) && !defined(MACOS_X_UNIX)
69 # include "macglue.h"
70 # include <CodeFragments.h>
71 #endif
72 #undef main /* Defined in python.h - aargh */
73 #undef HAVE_FCNTL_H /* Clash with os_win32.h */
74 
75 /* The "surrogateescape" error handler is new in Python 3.1 */
76 #if PY_VERSION_HEX >= 0x030100f0
77 # define CODEC_ERROR_HANDLER "surrogateescape"
78 #else
79 # define CODEC_ERROR_HANDLER NULL
80 #endif
81 
82 /* Python 3 does not support CObjects, always use Capsules */
83 #define PY_USE_CAPSULE
84 
85 #define PyInt Py_ssize_t
86 #ifndef PyString_Check
87 # define PyString_Check(obj) PyUnicode_Check(obj)
88 #endif
89 #define PyString_FromString(repr) \
90     PyUnicode_Decode(repr, STRLEN(repr), ENC_OPT, NULL)
91 #define PyString_FromFormat PyUnicode_FromFormat
92 #ifndef PyInt_Check
93 # define PyInt_Check(obj) PyLong_Check(obj)
94 #endif
95 #define PyInt_FromLong(i) PyLong_FromLong(i)
96 #define PyInt_AsLong(obj) PyLong_AsLong(obj)
97 #define Py_ssize_t_fmt "n"
98 #define Py_bytes_fmt "y"
99 
100 #define PyIntArgFunc	ssizeargfunc
101 #define PyIntObjArgProc	ssizeobjargproc
102 
103 /*
104  * PySlice_GetIndicesEx(): first argument type changed from PySliceObject
105  * to PyObject in Python 3.2 or later.
106  */
107 #if PY_VERSION_HEX >= 0x030200f0
108 typedef PyObject PySliceObject_T;
109 #else
110 typedef PySliceObject PySliceObject_T;
111 #endif
112 
113 #if defined(DYNAMIC_PYTHON3) || defined(PROTO)
114 
115 # ifndef WIN3264
116 #  include <dlfcn.h>
117 #  define FARPROC void*
118 #  define HINSTANCE void*
119 #  if defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)
120 #   define load_dll(n) dlopen((n), RTLD_LAZY)
121 #  else
122 #   define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL)
123 #  endif
124 #  define close_dll dlclose
125 #  define symbol_from_dll dlsym
126 # else
127 #  define load_dll vimLoadLib
128 #  define close_dll FreeLibrary
129 #  define symbol_from_dll GetProcAddress
130 # endif
131 /*
132  * Wrapper defines
133  */
134 # undef PyArg_Parse
135 # define PyArg_Parse py3_PyArg_Parse
136 # undef PyArg_ParseTuple
137 # define PyArg_ParseTuple py3_PyArg_ParseTuple
138 # define PyMem_Free py3_PyMem_Free
139 # define PyMem_Malloc py3_PyMem_Malloc
140 # define PyDict_SetItemString py3_PyDict_SetItemString
141 # define PyErr_BadArgument py3_PyErr_BadArgument
142 # define PyErr_Clear py3_PyErr_Clear
143 # define PyErr_Format py3_PyErr_Format
144 # define PyErr_PrintEx py3_PyErr_PrintEx
145 # define PyErr_NoMemory py3_PyErr_NoMemory
146 # define PyErr_Occurred py3_PyErr_Occurred
147 # define PyErr_SetNone py3_PyErr_SetNone
148 # define PyErr_SetString py3_PyErr_SetString
149 # define PyErr_SetObject py3_PyErr_SetObject
150 # define PyErr_ExceptionMatches py3_PyErr_ExceptionMatches
151 # define PyEval_InitThreads py3_PyEval_InitThreads
152 # define PyEval_RestoreThread py3_PyEval_RestoreThread
153 # define PyEval_SaveThread py3_PyEval_SaveThread
154 # define PyGILState_Ensure py3_PyGILState_Ensure
155 # define PyGILState_Release py3_PyGILState_Release
156 # define PyLong_AsLong py3_PyLong_AsLong
157 # define PyLong_FromLong py3_PyLong_FromLong
158 # define PyList_GetItem py3_PyList_GetItem
159 # define PyList_Append py3_PyList_Append
160 # define PyList_Insert py3_PyList_Insert
161 # define PyList_New py3_PyList_New
162 # define PyList_SetItem py3_PyList_SetItem
163 # define PyList_Size py3_PyList_Size
164 # define PySequence_Check py3_PySequence_Check
165 # define PySequence_Size py3_PySequence_Size
166 # define PySequence_GetItem py3_PySequence_GetItem
167 # define PySequence_Fast py3_PySequence_Fast
168 # define PyTuple_Size py3_PyTuple_Size
169 # define PyTuple_GetItem py3_PyTuple_GetItem
170 # define PySlice_GetIndicesEx py3_PySlice_GetIndicesEx
171 # define PyImport_ImportModule py3_PyImport_ImportModule
172 # define PyObject_Init py3__PyObject_Init
173 # define PyDict_New py3_PyDict_New
174 # define PyDict_GetItemString py3_PyDict_GetItemString
175 # define PyDict_Next py3_PyDict_Next
176 # define PyMapping_Check py3_PyMapping_Check
177 # ifndef PyMapping_Keys
178 #  define PyMapping_Keys py3_PyMapping_Keys
179 # endif
180 # define PyIter_Next py3_PyIter_Next
181 # define PyObject_GetIter py3_PyObject_GetIter
182 # define PyObject_Repr py3_PyObject_Repr
183 # define PyObject_GetItem py3_PyObject_GetItem
184 # define PyObject_IsTrue py3_PyObject_IsTrue
185 # define PyModule_GetDict py3_PyModule_GetDict
186 #undef PyRun_SimpleString
187 # define PyRun_SimpleString py3_PyRun_SimpleString
188 #undef PyRun_String
189 # define PyRun_String py3_PyRun_String
190 # define PyObject_GetAttrString py3_PyObject_GetAttrString
191 # define PyObject_HasAttrString py3_PyObject_HasAttrString
192 # define PyObject_SetAttrString py3_PyObject_SetAttrString
193 # define PyObject_CallFunctionObjArgs py3_PyObject_CallFunctionObjArgs
194 # define _PyObject_CallFunction_SizeT py3__PyObject_CallFunction_SizeT
195 # define PyObject_Call py3_PyObject_Call
196 # define PyEval_GetLocals py3_PyEval_GetLocals
197 # define PyEval_GetGlobals py3_PyEval_GetGlobals
198 # define PySys_SetObject py3_PySys_SetObject
199 # define PySys_GetObject py3_PySys_GetObject
200 # define PySys_SetArgv py3_PySys_SetArgv
201 # define PyType_Ready py3_PyType_Ready
202 #undef Py_BuildValue
203 # define Py_BuildValue py3_Py_BuildValue
204 # define Py_SetPythonHome py3_Py_SetPythonHome
205 # define Py_Initialize py3_Py_Initialize
206 # define Py_Finalize py3_Py_Finalize
207 # define Py_IsInitialized py3_Py_IsInitialized
208 # define _Py_NoneStruct (*py3__Py_NoneStruct)
209 # define _Py_FalseStruct (*py3__Py_FalseStruct)
210 # define _Py_TrueStruct (*py3__Py_TrueStruct)
211 # define _PyObject_NextNotImplemented (*py3__PyObject_NextNotImplemented)
212 # define PyModule_AddObject py3_PyModule_AddObject
213 # define PyImport_AppendInittab py3_PyImport_AppendInittab
214 # define PyImport_AddModule py3_PyImport_AddModule
215 # if PY_VERSION_HEX >= 0x030300f0
216 #  undef _PyUnicode_AsString
217 #  define _PyUnicode_AsString py3_PyUnicode_AsUTF8
218 # else
219 #  define _PyUnicode_AsString py3__PyUnicode_AsString
220 # endif
221 # undef PyUnicode_AsEncodedString
222 # define PyUnicode_AsEncodedString py3_PyUnicode_AsEncodedString
223 # undef PyBytes_AsString
224 # define PyBytes_AsString py3_PyBytes_AsString
225 # ifndef PyBytes_AsStringAndSize
226 #  define PyBytes_AsStringAndSize py3_PyBytes_AsStringAndSize
227 # endif
228 # undef PyBytes_FromString
229 # define PyBytes_FromString py3_PyBytes_FromString
230 # define PyFloat_FromDouble py3_PyFloat_FromDouble
231 # define PyFloat_AsDouble py3_PyFloat_AsDouble
232 # define PyObject_GenericGetAttr py3_PyObject_GenericGetAttr
233 # define PyType_Type (*py3_PyType_Type)
234 # define PySlice_Type (*py3_PySlice_Type)
235 # define PyFloat_Type (*py3_PyFloat_Type)
236 # define PyNumber_Check (*py3_PyNumber_Check)
237 # define PyNumber_Long (*py3_PyNumber_Long)
238 # define PyBool_Type (*py3_PyBool_Type)
239 # define PyErr_NewException py3_PyErr_NewException
240 # ifdef Py_DEBUG
241 #  define _Py_NegativeRefcount py3__Py_NegativeRefcount
242 #  define _Py_RefTotal (*py3__Py_RefTotal)
243 #  define _Py_Dealloc py3__Py_Dealloc
244 #  define PyModule_Create2TraceRefs py3_PyModule_Create2TraceRefs
245 # else
246 #  define PyModule_Create2 py3_PyModule_Create2
247 # endif
248 # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC)
249 #  define _PyObject_DebugMalloc py3__PyObject_DebugMalloc
250 #  define _PyObject_DebugFree py3__PyObject_DebugFree
251 # else
252 #  define PyObject_Malloc py3_PyObject_Malloc
253 #  define PyObject_Free py3_PyObject_Free
254 # endif
255 # define _PyObject_GC_New py3__PyObject_GC_New
256 # define PyObject_GC_Del py3_PyObject_GC_Del
257 # define PyObject_GC_UnTrack py3_PyObject_GC_UnTrack
258 # define PyType_GenericAlloc py3_PyType_GenericAlloc
259 # define PyType_GenericNew py3_PyType_GenericNew
260 # undef PyUnicode_FromString
261 # define PyUnicode_FromString py3_PyUnicode_FromString
262 # ifndef PyUnicode_FromFormat
263 #  define PyUnicode_FromFormat py3_PyUnicode_FromFormat
264 # else
265 #  define Py_UNICODE_USE_UCS_FUNCTIONS
266 #  ifdef Py_UNICODE_WIDE
267 #   define PyUnicodeUCS4_FromFormat py3_PyUnicodeUCS4_FromFormat
268 #  else
269 #   define PyUnicodeUCS2_FromFormat py3_PyUnicodeUCS2_FromFormat
270 #  endif
271 # endif
272 # undef PyUnicode_Decode
273 # define PyUnicode_Decode py3_PyUnicode_Decode
274 # define PyType_IsSubtype py3_PyType_IsSubtype
275 # define PyCapsule_New py3_PyCapsule_New
276 # define PyCapsule_GetPointer py3_PyCapsule_GetPointer
277 
278 # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC)
279 #  undef PyObject_NEW
280 #  define PyObject_NEW(type, typeobj) \
281 ( (type *) PyObject_Init( \
282 	(PyObject *) _PyObject_DebugMalloc( _PyObject_SIZE(typeobj) ), (typeobj)) )
283 # endif
284 
285 /*
286  * Pointers for dynamic link
287  */
288 static int (*py3_PySys_SetArgv)(int, wchar_t **);
289 static void (*py3_Py_SetPythonHome)(wchar_t *home);
290 static void (*py3_Py_Initialize)(void);
291 static PyObject* (*py3_PyList_New)(Py_ssize_t size);
292 static PyGILState_STATE (*py3_PyGILState_Ensure)(void);
293 static void (*py3_PyGILState_Release)(PyGILState_STATE);
294 static int (*py3_PySys_SetObject)(char *, PyObject *);
295 static PyObject* (*py3_PySys_GetObject)(char *);
296 static int (*py3_PyList_Append)(PyObject *, PyObject *);
297 static int (*py3_PyList_Insert)(PyObject *, int, PyObject *);
298 static Py_ssize_t (*py3_PyList_Size)(PyObject *);
299 static int (*py3_PySequence_Check)(PyObject *);
300 static Py_ssize_t (*py3_PySequence_Size)(PyObject *);
301 static PyObject* (*py3_PySequence_GetItem)(PyObject *, Py_ssize_t);
302 static PyObject* (*py3_PySequence_Fast)(PyObject *, const char *);
303 static Py_ssize_t (*py3_PyTuple_Size)(PyObject *);
304 static PyObject* (*py3_PyTuple_GetItem)(PyObject *, Py_ssize_t);
305 static int (*py3_PyMapping_Check)(PyObject *);
306 static PyObject* (*py3_PyMapping_Keys)(PyObject *);
307 static int (*py3_PySlice_GetIndicesEx)(PySliceObject_T *r, Py_ssize_t length,
308 		     Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step,
309 		     Py_ssize_t *slicelen);
310 static PyObject* (*py3_PyErr_NoMemory)(void);
311 static void (*py3_Py_Finalize)(void);
312 static void (*py3_PyErr_SetString)(PyObject *, const char *);
313 static void (*py3_PyErr_SetObject)(PyObject *, PyObject *);
314 static int (*py3_PyErr_ExceptionMatches)(PyObject *);
315 static int (*py3_PyRun_SimpleString)(char *);
316 static PyObject* (*py3_PyRun_String)(char *, int, PyObject *, PyObject *);
317 static PyObject* (*py3_PyObject_GetAttrString)(PyObject *, const char *);
318 static int (*py3_PyObject_HasAttrString)(PyObject *, const char *);
319 static int (*py3_PyObject_SetAttrString)(PyObject *, const char *, PyObject *);
320 static PyObject* (*py3_PyObject_CallFunctionObjArgs)(PyObject *, ...);
321 static PyObject* (*py3__PyObject_CallFunction_SizeT)(PyObject *, char *, ...);
322 static PyObject* (*py3_PyObject_Call)(PyObject *, PyObject *, PyObject *);
323 static PyObject* (*py3_PyEval_GetGlobals)();
324 static PyObject* (*py3_PyEval_GetLocals)();
325 static PyObject* (*py3_PyList_GetItem)(PyObject *, Py_ssize_t);
326 static PyObject* (*py3_PyImport_ImportModule)(const char *);
327 static PyObject* (*py3_PyImport_AddModule)(const char *);
328 static int (*py3_PyErr_BadArgument)(void);
329 static PyObject* (*py3_PyErr_Occurred)(void);
330 static PyObject* (*py3_PyModule_GetDict)(PyObject *);
331 static int (*py3_PyList_SetItem)(PyObject *, Py_ssize_t, PyObject *);
332 static PyObject* (*py3_PyDict_GetItemString)(PyObject *, const char *);
333 static int (*py3_PyDict_Next)(PyObject *, Py_ssize_t *, PyObject **, PyObject **);
334 static PyObject* (*py3_PyLong_FromLong)(long);
335 static PyObject* (*py3_PyDict_New)(void);
336 static PyObject* (*py3_PyIter_Next)(PyObject *);
337 static PyObject* (*py3_PyObject_GetIter)(PyObject *);
338 static PyObject* (*py3_PyObject_Repr)(PyObject *);
339 static PyObject* (*py3_PyObject_GetItem)(PyObject *, PyObject *);
340 static int (*py3_PyObject_IsTrue)(PyObject *);
341 static PyObject* (*py3_Py_BuildValue)(char *, ...);
342 static int (*py3_PyType_Ready)(PyTypeObject *type);
343 static int (*py3_PyDict_SetItemString)(PyObject *dp, char *key, PyObject *item);
344 static PyObject* (*py3_PyUnicode_FromString)(const char *u);
345 # ifndef Py_UNICODE_USE_UCS_FUNCTIONS
346 static PyObject* (*py3_PyUnicode_FromFormat)(const char *u, ...);
347 # else
348 #  ifdef Py_UNICODE_WIDE
349 static PyObject* (*py3_PyUnicodeUCS4_FromFormat)(const char *u, ...);
350 #  else
351 static PyObject* (*py3_PyUnicodeUCS2_FromFormat)(const char *u, ...);
352 #  endif
353 # endif
354 static PyObject* (*py3_PyUnicode_Decode)(const char *u, Py_ssize_t size,
355 	const char *encoding, const char *errors);
356 static long (*py3_PyLong_AsLong)(PyObject *);
357 static void (*py3_PyErr_SetNone)(PyObject *);
358 static void (*py3_PyEval_InitThreads)(void);
359 static void(*py3_PyEval_RestoreThread)(PyThreadState *);
360 static PyThreadState*(*py3_PyEval_SaveThread)(void);
361 static int (*py3_PyArg_Parse)(PyObject *, char *, ...);
362 static int (*py3_PyArg_ParseTuple)(PyObject *, char *, ...);
363 static int (*py3_PyMem_Free)(void *);
364 static void* (*py3_PyMem_Malloc)(size_t);
365 static int (*py3_Py_IsInitialized)(void);
366 static void (*py3_PyErr_Clear)(void);
367 static PyObject* (*py3_PyErr_Format)(PyObject *, const char *, ...);
368 static void (*py3_PyErr_PrintEx)(int);
369 static PyObject*(*py3__PyObject_Init)(PyObject *, PyTypeObject *);
370 static iternextfunc py3__PyObject_NextNotImplemented;
371 static PyObject* py3__Py_NoneStruct;
372 static PyObject* py3__Py_FalseStruct;
373 static PyObject* py3__Py_TrueStruct;
374 static int (*py3_PyModule_AddObject)(PyObject *m, const char *name, PyObject *o);
375 static int (*py3_PyImport_AppendInittab)(const char *name, PyObject* (*initfunc)(void));
376 # if PY_VERSION_HEX >= 0x030300f0
377 static char* (*py3_PyUnicode_AsUTF8)(PyObject *unicode);
378 # else
379 static char* (*py3__PyUnicode_AsString)(PyObject *unicode);
380 # endif
381 static PyObject* (*py3_PyUnicode_AsEncodedString)(PyObject *unicode, const char* encoding, const char* errors);
382 static char* (*py3_PyBytes_AsString)(PyObject *bytes);
383 static int (*py3_PyBytes_AsStringAndSize)(PyObject *bytes, char **buffer, Py_ssize_t *length);
384 static PyObject* (*py3_PyBytes_FromString)(char *str);
385 static PyObject* (*py3_PyFloat_FromDouble)(double num);
386 static double (*py3_PyFloat_AsDouble)(PyObject *);
387 static PyObject* (*py3_PyObject_GenericGetAttr)(PyObject *obj, PyObject *name);
388 static PyObject* (*py3_PyType_GenericAlloc)(PyTypeObject *type, Py_ssize_t nitems);
389 static PyObject* (*py3_PyType_GenericNew)(PyTypeObject *type, PyObject *args, PyObject *kwds);
390 static PyTypeObject* py3_PyType_Type;
391 static PyTypeObject* py3_PySlice_Type;
392 static PyTypeObject* py3_PyFloat_Type;
393 static PyTypeObject* py3_PyBool_Type;
394 static int (*py3_PyNumber_Check)(PyObject *);
395 static PyObject* (*py3_PyNumber_Long)(PyObject *);
396 static PyObject* (*py3_PyErr_NewException)(char *name, PyObject *base, PyObject *dict);
397 static PyObject* (*py3_PyCapsule_New)(void *, char *, PyCapsule_Destructor);
398 static void* (*py3_PyCapsule_GetPointer)(PyObject *, char *);
399 # ifdef Py_DEBUG
400 static void (*py3__Py_NegativeRefcount)(const char *fname, int lineno, PyObject *op);
401 static Py_ssize_t* py3__Py_RefTotal;
402 static void (*py3__Py_Dealloc)(PyObject *obj);
403 static PyObject* (*py3_PyModule_Create2TraceRefs)(struct PyModuleDef* module, int module_api_version);
404 # else
405 static PyObject* (*py3_PyModule_Create2)(struct PyModuleDef* module, int module_api_version);
406 # endif
407 # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC)
408 static void (*py3__PyObject_DebugFree)(void*);
409 static void* (*py3__PyObject_DebugMalloc)(size_t);
410 # else
411 static void (*py3_PyObject_Free)(void*);
412 static void* (*py3_PyObject_Malloc)(size_t);
413 # endif
414 static PyObject*(*py3__PyObject_GC_New)(PyTypeObject *);
415 static void(*py3_PyObject_GC_Del)(void *);
416 static void(*py3_PyObject_GC_UnTrack)(void *);
417 static int (*py3_PyType_IsSubtype)(PyTypeObject *, PyTypeObject *);
418 
419 static HINSTANCE hinstPy3 = 0; /* Instance of python.dll */
420 
421 /* Imported exception objects */
422 static PyObject *p3imp_PyExc_AttributeError;
423 static PyObject *p3imp_PyExc_IndexError;
424 static PyObject *p3imp_PyExc_KeyError;
425 static PyObject *p3imp_PyExc_KeyboardInterrupt;
426 static PyObject *p3imp_PyExc_TypeError;
427 static PyObject *p3imp_PyExc_ValueError;
428 static PyObject *p3imp_PyExc_SystemExit;
429 static PyObject *p3imp_PyExc_RuntimeError;
430 static PyObject *p3imp_PyExc_ImportError;
431 static PyObject *p3imp_PyExc_OverflowError;
432 
433 # define PyExc_AttributeError p3imp_PyExc_AttributeError
434 # define PyExc_IndexError p3imp_PyExc_IndexError
435 # define PyExc_KeyError p3imp_PyExc_KeyError
436 # define PyExc_KeyboardInterrupt p3imp_PyExc_KeyboardInterrupt
437 # define PyExc_TypeError p3imp_PyExc_TypeError
438 # define PyExc_ValueError p3imp_PyExc_ValueError
439 # define PyExc_SystemExit p3imp_PyExc_SystemExit
440 # define PyExc_RuntimeError p3imp_PyExc_RuntimeError
441 # define PyExc_ImportError p3imp_PyExc_ImportError
442 # define PyExc_OverflowError p3imp_PyExc_OverflowError
443 
444 /*
445  * Table of name to function pointer of python.
446  */
447 # define PYTHON_PROC FARPROC
448 static struct
449 {
450     char *name;
451     PYTHON_PROC *ptr;
452 } py3_funcname_table[] =
453 {
454     {"PySys_SetArgv", (PYTHON_PROC*)&py3_PySys_SetArgv},
455     {"Py_SetPythonHome", (PYTHON_PROC*)&py3_Py_SetPythonHome},
456     {"Py_Initialize", (PYTHON_PROC*)&py3_Py_Initialize},
457     {"_PyArg_ParseTuple_SizeT", (PYTHON_PROC*)&py3_PyArg_ParseTuple},
458     {"_Py_BuildValue_SizeT", (PYTHON_PROC*)&py3_Py_BuildValue},
459     {"PyMem_Free", (PYTHON_PROC*)&py3_PyMem_Free},
460     {"PyMem_Malloc", (PYTHON_PROC*)&py3_PyMem_Malloc},
461     {"PyList_New", (PYTHON_PROC*)&py3_PyList_New},
462     {"PyGILState_Ensure", (PYTHON_PROC*)&py3_PyGILState_Ensure},
463     {"PyGILState_Release", (PYTHON_PROC*)&py3_PyGILState_Release},
464     {"PySys_SetObject", (PYTHON_PROC*)&py3_PySys_SetObject},
465     {"PySys_GetObject", (PYTHON_PROC*)&py3_PySys_GetObject},
466     {"PyList_Append", (PYTHON_PROC*)&py3_PyList_Append},
467     {"PyList_Insert", (PYTHON_PROC*)&py3_PyList_Insert},
468     {"PyList_Size", (PYTHON_PROC*)&py3_PyList_Size},
469     {"PySequence_Check", (PYTHON_PROC*)&py3_PySequence_Check},
470     {"PySequence_Size", (PYTHON_PROC*)&py3_PySequence_Size},
471     {"PySequence_GetItem", (PYTHON_PROC*)&py3_PySequence_GetItem},
472     {"PySequence_Fast", (PYTHON_PROC*)&py3_PySequence_Fast},
473     {"PyTuple_Size", (PYTHON_PROC*)&py3_PyTuple_Size},
474     {"PyTuple_GetItem", (PYTHON_PROC*)&py3_PyTuple_GetItem},
475     {"PySlice_GetIndicesEx", (PYTHON_PROC*)&py3_PySlice_GetIndicesEx},
476     {"PyErr_NoMemory", (PYTHON_PROC*)&py3_PyErr_NoMemory},
477     {"Py_Finalize", (PYTHON_PROC*)&py3_Py_Finalize},
478     {"PyErr_SetString", (PYTHON_PROC*)&py3_PyErr_SetString},
479     {"PyErr_SetObject", (PYTHON_PROC*)&py3_PyErr_SetObject},
480     {"PyErr_ExceptionMatches", (PYTHON_PROC*)&py3_PyErr_ExceptionMatches},
481     {"PyRun_SimpleString", (PYTHON_PROC*)&py3_PyRun_SimpleString},
482     {"PyRun_String", (PYTHON_PROC*)&py3_PyRun_String},
483     {"PyObject_GetAttrString", (PYTHON_PROC*)&py3_PyObject_GetAttrString},
484     {"PyObject_HasAttrString", (PYTHON_PROC*)&py3_PyObject_HasAttrString},
485     {"PyObject_SetAttrString", (PYTHON_PROC*)&py3_PyObject_SetAttrString},
486     {"PyObject_CallFunctionObjArgs", (PYTHON_PROC*)&py3_PyObject_CallFunctionObjArgs},
487     {"_PyObject_CallFunction_SizeT", (PYTHON_PROC*)&py3__PyObject_CallFunction_SizeT},
488     {"PyObject_Call", (PYTHON_PROC*)&py3_PyObject_Call},
489     {"PyEval_GetGlobals", (PYTHON_PROC*)&py3_PyEval_GetGlobals},
490     {"PyEval_GetLocals", (PYTHON_PROC*)&py3_PyEval_GetLocals},
491     {"PyList_GetItem", (PYTHON_PROC*)&py3_PyList_GetItem},
492     {"PyImport_ImportModule", (PYTHON_PROC*)&py3_PyImport_ImportModule},
493     {"PyImport_AddModule", (PYTHON_PROC*)&py3_PyImport_AddModule},
494     {"PyErr_BadArgument", (PYTHON_PROC*)&py3_PyErr_BadArgument},
495     {"PyErr_Occurred", (PYTHON_PROC*)&py3_PyErr_Occurred},
496     {"PyModule_GetDict", (PYTHON_PROC*)&py3_PyModule_GetDict},
497     {"PyList_SetItem", (PYTHON_PROC*)&py3_PyList_SetItem},
498     {"PyDict_GetItemString", (PYTHON_PROC*)&py3_PyDict_GetItemString},
499     {"PyDict_Next", (PYTHON_PROC*)&py3_PyDict_Next},
500     {"PyMapping_Check", (PYTHON_PROC*)&py3_PyMapping_Check},
501     {"PyMapping_Keys", (PYTHON_PROC*)&py3_PyMapping_Keys},
502     {"PyIter_Next", (PYTHON_PROC*)&py3_PyIter_Next},
503     {"PyObject_GetIter", (PYTHON_PROC*)&py3_PyObject_GetIter},
504     {"PyObject_Repr", (PYTHON_PROC*)&py3_PyObject_Repr},
505     {"PyObject_GetItem", (PYTHON_PROC*)&py3_PyObject_GetItem},
506     {"PyObject_IsTrue", (PYTHON_PROC*)&py3_PyObject_IsTrue},
507     {"PyLong_FromLong", (PYTHON_PROC*)&py3_PyLong_FromLong},
508     {"PyDict_New", (PYTHON_PROC*)&py3_PyDict_New},
509     {"PyType_Ready", (PYTHON_PROC*)&py3_PyType_Ready},
510     {"PyDict_SetItemString", (PYTHON_PROC*)&py3_PyDict_SetItemString},
511     {"PyLong_AsLong", (PYTHON_PROC*)&py3_PyLong_AsLong},
512     {"PyErr_SetNone", (PYTHON_PROC*)&py3_PyErr_SetNone},
513     {"PyEval_InitThreads", (PYTHON_PROC*)&py3_PyEval_InitThreads},
514     {"PyEval_RestoreThread", (PYTHON_PROC*)&py3_PyEval_RestoreThread},
515     {"PyEval_SaveThread", (PYTHON_PROC*)&py3_PyEval_SaveThread},
516     {"_PyArg_Parse_SizeT", (PYTHON_PROC*)&py3_PyArg_Parse},
517     {"Py_IsInitialized", (PYTHON_PROC*)&py3_Py_IsInitialized},
518     {"_PyObject_NextNotImplemented", (PYTHON_PROC*)&py3__PyObject_NextNotImplemented},
519     {"_Py_NoneStruct", (PYTHON_PROC*)&py3__Py_NoneStruct},
520     {"_Py_FalseStruct", (PYTHON_PROC*)&py3__Py_FalseStruct},
521     {"_Py_TrueStruct", (PYTHON_PROC*)&py3__Py_TrueStruct},
522     {"PyErr_Clear", (PYTHON_PROC*)&py3_PyErr_Clear},
523     {"PyErr_Format", (PYTHON_PROC*)&py3_PyErr_Format},
524     {"PyErr_PrintEx", (PYTHON_PROC*)&py3_PyErr_PrintEx},
525     {"PyObject_Init", (PYTHON_PROC*)&py3__PyObject_Init},
526     {"PyModule_AddObject", (PYTHON_PROC*)&py3_PyModule_AddObject},
527     {"PyImport_AppendInittab", (PYTHON_PROC*)&py3_PyImport_AppendInittab},
528 # if PY_VERSION_HEX >= 0x030300f0
529     {"PyUnicode_AsUTF8", (PYTHON_PROC*)&py3_PyUnicode_AsUTF8},
530 # else
531     {"_PyUnicode_AsString", (PYTHON_PROC*)&py3__PyUnicode_AsString},
532 # endif
533 # ifndef Py_UNICODE_USE_UCS_FUNCTIONS
534     {"PyUnicode_FromFormat", (PYTHON_PROC*)&py3_PyUnicode_FromFormat},
535 # else
536 #  ifdef Py_UNICODE_WIDE
537     {"PyUnicodeUCS4_FromFormat", (PYTHON_PROC*)&py3_PyUnicodeUCS4_FromFormat},
538 #  else
539     {"PyUnicodeUCS2_FromFormat", (PYTHON_PROC*)&py3_PyUnicodeUCS2_FromFormat},
540 #  endif
541 # endif
542     {"PyBytes_AsString", (PYTHON_PROC*)&py3_PyBytes_AsString},
543     {"PyBytes_AsStringAndSize", (PYTHON_PROC*)&py3_PyBytes_AsStringAndSize},
544     {"PyBytes_FromString", (PYTHON_PROC*)&py3_PyBytes_FromString},
545     {"PyFloat_FromDouble", (PYTHON_PROC*)&py3_PyFloat_FromDouble},
546     {"PyFloat_AsDouble", (PYTHON_PROC*)&py3_PyFloat_AsDouble},
547     {"PyObject_GenericGetAttr", (PYTHON_PROC*)&py3_PyObject_GenericGetAttr},
548     {"PyType_GenericAlloc", (PYTHON_PROC*)&py3_PyType_GenericAlloc},
549     {"PyType_GenericNew", (PYTHON_PROC*)&py3_PyType_GenericNew},
550     {"PyType_Type", (PYTHON_PROC*)&py3_PyType_Type},
551     {"PySlice_Type", (PYTHON_PROC*)&py3_PySlice_Type},
552     {"PyFloat_Type", (PYTHON_PROC*)&py3_PyFloat_Type},
553     {"PyBool_Type", (PYTHON_PROC*)&py3_PyBool_Type},
554     {"PyNumber_Check", (PYTHON_PROC*)&py3_PyNumber_Check},
555     {"PyNumber_Long", (PYTHON_PROC*)&py3_PyNumber_Long},
556     {"PyErr_NewException", (PYTHON_PROC*)&py3_PyErr_NewException},
557 # ifdef Py_DEBUG
558     {"_Py_NegativeRefcount", (PYTHON_PROC*)&py3__Py_NegativeRefcount},
559     {"_Py_RefTotal", (PYTHON_PROC*)&py3__Py_RefTotal},
560     {"_Py_Dealloc", (PYTHON_PROC*)&py3__Py_Dealloc},
561     {"PyModule_Create2TraceRefs", (PYTHON_PROC*)&py3_PyModule_Create2TraceRefs},
562 # else
563     {"PyModule_Create2", (PYTHON_PROC*)&py3_PyModule_Create2},
564 # endif
565 # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC)
566     {"_PyObject_DebugFree", (PYTHON_PROC*)&py3__PyObject_DebugFree},
567     {"_PyObject_DebugMalloc", (PYTHON_PROC*)&py3__PyObject_DebugMalloc},
568 # else
569     {"PyObject_Malloc", (PYTHON_PROC*)&py3_PyObject_Malloc},
570     {"PyObject_Free", (PYTHON_PROC*)&py3_PyObject_Free},
571 # endif
572     {"_PyObject_GC_New", (PYTHON_PROC*)&py3__PyObject_GC_New},
573     {"PyObject_GC_Del", (PYTHON_PROC*)&py3_PyObject_GC_Del},
574     {"PyObject_GC_UnTrack", (PYTHON_PROC*)&py3_PyObject_GC_UnTrack},
575     {"PyType_IsSubtype", (PYTHON_PROC*)&py3_PyType_IsSubtype},
576     {"PyCapsule_New", (PYTHON_PROC*)&py3_PyCapsule_New},
577     {"PyCapsule_GetPointer", (PYTHON_PROC*)&py3_PyCapsule_GetPointer},
578     {"", NULL},
579 };
580 
581 /*
582  * Free python.dll
583  */
584     static void
585 end_dynamic_python3(void)
586 {
587     if (hinstPy3 != 0)
588     {
589 	close_dll(hinstPy3);
590 	hinstPy3 = 0;
591     }
592 }
593 
594 /*
595  * Load library and get all pointers.
596  * Parameter 'libname' provides name of DLL.
597  * Return OK or FAIL.
598  */
599     static int
600 py3_runtime_link_init(char *libname, int verbose)
601 {
602     int i;
603     void *ucs_from_string, *ucs_decode, *ucs_as_encoded_string;
604 
605 # if !(defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)) && defined(UNIX) && defined(FEAT_PYTHON)
606     /* Can't have Python and Python3 loaded at the same time.
607      * It cause a crash, because RTLD_GLOBAL is needed for
608      * standard C extension libraries of one or both python versions. */
609     if (python_loaded())
610     {
611 	if (verbose)
612 	    EMSG(_("E837: This Vim cannot execute :py3 after using :python"));
613 	return FAIL;
614     }
615 # endif
616 
617     if (hinstPy3 != 0)
618 	return OK;
619     hinstPy3 = load_dll(libname);
620 
621     if (!hinstPy3)
622     {
623 	if (verbose)
624 	    EMSG2(_(e_loadlib), libname);
625 	return FAIL;
626     }
627 
628     for (i = 0; py3_funcname_table[i].ptr; ++i)
629     {
630 	if ((*py3_funcname_table[i].ptr = symbol_from_dll(hinstPy3,
631 			py3_funcname_table[i].name)) == NULL)
632 	{
633 	    close_dll(hinstPy3);
634 	    hinstPy3 = 0;
635 	    if (verbose)
636 		EMSG2(_(e_loadfunc), py3_funcname_table[i].name);
637 	    return FAIL;
638 	}
639     }
640 
641     /* Load unicode functions separately as only the ucs2 or the ucs4 functions
642      * will be present in the library. */
643 # if PY_VERSION_HEX >= 0x030300f0
644     ucs_from_string = symbol_from_dll(hinstPy3, "PyUnicode_FromString");
645     ucs_decode = symbol_from_dll(hinstPy3, "PyUnicode_Decode");
646     ucs_as_encoded_string = symbol_from_dll(hinstPy3,
647 	    "PyUnicode_AsEncodedString");
648 # else
649     ucs_from_string = symbol_from_dll(hinstPy3, "PyUnicodeUCS2_FromString");
650     ucs_decode = symbol_from_dll(hinstPy3,
651 	    "PyUnicodeUCS2_Decode");
652     ucs_as_encoded_string = symbol_from_dll(hinstPy3,
653 	    "PyUnicodeUCS2_AsEncodedString");
654     if (!ucs_from_string || !ucs_decode || !ucs_as_encoded_string)
655     {
656 	ucs_from_string = symbol_from_dll(hinstPy3,
657 		"PyUnicodeUCS4_FromString");
658 	ucs_decode = symbol_from_dll(hinstPy3,
659 		"PyUnicodeUCS4_Decode");
660 	ucs_as_encoded_string = symbol_from_dll(hinstPy3,
661 		"PyUnicodeUCS4_AsEncodedString");
662     }
663 # endif
664     if (ucs_from_string && ucs_decode && ucs_as_encoded_string)
665     {
666 	py3_PyUnicode_FromString = ucs_from_string;
667 	py3_PyUnicode_Decode = ucs_decode;
668 	py3_PyUnicode_AsEncodedString = ucs_as_encoded_string;
669     }
670     else
671     {
672 	close_dll(hinstPy3);
673 	hinstPy3 = 0;
674 	if (verbose)
675 	    EMSG2(_(e_loadfunc), "PyUnicode_UCSX_*");
676 	return FAIL;
677     }
678 
679     return OK;
680 }
681 
682 /*
683  * If python is enabled (there is installed python on Windows system) return
684  * TRUE, else FALSE.
685  */
686     int
687 python3_enabled(int verbose)
688 {
689     return py3_runtime_link_init(DYNAMIC_PYTHON3_DLL, verbose) == OK;
690 }
691 
692 /* Load the standard Python exceptions - don't import the symbols from the
693  * DLL, as this can cause errors (importing data symbols is not reliable).
694  */
695 static void get_py3_exceptions __ARGS((void));
696 
697     static void
698 get_py3_exceptions()
699 {
700     PyObject *exmod = PyImport_ImportModule("builtins");
701     PyObject *exdict = PyModule_GetDict(exmod);
702     p3imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError");
703     p3imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError");
704     p3imp_PyExc_KeyError = PyDict_GetItemString(exdict, "KeyError");
705     p3imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt");
706     p3imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError");
707     p3imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError");
708     p3imp_PyExc_SystemExit = PyDict_GetItemString(exdict, "SystemExit");
709     p3imp_PyExc_RuntimeError = PyDict_GetItemString(exdict, "RuntimeError");
710     p3imp_PyExc_ImportError = PyDict_GetItemString(exdict, "ImportError");
711     p3imp_PyExc_OverflowError = PyDict_GetItemString(exdict, "OverflowError");
712     Py_XINCREF(p3imp_PyExc_AttributeError);
713     Py_XINCREF(p3imp_PyExc_IndexError);
714     Py_XINCREF(p3imp_PyExc_KeyError);
715     Py_XINCREF(p3imp_PyExc_KeyboardInterrupt);
716     Py_XINCREF(p3imp_PyExc_TypeError);
717     Py_XINCREF(p3imp_PyExc_ValueError);
718     Py_XINCREF(p3imp_PyExc_SystemExit);
719     Py_XINCREF(p3imp_PyExc_RuntimeError);
720     Py_XINCREF(p3imp_PyExc_ImportError);
721     Py_XINCREF(p3imp_PyExc_OverflowError);
722     Py_XDECREF(exmod);
723 }
724 #endif /* DYNAMIC_PYTHON3 */
725 
726 static int py3initialised = 0;
727 
728 #define PYINITIALISED py3initialised
729 
730 #define DESTRUCTOR_FINISH(self) Py_TYPE(self)->tp_free((PyObject*)self)
731 
732 #define WIN_PYTHON_REF(win) win->w_python3_ref
733 #define BUF_PYTHON_REF(buf) buf->b_python3_ref
734 #define TAB_PYTHON_REF(tab) tab->tp_python3_ref
735 
736     static void
737 call_PyObject_Free(void *p)
738 {
739 #if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC)
740     _PyObject_DebugFree(p);
741 #else
742     PyObject_Free(p);
743 #endif
744 }
745 
746     static PyObject *
747 call_PyType_GenericNew(PyTypeObject *type, PyObject *args, PyObject *kwds)
748 {
749     return PyType_GenericNew(type,args,kwds);
750 }
751 
752     static PyObject *
753 call_PyType_GenericAlloc(PyTypeObject *type, Py_ssize_t nitems)
754 {
755     return PyType_GenericAlloc(type,nitems);
756 }
757 
758 static PyObject *OutputGetattro(PyObject *, PyObject *);
759 static int OutputSetattro(PyObject *, PyObject *, PyObject *);
760 static PyObject *BufferGetattro(PyObject *, PyObject *);
761 static int BufferSetattro(PyObject *, PyObject *, PyObject *);
762 static PyObject *TabPageGetattro(PyObject *, PyObject *);
763 static PyObject *WindowGetattro(PyObject *, PyObject *);
764 static int WindowSetattro(PyObject *, PyObject *, PyObject *);
765 static PyObject *RangeGetattro(PyObject *, PyObject *);
766 static PyObject *CurrentGetattro(PyObject *, PyObject *);
767 static int CurrentSetattro(PyObject *, PyObject *, PyObject *);
768 static PyObject *DictionaryGetattro(PyObject *, PyObject *);
769 static int DictionarySetattro(PyObject *, PyObject *, PyObject *);
770 static PyObject *ListGetattro(PyObject *, PyObject *);
771 static int ListSetattro(PyObject *, PyObject *, PyObject *);
772 static PyObject *FunctionGetattro(PyObject *, PyObject *);
773 
774 static PyObject *VimPathHook(PyObject *, PyObject *);
775 
776 static struct PyModuleDef vimmodule;
777 
778 #define PY_CAN_RECURSE
779 
780 /*
781  * Include the code shared with if_python.c
782  */
783 #include "if_py_both.h"
784 
785 #define GET_ATTR_STRING(name, nameobj) \
786     char	*name = ""; \
787     if (PyUnicode_Check(nameobj)) \
788 	name = _PyUnicode_AsString(nameobj)
789 
790 #define PY3OBJ_DELETED(obj) (obj->ob_base.ob_refcnt<=0)
791 
792 /******************************************************
793  * Internal function prototypes.
794  */
795 
796 static PyObject *Py3Init_vim(void);
797 
798 /******************************************************
799  * 1. Python interpreter main program.
800  */
801 
802     void
803 python3_end()
804 {
805     static int recurse = 0;
806 
807     /* If a crash occurs while doing this, don't try again. */
808     if (recurse != 0)
809 	return;
810 
811     ++recurse;
812 
813 #ifdef DYNAMIC_PYTHON3
814     if (hinstPy3)
815 #endif
816     if (Py_IsInitialized())
817     {
818 	// acquire lock before finalizing
819 	PyGILState_Ensure();
820 
821 	Py_Finalize();
822     }
823 
824 #ifdef DYNAMIC_PYTHON3
825     end_dynamic_python3();
826 #endif
827 
828     --recurse;
829 }
830 
831 #if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON)) || defined(PROTO)
832     int
833 python3_loaded()
834 {
835     return (hinstPy3 != 0);
836 }
837 #endif
838 
839     static int
840 Python3_Init(void)
841 {
842     if (!py3initialised)
843     {
844 #ifdef DYNAMIC_PYTHON3
845 	if (!python3_enabled(TRUE))
846 	{
847 	    EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded."));
848 	    goto fail;
849 	}
850 #endif
851 
852 	init_structs();
853 
854 
855 #ifdef PYTHON3_HOME
856 	Py_SetPythonHome(PYTHON3_HOME);
857 #endif
858 
859 	PyImport_AppendInittab("vim", Py3Init_vim);
860 
861 #if !defined(MACOS) || defined(MACOS_X_UNIX)
862 	Py_Initialize();
863 #else
864 	PyMac_Initialize();
865 #endif
866 	/* Initialise threads, and below save the state using
867 	 * PyEval_SaveThread.  Without the call to PyEval_SaveThread, thread
868 	 * specific state (such as the system trace hook), will be lost
869 	 * between invocations of Python code. */
870 	PyEval_InitThreads();
871 #ifdef DYNAMIC_PYTHON3
872 	get_py3_exceptions();
873 #endif
874 
875 	if (PythonIO_Init_io())
876 	    goto fail;
877 
878 	globals = PyModule_GetDict(PyImport_AddModule("__main__"));
879 
880 	/* Remove the element from sys.path that was added because of our
881 	 * argv[0] value in Py3Init_vim().  Previously we used an empty
882 	 * string, but depending on the OS we then get an empty entry or
883 	 * the current directory in sys.path.
884 	 * Only after vim has been imported, the element does exist in
885 	 * sys.path.
886 	 */
887 	PyRun_SimpleString("import vim; import sys; sys.path = list(filter(lambda x: not x.endswith('must>not&exist'), sys.path))");
888 
889 	/* lock is created and acquired in PyEval_InitThreads() and thread
890 	 * state is created in Py_Initialize()
891 	 * there _PyGILState_NoteThreadState() also sets gilcounter to 1
892 	 * (python must have threads enabled!)
893 	 * so the following does both: unlock GIL and save thread state in TLS
894 	 * without deleting thread state
895 	 */
896 	PyEval_SaveThread();
897 
898 	py3initialised = 1;
899     }
900 
901     return 0;
902 
903 fail:
904     /* We call PythonIO_Flush() here to print any Python errors.
905      * This is OK, as it is possible to call this function even
906      * if PythonIO_Init_io() has not completed successfully (it will
907      * not do anything in this case).
908      */
909     PythonIO_Flush();
910     return -1;
911 }
912 
913 /*
914  * External interface
915  */
916     static void
917 DoPyCommand(const char *cmd, rangeinitializer init_range, runner run, void *arg)
918 {
919 #if defined(MACOS) && !defined(MACOS_X_UNIX)
920     GrafPtr		oldPort;
921 #endif
922 #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
923     char		*saved_locale;
924 #endif
925     PyObject		*cmdstr;
926     PyObject		*cmdbytes;
927     PyGILState_STATE	pygilstate;
928 
929 #if defined(MACOS) && !defined(MACOS_X_UNIX)
930     GetPort(&oldPort);
931     /* Check if the Python library is available */
932     if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress)
933 	goto theend;
934 #endif
935     if (Python3_Init())
936 	goto theend;
937 
938     init_range(arg);
939 
940     Python_Release_Vim();	    /* leave vim */
941 
942 #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
943     /* Python only works properly when the LC_NUMERIC locale is "C". */
944     saved_locale = setlocale(LC_NUMERIC, NULL);
945     if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0)
946 	saved_locale = NULL;
947     else
948     {
949 	/* Need to make a copy, value may change when setting new locale. */
950 	saved_locale = (char *)vim_strsave((char_u *)saved_locale);
951 	(void)setlocale(LC_NUMERIC, "C");
952     }
953 #endif
954 
955     pygilstate = PyGILState_Ensure();
956 
957     /* PyRun_SimpleString expects a UTF-8 string. Wrong encoding may cause
958      * SyntaxError (unicode error). */
959     cmdstr = PyUnicode_Decode(cmd, strlen(cmd),
960 					(char *)ENC_OPT, CODEC_ERROR_HANDLER);
961     cmdbytes = PyUnicode_AsEncodedString(cmdstr, "utf-8", CODEC_ERROR_HANDLER);
962     Py_XDECREF(cmdstr);
963 
964     run(PyBytes_AsString(cmdbytes), arg, &pygilstate);
965     Py_XDECREF(cmdbytes);
966 
967     PyGILState_Release(pygilstate);
968 
969 #if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
970     if (saved_locale != NULL)
971     {
972 	(void)setlocale(LC_NUMERIC, saved_locale);
973 	vim_free(saved_locale);
974     }
975 #endif
976 
977     Python_Lock_Vim();		    /* enter vim */
978     PythonIO_Flush();
979 #if defined(MACOS) && !defined(MACOS_X_UNIX)
980     SetPort(oldPort);
981 #endif
982 
983 theend:
984     return;	    /* keeps lint happy */
985 }
986 
987 /*
988  * ":py3"
989  */
990     void
991 ex_py3(exarg_T *eap)
992 {
993     char_u *script;
994 
995     script = script_get(eap, eap->arg);
996     if (!eap->skip)
997     {
998 	DoPyCommand(script == NULL ? (char *) eap->arg : (char *) script,
999 		(rangeinitializer) init_range_cmd,
1000 		(runner) run_cmd,
1001 		(void *) eap);
1002     }
1003     vim_free(script);
1004 }
1005 
1006 #define BUFFER_SIZE 2048
1007 
1008 /*
1009  * ":py3file"
1010  */
1011     void
1012 ex_py3file(exarg_T *eap)
1013 {
1014     static char buffer[BUFFER_SIZE];
1015     const char *file;
1016     char *p;
1017     int i;
1018 
1019     /* Have to do it like this. PyRun_SimpleFile requires you to pass a
1020      * stdio file pointer, but Vim and the Python DLL are compiled with
1021      * different options under Windows, meaning that stdio pointers aren't
1022      * compatible between the two. Yuk.
1023      *
1024      * construct: exec(compile(open('a_filename', 'rb').read(), 'a_filename', 'exec'))
1025      *
1026      * Using bytes so that Python can detect the source encoding as it normally
1027      * does. The doc does not say "compile" accept bytes, though.
1028      *
1029      * We need to escape any backslashes or single quotes in the file name, so that
1030      * Python won't mangle the file name.
1031      */
1032 
1033     strcpy(buffer, "exec(compile(open('");
1034     p = buffer + 19; /* size of "exec(compile(open('" */
1035 
1036     for (i=0; i<2; ++i)
1037     {
1038 	file = (char *)eap->arg;
1039 	while (*file && p < buffer + (BUFFER_SIZE - 3))
1040 	{
1041 	    if (*file == '\\' || *file == '\'')
1042 		*p++ = '\\';
1043 	    *p++ = *file++;
1044 	}
1045 	/* If we didn't finish the file name, we hit a buffer overflow */
1046 	if (*file != '\0')
1047 	    return;
1048 	if (i==0)
1049 	{
1050 	    strcpy(p,"','rb').read(),'");
1051 	    p += 16;
1052 	}
1053 	else
1054 	{
1055 	    strcpy(p,"','exec'))");
1056 	    p += 10;
1057 	}
1058     }
1059 
1060 
1061     /* Execute the file */
1062     DoPyCommand(buffer,
1063 	    (rangeinitializer) init_range_cmd,
1064 	    (runner) run_cmd,
1065 	    (void *) eap);
1066 }
1067 
1068     void
1069 ex_py3do(exarg_T *eap)
1070 {
1071     DoPyCommand((char *)eap->arg,
1072 	    (rangeinitializer)init_range_cmd,
1073 	    (runner)run_do,
1074 	    (void *)eap);
1075 }
1076 
1077 /******************************************************
1078  * 2. Python output stream: writes output via [e]msg().
1079  */
1080 
1081 /* Implementation functions
1082  */
1083 
1084     static PyObject *
1085 OutputGetattro(PyObject *self, PyObject *nameobj)
1086 {
1087     GET_ATTR_STRING(name, nameobj);
1088 
1089     if (strcmp(name, "softspace") == 0)
1090 	return PyLong_FromLong(((OutputObject *)(self))->softspace);
1091 
1092     return PyObject_GenericGetAttr(self, nameobj);
1093 }
1094 
1095     static int
1096 OutputSetattro(PyObject *self, PyObject *nameobj, PyObject *val)
1097 {
1098     GET_ATTR_STRING(name, nameobj);
1099 
1100     return OutputSetattr((OutputObject *)(self), name, val);
1101 }
1102 
1103 /******************************************************
1104  * 3. Implementation of the Vim module for Python
1105  */
1106 
1107 /* Window type - Implementation functions
1108  * --------------------------------------
1109  */
1110 
1111 #define WindowType_Check(obj) ((obj)->ob_base.ob_type == &WindowType)
1112 
1113 /* Buffer type - Implementation functions
1114  * --------------------------------------
1115  */
1116 
1117 #define BufferType_Check(obj) ((obj)->ob_base.ob_type == &BufferType)
1118 
1119 static PyObject* BufferSubscript(PyObject *self, PyObject *idx);
1120 static Py_ssize_t BufferAsSubscript(PyObject *self, PyObject *idx, PyObject *val);
1121 
1122 /* Line range type - Implementation functions
1123  * --------------------------------------
1124  */
1125 
1126 #define RangeType_Check(obj) ((obj)->ob_base.ob_type == &RangeType)
1127 
1128 static PyObject* RangeSubscript(PyObject *self, PyObject *idx);
1129 static Py_ssize_t RangeAsItem(PyObject *, Py_ssize_t, PyObject *);
1130 static Py_ssize_t RangeAsSubscript(PyObject *self, PyObject *idx, PyObject *val);
1131 
1132 /* Current objects type - Implementation functions
1133  * -----------------------------------------------
1134  */
1135 
1136 static PySequenceMethods BufferAsSeq = {
1137     (lenfunc)		BufferLength,	    /* sq_length,    len(x)   */
1138     (binaryfunc)	0,		    /* sq_concat,    x+y      */
1139     (ssizeargfunc)	0,		    /* sq_repeat,    x*n      */
1140     (ssizeargfunc)	BufferItem,	    /* sq_item,      x[i]     */
1141     0,					    /* was_sq_slice,	 x[i:j]   */
1142     0,					    /* sq_ass_item,  x[i]=v   */
1143     0,					    /* sq_ass_slice, x[i:j]=v */
1144     0,					    /* sq_contains */
1145     0,					    /* sq_inplace_concat */
1146     0,					    /* sq_inplace_repeat */
1147 };
1148 
1149 static PyMappingMethods BufferAsMapping = {
1150     /* mp_length	*/ (lenfunc)BufferLength,
1151     /* mp_subscript     */ (binaryfunc)BufferSubscript,
1152     /* mp_ass_subscript */ (objobjargproc)BufferAsSubscript,
1153 };
1154 
1155 
1156 /* Buffer object
1157  */
1158 
1159     static PyObject *
1160 BufferGetattro(PyObject *self, PyObject *nameobj)
1161 {
1162     PyObject *r;
1163 
1164     GET_ATTR_STRING(name, nameobj);
1165 
1166     if ((r = BufferAttrValid((BufferObject *)(self), name)))
1167 	return r;
1168 
1169     if (CheckBuffer((BufferObject *)(self)))
1170 	return NULL;
1171 
1172     r = BufferAttr((BufferObject *)(self), name);
1173     if (r || PyErr_Occurred())
1174 	return r;
1175     else
1176 	return PyObject_GenericGetAttr(self, nameobj);
1177 }
1178 
1179     static int
1180 BufferSetattro(PyObject *self, PyObject *nameobj, PyObject *val)
1181 {
1182     GET_ATTR_STRING(name, nameobj);
1183 
1184     return BufferSetattr((BufferObject *)(self), name, val);
1185 }
1186 
1187 /******************/
1188 
1189     static PyObject *
1190 BufferSubscript(PyObject *self, PyObject* idx)
1191 {
1192     if (PyLong_Check(idx))
1193     {
1194 	long _idx = PyLong_AsLong(idx);
1195 	return BufferItem((BufferObject *)(self), _idx);
1196     } else if (PySlice_Check(idx))
1197     {
1198 	Py_ssize_t start, stop, step, slicelen;
1199 
1200 	if (CheckBuffer((BufferObject *) self))
1201 	    return NULL;
1202 
1203 	if (PySlice_GetIndicesEx((PySliceObject_T *)idx,
1204 	      (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1205 	      &start, &stop,
1206 	      &step, &slicelen) < 0)
1207 	{
1208 	    return NULL;
1209 	}
1210 	return BufferSlice((BufferObject *)(self), start, stop);
1211     }
1212     else
1213     {
1214 	RAISE_INVALID_INDEX_TYPE(idx);
1215 	return NULL;
1216     }
1217 }
1218 
1219     static Py_ssize_t
1220 BufferAsSubscript(PyObject *self, PyObject* idx, PyObject* val)
1221 {
1222     if (PyLong_Check(idx))
1223     {
1224 	long n = PyLong_AsLong(idx);
1225 	return RBAsItem((BufferObject *)(self), n, val, 1,
1226 		    (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1227 		    NULL);
1228     } else if (PySlice_Check(idx))
1229     {
1230 	Py_ssize_t start, stop, step, slicelen;
1231 
1232 	if (CheckBuffer((BufferObject *) self))
1233 	    return -1;
1234 
1235 	if (PySlice_GetIndicesEx((PySliceObject_T *)idx,
1236 	      (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1237 	      &start, &stop,
1238 	      &step, &slicelen) < 0)
1239 	{
1240 	    return -1;
1241 	}
1242 	return RBAsSlice((BufferObject *)(self), start, stop, val, 1,
1243 			  (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count,
1244 			  NULL);
1245     }
1246     else
1247     {
1248 	RAISE_INVALID_INDEX_TYPE(idx);
1249 	return -1;
1250     }
1251 }
1252 
1253 static PySequenceMethods RangeAsSeq = {
1254     (lenfunc)		RangeLength,	 /* sq_length,	  len(x)   */
1255     (binaryfunc)	0,		 /* RangeConcat, sq_concat,  x+y   */
1256     (ssizeargfunc)	0,		 /* RangeRepeat, sq_repeat,  x*n   */
1257     (ssizeargfunc)	RangeItem,	 /* sq_item,	  x[i]	   */
1258     0,					 /* was_sq_slice,     x[i:j]   */
1259     (ssizeobjargproc)	RangeAsItem,	 /* sq_as_item,  x[i]=v   */
1260     0,					 /* sq_ass_slice, x[i:j]=v */
1261     0,					 /* sq_contains */
1262     0,					 /* sq_inplace_concat */
1263     0,					 /* sq_inplace_repeat */
1264 };
1265 
1266 static PyMappingMethods RangeAsMapping = {
1267     /* mp_length	*/ (lenfunc)RangeLength,
1268     /* mp_subscript     */ (binaryfunc)RangeSubscript,
1269     /* mp_ass_subscript */ (objobjargproc)RangeAsSubscript,
1270 };
1271 
1272 /* Line range object - Implementation
1273  */
1274 
1275     static PyObject *
1276 RangeGetattro(PyObject *self, PyObject *nameobj)
1277 {
1278     GET_ATTR_STRING(name, nameobj);
1279 
1280     if (strcmp(name, "start") == 0)
1281 	return Py_BuildValue("n", ((RangeObject *)(self))->start - 1);
1282     else if (strcmp(name, "end") == 0)
1283 	return Py_BuildValue("n", ((RangeObject *)(self))->end - 1);
1284     else
1285 	return PyObject_GenericGetAttr(self, nameobj);
1286 }
1287 
1288 /****************/
1289 
1290     static Py_ssize_t
1291 RangeAsItem(PyObject *self, Py_ssize_t n, PyObject *val)
1292 {
1293     return RBAsItem(((RangeObject *)(self))->buf, n, val,
1294 		    ((RangeObject *)(self))->start,
1295 		    ((RangeObject *)(self))->end,
1296 		    &((RangeObject *)(self))->end);
1297 }
1298 
1299     static Py_ssize_t
1300 RangeAsSlice(PyObject *self, Py_ssize_t lo, Py_ssize_t hi, PyObject *val)
1301 {
1302     return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val,
1303 		    ((RangeObject *)(self))->start,
1304 		    ((RangeObject *)(self))->end,
1305 		    &((RangeObject *)(self))->end);
1306 }
1307 
1308     static PyObject *
1309 RangeSubscript(PyObject *self, PyObject* idx)
1310 {
1311     if (PyLong_Check(idx))
1312     {
1313 	long _idx = PyLong_AsLong(idx);
1314 	return RangeItem((RangeObject *)(self), _idx);
1315     } else if (PySlice_Check(idx))
1316     {
1317 	Py_ssize_t start, stop, step, slicelen;
1318 
1319 	if (PySlice_GetIndicesEx((PySliceObject_T *)idx,
1320 		((RangeObject *)(self))->end-((RangeObject *)(self))->start+1,
1321 		&start, &stop,
1322 		&step, &slicelen) < 0)
1323 	{
1324 	    return NULL;
1325 	}
1326 	return RangeSlice((RangeObject *)(self), start, stop);
1327     }
1328     else
1329     {
1330 	RAISE_INVALID_INDEX_TYPE(idx);
1331 	return NULL;
1332     }
1333 }
1334 
1335     static Py_ssize_t
1336 RangeAsSubscript(PyObject *self, PyObject *idx, PyObject *val)
1337 {
1338     if (PyLong_Check(idx))
1339     {
1340 	long n = PyLong_AsLong(idx);
1341 	return RangeAsItem(self, n, val);
1342     } else if (PySlice_Check(idx))
1343     {
1344 	Py_ssize_t start, stop, step, slicelen;
1345 
1346 	if (PySlice_GetIndicesEx((PySliceObject_T *)idx,
1347 		((RangeObject *)(self))->end-((RangeObject *)(self))->start+1,
1348 		&start, &stop,
1349 		&step, &slicelen) < 0)
1350 	{
1351 	    return -1;
1352 	}
1353 	return RangeAsSlice(self, start, stop, val);
1354     }
1355     else
1356     {
1357 	RAISE_INVALID_INDEX_TYPE(idx);
1358 	return -1;
1359     }
1360 }
1361 
1362 /* TabPage object - Implementation
1363  */
1364 
1365     static PyObject *
1366 TabPageGetattro(PyObject *self, PyObject *nameobj)
1367 {
1368     PyObject *r;
1369 
1370     GET_ATTR_STRING(name, nameobj);
1371 
1372     if ((r = TabPageAttrValid((TabPageObject *)(self), name)))
1373 	return r;
1374 
1375     if (CheckTabPage((TabPageObject *)(self)))
1376 	return NULL;
1377 
1378     r = TabPageAttr((TabPageObject *)(self), name);
1379     if (r || PyErr_Occurred())
1380 	return r;
1381     else
1382 	return PyObject_GenericGetAttr(self, nameobj);
1383 }
1384 
1385 /* Window object - Implementation
1386  */
1387 
1388     static PyObject *
1389 WindowGetattro(PyObject *self, PyObject *nameobj)
1390 {
1391     PyObject *r;
1392 
1393     GET_ATTR_STRING(name, nameobj);
1394 
1395     if ((r = WindowAttrValid((WindowObject *)(self), name)))
1396 	return r;
1397 
1398     if (CheckWindow((WindowObject *)(self)))
1399 	return NULL;
1400 
1401     r = WindowAttr((WindowObject *)(self), name);
1402     if (r || PyErr_Occurred())
1403 	return r;
1404     else
1405 	return PyObject_GenericGetAttr(self, nameobj);
1406 }
1407 
1408     static int
1409 WindowSetattro(PyObject *self, PyObject *nameobj, PyObject *val)
1410 {
1411     GET_ATTR_STRING(name, nameobj);
1412 
1413     return WindowSetattr((WindowObject *)(self), name, val);
1414 }
1415 
1416 /* Tab page list object - Definitions
1417  */
1418 
1419 static PySequenceMethods TabListAsSeq = {
1420     (lenfunc)	     TabListLength,	    /* sq_length,    len(x)   */
1421     (binaryfunc)     0,			    /* sq_concat,    x+y      */
1422     (ssizeargfunc)   0,			    /* sq_repeat,    x*n      */
1423     (ssizeargfunc)   TabListItem,	    /* sq_item,      x[i]     */
1424     0,					    /* sq_slice,     x[i:j]   */
1425     (ssizeobjargproc)0,			    /* sq_as_item,  x[i]=v   */
1426     0,					    /* sq_ass_slice, x[i:j]=v */
1427     0,					    /* sq_contains */
1428     0,					    /* sq_inplace_concat */
1429     0,					    /* sq_inplace_repeat */
1430 };
1431 
1432 /* Window list object - Definitions
1433  */
1434 
1435 static PySequenceMethods WinListAsSeq = {
1436     (lenfunc)	     WinListLength,	    /* sq_length,    len(x)   */
1437     (binaryfunc)     0,			    /* sq_concat,    x+y      */
1438     (ssizeargfunc)   0,			    /* sq_repeat,    x*n      */
1439     (ssizeargfunc)   WinListItem,	    /* sq_item,      x[i]     */
1440     0,					    /* sq_slice,     x[i:j]   */
1441     (ssizeobjargproc)0,			    /* sq_as_item,  x[i]=v   */
1442     0,					    /* sq_ass_slice, x[i:j]=v */
1443     0,					    /* sq_contains */
1444     0,					    /* sq_inplace_concat */
1445     0,					    /* sq_inplace_repeat */
1446 };
1447 
1448 /* Current items object - Implementation
1449  */
1450     static PyObject *
1451 CurrentGetattro(PyObject *self, PyObject *nameobj)
1452 {
1453     PyObject	*r;
1454     GET_ATTR_STRING(name, nameobj);
1455     if (!(r = CurrentGetattr(self, name)))
1456 	return PyObject_GenericGetAttr(self, nameobj);
1457     return r;
1458 }
1459 
1460     static int
1461 CurrentSetattro(PyObject *self, PyObject *nameobj, PyObject *value)
1462 {
1463     GET_ATTR_STRING(name, nameobj);
1464     return CurrentSetattr(self, name, value);
1465 }
1466 
1467 /* Dictionary object - Definitions
1468  */
1469 
1470     static PyObject *
1471 DictionaryGetattro(PyObject *self, PyObject *nameobj)
1472 {
1473     DictionaryObject	*this = ((DictionaryObject *) (self));
1474 
1475     GET_ATTR_STRING(name, nameobj);
1476 
1477     if (strcmp(name, "locked") == 0)
1478 	return PyLong_FromLong(this->dict->dv_lock);
1479     else if (strcmp(name, "scope") == 0)
1480 	return PyLong_FromLong(this->dict->dv_scope);
1481 
1482     return PyObject_GenericGetAttr(self, nameobj);
1483 }
1484 
1485     static int
1486 DictionarySetattro(PyObject *self, PyObject *nameobj, PyObject *val)
1487 {
1488     GET_ATTR_STRING(name, nameobj);
1489     return DictionarySetattr((DictionaryObject *)(self), name, val);
1490 }
1491 
1492 /* List object - Definitions
1493  */
1494 
1495     static PyObject *
1496 ListGetattro(PyObject *self, PyObject *nameobj)
1497 {
1498     GET_ATTR_STRING(name, nameobj);
1499 
1500     if (strcmp(name, "locked") == 0)
1501 	return PyLong_FromLong(((ListObject *) (self))->list->lv_lock);
1502 
1503     return PyObject_GenericGetAttr(self, nameobj);
1504 }
1505 
1506     static int
1507 ListSetattro(PyObject *self, PyObject *nameobj, PyObject *val)
1508 {
1509     GET_ATTR_STRING(name, nameobj);
1510     return ListSetattr((ListObject *)(self), name, val);
1511 }
1512 
1513 /* Function object - Definitions
1514  */
1515 
1516     static PyObject *
1517 FunctionGetattro(PyObject *self, PyObject *nameobj)
1518 {
1519     FunctionObject	*this = (FunctionObject *)(self);
1520 
1521     GET_ATTR_STRING(name, nameobj);
1522 
1523     if (strcmp(name, "name") == 0)
1524 	return PyUnicode_FromString((char *)(this->name));
1525 
1526     return PyObject_GenericGetAttr(self, nameobj);
1527 }
1528 
1529 /* External interface
1530  */
1531 
1532     void
1533 python3_buffer_free(buf_T *buf)
1534 {
1535     if (BUF_PYTHON_REF(buf) != NULL)
1536     {
1537 	BufferObject *bp = BUF_PYTHON_REF(buf);
1538 	bp->buf = INVALID_BUFFER_VALUE;
1539 	BUF_PYTHON_REF(buf) = NULL;
1540     }
1541 }
1542 
1543 #if defined(FEAT_WINDOWS) || defined(PROTO)
1544     void
1545 python3_window_free(win_T *win)
1546 {
1547     if (WIN_PYTHON_REF(win) != NULL)
1548     {
1549 	WindowObject *wp = WIN_PYTHON_REF(win);
1550 	wp->win = INVALID_WINDOW_VALUE;
1551 	WIN_PYTHON_REF(win) = NULL;
1552     }
1553 }
1554 
1555     void
1556 python3_tabpage_free(tabpage_T *tab)
1557 {
1558     if (TAB_PYTHON_REF(tab) != NULL)
1559     {
1560 	TabPageObject *tp = TAB_PYTHON_REF(tab);
1561 	tp->tab = INVALID_TABPAGE_VALUE;
1562 	TAB_PYTHON_REF(tab) = NULL;
1563     }
1564 }
1565 #endif
1566 
1567     static PyObject *
1568 Py3Init_vim(void)
1569 {
1570     /* The special value is removed from sys.path in Python3_Init(). */
1571     static wchar_t *(argv[2]) = {L"/must>not&exist/foo", NULL};
1572 
1573     if (init_types())
1574 	return NULL;
1575 
1576     /* Set sys.argv[] to avoid a crash in warn(). */
1577     PySys_SetArgv(1, argv);
1578 
1579     if ((vim_module = PyModule_Create(&vimmodule)) == NULL)
1580 	return NULL;
1581 
1582     if (populate_module(vim_module))
1583 	return NULL;
1584 
1585     if (init_sys_path())
1586 	return NULL;
1587 
1588     return vim_module;
1589 }
1590 
1591 /*************************************************************************
1592  * 4. Utility functions for handling the interface between Vim and Python.
1593  */
1594 
1595 /* Convert a Vim line into a Python string.
1596  * All internal newlines are replaced by null characters.
1597  *
1598  * On errors, the Python exception data is set, and NULL is returned.
1599  */
1600     static PyObject *
1601 LineToString(const char *str)
1602 {
1603     PyObject *result;
1604     Py_ssize_t len = strlen(str);
1605     char *tmp,*p;
1606 
1607     tmp = (char *)alloc((unsigned)(len+1));
1608     p = tmp;
1609     if (p == NULL)
1610     {
1611 	PyErr_NoMemory();
1612 	return NULL;
1613     }
1614 
1615     while (*str)
1616     {
1617 	if (*str == '\n')
1618 	    *p = '\0';
1619 	else
1620 	    *p = *str;
1621 
1622 	++p;
1623 	++str;
1624     }
1625     *p = '\0';
1626 
1627     result = PyUnicode_Decode(tmp, len, (char *)ENC_OPT, CODEC_ERROR_HANDLER);
1628 
1629     vim_free(tmp);
1630     return result;
1631 }
1632 
1633     void
1634 do_py3eval (char_u *str, typval_T *rettv)
1635 {
1636     DoPyCommand((char *) str,
1637 	    (rangeinitializer) init_range_eval,
1638 	    (runner) run_eval,
1639 	    (void *) rettv);
1640     switch(rettv->v_type)
1641     {
1642 	case VAR_DICT: ++rettv->vval.v_dict->dv_refcount; break;
1643 	case VAR_LIST: ++rettv->vval.v_list->lv_refcount; break;
1644 	case VAR_FUNC: func_ref(rettv->vval.v_string);    break;
1645 	case VAR_UNKNOWN:
1646 	    rettv->v_type = VAR_NUMBER;
1647 	    rettv->vval.v_number = 0;
1648 	    break;
1649     }
1650 }
1651 
1652     void
1653 set_ref_in_python3 (int copyID)
1654 {
1655     set_ref_in_py(copyID);
1656 }
1657