Added idle_timeout attribute to Context

Author: MelindaShore

context.c

@@ -112,6 +112,36 @@ context_set_timeout(getdns_context *context, PyObject *py_value)
 }
 
 
+int
+context_set_idle_timeout(getdns_context *context, PyObject *py_value)
+{
+    getdns_return_t ret;
+    uint64_t value;
+    
+#if PY_MAJOR_VERSION >= 3
+    if (!PyLong_Check(py_value))  {
+#else
+    if (!PyInt_Check(py_value))  {
+#endif
+        PyErr_SetString(getdns_error, GETDNS_RETURN_INVALID_PARAMETER_TEXT);
+        return -1;
+    }
+#if PY_MAJOR_VERSION >= 3
+    if ((long)(value = PyLong_AsLong(py_value)) < 0)  {
+#else
+    if ((long)(value = PyInt_AsLong(py_value)) < 0)  {
+#endif
+        PyErr_SetString(getdns_error, GETDNS_RETURN_INVALID_PARAMETER_TEXT);
+        return -1;
+    }
+    if ((ret = getdns_context_set_idle_timeout(context, value)) != GETDNS_RETURN_GOOD)  {
+        PyErr_SetString(getdns_error, getdns_get_errorstr_by_id(ret));
+        return -1;
+    }
+    return 0;
+}
+
+
 int
 context_set_resolution_type(getdns_context *context, PyObject *py_value)
 {
@@ -830,6 +860,14 @@ context_getattro(PyObject *self, PyObject *nameobj)
         }
         return PyLong_FromLong((long)timeout);
     }
+    if (!strncmp(attrname, "idle_timeout", strlen("idle_timeout")))  {
+        uint64_t timeout;
+        if ((ret = getdns_context_get_idle_timeout(context, &timeout)) != GETDNS_RETURN_GOOD)  {
+            PyErr_SetString(getdns_error, getdns_get_errorstr_by_id(ret));
+            return NULL;
+        }
+        return PyLong_FromLong((long)timeout);
+    }
     if (!strncmp(attrname, "dns_transport_list", strlen("dns_transport_list")))  {
         getdns_transport_list_t *transports;
         PyObject *py_transports;
@@ -1055,6 +1093,9 @@ context_setattro(PyObject *self, PyObject *attrname, PyObject *py_value)
     if (!strncmp(name, "timeout", strlen("timeout")))  {
         return(context_set_timeout(context, py_value));
     }
+    if (!strncmp(name, "idle_timeout", strlen("idle_timeout")))  {
+        return(context_set_idle_timeout(context, py_value));
+    }
     if (!strncmp(name, "resolution_type", strlen("resolution_type")))  {
         return(context_set_resolution_type(context, py_value));
     }

getdns.c

@@ -196,6 +196,7 @@ PyMemberDef Context_members[] = {
       "string set by the implementer" },
     { "version_string", T_STRING|READONLY, offsetof(getdns_ContextObject, version_string), 0,
       "string set by the implementer" },
+    {"idle_timeout", T_INT, offsetof(getdns_ContextObject, idle_timeout), 0, "TCP idle timeout" },
     { NULL }
 };
 

pygetdns.h

@@ -73,6 +73,7 @@ typedef struct {
     PyObject_HEAD
     PyObject *py_context;       /* Python capsule containing getdns_context */
     uint64_t  timeout;          /* timeout attribute (milliseconds) */
+    uint64_t  idle_timeout;     /* TCP timeout attribute (milliseconds) */
     getdns_resolution_t resolution_type; /* stub or recursive? */
 #if 0
     getdns_transport_t dns_transport;    /* udp/tcp/etc */