| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 1 |  | 
 | 2 | /* New getargs implementation */ | 
 | 3 |  | 
 | 4 | /* XXX There are several unchecked sprintf or strcat calls in this file. | 
 | 5 |    XXX The only way these can become a danger is if some C code in the | 
 | 6 |    XXX Python source (or in an extension) uses ridiculously long names | 
| Thomas Wouters | 7e47402 | 2000-07-16 12:04:32 +0000 | [diff] [blame] | 7 |    XXX or ridiculously deep nesting in format strings. */ | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 8 |  | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 9 | #include "Python.h" | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 10 |  | 
| Guido van Rossum | c1d5053 | 1996-08-21 23:38:24 +0000 | [diff] [blame] | 11 | #include <ctype.h> | 
| Guido van Rossum | db67739 | 2000-07-01 01:09:43 +0000 | [diff] [blame] | 12 | #ifdef HAVE_LIMITS_H | 
 | 13 | #include <limits.h> | 
 | 14 | #endif | 
| Guido van Rossum | c1d5053 | 1996-08-21 23:38:24 +0000 | [diff] [blame] | 15 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 16 |  | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 17 | int PyArg_Parse(PyObject *, char *, ...); | 
 | 18 | int PyArg_ParseTuple(PyObject *, char *, ...); | 
 | 19 | int PyArg_VaParse(PyObject *, char *, va_list); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 20 |  | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 21 | int PyArg_ParseTupleAndKeywords(PyObject *, PyObject *, | 
 | 22 | 				char *, char **, ...); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 23 |  | 
 | 24 | /* Forward */ | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 25 | static int vgetargs1(PyObject *, char *, va_list *, int); | 
 | 26 | static void seterror(int, char *, int *, char *, char *); | 
 | 27 | static char *convertitem(PyObject *, char **, va_list *, int *, char *); | 
 | 28 | static char *converttuple(PyObject *, char **, va_list *, | 
 | 29 | 			  int *, char *, int); | 
 | 30 | static char *convertsimple(PyObject *, char **, va_list *, char *); | 
 | 31 | static char *convertsimple1(PyObject *, char **, va_list *); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 32 |  | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 33 | static int vgetargskeywords(PyObject *, PyObject *, | 
 | 34 | 			    char *, char **, va_list *); | 
 | 35 | static char *skipitem(char **, va_list *); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 36 |  | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 37 | int PyArg_Parse(PyObject *args, char *format, ...) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 38 | { | 
 | 39 | 	int retval; | 
 | 40 | 	va_list va; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 41 | 	 | 
 | 42 | 	va_start(va, format); | 
| Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 43 | 	retval = vgetargs1(args, format, &va, 1); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 44 | 	va_end(va); | 
 | 45 | 	return retval; | 
 | 46 | } | 
 | 47 |  | 
 | 48 |  | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 49 | int PyArg_ParseTuple(PyObject *args, char *format, ...) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 50 | { | 
 | 51 | 	int retval; | 
 | 52 | 	va_list va; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 53 | 	 | 
 | 54 | 	va_start(va, format); | 
| Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 55 | 	retval = vgetargs1(args, format, &va, 0); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 56 | 	va_end(va); | 
 | 57 | 	return retval; | 
 | 58 | } | 
 | 59 |  | 
 | 60 |  | 
 | 61 | int | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 62 | PyArg_VaParse(PyObject *args, char *format, va_list va) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 63 | { | 
| Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 64 | 	va_list lva; | 
 | 65 |  | 
 | 66 | #ifdef VA_LIST_IS_ARRAY | 
 | 67 | 	memcpy(lva, va, sizeof(va_list)); | 
 | 68 | #else | 
 | 69 | 	lva = va; | 
 | 70 | #endif | 
 | 71 |  | 
 | 72 | 	return vgetargs1(args, format, &lva, 0); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 73 | } | 
 | 74 |  | 
 | 75 |  | 
 | 76 | static int | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 77 | vgetargs1(PyObject *args, char *format, va_list *p_va, int compat) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 78 | { | 
 | 79 | 	char msgbuf[256]; | 
 | 80 | 	int levels[32]; | 
 | 81 | 	char *fname = NULL; | 
 | 82 | 	char *message = NULL; | 
 | 83 | 	int min = -1; | 
 | 84 | 	int max = 0; | 
 | 85 | 	int level = 0; | 
 | 86 | 	char *formatsave = format; | 
 | 87 | 	int i, len; | 
 | 88 | 	char *msg; | 
 | 89 | 	 | 
 | 90 | 	for (;;) { | 
 | 91 | 		int c = *format++; | 
 | 92 | 		if (c == '(' /* ')' */) { | 
 | 93 | 			if (level == 0) | 
 | 94 | 				max++; | 
 | 95 | 			level++; | 
 | 96 | 		} | 
 | 97 | 		else if (/* '(' */ c == ')') { | 
 | 98 | 			if (level == 0) | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 99 | 				Py_FatalError(/* '(' */ | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 100 | 				      "excess ')' in getargs format"); | 
 | 101 | 			else | 
 | 102 | 				level--; | 
 | 103 | 		} | 
 | 104 | 		else if (c == '\0') | 
 | 105 | 			break; | 
 | 106 | 		else if (c == ':') { | 
 | 107 | 			fname = format; | 
 | 108 | 			break; | 
 | 109 | 		} | 
 | 110 | 		else if (c == ';') { | 
 | 111 | 			message = format; | 
 | 112 | 			break; | 
 | 113 | 		} | 
 | 114 | 		else if (level != 0) | 
 | 115 | 			; /* Pass */ | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 116 | 		else if (c == 'e') | 
 | 117 | 			; /* Pass */ | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 118 | 		else if (isalpha(c)) | 
 | 119 | 			max++; | 
 | 120 | 		else if (c == '|') | 
 | 121 | 			min = max; | 
 | 122 | 	} | 
 | 123 | 	 | 
 | 124 | 	if (level != 0) | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 125 | 		Py_FatalError(/* '(' */ "missing ')' in getargs format"); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 126 | 	 | 
 | 127 | 	if (min < 0) | 
 | 128 | 		min = max; | 
 | 129 | 	 | 
 | 130 | 	format = formatsave; | 
 | 131 | 	 | 
 | 132 | 	if (compat) { | 
 | 133 | 		if (max == 0) { | 
 | 134 | 			if (args == NULL) | 
 | 135 | 				return 1; | 
 | 136 | 			sprintf(msgbuf, "%s requires no arguments", | 
 | 137 | 				fname==NULL ? "function" : fname); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 138 | 			PyErr_SetString(PyExc_TypeError, msgbuf); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 139 | 			return 0; | 
 | 140 | 		} | 
 | 141 | 		else if (min == 1 && max == 1) { | 
| Guido van Rossum | 13d0ed1 | 1994-11-10 22:35:48 +0000 | [diff] [blame] | 142 | 			if (args == NULL) { | 
 | 143 | 				sprintf(msgbuf, | 
 | 144 | 					"%s requires at least one argument", | 
 | 145 | 					fname==NULL ? "function" : fname); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 146 | 				PyErr_SetString(PyExc_TypeError, msgbuf); | 
| Guido van Rossum | 13d0ed1 | 1994-11-10 22:35:48 +0000 | [diff] [blame] | 147 | 				return 0; | 
 | 148 | 			} | 
| Guido van Rossum | 1ae940a | 1995-01-02 19:04:15 +0000 | [diff] [blame] | 149 | 			msg = convertitem(args, &format, p_va, levels, msgbuf); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 150 | 			if (msg == NULL) | 
 | 151 | 				return 1; | 
 | 152 | 			seterror(levels[0], msg, levels+1, fname, message); | 
 | 153 | 			return 0; | 
 | 154 | 		} | 
 | 155 | 		else { | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 156 | 			PyErr_SetString(PyExc_SystemError, | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 157 | 			    "old style getargs format uses new features"); | 
 | 158 | 			return 0; | 
 | 159 | 		} | 
 | 160 | 	} | 
 | 161 | 	 | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 162 | 	if (!PyTuple_Check(args)) { | 
 | 163 | 		PyErr_SetString(PyExc_SystemError, | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 164 | 		    "new style getargs format but argument is not a tuple"); | 
 | 165 | 		return 0; | 
 | 166 | 	} | 
 | 167 | 	 | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 168 | 	len = PyTuple_Size(args); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 169 | 	 | 
 | 170 | 	if (len < min || max < len) { | 
 | 171 | 		if (message == NULL) { | 
 | 172 | 			sprintf(msgbuf, | 
 | 173 | 				"%s requires %s %d argument%s; %d given", | 
 | 174 | 				fname==NULL ? "function" : fname, | 
 | 175 | 				min==max ? "exactly" | 
 | 176 | 				         : len < min ? "at least" : "at most", | 
 | 177 | 				len < min ? min : max, | 
 | 178 | 				(len < min ? min : max) == 1 ? "" : "s", | 
 | 179 | 				len); | 
 | 180 | 			message = msgbuf; | 
 | 181 | 		} | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 182 | 		PyErr_SetString(PyExc_TypeError, message); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 183 | 		return 0; | 
 | 184 | 	} | 
 | 185 | 	 | 
 | 186 | 	for (i = 0; i < len; i++) { | 
 | 187 | 		if (*format == '|') | 
 | 188 | 			format++; | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 189 | 		msg = convertitem(PyTuple_GetItem(args, i), &format, p_va, | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 190 | 				 levels, msgbuf); | 
 | 191 | 		if (msg) { | 
 | 192 | 			seterror(i+1, msg, levels, fname, message); | 
 | 193 | 			return 0; | 
 | 194 | 		} | 
 | 195 | 	} | 
| Guido van Rossum | 231a41e | 1997-12-09 20:36:39 +0000 | [diff] [blame] | 196 |  | 
| Guido van Rossum | 730806d | 1998-04-10 22:27:42 +0000 | [diff] [blame] | 197 | 	if (*format != '\0' && !isalpha((int)(*format)) && | 
| Guido van Rossum | 7d4f68c | 1997-12-19 04:25:23 +0000 | [diff] [blame] | 198 | 	    *format != '(' && | 
| Guido van Rossum | 231a41e | 1997-12-09 20:36:39 +0000 | [diff] [blame] | 199 | 	    *format != '|' && *format != ':' && *format != ';') { | 
 | 200 | 		PyErr_Format(PyExc_SystemError, | 
| Guido van Rossum | 0d6b49e | 1998-01-19 22:22:44 +0000 | [diff] [blame] | 201 | 			     "bad format string: %.200s", formatsave); | 
| Guido van Rossum | 231a41e | 1997-12-09 20:36:39 +0000 | [diff] [blame] | 202 | 		return 0; | 
 | 203 | 	} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 204 | 	 | 
 | 205 | 	return 1; | 
 | 206 | } | 
 | 207 |  | 
 | 208 |  | 
 | 209 |  | 
 | 210 | static void | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 211 | seterror(int iarg, char *msg, int *levels, char *fname, char *message) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 212 | { | 
 | 213 | 	char buf[256]; | 
 | 214 | 	int i; | 
 | 215 | 	char *p = buf; | 
 | 216 |  | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 217 | 	if (PyErr_Occurred()) | 
| Guido van Rossum | 64fc649 | 1995-01-21 14:09:37 +0000 | [diff] [blame] | 218 | 		return; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 219 | 	if (iarg == 0 && message == NULL) | 
 | 220 | 		message = msg; | 
 | 221 | 	else if (message == NULL) { | 
 | 222 | 		if (fname != NULL) { | 
 | 223 | 			sprintf(p, "%s, ", fname); | 
 | 224 | 			p += strlen(p); | 
 | 225 | 		} | 
 | 226 | 		sprintf(p, "argument %d", iarg); | 
 | 227 | 		i = 0; | 
 | 228 | 		p += strlen(p); | 
 | 229 | 		while (levels[i] > 0) { | 
 | 230 | 			sprintf(p, ", item %d", levels[i]-1); | 
 | 231 | 			p += strlen(p); | 
 | 232 | 			i++; | 
 | 233 | 		} | 
 | 234 | 		sprintf(p, ": expected %s found", msg); | 
 | 235 | 		message = buf; | 
 | 236 | 	} | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 237 | 	PyErr_SetString(PyExc_TypeError, message); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 238 | } | 
 | 239 |  | 
 | 240 |  | 
 | 241 | /* Convert a tuple argument. | 
 | 242 |    On entry, *p_format points to the character _after_ the opening '('. | 
 | 243 |    On successful exit, *p_format points to the closing ')'. | 
 | 244 |    If successful: | 
 | 245 |       *p_format and *p_va are updated, | 
 | 246 |       *levels and *msgbuf are untouched, | 
 | 247 |       and NULL is returned. | 
 | 248 |    If the argument is invalid: | 
 | 249 |       *p_format is unchanged, | 
 | 250 |       *p_va is undefined, | 
 | 251 |       *levels is a 0-terminated list of item numbers, | 
 | 252 |       *msgbuf contains an error message, whose format is: | 
 | 253 |          "<typename1>, <typename2>", where: | 
 | 254 |             <typename1> is the name of the expected type, and | 
 | 255 |             <typename2> is the name of the actual type, | 
 | 256 |          (so you can surround it by "expected ... found"), | 
 | 257 |       and msgbuf is returned. | 
 | 258 | */ | 
 | 259 |  | 
 | 260 | static char * | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 261 | converttuple(PyObject *arg, char **p_format, va_list *p_va, int *levels, | 
 | 262 | 	     char *msgbuf, int toplevel) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 263 | { | 
 | 264 | 	int level = 0; | 
 | 265 | 	int n = 0; | 
 | 266 | 	char *format = *p_format; | 
 | 267 | 	int i; | 
 | 268 | 	 | 
 | 269 | 	for (;;) { | 
 | 270 | 		int c = *format++; | 
 | 271 | 		if (c == '(') { | 
 | 272 | 			if (level == 0) | 
 | 273 | 				n++; | 
 | 274 | 			level++; | 
 | 275 | 		} | 
 | 276 | 		else if (c == ')') { | 
 | 277 | 			if (level == 0) | 
 | 278 | 				break; | 
 | 279 | 			level--; | 
 | 280 | 		} | 
 | 281 | 		else if (c == ':' || c == ';' || c == '\0') | 
 | 282 | 			break; | 
 | 283 | 		else if (level == 0 && isalpha(c)) | 
 | 284 | 			n++; | 
 | 285 | 	} | 
 | 286 | 	 | 
| Guido van Rossum | 66368cc | 1999-02-17 23:16:43 +0000 | [diff] [blame] | 287 | 	if (!PySequence_Check(arg)) { | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 288 | 		levels[0] = 0; | 
 | 289 | 		sprintf(msgbuf, | 
| Guido van Rossum | 66368cc | 1999-02-17 23:16:43 +0000 | [diff] [blame] | 290 | 			toplevel ? "%d arguments, %s" : "%d-sequence, %s", | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 291 | 			n, arg == Py_None ? "None" : arg->ob_type->tp_name); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 292 | 		return msgbuf; | 
 | 293 | 	} | 
 | 294 | 	 | 
| Jeremy Hylton | 03657cf | 2000-07-12 13:05:33 +0000 | [diff] [blame] | 295 | 	if ((i = PySequence_Size(arg)) != n) { | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 296 | 		levels[0] = 0; | 
 | 297 | 		sprintf(msgbuf, | 
| Guido van Rossum | 66368cc | 1999-02-17 23:16:43 +0000 | [diff] [blame] | 298 | 		    toplevel ? "%d arguments, %d" : "%d-sequence, %d-sequence", | 
 | 299 | 		    n, i); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 300 | 		return msgbuf; | 
 | 301 | 	} | 
 | 302 | 	 | 
 | 303 | 	format = *p_format; | 
 | 304 | 	for (i = 0; i < n; i++) { | 
 | 305 | 		char *msg; | 
| Guido van Rossum | 66368cc | 1999-02-17 23:16:43 +0000 | [diff] [blame] | 306 | 		PyObject *item; | 
 | 307 | 		item = PySequence_GetItem(arg, i); | 
 | 308 | 		msg = convertitem(item, &format, p_va, levels+1, msgbuf); | 
 | 309 | 		/* PySequence_GetItem calls tp->sq_item, which INCREFs */ | 
 | 310 | 		Py_XDECREF(item); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 311 | 		if (msg != NULL) { | 
 | 312 | 			levels[0] = i+1; | 
 | 313 | 			return msg; | 
 | 314 | 		} | 
 | 315 | 	} | 
 | 316 | 	 | 
 | 317 | 	*p_format = format; | 
 | 318 | 	return NULL; | 
 | 319 | } | 
 | 320 |  | 
 | 321 |  | 
 | 322 | /* Convert a single item. */ | 
 | 323 |  | 
 | 324 | static char * | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 325 | convertitem(PyObject *arg, char **p_format, va_list *p_va, int *levels, | 
 | 326 | 	    char *msgbuf) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 327 | { | 
 | 328 | 	char *msg; | 
 | 329 | 	char *format = *p_format; | 
 | 330 | 	 | 
 | 331 | 	if (*format == '(' /* ')' */) { | 
 | 332 | 		format++; | 
 | 333 | 		msg = converttuple(arg, &format, p_va, levels, msgbuf, 0); | 
 | 334 | 		if (msg == NULL) | 
 | 335 | 			format++; | 
 | 336 | 	} | 
 | 337 | 	else { | 
 | 338 | 		msg = convertsimple(arg, &format, p_va, msgbuf); | 
 | 339 | 		if (msg != NULL) | 
 | 340 | 			levels[0] = 0; | 
 | 341 | 	} | 
 | 342 | 	if (msg == NULL) | 
 | 343 | 		*p_format = format; | 
 | 344 | 	return msg; | 
 | 345 | } | 
 | 346 |  | 
 | 347 |  | 
 | 348 | /* Convert a non-tuple argument.  Adds to convertsimple1 functionality | 
 | 349 |    by appending ", <actual argument type>" to error message. */ | 
 | 350 |  | 
 | 351 | static char * | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 352 | convertsimple(PyObject *arg, char **p_format, va_list *p_va, char *msgbuf) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 353 | { | 
 | 354 | 	char *msg = convertsimple1(arg, p_format, p_va); | 
 | 355 | 	if (msg != NULL) { | 
 | 356 | 		sprintf(msgbuf, "%.50s, %.50s", msg, | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 357 | 			arg == Py_None ? "None" : arg->ob_type->tp_name); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 358 | 		msg = msgbuf; | 
 | 359 | 	} | 
 | 360 | 	return msg; | 
 | 361 | } | 
 | 362 |  | 
 | 363 |  | 
