| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 1 |  | 
 | 2 | /* New getargs implementation */ | 
 | 3 |  | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 4 | #include "Python.h" | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 5 |  | 
| Guido van Rossum | c1d5053 | 1996-08-21 23:38:24 +0000 | [diff] [blame] | 6 | #include <ctype.h> | 
 | 7 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 8 |  | 
| Anthony Baxter | 9730038 | 2006-04-12 04:38:54 +0000 | [diff] [blame] | 9 | #ifdef __cplusplus | 
 | 10 | extern "C" {  | 
 | 11 | #endif | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 12 | int PyArg_Parse(PyObject *, const char *, ...); | 
 | 13 | int PyArg_ParseTuple(PyObject *, const char *, ...); | 
 | 14 | int PyArg_VaParse(PyObject *, const char *, va_list); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 15 |  | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 16 | int PyArg_ParseTupleAndKeywords(PyObject *, PyObject *, | 
| Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 17 | 				const char *, char **, ...); | 
| Brett Cannon | 711e7d9 | 2004-07-10 22:20:32 +0000 | [diff] [blame] | 18 | int PyArg_VaParseTupleAndKeywords(PyObject *, PyObject *, | 
| Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 19 | 				const char *, char **, va_list); | 
| Brett Cannon | 711e7d9 | 2004-07-10 22:20:32 +0000 | [diff] [blame] | 20 |  | 
| Martin v. Löwis | 5cb6936 | 2006-04-14 09:08:42 +0000 | [diff] [blame] | 21 | #ifdef HAVE_DECLSPEC_DLL | 
 | 22 | /* Export functions */ | 
 | 23 | PyAPI_FUNC(int) _PyArg_Parse_SizeT(PyObject *, char *, ...); | 
 | 24 | PyAPI_FUNC(int) _PyArg_ParseTuple_SizeT(PyObject *, char *, ...); | 
 | 25 | PyAPI_FUNC(int) _PyArg_ParseTupleAndKeywords_SizeT(PyObject *, PyObject *, | 
 | 26 |                                                   const char *, char **, ...); | 
 | 27 | PyAPI_FUNC(PyObject *) _Py_BuildValue_SizeT(const char *, ...); | 
 | 28 | PyAPI_FUNC(int) _PyArg_VaParse_SizeT(PyObject *, char *, va_list); | 
 | 29 | PyAPI_FUNC(int) _PyArg_VaParseTupleAndKeywords_SizeT(PyObject *, PyObject *, | 
 | 30 |                                               const char *, char **, va_list); | 
 | 31 | #endif | 
 | 32 |  | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 33 | #define FLAG_COMPAT 1 | 
 | 34 | #define FLAG_SIZE_T 2 | 
 | 35 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 36 |  | 
 | 37 | /* Forward */ | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 38 | static int vgetargs1(PyObject *, const char *, va_list *, int); | 
 | 39 | static void seterror(int, const char *, int *, const char *, const char *); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 40 | static char *convertitem(PyObject *, const char **, va_list *, int, int *,  | 
 | 41 |                          char *, size_t, PyObject **); | 
 | 42 | static char *converttuple(PyObject *, const char **, va_list *, int, | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 43 | 			  int *, char *, size_t, int, PyObject **); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 44 | static char *convertsimple(PyObject *, const char **, va_list *, int, char *, | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 45 | 			   size_t, PyObject **); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 46 | static Py_ssize_t convertbuffer(PyObject *, void **p, char **); | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 47 | static int getbuffer(PyObject *, Py_buffer *, char**); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 48 |  | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 49 | static int vgetargskeywords(PyObject *, PyObject *, | 
| Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 50 | 			    const char *, char **, va_list *, int); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 51 | static char *skipitem(const char **, va_list *, int); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 52 |  | 
| Fred Drake | 563dfc2 | 2001-10-23 14:41:08 +0000 | [diff] [blame] | 53 | int | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 54 | PyArg_Parse(PyObject *args, const char *format, ...) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 55 | { | 
 | 56 | 	int retval; | 
 | 57 | 	va_list va; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 58 | 	 | 
 | 59 | 	va_start(va, format); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 60 | 	retval = vgetargs1(args, format, &va, FLAG_COMPAT); | 
 | 61 | 	va_end(va); | 
 | 62 | 	return retval; | 
 | 63 | } | 
 | 64 |  | 
 | 65 | int | 
 | 66 | _PyArg_Parse_SizeT(PyObject *args, char *format, ...) | 
 | 67 | { | 
 | 68 | 	int retval; | 
 | 69 | 	va_list va; | 
 | 70 | 	 | 
 | 71 | 	va_start(va, format); | 
 | 72 | 	retval = vgetargs1(args, format, &va, FLAG_COMPAT|FLAG_SIZE_T); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 73 | 	va_end(va); | 
 | 74 | 	return retval; | 
 | 75 | } | 
 | 76 |  | 
 | 77 |  | 
| Fred Drake | 563dfc2 | 2001-10-23 14:41:08 +0000 | [diff] [blame] | 78 | int | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 79 | PyArg_ParseTuple(PyObject *args, const char *format, ...) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 80 | { | 
 | 81 | 	int retval; | 
 | 82 | 	va_list va; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 83 | 	 | 
 | 84 | 	va_start(va, format); | 
| Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 85 | 	retval = vgetargs1(args, format, &va, 0); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 86 | 	va_end(va); | 
 | 87 | 	return retval; | 
 | 88 | } | 
 | 89 |  | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 90 | int | 
 | 91 | _PyArg_ParseTuple_SizeT(PyObject *args, char *format, ...) | 
 | 92 | { | 
 | 93 | 	int retval; | 
 | 94 | 	va_list va; | 
 | 95 | 	 | 
 | 96 | 	va_start(va, format); | 
 | 97 | 	retval = vgetargs1(args, format, &va, FLAG_SIZE_T); | 
 | 98 | 	va_end(va); | 
 | 99 | 	return retval; | 
 | 100 | } | 
 | 101 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 102 |  | 
 | 103 | int | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 104 | PyArg_VaParse(PyObject *args, const char *format, va_list va) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 105 | { | 
| Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 106 | 	va_list lva; | 
 | 107 |  | 
 | 108 | #ifdef VA_LIST_IS_ARRAY | 
 | 109 | 	memcpy(lva, va, sizeof(va_list)); | 
 | 110 | #else | 
| Martin v. Löwis | 75d2d94 | 2002-07-28 10:23:27 +0000 | [diff] [blame] | 111 | #ifdef __va_copy | 
 | 112 | 	__va_copy(lva, va); | 
 | 113 | #else | 
| Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 114 | 	lva = va; | 
 | 115 | #endif | 
| Martin v. Löwis | 75d2d94 | 2002-07-28 10:23:27 +0000 | [diff] [blame] | 116 | #endif | 
| Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 117 |  | 
 | 118 | 	return vgetargs1(args, format, &lva, 0); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 119 | } | 
 | 120 |  | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 121 | int | 
 | 122 | _PyArg_VaParse_SizeT(PyObject *args, char *format, va_list va) | 
 | 123 | { | 
 | 124 | 	va_list lva; | 
 | 125 |  | 
 | 126 | #ifdef VA_LIST_IS_ARRAY | 
 | 127 | 	memcpy(lva, va, sizeof(va_list)); | 
 | 128 | #else | 
 | 129 | #ifdef __va_copy | 
 | 130 | 	__va_copy(lva, va); | 
 | 131 | #else | 
 | 132 | 	lva = va; | 
 | 133 | #endif | 
 | 134 | #endif | 
 | 135 |  | 
 | 136 | 	return vgetargs1(args, format, &lva, FLAG_SIZE_T); | 
 | 137 | } | 
 | 138 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 139 |  | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 140 | /* Handle cleanup of allocated memory in case of exception */ | 
 | 141 |  | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 142 | static void | 
 | 143 | cleanup_ptr(void *ptr) | 
 | 144 | { | 
 | 145 | 	PyMem_FREE(ptr); | 
 | 146 | } | 
 | 147 |  | 
 | 148 | static void | 
 | 149 | cleanup_buffer(void *ptr) | 
 | 150 | { | 
 | 151 | 	PyBuffer_Release((Py_buffer *) ptr); | 
 | 152 | } | 
 | 153 |  | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 154 | static int | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 155 | addcleanup(void *ptr, PyObject **freelist, void (*destr)(void *)) | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 156 | { | 
 | 157 | 	PyObject *cobj; | 
 | 158 | 	if (!*freelist) { | 
 | 159 | 		*freelist = PyList_New(0); | 
 | 160 | 		if (!*freelist) { | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 161 | 			destr(ptr); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 162 | 			return -1; | 
 | 163 | 		} | 
 | 164 | 	} | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 165 | 	cobj = PyCObject_FromVoidPtr(ptr, destr); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 166 | 	if (!cobj) { | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 167 | 		destr(ptr); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 168 | 		return -1; | 
 | 169 | 	} | 
| Neal Norwitz | df6ac3d | 2008-02-26 05:23:51 +0000 | [diff] [blame] | 170 | 	if (PyList_Append(*freelist, cobj)) { | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 171 | 		Py_DECREF(cobj); | 
 | 172 | 		return -1; | 
 | 173 | 	} | 
 | 174 |         Py_DECREF(cobj); | 
 | 175 | 	return 0; | 
 | 176 | } | 
 | 177 |  | 
 | 178 | static int | 
 | 179 | cleanreturn(int retval, PyObject *freelist) | 
 | 180 | { | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 181 | 	if (freelist && retval != 0) { | 
 | 182 | 		/* We were successful, reset the destructors so that they | 
 | 183 | 		   don't get called. */ | 
 | 184 | 		Py_ssize_t len = PyList_GET_SIZE(freelist), i; | 
 | 185 | 		for (i = 0; i < len; i++) | 
 | 186 | 			((PyCObject *) PyList_GET_ITEM(freelist, i)) | 
 | 187 | 				->destructor = NULL; | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 188 | 	} | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 189 | 	Py_XDECREF(freelist); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 190 | 	return retval; | 
 | 191 | } | 
 | 192 |  | 
 | 193 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 194 | static int | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 195 | vgetargs1(PyObject *args, const char *format, va_list *p_va, int flags) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 196 | { | 
 | 197 | 	char msgbuf[256]; | 
 | 198 | 	int levels[32]; | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 199 | 	const char *fname = NULL; | 
 | 200 | 	const char *message = NULL; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 201 | 	int min = -1; | 
 | 202 | 	int max = 0; | 
 | 203 | 	int level = 0; | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 204 | 	int endfmt = 0; | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 205 | 	const char *formatsave = format; | 
| Martin v. Löwis | d96ee90 | 2006-02-16 14:37:16 +0000 | [diff] [blame] | 206 | 	Py_ssize_t i, len; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 207 | 	char *msg; | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 208 | 	PyObject *freelist = NULL; | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 209 | 	int compat = flags & FLAG_COMPAT; | 
 | 210 |  | 
| Tim Peters | 5c4d5bf | 2001-02-12 22:13:26 +0000 | [diff] [blame] | 211 | 	assert(compat || (args != (PyObject*)NULL)); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 212 | 	flags = flags & ~FLAG_COMPAT; | 
| Tim Peters | 5c4d5bf | 2001-02-12 22:13:26 +0000 | [diff] [blame] | 213 |  | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 214 | 	while (endfmt == 0) { | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 215 | 		int c = *format++; | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 216 | 		switch (c) { | 
 | 217 | 		case '(': | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 218 | 			if (level == 0) | 
 | 219 | 				max++; | 
 | 220 | 			level++; | 
| Georg Brandl | 209307e | 2006-08-09 07:03:22 +0000 | [diff] [blame] | 221 | 			if (level >= 30) | 
 | 222 | 				Py_FatalError("too many tuple nesting levels " | 
 | 223 | 					      "in argument format string"); | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 224 | 			break; | 
 | 225 | 		case ')': | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 226 | 			if (level == 0) | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 227 | 				Py_FatalError("excess ')' in getargs format"); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 228 | 			else | 
 | 229 | 				level--; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 230 | 			break; | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 231 | 		case '\0': | 
 | 232 | 			endfmt = 1; | 
 | 233 | 			break; | 
 | 234 | 		case ':': | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 235 | 			fname = format; | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 236 | 			endfmt = 1; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 237 | 			break; | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 238 | 		case ';': | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 239 | 			message = format; | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 240 | 			endfmt = 1; | 
 | 241 | 			break; | 
 | 242 | 		default: | 
 | 243 | 			if (level == 0) { | 
 | 244 | 				if (c == 'O') | 
 | 245 | 					max++; | 
| Neal Norwitz | 30b5c5d | 2005-12-19 06:05:18 +0000 | [diff] [blame] | 246 | 				else if (isalpha(Py_CHARMASK(c))) { | 
| Jeremy Hylton | 25916bd | 2001-05-29 17:46:19 +0000 | [diff] [blame] | 247 | 					if (c != 'e') /* skip encoded */ | 
 | 248 | 						max++; | 
 | 249 | 				} else if (c == '|') | 
 | 250 | 					min = max; | 
 | 251 | 			} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 252 | 			break; | 
 | 253 | 		} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 254 | 	} | 
 | 255 | 	 | 
 | 256 | 	if (level != 0) | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 257 | 		Py_FatalError(/* '(' */ "missing ')' in getargs format"); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 258 | 	 | 
 | 259 | 	if (min < 0) | 
 | 260 | 		min = max; | 
 | 261 | 	 | 
 | 262 | 	format = formatsave; | 
 | 263 | 	 | 
 | 264 | 	if (compat) { | 
 | 265 | 		if (max == 0) { | 
 | 266 | 			if (args == NULL) | 
 | 267 | 				return 1; | 
| Jeremy Hylton | 23ae987 | 2001-11-28 20:29:22 +0000 | [diff] [blame] | 268 | 			PyOS_snprintf(msgbuf, sizeof(msgbuf), | 
 | 269 | 				      "%.200s%s takes no arguments", | 
 | 270 | 				      fname==NULL ? "function" : fname, | 
 | 271 | 				      fname==NULL ? "" : "()"); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 272 | 			PyErr_SetString(PyExc_TypeError, msgbuf); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 273 | 			return 0; | 
 | 274 | 		} | 
 | 275 | 		else if (min == 1 && max == 1) { | 
| Guido van Rossum | 13d0ed1 | 1994-11-10 22:35:48 +0000 | [diff] [blame] | 276 | 			if (args == NULL) { | 
| Jeremy Hylton | 23ae987 | 2001-11-28 20:29:22 +0000 | [diff] [blame] | 277 | 				PyOS_snprintf(msgbuf, sizeof(msgbuf), | 
 | 278 | 				      "%.200s%s takes at least one argument", | 
 | 279 | 					      fname==NULL ? "function" : fname, | 
 | 280 | 					      fname==NULL ? "" : "()"); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 281 | 				PyErr_SetString(PyExc_TypeError, msgbuf); | 
| Guido van Rossum | 13d0ed1 | 1994-11-10 22:35:48 +0000 | [diff] [blame] | 282 | 				return 0; | 
 | 283 | 			} | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 284 | 			msg = convertitem(args, &format, p_va, flags, levels,  | 
 | 285 | 					  msgbuf, sizeof(msgbuf), &freelist); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 286 | 			if (msg == NULL) | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 287 | 				return cleanreturn(1, freelist); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 288 | 			seterror(levels[0], msg, levels+1, fname, message); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 289 | 			return cleanreturn(0, freelist); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 290 | 		} | 
 | 291 | 		else { | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 292 | 			PyErr_SetString(PyExc_SystemError, | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 293 | 			    "old style getargs format uses new features"); | 
 | 294 | 			return 0; | 
 | 295 | 		} | 
 | 296 | 	} | 
 | 297 | 	 | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 298 | 	if (!PyTuple_Check(args)) { | 
 | 299 | 		PyErr_SetString(PyExc_SystemError, | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 300 | 		    "new style getargs format but argument is not a tuple"); | 
 | 301 | 		return 0; | 
 | 302 | 	} | 
 | 303 | 	 | 
| Jeremy Hylton | 0f8117f | 2001-05-18 20:57:38 +0000 | [diff] [blame] | 304 | 	len = PyTuple_GET_SIZE(args); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 305 | 	 | 
 | 306 | 	if (len < min || max < len) { | 
 | 307 | 		if (message == NULL) { | 
| Jeremy Hylton | 23ae987 | 2001-11-28 20:29:22 +0000 | [diff] [blame] | 308 | 			PyOS_snprintf(msgbuf, sizeof(msgbuf), | 
 | 309 | 				      "%.150s%s takes %s %d argument%s " | 
| Neal Norwitz | 20dd93f | 2006-02-19 19:34:15 +0000 | [diff] [blame] | 310 | 				      "(%ld given)", | 
| Jeremy Hylton | 23ae987 | 2001-11-28 20:29:22 +0000 | [diff] [blame] | 311 | 				      fname==NULL ? "function" : fname, | 
 | 312 | 				      fname==NULL ? "" : "()", | 
 | 313 | 				      min==max ? "exactly" | 
 | 314 | 				      : len < min ? "at least" : "at most", | 
 | 315 | 				      len < min ? min : max, | 
 | 316 | 				      (len < min ? min : max) == 1 ? "" : "s", | 
| Neal Norwitz | 9a27617 | 2006-02-20 18:57:39 +0000 | [diff] [blame] | 317 | 				      Py_SAFE_DOWNCAST(len, Py_ssize_t, long)); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 318 | 			message = msgbuf; | 
 | 319 | 		} | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 320 | 		PyErr_SetString(PyExc_TypeError, message); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 321 | 		return 0; | 
 | 322 | 	} | 
 | 323 | 	 | 
 | 324 | 	for (i = 0; i < len; i++) { | 
 | 325 | 		if (*format == '|') | 
 | 326 | 			format++; | 
| Jeremy Hylton | 0f8117f | 2001-05-18 20:57:38 +0000 | [diff] [blame] | 327 | 		msg = convertitem(PyTuple_GET_ITEM(args, i), &format, p_va, | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 328 | 				  flags, levels, msgbuf,  | 
 | 329 | 				  sizeof(msgbuf), &freelist); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 330 | 		if (msg) { | 
| Sean Reifscheider | 9279e7d | 2009-08-01 23:54:55 +0000 | [diff] [blame] | 331 | 			seterror(i+1, msg, levels, fname, msg); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 332 | 			return cleanreturn(0, freelist); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 333 | 		} | 
 | 334 | 	} | 
| Guido van Rossum | 231a41e | 1997-12-09 20:36:39 +0000 | [diff] [blame] | 335 |  | 
| Neal Norwitz | 4ac13df | 2005-12-19 06:10:07 +0000 | [diff] [blame] | 336 | 	if (*format != '\0' && !isalpha(Py_CHARMASK(*format)) && | 
| Guido van Rossum | 7d4f68c | 1997-12-19 04:25:23 +0000 | [diff] [blame] | 337 | 	    *format != '(' && | 
| Guido van Rossum | 231a41e | 1997-12-09 20:36:39 +0000 | [diff] [blame] | 338 | 	    *format != '|' && *format != ':' && *format != ';') { | 
 | 339 | 		PyErr_Format(PyExc_SystemError, | 
| Guido van Rossum | 0d6b49e | 1998-01-19 22:22:44 +0000 | [diff] [blame] | 340 | 			     "bad format string: %.200s", formatsave); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 341 | 		return cleanreturn(0, freelist); | 
| Guido van Rossum | 231a41e | 1997-12-09 20:36:39 +0000 | [diff] [blame] | 342 | 	} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 343 | 	 | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 344 | 	return cleanreturn(1, freelist); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 345 | } | 
 | 346 |  | 
 | 347 |  | 
 | 348 |  | 
 | 349 | static void | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 350 | seterror(int iarg, const char *msg, int *levels, const char *fname, | 
 | 351 |          const char *message) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 352 | { | 
| Marc-André Lemburg | d4c0a9c | 2001-11-28 11:47:00 +0000 | [diff] [blame] | 353 | 	char buf[512]; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 354 | 	int i; | 
 | 355 | 	char *p = buf; | 
 | 356 |  | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 357 | 	if (PyErr_Occurred()) | 
| Guido van Rossum | 64fc649 | 1995-01-21 14:09:37 +0000 | [diff] [blame] | 358 | 		return; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 359 | 	else if (message == NULL) { | 
 | 360 | 		if (fname != NULL) { | 
| Jeremy Hylton | f16e05e | 2001-11-28 21:46:59 +0000 | [diff] [blame] | 361 | 			PyOS_snprintf(p, sizeof(buf), "%.200s() ", fname); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 362 | 			p += strlen(p); | 
 | 363 | 		} | 
| Ka-Ping Yee | 2057970 | 2001-01-15 22:14:16 +0000 | [diff] [blame] | 364 | 		if (iarg != 0) { | 
| Tim Peters | faad5ad | 2001-12-03 00:43:33 +0000 | [diff] [blame] | 365 | 			PyOS_snprintf(p, sizeof(buf) - (p - buf), | 
| Jeremy Hylton | f16e05e | 2001-11-28 21:46:59 +0000 | [diff] [blame] | 366 | 				      "argument %d", iarg); | 
| Ka-Ping Yee | 2057970 | 2001-01-15 22:14:16 +0000 | [diff] [blame] | 367 | 			i = 0; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 368 | 			p += strlen(p); | 
| Georg Brandl | 5f13578 | 2006-07-26 08:03:10 +0000 | [diff] [blame] | 369 | 			while (levels[i] > 0 && i < 32 && (int)(p-buf) < 220) { | 
 | 370 | 				PyOS_snprintf(p, sizeof(buf) - (p - buf), | 
| Jeremy Hylton | f16e05e | 2001-11-28 21:46:59 +0000 | [diff] [blame] | 371 | 					      ", item %d", levels[i]-1); | 
| Ka-Ping Yee | 2057970 | 2001-01-15 22:14:16 +0000 | [diff] [blame] | 372 | 				p += strlen(p); | 
 | 373 | 				i++; | 
 | 374 | 			} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 375 | 		} | 
| Ka-Ping Yee | 2057970 | 2001-01-15 22:14:16 +0000 | [diff] [blame] | 376 | 		else { | 
| Tim Peters | faad5ad | 2001-12-03 00:43:33 +0000 | [diff] [blame] | 377 | 			PyOS_snprintf(p, sizeof(buf) - (p - buf), "argument"); | 
| Ka-Ping Yee | 2057970 | 2001-01-15 22:14:16 +0000 | [diff] [blame] | 378 | 			p += strlen(p); | 
 | 379 | 		} | 
| Tim Peters | faad5ad | 2001-12-03 00:43:33 +0000 | [diff] [blame] | 380 | 		PyOS_snprintf(p, sizeof(buf) - (p - buf), " %.256s", msg); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 381 | 		message = buf; | 
 | 382 | 	} | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 383 | 	PyErr_SetString(PyExc_TypeError, message); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 384 | } | 
 | 385 |  | 
 | 386 |  | 
 | 387 | /* Convert a tuple argument. | 
 | 388 |    On entry, *p_format points to the character _after_ the opening '('. | 
 | 389 |    On successful exit, *p_format points to the closing ')'. | 
 | 390 |    If successful: | 
 | 391 |       *p_format and *p_va are updated, | 
 | 392 |       *levels and *msgbuf are untouched, | 
 | 393 |       and NULL is returned. | 
 | 394 |    If the argument is invalid: | 
 | 395 |       *p_format is unchanged, | 
 | 396 |       *p_va is undefined, | 
 | 397 |       *levels is a 0-terminated list of item numbers, | 
 | 398 |       *msgbuf contains an error message, whose format is: | 
| Ka-Ping Yee | 2057970 | 2001-01-15 22:14:16 +0000 | [diff] [blame] | 399 |          "must be <typename1>, not <typename2>", where: | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 400 |             <typename1> is the name of the expected type, and | 
 | 401 |             <typename2> is the name of the actual type, | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 402 |       and msgbuf is returned. | 
 | 403 | */ | 
 | 404 |  | 
 | 405 | static char * | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 406 | converttuple(PyObject *arg, const char **p_format, va_list *p_va, int flags, | 
 | 407 |              int *levels, char *msgbuf, size_t bufsize, int toplevel,  | 
 | 408 |              PyObject **freelist) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 409 | { | 
 | 410 | 	int level = 0; | 
 | 411 | 	int n = 0; | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 412 | 	const char *format = *p_format; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 413 | 	int i; | 
 | 414 | 	 | 
 | 415 | 	for (;;) { | 
 | 416 | 		int c = *format++; | 
 | 417 | 		if (c == '(') { | 
 | 418 | 			if (level == 0) | 
 | 419 | 				n++; | 
 | 420 | 			level++; | 
 | 421 | 		} | 
 | 422 | 		else if (c == ')') { | 
 | 423 | 			if (level == 0) | 
 | 424 | 				break; | 
 | 425 | 			level--; | 
 | 426 | 		} | 
 | 427 | 		else if (c == ':' || c == ';' || c == '\0') | 
 | 428 | 			break; | 
| Neal Norwitz | 30b5c5d | 2005-12-19 06:05:18 +0000 | [diff] [blame] | 429 | 		else if (level == 0 && isalpha(Py_CHARMASK(c))) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 430 | 			n++; | 
 | 431 | 	} | 
 | 432 | 	 | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 433 | 	if (!PySequence_Check(arg) || PyString_Check(arg)) { | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 434 | 		levels[0] = 0; | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 435 | 		PyOS_snprintf(msgbuf, bufsize, | 
| Jeremy Hylton | 23ae987 | 2001-11-28 20:29:22 +0000 | [diff] [blame] | 436 | 			      toplevel ? "expected %d arguments, not %.50s" : | 
 | 437 | 			              "must be %d-item sequence, not %.50s", | 
 | 438 | 			      n,  | 
 | 439 | 			      arg == Py_None ? "None" : arg->ob_type->tp_name); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 440 | 		return msgbuf; | 
 | 441 | 	} | 
 | 442 | 	 | 
| Jeremy Hylton | 03657cf | 2000-07-12 13:05:33 +0000 | [diff] [blame] | 443 | 	if ((i = PySequence_Size(arg)) != n) { | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 444 | 		levels[0] = 0; | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 445 | 		PyOS_snprintf(msgbuf, bufsize, | 
| Jeremy Hylton | 23ae987 | 2001-11-28 20:29:22 +0000 | [diff] [blame] | 446 | 			      toplevel ? "expected %d arguments, not %d" : | 
 | 447 | 			             "must be sequence of length %d, not %d", | 
 | 448 | 			      n, i); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 449 | 		return msgbuf; | 
 | 450 | 	} | 
| Ka-Ping Yee | 2057970 | 2001-01-15 22:14:16 +0000 | [diff] [blame] | 451 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 452 | 	format = *p_format; | 
 | 453 | 	for (i = 0; i < n; i++) { | 
 | 454 | 		char *msg; | 
| Guido van Rossum | 66368cc | 1999-02-17 23:16:43 +0000 | [diff] [blame] | 455 | 		PyObject *item; | 
 | 456 | 		item = PySequence_GetItem(arg, i); | 
| Georg Brandl | 5f13578 | 2006-07-26 08:03:10 +0000 | [diff] [blame] | 457 | 		if (item == NULL) { | 
 | 458 | 			PyErr_Clear(); | 
 | 459 | 			levels[0] = i+1; | 
 | 460 | 			levels[1] = 0; | 
 | 461 | 			strncpy(msgbuf, "is not retrievable", bufsize); | 
 | 462 | 			return msgbuf; | 
 | 463 | 		} | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 464 | 		msg = convertitem(item, &format, p_va, flags, levels+1,  | 
 | 465 | 				  msgbuf, bufsize, freelist); | 
| Guido van Rossum | 66368cc | 1999-02-17 23:16:43 +0000 | [diff] [blame] | 466 | 		/* PySequence_GetItem calls tp->sq_item, which INCREFs */ | 
 | 467 | 		Py_XDECREF(item); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 468 | 		if (msg != NULL) { | 
 | 469 | 			levels[0] = i+1; | 
 | 470 | 			return msg; | 
 | 471 | 		} | 
 | 472 | 	} | 
| Ka-Ping Yee | 2057970 | 2001-01-15 22:14:16 +0000 | [diff] [blame] | 473 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 474 | 	*p_format = format; | 
 | 475 | 	return NULL; | 
 | 476 | } | 
 | 477 |  | 
 | 478 |  | 
 | 479 | /* Convert a single item. */ | 
 | 480 |  | 
 | 481 | static char * | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 482 | convertitem(PyObject *arg, const char **p_format, va_list *p_va, int flags, | 
 | 483 |             int *levels, char *msgbuf, size_t bufsize, PyObject **freelist) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 484 | { | 
 | 485 | 	char *msg; | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 486 | 	const char *format = *p_format; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 487 | 	 | 
 | 488 | 	if (*format == '(' /* ')' */) { | 
 | 489 | 		format++; | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 490 | 		msg = converttuple(arg, &format, p_va, flags, levels, msgbuf,  | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 491 | 				   bufsize, 0, freelist); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 492 | 		if (msg == NULL) | 
 | 493 | 			format++; | 
 | 494 | 	} | 
 | 495 | 	else { | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 496 | 		msg = convertsimple(arg, &format, p_va, flags,  | 
 | 497 | 				    msgbuf, bufsize, freelist); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 498 | 		if (msg != NULL) | 
 | 499 | 			levels[0] = 0; | 
 | 500 | 	} | 
 | 501 | 	if (msg == NULL) | 
 | 502 | 		*p_format = format; | 
 | 503 | 	return msg; | 
 | 504 | } | 
 | 505 |  | 
 | 506 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 507 |  | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 508 | #define UNICODE_DEFAULT_ENCODING(arg) \ | 
 | 509 |         _PyUnicode_AsDefaultEncodedString(arg, NULL) | 
 | 510 |  | 
 | 511 | /* Format an error message generated by convertsimple(). */ | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 512 |  | 
 | 513 | static char * | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 514 | converterr(const char *expected, PyObject *arg, char *msgbuf, size_t bufsize) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 515 | { | 
| Jeremy Hylton | 77b8b67 | 2001-09-10 01:54:43 +0000 | [diff] [blame] | 516 | 	assert(expected != NULL); | 
 | 517 | 	assert(arg != NULL);  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 518 | 	PyOS_snprintf(msgbuf, bufsize, | 
 | 519 | 		      "must be %.50s, not %.50s", expected, | 
 | 520 | 		      arg == Py_None ? "None" : arg->ob_type->tp_name); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 521 | 	return msgbuf; | 
 | 522 | } | 
 | 523 |  | 
 | 524 | #define CONV_UNICODE "(unicode conversion error)" | 
 | 525 |  | 
| Neil Schemenauer | 5042da6 | 2003-02-04 20:59:40 +0000 | [diff] [blame] | 526 | /* explicitly check for float arguments when integers are expected.  For now | 
 | 527 |  * signal a warning.  Returns true if an exception was raised. */ | 
 | 528 | static int | 
| Mark Dickinson | 1b34d25 | 2010-01-01 17:27:30 +0000 | [diff] [blame^] | 529 | float_argument_warning(PyObject *arg) | 
| Neil Schemenauer | 5042da6 | 2003-02-04 20:59:40 +0000 | [diff] [blame] | 530 | { | 
 | 531 | 	if (PyFloat_Check(arg) && | 
 | 532 | 	    PyErr_Warn(PyExc_DeprecationWarning, | 
 | 533 | 		       "integer argument expected, got float" )) | 
 | 534 | 		return 1; | 
 | 535 | 	else | 
 | 536 | 		return 0; | 
 | 537 | } | 
 | 538 |  | 
| Mark Dickinson | 1b34d25 | 2010-01-01 17:27:30 +0000 | [diff] [blame^] | 539 | /* explicitly check for float arguments when integers are expected.  Raises | 
 | 540 |    TypeError and returns true for float arguments. */ | 
 | 541 | static int | 
 | 542 | float_argument_error(PyObject *arg) | 
 | 543 | { | 
 | 544 | 	if (PyFloat_Check(arg)) { | 
 | 545 | 		PyErr_SetString(PyExc_TypeError, | 
 | 546 | 				"integer argument expected, got float"); | 
 | 547 | 		return 1; | 
 | 548 | 	} | 
 | 549 | 	else | 
 | 550 | 		return 0; | 
 | 551 | } | 
 | 552 |  | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 553 | /* Convert a non-tuple argument.  Return NULL if conversion went OK, | 
 | 554 |    or a string with a message describing the failure.  The message is | 
 | 555 |    formatted as "must be <desired type>, not <actual type>". | 
 | 556 |    When failing, an exception may or may not have been raised. | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 557 |    Don't call if a tuple is expected. | 
 | 558 |  | 
 | 559 |    When you add new format codes, please don't forget poor skipitem() below. | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 560 | */ | 
 | 561 |  | 
 | 562 | static char * | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 563 | convertsimple(PyObject *arg, const char **p_format, va_list *p_va, int flags, | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 564 |               char *msgbuf, size_t bufsize, PyObject **freelist) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 565 | { | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 566 | 	/* For # codes */ | 
 | 567 | #define FETCH_SIZE	int *q=NULL;Py_ssize_t *q2=NULL;\ | 
 | 568 | 	if (flags & FLAG_SIZE_T) q2=va_arg(*p_va, Py_ssize_t*); \ | 
 | 569 | 	else q=va_arg(*p_va, int*); | 
 | 570 | #define STORE_SIZE(s)   if (flags & FLAG_SIZE_T) *q2=s; else *q=s; | 
 | 571 | #define BUFFER_LEN      ((flags & FLAG_SIZE_T) ? *q2:*q) | 
 | 572 |  | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 573 | 	const char *format = *p_format; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 574 | 	char c = *format++; | 
| Walter Dörwald | dffda2e | 2002-11-21 20:23:11 +0000 | [diff] [blame] | 575 | #ifdef Py_USING_UNICODE | 
| Jeremy Hylton | 77b8b67 | 2001-09-10 01:54:43 +0000 | [diff] [blame] | 576 | 	PyObject *uarg; | 
| Walter Dörwald | dffda2e | 2002-11-21 20:23:11 +0000 | [diff] [blame] | 577 | #endif | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 578 | 	 | 
 | 579 | 	switch (c) { | 
 | 580 | 	 | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 581 | 	case 'b': { /* unsigned byte -- very short int */ | 
 | 582 | 		char *p = va_arg(*p_va, char *); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 583 | 		long ival; | 
| Neil Schemenauer | 5042da6 | 2003-02-04 20:59:40 +0000 | [diff] [blame] | 584 | 		if (float_argument_error(arg)) | 
| Michael W. Hudson | 3455338 | 2004-08-07 17:57:16 +0000 | [diff] [blame] | 585 | 			return converterr("integer<b>", arg, msgbuf, bufsize); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 586 | 		ival = PyInt_AsLong(arg); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 587 | 		if (ival == -1 && PyErr_Occurred()) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 588 | 			return converterr("integer<b>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 589 | 		else if (ival < 0) { | 
 | 590 | 			PyErr_SetString(PyExc_OverflowError, | 
 | 591 | 			"unsigned byte integer is less than minimum"); | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 592 | 			return converterr("integer<b>", arg, msgbuf, bufsize); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 593 | 		} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 594 | 		else if (ival > UCHAR_MAX) { | 
 | 595 | 			PyErr_SetString(PyExc_OverflowError, | 
 | 596 | 			"unsigned byte integer is greater than maximum"); | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 597 | 			return converterr("integer<b>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 598 | 		} | 
 | 599 | 		else | 
 | 600 | 			*p = (unsigned char) ival; | 
 | 601 | 		break; | 
 | 602 | 	} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 603 | 	 | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 604 | 	case 'B': {/* byte sized bitfield - both signed and unsigned | 
 | 605 | 		      values allowed */   | 
 | 606 | 		char *p = va_arg(*p_va, char *); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 607 | 		long ival; | 
| Neil Schemenauer | 5042da6 | 2003-02-04 20:59:40 +0000 | [diff] [blame] | 608 | 		if (float_argument_error(arg)) | 
| Michael W. Hudson | 3455338 | 2004-08-07 17:57:16 +0000 | [diff] [blame] | 609 | 			return converterr("integer<B>", arg, msgbuf, bufsize); | 
| Thomas Heller | a4ea603 | 2003-04-17 18:55:45 +0000 | [diff] [blame] | 610 | 		ival = PyInt_AsUnsignedLongMask(arg); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 611 | 		if (ival == -1 && PyErr_Occurred()) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 612 | 			return converterr("integer<B>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 613 | 		else | 
 | 614 | 			*p = (unsigned char) ival; | 
 | 615 | 		break; | 
 | 616 | 	} | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 617 | 	 | 
| Guido van Rossum | fce26e7 | 2003-04-18 00:12:30 +0000 | [diff] [blame] | 618 | 	case 'h': {/* signed short int */ | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 619 | 		short *p = va_arg(*p_va, short *); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 620 | 		long ival; | 
| Neil Schemenauer | 5042da6 | 2003-02-04 20:59:40 +0000 | [diff] [blame] | 621 | 		if (float_argument_error(arg)) | 
| Michael W. Hudson | 3455338 | 2004-08-07 17:57:16 +0000 | [diff] [blame] | 622 | 			return converterr("integer<h>", arg, msgbuf, bufsize); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 623 | 		ival = PyInt_AsLong(arg); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 624 | 		if (ival == -1 && PyErr_Occurred()) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 625 | 			return converterr("integer<h>", arg, msgbuf, bufsize); | 
| Guido van Rossum | fce26e7 | 2003-04-18 00:12:30 +0000 | [diff] [blame] | 626 | 		else if (ival < SHRT_MIN) { | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 627 | 			PyErr_SetString(PyExc_OverflowError, | 
| Guido van Rossum | fce26e7 | 2003-04-18 00:12:30 +0000 | [diff] [blame] | 628 | 			"signed short integer is less than minimum"); | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 629 | 			return converterr("integer<h>", arg, msgbuf, bufsize); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 630 | 		} | 
| Guido van Rossum | fce26e7 | 2003-04-18 00:12:30 +0000 | [diff] [blame] | 631 | 		else if (ival > SHRT_MAX) { | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 632 | 			PyErr_SetString(PyExc_OverflowError, | 
| Guido van Rossum | fce26e7 | 2003-04-18 00:12:30 +0000 | [diff] [blame] | 633 | 			"signed short integer is greater than maximum"); | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 634 | 			return converterr("integer<h>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 635 | 		} | 
 | 636 | 		else | 
 | 637 | 			*p = (short) ival; | 
 | 638 | 		break; | 
 | 639 | 	} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 640 | 	 | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 641 | 	case 'H': { /* short int sized bitfield, both signed and | 
 | 642 | 		       unsigned allowed */  | 
 | 643 | 		unsigned short *p = va_arg(*p_va, unsigned short *); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 644 | 		long ival; | 
| Neil Schemenauer | 5042da6 | 2003-02-04 20:59:40 +0000 | [diff] [blame] | 645 | 		if (float_argument_error(arg)) | 
| Michael W. Hudson | 3455338 | 2004-08-07 17:57:16 +0000 | [diff] [blame] | 646 | 			return converterr("integer<H>", arg, msgbuf, bufsize); | 
| Thomas Heller | a4ea603 | 2003-04-17 18:55:45 +0000 | [diff] [blame] | 647 | 		ival = PyInt_AsUnsignedLongMask(arg); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 648 | 		if (ival == -1 && PyErr_Occurred()) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 649 | 			return converterr("integer<H>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 650 | 		else | 
 | 651 | 			*p = (unsigned short) ival; | 
 | 652 | 		break; | 
 | 653 | 	} | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 654 |  | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 655 | 	case 'i': {/* signed int */ | 
 | 656 | 		int *p = va_arg(*p_va, int *); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 657 | 		long ival; | 
| Neil Schemenauer | 5042da6 | 2003-02-04 20:59:40 +0000 | [diff] [blame] | 658 | 		if (float_argument_error(arg)) | 
| Michael W. Hudson | 3455338 | 2004-08-07 17:57:16 +0000 | [diff] [blame] | 659 | 			return converterr("integer<i>", arg, msgbuf, bufsize); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 660 | 		ival = PyInt_AsLong(arg); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 661 | 		if (ival == -1 && PyErr_Occurred()) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 662 | 			return converterr("integer<i>", arg, msgbuf, bufsize); | 
| Georg Brandl | 98251f8 | 2006-06-08 13:31:07 +0000 | [diff] [blame] | 663 | 		else if (ival > INT_MAX) { | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 664 | 			PyErr_SetString(PyExc_OverflowError, | 
 | 665 | 				"signed integer is greater than maximum"); | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 666 | 			return converterr("integer<i>", arg, msgbuf, bufsize); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 667 | 		} | 
| Georg Brandl | 98251f8 | 2006-06-08 13:31:07 +0000 | [diff] [blame] | 668 | 		else if (ival < INT_MIN) { | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 669 | 			PyErr_SetString(PyExc_OverflowError, | 
 | 670 | 				"signed integer is less than minimum"); | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 671 | 			return converterr("integer<i>", arg, msgbuf, bufsize); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 672 | 		} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 673 | 		else | 
 | 674 | 			*p = ival; | 
 | 675 | 		break; | 
 | 676 | 	} | 
 | 677 |  | 
| Thomas Heller | a4ea603 | 2003-04-17 18:55:45 +0000 | [diff] [blame] | 678 | 	case 'I': { /* int sized bitfield, both signed and | 
 | 679 | 		       unsigned allowed */  | 
 | 680 | 		unsigned int *p = va_arg(*p_va, unsigned int *); | 
 | 681 | 		unsigned int ival; | 
 | 682 | 		if (float_argument_error(arg)) | 
| Michael W. Hudson | 3455338 | 2004-08-07 17:57:16 +0000 | [diff] [blame] | 683 | 			return converterr("integer<I>", arg, msgbuf, bufsize); | 
| Skip Montanaro | b507972 | 2006-04-18 00:57:15 +0000 | [diff] [blame] | 684 | 		ival = (unsigned int)PyInt_AsUnsignedLongMask(arg); | 
 | 685 | 		if (ival == (unsigned int)-1 && PyErr_Occurred()) | 
| Thomas Heller | a4ea603 | 2003-04-17 18:55:45 +0000 | [diff] [blame] | 686 | 			return converterr("integer<I>", arg, msgbuf, bufsize); | 
 | 687 | 		else | 
 | 688 | 			*p = ival; | 
 | 689 | 		break; | 
 | 690 | 	} | 
 | 691 | 	 | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 692 | 	case 'n': /* Py_ssize_t */ | 
 | 693 | #if SIZEOF_SIZE_T != SIZEOF_LONG | 
 | 694 | 	{ | 
 | 695 | 		Py_ssize_t *p = va_arg(*p_va, Py_ssize_t *); | 
 | 696 | 		Py_ssize_t ival; | 
 | 697 | 		if (float_argument_error(arg)) | 
| Georg Brandl | 7f573f7 | 2006-04-13 07:59:30 +0000 | [diff] [blame] | 698 | 			return converterr("integer<n>", arg, msgbuf, bufsize); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 699 | 		ival = PyInt_AsSsize_t(arg); | 
 | 700 | 		if (ival == -1 && PyErr_Occurred()) | 
| Georg Brandl | 7f573f7 | 2006-04-13 07:59:30 +0000 | [diff] [blame] | 701 | 			return converterr("integer<n>", arg, msgbuf, bufsize); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 702 | 		*p = ival; | 
 | 703 | 		break; | 
 | 704 | 	} | 
 | 705 | #endif | 
 | 706 | 	/* Fall through from 'n' to 'l' if Py_ssize_t is int */ | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 707 | 	case 'l': {/* long int */ | 
 | 708 | 		long *p = va_arg(*p_va, long *); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 709 | 		long ival; | 
| Neil Schemenauer | 5042da6 | 2003-02-04 20:59:40 +0000 | [diff] [blame] | 710 | 		if (float_argument_error(arg)) | 
| Michael W. Hudson | 3455338 | 2004-08-07 17:57:16 +0000 | [diff] [blame] | 711 | 			return converterr("integer<l>", arg, msgbuf, bufsize); | 
| Neil Schemenauer | b808e99 | 2003-01-24 22:15:21 +0000 | [diff] [blame] | 712 | 		ival = PyInt_AsLong(arg); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 713 | 		if (ival == -1 && PyErr_Occurred()) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 714 | 			return converterr("integer<l>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 715 | 		else | 
 | 716 | 			*p = ival; | 
 | 717 | 		break; | 
 | 718 | 	} | 
| Thomas Heller | a4ea603 | 2003-04-17 18:55:45 +0000 | [diff] [blame] | 719 |  | 
 | 720 | 	case 'k': { /* long sized bitfield */ | 
 | 721 | 		unsigned long *p = va_arg(*p_va, unsigned long *); | 
 | 722 | 		unsigned long ival; | 
 | 723 | 		if (PyInt_Check(arg)) | 
 | 724 | 			ival = PyInt_AsUnsignedLongMask(arg); | 
 | 725 | 		else if (PyLong_Check(arg)) | 
 | 726 | 			ival = PyLong_AsUnsignedLongMask(arg); | 
 | 727 | 		else | 
 | 728 | 			return converterr("integer<k>", arg, msgbuf, bufsize); | 
 | 729 | 		*p = ival; | 
 | 730 | 		break; | 
 | 731 | 	} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 732 | 	 | 
| Guido van Rossum | 3dbba6e | 1999-01-25 21:48:56 +0000 | [diff] [blame] | 733 | #ifdef HAVE_LONG_LONG | 
| Martin v. Löwis | b9a0f91 | 2003-03-29 10:06:18 +0000 | [diff] [blame] | 734 | 	case 'L': {/* PY_LONG_LONG */ | 
 | 735 | 		PY_LONG_LONG *p = va_arg( *p_va, PY_LONG_LONG * ); | 
| Mark Dickinson | 1b34d25 | 2010-01-01 17:27:30 +0000 | [diff] [blame^] | 736 | 		PY_LONG_LONG ival; | 
 | 737 | 		if (float_argument_warning(arg)) | 
 | 738 | 			return converterr("long<L>", arg, msgbuf, bufsize); | 
 | 739 | 		ival = PyLong_AsLongLong(arg); | 
| Neal Norwitz | df6ac3d | 2008-02-26 05:23:51 +0000 | [diff] [blame] | 740 | 		if (ival == (PY_LONG_LONG)-1 && PyErr_Occurred() ) { | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 741 | 			return converterr("long<L>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 742 | 		} else { | 
 | 743 | 			*p = ival; | 
| Guido van Rossum | 1a8791e | 1998-08-04 22:46:29 +0000 | [diff] [blame] | 744 | 		} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 745 | 		break; | 
 | 746 | 	} | 
| Thomas Heller | a4ea603 | 2003-04-17 18:55:45 +0000 | [diff] [blame] | 747 |  | 
 | 748 | 	case 'K': { /* long long sized bitfield */ | 
 | 749 | 		unsigned PY_LONG_LONG *p = va_arg(*p_va, unsigned PY_LONG_LONG *); | 
 | 750 | 		unsigned PY_LONG_LONG ival; | 
| Thomas Heller | a4ea603 | 2003-04-17 18:55:45 +0000 | [diff] [blame] | 751 | 		if (PyInt_Check(arg)) | 
 | 752 | 			ival = PyInt_AsUnsignedLongMask(arg); | 
 | 753 | 		else if (PyLong_Check(arg)) | 
 | 754 | 			ival = PyLong_AsUnsignedLongLongMask(arg); | 
 | 755 | 		else | 
 | 756 | 			return converterr("integer<K>", arg, msgbuf, bufsize); | 
 | 757 | 		*p = ival; | 
 | 758 | 		break; | 
 | 759 | 	} | 
| Guido van Rossum | 1a8791e | 1998-08-04 22:46:29 +0000 | [diff] [blame] | 760 | #endif | 
 | 761 | 	 | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 762 | 	case 'f': {/* float */ | 
 | 763 | 		float *p = va_arg(*p_va, float *); | 
 | 764 | 		double dval = PyFloat_AsDouble(arg); | 
 | 765 | 		if (PyErr_Occurred()) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 766 | 			return converterr("float<f>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 767 | 		else | 
 | 768 | 			*p = (float) dval; | 
 | 769 | 		break; | 
 | 770 | 	} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 771 | 	 | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 772 | 	case 'd': {/* double */ | 
 | 773 | 		double *p = va_arg(*p_va, double *); | 
 | 774 | 		double dval = PyFloat_AsDouble(arg); | 
 | 775 | 		if (PyErr_Occurred()) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 776 | 			return converterr("float<d>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 777 | 		else | 
 | 778 | 			*p = dval; | 
 | 779 | 		break; | 
 | 780 | 	} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 781 | 	 | 
| Guido van Rossum | 530956d | 1996-07-21 02:27:43 +0000 | [diff] [blame] | 782 | #ifndef WITHOUT_COMPLEX | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 783 | 	case 'D': {/* complex double */ | 
 | 784 | 		Py_complex *p = va_arg(*p_va, Py_complex *); | 
 | 785 | 		Py_complex cval; | 
 | 786 | 		cval = PyComplex_AsCComplex(arg); | 
 | 787 | 		if (PyErr_Occurred()) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 788 | 			return converterr("complex<D>", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 789 | 		else | 
 | 790 | 			*p = cval; | 
 | 791 | 		break; | 
 | 792 | 	} | 
| Guido van Rossum | 530956d | 1996-07-21 02:27:43 +0000 | [diff] [blame] | 793 | #endif /* WITHOUT_COMPLEX */ | 
| Guido van Rossum | 8a5c5d2 | 1996-01-12 01:09:56 +0000 | [diff] [blame] | 794 | 	 | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 795 | 	case 'c': {/* char */ | 
 | 796 | 		char *p = va_arg(*p_va, char *); | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 797 | 		if (PyString_Check(arg) && PyString_Size(arg) == 1) | 
 | 798 | 			*p = PyString_AS_STRING(arg)[0]; | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 799 | 		else | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 800 | 			return converterr("char", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 801 | 		break; | 
 | 802 | 	} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 803 | 	 | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 804 | 	case 's': {/* string */ | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 805 | 		if (*format == '*') { | 
 | 806 | 			Py_buffer *p = (Py_buffer *)va_arg(*p_va, Py_buffer *); | 
 | 807 |  | 
 | 808 | 			if (PyString_Check(arg)) { | 
 | 809 | 				PyBuffer_FillInfo(p, arg, | 
 | 810 | 						  PyString_AS_STRING(arg), PyString_GET_SIZE(arg), | 
 | 811 | 						  1, 0); | 
 | 812 | 			} | 
 | 813 | #ifdef Py_USING_UNICODE | 
 | 814 | 			else if (PyUnicode_Check(arg)) { | 
 | 815 | 				uarg = UNICODE_DEFAULT_ENCODING(arg); | 
 | 816 | 				if (uarg == NULL) | 
 | 817 | 					return converterr(CONV_UNICODE, | 
 | 818 | 							  arg, msgbuf, bufsize); | 
 | 819 | 				PyBuffer_FillInfo(p, arg, | 
 | 820 | 						  PyString_AS_STRING(uarg), PyString_GET_SIZE(uarg), | 
 | 821 | 						  1, 0); | 
 | 822 | 			} | 
 | 823 | #endif | 
 | 824 | 			else { /* any buffer-like object */ | 
 | 825 | 				char *buf; | 
 | 826 | 				if (getbuffer(arg, p, &buf) < 0) | 
 | 827 | 					return converterr(buf, arg, msgbuf, bufsize); | 
 | 828 | 			} | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 829 | 			if (addcleanup(p, freelist, cleanup_buffer)) { | 
 | 830 | 				return converterr( | 
 | 831 | 					"(cleanup problem)", | 
 | 832 | 					arg, msgbuf, bufsize); | 
 | 833 | 			} | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 834 | 			format++; | 
 | 835 | 		} else if (*format == '#') { | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 836 | 			void **p = (void **)va_arg(*p_va, char **); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 837 | 			FETCH_SIZE; | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 838 | 			 | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 839 | 			if (PyString_Check(arg)) { | 
 | 840 | 				*p = PyString_AS_STRING(arg); | 
 | 841 | 				STORE_SIZE(PyString_GET_SIZE(arg)); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 842 | 			} | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 843 | #ifdef Py_USING_UNICODE | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 844 | 			else if (PyUnicode_Check(arg)) { | 
| Jeremy Hylton | 77b8b67 | 2001-09-10 01:54:43 +0000 | [diff] [blame] | 845 | 				uarg = UNICODE_DEFAULT_ENCODING(arg); | 
 | 846 | 				if (uarg == NULL) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 847 | 					return converterr(CONV_UNICODE, | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 848 | 							  arg, msgbuf, bufsize); | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 849 | 				*p = PyString_AS_STRING(uarg); | 
 | 850 | 				STORE_SIZE(PyString_GET_SIZE(uarg)); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 851 | 			} | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 852 | #endif | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 853 | 			else { /* any buffer-like object */ | 
 | 854 | 				char *buf; | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 855 | 				Py_ssize_t count = convertbuffer(arg, p, &buf); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 856 | 				if (count < 0) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 857 | 					return converterr(buf, arg, msgbuf, bufsize); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 858 | 				STORE_SIZE(count); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 859 | 			} | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 860 | 			format++; | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 861 | 		} else { | 
 | 862 | 			char **p = va_arg(*p_va, char **); | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 863 | 			 | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 864 | 			if (PyString_Check(arg)) | 
 | 865 | 				*p = PyString_AS_STRING(arg); | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 866 | #ifdef Py_USING_UNICODE | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 867 | 			else if (PyUnicode_Check(arg)) { | 
| Jeremy Hylton | 77b8b67 | 2001-09-10 01:54:43 +0000 | [diff] [blame] | 868 | 				uarg = UNICODE_DEFAULT_ENCODING(arg); | 
 | 869 | 				if (uarg == NULL) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 870 | 					return converterr(CONV_UNICODE, | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 871 | 							  arg, msgbuf, bufsize); | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 872 | 				*p = PyString_AS_STRING(uarg); | 
| Marc-André Lemburg | 6f15e57 | 2001-05-02 17:16:16 +0000 | [diff] [blame] | 873 | 			} | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 874 | #endif | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 875 | 			else | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 876 | 				return converterr("string", arg, msgbuf, bufsize); | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 877 | 			if ((Py_ssize_t)strlen(*p) != PyString_Size(arg)) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 878 | 				return converterr("string without null bytes", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 879 | 						  arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 880 | 		} | 
 | 881 | 		break; | 
 | 882 | 	} | 
 | 883 |  | 
 | 884 | 	case 'z': {/* string, may be NULL (None) */ | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 885 | 		if (*format == '*') { | 
 | 886 | 			Py_buffer *p = (Py_buffer *)va_arg(*p_va, Py_buffer *); | 
 | 887 |  | 
 | 888 | 			if (arg == Py_None) | 
 | 889 | 				PyBuffer_FillInfo(p, NULL, NULL, 0, 1, 0); | 
 | 890 | 			else if (PyString_Check(arg)) { | 
 | 891 | 				PyBuffer_FillInfo(p, arg, | 
 | 892 | 						  PyString_AS_STRING(arg), PyString_GET_SIZE(arg), | 
 | 893 | 						  1, 0); | 
 | 894 | 			} | 
 | 895 | #ifdef Py_USING_UNICODE | 
 | 896 | 			else if (PyUnicode_Check(arg)) { | 
 | 897 | 				uarg = UNICODE_DEFAULT_ENCODING(arg); | 
 | 898 | 				if (uarg == NULL) | 
 | 899 | 					return converterr(CONV_UNICODE, | 
 | 900 | 							  arg, msgbuf, bufsize); | 
 | 901 | 				PyBuffer_FillInfo(p, arg, | 
 | 902 | 						  PyString_AS_STRING(uarg), PyString_GET_SIZE(uarg), | 
 | 903 | 						  1, 0); | 
 | 904 | 			} | 
 | 905 | #endif | 
 | 906 | 			else { /* any buffer-like object */ | 
 | 907 | 				char *buf; | 
 | 908 | 				if (getbuffer(arg, p, &buf) < 0) | 
 | 909 | 					return converterr(buf, arg, msgbuf, bufsize); | 
 | 910 | 			} | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 911 | 			if (addcleanup(p, freelist, cleanup_buffer)) { | 
 | 912 | 				return converterr( | 
 | 913 | 					"(cleanup problem)", | 
 | 914 | 					arg, msgbuf, bufsize); | 
 | 915 | 			} | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 916 | 			format++; | 
 | 917 | 		} else if (*format == '#') { /* any buffer-like object */ | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 918 | 			void **p = (void **)va_arg(*p_va, char **); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 919 | 			FETCH_SIZE; | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 920 | 			 | 
 | 921 | 			if (arg == Py_None) { | 
 | 922 | 				*p = 0; | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 923 | 				STORE_SIZE(0); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 924 | 			} | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 925 | 			else if (PyString_Check(arg)) { | 
 | 926 | 				*p = PyString_AS_STRING(arg); | 
 | 927 | 				STORE_SIZE(PyString_GET_SIZE(arg)); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 928 | 			} | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 929 | #ifdef Py_USING_UNICODE | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 930 | 			else if (PyUnicode_Check(arg)) { | 
| Jeremy Hylton | 77b8b67 | 2001-09-10 01:54:43 +0000 | [diff] [blame] | 931 | 				uarg = UNICODE_DEFAULT_ENCODING(arg); | 
 | 932 | 				if (uarg == NULL) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 933 | 					return converterr(CONV_UNICODE, | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 934 | 							  arg, msgbuf, bufsize); | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 935 | 				*p = PyString_AS_STRING(uarg); | 
 | 936 | 				STORE_SIZE(PyString_GET_SIZE(uarg)); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 937 | 			} | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 938 | #endif | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 939 | 			else { /* any buffer-like object */ | 
 | 940 | 				char *buf; | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 941 | 				Py_ssize_t count = convertbuffer(arg, p, &buf); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 942 | 				if (count < 0) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 943 | 					return converterr(buf, arg, msgbuf, bufsize); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 944 | 				STORE_SIZE(count); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 945 | 			} | 
 | 946 | 			format++; | 
 | 947 | 		} else { | 
 | 948 | 			char **p = va_arg(*p_va, char **); | 
 | 949 | 			 | 
 | 950 | 			if (arg == Py_None) | 
 | 951 | 				*p = 0; | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 952 | 			else if (PyString_Check(arg)) | 
 | 953 | 				*p = PyString_AS_STRING(arg); | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 954 | #ifdef Py_USING_UNICODE | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 955 | 			else if (PyUnicode_Check(arg)) { | 
| Jeremy Hylton | 77b8b67 | 2001-09-10 01:54:43 +0000 | [diff] [blame] | 956 | 				uarg = UNICODE_DEFAULT_ENCODING(arg); | 
 | 957 | 				if (uarg == NULL) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 958 | 					return converterr(CONV_UNICODE, | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 959 | 							  arg, msgbuf, bufsize); | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 960 | 				*p = PyString_AS_STRING(uarg); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 961 | 			} | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 962 | #endif | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 963 | 			else | 
 | 964 | 				return converterr("string or None",  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 965 | 						  arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 966 | 			if (*format == '#') { | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 967 | 				FETCH_SIZE; | 
| Thomas Wouters | c3547a3 | 2006-03-01 21:31:21 +0000 | [diff] [blame] | 968 | 				assert(0); /* XXX redundant with if-case */ | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 969 | 				if (arg == Py_None) | 
 | 970 | 					*q = 0; | 
 | 971 | 				else | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 972 | 					*q = PyString_Size(arg); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 973 | 				format++; | 
 | 974 | 			} | 
 | 975 | 			else if (*p != NULL && | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 976 | 				 (Py_ssize_t)strlen(*p) != PyString_Size(arg)) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 977 | 				return converterr( | 
 | 978 | 					"string without null bytes or None",  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 979 | 					arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 980 | 		} | 
 | 981 | 		break; | 
 | 982 | 	} | 
 | 983 | 	 | 
 | 984 | 	case 'e': {/* encoded string */ | 
 | 985 | 		char **buffer; | 
 | 986 | 		const char *encoding; | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 987 | 		PyObject *s; | 
| Amaury Forgeot d'Arc | dafd32b | 2007-11-30 20:51:40 +0000 | [diff] [blame] | 988 | 		Py_ssize_t size; | 
 | 989 | 		int recode_strings; | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 990 |  | 
 | 991 | 		/* Get 'e' parameter: the encoding name */ | 
 | 992 | 		encoding = (const char *)va_arg(*p_va, const char *); | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 993 | #ifdef Py_USING_UNICODE | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 994 | 		if (encoding == NULL) | 
 | 995 | 			encoding = PyUnicode_GetDefaultEncoding(); | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 996 | #endif | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 997 | 			 | 
 | 998 | 		/* Get output buffer parameter: | 
 | 999 | 		   's' (recode all objects via Unicode) or | 
 | 1000 | 		   't' (only recode non-string objects)  | 
 | 1001 | 		*/ | 
 | 1002 | 		if (*format == 's') | 
 | 1003 | 			recode_strings = 1; | 
 | 1004 | 		else if (*format == 't') | 
 | 1005 | 			recode_strings = 0; | 
 | 1006 | 		else | 
 | 1007 | 			return converterr( | 
 | 1008 | 				"(unknown parser marker combination)", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1009 | 				arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1010 | 		buffer = (char **)va_arg(*p_va, char **); | 
 | 1011 | 		format++; | 
 | 1012 | 		if (buffer == NULL) | 
 | 1013 | 			return converterr("(buffer is NULL)",  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1014 | 					  arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1015 | 			 | 
 | 1016 | 		/* Encode object */ | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 1017 | 		if (!recode_strings && PyString_Check(arg)) { | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1018 | 			s = arg; | 
 | 1019 | 			Py_INCREF(s); | 
 | 1020 | 		} | 
 | 1021 | 		else { | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 1022 | #ifdef Py_USING_UNICODE | 
 | 1023 | 		    	PyObject *u; | 
 | 1024 |  | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 1025 | 			/* Convert object to Unicode */ | 
 | 1026 | 			u = PyUnicode_FromObject(arg); | 
 | 1027 | 			if (u == NULL) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1028 | 				return converterr( | 
 | 1029 | 					"string or unicode or text buffer",  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1030 | 					arg, msgbuf, bufsize); | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 1031 | 			 | 
 | 1032 | 			/* Encode object; use default error handling */ | 
 | 1033 | 			s = PyUnicode_AsEncodedString(u, | 
 | 1034 | 						      encoding, | 
 | 1035 | 						      NULL); | 
 | 1036 | 			Py_DECREF(u); | 
 | 1037 | 			if (s == NULL) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1038 | 				return converterr("(encoding failed)", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1039 | 						  arg, msgbuf, bufsize); | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 1040 | 			if (!PyString_Check(s)) { | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 1041 | 				Py_DECREF(s); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1042 | 				return converterr( | 
 | 1043 | 					"(encoder failed to return a string)", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1044 | 					arg, msgbuf, bufsize); | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 1045 | 			} | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 1046 | #else | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1047 | 			return converterr("string<e>", arg, msgbuf, bufsize); | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 1048 | #endif | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1049 | 		} | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 1050 | 		size = PyString_GET_SIZE(s); | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 1051 |  | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1052 | 		/* Write output; output is guaranteed to be 0-terminated */ | 
 | 1053 | 		if (*format == '#') {  | 
 | 1054 | 			/* Using buffer length parameter '#': | 
 | 1055 | 				    | 
 | 1056 | 			   - if *buffer is NULL, a new buffer of the | 
 | 1057 | 			   needed size is allocated and the data | 
 | 1058 | 			   copied into it; *buffer is updated to point | 
 | 1059 | 			   to the new buffer; the caller is | 
 | 1060 | 			   responsible for PyMem_Free()ing it after | 
 | 1061 | 			   usage  | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 1062 |  | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1063 | 			   - if *buffer is not NULL, the data is | 
 | 1064 | 			   copied to *buffer; *buffer_len has to be | 
 | 1065 | 			   set to the size of the buffer on input; | 
 | 1066 | 			   buffer overflow is signalled with an error; | 
 | 1067 | 			   buffer has to provide enough room for the | 
 | 1068 | 			   encoded string plus the trailing 0-byte | 
 | 1069 | 			    | 
 | 1070 | 			   - in both cases, *buffer_len is updated to | 
 | 1071 | 			   the size of the buffer /excluding/ the | 
 | 1072 | 			   trailing 0-byte  | 
 | 1073 | 			    | 
 | 1074 | 			*/ | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1075 | 			FETCH_SIZE; | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 1076 |  | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1077 | 			format++; | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1078 | 			if (q == NULL && q2 == NULL) { | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1079 | 				Py_DECREF(s); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1080 | 				return converterr( | 
 | 1081 | 					"(buffer_len is NULL)", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1082 | 					arg, msgbuf, bufsize); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1083 | 			} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1084 | 			if (*buffer == NULL) { | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 1085 | 				*buffer = PyMem_NEW(char, size + 1); | 
 | 1086 | 				if (*buffer == NULL) { | 
 | 1087 | 					Py_DECREF(s); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1088 | 					return converterr( | 
 | 1089 | 						"(memory error)", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1090 | 						arg, msgbuf, bufsize); | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 1091 | 				} | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 1092 | 				if (addcleanup(*buffer, freelist, cleanup_ptr)) { | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1093 | 					Py_DECREF(s); | 
 | 1094 | 					return converterr( | 
 | 1095 | 						"(cleanup problem)", | 
 | 1096 | 						arg, msgbuf, bufsize); | 
 | 1097 | 				} | 
| Fred Drake | 25871c0 | 2000-05-03 15:17:02 +0000 | [diff] [blame] | 1098 | 			} else { | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1099 | 				if (size + 1 > BUFFER_LEN) { | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1100 | 					Py_DECREF(s); | 
 | 1101 | 					return converterr( | 
 | 1102 | 						"(buffer overflow)",  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1103 | 						arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1104 | 				} | 
| Fred Drake | 25871c0 | 2000-05-03 15:17:02 +0000 | [diff] [blame] | 1105 | 			} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1106 | 			memcpy(*buffer, | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 1107 | 			       PyString_AS_STRING(s), | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1108 | 			       size + 1); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1109 | 			STORE_SIZE(size); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1110 | 		} else { | 
 | 1111 | 			/* Using a 0-terminated buffer: | 
 | 1112 | 				    | 
 | 1113 | 			   - the encoded string has to be 0-terminated | 
 | 1114 | 			   for this variant to work; if it is not, an | 
 | 1115 | 			   error raised  | 
| Fred Drake | 25871c0 | 2000-05-03 15:17:02 +0000 | [diff] [blame] | 1116 |  | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1117 | 			   - a new buffer of the needed size is | 
 | 1118 | 			   allocated and the data copied into it; | 
 | 1119 | 			   *buffer is updated to point to the new | 
 | 1120 | 			   buffer; the caller is responsible for | 
 | 1121 | 			   PyMem_Free()ing it after usage | 
 | 1122 |  | 
 | 1123 | 			*/ | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 1124 | 			if ((Py_ssize_t)strlen(PyString_AS_STRING(s)) | 
| Armin Rigo | 7ccbca9 | 2006-10-04 12:17:45 +0000 | [diff] [blame] | 1125 | 								!= size) { | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1126 | 				Py_DECREF(s); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1127 | 				return converterr( | 
| Georg Brandl | 6c59e72 | 2009-04-05 11:54:07 +0000 | [diff] [blame] | 1128 | 					"encoded string without NULL bytes", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1129 | 					arg, msgbuf, bufsize); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1130 | 			} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1131 | 			*buffer = PyMem_NEW(char, size + 1); | 
 | 1132 | 			if (*buffer == NULL) { | 
 | 1133 | 				Py_DECREF(s); | 
 | 1134 | 				return converterr("(memory error)", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1135 | 						  arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1136 | 			} | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 1137 | 			if (addcleanup(*buffer, freelist, cleanup_ptr)) { | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1138 | 				Py_DECREF(s); | 
 | 1139 | 				return converterr("(cleanup problem)", | 
 | 1140 | 						arg, msgbuf, bufsize); | 
 | 1141 | 			} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1142 | 			memcpy(*buffer, | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 1143 | 			       PyString_AS_STRING(s), | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1144 | 			       size + 1); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 1145 | 		} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1146 | 		Py_DECREF(s); | 
 | 1147 | 		break; | 
 | 1148 | 	} | 
 | 1149 |  | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 1150 | #ifdef Py_USING_UNICODE | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1151 | 	case 'u': {/* raw unicode buffer (Py_UNICODE *) */ | 
 | 1152 | 		if (*format == '#') { /* any buffer-like object */ | 
 | 1153 | 			void **p = (void **)va_arg(*p_va, char **); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1154 | 			FETCH_SIZE; | 
| Marc-André Lemburg | 3e3eacb | 2002-01-09 16:21:27 +0000 | [diff] [blame] | 1155 | 			if (PyUnicode_Check(arg)) { | 
 | 1156 | 			    	*p = PyUnicode_AS_UNICODE(arg); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1157 | 				STORE_SIZE(PyUnicode_GET_SIZE(arg)); | 
| Marc-André Lemburg | 3e3eacb | 2002-01-09 16:21:27 +0000 | [diff] [blame] | 1158 | 			} | 
 | 1159 | 			else { | 
| Neal Norwitz | 6154616 | 2006-04-14 05:20:28 +0000 | [diff] [blame] | 1160 | 				return converterr("cannot convert raw buffers", | 
 | 1161 | 						  arg, msgbuf, bufsize); | 
| Marc-André Lemburg | 3e3eacb | 2002-01-09 16:21:27 +0000 | [diff] [blame] | 1162 | 			} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1163 | 			format++; | 
 | 1164 | 		} else { | 
 | 1165 | 			Py_UNICODE **p = va_arg(*p_va, Py_UNICODE **); | 
| Guido van Rossum | e826ef0 | 2000-03-10 23:02:17 +0000 | [diff] [blame] | 1166 | 			if (PyUnicode_Check(arg)) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1167 | 				*p = PyUnicode_AS_UNICODE(arg); | 
 | 1168 | 			else | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1169 | 				return converterr("unicode", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1170 | 		} | 
 | 1171 | 		break; | 
 | 1172 | 	} | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 1173 | #endif | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1174 |  | 
 | 1175 | 	case 'S': { /* string object */ | 
 | 1176 | 		PyObject **p = va_arg(*p_va, PyObject **); | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 1177 | 		if (PyString_Check(arg)) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1178 | 			*p = arg; | 
 | 1179 | 		else | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1180 | 			return converterr("string", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1181 | 		break; | 
 | 1182 | 	} | 
 | 1183 | 	 | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 1184 | #ifdef Py_USING_UNICODE | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1185 | 	case 'U': { /* Unicode object */ | 
 | 1186 | 		PyObject **p = va_arg(*p_va, PyObject **); | 
 | 1187 | 		if (PyUnicode_Check(arg)) | 
 | 1188 | 			*p = arg; | 
 | 1189 | 		else | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1190 | 			return converterr("unicode", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1191 | 		break; | 
 | 1192 | 	} | 
| Martin v. Löwis | 339d0f7 | 2001-08-17 18:39:25 +0000 | [diff] [blame] | 1193 | #endif | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1194 | 	 | 
 | 1195 | 	case 'O': { /* object */ | 
 | 1196 | 		PyTypeObject *type; | 
 | 1197 | 		PyObject **p; | 
 | 1198 | 		if (*format == '!') { | 
 | 1199 | 			type = va_arg(*p_va, PyTypeObject*); | 
 | 1200 | 			p = va_arg(*p_va, PyObject **); | 
 | 1201 | 			format++; | 
| Guido van Rossum | cbfc855 | 2001-08-28 16:37:51 +0000 | [diff] [blame] | 1202 | 			if (PyType_IsSubtype(arg->ob_type, type)) | 
| Guido van Rossum | e826ef0 | 2000-03-10 23:02:17 +0000 | [diff] [blame] | 1203 | 				*p = arg; | 
 | 1204 | 			else | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1205 | 				return converterr(type->tp_name, arg, msgbuf, bufsize); | 
| Guido van Rossum | fccfe89 | 1998-05-15 22:04:07 +0000 | [diff] [blame] | 1206 |  | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1207 | 		} | 
 | 1208 | 		else if (*format == '?') { | 
 | 1209 | 			inquiry pred = va_arg(*p_va, inquiry); | 
 | 1210 | 			p = va_arg(*p_va, PyObject **); | 
 | 1211 | 			format++; | 
 | 1212 | 			if ((*pred)(arg))  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 1213 | 				*p = arg; | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1214 | 			else | 
 | 1215 | 				return converterr("(unspecified)",  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1216 | 						  arg, msgbuf, bufsize); | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 1217 | 				 | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 1218 | 		} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1219 | 		else if (*format == '&') { | 
 | 1220 | 			typedef int (*converter)(PyObject *, void *); | 
 | 1221 | 			converter convert = va_arg(*p_va, converter); | 
 | 1222 | 			void *addr = va_arg(*p_va, void *); | 
 | 1223 | 			format++; | 
 | 1224 | 			if (! (*convert)(arg, addr)) | 
 | 1225 | 				return converterr("(unspecified)",  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1226 | 						  arg, msgbuf, bufsize); | 
| Guido van Rossum | b317f8a | 1998-10-08 02:21:21 +0000 | [diff] [blame] | 1227 | 		} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1228 | 		else { | 
 | 1229 | 			p = va_arg(*p_va, PyObject **); | 
 | 1230 | 			*p = arg; | 
 | 1231 | 		} | 
 | 1232 | 		break; | 
 | 1233 | 	} | 
| Guido van Rossum | b317f8a | 1998-10-08 02:21:21 +0000 | [diff] [blame] | 1234 | 		 | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1235 | 		 | 
 | 1236 | 	case 'w': { /* memory buffer, read-write access */ | 
 | 1237 | 		void **p = va_arg(*p_va, void **); | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1238 | 		void *res; | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1239 | 		PyBufferProcs *pb = arg->ob_type->tp_as_buffer; | 
| Amaury Forgeot d'Arc | dafd32b | 2007-11-30 20:51:40 +0000 | [diff] [blame] | 1240 | 		Py_ssize_t count; | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1241 |  | 
 | 1242 | 		if (pb && pb->bf_releasebuffer && *format != '*') | 
 | 1243 | 			/* Buffer must be released, yet caller does not use | 
 | 1244 | 			   the Py_buffer protocol. */ | 
 | 1245 | 			return converterr("pinned buffer", arg, msgbuf, bufsize); | 
 | 1246 |  | 
 | 1247 | 		if (pb && pb->bf_getbuffer && *format == '*') { | 
 | 1248 | 			/* Caller is interested in Py_buffer, and the object | 
 | 1249 | 			   supports it directly. */ | 
 | 1250 | 			format++; | 
 | 1251 | 			if (pb->bf_getbuffer(arg, (Py_buffer*)p, PyBUF_WRITABLE) < 0) { | 
 | 1252 | 				PyErr_Clear(); | 
 | 1253 | 				return converterr("read-write buffer", arg, msgbuf, bufsize); | 
 | 1254 | 			} | 
| Antoine Pitrou | d4ae97b | 2008-08-29 18:39:48 +0000 | [diff] [blame] | 1255 | 			if (addcleanup(p, freelist, cleanup_buffer)) { | 
 | 1256 | 				return converterr( | 
 | 1257 | 					"(cleanup problem)", | 
 | 1258 | 					arg, msgbuf, bufsize); | 
 | 1259 | 			} | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1260 | 			if (!PyBuffer_IsContiguous((Py_buffer*)p, 'C')) | 
 | 1261 | 				return converterr("contiguous buffer", arg, msgbuf, bufsize); | 
 | 1262 | 			break; | 
 | 1263 | 		} | 
 | 1264 |  | 
 | 1265 | 		if (pb == NULL || | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1266 | 		    pb->bf_getwritebuffer == NULL || | 
 | 1267 | 		    pb->bf_getsegcount == NULL) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1268 | 			return converterr("read-write buffer", arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1269 | 		if ((*pb->bf_getsegcount)(arg, NULL) != 1) | 
 | 1270 | 			return converterr("single-segment read-write buffer",  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1271 | 					  arg, msgbuf, bufsize); | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1272 | 		if ((count = pb->bf_getwritebuffer(arg, 0, &res)) < 0) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1273 | 			return converterr("(unspecified)", arg, msgbuf, bufsize); | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1274 | 		if (*format == '*') { | 
 | 1275 | 			PyBuffer_FillInfo((Py_buffer*)p, arg, res, count, 1, 0); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1276 | 			format++; | 
 | 1277 | 		} | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1278 | 		else { | 
 | 1279 | 			*p = res; | 
 | 1280 | 			if (*format == '#') { | 
 | 1281 | 				FETCH_SIZE; | 
 | 1282 | 				STORE_SIZE(count); | 
 | 1283 | 				format++; | 
 | 1284 | 			} | 
 | 1285 | 		} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1286 | 		break; | 
 | 1287 | 	} | 
 | 1288 | 		 | 
 | 1289 | 	case 't': { /* 8-bit character buffer, read-only access */ | 
| Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 1290 | 		char **p = va_arg(*p_va, char **); | 
| Jeremy Hylton | 4819e97 | 2001-10-11 14:40:37 +0000 | [diff] [blame] | 1291 | 		PyBufferProcs *pb = arg->ob_type->tp_as_buffer; | 
| Amaury Forgeot d'Arc | dafd32b | 2007-11-30 20:51:40 +0000 | [diff] [blame] | 1292 | 		Py_ssize_t count; | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1293 | 		 | 
 | 1294 | 		if (*format++ != '#') | 
 | 1295 | 			return converterr( | 
 | 1296 | 				"invalid use of 't' format character",  | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1297 | 				arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1298 | 		if (!PyType_HasFeature(arg->ob_type, | 
| Jeremy Hylton | 4819e97 | 2001-10-11 14:40:37 +0000 | [diff] [blame] | 1299 | 				       Py_TPFLAGS_HAVE_GETCHARBUFFER) || | 
 | 1300 | 		    pb == NULL || pb->bf_getcharbuffer == NULL || | 
 | 1301 | 		    pb->bf_getsegcount == NULL) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1302 | 			return converterr( | 
 | 1303 | 				"string or read-only character buffer", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1304 | 				arg, msgbuf, bufsize); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1305 |  | 
| Jeremy Hylton | 4819e97 | 2001-10-11 14:40:37 +0000 | [diff] [blame] | 1306 | 		if (pb->bf_getsegcount(arg, NULL) != 1) | 
 | 1307 | 			return converterr( | 
 | 1308 | 				"string or single-segment read-only buffer", | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1309 | 				arg, msgbuf, bufsize); | 
| Jeremy Hylton | 4819e97 | 2001-10-11 14:40:37 +0000 | [diff] [blame] | 1310 |  | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1311 | 		if (pb->bf_releasebuffer) | 
 | 1312 | 			return converterr( | 
 | 1313 | 				"string or pinned buffer", | 
 | 1314 | 				arg, msgbuf, bufsize); | 
 | 1315 |  | 
| Jeremy Hylton | 4819e97 | 2001-10-11 14:40:37 +0000 | [diff] [blame] | 1316 | 		count = pb->bf_getcharbuffer(arg, 0, p); | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1317 | 		if (count < 0) | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1318 | 			return converterr("(unspecified)", arg, msgbuf, bufsize); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1319 | 		{ | 
 | 1320 | 			FETCH_SIZE; | 
 | 1321 | 			STORE_SIZE(count); | 
 | 1322 | 		} | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1323 | 		break; | 
 | 1324 | 	} | 
 | 1325 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 1326 | 	default: | 
| Jeremy Hylton | b048b26 | 2001-11-28 22:14:37 +0000 | [diff] [blame] | 1327 | 		return converterr("impossible<bad format char>", arg, msgbuf, bufsize); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 1328 | 	 | 
 | 1329 | 	} | 
 | 1330 | 	 | 
 | 1331 | 	*p_format = format; | 
 | 1332 | 	return NULL; | 
 | 1333 | } | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1334 |  | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1335 | static Py_ssize_t | 
| Fred Drake | 563dfc2 | 2001-10-23 14:41:08 +0000 | [diff] [blame] | 1336 | convertbuffer(PyObject *arg, void **p, char **errmsg) | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1337 | { | 
 | 1338 | 	PyBufferProcs *pb = arg->ob_type->tp_as_buffer; | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1339 | 	Py_ssize_t count; | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1340 | 	if (pb == NULL || | 
 | 1341 | 	    pb->bf_getreadbuffer == NULL || | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1342 | 	    pb->bf_getsegcount == NULL || | 
 | 1343 | 	    pb->bf_releasebuffer != NULL) { | 
| Jeremy Hylton | 1cb7aa3 | 2001-05-29 17:37:05 +0000 | [diff] [blame] | 1344 | 		*errmsg = "string or read-only buffer"; | 
 | 1345 | 		return -1; | 
 | 1346 | 	} | 
 | 1347 | 	if ((*pb->bf_getsegcount)(arg, NULL) != 1) { | 
 | 1348 | 		*errmsg = "string or single-segment read-only buffer"; | 
 | 1349 | 		return -1; | 
 | 1350 | 	} | 
 | 1351 | 	if ((count = (*pb->bf_getreadbuffer)(arg, 0, p)) < 0) { | 
 | 1352 | 		*errmsg = "(unspecified)"; | 
 | 1353 | 	} | 
 | 1354 | 	return count; | 
 | 1355 | } | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1356 |  | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1357 | static int | 
| Neal Norwitz | 18aa388 | 2008-08-24 05:04:52 +0000 | [diff] [blame] | 1358 | getbuffer(PyObject *arg, Py_buffer *view, char **errmsg) | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1359 | { | 
 | 1360 | 	void *buf; | 
 | 1361 | 	Py_ssize_t count; | 
 | 1362 | 	PyBufferProcs *pb = arg->ob_type->tp_as_buffer; | 
 | 1363 | 	if (pb == NULL) { | 
 | 1364 | 		*errmsg = "string or buffer"; | 
 | 1365 | 		return -1; | 
 | 1366 | 	} | 
 | 1367 | 	if (pb->bf_getbuffer) { | 
| Neal Norwitz | 18aa388 | 2008-08-24 05:04:52 +0000 | [diff] [blame] | 1368 | 		if (pb->bf_getbuffer(arg, view, 0) < 0) { | 
 | 1369 | 			*errmsg = "convertible to a buffer"; | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1370 | 			return -1; | 
| Neal Norwitz | 18aa388 | 2008-08-24 05:04:52 +0000 | [diff] [blame] | 1371 | 		} | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1372 | 		if (!PyBuffer_IsContiguous(view, 'C')) { | 
 | 1373 | 			*errmsg = "contiguous buffer"; | 
 | 1374 | 			return -1; | 
 | 1375 | 		} | 
 | 1376 | 		return 0; | 
 | 1377 | 	} | 
 | 1378 |  | 
 | 1379 | 	count = convertbuffer(arg, &buf, errmsg); | 
| Neal Norwitz | 18aa388 | 2008-08-24 05:04:52 +0000 | [diff] [blame] | 1380 | 	if (count < 0) { | 
 | 1381 | 		*errmsg = "convertible to a buffer"; | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1382 | 		return count; | 
| Neal Norwitz | 18aa388 | 2008-08-24 05:04:52 +0000 | [diff] [blame] | 1383 | 	} | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1384 | 	PyBuffer_FillInfo(view, NULL, buf, count, 1, 0); | 
 | 1385 | 	return 0; | 
 | 1386 | } | 
 | 1387 |  | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1388 | /* Support for keyword arguments donated by | 
 | 1389 |    Geoff Philbrick <philbric@delphi.hks.com> */ | 
 | 1390 |  | 
| Tim Peters | f8cd3e8 | 2001-10-27 04:26:57 +0000 | [diff] [blame] | 1391 | /* Return false (0) for error, else true. */ | 
| Fred Drake | 563dfc2 | 2001-10-23 14:41:08 +0000 | [diff] [blame] | 1392 | int | 
 | 1393 | PyArg_ParseTupleAndKeywords(PyObject *args, | 
 | 1394 | 			    PyObject *keywords, | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 1395 | 			    const char *format,  | 
| Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 1396 | 			    char **kwlist, ...) | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1397 | { | 
 | 1398 | 	int retval; | 
 | 1399 | 	va_list va; | 
| Tim Peters | 45772cd | 2001-10-27 03:58:40 +0000 | [diff] [blame] | 1400 |  | 
 | 1401 | 	if ((args == NULL || !PyTuple_Check(args)) || | 
 | 1402 | 	    (keywords != NULL && !PyDict_Check(keywords)) || | 
 | 1403 | 	    format == NULL || | 
 | 1404 | 	    kwlist == NULL) | 
 | 1405 | 	{ | 
 | 1406 | 		PyErr_BadInternalCall(); | 
| Tim Peters | f8cd3e8 | 2001-10-27 04:26:57 +0000 | [diff] [blame] | 1407 | 		return 0; | 
| Tim Peters | 45772cd | 2001-10-27 03:58:40 +0000 | [diff] [blame] | 1408 | 	} | 
 | 1409 |  | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1410 | 	va_start(va, kwlist); | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1411 | 	retval = vgetargskeywords(args, keywords, format, kwlist, &va, 0);	 | 
 | 1412 | 	va_end(va); | 
 | 1413 | 	return retval; | 
 | 1414 | } | 
 | 1415 |  | 
 | 1416 | int | 
 | 1417 | _PyArg_ParseTupleAndKeywords_SizeT(PyObject *args, | 
 | 1418 | 				  PyObject *keywords, | 
 | 1419 | 				  const char *format,  | 
| Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 1420 | 				  char **kwlist, ...) | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1421 | { | 
 | 1422 | 	int retval; | 
 | 1423 | 	va_list va; | 
 | 1424 |  | 
 | 1425 | 	if ((args == NULL || !PyTuple_Check(args)) || | 
 | 1426 | 	    (keywords != NULL && !PyDict_Check(keywords)) || | 
 | 1427 | 	    format == NULL || | 
 | 1428 | 	    kwlist == NULL) | 
 | 1429 | 	{ | 
 | 1430 | 		PyErr_BadInternalCall(); | 
 | 1431 | 		return 0; | 
 | 1432 | 	} | 
 | 1433 |  | 
 | 1434 | 	va_start(va, kwlist); | 
 | 1435 | 	retval = vgetargskeywords(args, keywords, format,  | 
 | 1436 | 				  kwlist, &va, FLAG_SIZE_T); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1437 | 	va_end(va); | 
 | 1438 | 	return retval; | 
 | 1439 | } | 
 | 1440 |  | 
 | 1441 |  | 
| Brett Cannon | 711e7d9 | 2004-07-10 22:20:32 +0000 | [diff] [blame] | 1442 | int | 
 | 1443 | PyArg_VaParseTupleAndKeywords(PyObject *args, | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 1444 |                               PyObject *keywords, | 
 | 1445 |                               const char *format,  | 
| Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 1446 |                               char **kwlist, va_list va) | 
| Brett Cannon | 711e7d9 | 2004-07-10 22:20:32 +0000 | [diff] [blame] | 1447 | { | 
 | 1448 | 	int retval; | 
 | 1449 | 	va_list lva; | 
 | 1450 |  | 
 | 1451 | 	if ((args == NULL || !PyTuple_Check(args)) || | 
 | 1452 | 	    (keywords != NULL && !PyDict_Check(keywords)) || | 
 | 1453 | 	    format == NULL || | 
 | 1454 | 	    kwlist == NULL) | 
 | 1455 | 	{ | 
 | 1456 | 		PyErr_BadInternalCall(); | 
 | 1457 | 		return 0; | 
 | 1458 | 	} | 
 | 1459 |  | 
 | 1460 | #ifdef VA_LIST_IS_ARRAY | 
 | 1461 | 	memcpy(lva, va, sizeof(va_list)); | 
 | 1462 | #else | 
 | 1463 | #ifdef __va_copy | 
 | 1464 | 	__va_copy(lva, va); | 
 | 1465 | #else | 
 | 1466 | 	lva = va; | 
 | 1467 | #endif | 
 | 1468 | #endif | 
 | 1469 |  | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1470 | 	retval = vgetargskeywords(args, keywords, format, kwlist, &lva, 0);	 | 
 | 1471 | 	return retval; | 
 | 1472 | } | 
 | 1473 |  | 
 | 1474 | int | 
 | 1475 | _PyArg_VaParseTupleAndKeywords_SizeT(PyObject *args, | 
 | 1476 | 				    PyObject *keywords, | 
 | 1477 | 				    const char *format,  | 
| Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 1478 | 				    char **kwlist, va_list va) | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1479 | { | 
 | 1480 | 	int retval; | 
 | 1481 | 	va_list lva; | 
 | 1482 |  | 
 | 1483 | 	if ((args == NULL || !PyTuple_Check(args)) || | 
 | 1484 | 	    (keywords != NULL && !PyDict_Check(keywords)) || | 
 | 1485 | 	    format == NULL || | 
 | 1486 | 	    kwlist == NULL) | 
 | 1487 | 	{ | 
 | 1488 | 		PyErr_BadInternalCall(); | 
 | 1489 | 		return 0; | 
 | 1490 | 	} | 
 | 1491 |  | 
 | 1492 | #ifdef VA_LIST_IS_ARRAY | 
 | 1493 | 	memcpy(lva, va, sizeof(va_list)); | 
 | 1494 | #else | 
 | 1495 | #ifdef __va_copy | 
 | 1496 | 	__va_copy(lva, va); | 
 | 1497 | #else | 
 | 1498 | 	lva = va; | 
 | 1499 | #endif | 
 | 1500 | #endif | 
 | 1501 |  | 
 | 1502 | 	retval = vgetargskeywords(args, keywords, format,  | 
 | 1503 | 				  kwlist, &lva, FLAG_SIZE_T); | 
| Brett Cannon | 711e7d9 | 2004-07-10 22:20:32 +0000 | [diff] [blame] | 1504 | 	return retval; | 
 | 1505 | } | 
 | 1506 |  | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1507 | #define IS_END_OF_FORMAT(c) (c == '\0' || c == ';' || c == ':') | 
| Brett Cannon | 711e7d9 | 2004-07-10 22:20:32 +0000 | [diff] [blame] | 1508 |  | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1509 | static int | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 1510 | vgetargskeywords(PyObject *args, PyObject *keywords, const char *format, | 
| Martin v. Löwis | 15e6274 | 2006-02-27 16:46:16 +0000 | [diff] [blame] | 1511 | 	         char **kwlist, va_list *p_va, int flags) | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1512 | { | 
| Tim Peters | dc5eff9 | 2001-10-27 06:53:00 +0000 | [diff] [blame] | 1513 | 	char msgbuf[512]; | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1514 | 	int levels[32]; | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1515 | 	const char *fname, *msg, *custom_msg, *keyword; | 
 | 1516 | 	int min = INT_MAX; | 
| Tim Peters | b639d49 | 2001-10-27 07:00:56 +0000 | [diff] [blame] | 1517 | 	int i, len, nargs, nkeywords; | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1518 | 	PyObject *freelist = NULL, *current_arg; | 
| Tim Peters | f4331c1 | 2001-10-27 00:17:34 +0000 | [diff] [blame] | 1519 |  | 
| Tim Peters | 45772cd | 2001-10-27 03:58:40 +0000 | [diff] [blame] | 1520 | 	assert(args != NULL && PyTuple_Check(args)); | 
 | 1521 | 	assert(keywords == NULL || PyDict_Check(keywords)); | 
 | 1522 | 	assert(format != NULL); | 
 | 1523 | 	assert(kwlist != NULL); | 
 | 1524 | 	assert(p_va != NULL); | 
 | 1525 |  | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1526 | 	/* grab the function name or custom error msg first (mutually exclusive) */ | 
 | 1527 | 	fname = strchr(format, ':'); | 
 | 1528 | 	if (fname) { | 
 | 1529 | 		fname++; | 
 | 1530 | 		custom_msg = NULL; | 
| Tim Peters | 62d48e1 | 2001-10-27 06:42:16 +0000 | [diff] [blame] | 1531 | 	} | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1532 | 	else { | 
 | 1533 | 		custom_msg = strchr(format,';'); | 
 | 1534 | 		if (custom_msg) | 
 | 1535 | 			custom_msg++; | 
| Tim Peters | 62d48e1 | 2001-10-27 06:42:16 +0000 | [diff] [blame] | 1536 | 	} | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1537 |  | 
 | 1538 | 	/* scan kwlist and get greatest possible nbr of args */ | 
 | 1539 | 	for (len=0; kwlist[len]; len++) | 
 | 1540 | 		continue; | 
| Tim Peters | f8cd3e8 | 2001-10-27 04:26:57 +0000 | [diff] [blame] | 1541 |  | 
| Tim Peters | 6fb2635 | 2001-10-27 04:38:11 +0000 | [diff] [blame] | 1542 | 	nargs = PyTuple_GET_SIZE(args); | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1543 | 	nkeywords = (keywords == NULL) ? 0 : PyDict_Size(keywords); | 
 | 1544 | 	if (nargs + nkeywords > len) { | 
 | 1545 | 		PyErr_Format(PyExc_TypeError, "%s%s takes at most %d " | 
 | 1546 | 			     "argument%s (%d given)", | 
 | 1547 | 			     (fname == NULL) ? "function" : fname, | 
 | 1548 | 			     (fname == NULL) ? "" : "()", | 
 | 1549 | 			     len, | 
 | 1550 | 			     (len == 1) ? "" : "s", | 
 | 1551 | 			     nargs + nkeywords); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1552 | 		return 0; | 
 | 1553 | 	} | 
| Tim Peters | c2f0112 | 2001-10-27 07:25:06 +0000 | [diff] [blame] | 1554 |  | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1555 | 	/* convert tuple args and keyword args in same loop, using kwlist to drive process */ | 
 | 1556 | 	for (i = 0; i < len; i++) { | 
 | 1557 | 		keyword = kwlist[i]; | 
 | 1558 | 		if (*format == '|') { | 
 | 1559 | 			min = i; | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1560 | 			format++; | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1561 | 		} | 
 | 1562 | 		if (IS_END_OF_FORMAT(*format)) { | 
 | 1563 | 			PyErr_Format(PyExc_RuntimeError, | 
 | 1564 | 				     "More keyword list entries (%d) than " | 
 | 1565 | 				     "format specifiers (%d)", len, i); | 
 | 1566 | 			return cleanreturn(0, freelist); | 
 | 1567 | 		} | 
 | 1568 | 		current_arg = NULL; | 
 | 1569 | 		if (nkeywords) { | 
 | 1570 | 			current_arg = PyDict_GetItemString(keywords, keyword); | 
 | 1571 | 		} | 
 | 1572 | 		if (current_arg) { | 
 | 1573 | 			--nkeywords; | 
 | 1574 | 			if (i < nargs) { | 
 | 1575 | 				/* arg present in tuple and in dict */ | 
 | 1576 | 				PyErr_Format(PyExc_TypeError, | 
 | 1577 | 					     "Argument given by name ('%s') " | 
 | 1578 | 					     "and position (%d)", | 
 | 1579 | 					     keyword, i+1); | 
 | 1580 | 				return cleanreturn(0, freelist); | 
 | 1581 | 			} | 
 | 1582 | 		} | 
 | 1583 | 		else if (nkeywords && PyErr_Occurred()) | 
 | 1584 | 			return cleanreturn(0, freelist); | 
 | 1585 | 		else if (i < nargs) | 
 | 1586 | 			current_arg = PyTuple_GET_ITEM(args, i); | 
 | 1587 | 			 | 
 | 1588 | 		if (current_arg) { | 
 | 1589 | 			msg = convertitem(current_arg, &format, p_va, flags, | 
 | 1590 | 				levels, msgbuf, sizeof(msgbuf), &freelist); | 
 | 1591 | 			if (msg) { | 
 | 1592 | 				seterror(i+1, msg, levels, fname, custom_msg); | 
 | 1593 | 				return cleanreturn(0, freelist); | 
 | 1594 | 			} | 
 | 1595 | 			continue; | 
 | 1596 | 		} | 
 | 1597 |  | 
 | 1598 | 		if (i < min) { | 
 | 1599 | 			PyErr_Format(PyExc_TypeError, "Required argument " | 
 | 1600 | 				     "'%s' (pos %d) not found", | 
 | 1601 | 				     keyword, i+1); | 
 | 1602 | 			return cleanreturn(0, freelist); | 
 | 1603 | 		} | 
 | 1604 | 		/* current code reports success when all required args | 
 | 1605 | 		 * fulfilled and no keyword args left, with no further | 
 | 1606 | 		 * validation. XXX Maybe skip this in debug build ? | 
 | 1607 | 		 */ | 
 | 1608 | 		if (!nkeywords) | 
 | 1609 | 			return cleanreturn(1, freelist); | 
 | 1610 |  | 
 | 1611 | 		/* We are into optional args, skip thru to any remaining | 
 | 1612 | 		 * keyword args */ | 
 | 1613 | 		msg = skipitem(&format, p_va, flags); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1614 | 		if (msg) { | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1615 | 			PyErr_Format(PyExc_RuntimeError, "%s: '%s'", msg, | 
 | 1616 | 				     format); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1617 | 			return cleanreturn(0, freelist); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1618 | 		} | 
 | 1619 | 	} | 
 | 1620 |  | 
| Benjamin Peterson | 4caef5c | 2008-12-22 20:51:15 +0000 | [diff] [blame] | 1621 | 	if (!IS_END_OF_FORMAT(*format) && *format != '|') { | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1622 | 		PyErr_Format(PyExc_RuntimeError, | 
 | 1623 | 			"more argument specifiers than keyword list entries " | 
 | 1624 | 			"(remaining format:'%s')", format); | 
 | 1625 | 		return cleanreturn(0, freelist); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1626 | 	} | 
| Tim Peters | b054be4 | 2001-10-27 05:07:41 +0000 | [diff] [blame] | 1627 |  | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1628 | 	/* make sure there are no extraneous keyword arguments */ | 
| Tim Peters | c2f0112 | 2001-10-27 07:25:06 +0000 | [diff] [blame] | 1629 | 	if (nkeywords > 0) { | 
 | 1630 | 		PyObject *key, *value; | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1631 | 		Py_ssize_t pos = 0; | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1632 | 		while (PyDict_Next(keywords, &pos, &key, &value)) { | 
| Tim Peters | c2f0112 | 2001-10-27 07:25:06 +0000 | [diff] [blame] | 1633 | 			int match = 0; | 
| Guido van Rossum | 5547476 | 2002-04-04 16:22:30 +0000 | [diff] [blame] | 1634 | 			char *ks; | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 1635 | 			if (!PyString_Check(key)) { | 
| Guido van Rossum | 5547476 | 2002-04-04 16:22:30 +0000 | [diff] [blame] | 1636 | 				PyErr_SetString(PyExc_TypeError,  | 
 | 1637 | 					        "keywords must be strings"); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1638 | 				return cleanreturn(0, freelist); | 
| Guido van Rossum | 5547476 | 2002-04-04 16:22:30 +0000 | [diff] [blame] | 1639 | 			} | 
| Gregory P. Smith | dd96db6 | 2008-06-09 04:58:54 +0000 | [diff] [blame] | 1640 | 			ks = PyString_AsString(key); | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1641 | 			for (i = 0; i < len; i++) { | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1642 | 				if (!strcmp(ks, kwlist[i])) { | 
 | 1643 | 					match = 1; | 
 | 1644 | 					break; | 
 | 1645 | 				} | 
 | 1646 | 			} | 
 | 1647 | 			if (!match) { | 
| Tim Peters | c2f0112 | 2001-10-27 07:25:06 +0000 | [diff] [blame] | 1648 | 				PyErr_Format(PyExc_TypeError, | 
 | 1649 | 					     "'%s' is an invalid keyword " | 
 | 1650 | 					     "argument for this function", | 
 | 1651 | 					     ks); | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1652 | 				return cleanreturn(0, freelist); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1653 | 			} | 
 | 1654 | 		} | 
 | 1655 | 	} | 
| Tim Peters | c2f0112 | 2001-10-27 07:25:06 +0000 | [diff] [blame] | 1656 |  | 
| Martin v. Löwis | e6bbb4d | 2003-05-03 10:00:22 +0000 | [diff] [blame] | 1657 | 	return cleanreturn(1, freelist); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1658 | } | 
 | 1659 |  | 
 | 1660 |  | 
 | 1661 | static char * | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1662 | skipitem(const char **p_format, va_list *p_va, int flags) | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1663 | { | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1664 | 	const char *format = *p_format; | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1665 | 	char c = *format++; | 
 | 1666 | 	 | 
 | 1667 | 	switch (c) { | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1668 |  | 
 | 1669 | 	/* simple codes | 
 | 1670 | 	 * The individual types (second arg of va_arg) are irrelevant */ | 
 | 1671 |  | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1672 | 	case 'b': /* byte -- very short int */ | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 1673 | 	case 'B': /* byte as bitfield */ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1674 | 	case 'h': /* short int */ | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 1675 | 	case 'H': /* short int as bitfield */ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1676 | 	case 'i': /* int */ | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1677 | 	case 'I': /* int sized bitfield */ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1678 | 	case 'l': /* long int */ | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1679 | 	case 'k': /* long int sized bitfield */ | 
| Guido van Rossum | 3dbba6e | 1999-01-25 21:48:56 +0000 | [diff] [blame] | 1680 | #ifdef HAVE_LONG_LONG | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1681 | 	case 'L': /* PY_LONG_LONG */ | 
 | 1682 | 	case 'K': /* PY_LONG_LONG sized bitfield */ | 
| Guido van Rossum | 1a8791e | 1998-08-04 22:46:29 +0000 | [diff] [blame] | 1683 | #endif | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1684 | 	case 'f': /* float */ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1685 | 	case 'd': /* double */ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1686 | #ifndef WITHOUT_COMPLEX | 
 | 1687 | 	case 'D': /* complex double */ | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1688 | #endif | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1689 | 	case 'c': /* char */ | 
 | 1690 | 		{ | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1691 | 			(void) va_arg(*p_va, void *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1692 | 			break; | 
 | 1693 | 		} | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1694 |  | 
 | 1695 | 	case 'n': /* Py_ssize_t */ | 
 | 1696 | 		{ | 
 | 1697 | 			(void) va_arg(*p_va, Py_ssize_t *); | 
 | 1698 | 			break; | 
 | 1699 | 		} | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1700 | 	 | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1701 | 	/* string codes */ | 
 | 1702 | 		 | 
 | 1703 | 	case 'e': /* string with encoding */ | 
 | 1704 | 		{ | 
 | 1705 | 			(void) va_arg(*p_va, const char *); | 
 | 1706 | 			if (!(*format == 's' || *format == 't')) | 
 | 1707 | 				/* after 'e', only 's' and 't' is allowed */ | 
 | 1708 | 				goto err; | 
 | 1709 | 			format++; | 
 | 1710 | 			/* explicit fallthrough to string cases */ | 
 | 1711 | 		} | 
 | 1712 | 	 | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1713 | 	case 's': /* string */ | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1714 | 	case 'z': /* string or None */ | 
 | 1715 | #ifdef Py_USING_UNICODE | 
 | 1716 | 	case 'u': /* unicode string */ | 
 | 1717 | #endif | 
 | 1718 | 	case 't': /* buffer, read-only */ | 
 | 1719 | 	case 'w': /* buffer, read-write */ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1720 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1721 | 			(void) va_arg(*p_va, char **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1722 | 			if (*format == '#') { | 
| Martin v. Löwis | 18e1655 | 2006-02-15 17:27:45 +0000 | [diff] [blame] | 1723 | 				if (flags & FLAG_SIZE_T) | 
 | 1724 | 					(void) va_arg(*p_va, Py_ssize_t *); | 
 | 1725 | 				else | 
 | 1726 | 					(void) va_arg(*p_va, int *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1727 | 				format++; | 
| Martin v. Löwis | f91d46a | 2008-08-12 14:49:50 +0000 | [diff] [blame] | 1728 | 			} else if ((c == 's' || c == 'z') && *format == '*') { | 
 | 1729 | 				format++; | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1730 | 			} | 
 | 1731 | 			break; | 
 | 1732 | 		} | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1733 |  | 
 | 1734 | 	/* object codes */ | 
 | 1735 |  | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1736 | 	case 'S': /* string object */ | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1737 | #ifdef Py_USING_UNICODE | 
 | 1738 | 	case 'U': /* unicode string object */ | 
 | 1739 | #endif | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1740 | 		{ | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 1741 | 			(void) va_arg(*p_va, PyObject **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1742 | 			break; | 
 | 1743 | 		} | 
 | 1744 | 	 | 
 | 1745 | 	case 'O': /* object */ | 
 | 1746 | 		{ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1747 | 			if (*format == '!') { | 
 | 1748 | 				format++; | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 1749 | 				(void) va_arg(*p_va, PyTypeObject*); | 
 | 1750 | 				(void) va_arg(*p_va, PyObject **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1751 | 			} | 
 | 1752 | #if 0 | 
 | 1753 | /* I don't know what this is for */ | 
 | 1754 | 			else if (*format == '?') { | 
 | 1755 | 				inquiry pred = va_arg(*p_va, inquiry); | 
 | 1756 | 				format++; | 
 | 1757 | 				if ((*pred)(arg)) { | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 1758 | 					(void) va_arg(*p_va, PyObject **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1759 | 				} | 
 | 1760 | 			} | 
 | 1761 | #endif | 
 | 1762 | 			else if (*format == '&') { | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 1763 | 				typedef int (*converter)(PyObject *, void *); | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1764 | 				(void) va_arg(*p_va, converter); | 
 | 1765 | 				(void) va_arg(*p_va, void *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1766 | 				format++; | 
 | 1767 | 			} | 
 | 1768 | 			else { | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 1769 | 				(void) va_arg(*p_va, PyObject **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1770 | 			} | 
 | 1771 | 			break; | 
 | 1772 | 		} | 
| Christian Heimes | ea83793 | 2008-02-26 17:23:51 +0000 | [diff] [blame] | 1773 |  | 
 | 1774 | 	case '(':	/* bypass tuple, not handled at all previously */ | 
 | 1775 | 		{ | 
 | 1776 | 			char *msg; | 
 | 1777 | 			for (;;) { | 
 | 1778 | 				if (*format==')') | 
 | 1779 | 					break; | 
 | 1780 | 				if (IS_END_OF_FORMAT(*format)) | 
 | 1781 | 					return "Unmatched left paren in format " | 
 | 1782 | 					       "string"; | 
 | 1783 | 				msg = skipitem(&format, p_va, flags); | 
 | 1784 | 				if (msg) | 
 | 1785 | 					return msg; | 
 | 1786 | 			} | 
 | 1787 | 			format++; | 
 | 1788 | 			break; | 
 | 1789 | 		} | 
 | 1790 |  | 
 | 1791 | 	case ')': | 
 | 1792 | 		return "Unmatched right paren in format string"; | 
 | 1793 |  | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1794 | 	default: | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1795 | err: | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1796 | 		return "impossible<bad format char>"; | 
 | 1797 | 	 | 
 | 1798 | 	} | 
| Georg Brandl | 6dd1461 | 2005-09-14 19:29:53 +0000 | [diff] [blame] | 1799 |  | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1800 | 	*p_format = format; | 
 | 1801 | 	return NULL; | 
 | 1802 | } | 
| Fred Drake | e4616e6 | 2001-10-23 21:09:29 +0000 | [diff] [blame] | 1803 |  | 
 | 1804 |  | 
 | 1805 | int | 
| Martin v. Löwis | 7624674 | 2006-03-01 04:06:10 +0000 | [diff] [blame] | 1806 | PyArg_UnpackTuple(PyObject *args, const char *name, Py_ssize_t min, Py_ssize_t max, ...) | 
| Fred Drake | e4616e6 | 2001-10-23 21:09:29 +0000 | [diff] [blame] | 1807 | { | 
| Martin v. Löwis | 7624674 | 2006-03-01 04:06:10 +0000 | [diff] [blame] | 1808 | 	Py_ssize_t i, l; | 
| Fred Drake | e4616e6 | 2001-10-23 21:09:29 +0000 | [diff] [blame] | 1809 | 	PyObject **o; | 
 | 1810 | 	va_list vargs; | 
 | 1811 |  | 
 | 1812 | #ifdef HAVE_STDARG_PROTOTYPES | 
 | 1813 | 	va_start(vargs, max); | 
 | 1814 | #else | 
 | 1815 | 	va_start(vargs); | 
 | 1816 | #endif | 
 | 1817 |  | 
 | 1818 | 	assert(min >= 0); | 
 | 1819 | 	assert(min <= max); | 
 | 1820 | 	if (!PyTuple_Check(args)) { | 
 | 1821 | 		PyErr_SetString(PyExc_SystemError, | 
 | 1822 | 		    "PyArg_UnpackTuple() argument list is not a tuple"); | 
 | 1823 | 		return 0; | 
 | 1824 | 	}	 | 
 | 1825 | 	l = PyTuple_GET_SIZE(args); | 
 | 1826 | 	if (l < min) { | 
 | 1827 | 		if (name != NULL) | 
 | 1828 | 			PyErr_Format( | 
 | 1829 | 			    PyExc_TypeError, | 
| Thomas Wouters | 572a9f3 | 2006-03-01 05:38:39 +0000 | [diff] [blame] | 1830 | 			    "%s expected %s%zd arguments, got %zd",  | 
| Fred Drake | e4616e6 | 2001-10-23 21:09:29 +0000 | [diff] [blame] | 1831 | 			    name, (min == max ? "" : "at least "), min, l); | 
 | 1832 | 		else | 
 | 1833 | 			PyErr_Format( | 
 | 1834 | 			    PyExc_TypeError, | 
| Thomas Wouters | 572a9f3 | 2006-03-01 05:38:39 +0000 | [diff] [blame] | 1835 | 			    "unpacked tuple should have %s%zd elements," | 
 | 1836 | 			    " but has %zd",  | 
| Fred Drake | e4616e6 | 2001-10-23 21:09:29 +0000 | [diff] [blame] | 1837 | 			    (min == max ? "" : "at least "), min, l); | 
 | 1838 | 		va_end(vargs); | 
 | 1839 | 		return 0; | 
 | 1840 | 	} | 
 | 1841 | 	if (l > max) { | 
 | 1842 | 		if (name != NULL) | 
 | 1843 | 			PyErr_Format( | 
 | 1844 | 			    PyExc_TypeError, | 
| Thomas Wouters | 572a9f3 | 2006-03-01 05:38:39 +0000 | [diff] [blame] | 1845 | 			    "%s expected %s%zd arguments, got %zd",  | 
| Fred Drake | e4616e6 | 2001-10-23 21:09:29 +0000 | [diff] [blame] | 1846 | 			    name, (min == max ? "" : "at most "), max, l); | 
 | 1847 | 		else | 
 | 1848 | 			PyErr_Format( | 
 | 1849 | 			    PyExc_TypeError, | 
| Thomas Wouters | 572a9f3 | 2006-03-01 05:38:39 +0000 | [diff] [blame] | 1850 | 			    "unpacked tuple should have %s%zd elements," | 
 | 1851 | 			    " but has %zd",  | 
| Fred Drake | e4616e6 | 2001-10-23 21:09:29 +0000 | [diff] [blame] | 1852 | 			    (min == max ? "" : "at most "), max, l); | 
 | 1853 | 		va_end(vargs); | 
 | 1854 | 		return 0; | 
 | 1855 | 	} | 
 | 1856 | 	for (i = 0; i < l; i++) { | 
 | 1857 | 		o = va_arg(vargs, PyObject **); | 
 | 1858 | 		*o = PyTuple_GET_ITEM(args, i); | 
 | 1859 | 	} | 
 | 1860 | 	va_end(vargs); | 
 | 1861 | 	return 1; | 
 | 1862 | } | 
| Georg Brandl | 02c4287 | 2005-08-26 06:42:30 +0000 | [diff] [blame] | 1863 |  | 
 | 1864 |  | 
 | 1865 | /* For type constructors that don't take keyword args | 
 | 1866 |  * | 
 | 1867 |  * Sets a TypeError and returns 0 if the kwds dict is  | 
| Walter Dörwald | d14bf61 | 2006-09-21 15:09:55 +0000 | [diff] [blame] | 1868 |  * not empty, returns 1 otherwise | 
| Georg Brandl | 02c4287 | 2005-08-26 06:42:30 +0000 | [diff] [blame] | 1869 |  */ | 
 | 1870 | int | 
| Jeremy Hylton | af68c87 | 2005-12-10 18:50:16 +0000 | [diff] [blame] | 1871 | _PyArg_NoKeywords(const char *funcname, PyObject *kw) | 
| Georg Brandl | 02c4287 | 2005-08-26 06:42:30 +0000 | [diff] [blame] | 1872 | { | 
 | 1873 | 	if (kw == NULL) | 
 | 1874 | 		return 1; | 
 | 1875 | 	if (!PyDict_CheckExact(kw)) { | 
 | 1876 | 		PyErr_BadInternalCall(); | 
 | 1877 | 		return 0; | 
 | 1878 | 	} | 
 | 1879 | 	if (PyDict_Size(kw) == 0) | 
 | 1880 | 		return 1; | 
 | 1881 | 	 | 
 | 1882 | 	PyErr_Format(PyExc_TypeError, "%s does not take keyword arguments",  | 
 | 1883 | 			funcname); | 
 | 1884 | 	return 0; | 
 | 1885 | } | 
| Anthony Baxter | 9730038 | 2006-04-12 04:38:54 +0000 | [diff] [blame] | 1886 | #ifdef __cplusplus | 
 | 1887 | }; | 
 | 1888 | #endif |