mirror of
https://github.com/linux-msm/qrtr.git
synced 2025-12-24 05:56:29 +01:00
74 lines
2.5 KiB
Python
Executable File
74 lines
2.5 KiB
Python
Executable File
#!/usr/bin/env python2.7
|
|
|
|
import ctypes
|
|
import collections
|
|
|
|
from ctypes import CDLL, CFUNCTYPE, POINTER, cast, py_object
|
|
from ctypes import c_char_p, c_void_p, c_int, pointer
|
|
|
|
_qrtr = CDLL("./libqrtr.so")
|
|
|
|
class qrtr:
|
|
Result = collections.namedtuple('Result', ['service', 'instance', 'addr'])
|
|
_cbtype = CFUNCTYPE(None, c_void_p, c_int, c_int, c_int, c_int)
|
|
def __init__(self, port=0):
|
|
self.sock = _qrtr.qrtr_open(port)
|
|
if self.sock < 0:
|
|
raise RuntimeError("unable to open qrtr socket")
|
|
self.service = None
|
|
|
|
def __del__(self):
|
|
if self.service != None:
|
|
_qrtr.qrtr_bye(self.sock, self.service[0], self.service[1])
|
|
_qrtr.qrtr_close(self.sock)
|
|
|
|
def _lookup_list_add(self, ptr, srv, instance, node, port):
|
|
res = qrtr.Result(srv, instance, (node, port))
|
|
cast(ptr, POINTER(py_object)).contents.value.append(res)
|
|
|
|
def lookup(self, srv, instance=0, ifilter=0):
|
|
results = []
|
|
err = _qrtr.qrtr_lookup(self.sock, srv, instance, ifilter,
|
|
qrtr._cbtype(self._lookup_list_add), cast(pointer(py_object(results)), c_void_p))
|
|
if err:
|
|
raise RuntimeError("query failed")
|
|
return results
|
|
|
|
def publish(self, service, instance):
|
|
err = _qrtr.qrtr_publish(self.sock, service, instance)
|
|
if err:
|
|
raise RuntimeError("publish failed")
|
|
self.service = (service, instance)
|
|
|
|
def send(self, addr, data):
|
|
node, port = addr
|
|
n = _qrtr.qrtr_sendto(self.sock, node, port, c_char_p(data), len(data))
|
|
if n:
|
|
raise RuntimeError("sendto failed")
|
|
|
|
def recv(self, sz=65536):
|
|
buf = ctypes.create_string_buffer(sz)
|
|
n = _qrtr.qrtr_recv(self.sock, c_char_p(ctypes.addressof(buf)), sz)
|
|
if n <= 0:
|
|
raise RuntimeError("recv failed")
|
|
return buf[0:n]
|
|
|
|
def recvfrom(self, sz=65536):
|
|
node = ctypes.c_int()
|
|
port = ctypes.c_int()
|
|
buf = ctypes.create_string_buffer(sz)
|
|
n = _qrtr.qrtr_recvfrom(self.sock, c_char_p(ctypes.addressof(buf)),
|
|
ctypes.byref(node), ctypes.byref(port))
|
|
if n <= 0:
|
|
raise RuntimeError("recvfrom failed")
|
|
return (buf[0:n], (node.value, port.value))
|
|
|
|
def poll(self, tout=0):
|
|
return _qrtr.qrtr_poll(self.sock, tout)
|
|
|
|
if __name__ == "__main__":
|
|
svcs = qrtr().lookup(15) # 15 is the test service
|
|
print " service instance addr"
|
|
for svc in svcs:
|
|
print "% 8d % 8d %s" % (svc.service, svc.instance, svc.addr)
|