Merge pull request #1768 from mhils/organize-examples

Organize Examples

1 files changed, 27 insertions, 0 deletions

diff --git a/examples/complex/tcp_message.py b/examples/complex/tcp_message.py
new file mode 100644
index 00000000..d7c9c42e
--- /dev/null
+++ b/examples/complex/tcp_message.py
@@ -0,0 +1,27 @@
+"""
+tcp_message Inline Script Hook API Demonstration
+------------------------------------------------
+
+* modifies packets containing "foo" to "bar"
+* prints various details for each packet.
+
+example cmdline invocation:
+mitmdump -T --host --tcp ".*" -q -s examples/tcp_message.py
+"""
+from mitmproxy.utils import strutils
+
+
+def tcp_message(tcp_msg):
+    modified_msg = tcp_msg.message.replace("foo", "bar")
+
+    is_modified = False if modified_msg == tcp_msg.message else True
+    tcp_msg.message = modified_msg
+
+    print(
+        "[tcp_message{}] from {} {} to {} {}:\r\n{}".format(
+            " (modified)" if is_modified else "",
+            "client" if tcp_msg.sender == tcp_msg.client_conn else "server",
+            tcp_msg.sender.address,
+            "server" if tcp_msg.receiver == tcp_msg.server_conn else "client",
+            tcp_msg.receiver.address, strutils.bytes_to_escaped_str(tcp_msg.message))
+    )