1edf3f97aSBram Moolenaar /* vi:set ts=8 sts=4 sw=4 noet: 2bd5e15fdSBram Moolenaar * 3bd5e15fdSBram Moolenaar * VIM - Vi IMproved by Bram Moolenaar 4bd5e15fdSBram Moolenaar * 5bd5e15fdSBram Moolenaar * Do ":help uganda" in Vim to read copying and usage conditions. 6bd5e15fdSBram Moolenaar * Do ":help credits" in Vim to see a list of people who contributed. 7bd5e15fdSBram Moolenaar * See README.txt for an overview of the Vim source code. 8bd5e15fdSBram Moolenaar */ 9bd5e15fdSBram Moolenaar /* 10bd5e15fdSBram Moolenaar * Python extensions by Paul Moore. 11bd5e15fdSBram Moolenaar * Changes for Unix by David Leonard. 12bd5e15fdSBram Moolenaar * 13bd5e15fdSBram Moolenaar * This consists of four parts: 14bd5e15fdSBram Moolenaar * 1. Python interpreter main program 15bd5e15fdSBram Moolenaar * 2. Python output stream: writes output via [e]msg(). 16bd5e15fdSBram Moolenaar * 3. Implementation of the Vim module for Python 17bd5e15fdSBram Moolenaar * 4. Utility functions for handling the interface between Vim and Python. 18bd5e15fdSBram Moolenaar */ 19bd5e15fdSBram Moolenaar 20bd5e15fdSBram Moolenaar /* 21bd5e15fdSBram Moolenaar * Roland Puntaier 2009/sept/16: 22bd5e15fdSBram Moolenaar * Adaptations to support both python3.x and python2.x 23bd5e15fdSBram Moolenaar */ 24bd5e15fdSBram Moolenaar 252ab2e860SBram Moolenaar // uncomment this if used with the debug version of python 262ab2e860SBram Moolenaar // #define Py_DEBUG 272ab2e860SBram Moolenaar // Note: most of time you can add -DPy_DEBUG to CFLAGS in place of uncommenting 282ab2e860SBram Moolenaar // uncomment this if used with the debug version of python, but without its 292ab2e860SBram Moolenaar // allocator 302ab2e860SBram Moolenaar // #define Py_DEBUG_NO_PYMALLOC 31bd5e15fdSBram Moolenaar 32bd5e15fdSBram Moolenaar #include "vim.h" 33bd5e15fdSBram Moolenaar 34bd5e15fdSBram Moolenaar #include <limits.h> 35bd5e15fdSBram Moolenaar 364f97475dSBram Moolenaar #if defined(MSWIN) && defined(HAVE_FCNTL_H) 37bd5e15fdSBram Moolenaar # undef HAVE_FCNTL_H 38bd5e15fdSBram Moolenaar #endif 39bd5e15fdSBram Moolenaar 40bd5e15fdSBram Moolenaar #ifdef _DEBUG 41bd5e15fdSBram Moolenaar # undef _DEBUG 42bd5e15fdSBram Moolenaar #endif 43bd5e15fdSBram Moolenaar 44bd5e15fdSBram Moolenaar #ifdef F_BLANK 45bd5e15fdSBram Moolenaar # undef F_BLANK 46bd5e15fdSBram Moolenaar #endif 47bd5e15fdSBram Moolenaar 486aa2cd4bSBram Moolenaar #ifdef HAVE_STRFTIME 496aa2cd4bSBram Moolenaar # undef HAVE_STRFTIME 506aa2cd4bSBram Moolenaar #endif 516aa2cd4bSBram Moolenaar #ifdef HAVE_STRING_H 526aa2cd4bSBram Moolenaar # undef HAVE_STRING_H 536aa2cd4bSBram Moolenaar #endif 546aa2cd4bSBram Moolenaar #ifdef HAVE_PUTENV 556aa2cd4bSBram Moolenaar # undef HAVE_PUTENV 566aa2cd4bSBram Moolenaar #endif 576df6f47dSBram Moolenaar #ifdef HAVE_STDARG_H 582ab2e860SBram Moolenaar # undef HAVE_STDARG_H // Python's config.h defines it as well. 596df6f47dSBram Moolenaar #endif 602ab2e860SBram Moolenaar #ifdef _POSIX_C_SOURCE // defined in feature.h 61bd5e15fdSBram Moolenaar # undef _POSIX_C_SOURCE 62bd5e15fdSBram Moolenaar #endif 636df6f47dSBram Moolenaar #ifdef _XOPEN_SOURCE 642ab2e860SBram Moolenaar # undef _XOPEN_SOURCE // pyconfig.h defines it as well. 656df6f47dSBram Moolenaar #endif 66bd5e15fdSBram Moolenaar 670bdda37fSBram Moolenaar #define PY_SSIZE_T_CLEAN 680bdda37fSBram Moolenaar 69bd5e15fdSBram Moolenaar #include <Python.h> 700bdda37fSBram Moolenaar 712ab2e860SBram Moolenaar #undef main // Defined in python.h - aargh 722ab2e860SBram Moolenaar #undef HAVE_FCNTL_H // Clash with os_win32.h 73bd5e15fdSBram Moolenaar 742ab2e860SBram Moolenaar // The "surrogateescape" error handler is new in Python 3.1 753d64a317SBram Moolenaar #if PY_VERSION_HEX >= 0x030100f0 763d64a317SBram Moolenaar # define CODEC_ERROR_HANDLER "surrogateescape" 773d64a317SBram Moolenaar #else 783d64a317SBram Moolenaar # define CODEC_ERROR_HANDLER NULL 793d64a317SBram Moolenaar #endif 803d64a317SBram Moolenaar 812ab2e860SBram Moolenaar // Python 3 does not support CObjects, always use Capsules 822afa3238SBram Moolenaar #define PY_USE_CAPSULE 832afa3238SBram Moolenaar 842e2f52a4SBram Moolenaar #define ERRORS_DECODE_ARG CODEC_ERROR_HANDLER 852e2f52a4SBram Moolenaar #define ERRORS_ENCODE_ARG ERRORS_DECODE_ARG 862e2f52a4SBram Moolenaar 87170bf1aeSBram Moolenaar #define PyInt Py_ssize_t 8832ac8cdaSBram Moolenaar #ifndef PyString_Check 89ca8a4dfeSBram Moolenaar # define PyString_Check(obj) PyUnicode_Check(obj) 9032ac8cdaSBram Moolenaar #endif 91e032461eSBram Moolenaar #define PyString_FromString(repr) \ 922e2f52a4SBram Moolenaar PyUnicode_Decode(repr, STRLEN(repr), ENC_OPT, ERRORS_DECODE_ARG) 931a3b5695SBram Moolenaar #define PyString_FromFormat PyUnicode_FromFormat 9432ac8cdaSBram Moolenaar #ifndef PyInt_Check 9577045658SBram Moolenaar # define PyInt_Check(obj) PyLong_Check(obj) 9632ac8cdaSBram Moolenaar #endif 9777045658SBram Moolenaar #define PyInt_FromLong(i) PyLong_FromLong(i) 9877045658SBram Moolenaar #define PyInt_AsLong(obj) PyLong_AsLong(obj) 994d1da49cSBram Moolenaar #define Py_ssize_t_fmt "n" 100a9922d62SBram Moolenaar #define Py_bytes_fmt "y" 101170bf1aeSBram Moolenaar 102063a46baSBram Moolenaar #define PyIntArgFunc ssizeargfunc 103063a46baSBram Moolenaar #define PyIntObjArgProc ssizeobjargproc 104063a46baSBram Moolenaar 105922a4664SBram Moolenaar /* 106922a4664SBram Moolenaar * PySlice_GetIndicesEx(): first argument type changed from PySliceObject 107922a4664SBram Moolenaar * to PyObject in Python 3.2 or later. 108922a4664SBram Moolenaar */ 109922a4664SBram Moolenaar #if PY_VERSION_HEX >= 0x030200f0 110922a4664SBram Moolenaar typedef PyObject PySliceObject_T; 111922a4664SBram Moolenaar #else 112922a4664SBram Moolenaar typedef PySliceObject PySliceObject_T; 113922a4664SBram Moolenaar #endif 114922a4664SBram Moolenaar 1150c1f3f4dSBram Moolenaar #if defined(DYNAMIC_PYTHON3) || defined(PROTO) 116bd5e15fdSBram Moolenaar 1174f97475dSBram Moolenaar # ifndef MSWIN 118bd5e15fdSBram Moolenaar # include <dlfcn.h> 119bd5e15fdSBram Moolenaar # define FARPROC void* 120bd5e15fdSBram Moolenaar # define HINSTANCE void* 121644d37b8SBram Moolenaar # if defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL) 122b61f95c3SBram Moolenaar # define load_dll(n) dlopen((n), RTLD_LAZY) 123b61f95c3SBram Moolenaar # else 124fa5d1e63SBram Moolenaar # define load_dll(n) dlopen((n), RTLD_LAZY|RTLD_GLOBAL) 125b61f95c3SBram Moolenaar # endif 126bd5e15fdSBram Moolenaar # define close_dll dlclose 127bd5e15fdSBram Moolenaar # define symbol_from_dll dlsym 128bd5e15fdSBram Moolenaar # else 129ebbcb824SBram Moolenaar # define load_dll vimLoadLib 130bd5e15fdSBram Moolenaar # define close_dll FreeLibrary 131bd5e15fdSBram Moolenaar # define symbol_from_dll GetProcAddress 132bd5e15fdSBram Moolenaar # endif 133bd5e15fdSBram Moolenaar /* 134bd5e15fdSBram Moolenaar * Wrapper defines 135bd5e15fdSBram Moolenaar */ 136bd5e15fdSBram Moolenaar # undef PyArg_Parse 137bd5e15fdSBram Moolenaar # define PyArg_Parse py3_PyArg_Parse 138bd5e15fdSBram Moolenaar # undef PyArg_ParseTuple 139bd5e15fdSBram Moolenaar # define PyArg_ParseTuple py3_PyArg_ParseTuple 14019e60943SBram Moolenaar # define PyMem_Free py3_PyMem_Free 141db913953SBram Moolenaar # define PyMem_Malloc py3_PyMem_Malloc 142bd5e15fdSBram Moolenaar # define PyDict_SetItemString py3_PyDict_SetItemString 143bd5e15fdSBram Moolenaar # define PyErr_BadArgument py3_PyErr_BadArgument 144bd5e15fdSBram Moolenaar # define PyErr_Clear py3_PyErr_Clear 145c476e52fSBram Moolenaar # define PyErr_Format py3_PyErr_Format 1464d36987cSBram Moolenaar # define PyErr_PrintEx py3_PyErr_PrintEx 147bd5e15fdSBram Moolenaar # define PyErr_NoMemory py3_PyErr_NoMemory 148bd5e15fdSBram Moolenaar # define PyErr_Occurred py3_PyErr_Occurred 149bd5e15fdSBram Moolenaar # define PyErr_SetNone py3_PyErr_SetNone 150bd5e15fdSBram Moolenaar # define PyErr_SetString py3_PyErr_SetString 1514d188da2SBram Moolenaar # define PyErr_SetObject py3_PyErr_SetObject 152c09a6d6cSBram Moolenaar # define PyErr_ExceptionMatches py3_PyErr_ExceptionMatches 153bd5e15fdSBram Moolenaar # define PyEval_InitThreads py3_PyEval_InitThreads 154bd5e15fdSBram Moolenaar # define PyEval_RestoreThread py3_PyEval_RestoreThread 155bd5e15fdSBram Moolenaar # define PyEval_SaveThread py3_PyEval_SaveThread 156bd5e15fdSBram Moolenaar # define PyGILState_Ensure py3_PyGILState_Ensure 157bd5e15fdSBram Moolenaar # define PyGILState_Release py3_PyGILState_Release 158bd5e15fdSBram Moolenaar # define PyLong_AsLong py3_PyLong_AsLong 159bd5e15fdSBram Moolenaar # define PyLong_FromLong py3_PyLong_FromLong 160bd5e15fdSBram Moolenaar # define PyList_GetItem py3_PyList_GetItem 161bd5e15fdSBram Moolenaar # define PyList_Append py3_PyList_Append 162c09a6d6cSBram Moolenaar # define PyList_Insert py3_PyList_Insert 163bd5e15fdSBram Moolenaar # define PyList_New py3_PyList_New 164bd5e15fdSBram Moolenaar # define PyList_SetItem py3_PyList_SetItem 165bd5e15fdSBram Moolenaar # define PyList_Size py3_PyList_Size 166db913953SBram Moolenaar # define PySequence_Check py3_PySequence_Check 167db913953SBram Moolenaar # define PySequence_Size py3_PySequence_Size 168db913953SBram Moolenaar # define PySequence_GetItem py3_PySequence_GetItem 169a9922d62SBram Moolenaar # define PySequence_Fast py3_PySequence_Fast 170db913953SBram Moolenaar # define PyTuple_Size py3_PyTuple_Size 171db913953SBram Moolenaar # define PyTuple_GetItem py3_PyTuple_GetItem 1723b48b11cSBram Moolenaar # if PY_VERSION_HEX >= 0x030601f0 1733b48b11cSBram Moolenaar # define PySlice_AdjustIndices py3_PySlice_AdjustIndices 1743b48b11cSBram Moolenaar # define PySlice_Unpack py3_PySlice_Unpack 1753b48b11cSBram Moolenaar # endif 1763b48b11cSBram Moolenaar # undef PySlice_GetIndicesEx 177bd5e15fdSBram Moolenaar # define PySlice_GetIndicesEx py3_PySlice_GetIndicesEx 178bd5e15fdSBram Moolenaar # define PyImport_ImportModule py3_PyImport_ImportModule 179bd5e15fdSBram Moolenaar # define PyObject_Init py3__PyObject_Init 180bd5e15fdSBram Moolenaar # define PyDict_New py3_PyDict_New 181bd5e15fdSBram Moolenaar # define PyDict_GetItemString py3_PyDict_GetItemString 182db913953SBram Moolenaar # define PyDict_Next py3_PyDict_Next 183db913953SBram Moolenaar # define PyMapping_Check py3_PyMapping_Check 18432ac8cdaSBram Moolenaar # ifndef PyMapping_Keys 185bcb40977SBram Moolenaar # define PyMapping_Keys py3_PyMapping_Keys 18632ac8cdaSBram Moolenaar # endif 187db913953SBram Moolenaar # define PyIter_Next py3_PyIter_Next 188db913953SBram Moolenaar # define PyObject_GetIter py3_PyObject_GetIter 189141be8a5SBram Moolenaar # define PyObject_Repr py3_PyObject_Repr 190bcb40977SBram Moolenaar # define PyObject_GetItem py3_PyObject_GetItem 19103db85b3SBram Moolenaar # define PyObject_IsTrue py3_PyObject_IsTrue 192bd5e15fdSBram Moolenaar # define PyModule_GetDict py3_PyModule_GetDict 193bd5e15fdSBram Moolenaar #undef PyRun_SimpleString 194bd5e15fdSBram Moolenaar # define PyRun_SimpleString py3_PyRun_SimpleString 195db913953SBram Moolenaar #undef PyRun_String 196db913953SBram Moolenaar # define PyRun_String py3_PyRun_String 1973dab2806SBram Moolenaar # define PyObject_GetAttrString py3_PyObject_GetAttrString 198a9922d62SBram Moolenaar # define PyObject_HasAttrString py3_PyObject_HasAttrString 1993dab2806SBram Moolenaar # define PyObject_SetAttrString py3_PyObject_SetAttrString 2003dab2806SBram Moolenaar # define PyObject_CallFunctionObjArgs py3_PyObject_CallFunctionObjArgs 20181c40c50SBram Moolenaar # define _PyObject_CallFunction_SizeT py3__PyObject_CallFunction_SizeT 202f4258308SBram Moolenaar # define PyObject_Call py3_PyObject_Call 2033dab2806SBram Moolenaar # define PyEval_GetLocals py3_PyEval_GetLocals 2043dab2806SBram Moolenaar # define PyEval_GetGlobals py3_PyEval_GetGlobals 205bd5e15fdSBram Moolenaar # define PySys_SetObject py3_PySys_SetObject 206c09a6d6cSBram Moolenaar # define PySys_GetObject py3_PySys_GetObject 207bd5e15fdSBram Moolenaar # define PySys_SetArgv py3_PySys_SetArgv 208bd5e15fdSBram Moolenaar # define PyType_Ready py3_PyType_Ready 209ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 210ee1b9316SBram Moolenaar # define PyType_GetFlags py3_PyType_GetFlags 211ee1b9316SBram Moolenaar # endif 212bd5e15fdSBram Moolenaar #undef Py_BuildValue 213bd5e15fdSBram Moolenaar # define Py_BuildValue py3_Py_BuildValue 214644d37b8SBram Moolenaar # define Py_SetPythonHome py3_Py_SetPythonHome 215bd5e15fdSBram Moolenaar # define Py_Initialize py3_Py_Initialize 216bd5e15fdSBram Moolenaar # define Py_Finalize py3_Py_Finalize 217bd5e15fdSBram Moolenaar # define Py_IsInitialized py3_Py_IsInitialized 218bd5e15fdSBram Moolenaar # define _Py_NoneStruct (*py3__Py_NoneStruct) 21966b7985eSBram Moolenaar # define _Py_FalseStruct (*py3__Py_FalseStruct) 22066b7985eSBram Moolenaar # define _Py_TrueStruct (*py3__Py_TrueStruct) 221db913953SBram Moolenaar # define _PyObject_NextNotImplemented (*py3__PyObject_NextNotImplemented) 222bd5e15fdSBram Moolenaar # define PyModule_AddObject py3_PyModule_AddObject 223bd5e15fdSBram Moolenaar # define PyImport_AppendInittab py3_PyImport_AppendInittab 2243dab2806SBram Moolenaar # define PyImport_AddModule py3_PyImport_AddModule 2257bc4f93cSBram Moolenaar # if PY_VERSION_HEX >= 0x030300f0 2267bc4f93cSBram Moolenaar # undef _PyUnicode_AsString 2279c9cbf13SBram Moolenaar # define _PyUnicode_AsString py3_PyUnicode_AsUTF8 2287bc4f93cSBram Moolenaar # else 229bd5e15fdSBram Moolenaar # define _PyUnicode_AsString py3__PyUnicode_AsString 2307bc4f93cSBram Moolenaar # endif 23119e60943SBram Moolenaar # undef PyUnicode_AsEncodedString 23219e60943SBram Moolenaar # define PyUnicode_AsEncodedString py3_PyUnicode_AsEncodedString 23319e60943SBram Moolenaar # undef PyBytes_AsString 23419e60943SBram Moolenaar # define PyBytes_AsString py3_PyBytes_AsString 23532ac8cdaSBram Moolenaar # ifndef PyBytes_AsStringAndSize 236cdab9051SBram Moolenaar # define PyBytes_AsStringAndSize py3_PyBytes_AsStringAndSize 23732ac8cdaSBram Moolenaar # endif 238db913953SBram Moolenaar # undef PyBytes_FromString 239db913953SBram Moolenaar # define PyBytes_FromString py3_PyBytes_FromString 2406e5ea8d2SBram Moolenaar # undef PyBytes_FromStringAndSize 2416e5ea8d2SBram Moolenaar # define PyBytes_FromStringAndSize py3_PyBytes_FromStringAndSize 242ee1b9316SBram Moolenaar # if defined(Py_DEBUG) || PY_VERSION_HEX >= 0x030900b0 243ee1b9316SBram Moolenaar # define _Py_Dealloc py3__Py_Dealloc 244ee1b9316SBram Moolenaar # endif 245db913953SBram Moolenaar # define PyFloat_FromDouble py3_PyFloat_FromDouble 246db913953SBram Moolenaar # define PyFloat_AsDouble py3_PyFloat_AsDouble 247bd5e15fdSBram Moolenaar # define PyObject_GenericGetAttr py3_PyObject_GenericGetAttr 24866b7985eSBram Moolenaar # define PyType_Type (*py3_PyType_Type) 249d4a8c98eSBram Moolenaar # define PyStdPrinter_Type (*py3_PyStdPrinter_Type) 250bd5e15fdSBram Moolenaar # define PySlice_Type (*py3_PySlice_Type) 251db913953SBram Moolenaar # define PyFloat_Type (*py3_PyFloat_Type) 252141be8a5SBram Moolenaar # define PyNumber_Check (*py3_PyNumber_Check) 253141be8a5SBram Moolenaar # define PyNumber_Long (*py3_PyNumber_Long) 25466b7985eSBram Moolenaar # define PyBool_Type (*py3_PyBool_Type) 25519e60943SBram Moolenaar # define PyErr_NewException py3_PyErr_NewException 256bd5e15fdSBram Moolenaar # ifdef Py_DEBUG 257bd5e15fdSBram Moolenaar # define _Py_NegativeRefcount py3__Py_NegativeRefcount 258bd5e15fdSBram Moolenaar # define _Py_RefTotal (*py3__Py_RefTotal) 2590014a53aSBram Moolenaar # define PyModule_Create2TraceRefs py3_PyModule_Create2TraceRefs 2600014a53aSBram Moolenaar # else 2610014a53aSBram Moolenaar # define PyModule_Create2 py3_PyModule_Create2 2620014a53aSBram Moolenaar # endif 2630014a53aSBram Moolenaar # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 264bd5e15fdSBram Moolenaar # define _PyObject_DebugMalloc py3__PyObject_DebugMalloc 265bd5e15fdSBram Moolenaar # define _PyObject_DebugFree py3__PyObject_DebugFree 266bd5e15fdSBram Moolenaar # else 267bd5e15fdSBram Moolenaar # define PyObject_Malloc py3_PyObject_Malloc 268bd5e15fdSBram Moolenaar # define PyObject_Free py3_PyObject_Free 269bd5e15fdSBram Moolenaar # endif 270774267bbSBram Moolenaar # define _PyObject_GC_New py3__PyObject_GC_New 271774267bbSBram Moolenaar # define PyObject_GC_Del py3_PyObject_GC_Del 272774267bbSBram Moolenaar # define PyObject_GC_UnTrack py3_PyObject_GC_UnTrack 273bd5e15fdSBram Moolenaar # define PyType_GenericAlloc py3_PyType_GenericAlloc 274bd5e15fdSBram Moolenaar # define PyType_GenericNew py3_PyType_GenericNew 275bd5e15fdSBram Moolenaar # undef PyUnicode_FromString 276bd5e15fdSBram Moolenaar # define PyUnicode_FromString py3_PyUnicode_FromString 2771a3b5695SBram Moolenaar # ifndef PyUnicode_FromFormat 2781a3b5695SBram Moolenaar # define PyUnicode_FromFormat py3_PyUnicode_FromFormat 2791a3b5695SBram Moolenaar # else 2801a3b5695SBram Moolenaar # define Py_UNICODE_USE_UCS_FUNCTIONS 2811a3b5695SBram Moolenaar # ifdef Py_UNICODE_WIDE 2821a3b5695SBram Moolenaar # define PyUnicodeUCS4_FromFormat py3_PyUnicodeUCS4_FromFormat 2831a3b5695SBram Moolenaar # else 2841a3b5695SBram Moolenaar # define PyUnicodeUCS2_FromFormat py3_PyUnicodeUCS2_FromFormat 2851a3b5695SBram Moolenaar # endif 2861a3b5695SBram Moolenaar # endif 28719e60943SBram Moolenaar # undef PyUnicode_Decode 28819e60943SBram Moolenaar # define PyUnicode_Decode py3_PyUnicode_Decode 289db913953SBram Moolenaar # define PyType_IsSubtype py3_PyType_IsSubtype 290db913953SBram Moolenaar # define PyCapsule_New py3_PyCapsule_New 291db913953SBram Moolenaar # define PyCapsule_GetPointer py3_PyCapsule_GetPointer 292bd5e15fdSBram Moolenaar 2930014a53aSBram Moolenaar # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 294bd5e15fdSBram Moolenaar # undef PyObject_NEW 295bd5e15fdSBram Moolenaar # define PyObject_NEW(type, typeobj) \ 296bd5e15fdSBram Moolenaar ( (type *) PyObject_Init( \ 297bd5e15fdSBram Moolenaar (PyObject *) _PyObject_DebugMalloc( _PyObject_SIZE(typeobj) ), (typeobj)) ) 298ee1b9316SBram Moolenaar # elif PY_VERSION_HEX >= 0x030900b0 299ee1b9316SBram Moolenaar # undef PyObject_NEW 300ee1b9316SBram Moolenaar # define PyObject_NEW(type, typeobj) \ 301ee1b9316SBram Moolenaar ((type *)py3__PyObject_New(typeobj)) 302bd5e15fdSBram Moolenaar # endif 303b61f95c3SBram Moolenaar 304bd5e15fdSBram Moolenaar /* 305bd5e15fdSBram Moolenaar * Pointers for dynamic link 306bd5e15fdSBram Moolenaar */ 307bd5e15fdSBram Moolenaar static int (*py3_PySys_SetArgv)(int, wchar_t **); 308644d37b8SBram Moolenaar static void (*py3_Py_SetPythonHome)(wchar_t *home); 309bd5e15fdSBram Moolenaar static void (*py3_Py_Initialize)(void); 310bd5e15fdSBram Moolenaar static PyObject* (*py3_PyList_New)(Py_ssize_t size); 311bd5e15fdSBram Moolenaar static PyGILState_STATE (*py3_PyGILState_Ensure)(void); 312bd5e15fdSBram Moolenaar static void (*py3_PyGILState_Release)(PyGILState_STATE); 313bd5e15fdSBram Moolenaar static int (*py3_PySys_SetObject)(char *, PyObject *); 314c09a6d6cSBram Moolenaar static PyObject* (*py3_PySys_GetObject)(char *); 315c09a6d6cSBram Moolenaar static int (*py3_PyList_Append)(PyObject *, PyObject *); 316c09a6d6cSBram Moolenaar static int (*py3_PyList_Insert)(PyObject *, int, PyObject *); 317bd5e15fdSBram Moolenaar static Py_ssize_t (*py3_PyList_Size)(PyObject *); 318db913953SBram Moolenaar static int (*py3_PySequence_Check)(PyObject *); 319db913953SBram Moolenaar static Py_ssize_t (*py3_PySequence_Size)(PyObject *); 320db913953SBram Moolenaar static PyObject* (*py3_PySequence_GetItem)(PyObject *, Py_ssize_t); 321a9922d62SBram Moolenaar static PyObject* (*py3_PySequence_Fast)(PyObject *, const char *); 322db913953SBram Moolenaar static Py_ssize_t (*py3_PyTuple_Size)(PyObject *); 323db913953SBram Moolenaar static PyObject* (*py3_PyTuple_GetItem)(PyObject *, Py_ssize_t); 324db913953SBram Moolenaar static int (*py3_PyMapping_Check)(PyObject *); 325bcb40977SBram Moolenaar static PyObject* (*py3_PyMapping_Keys)(PyObject *); 3263b48b11cSBram Moolenaar # if PY_VERSION_HEX >= 0x030601f0 3273b48b11cSBram Moolenaar static int (*py3_PySlice_AdjustIndices)(Py_ssize_t length, 3283b48b11cSBram Moolenaar Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t step); 3293b48b11cSBram Moolenaar static int (*py3_PySlice_Unpack)(PyObject *slice, 3303b48b11cSBram Moolenaar Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step); 3313b48b11cSBram Moolenaar # endif 332922a4664SBram Moolenaar static int (*py3_PySlice_GetIndicesEx)(PySliceObject_T *r, Py_ssize_t length, 333063a46baSBram Moolenaar Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step, 334063a46baSBram Moolenaar Py_ssize_t *slicelen); 335bd5e15fdSBram Moolenaar static PyObject* (*py3_PyErr_NoMemory)(void); 336bd5e15fdSBram Moolenaar static void (*py3_Py_Finalize)(void); 337bd5e15fdSBram Moolenaar static void (*py3_PyErr_SetString)(PyObject *, const char *); 3384d188da2SBram Moolenaar static void (*py3_PyErr_SetObject)(PyObject *, PyObject *); 339c09a6d6cSBram Moolenaar static int (*py3_PyErr_ExceptionMatches)(PyObject *); 340bd5e15fdSBram Moolenaar static int (*py3_PyRun_SimpleString)(char *); 341db913953SBram Moolenaar static PyObject* (*py3_PyRun_String)(char *, int, PyObject *, PyObject *); 3423dab2806SBram Moolenaar static PyObject* (*py3_PyObject_GetAttrString)(PyObject *, const char *); 343a9922d62SBram Moolenaar static int (*py3_PyObject_HasAttrString)(PyObject *, const char *); 3440b400087SBram Moolenaar static int (*py3_PyObject_SetAttrString)(PyObject *, const char *, PyObject *); 3453dab2806SBram Moolenaar static PyObject* (*py3_PyObject_CallFunctionObjArgs)(PyObject *, ...); 34681c40c50SBram Moolenaar static PyObject* (*py3__PyObject_CallFunction_SizeT)(PyObject *, char *, ...); 347f4258308SBram Moolenaar static PyObject* (*py3_PyObject_Call)(PyObject *, PyObject *, PyObject *); 3483dab2806SBram Moolenaar static PyObject* (*py3_PyEval_GetGlobals)(); 3493dab2806SBram Moolenaar static PyObject* (*py3_PyEval_GetLocals)(); 350bd5e15fdSBram Moolenaar static PyObject* (*py3_PyList_GetItem)(PyObject *, Py_ssize_t); 351bd5e15fdSBram Moolenaar static PyObject* (*py3_PyImport_ImportModule)(const char *); 352db913953SBram Moolenaar static PyObject* (*py3_PyImport_AddModule)(const char *); 353bd5e15fdSBram Moolenaar static int (*py3_PyErr_BadArgument)(void); 354bd5e15fdSBram Moolenaar static PyObject* (*py3_PyErr_Occurred)(void); 355bd5e15fdSBram Moolenaar static PyObject* (*py3_PyModule_GetDict)(PyObject *); 356bd5e15fdSBram Moolenaar static int (*py3_PyList_SetItem)(PyObject *, Py_ssize_t, PyObject *); 357bd5e15fdSBram Moolenaar static PyObject* (*py3_PyDict_GetItemString)(PyObject *, const char *); 358db913953SBram Moolenaar static int (*py3_PyDict_Next)(PyObject *, Py_ssize_t *, PyObject **, PyObject **); 359bd5e15fdSBram Moolenaar static PyObject* (*py3_PyLong_FromLong)(long); 360bd5e15fdSBram Moolenaar static PyObject* (*py3_PyDict_New)(void); 361db913953SBram Moolenaar static PyObject* (*py3_PyIter_Next)(PyObject *); 362db913953SBram Moolenaar static PyObject* (*py3_PyObject_GetIter)(PyObject *); 363141be8a5SBram Moolenaar static PyObject* (*py3_PyObject_Repr)(PyObject *); 364bcb40977SBram Moolenaar static PyObject* (*py3_PyObject_GetItem)(PyObject *, PyObject *); 36503db85b3SBram Moolenaar static int (*py3_PyObject_IsTrue)(PyObject *); 366bd5e15fdSBram Moolenaar static PyObject* (*py3_Py_BuildValue)(char *, ...); 367ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 368ee1b9316SBram Moolenaar static int (*py3_PyType_GetFlags)(PyTypeObject *o); 369ee1b9316SBram Moolenaar # endif 370bd5e15fdSBram Moolenaar static int (*py3_PyType_Ready)(PyTypeObject *type); 371bd5e15fdSBram Moolenaar static int (*py3_PyDict_SetItemString)(PyObject *dp, char *key, PyObject *item); 372bd5e15fdSBram Moolenaar static PyObject* (*py3_PyUnicode_FromString)(const char *u); 3731a3b5695SBram Moolenaar # ifndef Py_UNICODE_USE_UCS_FUNCTIONS 3741a3b5695SBram Moolenaar static PyObject* (*py3_PyUnicode_FromFormat)(const char *u, ...); 3751a3b5695SBram Moolenaar # else 3761a3b5695SBram Moolenaar # ifdef Py_UNICODE_WIDE 3771a3b5695SBram Moolenaar static PyObject* (*py3_PyUnicodeUCS4_FromFormat)(const char *u, ...); 3781a3b5695SBram Moolenaar # else 3791a3b5695SBram Moolenaar static PyObject* (*py3_PyUnicodeUCS2_FromFormat)(const char *u, ...); 3801a3b5695SBram Moolenaar # endif 3811a3b5695SBram Moolenaar # endif 38219e60943SBram Moolenaar static PyObject* (*py3_PyUnicode_Decode)(const char *u, Py_ssize_t size, 38319e60943SBram Moolenaar const char *encoding, const char *errors); 384bd5e15fdSBram Moolenaar static long (*py3_PyLong_AsLong)(PyObject *); 385bd5e15fdSBram Moolenaar static void (*py3_PyErr_SetNone)(PyObject *); 386bd5e15fdSBram Moolenaar static void (*py3_PyEval_InitThreads)(void); 387bd5e15fdSBram Moolenaar static void(*py3_PyEval_RestoreThread)(PyThreadState *); 388bd5e15fdSBram Moolenaar static PyThreadState*(*py3_PyEval_SaveThread)(void); 389bd5e15fdSBram Moolenaar static int (*py3_PyArg_Parse)(PyObject *, char *, ...); 390bd5e15fdSBram Moolenaar static int (*py3_PyArg_ParseTuple)(PyObject *, char *, ...); 39119e60943SBram Moolenaar static int (*py3_PyMem_Free)(void *); 392db913953SBram Moolenaar static void* (*py3_PyMem_Malloc)(size_t); 393bd5e15fdSBram Moolenaar static int (*py3_Py_IsInitialized)(void); 394bd5e15fdSBram Moolenaar static void (*py3_PyErr_Clear)(void); 395c476e52fSBram Moolenaar static PyObject* (*py3_PyErr_Format)(PyObject *, const char *, ...); 3964d36987cSBram Moolenaar static void (*py3_PyErr_PrintEx)(int); 397bd5e15fdSBram Moolenaar static PyObject*(*py3__PyObject_Init)(PyObject *, PyTypeObject *); 398db913953SBram Moolenaar static iternextfunc py3__PyObject_NextNotImplemented; 399bd5e15fdSBram Moolenaar static PyObject* py3__Py_NoneStruct; 40066b7985eSBram Moolenaar static PyObject* py3__Py_FalseStruct; 40166b7985eSBram Moolenaar static PyObject* py3__Py_TrueStruct; 402bd5e15fdSBram Moolenaar static int (*py3_PyModule_AddObject)(PyObject *m, const char *name, PyObject *o); 403bd5e15fdSBram Moolenaar static int (*py3_PyImport_AppendInittab)(const char *name, PyObject* (*initfunc)(void)); 4047bc4f93cSBram Moolenaar # if PY_VERSION_HEX >= 0x030300f0 4059c9cbf13SBram Moolenaar static char* (*py3_PyUnicode_AsUTF8)(PyObject *unicode); 4067bc4f93cSBram Moolenaar # else 407bd5e15fdSBram Moolenaar static char* (*py3__PyUnicode_AsString)(PyObject *unicode); 4087bc4f93cSBram Moolenaar # endif 40919e60943SBram Moolenaar static PyObject* (*py3_PyUnicode_AsEncodedString)(PyObject *unicode, const char* encoding, const char* errors); 41019e60943SBram Moolenaar static char* (*py3_PyBytes_AsString)(PyObject *bytes); 411808c2bc8SBram Moolenaar static int (*py3_PyBytes_AsStringAndSize)(PyObject *bytes, char **buffer, Py_ssize_t *length); 412db913953SBram Moolenaar static PyObject* (*py3_PyBytes_FromString)(char *str); 4136e5ea8d2SBram Moolenaar static PyObject* (*py3_PyBytes_FromStringAndSize)(char *str, Py_ssize_t length); 414ee1b9316SBram Moolenaar # if defined(Py_DEBUG) || PY_VERSION_HEX >= 0x030900b0 415ee1b9316SBram Moolenaar static void (*py3__Py_Dealloc)(PyObject *obj); 416ee1b9316SBram Moolenaar # endif 417ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 418ee1b9316SBram Moolenaar static PyObject* (*py3__PyObject_New)(PyTypeObject *); 419ee1b9316SBram Moolenaar # endif 420db913953SBram Moolenaar static PyObject* (*py3_PyFloat_FromDouble)(double num); 421db913953SBram Moolenaar static double (*py3_PyFloat_AsDouble)(PyObject *); 422bd5e15fdSBram Moolenaar static PyObject* (*py3_PyObject_GenericGetAttr)(PyObject *obj, PyObject *name); 423bd5e15fdSBram Moolenaar static PyObject* (*py3_PyType_GenericAlloc)(PyTypeObject *type, Py_ssize_t nitems); 424bd5e15fdSBram Moolenaar static PyObject* (*py3_PyType_GenericNew)(PyTypeObject *type, PyObject *args, PyObject *kwds); 42566b7985eSBram Moolenaar static PyTypeObject* py3_PyType_Type; 426d4a8c98eSBram Moolenaar static PyTypeObject* py3_PyStdPrinter_Type; 427bd5e15fdSBram Moolenaar static PyTypeObject* py3_PySlice_Type; 428db913953SBram Moolenaar static PyTypeObject* py3_PyFloat_Type; 42966b7985eSBram Moolenaar static PyTypeObject* py3_PyBool_Type; 430141be8a5SBram Moolenaar static int (*py3_PyNumber_Check)(PyObject *); 431141be8a5SBram Moolenaar static PyObject* (*py3_PyNumber_Long)(PyObject *); 43219e60943SBram Moolenaar static PyObject* (*py3_PyErr_NewException)(char *name, PyObject *base, PyObject *dict); 433db913953SBram Moolenaar static PyObject* (*py3_PyCapsule_New)(void *, char *, PyCapsule_Destructor); 434db913953SBram Moolenaar static void* (*py3_PyCapsule_GetPointer)(PyObject *, char *); 435bd5e15fdSBram Moolenaar # ifdef Py_DEBUG 436bd5e15fdSBram Moolenaar static void (*py3__Py_NegativeRefcount)(const char *fname, int lineno, PyObject *op); 437bd5e15fdSBram Moolenaar static Py_ssize_t* py3__Py_RefTotal; 4380014a53aSBram Moolenaar static PyObject* (*py3_PyModule_Create2TraceRefs)(struct PyModuleDef* module, int module_api_version); 4390014a53aSBram Moolenaar # else 4400014a53aSBram Moolenaar static PyObject* (*py3_PyModule_Create2)(struct PyModuleDef* module, int module_api_version); 4410014a53aSBram Moolenaar # endif 4420014a53aSBram Moolenaar # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 443bd5e15fdSBram Moolenaar static void (*py3__PyObject_DebugFree)(void*); 444bd5e15fdSBram Moolenaar static void* (*py3__PyObject_DebugMalloc)(size_t); 445bd5e15fdSBram Moolenaar # else 446bd5e15fdSBram Moolenaar static void (*py3_PyObject_Free)(void*); 447bd5e15fdSBram Moolenaar static void* (*py3_PyObject_Malloc)(size_t); 448bd5e15fdSBram Moolenaar # endif 449774267bbSBram Moolenaar static PyObject*(*py3__PyObject_GC_New)(PyTypeObject *); 450774267bbSBram Moolenaar static void(*py3_PyObject_GC_Del)(void *); 451774267bbSBram Moolenaar static void(*py3_PyObject_GC_UnTrack)(void *); 452db913953SBram Moolenaar static int (*py3_PyType_IsSubtype)(PyTypeObject *, PyTypeObject *); 453bd5e15fdSBram Moolenaar 4542ab2e860SBram Moolenaar static HINSTANCE hinstPy3 = 0; // Instance of python.dll 455bd5e15fdSBram Moolenaar 4562ab2e860SBram Moolenaar // Imported exception objects 457bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_AttributeError; 458bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_IndexError; 459af6abb9dSBram Moolenaar static PyObject *p3imp_PyExc_KeyError; 460bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_KeyboardInterrupt; 461bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_TypeError; 462bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_ValueError; 46341009374SBram Moolenaar static PyObject *p3imp_PyExc_SystemExit; 4648661b178SBram Moolenaar static PyObject *p3imp_PyExc_RuntimeError; 465c09a6d6cSBram Moolenaar static PyObject *p3imp_PyExc_ImportError; 466141be8a5SBram Moolenaar static PyObject *p3imp_PyExc_OverflowError; 467bd5e15fdSBram Moolenaar 468bd5e15fdSBram Moolenaar # define PyExc_AttributeError p3imp_PyExc_AttributeError 469bd5e15fdSBram Moolenaar # define PyExc_IndexError p3imp_PyExc_IndexError 470af6abb9dSBram Moolenaar # define PyExc_KeyError p3imp_PyExc_KeyError 471bd5e15fdSBram Moolenaar # define PyExc_KeyboardInterrupt p3imp_PyExc_KeyboardInterrupt 472bd5e15fdSBram Moolenaar # define PyExc_TypeError p3imp_PyExc_TypeError 473bd5e15fdSBram Moolenaar # define PyExc_ValueError p3imp_PyExc_ValueError 47441009374SBram Moolenaar # define PyExc_SystemExit p3imp_PyExc_SystemExit 4758661b178SBram Moolenaar # define PyExc_RuntimeError p3imp_PyExc_RuntimeError 476c09a6d6cSBram Moolenaar # define PyExc_ImportError p3imp_PyExc_ImportError 477141be8a5SBram Moolenaar # define PyExc_OverflowError p3imp_PyExc_OverflowError 478bd5e15fdSBram Moolenaar 479bd5e15fdSBram Moolenaar /* 480bd5e15fdSBram Moolenaar * Table of name to function pointer of python. 481bd5e15fdSBram Moolenaar */ 482bd5e15fdSBram Moolenaar # define PYTHON_PROC FARPROC 483bd5e15fdSBram Moolenaar static struct 484bd5e15fdSBram Moolenaar { 485bd5e15fdSBram Moolenaar char *name; 486bd5e15fdSBram Moolenaar PYTHON_PROC *ptr; 487bd5e15fdSBram Moolenaar } py3_funcname_table[] = 488bd5e15fdSBram Moolenaar { 489bd5e15fdSBram Moolenaar {"PySys_SetArgv", (PYTHON_PROC*)&py3_PySys_SetArgv}, 490644d37b8SBram Moolenaar {"Py_SetPythonHome", (PYTHON_PROC*)&py3_Py_SetPythonHome}, 491bd5e15fdSBram Moolenaar {"Py_Initialize", (PYTHON_PROC*)&py3_Py_Initialize}, 492e8cdcef8SBram Moolenaar {"_PyArg_ParseTuple_SizeT", (PYTHON_PROC*)&py3_PyArg_ParseTuple}, 493e8cdcef8SBram Moolenaar {"_Py_BuildValue_SizeT", (PYTHON_PROC*)&py3_Py_BuildValue}, 49419e60943SBram Moolenaar {"PyMem_Free", (PYTHON_PROC*)&py3_PyMem_Free}, 495db913953SBram Moolenaar {"PyMem_Malloc", (PYTHON_PROC*)&py3_PyMem_Malloc}, 496bd5e15fdSBram Moolenaar {"PyList_New", (PYTHON_PROC*)&py3_PyList_New}, 497bd5e15fdSBram Moolenaar {"PyGILState_Ensure", (PYTHON_PROC*)&py3_PyGILState_Ensure}, 498bd5e15fdSBram Moolenaar {"PyGILState_Release", (PYTHON_PROC*)&py3_PyGILState_Release}, 499bd5e15fdSBram Moolenaar {"PySys_SetObject", (PYTHON_PROC*)&py3_PySys_SetObject}, 500c09a6d6cSBram Moolenaar {"PySys_GetObject", (PYTHON_PROC*)&py3_PySys_GetObject}, 501bd5e15fdSBram Moolenaar {"PyList_Append", (PYTHON_PROC*)&py3_PyList_Append}, 502c09a6d6cSBram Moolenaar {"PyList_Insert", (PYTHON_PROC*)&py3_PyList_Insert}, 503bd5e15fdSBram Moolenaar {"PyList_Size", (PYTHON_PROC*)&py3_PyList_Size}, 504db913953SBram Moolenaar {"PySequence_Check", (PYTHON_PROC*)&py3_PySequence_Check}, 505db913953SBram Moolenaar {"PySequence_Size", (PYTHON_PROC*)&py3_PySequence_Size}, 506db913953SBram Moolenaar {"PySequence_GetItem", (PYTHON_PROC*)&py3_PySequence_GetItem}, 507a9922d62SBram Moolenaar {"PySequence_Fast", (PYTHON_PROC*)&py3_PySequence_Fast}, 508db913953SBram Moolenaar {"PyTuple_Size", (PYTHON_PROC*)&py3_PyTuple_Size}, 509db913953SBram Moolenaar {"PyTuple_GetItem", (PYTHON_PROC*)&py3_PyTuple_GetItem}, 5103b48b11cSBram Moolenaar # if PY_VERSION_HEX >= 0x030601f0 5113b48b11cSBram Moolenaar {"PySlice_AdjustIndices", (PYTHON_PROC*)&py3_PySlice_AdjustIndices}, 5123b48b11cSBram Moolenaar {"PySlice_Unpack", (PYTHON_PROC*)&py3_PySlice_Unpack}, 5133b48b11cSBram Moolenaar # endif 514bd5e15fdSBram Moolenaar {"PySlice_GetIndicesEx", (PYTHON_PROC*)&py3_PySlice_GetIndicesEx}, 515bd5e15fdSBram Moolenaar {"PyErr_NoMemory", (PYTHON_PROC*)&py3_PyErr_NoMemory}, 516bd5e15fdSBram Moolenaar {"Py_Finalize", (PYTHON_PROC*)&py3_Py_Finalize}, 517bd5e15fdSBram Moolenaar {"PyErr_SetString", (PYTHON_PROC*)&py3_PyErr_SetString}, 5184d188da2SBram Moolenaar {"PyErr_SetObject", (PYTHON_PROC*)&py3_PyErr_SetObject}, 519c09a6d6cSBram Moolenaar {"PyErr_ExceptionMatches", (PYTHON_PROC*)&py3_PyErr_ExceptionMatches}, 520bd5e15fdSBram Moolenaar {"PyRun_SimpleString", (PYTHON_PROC*)&py3_PyRun_SimpleString}, 521db913953SBram Moolenaar {"PyRun_String", (PYTHON_PROC*)&py3_PyRun_String}, 5223dab2806SBram Moolenaar {"PyObject_GetAttrString", (PYTHON_PROC*)&py3_PyObject_GetAttrString}, 523a9922d62SBram Moolenaar {"PyObject_HasAttrString", (PYTHON_PROC*)&py3_PyObject_HasAttrString}, 5243dab2806SBram Moolenaar {"PyObject_SetAttrString", (PYTHON_PROC*)&py3_PyObject_SetAttrString}, 5253dab2806SBram Moolenaar {"PyObject_CallFunctionObjArgs", (PYTHON_PROC*)&py3_PyObject_CallFunctionObjArgs}, 52681c40c50SBram Moolenaar {"_PyObject_CallFunction_SizeT", (PYTHON_PROC*)&py3__PyObject_CallFunction_SizeT}, 527f4258308SBram Moolenaar {"PyObject_Call", (PYTHON_PROC*)&py3_PyObject_Call}, 5283dab2806SBram Moolenaar {"PyEval_GetGlobals", (PYTHON_PROC*)&py3_PyEval_GetGlobals}, 5293dab2806SBram Moolenaar {"PyEval_GetLocals", (PYTHON_PROC*)&py3_PyEval_GetLocals}, 530bd5e15fdSBram Moolenaar {"PyList_GetItem", (PYTHON_PROC*)&py3_PyList_GetItem}, 531bd5e15fdSBram Moolenaar {"PyImport_ImportModule", (PYTHON_PROC*)&py3_PyImport_ImportModule}, 532db913953SBram Moolenaar {"PyImport_AddModule", (PYTHON_PROC*)&py3_PyImport_AddModule}, 533bd5e15fdSBram Moolenaar {"PyErr_BadArgument", (PYTHON_PROC*)&py3_PyErr_BadArgument}, 534bd5e15fdSBram Moolenaar {"PyErr_Occurred", (PYTHON_PROC*)&py3_PyErr_Occurred}, 535bd5e15fdSBram Moolenaar {"PyModule_GetDict", (PYTHON_PROC*)&py3_PyModule_GetDict}, 536bd5e15fdSBram Moolenaar {"PyList_SetItem", (PYTHON_PROC*)&py3_PyList_SetItem}, 537bd5e15fdSBram Moolenaar {"PyDict_GetItemString", (PYTHON_PROC*)&py3_PyDict_GetItemString}, 538db913953SBram Moolenaar {"PyDict_Next", (PYTHON_PROC*)&py3_PyDict_Next}, 539db913953SBram Moolenaar {"PyMapping_Check", (PYTHON_PROC*)&py3_PyMapping_Check}, 540bcb40977SBram Moolenaar {"PyMapping_Keys", (PYTHON_PROC*)&py3_PyMapping_Keys}, 541db913953SBram Moolenaar {"PyIter_Next", (PYTHON_PROC*)&py3_PyIter_Next}, 542db913953SBram Moolenaar {"PyObject_GetIter", (PYTHON_PROC*)&py3_PyObject_GetIter}, 543141be8a5SBram Moolenaar {"PyObject_Repr", (PYTHON_PROC*)&py3_PyObject_Repr}, 544bcb40977SBram Moolenaar {"PyObject_GetItem", (PYTHON_PROC*)&py3_PyObject_GetItem}, 54503db85b3SBram Moolenaar {"PyObject_IsTrue", (PYTHON_PROC*)&py3_PyObject_IsTrue}, 546bd5e15fdSBram Moolenaar {"PyLong_FromLong", (PYTHON_PROC*)&py3_PyLong_FromLong}, 547bd5e15fdSBram Moolenaar {"PyDict_New", (PYTHON_PROC*)&py3_PyDict_New}, 548ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 549ee1b9316SBram Moolenaar {"PyType_GetFlags", (PYTHON_PROC*)&py3_PyType_GetFlags}, 550ee1b9316SBram Moolenaar # endif 551bd5e15fdSBram Moolenaar {"PyType_Ready", (PYTHON_PROC*)&py3_PyType_Ready}, 552bd5e15fdSBram Moolenaar {"PyDict_SetItemString", (PYTHON_PROC*)&py3_PyDict_SetItemString}, 553bd5e15fdSBram Moolenaar {"PyLong_AsLong", (PYTHON_PROC*)&py3_PyLong_AsLong}, 554bd5e15fdSBram Moolenaar {"PyErr_SetNone", (PYTHON_PROC*)&py3_PyErr_SetNone}, 555bd5e15fdSBram Moolenaar {"PyEval_InitThreads", (PYTHON_PROC*)&py3_PyEval_InitThreads}, 556bd5e15fdSBram Moolenaar {"PyEval_RestoreThread", (PYTHON_PROC*)&py3_PyEval_RestoreThread}, 557bd5e15fdSBram Moolenaar {"PyEval_SaveThread", (PYTHON_PROC*)&py3_PyEval_SaveThread}, 5585f87b232SBram Moolenaar {"_PyArg_Parse_SizeT", (PYTHON_PROC*)&py3_PyArg_Parse}, 559bd5e15fdSBram Moolenaar {"Py_IsInitialized", (PYTHON_PROC*)&py3_Py_IsInitialized}, 560db913953SBram Moolenaar {"_PyObject_NextNotImplemented", (PYTHON_PROC*)&py3__PyObject_NextNotImplemented}, 561bd5e15fdSBram Moolenaar {"_Py_NoneStruct", (PYTHON_PROC*)&py3__Py_NoneStruct}, 56266b7985eSBram Moolenaar {"_Py_FalseStruct", (PYTHON_PROC*)&py3__Py_FalseStruct}, 56366b7985eSBram Moolenaar {"_Py_TrueStruct", (PYTHON_PROC*)&py3__Py_TrueStruct}, 564bd5e15fdSBram Moolenaar {"PyErr_Clear", (PYTHON_PROC*)&py3_PyErr_Clear}, 565c476e52fSBram Moolenaar {"PyErr_Format", (PYTHON_PROC*)&py3_PyErr_Format}, 5664d36987cSBram Moolenaar {"PyErr_PrintEx", (PYTHON_PROC*)&py3_PyErr_PrintEx}, 567bd5e15fdSBram Moolenaar {"PyObject_Init", (PYTHON_PROC*)&py3__PyObject_Init}, 568bd5e15fdSBram Moolenaar {"PyModule_AddObject", (PYTHON_PROC*)&py3_PyModule_AddObject}, 569bd5e15fdSBram Moolenaar {"PyImport_AppendInittab", (PYTHON_PROC*)&py3_PyImport_AppendInittab}, 5707bc4f93cSBram Moolenaar # if PY_VERSION_HEX >= 0x030300f0 5719c9cbf13SBram Moolenaar {"PyUnicode_AsUTF8", (PYTHON_PROC*)&py3_PyUnicode_AsUTF8}, 5727bc4f93cSBram Moolenaar # else 573bd5e15fdSBram Moolenaar {"_PyUnicode_AsString", (PYTHON_PROC*)&py3__PyUnicode_AsString}, 5747bc4f93cSBram Moolenaar # endif 5751a3b5695SBram Moolenaar # ifndef Py_UNICODE_USE_UCS_FUNCTIONS 5761a3b5695SBram Moolenaar {"PyUnicode_FromFormat", (PYTHON_PROC*)&py3_PyUnicode_FromFormat}, 5771a3b5695SBram Moolenaar # else 5781a3b5695SBram Moolenaar # ifdef Py_UNICODE_WIDE 5791a3b5695SBram Moolenaar {"PyUnicodeUCS4_FromFormat", (PYTHON_PROC*)&py3_PyUnicodeUCS4_FromFormat}, 5801a3b5695SBram Moolenaar # else 5811a3b5695SBram Moolenaar {"PyUnicodeUCS2_FromFormat", (PYTHON_PROC*)&py3_PyUnicodeUCS2_FromFormat}, 5821a3b5695SBram Moolenaar # endif 5831a3b5695SBram Moolenaar # endif 58419e60943SBram Moolenaar {"PyBytes_AsString", (PYTHON_PROC*)&py3_PyBytes_AsString}, 585cdab9051SBram Moolenaar {"PyBytes_AsStringAndSize", (PYTHON_PROC*)&py3_PyBytes_AsStringAndSize}, 586db913953SBram Moolenaar {"PyBytes_FromString", (PYTHON_PROC*)&py3_PyBytes_FromString}, 5876e5ea8d2SBram Moolenaar {"PyBytes_FromStringAndSize", (PYTHON_PROC*)&py3_PyBytes_FromStringAndSize}, 588ee1b9316SBram Moolenaar # if defined(Py_DEBUG) || PY_VERSION_HEX >= 0x030900b0 589ee1b9316SBram Moolenaar {"_Py_Dealloc", (PYTHON_PROC*)&py3__Py_Dealloc}, 590ee1b9316SBram Moolenaar # endif 591ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 592ee1b9316SBram Moolenaar {"_PyObject_New", (PYTHON_PROC*)&py3__PyObject_New}, 593ee1b9316SBram Moolenaar # endif 594db913953SBram Moolenaar {"PyFloat_FromDouble", (PYTHON_PROC*)&py3_PyFloat_FromDouble}, 595db913953SBram Moolenaar {"PyFloat_AsDouble", (PYTHON_PROC*)&py3_PyFloat_AsDouble}, 596bd5e15fdSBram Moolenaar {"PyObject_GenericGetAttr", (PYTHON_PROC*)&py3_PyObject_GenericGetAttr}, 597bd5e15fdSBram Moolenaar {"PyType_GenericAlloc", (PYTHON_PROC*)&py3_PyType_GenericAlloc}, 598bd5e15fdSBram Moolenaar {"PyType_GenericNew", (PYTHON_PROC*)&py3_PyType_GenericNew}, 59966b7985eSBram Moolenaar {"PyType_Type", (PYTHON_PROC*)&py3_PyType_Type}, 600d4a8c98eSBram Moolenaar {"PyStdPrinter_Type", (PYTHON_PROC*)&py3_PyStdPrinter_Type}, 601bd5e15fdSBram Moolenaar {"PySlice_Type", (PYTHON_PROC*)&py3_PySlice_Type}, 602db913953SBram Moolenaar {"PyFloat_Type", (PYTHON_PROC*)&py3_PyFloat_Type}, 60366b7985eSBram Moolenaar {"PyBool_Type", (PYTHON_PROC*)&py3_PyBool_Type}, 604141be8a5SBram Moolenaar {"PyNumber_Check", (PYTHON_PROC*)&py3_PyNumber_Check}, 605141be8a5SBram Moolenaar {"PyNumber_Long", (PYTHON_PROC*)&py3_PyNumber_Long}, 60619e60943SBram Moolenaar {"PyErr_NewException", (PYTHON_PROC*)&py3_PyErr_NewException}, 607bd5e15fdSBram Moolenaar # ifdef Py_DEBUG 608bd5e15fdSBram Moolenaar {"_Py_NegativeRefcount", (PYTHON_PROC*)&py3__Py_NegativeRefcount}, 609bd5e15fdSBram Moolenaar {"_Py_RefTotal", (PYTHON_PROC*)&py3__Py_RefTotal}, 6100014a53aSBram Moolenaar {"PyModule_Create2TraceRefs", (PYTHON_PROC*)&py3_PyModule_Create2TraceRefs}, 6110014a53aSBram Moolenaar # else 6120014a53aSBram Moolenaar {"PyModule_Create2", (PYTHON_PROC*)&py3_PyModule_Create2}, 6130014a53aSBram Moolenaar # endif 6140014a53aSBram Moolenaar # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 615bd5e15fdSBram Moolenaar {"_PyObject_DebugFree", (PYTHON_PROC*)&py3__PyObject_DebugFree}, 616bd5e15fdSBram Moolenaar {"_PyObject_DebugMalloc", (PYTHON_PROC*)&py3__PyObject_DebugMalloc}, 617bd5e15fdSBram Moolenaar # else 618bd5e15fdSBram Moolenaar {"PyObject_Malloc", (PYTHON_PROC*)&py3_PyObject_Malloc}, 619bd5e15fdSBram Moolenaar {"PyObject_Free", (PYTHON_PROC*)&py3_PyObject_Free}, 620bd5e15fdSBram Moolenaar # endif 621774267bbSBram Moolenaar {"_PyObject_GC_New", (PYTHON_PROC*)&py3__PyObject_GC_New}, 622774267bbSBram Moolenaar {"PyObject_GC_Del", (PYTHON_PROC*)&py3_PyObject_GC_Del}, 623774267bbSBram Moolenaar {"PyObject_GC_UnTrack", (PYTHON_PROC*)&py3_PyObject_GC_UnTrack}, 624db913953SBram Moolenaar {"PyType_IsSubtype", (PYTHON_PROC*)&py3_PyType_IsSubtype}, 625db913953SBram Moolenaar {"PyCapsule_New", (PYTHON_PROC*)&py3_PyCapsule_New}, 626db913953SBram Moolenaar {"PyCapsule_GetPointer", (PYTHON_PROC*)&py3_PyCapsule_GetPointer}, 627bd5e15fdSBram Moolenaar {"", NULL}, 628bd5e15fdSBram Moolenaar }; 629bd5e15fdSBram Moolenaar 63013a1f3fbSBram Moolenaar # if PY_VERSION_HEX >= 0x030800f0 63113a1f3fbSBram Moolenaar static inline void 63213a1f3fbSBram Moolenaar py3__Py_DECREF(const char *filename UNUSED, int lineno UNUSED, PyObject *op) 63313a1f3fbSBram Moolenaar { 63413a1f3fbSBram Moolenaar if (--op->ob_refcnt != 0) 63513a1f3fbSBram Moolenaar { 63613a1f3fbSBram Moolenaar # ifdef Py_REF_DEBUG 63713a1f3fbSBram Moolenaar if (op->ob_refcnt < 0) 63813a1f3fbSBram Moolenaar { 63913a1f3fbSBram Moolenaar _Py_NegativeRefcount(filename, lineno, op); 64013a1f3fbSBram Moolenaar } 64113a1f3fbSBram Moolenaar # endif 64213a1f3fbSBram Moolenaar } 64313a1f3fbSBram Moolenaar else 64413a1f3fbSBram Moolenaar { 64513a1f3fbSBram Moolenaar _Py_Dealloc(op); 64613a1f3fbSBram Moolenaar } 64713a1f3fbSBram Moolenaar } 64813a1f3fbSBram Moolenaar 64913a1f3fbSBram Moolenaar # undef Py_DECREF 65013a1f3fbSBram Moolenaar # define Py_DECREF(op) py3__Py_DECREF(__FILE__, __LINE__, _PyObject_CAST(op)) 65113a1f3fbSBram Moolenaar 65213a1f3fbSBram Moolenaar static inline void 65313a1f3fbSBram Moolenaar py3__Py_XDECREF(PyObject *op) 65413a1f3fbSBram Moolenaar { 65513a1f3fbSBram Moolenaar if (op != NULL) 65613a1f3fbSBram Moolenaar { 65713a1f3fbSBram Moolenaar Py_DECREF(op); 65813a1f3fbSBram Moolenaar } 65913a1f3fbSBram Moolenaar } 66013a1f3fbSBram Moolenaar 66113a1f3fbSBram Moolenaar # undef Py_XDECREF 66213a1f3fbSBram Moolenaar # define Py_XDECREF(op) py3__Py_XDECREF(_PyObject_CAST(op)) 66313a1f3fbSBram Moolenaar # endif 66413a1f3fbSBram Moolenaar 665ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 666ee1b9316SBram Moolenaar static inline int 667ee1b9316SBram Moolenaar py3_PyType_HasFeature(PyTypeObject *type, unsigned long feature) 668ee1b9316SBram Moolenaar { 669ee1b9316SBram Moolenaar return ((PyType_GetFlags(type) & feature) != 0); 670ee1b9316SBram Moolenaar } 671ee1b9316SBram Moolenaar # define PyType_HasFeature(t,f) py3_PyType_HasFeature(t,f) 672ee1b9316SBram Moolenaar # endif 673ee1b9316SBram Moolenaar 674*b2f9e0e2SBram Moolenaar # ifdef MSWIN 675*b2f9e0e2SBram Moolenaar /* 676*b2f9e0e2SBram Moolenaar * Look up the library "libname" using the InstallPath registry key. 677*b2f9e0e2SBram Moolenaar * Return NULL when failed. Return an allocated string when successful. 678*b2f9e0e2SBram Moolenaar */ 679*b2f9e0e2SBram Moolenaar static char * 680*b2f9e0e2SBram Moolenaar py3_get_system_libname(const char *libname) 681*b2f9e0e2SBram Moolenaar { 682*b2f9e0e2SBram Moolenaar const char *cp = libname; 683*b2f9e0e2SBram Moolenaar char subkey[128]; 684*b2f9e0e2SBram Moolenaar HKEY hKey; 685*b2f9e0e2SBram Moolenaar char installpath[MAXPATHL]; 686*b2f9e0e2SBram Moolenaar LONG len = sizeof(installpath); 687*b2f9e0e2SBram Moolenaar LSTATUS rc; 688*b2f9e0e2SBram Moolenaar size_t sysliblen; 689*b2f9e0e2SBram Moolenaar char *syslibname; 690*b2f9e0e2SBram Moolenaar 691*b2f9e0e2SBram Moolenaar while (*cp != '\0') 692*b2f9e0e2SBram Moolenaar { 693*b2f9e0e2SBram Moolenaar if (*cp == ':' || *cp == '\\' || *cp == '/') 694*b2f9e0e2SBram Moolenaar { 695*b2f9e0e2SBram Moolenaar // Bail out if "libname" contains path separator, assume it is 696*b2f9e0e2SBram Moolenaar // an absolute path. 697*b2f9e0e2SBram Moolenaar return NULL; 698*b2f9e0e2SBram Moolenaar } 699*b2f9e0e2SBram Moolenaar ++cp; 700*b2f9e0e2SBram Moolenaar } 701*b2f9e0e2SBram Moolenaar vim_snprintf(subkey, sizeof(subkey), 702*b2f9e0e2SBram Moolenaar # ifdef _WIN64 703*b2f9e0e2SBram Moolenaar "Software\\Python\\PythonCore\\%d.%d\\InstallPath", 704*b2f9e0e2SBram Moolenaar # else 705*b2f9e0e2SBram Moolenaar "Software\\Python\\PythonCore\\%d.%d-32\\InstallPath", 706*b2f9e0e2SBram Moolenaar # endif 707*b2f9e0e2SBram Moolenaar PY_MAJOR_VERSION, PY_MINOR_VERSION); 708*b2f9e0e2SBram Moolenaar if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, subkey, 0, KEY_QUERY_VALUE, &hKey) 709*b2f9e0e2SBram Moolenaar != ERROR_SUCCESS) 710*b2f9e0e2SBram Moolenaar return NULL; 711*b2f9e0e2SBram Moolenaar rc = RegQueryValueA(hKey, NULL, installpath, &len); 712*b2f9e0e2SBram Moolenaar RegCloseKey(hKey); 713*b2f9e0e2SBram Moolenaar if (ERROR_SUCCESS != rc) 714*b2f9e0e2SBram Moolenaar return NULL; 715*b2f9e0e2SBram Moolenaar cp = installpath + len; 716*b2f9e0e2SBram Moolenaar // Just in case registry value contains null terminators. 717*b2f9e0e2SBram Moolenaar while (cp > installpath && *(cp-1) == '\0') 718*b2f9e0e2SBram Moolenaar --cp; 719*b2f9e0e2SBram Moolenaar // Remove trailing path separators. 720*b2f9e0e2SBram Moolenaar while (cp > installpath && (*(cp-1) == '\\' || *(cp-1) == '/')) 721*b2f9e0e2SBram Moolenaar --cp; 722*b2f9e0e2SBram Moolenaar // Ignore if InstallPath is effectively empty. 723*b2f9e0e2SBram Moolenaar if (cp <= installpath) 724*b2f9e0e2SBram Moolenaar return NULL; 725*b2f9e0e2SBram Moolenaar sysliblen = (cp - installpath) + 1 + STRLEN(libname) + 1; 726*b2f9e0e2SBram Moolenaar syslibname = alloc(sysliblen); 727*b2f9e0e2SBram Moolenaar vim_snprintf(syslibname, sysliblen, "%.*s\\%s", 728*b2f9e0e2SBram Moolenaar (int)(cp - installpath), installpath, libname); 729*b2f9e0e2SBram Moolenaar return syslibname; 730*b2f9e0e2SBram Moolenaar } 731*b2f9e0e2SBram Moolenaar # endif 732*b2f9e0e2SBram Moolenaar 733bd5e15fdSBram Moolenaar /* 734bd5e15fdSBram Moolenaar * Load library and get all pointers. 735bd5e15fdSBram Moolenaar * Parameter 'libname' provides name of DLL. 736bd5e15fdSBram Moolenaar * Return OK or FAIL. 737bd5e15fdSBram Moolenaar */ 738170bf1aeSBram Moolenaar static int 739170bf1aeSBram Moolenaar py3_runtime_link_init(char *libname, int verbose) 740bd5e15fdSBram Moolenaar { 741bd5e15fdSBram Moolenaar int i; 7427b24ce08SBram Moolenaar PYTHON_PROC *ucs_from_string = (PYTHON_PROC *)&py3_PyUnicode_FromString; 7437b24ce08SBram Moolenaar PYTHON_PROC *ucs_decode = (PYTHON_PROC *)&py3_PyUnicode_Decode; 7447b24ce08SBram Moolenaar PYTHON_PROC *ucs_as_encoded_string = 7457b24ce08SBram Moolenaar (PYTHON_PROC *)&py3_PyUnicode_AsEncodedString; 746bd5e15fdSBram Moolenaar 747644d37b8SBram Moolenaar # if !(defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)) && defined(UNIX) && defined(FEAT_PYTHON) 7482ab2e860SBram Moolenaar // Can't have Python and Python3 loaded at the same time. 7492ab2e860SBram Moolenaar // It cause a crash, because RTLD_GLOBAL is needed for 7502ab2e860SBram Moolenaar // standard C extension libraries of one or both python versions. 7514c3a326cSBram Moolenaar if (python_loaded()) 7524c3a326cSBram Moolenaar { 7539dc93ae4SBram Moolenaar if (verbose) 754f9e3e09fSBram Moolenaar emsg(_("E837: This Vim cannot execute :py3 after using :python")); 7554c3a326cSBram Moolenaar return FAIL; 7564c3a326cSBram Moolenaar } 7574c3a326cSBram Moolenaar # endif 7584c3a326cSBram Moolenaar 7594c3a326cSBram Moolenaar if (hinstPy3 != 0) 760bd5e15fdSBram Moolenaar return OK; 761bd5e15fdSBram Moolenaar hinstPy3 = load_dll(libname); 762bd5e15fdSBram Moolenaar 763*b2f9e0e2SBram Moolenaar # ifdef MSWIN 764*b2f9e0e2SBram Moolenaar if (!hinstPy3) 765*b2f9e0e2SBram Moolenaar { 766*b2f9e0e2SBram Moolenaar // Attempt to use the path from InstallPath as stored in the registry. 767*b2f9e0e2SBram Moolenaar char *syslibname = py3_get_system_libname(libname); 768*b2f9e0e2SBram Moolenaar 769*b2f9e0e2SBram Moolenaar if (syslibname != NULL) 770*b2f9e0e2SBram Moolenaar { 771*b2f9e0e2SBram Moolenaar hinstPy3 = load_dll(syslibname); 772*b2f9e0e2SBram Moolenaar vim_free(syslibname); 773*b2f9e0e2SBram Moolenaar } 774*b2f9e0e2SBram Moolenaar } 775*b2f9e0e2SBram Moolenaar # endif 776*b2f9e0e2SBram Moolenaar 777bd5e15fdSBram Moolenaar if (!hinstPy3) 778bd5e15fdSBram Moolenaar { 779bd5e15fdSBram Moolenaar if (verbose) 780f9e3e09fSBram Moolenaar semsg(_(e_loadlib), libname); 781bd5e15fdSBram Moolenaar return FAIL; 782bd5e15fdSBram Moolenaar } 783bd5e15fdSBram Moolenaar 784bd5e15fdSBram Moolenaar for (i = 0; py3_funcname_table[i].ptr; ++i) 785bd5e15fdSBram Moolenaar { 786bd5e15fdSBram Moolenaar if ((*py3_funcname_table[i].ptr = symbol_from_dll(hinstPy3, 787bd5e15fdSBram Moolenaar py3_funcname_table[i].name)) == NULL) 788bd5e15fdSBram Moolenaar { 789bd5e15fdSBram Moolenaar close_dll(hinstPy3); 790bd5e15fdSBram Moolenaar hinstPy3 = 0; 791bd5e15fdSBram Moolenaar if (verbose) 792f9e3e09fSBram Moolenaar semsg(_(e_loadfunc), py3_funcname_table[i].name); 793bd5e15fdSBram Moolenaar return FAIL; 794bd5e15fdSBram Moolenaar } 795bd5e15fdSBram Moolenaar } 796bd5e15fdSBram Moolenaar 7972ab2e860SBram Moolenaar // Load unicode functions separately as only the ucs2 or the ucs4 functions 7982ab2e860SBram Moolenaar // will be present in the library. 7997bc4f93cSBram Moolenaar # if PY_VERSION_HEX >= 0x030300f0 8007b24ce08SBram Moolenaar *ucs_from_string = symbol_from_dll(hinstPy3, "PyUnicode_FromString"); 8017b24ce08SBram Moolenaar *ucs_decode = symbol_from_dll(hinstPy3, "PyUnicode_Decode"); 8027b24ce08SBram Moolenaar *ucs_as_encoded_string = symbol_from_dll(hinstPy3, 8037bc4f93cSBram Moolenaar "PyUnicode_AsEncodedString"); 8047bc4f93cSBram Moolenaar # else 8057b24ce08SBram Moolenaar *ucs_from_string = symbol_from_dll(hinstPy3, "PyUnicodeUCS2_FromString"); 8067b24ce08SBram Moolenaar *ucs_decode = symbol_from_dll(hinstPy3, 80719e60943SBram Moolenaar "PyUnicodeUCS2_Decode"); 8087b24ce08SBram Moolenaar *ucs_as_encoded_string = symbol_from_dll(hinstPy3, 80919e60943SBram Moolenaar "PyUnicodeUCS2_AsEncodedString"); 8107b24ce08SBram Moolenaar if (*ucs_from_string == NULL || *ucs_decode == NULL 8117b24ce08SBram Moolenaar || *ucs_as_encoded_string == NULL) 812bd5e15fdSBram Moolenaar { 8137b24ce08SBram Moolenaar *ucs_from_string = symbol_from_dll(hinstPy3, 814bd5e15fdSBram Moolenaar "PyUnicodeUCS4_FromString"); 8157b24ce08SBram Moolenaar *ucs_decode = symbol_from_dll(hinstPy3, 81619e60943SBram Moolenaar "PyUnicodeUCS4_Decode"); 8177b24ce08SBram Moolenaar *ucs_as_encoded_string = symbol_from_dll(hinstPy3, 81819e60943SBram Moolenaar "PyUnicodeUCS4_AsEncodedString"); 819bd5e15fdSBram Moolenaar } 8207bc4f93cSBram Moolenaar # endif 8217b24ce08SBram Moolenaar if (*ucs_from_string == NULL || *ucs_decode == NULL 8227b24ce08SBram Moolenaar || *ucs_as_encoded_string == NULL) 823bd5e15fdSBram Moolenaar { 824bd5e15fdSBram Moolenaar close_dll(hinstPy3); 825bd5e15fdSBram Moolenaar hinstPy3 = 0; 826bd5e15fdSBram Moolenaar if (verbose) 827f9e3e09fSBram Moolenaar semsg(_(e_loadfunc), "PyUnicode_UCSX_*"); 828bd5e15fdSBram Moolenaar return FAIL; 829bd5e15fdSBram Moolenaar } 830bd5e15fdSBram Moolenaar 831bd5e15fdSBram Moolenaar return OK; 832bd5e15fdSBram Moolenaar } 833bd5e15fdSBram Moolenaar 834bd5e15fdSBram Moolenaar /* 835bd5e15fdSBram Moolenaar * If python is enabled (there is installed python on Windows system) return 836bd5e15fdSBram Moolenaar * TRUE, else FALSE. 837bd5e15fdSBram Moolenaar */ 838170bf1aeSBram Moolenaar int 839170bf1aeSBram Moolenaar python3_enabled(int verbose) 840bd5e15fdSBram Moolenaar { 84125e4fcdeSBram Moolenaar return py3_runtime_link_init((char *)p_py3dll, verbose) == OK; 842bd5e15fdSBram Moolenaar } 843bd5e15fdSBram Moolenaar 8442ab2e860SBram Moolenaar /* 8452ab2e860SBram Moolenaar * Load the standard Python exceptions - don't import the symbols from the 846bd5e15fdSBram Moolenaar * DLL, as this can cause errors (importing data symbols is not reliable). 847bd5e15fdSBram Moolenaar */ 848170bf1aeSBram Moolenaar static void 84968c2f638SBram Moolenaar get_py3_exceptions(void) 850bd5e15fdSBram Moolenaar { 851bd5e15fdSBram Moolenaar PyObject *exmod = PyImport_ImportModule("builtins"); 852bd5e15fdSBram Moolenaar PyObject *exdict = PyModule_GetDict(exmod); 853bd5e15fdSBram Moolenaar p3imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError"); 854bd5e15fdSBram Moolenaar p3imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError"); 855af6abb9dSBram Moolenaar p3imp_PyExc_KeyError = PyDict_GetItemString(exdict, "KeyError"); 856bd5e15fdSBram Moolenaar p3imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt"); 857bd5e15fdSBram Moolenaar p3imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError"); 858bd5e15fdSBram Moolenaar p3imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError"); 85941009374SBram Moolenaar p3imp_PyExc_SystemExit = PyDict_GetItemString(exdict, "SystemExit"); 8608661b178SBram Moolenaar p3imp_PyExc_RuntimeError = PyDict_GetItemString(exdict, "RuntimeError"); 861c09a6d6cSBram Moolenaar p3imp_PyExc_ImportError = PyDict_GetItemString(exdict, "ImportError"); 862141be8a5SBram Moolenaar p3imp_PyExc_OverflowError = PyDict_GetItemString(exdict, "OverflowError"); 863bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_AttributeError); 864bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_IndexError); 865af6abb9dSBram Moolenaar Py_XINCREF(p3imp_PyExc_KeyError); 866bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_KeyboardInterrupt); 867bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_TypeError); 868bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_ValueError); 86941009374SBram Moolenaar Py_XINCREF(p3imp_PyExc_SystemExit); 8708661b178SBram Moolenaar Py_XINCREF(p3imp_PyExc_RuntimeError); 871c09a6d6cSBram Moolenaar Py_XINCREF(p3imp_PyExc_ImportError); 872141be8a5SBram Moolenaar Py_XINCREF(p3imp_PyExc_OverflowError); 873bd5e15fdSBram Moolenaar Py_XDECREF(exmod); 874bd5e15fdSBram Moolenaar } 8752ab2e860SBram Moolenaar #endif // DYNAMIC_PYTHON3 876bd5e15fdSBram Moolenaar 877db913953SBram Moolenaar static int py3initialised = 0; 878db913953SBram Moolenaar #define PYINITIALISED py3initialised 879c4f83380SBram Moolenaar static int python_end_called = FALSE; 880db913953SBram Moolenaar 881774267bbSBram Moolenaar #define DESTRUCTOR_FINISH(self) Py_TYPE(self)->tp_free((PyObject*)self) 882db913953SBram Moolenaar 883971db467SBram Moolenaar #define WIN_PYTHON_REF(win) win->w_python3_ref 884971db467SBram Moolenaar #define BUF_PYTHON_REF(buf) buf->b_python3_ref 8855e538ecdSBram Moolenaar #define TAB_PYTHON_REF(tab) tab->tp_python3_ref 886971db467SBram Moolenaar 887170bf1aeSBram Moolenaar static void 888170bf1aeSBram Moolenaar call_PyObject_Free(void *p) 889bd5e15fdSBram Moolenaar { 8900014a53aSBram Moolenaar #if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 891bd5e15fdSBram Moolenaar _PyObject_DebugFree(p); 892bd5e15fdSBram Moolenaar #else 893bd5e15fdSBram Moolenaar PyObject_Free(p); 894bd5e15fdSBram Moolenaar #endif 895bd5e15fdSBram Moolenaar } 896170bf1aeSBram Moolenaar 897170bf1aeSBram Moolenaar static PyObject * 898170bf1aeSBram Moolenaar call_PyType_GenericNew(PyTypeObject *type, PyObject *args, PyObject *kwds) 899bd5e15fdSBram Moolenaar { 900bd5e15fdSBram Moolenaar return PyType_GenericNew(type,args,kwds); 901bd5e15fdSBram Moolenaar } 902170bf1aeSBram Moolenaar 903170bf1aeSBram Moolenaar static PyObject * 904170bf1aeSBram Moolenaar call_PyType_GenericAlloc(PyTypeObject *type, Py_ssize_t nitems) 905bd5e15fdSBram Moolenaar { 906bd5e15fdSBram Moolenaar return PyType_GenericAlloc(type,nitems); 907bd5e15fdSBram Moolenaar } 908bd5e15fdSBram Moolenaar 9094d1da49cSBram Moolenaar static PyObject *OutputGetattro(PyObject *, PyObject *); 9104d1da49cSBram Moolenaar static int OutputSetattro(PyObject *, PyObject *, PyObject *); 9114d1da49cSBram Moolenaar static PyObject *BufferGetattro(PyObject *, PyObject *); 912e9ba516bSBram Moolenaar static int BufferSetattro(PyObject *, PyObject *, PyObject *); 9135e538ecdSBram Moolenaar static PyObject *TabPageGetattro(PyObject *, PyObject *); 9144d1da49cSBram Moolenaar static PyObject *WindowGetattro(PyObject *, PyObject *); 9154d1da49cSBram Moolenaar static int WindowSetattro(PyObject *, PyObject *, PyObject *); 9164d1da49cSBram Moolenaar static PyObject *RangeGetattro(PyObject *, PyObject *); 9174d1da49cSBram Moolenaar static PyObject *CurrentGetattro(PyObject *, PyObject *); 9184d1da49cSBram Moolenaar static int CurrentSetattro(PyObject *, PyObject *, PyObject *); 9194d1da49cSBram Moolenaar static PyObject *DictionaryGetattro(PyObject *, PyObject *); 9204d1da49cSBram Moolenaar static int DictionarySetattro(PyObject *, PyObject *, PyObject *); 9214d1da49cSBram Moolenaar static PyObject *ListGetattro(PyObject *, PyObject *); 9224d1da49cSBram Moolenaar static int ListSetattro(PyObject *, PyObject *, PyObject *); 9234d1da49cSBram Moolenaar static PyObject *FunctionGetattro(PyObject *, PyObject *); 9244d1da49cSBram Moolenaar 9254d1da49cSBram Moolenaar static struct PyModuleDef vimmodule; 9264d1da49cSBram Moolenaar 9272a0f3d3fSBram Moolenaar #define PY_CAN_RECURSE 9282a0f3d3fSBram Moolenaar 9294d1da49cSBram Moolenaar /* 9304d1da49cSBram Moolenaar * Include the code shared with if_python.c 9314d1da49cSBram Moolenaar */ 9324d1da49cSBram Moolenaar #include "if_py_both.h" 9334d1da49cSBram Moolenaar 934828bff1fSBram Moolenaar // NOTE: Must always be used at the start of a block, since it declares "name". 9354d1da49cSBram Moolenaar #define GET_ATTR_STRING(name, nameobj) \ 9364d1da49cSBram Moolenaar char *name = ""; \ 9374d1da49cSBram Moolenaar if (PyUnicode_Check(nameobj)) \ 938828bff1fSBram Moolenaar name = (char *)_PyUnicode_AsString(nameobj) 9394d1da49cSBram Moolenaar 9404d1da49cSBram Moolenaar #define PY3OBJ_DELETED(obj) (obj->ob_base.ob_refcnt<=0) 9414d1da49cSBram Moolenaar 9422ab2e860SBram Moolenaar /////////////////////////////////////////////////////// 9432ab2e860SBram Moolenaar // Internal function prototypes. 944bd5e15fdSBram Moolenaar 9457854e3abSBram Moolenaar static PyObject *Py3Init_vim(void); 946bd5e15fdSBram Moolenaar 9472ab2e860SBram Moolenaar /////////////////////////////////////////////////////// 9482ab2e860SBram Moolenaar // 1. Python interpreter main program. 949bd5e15fdSBram Moolenaar 950170bf1aeSBram Moolenaar void 95168c2f638SBram Moolenaar python3_end(void) 952bd5e15fdSBram Moolenaar { 953bd5e15fdSBram Moolenaar static int recurse = 0; 954bd5e15fdSBram Moolenaar 9552ab2e860SBram Moolenaar // If a crash occurs while doing this, don't try again. 956bd5e15fdSBram Moolenaar if (recurse != 0) 957bd5e15fdSBram Moolenaar return; 958bd5e15fdSBram Moolenaar 959c4f83380SBram Moolenaar python_end_called = TRUE; 960bd5e15fdSBram Moolenaar ++recurse; 961bd5e15fdSBram Moolenaar 962bd5e15fdSBram Moolenaar #ifdef DYNAMIC_PYTHON3 963bd5e15fdSBram Moolenaar if (hinstPy3) 964bd5e15fdSBram Moolenaar #endif 965bd5e15fdSBram Moolenaar if (Py_IsInitialized()) 966bd5e15fdSBram Moolenaar { 9672ab2e860SBram Moolenaar // acquire lock before finalizing 96871700b89SBram Moolenaar PyGILState_Ensure(); 969bd5e15fdSBram Moolenaar 970bd5e15fdSBram Moolenaar Py_Finalize(); 971bd5e15fdSBram Moolenaar } 972bd5e15fdSBram Moolenaar 973bd5e15fdSBram Moolenaar --recurse; 974bd5e15fdSBram Moolenaar } 975bd5e15fdSBram Moolenaar 976094454faSBram Moolenaar #if (defined(DYNAMIC_PYTHON3) && defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON) && defined(UNIX)) || defined(PROTO) 9774c3a326cSBram Moolenaar int 97868c2f638SBram Moolenaar python3_loaded(void) 9794c3a326cSBram Moolenaar { 9804c3a326cSBram Moolenaar return (hinstPy3 != 0); 9814c3a326cSBram Moolenaar } 9824c3a326cSBram Moolenaar #endif 9834c3a326cSBram Moolenaar 98494073167SBram Moolenaar static wchar_t *py_home_buf = NULL; 98594073167SBram Moolenaar 98656b8dc33SBram Moolenaar #if defined(MSWIN) && (PY_VERSION_HEX >= 0x030500f0) 987c6ed254dSBram Moolenaar /* 988c6ed254dSBram Moolenaar * Return TRUE if stdin is readable from Python 3. 989c6ed254dSBram Moolenaar */ 990c6ed254dSBram Moolenaar static BOOL 991c6ed254dSBram Moolenaar is_stdin_readable(void) 992c6ed254dSBram Moolenaar { 993c6ed254dSBram Moolenaar DWORD mode, eventnum; 994c6ed254dSBram Moolenaar struct _stat st; 995c6ed254dSBram Moolenaar int fd = fileno(stdin); 996c6ed254dSBram Moolenaar HANDLE hstdin = (HANDLE)_get_osfhandle(fd); 997c6ed254dSBram Moolenaar 998c6ed254dSBram Moolenaar // Check if stdin is connected to the console. 999c6ed254dSBram Moolenaar if (GetConsoleMode(hstdin, &mode)) 1000c6ed254dSBram Moolenaar // Check if it is opened as input. 1001c6ed254dSBram Moolenaar return GetNumberOfConsoleInputEvents(hstdin, &eventnum); 1002c6ed254dSBram Moolenaar 1003c6ed254dSBram Moolenaar return _fstat(fd, &st) == 0; 1004c6ed254dSBram Moolenaar } 1005c6ed254dSBram Moolenaar 1006c6ed254dSBram Moolenaar // Python 3.5 or later will abort inside Py_Initialize() when stdin has 1007c6ed254dSBram Moolenaar // been closed (i.e. executed by "vim -"). Reconnect stdin to CONIN$. 100856b8dc33SBram Moolenaar // Note that the python DLL is linked to its own stdio DLL which can be 100956b8dc33SBram Moolenaar // differ from Vim's stdio. 101056b8dc33SBram Moolenaar static void 101156b8dc33SBram Moolenaar reset_stdin(void) 101256b8dc33SBram Moolenaar { 101356b8dc33SBram Moolenaar FILE *(*py__acrt_iob_func)(unsigned) = NULL; 101456b8dc33SBram Moolenaar FILE *(*pyfreopen)(const char *, const char *, FILE *) = NULL; 101556b8dc33SBram Moolenaar HINSTANCE hinst; 101656b8dc33SBram Moolenaar 101756b8dc33SBram Moolenaar # ifdef DYNAMIC_PYTHON3 101856b8dc33SBram Moolenaar hinst = hinstPy3; 101956b8dc33SBram Moolenaar # else 102056b8dc33SBram Moolenaar hinst = GetModuleHandle(PYTHON3_DLL); 102156b8dc33SBram Moolenaar # endif 1022c6ed254dSBram Moolenaar if (hinst == NULL || is_stdin_readable()) 102356b8dc33SBram Moolenaar return; 102456b8dc33SBram Moolenaar 102556b8dc33SBram Moolenaar // Get "freopen" and "stdin" which are used in the python DLL. 102656b8dc33SBram Moolenaar // "stdin" is defined as "__acrt_iob_func(0)" in VC++ 2015 or later. 102756b8dc33SBram Moolenaar py__acrt_iob_func = get_dll_import_func(hinst, "__acrt_iob_func"); 102856b8dc33SBram Moolenaar if (py__acrt_iob_func) 102956b8dc33SBram Moolenaar { 103056b8dc33SBram Moolenaar HINSTANCE hpystdiodll = find_imported_module_by_funcname(hinst, 103156b8dc33SBram Moolenaar "__acrt_iob_func"); 103256b8dc33SBram Moolenaar if (hpystdiodll) 103356b8dc33SBram Moolenaar pyfreopen = (void *)GetProcAddress(hpystdiodll, "freopen"); 103456b8dc33SBram Moolenaar } 103556b8dc33SBram Moolenaar 1036c6ed254dSBram Moolenaar // Reconnect stdin to CONIN$. 1037253b16a4SBram Moolenaar if (pyfreopen != NULL) 1038c6ed254dSBram Moolenaar pyfreopen("CONIN$", "r", py__acrt_iob_func(0)); 103956b8dc33SBram Moolenaar else 1040c6ed254dSBram Moolenaar freopen("CONIN$", "r", stdin); 104156b8dc33SBram Moolenaar } 104256b8dc33SBram Moolenaar #else 104356b8dc33SBram Moolenaar # define reset_stdin() 104456b8dc33SBram Moolenaar #endif 104556b8dc33SBram Moolenaar 1046170bf1aeSBram Moolenaar static int 1047170bf1aeSBram Moolenaar Python3_Init(void) 1048bd5e15fdSBram Moolenaar { 1049bd5e15fdSBram Moolenaar if (!py3initialised) 1050bd5e15fdSBram Moolenaar { 1051bd5e15fdSBram Moolenaar #ifdef DYNAMIC_PYTHON3 1052bd5e15fdSBram Moolenaar if (!python3_enabled(TRUE)) 1053bd5e15fdSBram Moolenaar { 1054f9e3e09fSBram Moolenaar emsg(_("E263: Sorry, this command is disabled, the Python library could not be loaded.")); 1055bd5e15fdSBram Moolenaar goto fail; 1056bd5e15fdSBram Moolenaar } 1057bd5e15fdSBram Moolenaar #endif 1058bd5e15fdSBram Moolenaar 1059bd5e15fdSBram Moolenaar init_structs(); 1060bd5e15fdSBram Moolenaar 106194073167SBram Moolenaar if (*p_py3home != NUL) 106294073167SBram Moolenaar { 106394073167SBram Moolenaar size_t len = mbstowcs(NULL, (char *)p_py3home, 0) + 1; 1064644d37b8SBram Moolenaar 10652ab2e860SBram Moolenaar // The string must not change later, make a copy in static memory. 1066c799fe20SBram Moolenaar py_home_buf = ALLOC_MULT(wchar_t, len); 106794073167SBram Moolenaar if (py_home_buf != NULL && mbstowcs( 106894073167SBram Moolenaar py_home_buf, (char *)p_py3home, len) != (size_t)-1) 106994073167SBram Moolenaar Py_SetPythonHome(py_home_buf); 107094073167SBram Moolenaar } 1071644d37b8SBram Moolenaar #ifdef PYTHON3_HOME 107294073167SBram Moolenaar else if (mch_getenv((char_u *)"PYTHONHOME") == NULL) 1073644d37b8SBram Moolenaar Py_SetPythonHome(PYTHON3_HOME); 1074644d37b8SBram Moolenaar #endif 1075644d37b8SBram Moolenaar 10767bc4f93cSBram Moolenaar PyImport_AppendInittab("vim", Py3Init_vim); 10777bc4f93cSBram Moolenaar 107856b8dc33SBram Moolenaar reset_stdin(); 1079bd5e15fdSBram Moolenaar Py_Initialize(); 1080d057301bSBram Moolenaar 1081efc0d94aSBram Moolenaar #if PY_VERSION_HEX < 0x03090000 1082efc0d94aSBram Moolenaar // Initialise threads. This is deprecated since Python 3.9. 1083456f2bb2SBram Moolenaar PyEval_InitThreads(); 1084efc0d94aSBram Moolenaar #endif 1085bd5e15fdSBram Moolenaar #ifdef DYNAMIC_PYTHON3 1086bd5e15fdSBram Moolenaar get_py3_exceptions(); 1087bd5e15fdSBram Moolenaar #endif 1088bd5e15fdSBram Moolenaar 10891dc28783SBram Moolenaar if (PythonIO_Init_io()) 1090bd5e15fdSBram Moolenaar goto fail; 1091bd5e15fdSBram Moolenaar 1092db913953SBram Moolenaar globals = PyModule_GetDict(PyImport_AddModule("__main__")); 1093db913953SBram Moolenaar 10942ab2e860SBram Moolenaar // Remove the element from sys.path that was added because of our 10952ab2e860SBram Moolenaar // argv[0] value in Py3Init_vim(). Previously we used an empty 10962ab2e860SBram Moolenaar // string, but depending on the OS we then get an empty entry or 10972ab2e860SBram Moolenaar // the current directory in sys.path. 10982ab2e860SBram Moolenaar // Only after vim has been imported, the element does exist in 10992ab2e860SBram Moolenaar // sys.path. 110019e60943SBram Moolenaar PyRun_SimpleString("import vim; import sys; sys.path = list(filter(lambda x: not x.endswith('must>not&exist'), sys.path))"); 1101bd5e15fdSBram Moolenaar 1102efc0d94aSBram Moolenaar // Without the call to PyEval_SaveThread, thread specific state (such 1103efc0d94aSBram Moolenaar // as the system trace hook), will be lost between invocations of 1104efc0d94aSBram Moolenaar // Python code. 1105efc0d94aSBram Moolenaar // GIL may have been created and acquired in PyEval_InitThreads() and 1106efc0d94aSBram Moolenaar // thread state is created in Py_Initialize(); there 1107efc0d94aSBram Moolenaar // _PyGILState_NoteThreadState() also sets gilcounter to 1 (python must 1108efc0d94aSBram Moolenaar // have threads enabled!), so the following does both: unlock GIL and 1109efc0d94aSBram Moolenaar // save thread state in TLS without deleting thread state 111076d711c3SBram Moolenaar PyEval_SaveThread(); 1111bd5e15fdSBram Moolenaar 1112bd5e15fdSBram Moolenaar py3initialised = 1; 1113bd5e15fdSBram Moolenaar } 1114bd5e15fdSBram Moolenaar 1115bd5e15fdSBram Moolenaar return 0; 1116bd5e15fdSBram Moolenaar 1117bd5e15fdSBram Moolenaar fail: 11182ab2e860SBram Moolenaar // We call PythonIO_Flush() here to print any Python errors. 11192ab2e860SBram Moolenaar // This is OK, as it is possible to call this function even 11202ab2e860SBram Moolenaar // if PythonIO_Init_io() has not completed successfully (it will 11212ab2e860SBram Moolenaar // not do anything in this case). 1122bd5e15fdSBram Moolenaar PythonIO_Flush(); 1123bd5e15fdSBram Moolenaar return -1; 1124bd5e15fdSBram Moolenaar } 1125bd5e15fdSBram Moolenaar 1126bd5e15fdSBram Moolenaar /* 1127bd5e15fdSBram Moolenaar * External interface 1128bd5e15fdSBram Moolenaar */ 1129170bf1aeSBram Moolenaar static void 1130b52f4c02SBram Moolenaar DoPyCommand(const char *cmd, rangeinitializer init_range, runner run, void *arg) 1131bd5e15fdSBram Moolenaar { 1132bd5e15fdSBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 1133bd5e15fdSBram Moolenaar char *saved_locale; 1134bd5e15fdSBram Moolenaar #endif 113519e60943SBram Moolenaar PyObject *cmdstr; 113619e60943SBram Moolenaar PyObject *cmdbytes; 113771700b89SBram Moolenaar PyGILState_STATE pygilstate; 1138bd5e15fdSBram Moolenaar 1139c4f83380SBram Moolenaar if (python_end_called) 1140c4f83380SBram Moolenaar goto theend; 1141c4f83380SBram Moolenaar 1142bd5e15fdSBram Moolenaar if (Python3_Init()) 1143bd5e15fdSBram Moolenaar goto theend; 1144bd5e15fdSBram Moolenaar 1145b52f4c02SBram Moolenaar init_range(arg); 1146b52f4c02SBram Moolenaar 11472ab2e860SBram Moolenaar Python_Release_Vim(); // leave Vim 1148bd5e15fdSBram Moolenaar 1149bd5e15fdSBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 11502ab2e860SBram Moolenaar // Python only works properly when the LC_NUMERIC locale is "C". 1151bd5e15fdSBram Moolenaar saved_locale = setlocale(LC_NUMERIC, NULL); 1152bd5e15fdSBram Moolenaar if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0) 1153bd5e15fdSBram Moolenaar saved_locale = NULL; 1154bd5e15fdSBram Moolenaar else 1155bd5e15fdSBram Moolenaar { 11562ab2e860SBram Moolenaar // Need to make a copy, value may change when setting new locale. 1157bd5e15fdSBram Moolenaar saved_locale = (char *)vim_strsave((char_u *)saved_locale); 1158bd5e15fdSBram Moolenaar (void)setlocale(LC_NUMERIC, "C"); 1159bd5e15fdSBram Moolenaar } 1160bd5e15fdSBram Moolenaar #endif 1161bd5e15fdSBram Moolenaar 1162bd5e15fdSBram Moolenaar pygilstate = PyGILState_Ensure(); 1163bd5e15fdSBram Moolenaar 11642ab2e860SBram Moolenaar // PyRun_SimpleString expects a UTF-8 string. Wrong encoding may cause 11652ab2e860SBram Moolenaar // SyntaxError (unicode error). 11663d64a317SBram Moolenaar cmdstr = PyUnicode_Decode(cmd, strlen(cmd), 11672e2f52a4SBram Moolenaar (char *)ENC_OPT, ERRORS_DECODE_ARG); 11682e2f52a4SBram Moolenaar cmdbytes = PyUnicode_AsEncodedString(cmdstr, "utf-8", ERRORS_ENCODE_ARG); 116919e60943SBram Moolenaar Py_XDECREF(cmdstr); 1170db913953SBram Moolenaar 1171b52f4c02SBram Moolenaar run(PyBytes_AsString(cmdbytes), arg, &pygilstate); 117219e60943SBram Moolenaar Py_XDECREF(cmdbytes); 1173bd5e15fdSBram Moolenaar 1174bd5e15fdSBram Moolenaar PyGILState_Release(pygilstate); 1175bd5e15fdSBram Moolenaar 1176bd5e15fdSBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 1177bd5e15fdSBram Moolenaar if (saved_locale != NULL) 1178bd5e15fdSBram Moolenaar { 1179bd5e15fdSBram Moolenaar (void)setlocale(LC_NUMERIC, saved_locale); 1180bd5e15fdSBram Moolenaar vim_free(saved_locale); 1181bd5e15fdSBram Moolenaar } 1182bd5e15fdSBram Moolenaar #endif 1183bd5e15fdSBram Moolenaar 11842ab2e860SBram Moolenaar Python_Lock_Vim(); // enter Vim 1185bd5e15fdSBram Moolenaar PythonIO_Flush(); 1186bd5e15fdSBram Moolenaar 1187bd5e15fdSBram Moolenaar theend: 11882ab2e860SBram Moolenaar return; // keeps lint happy 1189bd5e15fdSBram Moolenaar } 1190bd5e15fdSBram Moolenaar 1191bd5e15fdSBram Moolenaar /* 1192368373e9SBram Moolenaar * ":py3" 1193bd5e15fdSBram Moolenaar */ 1194170bf1aeSBram Moolenaar void 1195170bf1aeSBram Moolenaar ex_py3(exarg_T *eap) 1196bd5e15fdSBram Moolenaar { 1197bd5e15fdSBram Moolenaar char_u *script; 1198bd5e15fdSBram Moolenaar 1199bd5e15fdSBram Moolenaar script = script_get(eap, eap->arg); 1200bd5e15fdSBram Moolenaar if (!eap->skip) 1201bd5e15fdSBram Moolenaar { 120214816ad6SBram Moolenaar if (p_pyx == 0) 120314816ad6SBram Moolenaar p_pyx = 3; 120414816ad6SBram Moolenaar 1205b52f4c02SBram Moolenaar DoPyCommand(script == NULL ? (char *) eap->arg : (char *) script, 1206b52f4c02SBram Moolenaar (rangeinitializer) init_range_cmd, 1207b52f4c02SBram Moolenaar (runner) run_cmd, 1208b52f4c02SBram Moolenaar (void *) eap); 1209bd5e15fdSBram Moolenaar } 1210bd5e15fdSBram Moolenaar vim_free(script); 1211bd5e15fdSBram Moolenaar } 1212bd5e15fdSBram Moolenaar 1213bd5e15fdSBram Moolenaar #define BUFFER_SIZE 2048 1214bd5e15fdSBram Moolenaar 1215bd5e15fdSBram Moolenaar /* 12166df6f47dSBram Moolenaar * ":py3file" 1217bd5e15fdSBram Moolenaar */ 1218bd5e15fdSBram Moolenaar void 1219bd5e15fdSBram Moolenaar ex_py3file(exarg_T *eap) 1220bd5e15fdSBram Moolenaar { 1221bd5e15fdSBram Moolenaar static char buffer[BUFFER_SIZE]; 1222bd5e15fdSBram Moolenaar const char *file; 1223bd5e15fdSBram Moolenaar char *p; 1224bd5e15fdSBram Moolenaar int i; 1225bd5e15fdSBram Moolenaar 1226f42dd3c3SBram Moolenaar if (p_pyx == 0) 1227f42dd3c3SBram Moolenaar p_pyx = 3; 1228f42dd3c3SBram Moolenaar 12292ab2e860SBram Moolenaar // Have to do it like this. PyRun_SimpleFile requires you to pass a 12302ab2e860SBram Moolenaar // stdio file pointer, but Vim and the Python DLL are compiled with 12312ab2e860SBram Moolenaar // different options under Windows, meaning that stdio pointers aren't 12322ab2e860SBram Moolenaar // compatible between the two. Yuk. 12332ab2e860SBram Moolenaar // 12342ab2e860SBram Moolenaar // construct: exec(compile(open('a_filename', 'rb').read(), 'a_filename', 'exec')) 12352ab2e860SBram Moolenaar // 12362ab2e860SBram Moolenaar // Using bytes so that Python can detect the source encoding as it normally 12372ab2e860SBram Moolenaar // does. The doc does not say "compile" accept bytes, though. 12382ab2e860SBram Moolenaar // 12392ab2e860SBram Moolenaar // We need to escape any backslashes or single quotes in the file name, so that 12402ab2e860SBram Moolenaar // Python won't mangle the file name. 1241bd5e15fdSBram Moolenaar 1242bd5e15fdSBram Moolenaar strcpy(buffer, "exec(compile(open('"); 12432ab2e860SBram Moolenaar p = buffer + 19; // size of "exec(compile(open('" 1244bd5e15fdSBram Moolenaar 1245bd5e15fdSBram Moolenaar for (i=0; i<2; ++i) 1246bd5e15fdSBram Moolenaar { 1247bd5e15fdSBram Moolenaar file = (char *)eap->arg; 1248bd5e15fdSBram Moolenaar while (*file && p < buffer + (BUFFER_SIZE - 3)) 1249bd5e15fdSBram Moolenaar { 1250bd5e15fdSBram Moolenaar if (*file == '\\' || *file == '\'') 1251bd5e15fdSBram Moolenaar *p++ = '\\'; 1252bd5e15fdSBram Moolenaar *p++ = *file++; 1253bd5e15fdSBram Moolenaar } 12542ab2e860SBram Moolenaar // If we didn't finish the file name, we hit a buffer overflow 1255bd5e15fdSBram Moolenaar if (*file != '\0') 1256bd5e15fdSBram Moolenaar return; 1257bd5e15fdSBram Moolenaar if (i==0) 1258bd5e15fdSBram Moolenaar { 125919e60943SBram Moolenaar strcpy(p,"','rb').read(),'"); 126019e60943SBram Moolenaar p += 16; 1261bd5e15fdSBram Moolenaar } 1262bd5e15fdSBram Moolenaar else 1263bd5e15fdSBram Moolenaar { 1264bd5e15fdSBram Moolenaar strcpy(p,"','exec'))"); 1265bd5e15fdSBram Moolenaar p += 10; 1266bd5e15fdSBram Moolenaar } 1267bd5e15fdSBram Moolenaar } 1268bd5e15fdSBram Moolenaar 1269bd5e15fdSBram Moolenaar 12702ab2e860SBram Moolenaar // Execute the file 1271b52f4c02SBram Moolenaar DoPyCommand(buffer, 1272b52f4c02SBram Moolenaar (rangeinitializer) init_range_cmd, 1273b52f4c02SBram Moolenaar (runner) run_cmd, 1274b52f4c02SBram Moolenaar (void *) eap); 1275bd5e15fdSBram Moolenaar } 1276bd5e15fdSBram Moolenaar 127754e8f005SBram Moolenaar void 127854e8f005SBram Moolenaar ex_py3do(exarg_T *eap) 12793dab2806SBram Moolenaar { 1280f42dd3c3SBram Moolenaar if (p_pyx == 0) 1281f42dd3c3SBram Moolenaar p_pyx = 3; 1282f42dd3c3SBram Moolenaar 1283b52f4c02SBram Moolenaar DoPyCommand((char *)eap->arg, 1284b52f4c02SBram Moolenaar (rangeinitializer)init_range_cmd, 1285b52f4c02SBram Moolenaar (runner)run_do, 1286b52f4c02SBram Moolenaar (void *)eap); 12873dab2806SBram Moolenaar } 12883dab2806SBram Moolenaar 12892ab2e860SBram Moolenaar /////////////////////////////////////////////////////// 12902ab2e860SBram Moolenaar // 2. Python output stream: writes output via [e]msg(). 1291bd5e15fdSBram Moolenaar 12922ab2e860SBram Moolenaar // Implementation functions 1293bd5e15fdSBram Moolenaar 1294170bf1aeSBram Moolenaar static PyObject * 1295170bf1aeSBram Moolenaar OutputGetattro(PyObject *self, PyObject *nameobj) 1296bd5e15fdSBram Moolenaar { 129777045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1298bd5e15fdSBram Moolenaar 1299bd5e15fdSBram Moolenaar if (strcmp(name, "softspace") == 0) 1300bd5e15fdSBram Moolenaar return PyLong_FromLong(((OutputObject *)(self))->softspace); 13016d4431e7SBram Moolenaar else if (strcmp(name, "errors") == 0) 13026d4431e7SBram Moolenaar return PyString_FromString("strict"); 13036d4431e7SBram Moolenaar else if (strcmp(name, "encoding") == 0) 13046d4431e7SBram Moolenaar return PyString_FromString(ENC_OPT); 1305bd5e15fdSBram Moolenaar 1306bd5e15fdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1307bd5e15fdSBram Moolenaar } 1308bd5e15fdSBram Moolenaar 1309170bf1aeSBram Moolenaar static int 1310170bf1aeSBram Moolenaar OutputSetattro(PyObject *self, PyObject *nameobj, PyObject *val) 1311bd5e15fdSBram Moolenaar { 131277045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1313bd5e15fdSBram Moolenaar 1314d6e39186SBram Moolenaar return OutputSetattr((OutputObject *)(self), name, val); 1315bd5e15fdSBram Moolenaar } 1316bd5e15fdSBram Moolenaar 13172ab2e860SBram Moolenaar /////////////////////////////////////////////////////// 13182ab2e860SBram Moolenaar // 3. Implementation of the Vim module for Python 1319bd5e15fdSBram Moolenaar 13202ab2e860SBram Moolenaar // Window type - Implementation functions 13212ab2e860SBram Moolenaar // -------------------------------------- 1322bd5e15fdSBram Moolenaar 1323bd5e15fdSBram Moolenaar #define WindowType_Check(obj) ((obj)->ob_base.ob_type == &WindowType) 1324bd5e15fdSBram Moolenaar 13252ab2e860SBram Moolenaar // Buffer type - Implementation functions 13262ab2e860SBram Moolenaar // -------------------------------------- 1327bd5e15fdSBram Moolenaar 1328bd5e15fdSBram Moolenaar #define BufferType_Check(obj) ((obj)->ob_base.ob_type == &BufferType) 1329bd5e15fdSBram Moolenaar 1330bd5e15fdSBram Moolenaar static PyObject* BufferSubscript(PyObject *self, PyObject *idx); 13314ce5fe4cSBram Moolenaar static int BufferAsSubscript(PyObject *self, PyObject *idx, PyObject *val); 1332bd5e15fdSBram Moolenaar 13332ab2e860SBram Moolenaar // Line range type - Implementation functions 13342ab2e860SBram Moolenaar // -------------------------------------- 1335bd5e15fdSBram Moolenaar 1336bd5e15fdSBram Moolenaar #define RangeType_Check(obj) ((obj)->ob_base.ob_type == &RangeType) 1337bd5e15fdSBram Moolenaar 1338bd5e15fdSBram Moolenaar static PyObject* RangeSubscript(PyObject *self, PyObject *idx); 13394ce5fe4cSBram Moolenaar static int RangeAsItem(PyObject *, Py_ssize_t, PyObject *); 13404ce5fe4cSBram Moolenaar static int RangeAsSubscript(PyObject *self, PyObject *idx, PyObject *val); 1341bd5e15fdSBram Moolenaar 13422ab2e860SBram Moolenaar // Current objects type - Implementation functions 13432ab2e860SBram Moolenaar // ----------------------------------------------- 1344bd5e15fdSBram Moolenaar 1345bd5e15fdSBram Moolenaar static PySequenceMethods BufferAsSeq = { 13462ab2e860SBram Moolenaar (lenfunc) BufferLength, // sq_length, len(x) 13472ab2e860SBram Moolenaar (binaryfunc) 0, // sq_concat, x+y 13482ab2e860SBram Moolenaar (ssizeargfunc) 0, // sq_repeat, x*n 13492ab2e860SBram Moolenaar (ssizeargfunc) BufferItem, // sq_item, x[i] 13502ab2e860SBram Moolenaar 0, // was_sq_slice, x[i:j] 13512ab2e860SBram Moolenaar 0, // sq_ass_item, x[i]=v 13522ab2e860SBram Moolenaar 0, // sq_ass_slice, x[i:j]=v 13532ab2e860SBram Moolenaar 0, // sq_contains 13542ab2e860SBram Moolenaar 0, // sq_inplace_concat 13552ab2e860SBram Moolenaar 0, // sq_inplace_repeat 1356bd5e15fdSBram Moolenaar }; 1357bd5e15fdSBram Moolenaar 13584d1da49cSBram Moolenaar static PyMappingMethods BufferAsMapping = { 1359bd5e15fdSBram Moolenaar /* mp_length */ (lenfunc)BufferLength, 1360bd5e15fdSBram Moolenaar /* mp_subscript */ (binaryfunc)BufferSubscript, 136119e60943SBram Moolenaar /* mp_ass_subscript */ (objobjargproc)BufferAsSubscript, 1362bd5e15fdSBram Moolenaar }; 1363bd5e15fdSBram Moolenaar 1364bd5e15fdSBram Moolenaar 13652ab2e860SBram Moolenaar // Buffer object 1366bd5e15fdSBram Moolenaar 1367170bf1aeSBram Moolenaar static PyObject * 1368170bf1aeSBram Moolenaar BufferGetattro(PyObject *self, PyObject *nameobj) 1369bd5e15fdSBram Moolenaar { 13704d1da49cSBram Moolenaar PyObject *r; 1371bd5e15fdSBram Moolenaar 137277045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1373bd5e15fdSBram Moolenaar 13749e822c00SBram Moolenaar if ((r = BufferAttrValid((BufferObject *)(self), name))) 13759e822c00SBram Moolenaar return r; 13769e822c00SBram Moolenaar 13774d1da49cSBram Moolenaar if (CheckBuffer((BufferObject *)(self))) 1378bd5e15fdSBram Moolenaar return NULL; 1379bd5e15fdSBram Moolenaar 13804d1da49cSBram Moolenaar r = BufferAttr((BufferObject *)(self), name); 13814d1da49cSBram Moolenaar if (r || PyErr_Occurred()) 13824d1da49cSBram Moolenaar return r; 1383bd5e15fdSBram Moolenaar else 1384bd5e15fdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1385bd5e15fdSBram Moolenaar } 1386bd5e15fdSBram Moolenaar 1387e9ba516bSBram Moolenaar static int 1388e9ba516bSBram Moolenaar BufferSetattro(PyObject *self, PyObject *nameobj, PyObject *val) 1389e9ba516bSBram Moolenaar { 1390e9ba516bSBram Moolenaar GET_ATTR_STRING(name, nameobj); 1391e9ba516bSBram Moolenaar 1392e9ba516bSBram Moolenaar return BufferSetattr((BufferObject *)(self), name, val); 1393e9ba516bSBram Moolenaar } 1394e9ba516bSBram Moolenaar 13952ab2e860SBram Moolenaar ////////////////// 1396bd5e15fdSBram Moolenaar 1397170bf1aeSBram Moolenaar static PyObject * 1398170bf1aeSBram Moolenaar BufferSubscript(PyObject *self, PyObject* idx) 1399bd5e15fdSBram Moolenaar { 1400db913953SBram Moolenaar if (PyLong_Check(idx)) 1401db913953SBram Moolenaar { 1402bd5e15fdSBram Moolenaar long _idx = PyLong_AsLong(idx); 1403d6e39186SBram Moolenaar return BufferItem((BufferObject *)(self), _idx); 1404db913953SBram Moolenaar } else if (PySlice_Check(idx)) 1405db913953SBram Moolenaar { 1406bd5e15fdSBram Moolenaar Py_ssize_t start, stop, step, slicelen; 1407bd5e15fdSBram Moolenaar 14088f1723deSBram Moolenaar if (CheckBuffer((BufferObject *) self)) 14098f1723deSBram Moolenaar return NULL; 14108f1723deSBram Moolenaar 1411922a4664SBram Moolenaar if (PySlice_GetIndicesEx((PySliceObject_T *)idx, 1412bd80f35bSBram Moolenaar (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count, 1413bd5e15fdSBram Moolenaar &start, &stop, 1414db913953SBram Moolenaar &step, &slicelen) < 0) 1415bd5e15fdSBram Moolenaar return NULL; 1416d6e39186SBram Moolenaar return BufferSlice((BufferObject *)(self), start, stop); 1417db913953SBram Moolenaar } 1418db913953SBram Moolenaar else 1419db913953SBram Moolenaar { 1420c476e52fSBram Moolenaar RAISE_INVALID_INDEX_TYPE(idx); 1421bd5e15fdSBram Moolenaar return NULL; 1422bd5e15fdSBram Moolenaar } 1423bd5e15fdSBram Moolenaar } 1424bd5e15fdSBram Moolenaar 14254ce5fe4cSBram Moolenaar static int 142619e60943SBram Moolenaar BufferAsSubscript(PyObject *self, PyObject* idx, PyObject* val) 142719e60943SBram Moolenaar { 1428db913953SBram Moolenaar if (PyLong_Check(idx)) 1429db913953SBram Moolenaar { 143019e60943SBram Moolenaar long n = PyLong_AsLong(idx); 1431ab589463SBram Moolenaar 1432ab589463SBram Moolenaar if (CheckBuffer((BufferObject *) self)) 1433ab589463SBram Moolenaar return -1; 1434ab589463SBram Moolenaar 143519e60943SBram Moolenaar return RBAsItem((BufferObject *)(self), n, val, 1, 143619e60943SBram Moolenaar (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count, 143719e60943SBram Moolenaar NULL); 1438db913953SBram Moolenaar } else if (PySlice_Check(idx)) 1439db913953SBram Moolenaar { 144019e60943SBram Moolenaar Py_ssize_t start, stop, step, slicelen; 144119e60943SBram Moolenaar 14428f1723deSBram Moolenaar if (CheckBuffer((BufferObject *) self)) 14438f1723deSBram Moolenaar return -1; 14448f1723deSBram Moolenaar 1445922a4664SBram Moolenaar if (PySlice_GetIndicesEx((PySliceObject_T *)idx, 1446bd80f35bSBram Moolenaar (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count, 144719e60943SBram Moolenaar &start, &stop, 1448db913953SBram Moolenaar &step, &slicelen) < 0) 144919e60943SBram Moolenaar return -1; 145019e60943SBram Moolenaar return RBAsSlice((BufferObject *)(self), start, stop, val, 1, 145119e60943SBram Moolenaar (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count, 145219e60943SBram Moolenaar NULL); 1453db913953SBram Moolenaar } 1454db913953SBram Moolenaar else 1455db913953SBram Moolenaar { 1456c476e52fSBram Moolenaar RAISE_INVALID_INDEX_TYPE(idx); 145719e60943SBram Moolenaar return -1; 145819e60943SBram Moolenaar } 145919e60943SBram Moolenaar } 146019e60943SBram Moolenaar 1461bd5e15fdSBram Moolenaar static PySequenceMethods RangeAsSeq = { 14622ab2e860SBram Moolenaar (lenfunc) RangeLength, // sq_length, len(x) 14632ab2e860SBram Moolenaar (binaryfunc) 0, // RangeConcat, sq_concat, x+y 14642ab2e860SBram Moolenaar (ssizeargfunc) 0, // RangeRepeat, sq_repeat, x*n 14652ab2e860SBram Moolenaar (ssizeargfunc) RangeItem, // sq_item, x[i] 14662ab2e860SBram Moolenaar 0, // was_sq_slice, x[i:j] 14672ab2e860SBram Moolenaar (ssizeobjargproc) RangeAsItem, // sq_as_item, x[i]=v 14682ab2e860SBram Moolenaar 0, // sq_ass_slice, x[i:j]=v 14692ab2e860SBram Moolenaar 0, // sq_contains 14702ab2e860SBram Moolenaar 0, // sq_inplace_concat 14712ab2e860SBram Moolenaar 0, // sq_inplace_repeat 1472bd5e15fdSBram Moolenaar }; 1473bd5e15fdSBram Moolenaar 14744d1da49cSBram Moolenaar static PyMappingMethods RangeAsMapping = { 1475bd5e15fdSBram Moolenaar /* mp_length */ (lenfunc)RangeLength, 1476bd5e15fdSBram Moolenaar /* mp_subscript */ (binaryfunc)RangeSubscript, 1477ba4897e6SBram Moolenaar /* mp_ass_subscript */ (objobjargproc)RangeAsSubscript, 1478bd5e15fdSBram Moolenaar }; 1479bd5e15fdSBram Moolenaar 14802ab2e860SBram Moolenaar // Line range object - Implementation 1481bd5e15fdSBram Moolenaar 1482170bf1aeSBram Moolenaar static PyObject * 1483170bf1aeSBram Moolenaar RangeGetattro(PyObject *self, PyObject *nameobj) 1484bd5e15fdSBram Moolenaar { 148577045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1486bd5e15fdSBram Moolenaar 1487bd5e15fdSBram Moolenaar if (strcmp(name, "start") == 0) 1488bd5e15fdSBram Moolenaar return Py_BuildValue("n", ((RangeObject *)(self))->start - 1); 1489bd5e15fdSBram Moolenaar else if (strcmp(name, "end") == 0) 1490bd5e15fdSBram Moolenaar return Py_BuildValue("n", ((RangeObject *)(self))->end - 1); 1491bd5e15fdSBram Moolenaar else 1492bd5e15fdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1493bd5e15fdSBram Moolenaar } 1494bd5e15fdSBram Moolenaar 14952ab2e860SBram Moolenaar //////////////// 1496bd5e15fdSBram Moolenaar 14974ce5fe4cSBram Moolenaar static int 1498170bf1aeSBram Moolenaar RangeAsItem(PyObject *self, Py_ssize_t n, PyObject *val) 1499bd5e15fdSBram Moolenaar { 1500bd5e15fdSBram Moolenaar return RBAsItem(((RangeObject *)(self))->buf, n, val, 1501bd5e15fdSBram Moolenaar ((RangeObject *)(self))->start, 1502bd5e15fdSBram Moolenaar ((RangeObject *)(self))->end, 1503bd5e15fdSBram Moolenaar &((RangeObject *)(self))->end); 1504bd5e15fdSBram Moolenaar } 1505bd5e15fdSBram Moolenaar 1506ba4897e6SBram Moolenaar static Py_ssize_t 1507ba4897e6SBram Moolenaar RangeAsSlice(PyObject *self, Py_ssize_t lo, Py_ssize_t hi, PyObject *val) 1508ba4897e6SBram Moolenaar { 1509ba4897e6SBram Moolenaar return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val, 1510ba4897e6SBram Moolenaar ((RangeObject *)(self))->start, 1511ba4897e6SBram Moolenaar ((RangeObject *)(self))->end, 1512ba4897e6SBram Moolenaar &((RangeObject *)(self))->end); 1513ba4897e6SBram Moolenaar } 1514ba4897e6SBram Moolenaar 1515170bf1aeSBram Moolenaar static PyObject * 1516170bf1aeSBram Moolenaar RangeSubscript(PyObject *self, PyObject* idx) 1517bd5e15fdSBram Moolenaar { 1518db913953SBram Moolenaar if (PyLong_Check(idx)) 1519db913953SBram Moolenaar { 1520bd5e15fdSBram Moolenaar long _idx = PyLong_AsLong(idx); 1521d6e39186SBram Moolenaar return RangeItem((RangeObject *)(self), _idx); 1522db913953SBram Moolenaar } else if (PySlice_Check(idx)) 1523db913953SBram Moolenaar { 1524bd5e15fdSBram Moolenaar Py_ssize_t start, stop, step, slicelen; 1525bd5e15fdSBram Moolenaar 1526922a4664SBram Moolenaar if (PySlice_GetIndicesEx((PySliceObject_T *)idx, 1527bd5e15fdSBram Moolenaar ((RangeObject *)(self))->end-((RangeObject *)(self))->start+1, 1528bd5e15fdSBram Moolenaar &start, &stop, 1529db913953SBram Moolenaar &step, &slicelen) < 0) 1530bd5e15fdSBram Moolenaar return NULL; 1531d6e39186SBram Moolenaar return RangeSlice((RangeObject *)(self), start, stop); 1532db913953SBram Moolenaar } 1533db913953SBram Moolenaar else 1534db913953SBram Moolenaar { 1535c476e52fSBram Moolenaar RAISE_INVALID_INDEX_TYPE(idx); 1536bd5e15fdSBram Moolenaar return NULL; 1537bd5e15fdSBram Moolenaar } 1538bd5e15fdSBram Moolenaar } 1539bd5e15fdSBram Moolenaar 15404ce5fe4cSBram Moolenaar static int 1541ba4897e6SBram Moolenaar RangeAsSubscript(PyObject *self, PyObject *idx, PyObject *val) 1542ba4897e6SBram Moolenaar { 1543db913953SBram Moolenaar if (PyLong_Check(idx)) 1544db913953SBram Moolenaar { 1545ba4897e6SBram Moolenaar long n = PyLong_AsLong(idx); 1546ba4897e6SBram Moolenaar return RangeAsItem(self, n, val); 1547abab0b0fSBram Moolenaar } 1548abab0b0fSBram Moolenaar else if (PySlice_Check(idx)) 1549db913953SBram Moolenaar { 1550ba4897e6SBram Moolenaar Py_ssize_t start, stop, step, slicelen; 1551ba4897e6SBram Moolenaar 1552922a4664SBram Moolenaar if (PySlice_GetIndicesEx((PySliceObject_T *)idx, 1553ba4897e6SBram Moolenaar ((RangeObject *)(self))->end-((RangeObject *)(self))->start+1, 1554ba4897e6SBram Moolenaar &start, &stop, 1555db913953SBram Moolenaar &step, &slicelen) < 0) 1556ba4897e6SBram Moolenaar return -1; 1557ba4897e6SBram Moolenaar return RangeAsSlice(self, start, stop, val); 1558db913953SBram Moolenaar } 1559db913953SBram Moolenaar else 1560db913953SBram Moolenaar { 1561c476e52fSBram Moolenaar RAISE_INVALID_INDEX_TYPE(idx); 1562ba4897e6SBram Moolenaar return -1; 1563ba4897e6SBram Moolenaar } 1564ba4897e6SBram Moolenaar } 1565ba4897e6SBram Moolenaar 15662ab2e860SBram Moolenaar // TabPage object - Implementation 15675e538ecdSBram Moolenaar 15685e538ecdSBram Moolenaar static PyObject * 15695e538ecdSBram Moolenaar TabPageGetattro(PyObject *self, PyObject *nameobj) 15705e538ecdSBram Moolenaar { 15715e538ecdSBram Moolenaar PyObject *r; 15725e538ecdSBram Moolenaar 15735e538ecdSBram Moolenaar GET_ATTR_STRING(name, nameobj); 15745e538ecdSBram Moolenaar 15759e822c00SBram Moolenaar if ((r = TabPageAttrValid((TabPageObject *)(self), name))) 15769e822c00SBram Moolenaar return r; 15779e822c00SBram Moolenaar 15785e538ecdSBram Moolenaar if (CheckTabPage((TabPageObject *)(self))) 15795e538ecdSBram Moolenaar return NULL; 15805e538ecdSBram Moolenaar 15815e538ecdSBram Moolenaar r = TabPageAttr((TabPageObject *)(self), name); 15825e538ecdSBram Moolenaar if (r || PyErr_Occurred()) 15835e538ecdSBram Moolenaar return r; 15845e538ecdSBram Moolenaar else 15855e538ecdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 15865e538ecdSBram Moolenaar } 15875e538ecdSBram Moolenaar 15882ab2e860SBram Moolenaar // Window object - Implementation 1589bd5e15fdSBram Moolenaar 1590170bf1aeSBram Moolenaar static PyObject * 1591170bf1aeSBram Moolenaar WindowGetattro(PyObject *self, PyObject *nameobj) 1592bd5e15fdSBram Moolenaar { 15934d1da49cSBram Moolenaar PyObject *r; 1594bd5e15fdSBram Moolenaar 159577045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1596bd5e15fdSBram Moolenaar 15979e822c00SBram Moolenaar if ((r = WindowAttrValid((WindowObject *)(self), name))) 15989e822c00SBram Moolenaar return r; 15999e822c00SBram Moolenaar 16004d1da49cSBram Moolenaar if (CheckWindow((WindowObject *)(self))) 1601bd5e15fdSBram Moolenaar return NULL; 1602bd5e15fdSBram Moolenaar 16034d1da49cSBram Moolenaar r = WindowAttr((WindowObject *)(self), name); 16044d1da49cSBram Moolenaar if (r || PyErr_Occurred()) 16054d1da49cSBram Moolenaar return r; 1606bd5e15fdSBram Moolenaar else 1607bd5e15fdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1608bd5e15fdSBram Moolenaar } 1609bd5e15fdSBram Moolenaar 1610170bf1aeSBram Moolenaar static int 1611170bf1aeSBram Moolenaar WindowSetattro(PyObject *self, PyObject *nameobj, PyObject *val) 1612bd5e15fdSBram Moolenaar { 161377045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1614bd5e15fdSBram Moolenaar 1615d6e39186SBram Moolenaar return WindowSetattr((WindowObject *)(self), name, val); 1616bd5e15fdSBram Moolenaar } 1617bd5e15fdSBram Moolenaar 16182ab2e860SBram Moolenaar // Tab page list object - Definitions 16195e538ecdSBram Moolenaar 16205e538ecdSBram Moolenaar static PySequenceMethods TabListAsSeq = { 16212ab2e860SBram Moolenaar (lenfunc) TabListLength, // sq_length, len(x) 16222ab2e860SBram Moolenaar (binaryfunc) 0, // sq_concat, x+y 16232ab2e860SBram Moolenaar (ssizeargfunc) 0, // sq_repeat, x*n 16242ab2e860SBram Moolenaar (ssizeargfunc) TabListItem, // sq_item, x[i] 16252ab2e860SBram Moolenaar 0, // sq_slice, x[i:j] 16262ab2e860SBram Moolenaar (ssizeobjargproc)0, // sq_as_item, x[i]=v 16272ab2e860SBram Moolenaar 0, // sq_ass_slice, x[i:j]=v 16282ab2e860SBram Moolenaar 0, // sq_contains 16292ab2e860SBram Moolenaar 0, // sq_inplace_concat 16302ab2e860SBram Moolenaar 0, // sq_inplace_repeat 16315e538ecdSBram Moolenaar }; 16325e538ecdSBram Moolenaar 16332ab2e860SBram Moolenaar // Window list object - Definitions 1634bd5e15fdSBram Moolenaar 1635bd5e15fdSBram Moolenaar static PySequenceMethods WinListAsSeq = { 16362ab2e860SBram Moolenaar (lenfunc) WinListLength, // sq_length, len(x) 16372ab2e860SBram Moolenaar (binaryfunc) 0, // sq_concat, x+y 16382ab2e860SBram Moolenaar (ssizeargfunc) 0, // sq_repeat, x*n 16392ab2e860SBram Moolenaar (ssizeargfunc) WinListItem, // sq_item, x[i] 16402ab2e860SBram Moolenaar 0, // sq_slice, x[i:j] 16412ab2e860SBram Moolenaar (ssizeobjargproc)0, // sq_as_item, x[i]=v 16422ab2e860SBram Moolenaar 0, // sq_ass_slice, x[i:j]=v 16432ab2e860SBram Moolenaar 0, // sq_contains 16442ab2e860SBram Moolenaar 0, // sq_inplace_concat 16452ab2e860SBram Moolenaar 0, // sq_inplace_repeat 1646bd5e15fdSBram Moolenaar }; 1647bd5e15fdSBram Moolenaar 16482ab2e860SBram Moolenaar /* 16492ab2e860SBram Moolenaar * Current items object - Implementation 1650bd5e15fdSBram Moolenaar */ 1651170bf1aeSBram Moolenaar static PyObject * 16524d1da49cSBram Moolenaar CurrentGetattro(PyObject *self, PyObject *nameobj) 1653bd5e15fdSBram Moolenaar { 1654dd8aca66SBram Moolenaar PyObject *r; 165577045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1656dd8aca66SBram Moolenaar if (!(r = CurrentGetattr(self, name))) 1657dd8aca66SBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1658dd8aca66SBram Moolenaar return r; 1659bd5e15fdSBram Moolenaar } 1660bd5e15fdSBram Moolenaar 1661170bf1aeSBram Moolenaar static int 16624d1da49cSBram Moolenaar CurrentSetattro(PyObject *self, PyObject *nameobj, PyObject *value) 1663bd5e15fdSBram Moolenaar { 16644d1da49cSBram Moolenaar GET_ATTR_STRING(name, nameobj); 16654d1da49cSBram Moolenaar return CurrentSetattr(self, name, value); 1666bd5e15fdSBram Moolenaar } 1667bd5e15fdSBram Moolenaar 16682ab2e860SBram Moolenaar // Dictionary object - Definitions 1669db913953SBram Moolenaar 167066b7985eSBram Moolenaar static PyObject * 167166b7985eSBram Moolenaar DictionaryGetattro(PyObject *self, PyObject *nameobj) 167266b7985eSBram Moolenaar { 167366b7985eSBram Moolenaar DictionaryObject *this = ((DictionaryObject *) (self)); 167466b7985eSBram Moolenaar 167566b7985eSBram Moolenaar GET_ATTR_STRING(name, nameobj); 167666b7985eSBram Moolenaar 167766b7985eSBram Moolenaar if (strcmp(name, "locked") == 0) 167866b7985eSBram Moolenaar return PyLong_FromLong(this->dict->dv_lock); 167966b7985eSBram Moolenaar else if (strcmp(name, "scope") == 0) 168066b7985eSBram Moolenaar return PyLong_FromLong(this->dict->dv_scope); 168166b7985eSBram Moolenaar 168266b7985eSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 168366b7985eSBram Moolenaar } 168466b7985eSBram Moolenaar 168566b7985eSBram Moolenaar static int 168666b7985eSBram Moolenaar DictionarySetattro(PyObject *self, PyObject *nameobj, PyObject *val) 168766b7985eSBram Moolenaar { 168866b7985eSBram Moolenaar GET_ATTR_STRING(name, nameobj); 1689d6e39186SBram Moolenaar return DictionarySetattr((DictionaryObject *)(self), name, val); 1690db913953SBram Moolenaar } 1691db913953SBram Moolenaar 16922ab2e860SBram Moolenaar // List object - Definitions 1693db913953SBram Moolenaar 169466b7985eSBram Moolenaar static PyObject * 169566b7985eSBram Moolenaar ListGetattro(PyObject *self, PyObject *nameobj) 169666b7985eSBram Moolenaar { 169766b7985eSBram Moolenaar GET_ATTR_STRING(name, nameobj); 169866b7985eSBram Moolenaar 169966b7985eSBram Moolenaar if (strcmp(name, "locked") == 0) 170066b7985eSBram Moolenaar return PyLong_FromLong(((ListObject *) (self))->list->lv_lock); 170166b7985eSBram Moolenaar 170266b7985eSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 170366b7985eSBram Moolenaar } 170466b7985eSBram Moolenaar 170566b7985eSBram Moolenaar static int 170666b7985eSBram Moolenaar ListSetattro(PyObject *self, PyObject *nameobj, PyObject *val) 170766b7985eSBram Moolenaar { 170866b7985eSBram Moolenaar GET_ATTR_STRING(name, nameobj); 1709d6e39186SBram Moolenaar return ListSetattr((ListObject *)(self), name, val); 1710db913953SBram Moolenaar } 1711db913953SBram Moolenaar 17122ab2e860SBram Moolenaar // Function object - Definitions 1713db913953SBram Moolenaar 1714db913953SBram Moolenaar static PyObject * 1715db913953SBram Moolenaar FunctionGetattro(PyObject *self, PyObject *nameobj) 1716db913953SBram Moolenaar { 17178110a091SBram Moolenaar PyObject *r; 1718db913953SBram Moolenaar FunctionObject *this = (FunctionObject *)(self); 171977045658SBram Moolenaar 172077045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1721db913953SBram Moolenaar 17228110a091SBram Moolenaar r = FunctionAttr(this, name); 17238110a091SBram Moolenaar if (r || PyErr_Occurred()) 17248110a091SBram Moolenaar return r; 17258110a091SBram Moolenaar else 1726db913953SBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1727db913953SBram Moolenaar } 1728db913953SBram Moolenaar 17292ab2e860SBram Moolenaar // External interface 1730bd5e15fdSBram Moolenaar 1731bd5e15fdSBram Moolenaar void 1732bd5e15fdSBram Moolenaar python3_buffer_free(buf_T *buf) 1733bd5e15fdSBram Moolenaar { 1734971db467SBram Moolenaar if (BUF_PYTHON_REF(buf) != NULL) 1735bd5e15fdSBram Moolenaar { 1736971db467SBram Moolenaar BufferObject *bp = BUF_PYTHON_REF(buf); 1737bd5e15fdSBram Moolenaar bp->buf = INVALID_BUFFER_VALUE; 1738971db467SBram Moolenaar BUF_PYTHON_REF(buf) = NULL; 1739bd5e15fdSBram Moolenaar } 1740bd5e15fdSBram Moolenaar } 1741bd5e15fdSBram Moolenaar 1742bd5e15fdSBram Moolenaar void 1743bd5e15fdSBram Moolenaar python3_window_free(win_T *win) 1744bd5e15fdSBram Moolenaar { 1745971db467SBram Moolenaar if (WIN_PYTHON_REF(win) != NULL) 1746bd5e15fdSBram Moolenaar { 1747971db467SBram Moolenaar WindowObject *wp = WIN_PYTHON_REF(win); 1748bd5e15fdSBram Moolenaar wp->win = INVALID_WINDOW_VALUE; 1749971db467SBram Moolenaar WIN_PYTHON_REF(win) = NULL; 1750bd5e15fdSBram Moolenaar } 1751bd5e15fdSBram Moolenaar } 17525e538ecdSBram Moolenaar 17535e538ecdSBram Moolenaar void 17545e538ecdSBram Moolenaar python3_tabpage_free(tabpage_T *tab) 17555e538ecdSBram Moolenaar { 17565e538ecdSBram Moolenaar if (TAB_PYTHON_REF(tab) != NULL) 17575e538ecdSBram Moolenaar { 17585e538ecdSBram Moolenaar TabPageObject *tp = TAB_PYTHON_REF(tab); 17595e538ecdSBram Moolenaar tp->tab = INVALID_TABPAGE_VALUE; 17605e538ecdSBram Moolenaar TAB_PYTHON_REF(tab) = NULL; 17615e538ecdSBram Moolenaar } 17625e538ecdSBram Moolenaar } 1763bd5e15fdSBram Moolenaar 17647854e3abSBram Moolenaar static PyObject * 17657854e3abSBram Moolenaar Py3Init_vim(void) 1766bd5e15fdSBram Moolenaar { 17672ab2e860SBram Moolenaar // The special value is removed from sys.path in Python3_Init(). 1768bd5e15fdSBram Moolenaar static wchar_t *(argv[2]) = {L"/must>not&exist/foo", NULL}; 1769bd5e15fdSBram Moolenaar 17701dc28783SBram Moolenaar if (init_types()) 17711dc28783SBram Moolenaar return NULL; 1772bd5e15fdSBram Moolenaar 17732ab2e860SBram Moolenaar // Set sys.argv[] to avoid a crash in warn(). 1774bd5e15fdSBram Moolenaar PySys_SetArgv(1, argv); 1775bd5e15fdSBram Moolenaar 1776c09a6d6cSBram Moolenaar if ((vim_module = PyModule_Create(&vimmodule)) == NULL) 177719e60943SBram Moolenaar return NULL; 1778bd5e15fdSBram Moolenaar 1779dee2e315SBram Moolenaar if (populate_module(vim_module)) 1780bd5e15fdSBram Moolenaar return NULL; 1781bd5e15fdSBram Moolenaar 1782c09a6d6cSBram Moolenaar if (init_sys_path()) 1783c09a6d6cSBram Moolenaar return NULL; 1784c09a6d6cSBram Moolenaar 1785c09a6d6cSBram Moolenaar return vim_module; 1786bd5e15fdSBram Moolenaar } 1787bd5e15fdSBram Moolenaar 17882ab2e860SBram Moolenaar ////////////////////////////////////////////////////////////////////////// 17892ab2e860SBram Moolenaar // 4. Utility functions for handling the interface between Vim and Python. 1790bd5e15fdSBram Moolenaar 17912ab2e860SBram Moolenaar /* 17922ab2e860SBram Moolenaar * Convert a Vim line into a Python string. 1793bd5e15fdSBram Moolenaar * All internal newlines are replaced by null characters. 1794bd5e15fdSBram Moolenaar * 1795bd5e15fdSBram Moolenaar * On errors, the Python exception data is set, and NULL is returned. 1796bd5e15fdSBram Moolenaar */ 1797170bf1aeSBram Moolenaar static PyObject * 1798170bf1aeSBram Moolenaar LineToString(const char *str) 1799bd5e15fdSBram Moolenaar { 1800bd5e15fdSBram Moolenaar PyObject *result; 1801bd5e15fdSBram Moolenaar Py_ssize_t len = strlen(str); 1802bd5e15fdSBram Moolenaar char *tmp, *p; 1803bd5e15fdSBram Moolenaar 1804c799fe20SBram Moolenaar tmp = alloc(len + 1); 1805bd5e15fdSBram Moolenaar p = tmp; 1806bd5e15fdSBram Moolenaar if (p == NULL) 1807bd5e15fdSBram Moolenaar { 1808bd5e15fdSBram Moolenaar PyErr_NoMemory(); 1809bd5e15fdSBram Moolenaar return NULL; 1810bd5e15fdSBram Moolenaar } 1811bd5e15fdSBram Moolenaar 1812bd5e15fdSBram Moolenaar while (*str) 1813bd5e15fdSBram Moolenaar { 1814bd5e15fdSBram Moolenaar if (*str == '\n') 1815bd5e15fdSBram Moolenaar *p = '\0'; 1816bd5e15fdSBram Moolenaar else 1817bd5e15fdSBram Moolenaar *p = *str; 1818bd5e15fdSBram Moolenaar 1819bd5e15fdSBram Moolenaar ++p; 1820bd5e15fdSBram Moolenaar ++str; 1821bd5e15fdSBram Moolenaar } 1822bd5e15fdSBram Moolenaar *p = '\0'; 1823bd5e15fdSBram Moolenaar 18242e2f52a4SBram Moolenaar result = PyUnicode_Decode(tmp, len, (char *)ENC_OPT, ERRORS_DECODE_ARG); 1825bd5e15fdSBram Moolenaar 1826bd5e15fdSBram Moolenaar vim_free(tmp); 1827bd5e15fdSBram Moolenaar return result; 1828bd5e15fdSBram Moolenaar } 1829bd5e15fdSBram Moolenaar 1830db913953SBram Moolenaar void 1831db913953SBram Moolenaar do_py3eval(char_u *str, typval_T *rettv) 1832db913953SBram Moolenaar { 1833b52f4c02SBram Moolenaar DoPyCommand((char *) str, 1834b52f4c02SBram Moolenaar (rangeinitializer) init_range_eval, 1835b52f4c02SBram Moolenaar (runner) run_eval, 1836b52f4c02SBram Moolenaar (void *) rettv); 18378e9a24a1SBram Moolenaar if (rettv->v_type == VAR_UNKNOWN) 1838db913953SBram Moolenaar { 183977fceb89SBram Moolenaar rettv->v_type = VAR_NUMBER; 184077fceb89SBram Moolenaar rettv->vval.v_number = 0; 1841db913953SBram Moolenaar } 1842db913953SBram Moolenaar } 1843db913953SBram Moolenaar 18442459a5ecSBram Moolenaar int 1845db913953SBram Moolenaar set_ref_in_python3(int copyID) 1846db913953SBram Moolenaar { 1847b641df49SBram Moolenaar return set_ref_in_py(copyID); 1848db913953SBram Moolenaar } 1849