mirror of
https://github.com/j93es/oauth-backend.git
synced 2026-06-04 10:41:52 +09:00
[Add] csrf 포팅
This commit is contained in:
parent
c0c6743423
commit
a5186a7e44
4 changed files with 210 additions and 19 deletions
|
|
@ -1,16 +1,31 @@
|
|||
from mitmproxy import http
|
||||
import asyncio
|
||||
from pkce_check import PKCEDowngradeChecker
|
||||
from csrf_check import CsrfChecker
|
||||
|
||||
class PKCEAddon:
|
||||
def __init__(self):
|
||||
self.checker = PKCEDowngradeChecker()
|
||||
|
||||
async def request(self, flow: http.HTTPFlow):
|
||||
print(f"[DEBUG] Processing request: {flow.request.method} {flow.request.pretty_url}")
|
||||
# print(f"[DEBUG] Processing request: {flow.request.method} {flow.request.pretty_url}")
|
||||
try:
|
||||
await self.checker.test(flow)
|
||||
except Exception as e:
|
||||
print(f"[ERROR] Addon failed: {e}")
|
||||
pass
|
||||
|
||||
|
||||
class CsrfAddon:
|
||||
def __init__(self):
|
||||
self.checker = CsrfChecker()
|
||||
|
||||
addons = [PKCEAddon()]
|
||||
async def response(self, flow: http.HTTPFlow):
|
||||
# print(f"[DEBUG] Processing request for CSRF check: {flow.request.method} {flow.request.pretty_url}")
|
||||
try:
|
||||
await self.checker.response(flow)
|
||||
except Exception as e:
|
||||
print(f"[ERROR] CSRF Addon failed: {e}")
|
||||
pass
|
||||
|
||||
addons = [PKCEAddon(), CsrfAddon()]
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue