71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
# Simple class for synchronous/blocking interface with a remote
|
|
# Osmocom program via the CTRL interface
|
|
|
|
# (C) 2020 by Harald Welte <laforge@gnumonks.org>
|
|
#
|
|
# Licensed under GNU General Public License, Version 2 or at your
|
|
# option, any later version.
|
|
|
|
from osmopy.osmo_ipa import Ctrl
|
|
import socket
|
|
|
|
class OsmoCtrlSimple:
|
|
'''Simple class for synchronous/blocking interface with a remote
|
|
Osmocom program via the CTRL interface'''
|
|
remote_host = "127.0.0.1"
|
|
remote_port = 0
|
|
sock = None
|
|
|
|
def __init__(self, host=None, port=None):
|
|
self.remote_host = host
|
|
self.remote_port = port
|
|
|
|
def connect(self, host=None, port=None):
|
|
if host:
|
|
self.remote_host = host
|
|
if port:
|
|
self.remote_port = port
|
|
self.sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
|
self.sock.setblocking(1)
|
|
self.sock.connect((self.remote_host, self.remote_port))
|
|
return self.sock
|
|
|
|
def disconnect(self):
|
|
self.sock.close()
|
|
self.sock = None
|
|
|
|
def _leftovers(self, fl):
|
|
"""
|
|
Read outstanding data if any according to flags
|
|
"""
|
|
try:
|
|
data = self.sock.recv(1024, fl)
|
|
except socket.error as _:
|
|
return False
|
|
if len(data) != 0:
|
|
tail = data
|
|
while True:
|
|
(head, tail) = Ctrl().split_combined(tail)
|
|
print("Got message:", Ctrl().rem_header(head))
|
|
if len(tail) == 0:
|
|
break
|
|
return True
|
|
return False
|
|
|
|
def do_set_get(self, var, value = None):
|
|
self._leftovers(socket.MSG_DONTWAIT)
|
|
(r, c) = Ctrl().cmd(var, value)
|
|
self.sock.send(c)
|
|
while True:
|
|
ret = self.sock.recv(4096)
|
|
# handle multiple messages, ignore TRAPs
|
|
ret = Ctrl().skip_traps(ret)
|
|
if ret != None:
|
|
(i, k, v) = Ctrl().parse(ret)
|
|
break;
|
|
return (Ctrl().rem_header(ret),) + Ctrl().verify(ret, r, var, value)
|
|
|
|
def set_var(self, var, val):
|
|
(a, _, _) = self.do_set_get(var, val)
|
|
return a
|