import { Component, OnInit, AfterContentInit, Output, EventEmitter, Input, OnDestroy } from '@angular/core'; import { Store, select, StateObservable } from '@ngrx/store'; import { RPCClientError } from '@loafer/ng-rpc'; import { DiscoveryStartInfo, DiscoverZone, Zone, DiscoverPort, DiscoverService } from '@overflow/commons-typescript/model/discovery'; import * as CIDR from 'ip-cidr'; import * as DiscoveredStore from '../../store/setting'; import { SettingSelector, DiscoverSelector } from '../../store'; import * as DiscoverStore from '../../store/discover'; import * as RegistStore from '../../store/regist'; import { Host } from '@overflow/commons-typescript/model/discovery'; import { Port } from '@overflow/commons-typescript/model/discovery'; import { Service } from '@overflow/commons-typescript/model/discovery'; import * as ProbeDetailStore from '@overflow/probe/store'; import { Probe } from '@overflow/commons-typescript/model/probe'; import { TreeNode } from 'primeng/primeng'; import { ListSelector as ProbeListSelector } from '@overflow/probe/store'; import * as ProbeListStore from '@overflow/probe/store/list'; import { AuthSelector } from '@overflow/member/store'; import { Domain } from '@overflow/commons-typescript/model/domain'; import { Subscription } from 'rxjs/Subscription'; @Component({ selector: 'of-discovery-setting', templateUrl: './setting.component.html', }) export class SettingComponent implements OnInit, AfterContentInit, OnDestroy { @Input() probe: Probe; /* To determine valid range. */ @Output() close = new EventEmitter(); settingSucceed$: any; discoveryResult$: any; discoveryReq$: any; discoveryStart$: any; discoveryEnd$: any; started = false; startIP; endIP; startPort; endPort; hostChecked = ['host']; portChecked = []; serviceChecked = []; tcpChecked = true; udpChecked = true; includeServices = []; treeNodes = []; selectedNodes = []; zones: Map = null; checkedSet = new Set(); testObj: any[] = testDiscovery; selectedDiscoveryResult: TreeNode[]; constructor( private discoverdstore: Store, private discoverstore: Store, private registStore: Store, private probeListStore: Store ) { this.settingSucceed$ = discoverdstore.pipe(select(SettingSelector.select('isStart'))); this.discoveryResult$ = discoverstore.pipe(select(DiscoverSelector.select('zones'))); this.discoveryReq$ = discoverstore.pipe(select(DiscoverSelector.select('isReq'))); this.discoveryStart$ = discoverstore.pipe(select(DiscoverSelector.select('isStart'))); this.discoveryEnd$ = discoverstore.pipe(select(DiscoverSelector.select('isEnd'))); } ngOnInit() { // this.settingSucceed$.subscribe( // (succeed: boolean) => { // if (succeed) { // this.started = true; // } // }, // (error: RPCClientError) => { // console.log(error.response.message); // } // ); // this.discoveryResult$.subscribe( // (zones: Map) => { // if (zones !== undefined && zones !== null) { // console.log('ZoneZoneZoneZoneZoneZoneZone'); // // console.log(JSON.stringify(zones)); // this.treeNodes = this.convertTreeViewZone(zones); // this.zones = zones; // } // }, // (error: RPCClientError) => { // console.log(error.response.message); // } // ); // this.discoveryReq$.subscribe( // (isReq: boolean) => { // if (isReq !== undefined && isReq !== null) { // console.log('isReqisReqisReqisReq'); // } // }, // (error: RPCClientError) => { // console.log(error.response.message); // } // ); // this.discoveryStart$.subscribe( // (isStart: boolean) => { // if (isStart !== undefined && isStart !== null) { // console.log('isStartisStartisStart'); // } // }, // (error: RPCClientError) => { // console.log(error.response.message); // } // ); // this.discoveryEnd$.subscribe( // (isEnd: boolean) => { // if (isEnd !== undefined && isEnd !== null) { // console.log('isEndisEndisEndisEnd'); // } // }, // (error: RPCClientError) => { // console.log(error.response.message); // } // ); } ngAfterContentInit() { if (this.probe === undefined || this.probe === null) { } else { this.hostRange(); this.portRange(); } } ngOnDestroy() { } onProbeSelect(event) { this.probe = event.value; this.hostRange(); this.portRange(); } closeDialog() { this.close.emit(); } isZone(zone: Zone): boolean { // for (let i = 0; i < this.treeNodes.length; ++i) { // if (zone.iface === this.treeNodes[i].iface) { // return true; // } // } return false; } hostRange() { const cidr = new CIDR(this.probe.cidr); if (!cidr.isValid()) { alert('Invalid cidr : ' + this.probe.cidr); return; } this.startIP = cidr.addressStart.address; this.endIP = cidr.addressEnd.address; } portRange() { this.startPort = '1'; this.endPort = '1024'; } handleHostCheckChange(e: any) { this.hostChecked = ['host']; } handlePortCheckChange(e: any) { this.serviceChecked = []; } handleServiceCheckChange(e: any) { if (e) { this.portChecked = ['port']; } } handleServiceChange(list: any) { this.includeServices = list; } discovery() { console.log(this.startIP); console.log(this.startPort); console.log(this.endPort); let discoverPort: DiscoverPort = null; let discoverService: DiscoverService = null; if (this.serviceChecked.length > 0) { const services = new Array(); for (const service of this.includeServices) { services.push(service.name); } discoverService = { includeServices: services, }; } if (this.portChecked.length > 0) { discoverPort = { firstScanRange: this.startPort, lastScanRange: this.endPort, includeTCP: this.tcpChecked, includeUDP: this.udpChecked, excludePorts: null, discoverService: discoverService }; } const discoverZone: DiscoverZone = { discoverHost: { firstScanRangev4: this.startIP, lastScanRangev4: this.endIP, discoverPort: discoverPort, }, }; console.log(discoverZone); // console.log('start discovery - ' + this.probe.probeKey); this.discoverstore.dispatch(new DiscoverStore.DiscoverZone( { probeID: this.probe.probeKey, discoverZone: discoverZone })); this.started = true; } checkDiscoveryResult(node) { node.toggleActivated(true); if (node.isActive) { if (node.data.obj !== undefined) { node.data.obj.target = true; } } else { if (node.data.obj !== undefined) { node.data.obj.target = false; } } console.log(this.zones); } convertMapToList(map: Map): Array { if (undefined === map || null === map || map.size < 0) { return null; } const retArr = new Array(); map.forEach((v: any, k: any) => { retArr.push(v); }); return retArr; } discoveryResultSave() { // console.log(this.checkedSet); // send discovery save this.zones; // this.registStore.dispatch(new DiscoverStore.DiscoverZone( // {probeID: '52abd6fd57e511e7ac52080027658d13', discoveryZone: discoveryZone})); // console.log(this.zones); // isTarget Check for (const sn of this.selectedNodes) { if (sn.obj !== undefined && sn.obj !== null) { if (sn.obj.target !== undefined && sn.obj.target !== null) { sn.obj.target = true; } } } this.zones.forEach((zone: Zone, key: string) => { if (zone.hosts !== undefined && zone.hosts !== null && zone.hosts.size > 0) { // FIXME: getHosts const hosts: Host[] = []; zone.hosts.forEach((h: Host, hKey: string) => { h.zone = null; if (h.ports !== undefined && h.ports !== null) { h.portList = []; h.ports.forEach((p: Port, pKey: number) => { p.host = null; if (p.services !== undefined && p.services !== null) { p.serviceList = []; p.services.forEach((s: Service, sKey: string) => { s.port = null; p.serviceList.push(s); }); } h.portList.push(p); }); } hosts.push(h); }); if (hosts.length > 0) { const sd: any = { hosts: hosts, probe: { id: 1 } }; console.log(JSON.stringify(sd)); console.log('--------'); this.registStore.dispatch(new RegistStore.DiscoverySaveAllTarget(sd)); } } }); } handleClick(event) { // console.log(this.filesTree4); this.checkTarget(this.selectedDiscoveryResult); console.log(this.testObj); } checkTarget(treeList: any[]) { if (treeList === undefined || treeList === null) { return; } for (let i = 0; i < treeList.length; ++i) { treeList[i].obj.target = true; this.checkTarget(treeList[i].children); } } convertTreeViewZone(zones: Map) { if (zones === undefined || zones === null) { return; } // console.log('convertTreeViewZone'); const treeNodes: any[] = []; zones.forEach((value: Zone, key: string, map) => { if (this.isZone(value)) { return; } const jZone: any = { label: 'Zone - ' + value.iface, // className: 'cn' + value.ip, expandedIcon: 'fa-folder-open', collapsedIcon: 'fa-folder', }; jZone.obj = value; jZone.children = this.convertViewHost(value.hosts); treeNodes.push(jZone); }); return treeNodes; } convertViewHost(hosts): any[] { if (hosts === undefined || hosts === null) { return null; } const hostNodes: any[] = []; hosts.forEach((host, hostKey) => { const jHost: any = { label: 'Host - ' + host.ipv4, // className: 'cn' + host.ip expandedIcon: 'fa-folder-open', collapsedIcon: 'fa-folder', }; jHost.obj = host; jHost.children = this.convertViewPort(host.ports); hostNodes.push(jHost); }); return hostNodes; } convertViewPort(ports): any[] { if (ports === undefined || ports === null || ports.size < 0) { return null; } const portChildren: any[] = []; ports.forEach((port, portKey) => { const jPort: any = { label: 'Port - ' + port.portNumber, // className: 'cn' + port.portNumber, expandedIcon: 'fa-folder-open', collapsedIcon: 'fa-folder', }; jPort.obj = port; jPort.children = this.convertViewService(port.services); portChildren.push(jPort); }); return portChildren; } convertViewService(services): any[] { if (services === undefined || services === null || services.size <= 0) { return null; } const serviceChildren: any[] = []; services.forEach((service, serviceKey) => { const jService: any = { label: 'Service - ' + service.serviceName, // className: 'cn' + service.serviceName, }; jService.obj = service; serviceChildren.push(jService); }); return serviceChildren; } } const testDiscovery: any[] = [ { id: 0, ip: '192.168.1.254', mac: '64:e5:99:63:e7:80', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: [ { id: 0, portType: 'UDP', portNumber: 53, discoveredDate: -62135596800000, host: null, services: [ { id: 0, cryptoType: null, serviceName: 'DNS', discoveredDate: -62135596800000, port: null, target: false } ] } ] }, { id: 0, ip: '192.168.1.99', mac: '00:25:b3:fa:ca:9b', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: [ { id: 0, portType: 'UDP', portNumber: 161, discoveredDate: -62135596800000, host: null, services: [ { id: 0, cryptoType: null, serviceName: 'SNMP_V2', discoveredDate: -62135596800000, port: null, target: false } ] }, { id: 0, portType: 'TCP', portNumber: 7, discoveredDate: -62135596800000, host: null, services: null } ] }, { id: 0, ip: '192.168.1.206', mac: '30:9c:23:15:a3:09', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: null }, { id: 0, ip: '192.168.1.50', mac: '6c:f0:49:7a:60:68', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: [ { id: 0, portType: 'TCP', portNumber: 22, discoveredDate: -62135596800000, host: null, services: [ { id: 0, cryptoType: 'TCP', serviceName: 'SSH', discoveredDate: -62135596800000, port: null, target: false } ] } ] }, { id: 0, ip: '192.168.1.15', mac: '00:19:b9:35:8e:a6', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: [ { id: 0, portType: 'TCP', portNumber: 22, discoveredDate: -62135596800000, host: null, services: null }, { id: 0, portType: 'TCP', portNumber: 25, discoveredDate: -62135596800000, host: null, services: null } ] }, { id: 0, ip: '192.168.1.100', mac: '00:17:08:8d:cf:f7', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: [ { id: 0, portType: 'UDP', portNumber: 161, discoveredDate: -62135596800000, host: null, services: [ { id: 0, cryptoType: null, serviceName: 'SNMP_V2', discoveredDate: -62135596800000, port: null, target: false } ] }, { id: 0, portType: 'TCP', portNumber: 23, discoveredDate: -62135596800000, host: null, services: null }, { id: 0, portType: 'TCP', portNumber: 21, discoveredDate: -62135596800000, host: null, services: null } ] }, { id: 0, ip: '192.168.1.16', mac: '00:19:b9:19:7e:c7', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: [ { id: 0, portType: 'TCP', portNumber: 22, discoveredDate: -62135596800000, host: null, services: null } ] }, { id: 0, ip: '192.168.1.106', mac: '44:8a:5b:f1:3a:7d', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: [ { id: 0, portType: 'UDP', portNumber: 161, discoveredDate: -62135596800000, host: null, services: [ { id: 0, cryptoType: null, serviceName: 'SNMP_V2', discoveredDate: -62135596800000, port: null, target: false } ] }, { id: 0, portType: 'TCP', portNumber: 21, discoveredDate: -62135596800000, host: null, services: null } ] }, { id: 0, ip: '192.168.1.101', mac: '44:8a:5b:f1:f1:f3', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: null }, { id: 0, ip: '192.168.1.102', mac: '44:8a:5b:f1:f2:d3', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: null }, { id: 0, ip: '192.168.1.205', mac: '40:98:ad:7b:d0:43', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: null }, { id: 0, ip: '192.168.1.103', mac: '44:8a:5b:44:8c:e8', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: [ { id: 0, portType: 'TCP', portNumber: 21, discoveredDate: -62135596800000, host: null, services: null } ] }, { id: 0, ip: '192.168.1.10', mac: 'd0:50:99:97:5d:99', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: [ { id: 0, portType: 'TCP', portNumber: 22, discoveredDate: -62135596800000, host: null, services: null } ] }, { id: 0, ip: '192.168.1.203', mac: 'a8:e5:39:5b:c9:62', os: null, discoveredDate: -62135596800000, target: false, zone: null, ports: null } ];