@sectester/repeater 0.16.1

package/src/request-runner/Response.d.ts

@@ -0,0 +1,17 @@
+import { Protocol } from '../models';
+export declare class Response {
+    readonly protocol: Protocol;
+    readonly statusCode?: number;
+    readonly headers?: Record<string, string | string[] | undefined>;
+    readonly body?: string;
+    readonly message?: string;
+    readonly errorCode?: string;
+    constructor({ protocol, statusCode, headers, body, message, errorCode }: {
+        protocol: Protocol;
+        statusCode?: number;
+        message?: string;
+        errorCode?: string;
+        headers?: Record<string, string | string[] | undefined>;
+        body?: string;
+    });
+}

package/src/request-runner/Response.js

@@ -0,0 +1,15 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Response = void 0;
+class Response {
+    constructor({ protocol, statusCode, headers, body, message, errorCode }) {
+        this.protocol = protocol;
+        this.statusCode = statusCode;
+        this.headers = headers;
+        this.body = body;
+        this.errorCode = errorCode;
+        this.message = message;
+    }
+}
+exports.Response = Response;
+//# sourceMappingURL=Response.js.map

package/src/request-runner/Response.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"Response.js","sourceRoot":"","sources":["../../../../../packages/repeater/src/request-runner/Response.ts"],"names":[],"mappings":";;;AAEA,MAAa,QAAQ;IAQnB,YAAY,EACV,QAAQ,EACR,UAAU,EACV,OAAO,EACP,IAAI,EACJ,OAAO,EACP,SAAS,EAQV;QACC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,UAAU,GAAG,UAAU,CAAC;QAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;QAC3B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;CACF;AA9BD,4BA8BC"}

package/src/request-runner/index.d.ts

@@ -0,0 +1,5 @@
+import './register';
+export * from './Request';
+export * from './RequestRunner';
+export * from './RequestRunnerOptions';
+export * from './Response';

package/src/request-runner/index.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+require("./register");
+(0, tslib_1.__exportStar)(require("./Request"), exports);
+(0, tslib_1.__exportStar)(require("./RequestRunner"), exports);
+(0, tslib_1.__exportStar)(require("./RequestRunnerOptions"), exports);
+(0, tslib_1.__exportStar)(require("./Response"), exports);
+//# sourceMappingURL=index.js.map

package/src/request-runner/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../packages/repeater/src/request-runner/index.ts"],"names":[],"mappings":";;;AAAA,sBAAoB;AAEpB,yDAA0B;AAC1B,+DAAgC;AAChC,sEAAuC;AACvC,0DAA2B"}

package/src/request-runner/protocols/HttpRequestRunner.d.ts

@@ -0,0 +1,29 @@
+import { Request } from '../Request';
+import { RequestRunner } from '../RequestRunner';
+import { RequestRunnerOptions } from '../RequestRunnerOptions';
+import { Response } from '../Response';
+import { Protocol } from '../../models';
+import { Logger } from '@sectester/core';
+export declare class HttpRequestRunner implements RequestRunner {
+    private readonly options;
+    private readonly logger;
+    private readonly proxy?;
+    private readonly httpAgent?;
+    private readonly httpsAgent?;
+    private readonly maxContentLength;
+    get protocol(): Protocol;
+    constructor(options: RequestRunnerOptions, logger: Logger);
+    run(options: Request): Promise<Response>;
+    private handleRequestError;
+    private request;
+    private getRequestAgent;
+    private truncateResponse;
+    private parseContentType;
+    private parseBody;
+    /**
+     * Allows to attack headers. Node.js does not accept any other characters
+     * which violate [rfc7230](https://tools.ietf.org/html/rfc7230#section-3.2.6).
+     * To override default behavior bypassing {@link OutgoingMessage.setHeader} method we have to set headers via internal symbol.
+     */
+    private setHeaders;
+}

package/src/request-runner/protocols/HttpRequestRunner.js

