2 /* New getargs implementation */
12 int PyArg_Parse(PyObject
*, const char *, ...);
13 int PyArg_ParseTuple(PyObject
*, const char *, ...);
14 int PyArg_VaParse(PyObject
*, const char *, va_list);
16 int PyArg_ParseTupleAndKeywords(PyObject
*, PyObject
*,
17 const char *, char **, ...);
18 int PyArg_VaParseTupleAndKeywords(PyObject
*, PyObject
*,
19 const char *, char **, va_list);
21 #ifdef HAVE_DECLSPEC_DLL
22 /* Export functions */
23 PyAPI_FUNC(int) _PyArg_Parse_SizeT(PyObject
*, char *, ...);
24 PyAPI_FUNC(int) _PyArg_ParseTuple_SizeT(PyObject
*, char *, ...);
25 PyAPI_FUNC(int) _PyArg_ParseTupleAndKeywords_SizeT(PyObject
*, PyObject
*,
26 const char *, char **, ...);
27 PyAPI_FUNC(PyObject
*) _Py_BuildValue_SizeT(const char *, ...);
28 PyAPI_FUNC(int) _PyArg_VaParse_SizeT(PyObject
*, char *, va_list);
29 PyAPI_FUNC(int) _PyArg_VaParseTupleAndKeywords_SizeT(PyObject
*, PyObject
*,
30 const char *, char **, va_list);
38 static int vgetargs1(PyObject
*, const char *, va_list *, int);
39 static void seterror(int, const char *, int *, const char *, const char *);
40 static char *convertitem(PyObject
*, const char **, va_list *, int, int *,
41 char *, size_t, PyObject
**);
42 static char *converttuple(PyObject
*, const char **, va_list *, int,
43 int *, char *, size_t, int, PyObject
**);
44 static char *convertsimple(PyObject
*, const char **, va_list *, int, char *,
46 static Py_ssize_t
convertbuffer(PyObject
*, void **p
, char **);
47 static int getbuffer(PyObject
*, Py_buffer
*, char**);
49 static int vgetargskeywords(PyObject
*, PyObject
*,
50 const char *, char **, va_list *, int);
51 static char *skipitem(const char **, va_list *, int);
54 PyArg_Parse(PyObject
*args
, const char *format
, ...)
60 retval
= vgetargs1(args
, format
, &va
, FLAG_COMPAT
);
66 _PyArg_Parse_SizeT(PyObject
*args
, char *format
, ...)
72 retval
= vgetargs1(args
, format
, &va
, FLAG_COMPAT
|FLAG_SIZE_T
);
79 PyArg_ParseTuple(PyObject
*args
, const char *format
, ...)
85 retval
= vgetargs1(args
, format
, &va
, 0);
91 _PyArg_ParseTuple_SizeT(PyObject
*args
, char *format
, ...)
97 retval
= vgetargs1(args
, format
, &va
, FLAG_SIZE_T
);
104 PyArg_VaParse(PyObject
*args
, const char *format
, va_list va
)
108 #ifdef VA_LIST_IS_ARRAY
109 memcpy(lva
, va
, sizeof(va_list));
118 return vgetargs1(args
, format
, &lva
, 0);
122 _PyArg_VaParse_SizeT(PyObject
*args
, char *format
, va_list va
)
126 #ifdef VA_LIST_IS_ARRAY
127 memcpy(lva
, va
, sizeof(va_list));
136 return vgetargs1(args
, format
, &lva
, FLAG_SIZE_T
);
140 /* Handle cleanup of allocated memory in case of exception */
142 #define GETARGS_CAPSULE_NAME_CLEANUP_PTR "getargs.cleanup_ptr"
143 #define GETARGS_CAPSULE_NAME_CLEANUP_BUFFER "getargs.cleanup_buffer"
146 cleanup_ptr(PyObject
*self
)
148 void *ptr
= PyCapsule_GetPointer(self
, GETARGS_CAPSULE_NAME_CLEANUP_PTR
);
155 cleanup_buffer(PyObject
*self
)
157 Py_buffer
*ptr
= (Py_buffer
*)PyCapsule_GetPointer(self
, GETARGS_CAPSULE_NAME_CLEANUP_BUFFER
);
159 PyBuffer_Release(ptr
);
164 addcleanup(void *ptr
, PyObject
**freelist
, PyCapsule_Destructor destr
)
170 *freelist
= PyList_New(0);
177 if (destr
== cleanup_ptr
) {
178 name
= GETARGS_CAPSULE_NAME_CLEANUP_PTR
;
179 } else if (destr
== cleanup_buffer
) {
180 name
= GETARGS_CAPSULE_NAME_CLEANUP_BUFFER
;
184 cobj
= PyCapsule_New(ptr
, name
, destr
);
189 if (PyList_Append(*freelist
, cobj
)) {
198 cleanreturn(int retval
, PyObject
*freelist
)
200 if (freelist
&& retval
!= 0) {
201 /* We were successful, reset the destructors so that they
203 Py_ssize_t len
= PyList_GET_SIZE(freelist
), i
;
204 for (i
= 0; i
< len
; i
++)
205 PyCapsule_SetDestructor(PyList_GET_ITEM(freelist
, i
), NULL
);
207 Py_XDECREF(freelist
);
213 vgetargs1(PyObject
*args
, const char *format
, va_list *p_va
, int flags
)
217 const char *fname
= NULL
;
218 const char *message
= NULL
;
223 const char *formatsave
= format
;
226 PyObject
*freelist
= NULL
;
227 int compat
= flags
& FLAG_COMPAT
;
229 assert(compat
|| (args
!= (PyObject
*)NULL
));
230 flags
= flags
& ~FLAG_COMPAT
;
232 while (endfmt
== 0) {
240 Py_FatalError("too many tuple nesting levels "
241 "in argument format string");
245 Py_FatalError("excess ')' in getargs format");
264 else if (isalpha(Py_CHARMASK(c
))) {
265 if (c
!= 'e') /* skip encoded */
275 Py_FatalError(/* '(' */ "missing ')' in getargs format");
286 PyOS_snprintf(msgbuf
, sizeof(msgbuf
),
287 "%.200s%s takes no arguments",
288 fname
==NULL
? "function" : fname
,
289 fname
==NULL
? "" : "()");
290 PyErr_SetString(PyExc_TypeError
, msgbuf
);
293 else if (min
== 1 && max
== 1) {
295 PyOS_snprintf(msgbuf
, sizeof(msgbuf
),
296 "%.200s%s takes at least one argument",
297 fname
==NULL
? "function" : fname
,
298 fname
==NULL
? "" : "()");
299 PyErr_SetString(PyExc_TypeError
, msgbuf
);
302 msg
= convertitem(args
, &format
, p_va
, flags
, levels
,
303 msgbuf
, sizeof(msgbuf
), &freelist
);
305 return cleanreturn(1, freelist
);
306 seterror(levels
[0], msg
, levels
+1, fname
, message
);
307 return cleanreturn(0, freelist
);
310 PyErr_SetString(PyExc_SystemError
,
311 "old style getargs format uses new features");
316 if (!PyTuple_Check(args
)) {
317 PyErr_SetString(PyExc_SystemError
,
318 "new style getargs format but argument is not a tuple");
322 len
= PyTuple_GET_SIZE(args
);
324 if (len
< min
|| max
< len
) {
325 if (message
== NULL
) {
326 PyOS_snprintf(msgbuf
, sizeof(msgbuf
),
327 "%.150s%s takes %s %d argument%s "
329 fname
==NULL
? "function" : fname
,
330 fname
==NULL
? "" : "()",
332 : len
< min
? "at least" : "at most",
333 len
< min
? min
: max
,
334 (len
< min
? min
: max
) == 1 ? "" : "s",
335 Py_SAFE_DOWNCAST(len
, Py_ssize_t
, long));
338 PyErr_SetString(PyExc_TypeError
, message
);
342 for (i
= 0; i
< len
; i
++) {
345 msg
= convertitem(PyTuple_GET_ITEM(args
, i
), &format
, p_va
,
346 flags
, levels
, msgbuf
,
347 sizeof(msgbuf
), &freelist
);
349 seterror(i
+1, msg
, levels
, fname
, msg
);
350 return cleanreturn(0, freelist
);
354 if (*format
!= '\0' && !isalpha(Py_CHARMASK(*format
)) &&
356 *format
!= '|' && *format
!= ':' && *format
!= ';') {
357 PyErr_Format(PyExc_SystemError
,
358 "bad format string: %.200s", formatsave
);
359 return cleanreturn(0, freelist
);
362 return cleanreturn(1, freelist
);
368 seterror(int iarg
, const char *msg
, int *levels
, const char *fname
,
375 if (PyErr_Occurred())
377 else if (message
== NULL
) {
379 PyOS_snprintf(p
, sizeof(buf
), "%.200s() ", fname
);
383 PyOS_snprintf(p
, sizeof(buf
) - (p
- buf
),
384 "argument %d", iarg
);
387 while (levels
[i
] > 0 && i
< 32 && (int)(p
-buf
) < 220) {
388 PyOS_snprintf(p
, sizeof(buf
) - (p
- buf
),
389 ", item %d", levels
[i
]-1);
395 PyOS_snprintf(p
, sizeof(buf
) - (p
- buf
), "argument");
398 PyOS_snprintf(p
, sizeof(buf
) - (p
- buf
), " %.256s", msg
);
401 PyErr_SetString(PyExc_TypeError
, message
);
405 /* Convert a tuple argument.
406 On entry, *p_format points to the character _after_ the opening '('.
407 On successful exit, *p_format points to the closing ')'.
409 *p_format and *p_va are updated,
410 *levels and *msgbuf are untouched,
411 and NULL is returned.
412 If the argument is invalid:
413 *p_format is unchanged,
415 *levels is a 0-terminated list of item numbers,
416 *msgbuf contains an error message, whose format is:
417 "must be <typename1>, not <typename2>", where:
418 <typename1> is the name of the expected type, and
419 <typename2> is the name of the actual type,
420 and msgbuf is returned.
424 converttuple(PyObject
*arg
, const char **p_format
, va_list *p_va
, int flags
,
425 int *levels
, char *msgbuf
, size_t bufsize
, int toplevel
,
430 const char *format
= *p_format
;
445 else if (c
== ':' || c
== ';' || c
== '\0')
447 else if (level
== 0 && isalpha(Py_CHARMASK(c
)))
451 if (!PySequence_Check(arg
) || PyString_Check(arg
)) {
453 PyOS_snprintf(msgbuf
, bufsize
,
454 toplevel
? "expected %d arguments, not %.50s" :
455 "must be %d-item sequence, not %.50s",
457 arg
== Py_None
? "None" : arg
->ob_type
->tp_name
);
461 if ((i
= PySequence_Size(arg
)) != n
) {
463 PyOS_snprintf(msgbuf
, bufsize
,
464 toplevel
? "expected %d arguments, not %d" :
465 "must be sequence of length %d, not %d",
471 for (i
= 0; i
< n
; i
++) {
474 item
= PySequence_GetItem(arg
, i
);
479 strncpy(msgbuf
, "is not retrievable", bufsize
);
482 msg
= convertitem(item
, &format
, p_va
, flags
, levels
+1,
483 msgbuf
, bufsize
, freelist
);
484 /* PySequence_GetItem calls tp->sq_item, which INCREFs */
497 /* Convert a single item. */
500 convertitem(PyObject
*arg
, const char **p_format
, va_list *p_va
, int flags
,
501 int *levels
, char *msgbuf
, size_t bufsize
, PyObject
**freelist
)
504 const char *format
= *p_format
;
506 if (*format
== '(' /* ')' */) {
508 msg
= converttuple(arg
, &format
, p_va
, flags
, levels
, msgbuf
,
509 bufsize
, 0, freelist
);
514 msg
= convertsimple(arg
, &format
, p_va
, flags
,
515 msgbuf
, bufsize
, freelist
);
526 #define UNICODE_DEFAULT_ENCODING(arg) \
527 _PyUnicode_AsDefaultEncodedString(arg, NULL)
529 /* Format an error message generated by convertsimple(). */
532 converterr(const char *expected
, PyObject
*arg
, char *msgbuf
, size_t bufsize
)
534 assert(expected
!= NULL
);
536 PyOS_snprintf(msgbuf
, bufsize
,
537 "must be %.50s, not %.50s", expected
,
538 arg
== Py_None
? "None" : arg
->ob_type
->tp_name
);
542 #define CONV_UNICODE "(unicode conversion error)"
544 /* explicitly check for float arguments when integers are expected. For now
545 * signal a warning. Returns true if an exception was raised. */
547 float_argument_warning(PyObject
*arg
)
549 if (PyFloat_Check(arg
) &&
550 PyErr_Warn(PyExc_DeprecationWarning
,
551 "integer argument expected, got float" ))
557 /* explicitly check for float arguments when integers are expected. Raises
558 TypeError and returns true for float arguments. */
560 float_argument_error(PyObject
*arg
)
562 if (PyFloat_Check(arg
)) {
563 PyErr_SetString(PyExc_TypeError
,
564 "integer argument expected, got float");
571 /* Convert a non-tuple argument. Return NULL if conversion went OK,
572 or a string with a message describing the failure. The message is
573 formatted as "must be <desired type>, not <actual type>".
574 When failing, an exception may or may not have been raised.
575 Don't call if a tuple is expected.
577 When you add new format codes, please don't forget poor skipitem() below.
581 convertsimple(PyObject
*arg
, const char **p_format
, va_list *p_va
, int flags
,
582 char *msgbuf
, size_t bufsize
, PyObject
**freelist
)
585 #define FETCH_SIZE int *q=NULL;Py_ssize_t *q2=NULL;\
586 if (flags & FLAG_SIZE_T) q2=va_arg(*p_va, Py_ssize_t*); \
587 else q=va_arg(*p_va, int*);
588 #define STORE_SIZE(s) if (flags & FLAG_SIZE_T) *q2=s; else *q=s;
589 #define BUFFER_LEN ((flags & FLAG_SIZE_T) ? *q2:*q)
591 const char *format
= *p_format
;
593 #ifdef Py_USING_UNICODE
599 case 'b': { /* unsigned byte -- very short int */
600 char *p
= va_arg(*p_va
, char *);
602 if (float_argument_error(arg
))
603 return converterr("integer<b>", arg
, msgbuf
, bufsize
);
604 ival
= PyInt_AsLong(arg
);
605 if (ival
== -1 && PyErr_Occurred())
606 return converterr("integer<b>", arg
, msgbuf
, bufsize
);
608 PyErr_SetString(PyExc_OverflowError
,
609 "unsigned byte integer is less than minimum");
610 return converterr("integer<b>", arg
, msgbuf
, bufsize
);
612 else if (ival
> UCHAR_MAX
) {
613 PyErr_SetString(PyExc_OverflowError
,
614 "unsigned byte integer is greater than maximum");
615 return converterr("integer<b>", arg
, msgbuf
, bufsize
);
618 *p
= (unsigned char) ival
;
622 case 'B': {/* byte sized bitfield - both signed and unsigned
624 char *p
= va_arg(*p_va
, char *);
626 if (float_argument_error(arg
))
627 return converterr("integer<B>", arg
, msgbuf
, bufsize
);
628 ival
= PyInt_AsUnsignedLongMask(arg
);
629 if (ival
== -1 && PyErr_Occurred())
630 return converterr("integer<B>", arg
, msgbuf
, bufsize
);
632 *p
= (unsigned char) ival
;
636 case 'h': {/* signed short int */
637 short *p
= va_arg(*p_va
, short *);
639 if (float_argument_error(arg
))
640 return converterr("integer<h>", arg
, msgbuf
, bufsize
);
641 ival
= PyInt_AsLong(arg
);
642 if (ival
== -1 && PyErr_Occurred())
643 return converterr("integer<h>", arg
, msgbuf
, bufsize
);
644 else if (ival
< SHRT_MIN
) {
645 PyErr_SetString(PyExc_OverflowError
,
646 "signed short integer is less than minimum");
647 return converterr("integer<h>", arg
, msgbuf
, bufsize
);
649 else if (ival
> SHRT_MAX
) {
650 PyErr_SetString(PyExc_OverflowError
,
651 "signed short integer is greater than maximum");
652 return converterr("integer<h>", arg
, msgbuf
, bufsize
);
659 case 'H': { /* short int sized bitfield, both signed and
661 unsigned short *p
= va_arg(*p_va
, unsigned short *);
663 if (float_argument_error(arg
))
664 return converterr("integer<H>", arg
, msgbuf
, bufsize
);
665 ival
= PyInt_AsUnsignedLongMask(arg
);
666 if (ival
== -1 && PyErr_Occurred())
667 return converterr("integer<H>", arg
, msgbuf
, bufsize
);
669 *p
= (unsigned short) ival
;
673 case 'i': {/* signed int */
674 int *p
= va_arg(*p_va
, int *);
676 if (float_argument_error(arg
))
677 return converterr("integer<i>", arg
, msgbuf
, bufsize
);
678 ival
= PyInt_AsLong(arg
);
679 if (ival
== -1 && PyErr_Occurred())
680 return converterr("integer<i>", arg
, msgbuf
, bufsize
);
681 else if (ival
> INT_MAX
) {
682 PyErr_SetString(PyExc_OverflowError
,
683 "signed integer is greater than maximum");
684 return converterr("integer<i>", arg
, msgbuf
, bufsize
);
686 else if (ival
< INT_MIN
) {
687 PyErr_SetString(PyExc_OverflowError
,
688 "signed integer is less than minimum");
689 return converterr("integer<i>", arg
, msgbuf
, bufsize
);
696 case 'I': { /* int sized bitfield, both signed and
698 unsigned int *p
= va_arg(*p_va
, unsigned int *);
700 if (float_argument_error(arg
))
701 return converterr("integer<I>", arg
, msgbuf
, bufsize
);
702 ival
= (unsigned int)PyInt_AsUnsignedLongMask(arg
);
703 if (ival
== (unsigned int)-1 && PyErr_Occurred())
704 return converterr("integer<I>", arg
, msgbuf
, bufsize
);
710 case 'n': /* Py_ssize_t */
711 #if SIZEOF_SIZE_T != SIZEOF_LONG
713 Py_ssize_t
*p
= va_arg(*p_va
, Py_ssize_t
*);
715 if (float_argument_error(arg
))
716 return converterr("integer<n>", arg
, msgbuf
, bufsize
);
717 ival
= PyInt_AsSsize_t(arg
);
718 if (ival
== -1 && PyErr_Occurred())
719 return converterr("integer<n>", arg
, msgbuf
, bufsize
);
724 /* Fall through from 'n' to 'l' if Py_ssize_t is int */
725 case 'l': {/* long int */
726 long *p
= va_arg(*p_va
, long *);
728 if (float_argument_error(arg
))
729 return converterr("integer<l>", arg
, msgbuf
, bufsize
);
730 ival
= PyInt_AsLong(arg
);
731 if (ival
== -1 && PyErr_Occurred())
732 return converterr("integer<l>", arg
, msgbuf
, bufsize
);
738 case 'k': { /* long sized bitfield */
739 unsigned long *p
= va_arg(*p_va
, unsigned long *);
741 if (PyInt_Check(arg
))
742 ival
= PyInt_AsUnsignedLongMask(arg
);
743 else if (PyLong_Check(arg
))
744 ival
= PyLong_AsUnsignedLongMask(arg
);
746 return converterr("integer<k>", arg
, msgbuf
, bufsize
);
751 #ifdef HAVE_LONG_LONG
752 case 'L': {/* PY_LONG_LONG */
753 PY_LONG_LONG
*p
= va_arg( *p_va
, PY_LONG_LONG
* );
755 if (float_argument_warning(arg
))
756 return converterr("long<L>", arg
, msgbuf
, bufsize
);
757 ival
= PyLong_AsLongLong(arg
);
758 if (ival
== (PY_LONG_LONG
)-1 && PyErr_Occurred() ) {
759 return converterr("long<L>", arg
, msgbuf
, bufsize
);
766 case 'K': { /* long long sized bitfield */
767 unsigned PY_LONG_LONG
*p
= va_arg(*p_va
, unsigned PY_LONG_LONG
*);
768 unsigned PY_LONG_LONG ival
;
769 if (PyInt_Check(arg
))
770 ival
= PyInt_AsUnsignedLongMask(arg
);
771 else if (PyLong_Check(arg
))
772 ival
= PyLong_AsUnsignedLongLongMask(arg
);
774 return converterr("integer<K>", arg
, msgbuf
, bufsize
);
780 case 'f': {/* float */
781 float *p
= va_arg(*p_va
, float *);
782 double dval
= PyFloat_AsDouble(arg
);
783 if (PyErr_Occurred())
784 return converterr("float<f>", arg
, msgbuf
, bufsize
);
790 case 'd': {/* double */
791 double *p
= va_arg(*p_va
, double *);
792 double dval
= PyFloat_AsDouble(arg
);
793 if (PyErr_Occurred())
794 return converterr("float<d>", arg
, msgbuf
, bufsize
);
800 #ifndef WITHOUT_COMPLEX
801 case 'D': {/* complex double */
802 Py_complex
*p
= va_arg(*p_va
, Py_complex
*);
804 cval
= PyComplex_AsCComplex(arg
);
805 if (PyErr_Occurred())
806 return converterr("complex<D>", arg
, msgbuf
, bufsize
);
811 #endif /* WITHOUT_COMPLEX */
813 case 'c': {/* char */
814 char *p
= va_arg(*p_va
, char *);
815 if (PyString_Check(arg
) && PyString_Size(arg
) == 1)
816 *p
= PyString_AS_STRING(arg
)[0];
818 return converterr("char", arg
, msgbuf
, bufsize
);
822 case 's': {/* string */
823 if (*format
== '*') {
824 Py_buffer
*p
= (Py_buffer
*)va_arg(*p_va
, Py_buffer
*);
826 if (PyString_Check(arg
)) {
827 PyBuffer_FillInfo(p
, arg
,
828 PyString_AS_STRING(arg
), PyString_GET_SIZE(arg
),
831 #ifdef Py_USING_UNICODE
832 else if (PyUnicode_Check(arg
)) {
833 uarg
= UNICODE_DEFAULT_ENCODING(arg
);
835 return converterr(CONV_UNICODE
,
836 arg
, msgbuf
, bufsize
);
837 PyBuffer_FillInfo(p
, arg
,
838 PyString_AS_STRING(uarg
), PyString_GET_SIZE(uarg
),
842 else { /* any buffer-like object */
844 if (getbuffer(arg
, p
, &buf
) < 0)
845 return converterr(buf
, arg
, msgbuf
, bufsize
);
847 if (addcleanup(p
, freelist
, cleanup_buffer
)) {
850 arg
, msgbuf
, bufsize
);
853 } else if (*format
== '#') {
854 void **p
= (void **)va_arg(*p_va
, char **);
857 if (PyString_Check(arg
)) {
858 *p
= PyString_AS_STRING(arg
);
859 STORE_SIZE(PyString_GET_SIZE(arg
));
861 #ifdef Py_USING_UNICODE
862 else if (PyUnicode_Check(arg
)) {
863 uarg
= UNICODE_DEFAULT_ENCODING(arg
);
865 return converterr(CONV_UNICODE
,
866 arg
, msgbuf
, bufsize
);
867 *p
= PyString_AS_STRING(uarg
);
868 STORE_SIZE(PyString_GET_SIZE(uarg
));
871 else { /* any buffer-like object */
873 Py_ssize_t count
= convertbuffer(arg
, p
, &buf
);
875 return converterr(buf
, arg
, msgbuf
, bufsize
);
880 char **p
= va_arg(*p_va
, char **);
882 if (PyString_Check(arg
))
883 *p
= PyString_AS_STRING(arg
);
884 #ifdef Py_USING_UNICODE
885 else if (PyUnicode_Check(arg
)) {
886 uarg
= UNICODE_DEFAULT_ENCODING(arg
);
888 return converterr(CONV_UNICODE
,
889 arg
, msgbuf
, bufsize
);
890 *p
= PyString_AS_STRING(uarg
);
894 return converterr("string", arg
, msgbuf
, bufsize
);
895 if ((Py_ssize_t
)strlen(*p
) != PyString_Size(arg
))
896 return converterr("string without null bytes",
897 arg
, msgbuf
, bufsize
);
902 case 'z': {/* string, may be NULL (None) */
903 if (*format
== '*') {
904 Py_buffer
*p
= (Py_buffer
*)va_arg(*p_va
, Py_buffer
*);
907 PyBuffer_FillInfo(p
, NULL
, NULL
, 0, 1, 0);
908 else if (PyString_Check(arg
)) {
909 PyBuffer_FillInfo(p
, arg
,
910 PyString_AS_STRING(arg
), PyString_GET_SIZE(arg
),
913 #ifdef Py_USING_UNICODE
914 else if (PyUnicode_Check(arg
)) {
915 uarg
= UNICODE_DEFAULT_ENCODING(arg
);
917 return converterr(CONV_UNICODE
,
918 arg
, msgbuf
, bufsize
);
919 PyBuffer_FillInfo(p
, arg
,
920 PyString_AS_STRING(uarg
), PyString_GET_SIZE(uarg
),
924 else { /* any buffer-like object */
926 if (getbuffer(arg
, p
, &buf
) < 0)
927 return converterr(buf
, arg
, msgbuf
, bufsize
);
929 if (addcleanup(p
, freelist
, cleanup_buffer
)) {
932 arg
, msgbuf
, bufsize
);
935 } else if (*format
== '#') { /* any buffer-like object */
936 void **p
= (void **)va_arg(*p_va
, char **);
939 if (arg
== Py_None
) {
943 else if (PyString_Check(arg
)) {
944 *p
= PyString_AS_STRING(arg
);
945 STORE_SIZE(PyString_GET_SIZE(arg
));
947 #ifdef Py_USING_UNICODE
948 else if (PyUnicode_Check(arg
)) {
949 uarg
= UNICODE_DEFAULT_ENCODING(arg
);
951 return converterr(CONV_UNICODE
,
952 arg
, msgbuf
, bufsize
);
953 *p
= PyString_AS_STRING(uarg
);
954 STORE_SIZE(PyString_GET_SIZE(uarg
));
957 else { /* any buffer-like object */
959 Py_ssize_t count
= convertbuffer(arg
, p
, &buf
);
961 return converterr(buf
, arg
, msgbuf
, bufsize
);
966 char **p
= va_arg(*p_va
, char **);
970 else if (PyString_Check(arg
))
971 *p
= PyString_AS_STRING(arg
);
972 #ifdef Py_USING_UNICODE
973 else if (PyUnicode_Check(arg
)) {
974 uarg
= UNICODE_DEFAULT_ENCODING(arg
);
976 return converterr(CONV_UNICODE
,
977 arg
, msgbuf
, bufsize
);
978 *p
= PyString_AS_STRING(uarg
);
982 return converterr("string or None",
983 arg
, msgbuf
, bufsize
);
984 if (*format
== '#') {
986 assert(0); /* XXX redundant with if-case */
990 *q
= PyString_Size(arg
);
993 else if (*p
!= NULL
&&
994 (Py_ssize_t
)strlen(*p
) != PyString_Size(arg
))
996 "string without null bytes or None",
997 arg
, msgbuf
, bufsize
);
1002 case 'e': {/* encoded string */
1004 const char *encoding
;
1009 /* Get 'e' parameter: the encoding name */
1010 encoding
= (const char *)va_arg(*p_va
, const char *);
1011 #ifdef Py_USING_UNICODE
1012 if (encoding
== NULL
)
1013 encoding
= PyUnicode_GetDefaultEncoding();
1016 /* Get output buffer parameter:
1017 's' (recode all objects via Unicode) or
1018 't' (only recode non-string objects)
1022 else if (*format
== 't')
1026 "(unknown parser marker combination)",
1027 arg
, msgbuf
, bufsize
);
1028 buffer
= (char **)va_arg(*p_va
, char **);
1031 return converterr("(buffer is NULL)",
1032 arg
, msgbuf
, bufsize
);
1035 if (!recode_strings
&& PyString_Check(arg
)) {
1040 #ifdef Py_USING_UNICODE
1043 /* Convert object to Unicode */
1044 u
= PyUnicode_FromObject(arg
);
1047 "string or unicode or text buffer",
1048 arg
, msgbuf
, bufsize
);
1050 /* Encode object; use default error handling */
1051 s
= PyUnicode_AsEncodedString(u
,
1056 return converterr("(encoding failed)",
1057 arg
, msgbuf
, bufsize
);
1058 if (!PyString_Check(s
)) {
1061 "(encoder failed to return a string)",
1062 arg
, msgbuf
, bufsize
);
1065 return converterr("string<e>", arg
, msgbuf
, bufsize
);
1068 size
= PyString_GET_SIZE(s
);
1070 /* Write output; output is guaranteed to be 0-terminated */
1071 if (*format
== '#') {
1072 /* Using buffer length parameter '#':
1074 - if *buffer is NULL, a new buffer of the
1075 needed size is allocated and the data
1076 copied into it; *buffer is updated to point
1077 to the new buffer; the caller is
1078 responsible for PyMem_Free()ing it after
1081 - if *buffer is not NULL, the data is
1082 copied to *buffer; *buffer_len has to be
1083 set to the size of the buffer on input;
1084 buffer overflow is signalled with an error;
1085 buffer has to provide enough room for the
1086 encoded string plus the trailing 0-byte
1088 - in both cases, *buffer_len is updated to
1089 the size of the buffer /excluding/ the
1096 if (q
== NULL
&& q2
== NULL
) {
1099 "(buffer_len is NULL)",
1100 arg
, msgbuf
, bufsize
);
1102 if (*buffer
== NULL
) {
1103 *buffer
= PyMem_NEW(char, size
+ 1);
1104 if (*buffer
== NULL
) {
1108 arg
, msgbuf
, bufsize
);
1110 if (addcleanup(*buffer
, freelist
, cleanup_ptr
)) {
1113 "(cleanup problem)",
1114 arg
, msgbuf
, bufsize
);
1117 if (size
+ 1 > BUFFER_LEN
) {
1120 "(buffer overflow)",
1121 arg
, msgbuf
, bufsize
);
1125 PyString_AS_STRING(s
),
1129 /* Using a 0-terminated buffer:
1131 - the encoded string has to be 0-terminated
1132 for this variant to work; if it is not, an
1135 - a new buffer of the needed size is
1136 allocated and the data copied into it;
1137 *buffer is updated to point to the new
1138 buffer; the caller is responsible for
1139 PyMem_Free()ing it after usage
1142 if ((Py_ssize_t
)strlen(PyString_AS_STRING(s
))
1146 "encoded string without NULL bytes",
1147 arg
, msgbuf
, bufsize
);
1149 *buffer
= PyMem_NEW(char, size
+ 1);
1150 if (*buffer
== NULL
) {
1152 return converterr("(memory error)",
1153 arg
, msgbuf
, bufsize
);
1155 if (addcleanup(*buffer
, freelist
, cleanup_ptr
)) {
1157 return converterr("(cleanup problem)",
1158 arg
, msgbuf
, bufsize
);
1161 PyString_AS_STRING(s
),
1168 #ifdef Py_USING_UNICODE
1169 case 'u': {/* raw unicode buffer (Py_UNICODE *) */
1170 if (*format
== '#') { /* any buffer-like object */
1171 void **p
= (void **)va_arg(*p_va
, char **);
1173 if (PyUnicode_Check(arg
)) {
1174 *p
= PyUnicode_AS_UNICODE(arg
);
1175 STORE_SIZE(PyUnicode_GET_SIZE(arg
));
1178 return converterr("cannot convert raw buffers",
1179 arg
, msgbuf
, bufsize
);
1183 Py_UNICODE
**p
= va_arg(*p_va
, Py_UNICODE
**);
1184 if (PyUnicode_Check(arg
))
1185 *p
= PyUnicode_AS_UNICODE(arg
);
1187 return converterr("unicode", arg
, msgbuf
, bufsize
);
1193 case 'S': { /* string object */
1194 PyObject
**p
= va_arg(*p_va
, PyObject
**);
1195 if (PyString_Check(arg
))
1198 return converterr("string", arg
, msgbuf
, bufsize
);
1202 #ifdef Py_USING_UNICODE
1203 case 'U': { /* Unicode object */
1204 PyObject
**p
= va_arg(*p_va
, PyObject
**);
1205 if (PyUnicode_Check(arg
))
1208 return converterr("unicode", arg
, msgbuf
, bufsize
);
1213 case 'O': { /* object */
1216 if (*format
== '!') {
1217 type
= va_arg(*p_va
, PyTypeObject
*);
1218 p
= va_arg(*p_va
, PyObject
**);
1220 if (PyType_IsSubtype(arg
->ob_type
, type
))
1223 return converterr(type
->tp_name
, arg
, msgbuf
, bufsize
);
1226 else if (*format
== '?') {
1227 inquiry pred
= va_arg(*p_va
, inquiry
);
1228 p
= va_arg(*p_va
, PyObject
**);
1233 return converterr("(unspecified)",
1234 arg
, msgbuf
, bufsize
);
1237 else if (*format
== '&') {
1238 typedef int (*converter
)(PyObject
*, void *);
1239 converter convert
= va_arg(*p_va
, converter
);
1240 void *addr
= va_arg(*p_va
, void *);
1242 if (! (*convert
)(arg
, addr
))
1243 return converterr("(unspecified)",
1244 arg
, msgbuf
, bufsize
);
1247 p
= va_arg(*p_va
, PyObject
**);
1254 case 'w': { /* memory buffer, read-write access */
1255 void **p
= va_arg(*p_va
, void **);
1257 PyBufferProcs
*pb
= arg
->ob_type
->tp_as_buffer
;
1260 if (pb
&& pb
->bf_releasebuffer
&& *format
!= '*')
1261 /* Buffer must be released, yet caller does not use
1262 the Py_buffer protocol. */
1263 return converterr("pinned buffer", arg
, msgbuf
, bufsize
);
1265 if (pb
&& pb
->bf_getbuffer
&& *format
== '*') {
1266 /* Caller is interested in Py_buffer, and the object
1267 supports it directly. */
1269 if (pb
->bf_getbuffer(arg
, (Py_buffer
*)p
, PyBUF_WRITABLE
) < 0) {
1271 return converterr("read-write buffer", arg
, msgbuf
, bufsize
);
1273 if (addcleanup(p
, freelist
, cleanup_buffer
)) {
1275 "(cleanup problem)",
1276 arg
, msgbuf
, bufsize
);
1278 if (!PyBuffer_IsContiguous((Py_buffer
*)p
, 'C'))
1279 return converterr("contiguous buffer", arg
, msgbuf
, bufsize
);
1284 pb
->bf_getwritebuffer
== NULL
||
1285 pb
->bf_getsegcount
== NULL
)
1286 return converterr("read-write buffer", arg
, msgbuf
, bufsize
);
1287 if ((*pb
->bf_getsegcount
)(arg
, NULL
) != 1)
1288 return converterr("single-segment read-write buffer",
1289 arg
, msgbuf
, bufsize
);
1290 if ((count
= pb
->bf_getwritebuffer(arg
, 0, &res
)) < 0)
1291 return converterr("(unspecified)", arg
, msgbuf
, bufsize
);
1292 if (*format
== '*') {
1293 PyBuffer_FillInfo((Py_buffer
*)p
, arg
, res
, count
, 1, 0);
1298 if (*format
== '#') {
1307 case 't': { /* 8-bit character buffer, read-only access */
1308 char **p
= va_arg(*p_va
, char **);
1309 PyBufferProcs
*pb
= arg
->ob_type
->tp_as_buffer
;
1312 if (*format
++ != '#')
1314 "invalid use of 't' format character",
1315 arg
, msgbuf
, bufsize
);
1316 if (!PyType_HasFeature(arg
->ob_type
,
1317 Py_TPFLAGS_HAVE_GETCHARBUFFER
) ||
1318 pb
== NULL
|| pb
->bf_getcharbuffer
== NULL
||
1319 pb
->bf_getsegcount
== NULL
)
1321 "string or read-only character buffer",
1322 arg
, msgbuf
, bufsize
);
1324 if (pb
->bf_getsegcount(arg
, NULL
) != 1)
1326 "string or single-segment read-only buffer",
1327 arg
, msgbuf
, bufsize
);
1329 if (pb
->bf_releasebuffer
)
1331 "string or pinned buffer",
1332 arg
, msgbuf
, bufsize
);
1334 count
= pb
->bf_getcharbuffer(arg
, 0, p
);
1336 return converterr("(unspecified)", arg
, msgbuf
, bufsize
);
1345 return converterr("impossible<bad format char>", arg
, msgbuf
, bufsize
);
1354 convertbuffer(PyObject
*arg
, void **p
, char **errmsg
)
1356 PyBufferProcs
*pb
= arg
->ob_type
->tp_as_buffer
;
1359 pb
->bf_getreadbuffer
== NULL
||
1360 pb
->bf_getsegcount
== NULL
||
1361 pb
->bf_releasebuffer
!= NULL
) {
1362 *errmsg
= "string or read-only buffer";
1365 if ((*pb
->bf_getsegcount
)(arg
, NULL
) != 1) {
1366 *errmsg
= "string or single-segment read-only buffer";
1369 if ((count
= (*pb
->bf_getreadbuffer
)(arg
, 0, p
)) < 0) {
1370 *errmsg
= "(unspecified)";
1376 getbuffer(PyObject
*arg
, Py_buffer
*view
, char **errmsg
)
1380 PyBufferProcs
*pb
= arg
->ob_type
->tp_as_buffer
;
1382 *errmsg
= "string or buffer";
1385 if (pb
->bf_getbuffer
) {
1386 if (pb
->bf_getbuffer(arg
, view
, 0) < 0) {
1387 *errmsg
= "convertible to a buffer";
1390 if (!PyBuffer_IsContiguous(view
, 'C')) {
1391 *errmsg
= "contiguous buffer";
1397 count
= convertbuffer(arg
, &buf
, errmsg
);
1399 *errmsg
= "convertible to a buffer";
1402 PyBuffer_FillInfo(view
, NULL
, buf
, count
, 1, 0);
1406 /* Support for keyword arguments donated by
1407 Geoff Philbrick <philbric@delphi.hks.com> */
1409 /* Return false (0) for error, else true. */
1411 PyArg_ParseTupleAndKeywords(PyObject
*args
,
1419 if ((args
== NULL
|| !PyTuple_Check(args
)) ||
1420 (keywords
!= NULL
&& !PyDict_Check(keywords
)) ||
1424 PyErr_BadInternalCall();
1428 va_start(va
, kwlist
);
1429 retval
= vgetargskeywords(args
, keywords
, format
, kwlist
, &va
, 0);
1435 _PyArg_ParseTupleAndKeywords_SizeT(PyObject
*args
,
1443 if ((args
== NULL
|| !PyTuple_Check(args
)) ||
1444 (keywords
!= NULL
&& !PyDict_Check(keywords
)) ||
1448 PyErr_BadInternalCall();
1452 va_start(va
, kwlist
);
1453 retval
= vgetargskeywords(args
, keywords
, format
,
1454 kwlist
, &va
, FLAG_SIZE_T
);
1461 PyArg_VaParseTupleAndKeywords(PyObject
*args
,
1464 char **kwlist
, va_list va
)
1469 if ((args
== NULL
|| !PyTuple_Check(args
)) ||
1470 (keywords
!= NULL
&& !PyDict_Check(keywords
)) ||
1474 PyErr_BadInternalCall();
1478 #ifdef VA_LIST_IS_ARRAY
1479 memcpy(lva
, va
, sizeof(va_list));
1488 retval
= vgetargskeywords(args
, keywords
, format
, kwlist
, &lva
, 0);
1493 _PyArg_VaParseTupleAndKeywords_SizeT(PyObject
*args
,
1496 char **kwlist
, va_list va
)
1501 if ((args
== NULL
|| !PyTuple_Check(args
)) ||
1502 (keywords
!= NULL
&& !PyDict_Check(keywords
)) ||
1506 PyErr_BadInternalCall();
1510 #ifdef VA_LIST_IS_ARRAY
1511 memcpy(lva
, va
, sizeof(va_list));
1520 retval
= vgetargskeywords(args
, keywords
, format
,
1521 kwlist
, &lva
, FLAG_SIZE_T
);
1525 #define IS_END_OF_FORMAT(c) (c == '\0' || c == ';' || c == ':')
1528 vgetargskeywords(PyObject
*args
, PyObject
*keywords
, const char *format
,
1529 char **kwlist
, va_list *p_va
, int flags
)
1533 const char *fname
, *msg
, *custom_msg
, *keyword
;
1535 int i
, len
, nargs
, nkeywords
;
1536 PyObject
*freelist
= NULL
, *current_arg
;
1538 assert(args
!= NULL
&& PyTuple_Check(args
));
1539 assert(keywords
== NULL
|| PyDict_Check(keywords
));
1540 assert(format
!= NULL
);
1541 assert(kwlist
!= NULL
);
1542 assert(p_va
!= NULL
);
1544 /* grab the function name or custom error msg first (mutually exclusive) */
1545 fname
= strchr(format
, ':');
1551 custom_msg
= strchr(format
,';');
1556 /* scan kwlist and get greatest possible nbr of args */
1557 for (len
=0; kwlist
[len
]; len
++)
1560 nargs
= PyTuple_GET_SIZE(args
);
1561 nkeywords
= (keywords
== NULL
) ? 0 : PyDict_Size(keywords
);
1562 if (nargs
+ nkeywords
> len
) {
1563 PyErr_Format(PyExc_TypeError
, "%s%s takes at most %d "
1564 "argument%s (%d given)",
1565 (fname
== NULL
) ? "function" : fname
,
1566 (fname
== NULL
) ? "" : "()",
1568 (len
== 1) ? "" : "s",
1573 /* convert tuple args and keyword args in same loop, using kwlist to drive process */
1574 for (i
= 0; i
< len
; i
++) {
1575 keyword
= kwlist
[i
];
1576 if (*format
== '|') {
1580 if (IS_END_OF_FORMAT(*format
)) {
1581 PyErr_Format(PyExc_RuntimeError
,
1582 "More keyword list entries (%d) than "
1583 "format specifiers (%d)", len
, i
);
1584 return cleanreturn(0, freelist
);
1588 current_arg
= PyDict_GetItemString(keywords
, keyword
);
1593 /* arg present in tuple and in dict */
1594 PyErr_Format(PyExc_TypeError
,
1595 "Argument given by name ('%s') "
1596 "and position (%d)",
1598 return cleanreturn(0, freelist
);
1601 else if (nkeywords
&& PyErr_Occurred())
1602 return cleanreturn(0, freelist
);
1604 current_arg
= PyTuple_GET_ITEM(args
, i
);
1607 msg
= convertitem(current_arg
, &format
, p_va
, flags
,
1608 levels
, msgbuf
, sizeof(msgbuf
), &freelist
);
1610 seterror(i
+1, msg
, levels
, fname
, custom_msg
);
1611 return cleanreturn(0, freelist
);
1617 PyErr_Format(PyExc_TypeError
, "Required argument "
1618 "'%s' (pos %d) not found",
1620 return cleanreturn(0, freelist
);
1622 /* current code reports success when all required args
1623 * fulfilled and no keyword args left, with no further
1624 * validation. XXX Maybe skip this in debug build ?
1627 return cleanreturn(1, freelist
);
1629 /* We are into optional args, skip thru to any remaining
1631 msg
= skipitem(&format
, p_va
, flags
);
1633 PyErr_Format(PyExc_RuntimeError
, "%s: '%s'", msg
,
1635 return cleanreturn(0, freelist
);
1639 if (!IS_END_OF_FORMAT(*format
) && *format
!= '|') {
1640 PyErr_Format(PyExc_RuntimeError
,
1641 "more argument specifiers than keyword list entries "
1642 "(remaining format:'%s')", format
);
1643 return cleanreturn(0, freelist
);
1646 /* make sure there are no extraneous keyword arguments */
1647 if (nkeywords
> 0) {
1648 PyObject
*key
, *value
;
1650 while (PyDict_Next(keywords
, &pos
, &key
, &value
)) {
1653 if (!PyString_Check(key
)) {
1654 PyErr_SetString(PyExc_TypeError
,
1655 "keywords must be strings");
1656 return cleanreturn(0, freelist
);
1658 ks
= PyString_AsString(key
);
1659 for (i
= 0; i
< len
; i
++) {
1660 if (!strcmp(ks
, kwlist
[i
])) {
1666 PyErr_Format(PyExc_TypeError
,
1667 "'%s' is an invalid keyword "
1668 "argument for this function",
1670 return cleanreturn(0, freelist
);
1675 return cleanreturn(1, freelist
);
1680 skipitem(const char **p_format
, va_list *p_va
, int flags
)
1682 const char *format
= *p_format
;
1688 * The individual types (second arg of va_arg) are irrelevant */
1690 case 'b': /* byte -- very short int */
1691 case 'B': /* byte as bitfield */
1692 case 'h': /* short int */
1693 case 'H': /* short int as bitfield */
1695 case 'I': /* int sized bitfield */
1696 case 'l': /* long int */
1697 case 'k': /* long int sized bitfield */
1698 #ifdef HAVE_LONG_LONG
1699 case 'L': /* PY_LONG_LONG */
1700 case 'K': /* PY_LONG_LONG sized bitfield */
1702 case 'f': /* float */
1703 case 'd': /* double */
1704 #ifndef WITHOUT_COMPLEX
1705 case 'D': /* complex double */
1707 case 'c': /* char */
1709 (void) va_arg(*p_va
, void *);
1713 case 'n': /* Py_ssize_t */
1715 (void) va_arg(*p_va
, Py_ssize_t
*);
1721 case 'e': /* string with encoding */
1723 (void) va_arg(*p_va
, const char *);
1724 if (!(*format
== 's' || *format
== 't'))
1725 /* after 'e', only 's' and 't' is allowed */
1728 /* explicit fallthrough to string cases */
1731 case 's': /* string */
1732 case 'z': /* string or None */
1733 #ifdef Py_USING_UNICODE
1734 case 'u': /* unicode string */
1736 case 't': /* buffer, read-only */
1737 case 'w': /* buffer, read-write */
1739 (void) va_arg(*p_va
, char **);
1740 if (*format
== '#') {
1741 if (flags
& FLAG_SIZE_T
)
1742 (void) va_arg(*p_va
, Py_ssize_t
*);
1744 (void) va_arg(*p_va
, int *);
1746 } else if ((c
== 's' || c
== 'z') && *format
== '*') {
1754 case 'S': /* string object */
1755 #ifdef Py_USING_UNICODE
1756 case 'U': /* unicode string object */
1759 (void) va_arg(*p_va
, PyObject
**);
1763 case 'O': /* object */
1765 if (*format
== '!') {
1767 (void) va_arg(*p_va
, PyTypeObject
*);
1768 (void) va_arg(*p_va
, PyObject
**);
1770 else if (*format
== '&') {
1771 typedef int (*converter
)(PyObject
*, void *);
1772 (void) va_arg(*p_va
, converter
);
1773 (void) va_arg(*p_va
, void *);
1777 (void) va_arg(*p_va
, PyObject
**);
1782 case '(': /* bypass tuple, not handled at all previously */
1788 if (IS_END_OF_FORMAT(*format
))
1789 return "Unmatched left paren in format "
1791 msg
= skipitem(&format
, p_va
, flags
);
1800 return "Unmatched right paren in format string";
1804 return "impossible<bad format char>";
1814 PyArg_UnpackTuple(PyObject
*args
, const char *name
, Py_ssize_t min
, Py_ssize_t max
, ...)
1820 #ifdef HAVE_STDARG_PROTOTYPES
1821 va_start(vargs
, max
);
1828 if (!PyTuple_Check(args
)) {
1829 PyErr_SetString(PyExc_SystemError
,
1830 "PyArg_UnpackTuple() argument list is not a tuple");
1833 l
= PyTuple_GET_SIZE(args
);
1838 "%s expected %s%zd arguments, got %zd",
1839 name
, (min
== max
? "" : "at least "), min
, l
);
1843 "unpacked tuple should have %s%zd elements,"
1845 (min
== max
? "" : "at least "), min
, l
);
1853 "%s expected %s%zd arguments, got %zd",
1854 name
, (min
== max
? "" : "at most "), max
, l
);
1858 "unpacked tuple should have %s%zd elements,"
1860 (min
== max
? "" : "at most "), max
, l
);
1864 for (i
= 0; i
< l
; i
++) {
1865 o
= va_arg(vargs
, PyObject
**);
1866 *o
= PyTuple_GET_ITEM(args
, i
);
1873 /* For type constructors that don't take keyword args
1875 * Sets a TypeError and returns 0 if the kwds dict is
1876 * not empty, returns 1 otherwise
1879 _PyArg_NoKeywords(const char *funcname
, PyObject
*kw
)
1883 if (!PyDict_CheckExact(kw
)) {
1884 PyErr_BadInternalCall();
1887 if (PyDict_Size(kw
) == 0)
1890 PyErr_Format(PyExc_TypeError
, "%s does not take keyword arguments",