diff options
author | Wojtek Kosior <koszko@koszko.org> | 2022-10-22 13:47:40 +0200 |
---|---|---|
committer | Wojtek Kosior <koszko@koszko.org> | 2022-10-22 14:00:35 +0200 |
commit | 59a4b477dd922d2f839a717fe199501e7cb71e6e (patch) | |
tree | d6120e13f493ef057ee5445c566e1aca11a9f7f5 /src/hydrilla/proxy/policies/payload_resource.py | |
parent | 1fc2dbe5a3e867eb5e6c2f759409f1c94146a794 (diff) | |
download | haketilo-hydrilla-59a4b477dd922d2f839a717fe199501e7cb71e6e.tar.gz haketilo-hydrilla-59a4b477dd922d2f839a717fe199501e7cb71e6e.zip |
[proxy] simplify usage of class variables of `Policy` subtypes
This commit also prevents popup script injection to Haketilo-served pages like `https://hkt.mitm.it/`.
Diffstat (limited to 'src/hydrilla/proxy/policies/payload_resource.py')
-rw-r--r-- | src/hydrilla/proxy/policies/payload_resource.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/src/hydrilla/proxy/policies/payload_resource.py b/src/hydrilla/proxy/policies/payload_resource.py index d8e5ea5..38cfd21 100644 --- a/src/hydrilla/proxy/policies/payload_resource.py +++ b/src/hydrilla/proxy/policies/payload_resource.py @@ -212,10 +212,9 @@ MessageInfo = t.Union[ @dc.dataclass(frozen=True) class PayloadResourcePolicy(PayloadAwarePolicy): - """....""" - _process_request: t.ClassVar[bool] = True + _process_request = base.MsgProcessOpt.MUST - priority: t.ClassVar[base.PolicyPriority] = base.PolicyPriority._THREE + priority = base.PolicyPriority._THREE def extract_resource_path(self, request_url: ParsedUrl) -> tuple[str, ...]: # Payload resource pattern has path of the form: @@ -370,10 +369,10 @@ resource_blocked_response = http_messages.ResponseInfo.make( @dc.dataclass(frozen=True) class BlockedResponsePolicy(base.Policy): - """....""" - _process_request: t.ClassVar[bool] = True + _process_request = base.MsgProcessOpt.MUST + _process_response = base.MsgProcessOpt.MUST_NOT - priority: t.ClassVar[base.PolicyPriority] = base.PolicyPriority._THREE + priority = base.PolicyPriority._THREE def consume_request(self, request_info: http_messages.RequestInfo) \ -> http_messages.ResponseInfo: |