mirror of
https://github.com/j93es/oauth-backend.git
synced 2026-06-04 06:51:51 +09:00
진행률 오류 수정
This commit is contained in:
parent
a4b14ab20f
commit
27e7a290ba
1 changed files with 5 additions and 7 deletions
|
|
@ -19,7 +19,6 @@ class RedirectRateLimiter:
|
|||
self.failure_count = 0
|
||||
self.consecutive_failures = 0
|
||||
self.blocked_until = 0
|
||||
self.pattern_index = 0 # 현재 테스트 중인 패턴 번호
|
||||
self.should_abandon_target = False
|
||||
|
||||
# 설정값 (전체 패턴 기준으로 최적화)
|
||||
|
|
@ -36,7 +35,6 @@ class RedirectRateLimiter:
|
|||
self.failure_count = 0
|
||||
self.consecutive_failures = 0
|
||||
self.blocked_until = 0
|
||||
self.pattern_index = 0
|
||||
self.should_abandon_target = False
|
||||
print("[RATE_LIMIT] 새로운 타겟을 위해 Rate Limiter 리셋됨")
|
||||
|
||||
|
|
@ -76,10 +74,6 @@ class RedirectRateLimiter:
|
|||
|
||||
self.last_request = time.time()
|
||||
self.request_count += 1
|
||||
self.pattern_index += 1
|
||||
|
||||
if self.pattern_index % 5 == 0:
|
||||
print(f"[PROGRESS] 📊 {self.pattern_index}/71 패턴 테스트 완료 ({(self.pattern_index/71)*100:.1f}%)")
|
||||
|
||||
return True
|
||||
|
||||
|
|
@ -1381,7 +1375,7 @@ class OpenRedirectChecker:
|
|||
|
||||
async def _test_oauth_redirect(self, url, parsed, query, test_type="UNKNOWN"):
|
||||
"""OAuth 리다이렉트 취약점 테스트"""
|
||||
|
||||
|
||||
# 테스트 타입별로 중복 방지
|
||||
endpoint_key = f"{test_type}:{parsed.netloc}{parsed.path}"
|
||||
|
||||
|
|
@ -1445,6 +1439,10 @@ class OpenRedirectChecker:
|
|||
for i, payload in enumerate(self.bypass_payloads, 1):
|
||||
print(f"[{i:2d}/{len(self.bypass_payloads)}] {payload.name}", end=" ... ")
|
||||
|
||||
if i % 10 == 0:
|
||||
progress = (i / len(self.bypass_payloads)) * 100
|
||||
print(f"\n[PROGRESS] 📊 {i}/{len(self.bypass_payloads)} 패턴 테스트 완료 ({progress:.1f}%)")
|
||||
|
||||
result = await self._test_bypass_pattern(
|
||||
url, query, parsed, original_value, payload, param_name
|
||||
)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue