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 187*90478f35SZdenek Dohnal # if PY_VERSION_HEX >= 0x030a00b2 188*90478f35SZdenek Dohnal # define PyIter_Check py3_PyIter_Check 189*90478f35SZdenek Dohnal # endif 190db913953SBram Moolenaar # define PyIter_Next py3_PyIter_Next 191db913953SBram Moolenaar # define PyObject_GetIter py3_PyObject_GetIter 192141be8a5SBram Moolenaar # define PyObject_Repr py3_PyObject_Repr 193bcb40977SBram Moolenaar # define PyObject_GetItem py3_PyObject_GetItem 19403db85b3SBram Moolenaar # define PyObject_IsTrue py3_PyObject_IsTrue 195bd5e15fdSBram Moolenaar # define PyModule_GetDict py3_PyModule_GetDict 196bd5e15fdSBram Moolenaar #undef PyRun_SimpleString 197bd5e15fdSBram Moolenaar # define PyRun_SimpleString py3_PyRun_SimpleString 198db913953SBram Moolenaar #undef PyRun_String 199db913953SBram Moolenaar # define PyRun_String py3_PyRun_String 2003dab2806SBram Moolenaar # define PyObject_GetAttrString py3_PyObject_GetAttrString 201a9922d62SBram Moolenaar # define PyObject_HasAttrString py3_PyObject_HasAttrString 2023dab2806SBram Moolenaar # define PyObject_SetAttrString py3_PyObject_SetAttrString 2033dab2806SBram Moolenaar # define PyObject_CallFunctionObjArgs py3_PyObject_CallFunctionObjArgs 20481c40c50SBram Moolenaar # define _PyObject_CallFunction_SizeT py3__PyObject_CallFunction_SizeT 205f4258308SBram Moolenaar # define PyObject_Call py3_PyObject_Call 2063dab2806SBram Moolenaar # define PyEval_GetLocals py3_PyEval_GetLocals 2073dab2806SBram Moolenaar # define PyEval_GetGlobals py3_PyEval_GetGlobals 208bd5e15fdSBram Moolenaar # define PySys_SetObject py3_PySys_SetObject 209c09a6d6cSBram Moolenaar # define PySys_GetObject py3_PySys_GetObject 210bd5e15fdSBram Moolenaar # define PySys_SetArgv py3_PySys_SetArgv 211bd5e15fdSBram Moolenaar # define PyType_Ready py3_PyType_Ready 212ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 213ee1b9316SBram Moolenaar # define PyType_GetFlags py3_PyType_GetFlags 214ee1b9316SBram Moolenaar # endif 215bd5e15fdSBram Moolenaar #undef Py_BuildValue 216bd5e15fdSBram Moolenaar # define Py_BuildValue py3_Py_BuildValue 217644d37b8SBram Moolenaar # define Py_SetPythonHome py3_Py_SetPythonHome 218bd5e15fdSBram Moolenaar # define Py_Initialize py3_Py_Initialize 219bd5e15fdSBram Moolenaar # define Py_Finalize py3_Py_Finalize 220bd5e15fdSBram Moolenaar # define Py_IsInitialized py3_Py_IsInitialized 221bd5e15fdSBram Moolenaar # define _Py_NoneStruct (*py3__Py_NoneStruct) 22266b7985eSBram Moolenaar # define _Py_FalseStruct (*py3__Py_FalseStruct) 22366b7985eSBram Moolenaar # define _Py_TrueStruct (*py3__Py_TrueStruct) 224db913953SBram Moolenaar # define _PyObject_NextNotImplemented (*py3__PyObject_NextNotImplemented) 225bd5e15fdSBram Moolenaar # define PyModule_AddObject py3_PyModule_AddObject 226bd5e15fdSBram Moolenaar # define PyImport_AppendInittab py3_PyImport_AppendInittab 2273dab2806SBram Moolenaar # define PyImport_AddModule py3_PyImport_AddModule 2287bc4f93cSBram Moolenaar # if PY_VERSION_HEX >= 0x030300f0 2297bc4f93cSBram Moolenaar # undef _PyUnicode_AsString 2309c9cbf13SBram Moolenaar # define _PyUnicode_AsString py3_PyUnicode_AsUTF8 2317bc4f93cSBram Moolenaar # else 232bd5e15fdSBram Moolenaar # define _PyUnicode_AsString py3__PyUnicode_AsString 2337bc4f93cSBram Moolenaar # endif 23419e60943SBram Moolenaar # undef PyUnicode_AsEncodedString 23519e60943SBram Moolenaar # define PyUnicode_AsEncodedString py3_PyUnicode_AsEncodedString 23619e60943SBram Moolenaar # undef PyBytes_AsString 23719e60943SBram Moolenaar # define PyBytes_AsString py3_PyBytes_AsString 23832ac8cdaSBram Moolenaar # ifndef PyBytes_AsStringAndSize 239cdab9051SBram Moolenaar # define PyBytes_AsStringAndSize py3_PyBytes_AsStringAndSize 24032ac8cdaSBram Moolenaar # endif 241db913953SBram Moolenaar # undef PyBytes_FromString 242db913953SBram Moolenaar # define PyBytes_FromString py3_PyBytes_FromString 2436e5ea8d2SBram Moolenaar # undef PyBytes_FromStringAndSize 2446e5ea8d2SBram Moolenaar # define PyBytes_FromStringAndSize py3_PyBytes_FromStringAndSize 245ee1b9316SBram Moolenaar # if defined(Py_DEBUG) || PY_VERSION_HEX >= 0x030900b0 246ee1b9316SBram Moolenaar # define _Py_Dealloc py3__Py_Dealloc 247ee1b9316SBram Moolenaar # endif 248db913953SBram Moolenaar # define PyFloat_FromDouble py3_PyFloat_FromDouble 249db913953SBram Moolenaar # define PyFloat_AsDouble py3_PyFloat_AsDouble 250bd5e15fdSBram Moolenaar # define PyObject_GenericGetAttr py3_PyObject_GenericGetAttr 25166b7985eSBram Moolenaar # define PyType_Type (*py3_PyType_Type) 252d4a8c98eSBram Moolenaar # define PyStdPrinter_Type (*py3_PyStdPrinter_Type) 253bd5e15fdSBram Moolenaar # define PySlice_Type (*py3_PySlice_Type) 254db913953SBram Moolenaar # define PyFloat_Type (*py3_PyFloat_Type) 255141be8a5SBram Moolenaar # define PyNumber_Check (*py3_PyNumber_Check) 256141be8a5SBram Moolenaar # define PyNumber_Long (*py3_PyNumber_Long) 25766b7985eSBram Moolenaar # define PyBool_Type (*py3_PyBool_Type) 25819e60943SBram Moolenaar # define PyErr_NewException py3_PyErr_NewException 259bd5e15fdSBram Moolenaar # ifdef Py_DEBUG 260bd5e15fdSBram Moolenaar # define _Py_NegativeRefcount py3__Py_NegativeRefcount 261bd5e15fdSBram Moolenaar # define _Py_RefTotal (*py3__Py_RefTotal) 2620014a53aSBram Moolenaar # define PyModule_Create2TraceRefs py3_PyModule_Create2TraceRefs 2630014a53aSBram Moolenaar # else 2640014a53aSBram Moolenaar # define PyModule_Create2 py3_PyModule_Create2 2650014a53aSBram Moolenaar # endif 2660014a53aSBram Moolenaar # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 267bd5e15fdSBram Moolenaar # define _PyObject_DebugMalloc py3__PyObject_DebugMalloc 268bd5e15fdSBram Moolenaar # define _PyObject_DebugFree py3__PyObject_DebugFree 269bd5e15fdSBram Moolenaar # else 270bd5e15fdSBram Moolenaar # define PyObject_Malloc py3_PyObject_Malloc 271bd5e15fdSBram Moolenaar # define PyObject_Free py3_PyObject_Free 272bd5e15fdSBram Moolenaar # endif 273774267bbSBram Moolenaar # define _PyObject_GC_New py3__PyObject_GC_New 274774267bbSBram Moolenaar # define PyObject_GC_Del py3_PyObject_GC_Del 275774267bbSBram Moolenaar # define PyObject_GC_UnTrack py3_PyObject_GC_UnTrack 276bd5e15fdSBram Moolenaar # define PyType_GenericAlloc py3_PyType_GenericAlloc 277bd5e15fdSBram Moolenaar # define PyType_GenericNew py3_PyType_GenericNew 278bd5e15fdSBram Moolenaar # undef PyUnicode_FromString 279bd5e15fdSBram Moolenaar # define PyUnicode_FromString py3_PyUnicode_FromString 2801a3b5695SBram Moolenaar # ifndef PyUnicode_FromFormat 2811a3b5695SBram Moolenaar # define PyUnicode_FromFormat py3_PyUnicode_FromFormat 2821a3b5695SBram Moolenaar # else 2831a3b5695SBram Moolenaar # define Py_UNICODE_USE_UCS_FUNCTIONS 2841a3b5695SBram Moolenaar # ifdef Py_UNICODE_WIDE 2851a3b5695SBram Moolenaar # define PyUnicodeUCS4_FromFormat py3_PyUnicodeUCS4_FromFormat 2861a3b5695SBram Moolenaar # else 2871a3b5695SBram Moolenaar # define PyUnicodeUCS2_FromFormat py3_PyUnicodeUCS2_FromFormat 2881a3b5695SBram Moolenaar # endif 2891a3b5695SBram Moolenaar # endif 29019e60943SBram Moolenaar # undef PyUnicode_Decode 29119e60943SBram Moolenaar # define PyUnicode_Decode py3_PyUnicode_Decode 292db913953SBram Moolenaar # define PyType_IsSubtype py3_PyType_IsSubtype 293db913953SBram Moolenaar # define PyCapsule_New py3_PyCapsule_New 294db913953SBram Moolenaar # define PyCapsule_GetPointer py3_PyCapsule_GetPointer 295bd5e15fdSBram Moolenaar 2960014a53aSBram Moolenaar # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 297bd5e15fdSBram Moolenaar # undef PyObject_NEW 298bd5e15fdSBram Moolenaar # define PyObject_NEW(type, typeobj) \ 299bd5e15fdSBram Moolenaar ( (type *) PyObject_Init( \ 300bd5e15fdSBram Moolenaar (PyObject *) _PyObject_DebugMalloc( _PyObject_SIZE(typeobj) ), (typeobj)) ) 301ee1b9316SBram Moolenaar # elif PY_VERSION_HEX >= 0x030900b0 302ee1b9316SBram Moolenaar # undef PyObject_NEW 303ee1b9316SBram Moolenaar # define PyObject_NEW(type, typeobj) \ 304ee1b9316SBram Moolenaar ((type *)py3__PyObject_New(typeobj)) 305bd5e15fdSBram Moolenaar # endif 306b61f95c3SBram Moolenaar 307bd5e15fdSBram Moolenaar /* 308bd5e15fdSBram Moolenaar * Pointers for dynamic link 309bd5e15fdSBram Moolenaar */ 310bd5e15fdSBram Moolenaar static int (*py3_PySys_SetArgv)(int, wchar_t **); 311644d37b8SBram Moolenaar static void (*py3_Py_SetPythonHome)(wchar_t *home); 312bd5e15fdSBram Moolenaar static void (*py3_Py_Initialize)(void); 313bd5e15fdSBram Moolenaar static PyObject* (*py3_PyList_New)(Py_ssize_t size); 314bd5e15fdSBram Moolenaar static PyGILState_STATE (*py3_PyGILState_Ensure)(void); 315bd5e15fdSBram Moolenaar static void (*py3_PyGILState_Release)(PyGILState_STATE); 316bd5e15fdSBram Moolenaar static int (*py3_PySys_SetObject)(char *, PyObject *); 317c09a6d6cSBram Moolenaar static PyObject* (*py3_PySys_GetObject)(char *); 318c09a6d6cSBram Moolenaar static int (*py3_PyList_Append)(PyObject *, PyObject *); 319c09a6d6cSBram Moolenaar static int (*py3_PyList_Insert)(PyObject *, int, PyObject *); 320bd5e15fdSBram Moolenaar static Py_ssize_t (*py3_PyList_Size)(PyObject *); 321db913953SBram Moolenaar static int (*py3_PySequence_Check)(PyObject *); 322db913953SBram Moolenaar static Py_ssize_t (*py3_PySequence_Size)(PyObject *); 323db913953SBram Moolenaar static PyObject* (*py3_PySequence_GetItem)(PyObject *, Py_ssize_t); 324a9922d62SBram Moolenaar static PyObject* (*py3_PySequence_Fast)(PyObject *, const char *); 325db913953SBram Moolenaar static Py_ssize_t (*py3_PyTuple_Size)(PyObject *); 326db913953SBram Moolenaar static PyObject* (*py3_PyTuple_GetItem)(PyObject *, Py_ssize_t); 327db913953SBram Moolenaar static int (*py3_PyMapping_Check)(PyObject *); 328bcb40977SBram Moolenaar static PyObject* (*py3_PyMapping_Keys)(PyObject *); 3293b48b11cSBram Moolenaar # if PY_VERSION_HEX >= 0x030601f0 3303b48b11cSBram Moolenaar static int (*py3_PySlice_AdjustIndices)(Py_ssize_t length, 3313b48b11cSBram Moolenaar Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t step); 3323b48b11cSBram Moolenaar static int (*py3_PySlice_Unpack)(PyObject *slice, 3333b48b11cSBram Moolenaar Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step); 3343b48b11cSBram Moolenaar # endif 335922a4664SBram Moolenaar static int (*py3_PySlice_GetIndicesEx)(PySliceObject_T *r, Py_ssize_t length, 336063a46baSBram Moolenaar Py_ssize_t *start, Py_ssize_t *stop, Py_ssize_t *step, 337063a46baSBram Moolenaar Py_ssize_t *slicelen); 338bd5e15fdSBram Moolenaar static PyObject* (*py3_PyErr_NoMemory)(void); 339bd5e15fdSBram Moolenaar static void (*py3_Py_Finalize)(void); 340bd5e15fdSBram Moolenaar static void (*py3_PyErr_SetString)(PyObject *, const char *); 3414d188da2SBram Moolenaar static void (*py3_PyErr_SetObject)(PyObject *, PyObject *); 342c09a6d6cSBram Moolenaar static int (*py3_PyErr_ExceptionMatches)(PyObject *); 343bd5e15fdSBram Moolenaar static int (*py3_PyRun_SimpleString)(char *); 344db913953SBram Moolenaar static PyObject* (*py3_PyRun_String)(char *, int, PyObject *, PyObject *); 3453dab2806SBram Moolenaar static PyObject* (*py3_PyObject_GetAttrString)(PyObject *, const char *); 346a9922d62SBram Moolenaar static int (*py3_PyObject_HasAttrString)(PyObject *, const char *); 3470b400087SBram Moolenaar static int (*py3_PyObject_SetAttrString)(PyObject *, const char *, PyObject *); 3483dab2806SBram Moolenaar static PyObject* (*py3_PyObject_CallFunctionObjArgs)(PyObject *, ...); 34981c40c50SBram Moolenaar static PyObject* (*py3__PyObject_CallFunction_SizeT)(PyObject *, char *, ...); 350f4258308SBram Moolenaar static PyObject* (*py3_PyObject_Call)(PyObject *, PyObject *, PyObject *); 3513dab2806SBram Moolenaar static PyObject* (*py3_PyEval_GetGlobals)(); 3523dab2806SBram Moolenaar static PyObject* (*py3_PyEval_GetLocals)(); 353bd5e15fdSBram Moolenaar static PyObject* (*py3_PyList_GetItem)(PyObject *, Py_ssize_t); 354bd5e15fdSBram Moolenaar static PyObject* (*py3_PyImport_ImportModule)(const char *); 355db913953SBram Moolenaar static PyObject* (*py3_PyImport_AddModule)(const char *); 356bd5e15fdSBram Moolenaar static int (*py3_PyErr_BadArgument)(void); 357bd5e15fdSBram Moolenaar static PyObject* (*py3_PyErr_Occurred)(void); 358bd5e15fdSBram Moolenaar static PyObject* (*py3_PyModule_GetDict)(PyObject *); 359bd5e15fdSBram Moolenaar static int (*py3_PyList_SetItem)(PyObject *, Py_ssize_t, PyObject *); 360bd5e15fdSBram Moolenaar static PyObject* (*py3_PyDict_GetItemString)(PyObject *, const char *); 361db913953SBram Moolenaar static int (*py3_PyDict_Next)(PyObject *, Py_ssize_t *, PyObject **, PyObject **); 362bd5e15fdSBram Moolenaar static PyObject* (*py3_PyLong_FromLong)(long); 363bd5e15fdSBram Moolenaar static PyObject* (*py3_PyDict_New)(void); 364*90478f35SZdenek Dohnal # if PY_VERSION_HEX >= 0x030a00b2 365*90478f35SZdenek Dohnal static int (*py3_PyIter_Check)(PyObject *o); 366*90478f35SZdenek Dohnal # endif 367db913953SBram Moolenaar static PyObject* (*py3_PyIter_Next)(PyObject *); 368db913953SBram Moolenaar static PyObject* (*py3_PyObject_GetIter)(PyObject *); 369141be8a5SBram Moolenaar static PyObject* (*py3_PyObject_Repr)(PyObject *); 370bcb40977SBram Moolenaar static PyObject* (*py3_PyObject_GetItem)(PyObject *, PyObject *); 37103db85b3SBram Moolenaar static int (*py3_PyObject_IsTrue)(PyObject *); 372bd5e15fdSBram Moolenaar static PyObject* (*py3_Py_BuildValue)(char *, ...); 373ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 374ee1b9316SBram Moolenaar static int (*py3_PyType_GetFlags)(PyTypeObject *o); 375ee1b9316SBram Moolenaar # endif 376bd5e15fdSBram Moolenaar static int (*py3_PyType_Ready)(PyTypeObject *type); 377bd5e15fdSBram Moolenaar static int (*py3_PyDict_SetItemString)(PyObject *dp, char *key, PyObject *item); 378bd5e15fdSBram Moolenaar static PyObject* (*py3_PyUnicode_FromString)(const char *u); 3791a3b5695SBram Moolenaar # ifndef Py_UNICODE_USE_UCS_FUNCTIONS 3801a3b5695SBram Moolenaar static PyObject* (*py3_PyUnicode_FromFormat)(const char *u, ...); 3811a3b5695SBram Moolenaar # else 3821a3b5695SBram Moolenaar # ifdef Py_UNICODE_WIDE 3831a3b5695SBram Moolenaar static PyObject* (*py3_PyUnicodeUCS4_FromFormat)(const char *u, ...); 3841a3b5695SBram Moolenaar # else 3851a3b5695SBram Moolenaar static PyObject* (*py3_PyUnicodeUCS2_FromFormat)(const char *u, ...); 3861a3b5695SBram Moolenaar # endif 3871a3b5695SBram Moolenaar # endif 38819e60943SBram Moolenaar static PyObject* (*py3_PyUnicode_Decode)(const char *u, Py_ssize_t size, 38919e60943SBram Moolenaar const char *encoding, const char *errors); 390bd5e15fdSBram Moolenaar static long (*py3_PyLong_AsLong)(PyObject *); 391bd5e15fdSBram Moolenaar static void (*py3_PyErr_SetNone)(PyObject *); 392bd5e15fdSBram Moolenaar static void (*py3_PyEval_InitThreads)(void); 393bd5e15fdSBram Moolenaar static void(*py3_PyEval_RestoreThread)(PyThreadState *); 394bd5e15fdSBram Moolenaar static PyThreadState*(*py3_PyEval_SaveThread)(void); 395bd5e15fdSBram Moolenaar static int (*py3_PyArg_Parse)(PyObject *, char *, ...); 396bd5e15fdSBram Moolenaar static int (*py3_PyArg_ParseTuple)(PyObject *, char *, ...); 39719e60943SBram Moolenaar static int (*py3_PyMem_Free)(void *); 398db913953SBram Moolenaar static void* (*py3_PyMem_Malloc)(size_t); 399bd5e15fdSBram Moolenaar static int (*py3_Py_IsInitialized)(void); 400bd5e15fdSBram Moolenaar static void (*py3_PyErr_Clear)(void); 401c476e52fSBram Moolenaar static PyObject* (*py3_PyErr_Format)(PyObject *, const char *, ...); 4024d36987cSBram Moolenaar static void (*py3_PyErr_PrintEx)(int); 403bd5e15fdSBram Moolenaar static PyObject*(*py3__PyObject_Init)(PyObject *, PyTypeObject *); 404db913953SBram Moolenaar static iternextfunc py3__PyObject_NextNotImplemented; 405bd5e15fdSBram Moolenaar static PyObject* py3__Py_NoneStruct; 40666b7985eSBram Moolenaar static PyObject* py3__Py_FalseStruct; 40766b7985eSBram Moolenaar static PyObject* py3__Py_TrueStruct; 408bd5e15fdSBram Moolenaar static int (*py3_PyModule_AddObject)(PyObject *m, const char *name, PyObject *o); 409bd5e15fdSBram Moolenaar static int (*py3_PyImport_AppendInittab)(const char *name, PyObject* (*initfunc)(void)); 4107bc4f93cSBram Moolenaar # if PY_VERSION_HEX >= 0x030300f0 4119c9cbf13SBram Moolenaar static char* (*py3_PyUnicode_AsUTF8)(PyObject *unicode); 4127bc4f93cSBram Moolenaar # else 413bd5e15fdSBram Moolenaar static char* (*py3__PyUnicode_AsString)(PyObject *unicode); 4147bc4f93cSBram Moolenaar # endif 41519e60943SBram Moolenaar static PyObject* (*py3_PyUnicode_AsEncodedString)(PyObject *unicode, const char* encoding, const char* errors); 41619e60943SBram Moolenaar static char* (*py3_PyBytes_AsString)(PyObject *bytes); 417808c2bc8SBram Moolenaar static int (*py3_PyBytes_AsStringAndSize)(PyObject *bytes, char **buffer, Py_ssize_t *length); 418db913953SBram Moolenaar static PyObject* (*py3_PyBytes_FromString)(char *str); 4196e5ea8d2SBram Moolenaar static PyObject* (*py3_PyBytes_FromStringAndSize)(char *str, Py_ssize_t length); 420ee1b9316SBram Moolenaar # if defined(Py_DEBUG) || PY_VERSION_HEX >= 0x030900b0 421ee1b9316SBram Moolenaar static void (*py3__Py_Dealloc)(PyObject *obj); 422ee1b9316SBram Moolenaar # endif 423ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 424ee1b9316SBram Moolenaar static PyObject* (*py3__PyObject_New)(PyTypeObject *); 425ee1b9316SBram Moolenaar # endif 426db913953SBram Moolenaar static PyObject* (*py3_PyFloat_FromDouble)(double num); 427db913953SBram Moolenaar static double (*py3_PyFloat_AsDouble)(PyObject *); 428bd5e15fdSBram Moolenaar static PyObject* (*py3_PyObject_GenericGetAttr)(PyObject *obj, PyObject *name); 429bd5e15fdSBram Moolenaar static PyObject* (*py3_PyType_GenericAlloc)(PyTypeObject *type, Py_ssize_t nitems); 430bd5e15fdSBram Moolenaar static PyObject* (*py3_PyType_GenericNew)(PyTypeObject *type, PyObject *args, PyObject *kwds); 43166b7985eSBram Moolenaar static PyTypeObject* py3_PyType_Type; 432d4a8c98eSBram Moolenaar static PyTypeObject* py3_PyStdPrinter_Type; 433bd5e15fdSBram Moolenaar static PyTypeObject* py3_PySlice_Type; 434db913953SBram Moolenaar static PyTypeObject* py3_PyFloat_Type; 43566b7985eSBram Moolenaar static PyTypeObject* py3_PyBool_Type; 436141be8a5SBram Moolenaar static int (*py3_PyNumber_Check)(PyObject *); 437141be8a5SBram Moolenaar static PyObject* (*py3_PyNumber_Long)(PyObject *); 43819e60943SBram Moolenaar static PyObject* (*py3_PyErr_NewException)(char *name, PyObject *base, PyObject *dict); 439db913953SBram Moolenaar static PyObject* (*py3_PyCapsule_New)(void *, char *, PyCapsule_Destructor); 440db913953SBram Moolenaar static void* (*py3_PyCapsule_GetPointer)(PyObject *, char *); 441bd5e15fdSBram Moolenaar # ifdef Py_DEBUG 442bd5e15fdSBram Moolenaar static void (*py3__Py_NegativeRefcount)(const char *fname, int lineno, PyObject *op); 443bd5e15fdSBram Moolenaar static Py_ssize_t* py3__Py_RefTotal; 4440014a53aSBram Moolenaar static PyObject* (*py3_PyModule_Create2TraceRefs)(struct PyModuleDef* module, int module_api_version); 4450014a53aSBram Moolenaar # else 4460014a53aSBram Moolenaar static PyObject* (*py3_PyModule_Create2)(struct PyModuleDef* module, int module_api_version); 4470014a53aSBram Moolenaar # endif 4480014a53aSBram Moolenaar # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 449bd5e15fdSBram Moolenaar static void (*py3__PyObject_DebugFree)(void*); 450bd5e15fdSBram Moolenaar static void* (*py3__PyObject_DebugMalloc)(size_t); 451bd5e15fdSBram Moolenaar # else 452bd5e15fdSBram Moolenaar static void (*py3_PyObject_Free)(void*); 453bd5e15fdSBram Moolenaar static void* (*py3_PyObject_Malloc)(size_t); 454bd5e15fdSBram Moolenaar # endif 455774267bbSBram Moolenaar static PyObject*(*py3__PyObject_GC_New)(PyTypeObject *); 456774267bbSBram Moolenaar static void(*py3_PyObject_GC_Del)(void *); 457774267bbSBram Moolenaar static void(*py3_PyObject_GC_UnTrack)(void *); 458db913953SBram Moolenaar static int (*py3_PyType_IsSubtype)(PyTypeObject *, PyTypeObject *); 459bd5e15fdSBram Moolenaar 4602ab2e860SBram Moolenaar static HINSTANCE hinstPy3 = 0; // Instance of python.dll 461bd5e15fdSBram Moolenaar 4622ab2e860SBram Moolenaar // Imported exception objects 463bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_AttributeError; 464bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_IndexError; 465af6abb9dSBram Moolenaar static PyObject *p3imp_PyExc_KeyError; 466bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_KeyboardInterrupt; 467bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_TypeError; 468bd5e15fdSBram Moolenaar static PyObject *p3imp_PyExc_ValueError; 46941009374SBram Moolenaar static PyObject *p3imp_PyExc_SystemExit; 4708661b178SBram Moolenaar static PyObject *p3imp_PyExc_RuntimeError; 471c09a6d6cSBram Moolenaar static PyObject *p3imp_PyExc_ImportError; 472141be8a5SBram Moolenaar static PyObject *p3imp_PyExc_OverflowError; 473bd5e15fdSBram Moolenaar 474bd5e15fdSBram Moolenaar # define PyExc_AttributeError p3imp_PyExc_AttributeError 475bd5e15fdSBram Moolenaar # define PyExc_IndexError p3imp_PyExc_IndexError 476af6abb9dSBram Moolenaar # define PyExc_KeyError p3imp_PyExc_KeyError 477bd5e15fdSBram Moolenaar # define PyExc_KeyboardInterrupt p3imp_PyExc_KeyboardInterrupt 478bd5e15fdSBram Moolenaar # define PyExc_TypeError p3imp_PyExc_TypeError 479bd5e15fdSBram Moolenaar # define PyExc_ValueError p3imp_PyExc_ValueError 48041009374SBram Moolenaar # define PyExc_SystemExit p3imp_PyExc_SystemExit 4818661b178SBram Moolenaar # define PyExc_RuntimeError p3imp_PyExc_RuntimeError 482c09a6d6cSBram Moolenaar # define PyExc_ImportError p3imp_PyExc_ImportError 483141be8a5SBram Moolenaar # define PyExc_OverflowError p3imp_PyExc_OverflowError 484bd5e15fdSBram Moolenaar 485bd5e15fdSBram Moolenaar /* 486bd5e15fdSBram Moolenaar * Table of name to function pointer of python. 487bd5e15fdSBram Moolenaar */ 488bd5e15fdSBram Moolenaar # define PYTHON_PROC FARPROC 489bd5e15fdSBram Moolenaar static struct 490bd5e15fdSBram Moolenaar { 491bd5e15fdSBram Moolenaar char *name; 492bd5e15fdSBram Moolenaar PYTHON_PROC *ptr; 493bd5e15fdSBram Moolenaar } py3_funcname_table[] = 494bd5e15fdSBram Moolenaar { 495bd5e15fdSBram Moolenaar {"PySys_SetArgv", (PYTHON_PROC*)&py3_PySys_SetArgv}, 496644d37b8SBram Moolenaar {"Py_SetPythonHome", (PYTHON_PROC*)&py3_Py_SetPythonHome}, 497bd5e15fdSBram Moolenaar {"Py_Initialize", (PYTHON_PROC*)&py3_Py_Initialize}, 498e8cdcef8SBram Moolenaar {"_PyArg_ParseTuple_SizeT", (PYTHON_PROC*)&py3_PyArg_ParseTuple}, 499e8cdcef8SBram Moolenaar {"_Py_BuildValue_SizeT", (PYTHON_PROC*)&py3_Py_BuildValue}, 50019e60943SBram Moolenaar {"PyMem_Free", (PYTHON_PROC*)&py3_PyMem_Free}, 501db913953SBram Moolenaar {"PyMem_Malloc", (PYTHON_PROC*)&py3_PyMem_Malloc}, 502bd5e15fdSBram Moolenaar {"PyList_New", (PYTHON_PROC*)&py3_PyList_New}, 503bd5e15fdSBram Moolenaar {"PyGILState_Ensure", (PYTHON_PROC*)&py3_PyGILState_Ensure}, 504bd5e15fdSBram Moolenaar {"PyGILState_Release", (PYTHON_PROC*)&py3_PyGILState_Release}, 505bd5e15fdSBram Moolenaar {"PySys_SetObject", (PYTHON_PROC*)&py3_PySys_SetObject}, 506c09a6d6cSBram Moolenaar {"PySys_GetObject", (PYTHON_PROC*)&py3_PySys_GetObject}, 507bd5e15fdSBram Moolenaar {"PyList_Append", (PYTHON_PROC*)&py3_PyList_Append}, 508c09a6d6cSBram Moolenaar {"PyList_Insert", (PYTHON_PROC*)&py3_PyList_Insert}, 509bd5e15fdSBram Moolenaar {"PyList_Size", (PYTHON_PROC*)&py3_PyList_Size}, 510db913953SBram Moolenaar {"PySequence_Check", (PYTHON_PROC*)&py3_PySequence_Check}, 511db913953SBram Moolenaar {"PySequence_Size", (PYTHON_PROC*)&py3_PySequence_Size}, 512db913953SBram Moolenaar {"PySequence_GetItem", (PYTHON_PROC*)&py3_PySequence_GetItem}, 513a9922d62SBram Moolenaar {"PySequence_Fast", (PYTHON_PROC*)&py3_PySequence_Fast}, 514db913953SBram Moolenaar {"PyTuple_Size", (PYTHON_PROC*)&py3_PyTuple_Size}, 515db913953SBram Moolenaar {"PyTuple_GetItem", (PYTHON_PROC*)&py3_PyTuple_GetItem}, 5163b48b11cSBram Moolenaar # if PY_VERSION_HEX >= 0x030601f0 5173b48b11cSBram Moolenaar {"PySlice_AdjustIndices", (PYTHON_PROC*)&py3_PySlice_AdjustIndices}, 5183b48b11cSBram Moolenaar {"PySlice_Unpack", (PYTHON_PROC*)&py3_PySlice_Unpack}, 5193b48b11cSBram Moolenaar # endif 520bd5e15fdSBram Moolenaar {"PySlice_GetIndicesEx", (PYTHON_PROC*)&py3_PySlice_GetIndicesEx}, 521bd5e15fdSBram Moolenaar {"PyErr_NoMemory", (PYTHON_PROC*)&py3_PyErr_NoMemory}, 522bd5e15fdSBram Moolenaar {"Py_Finalize", (PYTHON_PROC*)&py3_Py_Finalize}, 523bd5e15fdSBram Moolenaar {"PyErr_SetString", (PYTHON_PROC*)&py3_PyErr_SetString}, 5244d188da2SBram Moolenaar {"PyErr_SetObject", (PYTHON_PROC*)&py3_PyErr_SetObject}, 525c09a6d6cSBram Moolenaar {"PyErr_ExceptionMatches", (PYTHON_PROC*)&py3_PyErr_ExceptionMatches}, 526bd5e15fdSBram Moolenaar {"PyRun_SimpleString", (PYTHON_PROC*)&py3_PyRun_SimpleString}, 527db913953SBram Moolenaar {"PyRun_String", (PYTHON_PROC*)&py3_PyRun_String}, 5283dab2806SBram Moolenaar {"PyObject_GetAttrString", (PYTHON_PROC*)&py3_PyObject_GetAttrString}, 529a9922d62SBram Moolenaar {"PyObject_HasAttrString", (PYTHON_PROC*)&py3_PyObject_HasAttrString}, 5303dab2806SBram Moolenaar {"PyObject_SetAttrString", (PYTHON_PROC*)&py3_PyObject_SetAttrString}, 5313dab2806SBram Moolenaar {"PyObject_CallFunctionObjArgs", (PYTHON_PROC*)&py3_PyObject_CallFunctionObjArgs}, 53281c40c50SBram Moolenaar {"_PyObject_CallFunction_SizeT", (PYTHON_PROC*)&py3__PyObject_CallFunction_SizeT}, 533f4258308SBram Moolenaar {"PyObject_Call", (PYTHON_PROC*)&py3_PyObject_Call}, 5343dab2806SBram Moolenaar {"PyEval_GetGlobals", (PYTHON_PROC*)&py3_PyEval_GetGlobals}, 5353dab2806SBram Moolenaar {"PyEval_GetLocals", (PYTHON_PROC*)&py3_PyEval_GetLocals}, 536bd5e15fdSBram Moolenaar {"PyList_GetItem", (PYTHON_PROC*)&py3_PyList_GetItem}, 537bd5e15fdSBram Moolenaar {"PyImport_ImportModule", (PYTHON_PROC*)&py3_PyImport_ImportModule}, 538db913953SBram Moolenaar {"PyImport_AddModule", (PYTHON_PROC*)&py3_PyImport_AddModule}, 539bd5e15fdSBram Moolenaar {"PyErr_BadArgument", (PYTHON_PROC*)&py3_PyErr_BadArgument}, 540bd5e15fdSBram Moolenaar {"PyErr_Occurred", (PYTHON_PROC*)&py3_PyErr_Occurred}, 541bd5e15fdSBram Moolenaar {"PyModule_GetDict", (PYTHON_PROC*)&py3_PyModule_GetDict}, 542bd5e15fdSBram Moolenaar {"PyList_SetItem", (PYTHON_PROC*)&py3_PyList_SetItem}, 543bd5e15fdSBram Moolenaar {"PyDict_GetItemString", (PYTHON_PROC*)&py3_PyDict_GetItemString}, 544db913953SBram Moolenaar {"PyDict_Next", (PYTHON_PROC*)&py3_PyDict_Next}, 545db913953SBram Moolenaar {"PyMapping_Check", (PYTHON_PROC*)&py3_PyMapping_Check}, 546bcb40977SBram Moolenaar {"PyMapping_Keys", (PYTHON_PROC*)&py3_PyMapping_Keys}, 547*90478f35SZdenek Dohnal # if PY_VERSION_HEX >= 0x030a00b2 548*90478f35SZdenek Dohnal {"PyIter_Check", (PYTHON_PROC*)&py3_PyIter_Check}, 549*90478f35SZdenek Dohnal # endif 550db913953SBram Moolenaar {"PyIter_Next", (PYTHON_PROC*)&py3_PyIter_Next}, 551db913953SBram Moolenaar {"PyObject_GetIter", (PYTHON_PROC*)&py3_PyObject_GetIter}, 552141be8a5SBram Moolenaar {"PyObject_Repr", (PYTHON_PROC*)&py3_PyObject_Repr}, 553bcb40977SBram Moolenaar {"PyObject_GetItem", (PYTHON_PROC*)&py3_PyObject_GetItem}, 55403db85b3SBram Moolenaar {"PyObject_IsTrue", (PYTHON_PROC*)&py3_PyObject_IsTrue}, 555bd5e15fdSBram Moolenaar {"PyLong_FromLong", (PYTHON_PROC*)&py3_PyLong_FromLong}, 556bd5e15fdSBram Moolenaar {"PyDict_New", (PYTHON_PROC*)&py3_PyDict_New}, 557ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 558ee1b9316SBram Moolenaar {"PyType_GetFlags", (PYTHON_PROC*)&py3_PyType_GetFlags}, 559ee1b9316SBram Moolenaar # endif 560bd5e15fdSBram Moolenaar {"PyType_Ready", (PYTHON_PROC*)&py3_PyType_Ready}, 561bd5e15fdSBram Moolenaar {"PyDict_SetItemString", (PYTHON_PROC*)&py3_PyDict_SetItemString}, 562bd5e15fdSBram Moolenaar {"PyLong_AsLong", (PYTHON_PROC*)&py3_PyLong_AsLong}, 563bd5e15fdSBram Moolenaar {"PyErr_SetNone", (PYTHON_PROC*)&py3_PyErr_SetNone}, 564bd5e15fdSBram Moolenaar {"PyEval_InitThreads", (PYTHON_PROC*)&py3_PyEval_InitThreads}, 565bd5e15fdSBram Moolenaar {"PyEval_RestoreThread", (PYTHON_PROC*)&py3_PyEval_RestoreThread}, 566bd5e15fdSBram Moolenaar {"PyEval_SaveThread", (PYTHON_PROC*)&py3_PyEval_SaveThread}, 5675f87b232SBram Moolenaar {"_PyArg_Parse_SizeT", (PYTHON_PROC*)&py3_PyArg_Parse}, 568bd5e15fdSBram Moolenaar {"Py_IsInitialized", (PYTHON_PROC*)&py3_Py_IsInitialized}, 569db913953SBram Moolenaar {"_PyObject_NextNotImplemented", (PYTHON_PROC*)&py3__PyObject_NextNotImplemented}, 570bd5e15fdSBram Moolenaar {"_Py_NoneStruct", (PYTHON_PROC*)&py3__Py_NoneStruct}, 57166b7985eSBram Moolenaar {"_Py_FalseStruct", (PYTHON_PROC*)&py3__Py_FalseStruct}, 57266b7985eSBram Moolenaar {"_Py_TrueStruct", (PYTHON_PROC*)&py3__Py_TrueStruct}, 573bd5e15fdSBram Moolenaar {"PyErr_Clear", (PYTHON_PROC*)&py3_PyErr_Clear}, 574c476e52fSBram Moolenaar {"PyErr_Format", (PYTHON_PROC*)&py3_PyErr_Format}, 5754d36987cSBram Moolenaar {"PyErr_PrintEx", (PYTHON_PROC*)&py3_PyErr_PrintEx}, 576bd5e15fdSBram Moolenaar {"PyObject_Init", (PYTHON_PROC*)&py3__PyObject_Init}, 577bd5e15fdSBram Moolenaar {"PyModule_AddObject", (PYTHON_PROC*)&py3_PyModule_AddObject}, 578bd5e15fdSBram Moolenaar {"PyImport_AppendInittab", (PYTHON_PROC*)&py3_PyImport_AppendInittab}, 5797bc4f93cSBram Moolenaar # if PY_VERSION_HEX >= 0x030300f0 5809c9cbf13SBram Moolenaar {"PyUnicode_AsUTF8", (PYTHON_PROC*)&py3_PyUnicode_AsUTF8}, 5817bc4f93cSBram Moolenaar # else 582bd5e15fdSBram Moolenaar {"_PyUnicode_AsString", (PYTHON_PROC*)&py3__PyUnicode_AsString}, 5837bc4f93cSBram Moolenaar # endif 5841a3b5695SBram Moolenaar # ifndef Py_UNICODE_USE_UCS_FUNCTIONS 5851a3b5695SBram Moolenaar {"PyUnicode_FromFormat", (PYTHON_PROC*)&py3_PyUnicode_FromFormat}, 5861a3b5695SBram Moolenaar # else 5871a3b5695SBram Moolenaar # ifdef Py_UNICODE_WIDE 5881a3b5695SBram Moolenaar {"PyUnicodeUCS4_FromFormat", (PYTHON_PROC*)&py3_PyUnicodeUCS4_FromFormat}, 5891a3b5695SBram Moolenaar # else 5901a3b5695SBram Moolenaar {"PyUnicodeUCS2_FromFormat", (PYTHON_PROC*)&py3_PyUnicodeUCS2_FromFormat}, 5911a3b5695SBram Moolenaar # endif 5921a3b5695SBram Moolenaar # endif 59319e60943SBram Moolenaar {"PyBytes_AsString", (PYTHON_PROC*)&py3_PyBytes_AsString}, 594cdab9051SBram Moolenaar {"PyBytes_AsStringAndSize", (PYTHON_PROC*)&py3_PyBytes_AsStringAndSize}, 595db913953SBram Moolenaar {"PyBytes_FromString", (PYTHON_PROC*)&py3_PyBytes_FromString}, 5966e5ea8d2SBram Moolenaar {"PyBytes_FromStringAndSize", (PYTHON_PROC*)&py3_PyBytes_FromStringAndSize}, 597ee1b9316SBram Moolenaar # if defined(Py_DEBUG) || PY_VERSION_HEX >= 0x030900b0 598ee1b9316SBram Moolenaar {"_Py_Dealloc", (PYTHON_PROC*)&py3__Py_Dealloc}, 599ee1b9316SBram Moolenaar # endif 600ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 601ee1b9316SBram Moolenaar {"_PyObject_New", (PYTHON_PROC*)&py3__PyObject_New}, 602ee1b9316SBram Moolenaar # endif 603db913953SBram Moolenaar {"PyFloat_FromDouble", (PYTHON_PROC*)&py3_PyFloat_FromDouble}, 604db913953SBram Moolenaar {"PyFloat_AsDouble", (PYTHON_PROC*)&py3_PyFloat_AsDouble}, 605bd5e15fdSBram Moolenaar {"PyObject_GenericGetAttr", (PYTHON_PROC*)&py3_PyObject_GenericGetAttr}, 606bd5e15fdSBram Moolenaar {"PyType_GenericAlloc", (PYTHON_PROC*)&py3_PyType_GenericAlloc}, 607bd5e15fdSBram Moolenaar {"PyType_GenericNew", (PYTHON_PROC*)&py3_PyType_GenericNew}, 60866b7985eSBram Moolenaar {"PyType_Type", (PYTHON_PROC*)&py3_PyType_Type}, 609d4a8c98eSBram Moolenaar {"PyStdPrinter_Type", (PYTHON_PROC*)&py3_PyStdPrinter_Type}, 610bd5e15fdSBram Moolenaar {"PySlice_Type", (PYTHON_PROC*)&py3_PySlice_Type}, 611db913953SBram Moolenaar {"PyFloat_Type", (PYTHON_PROC*)&py3_PyFloat_Type}, 61266b7985eSBram Moolenaar {"PyBool_Type", (PYTHON_PROC*)&py3_PyBool_Type}, 613141be8a5SBram Moolenaar {"PyNumber_Check", (PYTHON_PROC*)&py3_PyNumber_Check}, 614141be8a5SBram Moolenaar {"PyNumber_Long", (PYTHON_PROC*)&py3_PyNumber_Long}, 61519e60943SBram Moolenaar {"PyErr_NewException", (PYTHON_PROC*)&py3_PyErr_NewException}, 616bd5e15fdSBram Moolenaar # ifdef Py_DEBUG 617bd5e15fdSBram Moolenaar {"_Py_NegativeRefcount", (PYTHON_PROC*)&py3__Py_NegativeRefcount}, 618bd5e15fdSBram Moolenaar {"_Py_RefTotal", (PYTHON_PROC*)&py3__Py_RefTotal}, 6190014a53aSBram Moolenaar {"PyModule_Create2TraceRefs", (PYTHON_PROC*)&py3_PyModule_Create2TraceRefs}, 6200014a53aSBram Moolenaar # else 6210014a53aSBram Moolenaar {"PyModule_Create2", (PYTHON_PROC*)&py3_PyModule_Create2}, 6220014a53aSBram Moolenaar # endif 6230014a53aSBram Moolenaar # if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 624bd5e15fdSBram Moolenaar {"_PyObject_DebugFree", (PYTHON_PROC*)&py3__PyObject_DebugFree}, 625bd5e15fdSBram Moolenaar {"_PyObject_DebugMalloc", (PYTHON_PROC*)&py3__PyObject_DebugMalloc}, 626bd5e15fdSBram Moolenaar # else 627bd5e15fdSBram Moolenaar {"PyObject_Malloc", (PYTHON_PROC*)&py3_PyObject_Malloc}, 628bd5e15fdSBram Moolenaar {"PyObject_Free", (PYTHON_PROC*)&py3_PyObject_Free}, 629bd5e15fdSBram Moolenaar # endif 630774267bbSBram Moolenaar {"_PyObject_GC_New", (PYTHON_PROC*)&py3__PyObject_GC_New}, 631774267bbSBram Moolenaar {"PyObject_GC_Del", (PYTHON_PROC*)&py3_PyObject_GC_Del}, 632774267bbSBram Moolenaar {"PyObject_GC_UnTrack", (PYTHON_PROC*)&py3_PyObject_GC_UnTrack}, 633db913953SBram Moolenaar {"PyType_IsSubtype", (PYTHON_PROC*)&py3_PyType_IsSubtype}, 634db913953SBram Moolenaar {"PyCapsule_New", (PYTHON_PROC*)&py3_PyCapsule_New}, 635db913953SBram Moolenaar {"PyCapsule_GetPointer", (PYTHON_PROC*)&py3_PyCapsule_GetPointer}, 636bd5e15fdSBram Moolenaar {"", NULL}, 637bd5e15fdSBram Moolenaar }; 638bd5e15fdSBram Moolenaar 63913a1f3fbSBram Moolenaar # if PY_VERSION_HEX >= 0x030800f0 64013a1f3fbSBram Moolenaar static inline void 64113a1f3fbSBram Moolenaar py3__Py_DECREF(const char *filename UNUSED, int lineno UNUSED, PyObject *op) 64213a1f3fbSBram Moolenaar { 64313a1f3fbSBram Moolenaar if (--op->ob_refcnt != 0) 64413a1f3fbSBram Moolenaar { 64513a1f3fbSBram Moolenaar # ifdef Py_REF_DEBUG 64613a1f3fbSBram Moolenaar if (op->ob_refcnt < 0) 64713a1f3fbSBram Moolenaar { 64813a1f3fbSBram Moolenaar _Py_NegativeRefcount(filename, lineno, op); 64913a1f3fbSBram Moolenaar } 65013a1f3fbSBram Moolenaar # endif 65113a1f3fbSBram Moolenaar } 65213a1f3fbSBram Moolenaar else 65313a1f3fbSBram Moolenaar { 65413a1f3fbSBram Moolenaar _Py_Dealloc(op); 65513a1f3fbSBram Moolenaar } 65613a1f3fbSBram Moolenaar } 65713a1f3fbSBram Moolenaar 65813a1f3fbSBram Moolenaar # undef Py_DECREF 65913a1f3fbSBram Moolenaar # define Py_DECREF(op) py3__Py_DECREF(__FILE__, __LINE__, _PyObject_CAST(op)) 66013a1f3fbSBram Moolenaar 66113a1f3fbSBram Moolenaar static inline void 66213a1f3fbSBram Moolenaar py3__Py_XDECREF(PyObject *op) 66313a1f3fbSBram Moolenaar { 66413a1f3fbSBram Moolenaar if (op != NULL) 66513a1f3fbSBram Moolenaar { 66613a1f3fbSBram Moolenaar Py_DECREF(op); 66713a1f3fbSBram Moolenaar } 66813a1f3fbSBram Moolenaar } 66913a1f3fbSBram Moolenaar 67013a1f3fbSBram Moolenaar # undef Py_XDECREF 67113a1f3fbSBram Moolenaar # define Py_XDECREF(op) py3__Py_XDECREF(_PyObject_CAST(op)) 67213a1f3fbSBram Moolenaar # endif 67313a1f3fbSBram Moolenaar 674ee1b9316SBram Moolenaar # if PY_VERSION_HEX >= 0x030900b0 675ee1b9316SBram Moolenaar static inline int 676ee1b9316SBram Moolenaar py3_PyType_HasFeature(PyTypeObject *type, unsigned long feature) 677ee1b9316SBram Moolenaar { 678ee1b9316SBram Moolenaar return ((PyType_GetFlags(type) & feature) != 0); 679ee1b9316SBram Moolenaar } 680ee1b9316SBram Moolenaar # define PyType_HasFeature(t,f) py3_PyType_HasFeature(t,f) 681ee1b9316SBram Moolenaar # endif 682ee1b9316SBram Moolenaar 683*90478f35SZdenek Dohnal # if PY_VERSION_HEX >= 0x030a00b2 684*90478f35SZdenek Dohnal static inline int 685*90478f35SZdenek Dohnal py3__PyObject_TypeCheck(PyObject *ob, PyTypeObject *type) 686*90478f35SZdenek Dohnal { 687*90478f35SZdenek Dohnal return Py_IS_TYPE(ob, type) || PyType_IsSubtype(Py_TYPE(ob), type); 688*90478f35SZdenek Dohnal } 689*90478f35SZdenek Dohnal # define _PyObject_TypeCheck(o,t) py3__PyObject_TypeCheck(o,t) 690*90478f35SZdenek Dohnal # endif 691*90478f35SZdenek Dohnal 692b2f9e0e2SBram Moolenaar # ifdef MSWIN 693b2f9e0e2SBram Moolenaar /* 694b2f9e0e2SBram Moolenaar * Look up the library "libname" using the InstallPath registry key. 695b2f9e0e2SBram Moolenaar * Return NULL when failed. Return an allocated string when successful. 696b2f9e0e2SBram Moolenaar */ 697b2f9e0e2SBram Moolenaar static char * 698b2f9e0e2SBram Moolenaar py3_get_system_libname(const char *libname) 699b2f9e0e2SBram Moolenaar { 700b2f9e0e2SBram Moolenaar const char *cp = libname; 701b2f9e0e2SBram Moolenaar char subkey[128]; 702b2f9e0e2SBram Moolenaar HKEY hKey; 703b2f9e0e2SBram Moolenaar char installpath[MAXPATHL]; 704b2f9e0e2SBram Moolenaar LONG len = sizeof(installpath); 705b2f9e0e2SBram Moolenaar LSTATUS rc; 706b2f9e0e2SBram Moolenaar size_t sysliblen; 707b2f9e0e2SBram Moolenaar char *syslibname; 708b2f9e0e2SBram Moolenaar 709b2f9e0e2SBram Moolenaar while (*cp != '\0') 710b2f9e0e2SBram Moolenaar { 711b2f9e0e2SBram Moolenaar if (*cp == ':' || *cp == '\\' || *cp == '/') 712b2f9e0e2SBram Moolenaar { 713b2f9e0e2SBram Moolenaar // Bail out if "libname" contains path separator, assume it is 714b2f9e0e2SBram Moolenaar // an absolute path. 715b2f9e0e2SBram Moolenaar return NULL; 716b2f9e0e2SBram Moolenaar } 717b2f9e0e2SBram Moolenaar ++cp; 718b2f9e0e2SBram Moolenaar } 719b2f9e0e2SBram Moolenaar vim_snprintf(subkey, sizeof(subkey), 720b2f9e0e2SBram Moolenaar # ifdef _WIN64 721b2f9e0e2SBram Moolenaar "Software\\Python\\PythonCore\\%d.%d\\InstallPath", 722b2f9e0e2SBram Moolenaar # else 723b2f9e0e2SBram Moolenaar "Software\\Python\\PythonCore\\%d.%d-32\\InstallPath", 724b2f9e0e2SBram Moolenaar # endif 725b2f9e0e2SBram Moolenaar PY_MAJOR_VERSION, PY_MINOR_VERSION); 726b2f9e0e2SBram Moolenaar if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, subkey, 0, KEY_QUERY_VALUE, &hKey) 727b2f9e0e2SBram Moolenaar != ERROR_SUCCESS) 728b2f9e0e2SBram Moolenaar return NULL; 729b2f9e0e2SBram Moolenaar rc = RegQueryValueA(hKey, NULL, installpath, &len); 730b2f9e0e2SBram Moolenaar RegCloseKey(hKey); 731b2f9e0e2SBram Moolenaar if (ERROR_SUCCESS != rc) 732b2f9e0e2SBram Moolenaar return NULL; 733b2f9e0e2SBram Moolenaar cp = installpath + len; 734b2f9e0e2SBram Moolenaar // Just in case registry value contains null terminators. 735b2f9e0e2SBram Moolenaar while (cp > installpath && *(cp-1) == '\0') 736b2f9e0e2SBram Moolenaar --cp; 737b2f9e0e2SBram Moolenaar // Remove trailing path separators. 738b2f9e0e2SBram Moolenaar while (cp > installpath && (*(cp-1) == '\\' || *(cp-1) == '/')) 739b2f9e0e2SBram Moolenaar --cp; 740b2f9e0e2SBram Moolenaar // Ignore if InstallPath is effectively empty. 741b2f9e0e2SBram Moolenaar if (cp <= installpath) 742b2f9e0e2SBram Moolenaar return NULL; 743b2f9e0e2SBram Moolenaar sysliblen = (cp - installpath) + 1 + STRLEN(libname) + 1; 744b2f9e0e2SBram Moolenaar syslibname = alloc(sysliblen); 745b2f9e0e2SBram Moolenaar vim_snprintf(syslibname, sysliblen, "%.*s\\%s", 746b2f9e0e2SBram Moolenaar (int)(cp - installpath), installpath, libname); 747b2f9e0e2SBram Moolenaar return syslibname; 748b2f9e0e2SBram Moolenaar } 749b2f9e0e2SBram Moolenaar # endif 750b2f9e0e2SBram Moolenaar 751bd5e15fdSBram Moolenaar /* 752bd5e15fdSBram Moolenaar * Load library and get all pointers. 753bd5e15fdSBram Moolenaar * Parameter 'libname' provides name of DLL. 754bd5e15fdSBram Moolenaar * Return OK or FAIL. 755bd5e15fdSBram Moolenaar */ 756170bf1aeSBram Moolenaar static int 757170bf1aeSBram Moolenaar py3_runtime_link_init(char *libname, int verbose) 758bd5e15fdSBram Moolenaar { 759bd5e15fdSBram Moolenaar int i; 7607b24ce08SBram Moolenaar PYTHON_PROC *ucs_from_string = (PYTHON_PROC *)&py3_PyUnicode_FromString; 7617b24ce08SBram Moolenaar PYTHON_PROC *ucs_decode = (PYTHON_PROC *)&py3_PyUnicode_Decode; 7627b24ce08SBram Moolenaar PYTHON_PROC *ucs_as_encoded_string = 7637b24ce08SBram Moolenaar (PYTHON_PROC *)&py3_PyUnicode_AsEncodedString; 764bd5e15fdSBram Moolenaar 765644d37b8SBram Moolenaar # if !(defined(PY_NO_RTLD_GLOBAL) && defined(PY3_NO_RTLD_GLOBAL)) && defined(UNIX) && defined(FEAT_PYTHON) 7662ab2e860SBram Moolenaar // Can't have Python and Python3 loaded at the same time. 7672ab2e860SBram Moolenaar // It cause a crash, because RTLD_GLOBAL is needed for 7682ab2e860SBram Moolenaar // standard C extension libraries of one or both python versions. 7694c3a326cSBram Moolenaar if (python_loaded()) 7704c3a326cSBram Moolenaar { 7719dc93ae4SBram Moolenaar if (verbose) 772f9e3e09fSBram Moolenaar emsg(_("E837: This Vim cannot execute :py3 after using :python")); 7734c3a326cSBram Moolenaar return FAIL; 7744c3a326cSBram Moolenaar } 7754c3a326cSBram Moolenaar # endif 7764c3a326cSBram Moolenaar 7774c3a326cSBram Moolenaar if (hinstPy3 != 0) 778bd5e15fdSBram Moolenaar return OK; 779bd5e15fdSBram Moolenaar hinstPy3 = load_dll(libname); 780bd5e15fdSBram Moolenaar 781b2f9e0e2SBram Moolenaar # ifdef MSWIN 782b2f9e0e2SBram Moolenaar if (!hinstPy3) 783b2f9e0e2SBram Moolenaar { 784b2f9e0e2SBram Moolenaar // Attempt to use the path from InstallPath as stored in the registry. 785b2f9e0e2SBram Moolenaar char *syslibname = py3_get_system_libname(libname); 786b2f9e0e2SBram Moolenaar 787b2f9e0e2SBram Moolenaar if (syslibname != NULL) 788b2f9e0e2SBram Moolenaar { 789b2f9e0e2SBram Moolenaar hinstPy3 = load_dll(syslibname); 790b2f9e0e2SBram Moolenaar vim_free(syslibname); 791b2f9e0e2SBram Moolenaar } 792b2f9e0e2SBram Moolenaar } 793b2f9e0e2SBram Moolenaar # endif 794b2f9e0e2SBram Moolenaar 795bd5e15fdSBram Moolenaar if (!hinstPy3) 796bd5e15fdSBram Moolenaar { 797bd5e15fdSBram Moolenaar if (verbose) 798f9e3e09fSBram Moolenaar semsg(_(e_loadlib), libname); 799bd5e15fdSBram Moolenaar return FAIL; 800bd5e15fdSBram Moolenaar } 801bd5e15fdSBram Moolenaar 802bd5e15fdSBram Moolenaar for (i = 0; py3_funcname_table[i].ptr; ++i) 803bd5e15fdSBram Moolenaar { 804bd5e15fdSBram Moolenaar if ((*py3_funcname_table[i].ptr = symbol_from_dll(hinstPy3, 805bd5e15fdSBram Moolenaar py3_funcname_table[i].name)) == NULL) 806bd5e15fdSBram Moolenaar { 807bd5e15fdSBram Moolenaar close_dll(hinstPy3); 808bd5e15fdSBram Moolenaar hinstPy3 = 0; 809bd5e15fdSBram Moolenaar if (verbose) 810f9e3e09fSBram Moolenaar semsg(_(e_loadfunc), py3_funcname_table[i].name); 811bd5e15fdSBram Moolenaar return FAIL; 812bd5e15fdSBram Moolenaar } 813bd5e15fdSBram Moolenaar } 814bd5e15fdSBram Moolenaar 8152ab2e860SBram Moolenaar // Load unicode functions separately as only the ucs2 or the ucs4 functions 8162ab2e860SBram Moolenaar // will be present in the library. 8177bc4f93cSBram Moolenaar # if PY_VERSION_HEX >= 0x030300f0 8187b24ce08SBram Moolenaar *ucs_from_string = symbol_from_dll(hinstPy3, "PyUnicode_FromString"); 8197b24ce08SBram Moolenaar *ucs_decode = symbol_from_dll(hinstPy3, "PyUnicode_Decode"); 8207b24ce08SBram Moolenaar *ucs_as_encoded_string = symbol_from_dll(hinstPy3, 8217bc4f93cSBram Moolenaar "PyUnicode_AsEncodedString"); 8227bc4f93cSBram Moolenaar # else 8237b24ce08SBram Moolenaar *ucs_from_string = symbol_from_dll(hinstPy3, "PyUnicodeUCS2_FromString"); 8247b24ce08SBram Moolenaar *ucs_decode = symbol_from_dll(hinstPy3, 82519e60943SBram Moolenaar "PyUnicodeUCS2_Decode"); 8267b24ce08SBram Moolenaar *ucs_as_encoded_string = symbol_from_dll(hinstPy3, 82719e60943SBram Moolenaar "PyUnicodeUCS2_AsEncodedString"); 8287b24ce08SBram Moolenaar if (*ucs_from_string == NULL || *ucs_decode == NULL 8297b24ce08SBram Moolenaar || *ucs_as_encoded_string == NULL) 830bd5e15fdSBram Moolenaar { 8317b24ce08SBram Moolenaar *ucs_from_string = symbol_from_dll(hinstPy3, 832bd5e15fdSBram Moolenaar "PyUnicodeUCS4_FromString"); 8337b24ce08SBram Moolenaar *ucs_decode = symbol_from_dll(hinstPy3, 83419e60943SBram Moolenaar "PyUnicodeUCS4_Decode"); 8357b24ce08SBram Moolenaar *ucs_as_encoded_string = symbol_from_dll(hinstPy3, 83619e60943SBram Moolenaar "PyUnicodeUCS4_AsEncodedString"); 837bd5e15fdSBram Moolenaar } 8387bc4f93cSBram Moolenaar # endif 8397b24ce08SBram Moolenaar if (*ucs_from_string == NULL || *ucs_decode == NULL 8407b24ce08SBram Moolenaar || *ucs_as_encoded_string == NULL) 841bd5e15fdSBram Moolenaar { 842bd5e15fdSBram Moolenaar close_dll(hinstPy3); 843bd5e15fdSBram Moolenaar hinstPy3 = 0; 844bd5e15fdSBram Moolenaar if (verbose) 845f9e3e09fSBram Moolenaar semsg(_(e_loadfunc), "PyUnicode_UCSX_*"); 846bd5e15fdSBram Moolenaar return FAIL; 847bd5e15fdSBram Moolenaar } 848bd5e15fdSBram Moolenaar 849bd5e15fdSBram Moolenaar return OK; 850bd5e15fdSBram Moolenaar } 851bd5e15fdSBram Moolenaar 852bd5e15fdSBram Moolenaar /* 853bd5e15fdSBram Moolenaar * If python is enabled (there is installed python on Windows system) return 854bd5e15fdSBram Moolenaar * TRUE, else FALSE. 855bd5e15fdSBram Moolenaar */ 856170bf1aeSBram Moolenaar int 857170bf1aeSBram Moolenaar python3_enabled(int verbose) 858bd5e15fdSBram Moolenaar { 85925e4fcdeSBram Moolenaar return py3_runtime_link_init((char *)p_py3dll, verbose) == OK; 860bd5e15fdSBram Moolenaar } 861bd5e15fdSBram Moolenaar 8622ab2e860SBram Moolenaar /* 8632ab2e860SBram Moolenaar * Load the standard Python exceptions - don't import the symbols from the 864bd5e15fdSBram Moolenaar * DLL, as this can cause errors (importing data symbols is not reliable). 865bd5e15fdSBram Moolenaar */ 866170bf1aeSBram Moolenaar static void 86768c2f638SBram Moolenaar get_py3_exceptions(void) 868bd5e15fdSBram Moolenaar { 869bd5e15fdSBram Moolenaar PyObject *exmod = PyImport_ImportModule("builtins"); 870bd5e15fdSBram Moolenaar PyObject *exdict = PyModule_GetDict(exmod); 871bd5e15fdSBram Moolenaar p3imp_PyExc_AttributeError = PyDict_GetItemString(exdict, "AttributeError"); 872bd5e15fdSBram Moolenaar p3imp_PyExc_IndexError = PyDict_GetItemString(exdict, "IndexError"); 873af6abb9dSBram Moolenaar p3imp_PyExc_KeyError = PyDict_GetItemString(exdict, "KeyError"); 874bd5e15fdSBram Moolenaar p3imp_PyExc_KeyboardInterrupt = PyDict_GetItemString(exdict, "KeyboardInterrupt"); 875bd5e15fdSBram Moolenaar p3imp_PyExc_TypeError = PyDict_GetItemString(exdict, "TypeError"); 876bd5e15fdSBram Moolenaar p3imp_PyExc_ValueError = PyDict_GetItemString(exdict, "ValueError"); 87741009374SBram Moolenaar p3imp_PyExc_SystemExit = PyDict_GetItemString(exdict, "SystemExit"); 8788661b178SBram Moolenaar p3imp_PyExc_RuntimeError = PyDict_GetItemString(exdict, "RuntimeError"); 879c09a6d6cSBram Moolenaar p3imp_PyExc_ImportError = PyDict_GetItemString(exdict, "ImportError"); 880141be8a5SBram Moolenaar p3imp_PyExc_OverflowError = PyDict_GetItemString(exdict, "OverflowError"); 881bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_AttributeError); 882bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_IndexError); 883af6abb9dSBram Moolenaar Py_XINCREF(p3imp_PyExc_KeyError); 884bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_KeyboardInterrupt); 885bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_TypeError); 886bd5e15fdSBram Moolenaar Py_XINCREF(p3imp_PyExc_ValueError); 88741009374SBram Moolenaar Py_XINCREF(p3imp_PyExc_SystemExit); 8888661b178SBram Moolenaar Py_XINCREF(p3imp_PyExc_RuntimeError); 889c09a6d6cSBram Moolenaar Py_XINCREF(p3imp_PyExc_ImportError); 890141be8a5SBram Moolenaar Py_XINCREF(p3imp_PyExc_OverflowError); 891bd5e15fdSBram Moolenaar Py_XDECREF(exmod); 892bd5e15fdSBram Moolenaar } 8932ab2e860SBram Moolenaar #endif // DYNAMIC_PYTHON3 894bd5e15fdSBram Moolenaar 895db913953SBram Moolenaar static int py3initialised = 0; 896db913953SBram Moolenaar #define PYINITIALISED py3initialised 897c4f83380SBram Moolenaar static int python_end_called = FALSE; 898db913953SBram Moolenaar 899774267bbSBram Moolenaar #define DESTRUCTOR_FINISH(self) Py_TYPE(self)->tp_free((PyObject*)self) 900db913953SBram Moolenaar 901971db467SBram Moolenaar #define WIN_PYTHON_REF(win) win->w_python3_ref 902971db467SBram Moolenaar #define BUF_PYTHON_REF(buf) buf->b_python3_ref 9035e538ecdSBram Moolenaar #define TAB_PYTHON_REF(tab) tab->tp_python3_ref 904971db467SBram Moolenaar 905170bf1aeSBram Moolenaar static void 906170bf1aeSBram Moolenaar call_PyObject_Free(void *p) 907bd5e15fdSBram Moolenaar { 9080014a53aSBram Moolenaar #if defined(Py_DEBUG) && !defined(Py_DEBUG_NO_PYMALLOC) 909bd5e15fdSBram Moolenaar _PyObject_DebugFree(p); 910bd5e15fdSBram Moolenaar #else 911bd5e15fdSBram Moolenaar PyObject_Free(p); 912bd5e15fdSBram Moolenaar #endif 913bd5e15fdSBram Moolenaar } 914170bf1aeSBram Moolenaar 915170bf1aeSBram Moolenaar static PyObject * 916170bf1aeSBram Moolenaar call_PyType_GenericNew(PyTypeObject *type, PyObject *args, PyObject *kwds) 917bd5e15fdSBram Moolenaar { 918bd5e15fdSBram Moolenaar return PyType_GenericNew(type,args,kwds); 919bd5e15fdSBram Moolenaar } 920170bf1aeSBram Moolenaar 921170bf1aeSBram Moolenaar static PyObject * 922170bf1aeSBram Moolenaar call_PyType_GenericAlloc(PyTypeObject *type, Py_ssize_t nitems) 923bd5e15fdSBram Moolenaar { 924bd5e15fdSBram Moolenaar return PyType_GenericAlloc(type,nitems); 925bd5e15fdSBram Moolenaar } 926bd5e15fdSBram Moolenaar 9274d1da49cSBram Moolenaar static PyObject *OutputGetattro(PyObject *, PyObject *); 9284d1da49cSBram Moolenaar static int OutputSetattro(PyObject *, PyObject *, PyObject *); 9294d1da49cSBram Moolenaar static PyObject *BufferGetattro(PyObject *, PyObject *); 930e9ba516bSBram Moolenaar static int BufferSetattro(PyObject *, PyObject *, PyObject *); 9315e538ecdSBram Moolenaar static PyObject *TabPageGetattro(PyObject *, PyObject *); 9324d1da49cSBram Moolenaar static PyObject *WindowGetattro(PyObject *, PyObject *); 9334d1da49cSBram Moolenaar static int WindowSetattro(PyObject *, PyObject *, PyObject *); 9344d1da49cSBram Moolenaar static PyObject *RangeGetattro(PyObject *, PyObject *); 9354d1da49cSBram Moolenaar static PyObject *CurrentGetattro(PyObject *, PyObject *); 9364d1da49cSBram Moolenaar static int CurrentSetattro(PyObject *, PyObject *, PyObject *); 9374d1da49cSBram Moolenaar static PyObject *DictionaryGetattro(PyObject *, PyObject *); 9384d1da49cSBram Moolenaar static int DictionarySetattro(PyObject *, PyObject *, PyObject *); 9394d1da49cSBram Moolenaar static PyObject *ListGetattro(PyObject *, PyObject *); 9404d1da49cSBram Moolenaar static int ListSetattro(PyObject *, PyObject *, PyObject *); 9414d1da49cSBram Moolenaar static PyObject *FunctionGetattro(PyObject *, PyObject *); 9424d1da49cSBram Moolenaar 9434d1da49cSBram Moolenaar static struct PyModuleDef vimmodule; 9444d1da49cSBram Moolenaar 9452a0f3d3fSBram Moolenaar #define PY_CAN_RECURSE 9462a0f3d3fSBram Moolenaar 9474d1da49cSBram Moolenaar /* 9484d1da49cSBram Moolenaar * Include the code shared with if_python.c 9494d1da49cSBram Moolenaar */ 9504d1da49cSBram Moolenaar #include "if_py_both.h" 9514d1da49cSBram Moolenaar 952828bff1fSBram Moolenaar // NOTE: Must always be used at the start of a block, since it declares "name". 9534d1da49cSBram Moolenaar #define GET_ATTR_STRING(name, nameobj) \ 9544d1da49cSBram Moolenaar char *name = ""; \ 9554d1da49cSBram Moolenaar if (PyUnicode_Check(nameobj)) \ 956828bff1fSBram Moolenaar name = (char *)_PyUnicode_AsString(nameobj) 9574d1da49cSBram Moolenaar 9584d1da49cSBram Moolenaar #define PY3OBJ_DELETED(obj) (obj->ob_base.ob_refcnt<=0) 9594d1da49cSBram Moolenaar 9602ab2e860SBram Moolenaar /////////////////////////////////////////////////////// 9612ab2e860SBram Moolenaar // Internal function prototypes. 962bd5e15fdSBram Moolenaar 9637854e3abSBram Moolenaar static PyObject *Py3Init_vim(void); 964bd5e15fdSBram Moolenaar 9652ab2e860SBram Moolenaar /////////////////////////////////////////////////////// 9662ab2e860SBram Moolenaar // 1. Python interpreter main program. 967bd5e15fdSBram Moolenaar 968170bf1aeSBram Moolenaar void 96968c2f638SBram Moolenaar python3_end(void) 970bd5e15fdSBram Moolenaar { 971bd5e15fdSBram Moolenaar static int recurse = 0; 972bd5e15fdSBram Moolenaar 9732ab2e860SBram Moolenaar // If a crash occurs while doing this, don't try again. 974bd5e15fdSBram Moolenaar if (recurse != 0) 975bd5e15fdSBram Moolenaar return; 976bd5e15fdSBram Moolenaar 977c4f83380SBram Moolenaar python_end_called = TRUE; 978bd5e15fdSBram Moolenaar ++recurse; 979bd5e15fdSBram Moolenaar 980bd5e15fdSBram Moolenaar #ifdef DYNAMIC_PYTHON3 981bd5e15fdSBram Moolenaar if (hinstPy3) 982bd5e15fdSBram Moolenaar #endif 983bd5e15fdSBram Moolenaar if (Py_IsInitialized()) 984bd5e15fdSBram Moolenaar { 9852ab2e860SBram Moolenaar // acquire lock before finalizing 98671700b89SBram Moolenaar PyGILState_Ensure(); 987bd5e15fdSBram Moolenaar 988bd5e15fdSBram Moolenaar Py_Finalize(); 989bd5e15fdSBram Moolenaar } 990bd5e15fdSBram Moolenaar 991bd5e15fdSBram Moolenaar --recurse; 992bd5e15fdSBram Moolenaar } 993bd5e15fdSBram Moolenaar 994094454faSBram Moolenaar #if (defined(DYNAMIC_PYTHON3) && defined(DYNAMIC_PYTHON) && defined(FEAT_PYTHON) && defined(UNIX)) || defined(PROTO) 9954c3a326cSBram Moolenaar int 99668c2f638SBram Moolenaar python3_loaded(void) 9974c3a326cSBram Moolenaar { 9984c3a326cSBram Moolenaar return (hinstPy3 != 0); 9994c3a326cSBram Moolenaar } 10004c3a326cSBram Moolenaar #endif 10014c3a326cSBram Moolenaar 100294073167SBram Moolenaar static wchar_t *py_home_buf = NULL; 100394073167SBram Moolenaar 100456b8dc33SBram Moolenaar #if defined(MSWIN) && (PY_VERSION_HEX >= 0x030500f0) 1005c6ed254dSBram Moolenaar /* 1006c6ed254dSBram Moolenaar * Return TRUE if stdin is readable from Python 3. 1007c6ed254dSBram Moolenaar */ 1008c6ed254dSBram Moolenaar static BOOL 1009c6ed254dSBram Moolenaar is_stdin_readable(void) 1010c6ed254dSBram Moolenaar { 1011c6ed254dSBram Moolenaar DWORD mode, eventnum; 1012c6ed254dSBram Moolenaar struct _stat st; 1013c6ed254dSBram Moolenaar int fd = fileno(stdin); 1014c6ed254dSBram Moolenaar HANDLE hstdin = (HANDLE)_get_osfhandle(fd); 1015c6ed254dSBram Moolenaar 1016c6ed254dSBram Moolenaar // Check if stdin is connected to the console. 1017c6ed254dSBram Moolenaar if (GetConsoleMode(hstdin, &mode)) 1018c6ed254dSBram Moolenaar // Check if it is opened as input. 1019c6ed254dSBram Moolenaar return GetNumberOfConsoleInputEvents(hstdin, &eventnum); 1020c6ed254dSBram Moolenaar 1021c6ed254dSBram Moolenaar return _fstat(fd, &st) == 0; 1022c6ed254dSBram Moolenaar } 1023c6ed254dSBram Moolenaar 1024c6ed254dSBram Moolenaar // Python 3.5 or later will abort inside Py_Initialize() when stdin has 1025c6ed254dSBram Moolenaar // been closed (i.e. executed by "vim -"). Reconnect stdin to CONIN$. 102656b8dc33SBram Moolenaar // Note that the python DLL is linked to its own stdio DLL which can be 102756b8dc33SBram Moolenaar // differ from Vim's stdio. 102856b8dc33SBram Moolenaar static void 102956b8dc33SBram Moolenaar reset_stdin(void) 103056b8dc33SBram Moolenaar { 103156b8dc33SBram Moolenaar FILE *(*py__acrt_iob_func)(unsigned) = NULL; 103256b8dc33SBram Moolenaar FILE *(*pyfreopen)(const char *, const char *, FILE *) = NULL; 103356b8dc33SBram Moolenaar HINSTANCE hinst; 103456b8dc33SBram Moolenaar 103556b8dc33SBram Moolenaar # ifdef DYNAMIC_PYTHON3 103656b8dc33SBram Moolenaar hinst = hinstPy3; 103756b8dc33SBram Moolenaar # else 103856b8dc33SBram Moolenaar hinst = GetModuleHandle(PYTHON3_DLL); 103956b8dc33SBram Moolenaar # endif 1040c6ed254dSBram Moolenaar if (hinst == NULL || is_stdin_readable()) 104156b8dc33SBram Moolenaar return; 104256b8dc33SBram Moolenaar 104356b8dc33SBram Moolenaar // Get "freopen" and "stdin" which are used in the python DLL. 104456b8dc33SBram Moolenaar // "stdin" is defined as "__acrt_iob_func(0)" in VC++ 2015 or later. 104556b8dc33SBram Moolenaar py__acrt_iob_func = get_dll_import_func(hinst, "__acrt_iob_func"); 104656b8dc33SBram Moolenaar if (py__acrt_iob_func) 104756b8dc33SBram Moolenaar { 104856b8dc33SBram Moolenaar HINSTANCE hpystdiodll = find_imported_module_by_funcname(hinst, 104956b8dc33SBram Moolenaar "__acrt_iob_func"); 105056b8dc33SBram Moolenaar if (hpystdiodll) 105156b8dc33SBram Moolenaar pyfreopen = (void *)GetProcAddress(hpystdiodll, "freopen"); 105256b8dc33SBram Moolenaar } 105356b8dc33SBram Moolenaar 1054c6ed254dSBram Moolenaar // Reconnect stdin to CONIN$. 1055253b16a4SBram Moolenaar if (pyfreopen != NULL) 1056c6ed254dSBram Moolenaar pyfreopen("CONIN$", "r", py__acrt_iob_func(0)); 105756b8dc33SBram Moolenaar else 1058c6ed254dSBram Moolenaar freopen("CONIN$", "r", stdin); 105956b8dc33SBram Moolenaar } 106056b8dc33SBram Moolenaar #else 106156b8dc33SBram Moolenaar # define reset_stdin() 106256b8dc33SBram Moolenaar #endif 106356b8dc33SBram Moolenaar 1064170bf1aeSBram Moolenaar static int 1065170bf1aeSBram Moolenaar Python3_Init(void) 1066bd5e15fdSBram Moolenaar { 1067bd5e15fdSBram Moolenaar if (!py3initialised) 1068bd5e15fdSBram Moolenaar { 1069bd5e15fdSBram Moolenaar #ifdef DYNAMIC_PYTHON3 1070bd5e15fdSBram Moolenaar if (!python3_enabled(TRUE)) 1071bd5e15fdSBram Moolenaar { 1072f9e3e09fSBram Moolenaar emsg(_("E263: Sorry, this command is disabled, the Python library could not be loaded.")); 1073bd5e15fdSBram Moolenaar goto fail; 1074bd5e15fdSBram Moolenaar } 1075bd5e15fdSBram Moolenaar #endif 1076bd5e15fdSBram Moolenaar 1077bd5e15fdSBram Moolenaar init_structs(); 1078bd5e15fdSBram Moolenaar 107994073167SBram Moolenaar if (*p_py3home != NUL) 108094073167SBram Moolenaar { 108194073167SBram Moolenaar size_t len = mbstowcs(NULL, (char *)p_py3home, 0) + 1; 1082644d37b8SBram Moolenaar 10832ab2e860SBram Moolenaar // The string must not change later, make a copy in static memory. 1084c799fe20SBram Moolenaar py_home_buf = ALLOC_MULT(wchar_t, len); 108594073167SBram Moolenaar if (py_home_buf != NULL && mbstowcs( 108694073167SBram Moolenaar py_home_buf, (char *)p_py3home, len) != (size_t)-1) 108794073167SBram Moolenaar Py_SetPythonHome(py_home_buf); 108894073167SBram Moolenaar } 1089644d37b8SBram Moolenaar #ifdef PYTHON3_HOME 109094073167SBram Moolenaar else if (mch_getenv((char_u *)"PYTHONHOME") == NULL) 1091644d37b8SBram Moolenaar Py_SetPythonHome(PYTHON3_HOME); 1092644d37b8SBram Moolenaar #endif 1093644d37b8SBram Moolenaar 10947bc4f93cSBram Moolenaar PyImport_AppendInittab("vim", Py3Init_vim); 10957bc4f93cSBram Moolenaar 109656b8dc33SBram Moolenaar reset_stdin(); 1097bd5e15fdSBram Moolenaar Py_Initialize(); 1098d057301bSBram Moolenaar 1099efc0d94aSBram Moolenaar #if PY_VERSION_HEX < 0x03090000 1100efc0d94aSBram Moolenaar // Initialise threads. This is deprecated since Python 3.9. 1101456f2bb2SBram Moolenaar PyEval_InitThreads(); 1102efc0d94aSBram Moolenaar #endif 1103bd5e15fdSBram Moolenaar #ifdef DYNAMIC_PYTHON3 1104bd5e15fdSBram Moolenaar get_py3_exceptions(); 1105bd5e15fdSBram Moolenaar #endif 1106bd5e15fdSBram Moolenaar 11071dc28783SBram Moolenaar if (PythonIO_Init_io()) 1108bd5e15fdSBram Moolenaar goto fail; 1109bd5e15fdSBram Moolenaar 1110db913953SBram Moolenaar globals = PyModule_GetDict(PyImport_AddModule("__main__")); 1111db913953SBram Moolenaar 11122ab2e860SBram Moolenaar // Remove the element from sys.path that was added because of our 11132ab2e860SBram Moolenaar // argv[0] value in Py3Init_vim(). Previously we used an empty 11142ab2e860SBram Moolenaar // string, but depending on the OS we then get an empty entry or 11152ab2e860SBram Moolenaar // the current directory in sys.path. 11162ab2e860SBram Moolenaar // Only after vim has been imported, the element does exist in 11172ab2e860SBram Moolenaar // sys.path. 111819e60943SBram Moolenaar PyRun_SimpleString("import vim; import sys; sys.path = list(filter(lambda x: not x.endswith('must>not&exist'), sys.path))"); 1119bd5e15fdSBram Moolenaar 1120efc0d94aSBram Moolenaar // Without the call to PyEval_SaveThread, thread specific state (such 1121efc0d94aSBram Moolenaar // as the system trace hook), will be lost between invocations of 1122efc0d94aSBram Moolenaar // Python code. 1123efc0d94aSBram Moolenaar // GIL may have been created and acquired in PyEval_InitThreads() and 1124efc0d94aSBram Moolenaar // thread state is created in Py_Initialize(); there 1125efc0d94aSBram Moolenaar // _PyGILState_NoteThreadState() also sets gilcounter to 1 (python must 1126efc0d94aSBram Moolenaar // have threads enabled!), so the following does both: unlock GIL and 1127efc0d94aSBram Moolenaar // save thread state in TLS without deleting thread state 112876d711c3SBram Moolenaar PyEval_SaveThread(); 1129bd5e15fdSBram Moolenaar 1130bd5e15fdSBram Moolenaar py3initialised = 1; 1131bd5e15fdSBram Moolenaar } 1132bd5e15fdSBram Moolenaar 1133bd5e15fdSBram Moolenaar return 0; 1134bd5e15fdSBram Moolenaar 1135bd5e15fdSBram Moolenaar fail: 11362ab2e860SBram Moolenaar // We call PythonIO_Flush() here to print any Python errors. 11372ab2e860SBram Moolenaar // This is OK, as it is possible to call this function even 11382ab2e860SBram Moolenaar // if PythonIO_Init_io() has not completed successfully (it will 11392ab2e860SBram Moolenaar // not do anything in this case). 1140bd5e15fdSBram Moolenaar PythonIO_Flush(); 1141bd5e15fdSBram Moolenaar return -1; 1142bd5e15fdSBram Moolenaar } 1143bd5e15fdSBram Moolenaar 1144bd5e15fdSBram Moolenaar /* 1145bd5e15fdSBram Moolenaar * External interface 1146bd5e15fdSBram Moolenaar */ 1147170bf1aeSBram Moolenaar static void 1148b52f4c02SBram Moolenaar DoPyCommand(const char *cmd, rangeinitializer init_range, runner run, void *arg) 1149bd5e15fdSBram Moolenaar { 1150bd5e15fdSBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 1151bd5e15fdSBram Moolenaar char *saved_locale; 1152bd5e15fdSBram Moolenaar #endif 115319e60943SBram Moolenaar PyObject *cmdstr; 115419e60943SBram Moolenaar PyObject *cmdbytes; 115571700b89SBram Moolenaar PyGILState_STATE pygilstate; 1156bd5e15fdSBram Moolenaar 1157c4f83380SBram Moolenaar if (python_end_called) 1158c4f83380SBram Moolenaar goto theend; 1159c4f83380SBram Moolenaar 1160bd5e15fdSBram Moolenaar if (Python3_Init()) 1161bd5e15fdSBram Moolenaar goto theend; 1162bd5e15fdSBram Moolenaar 1163b52f4c02SBram Moolenaar init_range(arg); 1164b52f4c02SBram Moolenaar 11652ab2e860SBram Moolenaar Python_Release_Vim(); // leave Vim 1166bd5e15fdSBram Moolenaar 1167bd5e15fdSBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 11682ab2e860SBram Moolenaar // Python only works properly when the LC_NUMERIC locale is "C". 1169bd5e15fdSBram Moolenaar saved_locale = setlocale(LC_NUMERIC, NULL); 1170bd5e15fdSBram Moolenaar if (saved_locale == NULL || STRCMP(saved_locale, "C") == 0) 1171bd5e15fdSBram Moolenaar saved_locale = NULL; 1172bd5e15fdSBram Moolenaar else 1173bd5e15fdSBram Moolenaar { 11742ab2e860SBram Moolenaar // Need to make a copy, value may change when setting new locale. 1175bd5e15fdSBram Moolenaar saved_locale = (char *)vim_strsave((char_u *)saved_locale); 1176bd5e15fdSBram Moolenaar (void)setlocale(LC_NUMERIC, "C"); 1177bd5e15fdSBram Moolenaar } 1178bd5e15fdSBram Moolenaar #endif 1179bd5e15fdSBram Moolenaar 1180bd5e15fdSBram Moolenaar pygilstate = PyGILState_Ensure(); 1181bd5e15fdSBram Moolenaar 11822ab2e860SBram Moolenaar // PyRun_SimpleString expects a UTF-8 string. Wrong encoding may cause 11832ab2e860SBram Moolenaar // SyntaxError (unicode error). 11843d64a317SBram Moolenaar cmdstr = PyUnicode_Decode(cmd, strlen(cmd), 11852e2f52a4SBram Moolenaar (char *)ENC_OPT, ERRORS_DECODE_ARG); 11862e2f52a4SBram Moolenaar cmdbytes = PyUnicode_AsEncodedString(cmdstr, "utf-8", ERRORS_ENCODE_ARG); 118719e60943SBram Moolenaar Py_XDECREF(cmdstr); 1188db913953SBram Moolenaar 1189b52f4c02SBram Moolenaar run(PyBytes_AsString(cmdbytes), arg, &pygilstate); 119019e60943SBram Moolenaar Py_XDECREF(cmdbytes); 1191bd5e15fdSBram Moolenaar 1192bd5e15fdSBram Moolenaar PyGILState_Release(pygilstate); 1193bd5e15fdSBram Moolenaar 1194bd5e15fdSBram Moolenaar #if defined(HAVE_LOCALE_H) || defined(X_LOCALE) 1195bd5e15fdSBram Moolenaar if (saved_locale != NULL) 1196bd5e15fdSBram Moolenaar { 1197bd5e15fdSBram Moolenaar (void)setlocale(LC_NUMERIC, saved_locale); 1198bd5e15fdSBram Moolenaar vim_free(saved_locale); 1199bd5e15fdSBram Moolenaar } 1200bd5e15fdSBram Moolenaar #endif 1201bd5e15fdSBram Moolenaar 12022ab2e860SBram Moolenaar Python_Lock_Vim(); // enter Vim 1203bd5e15fdSBram Moolenaar PythonIO_Flush(); 1204bd5e15fdSBram Moolenaar 1205bd5e15fdSBram Moolenaar theend: 12062ab2e860SBram Moolenaar return; // keeps lint happy 1207bd5e15fdSBram Moolenaar } 1208bd5e15fdSBram Moolenaar 1209bd5e15fdSBram Moolenaar /* 1210368373e9SBram Moolenaar * ":py3" 1211bd5e15fdSBram Moolenaar */ 1212170bf1aeSBram Moolenaar void 1213170bf1aeSBram Moolenaar ex_py3(exarg_T *eap) 1214bd5e15fdSBram Moolenaar { 1215bd5e15fdSBram Moolenaar char_u *script; 1216bd5e15fdSBram Moolenaar 1217bd5e15fdSBram Moolenaar script = script_get(eap, eap->arg); 1218bd5e15fdSBram Moolenaar if (!eap->skip) 1219bd5e15fdSBram Moolenaar { 122014816ad6SBram Moolenaar if (p_pyx == 0) 122114816ad6SBram Moolenaar p_pyx = 3; 122214816ad6SBram Moolenaar 1223b52f4c02SBram Moolenaar DoPyCommand(script == NULL ? (char *) eap->arg : (char *) script, 1224b52f4c02SBram Moolenaar (rangeinitializer) init_range_cmd, 1225b52f4c02SBram Moolenaar (runner) run_cmd, 1226b52f4c02SBram Moolenaar (void *) eap); 1227bd5e15fdSBram Moolenaar } 1228bd5e15fdSBram Moolenaar vim_free(script); 1229bd5e15fdSBram Moolenaar } 1230bd5e15fdSBram Moolenaar 1231bd5e15fdSBram Moolenaar #define BUFFER_SIZE 2048 1232bd5e15fdSBram Moolenaar 1233bd5e15fdSBram Moolenaar /* 12346df6f47dSBram Moolenaar * ":py3file" 1235bd5e15fdSBram Moolenaar */ 1236bd5e15fdSBram Moolenaar void 1237bd5e15fdSBram Moolenaar ex_py3file(exarg_T *eap) 1238bd5e15fdSBram Moolenaar { 1239bd5e15fdSBram Moolenaar static char buffer[BUFFER_SIZE]; 1240bd5e15fdSBram Moolenaar const char *file; 1241bd5e15fdSBram Moolenaar char *p; 1242bd5e15fdSBram Moolenaar int i; 1243bd5e15fdSBram Moolenaar 1244f42dd3c3SBram Moolenaar if (p_pyx == 0) 1245f42dd3c3SBram Moolenaar p_pyx = 3; 1246f42dd3c3SBram Moolenaar 12472ab2e860SBram Moolenaar // Have to do it like this. PyRun_SimpleFile requires you to pass a 12482ab2e860SBram Moolenaar // stdio file pointer, but Vim and the Python DLL are compiled with 12492ab2e860SBram Moolenaar // different options under Windows, meaning that stdio pointers aren't 12502ab2e860SBram Moolenaar // compatible between the two. Yuk. 12512ab2e860SBram Moolenaar // 12522ab2e860SBram Moolenaar // construct: exec(compile(open('a_filename', 'rb').read(), 'a_filename', 'exec')) 12532ab2e860SBram Moolenaar // 12542ab2e860SBram Moolenaar // Using bytes so that Python can detect the source encoding as it normally 12552ab2e860SBram Moolenaar // does. The doc does not say "compile" accept bytes, though. 12562ab2e860SBram Moolenaar // 12572ab2e860SBram Moolenaar // We need to escape any backslashes or single quotes in the file name, so that 12582ab2e860SBram Moolenaar // Python won't mangle the file name. 1259bd5e15fdSBram Moolenaar 1260bd5e15fdSBram Moolenaar strcpy(buffer, "exec(compile(open('"); 12612ab2e860SBram Moolenaar p = buffer + 19; // size of "exec(compile(open('" 1262bd5e15fdSBram Moolenaar 1263bd5e15fdSBram Moolenaar for (i=0; i<2; ++i) 1264bd5e15fdSBram Moolenaar { 1265bd5e15fdSBram Moolenaar file = (char *)eap->arg; 1266bd5e15fdSBram Moolenaar while (*file && p < buffer + (BUFFER_SIZE - 3)) 1267bd5e15fdSBram Moolenaar { 1268bd5e15fdSBram Moolenaar if (*file == '\\' || *file == '\'') 1269bd5e15fdSBram Moolenaar *p++ = '\\'; 1270bd5e15fdSBram Moolenaar *p++ = *file++; 1271bd5e15fdSBram Moolenaar } 12722ab2e860SBram Moolenaar // If we didn't finish the file name, we hit a buffer overflow 1273bd5e15fdSBram Moolenaar if (*file != '\0') 1274bd5e15fdSBram Moolenaar return; 1275bd5e15fdSBram Moolenaar if (i==0) 1276bd5e15fdSBram Moolenaar { 127719e60943SBram Moolenaar strcpy(p,"','rb').read(),'"); 127819e60943SBram Moolenaar p += 16; 1279bd5e15fdSBram Moolenaar } 1280bd5e15fdSBram Moolenaar else 1281bd5e15fdSBram Moolenaar { 1282bd5e15fdSBram Moolenaar strcpy(p,"','exec'))"); 1283bd5e15fdSBram Moolenaar p += 10; 1284bd5e15fdSBram Moolenaar } 1285bd5e15fdSBram Moolenaar } 1286bd5e15fdSBram Moolenaar 1287bd5e15fdSBram Moolenaar 12882ab2e860SBram Moolenaar // Execute the file 1289b52f4c02SBram Moolenaar DoPyCommand(buffer, 1290b52f4c02SBram Moolenaar (rangeinitializer) init_range_cmd, 1291b52f4c02SBram Moolenaar (runner) run_cmd, 1292b52f4c02SBram Moolenaar (void *) eap); 1293bd5e15fdSBram Moolenaar } 1294bd5e15fdSBram Moolenaar 129554e8f005SBram Moolenaar void 129654e8f005SBram Moolenaar ex_py3do(exarg_T *eap) 12973dab2806SBram Moolenaar { 1298f42dd3c3SBram Moolenaar if (p_pyx == 0) 1299f42dd3c3SBram Moolenaar p_pyx = 3; 1300f42dd3c3SBram Moolenaar 1301b52f4c02SBram Moolenaar DoPyCommand((char *)eap->arg, 1302b52f4c02SBram Moolenaar (rangeinitializer)init_range_cmd, 1303b52f4c02SBram Moolenaar (runner)run_do, 1304b52f4c02SBram Moolenaar (void *)eap); 13053dab2806SBram Moolenaar } 13063dab2806SBram Moolenaar 13072ab2e860SBram Moolenaar /////////////////////////////////////////////////////// 13082ab2e860SBram Moolenaar // 2. Python output stream: writes output via [e]msg(). 1309bd5e15fdSBram Moolenaar 13102ab2e860SBram Moolenaar // Implementation functions 1311bd5e15fdSBram Moolenaar 1312170bf1aeSBram Moolenaar static PyObject * 1313170bf1aeSBram Moolenaar OutputGetattro(PyObject *self, PyObject *nameobj) 1314bd5e15fdSBram Moolenaar { 131577045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1316bd5e15fdSBram Moolenaar 1317bd5e15fdSBram Moolenaar if (strcmp(name, "softspace") == 0) 1318bd5e15fdSBram Moolenaar return PyLong_FromLong(((OutputObject *)(self))->softspace); 13196d4431e7SBram Moolenaar else if (strcmp(name, "errors") == 0) 13206d4431e7SBram Moolenaar return PyString_FromString("strict"); 13216d4431e7SBram Moolenaar else if (strcmp(name, "encoding") == 0) 13226d4431e7SBram Moolenaar return PyString_FromString(ENC_OPT); 1323bd5e15fdSBram Moolenaar 1324bd5e15fdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1325bd5e15fdSBram Moolenaar } 1326bd5e15fdSBram Moolenaar 1327170bf1aeSBram Moolenaar static int 1328170bf1aeSBram Moolenaar OutputSetattro(PyObject *self, PyObject *nameobj, PyObject *val) 1329bd5e15fdSBram Moolenaar { 133077045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1331bd5e15fdSBram Moolenaar 1332d6e39186SBram Moolenaar return OutputSetattr((OutputObject *)(self), name, val); 1333bd5e15fdSBram Moolenaar } 1334bd5e15fdSBram Moolenaar 13352ab2e860SBram Moolenaar /////////////////////////////////////////////////////// 13362ab2e860SBram Moolenaar // 3. Implementation of the Vim module for Python 1337bd5e15fdSBram Moolenaar 13382ab2e860SBram Moolenaar // Window type - Implementation functions 13392ab2e860SBram Moolenaar // -------------------------------------- 1340bd5e15fdSBram Moolenaar 1341bd5e15fdSBram Moolenaar #define WindowType_Check(obj) ((obj)->ob_base.ob_type == &WindowType) 1342bd5e15fdSBram Moolenaar 13432ab2e860SBram Moolenaar // Buffer type - Implementation functions 13442ab2e860SBram Moolenaar // -------------------------------------- 1345bd5e15fdSBram Moolenaar 1346bd5e15fdSBram Moolenaar #define BufferType_Check(obj) ((obj)->ob_base.ob_type == &BufferType) 1347bd5e15fdSBram Moolenaar 1348bd5e15fdSBram Moolenaar static PyObject* BufferSubscript(PyObject *self, PyObject *idx); 13494ce5fe4cSBram Moolenaar static int BufferAsSubscript(PyObject *self, PyObject *idx, PyObject *val); 1350bd5e15fdSBram Moolenaar 13512ab2e860SBram Moolenaar // Line range type - Implementation functions 13522ab2e860SBram Moolenaar // -------------------------------------- 1353bd5e15fdSBram Moolenaar 1354bd5e15fdSBram Moolenaar #define RangeType_Check(obj) ((obj)->ob_base.ob_type == &RangeType) 1355bd5e15fdSBram Moolenaar 1356bd5e15fdSBram Moolenaar static PyObject* RangeSubscript(PyObject *self, PyObject *idx); 13574ce5fe4cSBram Moolenaar static int RangeAsItem(PyObject *, Py_ssize_t, PyObject *); 13584ce5fe4cSBram Moolenaar static int RangeAsSubscript(PyObject *self, PyObject *idx, PyObject *val); 1359bd5e15fdSBram Moolenaar 13602ab2e860SBram Moolenaar // Current objects type - Implementation functions 13612ab2e860SBram Moolenaar // ----------------------------------------------- 1362bd5e15fdSBram Moolenaar 1363bd5e15fdSBram Moolenaar static PySequenceMethods BufferAsSeq = { 13642ab2e860SBram Moolenaar (lenfunc) BufferLength, // sq_length, len(x) 13652ab2e860SBram Moolenaar (binaryfunc) 0, // sq_concat, x+y 13662ab2e860SBram Moolenaar (ssizeargfunc) 0, // sq_repeat, x*n 13672ab2e860SBram Moolenaar (ssizeargfunc) BufferItem, // sq_item, x[i] 13682ab2e860SBram Moolenaar 0, // was_sq_slice, x[i:j] 13692ab2e860SBram Moolenaar 0, // sq_ass_item, x[i]=v 13702ab2e860SBram Moolenaar 0, // sq_ass_slice, x[i:j]=v 13712ab2e860SBram Moolenaar 0, // sq_contains 13722ab2e860SBram Moolenaar 0, // sq_inplace_concat 13732ab2e860SBram Moolenaar 0, // sq_inplace_repeat 1374bd5e15fdSBram Moolenaar }; 1375bd5e15fdSBram Moolenaar 13764d1da49cSBram Moolenaar static PyMappingMethods BufferAsMapping = { 1377bd5e15fdSBram Moolenaar /* mp_length */ (lenfunc)BufferLength, 1378bd5e15fdSBram Moolenaar /* mp_subscript */ (binaryfunc)BufferSubscript, 137919e60943SBram Moolenaar /* mp_ass_subscript */ (objobjargproc)BufferAsSubscript, 1380bd5e15fdSBram Moolenaar }; 1381bd5e15fdSBram Moolenaar 1382bd5e15fdSBram Moolenaar 13832ab2e860SBram Moolenaar // Buffer object 1384bd5e15fdSBram Moolenaar 1385170bf1aeSBram Moolenaar static PyObject * 1386170bf1aeSBram Moolenaar BufferGetattro(PyObject *self, PyObject *nameobj) 1387bd5e15fdSBram Moolenaar { 13884d1da49cSBram Moolenaar PyObject *r; 1389bd5e15fdSBram Moolenaar 139077045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1391bd5e15fdSBram Moolenaar 13929e822c00SBram Moolenaar if ((r = BufferAttrValid((BufferObject *)(self), name))) 13939e822c00SBram Moolenaar return r; 13949e822c00SBram Moolenaar 13954d1da49cSBram Moolenaar if (CheckBuffer((BufferObject *)(self))) 1396bd5e15fdSBram Moolenaar return NULL; 1397bd5e15fdSBram Moolenaar 13984d1da49cSBram Moolenaar r = BufferAttr((BufferObject *)(self), name); 13994d1da49cSBram Moolenaar if (r || PyErr_Occurred()) 14004d1da49cSBram Moolenaar return r; 1401bd5e15fdSBram Moolenaar else 1402bd5e15fdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1403bd5e15fdSBram Moolenaar } 1404bd5e15fdSBram Moolenaar 1405e9ba516bSBram Moolenaar static int 1406e9ba516bSBram Moolenaar BufferSetattro(PyObject *self, PyObject *nameobj, PyObject *val) 1407e9ba516bSBram Moolenaar { 1408e9ba516bSBram Moolenaar GET_ATTR_STRING(name, nameobj); 1409e9ba516bSBram Moolenaar 1410e9ba516bSBram Moolenaar return BufferSetattr((BufferObject *)(self), name, val); 1411e9ba516bSBram Moolenaar } 1412e9ba516bSBram Moolenaar 14132ab2e860SBram Moolenaar ////////////////// 1414bd5e15fdSBram Moolenaar 1415170bf1aeSBram Moolenaar static PyObject * 1416170bf1aeSBram Moolenaar BufferSubscript(PyObject *self, PyObject* idx) 1417bd5e15fdSBram Moolenaar { 1418db913953SBram Moolenaar if (PyLong_Check(idx)) 1419db913953SBram Moolenaar { 1420bd5e15fdSBram Moolenaar long _idx = PyLong_AsLong(idx); 1421d6e39186SBram Moolenaar return BufferItem((BufferObject *)(self), _idx); 1422db913953SBram Moolenaar } else if (PySlice_Check(idx)) 1423db913953SBram Moolenaar { 1424bd5e15fdSBram Moolenaar Py_ssize_t start, stop, step, slicelen; 1425bd5e15fdSBram Moolenaar 14268f1723deSBram Moolenaar if (CheckBuffer((BufferObject *) self)) 14278f1723deSBram Moolenaar return NULL; 14288f1723deSBram Moolenaar 1429922a4664SBram Moolenaar if (PySlice_GetIndicesEx((PySliceObject_T *)idx, 1430bd80f35bSBram Moolenaar (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count, 1431bd5e15fdSBram Moolenaar &start, &stop, 1432db913953SBram Moolenaar &step, &slicelen) < 0) 1433bd5e15fdSBram Moolenaar return NULL; 1434d6e39186SBram Moolenaar return BufferSlice((BufferObject *)(self), start, stop); 1435db913953SBram Moolenaar } 1436db913953SBram Moolenaar else 1437db913953SBram Moolenaar { 1438c476e52fSBram Moolenaar RAISE_INVALID_INDEX_TYPE(idx); 1439bd5e15fdSBram Moolenaar return NULL; 1440bd5e15fdSBram Moolenaar } 1441bd5e15fdSBram Moolenaar } 1442bd5e15fdSBram Moolenaar 14434ce5fe4cSBram Moolenaar static int 144419e60943SBram Moolenaar BufferAsSubscript(PyObject *self, PyObject* idx, PyObject* val) 144519e60943SBram Moolenaar { 1446db913953SBram Moolenaar if (PyLong_Check(idx)) 1447db913953SBram Moolenaar { 144819e60943SBram Moolenaar long n = PyLong_AsLong(idx); 1449ab589463SBram Moolenaar 1450ab589463SBram Moolenaar if (CheckBuffer((BufferObject *) self)) 1451ab589463SBram Moolenaar return -1; 1452ab589463SBram Moolenaar 145319e60943SBram Moolenaar return RBAsItem((BufferObject *)(self), n, val, 1, 145419e60943SBram Moolenaar (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count, 145519e60943SBram Moolenaar NULL); 1456db913953SBram Moolenaar } else if (PySlice_Check(idx)) 1457db913953SBram Moolenaar { 145819e60943SBram Moolenaar Py_ssize_t start, stop, step, slicelen; 145919e60943SBram Moolenaar 14608f1723deSBram Moolenaar if (CheckBuffer((BufferObject *) self)) 14618f1723deSBram Moolenaar return -1; 14628f1723deSBram Moolenaar 1463922a4664SBram Moolenaar if (PySlice_GetIndicesEx((PySliceObject_T *)idx, 1464bd80f35bSBram Moolenaar (Py_ssize_t)((BufferObject *)(self))->buf->b_ml.ml_line_count, 146519e60943SBram Moolenaar &start, &stop, 1466db913953SBram Moolenaar &step, &slicelen) < 0) 146719e60943SBram Moolenaar return -1; 146819e60943SBram Moolenaar return RBAsSlice((BufferObject *)(self), start, stop, val, 1, 146919e60943SBram Moolenaar (PyInt)((BufferObject *)(self))->buf->b_ml.ml_line_count, 147019e60943SBram Moolenaar NULL); 1471db913953SBram Moolenaar } 1472db913953SBram Moolenaar else 1473db913953SBram Moolenaar { 1474c476e52fSBram Moolenaar RAISE_INVALID_INDEX_TYPE(idx); 147519e60943SBram Moolenaar return -1; 147619e60943SBram Moolenaar } 147719e60943SBram Moolenaar } 147819e60943SBram Moolenaar 1479bd5e15fdSBram Moolenaar static PySequenceMethods RangeAsSeq = { 14802ab2e860SBram Moolenaar (lenfunc) RangeLength, // sq_length, len(x) 14812ab2e860SBram Moolenaar (binaryfunc) 0, // RangeConcat, sq_concat, x+y 14822ab2e860SBram Moolenaar (ssizeargfunc) 0, // RangeRepeat, sq_repeat, x*n 14832ab2e860SBram Moolenaar (ssizeargfunc) RangeItem, // sq_item, x[i] 14842ab2e860SBram Moolenaar 0, // was_sq_slice, x[i:j] 14852ab2e860SBram Moolenaar (ssizeobjargproc) RangeAsItem, // sq_as_item, x[i]=v 14862ab2e860SBram Moolenaar 0, // sq_ass_slice, x[i:j]=v 14872ab2e860SBram Moolenaar 0, // sq_contains 14882ab2e860SBram Moolenaar 0, // sq_inplace_concat 14892ab2e860SBram Moolenaar 0, // sq_inplace_repeat 1490bd5e15fdSBram Moolenaar }; 1491bd5e15fdSBram Moolenaar 14924d1da49cSBram Moolenaar static PyMappingMethods RangeAsMapping = { 1493bd5e15fdSBram Moolenaar /* mp_length */ (lenfunc)RangeLength, 1494bd5e15fdSBram Moolenaar /* mp_subscript */ (binaryfunc)RangeSubscript, 1495ba4897e6SBram Moolenaar /* mp_ass_subscript */ (objobjargproc)RangeAsSubscript, 1496bd5e15fdSBram Moolenaar }; 1497bd5e15fdSBram Moolenaar 14982ab2e860SBram Moolenaar // Line range object - Implementation 1499bd5e15fdSBram Moolenaar 1500170bf1aeSBram Moolenaar static PyObject * 1501170bf1aeSBram Moolenaar RangeGetattro(PyObject *self, PyObject *nameobj) 1502bd5e15fdSBram Moolenaar { 150377045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1504bd5e15fdSBram Moolenaar 1505bd5e15fdSBram Moolenaar if (strcmp(name, "start") == 0) 1506bd5e15fdSBram Moolenaar return Py_BuildValue("n", ((RangeObject *)(self))->start - 1); 1507bd5e15fdSBram Moolenaar else if (strcmp(name, "end") == 0) 1508bd5e15fdSBram Moolenaar return Py_BuildValue("n", ((RangeObject *)(self))->end - 1); 1509bd5e15fdSBram Moolenaar else 1510bd5e15fdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1511bd5e15fdSBram Moolenaar } 1512bd5e15fdSBram Moolenaar 15132ab2e860SBram Moolenaar //////////////// 1514bd5e15fdSBram Moolenaar 15154ce5fe4cSBram Moolenaar static int 1516170bf1aeSBram Moolenaar RangeAsItem(PyObject *self, Py_ssize_t n, PyObject *val) 1517bd5e15fdSBram Moolenaar { 1518bd5e15fdSBram Moolenaar return RBAsItem(((RangeObject *)(self))->buf, n, val, 1519bd5e15fdSBram Moolenaar ((RangeObject *)(self))->start, 1520bd5e15fdSBram Moolenaar ((RangeObject *)(self))->end, 1521bd5e15fdSBram Moolenaar &((RangeObject *)(self))->end); 1522bd5e15fdSBram Moolenaar } 1523bd5e15fdSBram Moolenaar 1524ba4897e6SBram Moolenaar static Py_ssize_t 1525ba4897e6SBram Moolenaar RangeAsSlice(PyObject *self, Py_ssize_t lo, Py_ssize_t hi, PyObject *val) 1526ba4897e6SBram Moolenaar { 1527ba4897e6SBram Moolenaar return RBAsSlice(((RangeObject *)(self))->buf, lo, hi, val, 1528ba4897e6SBram Moolenaar ((RangeObject *)(self))->start, 1529ba4897e6SBram Moolenaar ((RangeObject *)(self))->end, 1530ba4897e6SBram Moolenaar &((RangeObject *)(self))->end); 1531ba4897e6SBram Moolenaar } 1532ba4897e6SBram Moolenaar 1533170bf1aeSBram Moolenaar static PyObject * 1534170bf1aeSBram Moolenaar RangeSubscript(PyObject *self, PyObject* idx) 1535bd5e15fdSBram Moolenaar { 1536db913953SBram Moolenaar if (PyLong_Check(idx)) 1537db913953SBram Moolenaar { 1538bd5e15fdSBram Moolenaar long _idx = PyLong_AsLong(idx); 1539d6e39186SBram Moolenaar return RangeItem((RangeObject *)(self), _idx); 1540db913953SBram Moolenaar } else if (PySlice_Check(idx)) 1541db913953SBram Moolenaar { 1542bd5e15fdSBram Moolenaar Py_ssize_t start, stop, step, slicelen; 1543bd5e15fdSBram Moolenaar 1544922a4664SBram Moolenaar if (PySlice_GetIndicesEx((PySliceObject_T *)idx, 1545bd5e15fdSBram Moolenaar ((RangeObject *)(self))->end-((RangeObject *)(self))->start+1, 1546bd5e15fdSBram Moolenaar &start, &stop, 1547db913953SBram Moolenaar &step, &slicelen) < 0) 1548bd5e15fdSBram Moolenaar return NULL; 1549d6e39186SBram Moolenaar return RangeSlice((RangeObject *)(self), start, stop); 1550db913953SBram Moolenaar } 1551db913953SBram Moolenaar else 1552db913953SBram Moolenaar { 1553c476e52fSBram Moolenaar RAISE_INVALID_INDEX_TYPE(idx); 1554bd5e15fdSBram Moolenaar return NULL; 1555bd5e15fdSBram Moolenaar } 1556bd5e15fdSBram Moolenaar } 1557bd5e15fdSBram Moolenaar 15584ce5fe4cSBram Moolenaar static int 1559ba4897e6SBram Moolenaar RangeAsSubscript(PyObject *self, PyObject *idx, PyObject *val) 1560ba4897e6SBram Moolenaar { 1561db913953SBram Moolenaar if (PyLong_Check(idx)) 1562db913953SBram Moolenaar { 1563ba4897e6SBram Moolenaar long n = PyLong_AsLong(idx); 1564ba4897e6SBram Moolenaar return RangeAsItem(self, n, val); 1565abab0b0fSBram Moolenaar } 1566abab0b0fSBram Moolenaar else if (PySlice_Check(idx)) 1567db913953SBram Moolenaar { 1568ba4897e6SBram Moolenaar Py_ssize_t start, stop, step, slicelen; 1569ba4897e6SBram Moolenaar 1570922a4664SBram Moolenaar if (PySlice_GetIndicesEx((PySliceObject_T *)idx, 1571ba4897e6SBram Moolenaar ((RangeObject *)(self))->end-((RangeObject *)(self))->start+1, 1572ba4897e6SBram Moolenaar &start, &stop, 1573db913953SBram Moolenaar &step, &slicelen) < 0) 1574ba4897e6SBram Moolenaar return -1; 1575ba4897e6SBram Moolenaar return RangeAsSlice(self, start, stop, val); 1576db913953SBram Moolenaar } 1577db913953SBram Moolenaar else 1578db913953SBram Moolenaar { 1579c476e52fSBram Moolenaar RAISE_INVALID_INDEX_TYPE(idx); 1580ba4897e6SBram Moolenaar return -1; 1581ba4897e6SBram Moolenaar } 1582ba4897e6SBram Moolenaar } 1583ba4897e6SBram Moolenaar 15842ab2e860SBram Moolenaar // TabPage object - Implementation 15855e538ecdSBram Moolenaar 15865e538ecdSBram Moolenaar static PyObject * 15875e538ecdSBram Moolenaar TabPageGetattro(PyObject *self, PyObject *nameobj) 15885e538ecdSBram Moolenaar { 15895e538ecdSBram Moolenaar PyObject *r; 15905e538ecdSBram Moolenaar 15915e538ecdSBram Moolenaar GET_ATTR_STRING(name, nameobj); 15925e538ecdSBram Moolenaar 15939e822c00SBram Moolenaar if ((r = TabPageAttrValid((TabPageObject *)(self), name))) 15949e822c00SBram Moolenaar return r; 15959e822c00SBram Moolenaar 15965e538ecdSBram Moolenaar if (CheckTabPage((TabPageObject *)(self))) 15975e538ecdSBram Moolenaar return NULL; 15985e538ecdSBram Moolenaar 15995e538ecdSBram Moolenaar r = TabPageAttr((TabPageObject *)(self), name); 16005e538ecdSBram Moolenaar if (r || PyErr_Occurred()) 16015e538ecdSBram Moolenaar return r; 16025e538ecdSBram Moolenaar else 16035e538ecdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 16045e538ecdSBram Moolenaar } 16055e538ecdSBram Moolenaar 16062ab2e860SBram Moolenaar // Window object - Implementation 1607bd5e15fdSBram Moolenaar 1608170bf1aeSBram Moolenaar static PyObject * 1609170bf1aeSBram Moolenaar WindowGetattro(PyObject *self, PyObject *nameobj) 1610bd5e15fdSBram Moolenaar { 16114d1da49cSBram Moolenaar PyObject *r; 1612bd5e15fdSBram Moolenaar 161377045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1614bd5e15fdSBram Moolenaar 16159e822c00SBram Moolenaar if ((r = WindowAttrValid((WindowObject *)(self), name))) 16169e822c00SBram Moolenaar return r; 16179e822c00SBram Moolenaar 16184d1da49cSBram Moolenaar if (CheckWindow((WindowObject *)(self))) 1619bd5e15fdSBram Moolenaar return NULL; 1620bd5e15fdSBram Moolenaar 16214d1da49cSBram Moolenaar r = WindowAttr((WindowObject *)(self), name); 16224d1da49cSBram Moolenaar if (r || PyErr_Occurred()) 16234d1da49cSBram Moolenaar return r; 1624bd5e15fdSBram Moolenaar else 1625bd5e15fdSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1626bd5e15fdSBram Moolenaar } 1627bd5e15fdSBram Moolenaar 1628170bf1aeSBram Moolenaar static int 1629170bf1aeSBram Moolenaar WindowSetattro(PyObject *self, PyObject *nameobj, PyObject *val) 1630bd5e15fdSBram Moolenaar { 163177045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1632bd5e15fdSBram Moolenaar 1633d6e39186SBram Moolenaar return WindowSetattr((WindowObject *)(self), name, val); 1634bd5e15fdSBram Moolenaar } 1635bd5e15fdSBram Moolenaar 16362ab2e860SBram Moolenaar // Tab page list object - Definitions 16375e538ecdSBram Moolenaar 16385e538ecdSBram Moolenaar static PySequenceMethods TabListAsSeq = { 16392ab2e860SBram Moolenaar (lenfunc) TabListLength, // sq_length, len(x) 16402ab2e860SBram Moolenaar (binaryfunc) 0, // sq_concat, x+y 16412ab2e860SBram Moolenaar (ssizeargfunc) 0, // sq_repeat, x*n 16422ab2e860SBram Moolenaar (ssizeargfunc) TabListItem, // sq_item, x[i] 16432ab2e860SBram Moolenaar 0, // sq_slice, x[i:j] 16442ab2e860SBram Moolenaar (ssizeobjargproc)0, // sq_as_item, x[i]=v 16452ab2e860SBram Moolenaar 0, // sq_ass_slice, x[i:j]=v 16462ab2e860SBram Moolenaar 0, // sq_contains 16472ab2e860SBram Moolenaar 0, // sq_inplace_concat 16482ab2e860SBram Moolenaar 0, // sq_inplace_repeat 16495e538ecdSBram Moolenaar }; 16505e538ecdSBram Moolenaar 16512ab2e860SBram Moolenaar // Window list object - Definitions 1652bd5e15fdSBram Moolenaar 1653bd5e15fdSBram Moolenaar static PySequenceMethods WinListAsSeq = { 16542ab2e860SBram Moolenaar (lenfunc) WinListLength, // sq_length, len(x) 16552ab2e860SBram Moolenaar (binaryfunc) 0, // sq_concat, x+y 16562ab2e860SBram Moolenaar (ssizeargfunc) 0, // sq_repeat, x*n 16572ab2e860SBram Moolenaar (ssizeargfunc) WinListItem, // sq_item, x[i] 16582ab2e860SBram Moolenaar 0, // sq_slice, x[i:j] 16592ab2e860SBram Moolenaar (ssizeobjargproc)0, // sq_as_item, x[i]=v 16602ab2e860SBram Moolenaar 0, // sq_ass_slice, x[i:j]=v 16612ab2e860SBram Moolenaar 0, // sq_contains 16622ab2e860SBram Moolenaar 0, // sq_inplace_concat 16632ab2e860SBram Moolenaar 0, // sq_inplace_repeat 1664bd5e15fdSBram Moolenaar }; 1665bd5e15fdSBram Moolenaar 16662ab2e860SBram Moolenaar /* 16672ab2e860SBram Moolenaar * Current items object - Implementation 1668bd5e15fdSBram Moolenaar */ 1669170bf1aeSBram Moolenaar static PyObject * 16704d1da49cSBram Moolenaar CurrentGetattro(PyObject *self, PyObject *nameobj) 1671bd5e15fdSBram Moolenaar { 1672dd8aca66SBram Moolenaar PyObject *r; 167377045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1674dd8aca66SBram Moolenaar if (!(r = CurrentGetattr(self, name))) 1675dd8aca66SBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1676dd8aca66SBram Moolenaar return r; 1677bd5e15fdSBram Moolenaar } 1678bd5e15fdSBram Moolenaar 1679170bf1aeSBram Moolenaar static int 16804d1da49cSBram Moolenaar CurrentSetattro(PyObject *self, PyObject *nameobj, PyObject *value) 1681bd5e15fdSBram Moolenaar { 16824d1da49cSBram Moolenaar GET_ATTR_STRING(name, nameobj); 16834d1da49cSBram Moolenaar return CurrentSetattr(self, name, value); 1684bd5e15fdSBram Moolenaar } 1685bd5e15fdSBram Moolenaar 16862ab2e860SBram Moolenaar // Dictionary object - Definitions 1687db913953SBram Moolenaar 168866b7985eSBram Moolenaar static PyObject * 168966b7985eSBram Moolenaar DictionaryGetattro(PyObject *self, PyObject *nameobj) 169066b7985eSBram Moolenaar { 169166b7985eSBram Moolenaar DictionaryObject *this = ((DictionaryObject *) (self)); 169266b7985eSBram Moolenaar 169366b7985eSBram Moolenaar GET_ATTR_STRING(name, nameobj); 169466b7985eSBram Moolenaar 169566b7985eSBram Moolenaar if (strcmp(name, "locked") == 0) 169666b7985eSBram Moolenaar return PyLong_FromLong(this->dict->dv_lock); 169766b7985eSBram Moolenaar else if (strcmp(name, "scope") == 0) 169866b7985eSBram Moolenaar return PyLong_FromLong(this->dict->dv_scope); 169966b7985eSBram Moolenaar 170066b7985eSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 170166b7985eSBram Moolenaar } 170266b7985eSBram Moolenaar 170366b7985eSBram Moolenaar static int 170466b7985eSBram Moolenaar DictionarySetattro(PyObject *self, PyObject *nameobj, PyObject *val) 170566b7985eSBram Moolenaar { 170666b7985eSBram Moolenaar GET_ATTR_STRING(name, nameobj); 1707d6e39186SBram Moolenaar return DictionarySetattr((DictionaryObject *)(self), name, val); 1708db913953SBram Moolenaar } 1709db913953SBram Moolenaar 17102ab2e860SBram Moolenaar // List object - Definitions 1711db913953SBram Moolenaar 171266b7985eSBram Moolenaar static PyObject * 171366b7985eSBram Moolenaar ListGetattro(PyObject *self, PyObject *nameobj) 171466b7985eSBram Moolenaar { 171566b7985eSBram Moolenaar GET_ATTR_STRING(name, nameobj); 171666b7985eSBram Moolenaar 171766b7985eSBram Moolenaar if (strcmp(name, "locked") == 0) 171866b7985eSBram Moolenaar return PyLong_FromLong(((ListObject *) (self))->list->lv_lock); 171966b7985eSBram Moolenaar 172066b7985eSBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 172166b7985eSBram Moolenaar } 172266b7985eSBram Moolenaar 172366b7985eSBram Moolenaar static int 172466b7985eSBram Moolenaar ListSetattro(PyObject *self, PyObject *nameobj, PyObject *val) 172566b7985eSBram Moolenaar { 172666b7985eSBram Moolenaar GET_ATTR_STRING(name, nameobj); 1727d6e39186SBram Moolenaar return ListSetattr((ListObject *)(self), name, val); 1728db913953SBram Moolenaar } 1729db913953SBram Moolenaar 17302ab2e860SBram Moolenaar // Function object - Definitions 1731db913953SBram Moolenaar 1732db913953SBram Moolenaar static PyObject * 1733db913953SBram Moolenaar FunctionGetattro(PyObject *self, PyObject *nameobj) 1734db913953SBram Moolenaar { 17358110a091SBram Moolenaar PyObject *r; 1736db913953SBram Moolenaar FunctionObject *this = (FunctionObject *)(self); 173777045658SBram Moolenaar 173877045658SBram Moolenaar GET_ATTR_STRING(name, nameobj); 1739db913953SBram Moolenaar 17408110a091SBram Moolenaar r = FunctionAttr(this, name); 17418110a091SBram Moolenaar if (r || PyErr_Occurred()) 17428110a091SBram Moolenaar return r; 17438110a091SBram Moolenaar else 1744db913953SBram Moolenaar return PyObject_GenericGetAttr(self, nameobj); 1745db913953SBram Moolenaar } 1746db913953SBram Moolenaar 17472ab2e860SBram Moolenaar // External interface 1748bd5e15fdSBram Moolenaar 1749bd5e15fdSBram Moolenaar void 1750bd5e15fdSBram Moolenaar python3_buffer_free(buf_T *buf) 1751bd5e15fdSBram Moolenaar { 1752971db467SBram Moolenaar if (BUF_PYTHON_REF(buf) != NULL) 1753bd5e15fdSBram Moolenaar { 1754971db467SBram Moolenaar BufferObject *bp = BUF_PYTHON_REF(buf); 1755bd5e15fdSBram Moolenaar bp->buf = INVALID_BUFFER_VALUE; 1756971db467SBram Moolenaar BUF_PYTHON_REF(buf) = NULL; 1757bd5e15fdSBram Moolenaar } 1758bd5e15fdSBram Moolenaar } 1759bd5e15fdSBram Moolenaar 1760bd5e15fdSBram Moolenaar void 1761bd5e15fdSBram Moolenaar python3_window_free(win_T *win) 1762bd5e15fdSBram Moolenaar { 1763971db467SBram Moolenaar if (WIN_PYTHON_REF(win) != NULL) 1764bd5e15fdSBram Moolenaar { 1765971db467SBram Moolenaar WindowObject *wp = WIN_PYTHON_REF(win); 1766bd5e15fdSBram Moolenaar wp->win = INVALID_WINDOW_VALUE; 1767971db467SBram Moolenaar WIN_PYTHON_REF(win) = NULL; 1768bd5e15fdSBram Moolenaar } 1769bd5e15fdSBram Moolenaar } 17705e538ecdSBram Moolenaar 17715e538ecdSBram Moolenaar void 17725e538ecdSBram Moolenaar python3_tabpage_free(tabpage_T *tab) 17735e538ecdSBram Moolenaar { 17745e538ecdSBram Moolenaar if (TAB_PYTHON_REF(tab) != NULL) 17755e538ecdSBram Moolenaar { 17765e538ecdSBram Moolenaar TabPageObject *tp = TAB_PYTHON_REF(tab); 17775e538ecdSBram Moolenaar tp->tab = INVALID_TABPAGE_VALUE; 17785e538ecdSBram Moolenaar TAB_PYTHON_REF(tab) = NULL; 17795e538ecdSBram Moolenaar } 17805e538ecdSBram Moolenaar } 1781bd5e15fdSBram Moolenaar 17827854e3abSBram Moolenaar static PyObject * 17837854e3abSBram Moolenaar Py3Init_vim(void) 1784bd5e15fdSBram Moolenaar { 17852ab2e860SBram Moolenaar // The special value is removed from sys.path in Python3_Init(). 1786bd5e15fdSBram Moolenaar static wchar_t *(argv[2]) = {L"/must>not&exist/foo", NULL}; 1787bd5e15fdSBram Moolenaar 17881dc28783SBram Moolenaar if (init_types()) 17891dc28783SBram Moolenaar return NULL; 1790bd5e15fdSBram Moolenaar 17912ab2e860SBram Moolenaar // Set sys.argv[] to avoid a crash in warn(). 1792bd5e15fdSBram Moolenaar PySys_SetArgv(1, argv); 1793bd5e15fdSBram Moolenaar 1794c09a6d6cSBram Moolenaar if ((vim_module = PyModule_Create(&vimmodule)) == NULL) 179519e60943SBram Moolenaar return NULL; 1796bd5e15fdSBram Moolenaar 1797dee2e315SBram Moolenaar if (populate_module(vim_module)) 1798bd5e15fdSBram Moolenaar return NULL; 1799bd5e15fdSBram Moolenaar 1800c09a6d6cSBram Moolenaar if (init_sys_path()) 1801c09a6d6cSBram Moolenaar return NULL; 1802c09a6d6cSBram Moolenaar 1803c09a6d6cSBram Moolenaar return vim_module; 1804bd5e15fdSBram Moolenaar } 1805bd5e15fdSBram Moolenaar 18062ab2e860SBram Moolenaar ////////////////////////////////////////////////////////////////////////// 18072ab2e860SBram Moolenaar // 4. Utility functions for handling the interface between Vim and Python. 1808bd5e15fdSBram Moolenaar 18092ab2e860SBram Moolenaar /* 18102ab2e860SBram Moolenaar * Convert a Vim line into a Python string. 1811bd5e15fdSBram Moolenaar * All internal newlines are replaced by null characters. 1812bd5e15fdSBram Moolenaar * 1813bd5e15fdSBram Moolenaar * On errors, the Python exception data is set, and NULL is returned. 1814bd5e15fdSBram Moolenaar */ 1815170bf1aeSBram Moolenaar static PyObject * 1816170bf1aeSBram Moolenaar LineToString(const char *str) 1817bd5e15fdSBram Moolenaar { 1818bd5e15fdSBram Moolenaar PyObject *result; 1819bd5e15fdSBram Moolenaar Py_ssize_t len = strlen(str); 1820bd5e15fdSBram Moolenaar char *tmp, *p; 1821bd5e15fdSBram Moolenaar 1822c799fe20SBram Moolenaar tmp = alloc(len + 1); 1823bd5e15fdSBram Moolenaar p = tmp; 1824bd5e15fdSBram Moolenaar if (p == NULL) 1825bd5e15fdSBram Moolenaar { 1826bd5e15fdSBram Moolenaar PyErr_NoMemory(); 1827bd5e15fdSBram Moolenaar return NULL; 1828bd5e15fdSBram Moolenaar } 1829bd5e15fdSBram Moolenaar 1830bd5e15fdSBram Moolenaar while (*str) 1831bd5e15fdSBram Moolenaar { 1832bd5e15fdSBram Moolenaar if (*str == '\n') 1833bd5e15fdSBram Moolenaar *p = '\0'; 1834bd5e15fdSBram Moolenaar else 1835bd5e15fdSBram Moolenaar *p = *str; 1836bd5e15fdSBram Moolenaar 1837bd5e15fdSBram Moolenaar ++p; 1838bd5e15fdSBram Moolenaar ++str; 1839bd5e15fdSBram Moolenaar } 1840bd5e15fdSBram Moolenaar *p = '\0'; 1841bd5e15fdSBram Moolenaar 18422e2f52a4SBram Moolenaar result = PyUnicode_Decode(tmp, len, (char *)ENC_OPT, ERRORS_DECODE_ARG); 1843bd5e15fdSBram Moolenaar 1844bd5e15fdSBram Moolenaar vim_free(tmp); 1845bd5e15fdSBram Moolenaar return result; 1846bd5e15fdSBram Moolenaar } 1847bd5e15fdSBram Moolenaar 1848db913953SBram Moolenaar void 1849db913953SBram Moolenaar do_py3eval(char_u *str, typval_T *rettv) 1850db913953SBram Moolenaar { 1851b52f4c02SBram Moolenaar DoPyCommand((char *) str, 1852b52f4c02SBram Moolenaar (rangeinitializer) init_range_eval, 1853b52f4c02SBram Moolenaar (runner) run_eval, 1854b52f4c02SBram Moolenaar (void *) rettv); 18558e9a24a1SBram Moolenaar if (rettv->v_type == VAR_UNKNOWN) 1856db913953SBram Moolenaar { 185777fceb89SBram Moolenaar rettv->v_type = VAR_NUMBER; 185877fceb89SBram Moolenaar rettv->vval.v_number = 0; 1859db913953SBram Moolenaar } 1860db913953SBram Moolenaar } 1861db913953SBram Moolenaar 18622459a5ecSBram Moolenaar int 1863db913953SBram Moolenaar set_ref_in_python3(int copyID) 1864db913953SBram Moolenaar { 1865b641df49SBram Moolenaar return set_ref_in_py(copyID); 1866db913953SBram Moolenaar } 1867