From c7bf94e9f128e9a76a76d355a8ce77c3cc3ca6d9 Mon Sep 17 00:00:00 2001 From: lns Date: Sun, 17 Apr 2022 18:56:30 +0200 Subject: nDPIsrvd.(h|py): Added socket read/recv timeout. * nDPIsrvd.h: support for O_NONBLOCK nDPIsrvd_socket Signed-off-by: lns --- examples/py-flow-info/flow-info.py | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) (limited to 'examples/py-flow-info/flow-info.py') diff --git a/examples/py-flow-info/flow-info.py b/examples/py-flow-info/flow-info.py index 0dbc5e63c..f11750351 100755 --- a/examples/py-flow-info/flow-info.py +++ b/examples/py-flow-info/flow-info.py @@ -426,8 +426,16 @@ if __name__ == '__main__': nsock = nDPIsrvdSocket() nsock.connect(address) + nsock.timeout(1.0) stats = Stats(nsock) - try: - nsock.loop(onJsonLineRecvd, onFlowCleanup, stats) - except KeyboardInterrupt: - print('\n\nKeyboard Interrupt: cleaned up {} flows.'.format(len(nsock.shutdown()))) + + while True: + try: + nsock.loop(onJsonLineRecvd, onFlowCleanup, stats) + except KeyboardInterrupt: + print('\n\nKeyboard Interrupt: cleaned up {} flows.'.format(len(nsock.shutdown()))) + break + except TimeoutError: + stats.updateSpinner() + stats.resetStatus() + stats.printStatus() -- cgit v1.2.3