blob: c9a5273c845c35ad40e00848910ec4f66829b4dc [file] [log] [blame]
Guido van Rossumfe3f1a21994-09-29 09:42:55 +00001
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 Wouters7e474022000-07-16 12:04:32 +00007 XXX or ridiculously deep nesting in format strings. */
Guido van Rossumfe3f1a21994-09-29 09:42:55 +00008
Guido van Rossum79f25d91997-04-29 20:08:16 +00009#include "Python.h"
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000010
Guido van Rossumc1d50531996-08-21 23:38:24 +000011#include <ctype.h>
12
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000013
Tim Petersdbd9ba62000-07-09 03:09:57 +000014int PyArg_Parse(PyObject *, char *, ...);
15int PyArg_ParseTuple(PyObject *, char *, ...);
16int PyArg_VaParse(PyObject *, char *, va_list);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000017
Tim Petersdbd9ba62000-07-09 03:09:57 +000018int PyArg_ParseTupleAndKeywords(PyObject *, PyObject *,
19 char *, char **, ...);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000020
21/* Forward */
Tim Petersdbd9ba62000-07-09 03:09:57 +000022static int vgetargs1(PyObject *, char *, va_list *, int);
23static void seterror(int, char *, int *, char *, char *);
24static char *convertitem(PyObject *, char **, va_list *, int *, char *);
25static char *converttuple(PyObject *, char **, va_list *,
26 int *, char *, int);
27static char *convertsimple(PyObject *, char **, va_list *, char *);
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +000028static int convertbuffer(PyObject *, void **p, char **);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000029
Tim Petersdbd9ba62000-07-09 03:09:57 +000030static int vgetargskeywords(PyObject *, PyObject *,
31 char *, char **, va_list *);
32static char *skipitem(char **, va_list *);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000033
Guido van Rossum79f25d91997-04-29 20:08:16 +000034int PyArg_Parse(PyObject *args, char *format, ...)
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000035{
36 int retval;
37 va_list va;
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000038
39 va_start(va, format);
Guido van Rossum1ae940a1995-01-02 19:04:15 +000040 retval = vgetargs1(args, format, &va, 1);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000041 va_end(va);
42 return retval;
43}
44
45
Guido van Rossum79f25d91997-04-29 20:08:16 +000046int PyArg_ParseTuple(PyObject *args, char *format, ...)
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000047{
48 int retval;
49 va_list va;
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000050
51 va_start(va, format);
Guido van Rossum1ae940a1995-01-02 19:04:15 +000052 retval = vgetargs1(args, format, &va, 0);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000053 va_end(va);
54 return retval;
55}
56
57
58int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000059PyArg_VaParse(PyObject *args, char *format, va_list va)
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000060{
Guido van Rossum1ae940a1995-01-02 19:04:15 +000061 va_list lva;
62
63#ifdef VA_LIST_IS_ARRAY
64 memcpy(lva, va, sizeof(va_list));
65#else
66 lva = va;
67#endif
68
69 return vgetargs1(args, format, &lva, 0);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000070}
71
72
73static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +000074vgetargs1(PyObject *args, char *format, va_list *p_va, int compat)
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000075{
76 char msgbuf[256];
77 int levels[32];
78 char *fname = NULL;
79 char *message = NULL;
80 int min = -1;
81 int max = 0;
82 int level = 0;
Jeremy Hylton25916bd2001-05-29 17:46:19 +000083 int endfmt = 0;
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000084 char *formatsave = format;
85 int i, len;
86 char *msg;
87
Tim Peters5c4d5bf2001-02-12 22:13:26 +000088 assert(compat || (args != (PyObject*)NULL));
89
Jeremy Hylton25916bd2001-05-29 17:46:19 +000090 while (endfmt == 0) {
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000091 int c = *format++;
Jeremy Hylton25916bd2001-05-29 17:46:19 +000092 switch (c) {
93 case '(':
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000094 if (level == 0)
95 max++;
96 level++;
Jeremy Hylton25916bd2001-05-29 17:46:19 +000097 break;
98 case ')':
Guido van Rossumfe3f1a21994-09-29 09:42:55 +000099 if (level == 0)
Jeremy Hylton25916bd2001-05-29 17:46:19 +0000100 Py_FatalError("excess ')' in getargs format");
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000101 else
102 level--;
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000103 break;
Jeremy Hylton25916bd2001-05-29 17:46:19 +0000104 case '\0':
105 endfmt = 1;
106 break;
107 case ':':
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000108 fname = format;
Jeremy Hylton25916bd2001-05-29 17:46:19 +0000109 endfmt = 1;
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000110 break;
Jeremy Hylton25916bd2001-05-29 17:46:19 +0000111 case ';':
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000112 message = format;
Jeremy Hylton25916bd2001-05-29 17:46:19 +0000113 endfmt = 1;
114 break;
115 default:
116 if (level == 0) {
117 if (c == 'O')
118 max++;
119 else if (isalpha(c)) {
120 if (c != 'e') /* skip encoded */
121 max++;
122 } else if (c == '|')
123 min = max;
124 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000125 break;
126 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000127 }
128
129 if (level != 0)
Guido van Rossum79f25d91997-04-29 20:08:16 +0000130 Py_FatalError(/* '(' */ "missing ')' in getargs format");
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000131
132 if (min < 0)
133 min = max;
134
135 format = formatsave;
136
137 if (compat) {
138 if (max == 0) {
139 if (args == NULL)
140 return 1;
Ka-Ping Yee20579702001-01-15 22:14:16 +0000141 sprintf(msgbuf, "%s%s takes no arguments",
142 fname==NULL ? "function" : fname,
143 fname==NULL ? "" : "()");
Guido van Rossum79f25d91997-04-29 20:08:16 +0000144 PyErr_SetString(PyExc_TypeError, msgbuf);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000145 return 0;
146 }
147 else if (min == 1 && max == 1) {
Guido van Rossum13d0ed11994-11-10 22:35:48 +0000148 if (args == NULL) {
149 sprintf(msgbuf,
Ka-Ping Yee20579702001-01-15 22:14:16 +0000150 "%s%s takes at least one argument",
151 fname==NULL ? "function" : fname,
152 fname==NULL ? "" : "()");
Guido van Rossum79f25d91997-04-29 20:08:16 +0000153 PyErr_SetString(PyExc_TypeError, msgbuf);
Guido van Rossum13d0ed11994-11-10 22:35:48 +0000154 return 0;
155 }
Guido van Rossum1ae940a1995-01-02 19:04:15 +0000156 msg = convertitem(args, &format, p_va, levels, msgbuf);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000157 if (msg == NULL)
158 return 1;
159 seterror(levels[0], msg, levels+1, fname, message);
160 return 0;
161 }
162 else {
Guido van Rossum79f25d91997-04-29 20:08:16 +0000163 PyErr_SetString(PyExc_SystemError,
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000164 "old style getargs format uses new features");
165 return 0;
166 }
167 }
168
Guido van Rossum79f25d91997-04-29 20:08:16 +0000169 if (!PyTuple_Check(args)) {
170 PyErr_SetString(PyExc_SystemError,
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000171 "new style getargs format but argument is not a tuple");
172 return 0;
173 }
174
Jeremy Hylton0f8117f2001-05-18 20:57:38 +0000175 len = PyTuple_GET_SIZE(args);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000176
177 if (len < min || max < len) {
178 if (message == NULL) {
179 sprintf(msgbuf,
Ka-Ping Yee20579702001-01-15 22:14:16 +0000180 "%s%s takes %s %d argument%s (%d given)",
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000181 fname==NULL ? "function" : fname,
Ka-Ping Yee20579702001-01-15 22:14:16 +0000182 fname==NULL ? "" : "()",
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000183 min==max ? "exactly"
184 : len < min ? "at least" : "at most",
185 len < min ? min : max,
186 (len < min ? min : max) == 1 ? "" : "s",
187 len);
188 message = msgbuf;
189 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000190 PyErr_SetString(PyExc_TypeError, message);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000191 return 0;
192 }
193
194 for (i = 0; i < len; i++) {
195 if (*format == '|')
196 format++;
Jeremy Hylton0f8117f2001-05-18 20:57:38 +0000197 msg = convertitem(PyTuple_GET_ITEM(args, i), &format, p_va,
198 levels, msgbuf);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000199 if (msg) {
200 seterror(i+1, msg, levels, fname, message);
201 return 0;
202 }
203 }
Guido van Rossum231a41e1997-12-09 20:36:39 +0000204
Guido van Rossum730806d1998-04-10 22:27:42 +0000205 if (*format != '\0' && !isalpha((int)(*format)) &&
Guido van Rossum7d4f68c1997-12-19 04:25:23 +0000206 *format != '(' &&
Guido van Rossum231a41e1997-12-09 20:36:39 +0000207 *format != '|' && *format != ':' && *format != ';') {
208 PyErr_Format(PyExc_SystemError,
Guido van Rossum0d6b49e1998-01-19 22:22:44 +0000209 "bad format string: %.200s", formatsave);
Guido van Rossum231a41e1997-12-09 20:36:39 +0000210 return 0;
211 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000212
213 return 1;
214}
215
216
217
218static void
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000219seterror(int iarg, char *msg, int *levels, char *fname, char *message)
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000220{
221 char buf[256];
222 int i;
223 char *p = buf;
224
Guido van Rossum79f25d91997-04-29 20:08:16 +0000225 if (PyErr_Occurred())
Guido van Rossum64fc6491995-01-21 14:09:37 +0000226 return;
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000227 else if (message == NULL) {
228 if (fname != NULL) {
Ka-Ping Yee20579702001-01-15 22:14:16 +0000229 sprintf(p, "%s() ", fname);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000230 p += strlen(p);
231 }
Ka-Ping Yee20579702001-01-15 22:14:16 +0000232 if (iarg != 0) {
233 sprintf(p, "argument %d", iarg);
234 i = 0;
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000235 p += strlen(p);
Ka-Ping Yee20579702001-01-15 22:14:16 +0000236 while (levels[i] > 0) {
237 sprintf(p, ", item %d", levels[i]-1);
238 p += strlen(p);
239 i++;
240 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000241 }
Ka-Ping Yee20579702001-01-15 22:14:16 +0000242 else {
243 sprintf(p, "argument");
244 p += strlen(p);
245 }
246 sprintf(p, " %s", msg);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000247 message = buf;
248 }
Guido van Rossum79f25d91997-04-29 20:08:16 +0000249 PyErr_SetString(PyExc_TypeError, message);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000250}
251
252
253/* Convert a tuple argument.
254 On entry, *p_format points to the character _after_ the opening '('.
255 On successful exit, *p_format points to the closing ')'.
256 If successful:
257 *p_format and *p_va are updated,
258 *levels and *msgbuf are untouched,
259 and NULL is returned.
260 If the argument is invalid:
261 *p_format is unchanged,
262 *p_va is undefined,
263 *levels is a 0-terminated list of item numbers,
264 *msgbuf contains an error message, whose format is:
Ka-Ping Yee20579702001-01-15 22:14:16 +0000265 "must be <typename1>, not <typename2>", where:
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000266 <typename1> is the name of the expected type, and
267 <typename2> is the name of the actual type,
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000268 and msgbuf is returned.
269*/
270
271static char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000272converttuple(PyObject *arg, char **p_format, va_list *p_va, int *levels,
273 char *msgbuf, int toplevel)
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000274{
275 int level = 0;
276 int n = 0;
277 char *format = *p_format;
278 int i;
279
280 for (;;) {
281 int c = *format++;
282 if (c == '(') {
283 if (level == 0)
284 n++;
285 level++;
286 }
287 else if (c == ')') {
288 if (level == 0)
289 break;
290 level--;
291 }
292 else if (c == ':' || c == ';' || c == '\0')
293 break;
294 else if (level == 0 && isalpha(c))
295 n++;
296 }
297
Ka-Ping Yee20579702001-01-15 22:14:16 +0000298 if (!PySequence_Check(arg) || PyString_Check(arg)) {
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000299 levels[0] = 0;
300 sprintf(msgbuf,
Ka-Ping Yee20579702001-01-15 22:14:16 +0000301 toplevel ? "expected %d arguments, not %s" :
302 "must be %d-item sequence, not %s",
Guido van Rossum79f25d91997-04-29 20:08:16 +0000303 n, arg == Py_None ? "None" : arg->ob_type->tp_name);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000304 return msgbuf;
305 }
306
Jeremy Hylton03657cf2000-07-12 13:05:33 +0000307 if ((i = PySequence_Size(arg)) != n) {
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000308 levels[0] = 0;
309 sprintf(msgbuf,
Ka-Ping Yee20579702001-01-15 22:14:16 +0000310 toplevel ? "expected %d arguments, not %d" :
311 "must be sequence of length %d, not %d",
312 n, i);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000313 return msgbuf;
314 }
Ka-Ping Yee20579702001-01-15 22:14:16 +0000315
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000316 format = *p_format;
317 for (i = 0; i < n; i++) {
318 char *msg;
Guido van Rossum66368cc1999-02-17 23:16:43 +0000319 PyObject *item;
320 item = PySequence_GetItem(arg, i);
321 msg = convertitem(item, &format, p_va, levels+1, msgbuf);
322 /* PySequence_GetItem calls tp->sq_item, which INCREFs */
323 Py_XDECREF(item);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000324 if (msg != NULL) {
325 levels[0] = i+1;
326 return msg;
327 }
328 }
Ka-Ping Yee20579702001-01-15 22:14:16 +0000329
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000330 *p_format = format;
331 return NULL;
332}
333
334
335/* Convert a single item. */
336
337static char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000338convertitem(PyObject *arg, char **p_format, va_list *p_va, int *levels,
339 char *msgbuf)
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000340{
341 char *msg;
342 char *format = *p_format;
343
344 if (*format == '(' /* ')' */) {
345 format++;
346 msg = converttuple(arg, &format, p_va, levels, msgbuf, 0);
347 if (msg == NULL)
348 format++;
349 }
350 else {
351 msg = convertsimple(arg, &format, p_va, msgbuf);
352 if (msg != NULL)
353 levels[0] = 0;
354 }
355 if (msg == NULL)
356 *p_format = format;
357 return msg;
358}
359
360
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000361
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000362#define UNICODE_DEFAULT_ENCODING(arg) \
363 _PyUnicode_AsDefaultEncodedString(arg, NULL)
364
365/* Format an error message generated by convertsimple(). */
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000366
367static char *
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000368converterr(char *expected, PyObject *arg, char *msgbuf)
369{
370 assert (expected != NULL);
371 sprintf(msgbuf, "must be %.50s, not %.50s", expected,
372 arg == Py_None ? "None" : arg->ob_type->tp_name);
373 return msgbuf;
374}
375
376#define CONV_UNICODE "(unicode conversion error)"
377
378/* Convert a non-tuple argument. Return NULL if conversion went OK,
379 or a string with a message describing the failure. The message is
380 formatted as "must be <desired type>, not <actual type>".
381 When failing, an exception may or may not have been raised.
382 Don't call if a tuple is expected.
383*/
384
385static char *
386convertsimple(PyObject *arg, char **p_format, va_list *p_va, char *msgbuf)
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000387{
388 char *format = *p_format;
389 char c = *format++;
390
391 switch (c) {
392
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000393 case 'b': { /* unsigned byte -- very short int */
394 char *p = va_arg(*p_va, char *);
395 long ival = PyInt_AsLong(arg);
396 if (ival == -1 && PyErr_Occurred())
397 return converterr("integer<b>", arg, msgbuf);
398 else if (ival < 0) {
399 PyErr_SetString(PyExc_OverflowError,
400 "unsigned byte integer is less than minimum");
401 return converterr("integer<b>", arg, msgbuf);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000402 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000403 else if (ival > UCHAR_MAX) {
404 PyErr_SetString(PyExc_OverflowError,
405 "unsigned byte integer is greater than maximum");
406 return converterr("integer<b>", arg, msgbuf);
407 }
408 else
409 *p = (unsigned char) ival;
410 break;
411 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000412
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000413 case 'B': {/* byte sized bitfield - both signed and unsigned
414 values allowed */
415 char *p = va_arg(*p_va, char *);
416 long ival = PyInt_AsLong(arg);
417 if (ival == -1 && PyErr_Occurred())
418 return converterr("integer<b>", arg, msgbuf);
419 else if (ival < SCHAR_MIN) {
420 PyErr_SetString(PyExc_OverflowError,
421 "byte-sized integer bitfield is less than minimum");
422 return converterr("integer<B>", arg, msgbuf);
Jack Jansencc22fbe2000-08-05 21:29:58 +0000423 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000424 else if (ival > (int)UCHAR_MAX) {
425 PyErr_SetString(PyExc_OverflowError,
426 "byte-sized integer bitfield is greater than maximum");
427 return converterr("integer<B>", arg, msgbuf);
428 }
429 else
430 *p = (unsigned char) ival;
431 break;
432 }
Jack Jansencc22fbe2000-08-05 21:29:58 +0000433
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000434 case 'h': {/* signed short int */
435 short *p = va_arg(*p_va, short *);
436 long ival = PyInt_AsLong(arg);
437 if (ival == -1 && PyErr_Occurred())
438 return converterr("integer<h>", arg, msgbuf);
439 else if (ival < SHRT_MIN) {
440 PyErr_SetString(PyExc_OverflowError,
441 "signed short integer is less than minimum");
442 return converterr("integer<h>", arg, msgbuf);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000443 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000444 else if (ival > SHRT_MAX) {
445 PyErr_SetString(PyExc_OverflowError,
446 "signed short integer is greater than maximum");
447 return converterr("integer<h>", arg, msgbuf);
448 }
449 else
450 *p = (short) ival;
451 break;
452 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000453
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000454 case 'H': { /* short int sized bitfield, both signed and
455 unsigned allowed */
456 unsigned short *p = va_arg(*p_va, unsigned short *);
457 long ival = PyInt_AsLong(arg);
458 if (ival == -1 && PyErr_Occurred())
459 return converterr("integer<H>", arg, msgbuf);
460 else if (ival < SHRT_MIN) {
461 PyErr_SetString(PyExc_OverflowError,
462 "short integer bitfield is less than minimum");
463 return converterr("integer<H>", arg, msgbuf);
Jack Jansend50338f2000-07-06 12:22:00 +0000464 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000465 else if (ival > USHRT_MAX) {
466 PyErr_SetString(PyExc_OverflowError,
467 "short integer bitfield is greater than maximum");
468 return converterr("integer<H>", arg, msgbuf);
469 }
470 else
471 *p = (unsigned short) ival;
472 break;
473 }
Jack Jansend50338f2000-07-06 12:22:00 +0000474
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000475 case 'i': {/* signed int */
476 int *p = va_arg(*p_va, int *);
477 long ival = PyInt_AsLong(arg);
478 if (ival == -1 && PyErr_Occurred())
479 return converterr("integer<i>", arg, msgbuf);
480 else if (ival > INT_MAX) {
481 PyErr_SetString(PyExc_OverflowError,
482 "signed integer is greater than maximum");
483 return converterr("integer<i>", arg, msgbuf);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000484 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000485 else if (ival < INT_MIN) {
486 PyErr_SetString(PyExc_OverflowError,
487 "signed integer is less than minimum");
488 return converterr("integer<i>", arg, msgbuf);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000489 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000490 else
491 *p = ival;
492 break;
493 }
494
495 case 'l': {/* long int */
496 long *p = va_arg(*p_va, long *);
497 long ival = PyInt_AsLong(arg);
498 if (ival == -1 && PyErr_Occurred())
499 return converterr("integer<l>", arg, msgbuf);
500 else
501 *p = ival;
502 break;
503 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000504
Guido van Rossum3dbba6e1999-01-25 21:48:56 +0000505#ifdef HAVE_LONG_LONG
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000506 case 'L': {/* LONG_LONG */
507 LONG_LONG *p = va_arg( *p_va, LONG_LONG * );
508 LONG_LONG ival = PyLong_AsLongLong( arg );
509 if( ival == (LONG_LONG)-1 && PyErr_Occurred() ) {
510 return converterr("long<L>", arg, msgbuf);
511 } else {
512 *p = ival;
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000513 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000514 break;
515 }
Guido van Rossum1a8791e1998-08-04 22:46:29 +0000516#endif
517
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000518 case 'f': {/* float */
519 float *p = va_arg(*p_va, float *);
520 double dval = PyFloat_AsDouble(arg);
521 if (PyErr_Occurred())
522 return converterr("float<f>", arg, msgbuf);
523 else
524 *p = (float) dval;
525 break;
526 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000527
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000528 case 'd': {/* double */
529 double *p = va_arg(*p_va, double *);
530 double dval = PyFloat_AsDouble(arg);
531 if (PyErr_Occurred())
532 return converterr("float<d>", arg, msgbuf);
533 else
534 *p = dval;
535 break;
536 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000537
Guido van Rossum530956d1996-07-21 02:27:43 +0000538#ifndef WITHOUT_COMPLEX
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000539 case 'D': {/* complex double */
540 Py_complex *p = va_arg(*p_va, Py_complex *);
541 Py_complex cval;
542 cval = PyComplex_AsCComplex(arg);
543 if (PyErr_Occurred())
544 return converterr("complex<D>", arg, msgbuf);
545 else
546 *p = cval;
547 break;
548 }
Guido van Rossum530956d1996-07-21 02:27:43 +0000549#endif /* WITHOUT_COMPLEX */
Guido van Rossum8a5c5d21996-01-12 01:09:56 +0000550
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000551 case 'c': {/* char */
552 char *p = va_arg(*p_va, char *);
553 if (PyString_Check(arg) && PyString_Size(arg) == 1)
554 *p = PyString_AsString(arg)[0];
555 else
556 return converterr("char", arg, msgbuf);
557 break;
558 }
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000559
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000560 case 's': {/* string */
561 if (*format == '#') {
562 void **p = (void **)va_arg(*p_va, char **);
563 int *q = va_arg(*p_va, int *);
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000564
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000565 if (PyString_Check(arg)) {
566 *p = PyString_AS_STRING(arg);
567 *q = PyString_GET_SIZE(arg);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000568 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000569 else if (PyUnicode_Check(arg)) {
570 arg = UNICODE_DEFAULT_ENCODING(arg);
571 if (arg == NULL)
572 return converterr(CONV_UNICODE,
573 arg, msgbuf);
574 *p = PyString_AS_STRING(arg);
575 *q = PyString_GET_SIZE(arg);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000576 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000577 else { /* any buffer-like object */
578 char *buf;
579 int count = convertbuffer(arg, p, &buf);
580 if (count < 0)
581 return converterr(buf, arg, msgbuf);
582 *q = count;
583 }
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000584 format++;
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000585 } else {
586 char **p = va_arg(*p_va, char **);
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000587
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000588 if (PyString_Check(arg))
589 *p = PyString_AS_STRING(arg);
590 else if (PyUnicode_Check(arg)) {
591 arg = UNICODE_DEFAULT_ENCODING(arg);
592 if (arg == NULL)
593 return converterr(CONV_UNICODE,
594 arg, msgbuf);
595 *p = PyString_AS_STRING(arg);
Marc-André Lemburg6f15e572001-05-02 17:16:16 +0000596 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000597 else
598 return converterr("string", arg, msgbuf);
599 if ((int)strlen(*p) != PyString_Size(arg))
600 return converterr("string without null bytes",
601 arg, msgbuf);
602 }
603 break;
604 }
605
606 case 'z': {/* string, may be NULL (None) */
607 if (*format == '#') { /* any buffer-like object */
608 void **p = (void **)va_arg(*p_va, char **);
609 int *q = va_arg(*p_va, int *);
610
611 if (arg == Py_None) {
612 *p = 0;
613 *q = 0;
614 }
615 else if (PyString_Check(arg)) {
616 *p = PyString_AS_STRING(arg);
617 *q = PyString_GET_SIZE(arg);
618 }
619 else if (PyUnicode_Check(arg)) {
620 arg = UNICODE_DEFAULT_ENCODING(arg);
621 if (arg == NULL)
622 return converterr(CONV_UNICODE,
623 arg, msgbuf);
624 *p = PyString_AS_STRING(arg);
625 *q = PyString_GET_SIZE(arg);
626 }
627 else { /* any buffer-like object */
628 char *buf;
629 int count = convertbuffer(arg, p, &buf);
630
631 if (count < 0)
632 return converterr(buf, arg, msgbuf);
633 *q = count;
634 }
635 format++;
636 } else {
637 char **p = va_arg(*p_va, char **);
638
639 if (arg == Py_None)
640 *p = 0;
641 else if (PyString_Check(arg))
642 *p = PyString_AsString(arg);
643 else if (PyUnicode_Check(arg)) {
644 arg = UNICODE_DEFAULT_ENCODING(arg);
645 if (arg == NULL)
646 return converterr(CONV_UNICODE,
647 arg, msgbuf);
648 *p = PyString_AS_STRING(arg);
649 }
650 else
651 return converterr("string or None",
652 arg, msgbuf);
653 if (*format == '#') {
654 int *q = va_arg(*p_va, int *);
655 if (arg == Py_None)
656 *q = 0;
657 else
658 *q = PyString_Size(arg);
659 format++;
660 }
661 else if (*p != NULL &&
662 (int)strlen(*p) != PyString_Size(arg))
663 return converterr(
664 "string without null bytes or None",
665 arg, msgbuf);
666 }
667 break;
668 }
669
670 case 'e': {/* encoded string */
671 char **buffer;
672 const char *encoding;
673 PyObject *u, *s;
674 int size, recode_strings;
675
676 /* Get 'e' parameter: the encoding name */
677 encoding = (const char *)va_arg(*p_va, const char *);
678 if (encoding == NULL)
679 encoding = PyUnicode_GetDefaultEncoding();
680
681 /* Get output buffer parameter:
682 's' (recode all objects via Unicode) or
683 't' (only recode non-string objects)
684 */
685 if (*format == 's')
686 recode_strings = 1;
687 else if (*format == 't')
688 recode_strings = 0;
689 else
690 return converterr(
691 "(unknown parser marker combination)",
692 arg, msgbuf);
693 buffer = (char **)va_arg(*p_va, char **);
694 format++;
695 if (buffer == NULL)
696 return converterr("(buffer is NULL)",
697 arg, msgbuf);
698
699 /* Encode object */
700 if (!recode_strings && PyString_Check(arg)) {
701 s = arg;
702 Py_INCREF(s);
703 }
704 else {
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000705 /* Convert object to Unicode */
706 u = PyUnicode_FromObject(arg);
707 if (u == NULL)
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000708 return converterr(
709 "string or unicode or text buffer",
710 arg, msgbuf);
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000711
712 /* Encode object; use default error handling */
713 s = PyUnicode_AsEncodedString(u,
714 encoding,
715 NULL);
716 Py_DECREF(u);
717 if (s == NULL)
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000718 return converterr("(encoding failed)",
719 arg, msgbuf);
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000720 if (!PyString_Check(s)) {
721 Py_DECREF(s);
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000722 return converterr(
723 "(encoder failed to return a string)",
724 arg, msgbuf);
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000725 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000726 }
727 size = PyString_GET_SIZE(s);
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000728
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000729 /* Write output; output is guaranteed to be 0-terminated */
730 if (*format == '#') {
731 /* Using buffer length parameter '#':
732
733 - if *buffer is NULL, a new buffer of the
734 needed size is allocated and the data
735 copied into it; *buffer is updated to point
736 to the new buffer; the caller is
737 responsible for PyMem_Free()ing it after
738 usage
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000739
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000740 - if *buffer is not NULL, the data is
741 copied to *buffer; *buffer_len has to be
742 set to the size of the buffer on input;
743 buffer overflow is signalled with an error;
744 buffer has to provide enough room for the
745 encoded string plus the trailing 0-byte
746
747 - in both cases, *buffer_len is updated to
748 the size of the buffer /excluding/ the
749 trailing 0-byte
750
751 */
752 int *buffer_len = va_arg(*p_va, int *);
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000753
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000754 format++;
755 if (buffer_len == NULL)
756 return converterr(
757 "(buffer_len is NULL)",
758 arg, msgbuf);
759 if (*buffer == NULL) {
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000760 *buffer = PyMem_NEW(char, size + 1);
761 if (*buffer == NULL) {
762 Py_DECREF(s);
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000763 return converterr(
764 "(memory error)",
765 arg, msgbuf);
Guido van Rossumd8855fd2000-03-24 22:14:19 +0000766 }
Fred Drake25871c02000-05-03 15:17:02 +0000767 } else {
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000768 if (size + 1 > *buffer_len) {
769 Py_DECREF(s);
770 return converterr(
771 "(buffer overflow)",
772 arg, msgbuf);
773 }
Fred Drake25871c02000-05-03 15:17:02 +0000774 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000775 memcpy(*buffer,
776 PyString_AS_STRING(s),
777 size + 1);
778 *buffer_len = size;
779 } else {
780 /* Using a 0-terminated buffer:
781
782 - the encoded string has to be 0-terminated
783 for this variant to work; if it is not, an
784 error raised
Fred Drake25871c02000-05-03 15:17:02 +0000785
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000786 - a new buffer of the needed size is
787 allocated and the data copied into it;
788 *buffer is updated to point to the new
789 buffer; the caller is responsible for
790 PyMem_Free()ing it after usage
791
792 */
793 if ((int)strlen(PyString_AS_STRING(s)) != size)
794 return converterr(
795 "(encoded string without NULL bytes)",
796 arg, msgbuf);
797 *buffer = PyMem_NEW(char, size + 1);
798 if (*buffer == NULL) {
799 Py_DECREF(s);
800 return converterr("(memory error)",
801 arg, msgbuf);
802 }
803 memcpy(*buffer,
804 PyString_AS_STRING(s),
805 size + 1);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000806 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000807 Py_DECREF(s);
808 break;
809 }
810
811 case 'u': {/* raw unicode buffer (Py_UNICODE *) */
812 if (*format == '#') { /* any buffer-like object */
813 void **p = (void **)va_arg(*p_va, char **);
814 int *q = va_arg(*p_va, int *);
815 char *buf;
816 int count = convertbuffer(arg, p, &buf);
817
818 if (count < 0)
819 return converterr(buf, arg, msgbuf);
820 *q = count/(sizeof(Py_UNICODE));
821 format++;
822 } else {
823 Py_UNICODE **p = va_arg(*p_va, Py_UNICODE **);
824
Guido van Rossume826ef02000-03-10 23:02:17 +0000825 if (PyUnicode_Check(arg))
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000826 *p = PyUnicode_AS_UNICODE(arg);
827 else
828 return converterr("unicode", arg, msgbuf);
829 }
830 break;
831 }
832
833 case 'S': { /* string object */
834 PyObject **p = va_arg(*p_va, PyObject **);
835 if (PyString_Check(arg))
836 *p = arg;
837 else
838 return converterr("string", arg, msgbuf);
839 break;
840 }
841
842 case 'U': { /* Unicode object */
843 PyObject **p = va_arg(*p_va, PyObject **);
844 if (PyUnicode_Check(arg))
845 *p = arg;
846 else
847 return converterr("unicode", arg, msgbuf);
848 break;
849 }
850
851 case 'O': { /* object */
852 PyTypeObject *type;
853 PyObject **p;
854 if (*format == '!') {
855 type = va_arg(*p_va, PyTypeObject*);
856 p = va_arg(*p_va, PyObject **);
857 format++;
858 if (arg->ob_type == type)
Guido van Rossume826ef02000-03-10 23:02:17 +0000859 *p = arg;
860 else
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000861 return converterr(type->tp_name, arg, msgbuf);
Guido van Rossumfccfe891998-05-15 22:04:07 +0000862
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000863 }
864 else if (*format == '?') {
865 inquiry pred = va_arg(*p_va, inquiry);
866 p = va_arg(*p_va, PyObject **);
867 format++;
868 if ((*pred)(arg))
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000869 *p = arg;
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000870 else
871 return converterr("(unspecified)",
872 arg, msgbuf);
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000873
Guido van Rossumfdf95dd1997-05-05 22:15:02 +0000874 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000875 else if (*format == '&') {
876 typedef int (*converter)(PyObject *, void *);
877 converter convert = va_arg(*p_va, converter);
878 void *addr = va_arg(*p_va, void *);
879 format++;
880 if (! (*convert)(arg, addr))
881 return converterr("(unspecified)",
882 arg, msgbuf);
Guido van Rossumb317f8a1998-10-08 02:21:21 +0000883 }
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000884 else {
885 p = va_arg(*p_va, PyObject **);
886 *p = arg;
887 }
888 break;
889 }
Guido van Rossumb317f8a1998-10-08 02:21:21 +0000890
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000891
892 case 'w': { /* memory buffer, read-write access */
893 void **p = va_arg(*p_va, void **);
894 PyBufferProcs *pb = arg->ob_type->tp_as_buffer;
895 int count;
896
897 if (pb == NULL ||
898 pb->bf_getwritebuffer == NULL ||
899 pb->bf_getsegcount == NULL)
900 return converterr("read-write buffer", arg, msgbuf);
901 if ((*pb->bf_getsegcount)(arg, NULL) != 1)
902 return converterr("single-segment read-write buffer",
903 arg, msgbuf);
904 if ((count = pb->bf_getwritebuffer(arg, 0, p)) < 0)
905 return converterr("(unspecified)", arg, msgbuf);
906 if (*format == '#') {
907 int *q = va_arg(*p_va, int *);
908
909 *q = count;
910 format++;
911 }
912 break;
913 }
914
915 case 't': { /* 8-bit character buffer, read-only access */
916 const char **p = va_arg(*p_va, const char **);
917 char *buf;
918 int count;
919
920 if (*format++ != '#')
921 return converterr(
922 "invalid use of 't' format character",
923 arg, msgbuf);
924 if (!PyType_HasFeature(arg->ob_type,
925 Py_TPFLAGS_HAVE_GETCHARBUFFER))
926 return converterr(
927 "string or read-only character buffer",
928 arg, msgbuf);
929
930 count = convertbuffer(arg, (void **)p, &buf);
931 if (count < 0)
932 return converterr(buf, arg, msgbuf);
933 *va_arg(*p_va, int *) = count;
934 break;
935 }
936
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000937 default:
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000938 return converterr("impossible<bad format char>", arg, msgbuf);
Guido van Rossumfe3f1a21994-09-29 09:42:55 +0000939
940 }
941
942 *p_format = format;
943 return NULL;
944}
Guido van Rossumaa354651996-08-19 19:32:04 +0000945
Jeremy Hylton1cb7aa32001-05-29 17:37:05 +0000946int convertbuffer(PyObject *arg, void **p, char **errmsg)
947{
948 PyBufferProcs *pb = arg->ob_type->tp_as_buffer;
949 int count;
950 if (pb == NULL ||
951 pb->bf_getreadbuffer == NULL ||
952 pb->bf_getsegcount == NULL) {
953 *errmsg = "string or read-only buffer";
954 return -1;
955 }
956 if ((*pb->bf_getsegcount)(arg, NULL) != 1) {
957 *errmsg = "string or single-segment read-only buffer";
958 return -1;
959 }
960 if ((count = (*pb->bf_getreadbuffer)(arg, 0, p)) < 0) {
961 *errmsg = "(unspecified)";
962 }
963 return count;
964}
Guido van Rossumaa354651996-08-19 19:32:04 +0000965
966/* Support for keyword arguments donated by
967 Geoff Philbrick <philbric@delphi.hks.com> */
968
Guido van Rossum79f25d91997-04-29 20:08:16 +0000969int PyArg_ParseTupleAndKeywords(PyObject *args,
970 PyObject *keywords,
Guido van Rossumaa354651996-08-19 19:32:04 +0000971 char *format,
972 char **kwlist, ...)
Guido van Rossumaa354651996-08-19 19:32:04 +0000973{
974 int retval;
975 va_list va;
Guido van Rossumaa354651996-08-19 19:32:04 +0000976
977 va_start(va, kwlist);
Guido van Rossumaa354651996-08-19 19:32:04 +0000978 retval = vgetargskeywords(args, keywords, format, kwlist, &va);
979 va_end(va);
980 return retval;
981}
982
983
984static int
Thomas Woutersf70ef4f2000-07-22 18:47:25 +0000985vgetargskeywords(PyObject *args, PyObject *keywords, char *format,
986 char **kwlist, va_list *p_va)
Guido van Rossumaa354651996-08-19 19:32:04 +0000987{
988 char msgbuf[256];
989 int levels[32];
990 char *fname = NULL;
991 char *message = NULL;
992 int min = -1;
993 int max = 0;
Guido van Rossumaa354651996-08-19 19:32:04 +0000994 char *formatsave = format;
995 int i, len, tplen, kwlen;
996 char *msg, *ks, **p;
997 int nkwds, pos, match, converted;
Guido van Rossum79f25d91997-04-29 20:08:16 +0000998 PyObject *key, *value;
Guido van Rossumaa354651996-08-19 19:32:04 +0000999
1000 /* nested tuples cannot be parsed when using keyword arguments */
1001
1002 for (;;) {
1003 int c = *format++;
1004 if (c == '(') {
1005 PyErr_SetString(PyExc_SystemError,
1006 "tuple found in format when using keyword arguments");
1007 return 0;
1008 }
1009 else if (c == '\0')
1010 break;
1011 else if (c == ':') {
1012 fname = format;
1013 break;
Jeremy Hylton0f8117f2001-05-18 20:57:38 +00001014 } else if (c == ';') {
Guido van Rossumaa354651996-08-19 19:32:04 +00001015 message = format;
1016 break;
Jeremy Hylton0f8117f2001-05-18 20:57:38 +00001017 } else if (c == 'e')
Marc-André Lemburgbbcf2a72000-09-08 11:49:37 +00001018 ; /* Pass */
Guido van Rossumaa354651996-08-19 19:32:04 +00001019 else if (isalpha(c))
1020 max++;
1021 else if (c == '|')
1022 min = max;
1023 }
1024
1025 if (min < 0)
1026 min = max;
1027
1028 format = formatsave;
1029
1030 if (!PyTuple_Check(args)) {
1031 PyErr_SetString(PyExc_SystemError,
1032 "new style getargs format but argument is not a tuple");
1033 return 0;
1034 }
1035
Jeremy Hylton0f8117f2001-05-18 20:57:38 +00001036 tplen = PyTuple_GET_SIZE(args);
Guido van Rossumaa354651996-08-19 19:32:04 +00001037
1038 /* do a cursory check of the keywords just to see how many we got */
1039
1040 if (keywords) {
1041 if (!PyDict_Check(keywords)) {
Jeremy Hyltona0ac40c2001-01-25 20:13:10 +00001042 if (keywords == NULL)
1043 PyErr_SetString(PyExc_SystemError,
1044 "NULL received when keyword dictionary expected");
1045 else
1046 PyErr_Format(PyExc_SystemError,
1047 "%s received when keyword dictionary expected",
1048 keywords->ob_type->tp_name);
Guido van Rossumaa354651996-08-19 19:32:04 +00001049 return 0;
1050 }
1051 kwlen = PyDict_Size(keywords);
1052 }
1053 else {
1054 kwlen = 0;
1055 }
1056
1057 /* make sure there are no duplicate values for an argument;
1058 its not clear when to use the term "keyword argument vs.
1059 keyword parameter in messages */
1060
1061 if (keywords) {
1062 for (i = 0; i < tplen; i++) {
1063 if (PyMapping_HasKeyString(keywords, kwlist[i])) {
1064 sprintf(msgbuf,
1065 "keyword parameter %s redefined",
1066 kwlist[i]);
1067 PyErr_SetString(PyExc_TypeError, msgbuf);
1068 return 0;
1069 }
1070 }
1071 }
1072 PyErr_Clear(); /* I'm not which Py functions set the error string */
1073
1074 /* required arguments missing from args can be supplied by keyword
1075 arguments */
1076
1077 len = tplen;
1078 if (keywords && tplen < min) {
1079 for (i = tplen; i < min; i++) {
1080 if (PyMapping_HasKeyString(keywords, kwlist[i])) {
1081 len++;
1082 }
1083 }
1084 }
1085 PyErr_Clear();
1086
1087 /* make sure we got an acceptable number of arguments; the message
1088 is a little confusing with keywords since keyword arguments
1089 which are supplied, but don't match the required arguments
1090 are not included in the "%d given" part of the message */
1091
1092 if (len < min || max < len) {
1093 if (message == NULL) {
1094 sprintf(msgbuf,
Ka-Ping Yee20579702001-01-15 22:14:16 +00001095 "%s%s takes %s %d argument%s (%d given)",
Guido van Rossumaa354651996-08-19 19:32:04 +00001096 fname==NULL ? "function" : fname,
Ka-Ping Yee20579702001-01-15 22:14:16 +00001097 fname==NULL ? "" : "()",
Guido van Rossumaa354651996-08-19 19:32:04 +00001098 min==max ? "exactly"
1099 : len < min ? "at least" : "at most",
1100 len < min ? min : max,
1101 (len < min ? min : max) == 1 ? "" : "s",
1102 len);
1103 message = msgbuf;
1104 }
1105 PyErr_SetString(PyExc_TypeError, message);
1106 return 0;
1107 }
1108
1109 for (i = 0; i < tplen; i++) {
1110 if (*format == '|')
1111 format++;
Jeremy Hylton0f8117f2001-05-18 20:57:38 +00001112 msg = convertitem(PyTuple_GET_ITEM(args, i), &format, p_va,
Guido van Rossumaa354651996-08-19 19:32:04 +00001113 levels, msgbuf);
1114 if (msg) {
1115 seterror(i+1, msg, levels, fname, message);
1116 return 0;
1117 }
1118 }
1119
1120 /* handle no keyword parameters in call */
1121
1122 if (!keywords) return 1;
1123
1124 /* make sure the number of keywords in the keyword list matches the
1125 number of items in the format string */
1126
1127 nkwds = 0;
1128 p = kwlist;
1129 for (;;) {
1130 if (!*(p++)) break;
1131 nkwds++;
1132 }
1133
1134 if (nkwds != max) {
1135 PyErr_SetString(PyExc_SystemError,
1136 "number of items in format string and keyword list do not match");
1137 return 0;
1138 }
1139
1140 /* convert the keyword arguments; this uses the format
1141 string where it was left after processing args */
1142
1143 converted = 0;
1144 for (i = tplen; i < nkwds; i++) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001145 PyObject *item;
Guido van Rossumaa354651996-08-19 19:32:04 +00001146 if (*format == '|')
1147 format++;
Guido van Rossum80bb9651996-12-05 23:27:02 +00001148 item = PyMapping_GetItemString(keywords, kwlist[i]);
1149 if (item != NULL) {
Guido van Rossumaa354651996-08-19 19:32:04 +00001150 msg = convertitem(item, &format, p_va, levels, msgbuf);
1151 if (msg) {
1152 seterror(i+1, msg, levels, fname, message);
1153 return 0;
1154 }
1155 converted++;
Barry Warsaw07050282000-12-11 20:01:55 +00001156 Py_DECREF(item);
Guido van Rossumaa354651996-08-19 19:32:04 +00001157 }
1158 else {
1159 PyErr_Clear();
1160 msg = skipitem(&format, p_va);
1161 if (msg) {
1162 seterror(i+1, msg, levels, fname, message);
1163 return 0;
1164 }
1165 }
1166 }
1167
1168 /* make sure there are no extraneous keyword arguments */
1169
1170 pos = 0;
1171 if (converted < kwlen) {
1172 while (PyDict_Next(keywords, &pos, &key, &value)) {
1173 match = 0;
1174 ks = PyString_AsString(key);
1175 for (i = 0; i < nkwds; i++) {
1176 if (!strcmp(ks, kwlist[i])) {
1177 match = 1;
1178 break;
1179 }
1180 }
1181 if (!match) {
1182 sprintf(msgbuf,
Guido van Rossum80dc16b2000-05-08 14:02:41 +00001183 "%s is an invalid keyword argument for this function",
Guido van Rossumaa354651996-08-19 19:32:04 +00001184 ks);
1185 PyErr_SetString(PyExc_TypeError, msgbuf);
1186 return 0;
1187 }
1188 }
1189 }
1190
1191 return 1;
1192}
1193
1194
1195static char *
Thomas Woutersf70ef4f2000-07-22 18:47:25 +00001196skipitem(char **p_format, va_list *p_va)
Guido van Rossumaa354651996-08-19 19:32:04 +00001197{
1198 char *format = *p_format;
1199 char c = *format++;
1200
1201 switch (c) {
1202
1203 case 'b': /* byte -- very short int */
Jack Jansencc22fbe2000-08-05 21:29:58 +00001204 case 'B': /* byte as bitfield */
Guido van Rossumaa354651996-08-19 19:32:04 +00001205 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001206 (void) va_arg(*p_va, char *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001207 break;
1208 }
1209
1210 case 'h': /* short int */
1211 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001212 (void) va_arg(*p_va, short *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001213 break;
1214 }
1215
Jack Jansencc22fbe2000-08-05 21:29:58 +00001216 case 'H': /* short int as bitfield */
Jack Jansend50338f2000-07-06 12:22:00 +00001217 {
1218 (void) va_arg(*p_va, unsigned short *);
1219 break;
1220 }
1221
Guido van Rossumaa354651996-08-19 19:32:04 +00001222 case 'i': /* int */
1223 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001224 (void) va_arg(*p_va, int *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001225 break;
1226 }
1227
1228 case 'l': /* long int */
1229 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001230 (void) va_arg(*p_va, long *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001231 break;
1232 }
1233
Guido van Rossum3dbba6e1999-01-25 21:48:56 +00001234#ifdef HAVE_LONG_LONG
Guido van Rossum3293b071998-08-25 16:07:15 +00001235 case 'L': /* LONG_LONG int */
Guido van Rossum1a8791e1998-08-04 22:46:29 +00001236 {
Guido van Rossum3293b071998-08-25 16:07:15 +00001237 (void) va_arg(*p_va, LONG_LONG *);
Guido van Rossum1a8791e1998-08-04 22:46:29 +00001238 break;
1239 }
1240#endif
1241
Guido van Rossumaa354651996-08-19 19:32:04 +00001242 case 'f': /* float */
1243 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001244 (void) va_arg(*p_va, float *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001245 break;
1246 }
1247
1248 case 'd': /* double */
1249 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001250 (void) va_arg(*p_va, double *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001251 break;
1252 }
1253
1254#ifndef WITHOUT_COMPLEX
1255 case 'D': /* complex double */
1256 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001257 (void) va_arg(*p_va, Py_complex *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001258 break;
1259 }
1260#endif /* WITHOUT_COMPLEX */
1261
1262 case 'c': /* char */
1263 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001264 (void) va_arg(*p_va, char *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001265 break;
1266 }
1267
1268 case 's': /* string */
1269 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001270 (void) va_arg(*p_va, char **);
Guido van Rossumaa354651996-08-19 19:32:04 +00001271 if (*format == '#') {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001272 (void) va_arg(*p_va, int *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001273 format++;
1274 }
1275 break;
1276 }
1277
1278 case 'z': /* string */
1279 {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001280 (void) va_arg(*p_va, char **);
Guido van Rossumaa354651996-08-19 19:32:04 +00001281 if (*format == '#') {
Guido van Rossum80bb9651996-12-05 23:27:02 +00001282 (void) va_arg(*p_va, int *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001283 format++;
1284 }
1285 break;
1286 }
1287
1288 case 'S': /* string object */
1289 {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001290 (void) va_arg(*p_va, PyObject **);
Guido van Rossumaa354651996-08-19 19:32:04 +00001291 break;
1292 }
1293
1294 case 'O': /* object */
1295 {
Guido van Rossumaa354651996-08-19 19:32:04 +00001296 if (*format == '!') {
1297 format++;
Guido van Rossum79f25d91997-04-29 20:08:16 +00001298 (void) va_arg(*p_va, PyTypeObject*);
1299 (void) va_arg(*p_va, PyObject **);
Guido van Rossumaa354651996-08-19 19:32:04 +00001300 }
1301#if 0
1302/* I don't know what this is for */
1303 else if (*format == '?') {
1304 inquiry pred = va_arg(*p_va, inquiry);
1305 format++;
1306 if ((*pred)(arg)) {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001307 (void) va_arg(*p_va, PyObject **);
Guido van Rossumaa354651996-08-19 19:32:04 +00001308 }
1309 }
1310#endif
1311 else if (*format == '&') {
Tim Petersdbd9ba62000-07-09 03:09:57 +00001312 typedef int (*converter)(PyObject *, void *);
Guido van Rossum80bb9651996-12-05 23:27:02 +00001313 (void) va_arg(*p_va, converter);
1314 (void) va_arg(*p_va, void *);
Guido van Rossumaa354651996-08-19 19:32:04 +00001315 format++;
1316 }
1317 else {
Guido van Rossum79f25d91997-04-29 20:08:16 +00001318 (void) va_arg(*p_va, PyObject **);
Guido van Rossumaa354651996-08-19 19:32:04 +00001319 }
1320 break;
1321 }
1322
1323 default:
1324 return "impossible<bad format char>";
1325
1326 }
1327
1328 *p_format = format;
1329 return NULL;
1330}