@@ -0,0 +1,180 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.HttpRequestRunner = void 0;
+const tslib_1 = require("tslib");
+const RequestRunnerOptions_1 = require("../RequestRunnerOptions");
+const Response_1 = require("../Response");
+const models_1 = require("../../models");
+const core_1 = require("@sectester/core");
+const request_promise_1 = (0, tslib_1.__importDefault)(require("request-promise"));
+const socks_proxy_agent_1 = require("socks-proxy-agent");
+const tsyringe_1 = require("tsyringe");
+const content_type_1 = require("content-type");
+const url_1 = require("url");
+const http_1 = (0, tslib_1.__importDefault)(require("http"));
+const https_1 = (0, tslib_1.__importDefault)(require("https"));
+let HttpRequestRunner = class HttpRequestRunner {
+    constructor(options, logger) {
+        this.options = options;
+        this.logger = logger;
+        if (this.options.proxyUrl) {
+            this.proxy = new socks_proxy_agent_1.SocksProxyAgent({
+                ...(0, url_1.parse)(this.options.proxyUrl)
+            });
+        }
+        this.maxContentLength =
+            typeof this.options.maxContentLength === 'number'
+                ? this.options.maxContentLength
+                : -1;
+        if (this.options.reuseConnection) {
+            const agentOptions = {
+                keepAlive: true,
+                maxSockets: 100,
+                timeout: this.options.timeout
+            };
+            this.httpsAgent = new https_1.default.Agent(agentOptions);
+            this.httpAgent = new http_1.default.Agent(agentOptions);
+        }
+    }
+    get protocol() {
+        return models_1.Protocol.HTTP;
+    }
+    async run(options) {
+        try {
+            if (this.options.headers) {
+                options.setHeaders(this.options.headers);
+            }
+            this.logger.debug('Executing HTTP request with following params: %j', options);
+            const response = await this.request(options);
+            return new Response_1.Response({
+                protocol: this.protocol,
+                statusCode: response.statusCode,
+                headers: response.headers,
+                body: response.body
+            });
+        }
+        catch (err) {
+            return this.handleRequestError(err, options);
+        }
+    }
+    handleRequestError(err, options) {
+        var _a, _b, _c, _d, _e, _f;
+        if (err.response) {
+            const { response } = err;
+            return new Response_1.Response({
+                protocol: this.protocol,
+                statusCode: response.statusCode,
+                headers: response.headers,
+                body: response.body
+            });
+        }
+        const message = (_b = (_a = err.cause) === null || _a === void 0 ? void 0 : _a.message) !== null && _b !== void 0 ? _b : err.message;
+        const errorCode = (_f = (_d = (_c = err.cause) === null || _c === void 0 ? void 0 : _c.code) !== null && _d !== void 0 ? _d : (_e = err.error) === null || _e === void 0 ? void 0 : _e.syscall) !== null && _f !== void 0 ? _f : err.name;
+        this.logger.error('Error executing request: "%s %s HTTP/1.1"', options.method, options.url);
+        this.logger.error('Cause: %s', message);
+        return new Response_1.Response({
+            message,
+            errorCode,
+            protocol: this.protocol
+        });
+    }
+    async request(options) {
+        let responseTruncatePromise;
+        const res = await (0, request_promise_1.default)({
+            agent: this.getRequestAgent(options),
+            body: options.body,
+            followRedirect: false,
+            gzip: true,
+            method: options.method,
+            resolveWithFullResponse: true,
+            strictSSL: false,
+            rejectUnauthorized: false,
+            timeout: this.options.timeout,
+            url: options.url
+        })
+            .on('request', (req) => this.setHeaders(req, options))
+            .on('response', (response) => (responseTruncatePromise = this.truncateResponse(response)));
+        // eslint-disable-next-line @typescript-eslint/no-non-null-assertion
+        const body = await responseTruncatePromise;
+        if (body) {
+            res.body = body.toString();
+            res.headers['content-length'] = String(body.byteLength);
+        }
+        return res;
+    }
+    getRequestAgent(options) {
+        var _a;
+        return ((_a = this.proxy) !== null && _a !== void 0 ? _a : (options.url.startsWith('https') ? this.httpsAgent : this.httpAgent));
+    }
+    async truncateResponse(res) {
+        var _a;
+        if (res.statusCode === 204 || res.method === 'HEAD') {
+            return;
+        }
+        const type = this.parseContentType(res);
+        const maxBodySize = this.maxContentLength * 1024;
+        const requiresTruncating = !((_a = this.options.allowedMimes) === null || _a === void 0 ? void 0 : _a.some((mime) => type.startsWith(mime)));
+        return this.parseBody(res, { maxBodySize, requiresTruncating });
+    }
+    parseContentType(res) {
+        let type = res.headers['content-type'] || 'text/plain';
+        try {
+            ({ type } = (0, content_type_1.parse)(type));
+        }
+        catch {
+            // noop
+        }
+        return type;
+    }
+    async parseBody(res, options) {
+        let truncated = false;
+        const chunks = [];
+        let chunksByteLength = 0;
+        for await (const chunk of res) {
+            chunks.push(chunk);
+            chunksByteLength += Buffer.from(chunk).byteLength;
+            truncated =
+                this.maxContentLength !== -1 &&
+                    chunksByteLength > options.maxBodySize &&
+                    options.requiresTruncating;
+            if (truncated) {
+                res.destroy();
+                break;
+            }
+        }
+        const body = Buffer.concat(chunks);
+        if (truncated) {
+            this.logger.debug('Truncate original response body to %i bytes', options.maxBodySize);
+            return body.slice(0, options.maxBodySize);
+        }
+        return body;
+    }
+    /**
+     * Allows to attack headers. Node.js does not accept any other characters
+     * which violate [rfc7230](https://tools.ietf.org/html/rfc7230#section-3.2.6).
+     * To override default behavior bypassing {@link OutgoingMessage.setHeader} method we have to set headers via internal symbol.
+     */
+    setHeaders(req, options) {
+        var _a;
+        const symbols = Object.getOwnPropertySymbols(req);
+        const headersSymbol = symbols.find(
+        // ADHOC: Node.js version < 12 uses "outHeadersKey" symbol to set headers
+        item => ['Symbol(kOutHeaders)', 'Symbol(outHeadersKey)'].includes(item.toString()));
+        if (!req.headersSent && headersSymbol && options.headers) {
+            const headers = (req[headersSymbol] =
+                (_a = req[headersSymbol]) !== null && _a !== void 0 ? _a : Object.create(null));
+            Object.entries(options.headers).forEach(([key, value]) => {
+                if (key) {
+                    headers[key.toLowerCase()] = [key.toLowerCase(), value !== null && value !== void 0 ? value : ''];
+                }
+            });
+        }
+    }
+};
+HttpRequestRunner = (0, tslib_1.__decorate)([
+    (0, tsyringe_1.injectable)(),
+    (0, tslib_1.__param)(0, (0, tsyringe_1.inject)(RequestRunnerOptions_1.RequestRunnerOptions)),
+    (0, tslib_1.__metadata)("design:paramtypes", [Object, core_1.Logger])
+], HttpRequestRunner);
+exports.HttpRequestRunner = HttpRequestRunner;
+//# sourceMappingURL=HttpRequestRunner.js.map