| Guido van Rossum | 700c6ff | 2000-04-27 20:13:18 +0000 | [diff] [blame] | 364 | /* Internal API needed by convertsimple1(): */ | 
 | 365 | extern  | 
| Marc-André Lemburg | bff879c | 2000-08-03 18:46:08 +0000 | [diff] [blame] | 366 | PyObject *_PyUnicode_AsDefaultEncodedString(PyObject *unicode, | 
| Guido van Rossum | 700c6ff | 2000-04-27 20:13:18 +0000 | [diff] [blame] | 367 | 				  const char *errors); | 
 | 368 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 369 | /* Convert a non-tuple argument.  Return NULL if conversion went OK, | 
 | 370 |    or a string representing the expected type if the conversion failed. | 
 | 371 |    When failing, an exception may or may not have been raised. | 
 | 372 |    Don't call if a tuple is expected. */ | 
 | 373 |  | 
 | 374 | static char * | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 375 | convertsimple1(PyObject *arg, char **p_format, va_list *p_va) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 376 | { | 
 | 377 | 	char *format = *p_format; | 
 | 378 | 	char c = *format++; | 
 | 379 | 	 | 
 | 380 | 	switch (c) { | 
 | 381 | 	 | 
| Fred Drake | 230cae7 | 2000-05-09 21:50:00 +0000 | [diff] [blame] | 382 | 	case 'b': /* unsigned byte -- very short int */ | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 383 | 		{ | 
 | 384 | 			char *p = va_arg(*p_va, char *); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 385 | 			long ival = PyInt_AsLong(arg); | 
 | 386 | 			if (ival == -1 && PyErr_Occurred()) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 387 | 				return "integer<b>"; | 
| Fred Drake | 230cae7 | 2000-05-09 21:50:00 +0000 | [diff] [blame] | 388 | 			else if (ival < 0) { | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 389 | 				PyErr_SetString(PyExc_OverflowError, | 
| Guido van Rossum | 5e08cb8 | 2000-06-28 23:53:56 +0000 | [diff] [blame] | 390 | 			      "unsigned byte integer is less than minimum"); | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 391 | 				return "integer<b>"; | 
 | 392 | 			} | 
| Fred Drake | 230cae7 | 2000-05-09 21:50:00 +0000 | [diff] [blame] | 393 | 			else if (ival > UCHAR_MAX) { | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 394 | 				PyErr_SetString(PyExc_OverflowError, | 
| Guido van Rossum | 5e08cb8 | 2000-06-28 23:53:56 +0000 | [diff] [blame] | 395 | 			      "unsigned byte integer is greater than maximum"); | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 396 | 				return "integer<b>"; | 
 | 397 | 			} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 398 | 			else | 
| Fred Drake | 230cae7 | 2000-05-09 21:50:00 +0000 | [diff] [blame] | 399 | 				*p = (unsigned char) ival; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 400 | 			break; | 
 | 401 | 		} | 
 | 402 | 	 | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 403 | 	case 'B': /* byte sized bitfield - both signed and unsigned values allowed */ | 
 | 404 | 		{ | 
 | 405 | 			char *p = va_arg(*p_va, char *); | 
 | 406 | 			long ival = PyInt_AsLong(arg); | 
 | 407 | 			if (ival == -1 && PyErr_Occurred()) | 
 | 408 | 				return "integer<b>"; | 
 | 409 | 			else if (ival < SCHAR_MIN) { | 
 | 410 | 				PyErr_SetString(PyExc_OverflowError, | 
 | 411 | 			      "byte-sized integer bitfield is less than minimum"); | 
 | 412 | 				return "integer<B>"; | 
 | 413 | 			} | 
| Jack Jansen | a454ebd | 2000-09-15 12:52:19 +0000 | [diff] [blame] | 414 | 			else if (ival > (int)UCHAR_MAX) { | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 415 | 				PyErr_SetString(PyExc_OverflowError, | 
 | 416 | 			      "byte-sized integer bitfield is greater than maximum"); | 
 | 417 | 				return "integer<B>"; | 
 | 418 | 			} | 
 | 419 | 			else | 
 | 420 | 				*p = (unsigned char) ival; | 
 | 421 | 			break; | 
 | 422 | 		} | 
 | 423 | 	 | 
| Fred Drake | 230cae7 | 2000-05-09 21:50:00 +0000 | [diff] [blame] | 424 | 	case 'h': /* signed short int */ | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 425 | 		{ | 
 | 426 | 			short *p = va_arg(*p_va, short *); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 427 | 			long ival = PyInt_AsLong(arg); | 
 | 428 | 			if (ival == -1 && PyErr_Occurred()) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 429 | 				return "integer<h>"; | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 430 | 			else if (ival < SHRT_MIN) { | 
 | 431 | 				PyErr_SetString(PyExc_OverflowError, | 
| Guido van Rossum | 5e08cb8 | 2000-06-28 23:53:56 +0000 | [diff] [blame] | 432 | 			      "signed short integer is less than minimum"); | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 433 | 				return "integer<h>"; | 
 | 434 | 			} | 
 | 435 | 			else if (ival > SHRT_MAX) { | 
 | 436 | 				PyErr_SetString(PyExc_OverflowError, | 
| Guido van Rossum | 5e08cb8 | 2000-06-28 23:53:56 +0000 | [diff] [blame] | 437 | 			      "signed short integer is greater than maximum"); | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 438 | 				return "integer<h>"; | 
 | 439 | 			} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 440 | 			else | 
| Guido van Rossum | 6bf62da | 1997-04-11 20:37:35 +0000 | [diff] [blame] | 441 | 				*p = (short) ival; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 442 | 			break; | 
 | 443 | 		} | 
 | 444 | 	 | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 445 | 	case 'H': /* short int sized bitfield, both signed and unsigned allowed */ | 
| Jack Jansen | d50338f | 2000-07-06 12:22:00 +0000 | [diff] [blame] | 446 | 		{ | 
 | 447 | 			unsigned short *p = va_arg(*p_va, unsigned short *); | 
 | 448 | 			long ival = PyInt_AsLong(arg); | 
 | 449 | 			if (ival == -1 && PyErr_Occurred()) | 
 | 450 | 				return "integer<H>"; | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 451 | 			else if (ival < SHRT_MIN) { | 
| Jack Jansen | d50338f | 2000-07-06 12:22:00 +0000 | [diff] [blame] | 452 | 				PyErr_SetString(PyExc_OverflowError, | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 453 | 			      "short integer bitfield is less than minimum"); | 
| Jack Jansen | d50338f | 2000-07-06 12:22:00 +0000 | [diff] [blame] | 454 | 				return "integer<H>"; | 
 | 455 | 			} | 
 | 456 | 			else if (ival > USHRT_MAX) { | 
 | 457 | 				PyErr_SetString(PyExc_OverflowError, | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 458 | 			      "short integer bitfield is greater than maximum"); | 
| Jack Jansen | d50338f | 2000-07-06 12:22:00 +0000 | [diff] [blame] | 459 | 				return "integer<H>"; | 
 | 460 | 			} | 
 | 461 | 			else | 
 | 462 | 				*p = (unsigned short) ival; | 
 | 463 | 			break; | 
 | 464 | 		} | 
 | 465 | 	 | 
| Fred Drake | 230cae7 | 2000-05-09 21:50:00 +0000 | [diff] [blame] | 466 | 	case 'i': /* signed int */ | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 467 | 		{ | 
 | 468 | 			int *p = va_arg(*p_va, int *); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 469 | 			long ival = PyInt_AsLong(arg); | 
 | 470 | 			if (ival == -1 && PyErr_Occurred()) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 471 | 				return "integer<i>"; | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 472 | 			else if (ival > INT_MAX) { | 
 | 473 | 				PyErr_SetString(PyExc_OverflowError, | 
| Guido van Rossum | 5e08cb8 | 2000-06-28 23:53:56 +0000 | [diff] [blame] | 474 | 				    "signed integer is greater than maximum"); | 
 | 475 | 				return "integer<i>"; | 
 | 476 | 			} | 
 | 477 | 			else if (ival < INT_MIN) { | 
 | 478 | 				PyErr_SetString(PyExc_OverflowError, | 
 | 479 | 				    "signed integer is less than minimum"); | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 480 | 				return "integer<i>"; | 
 | 481 | 			} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 482 | 			else | 
 | 483 | 				*p = ival; | 
 | 484 | 			break; | 
 | 485 | 		} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 486 | 	case 'l': /* long int */ | 
 | 487 | 		{ | 
 | 488 | 			long *p = va_arg(*p_va, long *); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 489 | 			long ival = PyInt_AsLong(arg); | 
 | 490 | 			if (ival == -1 && PyErr_Occurred()) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 491 | 				return "integer<l>"; | 
 | 492 | 			else | 
 | 493 | 				*p = ival; | 
 | 494 | 			break; | 
 | 495 | 		} | 
 | 496 | 	 | 
| Guido van Rossum | 3dbba6e | 1999-01-25 21:48:56 +0000 | [diff] [blame] | 497 | #ifdef HAVE_LONG_LONG | 
| Guido van Rossum | 3293b07 | 1998-08-25 16:07:15 +0000 | [diff] [blame] | 498 | 	case 'L': /* LONG_LONG */ | 
| Guido van Rossum | 1a8791e | 1998-08-04 22:46:29 +0000 | [diff] [blame] | 499 | 		{ | 
| Guido van Rossum | 3293b07 | 1998-08-25 16:07:15 +0000 | [diff] [blame] | 500 | 			LONG_LONG *p = va_arg( *p_va, LONG_LONG * ); | 
 | 501 | 			LONG_LONG ival = PyLong_AsLongLong( arg ); | 
 | 502 | 			if( ival == (LONG_LONG)-1 && PyErr_Occurred() ) { | 
| Guido van Rossum | 1a8791e | 1998-08-04 22:46:29 +0000 | [diff] [blame] | 503 | 				return "long<L>"; | 
 | 504 | 			} else { | 
 | 505 | 				*p = ival; | 
 | 506 | 			} | 
 | 507 | 			break; | 
 | 508 | 		} | 
 | 509 | #endif | 
 | 510 | 	 | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 511 | 	case 'f': /* float */ | 
 | 512 | 		{ | 
 | 513 | 			float *p = va_arg(*p_va, float *); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 514 | 			double dval = PyFloat_AsDouble(arg); | 
 | 515 | 			if (PyErr_Occurred()) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 516 | 				return "float<f>"; | 
 | 517 | 			else | 
| Guido van Rossum | 6bf62da | 1997-04-11 20:37:35 +0000 | [diff] [blame] | 518 | 				*p = (float) dval; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 519 | 			break; | 
 | 520 | 		} | 
 | 521 | 	 | 
 | 522 | 	case 'd': /* double */ | 
 | 523 | 		{ | 
 | 524 | 			double *p = va_arg(*p_va, double *); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 525 | 			double dval = PyFloat_AsDouble(arg); | 
 | 526 | 			if (PyErr_Occurred()) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 527 | 				return "float<d>"; | 
 | 528 | 			else | 
 | 529 | 				*p = dval; | 
 | 530 | 			break; | 
 | 531 | 		} | 
 | 532 | 	 | 
| Guido van Rossum | 530956d | 1996-07-21 02:27:43 +0000 | [diff] [blame] | 533 | #ifndef WITHOUT_COMPLEX | 
| Guido van Rossum | 8a5c5d2 | 1996-01-12 01:09:56 +0000 | [diff] [blame] | 534 | 	case 'D': /* complex double */ | 
 | 535 | 		{ | 
| Guido van Rossum | 530956d | 1996-07-21 02:27:43 +0000 | [diff] [blame] | 536 | 			Py_complex *p = va_arg(*p_va, Py_complex *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 537 | 			Py_complex cval; | 
 | 538 | 			cval = PyComplex_AsCComplex(arg); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 539 | 			if (PyErr_Occurred()) | 
| Guido van Rossum | 8a5c5d2 | 1996-01-12 01:09:56 +0000 | [diff] [blame] | 540 | 				return "complex<D>"; | 
 | 541 | 			else | 
 | 542 | 				*p = cval; | 
 | 543 | 			break; | 
 | 544 | 		} | 
| Guido van Rossum | 530956d | 1996-07-21 02:27:43 +0000 | [diff] [blame] | 545 | #endif /* WITHOUT_COMPLEX */ | 
| Guido van Rossum | 8a5c5d2 | 1996-01-12 01:09:56 +0000 | [diff] [blame] | 546 | 	 | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 547 | 	case 'c': /* char */ | 
 | 548 | 		{ | 
 | 549 | 			char *p = va_arg(*p_va, char *); | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 550 | 			if (PyString_Check(arg) && PyString_Size(arg) == 1) | 
 | 551 | 				*p = PyString_AsString(arg)[0]; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 552 | 			else | 
 | 553 | 				return "char"; | 
 | 554 | 			break; | 
 | 555 | 		} | 
 | 556 | 	 | 
 | 557 | 	case 's': /* string */ | 
 | 558 | 		{ | 
| Marc-André Lemburg | 0afff38 | 2000-09-21 21:08:30 +0000 | [diff] [blame] | 559 | 			if (*format == '#') { | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 560 | 				void **p = (void **)va_arg(*p_va, char **); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 561 | 				int *q = va_arg(*p_va, int *); | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 562 |  | 
| Marc-André Lemburg | 0afff38 | 2000-09-21 21:08:30 +0000 | [diff] [blame] | 563 | 				if (PyString_Check(arg)) { | 
 | 564 | 				    *p = PyString_AS_STRING(arg); | 
 | 565 | 				    *q = PyString_GET_SIZE(arg); | 
 | 566 | 				} | 
 | 567 | 				else if (PyUnicode_Check(arg)) { | 
 | 568 | 				    arg = _PyUnicode_AsDefaultEncodedString( | 
 | 569 | 							            arg, NULL); | 
 | 570 | 				    if (arg == NULL) | 
 | 571 | 					return "unicode conversion error"; | 
 | 572 | 				    *p = PyString_AS_STRING(arg); | 
 | 573 | 				    *q = PyString_GET_SIZE(arg); | 
 | 574 | 				} | 
 | 575 | 				else { /* any buffer-like object */ | 
 | 576 | 				    PyBufferProcs *pb = arg->ob_type->tp_as_buffer; | 
 | 577 | 				    int count; | 
 | 578 | 				    if ( pb == NULL || | 
 | 579 | 					 pb->bf_getreadbuffer == NULL || | 
 | 580 | 					 pb->bf_getsegcount == NULL ) | 
 | 581 | 					return "read-only buffer"; | 
 | 582 | 				    if ( (*pb->bf_getsegcount)(arg, NULL) != 1 ) | 
 | 583 | 					return "single-segment read-only buffer"; | 
 | 584 | 				    if ( (count = | 
 | 585 | 					  (*pb->bf_getreadbuffer)(arg, 0, p)) < 0 ) | 
 | 586 | 					return "(unspecified)"; | 
 | 587 | 				    *q = count; | 
 | 588 | 				} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 589 | 				format++; | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 590 | 			} else { | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 591 | 				char **p = va_arg(*p_va, char **); | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 592 | 			 | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 593 | 				if (PyString_Check(arg)) | 
| Guido van Rossum | e826ef0 | 2000-03-10 23:02:17 +0000 | [diff] [blame] | 594 | 				    *p = PyString_AS_STRING(arg); | 
 | 595 | 				else if (PyUnicode_Check(arg)) { | 
| Marc-André Lemburg | bff879c | 2000-08-03 18:46:08 +0000 | [diff] [blame] | 596 | 				    arg = _PyUnicode_AsDefaultEncodedString( | 
 | 597 | 							            arg, NULL); | 
| Guido van Rossum | e826ef0 | 2000-03-10 23:02:17 +0000 | [diff] [blame] | 598 | 				    if (arg == NULL) | 
 | 599 | 					return "unicode conversion error"; | 
 | 600 | 				    *p = PyString_AS_STRING(arg); | 
 | 601 | 				} | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 602 | 				else | 
 | 603 | 				  return "string"; | 
 | 604 | 				if ((int)strlen(*p) != PyString_Size(arg)) | 
 | 605 | 				  return "string without null bytes"; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 606 | 			} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 607 | 			break; | 
 | 608 | 		} | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 609 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 610 | 	case 'z': /* string, may be NULL (None) */ | 
 | 611 | 		{ | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 612 | 			if (*format == '#') { /* any buffer-like object */ | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 613 | 				void **p = (void **)va_arg(*p_va, char **); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 614 | 				int *q = va_arg(*p_va, int *); | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 615 |  | 
 | 616 | 				if (arg == Py_None) { | 
 | 617 | 				  *p = 0; | 
 | 618 | 				  *q = 0; | 
| Marc-André Lemburg | 0afff38 | 2000-09-21 21:08:30 +0000 | [diff] [blame] | 619 | 				} | 
 | 620 | 				else if (PyString_Check(arg)) { | 
 | 621 | 				    *p = PyString_AS_STRING(arg); | 
 | 622 | 				    *q = PyString_GET_SIZE(arg); | 
 | 623 | 				} | 
 | 624 | 				else if (PyUnicode_Check(arg)) { | 
 | 625 | 				    arg = _PyUnicode_AsDefaultEncodedString( | 
 | 626 | 							            arg, NULL); | 
 | 627 | 				    if (arg == NULL) | 
 | 628 | 					return "unicode conversion error"; | 
 | 629 | 				    *p = PyString_AS_STRING(arg); | 
 | 630 | 				    *q = PyString_GET_SIZE(arg); | 
 | 631 | 				} | 
 | 632 | 				else { /* any buffer-like object */ | 
 | 633 | 				    PyBufferProcs *pb = arg->ob_type->tp_as_buffer; | 
 | 634 | 				    int count; | 
 | 635 | 				    if ( pb == NULL || | 
 | 636 | 					 pb->bf_getreadbuffer == NULL || | 
 | 637 | 					 pb->bf_getsegcount == NULL ) | 
 | 638 | 					return "read-only buffer"; | 
 | 639 | 				    if ( (*pb->bf_getsegcount)(arg, NULL) != 1 ) | 
 | 640 | 					return "single-segment read-only buffer"; | 
 | 641 | 				    if ( (count = | 
 | 642 | 					  (*pb->bf_getreadbuffer)(arg, 0, p)) < 0 ) | 
 | 643 | 					return "(unspecified)"; | 
 | 644 | 				    *q = count; | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 645 | 				} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 646 | 				format++; | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 647 | 			} else { | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 648 | 				char **p = va_arg(*p_va, char **); | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 649 | 			 | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 650 | 				if (arg == Py_None) | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 651 | 				  *p = 0; | 
 | 652 | 				else if (PyString_Check(arg)) | 
 | 653 | 				  *p = PyString_AsString(arg); | 
| Guido van Rossum | e826ef0 | 2000-03-10 23:02:17 +0000 | [diff] [blame] | 654 | 				else if (PyUnicode_Check(arg)) { | 
| Marc-André Lemburg | bff879c | 2000-08-03 18:46:08 +0000 | [diff] [blame] | 655 | 				  arg = _PyUnicode_AsDefaultEncodedString( | 
 | 656 | 								  arg, NULL); | 
| Guido van Rossum | e826ef0 | 2000-03-10 23:02:17 +0000 | [diff] [blame] | 657 | 				  if (arg == NULL) | 
 | 658 | 				      return "unicode conversion error"; | 
 | 659 | 				  *p = PyString_AS_STRING(arg); | 
 | 660 | 				} | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 661 | 				else | 
 | 662 | 				  return "None or string"; | 
 | 663 | 				if (*format == '#') { | 
 | 664 | 				  int *q = va_arg(*p_va, int *); | 
 | 665 | 				  if (arg == Py_None) | 
 | 666 | 				    *q = 0; | 
 | 667 | 				  else | 
 | 668 | 				    *q = PyString_Size(arg); | 
 | 669 | 				  format++; | 
 | 670 | 				} | 
 | 671 | 				else if (*p != NULL && | 
 | 672 | 					 (int)strlen(*p) != PyString_Size(arg)) | 
 | 673 | 				  return "None or string without null bytes"; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 674 | 			} | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 675 | 			break; | 
 | 676 | 		} | 
 | 677 | 	 | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 678 | 	case 'e': /* encoded string */ | 
 | 679 | 		{ | 
 | 680 | 			char **buffer; | 
 | 681 | 			const char *encoding; | 
 | 682 | 			PyObject *u, *s; | 
 | 683 | 			int size; | 
 | 684 |  | 
 | 685 | 			/* Get 'e' parameter: the encoding name */ | 
 | 686 | 			encoding = (const char *)va_arg(*p_va, const char *); | 
 | 687 | 			if (encoding == NULL) | 
| Marc-André Lemburg | bff879c | 2000-08-03 18:46:08 +0000 | [diff] [blame] | 688 | 			    	encoding = PyUnicode_GetDefaultEncoding(); | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 689 | 			 | 
 | 690 | 			/* Get 's' parameter: the output buffer to use */ | 
 | 691 | 			if (*format != 's') | 
| Guido van Rossum | 50fbb15 | 2000-03-28 02:00:29 +0000 | [diff] [blame] | 692 | 				return "(unknown parser marker combination)"; | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 693 | 			buffer = (char **)va_arg(*p_va, char **); | 
 | 694 | 			format++; | 
 | 695 | 			if (buffer == NULL) | 
 | 696 | 				return "(buffer is NULL)"; | 
 | 697 | 			 | 
 | 698 | 			/* Convert object to Unicode */ | 
 | 699 | 			u = PyUnicode_FromObject(arg); | 
 | 700 | 			if (u == NULL) | 
 | 701 | 				return "string, unicode or text buffer"; | 
 | 702 | 			 | 
 | 703 | 			/* Encode object; use default error handling */ | 
 | 704 | 			s = PyUnicode_AsEncodedString(u, | 
 | 705 | 						      encoding, | 
 | 706 | 						      NULL); | 
 | 707 | 			Py_DECREF(u); | 
 | 708 | 			if (s == NULL) | 
 | 709 | 				return "(encoding failed)"; | 
 | 710 | 			if (!PyString_Check(s)) { | 
 | 711 | 				Py_DECREF(s); | 
 | 712 | 				return "(encoder failed to return a string)"; | 
 | 713 | 			} | 
 | 714 | 			size = PyString_GET_SIZE(s); | 
 | 715 |  | 
 | 716 | 			/* Write output; output is guaranteed to be | 
 | 717 | 			   0-terminated */ | 
 | 718 | 			if (*format == '#') {  | 
 | 719 | 				/* Using buffer length parameter '#': | 
 | 720 |  | 
 | 721 | 				   - if *buffer is NULL, a new buffer | 
 | 722 | 				   of the needed size is allocated and | 
 | 723 | 				   the data copied into it; *buffer is | 
 | 724 | 				   updated to point to the new buffer; | 
 | 725 | 				   the caller is responsible for | 
| Guido van Rossum | 24bdb04 | 2000-03-28 20:29:59 +0000 | [diff] [blame] | 726 | 				   PyMem_Free()ing it after usage | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 727 |  | 
 | 728 | 				   - if *buffer is not NULL, the data | 
 | 729 | 				   is copied to *buffer; *buffer_len | 
 | 730 | 				   has to be set to the size of the | 
 | 731 | 				   buffer on input; buffer overflow is | 
 | 732 | 				   signalled with an error; buffer has | 
 | 733 | 				   to provide enough room for the | 
 | 734 | 				   encoded string plus the trailing | 
 | 735 | 				   0-byte | 
 | 736 |  | 
 | 737 | 				   - in both cases, *buffer_len is | 
 | 738 | 				   updated to the size of the buffer | 
 | 739 | 				   /excluding/ the trailing 0-byte | 
 | 740 |  | 
 | 741 | 				*/ | 
 | 742 | 				int *buffer_len = va_arg(*p_va, int *); | 
 | 743 |  | 
 | 744 | 				format++; | 
 | 745 | 				if (buffer_len == NULL) | 
 | 746 | 					return "(buffer_len is NULL)"; | 
 | 747 | 				if (*buffer == NULL) { | 
 | 748 | 					*buffer = PyMem_NEW(char, size + 1); | 
 | 749 | 					if (*buffer == NULL) { | 
 | 750 | 						Py_DECREF(s); | 
 | 751 | 						return "(memory error)"; | 
 | 752 | 					} | 
 | 753 | 				} else { | 
 | 754 | 					if (size + 1 > *buffer_len) { | 
 | 755 | 						Py_DECREF(s); | 
 | 756 | 						return "(buffer overflow)"; | 
 | 757 | 					} | 
 | 758 | 				} | 
 | 759 | 				memcpy(*buffer, | 
 | 760 | 				       PyString_AS_STRING(s), | 
 | 761 | 				       size + 1); | 
 | 762 | 				*buffer_len = size; | 
 | 763 | 			} else { | 
 | 764 | 				/* Using a 0-terminated buffer: | 
 | 765 |  | 
 | 766 | 				   - the encoded string has to be | 
 | 767 | 				   0-terminated for this variant to | 
 | 768 | 				   work; if it is not, an error raised | 
 | 769 |  | 
 | 770 | 				   - a new buffer of the needed size | 
 | 771 | 				   is allocated and the data copied | 
 | 772 | 				   into it; *buffer is updated to | 
 | 773 | 				   point to the new buffer; the caller | 
| Guido van Rossum | 24bdb04 | 2000-03-28 20:29:59 +0000 | [diff] [blame] | 774 | 				   is responsible for PyMem_Free()ing it | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 775 | 				   after usage | 
 | 776 |  | 
 | 777 | 				 */ | 
| Guido van Rossum | 50fbb15 | 2000-03-28 02:00:29 +0000 | [diff] [blame] | 778 | 				if ((int)strlen(PyString_AS_STRING(s)) != size) | 
| Guido van Rossum | d8855fd | 2000-03-24 22:14:19 +0000 | [diff] [blame] | 779 | 					return "(encoded string without "\ | 
 | 780 | 					       "NULL bytes)"; | 
 | 781 | 				*buffer = PyMem_NEW(char, size + 1); | 
 | 782 | 				if (*buffer == NULL) { | 
 | 783 | 					Py_DECREF(s); | 
 | 784 | 					return "(memory error)"; | 
 | 785 | 				} | 
 | 786 | 				memcpy(*buffer, | 
 | 787 | 				       PyString_AS_STRING(s), | 
 | 788 | 				       size + 1); | 
 | 789 | 			} | 
 | 790 | 			Py_DECREF(s); | 
 | 791 | 			break; | 
 | 792 | 		} | 
 | 793 |  | 
| Fred Drake | 25871c0 | 2000-05-03 15:17:02 +0000 | [diff] [blame] | 794 | 	case 'u': /* raw unicode buffer (Py_UNICODE *) */ | 
 | 795 | 		{ | 
 | 796 | 			if (*format == '#') { /* any buffer-like object */ | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 797 | 				void **p = (void **)va_arg(*p_va, char **); | 
 | 798 | 				PyBufferProcs *pb = arg->ob_type->tp_as_buffer; | 
| Fred Drake | 25871c0 | 2000-05-03 15:17:02 +0000 | [diff] [blame] | 799 | 				int *q = va_arg(*p_va, int *); | 
 | 800 | 				int count; | 
 | 801 |  | 
 | 802 | 				if ( pb == NULL || | 
 | 803 | 				     pb->bf_getreadbuffer == NULL || | 
 | 804 | 				     pb->bf_getsegcount == NULL ) | 
 | 805 | 				  return "read-only buffer"; | 
 | 806 | 				if ( (*pb->bf_getsegcount)(arg, NULL) != 1 ) | 
 | 807 | 				  return "single-segment read-only buffer"; | 
 | 808 | 				if ( (count = | 
 | 809 | 				      (*pb->bf_getreadbuffer)(arg, 0, p)) < 0 ) | 
 | 810 | 				  return "(unspecified)"; | 
 | 811 | 				/* buffer interface returns bytes, we want | 
 | 812 | 				   length in characters */ | 
 | 813 | 				*q = count/(sizeof(Py_UNICODE));  | 
 | 814 | 				format++; | 
 | 815 | 			} else { | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 816 | 				Py_UNICODE **p = va_arg(*p_va, Py_UNICODE **); | 
| Fred Drake | 25871c0 | 2000-05-03 15:17:02 +0000 | [diff] [blame] | 817 | 			 | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 818 | 				if (PyUnicode_Check(arg)) | 
| Fred Drake | 25871c0 | 2000-05-03 15:17:02 +0000 | [diff] [blame] | 819 | 				    *p = PyUnicode_AS_UNICODE(arg); | 
 | 820 | 				else | 
 | 821 | 				  return "unicode"; | 
 | 822 | 			} | 
 | 823 | 			break; | 
 | 824 | 		} | 
 | 825 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 826 | 	case 'S': /* string object */ | 
 | 827 | 		{ | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 828 | 			PyObject **p = va_arg(*p_va, PyObject **); | 
 | 829 | 			if (PyString_Check(arg)) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 830 | 				*p = arg; | 
 | 831 | 			else | 
 | 832 | 				return "string"; | 
 | 833 | 			break; | 
 | 834 | 		} | 
 | 835 | 	 | 
| Guido van Rossum | e826ef0 | 2000-03-10 23:02:17 +0000 | [diff] [blame] | 836 | 	case 'U': /* Unicode object */ | 
 | 837 | 		{ | 
 | 838 | 			PyObject **p = va_arg(*p_va, PyObject **); | 
 | 839 | 			if (PyUnicode_Check(arg)) | 
 | 840 | 				*p = arg; | 
 | 841 | 			else | 
 | 842 | 				return "unicode"; | 
 | 843 | 			break; | 
 | 844 | 		} | 
 | 845 | 	 | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 846 | 	case 'O': /* object */ | 
 | 847 | 		{ | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 848 | 			PyTypeObject *type; | 
 | 849 | 			PyObject **p; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 850 | 			if (*format == '!') { | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 851 | 				type = va_arg(*p_va, PyTypeObject*); | 
| Guido van Rossum | fccfe89 | 1998-05-15 22:04:07 +0000 | [diff] [blame] | 852 | 				p = va_arg(*p_va, PyObject **); | 
 | 853 | 				format++; | 
 | 854 | 				if (arg->ob_type == type) | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 855 | 					*p = arg; | 
| Guido van Rossum | fccfe89 | 1998-05-15 22:04:07 +0000 | [diff] [blame] | 856 | 				else | 
 | 857 | 					return type->tp_name; | 
 | 858 |  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 859 | 			} | 
 | 860 | 			else if (*format == '?') { | 
 | 861 | 				inquiry pred = va_arg(*p_va, inquiry); | 
| Guido van Rossum | fccfe89 | 1998-05-15 22:04:07 +0000 | [diff] [blame] | 862 | 				p = va_arg(*p_va, PyObject **); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 863 | 				format++; | 
| Guido van Rossum | fccfe89 | 1998-05-15 22:04:07 +0000 | [diff] [blame] | 864 | 				if ((*pred)(arg))  | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 865 | 					*p = arg; | 
| Guido van Rossum | fccfe89 | 1998-05-15 22:04:07 +0000 | [diff] [blame] | 866 | 				else | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 867 | 					return "(unspecified)"; | 
| Guido van Rossum | fccfe89 | 1998-05-15 22:04:07 +0000 | [diff] [blame] | 868 | 				 | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 869 | 			} | 
 | 870 | 			else if (*format == '&') { | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 871 | 				typedef int (*converter)(PyObject *, void *); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 872 | 				converter convert = va_arg(*p_va, converter); | 
 | 873 | 				void *addr = va_arg(*p_va, void *); | 
 | 874 | 				format++; | 
 | 875 | 				if (! (*convert)(arg, addr)) | 
| Guido van Rossum | 64fc649 | 1995-01-21 14:09:37 +0000 | [diff] [blame] | 876 | 					return "(unspecified)"; | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 877 | 			} | 
 | 878 | 			else { | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 879 | 				p = va_arg(*p_va, PyObject **); | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 880 | 				*p = arg; | 
 | 881 | 			} | 
 | 882 | 			break; | 
 | 883 | 		} | 
| Guido van Rossum | fdf95dd | 1997-05-05 22:15:02 +0000 | [diff] [blame] | 884 | 		 | 
 | 885 | 		 | 
 | 886 | 	case 'w': /* memory buffer, read-write access */ | 
 | 887 | 		{ | 
 | 888 | 			void **p = va_arg(*p_va, void **); | 
 | 889 | 			PyBufferProcs *pb = arg->ob_type->tp_as_buffer; | 
 | 890 | 			int count; | 
 | 891 | 			 | 
 | 892 | 			if ( pb == NULL || pb->bf_getwritebuffer == NULL || | 
 | 893 | 					pb->bf_getsegcount == NULL ) | 
 | 894 | 				return "read-write buffer"; | 
 | 895 | 			if ( (*pb->bf_getsegcount)(arg, NULL) != 1 ) | 
 | 896 | 				return "single-segment read-write buffer"; | 
 | 897 | 			if ( (count = pb->bf_getwritebuffer(arg, 0, p)) < 0 ) | 
 | 898 | 				return "(unspecified)"; | 
 | 899 | 			if (*format == '#') { | 
 | 900 | 				int *q = va_arg(*p_va, int *); | 
 | 901 | 				 | 
 | 902 | 				*q = count; | 
 | 903 | 				format++; | 
 | 904 | 			} | 
 | 905 | 			break; | 
 | 906 | 		} | 
 | 907 | 		 | 
| Guido van Rossum | b317f8a | 1998-10-08 02:21:21 +0000 | [diff] [blame] | 908 | 	case 't': /* 8-bit character buffer, read-only access */ | 
 | 909 | 		{ | 
 | 910 | 			const char **p = va_arg(*p_va, const char **); | 
 | 911 | 			PyBufferProcs *pb = arg->ob_type->tp_as_buffer; | 
 | 912 | 			int count; | 
 | 913 |  | 
 | 914 | 			if ( *format++ != '#' ) | 
 | 915 | 				return "invalid use of 't' format character"; | 
 | 916 | 			if ( !PyType_HasFeature( | 
 | 917 | 				arg->ob_type, | 
 | 918 | 				Py_TPFLAGS_HAVE_GETCHARBUFFER) || | 
 | 919 | 			     pb == NULL || | 
 | 920 | 			     pb->bf_getcharbuffer == NULL || | 
 | 921 | 			     pb->bf_getsegcount == NULL ) | 
 | 922 | 				return "read-only character buffer"; | 
 | 923 | 			if ( (*pb->bf_getsegcount)(arg, NULL) != 1 ) | 
 | 924 | 				return "single-segment read-only buffer"; | 
 | 925 | 			if ( (count = pb->bf_getcharbuffer(arg, 0, p)) < 0 ) | 
 | 926 | 				return "(unspecified)"; | 
 | 927 |  | 
 | 928 | 			*va_arg(*p_va, int *) = count; | 
 | 929 |  | 
 | 930 | 			break; | 
 | 931 | 		} | 
 | 932 | 		 | 
| Guido van Rossum | fe3f1a2 | 1994-09-29 09:42:55 +0000 | [diff] [blame] | 933 | 	 | 
 | 934 | 	default: | 
 | 935 | 		return "impossible<bad format char>"; | 
 | 936 | 	 | 
 | 937 | 	} | 
 | 938 | 	 | 
 | 939 | 	*p_format = format; | 
 | 940 | 	return NULL; | 
 | 941 | } | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 942 |  | 
 | 943 |  | 
 | 944 | /* Support for keyword arguments donated by | 
 | 945 |    Geoff Philbrick <philbric@delphi.hks.com> */ | 
 | 946 |  | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 947 | int PyArg_ParseTupleAndKeywords(PyObject *args, | 
 | 948 | 				PyObject *keywords, | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 949 | 				char *format,  | 
 | 950 | 				char **kwlist, ...) | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 951 | { | 
 | 952 | 	int retval; | 
 | 953 | 	va_list va; | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 954 | 	 | 
 | 955 | 	va_start(va, kwlist); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 956 | 	retval = vgetargskeywords(args, keywords, format, kwlist, &va);	 | 
 | 957 | 	va_end(va); | 
 | 958 | 	return retval; | 
 | 959 | } | 
 | 960 |  | 
 | 961 |  | 
 | 962 | static int | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 963 | vgetargskeywords(PyObject *args, PyObject *keywords, char *format, | 
 | 964 | 	         char **kwlist, va_list *p_va) | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 965 | { | 
 | 966 | 	char msgbuf[256]; | 
 | 967 | 	int levels[32]; | 
 | 968 | 	char *fname = NULL; | 
 | 969 | 	char *message = NULL; | 
 | 970 | 	int min = -1; | 
 | 971 | 	int max = 0; | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 972 | 	char *formatsave = format; | 
 | 973 | 	int i, len, tplen, kwlen; | 
 | 974 | 	char *msg, *ks, **p; | 
 | 975 | 	int nkwds, pos, match, converted; | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 976 | 	PyObject *key, *value; | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 977 | 	 | 
 | 978 | 	/* nested tuples cannot be parsed when using keyword arguments */ | 
 | 979 | 	 | 
 | 980 | 	for (;;) { | 
 | 981 | 		int c = *format++; | 
 | 982 | 		if (c == '(') { | 
 | 983 | 			PyErr_SetString(PyExc_SystemError, | 
 | 984 | 		      "tuple found in format when using keyword arguments"); | 
 | 985 | 			return 0; | 
 | 986 | 		} | 
 | 987 | 		else if (c == '\0') | 
 | 988 | 			break; | 
 | 989 | 		else if (c == ':') { | 
 | 990 | 			fname = format; | 
 | 991 | 			break; | 
 | 992 | 		} | 
 | 993 | 		else if (c == ';') { | 
 | 994 | 			message = format; | 
 | 995 | 			break; | 
 | 996 | 		} | 
| Marc-André Lemburg | bbcf2a7 | 2000-09-08 11:49:37 +0000 | [diff] [blame] | 997 | 		else if (c == 'e') | 
 | 998 | 			; /* Pass */ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 999 | 		else if (isalpha(c)) | 
 | 1000 | 			max++; | 
 | 1001 | 		else if (c == '|') | 
 | 1002 | 			min = max; | 
 | 1003 | 	}	 | 
 | 1004 | 	 | 
 | 1005 | 	if (min < 0) | 
 | 1006 | 		min = max; | 
 | 1007 | 	 | 
 | 1008 | 	format = formatsave; | 
 | 1009 | 	 | 
 | 1010 | 	if (!PyTuple_Check(args)) { | 
 | 1011 | 		PyErr_SetString(PyExc_SystemError, | 
 | 1012 | 		    "new style getargs format but argument is not a tuple"); | 
 | 1013 | 		return 0; | 
 | 1014 | 	}	 | 
 | 1015 | 	 | 
 | 1016 | 	tplen = PyTuple_Size(args); | 
 | 1017 | 	 | 
 | 1018 | 	/* do a cursory check of the keywords just to see how many we got */ | 
 | 1019 | 	    | 
 | 1020 | 	if (keywords) { 	 | 
 | 1021 | 		if (!PyDict_Check(keywords)) { | 
 | 1022 | 			PyErr_SetString(PyExc_SystemError, | 
 | 1023 | 	  "non-dictionary object received when keyword dictionary expected"); | 
 | 1024 | 			return 0; | 
 | 1025 | 		}	 | 
 | 1026 | 		kwlen = PyDict_Size(keywords); | 
 | 1027 | 	} | 
 | 1028 | 	else { | 
 | 1029 | 		kwlen = 0; | 
 | 1030 | 	} | 
 | 1031 | 			 | 
 | 1032 | 	/* make sure there are no duplicate values for an argument; | 
 | 1033 | 	   its not clear when to use the term "keyword argument vs.  | 
 | 1034 | 	   keyword parameter in messages */ | 
 | 1035 | 	 | 
 | 1036 | 	if (keywords) { | 
 | 1037 | 		for (i = 0; i < tplen; i++) { | 
 | 1038 | 			if (PyMapping_HasKeyString(keywords, kwlist[i])) { | 
 | 1039 | 				sprintf(msgbuf, | 
 | 1040 | 					"keyword parameter %s redefined", | 
 | 1041 | 					kwlist[i]); | 
 | 1042 | 				PyErr_SetString(PyExc_TypeError, msgbuf); | 
 | 1043 | 				return 0; | 
 | 1044 | 			} | 
 | 1045 | 		} | 
 | 1046 | 	} | 
 | 1047 | 	PyErr_Clear(); /* I'm not which Py functions set the error string */ | 
 | 1048 | 		 | 
 | 1049 | 	/* required arguments missing from args can be supplied by keyword  | 
 | 1050 | 	   arguments */ | 
 | 1051 | 	 | 
 | 1052 | 	len = tplen; | 
 | 1053 | 	if (keywords && tplen < min) { | 
 | 1054 | 		for (i = tplen; i < min; i++) { | 
 | 1055 | 		  if (PyMapping_HasKeyString(keywords, kwlist[i])) { | 
 | 1056 | 				len++; | 
 | 1057 | 		  } | 
 | 1058 | 		} | 
 | 1059 | 	} | 
 | 1060 | 	PyErr_Clear();	 | 
 | 1061 | 	 | 
 | 1062 | 	/* make sure we got an acceptable number of arguments; the message | 
 | 1063 | 	   is a little confusing with keywords since keyword arguments | 
 | 1064 | 	   which are supplied, but don't match the required arguments | 
 | 1065 | 	   are not included in the "%d given" part of the message */ | 
 | 1066 |  | 
 | 1067 | 	if (len < min || max < len) { | 
 | 1068 | 		if (message == NULL) { | 
 | 1069 | 			sprintf(msgbuf, | 
 | 1070 | 				"%s requires %s %d argument%s; %d given", | 
 | 1071 | 				fname==NULL ? "function" : fname, | 
 | 1072 | 				min==max ? "exactly" | 
 | 1073 | 				         : len < min ? "at least" : "at most", | 
 | 1074 | 				len < min ? min : max, | 
 | 1075 | 				(len < min ? min : max) == 1 ? "" : "s", | 
 | 1076 | 				len); | 
 | 1077 | 			message = msgbuf; | 
 | 1078 | 		} | 
 | 1079 | 		PyErr_SetString(PyExc_TypeError, message); | 
 | 1080 | 		return 0; | 
 | 1081 | 	} | 
 | 1082 | 	 | 
 | 1083 | 	for (i = 0; i < tplen; i++) { | 
 | 1084 | 		if (*format == '|') | 
 | 1085 | 			format++; | 
 | 1086 | 		msg = convertitem(PyTuple_GetItem(args, i), &format, p_va, | 
 | 1087 | 				 levels, msgbuf); | 
 | 1088 | 		if (msg) { | 
 | 1089 | 			seterror(i+1, msg, levels, fname, message); | 
 | 1090 | 			return 0; | 
 | 1091 | 		} | 
 | 1092 | 	} | 
 | 1093 |  | 
 | 1094 | 	/* handle no keyword parameters in call  */	 | 
 | 1095 | 	   	    | 
 | 1096 | 	if (!keywords) return 1;  | 
 | 1097 | 		 | 
 | 1098 | 	/* make sure the number of keywords in the keyword list matches the  | 
 | 1099 | 	   number of items in the format string */ | 
 | 1100 | 	   | 
 | 1101 | 	nkwds = 0; | 
 | 1102 | 	p =  kwlist; | 
 | 1103 | 	for (;;) { | 
 | 1104 | 		if (!*(p++)) break; | 
 | 1105 | 		nkwds++; | 
 | 1106 | 	} | 
 | 1107 |  | 
 | 1108 | 	if (nkwds != max) { | 
 | 1109 | 		PyErr_SetString(PyExc_SystemError, | 
 | 1110 | 	  "number of items in format string and keyword list do not match"); | 
 | 1111 | 		return 0; | 
 | 1112 | 	}	  	   | 
 | 1113 | 			 | 
 | 1114 | 	/* convert the keyword arguments; this uses the format  | 
 | 1115 | 	   string where it was left after processing args */ | 
 | 1116 | 	 | 
 | 1117 | 	converted = 0; | 
 | 1118 | 	for (i = tplen; i < nkwds; i++) { | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 1119 | 		PyObject *item; | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1120 | 		if (*format == '|') | 
 | 1121 | 			format++; | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1122 | 		item = PyMapping_GetItemString(keywords, kwlist[i]); | 
 | 1123 | 		if (item != NULL) { | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1124 | 			msg = convertitem(item, &format, p_va, levels, msgbuf); | 
 | 1125 | 			if (msg) { | 
 | 1126 | 				seterror(i+1, msg, levels, fname, message); | 
 | 1127 | 				return 0; | 
 | 1128 | 			} | 
 | 1129 | 			converted++; | 
 | 1130 | 		} | 
 | 1131 | 		else { | 
 | 1132 | 			PyErr_Clear(); | 
 | 1133 | 			msg = skipitem(&format, p_va); | 
 | 1134 | 			if (msg) { | 
 | 1135 | 				seterror(i+1, msg, levels, fname, message); | 
 | 1136 | 				return 0; | 
 | 1137 | 			} | 
 | 1138 | 		} | 
 | 1139 | 	} | 
 | 1140 | 	 | 
 | 1141 | 	/* make sure there are no extraneous keyword arguments */ | 
 | 1142 | 	 | 
 | 1143 | 	pos = 0; | 
 | 1144 | 	if (converted < kwlen) { | 
 | 1145 | 		while (PyDict_Next(keywords, &pos, &key, &value)) { | 
 | 1146 | 			match = 0; | 
 | 1147 | 			ks = PyString_AsString(key); | 
 | 1148 | 			for (i = 0; i < nkwds; i++) { | 
 | 1149 | 				if (!strcmp(ks, kwlist[i])) { | 
 | 1150 | 					match = 1; | 
 | 1151 | 					break; | 
 | 1152 | 				} | 
 | 1153 | 			} | 
 | 1154 | 			if (!match) { | 
 | 1155 | 				sprintf(msgbuf, | 
| Guido van Rossum | 80dc16b | 2000-05-08 14:02:41 +0000 | [diff] [blame] | 1156 | 			"%s is an invalid keyword argument for this function", | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1157 | 					ks); | 
 | 1158 | 				PyErr_SetString(PyExc_TypeError, msgbuf); | 
 | 1159 | 				return 0; | 
 | 1160 | 			} | 
 | 1161 | 		} | 
 | 1162 | 	} | 
 | 1163 | 	 | 
 | 1164 | 	return 1; | 
 | 1165 | } | 
 | 1166 |  | 
 | 1167 |  | 
 | 1168 | static char * | 
| Thomas Wouters | f70ef4f | 2000-07-22 18:47:25 +0000 | [diff] [blame] | 1169 | skipitem(char **p_format, va_list *p_va) | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1170 | { | 
 | 1171 | 	char *format = *p_format; | 
 | 1172 | 	char c = *format++; | 
 | 1173 | 	 | 
 | 1174 | 	switch (c) { | 
 | 1175 | 	 | 
 | 1176 | 	case 'b': /* byte -- very short int */ | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 1177 | 	case 'B': /* byte as bitfield */ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1178 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1179 | 			(void) va_arg(*p_va, char *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1180 | 			break; | 
 | 1181 | 		} | 
 | 1182 | 	 | 
 | 1183 | 	case 'h': /* short int */ | 
 | 1184 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1185 | 			(void) va_arg(*p_va, short *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1186 | 			break; | 
 | 1187 | 		} | 
 | 1188 | 	 | 
| Jack Jansen | cc22fbe | 2000-08-05 21:29:58 +0000 | [diff] [blame] | 1189 | 	case 'H': /* short int as bitfield */ | 
| Jack Jansen | d50338f | 2000-07-06 12:22:00 +0000 | [diff] [blame] | 1190 | 		{ | 
 | 1191 | 			(void) va_arg(*p_va, unsigned short *); | 
 | 1192 | 			break; | 
 | 1193 | 		} | 
 | 1194 | 	 | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1195 | 	case 'i': /* int */ | 
 | 1196 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1197 | 			(void) va_arg(*p_va, int *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1198 | 			break; | 
 | 1199 | 		} | 
 | 1200 | 	 | 
 | 1201 | 	case 'l': /* long int */ | 
 | 1202 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1203 | 			(void) va_arg(*p_va, long *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1204 | 			break; | 
 | 1205 | 		} | 
 | 1206 | 	 | 
| Guido van Rossum | 3dbba6e | 1999-01-25 21:48:56 +0000 | [diff] [blame] | 1207 | #ifdef HAVE_LONG_LONG | 
| Guido van Rossum | 3293b07 | 1998-08-25 16:07:15 +0000 | [diff] [blame] | 1208 | 	case 'L': /* LONG_LONG int */ | 
| Guido van Rossum | 1a8791e | 1998-08-04 22:46:29 +0000 | [diff] [blame] | 1209 | 		{ | 
| Guido van Rossum | 3293b07 | 1998-08-25 16:07:15 +0000 | [diff] [blame] | 1210 | 			(void) va_arg(*p_va, LONG_LONG *); | 
| Guido van Rossum | 1a8791e | 1998-08-04 22:46:29 +0000 | [diff] [blame] | 1211 | 			break; | 
 | 1212 | 		} | 
 | 1213 | #endif | 
 | 1214 | 	 | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1215 | 	case 'f': /* float */ | 
 | 1216 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1217 | 			(void) va_arg(*p_va, float *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1218 | 			break; | 
 | 1219 | 		} | 
 | 1220 | 	 | 
 | 1221 | 	case 'd': /* double */ | 
 | 1222 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1223 | 			(void) va_arg(*p_va, double *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1224 | 			break; | 
 | 1225 | 		} | 
 | 1226 | 	 | 
 | 1227 | #ifndef WITHOUT_COMPLEX | 
 | 1228 | 	case 'D': /* complex double */ | 
 | 1229 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1230 | 			(void) va_arg(*p_va, Py_complex *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1231 | 			break; | 
 | 1232 | 		} | 
 | 1233 | #endif /* WITHOUT_COMPLEX */ | 
 | 1234 | 	 | 
 | 1235 | 	case 'c': /* char */ | 
 | 1236 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1237 | 			(void) va_arg(*p_va, char *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1238 | 			break; | 
 | 1239 | 		} | 
 | 1240 | 	 | 
 | 1241 | 	case 's': /* string */ | 
 | 1242 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1243 | 			(void) va_arg(*p_va, char **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1244 | 			if (*format == '#') { | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1245 | 				(void) va_arg(*p_va, int *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1246 | 				format++; | 
 | 1247 | 			} | 
 | 1248 | 			break; | 
 | 1249 | 		} | 
 | 1250 | 	 | 
 | 1251 | 	case 'z': /* string */ | 
 | 1252 | 		{ | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1253 | 			(void) va_arg(*p_va, char **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1254 | 			if (*format == '#') { | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1255 | 				(void) va_arg(*p_va, int *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1256 | 				format++; | 
 | 1257 | 			} | 
 | 1258 | 			break; | 
 | 1259 | 		} | 
 | 1260 | 	 | 
 | 1261 | 	case 'S': /* string object */ | 
 | 1262 | 		{ | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 1263 | 			(void) va_arg(*p_va, PyObject **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1264 | 			break; | 
 | 1265 | 		} | 
 | 1266 | 	 | 
 | 1267 | 	case 'O': /* object */ | 
 | 1268 | 		{ | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1269 | 			if (*format == '!') { | 
 | 1270 | 				format++; | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 1271 | 				(void) va_arg(*p_va, PyTypeObject*); | 
 | 1272 | 				(void) va_arg(*p_va, PyObject **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1273 | 			} | 
 | 1274 | #if 0 | 
 | 1275 | /* I don't know what this is for */ | 
 | 1276 | 			else if (*format == '?') { | 
 | 1277 | 				inquiry pred = va_arg(*p_va, inquiry); | 
 | 1278 | 				format++; | 
 | 1279 | 				if ((*pred)(arg)) { | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 1280 | 					(void) va_arg(*p_va, PyObject **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1281 | 				} | 
 | 1282 | 			} | 
 | 1283 | #endif | 
 | 1284 | 			else if (*format == '&') { | 
| Tim Peters | dbd9ba6 | 2000-07-09 03:09:57 +0000 | [diff] [blame] | 1285 | 				typedef int (*converter)(PyObject *, void *); | 
| Guido van Rossum | 80bb965 | 1996-12-05 23:27:02 +0000 | [diff] [blame] | 1286 | 				(void) va_arg(*p_va, converter); | 
 | 1287 | 				(void) va_arg(*p_va, void *); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1288 | 				format++; | 
 | 1289 | 			} | 
 | 1290 | 			else { | 
| Guido van Rossum | 79f25d9 | 1997-04-29 20:08:16 +0000 | [diff] [blame] | 1291 | 				(void) va_arg(*p_va, PyObject **); | 
| Guido van Rossum | aa35465 | 1996-08-19 19:32:04 +0000 | [diff] [blame] | 1292 | 			} | 
 | 1293 | 			break; | 
 | 1294 | 		} | 
 | 1295 | 	 | 
 | 1296 | 	default: | 
 | 1297 | 		return "impossible<bad format char>"; | 
 | 1298 | 	 | 
 | 1299 | 	} | 
 | 1300 | 	 | 
 | 1301 | 	*p_format = format; | 
 | 1302 | 	return NULL; | 
 | 1303 | } |