Added functions CFObj_New and CFObj_Convert, general functions to convert

between CF objects and their Python representation. Fixes 734695.

diff --git a/Include/pymactoolbox.h b/Include/pymactoolbox.h
index 2286ebe1a99efe576239dd572e773582736953fc..916dac71486622e38be72c6e7c8ce21b06aa91ac 100644 (file)
--- a/Include/pymactoolbox.h
+++ b/Include/pymactoolbox.h
@@ -178,6 +178,8 @@ extern int WinObj_Convert(PyObject *, WindowPtr *);
 extern PyObject *WinObj_WhichWindow(WindowPtr);
 
 /* CF exports */
+extern PyObject *CFObj_New(CFTypeRef);
+extern int CFObj_Convert(PyObject *, CFTypeRef *);
 extern PyObject *CFTypeRefObj_New(CFTypeRef);
 extern int CFTypeRefObj_Convert(PyObject *, CFTypeRef *);
 extern PyObject *CFStringRefObj_New(CFStringRef);

diff --git a/Mac/Modules/cf/_CFmodule.c b/Mac/Modules/cf/_CFmodule.c
index 8473eb56bade5b4b39e020a2f157ee5207fbb23d..b6e434af77ab908ff61dee6c9876597962c39629 100644 (file)
--- a/Mac/Modules/cf/_CFmodule.c
+++ b/Mac/Modules/cf/_CFmodule.c
@@ -36,6 +36,11 @@
 #include "pycfbridge.h"
 
 #ifdef USE_TOOLBOX_OBJECT_GLUE
+extern PyObject *_CFObj_New(CFTypeRef);
+extern int _CFObj_Convert(PyObject *, CFTypeRef *);
+#define CFObj_New _CFObj_New
+#define CFObj_Convert _CFObj_Convert
+
 extern PyObject *_CFTypeRefObj_New(CFTypeRef);
 extern int _CFTypeRefObj_Convert(PyObject *, CFTypeRef *);
 #define CFTypeRefObj_New _CFTypeRefObj_New
@@ -121,7 +126,6 @@ OptionalCFURLRefObj_Convert(PyObject *v, CFURLRef *p_itself)
     return CFURLRefObj_Convert(v, p_itself);
 }
 
-
 static PyObject *CF_Error;
 
 /* --------------------- Object type CFTypeRef ---------------------- */
@@ -1457,7 +1461,9 @@ int CFStringRefObj_Convert(PyObject *v, CFStringRef *p_itself)
 
        if (v == Py_None) { *p_itself = NULL; return 1; }
        if (PyString_Check(v)) {
-           char *cStr = PyString_AsString(v);
+           char *cStr;
+           if (!PyArg_Parse(v, "es", "ascii", &cStr))
+               return NULL;
                *p_itself = CFStringCreateWithCString((CFAllocatorRef)NULL, cStr, kCFStringEncodingASCII);
                return 1;
        }