package/src/request-runner/protocols/HttpRequestRunner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"HttpRequestRunner.js","sourceRoot":"","sources":["../../../../../../packages/repeater/src/request-runner/protocols/HttpRequestRunner.ts"],"names":[],"mappings":";;;;AAEA,kEAA+D;AAC/D,0CAAuC;AACvC,yCAAwC;AACxC,0CAAyC;AACzC,mFAAsC;AAEtC,yDAAoD;AACpD,uCAA8C;AAC9C,+CAAyD;AACzD,6BAA4B;AAC5B,6DAA6C;AAC7C,+DAA4C;AAG5C,IAAa,iBAAiB,GAA9B,MAAa,iBAAiB;IAU5B,YAEmB,OAA6B,EAC7B,MAAc;QADd,YAAO,GAAP,OAAO,CAAsB;QAC7B,WAAM,GAAN,MAAM,CAAQ;QAE/B,IAAI,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE;YACzB,IAAI,CAAC,KAAK,GAAG,IAAI,mCAAe,CAAC;gBAC/B,GAAG,IAAA,WAAK,EAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;aAChC,CAAC,CAAC;SACJ;QAED,IAAI,CAAC,gBAAgB;YACnB,OAAO,IAAI,CAAC,OAAO,CAAC,gBAAgB,KAAK,QAAQ;gBAC/C,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,gBAAgB;gBAC/B,CAAC,CAAC,CAAC,CAAC,CAAC;QAET,IAAI,IAAI,CAAC,OAAO,CAAC,eAAe,EAAE;YAChC,MAAM,YAAY,GAAiB;gBACjC,SAAS,EAAE,IAAI;gBACf,UAAU,EAAE,GAAG;gBACf,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;aAC9B,CAAC;YAEF,IAAI,CAAC,UAAU,GAAG,IAAI,eAAK,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;YAChD,IAAI,CAAC,SAAS,GAAG,IAAI,cAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;SAC/C;IACH,CAAC;IA9BD,IAAI,QAAQ;QACV,OAAO,iBAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IA8BM,KAAK,CAAC,GAAG,CAAC,OAAgB;QAC/B,IAAI;YACF,IAAI,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE;gBACxB,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;aAC1C;YAED,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,kDAAkD,EAClD,OAAO,CACR,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAE7C,OAAO,IAAI,mBAAQ,CAAC;gBAClB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,IAAI,EAAE,QAAQ,CAAC,IAAI;aACpB,CAAC,CAAC;SACJ;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;SAC9C;IACH,CAAC;IAEO,kBAAkB,CAAC,GAAQ,EAAE,OAAgB;;QACnD,IAAI,GAAG,CAAC,QAAQ,EAAE;YAChB,MAAM,EAAE,QAAQ,EAAE,GAAG,GAAG,CAAC;YAEzB,OAAO,IAAI,mBAAQ,CAAC;gBAClB,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,UAAU,EAAE,QAAQ,CAAC,UAAU;gBAC/B,OAAO,EAAE,QAAQ,CAAC,OAAO;gBACzB,IAAI,EAAE,QAAQ,CAAC,IAAI;aACpB,CAAC,CAAC;SACJ;QAED,MAAM,OAAO,GAAG,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,OAAO,mCAAI,GAAG,CAAC,OAAO,CAAC;QAClD,MAAM,SAAS,GAAG,MAAA,MAAA,MAAA,GAAG,CAAC,KAAK,0CAAE,IAAI,mCAAI,MAAA,GAAG,CAAC,KAAK,0CAAE,OAAO,mCAAI,GAAG,CAAC,IAAI,CAAC;QAEpE,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,2CAA2C,EAC3C,OAAO,CAAC,MAAM,EACd,OAAO,CAAC,GAAG,CACZ,CAAC;QACF,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAExC,OAAO,IAAI,mBAAQ,CAAC;YAClB,OAAO;YACP,SAAS;YACT,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,OAAgB;QACpC,IAAI,uBAAoD,CAAC;QAEzD,MAAM,GAAG,GAAG,MAAM,IAAA,yBAAO,EAAC;YACxB,KAAK,EAAE,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC;YACpC,IAAI,EAAE,OAAO,CAAC,IAAI;YAClB,cAAc,EAAE,KAAK;YACrB,IAAI,EAAE,IAAI;YACV,MAAM,EAAE,OAAO,CAAC,MAAM;YACtB,uBAAuB,EAAE,IAAI;YAC7B,SAAS,EAAE,KAAK;YAChB,kBAAkB,EAAE,KAAK;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,GAAG,EAAE,OAAO,CAAC,GAAG;SACjB,CAAC;aACC,EAAE,CAAC,SAAS,EAAE,CAAC,GAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;aACtE,EAAE,CACD,UAAU,EACV,CAAC,QAA0B,EAAE,EAAE,CAC7B,CAAC,uBAAuB,GAAG,IAAI,CAAC,gBAAgB,CAAC,QAAQ,CAAC,CAAC,CAC9D,CAAC;QAEJ,oEAAoE;QACpE,MAAM,IAAI,GAAG,MAAM,uBAAwB,CAAC;QAE5C,IAAI,IAAI,EAAE;YACR,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC3B,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;SACzD;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,eAAe,CAAC,OAAgB;;QACtC,OAAO,CACL,MAAA,IAAI,CAAC,KAAK,mCACV,CAAC,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CACrE,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC5B,GAAqB;;QAErB,IAAI,GAAG,CAAC,UAAU,KAAK,GAAG,IAAI,GAAG,CAAC,MAAM,KAAK,MAAM,EAAE;YACnD,OAAO;SACR;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC;QACxC,MAAM,WAAW,GAAG,IAAI,CAAC,gBAAgB,GAAG,IAAI,CAAC;QAEjD,MAAM,kBAAkB,GAAG,CAAC,CAAA,MAAA,IAAI,CAAC,OAAO,CAAC,YAAY,0CAAE,IAAI,CACzD,CAAC,IAAY,EAAE,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CACxC,CAAA,CAAC;QAEF,OAAO,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC,CAAC;IAClE,CAAC;IAEO,gBAAgB,CAAC,GAAqB;QAC5C,IAAI,IAAI,GAAG,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,YAAY,CAAC;QAEvD,IAAI;YACF,CAAC,EAAE,IAAI,EAAE,GAAG,IAAA,oBAAgB,EAAC,IAAI,CAAC,CAAC,CAAC;SACrC;QAAC,MAAM;YACN,OAAO;SACR;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,SAAS,CACrB,GAAqB,EACrB,OAA6D;QAE7D,IAAI,SAAS,GAAG,KAAK,CAAC;QAEtB,MAAM,MAAM,GAAG,EAAE,CAAC;QAClB,IAAI,gBAAgB,GAAG,CAAC,CAAC;QAEzB,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,GAAG,EAAE;YAC7B,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAEnB,gBAAgB,IAAI,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,UAAU,CAAC;YAClD,SAAS;gBACP,IAAI,CAAC,gBAAgB,KAAK,CAAC,CAAC;oBAC5B,gBAAgB,GAAG,OAAO,CAAC,WAAW;oBACtC,OAAO,CAAC,kBAAkB,CAAC;YAE7B,IAAI,SAAS,EAAE;gBACb,GAAG,CAAC,OAAO,EAAE,CAAC;gBACd,MAAM;aACP;SACF;QAED,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEnC,IAAI,SAAS,EAAE;YACb,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,6CAA6C,EAC7C,OAAO,CAAC,WAAW,CACpB,CAAC;YAEF,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,OAAO,CAAC,WAAW,CAAC,CAAC;SAC3C;QAED,OAAO,IAAI,CAAC;IACd,CAAC;IAED;;;;OAIG;IACK,UAAU,CAAC,GAAoB,EAAE,OAAgB;;QACvD,MAAM,OAAO,GAAa,MAAM,CAAC,qBAAqB,CAAC,GAAG,CAAC,CAAC;QAC5D,MAAM,aAAa,GAAuB,OAAO,CAAC,IAAI;QACpD,yEAAyE;QACzE,IAAI,CAAC,EAAE,CACL,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,CAAC,QAAQ,CACvD,IAAI,CAAC,QAAQ,EAAE,CAChB,CACJ,CAAC;QAEF,IAAI,CAAC,GAAG,CAAC,WAAW,IAAI,aAAa,IAAI,OAAO,CAAC,OAAO,EAAE;YACxD,MAAM,OAAO,GAAG,CAAC,GAAG,CAAC,aAAa,CAAC;gBACjC,MAAA,GAAG,CAAC,aAAa,CAAC,mCAAI,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC;YAE7C,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,OAAO,CACrC,CAAC,CAAC,GAAG,EAAE,KAAK,CAA8B,EAAE,EAAE;gBAC5C,IAAI,GAAG,EAAE;oBACP,OAAO,CAAC,GAAG,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,GAAG,CAAC,WAAW,EAAE,EAAE,KAAK,aAAL,KAAK,cAAL,KAAK,GAAI,EAAE,CAAC,CAAC;iBAC/D;YACH,CAAC,CACF,CAAC;SACH;IACH,CAAC;CACF,CAAA;AAlOY,iBAAiB;IAD7B,IAAA,qBAAU,GAAE;IAYR,wBAAA,IAAA,iBAAM,EAAC,2CAAoB,CAAC,CAAA;0DAEJ,aAAM;GAbtB,iBAAiB,CAkO7B;AAlOY,8CAAiB"}

package/src/request-runner/protocols/WsRequestRunner.d.ts

@@ -0,0 +1,24 @@
+import { Request } from '../Request';
+import { RequestRunner } from '../RequestRunner';
+import { RequestRunnerOptions } from '../RequestRunnerOptions';
+import { Response } from '../Response';
+import { Protocol } from '../../models';
+import { Logger } from '@sectester/core';
+export declare class WsRequestRunner implements RequestRunner {
+    private readonly options;
+    private readonly logger;
+    static readonly FORBIDDEN_HEADERS: ReadonlySet<string>;
+    private readonly agent?;
+    constructor(options: RequestRunnerOptions, logger: Logger);
+    get protocol(): Protocol;
+    run(options: Request): Promise<Response>;
+    private createWebSocketClient;
+    private sendMessage;
+    private createWsResponse;
+    private handleRequestError;
+    private setTimeout;
+    private consume;
+    private waitForResponse;
+    private connect;
+    private normalizeHeaders;
+}

package/src/request-runner/protocols/WsRequestRunner.js

@@ -0,0 +1,148 @@
+"use strict";
+var WsRequestRunner_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.WsRequestRunner = void 0;
+const tslib_1 = require("tslib");
+const RequestRunnerOptions_1 = require("../RequestRunnerOptions");
+const Response_1 = require("../Response");
+const models_1 = require("../../models");
+const core_1 = require("@sectester/core");
+const ws_1 = (0, tslib_1.__importDefault)(require("ws"));
+const tsyringe_1 = require("tsyringe");
+const socks_proxy_agent_1 = require("socks-proxy-agent");
+const events_1 = require("events");
+const url_1 = require("url");
+const util_1 = require("util");
+let WsRequestRunner = WsRequestRunner_1 = class WsRequestRunner {
+    constructor(options, logger) {
+        this.options = options;
+        this.logger = logger;
+        this.agent = this.options.proxyUrl
+            ? new socks_proxy_agent_1.SocksProxyAgent({
+                ...(0, url_1.parse)(this.options.proxyUrl)
+            })
+            : undefined;
+    }
+    get protocol() {
+        return models_1.Protocol.WS;
+    }
+    async run(options) {
+        let timeout;
+        let client;
+        try {
+            this.logger.debug('Executing WS request with following params: %j', options);
+            client = this.createWebSocketClient(options);
+            const connectRes = await this.connect(client);
+            timeout = this.setTimeout(client);
+            const msg = await this.sendMessage(client, options);
+            return this.createWsResponse(msg, connectRes);
+        }
+        catch (err) {
+            return this.handleRequestError(err, options);
+        }
+        finally {
+            if (timeout) {
+                clearTimeout(timeout);
+            }
+            if ((client === null || client === void 0 ? void 0 : client.readyState) === ws_1.default.OPEN) {
+                client.close(1000);
+            }
+        }
+    }
+    createWebSocketClient(options) {
+        return new ws_1.default(options.url, {
+            agent: this.agent,
+            rejectUnauthorized: false,
+            timeout: this.options.timeout,
+            headers: this.normalizeHeaders(options.headers)
+        });
+    }
+    async sendMessage(client, options) {
+        // @ts-expect-error TS infers a wrong type here
+        await (0, util_1.promisify)(client.send.bind(client))(options.body);
+        const message = await this.consume(client, options.correlationIdRegex);
+        return message;
+    }
+    createWsResponse(msg, connectRes) {
+        var _a;
+        return new Response_1.Response({
+            protocol: this.protocol,
+            statusCode: (_a = msg === null || msg === void 0 ? void 0 : msg.code) !== null && _a !== void 0 ? _a : connectRes.statusCode,
+            headers: connectRes.headers,
+            body: msg === null || msg === void 0 ? void 0 : msg.body
+        });
+    }
+    handleRequestError(err, options) {
+        var _a, _b;
+        const message = (_a = err.info) !== null && _a !== void 0 ? _a : err.message;
+        const errorCode = (_b = err.code) !== null && _b !== void 0 ? _b : err.syscall;
+        this.logger.error('Error executing request: %s', options.url);
+        this.logger.error('Cause: %s', message);
+        return new Response_1.Response({
+            message,
+            errorCode,
+            protocol: this.protocol
+        });
+    }
+    setTimeout(client) {
+        const timeout = setTimeout(() => client.emit('error', Object.assign(new Error('Waiting frame has timed out'), {
+            code: 'ETIMEDOUT'
+        })), this.options.timeout);
+        timeout.unref();
+        return timeout;
+    }
+    async consume(client, matcher) {
+        const result = (await Promise.race([
+            this.waitForResponse(client, matcher),
+            (0, events_1.once)(client, 'close')
+        ]));
+        let msg;
+        if (result.length) {
+            const [data, reason] = result;
+            const body = typeof data === 'string' ? data : reason;
+            const code = typeof data === 'number' ? data : undefined;
+            msg = {
+                body,
+                code
+            };
+        }
+        return msg;
+    }
+    waitForResponse(client, matcher) {
+        return new Promise(resolve => {
+            client.on('message', (data) => {
+                const dataString = String(data);
+                !matcher || matcher.test(dataString)
+                    ? resolve([dataString])
+                    : undefined;
+            });
+        });
+    }
+    async connect(client) {
+        const opening = (0, events_1.once)(client, 'open');
+        const upgrading = (0, events_1.once)(client, 'upgrade');
+        await opening;
+        const [res] = await upgrading;
+        return res;
+    }
+    normalizeHeaders(headers) {
+        return Object.entries(headers).reduce((result, [key, value]) => {
+            const headerName = key.trim().toLowerCase();
+            if (!WsRequestRunner_1.FORBIDDEN_HEADERS.has(headerName)) {
+                result[key] = value;
+            }
+            return result;
+        }, {});
+    }
+};
+WsRequestRunner.FORBIDDEN_HEADERS = new Set([
+    'sec-websocket-version',
+    'sec-websocket-key'
+]);
+WsRequestRunner = WsRequestRunner_1 = (0, tslib_1.__decorate)([
+    (0, tsyringe_1.injectable)(),
+    (0, tslib_1.__param)(0, (0, tsyringe_1.inject)(RequestRunnerOptions_1.RequestRunnerOptions)),
+    (0, tslib_1.__metadata)("design:paramtypes", [Object, core_1.Logger])
+], WsRequestRunner);
+exports.WsRequestRunner = WsRequestRunner;
+//# sourceMappingURL=WsRequestRunner.js.map

