greybel-mock-environment 1.2.2 → 1.2.4

package/dist/environment.d.ts

@@ -3,8 +3,14 @@ import EmailGenerator from './generators/email';
 import NetworkGenerator from './generators/network';
 import UserGenerator from './generators/user';
 import VulnerabilityGenerator from './generators/vulnerability';
-import { Computer, EMail, Port, Router, Session, Switch } from './types';
+import { Computer, Device, EMail, Location, Port, Router, Session, Switch } from './types';
+export interface RouterLocation {
+    distance: number;
+    percentage: number;
+    router: Router;
+}
 export default class MockEnvironment {
+    readonly WIFI_MAX_DISTANCE: number;
     seed: string;
     basicGenerator: BasicGenerator;
     emailGenerator: EmailGenerator;
@@ -29,7 +35,9 @@ export default class MockEnvironment {
     setupLocalSession(username: string, password: string): void;
     getLatestSession(): Session;
     getLocal(): Session;
+    connect(router: Router, device: Device): void;
     connectLocal(router: Router): void;
+    findRoutersCloseToLocation(loc: Location): RouterLocation[];
     getRouterByIp(ipAddress: string): Router | null;
     getSwitchByIp(ipAddress: string): Switch | null;
     findRouterViaNS(name: string): Router;

package/dist/environment.js

@@ -40,6 +40,7 @@ var utils_1 = require("./utils");
 var MockEnvironment = /** @class */ (function () {
     function MockEnvironment(seed, localUser) {
         if (seed === void 0) { seed = 'test'; }
+        this.WIFI_MAX_DISTANCE = 10;
         var me = this;
         me.seed = seed;
         me.basicGenerator = new basic_1.default({ seed: seed });
@@ -164,21 +165,58 @@ var MockEnvironment = /** @class */ (function () {
     MockEnvironment.prototype.getLocal = function () {
         return this.localSession;
     };
-    MockEnvironment.prototype.connectLocal = function (router) {
+    MockEnvironment.prototype.connect = function (router, device) {
+        if (!(device instanceof types_1.Computer) && !(device instanceof types_1.Switch)) {
+            throw new Error('Cannot connect router to router.');
+        }
+        if (device.router) {
+            device.router.devices.delete(device.localIp);
+        }
         var me = this;
         var serverMap = me.networkGenerator.serverMap;
         if (!serverMap.has(router.publicIp)) {
             throw new Error("Router with ip ".concat(router.publicIp, " is unknown in serverMap."));
         }
-        var computer = me.localSession.computer;
-        if (computer.router) {
-            computer.router.devices.delete(computer.localIp);
-        }
         var serverMapGenerator = serverMap.get(router.publicIp);
         var localIp = serverMapGenerator.generateLocalIp(router);
-        computer.router = router;
-        computer.localIp = localIp;
-        router.devices.set(localIp, computer);
+        device.router = router;
+        device.localIp = localIp;
+        router.devices.set(localIp, device);
+    };
+    MockEnvironment.prototype.connectLocal = function (router) {
+        this.connect(router, this.localSession.computer);
+    };
+    MockEnvironment.prototype.findRoutersCloseToLocation = function (loc) {
+        var e_1, _a;
+        var me = this;
+        var getDistance = function (routerLoc) {
+            var y = routerLoc.x - loc.x;
+            var x = routerLoc.y - loc.y;
+            return Math.sqrt(x * x + y * y);
+        };
+        var result = [];
+        try {
+            for (var _b = __values(me.networkGenerator.serverMap.values()), _c = _b.next(); !_c.done; _c = _b.next()) {
+                var serverMapGenerator = _c.value;
+                var distance = getDistance(serverMapGenerator.router.location);
+                if (distance < me.WIFI_MAX_DISTANCE) {
+                    var percentage = Math.round(((me.WIFI_MAX_DISTANCE - distance) * 100) / me.WIFI_MAX_DISTANCE);
+                    result.push({
+                        percentage: percentage,
+                        distance: distance,
+                        router: serverMapGenerator.router
+                    });
+                }
+            }
+        }
+        catch (e_1_1) { e_1 = { error: e_1_1 }; }
+        finally {
+            try {
+                if (_c && !_c.done && (_a = _b.return)) _a.call(_b);
+            }
+            finally { if (e_1) throw e_1.error; }
+        }
+        return result;
     };
     MockEnvironment.prototype.getRouterByIp = function (ipAddress) {
         var me = this;
@@ -194,7 +232,7 @@ var MockEnvironment = /** @class */ (function () {
         });
     };
     MockEnvironment.prototype.getSwitchByIp = function (ipAddress) {
-        var e_1, _a;
+        var e_2, _a;
         var me = this;
         if (!(0, utils_1.isValidIp)(ipAddress) || !(0, utils_1.isLanIp)(ipAddress)) {
             return null;
@@ -208,12 +246,12 @@ var MockEnvironment = /** @class */ (function () {
                 }
             }
         }
-        catch (e_1_1) { e_1 = { error: e_1_1 }; }
+        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_1) throw e_1.error; }
+            finally { if (e_2) throw e_2.error; }
         }
         return null;
     };

package/dist/generators/basic.d.ts

@@ -1,4 +1,5 @@
 import { RandomSeed } from 'random-seed';
+import { Location } from '../types/location';
 export interface BasicGeneratorOptions {
     seed: string;
 }
@@ -9,6 +10,7 @@ export default class BasicGenerator {
     bankIdRng: RandomSeed;
     macAddressRng: RandomSeed;
     uuidRng: RandomSeed;
+    locationRng: RandomSeed;
     constructor({ seed }: BasicGeneratorOptions);
     generateDomain(): string;
     generateUsername(): string;
@@ -16,4 +18,5 @@ export default class BasicGenerator {
     generateBankNumber(): string;
     generateMAC(): string;
     generateUUID(): string;
+    generateLocation(): Location;
 }

package/dist/generators/basic.js

@@ -29,6 +29,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 var random_seed_1 = __importDefault(require("random-seed"));
+var location_1 = require("../types/location");
 var randomUsernameGenerator = require('random-username-generator');
 var BasicGenerator = /** @class */ (function () {
     function BasicGenerator(_a) {
@@ -39,6 +40,7 @@ var BasicGenerator = /** @class */ (function () {
         this.tldRng = random_seed_1.default.create("".concat(seed, "-tld"));
         this.bankIdRng = random_seed_1.default.create("".concat(seed, "-bank-number"));
         this.uuidRng = random_seed_1.default.create("".concat(seed, "-uuid"));
+        this.locationRng = random_seed_1.default.create("".concat(seed, "-location"));
     }
     BasicGenerator.prototype.generateDomain = function () {
         var me = this;
@@ -78,6 +80,12 @@ var BasicGenerator = /** @class */ (function () {
             return (c === 'x' ? r : (r & 0x3) | 0x8).toString(16);
         });
     };
+    BasicGenerator.prototype.generateLocation = function () {
+        return new location_1.Location({
+            x: this.locationRng.floatBetween(0, 50),
+            y: this.locationRng.floatBetween(0, 50)
+        });
+    };
     return BasicGenerator;
 }());
 exports.default = BasicGenerator;

package/dist/generators/network.d.ts

@@ -1,6 +1,6 @@
 import { RandomSeed } from 'random-seed';
 import { FSCreationVariation } from '../fs';
-import { Computer, NetworkDevice, Port, Router, Switch, User, WifiNetwork } from '../types';
+import { Computer, NetworkDevice, Port, Router, Switch, User } from '../types';
 import BasicGenerator from './basic';
 import EmailGenerator from './email';
 import UserGenerator from './user';
@@ -39,12 +39,10 @@ export default class NetworkGenerator {
     networkRng: RandomSeed;
     portRng: RandomSeed;
     serverMap: Map<string, ServerMapGenerator>;
-    wifiNetworks: WifiNetwork[];
     constructor({ seed, basicGenerator, userGenerator, emailGenerator }: NetworkGeneratorOptions);
     generateIp(): string;
     generateDifficulty(): FSCreationVariation;
     generateLocalComputer(users: User[], rootPassword?: string): Computer;
     generateRouter(options?: Partial<Router>): Router;
     generateNetworkDevice(): NetworkDevice;
-    generateWifiNetwork(router: Router, options?: Partial<WifiNetwork>): WifiNetwork;
 }

package/dist/generators/network.js

@@ -32,6 +32,7 @@ exports.ServerMapGenerator = void 0;
 var random_seed_1 = __importDefault(require("random-seed"));
 var fs_1 = require("../fs");
 var types_1 = require("../types");
+var net_card_1 = require("../types/net-card");
 var ServerMapGenerator = /** @class */ (function () {
     function ServerMapGenerator(_a) {
         var router = _a.router, networkRng = _a.networkRng, portRng = _a.portRng, basicGenerator = _a.basicGenerator, userGenerator = _a.userGenerator, networkGenerator = _a.networkGenerator;
@@ -78,7 +79,13 @@ var ServerMapGenerator = /** @class */ (function () {
                     ownerType: fs_1.FSDeviceOwnerType.Npc
                 }),
             bssid: options.bssid || basicGenerator.generateUUID(),
-            essid: options.essid || basicGenerator.generateUUID()
+            essid: options.essid || basicGenerator.generateUUID(),
+            mac: options.mac || basicGenerator.generateMAC(),
+            location: options.location || basicGenerator.generateLocation(),
+            wifi: {
+                credentials: switchUsers[1],
+                name: basicGenerator.generateUsername()
+            }
         });
         for (var index = me.networkRng.intBetween(2, 4); index >= 0; index--) {
             me.generateComputer(subSwitch, [
@@ -112,7 +119,9 @@ var ServerMapGenerator = /** @class */ (function () {
                 type: fs_1.FSDeviceType.Computer,
                 difficulty: networkGenerator.generateDifficulty(),
                 ports: ports
-            })
+            }),
+            mac: basicGenerator.generateMAC(),
+            location: basicGenerator.generateLocation()
         });
         router.devices.set(computer.localIp, computer);
         return computer;
@@ -151,7 +160,6 @@ var NetworkGenerator = /** @class */ (function () {
         this.networkRng = random_seed_1.default.create("".concat(seed, "-network"));
         this.portRng = random_seed_1.default.create("".concat(seed, "-port"));
         this.serverMap = new Map();
-        this.wifiNetworks = [];
     }
     NetworkGenerator.prototype.generateIp = function () {
         var me = this;
@@ -186,7 +194,9 @@ var NetworkGenerator = /** @class */ (function () {
                 users: computerUsers,
                 ownerType: fs_1.FSDeviceOwnerType.Player,
                 type: fs_1.FSDeviceType.Computer
-            })
+            }),
+            mac: basicGenerator.generateMAC(),
+            location: basicGenerator.generateLocation()
         });
         return computer;
     };
@@ -226,7 +236,13 @@ var NetworkGenerator = /** @class */ (function () {
                     difficulty: me.generateDifficulty()
                 }),
             bssid: options.bssid || basicGenerator.generateUUID(),
-            essid: options.essid || basicGenerator.generateUUID()
+            essid: options.essid || basicGenerator.generateUUID(),
+            mac: options.mac || basicGenerator.generateMAC(),
+            location: options.location || basicGenerator.generateLocation(),
+            wifi: {
+                credentials: routerUsers[1],
+                name: basicGenerator.generateUsername()
+            }
         });
         var serverMapGenerator = new ServerMapGenerator({
             router: router,
@@ -249,25 +265,13 @@ var NetworkGenerator = /** @class */ (function () {
     };
     NetworkGenerator.prototype.generateNetworkDevice = function () {
         return {
-            type: this.networkRng.intBetween(0, 1) === 0 ? 'wlan0' : 'eth0',
+            type: this.networkRng.intBetween(0, 1) === 0
+                ? net_card_1.NetCard.Wifi
+                : net_card_1.NetCard.Ethernet,
             id: this.networkRng.random().toString(36).substring(2, 10),
             active: true
         };
     };
-    NetworkGenerator.prototype.generateWifiNetwork = function (router, options) {
-        if (options === void 0) { options = {}; }
-        var me = this;
-        var basicGenerator = me.basicGenerator;
-        var wifiNetwork = new types_1.WifiNetwork({
-            mac: options.mac || basicGenerator.generateMAC(),
-            percentage: options.percentage || me.networkRng.intBetween(20, 70),
-            name: options.name || basicGenerator.generateUsername(),
-            password: options.password || basicGenerator.generatePassword(),
-            router: router
-        });
-        me.wifiNetworks.push(wifiNetwork);
-        return wifiNetwork;
-    };
     return NetworkGenerator;
 }());
 exports.default = NetworkGenerator;

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-export { default as MockEnvironment } from './environment';
+export { default as MockEnvironment, RouterLocation } from './environment';
 export * as FS from './fs';
 export * as Generators from './generators';
 export { default as MockTranspiler } from './mock-transpiler';

package/dist/types/device.d.ts

@@ -1,27 +1,47 @@
 import { FSDeviceOwnerType, FSDeviceType } from '../fs';
 import { Entity, File, Folder } from './file-system';
+import { Location } from './location';
+import { NetCard } from './net-card';
 import { NetworkDevice } from './network-device';
 import { Port } from './port';
+import { Process, ProcessOptions } from './process';
 import { User } from './user';
 import { VulnerabilityActionUser } from './vulnerabilities';
 export interface DeviceOptions {
     localIp: string;
-    activeNetCard: string;
+    activeNetCard: NetCard;
+    location: Location;
     networkDevices?: NetworkDevice[];
     users?: User[];
     fileSystem: Folder;
     ports?: Map<number, Port>;
     ownerType?: FSDeviceOwnerType;
+    mac: string;
 }
 export declare class Device {
     localIp: string;
-    activeNetCard: string;
+    activeNetCard: NetCard;
+    location: Location;
     networkDevices: NetworkDevice[];
     users: User[];
     fileSystem: Folder;
     ports: Map<number, Port>;
     ownerType: FSDeviceOwnerType;
+    processes: Map<number, Process>;
+    groups: Map<string, Set<string>>;
+    mac: string;
     constructor(options: DeviceOptions);
+    initialize(): void;
+    getNetworkDeviceMap(): Map<string, NetworkDevice>;
+    findUser(userName: string): User;
+    addUser(userName: string, password: string): boolean;
+    removeUser(userName: string): boolean;
+    updatePasswd(): boolean;
+    createUserFolder(userName: string): boolean;
+    addProcess(options: Omit<ProcessOptions, 'pid'>): Device;
+    removeProcess(pid: number): boolean;
+    addGroup(userName: string, groupName: string): boolean;
+    removeGroup(userName: string, groupName: string): boolean;
     getDeviceType(): FSDeviceType;
     findByLanIp(_ip: string): Device;
     getRouter(): Device;

package/dist/types/device.js

@@ -31,19 +31,166 @@ exports.Device = void 0;
 var fs_1 = require("../fs");
 var utils_1 = require("../utils");
 var file_system_1 = require("./file-system");
+var net_card_1 = require("./net-card");
+var process_1 = require("./process");
 var service_1 = require("./service");
 var user_1 = require("./user");
 var vulnerabilities_1 = require("./vulnerabilities");
 var Device = /** @class */ (function () {
     function Device(options) {
         this.localIp = options.localIp;
-        this.activeNetCard = options.activeNetCard;
+        this.activeNetCard = options.activeNetCard || net_card_1.NetCard.None;
+        this.location = options.location;
         this.networkDevices = options.networkDevices || [];
         this.users = options.users || [];
         this.fileSystem = options.fileSystem;
         this.ports = options.ports || new Map();
         this.ownerType = options.ownerType || fs_1.FSDeviceOwnerType.Npc;
+        this.mac = options.mac;
+        this.processes = new Map();
+        this.groups = new Map();
+        this.initialize();
     }
+    Device.prototype.initialize = function () {
+        this.addProcess({
+            owner: this.findUser('root'),
+            command: 'kernel_task',
+            protected: true
+        }).addProcess({
+            owner: this.findUser('root'),
+            command: 'Xorg',
+            protected: true
+        });
+    };
+    Device.prototype.getNetworkDeviceMap = function () {
+        var e_1, _a;
+        var result = new Map();
+        var wifiIndex = 0;
+        var ethIndex = 0;
+        try {
+            for (var _b = __values(this.networkDevices), _c = _b.next(); !_c.done; _c = _b.next()) {
+                var networkDevice = _c.value;
+                var type = void 0;
+                if (networkDevice.type === net_card_1.NetCard.Wifi) {
+                    type = "wlan".concat(wifiIndex++);
+                }
+                else {
+                    type = "eth".concat(ethIndex++);
+                }
+                result.set(type, networkDevice);
+            }
+        }
+        catch (e_1_1) { e_1 = { error: e_1_1 }; }
+        finally {
+            try {
+                if (_c && !_c.done && (_a = _b.return)) _a.call(_b);
+            }
+            finally { if (e_1) throw e_1.error; }
+        }
+        return result;
+    };
+    Device.prototype.findUser = function (userName) {
+        return this.users.find(function (u) { return u.username === userName; }) || null;
+    };
+    Device.prototype.addUser = function (userName, password) {
+        var existing = this.findUser(userName);
+        if (existing !== null) {
+            return false;
+        }
+        var user = new user_1.User({
+            username: userName,
+            password: password
+        });
+        this.users.push(user);
+        return true;
+    };
+    Device.prototype.removeUser = function (userName) {
+        var index = this.users.findIndex(function (u) { return u.username === userName; });
+        if (index === -1) {
+            return false;
+        }
+        this.users.splice(index, 1);
+        return true;
+    };
+    Device.prototype.updatePasswd = function () {
+        var etcFolder = this.getFile(['etc']);
+        if (etcFolder instanceof file_system_1.Folder) {
+            var file = new file_system_1.File({
+                name: 'passwd',
+                permissions: '-rw-r-----',
+                owner: 'root',
+                content: this.users
+                    .map(function (v) { return "".concat(v.username, ":").concat(v.passwordHashed); })
+                    .join('\n'),
+                type: file_system_1.FileType.Plain
+            });
+            etcFolder.putFile(file);
+            return true;
+        }
+        return false;
+    };
+    Device.prototype.createUserFolder = function (userName) {
+        var homeFolder = this.getFile(['home']);
+        if (homeFolder instanceof file_system_1.Folder) {
+            var userFolder = (0, fs_1.getUserFolder)({
+                parent: homeFolder,
+                users: this.users,
+                type: this.getDeviceType(),
+                ownerType: fs_1.FSDeviceOwnerType.Player
+            }, userName);
+            homeFolder.putFolder(userFolder);
+            return true;
+        }
+        return false;
+    };
+    Device.prototype.addProcess = function (options) {
+        var pid = (0, utils_1.getRandomInt)(10, 9000);
+        var process = new process_1.Process({
+            owner: options.owner,
+            pid: pid,
+            command: options.command,
+            protected: options.protected
+        });
+        this.processes.set(pid, process);
+        return this;
+    };
+    Device.prototype.removeProcess = function (pid) {
+        if (this.processes.has(pid)) {
+            var process_2 = this.processes.get(pid);
+            if (!process_2.protected) {
+                this.processes.delete(pid);
+                return true;
+            }
+        }
+        return false;
+    };
+    Device.prototype.addGroup = function (userName, groupName) {
+        var user = this.findUser(userName);
+        if (user === null) {
+            return false;
+        }
+        if (!this.groups.has(groupName)) {
+            this.groups.set(groupName, new Set());
+        }
+        var groups = this.groups.get(groupName);
+        groups.add(user.username);
+        return true;
+    };
+    Device.prototype.removeGroup = function (userName, groupName) {
+        var user = this.findUser(userName);
+        if (user === null) {
+            return false;
+        }
+        if (!this.groups.has(groupName)) {
+            return false;
+        }
+        var groups = this.groups.get(groupName);
+        groups.delete(user.username);
+        if (groups.size === 0) {
+            this.groups.delete(groupName);
+        }
+        return true;
+    };
     Device.prototype.getDeviceType = function () {
         return fs_1.FSDeviceType.Unknown;
     };
@@ -96,7 +243,7 @@ var Device = /** @class */ (function () {
         });
     };
     Device.prototype.findPort = function (portNumber) {
-        var e_1, _a;
+        var e_2, _a;
         try {
             for (var _b = __values(this.ports), _c = _b.next(); !_c.done; _c = _b.next()) {
                 var _d = __read(_c.value, 2), pn = _d[0], port = _d[1];
@@ -104,12 +251,12 @@ var Device = /** @class */ (function () {
                     return port;
             }
         }
-        catch (e_1_1) { e_1 = { error: e_1_1 }; }
+        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_1) throw e_1.error; }
+            finally { if (e_2) throw e_2.error; }
         }
         return null;
     };

package/dist/types/file-system.d.ts

@@ -2,6 +2,7 @@ import { PermissionMap, PermissionSegment } from '../utils';
 import { Library } from './library';
 import { User } from './user';
 import { Version } from './version';
+export declare const FILE_CONTENT_LIMIT = 160000;
 export declare enum FileType {
     Plain = 0,
     Bin = 1,
@@ -25,6 +26,7 @@ export declare enum FileType {
 }
 export interface EntityOptions {
     name: string;
+    group?: string;
     permissions: string;
     owner: string;
     isFolder?: boolean;
@@ -36,15 +38,17 @@ export declare class Entity {
     name: string;
     permissions: string;
     owner: string;
+    group: string;
     isFolder?: boolean;
     isProtected?: boolean;
     deleted?: boolean;
     constructor(options: EntityOptions, parent?: Entity);
+    getSize(): number;
     getPath(): string[] | null;
     getEntity(_path: string[]): Entity | null;
-    copyEntity(toFolder: Entity): Entity;
+    copy(): Entity;
     parsePermissions(): PermissionMap;
-    getPermissions(user: User): PermissionSegment;
+    getPermissions(user: User, groups: Map<string, Set<string>>): PermissionSegment;
 }
 export interface FileOptions extends Omit<EntityOptions, 'isFolder'> {
     content?: string;
@@ -58,7 +62,8 @@ export declare class File extends Entity {
     allowImport: boolean;
     version: Version;
     constructor(options: FileOptions, parent?: Entity);
-    copyEntity(toFolder: Entity): File;
+    getSize(): number;
+    copy(): File;
     getLibraryType(): Library | null;
 }
 export interface FolderOptions extends Omit<EntityOptions, 'isFolder'> {
@@ -76,10 +81,11 @@ export declare class Folder extends Entity {
     getEntity(path: string[]): Entity | null;
     getEntityFileIndex(entityName: string): FolderIndexResult;
     getEntityFolderIndex(entityName: string): FolderIndexResult;
+    copy(recursive?: boolean): Folder;
     getEntityIndex(entityName: string): FolderIndexResult;
     hasFile(entityName: string): boolean;
     putFile(file: File): File;
-    removeFile(entityName: string): boolean;
-    copyEntity(toFolder: Entity): Folder;
-    traverseChildren(callback: (v: Entity) => void, skip?: boolean): void;
+    putFolder(folder: Folder): Folder;
+    removeEntity(entityName: string): boolean;
+    traverseChildren(callback: (v: Entity) => void): void;
 }

package/dist/types/file-system.js

@@ -42,9 +42,10 @@ var __read = (this && this.__read) || function (o, n) {
     return ar;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.Folder = exports.File = exports.Entity = exports.FileType = void 0;
+exports.Folder = exports.File = exports.Entity = exports.FileType = exports.FILE_CONTENT_LIMIT = void 0;
 var library_1 = require("./library");
 var version_1 = require("./version");
+exports.FILE_CONTENT_LIMIT = 160000;
 var FileType;
 (function (FileType) {
     FileType[FileType["Plain"] = 0] = "Plain";
@@ -73,10 +74,14 @@ var Entity = /** @class */ (function () {
         this.name = options.name;
         this.permissions = options.permissions;
         this.owner = options.owner;
+        this.group = options.group || options.owner;
         this.isFolder = options.isFolder;
         this.isProtected = options.isProtected;
         this.deleted = options.deleted;
     }
+    Entity.prototype.getSize = function () {
+        return 0;
+    };
     Entity.prototype.getPath = function () {
         var path = [this.name];
         var current = this.parent;
@@ -89,13 +94,13 @@ var Entity = /** @class */ (function () {
     Entity.prototype.getEntity = function (_path) {
         return null;
     };
-    Entity.prototype.copyEntity = function (toFolder) {
+    Entity.prototype.copy = function () {
         var newEntity = new Entity({
             name: this.name,
             permissions: this.permissions,
             owner: this.owner,
             isFolder: this.isFolder
-        }, toFolder);
+        });
         return newEntity;
     };
     Entity.prototype.parsePermissions = function () {
@@ -122,13 +127,22 @@ var Entity = /** @class */ (function () {
             }
         };
     };
-    Entity.prototype.getPermissions = function (user) {
-        // g is ignored for now
-        // todo: add group logic
+    Entity.prototype.getPermissions = function (user, groups) {
+        if (user.username === 'root') {
+            return {
+                r: true,
+                w: true,
+                x: true
+            };
+        }
         var parsedPermissions = this.parsePermissions();
         if (this.owner === user.username) {
             return parsedPermissions.u;
         }
+        else if (groups.has(this.owner) &&
+            groups.get(this.owner).has(user.username)) {
+            return parsedPermissions.g;
+        }
         return parsedPermissions.o;
     };
     return Entity;
@@ -144,15 +158,20 @@ var File = /** @class */ (function (_super) {
         _this.version = new version_1.Version({ version: options.version });
         return _this;
     }
-    File.prototype.copyEntity = function (toFolder) {
-        var newFile = new File({
+    File.prototype.getSize = function () {
+        return Math.round((100 * this.content.length) / exports.FILE_CONTENT_LIMIT) * 100;
+    };
+    File.prototype.copy = function () {
+        var newEntity = new File({
             name: this.name,
             permissions: this.permissions,
             owner: this.owner,
             content: this.content,
-            type: this.type
-        }, toFolder);
-        return newFile;
+            type: this.type,
+            allowImport: this.allowImport,
+            version: this.version.toString()
+        });
+        return newEntity;
     };
     File.prototype.getLibraryType = function () {
         switch (this.type) {
@@ -222,6 +241,16 @@ var Folder = /** @class */ (function (_super) {
         }
         return null;
     };
+    Folder.prototype.copy = function (recursive) {
+        var newEntity = new Folder({
+            name: this.name,
+            permissions: this.permissions,
+            owner: this.owner,
+            folders: recursive ? this.folders.map(function (f) { return f.copy(); }) : [],
+            files: recursive ? this.files.map(function (f) { return f.copy(); }) : []
+        });
+        return newEntity;
+    };
     Folder.prototype.getEntityIndex = function (entityName) {
         return (this.getEntityFileIndex(entityName) ||
             this.getEntityFolderIndex(entityName));
@@ -230,12 +259,18 @@ var Folder = /** @class */ (function (_super) {
         return !!this.getEntityIndex(entityName);
     };
     Folder.prototype.putFile = function (file) {
-        this.removeFile(file.name);
+        this.removeEntity(file.name);
         file.parent = this;
         this.files.push(file);
         return file;
     };
-    Folder.prototype.removeFile = function (entityName) {
+    Folder.prototype.putFolder = function (folder) {
+        this.removeEntity(folder.name);
+        folder.parent = this;
+        this.folders.push(folder);
+        return folder;
+    };
+    Folder.prototype.removeEntity = function (entityName) {
         var result = this.getEntityIndex(entityName);
         if (!result) {
             return false;
@@ -256,20 +291,8 @@ var Folder = /** @class */ (function (_super) {
         }
         return true;
     };
-    Folder.prototype.copyEntity = function (toFolder) {
-        var newFolder = new Folder({
-            name: this.name,
-            permissions: this.permissions,
-            owner: this.owner
-        }, toFolder);
-        this.files = this.files.map(function (item) { return item.copyEntity(newFolder); });
-        this.folders = this.folders.map(function (item) { return item.copyEntity(newFolder); });
-        return newFolder;
-    };
-    Folder.prototype.traverseChildren = function (callback, skip) {
-        if (!skip) {
-            callback(this);
-        }
+    Folder.prototype.traverseChildren = function (callback) {
+        callback(this);
         this.files.forEach(function (item) { return callback(item); });
         this.folders.forEach(function (item) { return item.traverseChildren(callback); });
     };

package/dist/types/index.d.ts

@@ -3,12 +3,14 @@ export { Device } from './device';
 export { EMail, EMailMessage } from './email';
 export { File, FileType, Folder, Entity as FSEntity } from './file-system';
 export { Library } from './library';
+export { Location } from './location';
+export { NetCard } from './net-card';
 export { NetworkDevice } from './network-device';
 export { Port } from './port';
+export { Process } from './process';
 export { Router } from './router';
 export { Service, ServiceList } from './service';
 export { Session } from './session';
 export { Switch } from './switch';
 export { User } from './user';
 export { Vulnerability, VulnerabilityAction, VulnerabilityActionList, VulnerabilityActionUser, VulnerabilityRequirementList, VulnerabilityRequirements } from './vulnerabilities';
-export { WifiNetwork } from './wifi-network';

package/dist/types/index.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.WifiNetwork = exports.VulnerabilityRequirements = exports.VulnerabilityRequirementList = exports.VulnerabilityActionUser = exports.VulnerabilityActionList = exports.VulnerabilityAction = exports.Vulnerability = exports.User = exports.Switch = exports.Session = exports.ServiceList = exports.Service = exports.Router = exports.Port = exports.NetworkDevice = exports.Library = exports.FSEntity = exports.Folder = exports.FileType = exports.File = exports.EMailMessage = exports.EMail = exports.Device = exports.Computer = void 0;
+exports.VulnerabilityRequirements = exports.VulnerabilityRequirementList = exports.VulnerabilityActionUser = exports.VulnerabilityActionList = exports.VulnerabilityAction = exports.Vulnerability = exports.User = exports.Switch = exports.Session = exports.ServiceList = exports.Service = exports.Router = exports.Process = exports.Port = exports.NetworkDevice = exports.NetCard = exports.Location = exports.Library = exports.FSEntity = exports.Folder = exports.FileType = exports.File = exports.EMailMessage = exports.EMail = exports.Device = exports.Computer = void 0;
 var computer_1 = require("./computer");
 Object.defineProperty(exports, "Computer", { enumerable: true, get: function () { return computer_1.Computer; } });
 var device_1 = require("./device");
@@ -15,10 +15,16 @@ Object.defineProperty(exports, "Folder", { enumerable: true, get: function () {
 Object.defineProperty(exports, "FSEntity", { enumerable: true, get: function () { return file_system_1.Entity; } });
 var library_1 = require("./library");
 Object.defineProperty(exports, "Library", { enumerable: true, get: function () { return library_1.Library; } });
+var location_1 = require("./location");
+Object.defineProperty(exports, "Location", { enumerable: true, get: function () { return location_1.Location; } });
+var net_card_1 = require("./net-card");
+Object.defineProperty(exports, "NetCard", { enumerable: true, get: function () { return net_card_1.NetCard; } });
 var network_device_1 = require("./network-device");
 Object.defineProperty(exports, "NetworkDevice", { enumerable: true, get: function () { return network_device_1.NetworkDevice; } });
 var port_1 = require("./port");
 Object.defineProperty(exports, "Port", { enumerable: true, get: function () { return port_1.Port; } });
+var process_1 = require("./process");
+Object.defineProperty(exports, "Process", { enumerable: true, get: function () { return process_1.Process; } });
 var router_1 = require("./router");
 Object.defineProperty(exports, "Router", { enumerable: true, get: function () { return router_1.Router; } });
 var service_1 = require("./service");
@@ -37,5 +43,3 @@ Object.defineProperty(exports, "VulnerabilityActionList", { enumerable: true, ge
 Object.defineProperty(exports, "VulnerabilityActionUser", { enumerable: true, get: function () { return vulnerabilities_1.VulnerabilityActionUser; } });
 Object.defineProperty(exports, "VulnerabilityRequirementList", { enumerable: true, get: function () { return vulnerabilities_1.VulnerabilityRequirementList; } });
 Object.defineProperty(exports, "VulnerabilityRequirements", { enumerable: true, get: function () { return vulnerabilities_1.VulnerabilityRequirements; } });
-var wifi_network_1 = require("./wifi-network");
-Object.defineProperty(exports, "WifiNetwork", { enumerable: true, get: function () { return wifi_network_1.WifiNetwork; } });

package/dist/types/location.d.ts

@@ -0,0 +1,11 @@
+export interface LocationOptions {
+    x: number;
+    y: number;
+}
+export declare class Location {
+    x: number;
+    y: number;
+    constructor(options: LocationOptions);
+    offset(x: number, y: number): Location;
+    fork(): Location;
+}

package/dist/types/location.js

@@ -0,0 +1,23 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Location = void 0;
+var Location = /** @class */ (function () {
+    function Location(options) {
+        this.x = options.x;
+        this.y = options.y;
+    }
+    Location.prototype.offset = function (x, y) {
+        return new Location({
+            x: this.x + x,
+            y: this.y + y
+        });
+    };
+    Location.prototype.fork = function () {
+        return new Location({
+            x: this.x,
+            y: this.y
+        });
+    };
+    return Location;
+}());
+exports.Location = Location;

package/dist/types/net-card.d.ts

@@ -0,0 +1,5 @@
+export declare enum NetCard {
+    Ethernet = "ETHERNET",
+    Wifi = "WIFI",
+    None = "NONE"
+}

package/dist/types/net-card.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.NetCard = void 0;
+var NetCard;
+(function (NetCard) {
+    NetCard["Ethernet"] = "ETHERNET";
+    NetCard["Wifi"] = "WIFI";
+    NetCard["None"] = "NONE";
+})(NetCard = exports.NetCard || (exports.NetCard = {}));

package/dist/types/network-device.d.ts

@@ -1,10 +1,11 @@
+import { NetCard } from './net-card';
 export interface NetworkDeviceOptions {
-    type: string;
+    type: NetCard;
     id: string;
     active: boolean;
 }
 export declare class NetworkDevice {
-    type: string;
+    type: NetCard;
     id: string;
     active: boolean;
     constructor(options: NetworkDeviceOptions);

package/dist/types/process.d.ts

@@ -0,0 +1,16 @@
+import { User } from './user';
+export declare class ProcessOptions {
+    owner: User;
+    pid: number;
+    command: string;
+    protected?: boolean;
+}
+export declare class Process {
+    owner: User;
+    pid: number;
+    cpu: number;
+    mem: number;
+    command: string;
+    protected: boolean;
+    constructor(options: ProcessOptions);
+}

package/dist/types/process.js

@@ -0,0 +1,22 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.Process = exports.ProcessOptions = void 0;
+var utils_1 = require("../utils");
+var ProcessOptions = /** @class */ (function () {
+    function ProcessOptions() {
+    }
+    return ProcessOptions;
+}());
+exports.ProcessOptions = ProcessOptions;
+var Process = /** @class */ (function () {
+    function Process(options) {
+        this.owner = options.owner;
+        this.pid = options.pid;
+        this.cpu = (0, utils_1.getRandom)(2, 15);
+        this.mem = (0, utils_1.getRandom)(5, 33);
+        this.command = options.command;
+        this.protected = options.protected;
+    }
+    return Process;
+}());
+exports.Process = Process;

package/dist/types/router.d.ts

@@ -2,12 +2,17 @@ import { FSDeviceType } from '../fs';
 import { Computer } from './computer';
 import { Device, DeviceOptions } from './device';
 import { Port } from './port';
+import { User } from './user';
 export interface RouterOptions extends DeviceOptions {
     bssid: string;
     essid: string;
     publicIp?: string;
     domain: string;
     whoisDescription: string;
+    wifi: {
+        credentials: User;
+        name: string;
+    };
 }
 export declare class Router extends Device {
     bssid: string;
@@ -16,7 +21,12 @@ export declare class Router extends Device {
     domain: string;
     whoisDescription: string;
     devices: Map<string, Device>;
+    wifi: {
+        credentials: User;
+        name: string;
+    };
     constructor(options: RouterOptions);
+    changeIp(sourceIp: string, targetIp: string): boolean;
     getDeviceType(): FSDeviceType;
     getRouter(): Device;
     findByLanIp(ip: string): Device;

package/dist/types/router.js

@@ -56,8 +56,18 @@ var Router = /** @class */ (function (_super) {
         _this.domain = options.domain;
         _this.whoisDescription = options.whoisDescription;
         _this.devices = new Map();
+        _this.wifi = options.wifi;
         return _this;
     }
+    Router.prototype.changeIp = function (sourceIp, targetIp) {
+        if (!this.devices.has(sourceIp) || this.devices.has(targetIp)) {
+            return false;
+        }
+        var device = this.devices.get(sourceIp);
+        this.devices.delete(sourceIp);
+        this.devices.set(targetIp, device);
+        device.localIp = targetIp;
+    };
     Router.prototype.getDeviceType = function () {
         return fs_1.FSDeviceType.Router;
     };

package/dist/types/user.d.ts

@@ -1,8 +1,8 @@
 export interface UserOptions {
     username: string;
     password: string;
-    email: string;
-    bankNumber: string;
+    email?: string;
+    bankNumber?: string;
 }
 export declare class User {
     username: string;

package/dist/utils.d.ts

@@ -14,9 +14,11 @@ export interface PermissionMap {
     o: PermissionSegment;
 }
 export declare function getServiceLibrary(service: Service): Library | null;
-export declare function getPermissionSegmentValueByString(flag: PermissionSegment, flagType: string): boolean;
-export declare function setPermissionSegmentValueByString(flag: PermissionSegment, flagType: string, value: boolean): void;
+export declare function getFlagTypeByString(flag: PermissionSegment, flagType: string): boolean;
+export declare function setFlagTypeByString(flag: PermissionSegment, flagType: string, value: boolean): void;
 export declare function getPermissionSegmentByString(flag: PermissionMap, userType: string): PermissionSegment;
-export declare function transformFlagsToPermissions(permissionMap: PermissionMap): string;
+export declare function transformFlagsToString(permissionMap: PermissionMap): string;
 export declare function getUserPath(user: User): string;
 export declare function getTraversalPath(path: string | null, currentLocation?: string[]): null | string[];
+export declare function getRandom(min: number, max: number): number;
+export declare function getRandomInt(min: number, max: number): number;

package/dist/utils.js

@@ -1,6 +1,10 @@
 "use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getTraversalPath = exports.getUserPath = exports.transformFlagsToPermissions = exports.getPermissionSegmentByString = exports.setPermissionSegmentValueByString = exports.getPermissionSegmentValueByString = exports.getServiceLibrary = exports.isLanIp = exports.isValidIp = void 0;
+exports.getRandomInt = exports.getRandom = exports.getTraversalPath = exports.getUserPath = exports.transformFlagsToString = exports.getPermissionSegmentByString = exports.setFlagTypeByString = exports.getFlagTypeByString = exports.getServiceLibrary = exports.isLanIp = exports.isValidIp = void 0;
+var random_seed_1 = __importDefault(require("random-seed"));
 var library_1 = require("./types/library");
 var service_1 = require("./types/service");
 function isValidIp(target) {
@@ -30,7 +34,7 @@ function getServiceLibrary(service) {
     return null;
 }
 exports.getServiceLibrary = getServiceLibrary;
-function getPermissionSegmentValueByString(flag, flagType) {
+function getFlagTypeByString(flag, flagType) {
     switch (flagType) {
         case 'x':
             return flag.x;
@@ -42,8 +46,8 @@ function getPermissionSegmentValueByString(flag, flagType) {
             throw new Error('Invalid permission flag type.');
     }
 }
-exports.getPermissionSegmentValueByString = getPermissionSegmentValueByString;
-function setPermissionSegmentValueByString(flag, flagType, value) {
+exports.getFlagTypeByString = getFlagTypeByString;
+function setFlagTypeByString(flag, flagType, value) {
     switch (flagType) {
         case 'x': {
             flag.x = value;
@@ -61,7 +65,7 @@ function setPermissionSegmentValueByString(flag, flagType, value) {
             throw new Error('Invalid permission flag type.');
     }
 }
-exports.setPermissionSegmentValueByString = setPermissionSegmentValueByString;
+exports.setFlagTypeByString = setFlagTypeByString;
 function getPermissionSegmentByString(flag, userType) {
     switch (userType) {
         case 'u':
@@ -75,13 +79,13 @@ function getPermissionSegmentByString(flag, userType) {
     }
 }
 exports.getPermissionSegmentByString = getPermissionSegmentByString;
-function transformFlagsToPermissions(permissionMap) {
+function transformFlagsToString(permissionMap) {
     var segments = [permissionMap.u, permissionMap.g, permissionMap.o].map(function (map) {
         return "".concat(map.r ? 'r' : '-').concat(map.w ? 'w' : '-').concat(map.x ? 'x' : '-');
     });
-    return segments.join('');
+    return "d".concat(segments.join(''));
 }
-exports.transformFlagsToPermissions = transformFlagsToPermissions;
+exports.transformFlagsToString = transformFlagsToString;
 function getUserPath(user) {
     switch (user.username) {
         case 'root':
@@ -110,3 +114,12 @@ function getTraversalPath(path, currentLocation) {
     return path.split('/');
 }
 exports.getTraversalPath = getTraversalPath;
+var utilRng = random_seed_1.default.create('util');
+function getRandom(min, max) {
+    return utilRng.floatBetween(min, max);
+}
+exports.getRandom = getRandom;
+function getRandomInt(min, max) {
+    return utilRng.intBetween(min, max);
+}
+exports.getRandomInt = getRandomInt;

package/package.json

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

package/dist/types/wifi-network.d.ts

@@ -1,16 +0,0 @@
-import { Router } from './router';
-export interface WifiNetworkOptions {
-    password: string;
-    router: Router;
-    mac: string;
-    percentage: number;
-    name: string;
-}
-export declare class WifiNetwork {
-    password: string;
-    router: Router;
-    mac: string;
-    percentage: number;
-    name: string;
-    constructor(options: WifiNetworkOptions);
-}

package/dist/types/wifi-network.js

@@ -1,14 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.WifiNetwork = void 0;
-var WifiNetwork = /** @class */ (function () {
-    function WifiNetwork(options) {
-        this.password = options.password;
-        this.router = options.router;
-        this.mac = options.mac;
-        this.percentage = options.percentage;
-        this.name = options.name;
-    }
-    return WifiNetwork;
-}());
-exports.WifiNetwork = WifiNetwork;