Add ToxDNS core functions

Author: abbat

examples/toxdns.py

@@ -0,0 +1,86 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+__title__    = "toxdns"
+__version__  = "0.0.17"
+__author__   = "Anton Batenev"
+__license__  = "BSD"
+
+import dns.resolver
+
+from pytoxcore import ToxDNS, ToxDNSException
+
+
+def last_txt_record(host):
+    response = dns.resolver.query(host, "TXT")
+    if response:
+        result = None
+        for record in response:
+           result = record
+        return str(result).replace('"', "")
+    return None
+
+
+def tox_dns3_id(record):
+    parts = record.split(";")
+    for part in parts:
+        kv = part.split("=")
+        if len(kv) != 2:
+            return None
+
+        k = kv[0]
+        v = kv[1]
+        if k == "v" and v != "tox3":
+            return None
+        elif k == "id":
+            return v
+
+    return None
+
+
+if __name__ == "__main__":
+    request = "[email protected]"
+
+    print("Request: {0}".format(request))
+
+    parts = request.split("@")
+    if len(parts) != 2:
+        exit("User name must be user@domain")
+
+    user   = parts[0]
+    domain = parts[1]
+
+    print("User: {0}".format(user))
+    print("Domain: {0}".format(domain))
+
+    host = "_tox.{0}".format(domain)
+    key  = last_txt_record(host)
+    if key == None:
+        exit("No TXT record found for {0}".format(host))
+    else:
+        print("Public key: {0}".format(key))
+
+    if len(key) != ToxDNS.TOX_CLIENT_ID_SIZE * 2:
+        exit("Public key length = {0}, but {1} required".format(len(key), ToxDNS.TOX_CLIENT_ID_SIZE * 2))
+
+    tox_dns = ToxDNS(key)
+
+    id_record, request_id = tox_dns.tox_generate_dns3_string(user)
+
+    print("Request record: {0}".format(id_record))
+    print("Request Id: {0}".format(request_id))
+
+    host   = "_{0}._tox.{1}".format(id_record, domain)
+    record = last_txt_record(host)
+    if record == None:
+        exit("No TXT record found for {0}".format(host))
+    else:
+        print("Record: {0}".format(record))
+
+    record = tox_dns3_id(record)
+    if record == None:
+        exit("Can not parse record")
+
+    address = tox_dns.tox_decrypt_dns3_TXT(record, request_id)
+
+    print("Address: {0}".format(address))

pytox.c

@@ -18,6 +18,7 @@
  */
 //----------------------------------------------------------------------------------------------
 #include "pytoxav.h"
+#include "pytoxdns.h"
 //----------------------------------------------------------------------------------------------
 
 void bytes_to_hex_string(const uint8_t* digest, int length, uint8_t* hex_digest)
@@ -139,6 +140,23 @@ PyMODINIT_FUNC initpytoxcore(void)
     ToxAVException = PyErr_NewException("pytoxcore.ToxAVException", NULL, NULL);
     PyModule_AddObject(module, "ToxAVException", (PyObject*)ToxAVException);
 
+    //
+    // initialize pytoxdns
+    //
+
+    ToxDNS_install_dict();
+
+    if (PyType_Ready(&ToxDNSType) < 0) {
+        fprintf(stderr, "Invalid PyTypeObject 'ToxDNSType'\n");
+        goto error;
+    }
+
+    Py_INCREF(&ToxDNSType);
+    PyModule_AddObject(module, "ToxDNS", (PyObject*)&ToxDNSType);
+
+    ToxDNSException = PyErr_NewException("pytoxcore.ToxDNSException", NULL, NULL);
+    PyModule_AddObject(module, "ToxDNSException", (PyObject*)ToxDNSException);
+
 #if PY_MAJOR_VERSION >= 3
     return module;
 #endif

pytox.h

@@ -28,6 +28,7 @@
 #include <pthread.h>
 #include <tox/tox.h>
 #include <tox/toxav.h>
+#include <tox/toxdns.h>
 #include <sys/param.h>
 #include <arpa/inet.h>
 #include <vpx/vpx_image.h>

pytoxav.c

@@ -1187,7 +1187,7 @@ PyTypeObject ToxAVType = {
     0,                                          /* tp_dictoffset     */
     (initproc)ToxAV_init,                       /* tp_init           */
     0,                                          /* tp_alloc          */
-    ToxAV_new,                                  /* tp_new            */
+    ToxAV_new                                   /* tp_new            */
 };
 //----------------------------------------------------------------------------------------------
 

pytoxav.h

@@ -46,5 +46,5 @@ extern PyObject* ToxAVException;
 //----------------------------------------------------------------------------------------------
 void ToxAV_install_dict(void);
 //----------------------------------------------------------------------------------------------
-#endif   // _pytoxavh_
+#endif   // _pytoxav_h_
 //----------------------------------------------------------------------------------------------

pytoxcore.c

@@ -2296,7 +2296,7 @@ PyTypeObject ToxCoreType = {
     0,                                          /* tp_dictoffset     */
     (initproc)ToxCore_init,                     /* tp_init           */
     0,                                          /* tp_alloc          */
-    ToxCore_new,                                /* tp_new            */
+    ToxCore_new                                 /* tp_new            */
 };
 //----------------------------------------------------------------------------------------------