package/src/request-runner/protocols/WsRequestRunner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WsRequestRunner.js","sourceRoot":"","sources":["../../../../../../packages/repeater/src/request-runner/protocols/WsRequestRunner.ts"],"names":[],"mappings":";;;;;AAEA,kEAA+D;AAC/D,0CAAuC;AACvC,yCAAwC;AACxC,0CAAyC;AACzC,yDAA2B;AAC3B,uCAA8C;AAC9C,yDAAoD;AACpD,mCAA8B;AAE9B,6BAA4B;AAC5B,+BAAiC;AAQjC,IAAa,eAAe,uBAA5B,MAAa,eAAe;IAQ1B,YAEmB,OAA6B,EAC7B,MAAc;QADd,YAAO,GAAP,OAAO,CAAsB;QAC7B,WAAM,GAAN,MAAM,CAAQ;QAE/B,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ;YAChC,CAAC,CAAC,IAAI,mCAAe,CAAC;gBAClB,GAAG,IAAA,WAAK,EAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC;aAChC,CAAC;YACJ,CAAC,CAAC,SAAS,CAAC;IAChB,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,iBAAQ,CAAC,EAAE,CAAC;IACrB,CAAC;IAEM,KAAK,CAAC,GAAG,CAAC,OAAgB;QAC/B,IAAI,OAAmC,CAAC;QACxC,IAAI,MAA6B,CAAC;QAElC,IAAI;YACF,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,gDAAgD,EAChD,OAAO,CACR,CAAC;YAEF,MAAM,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;YAC7C,MAAM,UAAU,GAAoB,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAE/D,OAAO,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YAClC,MAAM,GAAG,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;YAEpD,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;SAC/C;QAAC,OAAO,GAAG,EAAE;YACZ,OAAO,IAAI,CAAC,kBAAkB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;SAC9C;gBAAS;YACR,IAAI,OAAO,EAAE;gBACX,YAAY,CAAC,OAAO,CAAC,CAAC;aACvB;YAED,IAAI,CAAA,MAAM,aAAN,MAAM,uBAAN,MAAM,CAAE,UAAU,MAAK,YAAS,CAAC,IAAI,EAAE;gBACzC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;aACpB;SACF;IACH,CAAC;IAEO,qBAAqB,CAAC,OAAgB;QAC5C,OAAO,IAAI,YAAS,CAAC,OAAO,CAAC,GAAG,EAAE;YAChC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,kBAAkB,EAAE,KAAK;YACzB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,OAAO,CAAC;SAChD,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,MAAiB,EACjB,OAAgB;QAEhB,+CAA+C;QAC/C,MAAM,IAAA,gBAAS,EAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAExD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,kBAAkB,CAAC,CAAC;QAEvE,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,gBAAgB,CACtB,GAA0B,EAC1B,UAA2B;;QAE3B,OAAO,IAAI,mBAAQ,CAAC;YAClB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,UAAU,EAAE,MAAA,GAAG,aAAH,GAAG,uBAAH,GAAG,CAAE,IAAI,mCAAI,UAAU,CAAC,UAAU;YAC9C,OAAO,EAAE,UAAU,CAAC,OAAO;YAC3B,IAAI,EAAE,GAAG,aAAH,GAAG,uBAAH,GAAG,CAAE,IAAI;SAChB,CAAC,CAAC;IACL,CAAC;IAEO,kBAAkB,CAAC,GAAQ,EAAE,OAAgB;;QACnD,MAAM,OAAO,GAAG,MAAA,GAAG,CAAC,IAAI,mCAAI,GAAG,CAAC,OAAO,CAAC;QACxC,MAAM,SAAS,GAAG,MAAA,GAAG,CAAC,IAAI,mCAAI,GAAG,CAAC,OAAO,CAAC;QAE1C,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,6BAA6B,EAAE,OAAO,CAAC,GAAG,CAAC,CAAC;QAC9D,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;QAExC,OAAO,IAAI,mBAAQ,CAAC;YAClB,OAAO;YACP,SAAS;YACT,QAAQ,EAAE,IAAI,CAAC,QAAQ;SACxB,CAAC,CAAC;IACL,CAAC;IAEO,UAAU,CAAC,MAAiB;QAClC,MAAM,OAAO,GAAG,UAAU,CACxB,GAAG,EAAE,CACH,MAAM,CAAC,IAAI,CACT,OAAO,EACP,MAAM,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,6BAA6B,CAAC,EAAE;YACtD,IAAI,EAAE,WAAW;SAClB,CAAC,CACH,EACH,IAAI,CAAC,OAAO,CAAC,OAAO,CACrB,CAAC;QAEF,OAAO,CAAC,KAAK,EAAE,CAAC;QAEhB,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,OAAO,CACnB,MAAiB,EACjB,OAAgB;QAEhB,MAAM,MAAM,GAAG,CAAC,MAAM,OAAO,CAAC,IAAI,CAAC;YACjC,IAAI,CAAC,eAAe,CAAC,MAAM,EAAE,OAAO,CAAC;YACrC,IAAA,aAAI,EAAC,MAAM,EAAE,OAAO,CAAC;SACtB,CAAC,CAA0C,CAAC;QAE7C,IAAI,GAA0B,CAAC;QAE/B,IAAI,MAAM,CAAC,MAAM,EAAE;YACjB,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,GAA0C,MAAM,CAAC;YACrE,MAAM,IAAI,GAAG,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC;YACtD,MAAM,IAAI,GAAG,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;YAEzD,GAAG,GAAG;gBACJ,IAAI;gBACJ,IAAI;aACL,CAAC;SACH;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,eAAe,CACrB,MAAiB,EACjB,OAA2B;QAE3B,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE;YAC3B,MAAM,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,IAAoB,EAAE,EAAE;gBAC5C,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;gBAChC,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC;oBAClC,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC;oBACvB,CAAC,CAAC,SAAS,CAAC;YAChB,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,MAAiB;QACrC,MAAM,OAAO,GAAG,IAAA,aAAI,EAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACrC,MAAM,SAAS,GAAG,IAAA,aAAI,EAAC,MAAM,EAAE,SAAS,CAA+B,CAAC;QAExE,MAAM,OAAO,CAAC;QAEd,MAAM,CAAC,GAAG,CAAC,GAAsB,MAAM,SAAS,CAAC;QAEjD,OAAO,GAAG,CAAC;IACb,CAAC;IAEO,gBAAgB,CACtB,OAA0C;QAE1C,OAAO,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,MAAM,CACnC,CACE,MAAyC,EACzC,CAAC,GAAG,EAAE,KAAK,CAA8B,EACzC,EAAE;YACF,MAAM,UAAU,GAAG,GAAG,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YAC5C,IAAI,CAAC,iBAAe,CAAC,iBAAiB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE;gBACtD,MAAM,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;aACrB;YAED,OAAO,MAAM,CAAC;QAChB,CAAC,EACD,EAAE,CACH,CAAC;IACJ,CAAC;CACF,CAAA;AAzLwB,iCAAiB,GAAwB,IAAI,GAAG,CAAC;IACtE,uBAAuB;IACvB,mBAAmB;CACpB,CAAE,CAAA;AAJQ,eAAe;IAD3B,IAAA,qBAAU,GAAE;IAUR,wBAAA,IAAA,iBAAM,EAAC,2CAAoB,CAAC,CAAA;0DAEJ,aAAM;GAXtB,eAAe,CA0L3B;AA1LY,0CAAe"}