@@ -4292,6 +4298,48 @@ static PyMethodDef CF_methods[] = {
 
 
 
+/* Routines to convert any CF type to/from the corresponding CFxxxObj */
+PyObject *CFObj_New(CFTypeRef itself)
+{
+       if (itself == NULL)
+       {
+               PyErr_SetString(PyExc_RuntimeError, "cannot wrap NULL");
+               return NULL;
+       }
+       if (CFGetTypeID(itself) == CFArrayGetTypeID()) return CFArrayRefObj_New((CFArrayRef)itself);
+       if (CFGetTypeID(itself) == CFDictionaryGetTypeID()) return CFDictionaryRefObj_New((CFDictionaryRef)itself);
+       if (CFGetTypeID(itself) == CFDataGetTypeID()) return CFDataRefObj_New((CFDataRef)itself);
+       if (CFGetTypeID(itself) == CFStringGetTypeID()) return CFStringRefObj_New((CFStringRef)itself);
+       if (CFGetTypeID(itself) == CFURLGetTypeID()) return CFURLRefObj_New((CFURLRef)itself);
+       /* XXXX Or should we use PyCF_CF2Python here?? */
+       return CFTypeRefObj_New(itself);
+}
+int CFObj_Convert(PyObject *v, CFTypeRef *p_itself)
+{
+
+       if (v == Py_None) { *p_itself = NULL; return 1; }
+       /* Check for other CF objects here */
+
+       if (!CFTypeRefObj_Check(v) &&
+               !CFArrayRefObj_Check(v) &&
+               !CFMutableArrayRefObj_Check(v) &&
+               !CFDictionaryRefObj_Check(v) &&
+               !CFMutableDictionaryRefObj_Check(v) &&
+               !CFDataRefObj_Check(v) &&
+               !CFMutableDataRefObj_Check(v) &&
+               !CFStringRefObj_Check(v) &&
+               !CFMutableStringRefObj_Check(v) &&
+               !CFURLRefObj_Check(v) )
+       {
+               /* XXXX Or should we use PyCF_Python2CF here?? */
+               PyErr_SetString(PyExc_TypeError, "CF object required");
+               return 0;
+       }
+       *p_itself = ((CFTypeRefObject *)v)->ob_itself;
+       return 1;
+}
+
+
 void init_CF(void)
 {
        PyObject *m;

diff --git a/Mac/Modules/cf/cfsupport.py b/Mac/Modules/cf/cfsupport.py
index e8fb710310ec9a4446d6543f572a5d31621efd33..7e0a5d9f2367f4e234e17b91e4dc4fad6b6290bb 100644 (file)
--- a/Mac/Modules/cf/cfsupport.py
+++ b/Mac/Modules/cf/cfsupport.py
@@ -58,6 +58,11 @@ includestuff = includestuff + """
 #include "pycfbridge.h"
 
 #ifdef USE_TOOLBOX_OBJECT_GLUE
+extern PyObject *_CFObj_New(CFTypeRef);
+extern int _CFObj_Convert(PyObject *, CFTypeRef *);
+#define CFObj_New _CFObj_New
+#define CFObj_Convert _CFObj_Convert
+
 extern PyObject *_CFTypeRefObj_New(CFTypeRef);
 extern int _CFTypeRefObj_Convert(PyObject *, CFTypeRef *);
 #define CFTypeRefObj_New _CFTypeRefObj_New
@@ -142,7 +147,50 @@ OptionalCFURLRefObj_Convert(PyObject *v, CFURLRef *p_itself)
     }
     return CFURLRefObj_Convert(v, p_itself);
 }
+"""
+
+finalstuff = finalstuff + """
+
+/* Routines to convert any CF type to/from the corresponding CFxxxObj */
+PyObject *CFObj_New(CFTypeRef itself)
+{
+       if (itself == NULL)
+       {
+               PyErr_SetString(PyExc_RuntimeError, "cannot wrap NULL");
+               return NULL;
+       }
+       if (CFGetTypeID(itself) == CFArrayGetTypeID()) return CFArrayRefObj_New((CFArrayRef)itself);
+       if (CFGetTypeID(itself) == CFDictionaryGetTypeID()) return CFDictionaryRefObj_New((CFDictionaryRef)itself);
+       if (CFGetTypeID(itself) == CFDataGetTypeID()) return CFDataRefObj_New((CFDataRef)itself);
+       if (CFGetTypeID(itself) == CFStringGetTypeID()) return CFStringRefObj_New((CFStringRef)itself);
+       if (CFGetTypeID(itself) == CFURLGetTypeID()) return CFURLRefObj_New((CFURLRef)itself);
+       /* XXXX Or should we use PyCF_CF2Python here?? */
+       return CFTypeRefObj_New(itself);
+}
+int CFObj_Convert(PyObject *v, CFTypeRef *p_itself)
+{
 
+       if (v == Py_None) { *p_itself = NULL; return 1; }
+       /* Check for other CF objects here */
+
+       if (!CFTypeRefObj_Check(v) &&
+               !CFArrayRefObj_Check(v) &&
+               !CFMutableArrayRefObj_Check(v) &&
+               !CFDictionaryRefObj_Check(v) &&
+               !CFMutableDictionaryRefObj_Check(v) &&
+               !CFDataRefObj_Check(v) &&
+               !CFMutableDataRefObj_Check(v) &&
+               !CFStringRefObj_Check(v) &&
+               !CFMutableStringRefObj_Check(v) &&
+               !CFURLRefObj_Check(v) )
+       {
+               /* XXXX Or should we use PyCF_Python2CF here?? */
+               PyErr_SetString(PyExc_TypeError, "CF object required");
+               return 0;
+       }
+       *p_itself = ((CFTypeRefObject *)v)->ob_itself;
+       return 1;
+}
 """
 
 initstuff = initstuff + """

diff --git a/Python/mactoolboxglue.c b/Python/mactoolboxglue.c
index a939ee5dcf93f2ceb535e844424323ef97a921b8..8b31f7f619e4be5ebf0a3dfec823250194c61a7a 100644 (file)
--- a/Python/mactoolboxglue.c
+++ b/Python/mactoolboxglue.c
@@ -593,6 +593,9 @@ GLUE_NEW(WindowPtr, WinObj_New, "Carbon.Win")
 GLUE_CONVERT(WindowPtr, WinObj_Convert, "Carbon.Win")
 GLUE_NEW(WindowPtr, WinObj_WhichWindow, "Carbon.Win")
 
+GLUE_CONVERT(CFTypeRef, CFObj_Convert, "Carbon.CF")
+GLUE_NEW(CFTypeRef, CFObj_New, "Carbon.CF")
+
 GLUE_CONVERT(CFTypeRef, CFTypeRefObj_Convert, "Carbon.CF")
 GLUE_NEW(CFTypeRef, CFTypeRefObj_New, "Carbon.CF")