Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 1 | |
| 2 | /* Support for dynamic loading of extension modules */ |
Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 3 | |
Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 4 | #include "Python.h" |
Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 5 | |
Guido van Rossum | 96a8fb7 | 1999-12-22 14:09:35 +0000 | [diff] [blame] | 6 | /* ./configure sets HAVE_DYNAMIC_LOADING if dynamic loading of modules is |
| 7 | supported on this platform. configure will then compile and link in one |
| 8 | of the dynload_*.c files, as appropriate. We will call a function in |
| 9 | those modules to get a function pointer to the module's init function. |
Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 10 | */ |
Guido van Rossum | 6ea9092 | 1999-12-20 21:20:42 +0000 | [diff] [blame] | 11 | #ifdef HAVE_DYNAMIC_LOADING |
Guido van Rossum | ff4af06 | 1996-01-12 01:17:50 +0000 | [diff] [blame] | 12 | |
Guido van Rossum | 96a8fb7 | 1999-12-22 14:09:35 +0000 | [diff] [blame] | 13 | #include "importdl.h" |
| 14 | |
Victor Stinner | 2d32227 | 2011-04-04 23:05:53 +0200 | [diff] [blame] | 15 | #ifdef MS_WINDOWS |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 16 | extern dl_funcptr _PyImport_FindSharedFuncptrWindows(const char *prefix, |
| 17 | const char *shortname, |
| 18 | PyObject *pathname, |
| 19 | FILE *fp); |
Victor Stinner | 2d32227 | 2011-04-04 23:05:53 +0200 | [diff] [blame] | 20 | #else |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 21 | extern dl_funcptr _PyImport_FindSharedFuncptr(const char *prefix, |
| 22 | const char *shortname, |
| 23 | const char *pathname, FILE *fp); |
Victor Stinner | 2d32227 | 2011-04-04 23:05:53 +0200 | [diff] [blame] | 24 | #endif |
Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 25 | |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 26 | static const char *ascii_only_prefix = "PyInit"; |
| 27 | static const char *nonascii_prefix = "PyInitU"; |
| 28 | |
| 29 | /* Get the variable part of a module's export symbol name. |
| 30 | * Returns a bytes instance. For non-ASCII-named modules, the name is |
| 31 | * encoded as per PEP 489. |
| 32 | * The hook_prefix pointer is set to either ascii_only_prefix or |
| 33 | * nonascii_prefix, as appropriate. |
| 34 | */ |
| 35 | static PyObject * |
| 36 | get_encoded_name(PyObject *name, const char **hook_prefix) { |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 37 | PyObject *tmp; |
| 38 | PyObject *encoded = NULL; |
Benjamin Peterson | e20056c | 2015-05-29 17:10:30 -0500 | [diff] [blame] | 39 | PyObject *modname = NULL; |
| 40 | Py_ssize_t name_len, lastdot; |
| 41 | _Py_IDENTIFIER(replace); |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 42 | |
| 43 | /* Get the short name (substring after last dot) */ |
| 44 | name_len = PyUnicode_GetLength(name); |
| 45 | lastdot = PyUnicode_FindChar(name, '.', 0, name_len, -1); |
| 46 | if (lastdot < -1) { |
| 47 | return NULL; |
| 48 | } else if (lastdot >= 0) { |
Nick Coghlan | 55871f0 | 2015-05-26 21:48:17 +1000 | [diff] [blame] | 49 | tmp = PyUnicode_Substring(name, lastdot + 1, name_len); |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 50 | if (tmp == NULL) |
| 51 | return NULL; |
| 52 | name = tmp; |
| 53 | /* "name" now holds a new reference to the substring */ |
| 54 | } else { |
| 55 | Py_INCREF(name); |
| 56 | } |
| 57 | |
| 58 | /* Encode to ASCII or Punycode, as needed */ |
| 59 | encoded = PyUnicode_AsEncodedString(name, "ascii", NULL); |
| 60 | if (encoded != NULL) { |
| 61 | *hook_prefix = ascii_only_prefix; |
| 62 | } else { |
| 63 | if (PyErr_ExceptionMatches(PyExc_UnicodeEncodeError)) { |
| 64 | PyErr_Clear(); |
| 65 | encoded = PyUnicode_AsEncodedString(name, "punycode", NULL); |
| 66 | if (encoded == NULL) { |
| 67 | goto error; |
| 68 | } |
| 69 | *hook_prefix = nonascii_prefix; |
| 70 | } else { |
| 71 | goto error; |
| 72 | } |
| 73 | } |
| 74 | |
Benjamin Peterson | e20056c | 2015-05-29 17:10:30 -0500 | [diff] [blame] | 75 | /* Replace '-' by '_' */ |
| 76 | modname = _PyObject_CallMethodId(encoded, &PyId_replace, "cc", '-', '_'); |
| 77 | if (modname == NULL) |
| 78 | goto error; |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 79 | |
| 80 | Py_DECREF(name); |
Benjamin Peterson | e20056c | 2015-05-29 17:10:30 -0500 | [diff] [blame] | 81 | Py_DECREF(encoded); |
| 82 | return modname; |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 83 | error: |
| 84 | Py_DECREF(name); |
| 85 | Py_XDECREF(encoded); |
| 86 | return NULL; |
| 87 | } |
| 88 | |
Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 89 | PyObject * |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 90 | _PyImport_LoadDynamicModuleWithSpec(PyObject *spec, FILE *fp) |
Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 91 | { |
Victor Stinner | 2d32227 | 2011-04-04 23:05:53 +0200 | [diff] [blame] | 92 | #ifndef MS_WINDOWS |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 93 | PyObject *pathbytes = NULL; |
Victor Stinner | 2d32227 | 2011-04-04 23:05:53 +0200 | [diff] [blame] | 94 | #endif |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 95 | PyObject *name_unicode = NULL, *name = NULL, *path = NULL, *m = NULL; |
| 96 | const char *name_buf, *hook_prefix; |
| 97 | char *oldcontext; |
| 98 | dl_funcptr exportfunc; |
| 99 | PyModuleDef *def; |
| 100 | PyObject *(*p0)(void); |
Guido van Rossum | 6ea9092 | 1999-12-20 21:20:42 +0000 | [diff] [blame] | 101 | |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 102 | name_unicode = PyObject_GetAttrString(spec, "name"); |
| 103 | if (name_unicode == NULL) { |
| 104 | return NULL; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 105 | } |
Victor Stinner | fefd70c | 2011-03-14 15:54:07 -0400 | [diff] [blame] | 106 | |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 107 | name = get_encoded_name(name_unicode, &hook_prefix); |
| 108 | if (name == NULL) { |
| 109 | goto error; |
| 110 | } |
| 111 | name_buf = PyBytes_AS_STRING(name); |
Victor Stinner | 65b7eff | 2011-05-07 12:46:05 +0200 | [diff] [blame] | 112 | |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 113 | path = PyObject_GetAttrString(spec, "origin"); |
| 114 | if (path == NULL) |
Victor Stinner | 65b7eff | 2011-05-07 12:46:05 +0200 | [diff] [blame] | 115 | goto error; |
| 116 | |
Victor Stinner | 2d32227 | 2011-04-04 23:05:53 +0200 | [diff] [blame] | 117 | #ifdef MS_WINDOWS |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 118 | exportfunc = _PyImport_FindSharedFuncptrWindows(hook_prefix, name_buf, |
| 119 | path, fp); |
Victor Stinner | 2d32227 | 2011-04-04 23:05:53 +0200 | [diff] [blame] | 120 | #else |
Victor Stinner | fefd70c | 2011-03-14 15:54:07 -0400 | [diff] [blame] | 121 | pathbytes = PyUnicode_EncodeFSDefault(path); |
| 122 | if (pathbytes == NULL) |
Victor Stinner | 65b7eff | 2011-05-07 12:46:05 +0200 | [diff] [blame] | 123 | goto error; |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 124 | exportfunc = _PyImport_FindSharedFuncptr(hook_prefix, name_buf, |
| 125 | PyBytes_AS_STRING(pathbytes), |
| 126 | fp); |
Victor Stinner | fefd70c | 2011-03-14 15:54:07 -0400 | [diff] [blame] | 127 | Py_DECREF(pathbytes); |
Victor Stinner | 2d32227 | 2011-04-04 23:05:53 +0200 | [diff] [blame] | 128 | #endif |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 129 | |
| 130 | if (exportfunc == NULL) { |
| 131 | if (!PyErr_Occurred()) { |
| 132 | PyObject *msg; |
| 133 | msg = PyUnicode_FromFormat( |
| 134 | "dynamic module does not define " |
| 135 | "module export function (%s_%s)", |
| 136 | hook_prefix, name_buf); |
| 137 | if (msg == NULL) |
| 138 | goto error; |
| 139 | PyErr_SetImportError(msg, name_unicode, path); |
| 140 | Py_DECREF(msg); |
| 141 | } |
Victor Stinner | 65b7eff | 2011-05-07 12:46:05 +0200 | [diff] [blame] | 142 | goto error; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 143 | } |
Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 144 | |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 145 | p0 = (PyObject *(*)(void))exportfunc; |
| 146 | |
| 147 | /* Package context is needed for single-phase init */ |
| 148 | oldcontext = _Py_PackageContext; |
| 149 | _Py_PackageContext = PyUnicode_AsUTF8(name_unicode); |
| 150 | m = p0(); |
| 151 | _Py_PackageContext = oldcontext; |
| 152 | |
| 153 | if (m == NULL) { |
| 154 | if (!PyErr_Occurred()) { |
| 155 | PyErr_Format( |
| 156 | PyExc_SystemError, |
| 157 | "initialization of %s failed without raising an exception", |
| 158 | name_buf); |
| 159 | } |
| 160 | goto error; |
| 161 | } else if (PyErr_Occurred()) { |
| 162 | PyErr_Clear(); |
| 163 | PyErr_Format( |
| 164 | PyExc_SystemError, |
| 165 | "initialization of %s raised unreported exception", |
| 166 | name_buf); |
| 167 | m = NULL; |
| 168 | goto error; |
| 169 | } |
| 170 | if (Py_TYPE(m) == NULL) { |
| 171 | /* This can happen when a PyModuleDef is returned without calling |
| 172 | * PyModuleDef_Init on it |
| 173 | */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 174 | PyErr_Format(PyExc_SystemError, |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 175 | "init function of %s returned uninitialized object", |
| 176 | name_buf); |
| 177 | m = NULL; /* prevent segfault in DECREF */ |
| 178 | goto error; |
| 179 | } |
| 180 | if (PyObject_TypeCheck(m, &PyModuleDef_Type)) { |
| 181 | Py_DECREF(name_unicode); |
| 182 | Py_DECREF(name); |
| 183 | Py_DECREF(path); |
| 184 | return PyModule_FromDefAndSpec((PyModuleDef*)m, spec); |
| 185 | } |
| 186 | |
| 187 | /* Fall back to single-phase init mechanism */ |
| 188 | |
| 189 | if (hook_prefix == nonascii_prefix) { |
| 190 | /* don't allow legacy init for non-ASCII module names */ |
| 191 | PyErr_Format( |
| 192 | PyExc_SystemError, |
| 193 | "initialization of * did not return PyModuleDef", |
| 194 | name_buf); |
Victor Stinner | 65b7eff | 2011-05-07 12:46:05 +0200 | [diff] [blame] | 195 | goto error; |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 196 | } |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 197 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 198 | /* Remember pointer to module init function. */ |
| 199 | def = PyModule_GetDef(m); |
Christian Heimes | 7ce57d6 | 2013-07-11 13:02:30 +0200 | [diff] [blame] | 200 | if (def == NULL) { |
| 201 | PyErr_Format(PyExc_SystemError, |
| 202 | "initialization of %s did not return an extension " |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 203 | "module", name_buf); |
Christian Heimes | 848ee09 | 2013-07-11 11:22:21 +0200 | [diff] [blame] | 204 | goto error; |
Christian Heimes | 7ce57d6 | 2013-07-11 13:02:30 +0200 | [diff] [blame] | 205 | } |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 206 | def->m_base.m_init = p0; |
Martin v. Löwis | 1a21451 | 2008-06-11 05:26:20 +0000 | [diff] [blame] | 207 | |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 208 | /* Remember the filename as the __file__ attribute */ |
Antoine Pitrou | f95a1b3 | 2010-05-09 15:52:27 +0000 | [diff] [blame] | 209 | if (PyModule_AddObject(m, "__file__", path) < 0) |
| 210 | PyErr_Clear(); /* Not important enough to report */ |
Victor Stinner | 49d3f25 | 2010-10-17 01:24:53 +0000 | [diff] [blame] | 211 | else |
| 212 | Py_INCREF(path); |
Martin v. Löwis | e81e9b1 | 2003-09-04 18:45:59 +0000 | [diff] [blame] | 213 | |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 214 | if (_PyImport_FixupExtensionObject(m, name_unicode, path) < 0) |
Victor Stinner | 65b7eff | 2011-05-07 12:46:05 +0200 | [diff] [blame] | 215 | goto error; |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 216 | |
| 217 | Py_DECREF(name_unicode); |
| 218 | Py_DECREF(name); |
| 219 | Py_DECREF(path); |
| 220 | |
Victor Stinner | fefd70c | 2011-03-14 15:54:07 -0400 | [diff] [blame] | 221 | return m; |
Victor Stinner | 65b7eff | 2011-05-07 12:46:05 +0200 | [diff] [blame] | 222 | |
| 223 | error: |
Nick Coghlan | d5cacbb | 2015-05-23 22:24:10 +1000 | [diff] [blame] | 224 | Py_DECREF(name_unicode); |
| 225 | Py_XDECREF(name); |
| 226 | Py_XDECREF(path); |
Victor Stinner | 65b7eff | 2011-05-07 12:46:05 +0200 | [diff] [blame] | 227 | Py_XDECREF(m); |
| 228 | return NULL; |
Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 229 | } |
Guido van Rossum | 96a8fb7 | 1999-12-22 14:09:35 +0000 | [diff] [blame] | 230 | |
| 231 | #endif /* HAVE_DYNAMIC_LOADING */ |