noauthprobe authorization with infras
This commit is contained in:
insanity
parent
548ed630b9
commit
8693503077
|
|
@ -7,20 +7,34 @@ import com.loafle.overflow.central.commons.stereotype.ProbeAPI;
|
|||
import com.loafle.overflow.central.commons.stereotype.WebappAPI;
|
||||
import com.loafle.overflow.central.commons.utils.GenerateKey;
|
||||
import com.loafle.overflow.central.commons.utils.SessionMetadata;
|
||||
import com.loafle.overflow.central.commons.utils.StringConvertor;
|
||||
import com.loafle.overflow.central.module.infra.service.CentralInfraMachineService;
|
||||
import com.loafle.overflow.central.module.infra.service.CentralInfraOSService;
|
||||
import com.loafle.overflow.central.module.noauthprobe.dao.NoAuthProbeDAO;
|
||||
import com.loafle.overflow.core.exception.OverflowException;
|
||||
import com.loafle.overflow.model.apikey.ApiKey;
|
||||
import com.loafle.overflow.model.domain.Domain;
|
||||
import com.loafle.overflow.model.domain.DomainMember;
|
||||
import com.loafle.overflow.model.infra.InfraHost;
|
||||
import com.loafle.overflow.model.infra.InfraMachine;
|
||||
import com.loafle.overflow.model.infra.InfraOS;
|
||||
import com.loafle.overflow.model.member.Member;
|
||||
import com.loafle.overflow.model.meta.MetaInfraType;
|
||||
import com.loafle.overflow.model.meta.MetaInfraVendor;
|
||||
import com.loafle.overflow.model.meta.MetaNoAuthProbeStatus;
|
||||
import com.loafle.overflow.model.meta.MetaProbeStatus;
|
||||
import com.loafle.overflow.model.noauthprobe.NoAuthProbe;
|
||||
import com.loafle.overflow.model.probe.Probe;
|
||||
import com.loafle.overflow.model.probe.ProbeHost;
|
||||
import com.loafle.overflow.service.central.apikey.ApiKeyService;
|
||||
import com.loafle.overflow.service.central.domain.DomainMemberService;
|
||||
import com.loafle.overflow.service.central.infra.InfraHostService;
|
||||
import com.loafle.overflow.service.central.infra.InfraMachineService;
|
||||
import com.loafle.overflow.service.central.infra.InfraOSService;
|
||||
import com.loafle.overflow.service.central.noauthprobe.NoAuthProbeService;
|
||||
import com.loafle.overflow.service.central.probe.ProbeHostService;
|
||||
import com.loafle.overflow.service.central.probe.ProbeService;
|
||||
import com.sun.jna.platform.win32.Variant.VARIANT._VARIANT.__VARIANT;
|
||||
|
||||
import org.springframework.beans.factory.annotation.Autowired;
|
||||
import org.springframework.security.crypto.bcrypt.BCryptPasswordEncoder;
|
||||
|
|
@ -31,6 +45,8 @@ import java.util.List;
|
|||
import java.util.Map;
|
||||
import java.util.UUID;
|
||||
|
||||
import javax.transaction.Transactional;
|
||||
|
||||
/**
|
||||
* Created by snoop on 17. 6. 28.
|
||||
*/
|
||||
|
|
@ -55,6 +71,15 @@ public class CentralNoAuthProbeService implements NoAuthProbeService{
|
|||
@Autowired
|
||||
private Gson gson;
|
||||
|
||||
@Autowired
|
||||
private InfraMachineService infraMachineService;
|
||||
@Autowired
|
||||
private InfraOSService infraOSService;
|
||||
@Autowired
|
||||
private InfraHostService infraHostService;
|
||||
@Autowired
|
||||
private ProbeHostService probeHostService;
|
||||
|
||||
@ProbeAPI
|
||||
public NoAuthProbe regist(NoAuthProbe noAuthProbe) throws OverflowException {
|
||||
|
||||
|
|
@ -69,7 +94,6 @@ public class CentralNoAuthProbeService implements NoAuthProbeService{
|
|||
return this.noAuthProbeDAO.save(noAuthProbe);
|
||||
}
|
||||
|
||||
|
||||
public List<NoAuthProbe> readAllByDomain(Domain domain) throws OverflowException {
|
||||
|
||||
return this.noAuthProbeDAO.findAllByDomain(domain);
|
||||
|
|
@ -80,47 +104,101 @@ public class CentralNoAuthProbeService implements NoAuthProbeService{
|
|||
}
|
||||
|
||||
@WebappAPI
|
||||
@Transactional
|
||||
public List<NoAuthProbe> acceptNoAuthProbe(NoAuthProbe noAuthProbe) throws OverflowException {
|
||||
String memberEmail = SessionMetadata.getTargetID();
|
||||
|
||||
// Todo domain injection & member injection
|
||||
HashMap<String, Object> objMap = this.gson.fromJson(noAuthProbe.getDescription(),
|
||||
TypeToken.getParameterized(HashMap.class, String.class, Object.class).getType());
|
||||
Map<String, String> hostMap = (Map<String, String>) objMap.get("host");
|
||||
Map<String, String> netMap = (Map<String, String>) objMap.get("network");
|
||||
|
||||
Probe probe = this.newProbe(noAuthProbe, hostMap, netMap);
|
||||
InfraMachine machine = this.newInfraMachine(noAuthProbe);
|
||||
InfraOS os = this.newInfraOS(machine, hostMap);
|
||||
InfraHost host = this.newInfraHost(os, netMap);
|
||||
this.newProbeHost(host, probe);
|
||||
|
||||
noAuthProbe.setStatus(new MetaNoAuthProbeStatus((short) 1));
|
||||
this.noAuthProbeDAO.save(noAuthProbe);
|
||||
|
||||
messagePublisher.publishToNoAuthProbe(noAuthProbe.getTempProbeKey(), "NoAuthProbeService.Accept",
|
||||
probe.getProbeKey());
|
||||
|
||||
return this.readAllByDomain(noAuthProbe.getDomain());
|
||||
}
|
||||
|
||||
private Probe newProbe(NoAuthProbe noauthprobe, Map<String, String> hostMap, Map<String, String> netMap) throws OverflowException {
|
||||
|
||||
BCryptPasswordEncoder passwordEncoder = new BCryptPasswordEncoder();
|
||||
String encryptKey = passwordEncoder.encode(UUID.randomUUID().toString());
|
||||
|
||||
ApiKey apiKey = apiKeyService.readByApiKey(noAuthProbe.getApiKey());
|
||||
ApiKey apiKey = apiKeyService.readByApiKey(noauthprobe.getApiKey());
|
||||
String memberEmail = SessionMetadata.getTargetID();
|
||||
DomainMember domainMember = domainMemberService.readByMemberEmail(memberEmail);
|
||||
|
||||
Probe probe = new Probe();
|
||||
probe.setEncryptionKey(encryptKey);
|
||||
probe.setProbeKey(GenerateKey.getKey());
|
||||
|
||||
probe.setDomain(new Domain(apiKey.getDomain().getId()));
|
||||
probe.setAuthorizeMember(new Member(domainMember.getMember().getId()));
|
||||
probe.setStatus(new MetaProbeStatus((short) 1));
|
||||
|
||||
HashMap<String, Object> objMap = this.gson.fromJson(noAuthProbe.getDescription(), TypeToken.getParameterized(HashMap.class, String.class, Object.class).getType());
|
||||
Map<String, String> hostMap = (Map<String, String>) objMap.get("host");
|
||||
Map<String, String> netMap = (Map<String, String>) objMap.get("network");
|
||||
|
||||
String dispName = hostMap.get("name");
|
||||
dispName += " probe";
|
||||
probe.setDisplayName(dispName);
|
||||
|
||||
String addrStr = netMap.get("address");
|
||||
|
||||
String[] addrArr = addrStr.split("\\|");
|
||||
|
||||
probe.setCidr(addrArr[0]);
|
||||
|
||||
noAuthProbe.setStatus(new MetaNoAuthProbeStatus((short) 1));
|
||||
return this.probeService.regist(probe);
|
||||
}
|
||||
|
||||
this.probeService.regist(probe);
|
||||
this.noAuthProbeDAO.save(noAuthProbe);
|
||||
private InfraMachine newInfraMachine(NoAuthProbe noauthprobe) throws OverflowException {
|
||||
MetaInfraType infraType = new MetaInfraType();
|
||||
infraType.setId(1);
|
||||
|
||||
messagePublisher.publishToNoAuthProbe(noAuthProbe.getTempProbeKey(), "NoAuthProbeService.Accept", probe.getProbeKey());
|
||||
InfraMachine infraMachine = new InfraMachine();
|
||||
infraMachine.setMeta(noauthprobe.getDescription());
|
||||
infraMachine.setInfraType(infraType);
|
||||
|
||||
return this.readAllByDomain(noAuthProbe.getDomain());
|
||||
return this.infraMachineService.regist(infraMachine);
|
||||
}
|
||||
|
||||
private InfraOS newInfraOS(InfraMachine infraMachine, Map<String, String> hostMap) throws OverflowException {
|
||||
MetaInfraType infraType = new MetaInfraType();
|
||||
infraType.setId(3);
|
||||
|
||||
InfraOS infraOS = new InfraOS();
|
||||
infraOS.setMachine(infraMachine);
|
||||
infraOS.setInfraType(infraType);
|
||||
infraOS.setVendor(MetaInfraVendor.CreateInfraVendorByOS(hostMap.get("os")));
|
||||
|
||||
return this.infraOSService.regist(infraOS);
|
||||
}
|
||||
|
||||
private InfraHost newInfraHost(InfraOS infraOS, Map<String, String> netMap) throws OverflowException {
|
||||
MetaInfraType infraType = new MetaInfraType();
|
||||
infraType.setId(2);
|
||||
|
||||
InfraHost infraHost = new InfraHost();
|
||||
String addrStr = netMap.get("address");
|
||||
String[] addrArr = addrStr.split("\\|");
|
||||
|
||||
infraHost.setIp(StringConvertor.ipToLong(addrArr[1]));
|
||||
infraHost.setMac(StringConvertor.macStrToLong(netMap.get("macAddress")));
|
||||
infraHost.setOs(infraOS);
|
||||
infraHost.setInfraType(infraType);
|
||||
|
||||
return this.infraHostService.regist(infraHost);
|
||||
}
|
||||
|
||||
private void newProbeHost(InfraHost infraHost, Probe probe) throws OverflowException {
|
||||
ProbeHost probeHost = new ProbeHost();
|
||||
probeHost.setHost(infraHost);
|
||||
probeHost.setProbe(probe);
|
||||
|
||||
this.probeHostService.regist(probeHost);
|
||||
}
|
||||
|
||||
@WebappAPI
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user