From f4e90b5770d54af951e73fa8a684a7db3a88235f Mon Sep 17 00:00:00 2001
From: Matthias Tafelmeier <matthias.tafelmeier@gmx.net>
Date: Fri, 24 Nov 2023 16:49:46 +0100
Subject: [PATCH] ss2 - offer a flag to discern ss2 module run

---
 pyroute2/netlink/diag/ss2.py | 17 ++++++++++++-----
 1 file changed, 12 insertions(+), 5 deletions(-)
 mode change 100644 => 100755 pyroute2/netlink/diag/ss2.py

diff --git a/pyroute2/netlink/diag/ss2.py b/pyroute2/netlink/diag/ss2.py
old mode 100644
new mode 100755
index 978ddc30d..064f28ef5
--- a/pyroute2/netlink/diag/ss2.py
+++ b/pyroute2/netlink/diag/ss2.py
@@ -56,6 +56,7 @@
 # UDIAG_SHOW_RQLEN,
 # UDIAG_SHOW_MEMINFO
 
+RUN_AS_MODULE=False
 
 class UserCtxtMap(Mapping):
     _sk_inode_re = re.compile(r"socket:\[(?P<ino>\d+)\]")
@@ -210,9 +211,8 @@ def __call__(self, nl_diag_sk, args, usr_ctxt):
             show=(UDIAG_SHOW_NAME | UDIAG_SHOW_VFS | UDIAG_SHOW_PEER),
         )
         refined_stats = self._refine_diag_raw(sstats, usr_ctxt)
-        printable = self._fmt(refined_stats)
 
-        print(printable)
+        return refined_stats
 
     def _refine_diag_raw(self, raw_stats, usr_ctxt):
         refined = {"UNIX": {"flows": []}}
@@ -289,9 +289,9 @@ def __call__(self, nl_diag_sk, args, usr_ctxt):
             states=self._states, family=AF_INET, extensions=self.ext_f
         )
         refined_stats = self._refine_diag_raw(sstats, args.resolve, usr_ctxt)
-        printable = self._fmt(refined_stats)
 
-        print(printable)
+        return refined_stats
+
 
     def _refine_diag_raw(self, raw_stats, do_resolve, usr_ctxt):
         refined = {"TCP": {"flows": []}}
@@ -616,11 +616,18 @@ def run(args=None):
     if args.process:
         _user_ctxt_map = UserCtxtMap()
 
+    result = list()
+
     with DiagSocket() as ds:
         ds.bind()
         for p in protocols:
-            p(ds, args, _user_ctxt_map)
+            sub_statistics = p(ds, args, _user_ctxt_map)
+            result.append(sub_statistics)
 
+    if RUN_AS_MODULE:
+        return result
+    else:
+        print(json.dumps(result, indent=4))
 
 if __name__ == "__main__":
     run()