rtpbleed: added -o param

Pepelux · Sep 20, 2024 · 176e511 · 176e511
1 parent ced8d61
commit 176e511
Show file tree

Hide file tree

Showing 5 changed files with 64 additions and 8 deletions.
diff --git a/bin/sippts b/bin/sippts
@@ -667,7 +667,7 @@ def main():
         s.rtp = rtp
         s.auth = auth
     elif params[0] == "rtpbleed":
-        (cmd, ip, start_port, end_port, loops, payload, delay) = params
+        (cmd, ip, start_port, end_port, loops, payload, delay, ofile) = params
 
         from sippts.rtpbleed import RTPBleed
 
@@ -678,8 +678,9 @@ def main():
         s.loops = loops
         s.payload = payload
         s.delay = delay
+        s.ofile = ofile
     elif params[0] == "rtcpbleed":
-        (cmd, ip, start_port, end_port, delay) = params
+        (cmd, ip, start_port, end_port, delay, ofile) = params
 
         from sippts.rtcpbleed import RTCPBleed
 
@@ -688,6 +689,7 @@ def main():
         s.start_port = start_port
         s.end_port = end_port
         s.delay = delay
+        s.ofile = ofile
     elif params[0] == "rtpbleedflood":
         (cmd, ip, port, payload, verbose) = params
 

diff --git a/src/sippts/lib/params.py b/src/sippts/lib/params.py
@@ -26,7 +26,7 @@
 BWHITE = "\033[1;37;20m"
 WHITE = "\033[0;37;20m"
 
-local_version = "4.0.14"
+local_version = "4.0.15"
 
 
 def get_sippts_args():
@@ -2747,6 +2747,16 @@ def get_sippts_args():
         "-i", metavar="IP", type=str, help="Target IP address", dest="ipaddr"
     )
 
+    log = parser_rtpbleed.add_argument_group("Log")
+    log.add_argument(
+        "-o",
+        metavar="FILE",
+        type=str,
+        help="Save data into a log file",
+        dest="ofile",
+        default="",
+    )
+
     other = parser_rtpbleed.add_argument_group("Other options")
     other.add_argument(
         "-s",
@@ -2817,6 +2827,16 @@ def get_sippts_args():
         "-i", metavar="IP", type=str, help="Target IP address", dest="ipaddr"
     )
 
+    log = parser_rtcpbleed.add_argument_group("Log")
+    log.add_argument(
+        "-o",
+        metavar="FILE",
+        type=str,
+        help="Save data into a log file",
+        dest="ofile",
+        default="",
+    )
+
     other = parser_rtcpbleed.add_argument_group("Other options")
     other.add_argument(
         "-s",
@@ -3989,10 +4009,11 @@ def get_sippts_args():
             SP = SP + 1
         if EP % 2 != 0:
             EP = EP + 1
-        LOOPS = args.loops
+        LOOPS = args.loops        
         PAYLOAD = args.payload
         DELAY = args.delay
-        return COMMAND, IPADDR, SP, EP, LOOPS, PAYLOAD, DELAY
+        OFILE = args.ofile
+        return COMMAND, IPADDR, SP, EP, LOOPS, PAYLOAD, DELAY, OFILE
     elif COMMAND == "rtcpbleed":
         if args.help == 1:
             parser_rtcpbleed.print_help()
@@ -4014,7 +4035,8 @@ def get_sippts_args():
         if EP % 2 == 0:
             EP = EP + 1
         DELAY = args.delay
-        return COMMAND, IPADDR, SP, EP, DELAY
+        OFILE = args.ofile
+        return COMMAND, IPADDR, SP, EP, DELAY, OFILE
     elif COMMAND == "rtpbleedflood":
         if args.help == 1:
             parser_rtpbleedflood.print_help()

diff --git a/src/sippts/rtcpbleed.py b/src/sippts/rtcpbleed.py
@@ -24,6 +24,7 @@ def __init__(self):
         self.start_port = "10000"
         self.end_port = "20000"
         self.delay = "1"
+        self.ofile = ""
 
         self.c = Color()
 
@@ -44,6 +45,10 @@ def start(self):
         )
         print(self.c.WHITE)
 
+        if self.ofile != "":
+            f = open(self.ofile, "a+")
+            f.write(f"Target IP: {self.ip}\n")
+
         # Create a UDP socket
         try:
             sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
@@ -57,7 +62,8 @@ def start(self):
         byte_array = bytearray.fromhex(message)
         port = self.start_port
 
-        while True:
+        # while True:
+        while port < self.end_port + 2:
             try:
                 host = (str(self.ip), port)
 
@@ -76,16 +82,28 @@ def start(self):
                         print(
                             f"{self.c.WHITE}received {str(size)} bytes from target port {str(rport)}"
                         )
+
+                        if self.ofile != "":
+                            f.write(f"received {str(size)} bytes from target port {str(rport)}\n")
                 except:
                     # No data available
                     pass
             except KeyboardInterrupt:
                 print(f"{self.c.YELLOW}\nYou pressed Ctrl+C!")
                 print(self.c.WHITE)
+                if self.ofile != "":
+                    f.write("\n")
+                    f.close()
                 exit()
             except:
                 pass
 
             port += 2
             if port > self.end_port:
                 port = self.start_port
+
+        print(self.c.WHITE)
+
+        if self.ofile != "":
+            f.write("\n")
+            f.close()
diff --git a/src/sippts/rtpbleed.py b/src/sippts/rtpbleed.py
@@ -26,6 +26,7 @@ def __init__(self):
         self.loops = "4"
         self.payload = "0"
         self.delay = "50"
+        self.ofile = ""
 
         self.c = Color()
 
@@ -52,6 +53,10 @@ def start(self):
         )
         print(self.c.WHITE)
 
+        if self.ofile != "":
+            f = open(self.ofile, "a+")
+            f.write(f"Target IP: {self.ip}\n")
+
         # Create a UDP socket
         try:
             sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
@@ -117,16 +122,25 @@ def start(self):
                             print(
                                 f"{self.c.WHITE}    [-] SSRC: {ssrc} - Timestamp: {timestamp} - Seq number: {seq}"
                             )
+                            if self.ofile != "":
+                                f.write(f"received {str(size)} bytes from target port {str(rport)} - loop {str(loop)} - SSRC: {ssrc} - Timestamp: {timestamp} - Seq number: {seq}\n")
                     except:
                         # No data available
                         continue
             except KeyboardInterrupt:
                 print(f"{self.c.YELLOW}\nYou pressed Ctrl+C!")
                 print(self.c.WHITE)
+                if self.ofile != "":
+                    f.write("\n")
+                    f.close()
                 exit()
             except:
                 pass
 
             port += 2
 
         print(self.c.WHITE)
+
+        if self.ofile != "":
+            f.write("\n")
+            f.close()
diff --git a/version b/version
@@ -1 +1 @@
-4.0.14
+4.0.15