From 858b8d2274bd2fc50f9537cd503385ce1737f8b1 Mon Sep 17 00:00:00 2001 From: lqqyt2423 <974923609@qq.com> Date: Tue, 17 Jan 2023 17:15:33 +0800 Subject: [PATCH] web flow filter refactoring --- web/client/src/lib/filter.ts | 125 +++++++++++++++++++++++++++++++++++ web/client/src/lib/flow.ts | 125 +---------------------------------- 2 files changed, 126 insertions(+), 124 deletions(-) create mode 100644 web/client/src/lib/filter.ts diff --git a/web/client/src/lib/filter.ts b/web/client/src/lib/filter.ts new file mode 100644 index 0000000..0dd5b17 --- /dev/null +++ b/web/client/src/lib/filter.ts @@ -0,0 +1,125 @@ +import type { Flow, Header } from './flow' + +const FLOW_FILTER_SCOPES = ['url', 'method', 'code', 'header', 'reqheader', 'resheader', 'body', 'reqbody', 'resbody', 'all'] as const +type FlowFilterScope = typeof FLOW_FILTER_SCOPES[number] + +export class FlowFilter { + private keyword: string | RegExp | undefined + private scope: FlowFilterScope = 'url' + + constructor(text: string) { + text = text.trim() + if (!text) return + + for (const scope of FLOW_FILTER_SCOPES) { + if (text.startsWith(`${scope}:`)) { + this.scope = scope + text = text.replace(`${scope}:`, '').trim() + break + } + } + if (!text) return + + // regexp + if (text.startsWith('/') && (text.endsWith('/') || text.endsWith('/i'))) { + let flags: string | undefined + if (text.endsWith('i')) { + flags = 'i' + text = text.slice(0, -1) + } + text = text.slice(1, -1).trim() + if (!text) return + this.keyword = new RegExp(text, flags) + } + // string + else { + this.keyword = text + } + } + + public match(flow: Flow): boolean { + switch (this.scope) { + case 'url': + return this.matchUrl(flow) + case 'method': + return this.matchMethod(flow) + case 'code': + return this.matchCode(flow) + case 'reqheader': + return this.matchReqHeader(flow) + case 'resheader': + return this.matchResHeader(flow) + case 'header': + return this.matchHeader(flow) + case 'reqbody': + return this.matchReqBody(flow) + case 'resbody': + return this.matchResBody(flow) + case 'body': + return this.matchBody(flow) + case 'all': + return this.matchAll(flow) + default: + throw new Error(`invalid scope ${this.scope}`) + } + } + + private matchUrl(flow: Flow): boolean { + return this.matchKeyword(flow.request.url) + } + + private matchMethod(flow: Flow): boolean { + return this.matchKeyword(flow.request.method) || this.matchKeyword(flow.request.method.toLowerCase()) + } + + private matchCode(flow: Flow): boolean { + if (!flow.response) return false + return this.matchKeyword(flow.response.statusCode.toString()) + } + + private _matchHeader(header: Header): boolean { + return Object.entries(header).some(([key, vals]) => { + return [key].concat(vals).some(text => this.matchKeyword(text)) + }) + } + + private matchReqHeader(flow: Flow): boolean { + return this._matchHeader(flow.request.header) + } + + private matchResHeader(flow: Flow): boolean { + if (!flow.response) return false + return this._matchHeader(flow.response.header) + } + + private matchHeader(flow: Flow): boolean { + return this.matchReqHeader(flow) || this.matchResHeader(flow) + } + + private matchReqBody(flow: Flow): boolean { + const body = flow.requestBody() + if (!body) return false + return this.matchKeyword(body) + } + + private matchResBody(flow: Flow): boolean { + const body = flow.responseBody() + if (!body) return false + return this.matchKeyword(body) + } + + private matchBody(flow: Flow): boolean { + return this.matchReqBody(flow) || this.matchResBody(flow) + } + + private matchAll(flow: Flow): boolean { + return this.matchUrl(flow) || this.matchMethod(flow) || this.matchHeader(flow) || this.matchBody(flow) + } + + private matchKeyword(text: string): boolean { + if (!this.keyword) return true + if (!text) return false + if (this.keyword instanceof RegExp) return this.keyword.test(text) + return text.includes(this.keyword) + } +} diff --git a/web/client/src/lib/flow.ts b/web/client/src/lib/flow.ts index 7e9b99c..df0f4d2 100644 --- a/web/client/src/lib/flow.ts +++ b/web/client/src/lib/flow.ts @@ -1,6 +1,7 @@ import type { ConnectionManager, IConnection } from './connection' import { IMessage, MessageType } from './message' import { arrayBufferToBase64, bufHexView, getSize, isTextBody } from './utils' +import { FlowFilter } from './filter' export type Header = Record @@ -262,130 +263,6 @@ export class Flow { } } -const FLOW_FILTER_SCOPES = ['url', 'method', 'code', 'header', 'reqheader', 'resheader', 'body', 'reqbody', 'resbody', 'all'] as const -type FlowFilterScope = typeof FLOW_FILTER_SCOPES[number] - -class FlowFilter { - private keyword: string | RegExp | undefined - private scope: FlowFilterScope = 'url' - - constructor(text: string) { - text = text.trim() - if (!text) return - - for (const scope of FLOW_FILTER_SCOPES) { - if (text.startsWith(`${scope}:`)) { - this.scope = scope - text = text.replace(`${scope}:`, '').trim() - break - } - } - if (!text) return - - // regexp - if (text.startsWith('/') && (text.endsWith('/') || text.endsWith('/i'))) { - let flags: string | undefined - if (text.endsWith('i')) { - flags = 'i' - text = text.slice(0, -1) - } - text = text.slice(1, -1).trim() - if (!text) return - this.keyword = new RegExp(text, flags) - } - // string - else { - this.keyword = text - } - } - - public match(flow: Flow): boolean { - switch (this.scope) { - case 'url': - return this.matchUrl(flow) - case 'method': - return this.matchMethod(flow) - case 'code': - return this.matchCode(flow) - case 'reqheader': - return this.matchReqHeader(flow) - case 'resheader': - return this.matchResHeader(flow) - case 'header': - return this.matchHeader(flow) - case 'reqbody': - return this.matchReqBody(flow) - case 'resbody': - return this.matchResBody(flow) - case 'body': - return this.matchBody(flow) - case 'all': - return this.matchAll(flow) - default: - throw new Error(`invalid scope ${this.scope}`) - } - } - - private matchUrl(flow: Flow): boolean { - return this.matchKeyword(flow.request.url) - } - - private matchMethod(flow: Flow): boolean { - return this.matchKeyword(flow.request.method) || this.matchKeyword(flow.request.method.toLowerCase()) - } - - private matchCode(flow: Flow): boolean { - if (!flow.response) return false - return this.matchKeyword(flow.response.statusCode.toString()) - } - - private _matchHeader(header: Header): boolean { - return Object.entries(header).some(([key, vals]) => { - return [key].concat(vals).some(text => this.matchKeyword(text)) - }) - } - - private matchReqHeader(flow: Flow): boolean { - return this._matchHeader(flow.request.header) - } - - private matchResHeader(flow: Flow): boolean { - if (!flow.response) return false - return this._matchHeader(flow.response.header) - } - - private matchHeader(flow: Flow): boolean { - return this.matchReqHeader(flow) || this.matchResHeader(flow) - } - - private matchReqBody(flow: Flow): boolean { - const body = flow.requestBody() - if (!body) return false - return this.matchKeyword(body) - } - - private matchResBody(flow: Flow): boolean { - const body = flow.responseBody() - if (!body) return false - return this.matchKeyword(body) - } - - private matchBody(flow: Flow): boolean { - return this.matchReqBody(flow) || this.matchResBody(flow) - } - - private matchAll(flow: Flow): boolean { - return this.matchUrl(flow) || this.matchMethod(flow) || this.matchHeader(flow) || this.matchBody(flow) - } - - private matchKeyword(text: string): boolean { - if (!this.keyword) return true - if (!text) return false - if (this.keyword instanceof RegExp) return this.keyword.test(text) - return text.includes(this.keyword) - } -} - export class FlowManager { private items: Flow[] private _map: Map