aboutsummaryrefslogtreecommitdiffstats
path: root/libmproxy/protocol/primitives.py
diff options
context:
space:
mode:
authorJake Drahos <drahos.jake@gmail.com>2015-06-11 10:27:48 -0500
committerJake Drahos <drahos.jake@gmail.com>2015-06-11 10:27:48 -0500
commit8b998cfbeace0777293f3cef804c1bf239758273 (patch)
tree58d18756e5de9f6f74a3fd2fd352f3ef601f7f67 /libmproxy/protocol/primitives.py
parentd389b9c59d7f4a4918a15b5a40d771ccc154d751 (diff)
downloadmitmproxy-8b998cfbeace0777293f3cef804c1bf239758273.tar.gz
mitmproxy-8b998cfbeace0777293f3cef804c1bf239758273.tar.bz2
mitmproxy-8b998cfbeace0777293f3cef804c1bf239758273.zip
Implemented basic marking of flows
- Press m to toggle flow mark - Flow mark is set in libmproxy/console/common.py. Currently set to "==="
Diffstat (limited to 'libmproxy/protocol/primitives.py')
-rw-r--r--libmproxy/protocol/primitives.py7
1 files changed, 7 insertions, 0 deletions
diff --git a/libmproxy/protocol/primitives.py b/libmproxy/protocol/primitives.py
index 2f8ea3e0..1d2389c3 100644
--- a/libmproxy/protocol/primitives.py
+++ b/libmproxy/protocol/primitives.py
@@ -77,6 +77,7 @@ class Flow(stateobject.StateObject):
"""@type: bool"""
self._backup = None
self.reply = None
+ self.marked = False
_stateobject_attributes = dict(
id=str,
@@ -165,6 +166,12 @@ class Flow(stateobject.StateObject):
self.intercepted = False
self.reply()
master.handle_accept_intercept(self)
+
+ def toggle_mark(self):
+ if self.marked:
+ self.marked = False
+ else:
+ self.marked = True
class ProtocolHandler(object):