package/src/request-runner/protocols/index.d.ts

@@ -0,0 +1,2 @@
+export * from './HttpRequestRunner';
+export * from './WsRequestRunner';

package/src/request-runner/protocols/index.js

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const tslib_1 = require("tslib");
+(0, tslib_1.__exportStar)(require("./HttpRequestRunner"), exports);
+(0, tslib_1.__exportStar)(require("./WsRequestRunner"), exports);
+//# sourceMappingURL=index.js.map

package/src/request-runner/protocols/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../../packages/repeater/src/request-runner/protocols/index.ts"],"names":[],"mappings":";;;AAAA,mEAAoC;AACpC,iEAAkC"}

package/src/request-runner/register.d.ts

@@ -0,0 +1 @@
+export {};

package/src/request-runner/register.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const RequestRunner_1 = require("./RequestRunner");
+const protocols_1 = require("./protocols");
+const tsyringe_1 = require("tsyringe");
+tsyringe_1.container.register(RequestRunner_1.RequestRunner, {
+    useClass: protocols_1.HttpRequestRunner
+});
+tsyringe_1.container.register(RequestRunner_1.RequestRunner, {
+    useClass: protocols_1.WsRequestRunner
+});
+//# sourceMappingURL=register.js.map

package/src/request-runner/register.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"register.js","sourceRoot":"","sources":["../../../../../packages/repeater/src/request-runner/register.ts"],"names":[],"mappings":";;AAAA,mDAAgD;AAChD,2CAAiE;AACjE,uCAAqC;AAErC,oBAAS,CAAC,QAAQ,CAAC,6BAAa,EAAE;IAChC,QAAQ,EAAE,6BAAiB;CAC5B,CAAC,CAAC;AAEH,oBAAS,CAAC,QAAQ,CAAC,6BAAa,EAAE;IAChC,QAAQ,EAAE,2BAAe;CAC1B,CAAC,CAAC"}