REFACTOR : findings를index가 아닌 모듈애서 만들도록 수정
This commit is contained in:
parent
7b704cacf4
commit
f1b5ef5f9b
2 changed files with 40 additions and 12 deletions
|
|
@ -1,22 +1,46 @@
|
|||
import type { Request, Response } from "caido:utils";
|
||||
import type { SDK, DefineAPI } from "caido:plugin";
|
||||
|
||||
// 토큰 누출 검사 결과를 담는 구조
|
||||
export interface TokenLeakResult {
|
||||
found: boolean; // 토큰이 발견되었는지 여부 (true/false)
|
||||
location: 'url' | 'body' | 'header'; // 토큰이 발견된 위치 (url, body, header 중 하나)
|
||||
title: string; // 경고 제목
|
||||
description: string; // 상세 설명
|
||||
value?: string; // 실제 발견된 값 (선택적)
|
||||
found: boolean; // 토큰이 발견되었는지 여부 (true/false)
|
||||
location: 'url' | 'body' | 'header'; // 토큰이 발견된 위치 (url, body, header 중 하나)
|
||||
title: string; // 경고 제목
|
||||
description: string; // 상세 설명
|
||||
value?: string; // 실제 발견된 값 (선택적)
|
||||
}
|
||||
|
||||
// 액세스 토큰 누출 검사 클래스
|
||||
export class AccessTokenLeakController {
|
||||
|
||||
/**
|
||||
* @param request - 검사할 HTTP 요청 객체
|
||||
* @returns 토큰이 발견되면 결과 객체, 없으면 null
|
||||
*/
|
||||
async testReq(request: Request): Promise<TokenLeakResult | null> {
|
||||
async testReq(sdk: SDK<DefineAPI<{}>>, request: Request): Promise<void> {
|
||||
const result = await this._scanRequest(request);
|
||||
if (result) {
|
||||
await sdk.findings.create({
|
||||
title: result.title,
|
||||
description: result.description,
|
||||
request,
|
||||
reporter: "",
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
async testResp(sdk: SDK<DefineAPI<{}>>, response: Response, request: Request): Promise<void> {
|
||||
const result = await this._scanResponse(response);
|
||||
if (result) {
|
||||
await sdk.findings.create({
|
||||
title: result.title,
|
||||
description: result.description,
|
||||
request,
|
||||
reporter: "",
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* @param request - 검사할 HTTP 요청 객체
|
||||
* @returns 토큰이 발견되면 결과 객체, 없으면 null
|
||||
*/
|
||||
async _scanRequest(request: Request): Promise<TokenLeakResult | null> {
|
||||
|
||||
// === 1. URL에서 토큰 검사 ===
|
||||
const url = request.getUrl();
|
||||
|
|
@ -60,7 +84,7 @@ export class AccessTokenLeakController {
|
|||
* @param response - 검사할 HTTP 응답 객체
|
||||
* @returns 토큰이 발견되면 결과 객체, 없으면 null
|
||||
*/
|
||||
async testResp(response: Response): Promise<TokenLeakResult | null> {
|
||||
async _scanResponse(response: Response): Promise<TokenLeakResult | null> {
|
||||
|
||||
// === 1. Location 헤더에서 토큰 검사 ===
|
||||
const locationHeader = response.getHeader("Location");
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue