gfwleak/zhangyang-libzt
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add address getter method to Python wrapper

Browse Source
This commit is contained in:
Joseph Henry
2021-05-19 21:33:08 -07:00
parent 111dc7f5d9
commit 4e0813d167
11 changed files with 15884 additions and 21731 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
build.sh
View File

@@ -661,8 +661,11 @@ format-code()
if [[ $1 = *"python"* ]]; then
if [[ ! $($PIP list | grep black) = "" ]];
then
$PYTHON -m black src/bindings/python/*.py
$PYTHON -m black examples/python/*.py
$PYTHON -m black src/bindings/python/libzt.py
$PYTHON -m black src/bindings/python/node.py
$PYTHON -m black src/bindings/python/sockets.py
$PYTHON -m black examples/python/
$PYTHON -m black test/selftest.py
else
echo "Please install python module (black)"
fi

61
include/ZeroTierSockets.h
View File

@@ -996,48 +996,6 @@ typedef struct {
int len;
} zts_event_msg_t;
//----------------------------------------------------------------------------//
// Python Bindings (Subset of regular socket API) //
//----------------------------------------------------------------------------//
#ifdef ZTS_ENABLE_PYTHON
#include "Python.h"
/**
* Abstract class used as a director. Pointer to an instance of this class
* is provided to the Python layer.
*/
class PythonDirectorCallbackClass {
public:
/**
* Called by native code on event. Implemented in Python
*/
virtual void on_zerotier_event(zts_event_msg_t* msg);
virtual ~PythonDirectorCallbackClass() {};
};
extern PythonDirectorCallbackClass* _userEventCallback;
int zts_py_bind(int fd, int family, int type, PyObject* addro);
int zts_py_connect(int fd, int family, int type, PyObject* addro);
PyObject* zts_py_accept(int fd);
int zts_py_listen(int fd, int backlog);
PyObject* zts_py_recv(int fd, int len, int flags);
int zts_py_send(int fd, PyObject* buf, int flags);
int zts_py_close(int fd);
int zts_py_setblocking(int fd, int flag);
int zts_py_getblocking(int fd);
#endif // ZTS_ENABLE_PYTHON
//----------------------------------------------------------------------------//
// ZeroTier Service and Network Controls //
//----------------------------------------------------------------------------//
@@ -1274,6 +1232,24 @@ ZTS_API int ZTCALL zts_init_from_storage(const char* path);
*/
ZTS_API int ZTCALL zts_init_from_memory(const char* key, unsigned int len);
#ifdef ZTS_ENABLE_PYTHON
#include "Python.h"
/**
* Abstract class used as a director. Pointer to an instance of this class
* is provided to the Python layer.
*/
class PythonDirectorCallbackClass {
public:
/**
* Called by native code on event. Implemented in Python
*/
virtual void on_zerotier_event(zts_event_msg_t* msg);
virtual ~PythonDirectorCallbackClass() {};
};
extern PythonDirectorCallbackClass* _userEventCallback;
/**
* @brief Set the event handler function. This is an initialization function that can only be called
* before `zts_node_start()`.
@@ -1283,7 +1259,6 @@ ZTS_API int ZTCALL zts_init_from_memory(const char* key, unsigned int len);
* @return `ZTS_ERR_OK` if successful, `ZTS_ERR_SERVICE` if the node
* experiences a problem, `ZTS_ERR_ARG` if invalid argument.
*/
#ifdef ZTS_ENABLE_PYTHON
ZTS_API int ZTCALL zts_init_set_event_handler(PythonDirectorCallbackClass* callback);
#endif
#ifdef ZTS_ENABLE_PINVOKE

3
pkg/pypi/setup.py
View File

@@ -110,8 +110,7 @@ setup(
keywords = 'zerotier p2p peer-to-peer sdwan sdn virtual network socket tcp udp zt encryption encrypted',
py_modules = ['libzt'],
packages = ['libzt'],
classifiers = ['Development Status :: 4 - Beta',
'Topic :: Internet',
classifiers = ['Topic :: Internet',
'Topic :: System :: Networking',
'Topic :: Security :: Cryptography',
'Intended Audience :: Developers',

11
src/bindings/python/PythonSockets.cxx
View File

@@ -200,4 +200,15 @@ int zts_py_close(int fd)
Py_END_ALLOW_THREADS return err;
}
PyObject* zts_py_addr_get_str(uint64_t net_id, int family)
{
char addr_str[ZTS_IP_MAX_STR_LEN] = { 0 };
if (zts_addr_get_str(net_id, family, addr_str, ZTS_IP_MAX_STR_LEN) < 0) {
PyErr_SetString(PyExc_Warning, "No address of the given type has been assigned by the network");
return NULL;
}
PyObject* t = PyUnicode_FromString(addr_str);
return t;
}
#endif // ZTS_ENABLE_PYTHON

42
src/bindings/python/PythonSockets.h Normal file
View File

@@ -0,0 +1,42 @@
/*
* Copyright (c)2013-2021 ZeroTier, Inc.
*
* Use of this software is governed by the Business Source License included
* in the LICENSE.TXT file in the project's root directory.
*
* Change Date: 2026-01-01
*
* On the date above, in accordance with the Business Source License, use
* of this software will be governed by version 2.0 of the Apache License.
*/
/****/
#ifndef ZTS_PYTHON_SOCKETS_H
#define ZTS_PYTHON_SOCKETS_H
#ifdef ZTS_ENABLE_PYTHON
#include "Python.h"
int zts_py_bind(int fd, int family, int type, PyObject* addro);
int zts_py_connect(int fd, int family, int type, PyObject* addro);
PyObject* zts_py_accept(int fd);
int zts_py_listen(int fd, int backlog);
PyObject* zts_py_recv(int fd, int len, int flags);
int zts_py_send(int fd, PyObject* buf, int flags);
int zts_py_close(int fd);
int zts_py_setblocking(int fd, int flag);
int zts_py_getblocking(int fd);
PyObject* zts_py_addr_get_str(uint64_t net_id, int family);
#endif // ZTS_ENABLE_PYTHON
#endif // ZTS_PYTHON_SOCKETS_H

115
src/bindings/python/libzt.py
View File

@@ -532,6 +532,25 @@ class zts_event_msg_t(object):
# Register zts_event_msg_t in _libzt:
_libzt.zts_event_msg_t_swigregister(zts_event_msg_t)
ZTS_DISABLE_CENTRAL_API = _libzt.ZTS_DISABLE_CENTRAL_API
ZTS_ID_STR_BUF_LEN = _libzt.ZTS_ID_STR_BUF_LEN
def zts_id_new(key, key_buf_len):
return _libzt.zts_id_new(key, key_buf_len)
def zts_id_pair_is_valid(key, len):
return _libzt.zts_id_pair_is_valid(key, len)
def zts_init_from_storage(path):
return _libzt.zts_init_from_storage(path)
def zts_init_from_memory(key, len):
return _libzt.zts_init_from_memory(key, len)
class PythonDirectorCallbackClass(object):
thisown = property(
@@ -566,62 +585,6 @@ class PythonDirectorCallbackClass(object):
_libzt.PythonDirectorCallbackClass_swigregister(PythonDirectorCallbackClass)
def zts_py_bind(fd, family, type, addro):
return _libzt.zts_py_bind(fd, family, type, addro)
def zts_py_connect(fd, family, type, addro):
return _libzt.zts_py_connect(fd, family, type, addro)
def zts_py_accept(fd):
return _libzt.zts_py_accept(fd)
def zts_py_listen(fd, backlog):
return _libzt.zts_py_listen(fd, backlog)
def zts_py_recv(fd, len, flags):
return _libzt.zts_py_recv(fd, len, flags)
def zts_py_send(fd, buf, flags):
return _libzt.zts_py_send(fd, buf, flags)
def zts_py_close(fd):
return _libzt.zts_py_close(fd)
def zts_py_setblocking(fd, flag):
return _libzt.zts_py_setblocking(fd, flag)
def zts_py_getblocking(fd):
return _libzt.zts_py_getblocking(fd)
ZTS_DISABLE_CENTRAL_API = _libzt.ZTS_DISABLE_CENTRAL_API
ZTS_ID_STR_BUF_LEN = _libzt.ZTS_ID_STR_BUF_LEN
def zts_id_new(key, key_buf_len):
return _libzt.zts_id_new(key, key_buf_len)
def zts_id_pair_is_valid(key, len):
return _libzt.zts_id_pair_is_valid(key, len)
def zts_init_from_storage(path):
return _libzt.zts_init_from_storage(path)
def zts_init_from_memory(key, len):
return _libzt.zts_init_from_memory(key, len)
def zts_init_set_event_handler(callback):
return _libzt.zts_init_set_event_handler(callback)
@@ -1509,3 +1472,43 @@ def zts_inet_ntop(family, src, dst, size):
def zts_inet_pton(family, src, dst):
return _libzt.zts_inet_pton(family, src, dst)
def zts_py_bind(fd, family, type, addro):
return _libzt.zts_py_bind(fd, family, type, addro)
def zts_py_connect(fd, family, type, addro):
return _libzt.zts_py_connect(fd, family, type, addro)
def zts_py_accept(fd):
return _libzt.zts_py_accept(fd)
def zts_py_listen(fd, backlog):
return _libzt.zts_py_listen(fd, backlog)
def zts_py_recv(fd, len, flags):
return _libzt.zts_py_recv(fd, len, flags)
def zts_py_send(fd, buf, flags):
return _libzt.zts_py_send(fd, buf, flags)
def zts_py_close(fd):
return _libzt.zts_py_close(fd)
def zts_py_setblocking(fd, flag):
return _libzt.zts_py_setblocking(fd, flag)
def zts_py_getblocking(fd):
return _libzt.zts_py_getblocking(fd)
def zts_py_addr_get_str(net_id, family):
return _libzt.zts_py_addr_get_str(net_id, family)

6
src/bindings/python/node.py
View File

@@ -77,3 +77,9 @@ class ZeroTierNode:
def net_transport_is_ready(self, net_id):
return libzt.zts_net_transport_is_ready(net_id)
def addr_get_ipv4(self, net_id):
return libzt.zts_py_addr_get_str(net_id, libzt.ZTS_AF_INET)
def addr_get_ipv6(self, net_id):
return libzt.zts_py_addr_get_str(net_id, libzt.ZTS_AF_INET6)

2
src/bindings/python/zt.i
View File

@@ -13,6 +13,7 @@
%module libzt
%{
#include "ZeroTierSockets.h"
#include "PythonSockets.h"
%}
%feature("director") PythonDirectorCallbackClass;
@@ -35,3 +36,4 @@
%ignore zts_msghdr;
%include "ZeroTierSockets.h"
%include "PythonSockets.h"

13355
src/bindings/python/zt_wrap.cxx
View File

File diff suppressed because it is too large Load Diff

19
src/bindings/python/zt_wrap.h
View File

@@ -14,9 +14,9 @@
#include <map>
#include <string>
class SwigDirector_PythonDirectorCallbackClass
: public PythonDirectorCallbackClass
, public Swig::Director {
class SwigDirector_PythonDirectorCallbackClass : public PythonDirectorCallbackClass, public Swig::Director {
public:
SwigDirector_PythonDirectorCallbackClass(PyObject *self);
virtual void on_zerotier_event(zts_event_msg_t *msg);
@@ -24,23 +24,19 @@ class SwigDirector_PythonDirectorCallbackClass
/* Internal director utilities */
public:
bool swig_get_inner(const char* swig_protected_method_name) const
{
bool swig_get_inner(const char *swig_protected_method_name) const {
std::map<std::string, bool>::const_iterator iv = swig_inner.find(swig_protected_method_name);
return (iv != swig_inner.end() ? iv->second : false);
}
void swig_set_inner(const char* swig_protected_method_name, bool swig_val) const
{
void swig_set_inner(const char *swig_protected_method_name, bool swig_val) const {
swig_inner[swig_protected_method_name] = swig_val;
}
private:
mutable std::map<std::string, bool> swig_inner;
#if defined(SWIG_PYTHON_DIRECTOR_VTABLE)
/* VTable implementation */
PyObject* swig_get_method(size_t method_index, const char* method_name) const
{
PyObject *swig_get_method(size_t method_index, const char *method_name) const {
PyObject *method = vtable[method_index];
if (!method) {
swig::SwigVar_PyObject name = SWIG_Python_str_FromChar(method_name);
@@ -54,10 +50,11 @@ class SwigDirector_PythonDirectorCallbackClass
}
return method;
}
private:
mutable swig::SwigVar_PyObject vtable[1];
#endif
};
#endif

2
test/selftest.py
View File

@@ -89,6 +89,8 @@ def main():
time.sleep(1)
print("Joined network")
print("addr = ", n.addr_get_ipv4(net_id))
#
# Example server
#