blob: f8e72f5fcc44a62d815ff484b46276c0fb042525 [file] [log] [blame]
Bram Moolenaardb913952012-06-29 12:54:53 +02001/* vi:set ts=8 sts=4 sw=4 noet:
Bram Moolenaar071d4272004-06-13 20:20:40 +00002 *
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#include "vim.h"
21
Bram Moolenaar071d4272004-06-13 20:20:40 +000022#include <limits.h>
23
Bram Moolenaar0014a532013-05-29 21:33:39 +020024/* uncomment this if used with the debug version of python.
25 * Checked on 2.7.4. */
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
Bram Moolenaar071d4272004-06-13 20:20:40 +000033/* Python.h defines _POSIX_THREADS itself (if needed) */
34#ifdef _POSIX_THREADS
35# undef _POSIX_THREADS
36#endif
37
Bram Moolenaar860cae12010-06-05 23:22:07 +020038#if defined(_WIN32) && defined(HAVE_FCNTL_H)
Bram Moolenaar071d4272004-06-13 20:20:40 +000039# undef HAVE_FCNTL_H
40#endif
41
42#ifdef _DEBUG
43# undef _DEBUG
44#endif
45
46#ifdef HAVE_STDARG_H
47# undef HAVE_STDARG_H /* Python's config.h defines it as well. */
48#endif
Bram Moolenaarbe2c9ae2009-11-11 14:06:59 +000049#ifdef _POSIX_C_SOURCE
50# undef _POSIX_C_SOURCE /* pyconfig.h defines it as well. */
51#endif
52#ifdef _XOPEN_SOURCE
53# undef _XOPEN_SOURCE /* pyconfig.h defines it as well. */
54#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +000055
56#include <Python.h>
57#if defined(MACOS) && !defined(MACOS_X_UNIX)
58# include "macglue.h"
59# include <CodeFragments.h>
60#endif
61#undef main /* Defined in python.h - aargh */
62#undef HAVE_FCNTL_H /* Clash with os_win32.h */
63
Bram Moolenaare8cdcef2012-09-12 20:21:43 +020064#if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02050000
65# define PY_SSIZE_T_CLEAN
66#endif
67
Bram Moolenaardb913952012-06-29 12:54:53 +020068#define PyBytes_FromString PyString_FromString
Bram Moolenaar335e0b62013-04-24 13:47:45 +020069#define PyBytes_Check PyString_Check
Bram Moolenaardb913952012-06-29 12:54:53 +020070
Bram Moolenaar19e60942011-06-19 00:27:51 +020071/* No-op conversion functions, use with care! */
72#define PyString_AsBytes(obj) (obj)
73#define PyString_FreeBytes(obj)
74
Bram Moolenaar071d4272004-06-13 20:20:40 +000075#if !defined(FEAT_PYTHON) && defined(PROTO)
76/* Use this to be able to generate prototypes without python being used. */
Bram Moolenaare7cb9cf2008-06-20 14:32:41 +000077# define PyObject Py_ssize_t
78# define PyThreadState Py_ssize_t
79# define PyTypeObject Py_ssize_t
80struct PyMethodDef { Py_ssize_t a; };
81# define PySequenceMethods Py_ssize_t
Bram Moolenaar071d4272004-06-13 20:20:40 +000082#endif
83
Bram Moolenaar2afa3232012-06-29 16:28:28 +020084#if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02070000
85# define PY_USE_CAPSULE
86#endif
87
Bram Moolenaar2c45e942008-06-04 11:35:26 +000088#if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02050000
89# define PyInt Py_ssize_t
90# define PyInquiry lenfunc
91# define PyIntArgFunc ssizeargfunc
92# define PyIntIntArgFunc ssizessizeargfunc
93# define PyIntObjArgProc ssizeobjargproc
94# define PyIntIntObjArgProc ssizessizeobjargproc
Bram Moolenaare7cb9cf2008-06-20 14:32:41 +000095# define Py_ssize_t_fmt "n"
Bram Moolenaar2c45e942008-06-04 11:35:26 +000096#else
97# define PyInt int
Bram Moolenaar4d1da492013-04-24 13:39:15 +020098# define lenfunc inquiry
Bram Moolenaar2c45e942008-06-04 11:35:26 +000099# define PyInquiry inquiry
100# define PyIntArgFunc intargfunc
101# define PyIntIntArgFunc intintargfunc
102# define PyIntObjArgProc intobjargproc
103# define PyIntIntObjArgProc intintobjargproc
Bram Moolenaare7cb9cf2008-06-20 14:32:41 +0000104# define Py_ssize_t_fmt "i"
Bram Moolenaar2c45e942008-06-04 11:35:26 +0000105#endif
106
Bram Moolenaar071d4272004-06-13 20:20:40 +0000107/* Parser flags */
108#define single_input 256
109#define file_input 257
110#define eval_input 258
111
112#if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x020300F0
113 /* Python 2.3: can invoke ":python" recursively. */
114# define PY_CAN_RECURSE
115#endif
116
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200117# if defined(DYNAMIC_PYTHON) || defined(PROTO)
118# ifndef DYNAMIC_PYTHON
119# define HINSTANCE long_u /* for generating prototypes */
120# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000121
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200122# ifndef WIN3264
123# include <dlfcn.h>
124# define FARPROC void*
125# define HINSTANCE void*
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100126# if defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200127# define load_dll(n) dlopen((n), RTLD_LAZY)
128# else
129# define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL)
130# endif
131# define close_dll dlclose
132# define symbol_from_dll dlsym
133# else
Bram Moolenaarebbcb822010-10-23 14:02:54 +0200134# define load_dll vimLoadLib
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200135# define close_dll FreeLibrary
136# define symbol_from_dll GetProcAddress
137# endif
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200138
Bram Moolenaare7cb9cf2008-06-20 14:32:41 +0000139/* This makes if_python.c compile without warnings against Python 2.5
140 * on Win32 and Win64. */
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200141# undef PyRun_SimpleString
Bram Moolenaardb913952012-06-29 12:54:53 +0200142# undef PyRun_String
Bram Moolenaarb61f95c2010-08-09 22:06:13 +0200143# undef PyArg_Parse
144# undef PyArg_ParseTuple
145# undef Py_BuildValue
146# undef Py_InitModule4
147# undef Py_InitModule4_64
Bram Moolenaardb913952012-06-29 12:54:53 +0200148# undef PyObject_CallMethod
Bram Moolenaare7cb9cf2008-06-20 14:32:41 +0000149
Bram Moolenaar071d4272004-06-13 20:20:40 +0000150/*
151 * Wrapper defines
152 */
153# define PyArg_Parse dll_PyArg_Parse
154# define PyArg_ParseTuple dll_PyArg_ParseTuple
Bram Moolenaar19e60942011-06-19 00:27:51 +0200155# define PyMem_Free dll_PyMem_Free
Bram Moolenaardb913952012-06-29 12:54:53 +0200156# define PyMem_Malloc dll_PyMem_Malloc
Bram Moolenaar071d4272004-06-13 20:20:40 +0000157# define PyDict_SetItemString dll_PyDict_SetItemString
158# define PyErr_BadArgument dll_PyErr_BadArgument
Bram Moolenaard5f729c2013-05-15 16:04:40 +0200159# define PyErr_NewException dll_PyErr_NewException
Bram Moolenaar071d4272004-06-13 20:20:40 +0000160# define PyErr_Clear dll_PyErr_Clear
Bram Moolenaar4d369872013-02-20 16:09:43 +0100161# define PyErr_PrintEx dll_PyErr_PrintEx
Bram Moolenaar071d4272004-06-13 20:20:40 +0000162# define PyErr_NoMemory dll_PyErr_NoMemory
163# define PyErr_Occurred dll_PyErr_Occurred
164# define PyErr_SetNone dll_PyErr_SetNone
165# define PyErr_SetString dll_PyErr_SetString
Bram Moolenaar4d188da2013-05-15 15:35:09 +0200166# define PyErr_SetObject dll_PyErr_SetObject
Bram Moolenaar071d4272004-06-13 20:20:40 +0000167# define PyEval_InitThreads dll_PyEval_InitThreads
168# define PyEval_RestoreThread dll_PyEval_RestoreThread
169# define PyEval_SaveThread dll_PyEval_SaveThread
170# ifdef PY_CAN_RECURSE
171# define PyGILState_Ensure dll_PyGILState_Ensure
172# define PyGILState_Release dll_PyGILState_Release
173# endif
174# define PyInt_AsLong dll_PyInt_AsLong
175# define PyInt_FromLong dll_PyInt_FromLong
Bram Moolenaardb913952012-06-29 12:54:53 +0200176# define PyLong_AsLong dll_PyLong_AsLong
177# define PyLong_FromLong dll_PyLong_FromLong
Bram Moolenaar66b79852012-09-21 14:00:35 +0200178# define PyBool_Type (*dll_PyBool_Type)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000179# define PyInt_Type (*dll_PyInt_Type)
Bram Moolenaardb913952012-06-29 12:54:53 +0200180# define PyLong_Type (*dll_PyLong_Type)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000181# define PyList_GetItem dll_PyList_GetItem
Bram Moolenaar0ac93792006-01-21 22:16:51 +0000182# define PyList_Append dll_PyList_Append
Bram Moolenaar071d4272004-06-13 20:20:40 +0000183# define PyList_New dll_PyList_New
184# define PyList_SetItem dll_PyList_SetItem
185# define PyList_Size dll_PyList_Size
186# define PyList_Type (*dll_PyList_Type)
Bram Moolenaardb913952012-06-29 12:54:53 +0200187# define PySequence_Check dll_PySequence_Check
188# define PySequence_Size dll_PySequence_Size
189# define PySequence_GetItem dll_PySequence_GetItem
190# define PyTuple_Size dll_PyTuple_Size
191# define PyTuple_GetItem dll_PyTuple_GetItem
192# define PyTuple_Type (*dll_PyTuple_Type)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000193# define PyImport_ImportModule dll_PyImport_ImportModule
Bram Moolenaar0ac93792006-01-21 22:16:51 +0000194# define PyDict_New dll_PyDict_New
Bram Moolenaar071d4272004-06-13 20:20:40 +0000195# define PyDict_GetItemString dll_PyDict_GetItemString
Bram Moolenaardb913952012-06-29 12:54:53 +0200196# define PyDict_Next dll_PyDict_Next
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200197# define PyDict_Type (*dll_PyDict_Type)
Bram Moolenaardb913952012-06-29 12:54:53 +0200198# ifdef PyMapping_Items
199# define PY_NO_MAPPING_ITEMS
200# else
201# define PyMapping_Items dll_PyMapping_Items
202# endif
203# define PyObject_CallMethod dll_PyObject_CallMethod
204# define PyMapping_Check dll_PyMapping_Check
205# define PyIter_Next dll_PyIter_Next
Bram Moolenaar071d4272004-06-13 20:20:40 +0000206# define PyModule_GetDict dll_PyModule_GetDict
207# define PyRun_SimpleString dll_PyRun_SimpleString
Bram Moolenaardb913952012-06-29 12:54:53 +0200208# define PyRun_String dll_PyRun_String
Bram Moolenaard620aa92013-05-17 16:40:06 +0200209# define PyObject_GetAttrString dll_PyObject_GetAttrString
210# define PyObject_SetAttrString dll_PyObject_SetAttrString
211# define PyObject_CallFunctionObjArgs dll_PyObject_CallFunctionObjArgs
Bram Moolenaar071d4272004-06-13 20:20:40 +0000212# define PyString_AsString dll_PyString_AsString
Bram Moolenaarcdab9052012-09-05 19:03:56 +0200213# define PyString_AsStringAndSize dll_PyString_AsStringAndSize
Bram Moolenaar071d4272004-06-13 20:20:40 +0000214# define PyString_FromString dll_PyString_FromString
215# define PyString_FromStringAndSize dll_PyString_FromStringAndSize
216# define PyString_Size dll_PyString_Size
217# define PyString_Type (*dll_PyString_Type)
Bram Moolenaardb913952012-06-29 12:54:53 +0200218# define PyUnicode_Type (*dll_PyUnicode_Type)
Bram Moolenaarcc3e85f2012-06-29 19:14:52 +0200219# undef PyUnicode_AsEncodedString
220# define PyUnicode_AsEncodedString py_PyUnicode_AsEncodedString
Bram Moolenaardb913952012-06-29 12:54:53 +0200221# define PyFloat_AsDouble dll_PyFloat_AsDouble
222# define PyFloat_FromDouble dll_PyFloat_FromDouble
223# define PyFloat_Type (*dll_PyFloat_Type)
224# define PyImport_AddModule (*dll_PyImport_AddModule)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000225# define PySys_SetObject dll_PySys_SetObject
226# define PySys_SetArgv dll_PySys_SetArgv
227# define PyType_Type (*dll_PyType_Type)
Bram Moolenaar30fec7b2011-03-26 18:32:05 +0100228# define PyType_Ready (*dll_PyType_Ready)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000229# define Py_BuildValue dll_Py_BuildValue
230# define Py_FindMethod dll_Py_FindMethod
231# define Py_InitModule4 dll_Py_InitModule4
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100232# define Py_SetPythonHome dll_Py_SetPythonHome
Bram Moolenaar071d4272004-06-13 20:20:40 +0000233# define Py_Initialize dll_Py_Initialize
Bram Moolenaar0e21a3f2005-04-17 20:28:32 +0000234# define Py_Finalize dll_Py_Finalize
235# define Py_IsInitialized dll_Py_IsInitialized
Bram Moolenaar071d4272004-06-13 20:20:40 +0000236# define _PyObject_New dll__PyObject_New
Bram Moolenaar774267b2013-05-21 20:51:59 +0200237# define _PyObject_GC_New dll__PyObject_GC_New
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200238# ifdef PyObject_GC_Del
239# define Py_underscore_GC
240# define _PyObject_GC_Del dll__PyObject_GC_Del
241# define _PyObject_GC_UnTrack dll__PyObject_GC_UnTrack
242# else
243# define PyObject_GC_Del dll_PyObject_GC_Del
244# define PyObject_GC_UnTrack dll_PyObject_GC_UnTrack
245# endif
Bram Moolenaare7211222012-06-30 13:21:08 +0200246# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02070000
247# define _PyObject_NextNotImplemented (*dll__PyObject_NextNotImplemented)
248# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000249# define _Py_NoneStruct (*dll__Py_NoneStruct)
Bram Moolenaar66b79852012-09-21 14:00:35 +0200250# define _Py_ZeroStruct (*dll__Py_ZeroStruct)
251# define _Py_TrueStruct (*dll__Py_TrueStruct)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000252# define PyObject_Init dll__PyObject_Init
Bram Moolenaardb913952012-06-29 12:54:53 +0200253# define PyObject_GetIter dll_PyObject_GetIter
Bram Moolenaar03db85b2013-05-15 14:51:35 +0200254# define PyObject_IsTrue dll_PyObject_IsTrue
Bram Moolenaar071d4272004-06-13 20:20:40 +0000255# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000
256# define PyType_IsSubtype dll_PyType_IsSubtype
Bram Moolenaar0014a532013-05-29 21:33:39 +0200257# ifdef Py_DEBUG
258# define _Py_NegativeRefcount dll__Py_NegativeRefcount
259# define _Py_RefTotal (*dll__Py_RefTotal)
260# define _Py_Dealloc dll__Py_Dealloc
261# endif
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200262# endif
263# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000
Bram Moolenaar0014a532013-05-29 21:33:39 +0200264# if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC)
265# define _PyObject_DebugMalloc dll__PyObject_DebugMalloc
266# define _PyObject_DebugFree dll__PyObject_DebugFree
267# else
268# define PyObject_Malloc dll_PyObject_Malloc
269# define PyObject_Free dll_PyObject_Free
270# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000271# endif
Bram Moolenaar2afa3232012-06-29 16:28:28 +0200272# ifdef PY_USE_CAPSULE
273# define PyCapsule_New dll_PyCapsule_New
274# define PyCapsule_GetPointer dll_PyCapsule_GetPointer
275# else
276# define PyCObject_FromVoidPtr dll_PyCObject_FromVoidPtr
277# define PyCObject_AsVoidPtr dll_PyCObject_AsVoidPtr
278# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000279
280/*
281 * Pointers for dynamic link
282 */
283static int(*dll_PyArg_Parse)(PyObject *, char *, ...);
284static int(*dll_PyArg_ParseTuple)(PyObject *, char *, ...);
Bram Moolenaar19e60942011-06-19 00:27:51 +0200285static int(*dll_PyMem_Free)(void *);
Bram Moolenaardb913952012-06-29 12:54:53 +0200286static void* (*dll_PyMem_Malloc)(size_t);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000287static int(*dll_PyDict_SetItemString)(PyObject *dp, char *key, PyObject *item);
288static int(*dll_PyErr_BadArgument)(void);
Bram Moolenaard5f729c2013-05-15 16:04:40 +0200289static PyObject *(*dll_PyErr_NewException)(char *, PyObject *, PyObject *);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000290static void(*dll_PyErr_Clear)(void);
Bram Moolenaar4d369872013-02-20 16:09:43 +0100291static void(*dll_PyErr_PrintEx)(int);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000292static PyObject*(*dll_PyErr_NoMemory)(void);
293static PyObject*(*dll_PyErr_Occurred)(void);
294static void(*dll_PyErr_SetNone)(PyObject *);
295static void(*dll_PyErr_SetString)(PyObject *, const char *);
Bram Moolenaar4d188da2013-05-15 15:35:09 +0200296static void(*dll_PyErr_SetObject)(PyObject *, PyObject *);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000297static void(*dll_PyEval_InitThreads)(void);
298static void(*dll_PyEval_RestoreThread)(PyThreadState *);
299static PyThreadState*(*dll_PyEval_SaveThread)(void);
300# ifdef PY_CAN_RECURSE
301static PyGILState_STATE (*dll_PyGILState_Ensure)(void);
302static void (*dll_PyGILState_Release)(PyGILState_STATE);
Bram Moolenaardb913952012-06-29 12:54:53 +0200303# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000304static long(*dll_PyInt_AsLong)(PyObject *);
305static PyObject*(*dll_PyInt_FromLong)(long);
Bram Moolenaardb913952012-06-29 12:54:53 +0200306static long(*dll_PyLong_AsLong)(PyObject *);
307static PyObject*(*dll_PyLong_FromLong)(long);
Bram Moolenaar66b79852012-09-21 14:00:35 +0200308static PyTypeObject* dll_PyBool_Type;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000309static PyTypeObject* dll_PyInt_Type;
Bram Moolenaardb913952012-06-29 12:54:53 +0200310static PyTypeObject* dll_PyLong_Type;
Bram Moolenaar2c45e942008-06-04 11:35:26 +0000311static PyObject*(*dll_PyList_GetItem)(PyObject *, PyInt);
Bram Moolenaar0ac93792006-01-21 22:16:51 +0000312static PyObject*(*dll_PyList_Append)(PyObject *, PyObject *);
Bram Moolenaar2c45e942008-06-04 11:35:26 +0000313static PyObject*(*dll_PyList_New)(PyInt size);
314static int(*dll_PyList_SetItem)(PyObject *, PyInt, PyObject *);
315static PyInt(*dll_PyList_Size)(PyObject *);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000316static PyTypeObject* dll_PyList_Type;
Bram Moolenaardb913952012-06-29 12:54:53 +0200317static int (*dll_PySequence_Check)(PyObject *);
318static PyInt(*dll_PySequence_Size)(PyObject *);
319static PyObject*(*dll_PySequence_GetItem)(PyObject *, PyInt);
320static PyInt(*dll_PyTuple_Size)(PyObject *);
321static PyObject*(*dll_PyTuple_GetItem)(PyObject *, PyInt);
322static PyTypeObject* dll_PyTuple_Type;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000323static PyObject*(*dll_PyImport_ImportModule)(const char *);
Bram Moolenaar0ac93792006-01-21 22:16:51 +0000324static PyObject*(*dll_PyDict_New)(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000325static PyObject*(*dll_PyDict_GetItemString)(PyObject *, const char *);
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200326static int (*dll_PyDict_Next)(PyObject *, PyInt *, PyObject **, PyObject **);
327static PyTypeObject* dll_PyDict_Type;
Bram Moolenaardb913952012-06-29 12:54:53 +0200328# ifndef PY_NO_MAPPING_ITEMS
329static PyObject* (*dll_PyMapping_Items)(PyObject *);
330# endif
331static PyObject* (*dll_PyObject_CallMethod)(PyObject *, char *, PyObject *);
332static int (*dll_PyMapping_Check)(PyObject *);
333static PyObject* (*dll_PyIter_Next)(PyObject *);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000334static PyObject*(*dll_PyModule_GetDict)(PyObject *);
335static int(*dll_PyRun_SimpleString)(char *);
Bram Moolenaardb913952012-06-29 12:54:53 +0200336static PyObject *(*dll_PyRun_String)(char *, int, PyObject *, PyObject *);
Bram Moolenaard620aa92013-05-17 16:40:06 +0200337static PyObject* (*dll_PyObject_GetAttrString)(PyObject *, const char *);
338static PyObject* (*dll_PyObject_SetAttrString)(PyObject *, const char *, PyObject *);
339static PyObject* (*dll_PyObject_CallFunctionObjArgs)(PyObject *, ...);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000340static char*(*dll_PyString_AsString)(PyObject *);
Bram Moolenaarcdab9052012-09-05 19:03:56 +0200341static int(*dll_PyString_AsStringAndSize)(PyObject *, char **, int *);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000342static PyObject*(*dll_PyString_FromString)(const char *);
Bram Moolenaar2c45e942008-06-04 11:35:26 +0000343static PyObject*(*dll_PyString_FromStringAndSize)(const char *, PyInt);
344static PyInt(*dll_PyString_Size)(PyObject *);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000345static PyTypeObject* dll_PyString_Type;
Bram Moolenaardb913952012-06-29 12:54:53 +0200346static PyTypeObject* dll_PyUnicode_Type;
Bram Moolenaarcc3e85f2012-06-29 19:14:52 +0200347static PyObject *(*py_PyUnicode_AsEncodedString)(PyObject *, char *, char *);
Bram Moolenaardb913952012-06-29 12:54:53 +0200348static double(*dll_PyFloat_AsDouble)(PyObject *);
349static PyObject*(*dll_PyFloat_FromDouble)(double);
350static PyTypeObject* dll_PyFloat_Type;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000351static int(*dll_PySys_SetObject)(char *, PyObject *);
352static int(*dll_PySys_SetArgv)(int, char **);
353static PyTypeObject* dll_PyType_Type;
Bram Moolenaar30fec7b2011-03-26 18:32:05 +0100354static int (*dll_PyType_Ready)(PyTypeObject *type);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000355static PyObject*(*dll_Py_BuildValue)(char *, ...);
356static PyObject*(*dll_Py_FindMethod)(struct PyMethodDef[], PyObject *, char *);
357static PyObject*(*dll_Py_InitModule4)(char *, struct PyMethodDef *, char *, PyObject *, int);
Bram Moolenaardb913952012-06-29 12:54:53 +0200358static PyObject*(*dll_PyImport_AddModule)(char *);
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100359static void(*dll_Py_SetPythonHome)(char *home);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000360static void(*dll_Py_Initialize)(void);
Bram Moolenaar0e21a3f2005-04-17 20:28:32 +0000361static void(*dll_Py_Finalize)(void);
362static int(*dll_Py_IsInitialized)(void);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000363static PyObject*(*dll__PyObject_New)(PyTypeObject *, PyObject *);
Bram Moolenaar774267b2013-05-21 20:51:59 +0200364static PyObject*(*dll__PyObject_GC_New)(PyTypeObject *);
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200365# ifdef Py_underscore_GC
366static void(*dll__PyObject_GC_Del)(void *);
367static void(*dll__PyObject_GC_UnTrack)(void *);
368# else
Bram Moolenaar774267b2013-05-21 20:51:59 +0200369static void(*dll_PyObject_GC_Del)(void *);
370static void(*dll_PyObject_GC_UnTrack)(void *);
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200371# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000372static PyObject*(*dll__PyObject_Init)(PyObject *, PyTypeObject *);
Bram Moolenaardb913952012-06-29 12:54:53 +0200373static PyObject* (*dll_PyObject_GetIter)(PyObject *);
Bram Moolenaar03db85b2013-05-15 14:51:35 +0200374static int (*dll_PyObject_IsTrue)(PyObject *);
Bram Moolenaare7211222012-06-30 13:21:08 +0200375# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02070000
Bram Moolenaardb913952012-06-29 12:54:53 +0200376static iternextfunc dll__PyObject_NextNotImplemented;
Bram Moolenaare7211222012-06-30 13:21:08 +0200377# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000378static PyObject* dll__Py_NoneStruct;
Bram Moolenaar66b79852012-09-21 14:00:35 +0200379static PyObject* _Py_ZeroStruct;
380static PyObject* dll__Py_TrueStruct;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000381# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000
382static int (*dll_PyType_IsSubtype)(PyTypeObject *, PyTypeObject *);
Bram Moolenaar0014a532013-05-29 21:33:39 +0200383# ifdef Py_DEBUG
384static void (*dll__Py_NegativeRefcount)(const char *fname, int lineno, PyObject *op);
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200385static PyInt* dll__Py_RefTotal;
Bram Moolenaar0014a532013-05-29 21:33:39 +0200386static void (*dll__Py_Dealloc)(PyObject *obj);
387# endif
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200388# endif
389# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000
Bram Moolenaar0014a532013-05-29 21:33:39 +0200390# if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC)
391static void (*dll__PyObject_DebugFree)(void*);
392static void* (*dll__PyObject_DebugMalloc)(size_t);
393# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000394static void* (*dll_PyObject_Malloc)(size_t);
395static void (*dll_PyObject_Free)(void*);
Bram Moolenaar0014a532013-05-29 21:33:39 +0200396# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000397# endif
Bram Moolenaar2afa3232012-06-29 16:28:28 +0200398# ifdef PY_USE_CAPSULE
Bram Moolenaardb913952012-06-29 12:54:53 +0200399static PyObject* (*dll_PyCapsule_New)(void *, char *, PyCapsule_Destructor);
400static void* (*dll_PyCapsule_GetPointer)(PyObject *, char *);
Bram Moolenaar2afa3232012-06-29 16:28:28 +0200401# else
Bram Moolenaar221d6872012-06-30 13:34:34 +0200402static PyObject* (*dll_PyCObject_FromVoidPtr)(void *cobj, void (*destr)(void *));
403static void* (*dll_PyCObject_AsVoidPtr)(PyObject *);
Bram Moolenaar2afa3232012-06-29 16:28:28 +0200404# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000405
406static HINSTANCE hinstPython = 0; /* Instance of python.dll */
407
408/* Imported exception objects */
409static PyObject *imp_PyExc_AttributeError;
410static PyObject *imp_PyExc_IndexError;
Bram Moolenaaraf6abb92013-04-24 13:04:26 +0200411static PyObject *imp_PyExc_KeyError;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000412static PyObject *imp_PyExc_KeyboardInterrupt;
413static PyObject *imp_PyExc_TypeError;
414static PyObject *imp_PyExc_ValueError;
Bram Moolenaar8661b172013-05-15 15:44:28 +0200415static PyObject *imp_PyExc_RuntimeError;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000416
417# define PyExc_AttributeError imp_PyExc_AttributeError
418# define PyExc_IndexError imp_PyExc_IndexError
Bram Moolenaaraf6abb92013-04-24 13:04:26 +0200419# define PyExc_KeyError imp_PyExc_KeyError
Bram Moolenaar071d4272004-06-13 20:20:40 +0000420# define PyExc_KeyboardInterrupt imp_PyExc_KeyboardInterrupt
421# define PyExc_TypeError imp_PyExc_TypeError
422# define PyExc_ValueError imp_PyExc_ValueError
Bram Moolenaar8661b172013-05-15 15:44:28 +0200423# define PyExc_RuntimeError imp_PyExc_RuntimeError
Bram Moolenaar071d4272004-06-13 20:20:40 +0000424
425/*
426 * Table of name to function pointer of python.
427 */
428# define PYTHON_PROC FARPROC
429static struct
430{
431 char *name;
432 PYTHON_PROC *ptr;
433} python_funcname_table[] =
434{
Bram Moolenaare8cdcef2012-09-12 20:21:43 +0200435#ifndef PY_SSIZE_T_CLEAN
Bram Moolenaar071d4272004-06-13 20:20:40 +0000436 {"PyArg_Parse", (PYTHON_PROC*)&dll_PyArg_Parse},
437 {"PyArg_ParseTuple", (PYTHON_PROC*)&dll_PyArg_ParseTuple},
Bram Moolenaare8cdcef2012-09-12 20:21:43 +0200438 {"Py_BuildValue", (PYTHON_PROC*)&dll_Py_BuildValue},
439#else
440 {"_PyArg_Parse_SizeT", (PYTHON_PROC*)&dll_PyArg_Parse},
441 {"_PyArg_ParseTuple_SizeT", (PYTHON_PROC*)&dll_PyArg_ParseTuple},
442 {"_Py_BuildValue_SizeT", (PYTHON_PROC*)&dll_Py_BuildValue},
443#endif
Bram Moolenaar19e60942011-06-19 00:27:51 +0200444 {"PyMem_Free", (PYTHON_PROC*)&dll_PyMem_Free},
Bram Moolenaardb913952012-06-29 12:54:53 +0200445 {"PyMem_Malloc", (PYTHON_PROC*)&dll_PyMem_Malloc},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000446 {"PyDict_SetItemString", (PYTHON_PROC*)&dll_PyDict_SetItemString},
447 {"PyErr_BadArgument", (PYTHON_PROC*)&dll_PyErr_BadArgument},
Bram Moolenaard5f729c2013-05-15 16:04:40 +0200448 {"PyErr_NewException", (PYTHON_PROC*)&dll_PyErr_NewException},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000449 {"PyErr_Clear", (PYTHON_PROC*)&dll_PyErr_Clear},
Bram Moolenaar4d369872013-02-20 16:09:43 +0100450 {"PyErr_PrintEx", (PYTHON_PROC*)&dll_PyErr_PrintEx},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000451 {"PyErr_NoMemory", (PYTHON_PROC*)&dll_PyErr_NoMemory},
452 {"PyErr_Occurred", (PYTHON_PROC*)&dll_PyErr_Occurred},
453 {"PyErr_SetNone", (PYTHON_PROC*)&dll_PyErr_SetNone},
454 {"PyErr_SetString", (PYTHON_PROC*)&dll_PyErr_SetString},
Bram Moolenaar4d188da2013-05-15 15:35:09 +0200455 {"PyErr_SetObject", (PYTHON_PROC*)&dll_PyErr_SetObject},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000456 {"PyEval_InitThreads", (PYTHON_PROC*)&dll_PyEval_InitThreads},
457 {"PyEval_RestoreThread", (PYTHON_PROC*)&dll_PyEval_RestoreThread},
458 {"PyEval_SaveThread", (PYTHON_PROC*)&dll_PyEval_SaveThread},
459# ifdef PY_CAN_RECURSE
460 {"PyGILState_Ensure", (PYTHON_PROC*)&dll_PyGILState_Ensure},
461 {"PyGILState_Release", (PYTHON_PROC*)&dll_PyGILState_Release},
462# endif
463 {"PyInt_AsLong", (PYTHON_PROC*)&dll_PyInt_AsLong},
464 {"PyInt_FromLong", (PYTHON_PROC*)&dll_PyInt_FromLong},
Bram Moolenaardb913952012-06-29 12:54:53 +0200465 {"PyLong_AsLong", (PYTHON_PROC*)&dll_PyLong_AsLong},
466 {"PyLong_FromLong", (PYTHON_PROC*)&dll_PyLong_FromLong},
Bram Moolenaar66b79852012-09-21 14:00:35 +0200467 {"PyBool_Type", (PYTHON_PROC*)&dll_PyBool_Type},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000468 {"PyInt_Type", (PYTHON_PROC*)&dll_PyInt_Type},
Bram Moolenaardb913952012-06-29 12:54:53 +0200469 {"PyLong_Type", (PYTHON_PROC*)&dll_PyLong_Type},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000470 {"PyList_GetItem", (PYTHON_PROC*)&dll_PyList_GetItem},
Bram Moolenaar0ac93792006-01-21 22:16:51 +0000471 {"PyList_Append", (PYTHON_PROC*)&dll_PyList_Append},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000472 {"PyList_New", (PYTHON_PROC*)&dll_PyList_New},
473 {"PyList_SetItem", (PYTHON_PROC*)&dll_PyList_SetItem},
474 {"PyList_Size", (PYTHON_PROC*)&dll_PyList_Size},
475 {"PyList_Type", (PYTHON_PROC*)&dll_PyList_Type},
Bram Moolenaardb913952012-06-29 12:54:53 +0200476 {"PySequence_GetItem", (PYTHON_PROC*)&dll_PySequence_GetItem},
477 {"PySequence_Size", (PYTHON_PROC*)&dll_PySequence_Size},
478 {"PySequence_Check", (PYTHON_PROC*)&dll_PySequence_Check},
479 {"PyTuple_GetItem", (PYTHON_PROC*)&dll_PyTuple_GetItem},
480 {"PyTuple_Size", (PYTHON_PROC*)&dll_PyTuple_Size},
481 {"PyTuple_Type", (PYTHON_PROC*)&dll_PyTuple_Type},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000482 {"PyImport_ImportModule", (PYTHON_PROC*)&dll_PyImport_ImportModule},
483 {"PyDict_GetItemString", (PYTHON_PROC*)&dll_PyDict_GetItemString},
Bram Moolenaardb913952012-06-29 12:54:53 +0200484 {"PyDict_Next", (PYTHON_PROC*)&dll_PyDict_Next},
Bram Moolenaar0ac93792006-01-21 22:16:51 +0000485 {"PyDict_New", (PYTHON_PROC*)&dll_PyDict_New},
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200486 {"PyDict_Type", (PYTHON_PROC*)&dll_PyDict_Type},
Bram Moolenaardb913952012-06-29 12:54:53 +0200487# ifndef PY_NO_MAPPING_ITEMS
488 {"PyMapping_Items", (PYTHON_PROC*)&dll_PyMapping_Items},
489# endif
490 {"PyObject_CallMethod", (PYTHON_PROC*)&dll_PyObject_CallMethod},
491 {"PyMapping_Check", (PYTHON_PROC*)&dll_PyMapping_Check},
492 {"PyIter_Next", (PYTHON_PROC*)&dll_PyIter_Next},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000493 {"PyModule_GetDict", (PYTHON_PROC*)&dll_PyModule_GetDict},
494 {"PyRun_SimpleString", (PYTHON_PROC*)&dll_PyRun_SimpleString},
Bram Moolenaardb913952012-06-29 12:54:53 +0200495 {"PyRun_String", (PYTHON_PROC*)&dll_PyRun_String},
Bram Moolenaard620aa92013-05-17 16:40:06 +0200496 {"PyObject_GetAttrString", (PYTHON_PROC*)&dll_PyObject_GetAttrString},
497 {"PyObject_SetAttrString", (PYTHON_PROC*)&dll_PyObject_SetAttrString},
498 {"PyObject_CallFunctionObjArgs", (PYTHON_PROC*)&dll_PyObject_CallFunctionObjArgs},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000499 {"PyString_AsString", (PYTHON_PROC*)&dll_PyString_AsString},
Bram Moolenaarcdab9052012-09-05 19:03:56 +0200500 {"PyString_AsStringAndSize", (PYTHON_PROC*)&dll_PyString_AsStringAndSize},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000501 {"PyString_FromString", (PYTHON_PROC*)&dll_PyString_FromString},
502 {"PyString_FromStringAndSize", (PYTHON_PROC*)&dll_PyString_FromStringAndSize},
503 {"PyString_Size", (PYTHON_PROC*)&dll_PyString_Size},
504 {"PyString_Type", (PYTHON_PROC*)&dll_PyString_Type},
Bram Moolenaardb913952012-06-29 12:54:53 +0200505 {"PyUnicode_Type", (PYTHON_PROC*)&dll_PyUnicode_Type},
Bram Moolenaardb913952012-06-29 12:54:53 +0200506 {"PyFloat_Type", (PYTHON_PROC*)&dll_PyFloat_Type},
507 {"PyFloat_AsDouble", (PYTHON_PROC*)&dll_PyFloat_AsDouble},
508 {"PyFloat_FromDouble", (PYTHON_PROC*)&dll_PyFloat_FromDouble},
509 {"PyImport_AddModule", (PYTHON_PROC*)&dll_PyImport_AddModule},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000510 {"PySys_SetObject", (PYTHON_PROC*)&dll_PySys_SetObject},
511 {"PySys_SetArgv", (PYTHON_PROC*)&dll_PySys_SetArgv},
512 {"PyType_Type", (PYTHON_PROC*)&dll_PyType_Type},
Bram Moolenaar30fec7b2011-03-26 18:32:05 +0100513 {"PyType_Ready", (PYTHON_PROC*)&dll_PyType_Ready},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000514 {"Py_FindMethod", (PYTHON_PROC*)&dll_Py_FindMethod},
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100515 {"Py_SetPythonHome", (PYTHON_PROC*)&dll_Py_SetPythonHome},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000516 {"Py_Initialize", (PYTHON_PROC*)&dll_Py_Initialize},
Bram Moolenaar0e21a3f2005-04-17 20:28:32 +0000517 {"Py_Finalize", (PYTHON_PROC*)&dll_Py_Finalize},
518 {"Py_IsInitialized", (PYTHON_PROC*)&dll_Py_IsInitialized},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000519 {"_PyObject_New", (PYTHON_PROC*)&dll__PyObject_New},
Bram Moolenaar774267b2013-05-21 20:51:59 +0200520 {"_PyObject_GC_New", (PYTHON_PROC*)&dll__PyObject_GC_New},
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200521# ifdef Py_underscore_GC
522 {"_PyObject_GC_Del", (PYTHON_PROC*)&dll__PyObject_GC_Del},
523 {"_PyObject_GC_UnTrack", (PYTHON_PROC*)&dll__PyObject_GC_UnTrack},
524# else
Bram Moolenaar774267b2013-05-21 20:51:59 +0200525 {"PyObject_GC_Del", (PYTHON_PROC*)&dll_PyObject_GC_Del},
526 {"PyObject_GC_UnTrack", (PYTHON_PROC*)&dll_PyObject_GC_UnTrack},
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200527# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000528 {"PyObject_Init", (PYTHON_PROC*)&dll__PyObject_Init},
Bram Moolenaardb913952012-06-29 12:54:53 +0200529 {"PyObject_GetIter", (PYTHON_PROC*)&dll_PyObject_GetIter},
Bram Moolenaar03db85b2013-05-15 14:51:35 +0200530 {"PyObject_IsTrue", (PYTHON_PROC*)&dll_PyObject_IsTrue},
Bram Moolenaare7211222012-06-30 13:21:08 +0200531# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02070000
Bram Moolenaardb913952012-06-29 12:54:53 +0200532 {"_PyObject_NextNotImplemented", (PYTHON_PROC*)&dll__PyObject_NextNotImplemented},
Bram Moolenaare7211222012-06-30 13:21:08 +0200533# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000534 {"_Py_NoneStruct", (PYTHON_PROC*)&dll__Py_NoneStruct},
Bram Moolenaar66b79852012-09-21 14:00:35 +0200535 {"_Py_ZeroStruct", (PYTHON_PROC*)&dll__Py_ZeroStruct},
536 {"_Py_TrueStruct", (PYTHON_PROC*)&dll__Py_TrueStruct},
Bram Moolenaar071d4272004-06-13 20:20:40 +0000537# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02020000
Bram Moolenaar0014a532013-05-29 21:33:39 +0200538# ifdef Py_DEBUG
539 {"_Py_NegativeRefcount", (PYTHON_PROC*)&dll__Py_NegativeRefcount},
540 {"_Py_RefTotal", (PYTHON_PROC*)&dll__Py_RefTotal},
541 {"_Py_Dealloc", (PYTHON_PROC*)&dll__Py_Dealloc},
542# endif
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200543 {"PyType_IsSubtype", (PYTHON_PROC*)&dll_PyType_IsSubtype},
544# endif
545# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02030000
Bram Moolenaar0014a532013-05-29 21:33:39 +0200546# if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC)
547 {"_PyObject_DebugFree", (PYTHON_PROC*)&dll__PyObject_DebugFree},
548 {"_PyObject_DebugMalloc", (PYTHON_PROC*)&dll__PyObject_DebugMalloc},
549# else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000550 {"PyObject_Malloc", (PYTHON_PROC*)&dll_PyObject_Malloc},
551 {"PyObject_Free", (PYTHON_PROC*)&dll_PyObject_Free},
Bram Moolenaar0014a532013-05-29 21:33:39 +0200552# endif
553# endif
554# if defined(PY_VERSION_HEX) && PY_VERSION_HEX >= 0x02050000 \
555 && SIZEOF_SIZE_T != SIZEOF_INT
556# ifdef Py_DEBUG
557 {"Py_InitModule4TraceRefs_64", (PYTHON_PROC*)&dll_Py_InitModule4},
558# else
559 {"Py_InitModule4_64", (PYTHON_PROC*)&dll_Py_InitModule4},
560# endif
561# else
562# ifdef Py_DEBUG
563 {"Py_InitModule4TraceRefs", (PYTHON_PROC*)&dll_Py_InitModule4},
564# else
565 {"Py_InitModule4", (PYTHON_PROC*)&dll_Py_InitModule4},
566# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000567# endif
Bram Moolenaar2afa3232012-06-29 16:28:28 +0200568# ifdef PY_USE_CAPSULE
Bram Moolenaardb913952012-06-29 12:54:53 +0200569 {"PyCapsule_New", (PYTHON_PROC*)&dll_PyCapsule_New},
570 {"PyCapsule_GetPointer", (PYTHON_PROC*)&dll_PyCapsule_GetPointer},
Bram Moolenaar2afa3232012-06-29 16:28:28 +0200571# else
572 {"PyCObject_FromVoidPtr", (PYTHON_PROC*)&dll_PyCObject_FromVoidPtr},
573 {"PyCObject_AsVoidPtr", (PYTHON_PROC*)&dll_PyCObject_AsVoidPtr},
574# endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000575 {"", NULL},
576};
577
578/*
579 * Free python.dll
580 */
581 static void
582end_dynamic_python(void)
583{
584 if (hinstPython)
585 {
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200586 close_dll(hinstPython);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000587 hinstPython = 0;
588 }
589}
590
591/*
592 * Load library and get all pointers.
593 * Parameter 'libname' provides name of DLL.
594 * Return OK or FAIL.
595 */
596 static int
597python_runtime_link_init(char *libname, int verbose)
598{
599 int i;
Bram Moolenaarcc3e85f2012-06-29 19:14:52 +0200600 void *ucs_as_encoded_string;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000601
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100602#if !(defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)) && defined(UNIX) && defined(FEAT_PYTHON3)
Bram Moolenaarb744b2f2010-08-13 16:22:57 +0200603 /* Can't have Python and Python3 loaded at the same time.
604 * It cause a crash, because RTLD_GLOBAL is needed for
605 * standard C extension libraries of one or both python versions. */
Bram Moolenaar4c3a3262010-07-24 15:42:14 +0200606 if (python3_loaded())
607 {
Bram Moolenaar9dc93ae2011-08-28 16:00:19 +0200608 if (verbose)
609 EMSG(_("E836: This Vim cannot execute :python after using :py3"));
Bram Moolenaar4c3a3262010-07-24 15:42:14 +0200610 return FAIL;
611 }
612#endif
613
Bram Moolenaar071d4272004-06-13 20:20:40 +0000614 if (hinstPython)
615 return OK;
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200616 hinstPython = load_dll(libname);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000617 if (!hinstPython)
618 {
619 if (verbose)
620 EMSG2(_(e_loadlib), libname);
621 return FAIL;
622 }
623
624 for (i = 0; python_funcname_table[i].ptr; ++i)
625 {
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200626 if ((*python_funcname_table[i].ptr = symbol_from_dll(hinstPython,
Bram Moolenaar071d4272004-06-13 20:20:40 +0000627 python_funcname_table[i].name)) == NULL)
628 {
Bram Moolenaarbd5e15f2010-07-17 21:19:38 +0200629 close_dll(hinstPython);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000630 hinstPython = 0;
631 if (verbose)
632 EMSG2(_(e_loadfunc), python_funcname_table[i].name);
633 return FAIL;
634 }
635 }
Bram Moolenaarcc3e85f2012-06-29 19:14:52 +0200636
637 /* Load unicode functions separately as only the ucs2 or the ucs4 functions
638 * will be present in the library. */
639 ucs_as_encoded_string = symbol_from_dll(hinstPython,
640 "PyUnicodeUCS2_AsEncodedString");
641 if (ucs_as_encoded_string == NULL)
642 ucs_as_encoded_string = symbol_from_dll(hinstPython,
643 "PyUnicodeUCS4_AsEncodedString");
644 if (ucs_as_encoded_string != NULL)
645 py_PyUnicode_AsEncodedString = ucs_as_encoded_string;
646 else
647 {
648 close_dll(hinstPython);
649 hinstPython = 0;
650 if (verbose)
651 EMSG2(_(e_loadfunc), "PyUnicode_UCSX_*");
652 return FAIL;
653 }
654
Bram Moolenaar071d4272004-06-13 20:20:40 +0000655 return OK;
656}
657
658/*
659 * If python is enabled (there is installed python on Windows system) return
660 * TRUE, else FALSE.
661 */
662 int
Bram Moolenaare7cb9cf2008-06-20 14:32:41 +0000663python_enabled(int verbose)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000664{
665 return python_runtime_link_init(DYNAMIC_PYTHON_DLL, verbose) == OK;
666}
667
Bram Moolenaarca8a4df2010-07-31 19:54:14 +0200668/*
669 * Load the standard Python exceptions - don't import the symbols from the
Bram Moolenaar071d4272004-06-13 20:20:40 +0000670 * DLL, as this can cause errors (importing data symbols is not reliable).
671 */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000672 static void
Bram Moolenaarca8a4df2010-07-31 19:54:14 +0200673get_exceptions(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000674{
675 PyObject *exmod = PyImport_ImportModule("exceptions");
676 PyObject *exdict = PyModule_GetDict(exmod);
677 imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError");
678 imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError");
Bram Moolenaaraf6abb92013-04-24 13:04:26 +0200679 imp_PyExc_KeyError = PyDict_GetItemString(exdict, "KeyError");
Bram Moolenaar071d4272004-06-13 20:20:40 +0000680 imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt");
681 imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError");
682 imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError");
Bram Moolenaar8661b172013-05-15 15:44:28 +0200683 imp_PyExc_RuntimeError = PyDict_GetItemString(exdict, "RuntimeError");
Bram Moolenaar071d4272004-06-13 20:20:40 +0000684 Py_XINCREF(imp_PyExc_AttributeError);
685 Py_XINCREF(imp_PyExc_IndexError);
Bram Moolenaaraf6abb92013-04-24 13:04:26 +0200686 Py_XINCREF(imp_PyExc_KeyError);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000687 Py_XINCREF(imp_PyExc_KeyboardInterrupt);
688 Py_XINCREF(imp_PyExc_TypeError);
689 Py_XINCREF(imp_PyExc_ValueError);
Bram Moolenaar8661b172013-05-15 15:44:28 +0200690 Py_XINCREF(imp_PyExc_RuntimeError);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000691 Py_XDECREF(exmod);
692}
693#endif /* DYNAMIC_PYTHON */
694
Bram Moolenaardb913952012-06-29 12:54:53 +0200695static int initialised = 0;
696#define PYINITIALISED initialised
697
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200698#define DESTRUCTOR_FINISH(self) self->ob_type->tp_free((PyObject*)self);
Bram Moolenaar4d1da492013-04-24 13:39:15 +0200699
Bram Moolenaar971db462013-05-12 18:44:48 +0200700#define WIN_PYTHON_REF(win) win->w_python_ref
701#define BUF_PYTHON_REF(buf) buf->b_python_ref
Bram Moolenaar5e538ec2013-05-15 15:12:29 +0200702#define TAB_PYTHON_REF(tab) tab->tp_python_ref
Bram Moolenaar971db462013-05-12 18:44:48 +0200703
Bram Moolenaar4d1da492013-04-24 13:39:15 +0200704static PyObject *OutputGetattr(PyObject *, char *);
705static PyObject *BufferGetattr(PyObject *, char *);
706static PyObject *WindowGetattr(PyObject *, char *);
Bram Moolenaar5e538ec2013-05-15 15:12:29 +0200707static PyObject *TabPageGetattr(PyObject *, char *);
Bram Moolenaar4d1da492013-04-24 13:39:15 +0200708static PyObject *RangeGetattr(PyObject *, char *);
709static PyObject *DictionaryGetattr(PyObject *, char*);
710static PyObject *ListGetattr(PyObject *, char *);
711static PyObject *FunctionGetattr(PyObject *, char *);
712
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200713#ifndef Py_VISIT
714# define Py_VISIT(obj) visit(obj, arg)
715#endif
716#ifndef Py_CLEAR
717# define Py_CLEAR(obj) \
Bram Moolenaar3e734ea2013-05-29 22:05:55 +0200718 { \
719 Py_XDECREF(obj); \
720 obj = NULL; \
721 }
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200722#endif
723
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200724/*
725 * Include the code shared with if_python3.c
726 */
727#include "if_py_both.h"
728
729
Bram Moolenaar071d4272004-06-13 20:20:40 +0000730/******************************************************
731 * Internal function prototypes.
732 */
733
Bram Moolenaar071d4272004-06-13 20:20:40 +0000734static int PythonMod_Init(void);
735
Bram Moolenaar071d4272004-06-13 20:20:40 +0000736
737/******************************************************
738 * 1. Python interpreter main program.
739 */
740
Bram Moolenaar071d4272004-06-13 20:20:40 +0000741#if PYTHON_API_VERSION < 1007 /* Python 1.4 */
742typedef PyObject PyThreadState;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +0000743#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000744
Bram Moolenaar71700b82013-05-15 17:49:05 +0200745#ifndef PY_CAN_RECURSE
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000746static PyThreadState *saved_python_thread = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000747
748/*
749 * Suspend a thread of the Python interpreter, other threads are allowed to
750 * run.
751 */
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000752 static void
753Python_SaveThread(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000754{
755 saved_python_thread = PyEval_SaveThread();
756}
757
758/*
759 * Restore a thread of the Python interpreter, waits for other threads to
760 * block.
761 */
Bram Moolenaar293ee4d2004-12-09 21:34:53 +0000762 static void
763Python_RestoreThread(void)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000764{
765 PyEval_RestoreThread(saved_python_thread);
766 saved_python_thread = NULL;
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +0000767}
Bram Moolenaar71700b82013-05-15 17:49:05 +0200768#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000769
Bram Moolenaar071d4272004-06-13 20:20:40 +0000770 void
771python_end()
772{
Bram Moolenaara5792f52005-11-23 21:25:05 +0000773 static int recurse = 0;
774
775 /* If a crash occurs while doing this, don't try again. */
776 if (recurse != 0)
777 return;
778
779 ++recurse;
780
Bram Moolenaar071d4272004-06-13 20:20:40 +0000781#ifdef DYNAMIC_PYTHON
Bram Moolenaar0e21a3f2005-04-17 20:28:32 +0000782 if (hinstPython && Py_IsInitialized())
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +0000783 {
Bram Moolenaar71700b82013-05-15 17:49:05 +0200784# ifdef PY_CAN_RECURSE
785 PyGILState_Ensure();
786# else
Bram Moolenaarc9b4b052006-04-30 18:54:39 +0000787 Python_RestoreThread(); /* enter python */
Bram Moolenaar71700b82013-05-15 17:49:05 +0200788# endif
Bram Moolenaarc9b4b052006-04-30 18:54:39 +0000789 Py_Finalize();
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +0000790 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000791 end_dynamic_python();
Bram Moolenaar0e21a3f2005-04-17 20:28:32 +0000792#else
793 if (Py_IsInitialized())
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +0000794 {
Bram Moolenaar71700b82013-05-15 17:49:05 +0200795# ifdef PY_CAN_RECURSE
796 PyGILState_Ensure();
797# else
Bram Moolenaarc9b4b052006-04-30 18:54:39 +0000798 Python_RestoreThread(); /* enter python */
Bram Moolenaar71700b82013-05-15 17:49:05 +0200799# endif
Bram Moolenaarc9b4b052006-04-30 18:54:39 +0000800 Py_Finalize();
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +0000801 }
Bram Moolenaar071d4272004-06-13 20:20:40 +0000802#endif
Bram Moolenaara5792f52005-11-23 21:25:05 +0000803
804 --recurse;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000805}
806
Bram Moolenaar4c3a3262010-07-24 15:42:14 +0200807#if (defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON3)) || defined(PROTO)
808 int
809python_loaded()
810{
811 return (hinstPython != 0);
812}
813#endif
814
Bram Moolenaar071d4272004-06-13 20:20:40 +0000815 static int
816Python_Init(void)
817{
818 if (!initialised)
819 {
820#ifdef DYNAMIC_PYTHON
821 if (!python_enabled(TRUE))
822 {
823 EMSG(_("E263: Sorry, this command is disabled, the Python library could not be loaded."));
824 goto fail;
825 }
826#endif
827
Bram Moolenaar644d37b2010-11-16 19:26:02 +0100828#ifdef PYTHON_HOME
829 Py_SetPythonHome(PYTHON_HOME);
830#endif
831
Bram Moolenaar170bf1a2010-07-24 23:51:45 +0200832 init_structs();
833
Bram Moolenaar071d4272004-06-13 20:20:40 +0000834#if !defined(MACOS) || defined(MACOS_X_UNIX)
835 Py_Initialize();
836#else
837 PyMac_Initialize();
838#endif
Bram Moolenaar02366252013-01-30 11:44:39 +0100839 /* Initialise threads, and below save the state using
Bram Moolenaar76d711c2013-02-13 14:17:08 +0100840 * PyEval_SaveThread. Without the call to PyEval_SaveThread, thread
Bram Moolenaar02366252013-01-30 11:44:39 +0100841 * specific state (such as the system trace hook), will be lost
842 * between invocations of Python code. */
Bram Moolenaar071d4272004-06-13 20:20:40 +0000843 PyEval_InitThreads();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000844#ifdef DYNAMIC_PYTHON
845 get_exceptions();
846#endif
847
Bram Moolenaar1dc28782013-05-21 19:11:01 +0200848 if (PythonIO_Init_io())
Bram Moolenaar071d4272004-06-13 20:20:40 +0000849 goto fail;
850
851 if (PythonMod_Init())
852 goto fail;
853
Bram Moolenaardb913952012-06-29 12:54:53 +0200854 globals = PyModule_GetDict(PyImport_AddModule("__main__"));
855
Bram Moolenaar9774ecc2008-11-20 10:04:53 +0000856 /* Remove the element from sys.path that was added because of our
857 * argv[0] value in PythonMod_Init(). Previously we used an empty
Bram Moolenaar84a05ac2013-05-06 04:24:17 +0200858 * string, but depending on the OS we then get an empty entry or
Bram Moolenaar9774ecc2008-11-20 10:04:53 +0000859 * the current directory in sys.path. */
860 PyRun_SimpleString("import sys; sys.path = filter(lambda x: x != '/must>not&exist', sys.path)");
861
Bram Moolenaar76d711c2013-02-13 14:17:08 +0100862 /* lock is created and acquired in PyEval_InitThreads() and thread
863 * state is created in Py_Initialize()
864 * there _PyGILState_NoteThreadState() also sets gilcounter to 1
865 * (python must have threads enabled!)
866 * so the following does both: unlock GIL and save thread state in TLS
867 * without deleting thread state
868 */
Bram Moolenaar03db85b2013-05-15 14:51:35 +0200869#ifndef PY_CAN_RECURSE
870 saved_python_thread =
871#endif
872 PyEval_SaveThread();
Bram Moolenaar071d4272004-06-13 20:20:40 +0000873
874 initialised = 1;
875 }
876
877 return 0;
878
879fail:
880 /* We call PythonIO_Flush() here to print any Python errors.
881 * This is OK, as it is possible to call this function even
Bram Moolenaar1dc28782013-05-21 19:11:01 +0200882 * if PythonIO_Init_io() has not completed successfully (it will
Bram Moolenaar071d4272004-06-13 20:20:40 +0000883 * not do anything in this case).
884 */
885 PythonIO_Flush();
886 return -1;
887}
888
889/*
890 * External interface
891 */
892 static void
Bram Moolenaarb52f4c02013-05-21 18:19:38 +0200893DoPyCommand(const char *cmd, rangeinitializer init_range, runner run, void *arg)
Bram Moolenaar071d4272004-06-13 20:20:40 +0000894{
Bram Moolenaar9ba0eb82005-06-13 22:28:56 +0000895#ifndef PY_CAN_RECURSE
Bram Moolenaar071d4272004-06-13 20:20:40 +0000896 static int recursive = 0;
897#endif
898#if defined(MACOS) && !defined(MACOS_X_UNIX)
899 GrafPtr oldPort;
900#endif
901#if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
902 char *saved_locale;
903#endif
Bram Moolenaar71700b82013-05-15 17:49:05 +0200904#ifdef PY_CAN_RECURSE
905 PyGILState_STATE pygilstate;
906#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000907
908#ifndef PY_CAN_RECURSE
909 if (recursive)
910 {
911 EMSG(_("E659: Cannot invoke Python recursively"));
912 return;
913 }
914 ++recursive;
915#endif
916
917#if defined(MACOS) && !defined(MACOS_X_UNIX)
918 GetPort(&oldPort);
919 /* Check if the Python library is available */
920 if ((Ptr)PyMac_Initialize == (Ptr)kUnresolvedCFragSymbolAddress)
921 goto theend;
922#endif
923 if (Python_Init())
924 goto theend;
925
Bram Moolenaarb52f4c02013-05-21 18:19:38 +0200926 init_range(arg);
927
Bram Moolenaar071d4272004-06-13 20:20:40 +0000928 Python_Release_Vim(); /* leave vim */
929
930#if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
931 /* Python only works properly when the LC_NUMERIC locale is "C". */
932 saved_locale = setlocale(LC_NUMERIC, NULL);
933 if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0)
934 saved_locale = NULL;
935 else
936 {
937 /* Need to make a copy, value may change when setting new locale. */
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200938 saved_locale = (char *) PY_STRSAVE(saved_locale);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000939 (void)setlocale(LC_NUMERIC, "C");
940 }
941#endif
942
Bram Moolenaar71700b82013-05-15 17:49:05 +0200943#ifdef PY_CAN_RECURSE
944 pygilstate = PyGILState_Ensure();
945#else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000946 Python_RestoreThread(); /* enter python */
Bram Moolenaar71700b82013-05-15 17:49:05 +0200947#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000948
Bram Moolenaar2a0f3d32013-05-21 22:23:56 +0200949 run((char *) cmd, arg
950#ifdef PY_CAN_RECURSE
951 , &pygilstate
952#endif
953 );
Bram Moolenaar071d4272004-06-13 20:20:40 +0000954
Bram Moolenaar71700b82013-05-15 17:49:05 +0200955#ifdef PY_CAN_RECURSE
956 PyGILState_Release(pygilstate);
957#else
Bram Moolenaar071d4272004-06-13 20:20:40 +0000958 Python_SaveThread(); /* leave python */
Bram Moolenaar71700b82013-05-15 17:49:05 +0200959#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +0000960
961#if defined(HAVE_LOCALE_H) || defined(X_LOCALE)
962 if (saved_locale != NULL)
963 {
964 (void)setlocale(LC_NUMERIC, saved_locale);
Bram Moolenaare9ba5162013-05-29 22:02:22 +0200965 PyMem_Free(saved_locale);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000966 }
967#endif
968
969 Python_Lock_Vim(); /* enter vim */
970 PythonIO_Flush();
971#if defined(MACOS) && !defined(MACOS_X_UNIX)
972 SetPort(oldPort);
973#endif
974
975theend:
976#ifndef PY_CAN_RECURSE
977 --recursive;
978#endif
Bram Moolenaardb913952012-06-29 12:54:53 +0200979 return;
Bram Moolenaar071d4272004-06-13 20:20:40 +0000980}
981
982/*
983 * ":python"
984 */
985 void
986ex_python(exarg_T *eap)
987{
988 char_u *script;
989
990 script = script_get(eap, eap->arg);
991 if (!eap->skip)
992 {
Bram Moolenaarb52f4c02013-05-21 18:19:38 +0200993 DoPyCommand(script == NULL ? (char *) eap->arg : (char *) script,
994 (rangeinitializer) init_range_cmd,
995 (runner) run_cmd,
996 (void *) eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +0000997 }
998 vim_free(script);
999}
1000
1001#define BUFFER_SIZE 1024
1002
1003/*
1004 * ":pyfile"
1005 */
1006 void
1007ex_pyfile(exarg_T *eap)
1008{
1009 static char buffer[BUFFER_SIZE];
1010 const char *file = (char *)eap->arg;
1011 char *p;
1012
1013 /* Have to do it like this. PyRun_SimpleFile requires you to pass a
1014 * stdio file pointer, but Vim and the Python DLL are compiled with
1015 * different options under Windows, meaning that stdio pointers aren't
1016 * compatible between the two. Yuk.
1017 *
1018 * Put the string "execfile('file')" into buffer. But, we need to
1019 * escape any backslashes or single quotes in the file name, so that
1020 * Python won't mangle the file name.
1021 */
1022 strcpy(buffer, "execfile('");
1023 p = buffer + 10; /* size of "execfile('" */
1024
1025 while (*file && p < buffer + (BUFFER_SIZE - 3))
1026 {
1027 if (*file == '\\' || *file == '\'')
1028 *p++ = '\\';
1029 *p++ = *file++;
1030 }
1031
1032 /* If we didn't finish the file name, we hit a buffer overflow */
1033 if (*file != '\0')
1034 return;
1035
1036 /* Put in the terminating "')" and a null */
1037 *p++ = '\'';
1038 *p++ = ')';
1039 *p++ = '\0';
1040
1041 /* Execute the file */
Bram Moolenaarb52f4c02013-05-21 18:19:38 +02001042 DoPyCommand(buffer,
1043 (rangeinitializer) init_range_cmd,
1044 (runner) run_cmd,
1045 (void *) eap);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001046}
1047
Bram Moolenaard620aa92013-05-17 16:40:06 +02001048 void
1049ex_pydo(exarg_T *eap)
1050{
Bram Moolenaarb52f4c02013-05-21 18:19:38 +02001051 DoPyCommand((char *)eap->arg,
1052 (rangeinitializer) init_range_cmd,
1053 (runner)run_do,
1054 (void *)eap);
Bram Moolenaard620aa92013-05-17 16:40:06 +02001055}
1056
Bram Moolenaar071d4272004-06-13 20:20:40 +00001057/******************************************************
1058 * 2. Python output stream: writes output via [e]msg().
1059 */
1060
1061/* Implementation functions
1062 */
1063
Bram Moolenaar071d4272004-06-13 20:20:40 +00001064 static PyObject *
1065OutputGetattr(PyObject *self, char *name)
1066{
1067 if (strcmp(name, "softspace") == 0)
1068 return PyInt_FromLong(((OutputObject *)(self))->softspace);
1069
1070 return Py_FindMethod(OutputMethods, self, name);
1071}
1072
Bram Moolenaar071d4272004-06-13 20:20:40 +00001073/******************************************************
1074 * 3. Implementation of the Vim module for Python
1075 */
1076
Bram Moolenaar071d4272004-06-13 20:20:40 +00001077/* Window type - Implementation functions
1078 * --------------------------------------
1079 */
1080
Bram Moolenaar071d4272004-06-13 20:20:40 +00001081#define WindowType_Check(obj) ((obj)->ob_type == &WindowType)
1082
Bram Moolenaar071d4272004-06-13 20:20:40 +00001083/* Buffer type - Implementation functions
1084 * --------------------------------------
1085 */
1086
Bram Moolenaar071d4272004-06-13 20:20:40 +00001087#define BufferType_Check(obj) ((obj)->ob_type == &BufferType)
1088
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001089static PyInt BufferAssItem(PyObject *, PyInt, PyObject *);
1090static PyInt BufferAssSlice(PyObject *, PyInt, PyInt, PyObject *);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001091
Bram Moolenaar071d4272004-06-13 20:20:40 +00001092/* Line range type - Implementation functions
1093 * --------------------------------------
1094 */
1095
Bram Moolenaar071d4272004-06-13 20:20:40 +00001096#define RangeType_Check(obj) ((obj)->ob_type == &RangeType)
1097
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001098static PyInt RangeAssItem(PyObject *, PyInt, PyObject *);
1099static PyInt RangeAssSlice(PyObject *, PyInt, PyInt, PyObject *);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001100
Bram Moolenaar071d4272004-06-13 20:20:40 +00001101/* Current objects type - Implementation functions
1102 * -----------------------------------------------
1103 */
1104
Bram Moolenaar071d4272004-06-13 20:20:40 +00001105static PySequenceMethods BufferAsSeq = {
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001106 (PyInquiry) BufferLength, /* sq_length, len(x) */
Bram Moolenaar77fceb82012-09-05 18:54:48 +02001107 (binaryfunc) 0, /* BufferConcat, sq_concat, x+y */
1108 (PyIntArgFunc) 0, /* BufferRepeat, sq_repeat, x*n */
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001109 (PyIntArgFunc) BufferItem, /* sq_item, x[i] */
1110 (PyIntIntArgFunc) BufferSlice, /* sq_slice, x[i:j] */
1111 (PyIntObjArgProc) BufferAssItem, /* sq_ass_item, x[i]=v */
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001112 (PyIntIntObjArgProc) BufferAssSlice, /* sq_ass_slice, x[i:j]=v */
1113 (objobjproc) 0,
1114#if PY_MAJOR_VERSION >= 2
1115 (binaryfunc) 0,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001116 0,
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001117#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001118};
1119
1120/* Buffer object - Implementation
1121 */
1122
1123 static PyObject *
Bram Moolenaar071d4272004-06-13 20:20:40 +00001124BufferGetattr(PyObject *self, char *name)
1125{
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001126 PyObject *r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001127
Bram Moolenaar9e822c02013-05-29 22:15:30 +02001128 if ((r = BufferAttrValid((BufferObject *)(self), name)))
1129 return r;
1130
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001131 if (CheckBuffer((BufferObject *)(self)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001132 return NULL;
1133
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001134 r = BufferAttr((BufferObject *)(self), name);
1135 if (r || PyErr_Occurred())
1136 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001137 else
1138 return Py_FindMethod(BufferMethods, self, name);
1139}
1140
Bram Moolenaar071d4272004-06-13 20:20:40 +00001141/******************/
1142
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001143 static PyInt
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001144BufferAssItem(PyObject *self, PyInt n, PyObject *val)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001145{
Bram Moolenaar8f1723d2013-05-12 20:36:14 +02001146 return RBAsItem((BufferObject *)(self), n, val, 1, -1, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001147}
1148
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001149 static PyInt
1150BufferAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001151{
Bram Moolenaar8f1723d2013-05-12 20:36:14 +02001152 return RBAsSlice((BufferObject *)(self), lo, hi, val, 1, -1, NULL);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001153}
1154
Bram Moolenaar071d4272004-06-13 20:20:40 +00001155static PySequenceMethods RangeAsSeq = {
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001156 (PyInquiry) RangeLength, /* sq_length, len(x) */
1157 (binaryfunc) 0, /* RangeConcat, */ /* sq_concat, x+y */
1158 (PyIntArgFunc) 0, /* RangeRepeat, */ /* sq_repeat, x*n */
1159 (PyIntArgFunc) RangeItem, /* sq_item, x[i] */
1160 (PyIntIntArgFunc) RangeSlice, /* sq_slice, x[i:j] */
1161 (PyIntObjArgProc) RangeAssItem, /* sq_ass_item, x[i]=v */
1162 (PyIntIntObjArgProc) RangeAssSlice, /* sq_ass_slice, x[i:j]=v */
1163 (objobjproc) 0,
1164#if PY_MAJOR_VERSION >= 2
1165 (binaryfunc) 0,
1166 0,
1167#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001168};
1169
Bram Moolenaar071d4272004-06-13 20:20:40 +00001170/* Line range object - Implementation
1171 */
1172
Bram Moolenaar071d4272004-06-13 20:20:40 +00001173 static PyObject *
1174RangeGetattr(PyObject *self, char *name)
1175{
1176 if (strcmp(name, "start") == 0)
Bram Moolenaare7cb9cf2008-06-20 14:32:41 +00001177 return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->start - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001178 else if (strcmp(name, "end") == 0)
Bram Moolenaare7cb9cf2008-06-20 14:32:41 +00001179 return Py_BuildValue(Py_ssize_t_fmt, ((RangeObject *)(self))->end - 1);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001180 else
1181 return Py_FindMethod(RangeMethods, self, name);
1182}
1183
Bram Moolenaar071d4272004-06-13 20:20:40 +00001184/****************/
1185
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001186 static PyInt
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001187RangeAssItem(PyObject *self, PyInt n, PyObject *val)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001188{
Bram Moolenaarca8a4df2010-07-31 19:54:14 +02001189 return RBAsItem(((RangeObject *)(self))->buf, n, val,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001190 ((RangeObject *)(self))->start,
1191 ((RangeObject *)(self))->end,
1192 &((RangeObject *)(self))->end);
1193}
1194
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001195 static PyInt
1196RangeAssSlice(PyObject *self, PyInt lo, PyInt hi, PyObject *val)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001197{
Bram Moolenaar19e60942011-06-19 00:27:51 +02001198 return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val,
Bram Moolenaar071d4272004-06-13 20:20:40 +00001199 ((RangeObject *)(self))->start,
1200 ((RangeObject *)(self))->end,
1201 &((RangeObject *)(self))->end);
1202}
1203
Bram Moolenaar5e538ec2013-05-15 15:12:29 +02001204/* TabPage object - Implementation
1205 */
1206
1207 static PyObject *
1208TabPageGetattr(PyObject *self, char *name)
1209{
1210 PyObject *r;
1211
Bram Moolenaar9e822c02013-05-29 22:15:30 +02001212 if ((r = TabPageAttrValid((TabPageObject *)(self), name)))
1213 return r;
1214
Bram Moolenaar5e538ec2013-05-15 15:12:29 +02001215 if (CheckTabPage((TabPageObject *)(self)))
1216 return NULL;
1217
1218 r = TabPageAttr((TabPageObject *)(self), name);
1219 if (r || PyErr_Occurred())
1220 return r;
1221 else
1222 return Py_FindMethod(TabPageMethods, self, name);
1223}
1224
Bram Moolenaar071d4272004-06-13 20:20:40 +00001225/* Window object - Implementation
1226 */
1227
1228 static PyObject *
Bram Moolenaar071d4272004-06-13 20:20:40 +00001229WindowGetattr(PyObject *self, char *name)
1230{
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001231 PyObject *r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001232
Bram Moolenaar9e822c02013-05-29 22:15:30 +02001233 if ((r = WindowAttrValid((WindowObject *)(self), name)))
1234 return r;
1235
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001236 if (CheckWindow((WindowObject *)(self)))
Bram Moolenaar071d4272004-06-13 20:20:40 +00001237 return NULL;
1238
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001239 r = WindowAttr((WindowObject *)(self), name);
1240 if (r || PyErr_Occurred())
1241 return r;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001242 else
1243 return Py_FindMethod(WindowMethods, self, name);
1244}
1245
Bram Moolenaar5e538ec2013-05-15 15:12:29 +02001246/* Tab page list object - Definitions
1247 */
1248
1249static PySequenceMethods TabListAsSeq = {
1250 (PyInquiry) TabListLength, /* sq_length, len(x) */
1251 (binaryfunc) 0, /* sq_concat, x+y */
1252 (PyIntArgFunc) 0, /* sq_repeat, x*n */
1253 (PyIntArgFunc) TabListItem, /* sq_item, x[i] */
1254 (PyIntIntArgFunc) 0, /* sq_slice, x[i:j] */
1255 (PyIntObjArgProc) 0, /* sq_ass_item, x[i]=v */
1256 (PyIntIntObjArgProc) 0, /* sq_ass_slice, x[i:j]=v */
1257 (objobjproc) 0,
1258#if PY_MAJOR_VERSION >= 2
1259 (binaryfunc) 0,
1260 0,
1261#endif
1262};
1263
Bram Moolenaar071d4272004-06-13 20:20:40 +00001264/* Window list object - Definitions
1265 */
1266
Bram Moolenaar071d4272004-06-13 20:20:40 +00001267static PySequenceMethods WinListAsSeq = {
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001268 (PyInquiry) WinListLength, /* sq_length, len(x) */
Bram Moolenaar071d4272004-06-13 20:20:40 +00001269 (binaryfunc) 0, /* sq_concat, x+y */
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001270 (PyIntArgFunc) 0, /* sq_repeat, x*n */
1271 (PyIntArgFunc) WinListItem, /* sq_item, x[i] */
1272 (PyIntIntArgFunc) 0, /* sq_slice, x[i:j] */
1273 (PyIntObjArgProc) 0, /* sq_ass_item, x[i]=v */
Bram Moolenaar4d1da492013-04-24 13:39:15 +02001274 (PyIntIntObjArgProc) 0, /* sq_ass_slice, x[i:j]=v */
1275 (objobjproc) 0,
1276#if PY_MAJOR_VERSION >= 2
1277 (binaryfunc) 0,
1278 0,
1279#endif
Bram Moolenaar071d4272004-06-13 20:20:40 +00001280};
1281
Bram Moolenaar071d4272004-06-13 20:20:40 +00001282/* External interface
1283 */
1284
1285 void
1286python_buffer_free(buf_T *buf)
1287{
Bram Moolenaar971db462013-05-12 18:44:48 +02001288 if (BUF_PYTHON_REF(buf) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001289 {
Bram Moolenaar971db462013-05-12 18:44:48 +02001290 BufferObject *bp = BUF_PYTHON_REF(buf);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001291 bp->buf = INVALID_BUFFER_VALUE;
Bram Moolenaar971db462013-05-12 18:44:48 +02001292 BUF_PYTHON_REF(buf) = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001293 }
1294}
1295
1296#if defined(FEAT_WINDOWS) || defined(PROTO)
1297 void
1298python_window_free(win_T *win)
1299{
Bram Moolenaar971db462013-05-12 18:44:48 +02001300 if (WIN_PYTHON_REF(win) != NULL)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001301 {
Bram Moolenaar971db462013-05-12 18:44:48 +02001302 WindowObject *wp = WIN_PYTHON_REF(win);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001303 wp->win = INVALID_WINDOW_VALUE;
Bram Moolenaar971db462013-05-12 18:44:48 +02001304 WIN_PYTHON_REF(win) = NULL;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001305 }
1306}
Bram Moolenaar5e538ec2013-05-15 15:12:29 +02001307
1308 void
1309python_tabpage_free(tabpage_T *tab)
1310{
1311 if (TAB_PYTHON_REF(tab) != NULL)
1312 {
1313 TabPageObject *tp = TAB_PYTHON_REF(tab);
1314 tp->tab = INVALID_TABPAGE_VALUE;
1315 TAB_PYTHON_REF(tab) = NULL;
1316 }
1317}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001318#endif
1319
Bram Moolenaar1dc28782013-05-21 19:11:01 +02001320 static int
1321add_object(PyObject *dict, const char *name, PyObject *object)
Bram Moolenaar071d4272004-06-13 20:20:40 +00001322{
Bram Moolenaar1dc28782013-05-21 19:11:01 +02001323 if (PyDict_SetItemString(dict, (char *) name, object))
1324 return -1;
1325 Py_DECREF(object);
1326 return 0;
1327}
Bram Moolenaar5e538ec2013-05-15 15:12:29 +02001328
Bram Moolenaar071d4272004-06-13 20:20:40 +00001329 static int
1330PythonMod_Init(void)
1331{
1332 PyObject *mod;
1333 PyObject *dict;
Bram Moolenaar1dc28782013-05-21 19:11:01 +02001334
Bram Moolenaar9774ecc2008-11-20 10:04:53 +00001335 /* The special value is removed from sys.path in Python_Init(). */
1336 static char *(argv[2]) = {"/must>not&exist/foo", NULL};
Bram Moolenaar071d4272004-06-13 20:20:40 +00001337
Bram Moolenaar1dc28782013-05-21 19:11:01 +02001338 if (init_types())
1339 return -1;
Bram Moolenaar071d4272004-06-13 20:20:40 +00001340
1341 /* Set sys.argv[] to avoid a crash in warn(). */
1342 PySys_SetArgv(1, argv);
1343
Bram Moolenaare7cb9cf2008-06-20 14:32:41 +00001344 mod = Py_InitModule4("vim", VimMethods, (char *)NULL, (PyObject *)NULL, PYTHON_API_VERSION);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001345 dict = PyModule_GetDict(mod);
1346
Bram Moolenaar1dc28782013-05-21 19:11:01 +02001347 return populate_module(dict, add_object);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001348}
1349
1350/*************************************************************************
1351 * 4. Utility functions for handling the interface between Vim and Python.
1352 */
1353
Bram Moolenaar071d4272004-06-13 20:20:40 +00001354/* Convert a Vim line into a Python string.
1355 * All internal newlines are replaced by null characters.
1356 *
1357 * On errors, the Python exception data is set, and NULL is returned.
1358 */
1359 static PyObject *
1360LineToString(const char *str)
1361{
1362 PyObject *result;
Bram Moolenaar2c45e942008-06-04 11:35:26 +00001363 PyInt len = strlen(str);
Bram Moolenaar071d4272004-06-13 20:20:40 +00001364 char *p;
1365
1366 /* Allocate an Python string object, with uninitialised contents. We
1367 * must do it this way, so that we can modify the string in place
1368 * later. See the Python source, Objects/stringobject.c for details.
1369 */
1370 result = PyString_FromStringAndSize(NULL, len);
1371 if (result == NULL)
1372 return NULL;
1373
1374 p = PyString_AsString(result);
1375
1376 while (*str)
1377 {
1378 if (*str == '\n')
1379 *p = '\0';
1380 else
1381 *p = *str;
1382
1383 ++p;
1384 ++str;
1385 }
1386
1387 return result;
1388}
1389
Bram Moolenaardb913952012-06-29 12:54:53 +02001390 static PyObject *
1391DictionaryGetattr(PyObject *self, char *name)
1392{
Bram Moolenaar66b79852012-09-21 14:00:35 +02001393 DictionaryObject *this = ((DictionaryObject *) (self));
1394
1395 if (strcmp(name, "locked") == 0)
1396 return PyInt_FromLong(this->dict->dv_lock);
1397 else if (strcmp(name, "scope") == 0)
1398 return PyInt_FromLong(this->dict->dv_scope);
1399
Bram Moolenaardb913952012-06-29 12:54:53 +02001400 return Py_FindMethod(DictionaryMethods, self, name);
1401}
1402
Bram Moolenaardb913952012-06-29 12:54:53 +02001403static PySequenceMethods ListAsSeq = {
1404 (PyInquiry) ListLength,
1405 (binaryfunc) 0,
1406 (PyIntArgFunc) 0,
1407 (PyIntArgFunc) ListItem,
1408 (PyIntIntArgFunc) ListSlice,
1409 (PyIntObjArgProc) ListAssItem,
1410 (PyIntIntObjArgProc) ListAssSlice,
1411 (objobjproc) 0,
1412#if PY_MAJOR_VERSION >= 2
1413 (binaryfunc) ListConcatInPlace,
1414 0,
1415#endif
1416};
1417
Bram Moolenaardb913952012-06-29 12:54:53 +02001418 static PyObject *
1419ListGetattr(PyObject *self, char *name)
1420{
Bram Moolenaar66b79852012-09-21 14:00:35 +02001421 if (strcmp(name, "locked") == 0)
1422 return PyInt_FromLong(((ListObject *)(self))->list->lv_lock);
1423
Bram Moolenaardb913952012-06-29 12:54:53 +02001424 return Py_FindMethod(ListMethods, self, name);
1425}
1426
Bram Moolenaardb913952012-06-29 12:54:53 +02001427 static PyObject *
1428FunctionGetattr(PyObject *self, char *name)
1429{
1430 FunctionObject *this = (FunctionObject *)(self);
1431
1432 if (strcmp(name, "name") == 0)
1433 return PyString_FromString((char *)(this->name));
1434 else
1435 return Py_FindMethod(FunctionMethods, self, name);
1436}
1437
1438 void
1439do_pyeval (char_u *str, typval_T *rettv)
1440{
Bram Moolenaarb52f4c02013-05-21 18:19:38 +02001441 DoPyCommand((char *) str,
1442 (rangeinitializer) init_range_eval,
1443 (runner) run_eval,
1444 (void *) rettv);
Bram Moolenaardb913952012-06-29 12:54:53 +02001445 switch(rettv->v_type)
1446 {
1447 case VAR_DICT: ++rettv->vval.v_dict->dv_refcount; break;
1448 case VAR_LIST: ++rettv->vval.v_list->lv_refcount; break;
1449 case VAR_FUNC: func_ref(rettv->vval.v_string); break;
Bram Moolenaar77fceb82012-09-05 18:54:48 +02001450 case VAR_UNKNOWN:
1451 rettv->v_type = VAR_NUMBER;
1452 rettv->vval.v_number = 0;
1453 break;
Bram Moolenaardb913952012-06-29 12:54:53 +02001454 }
1455}
Bram Moolenaar071d4272004-06-13 20:20:40 +00001456
1457/* Don't generate a prototype for the next function, it generates an error on
1458 * newer Python versions. */
1459#if PYTHON_API_VERSION < 1007 /* Python 1.4 */ && !defined(PROTO)
1460
1461 char *
1462Py_GetProgramName(void)
1463{
1464 return "vim";
1465}
1466#endif /* Python 1.4 */
Bram Moolenaar170bf1a2010-07-24 23:51:45 +02001467
Bram Moolenaardb913952012-06-29 12:54:53 +02001468 void
1469set_ref_in_python (int copyID)
1470{
1471 set_ref_in_py(copyID);
1472}