--- /dev/null
+/* $Id$ */
+
+/* PycURL -- cURL Python module
+ *
+ * Authors:
+ * Copyright (C) 2001-2005 by Kjetil Jacobsen <kjetilja at cs.uit.no>
+ * Copyright (C) 2001-2005 by Markus F.X.J. Oberhumer <markus at oberhumer.com>
+ *
+ * Contributions:
+ * Tino Lange <Tino.Lange at gmx.de>
+ * Matt King <matt at gnik.com>
+ * Conrad Steenberg <conrad at hep.caltech.edu>
+ * Amit Mongia <amit_mongia at hotmail.com>
+ * Eric S. Raymond <esr at thyrsus.com>
+ * Martin Muenstermann <mamuema at sourceforge.net>
+ * Domenico Andreoli <cavok at libero.it>
+ *
+ * See file COPYING for license information.
+ *
+ * Some quick info on Python's refcount:
+ * Py_BuildValue does incref the item(s)
+ * PyArg_ParseTuple does NOT incref the item
+ * PyList_Append does incref the item
+ * PyTuple_SET_ITEM does NOT incref the item
+ * PyTuple_SetItem does NOT incref the item
+ * PyXXX_GetItem returns a borrowed reference
+ */
+
+#if (defined(_WIN32) || defined(__WIN32__)) && !defined(WIN32)
+# define WIN32 1
+#endif
+#if defined(WIN32)
+# define CURL_STATICLIB 1
+#endif
+#include <Python.h>
+#include <sys/types.h>
+#include <stddef.h>
+#include <stdlib.h>
+#include <stdio.h>
+#include <string.h>
+#include <limits.h>
+#include <curl/curl.h>
+#include <curl/multi.h>
+#undef NDEBUG
+#include <assert.h>
+
+/* Ensure we have updated versions */
+#if !defined(PY_VERSION_HEX) || (PY_VERSION_HEX < 0x02020000)
+# error "Need Python version 2.2 or greater to compile pycurl."
+#endif
+#if !defined(LIBCURL_VERSION_NUM) || (LIBCURL_VERSION_NUM < 0x070d01)
+# error "Need libcurl version 7.13.1 or greater to compile pycurl."
+#endif
+
+#undef UNUSED
+#define UNUSED(var) ((void)&var)
+
+#undef COMPILE_TIME_ASSERT
+#define COMPILE_TIME_ASSERT(expr) \
+ { typedef int compile_time_assert_fail__[1 - 2 * !(expr)]; }
+
+
+/* Calculate the number of OBJECTPOINT options we need to store */
+#define OPTIONS_SIZE ((int)CURLOPT_LASTENTRY % 10000)
+static int OPT_INDEX(int o)
+{
+ assert(o >= CURLOPTTYPE_OBJECTPOINT);
+ assert(o < CURLOPTTYPE_OBJECTPOINT + OPTIONS_SIZE);
+ return o - CURLOPTTYPE_OBJECTPOINT;
+}
+
+
+static PyObject *ErrorObject = NULL;
+static PyTypeObject *p_Curl_Type = NULL;
+static PyTypeObject *p_CurlMulti_Type = NULL;
+
+typedef struct {
+ PyObject_HEAD
+ PyObject *dict; /* Python attributes dictionary */
+ CURLM *multi_handle;
+ PyThreadState *state;
+ fd_set read_fd_set;
+ fd_set write_fd_set;
+ fd_set exc_fd_set;
+} CurlMultiObject;
+
+typedef struct {
+ PyObject_HEAD
+ PyObject *dict; /* Python attributes dictionary */
+ CURL *handle;
+ PyThreadState *state;
+ CurlMultiObject *multi_stack;
+ struct curl_httppost *httppost;
+ struct curl_slist *httpheader;
+ struct curl_slist *http200aliases;
+ struct curl_slist *quote;
+ struct curl_slist *postquote;
+ struct curl_slist *prequote;
+ struct curl_slist *source_prequote;
+ struct curl_slist *source_postquote;
+ /* callbacks */
+ PyObject *w_cb;
+ PyObject *h_cb;
+ PyObject *r_cb;
+ PyObject *pro_cb;
+ PyObject *debug_cb;
+ PyObject *ioctl_cb;
+ /* file objects */
+ PyObject *readdata_fp;
+ PyObject *writedata_fp;
+ PyObject *writeheader_fp;
+ /* misc */
+ void *options[OPTIONS_SIZE]; /* for OBJECTPOINT options */
+ char error[CURL_ERROR_SIZE+1];
+} CurlObject;
+
+/* Throw exception based on return value `res' and `self->error' */
+#define CURLERROR_RETVAL() do {\
+ PyObject *v; \
+ self->error[sizeof(self->error) - 1] = 0; \
+ v = Py_BuildValue("(is)", (int) (res), self->error); \
+ if (v != NULL) { PyErr_SetObject(ErrorObject, v); Py_DECREF(v); } \
+ return NULL; \
+} while (0)
+
+/* Throw exception based on return value `res' and custom message */
+#define CURLERROR_MSG(msg) do {\
+ PyObject *v; const char *m = (msg); \
+ v = Py_BuildValue("(is)", (int) (res), (m)); \
+ if (v != NULL) { PyErr_SetObject(ErrorObject, v); Py_DECREF(v); } \
+ return NULL; \
+} while (0)
+
+
+/* Safe XDECREF for object states that handles nested deallocations */
+#define ZAP(v) do {\
+ PyObject *tmp = (PyObject *)(v); \
+ (v) = NULL; \
+ Py_XDECREF(tmp); \
+} while (0)
+
+
+/*************************************************************************
+// python utility functions
+**************************************************************************/
+
+#if (PY_VERSION_HEX < 0x02030000) && !defined(PY_LONG_LONG)
+# define PY_LONG_LONG LONG_LONG
+#endif
+
+/* Like PyString_AsString(), but set an exception if the string contains
+ * embedded NULs. Actually PyString_AsStringAndSize() already does that for
+ * us if the `len' parameter is NULL - see Objects/stringobject.c.
+ */
+
+static char *PyString_AsString_NoNUL(PyObject *obj)
+{
+ char *s = NULL;
+ int r;
+ r = PyString_AsStringAndSize(obj, &s, NULL);
+ if (r != 0)
+ return NULL; /* exception already set */
+ assert(s != NULL);
+ return s;
+}
+
+
+/* Convert a curl slist (a list of strings) to a Python list.
+ * In case of error return NULL with an exception set.
+ */
+static PyObject *convert_slist(struct curl_slist *slist, int free_flags)
+{
+ PyObject *ret = NULL;
+
+ ret = PyList_New(0);
+ if (ret == NULL) goto error;
+
+ for ( ; slist != NULL; slist = slist->next) {
+ PyObject *v = NULL;
+
+ if (slist->data != NULL) {
+ v = PyString_FromString(slist->data);
+ if (v == NULL || PyList_Append(ret, v) != 0) {
+ Py_XDECREF(v);
+ goto error;
+ }
+ Py_DECREF(v);
+ }
+ }
+
+ if ((free_flags & 1) && slist)
+ curl_slist_free_all(slist);
+ return ret;
+
+error:
+ Py_XDECREF(ret);
+ if ((free_flags & 2) && slist)
+ curl_slist_free_all(slist);
+ return NULL;
+}
+
+
+/*************************************************************************
+// static utility functions
+**************************************************************************/
+
+static PyThreadState *
+get_thread_state(const CurlObject *self)
+{
+ /* Get the thread state for callbacks to run in.
+ * This is either `self->state' when running inside perform() or
+ * `self->multi_stack->state' when running inside multi_perform().
+ * When the result is != NULL we also implicitly assert
+ * a valid `self->handle'.
+ */
+ if (self == NULL)
+ return NULL;
+ assert(self->ob_type == p_Curl_Type);
+ if (self->state != NULL)
+ {
+ /* inside perform() */
+ assert(self->handle != NULL);
+ if (self->multi_stack != NULL) {
+ assert(self->multi_stack->state == NULL);
+ }
+ return self->state;
+ }
+ if (self->multi_stack != NULL && self->multi_stack->state != NULL)
+ {
+ /* inside multi_perform() */
+ assert(self->handle != NULL);
+ assert(self->multi_stack->multi_handle != NULL);
+ assert(self->state == NULL);
+ return self->multi_stack->state;
+ }
+ return NULL;
+}
+
+
+/* assert some CurlObject invariants */
+static void
+assert_curl_state(const CurlObject *self)
+{
+ assert(self != NULL);
+ assert(self->ob_type == p_Curl_Type);
+ (void) get_thread_state(self);
+}
+
+
+/* assert some CurlMultiObject invariants */
+static void
+assert_multi_state(const CurlMultiObject *self)
+{
+ assert(self != NULL);
+ assert(self->ob_type == p_CurlMulti_Type);
+ if (self->state != NULL) {
+ assert(self->multi_handle != NULL);
+ }
+}
+
+
+/* check state for methods */
+static int
+check_curl_state(const CurlObject *self, int flags, const char *name)
+{
+ assert_curl_state(self);
+ if ((flags & 1) && self->handle == NULL) {
+ PyErr_Format(ErrorObject, "cannot invoke %s() - no curl handle", name);
+ return -1;
+ }
+ if ((flags & 2) && get_thread_state(self) != NULL) {
+ PyErr_Format(ErrorObject, "cannot invoke %s() - perform() is currently running", name);
+ return -1;
+ }
+ return 0;
+}
+
+static int
+check_multi_state(const CurlMultiObject *self, int flags, const char *name)
+{
+ assert_multi_state(self);
+ if ((flags & 1) && self->multi_handle == NULL) {
+ PyErr_Format(ErrorObject, "cannot invoke %s() - no multi handle", name);
+ return -1;
+ }
+ if ((flags & 2) && self->state != NULL) {
+ PyErr_Format(ErrorObject, "cannot invoke %s() - multi_perform() is currently running", name);
+ return -1;
+ }
+ return 0;
+}
+
+
+/*************************************************************************
+// CurlObject
+**************************************************************************/
+
+/* --------------- construct/destruct (i.e. open/close) --------------- */
+
+/* Allocate a new python curl object */
+static CurlObject *
+util_curl_new(void)
+{
+ CurlObject *self;
+
+ self = (CurlObject *) PyObject_GC_New(CurlObject, p_Curl_Type);
+ if (self == NULL)
+ return NULL;
+ PyObject_GC_Track(self);
+
+ /* Set python curl object initial values */
+ self->dict = NULL;
+ self->handle = NULL;
+ self->state = NULL;
+ self->multi_stack = NULL;
+ self->httppost = NULL;
+ self->httpheader = NULL;
+ self->http200aliases = NULL;
+ self->quote = NULL;
+ self->postquote = NULL;
+ self->prequote = NULL;
+ self->source_postquote = NULL;
+ self->source_prequote = NULL;
+
+ /* Set callback pointers to NULL by default */
+ self->w_cb = NULL;
+ self->h_cb = NULL;
+ self->r_cb = NULL;
+ self->pro_cb = NULL;
+ self->debug_cb = NULL;
+ self->ioctl_cb = NULL;
+
+ /* Set file object pointers to NULL by default */
+ self->readdata_fp = NULL;
+ self->writedata_fp = NULL;
+ self->writeheader_fp = NULL;
+
+ /* Zero string pointer memory buffer used by setopt */
+ memset(self->options, 0, sizeof(self->options));
+ memset(self->error, 0, sizeof(self->error));
+
+ return self;
+}
+
+
+/* constructor - this is a module-level function returning a new instance */
+static CurlObject *
+do_curl_new(PyObject *dummy)
+{
+ CurlObject *self = NULL;
+ int res;
+ char *s = NULL;
+
+ UNUSED(dummy);
+
+ /* Allocate python curl object */
+ self = util_curl_new();
+ if (self == NULL)
+ return NULL;
+
+ /* Initialize curl handle */
+ self->handle = curl_easy_init();
+ if (self->handle == NULL)
+ goto error;
+
+ /* Set curl error buffer and zero it */
+ res = curl_easy_setopt(self->handle, CURLOPT_ERRORBUFFER, self->error);
+ if (res != CURLE_OK)
+ goto error;
+ memset(self->error, 0, sizeof(self->error));
+
+ /* Set backreference */
+ res = curl_easy_setopt(self->handle, CURLOPT_PRIVATE, (char *) self);
+ if (res != CURLE_OK)
+ goto error;
+
+ /* Enable NOPROGRESS by default, i.e. no progress output */
+ res = curl_easy_setopt(self->handle, CURLOPT_NOPROGRESS, (long)1);
+ if (res != CURLE_OK)
+ goto error;
+
+ /* Disable VERBOSE by default, i.e. no verbose output */
+ res = curl_easy_setopt(self->handle, CURLOPT_VERBOSE, (long)0);
+ if (res != CURLE_OK)
+ goto error;
+
+ /* Set FTP_ACCOUNT to NULL by default */
+ res = curl_easy_setopt(self->handle, CURLOPT_FTP_ACCOUNT, NULL);
+ if (res != CURLE_OK)
+ goto error;
+
+ /* Set default USERAGENT */
+ s = (char *) malloc(7 + strlen(LIBCURL_VERSION) + 1);
+ if (s == NULL)
+ goto error;
+ strcpy(s, "PycURL/"); strcpy(s+7, LIBCURL_VERSION);
+ res = curl_easy_setopt(self->handle, CURLOPT_USERAGENT, (char *) s);
+ if (res != CURLE_OK) {
+ free(s);
+ goto error;
+ }
+ self->options[ OPT_INDEX(CURLOPT_USERAGENT) ] = s; s = NULL;
+
+ /* Success - return new object */
+ return self;
+
+error:
+ Py_DECREF(self); /* this also closes self->handle */
+ PyErr_SetString(ErrorObject, "initializing curl failed");
+ return NULL;
+}
+
+
+/* util function shared by close() and clear() */
+static void
+util_curl_xdecref(CurlObject *self, int flags, CURL *handle)
+{
+ if (flags & 1) {
+ /* Decrement refcount for attributes dictionary. */
+ ZAP(self->dict);
+ }
+
+ if (flags & 2) {
+ /* Decrement refcount for multi_stack. */
+ if (self->multi_stack != NULL) {
+ CurlMultiObject *multi_stack = self->multi_stack;
+ self->multi_stack = NULL;
+ if (multi_stack->multi_handle != NULL && handle != NULL) {
+ (void) curl_multi_remove_handle(multi_stack->multi_handle, handle);
+ }
+ Py_DECREF(multi_stack);
+ }
+ }
+
+ if (flags & 4) {
+ /* Decrement refcount for python callbacks. */
+ ZAP(self->w_cb);
+ ZAP(self->h_cb);
+ ZAP(self->r_cb);
+ ZAP(self->pro_cb);
+ ZAP(self->debug_cb);
+ ZAP(self->ioctl_cb);
+ }
+
+ if (flags & 8) {
+ /* Decrement refcount for python file objects. */
+ ZAP(self->readdata_fp);
+ ZAP(self->writedata_fp);
+ ZAP(self->writeheader_fp);
+ }
+}
+
+
+static void
+util_curl_close(CurlObject *self)
+{
+ CURL *handle;
+ int i;
+
+ /* Zero handle and thread-state to disallow any operations to be run
+ * from now on */
+ assert(self != NULL);
+ assert(self->ob_type == p_Curl_Type);
+ handle = self->handle;
+ self->handle = NULL;
+ if (handle == NULL) {
+ /* Some paranoia assertions just to make sure the object
+ * deallocation problem is finally really fixed... */
+ assert(self->state == NULL);
+ assert(self->multi_stack == NULL);
+ return; /* already closed */
+ }
+ self->state = NULL;
+
+ /* Decref multi stuff which uses this handle */
+ util_curl_xdecref(self, 2, handle);
+
+ /* Cleanup curl handle - must be done without the gil */
+ Py_BEGIN_ALLOW_THREADS
+ curl_easy_cleanup(handle);
+ Py_END_ALLOW_THREADS
+ handle = NULL;
+
+ /* Decref callbacks and file handles */
+ util_curl_xdecref(self, 4 | 8, handle);
+
+ /* Free all variables allocated by setopt */
+#undef SFREE
+#define SFREE(v) if ((v) != NULL) (curl_formfree(v), (v) = NULL)
+ SFREE(self->httppost);
+#undef SFREE
+#define SFREE(v) if ((v) != NULL) (curl_slist_free_all(v), (v) = NULL)
+ SFREE(self->httpheader);
+ SFREE(self->http200aliases);
+ SFREE(self->quote);
+ SFREE(self->postquote);
+ SFREE(self->prequote);
+ SFREE(self->source_postquote);
+ SFREE(self->source_prequote);
+#undef SFREE
+
+ /* Last, free the options. This must be done after the curl handle
+ * is closed since libcurl assumes that some options are valid when
+ * invoking curl_easy_cleanup(). */
+ for (i = 0; i < OPTIONS_SIZE; i++) {
+ if (self->options[i] != NULL) {
+ free(self->options[i]);
+ self->options[i] = NULL;
+ }
+ }
+}
+
+
+static void
+do_curl_dealloc(CurlObject *self)
+{
+ PyObject_GC_UnTrack(self);
+ Py_TRASHCAN_SAFE_BEGIN(self)
+
+ ZAP(self->dict);
+ util_curl_close(self);
+
+ PyObject_GC_Del(self);
+ Py_TRASHCAN_SAFE_END(self)
+}
+
+
+static PyObject *
+do_curl_close(CurlObject *self)
+{
+ if (check_curl_state(self, 2, "close") != 0) {
+ return NULL;
+ }
+ util_curl_close(self);
+ Py_INCREF(Py_None);
+ return Py_None;
+}
+
+
+static PyObject *
+do_curl_errstr(CurlObject *self)
+{
+ if (check_curl_state(self, 1 | 2, "errstr") != 0) {
+ return NULL;
+ }
+ self->error[sizeof(self->error) - 1] = 0;
+ return PyString_FromString(self->error);
+}
+
+
+/* --------------- GC support --------------- */
+
+/* Drop references that may have created reference cycles. */
+static int
+do_curl_clear(CurlObject *self)
+{
+ assert(get_thread_state(self) == NULL);
+ util_curl_xdecref(self, 1 | 2 | 4 | 8, self->handle);
+ return 0;
+}
+
+/* Traverse all refcounted objects. */
+static int
+do_curl_traverse(CurlObject *self, visitproc visit, void *arg)
+{
+ int err;
+#undef VISIT
+#define VISIT(v) if ((v) != NULL && ((err = visit(v, arg)) != 0)) return err
+
+ VISIT(self->dict);
+ VISIT((PyObject *) self->multi_stack);
+
+ VISIT(self->w_cb);
+ VISIT(self->h_cb);
+ VISIT(self->r_cb);
+ VISIT(self->pro_cb);
+ VISIT(self->debug_cb);
+ VISIT(self->ioctl_cb);
+
+ VISIT(self->readdata_fp);
+ VISIT(self->writedata_fp);
+ VISIT(self->writeheader_fp);
+
+ return 0;
+#undef VISIT
+}
+
+
+/* --------------- perform --------------- */
+
+static PyObject *
+do_curl_perform(CurlObject *self)
+{
+ int res;
+
+ if (check_curl_state(self, 1 | 2, "perform") != 0) {
+ return NULL;
+ }
+
+ /* Save handle to current thread (used as context for python callbacks) */
+ self->state = PyThreadState_Get();
+ assert(self->state != NULL);
+
+ /* Release global lock and start */
+ Py_BEGIN_ALLOW_THREADS
+ res = curl_easy_perform(self->handle);
+ Py_END_ALLOW_THREADS
+
+ /* Zero thread-state to disallow callbacks to be run from now on */
+ self->state = NULL;
+
+ if (res != CURLE_OK) {
+ CURLERROR_RETVAL();
+ }
+ Py_INCREF(Py_None);
+ return Py_None;
+}
+
+
+/* --------------- callback handlers --------------- */
+
+/* IMPORTANT NOTE: due to threading issues, we cannot call _any_ Python
+ * function without acquiring the thread state in the callback handlers.
+ */
+
+static size_t
+util_write_callback(int flags, char *ptr, size_t size, size_t nmemb, void *stream)
+{
+ CurlObject *self;
+ PyThreadState *tmp_state;
+ PyObject *arglist;
+ PyObject *result = NULL;
+ size_t ret = 0; /* assume error */
+ PyObject *cb;
+ int total_size;
+
+ /* acquire thread */
+ self = (CurlObject *)stream;
+ tmp_state = get_thread_state(self);
+ if (tmp_state == NULL)
+ return ret;
+ PyEval_AcquireThread(tmp_state);
+
+ /* check args */
+ cb = flags ? self->h_cb : self->w_cb;
+ if (cb == NULL)
+ goto silent_error;
+ if (size <= 0 || nmemb <= 0)
+ goto done;
+ total_size = (int)(size * nmemb);
+ if (total_size < 0 || (size_t)total_size / size != nmemb) {
+ PyErr_SetString(ErrorObject, "integer overflow in write callback");
+ goto verbose_error;
+ }
+
+ /* run callback */
+ arglist = Py_BuildValue("(s#)", ptr, total_size);
+ if (arglist == NULL)
+ goto verbose_error;
+ result = PyEval_CallObject(cb, arglist);
+ Py_DECREF(arglist);
+ if (result == NULL)
+ goto verbose_error;
+
+ /* handle result */
+ if (result == Py_None) {
+ ret = total_size; /* None means success */
+ }
+ else if (PyInt_Check(result)) {
+ long obj_size = PyInt_AsLong(result);
+ if (obj_size < 0 || obj_size > total_size) {
+ PyErr_Format(ErrorObject, "invalid return value for write callback %ld %ld", (long)obj_size, (long)total_size);
+ goto verbose_error;
+ }
+ ret = (size_t) obj_size; /* success */
+ }
+ else if (PyLong_Check(result)) {
+ long obj_size = PyLong_AsLong(result);
+ if (obj_size < 0 || obj_size > total_size) {
+ PyErr_Format(ErrorObject, "invalid return value for write callback %ld %ld", (long)obj_size, (long)total_size);
+ goto verbose_error;
+ }
+ ret = (size_t) obj_size; /* success */
+ }
+ else {
+ PyErr_SetString(ErrorObject, "write callback must return int or None");
+ goto verbose_error;
+ }
+
+done:
+silent_error:
+ Py_XDECREF(result);
+ PyEval_ReleaseThread(tmp_state);
+ return ret;
+verbose_error:
+ PyErr_Print();
+ goto silent_error;
+}
+
+
+static size_t
+write_callback(char *ptr, size_t size, size_t nmemb, void *stream)
+{
+ return util_write_callback(0, ptr, size, nmemb, stream);
+}
+
+static size_t
+header_callback(char *ptr, size_t size, size_t nmemb, void *stream)
+{
+ return util_write_callback(1, ptr, size, nmemb, stream);
+}
+
+
+static size_t
+read_callback(char *ptr, size_t size, size_t nmemb, void *stream)
+{
+ CurlObject *self;
+ PyThreadState *tmp_state;
+ PyObject *arglist;
+ PyObject *result = NULL;
+
+ size_t ret = CURL_READFUNC_ABORT; /* assume error, this actually works */
+ int total_size;
+
+ /* acquire thread */
+ self = (CurlObject *)stream;
+ tmp_state = get_thread_state(self);
+ if (tmp_state == NULL)
+ return ret;
+ PyEval_AcquireThread(tmp_state);
+
+ /* check args */
+ if (self->r_cb == NULL)
+ goto silent_error;
+ if (size <= 0 || nmemb <= 0)
+ goto done;
+ total_size = (int)(size * nmemb);
+ if (total_size < 0 || (size_t)total_size / size != nmemb) {
+ PyErr_SetString(ErrorObject, "integer overflow in read callback");
+ goto verbose_error;
+ }
+
+ /* run callback */
+ arglist = Py_BuildValue("(i)", total_size);
+ if (arglist == NULL)
+ goto verbose_error;
+ result = PyEval_CallObject(self->r_cb, arglist);
+ Py_DECREF(arglist);
+ if (result == NULL)
+ goto verbose_error;
+
+ /* handle result */
+ if (PyString_Check(result)) {
+ char *buf = NULL;
+ int obj_size = -1;
+ int r;
+ r = PyString_AsStringAndSize(result, &buf, &obj_size);
+ if (r != 0 || obj_size < 0 || obj_size > total_size) {
+ PyErr_Format(ErrorObject, "invalid return value for read callback %ld %ld", (long)obj_size, (long)total_size);
+ goto verbose_error;
+ }
+ memcpy(ptr, buf, obj_size);
+ ret = obj_size; /* success */
+ }
+ else if (PyInt_Check(result)) {
+ long r = PyInt_AsLong(result);
+ if (r != CURL_READFUNC_ABORT) {
+ goto type_error;
+ }
+ /* ret is CURL_READUNC_ABORT */
+ }
+ else if (PyLong_Check(result)) {
+ long r = PyLong_AsLong(result);
+ if (r != CURL_READFUNC_ABORT) {
+ goto type_error;
+ }
+ /* ret is CURL_READUNC_ABORT */
+ }
+ else {
+ type_error:
+ PyErr_SetString(ErrorObject, "read callback must return string");
+ goto verbose_error;
+ }
+
+done:
+silent_error:
+ Py_XDECREF(result);
+ PyEval_ReleaseThread(tmp_state);
+ return ret;
+verbose_error:
+ PyErr_Print();
+ goto silent_error;
+}
+
+
+static int
+progress_callback(void *stream,
+ double dltotal, double dlnow, double ultotal, double ulnow)
+{
+ CurlObject *self;
+ PyThreadState *tmp_state;
+ PyObject *arglist;
+ PyObject *result = NULL;
+ int ret = 1; /* assume error */
+
+ /* acquire thread */
+ self = (CurlObject *)stream;
+ tmp_state = get_thread_state(self);
+ if (tmp_state == NULL)
+ return ret;
+ PyEval_AcquireThread(tmp_state);
+
+ /* check args */
+ if (self->pro_cb == NULL)
+ goto silent_error;
+
+ /* run callback */
+ arglist = Py_BuildValue("(dddd)", dltotal, dlnow, ultotal, ulnow);
+ if (arglist == NULL)
+ goto verbose_error;
+ result = PyEval_CallObject(self->pro_cb, arglist);
+ Py_DECREF(arglist);
+ if (result == NULL)
+ goto verbose_error;
+
+ /* handle result */
+ if (result == Py_None) {
+ ret = 0; /* None means success */
+ }
+ else if (PyInt_Check(result)) {
+ ret = (int) PyInt_AsLong(result);
+ }
+ else {
+ ret = PyObject_IsTrue(result); /* FIXME ??? */
+ }
+
+silent_error:
+ Py_XDECREF(result);
+ PyEval_ReleaseThread(tmp_state);
+ return ret;
+verbose_error:
+ PyErr_Print();
+ goto silent_error;
+}
+
+
+static int
+debug_callback(CURL *curlobj, curl_infotype type,
+ char *buffer, size_t total_size, void *stream)
+{
+ CurlObject *self;
+ PyThreadState *tmp_state;
+ PyObject *arglist;
+ PyObject *result = NULL;
+ int ret = 0; /* always success */
+
+ UNUSED(curlobj);
+
+ /* acquire thread */
+ self = (CurlObject *)stream;
+ tmp_state = get_thread_state(self);
+ if (tmp_state == NULL)
+ return ret;
+ PyEval_AcquireThread(tmp_state);
+
+ /* check args */
+ if (self->debug_cb == NULL)
+ goto silent_error;
+ if ((int)total_size < 0 || (size_t)((int)total_size) != total_size) {
+ PyErr_SetString(ErrorObject, "integer overflow in debug callback");
+ goto verbose_error;
+ }
+
+ /* run callback */
+ arglist = Py_BuildValue("(is#)", (int)type, buffer, (int)total_size);
+ if (arglist == NULL)
+ goto verbose_error;
+ result = PyEval_CallObject(self->debug_cb, arglist);
+ Py_DECREF(arglist);
+ if (result == NULL)
+ goto verbose_error;
+
+ /* return values from debug callbacks should be ignored */
+
+silent_error:
+ Py_XDECREF(result);
+ PyEval_ReleaseThread(tmp_state);
+ return ret;
+verbose_error:
+ PyErr_Print();
+ goto silent_error;
+}
+
+
+static curlioerr
+ioctl_callback(CURL *curlobj, int cmd, void *stream)
+{
+ CurlObject *self;
+ PyThreadState *tmp_state;
+ PyObject *arglist;
+ PyObject *result = NULL;
+ int ret = CURLIOE_FAILRESTART; /* assume error */
+
+ UNUSED(curlobj);
+
+ /* acquire thread */
+ self = (CurlObject *)stream;
+ tmp_state = get_thread_state(self);
+ if (tmp_state == NULL)
+ return (curlioerr) ret;
+ PyEval_AcquireThread(tmp_state);
+
+ /* check args */
+ if (self->ioctl_cb == NULL)
+ goto silent_error;
+
+ /* run callback */
+ arglist = Py_BuildValue("(i)", (int)cmd);
+ if (arglist == NULL)
+ goto verbose_error;
+ result = PyEval_CallObject(self->ioctl_cb, arglist);
+ Py_DECREF(arglist);
+ if (result == NULL)
+ goto verbose_error;
+
+ /* handle result */
+ if (result == Py_None) {
+ ret = CURLIOE_OK; /* None means success */
+ }
+ else if (PyInt_Check(result)) {
+ ret = (int) PyInt_AsLong(result);
+ if (ret >= CURLIOE_LAST || ret < 0) {
+ PyErr_SetString(ErrorObject, "ioctl callback returned invalid value");
+ goto verbose_error;
+ }
+ }
+
+silent_error:
+ Py_XDECREF(result);
+ PyEval_ReleaseThread(tmp_state);
+ return (curlioerr) ret;
+verbose_error:
+ PyErr_Print();
+ goto silent_error;
+}
+
+
+/* --------------- unsetopt/setopt/getinfo --------------- */
+
+static PyObject *
+util_curl_unsetopt(CurlObject *self, int option)
+{
+ int res;
+ int opt_index = -1;
+
+#define SETOPT2(o,x) \
+ if ((res = curl_easy_setopt(self->handle, (o), (x))) != CURLE_OK) goto error
+#define SETOPT(x) SETOPT2((CURLoption)option, (x))
+
+ /* FIXME: implement more options. Have to carefully check lib/url.c in the
+ * libcurl source code to see if it's actually safe to simply
+ * unset the option. */
+ switch (option)
+ {
+ case CURLOPT_HTTPPOST:
+ SETOPT((void *) 0);
+ curl_formfree(self->httppost);
+ self->httppost = NULL;
+ /* FIXME: what about data->set.httpreq ?? */
+ break;
+ case CURLOPT_INFILESIZE:
+ SETOPT((long) -1);
+ break;
+ case CURLOPT_WRITEHEADER:
+ SETOPT((void *) 0);
+ ZAP(self->writeheader_fp);
+ break;
+ case CURLOPT_CAINFO:
+ case CURLOPT_CAPATH:
+ case CURLOPT_COOKIE:
+ case CURLOPT_COOKIEJAR:
+ case CURLOPT_CUSTOMREQUEST:
+ case CURLOPT_EGDSOCKET:
+ case CURLOPT_FTPPORT:
+ case CURLOPT_PROXYUSERPWD:
+ case CURLOPT_RANDOM_FILE:
+ case CURLOPT_SSL_CIPHER_LIST:
+ case CURLOPT_USERPWD:
+ SETOPT((char *) 0);
+ opt_index = OPT_INDEX(option);
+ break;
+
+ /* info: we explicitly list unsupported options here */
+ case CURLOPT_COOKIEFILE:
+ default:
+ PyErr_SetString(PyExc_TypeError, "unsetopt() is not supported for this option");
+ return NULL;
+ }
+
+ if (opt_index >= 0 && self->options[opt_index] != NULL) {
+ free(self->options[opt_index]);
+ self->options[opt_index] = NULL;
+ }
+
+ Py_INCREF(Py_None);
+ return Py_None;
+
+error:
+ CURLERROR_RETVAL();
+
+#undef SETOPT
+#undef SETOPT2
+}
+
+
+static PyObject *
+do_curl_unsetopt(CurlObject *self, PyObject *args)
+{
+ int option;
+
+ if (!PyArg_ParseTuple(args, "i:unsetopt", &option)) {
+ return NULL;
+ }
+ if (check_curl_state(self, 1 | 2, "unsetopt") != 0) {
+ return NULL;
+ }
+
+ /* early checks of option value */
+ if (option <= 0)
+ goto error;
+ if (option >= (int)CURLOPTTYPE_OFF_T + OPTIONS_SIZE)
+ goto error;
+ if (option % 10000 >= OPTIONS_SIZE)
+ goto error;
+
+ return util_curl_unsetopt(self, option);
+
+error:
+ PyErr_SetString(PyExc_TypeError, "invalid arguments to unsetopt");
+ return NULL;
+}
+
+
+static PyObject *
+do_curl_setopt(CurlObject *self, PyObject *args)
+{
+ int option;
+ PyObject *obj;
+ int res;
+
+ if (!PyArg_ParseTuple(args, "iO:setopt", &option, &obj))
+ return NULL;
+ if (check_curl_state(self, 1 | 2, "setopt") != 0)
+ return NULL;
+
+ /* early checks of option value */
+ if (option <= 0)
+ goto error;
+ if (option >= (int)CURLOPTTYPE_OFF_T + OPTIONS_SIZE)
+ goto error;
+ if (option % 10000 >= OPTIONS_SIZE)
+ goto error;
+
+#if 0 /* XXX - should we ??? */
+ /* Handle the case of None */
+ if (obj == Py_None) {
+ return util_curl_unsetopt(self, option);
+ }
+#endif
+
+ /* Handle the case of string arguments */
+ if (PyString_Check(obj)) {
+ char *str = NULL;
+ int len = -1;
+ char *buf;
+ int opt_index;
+
+ /* Check that the option specified a string as well as the input */
+ switch (option) {
+ case CURLOPT_CAINFO:
+ case CURLOPT_CAPATH:
+ case CURLOPT_COOKIE:
+ case CURLOPT_COOKIEFILE:
+ case CURLOPT_COOKIEJAR:
+ case CURLOPT_CUSTOMREQUEST:
+ case CURLOPT_EGDSOCKET:
+ case CURLOPT_ENCODING:
+ case CURLOPT_FTPPORT:
+ case CURLOPT_INTERFACE:
+ case CURLOPT_KRB4LEVEL:
+ case CURLOPT_NETRC_FILE:
+ case CURLOPT_PROXY:
+ case CURLOPT_PROXYUSERPWD:
+ case CURLOPT_RANDOM_FILE:
+ case CURLOPT_RANGE:
+ case CURLOPT_REFERER:
+ case CURLOPT_SSLCERT:
+ case CURLOPT_SSLCERTTYPE:
+ case CURLOPT_SSLENGINE:
+ case CURLOPT_SSLKEY:
+ case CURLOPT_SSLKEYPASSWD:
+ case CURLOPT_SSLKEYTYPE:
+ case CURLOPT_SSL_CIPHER_LIST:
+ case CURLOPT_URL:
+ case CURLOPT_USERAGENT:
+ case CURLOPT_USERPWD:
+ case CURLOPT_SOURCE_HOST:
+ case CURLOPT_SOURCE_USERPWD:
+ case CURLOPT_SOURCE_PATH:
+/* FIXME: check if more of these options allow binary data */
+ str = PyString_AsString_NoNUL(obj);
+ if (str == NULL)
+ return NULL;
+ break;
+ case CURLOPT_POSTFIELDS:
+ if (PyString_AsStringAndSize(obj, &str, &len) != 0)
+ return NULL;
+ /* automatically set POSTFIELDSIZE */
+ res = curl_easy_setopt(self->handle, CURLOPT_POSTFIELDSIZE, (long)len);
+ if (res != CURLE_OK) {
+ CURLERROR_RETVAL();
+ }
+ break;
+ default:
+ PyErr_SetString(PyExc_TypeError, "strings are not supported for this option");
+ return NULL;
+ }
+ /* Allocate memory to hold the string */
+ assert(str != NULL);
+ if (len <= 0)
+ buf = strdup(str);
+ else {
+ buf = (char *) malloc(len);
+ if (buf) memcpy(buf, str, len);
+ }
+ if (buf == NULL)
+ return PyErr_NoMemory();
+ /* Call setopt */
+ res = curl_easy_setopt(self->handle, (CURLoption)option, buf);
+ /* Check for errors */
+ if (res != CURLE_OK) {
+ free(buf);
+ CURLERROR_RETVAL();
+ }
+ /* Save allocated option buffer */
+ opt_index = OPT_INDEX(option);
+ if (self->options[opt_index] != NULL) {
+ free(self->options[opt_index]);
+ self->options[opt_index] = NULL;
+ }
+ self->options[opt_index] = buf;
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+
+#define IS_LONG_OPTION(o) (o < CURLOPTTYPE_OBJECTPOINT)
+#define IS_OFF_T_OPTION(o) (o >= CURLOPTTYPE_OFF_T)
+
+ /* Handle the case of integer arguments */
+ if (PyInt_Check(obj)) {
+ long d = PyInt_AsLong(obj);
+
+ if (IS_LONG_OPTION(option))
+ res = curl_easy_setopt(self->handle, (CURLoption)option, (long)d);
+ else if (IS_OFF_T_OPTION(option))
+ res = curl_easy_setopt(self->handle, (CURLoption)option, (curl_off_t)d);
+ else {
+ PyErr_SetString(PyExc_TypeError, "integers are not supported for this option");
+ return NULL;
+ }
+ if (res != CURLE_OK) {
+ CURLERROR_RETVAL();
+ }
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+
+ /* Handle the case of long arguments (used by *_LARGE options) */
+ if (PyLong_Check(obj)) {
+ PY_LONG_LONG d = PyLong_AsLongLong(obj);
+ if (d == -1 && PyErr_Occurred())
+ return NULL;
+
+ if (IS_LONG_OPTION(option) && (long)d == d)
+ res = curl_easy_setopt(self->handle, (CURLoption)option, (long)d);
+ else if (IS_OFF_T_OPTION(option) && (curl_off_t)d == d)
+ res = curl_easy_setopt(self->handle, (CURLoption)option, (curl_off_t)d);
+ else {
+ PyErr_SetString(PyExc_TypeError, "longs are not supported for this option");
+ return NULL;
+ }
+ if (res != CURLE_OK) {
+ CURLERROR_RETVAL();
+ }
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+
+#undef IS_LONG_OPTION
+#undef IS_OFF_T_OPTION
+
+ /* Handle the case of file objects */
+ if (PyFile_Check(obj)) {
+ FILE *fp;
+
+ /* Ensure the option specified a file as well as the input */
+ switch (option) {
+ case CURLOPT_READDATA:
+ case CURLOPT_WRITEDATA:
+ break;
+ case CURLOPT_WRITEHEADER:
+ if (self->w_cb != NULL) {
+ PyErr_SetString(ErrorObject, "cannot combine WRITEHEADER with WRITEFUNCTION.");
+ return NULL;
+ }
+ break;
+ default:
+ PyErr_SetString(PyExc_TypeError, "files are not supported for this option");
+ return NULL;
+ }
+
+ fp = PyFile_AsFile(obj);
+ if (fp == NULL) {
+ PyErr_SetString(PyExc_TypeError, "second argument must be open file");
+ return NULL;
+ }
+ res = curl_easy_setopt(self->handle, (CURLoption)option, fp);
+ if (res != CURLE_OK) {
+ CURLERROR_RETVAL();
+ }
+ Py_INCREF(obj);
+
+ switch (option) {
+ case CURLOPT_READDATA:
+ ZAP(self->readdata_fp);
+ self->readdata_fp = obj;
+ break;
+ case CURLOPT_WRITEDATA:
+ ZAP(self->writedata_fp);
+ self->writedata_fp = obj;
+ break;
+ case CURLOPT_WRITEHEADER:
+ ZAP(self->writeheader_fp);
+ self->writeheader_fp = obj;
+ break;
+ default:
+ assert(0);
+ break;
+ }
+ /* Return success */
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+
+ /* Handle the case of list objects */
+ if (PyList_Check(obj)) {
+ struct curl_slist **old_slist = NULL;
+ struct curl_slist *slist = NULL;
+ int i, len;
+
+ switch (option) {
+ case CURLOPT_HTTP200ALIASES:
+ old_slist = &self->http200aliases;
+ break;
+ case CURLOPT_HTTPHEADER:
+ old_slist = &self->httpheader;
+ break;
+ case CURLOPT_QUOTE:
+ old_slist = &self->quote;
+ break;
+ case CURLOPT_POSTQUOTE:
+ old_slist = &self->postquote;
+ break;
+ case CURLOPT_PREQUOTE:
+ old_slist = &self->prequote;
+ break;
+ case CURLOPT_SOURCE_PREQUOTE:
+ old_slist = &self->source_prequote;
+ break;
+ case CURLOPT_SOURCE_POSTQUOTE:
+ old_slist = &self->source_postquote;
+ break;
+ case CURLOPT_HTTPPOST:
+ break;
+ default:
+ /* None of the list options were recognized, throw exception */
+ PyErr_SetString(PyExc_TypeError, "lists are not supported for this option");
+ return NULL;
+ }
+
+ len = PyList_Size(obj);
+ if (len == 0) {
+ /* Empty list - do nothing */
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+
+ /* Handle HTTPPOST different since we construct a HttpPost form struct */
+ if (option == CURLOPT_HTTPPOST) {
+ struct curl_httppost *post = NULL;
+ struct curl_httppost *last = NULL;
+
+ for (i = 0; i < len; i++) {
+ char *nstr = NULL, *cstr = NULL;
+ int nlen = -1, clen = -1;
+ PyObject *listitem = PyList_GetItem(obj, i);
+
+ if (!PyTuple_Check(listitem)) {
+ curl_formfree(post);
+ PyErr_SetString(PyExc_TypeError, "list items must be tuple objects");
+ return NULL;
+ }
+ if (PyTuple_GET_SIZE(listitem) != 2) {
+ curl_formfree(post);
+ PyErr_SetString(PyExc_TypeError, "tuple must contain two elements (name, value)");
+ return NULL;
+ }
+ if (PyString_AsStringAndSize(PyTuple_GET_ITEM(listitem, 0), &nstr, &nlen) != 0) {
+ curl_formfree(post);
+ PyErr_SetString(PyExc_TypeError, "tuple must contain string as first element");
+ return NULL;
+ }
+ if (PyString_Check(PyTuple_GET_ITEM(listitem, 1))) {
+ /* Handle strings as second argument for backwards compatibility */
+ PyString_AsStringAndSize(PyTuple_GET_ITEM(listitem, 1), &cstr, &clen);
+ /* INFO: curl_formadd() internally does memdup() the data, so
+ * embedded NUL characters _are_ allowed here. */
+ res = curl_formadd(&post, &last,
+ CURLFORM_COPYNAME, nstr,
+ CURLFORM_NAMELENGTH, (long) nlen,
+ CURLFORM_COPYCONTENTS, cstr,
+ CURLFORM_CONTENTSLENGTH, (long) clen,
+ CURLFORM_END);
+ if (res != CURLE_OK) {
+ curl_formfree(post);
+ CURLERROR_RETVAL();
+ }
+ }
+ else if (PyTuple_Check(PyTuple_GET_ITEM(listitem, 1))) {
+ /* Supports content, file and content-type */
+ PyObject *t = PyTuple_GET_ITEM(listitem, 1);
+ int tlen = PyTuple_Size(t);
+ int j, k, l;
+ struct curl_forms *forms = NULL;
+
+ /* Sanity check that there are at least two tuple items */
+ if (tlen < 2) {
+ curl_formfree(post);
+ PyErr_SetString(PyExc_TypeError, "tuple must contain at least one option and one value");
+ return NULL;
+ }
+
+ /* Allocate enough space to accommodate length options for content */
+ forms = PyMem_Malloc(sizeof(struct curl_forms) * ((tlen*2) + 1));
+ if (forms == NULL) {
+ curl_formfree(post);
+ PyErr_NoMemory();
+ return NULL;
+ }
+
+ /* Iterate all the tuple members pairwise */
+ for (j = 0, k = 0, l = 0; j < tlen; j += 2, l++) {
+ char *ostr;
+ int olen, val;
+
+ if (j == (tlen-1)) {
+ PyErr_SetString(PyExc_TypeError, "expected value");
+ PyMem_Free(forms);
+ curl_formfree(post);
+ return NULL;
+ }
+ if (!PyInt_Check(PyTuple_GET_ITEM(t, j))) {
+ PyErr_SetString(PyExc_TypeError, "option must be long");
+ PyMem_Free(forms);
+ curl_formfree(post);
+ return NULL;
+ }
+ if (!PyString_Check(PyTuple_GET_ITEM(t, j+1))) {
+ PyErr_SetString(PyExc_TypeError, "value must be string");
+ PyMem_Free(forms);
+ curl_formfree(post);
+ return NULL;
+ }
+
+ val = PyLong_AsLong(PyTuple_GET_ITEM(t, j));
+ if (val != CURLFORM_COPYCONTENTS &&
+ val != CURLFORM_FILE &&
+ val != CURLFORM_CONTENTTYPE)
+ {
+ PyErr_SetString(PyExc_TypeError, "unsupported option");
+ PyMem_Free(forms);
+ curl_formfree(post);
+ return NULL;
+ }
+ PyString_AsStringAndSize(PyTuple_GET_ITEM(t, j+1), &ostr, &olen);
+ forms[k].option = val;
+ forms[k].value = ostr;
+ ++k;
+ if (val == CURLFORM_COPYCONTENTS) {
+ /* Contents can contain \0 bytes so we specify the length */
+ forms[k].option = CURLFORM_CONTENTSLENGTH;
+ forms[k].value = (char *)olen;
+ ++k;
+ }
+ }
+ forms[k].option = CURLFORM_END;
+ res = curl_formadd(&post, &last,
+ CURLFORM_COPYNAME, nstr,
+ CURLFORM_NAMELENGTH, (long) nlen,
+ CURLFORM_ARRAY, forms,
+ CURLFORM_END);
+ PyMem_Free(forms);
+ if (res != CURLE_OK) {
+ curl_formfree(post);
+ CURLERROR_RETVAL();
+ }
+ } else {
+ /* Some other type was given, ignore */
+ curl_formfree(post);
+ PyErr_SetString(PyExc_TypeError, "unsupported second type in tuple");
+ return NULL;
+ }
+ }
+ res = curl_easy_setopt(self->handle, CURLOPT_HTTPPOST, post);
+ /* Check for errors */
+ if (res != CURLE_OK) {
+ curl_formfree(post);
+ CURLERROR_RETVAL();
+ }
+ /* Finally, free previously allocated httppost and update */
+ curl_formfree(self->httppost);
+ self->httppost = post;
+
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+
+ /* Just to be sure we do not bug off here */
+ assert(old_slist != NULL && slist == NULL);
+
+ /* Handle regular list operations on the other options */
+ for (i = 0; i < len; i++) {
+ PyObject *listitem = PyList_GetItem(obj, i);
+ struct curl_slist *nlist;
+ char *str;
+
+ if (!PyString_Check(listitem)) {
+ curl_slist_free_all(slist);
+ PyErr_SetString(PyExc_TypeError, "list items must be string objects");
+ return NULL;
+ }
+ /* INFO: curl_slist_append() internally does strdup() the data, so
+ * no embedded NUL characters allowed here. */
+ str = PyString_AsString_NoNUL(listitem);
+ if (str == NULL) {
+ curl_slist_free_all(slist);
+ return NULL;
+ }
+ nlist = curl_slist_append(slist, str);
+ if (nlist == NULL || nlist->data == NULL) {
+ curl_slist_free_all(slist);
+ return PyErr_NoMemory();
+ }
+ slist = nlist;
+ }
+ res = curl_easy_setopt(self->handle, (CURLoption)option, slist);
+ /* Check for errors */
+ if (res != CURLE_OK) {
+ curl_slist_free_all(slist);
+ CURLERROR_RETVAL();
+ }
+ /* Finally, free previously allocated list and update */
+ curl_slist_free_all(*old_slist);
+ *old_slist = slist;
+
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+
+ /* Handle the case of function objects for callbacks */
+ if (PyFunction_Check(obj) || PyCFunction_Check(obj) || PyMethod_Check(obj)) {
+ /* We use function types here to make sure that our callback
+ * definitions exactly match the <curl/curl.h> interface.
+ */
+ const curl_write_callback w_cb = write_callback;
+ const curl_write_callback h_cb = header_callback;
+ const curl_read_callback r_cb = read_callback;
+ const curl_progress_callback pro_cb = progress_callback;
+ const curl_debug_callback debug_cb = debug_callback;
+ const curl_ioctl_callback ioctl_cb = ioctl_callback;
+
+ switch(option) {
+ case CURLOPT_WRITEFUNCTION:
+ if (self->writeheader_fp != NULL) {
+ PyErr_SetString(ErrorObject, "cannot combine WRITEFUNCTION with WRITEHEADER option.");
+ return NULL;
+ }
+ Py_INCREF(obj);
+ ZAP(self->writedata_fp);
+ ZAP(self->w_cb);
+ self->w_cb = obj;
+ curl_easy_setopt(self->handle, CURLOPT_WRITEFUNCTION, w_cb);
+ curl_easy_setopt(self->handle, CURLOPT_WRITEDATA, self);
+ break;
+ case CURLOPT_HEADERFUNCTION:
+ Py_INCREF(obj);
+ ZAP(self->h_cb);
+ self->h_cb = obj;
+ curl_easy_setopt(self->handle, CURLOPT_HEADERFUNCTION, h_cb);
+ curl_easy_setopt(self->handle, CURLOPT_WRITEHEADER, self);
+ break;
+ case CURLOPT_READFUNCTION:
+ Py_INCREF(obj);
+ ZAP(self->readdata_fp);
+ ZAP(self->r_cb);
+ self->r_cb = obj;
+ curl_easy_setopt(self->handle, CURLOPT_READFUNCTION, r_cb);
+ curl_easy_setopt(self->handle, CURLOPT_READDATA, self);
+ break;
+ case CURLOPT_PROGRESSFUNCTION:
+ Py_INCREF(obj);
+ ZAP(self->pro_cb);
+ self->pro_cb = obj;
+ curl_easy_setopt(self->handle, CURLOPT_PROGRESSFUNCTION, pro_cb);
+ curl_easy_setopt(self->handle, CURLOPT_PROGRESSDATA, self);
+ break;
+ case CURLOPT_DEBUGFUNCTION:
+ Py_INCREF(obj);
+ ZAP(self->debug_cb);
+ self->debug_cb = obj;
+ curl_easy_setopt(self->handle, CURLOPT_DEBUGFUNCTION, debug_cb);
+ curl_easy_setopt(self->handle, CURLOPT_DEBUGDATA, self);
+ break;
+ case CURLOPT_IOCTLFUNCTION:
+ Py_INCREF(obj);
+ ZAP(self->ioctl_cb);
+ self->ioctl_cb = obj;
+ curl_easy_setopt(self->handle, CURLOPT_IOCTLFUNCTION, ioctl_cb);
+ curl_easy_setopt(self->handle, CURLOPT_IOCTLDATA, self);
+ break;
+
+ default:
+ /* None of the function options were recognized, throw exception */
+ PyErr_SetString(PyExc_TypeError, "functions are not supported for this option");
+ return NULL;
+ }
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+
+ /* Failed to match any of the function signatures -- return error */
+error:
+ PyErr_SetString(PyExc_TypeError, "invalid arguments to setopt");
+ return NULL;
+}
+
+
+static PyObject *
+do_curl_getinfo(CurlObject *self, PyObject *args)
+{
+ int option;
+ int res;
+
+ if (!PyArg_ParseTuple(args, "i:getinfo", &option)) {
+ return NULL;
+ }
+ if (check_curl_state(self, 1 | 2, "getinfo") != 0) {
+ return NULL;
+ }
+
+ switch (option) {
+ case CURLINFO_FILETIME:
+ case CURLINFO_HEADER_SIZE:
+ case CURLINFO_HTTP_CODE:
+ case CURLINFO_REDIRECT_COUNT:
+ case CURLINFO_REQUEST_SIZE:
+ case CURLINFO_SSL_VERIFYRESULT:
+ case CURLINFO_HTTP_CONNECTCODE:
+ case CURLINFO_HTTPAUTH_AVAIL:
+ case CURLINFO_PROXYAUTH_AVAIL:
+ case CURLINFO_OS_ERRNO:
+ case CURLINFO_NUM_CONNECTS:
+ {
+ /* Return PyInt as result */
+ long l_res = -1;
+
+ res = curl_easy_getinfo(self->handle, (CURLINFO)option, &l_res);
+ /* Check for errors and return result */
+ if (res != CURLE_OK) {
+ CURLERROR_RETVAL();
+ }
+ return PyInt_FromLong(l_res);
+ }
+
+ case CURLINFO_CONTENT_TYPE:
+ case CURLINFO_EFFECTIVE_URL:
+ {
+ /* Return PyString as result */
+ char *s_res = NULL;
+
+ res = curl_easy_getinfo(self->handle, (CURLINFO)option, &s_res);
+ if (res != CURLE_OK) {
+ CURLERROR_RETVAL();
+ }
+ /* If the resulting string is NULL, return None */
+ if (s_res == NULL) {
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+ return PyString_FromString(s_res);
+ }
+
+ case CURLINFO_CONNECT_TIME:
+ case CURLINFO_CONTENT_LENGTH_DOWNLOAD:
+ case CURLINFO_CONTENT_LENGTH_UPLOAD:
+ case CURLINFO_NAMELOOKUP_TIME:
+ case CURLINFO_PRETRANSFER_TIME:
+ case CURLINFO_REDIRECT_TIME:
+ case CURLINFO_SIZE_DOWNLOAD:
+ case CURLINFO_SIZE_UPLOAD:
+ case CURLINFO_SPEED_DOWNLOAD:
+ case CURLINFO_SPEED_UPLOAD:
+ case CURLINFO_STARTTRANSFER_TIME:
+ case CURLINFO_TOTAL_TIME:
+ {
+ /* Return PyFloat as result */
+ double d_res = 0.0;
+
+ res = curl_easy_getinfo(self->handle, (CURLINFO)option, &d_res);
+ if (res != CURLE_OK) {
+ CURLERROR_RETVAL();
+ }
+ return PyFloat_FromDouble(d_res);
+ }
+
+ case CURLINFO_SSL_ENGINES:
+ {
+ /* Return a list of strings */
+ struct curl_slist *slist = NULL;
+
+ res = curl_easy_getinfo(self->handle, (CURLINFO)option, &slist);
+ if (res != CURLE_OK) {
+ CURLERROR_RETVAL();
+ }
+ return convert_slist(slist, 1 | 2);
+ }
+ }
+
+ /* Got wrong option on the method call */
+ PyErr_SetString(PyExc_ValueError, "invalid argument to getinfo");
+ return NULL;
+}
+
+
+/*************************************************************************
+// CurlMultiObject
+**************************************************************************/
+
+/* --------------- construct/destruct (i.e. open/close) --------------- */
+
+/* constructor - this is a module-level function returning a new instance */
+static CurlMultiObject *
+do_multi_new(PyObject *dummy)
+{
+ CurlMultiObject *self;
+
+ UNUSED(dummy);
+
+ /* Allocate python curl-multi object */
+ self = (CurlMultiObject *) PyObject_GC_New(CurlMultiObject, p_CurlMulti_Type);
+ if (self) {
+ PyObject_GC_Track(self);
+ }
+ else {
+ return NULL;
+ }
+
+ /* Initialize object attributes */
+ self->dict = NULL;
+ self->state = NULL;
+
+ /* Allocate libcurl multi handle */
+ self->multi_handle = curl_multi_init();
+ if (self->multi_handle == NULL) {
+ Py_DECREF(self);
+ PyErr_SetString(ErrorObject, "initializing curl-multi failed");
+ return NULL;
+ }
+ return self;
+}
+
+
+static void
+util_multi_close(CurlMultiObject *self)
+{
+ assert(self != NULL);
+ self->state = NULL;
+ if (self->multi_handle != NULL) {
+ CURLM *multi_handle = self->multi_handle;
+ self->multi_handle = NULL;
+ curl_multi_cleanup(multi_handle);
+ }
+}
+
+
+static void
+do_multi_dealloc(CurlMultiObject *self)
+{
+ PyObject_GC_UnTrack(self);
+ Py_TRASHCAN_SAFE_BEGIN(self)
+
+ ZAP(self->dict);
+ util_multi_close(self);
+
+ PyObject_GC_Del(self);
+ Py_TRASHCAN_SAFE_END(self)
+}
+
+
+static PyObject *
+do_multi_close(CurlMultiObject *self)
+{
+ if (check_multi_state(self, 2, "close") != 0) {
+ return NULL;
+ }
+ util_multi_close(self);
+ Py_INCREF(Py_None);
+ return Py_None;
+}
+
+
+/* --------------- GC support --------------- */
+
+/* Drop references that may have created reference cycles. */
+static int
+do_multi_clear(CurlMultiObject *self)
+{
+ ZAP(self->dict);
+ return 0;
+}
+
+static int
+do_multi_traverse(CurlMultiObject *self, visitproc visit, void *arg)
+{
+ int err;
+#undef VISIT
+#define VISIT(v) if ((v) != NULL && ((err = visit(v, arg)) != 0)) return err
+
+ VISIT(self->dict);
+
+ return 0;
+#undef VISIT
+}
+
+/* --------------- perform --------------- */
+
+
+static PyObject *
+do_multi_perform(CurlMultiObject *self)
+{
+ CURLMcode res;
+ int running = -1;
+
+ if (check_multi_state(self, 1 | 2, "perform") != 0) {
+ return NULL;
+ }
+
+ /* Release global lock and start */
+ self->state = PyThreadState_Get();
+ assert(self->state != NULL);
+ Py_BEGIN_ALLOW_THREADS
+ res = curl_multi_perform(self->multi_handle, &running);
+ Py_END_ALLOW_THREADS
+ self->state = NULL;
+
+ /* We assume these errors are ok, otherwise throw exception */
+ if (res != CURLM_OK && res != CURLM_CALL_MULTI_PERFORM) {
+ CURLERROR_MSG("perform failed");
+ }
+
+ /* Return a tuple with the result and the number of running handles */
+ return Py_BuildValue("(ii)", (int)res, running);
+}
+
+
+/* --------------- add_handle/remove_handle --------------- */
+
+/* static utility function */
+static int
+check_multi_add_remove(const CurlMultiObject *self, const CurlObject *obj)
+{
+ /* check CurlMultiObject status */
+ assert_multi_state(self);
+ if (self->multi_handle == NULL) {
+ PyErr_SetString(ErrorObject, "cannot add/remove handle - multi-stack is closed");
+ return -1;
+ }
+ if (self->state != NULL) {
+ PyErr_SetString(ErrorObject, "cannot add/remove handle - multi_perform() already running");
+ return -1;
+ }
+ /* check CurlObject status */
+ assert_curl_state(obj);
+ if (obj->state != NULL) {
+ PyErr_SetString(ErrorObject, "cannot add/remove handle - perform() of curl object already running");
+ return -1;
+ }
+ if (obj->multi_stack != NULL && obj->multi_stack != self) {
+ PyErr_SetString(ErrorObject, "cannot add/remove handle - curl object already on another multi-stack");
+ return -1;
+ }
+ return 0;
+}
+
+
+static PyObject *
+do_multi_add_handle(CurlMultiObject *self, PyObject *args)
+{
+ CurlObject *obj;
+ CURLMcode res;
+
+ if (!PyArg_ParseTuple(args, "O!:add_handle", p_Curl_Type, &obj)) {
+ return NULL;
+ }
+ if (check_multi_add_remove(self, obj) != 0) {
+ return NULL;
+ }
+ if (obj->handle == NULL) {
+ PyErr_SetString(ErrorObject, "curl object already closed");
+ return NULL;
+ }
+ if (obj->multi_stack == self) {
+ PyErr_SetString(ErrorObject, "curl object already on this multi-stack");
+ return NULL;
+ }
+ assert(obj->multi_stack == NULL);
+ res = curl_multi_add_handle(self->multi_handle, obj->handle);
+ if (res != CURLM_OK) {
+ CURLERROR_MSG("curl_multi_add_handle() failed due to internal errors");
+ }
+ obj->multi_stack = self;
+ Py_INCREF(self);
+ Py_INCREF(Py_None);
+ return Py_None;
+}
+
+
+static PyObject *
+do_multi_remove_handle(CurlMultiObject *self, PyObject *args)
+{
+ CurlObject *obj;
+ CURLMcode res;
+
+ if (!PyArg_ParseTuple(args, "O!:remove_handle", p_Curl_Type, &obj)) {
+ return NULL;
+ }
+ if (check_multi_add_remove(self, obj) != 0) {
+ return NULL;
+ }
+ if (obj->handle == NULL) {
+ /* CurlObject handle already closed -- ignore */
+ goto done;
+ }
+ if (obj->multi_stack != self) {
+ PyErr_SetString(ErrorObject, "curl object not on this multi-stack");
+ return NULL;
+ }
+ res = curl_multi_remove_handle(self->multi_handle, obj->handle);
+ if (res != CURLM_OK) {
+ CURLERROR_MSG("curl_multi_remove_handle() failed due to internal errors");
+ }
+ assert(obj->multi_stack == self);
+ obj->multi_stack = NULL;
+ Py_DECREF(self);
+done:
+ Py_INCREF(Py_None);
+ return Py_None;
+}
+
+
+/* --------------- fdset ---------------------- */
+
+static PyObject *
+do_multi_fdset(CurlMultiObject *self)
+{
+ CURLMcode res;
+ int max_fd = -1, fd;
+ PyObject *ret = NULL;
+ PyObject *read_list = NULL, *write_list = NULL, *except_list = NULL;
+ PyObject *py_fd = NULL;
+
+ if (check_multi_state(self, 1 | 2, "fdset") != 0) {
+ return NULL;
+ }
+
+ /* Clear file descriptor sets */
+ FD_ZERO(&self->read_fd_set);
+ FD_ZERO(&self->write_fd_set);
+ FD_ZERO(&self->exc_fd_set);
+
+ /* Don't bother releasing the gil as this is just a data structure operation */
+ res = curl_multi_fdset(self->multi_handle, &self->read_fd_set,
+ &self->write_fd_set, &self->exc_fd_set, &max_fd);
+ if (res != CURLM_OK || max_fd < 0) {
+ CURLERROR_MSG("curl_multi_fdset() failed due to internal errors");
+ }
+
+ /* Allocate lists. */
+ if ((read_list = PyList_New(0)) == NULL) goto error;
+ if ((write_list = PyList_New(0)) == NULL) goto error;
+ if ((except_list = PyList_New(0)) == NULL) goto error;
+
+ /* Populate lists */
+ for (fd = 0; fd < max_fd + 1; fd++) {
+ if (FD_ISSET(fd, &self->read_fd_set)) {
+ if ((py_fd = PyInt_FromLong((long)fd)) == NULL) goto error;
+ if (PyList_Append(read_list, py_fd) != 0) goto error;
+ Py_DECREF(py_fd);
+ py_fd = NULL;
+ }
+ if (FD_ISSET(fd, &self->write_fd_set)) {
+ if ((py_fd = PyInt_FromLong((long)fd)) == NULL) goto error;
+ if (PyList_Append(write_list, py_fd) != 0) goto error;
+ Py_DECREF(py_fd);
+ py_fd = NULL;
+ }
+ if (FD_ISSET(fd, &self->exc_fd_set)) {
+ if ((py_fd = PyInt_FromLong((long)fd)) == NULL) goto error;
+ if (PyList_Append(except_list, py_fd) != 0) goto error;
+ Py_DECREF(py_fd);
+ py_fd = NULL;
+ }
+ }
+
+ /* Return a tuple with the 3 lists */
+ ret = Py_BuildValue("(OOO)", read_list, write_list, except_list);
+error:
+ Py_XDECREF(py_fd);
+ Py_XDECREF(except_list);
+ Py_XDECREF(write_list);
+ Py_XDECREF(read_list);
+ return ret;
+}
+
+
+/* --------------- info_read --------------- */
+
+static PyObject *
+do_multi_info_read(CurlMultiObject *self, PyObject *args)
+{
+ PyObject *ret = NULL;
+ PyObject *ok_list = NULL, *err_list = NULL;
+ CURLMsg *msg;
+ int in_queue = 0, num_results = INT_MAX;
+
+ /* Sanity checks */
+ if (!PyArg_ParseTuple(args, "|i:info_read", &num_results)) {
+ return NULL;
+ }
+ if (num_results <= 0) {
+ PyErr_SetString(ErrorObject, "argument to info_read must be greater than zero");
+ return NULL;
+ }
+ if (check_multi_state(self, 1 | 2, "info_read") != 0) {
+ return NULL;
+ }
+
+ if ((ok_list = PyList_New(0)) == NULL) goto error;
+ if ((err_list = PyList_New(0)) == NULL) goto error;
+
+ /* Loop through all messages */
+ while ((msg = curl_multi_info_read(self->multi_handle, &in_queue)) != NULL) {
+ CURLcode res;
+ CurlObject *co = NULL;
+
+ /* Check for termination as specified by the user */
+ if (num_results-- <= 0) {
+ break;
+ }
+
+ /* Fetch the curl object that corresponds to the curl handle in the message */
+ res = curl_easy_getinfo(msg->easy_handle, CURLINFO_PRIVATE, &co);
+ if (res != CURLE_OK || co == NULL) {
+ Py_DECREF(err_list);
+ Py_DECREF(ok_list);
+ CURLERROR_MSG("Unable to fetch curl handle from curl object");
+ }
+ assert(co->ob_type == p_Curl_Type);
+ if (msg->msg != CURLMSG_DONE) {
+ /* FIXME: what does this mean ??? */
+ }
+ if (msg->data.result == CURLE_OK) {
+ /* Append curl object to list of objects which succeeded */
+ if (PyList_Append(ok_list, (PyObject *)co) != 0) {
+ goto error;
+ }
+ }
+ else {
+ /* Create a result tuple that will get added to err_list. */
+ PyObject *v = Py_BuildValue("(Ois)", (PyObject *)co, (int)msg->data.result, co->error);
+ /* Append curl object to list of objects which failed */
+ if (v == NULL || PyList_Append(err_list, v) != 0) {
+ Py_XDECREF(v);
+ goto error;
+ }
+ Py_DECREF(v);
+ }
+ }
+ /* Return (number of queued messages, [ok_objects], [error_objects]) */
+ ret = Py_BuildValue("(iOO)", in_queue, ok_list, err_list);
+error:
+ Py_XDECREF(err_list);
+ Py_XDECREF(ok_list);
+ return ret;
+}
+
+
+/* --------------- select --------------- */
+
+static PyObject *
+do_multi_select(CurlMultiObject *self, PyObject *args)
+{
+ int max_fd = -1, n;
+ double timeout = -1.0;
+ struct timeval tv, *tvp;
+ CURLMcode res;
+
+ if (!PyArg_ParseTuple(args, "|d:select", &timeout)) {
+ return NULL;
+ }
+ if (check_multi_state(self, 1 | 2, "select") != 0) {
+ return NULL;
+ }
+
+ if (timeout == -1.0) {
+ /* no timeout given - wait forever */
+ tvp = NULL;
+ } else if (timeout < 0 || timeout >= 365 * 24 * 60 * 60) {
+ PyErr_SetString(PyExc_OverflowError, "invalid timeout period");
+ return NULL;
+ } else {
+ long seconds = (long)timeout;
+ timeout = timeout - (double)seconds;
+ assert(timeout >= 0.0); assert(timeout < 1.0);
+ tv.tv_sec = seconds;
+ tv.tv_usec = (long)(timeout*1000000.0);
+ tvp = &tv;
+ }
+
+ FD_ZERO(&self->read_fd_set);
+ FD_ZERO(&self->write_fd_set);
+ FD_ZERO(&self->exc_fd_set);
+
+ res = curl_multi_fdset(self->multi_handle, &self->read_fd_set,
+ &self->write_fd_set, &self->exc_fd_set, &max_fd);
+ if (res != CURLM_OK) {
+ CURLERROR_MSG("multi_fdset failed");
+ }
+
+ if (max_fd < 0) {
+ n = 0;
+ }
+ else {
+ Py_BEGIN_ALLOW_THREADS
+ n = select(max_fd + 1, &self->read_fd_set, &self->write_fd_set, &self->exc_fd_set, tvp);
+ Py_END_ALLOW_THREADS
+ /* info: like Python's socketmodule.c we do not raise an exception
+ * if select() fails - we'll leave it to the actual libcurl
+ * socket code to report any errors.
+ */
+ }
+
+ return PyInt_FromLong(n);
+}
+
+
+/*************************************************************************
+// type definitions
+**************************************************************************/
+
+/* --------------- methods --------------- */
+
+static char co_close_doc [] = "close() -> None. Close handle and end curl session.\n";
+static char co_errstr_doc [] = "errstr() -> String. Return the internal libcurl error buffer string.\n";
+static char co_getinfo_doc [] = "getinfo(info) -> Res. Extract and return information from a curl session. Throws pycurl.error exception upon failure.\n";
+static char co_perform_doc [] = "perform() -> None. Perform a file transfer. Throws pycurl.error exception upon failure.\n";
+static char co_setopt_doc [] = "setopt(option, parameter) -> None. Set curl session option. Throws pycurl.error exception upon failure.\n";
+static char co_unsetopt_doc [] = "unsetopt(option) -> None. Reset curl session option to default value. Throws pycurl.error exception upon failure.\n";
+
+static char co_multi_fdset_doc [] = "fdset() -> Tuple. Returns a tuple of three lists that can be passed to the select.select() method .\n";
+static char co_multi_info_read_doc [] = "info_read([max_objects]) -> Tuple. Returns a tuple (number of queued handles, [curl objects]).\n";
+static char co_multi_select_doc [] = "select([timeout]) -> Int. Returns result from doing a select() on the curl multi file descriptor with the given timeout.\n";
+
+static PyMethodDef curlobject_methods[] = {
+ {"close", (PyCFunction)do_curl_close, METH_NOARGS, co_close_doc},
+ {"errstr", (PyCFunction)do_curl_errstr, METH_NOARGS, co_errstr_doc},
+ {"getinfo", (PyCFunction)do_curl_getinfo, METH_VARARGS, co_getinfo_doc},
+ {"perform", (PyCFunction)do_curl_perform, METH_NOARGS, co_perform_doc},
+ {"setopt", (PyCFunction)do_curl_setopt, METH_VARARGS, co_setopt_doc},
+ {"unsetopt", (PyCFunction)do_curl_unsetopt, METH_VARARGS, co_unsetopt_doc},
+ {NULL, NULL, 0, NULL}
+};
+
+static PyMethodDef curlmultiobject_methods[] = {
+ {"add_handle", (PyCFunction)do_multi_add_handle, METH_VARARGS, NULL},
+ {"close", (PyCFunction)do_multi_close, METH_NOARGS, NULL},
+ {"fdset", (PyCFunction)do_multi_fdset, METH_NOARGS, co_multi_fdset_doc},
+ {"info_read", (PyCFunction)do_multi_info_read, METH_VARARGS, co_multi_info_read_doc},
+ {"perform", (PyCFunction)do_multi_perform, METH_NOARGS, NULL},
+ {"remove_handle", (PyCFunction)do_multi_remove_handle, METH_VARARGS, NULL},
+ {"select", (PyCFunction)do_multi_select, METH_VARARGS, co_multi_select_doc},
+ {NULL, NULL, 0, NULL}
+};
+
+
+/* --------------- setattr/getattr --------------- */
+
+static PyObject *curlobject_constants = NULL;
+static PyObject *curlmultiobject_constants = NULL;
+
+static int
+my_setattr(PyObject **dict, char *name, PyObject *v)
+{
+ if (v == NULL) {
+ int rv = -1;
+ if (*dict != NULL)
+ rv = PyDict_DelItemString(*dict, name);
+ if (rv < 0)
+ PyErr_SetString(PyExc_AttributeError, "delete non-existing attribute");
+ return rv;
+ }
+ if (*dict == NULL) {
+ *dict = PyDict_New();
+ if (*dict == NULL)
+ return -1;
+ }
+ return PyDict_SetItemString(*dict, name, v);
+}
+
+static PyObject *
+my_getattr(PyObject *co, char *name, PyObject *dict1, PyObject *dict2, PyMethodDef *m)
+{
+ PyObject *v = NULL;
+ if (v == NULL && dict1 != NULL)
+ v = PyDict_GetItemString(dict1, name);
+ if (v == NULL && dict2 != NULL)
+ v = PyDict_GetItemString(dict2, name);
+ if (v != NULL) {
+ Py_INCREF(v);
+ return v;
+ }
+ return Py_FindMethod(m, co, name);
+}
+
+static int
+do_curl_setattr(CurlObject *co, char *name, PyObject *v)
+{
+ assert_curl_state(co);
+ return my_setattr(&co->dict, name, v);
+}
+
+static int
+do_multi_setattr(CurlMultiObject *co, char *name, PyObject *v)
+{
+ assert_multi_state(co);
+ return my_setattr(&co->dict, name, v);
+}
+
+static PyObject *
+do_curl_getattr(CurlObject *co, char *name)
+{
+ assert_curl_state(co);
+ return my_getattr((PyObject *)co, name, co->dict,
+ curlobject_constants, curlobject_methods);
+}
+
+static PyObject *
+do_multi_getattr(CurlMultiObject *co, char *name)
+{
+ assert_multi_state(co);
+ return my_getattr((PyObject *)co, name, co->dict,
+ curlmultiobject_constants, curlmultiobject_methods);
+}
+
+
+/* --------------- actual type definitions --------------- */
+
+static PyTypeObject Curl_Type = {
+ PyObject_HEAD_INIT(NULL)
+ 0, /* ob_size */
+ "pycurl.Curl", /* tp_name */
+ sizeof(CurlObject), /* tp_basicsize */
+ 0, /* tp_itemsize */
+ /* Methods */
+ (destructor)do_curl_dealloc, /* tp_dealloc */
+ 0, /* tp_print */
+ (getattrfunc)do_curl_getattr, /* tp_getattr */
+ (setattrfunc)do_curl_setattr, /* tp_setattr */
+ 0, /* tp_compare */
+ 0, /* tp_repr */
+ 0, /* tp_as_number */
+ 0, /* tp_as_sequence */
+ 0, /* tp_as_mapping */
+ 0, /* tp_hash */
+ 0, /* tp_call */
+ 0, /* tp_str */
+ 0, /* tp_getattro */
+ 0, /* tp_setattro */
+ 0, /* tp_as_buffer */
+ Py_TPFLAGS_HAVE_GC, /* tp_flags */
+ 0, /* tp_doc */
+ (traverseproc)do_curl_traverse, /* tp_traverse */
+ (inquiry)do_curl_clear /* tp_clear */
+ /* More fields follow here, depending on your Python version. You can
+ * safely ignore any compiler warnings about missing initializers.
+ */
+};
+
+static PyTypeObject CurlMulti_Type = {
+ PyObject_HEAD_INIT(NULL)
+ 0, /* ob_size */
+ "pycurl.CurlMulti", /* tp_name */
+ sizeof(CurlMultiObject), /* tp_basicsize */
+ 0, /* tp_itemsize */
+ /* Methods */
+ (destructor)do_multi_dealloc, /* tp_dealloc */
+ 0, /* tp_print */
+ (getattrfunc)do_multi_getattr, /* tp_getattr */
+ (setattrfunc)do_multi_setattr, /* tp_setattr */
+ 0, /* tp_compare */
+ 0, /* tp_repr */
+ 0, /* tp_as_number */
+ 0, /* tp_as_sequence */
+ 0, /* tp_as_mapping */
+ 0, /* tp_hash */
+ 0, /* tp_call */
+ 0, /* tp_str */
+ 0, /* tp_getattro */
+ 0, /* tp_setattro */
+ 0, /* tp_as_buffer */
+ Py_TPFLAGS_HAVE_GC, /* tp_flags */
+ 0, /* tp_doc */
+ (traverseproc)do_multi_traverse, /* tp_traverse */
+ (inquiry)do_multi_clear /* tp_clear */
+ /* More fields follow here, depending on your Python version. You can
+ * safely ignore any compiler warnings about missing initializers.
+ */
+};
+
+
+/*************************************************************************
+// module level
+// Note that the object constructors (do_curl_new, do_multi_new)
+// are module-level functions as well.
+**************************************************************************/
+
+static PyObject *
+do_global_init(PyObject *dummy, PyObject *args)
+{
+ int res, option;
+
+ UNUSED(dummy);
+ if (!PyArg_ParseTuple(args, "i:global_init", &option)) {
+ return NULL;
+ }
+
+ if (!(option == CURL_GLOBAL_SSL ||
+ option == CURL_GLOBAL_WIN32 ||
+ option == CURL_GLOBAL_ALL ||
+ option == CURL_GLOBAL_NOTHING)) {
+ PyErr_SetString(PyExc_ValueError, "invalid option to global_init");
+ return NULL;
+ }
+
+ res = curl_global_init(option);
+ if (res != CURLE_OK) {
+ PyErr_SetString(ErrorObject, "unable to set global option");
+ return NULL;
+ }
+
+ Py_INCREF(Py_None);
+ return Py_None;
+}
+
+
+static PyObject *
+do_global_cleanup(PyObject *dummy)
+{
+ UNUSED(dummy);
+ curl_global_cleanup();
+ Py_INCREF(Py_None);
+ return Py_None;
+}
+
+
+
+static PyObject *vi_str(const char *s)
+{
+ if (s == NULL) {
+ Py_INCREF(Py_None);
+ return Py_None;
+ }
+ while (*s == ' ' || *s == '\t')
+ s++;
+ return PyString_FromString(s);
+}
+
+static PyObject *
+do_version_info(PyObject *dummy, PyObject *args)
+{
+ const curl_version_info_data *vi;
+ PyObject *ret = NULL;
+ PyObject *protocols = NULL;
+ PyObject *tmp;
+ int i;
+ int stamp = CURLVERSION_NOW;
+
+ UNUSED(dummy);
+ if (!PyArg_ParseTuple(args, "|i:version_info", &stamp)) {
+ return NULL;
+ }
+ vi = curl_version_info((CURLversion) stamp);
+ if (vi == NULL) {
+ PyErr_SetString(ErrorObject, "unable to get version info");
+ return NULL;
+ }
+
+ /* Note: actually libcurl in lib/version.c does ignore
+ * the "stamp" parm, and so do we */
+
+ for (i = 0; vi->protocols[i] != NULL; )
+ i++;
+ protocols = PyTuple_New(i);
+ if (protocols == NULL)
+ goto error;
+ for (i = 0; vi->protocols[i] != NULL; i++) {
+ tmp = vi_str(vi->protocols[i]);
+ if (tmp == NULL)
+ goto error;
+ PyTuple_SET_ITEM(protocols, i, tmp);
+ }
+ ret = PyTuple_New(12);
+ if (ret == NULL)
+ goto error;
+
+#define SET(i, v) \
+ tmp = (v); if (tmp == NULL) goto error; PyTuple_SET_ITEM(ret, i, tmp)
+ SET(0, PyInt_FromLong((long) vi->age));
+ SET(1, vi_str(vi->version));
+ SET(2, PyInt_FromLong(vi->version_num));
+ SET(3, vi_str(vi->host));
+ SET(4, PyInt_FromLong(vi->features));
+ SET(5, vi_str(vi->ssl_version));
+ SET(6, PyInt_FromLong(vi->ssl_version_num));
+ SET(7, vi_str(vi->libz_version));
+ SET(8, protocols);
+ SET(9, vi_str(vi->ares));
+ SET(10, PyInt_FromLong(vi->ares_num));
+ SET(11, vi_str(vi->libidn));
+#undef SET
+ return ret;
+
+error:
+ Py_XDECREF(ret);
+ Py_XDECREF(protocols);
+ return NULL;
+}
+
+
+/* Per function docstrings */
+static char pycurl_global_init_doc [] =
+"global_init(option) -> None. Initialize curl environment.\n";
+
+static char pycurl_global_cleanup_doc [] =
+"global_cleanup() -> None. Cleanup curl environment.\n";
+
+static char pycurl_version_info_doc [] =
+"version_info() -> tuple. Returns a 12-tuple with the version info.\n";
+
+static char pycurl_curl_new_doc [] =
+"Curl() -> New curl object. Implicitly calls global_init() if not called.\n";
+
+static char pycurl_multi_new_doc [] =
+"CurlMulti() -> New curl multi-object.\n";
+
+
+/* List of functions defined in this module */
+static PyMethodDef curl_methods[] = {
+ {"global_init", (PyCFunction)do_global_init, METH_VARARGS, pycurl_global_init_doc},
+ {"global_cleanup", (PyCFunction)do_global_cleanup, METH_NOARGS, pycurl_global_cleanup_doc},
+ {"version_info", (PyCFunction)do_version_info, METH_VARARGS, pycurl_version_info_doc},
+ {"Curl", (PyCFunction)do_curl_new, METH_NOARGS, pycurl_curl_new_doc},
+ {"CurlMulti", (PyCFunction)do_multi_new, METH_NOARGS, pycurl_multi_new_doc},
+ {NULL, NULL, 0, NULL}
+};
+
+
+/* Module docstring */
+static char module_doc [] =
+"This module implements an interface to the cURL library.\n"
+"\n"
+"Types:\n"
+"\n"
+"Curl() -> New object. Create a new curl object.\n"
+"CurlMulti() -> New object. Create a new curl multi-object.\n"
+"\n"
+"Functions:\n"
+"\n"
+"global_init(option) -> None. Initialize curl environment.\n"
+"global_cleanup() -> None. Cleanup curl environment.\n"
+"version_info() -> tuple. Return version information.\n"
+;
+
+
+/* Helper functions for inserting constants into the module namespace */
+
+static void
+insobj2(PyObject *dict1, PyObject *dict2, char *name, PyObject *value)
+{
+ /* Insert an object into one or two dicts. Eats a reference to value.
+ * See also the implementation of PyDict_SetItemString(). */
+ PyObject *key = NULL;
+
+ if (dict1 == NULL && dict2 == NULL)
+ goto error;
+ if (value == NULL)
+ goto error;
+ key = PyString_FromString(name);
+ if (key == NULL)
+ goto error;
+#if 0
+ PyString_InternInPlace(&key); /* XXX Should we really? */
+#endif
+ if (dict1 != NULL) {
+ assert(PyDict_GetItem(dict1, key) == NULL);
+ if (PyDict_SetItem(dict1, key, value) != 0)
+ goto error;
+ }
+ if (dict2 != NULL && dict2 != dict1) {
+ assert(PyDict_GetItem(dict2, key) == NULL);
+ if (PyDict_SetItem(dict2, key, value) != 0)
+ goto error;
+ }
+ Py_DECREF(key);
+ Py_DECREF(value);
+ return;
+error:
+ Py_FatalError("pycurl: FATAL: insobj2() failed");
+ assert(0);
+}
+
+static void
+insstr(PyObject *d, char *name, char *value)
+{
+ PyObject *v = PyString_FromString(value);
+ insobj2(d, NULL, name, v);
+}
+
+static void
+insint(PyObject *d, char *name, long value)
+{
+ PyObject *v = PyInt_FromLong(value);
+ insobj2(d, NULL, name, v);
+}
+
+static void
+insint_c(PyObject *d, char *name, long value)
+{
+ PyObject *v = PyInt_FromLong(value);
+ insobj2(d, curlobject_constants, name, v);
+}
+
+static void
+insint_m(PyObject *d, char *name, long value)
+{
+ PyObject *v = PyInt_FromLong(value);
+ insobj2(d, curlmultiobject_constants, name, v);
+}
+
+
+/* Initialization function for the module */
+#if defined(PyMODINIT_FUNC)
+PyMODINIT_FUNC
+#else
+#if defined(__cplusplus)
+extern "C"
+#endif
+DL_EXPORT(void)
+#endif
+initpycurl(void)
+{
+ PyObject *m, *d;
+ const curl_version_info_data *vi;
+
+ /* Initialize the type of the new type objects here; doing it here
+ * is required for portability to Windows without requiring C++. */
+ p_Curl_Type = &Curl_Type;
+ p_CurlMulti_Type = &CurlMulti_Type;
+ Curl_Type.ob_type = &PyType_Type;
+ CurlMulti_Type.ob_type = &PyType_Type;
+
+ /* Create the module and add the functions */
+ m = Py_InitModule3("pycurl", curl_methods, module_doc);
+ assert(m != NULL && PyModule_Check(m));
+
+ /* Add error object to the module */
+ d = PyModule_GetDict(m);
+ assert(d != NULL);
+ ErrorObject = PyErr_NewException("pycurl.error", NULL, NULL);
+ assert(ErrorObject != NULL);
+ PyDict_SetItemString(d, "error", ErrorObject);
+
+ curlobject_constants = PyDict_New();
+ assert(curlobject_constants != NULL);
+
+ /* Add version strings to the module */
+ insstr(d, "version", curl_version());
+ insstr(d, "COMPILE_DATE", __DATE__ " " __TIME__);
+ insint(d, "COMPILE_PY_VERSION_HEX", PY_VERSION_HEX);
+ insint(d, "COMPILE_LIBCURL_VERSION_NUM", LIBCURL_VERSION_NUM);
+
+ /**
+ ** the order of these constants mostly follows <curl/curl.h>
+ **/
+
+ /* Abort curl_read_callback(). */
+ insint_c(d, "READFUNC_ABORT", CURL_READFUNC_ABORT);
+
+ /* constants for ioctl callback return values */
+ insint_c(d, "IOE_OK", CURLIOE_OK);
+ insint_c(d, "IOE_UNKNOWNCMD", CURLIOE_UNKNOWNCMD);
+ insint_c(d, "IOE_FAILRESTART", CURLIOE_FAILRESTART);
+
+ /* curl_infotype: the kind of data that is passed to information_callback */
+/* XXX do we actually need curl_infotype in pycurl ??? */
+ insint_c(d, "INFOTYPE_TEXT", CURLINFO_TEXT);
+ insint_c(d, "INFOTYPE_HEADER_IN", CURLINFO_HEADER_IN);
+ insint_c(d, "INFOTYPE_HEADER_OUT", CURLINFO_HEADER_OUT);
+ insint_c(d, "INFOTYPE_DATA_IN", CURLINFO_DATA_IN);
+ insint_c(d, "INFOTYPE_DATA_OUT", CURLINFO_DATA_OUT);
+ insint_c(d, "INFOTYPE_SSL_DATA_IN", CURLINFO_SSL_DATA_IN);
+ insint_c(d, "INFOTYPE_SSL_DATA_OUT", CURLINFO_SSL_DATA_OUT);
+
+ /* CURLcode: error codes */
+/* FIXME: lots of error codes are missing */
+ insint_c(d, "E_OK", CURLE_OK);
+ insint_c(d, "E_UNSUPPORTED_PROTOCOL", CURLE_UNSUPPORTED_PROTOCOL);
+
+ /* curl_proxytype: constants for setopt(PROXYTYPE, x) */
+ insint_c(d, "PROXYTYPE_HTTP", CURLPROXY_HTTP);
+ insint_c(d, "PROXYTYPE_SOCKS4", CURLPROXY_SOCKS4);
+ insint_c(d, "PROXYTYPE_SOCKS5", CURLPROXY_SOCKS5);
+
+ /* curl_httpauth: constants for setopt(HTTPAUTH, x) */
+ insint_c(d, "HTTPAUTH_NONE", CURLAUTH_NONE);
+ insint_c(d, "HTTPAUTH_BASIC", CURLAUTH_BASIC);
+ insint_c(d, "HTTPAUTH_DIGEST", CURLAUTH_DIGEST);
+ insint_c(d, "HTTPAUTH_GSSNEGOTIATE", CURLAUTH_GSSNEGOTIATE);
+ insint_c(d, "HTTPAUTH_NTLM", CURLAUTH_NTLM);
+ insint_c(d, "HTTPAUTH_ANY", CURLAUTH_ANY);
+ insint_c(d, "HTTPAUTH_ANYSAFE", CURLAUTH_ANYSAFE);
+
+ /* curl_ftpssl: constants for setopt(FTP_SSL, x) */
+ insint_c(d, "FTPSSL_NONE", CURLFTPSSL_NONE);
+ insint_c(d, "FTPSSL_TRY", CURLFTPSSL_TRY);
+ insint_c(d, "FTPSSL_CONTROL", CURLFTPSSL_CONTROL);
+ insint_c(d, "FTPSSL_ALL", CURLFTPSSL_ALL);
+
+ /* curl_ftpauth: constants for setopt(FTPSSLAUTH, x) */
+ insint_c(d, "FTPAUTH_DEFAULT", CURLFTPAUTH_DEFAULT);
+ insint_c(d, "FTPAUTH_SSL", CURLFTPAUTH_SSL);
+ insint_c(d, "FTPAUTH_TLS", CURLFTPAUTH_TLS);
+
+ /* curl_ftpauth: constants for setopt(FTPSSLAUTH, x) */
+ insint_c(d, "FORM_CONTENTS", CURLFORM_COPYCONTENTS);
+ insint_c(d, "FORM_FILE", CURLFORM_FILE);
+ insint_c(d, "FORM_CONTENTTYPE", CURLFORM_CONTENTTYPE);
+
+ /* CURLoption: symbolic constants for setopt() */
+/* FIXME: reorder these to match <curl/curl.h> */
+ insint_c(d, "FILE", CURLOPT_WRITEDATA);
+ insint_c(d, "URL", CURLOPT_URL);
+ insint_c(d, "PORT", CURLOPT_PORT);
+ insint_c(d, "PROXY", CURLOPT_PROXY);
+ insint_c(d, "USERPWD", CURLOPT_USERPWD);
+ insint_c(d, "PROXYUSERPWD", CURLOPT_PROXYUSERPWD);
+ insint_c(d, "RANGE", CURLOPT_RANGE);
+ insint_c(d, "INFILE", CURLOPT_READDATA);
+ /* ERRORBUFFER is not supported */
+ insint_c(d, "WRITEFUNCTION", CURLOPT_WRITEFUNCTION);
+ insint_c(d, "READFUNCTION", CURLOPT_READFUNCTION);
+ insint_c(d, "TIMEOUT", CURLOPT_TIMEOUT);
+ insint_c(d, "INFILESIZE", CURLOPT_INFILESIZE_LARGE); /* _LARGE ! */
+ insint_c(d, "POSTFIELDS", CURLOPT_POSTFIELDS);
+ insint_c(d, "REFERER", CURLOPT_REFERER);
+ insint_c(d, "FTPPORT", CURLOPT_FTPPORT);
+ insint_c(d, "USERAGENT", CURLOPT_USERAGENT);
+ insint_c(d, "LOW_SPEED_LIMIT", CURLOPT_LOW_SPEED_LIMIT);
+ insint_c(d, "LOW_SPEED_TIME", CURLOPT_LOW_SPEED_TIME);
+ insint_c(d, "RESUME_FROM", CURLOPT_RESUME_FROM_LARGE); /* _LARGE ! */
+ insint_c(d, "WRITEDATA", CURLOPT_WRITEDATA);
+ insint_c(d, "READDATA", CURLOPT_READDATA);
+ insint_c(d, "PROXYPORT", CURLOPT_PROXYPORT);
+ insint_c(d, "HTTPPROXYTUNNEL", CURLOPT_HTTPPROXYTUNNEL);
+ insint_c(d, "VERBOSE", CURLOPT_VERBOSE);
+ insint_c(d, "HEADER", CURLOPT_HEADER);
+ insint_c(d, "NOPROGRESS", CURLOPT_NOPROGRESS);
+ insint_c(d, "NOBODY", CURLOPT_NOBODY);
+ insint_c(d, "FAILONERROR", CURLOPT_FAILONERROR);
+ insint_c(d, "UPLOAD", CURLOPT_UPLOAD);
+ insint_c(d, "POST", CURLOPT_POST);
+ insint_c(d, "FTPLISTONLY", CURLOPT_FTPLISTONLY);
+ insint_c(d, "FTPAPPEND", CURLOPT_FTPAPPEND);
+ insint_c(d, "NETRC", CURLOPT_NETRC);
+ insint_c(d, "FOLLOWLOCATION", CURLOPT_FOLLOWLOCATION);
+ insint_c(d, "TRANSFERTEXT", CURLOPT_TRANSFERTEXT);
+ insint_c(d, "PUT", CURLOPT_PUT);
+ insint_c(d, "POSTFIELDSIZE", CURLOPT_POSTFIELDSIZE_LARGE); /* _LARGE ! */
+ insint_c(d, "COOKIE", CURLOPT_COOKIE);
+ insint_c(d, "HTTPHEADER", CURLOPT_HTTPHEADER);
+ insint_c(d, "HTTPPOST", CURLOPT_HTTPPOST);
+ insint_c(d, "SSLCERT", CURLOPT_SSLCERT);
+ insint_c(d, "SSLCERTPASSWD", CURLOPT_SSLCERTPASSWD);
+ insint_c(d, "CRLF", CURLOPT_CRLF);
+ insint_c(d, "QUOTE", CURLOPT_QUOTE);
+ insint_c(d, "POSTQUOTE", CURLOPT_POSTQUOTE);
+ insint_c(d, "PREQUOTE", CURLOPT_PREQUOTE);
+ insint_c(d, "WRITEHEADER", CURLOPT_WRITEHEADER);
+ insint_c(d, "HEADERFUNCTION", CURLOPT_HEADERFUNCTION);
+ insint_c(d, "COOKIEFILE", CURLOPT_COOKIEFILE);
+ insint_c(d, "SSLVERSION", CURLOPT_SSLVERSION);
+ insint_c(d, "TIMECONDITION", CURLOPT_TIMECONDITION);
+ insint_c(d, "TIMEVALUE", CURLOPT_TIMEVALUE);
+ insint_c(d, "CUSTOMREQUEST", CURLOPT_CUSTOMREQUEST);
+ insint_c(d, "STDERR", CURLOPT_STDERR);
+ insint_c(d, "INTERFACE", CURLOPT_INTERFACE);
+ insint_c(d, "KRB4LEVEL", CURLOPT_KRB4LEVEL);
+ insint_c(d, "PROGRESSFUNCTION", CURLOPT_PROGRESSFUNCTION);
+ insint_c(d, "SSL_VERIFYPEER", CURLOPT_SSL_VERIFYPEER);
+ insint_c(d, "CAPATH", CURLOPT_CAPATH);
+ insint_c(d, "CAINFO", CURLOPT_CAINFO);
+ insint_c(d, "OPT_FILETIME", CURLOPT_FILETIME);
+ insint_c(d, "MAXREDIRS", CURLOPT_MAXREDIRS);
+ insint_c(d, "MAXCONNECTS", CURLOPT_MAXCONNECTS);
+ insint_c(d, "CLOSEPOLICY", CURLOPT_CLOSEPOLICY);
+ insint_c(d, "FRESH_CONNECT", CURLOPT_FRESH_CONNECT);
+ insint_c(d, "FORBID_REUSE", CURLOPT_FORBID_REUSE);
+ insint_c(d, "RANDOM_FILE", CURLOPT_RANDOM_FILE);
+ insint_c(d, "EGDSOCKET", CURLOPT_EGDSOCKET);
+ insint_c(d, "CONNECTTIMEOUT", CURLOPT_CONNECTTIMEOUT);
+ insint_c(d, "HTTPGET", CURLOPT_HTTPGET);
+ insint_c(d, "SSL_VERIFYHOST", CURLOPT_SSL_VERIFYHOST);
+ insint_c(d, "COOKIEJAR", CURLOPT_COOKIEJAR);
+ insint_c(d, "SSL_CIPHER_LIST", CURLOPT_SSL_CIPHER_LIST);
+ insint_c(d, "HTTP_VERSION", CURLOPT_HTTP_VERSION);
+ insint_c(d, "FTP_USE_EPSV", CURLOPT_FTP_USE_EPSV);
+ insint_c(d, "SSLCERTTYPE", CURLOPT_SSLCERTTYPE);
+ insint_c(d, "SSLKEY", CURLOPT_SSLKEY);
+ insint_c(d, "SSLKEYTYPE", CURLOPT_SSLKEYTYPE);
+ insint_c(d, "SSLKEYPASSWD", CURLOPT_SSLKEYPASSWD);
+ insint_c(d, "SSLENGINE", CURLOPT_SSLENGINE);
+ insint_c(d, "SSLENGINE_DEFAULT", CURLOPT_SSLENGINE_DEFAULT);
+ insint_c(d, "DNS_CACHE_TIMEOUT", CURLOPT_DNS_CACHE_TIMEOUT);
+ insint_c(d, "DNS_USE_GLOBAL_CACHE", CURLOPT_DNS_USE_GLOBAL_CACHE);
+ insint_c(d, "DEBUGFUNCTION", CURLOPT_DEBUGFUNCTION);
+ insint_c(d, "BUFFERSIZE", CURLOPT_BUFFERSIZE);
+ insint_c(d, "NOSIGNAL", CURLOPT_NOSIGNAL);
+ insint_c(d, "SHARE", CURLOPT_SHARE);
+ insint_c(d, "PROXYTYPE", CURLOPT_PROXYTYPE);
+ insint_c(d, "ENCODING", CURLOPT_ENCODING);
+ insint_c(d, "HTTP200ALIASES", CURLOPT_HTTP200ALIASES);
+ insint_c(d, "UNRESTRICTED_AUTH", CURLOPT_UNRESTRICTED_AUTH);
+ insint_c(d, "FTP_USE_EPRT", CURLOPT_FTP_USE_EPRT);
+ insint_c(d, "HTTPAUTH", CURLOPT_HTTPAUTH);
+ insint_c(d, "FTP_CREATE_MISSING_DIRS", CURLOPT_FTP_CREATE_MISSING_DIRS);
+ insint_c(d, "PROXYAUTH", CURLOPT_PROXYAUTH);
+ insint_c(d, "FTP_RESPONSE_TIMEOUT", CURLOPT_FTP_RESPONSE_TIMEOUT);
+ insint_c(d, "IPRESOLVE", CURLOPT_IPRESOLVE);
+ insint_c(d, "MAXFILESIZE", CURLOPT_MAXFILESIZE_LARGE); /* _LARGE ! */
+ insint_c(d, "INFILESIZE_LARGE", CURLOPT_INFILESIZE_LARGE);
+ insint_c(d, "RESUME_FROM_LARGE", CURLOPT_RESUME_FROM_LARGE);
+ insint_c(d, "MAXFILESIZE_LARGE", CURLOPT_MAXFILESIZE_LARGE);
+ insint_c(d, "NETRC_FILE", CURLOPT_NETRC_FILE);
+ insint_c(d, "FTP_SSL", CURLOPT_FTP_SSL);
+ insint_c(d, "POSTFIELDSIZE_LARGE", CURLOPT_POSTFIELDSIZE_LARGE);
+ insint_c(d, "TCP_NODELAY", CURLOPT_TCP_NODELAY);
+ insint_c(d, "SOURCE_USERPWD", CURLOPT_SOURCE_USERPWD);
+ insint_c(d, "SOURCE_PREQUOTE", CURLOPT_SOURCE_PREQUOTE);
+ insint_c(d, "SOURCE_POSTQUOTE", CURLOPT_SOURCE_POSTQUOTE);
+ insint_c(d, "FTPSSLAUTH", CURLOPT_FTPSSLAUTH);
+ insint_c(d, "IOCTLFUNCTION", CURLOPT_IOCTLFUNCTION);
+ insint_c(d, "IOCTLDATA", CURLOPT_IOCTLDATA);
+ insint_c(d, "SOURCE_URL", CURLOPT_SOURCE_URL);
+ insint_c(d, "SOURCE_QUOTE", CURLOPT_SOURCE_QUOTE);
+ insint_c(d, "FTP_ACCOUNT", CURLOPT_FTP_ACCOUNT);
+
+ /* constants for setopt(IPRESOLVE, x) */
+ insint_c(d, "IPRESOLVE_WHATEVER", CURL_IPRESOLVE_WHATEVER);
+ insint_c(d, "IPRESOLVE_V4", CURL_IPRESOLVE_V4);
+ insint_c(d, "IPRESOLVE_V6", CURL_IPRESOLVE_V6);
+
+ /* constants for setopt(HTTP_VERSION, x) */
+ insint_c(d, "CURL_HTTP_VERSION_NONE", CURL_HTTP_VERSION_NONE);
+ insint_c(d, "CURL_HTTP_VERSION_1_0", CURL_HTTP_VERSION_1_0);
+ insint_c(d, "CURL_HTTP_VERSION_1_1", CURL_HTTP_VERSION_1_1);
+ insint_c(d, "CURL_HTTP_VERSION_LAST", CURL_HTTP_VERSION_LAST);
+
+ /* CURL_NETRC_OPTION: constants for setopt(NETRC, x) */
+ insint_c(d, "NETRC_OPTIONAL", CURL_NETRC_OPTIONAL);
+ insint_c(d, "NETRC_IGNORED", CURL_NETRC_IGNORED);
+ insint_c(d, "NETRC_REQUIRED", CURL_NETRC_REQUIRED);
+
+ /* constants for setopt(SSLVERSION, x) */
+ insint_c(d, "SSLVERSION_DEFAULT", CURL_SSLVERSION_DEFAULT);
+ insint_c(d, "SSLVERSION_TLSv1", CURL_SSLVERSION_TLSv1);
+ insint_c(d, "SSLVERSION_SSLv2", CURL_SSLVERSION_SSLv2);
+ insint_c(d, "SSLVERSION_SSLv3", CURL_SSLVERSION_SSLv3);
+
+ /* curl_TimeCond: constants for setopt(TIMECONDITION, x) */
+ insint_c(d, "TIMECONDITION_NONE", CURL_TIMECOND_NONE);
+ insint_c(d, "TIMECONDITION_IFMODSINCE", CURL_TIMECOND_IFMODSINCE);
+ insint_c(d, "TIMECONDITION_IFUNMODSINCE", CURL_TIMECOND_IFUNMODSINCE);
+ insint_c(d, "TIMECONDITION_LASTMOD", CURL_TIMECOND_LASTMOD);
+
+ /* CURLINFO: symbolic constants for getinfo(x) */
+ insint_c(d, "EFFECTIVE_URL", CURLINFO_EFFECTIVE_URL);
+ insint_c(d, "HTTP_CODE", CURLINFO_HTTP_CODE);
+ insint_c(d, "RESPONSE_CODE", CURLINFO_HTTP_CODE);
+ insint_c(d, "TOTAL_TIME", CURLINFO_TOTAL_TIME);
+ insint_c(d, "NAMELOOKUP_TIME", CURLINFO_NAMELOOKUP_TIME);
+ insint_c(d, "CONNECT_TIME", CURLINFO_CONNECT_TIME);
+ insint_c(d, "PRETRANSFER_TIME", CURLINFO_PRETRANSFER_TIME);
+ insint_c(d, "SIZE_UPLOAD", CURLINFO_SIZE_UPLOAD);
+ insint_c(d, "SIZE_DOWNLOAD", CURLINFO_SIZE_DOWNLOAD);
+ insint_c(d, "SPEED_DOWNLOAD", CURLINFO_SPEED_DOWNLOAD);
+ insint_c(d, "SPEED_UPLOAD", CURLINFO_SPEED_UPLOAD);
+ insint_c(d, "HEADER_SIZE", CURLINFO_HEADER_SIZE);
+ insint_c(d, "REQUEST_SIZE", CURLINFO_REQUEST_SIZE);
+ insint_c(d, "SSL_VERIFYRESULT", CURLINFO_SSL_VERIFYRESULT);
+ insint_c(d, "INFO_FILETIME", CURLINFO_FILETIME);
+ insint_c(d, "CONTENT_LENGTH_DOWNLOAD", CURLINFO_CONTENT_LENGTH_DOWNLOAD);
+ insint_c(d, "CONTENT_LENGTH_UPLOAD", CURLINFO_CONTENT_LENGTH_UPLOAD);
+ insint_c(d, "STARTTRANSFER_TIME", CURLINFO_STARTTRANSFER_TIME);
+ insint_c(d, "CONTENT_TYPE", CURLINFO_CONTENT_TYPE);
+ insint_c(d, "REDIRECT_TIME", CURLINFO_REDIRECT_TIME);
+ insint_c(d, "REDIRECT_COUNT", CURLINFO_REDIRECT_COUNT);
+ insint_c(d, "HTTP_CONNECTCODE", CURLINFO_HTTP_CONNECTCODE);
+ insint_c(d, "HTTPAUTH_AVAIL", CURLINFO_HTTPAUTH_AVAIL);
+ insint_c(d, "PROXYAUTH_AVAIL", CURLINFO_PROXYAUTH_AVAIL);
+ insint_c(d, "OS_ERRNO", CURLINFO_OS_ERRNO);
+ insint_c(d, "NUM_CONNECTS", CURLINFO_NUM_CONNECTS);
+ insint_c(d, "SSL_ENGINES", CURLINFO_SSL_ENGINES);
+
+ /* curl_closepolicy: constants for setopt(CLOSEPOLICY, x) */
+ insint_c(d, "CLOSEPOLICY_OLDEST", CURLCLOSEPOLICY_OLDEST);
+ insint_c(d, "CLOSEPOLICY_LEAST_RECENTLY_USED", CURLCLOSEPOLICY_LEAST_RECENTLY_USED);
+ insint_c(d, "CLOSEPOLICY_LEAST_TRAFFIC", CURLCLOSEPOLICY_LEAST_TRAFFIC);
+ insint_c(d, "CLOSEPOLICY_SLOWEST", CURLCLOSEPOLICY_SLOWEST);
+ insint_c(d, "CLOSEPOLICY_CALLBACK", CURLCLOSEPOLICY_CALLBACK);
+
+ /* options for global_init() */
+ insint(d, "GLOBAL_SSL", CURL_GLOBAL_SSL);
+ insint(d, "GLOBAL_WIN32", CURL_GLOBAL_WIN32);
+ insint(d, "GLOBAL_ALL", CURL_GLOBAL_ALL);
+ insint(d, "GLOBAL_NOTHING", CURL_GLOBAL_NOTHING);
+ insint(d, "GLOBAL_DEFAULT", CURL_GLOBAL_DEFAULT);
+
+ /* curl_lock_data: XXX do we need this in pycurl ??? */
+ /* curl_lock_access: XXX do we need this in pycurl ??? */
+ /* CURLSHcode: XXX do we need this in pycurl ??? */
+ /* CURLSHoption: XXX do we need this in pycurl ??? */
+
+ /* CURLversion: constants for curl_version_info(x) */
+#if 0
+ /* XXX - do we need these ?? */
+ insint(d, "VERSION_FIRST", CURLVERSION_FIRST);
+ insint(d, "VERSION_SECOND", CURLVERSION_SECOND);
+ insint(d, "VERSION_THIRD", CURLVERSION_THIRD);
+ insint(d, "VERSION_NOW", CURLVERSION_NOW);
+#endif
+
+ /* version features - bitmasks for curl_version_info_data.features */
+#if 0
+ /* XXX - do we need these ?? */
+ /* XXX - should we really rename these ?? */
+ insint(d, "VERSION_FEATURE_IPV6", CURL_VERSION_IPV6);
+ insint(d, "VERSION_FEATURE_KERBEROS4", CURL_VERSION_KERBEROS4);
+ insint(d, "VERSION_FEATURE_SSL", CURL_VERSION_SSL);
+ insint(d, "VERSION_FEATURE_LIBZ", CURL_VERSION_LIBZ);
+ insint(d, "VERSION_FEATURE_NTLM", CURL_VERSION_NTLM);
+ insint(d, "VERSION_FEATURE_GSSNEGOTIATE", CURL_VERSION_GSSNEGOTIATE);
+ insint(d, "VERSION_FEATURE_DEBUG", CURL_VERSION_DEBUG);
+ insint(d, "VERSION_FEATURE_ASYNCHDNS", CURL_VERSION_ASYNCHDNS);
+ insint(d, "VERSION_FEATURE_SPNEGO", CURL_VERSION_SPNEGO);
+ insint(d, "VERSION_FEATURE_LARGEFILE", CURL_VERSION_LARGEFILE);
+ insint(d, "VERSION_FEATURE_IDN", CURL_VERSION_IDN);
+#endif
+
+ /**
+ ** the order of these constants mostly follows <curl/multi.h>
+ **/
+
+ /* CURLMcode: multi error codes */
+ insint_m(d, "E_CALL_MULTI_PERFORM", CURLM_CALL_MULTI_PERFORM);
+ insint_m(d, "E_MULTI_OK", CURLM_OK);
+ insint_m(d, "E_MULTI_BAD_HANDLE", CURLM_BAD_HANDLE);
+ insint_m(d, "E_MULTI_BAD_EASY_HANDLE", CURLM_BAD_EASY_HANDLE);
+ insint_m(d, "E_MULTI_OUT_OF_MEMORY", CURLM_OUT_OF_MEMORY);
+ insint_m(d, "E_MULTI_INTERNAL_ERROR", CURLM_INTERNAL_ERROR);
+
+ /* Check the version, as this has caused nasty problems in
+ * some cases. */
+ vi = curl_version_info(CURLVERSION_NOW);
+ if (vi == NULL) {
+ Py_FatalError("pycurl: FATAL: curl_version_info() failed");
+ assert(0);
+ }
+ if (vi->version_num < LIBCURL_VERSION_NUM) {
+ Py_FatalError("pycurl: FATAL: libcurl link-time version is older than compile-time version");
+ assert(0);
+ }
+
+ /* Finally initialize global interpreter lock */
+ PyEval_InitThreads();
+}
+
+/* vi:ts=4:et:nowrap
+ */
git://git.onelab.eu / plcapi.git / blobdiff