greybel-mock-environment 1.1.9 → 1.2.1

package/dist/fs.d.ts

@@ -1,4 +1,5 @@
-import { Entity, Folder } from './types/file-system';
+import { Service } from './types';
+import { Entity, File, Folder } from './types/file-system';
 import { Port } from './types/port';
 import { User } from './types/user';
 export declare enum FSDeviceOwnerType {
@@ -24,5 +25,6 @@ export interface FSOptions {
     parent?: Entity;
     ports?: Map<number, Port>;
 }
+export declare function getLibraryFile(service: Service, parent?: Entity): File;
 export declare function getUserFolder(options: FSOptions, user: string): Folder;
 export declare function getDefaultFileSystem(options: FSOptions): Folder;

package/dist/fs.js

@@ -47,7 +47,7 @@ var __spreadArray = (this && this.__spreadArray) || function (to, from, pack) {
     return to.concat(ar || Array.prototype.slice.call(from));
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getDefaultFileSystem = exports.getUserFolder = exports.FSDeviceType = exports.FSCreationVariation = exports.FSDeviceOwnerType = void 0;
+exports.getDefaultFileSystem = exports.getUserFolder = exports.getLibraryFile = exports.FSDeviceType = exports.FSCreationVariation = exports.FSDeviceOwnerType = void 0;
 var types_1 = require("./types");
 var file_system_1 = require("./types/file-system");
 var FSDeviceOwnerType;
@@ -91,6 +91,48 @@ function getEtcAptFolder(options) {
     aptFolder.files = getEtcAptFiles(newOptions);
     return aptFolder;
 }
+function getLibraryFile(service, parent) {
+    switch (service) {
+        case types_1.Service.SSH: {
+            return new file_system_1.File({
+                name: 'ssh.so',
+                permissions: '-rw-r--r--',
+                owner: 'root',
+                type: file_system_1.FileType.SSH,
+                version: '1.0.0.0'
+            }, parent);
+        }
+        case types_1.Service.FTP: {
+            return new file_system_1.File({
+                name: 'ftp.so',
+                permissions: '-rw-r--r--',
+                owner: 'root',
+                type: file_system_1.FileType.FTP,
+                version: '1.0.0.0'
+            }, parent);
+        }
+        case types_1.Service.HTTP: {
+            return new file_system_1.File({
+                name: 'http.so',
+                permissions: '-rw-r--r--',
+                owner: 'root',
+                type: file_system_1.FileType.HTTP,
+                version: '1.0.0.0'
+            }, parent);
+        }
+        case types_1.Service.RSHELL: {
+            return new file_system_1.File({
+                name: 'rshell.so',
+                permissions: '-rw-r--r--',
+                owner: 'root',
+                type: file_system_1.FileType.RShell,
+                version: '1.0.0.0'
+            }, parent);
+        }
+    }
+    return null;
+}
+exports.getLibraryFile = getLibraryFile;
 function getEtcFiles(options) {
     return [
         new file_system_1.File({
@@ -202,47 +244,9 @@ function getLibFiles(options) {
     try {
         for (var services_1 = __values(services), services_1_1 = services_1.next(); !services_1_1.done; services_1_1 = services_1.next()) {
             var service = services_1_1.value;
-            switch (service) {
-                case types_1.Service.SSH: {
-                    libFiles.push(new file_system_1.File({
-                        name: 'ssh.so',
-                        permissions: '-rw-r--r--',
-                        owner: 'root',
-                        type: file_system_1.FileType.SSH,
-                        version: '1.0.0.0'
-                    }, options.parent));
-                    break;
-                }
-                case types_1.Service.FTP: {
-                    libFiles.push(new file_system_1.File({
-                        name: 'ftp.so',
-                        permissions: '-rw-r--r--',
-                        owner: 'root',
-                        type: file_system_1.FileType.FTP,
-                        version: '1.0.0.0'
-                    }, options.parent));
-                    break;
-                }
-                case types_1.Service.HTTP: {
-                    libFiles.push(new file_system_1.File({
-                        name: 'http.so',
-                        permissions: '-rw-r--r--',
-                        owner: 'root',
-                        type: file_system_1.FileType.HTTP,
-                        version: '1.0.0.0'
-                    }, options.parent));
-                    break;
-                }
-                case types_1.Service.RSHELL: {
-                    libFiles.push(new file_system_1.File({
-                        name: 'rshell.so',
-                        permissions: '-rw-r--r--',
-                        owner: 'root',
-                        type: file_system_1.FileType.RShell,
-                        version: '1.0.0.0'
-                    }, options.parent));
-                    break;
-                }
+            var file = getLibraryFile(service, options.parent);
+            if (file !== null) {
+                libFiles.push(file);
             }
         }
     }

package/dist/types/computer.d.ts

@@ -1,5 +1,6 @@
 import { FSDeviceType } from '../fs';
 import { Device, DeviceOptions } from './device';
+import { Port } from './port';
 import { Router } from './router';
 export interface ComputerOptions extends DeviceOptions {
     router?: Router;
@@ -9,5 +10,6 @@ export declare class Computer extends Device {
     constructor(options: ComputerOptions);
     getDeviceType(): FSDeviceType;
     getRouter(): Device;
+    addPort(port: Port): void;
     findByLanIp(ip: string): Device;
 }

package/dist/types/computer.js

@@ -31,6 +31,10 @@ var Computer = /** @class */ (function (_super) {
     Computer.prototype.getRouter = function () {
         return this.router;
     };
+    Computer.prototype.addPort = function (port) {
+        _super.prototype.addPort.call(this, port);
+        this.router.ports.set(port.port, port);
+    };
     Computer.prototype.findByLanIp = function (ip) {
         return this.getRouter().findByLanIp(ip);
     };

package/dist/types/device.d.ts

@@ -31,6 +31,7 @@ export declare class Device {
     changePassword(username: string, password: string): boolean;
     getUserByVulnerability(vulActionUser: VulnerabilityActionUser): User;
     findPort(portNumber: number): Port;
+    addPort(port: Port): void;
     findLibraryFileByPort(port: Port): File;
     getKernel(): File;
 }

package/dist/types/device.js

@@ -113,6 +113,16 @@ var Device = /** @class */ (function () {
         }
         return null;
     };
+    Device.prototype.addPort = function (port) {
+        this.ports.set(port.port, port);
+        var addedPort = this.ports.get(port.port);
+        var libFolder = this.getFile(['lib']);
+        if (!(libFolder instanceof file_system_1.Folder)) {
+            return;
+        }
+        var libaryFile = (0, fs_1.getLibraryFile)(addedPort.service, libFolder);
+        libFolder.putFile(libaryFile);
+    };
     Device.prototype.findLibraryFileByPort = function (port) {
         var _this = this;
         var get = function (fileName, type) {

package/dist/types/router.d.ts

@@ -1,6 +1,7 @@
 import { FSDeviceType } from '../fs';
 import { Computer } from './computer';
 import { Device, DeviceOptions } from './device';
+import { Port } from './port';
 export interface RouterOptions extends DeviceOptions {
     bssid: string;
     essid: string;
@@ -20,4 +21,5 @@ export declare class Router extends Device {
     getRouter(): Device;
     findByLanIp(ip: string): Device;
     getComputers(): Computer[];
+    findDeviceByPort(port: Port): Device;
 }

package/dist/types/router.js

@@ -89,6 +89,26 @@ var Router = /** @class */ (function (_super) {
             return item instanceof computer_1.Computer;
         });
     };
+    Router.prototype.findDeviceByPort = function (port) {
+        var e_2, _a;
+        try {
+            for (var _b = __values(this.devices.values()), _c = _b.next(); !_c.done; _c = _b.next()) {
+                var device = _c.value;
+                var p = device.findPort(port.port);
+                if (p !== null) {
+                    return device;
+                }
+            }
+        }
+        catch (e_2_1) { e_2 = { error: e_2_1 }; }
+        finally {
+            try {
+                if (_c && !_c.done && (_a = _b.return)) _a.call(_b);
+            }
+            finally { if (e_2) throw e_2.error; }
+        }
+        return null;
+    };
     return Router;
 }(device_1.Device));
 exports.Router = Router;

package/dist/types/switch.d.ts

@@ -1,5 +1,6 @@
 import { FSDeviceType } from '../fs';
 import { Device } from './device';
+import { Port } from './port';
 import { Router, RouterOptions } from './router';
 export interface SwitchOptions extends RouterOptions {
     router: Router;
@@ -7,6 +8,7 @@ export interface SwitchOptions extends RouterOptions {
 export declare class Switch extends Router {
     router?: Router;
     constructor(options: SwitchOptions);
+    addPort(port: Port): void;
     getDeviceType(): FSDeviceType;
     getRouter(): Device;
 }

package/dist/types/switch.js

@@ -25,6 +25,10 @@ var Switch = /** @class */ (function (_super) {
         _this.router = options.router;
         return _this;
     }
+    Switch.prototype.addPort = function (port) {
+        _super.prototype.addPort.call(this, port);
+        this.router.ports.set(port.port, port);
+    };
     Switch.prototype.getDeviceType = function () {
         return fs_1.FSDeviceType.Switch;
     };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "greybel-mock-environment",
-  "version": "1.1.9",
+  "version": "1.2.1",
   "description": "Mock environment",
   "main": "dist/index",
   "typings": "dist/index",