1 /* 2 * Extension module used by multiprocessing package 3 * 4 * multiprocessing.c 5 * 6 * Copyright (c) 2006-2008, R Oudkerk 7 * Licensed to PSF under a Contributor Agreement. 8 */ 9 10 #include "multiprocessing.h" 11 12 13 /* 14 * Function which raises exceptions based on error codes 15 */ 16 17 PyObject * 18 _PyMp_SetError(PyObject *Type, int num) 19 { 20 switch (num) { 21 #ifdef MS_WINDOWS 22 case MP_STANDARD_ERROR: 23 if (Type == NULL) 24 Type = PyExc_OSError; 25 PyErr_SetExcFromWindowsErr(Type, 0); 26 break; 27 case MP_SOCKET_ERROR: 28 if (Type == NULL) 29 Type = PyExc_OSError; 30 PyErr_SetExcFromWindowsErr(Type, WSAGetLastError()); 31 break; 32 #else /* !MS_WINDOWS */ 33 case MP_STANDARD_ERROR: 34 case MP_SOCKET_ERROR: 35 if (Type == NULL) 36 Type = PyExc_OSError; 37 PyErr_SetFromErrno(Type); 38 break; 39 #endif /* !MS_WINDOWS */ 40 case MP_MEMORY_ERROR: 41 PyErr_NoMemory(); 42 break; 43 case MP_EXCEPTION_HAS_BEEN_SET: 44 break; 45 default: 46 PyErr_Format(PyExc_RuntimeError, 47 "unknown error number %d", num); 48 } 49 return NULL; 50 } 51 52 #ifdef MS_WINDOWS 53 static PyObject * 54 multiprocessing_closesocket(PyObject *self, PyObject *args) 55 { 56 HANDLE handle; 57 int ret; 58 59 if (!PyArg_ParseTuple(args, F_HANDLE ":closesocket" , &handle)) 60 return NULL; 61 62 Py_BEGIN_ALLOW_THREADS 63 ret = closesocket((SOCKET) handle); 64 Py_END_ALLOW_THREADS 65 66 if (ret) 67 return PyErr_SetExcFromWindowsErr(PyExc_IOError, WSAGetLastError()); 68 Py_RETURN_NONE; 69 } 70 71 static PyObject * 72 multiprocessing_recv(PyObject *self, PyObject *args) 73 { 74 HANDLE handle; 75 int size, nread; 76 PyObject *buf; 77 78 if (!PyArg_ParseTuple(args, F_HANDLE "i:recv" , &handle, &size)) 79 return NULL; 80 81 buf = PyBytes_FromStringAndSize(NULL, size); 82 if (!buf) 83 return NULL; 84 85 Py_BEGIN_ALLOW_THREADS 86 nread = recv((SOCKET) handle, PyBytes_AS_STRING(buf), size, 0); 87 Py_END_ALLOW_THREADS 88 89 if (nread < 0) { 90 Py_DECREF(buf); 91 return PyErr_SetExcFromWindowsErr(PyExc_IOError, WSAGetLastError()); 92 } 93 _PyBytes_Resize(&buf, nread); 94 return buf; 95 } 96 97 static PyObject * 98 multiprocessing_send(PyObject *self, PyObject *args) 99 { 100 HANDLE handle; 101 Py_buffer buf; 102 int ret, length; 103 104 if (!PyArg_ParseTuple(args, F_HANDLE "y*:send" , &handle, &buf)) 105 return NULL; 106 107 length = (int)Py_MIN(buf.len, INT_MAX); 108 109 Py_BEGIN_ALLOW_THREADS 110 ret = send((SOCKET) handle, buf.buf, length, 0); 111 Py_END_ALLOW_THREADS 112 113 PyBuffer_Release(&buf); 114 if (ret < 0) 115 return PyErr_SetExcFromWindowsErr(PyExc_IOError, WSAGetLastError()); 116 return PyLong_FromLong(ret); 117 } 118 119 #endif 120 121 /* 122 * Function table 123 */ 124 125 static PyMethodDef module_methods[] = { 126 #ifdef MS_WINDOWS 127 {"closesocket", multiprocessing_closesocket, METH_VARARGS, ""}, 128 {"recv", multiprocessing_recv, METH_VARARGS, ""}, 129 {"send", multiprocessing_send, METH_VARARGS, ""}, 130 #endif 131 #if !defined(POSIX_SEMAPHORES_NOT_ENABLED) && !defined(__ANDROID__) 132 {"sem_unlink", _PyMp_sem_unlink, METH_VARARGS, ""}, 133 #endif 134 {NULL} 135 }; 136 137 138 /* 139 * Initialize 140 */ 141 142 static struct PyModuleDef multiprocessing_module = { 143 PyModuleDef_HEAD_INIT, 144 "_multiprocessing", 145 NULL, 146 -1, 147 module_methods, 148 NULL, 149 NULL, 150 NULL, 151 NULL 152 }; 153 154 155 PyMODINIT_FUNC 156 PyInit__multiprocessing(void) 157 { 158 PyObject *module, *temp, *value = NULL; 159 160 /* Initialize module */ 161 module = PyModule_Create(&multiprocessing_module); 162 if (!module) 163 return NULL; 164 165 #if defined(MS_WINDOWS) || \ 166 (defined(HAVE_SEM_OPEN) && !defined(POSIX_SEMAPHORES_NOT_ENABLED)) 167 /* Add _PyMp_SemLock type to module */ 168 if (PyType_Ready(&_PyMp_SemLockType) < 0) 169 return NULL; 170 Py_INCREF(&_PyMp_SemLockType); 171 { 172 PyObject *py_sem_value_max; 173 /* Some systems define SEM_VALUE_MAX as an unsigned value that 174 * causes it to be negative when used as an int (NetBSD). */ 175 if ((int)(SEM_VALUE_MAX) < 0) 176 py_sem_value_max = PyLong_FromLong(INT_MAX); 177 else 178 py_sem_value_max = PyLong_FromLong(SEM_VALUE_MAX); 179 if (py_sem_value_max == NULL) 180 return NULL; 181 PyDict_SetItemString(_PyMp_SemLockType.tp_dict, "SEM_VALUE_MAX", 182 py_sem_value_max); 183 } 184 PyModule_AddObject(module, "SemLock", (PyObject*)&_PyMp_SemLockType); 185 #endif 186 187 /* Add configuration macros */ 188 temp = PyDict_New(); 189 if (!temp) 190 return NULL; 191 192 #define ADD_FLAG(name) \ 193 value = Py_BuildValue("i", name); \ 194 if (value == NULL) { Py_DECREF(temp); return NULL; } \ 195 if (PyDict_SetItemString(temp, #name, value) < 0) { \ 196 Py_DECREF(temp); Py_DECREF(value); return NULL; } \ 197 Py_DECREF(value) 198 199 #if defined(HAVE_SEM_OPEN) && !defined(POSIX_SEMAPHORES_NOT_ENABLED) 200 ADD_FLAG(HAVE_SEM_OPEN); 201 #endif 202 #ifdef HAVE_SEM_TIMEDWAIT 203 ADD_FLAG(HAVE_SEM_TIMEDWAIT); 204 #endif 205 #ifdef HAVE_BROKEN_SEM_GETVALUE 206 ADD_FLAG(HAVE_BROKEN_SEM_GETVALUE); 207 #endif 208 #ifdef HAVE_BROKEN_SEM_UNLINK 209 ADD_FLAG(HAVE_BROKEN_SEM_UNLINK); 210 #endif 211 212 if (PyModule_AddObject(module, "flags", temp) < 0) 213 return NULL; 214 215 return module; 216 } 217
