content: // Licensed to the Apache Software Foundation (ASF) under one
  // or more contributor license agreements.  See the NOTICE file
  // distributed with this work for additional information
  // regarding copyright ownership.  The ASF licenses this file
  // to you under the Apache License, Version 2.0 (the
  // "License"); you may not use this file except in compliance
  // with the License.  You may obtain a copy of the License at
  //
  //   http://www.apache.org/licenses/LICENSE-2.0
  //
  // Unless required by applicable law or agreed to in writing,
  // software distributed under the License is distributed on an
  // "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  // KIND, either express or implied.  See the License for the
  // specific language governing permissions and limitations
  // under the License.
  
  package com.cloud.hypervisor.ovm3.resources;
  
  import java.io.IOException;
  import java.net.InetAddress;
  import java.net.URI;
  import java.net.UnknownHostException;
  import java.util.HashMap;
  import java.util.List;
  import java.util.Map;
  import java.util.UUID;
  
  import javax.inject.Inject;
  import javax.naming.ConfigurationException;
  
  
  import com.cloud.agent.AgentManager;
  import com.cloud.agent.Listener;
  import com.cloud.agent.api.AgentControlAnswer;
  import com.cloud.agent.api.AgentControlCommand;
  import com.cloud.agent.api.Answer;
  import com.cloud.agent.api.Command;
  import com.cloud.agent.api.StartupCommand;
  import com.cloud.agent.api.StartupRoutingCommand;
  import com.cloud.configuration.Config;
  import com.cloud.dc.ClusterDetailsDao;
  import com.cloud.dc.ClusterVO;
  import com.cloud.dc.dao.ClusterDao;
  import com.cloud.exception.DiscoveryException;
  import com.cloud.host.Host;
  import com.cloud.host.HostInfo;
  import com.cloud.host.HostVO;
  import com.cloud.host.Status;
  import com.cloud.host.dao.HostDao;
  import com.cloud.hypervisor.Hypervisor.HypervisorType;
  import com.cloud.hypervisor.ovm3.objects.Connection;
  import com.cloud.hypervisor.ovm3.objects.Linux;
  import com.cloud.hypervisor.ovm3.objects.Ovm3ResourceException;
  import com.cloud.resource.Discoverer;
  import com.cloud.resource.DiscovererBase;
  import com.cloud.resource.ResourceManager;
  import com.cloud.resource.ResourceStateAdapter;
  import com.cloud.resource.ServerResource;
  import com.cloud.resource.UnableDeleteHostException;
  import com.cloud.utils.UuidUtils;
  import com.cloud.utils.db.QueryBuilder;
  import com.cloud.utils.db.SearchCriteria;
  import com.cloud.utils.exception.CloudRuntimeException;
  import com.cloud.utils.ssh.SSHCmdHelper;
  
  public class Ovm3Discoverer extends DiscovererBase implements Discoverer,
          Listener, ResourceStateAdapter {
      protected String publicNetworkDevice;
      protected String privateNetworkDevice;
      protected String guestNetworkDevice;
      protected String storageNetworkDevice;
  
      @Inject
      ClusterDao clusterDao;
      @Inject
      ClusterDetailsDao clusterDetailsDao;
      @Inject
      ResourceManager resourceMgr;
      @Inject
      AgentManager agentMgr;
      @Inject
      HostDao hostDao = null;
  
      protected Ovm3Discoverer() {
      }
  
      @Override
      public boolean configure(String name, Map<String, Object> params)
              throws ConfigurationException {
          boolean success = super.configure(name, params);
          if (!success) {
              return false;
          }
  
          /* these are in Config.java */
          publicNetworkDevice = _params.get(Config.Ovm3PublicNetwork.key());
          privateNetworkDevice = _params.get(Config.Ovm3PrivateNetwork.key());
          guestNetworkDevice = _params.get(Config.Ovm3GuestNetwork.key());
          storageNetworkDevice = _params.get(Config.Ovm3StorageNetwork.key());
          resourceMgr.registerResourceStateAdapter(this.getClass()
                  .getSimpleName(), this);
          return true;
      }
  
      @Override
      public boolean stop() {
          resourceMgr.unregisterResourceStateAdapter(this.getClass()
                  .getSimpleName());
          return super.stop();
      }
  
      private boolean checkIfExisted(String guid) {
          QueryBuilder<HostVO> sc = QueryBuilder.create(HostVO.class);
          sc.and(sc.entity().getGuid(), SearchCriteria.Op.EQ, guid);
          sc.and(sc.entity().getHypervisorType(), SearchCriteria.Op.EQ,
                  HypervisorType.Ovm3);
          List<HostVO> hosts = sc.list();
          return !hosts.isEmpty();
      }
  
      private boolean CheckUrl(URI url) throws DiscoveryException {
          if ("http".equals(url.getScheme()) || "https".equals(url.getScheme())) {
              String msg = "Discovering " + url + ": " + _params;
              logger.debug(msg);
          } else {
              String msg = "urlString is not http(s) so we're not taking care of the discovery for this: "
                      + url;
              logger.info(msg);
              throw new DiscoveryException(msg);
          }
          return true;
      }
  
      @Override
      public Map<? extends ServerResource, Map<String, String>> find(long dcId,
              Long podId, Long clusterId, URI url, String username,
              String password, List<String> hostTags) throws DiscoveryException {
          Connection c = null;
  
          CheckUrl(url);
          if (clusterId == null) {
              String msg = "must specify cluster Id when add host";
              logger.info(msg);
              throw new DiscoveryException(msg);
          }
  
          if (podId == null) {
              String msg = "must specify pod Id when add host";
              logger.info(msg);
              throw new DiscoveryException(msg);
          }
  
          ClusterVO cluster = clusterDao.findById(clusterId);
          if (cluster == null
                  || (cluster.getHypervisorType() != HypervisorType.Ovm3)) {
              String msg = "invalid cluster id or cluster is not for Ovm3 hypervisors";
              logger.info(msg);
              throw new DiscoveryException(msg);
          } else {
              logger.debug("cluster: " + cluster);
          }
  
          String agentUsername = _params.get("agentusername");
          if (agentUsername == null) {
              String msg = "Agent user name must be specified";
              logger.info(msg);
              throw new DiscoveryException(msg);
          }
  
          String agentPassword = _params.get("agentpassword");
          if (agentPassword == null) {
              String msg = "Agent password must be specified";
              logger.info(msg);
              throw new DiscoveryException(msg);
          }
  
          String agentPort = _params.get("agentport");
          if (agentPort == null) {
              String msg = "Agent port must be specified";
              logger.info(msg);
              throw new DiscoveryException(msg);
          }
  
          try {
              String hostname = url.getHost();
  
              InetAddress ia = InetAddress.getByName(hostname);
              String hostIp = ia.getHostAddress();
              String guid = UuidUtils.nameUUIDFromBytes(hostIp.getBytes("UTF8"))
                      .toString();
  
              if (checkIfExisted(guid)) {
                  String msg = "The host " + hostIp + " has been added before";
                  logger.info(msg);
                  throw new DiscoveryException(msg);
              }
  
              logger.debug("Ovm3 discover is going to disover host having guid "
                      + guid);
  
              ClusterVO clu = clusterDao.findById(clusterId);
              if (clu.getGuid() == null) {
                  clu.setGuid(UUID.randomUUID().toString());
              }
              clusterDao.update(clusterId, clu);
              Map<String, String> clusterDetails = clusterDetailsDao
                      .findDetails(clusterId);
              String ovm3vip = (clusterDetails.get("ovm3vip") == null) ? ""
                      : clusterDetails.get("ovm3vip");
              String ovm3pool = (clusterDetails.get("ovm3pool") == null) ? "false"
                      : clusterDetails.get("ovm3pool");
              String ovm3cluster = (clusterDetails.get("ovm3cluster") == null) ? "false"
                      : clusterDetails.get("ovm3cluster");
  
              /* should perhaps only make this connect to the agent port ? */
              com.trilead.ssh2.Connection sshConnection = new com.trilead.ssh2.Connection(
                      hostIp, 22);
              sshConnection.connect(null, 60000, 60000);
              sshConnection = SSHCmdHelper.acquireAuthorizedConnection(hostIp,
                      username, password);
              if (sshConnection == null) {
                  String msg = "Cannot Ssh to Ovm3 host(IP=" + hostIp
                          + ", username=" + username
                          + ", password=*******), discovery failed";
                  logger.warn(msg);
                  throw new DiscoveryException(msg);
              }
  
              Map<String, String> details = new HashMap<String, String>();
              Ovm3HypervisorResource ovmResource = new Ovm3HypervisorResource();
              details.put("ip", hostIp);
              details.put("host", hostname);
              details.put("username", username);
              details.put("password", password);
              details.put("zone", Long.toString(dcId));
              details.put("guid", guid);
              details.put("pod", Long.toString(podId));
              details.put("cluster", Long.toString(clusterId));
              details.put("agentusername", agentUsername);
              details.put("agentpassword", agentPassword);
              details.put("agentport", agentPort.toString());
              details.put("ovm3vip", ovm3vip);
              details.put("ovm3pool", ovm3pool);
              details.put("ovm3cluster", ovm3cluster);
  
              if (publicNetworkDevice != null) {
                  details.put("public.network.device", publicNetworkDevice);
              }
              if (privateNetworkDevice != null) {
                  details.put("private.network.device", privateNetworkDevice);
              }
              if (guestNetworkDevice != null) {
                  details.put("guest.network.device", guestNetworkDevice);
              }
              if (storageNetworkDevice != null) {
                  details.put("storage.network.device", storageNetworkDevice);
              }
  
              Map<String, Object> params = new HashMap<String, Object>();
              params.putAll(details);
  
              ovmResource.configure(hostname, params);
              ovmResource.start();
  
              c = new Connection(hostIp, Integer.parseInt(agentPort),
                      agentUsername, agentPassword);
  
              /* After resource start, we are able to execute our agent api */
              Linux host = new Linux(c);
              details.put("agentVersion", host.getAgentVersion());
              details.put(HostInfo.HOST_OS_KERNEL_VERSION,
                      host.getHostKernelRelease());
              details.put(HostInfo.HOST_OS, host.getHostOs());
              details.put(HostInfo.HOST_OS_VERSION, host.getHostOsVersion());
              details.put(HostInfo.HYPERVISOR_VERSION,
                      host.getHypervisorVersion());
  
              Map<Ovm3HypervisorResource, Map<String, String>> resources = new HashMap<Ovm3HypervisorResource, Map<String, String>>();
              resources.put(ovmResource, details);
              return resources;
          } catch (UnknownHostException e) {
              logger.error(
                      "Host name resolve failed exception, Unable to discover Ovm3 host: "
                              + url.getHost(), e);
              return null;
          } catch (ConfigurationException e) {
              logger.error(
                      "Configure resource failed, Unable to discover Ovm3 host: "
                              + url.getHost(), e);
              return null;
          } catch (IOException | Ovm3ResourceException e) {
              logger.error("Unable to discover Ovm3 host: " + url.getHost(), e);
              return null;
          }
      }
  
      @Override
      public void postDiscovery(List<HostVO> hosts, long msId)
              throws CloudRuntimeException {
          logger.debug("postDiscovery: " + hosts);
      }
  
      @Override
      public boolean matchHypervisor(String hypervisor) {
          return HypervisorType.Ovm3.toString().equalsIgnoreCase(hypervisor);
      }
  
      @Override
      public HypervisorType getHypervisorType() {
          return HypervisorType.Ovm3;
      }
  
      @Override
      public HostVO createHostVOForConnectedAgent(HostVO host,
              StartupCommand[] cmd) {
          logger.debug("createHostVOForConnectedAgent: " + host);
          return null;
      }
  
      @Override
      public boolean processAnswers(long agentId, long seq, Answer[] answers) {
          logger.debug("processAnswers: " + agentId);
          return false;
      }
  
      @Override
      public boolean processCommands(long agentId, long seq, Command[] commands) {
          logger.debug("processCommands: " + agentId);
          return false;
      }
  
      @Override
      public AgentControlAnswer processControlCommand(long agentId,
              AgentControlCommand cmd) {
          logger.debug("processControlCommand: " + agentId);
          return null;
      }
  
      @Override
      public void processHostAdded(long hostId) {
      }
  
      /* for reconnecting */
      @Override
      public void processConnect(Host host, StartupCommand cmd,
              boolean forRebalance) {
          logger.debug("processConnect");
      }
  
      @Override
      public boolean processDisconnect(long agentId, Status state) {
          logger.debug("processDisconnect");
          return false;
      }
  
      @Override
      public void processHostAboutToBeRemoved(long hostId) {
      }
  
      @Override
      public void processHostRemoved(long hostId, long clusterId) {
      }
  
      @Override
      public boolean isRecurring() {
          return false;
      }
  
      @Override
      public int getTimeout() {
          logger.debug("getTimeout");
          return 0;
      }
  
      @Override
      public boolean processTimeout(long agentId, long seq) {
          logger.debug("processTimeout: " + agentId);
          return false;
      }
  
      @Override
      public HostVO createHostVOForDirectConnectAgent(HostVO host,
              StartupCommand[] startup, ServerResource resource,
              Map<String, String> details, List<String> hostTags) {
          logger.debug("createHostVOForDirectConnectAgent: " + host);
          StartupCommand firstCmd = startup[0];
          if (!(firstCmd instanceof StartupRoutingCommand)) {
              return null;
          }
  
          StartupRoutingCommand ssCmd = (StartupRoutingCommand) firstCmd;
          if (ssCmd.getHypervisorType() != HypervisorType.Ovm3) {
              return null;
          }
  
          return resourceMgr.fillRoutingHostVO(host, ssCmd, HypervisorType.Ovm3,
                  details, hostTags);
      }
  
      @Override
      public DeleteHostAnswer deleteHost(HostVO host, boolean isForced,
              boolean isForceDeleteStorage) throws UnableDeleteHostException {
          logger.debug("deleteHost: " + host);
          if (host.getType() != com.cloud.host.Host.Type.Routing
                  || host.getHypervisorType() != HypervisorType.Ovm3) {
              return null;
          }
  
          resourceMgr.deleteRoutingHost(host, isForced, isForceDeleteStorage);
          return new DeleteHostAnswer(true);
      }
  
  }
  