| // 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.network.guru; |
| |
| import javax.inject.Inject; |
| |
| import org.apache.cloudstack.engine.orchestration.service.NetworkOrchestrationService; |
| import org.apache.log4j.Logger; |
| |
| import com.cloud.dc.DataCenter; |
| import com.cloud.dc.Vlan.VlanType; |
| import com.cloud.dc.dao.DataCenterDao; |
| import com.cloud.dc.dao.VlanDao; |
| import com.cloud.deploy.DeployDestination; |
| import com.cloud.deploy.DeploymentPlan; |
| import com.cloud.exception.ConcurrentOperationException; |
| import com.cloud.exception.InsufficientAddressCapacityException; |
| import com.cloud.exception.InsufficientVirtualNetworkCapacityException; |
| import com.cloud.network.IpAddressManager; |
| import com.cloud.network.Network; |
| import com.cloud.network.Network.State; |
| import com.cloud.network.NetworkProfile; |
| import com.cloud.network.Networks.AddressFormat; |
| import com.cloud.network.Networks.BroadcastDomainType; |
| import com.cloud.network.Networks.IsolationType; |
| import com.cloud.network.Networks.Mode; |
| import com.cloud.network.Networks.TrafficType; |
| import com.cloud.network.addr.PublicIp; |
| import com.cloud.network.dao.IPAddressDao; |
| import com.cloud.network.dao.IPAddressVO; |
| import com.cloud.network.dao.NetworkVO; |
| import com.cloud.offering.NetworkOffering; |
| import com.cloud.user.Account; |
| import com.cloud.utils.component.AdapterBase; |
| import com.cloud.utils.db.DB; |
| import com.cloud.utils.db.Transaction; |
| import com.cloud.utils.db.TransactionCallbackNoReturn; |
| import com.cloud.utils.db.TransactionStatus; |
| import com.cloud.utils.exception.CloudRuntimeException; |
| import com.cloud.vm.Nic.ReservationStrategy; |
| import com.cloud.vm.NicProfile; |
| import com.cloud.vm.ReservationContext; |
| import com.cloud.vm.VirtualMachine; |
| import com.cloud.vm.VirtualMachineProfile; |
| |
| public class PublicNetworkGuru extends AdapterBase implements NetworkGuru { |
| private static final Logger s_logger = Logger.getLogger(PublicNetworkGuru.class); |
| |
| @Inject |
| DataCenterDao _dcDao; |
| @Inject |
| VlanDao _vlanDao; |
| @Inject |
| NetworkOrchestrationService _networkMgr; |
| @Inject |
| IPAddressDao _ipAddressDao; |
| @Inject |
| IpAddressManager _ipAddrMgr; |
| |
| private static final TrafficType[] TrafficTypes = {TrafficType.Public}; |
| |
| @Override |
| public boolean isMyTrafficType(TrafficType type) { |
| for (TrafficType t : TrafficTypes) { |
| if (t == type) { |
| return true; |
| } |
| } |
| return false; |
| } |
| |
| @Override |
| public TrafficType[] getSupportedTrafficType() { |
| return TrafficTypes; |
| } |
| |
| protected boolean canHandle(NetworkOffering offering) { |
| return isMyTrafficType(offering.getTrafficType()) && offering.isSystemOnly(); |
| } |
| |
| @Override |
| public Network design(NetworkOffering offering, DeploymentPlan plan, Network network, Account owner) { |
| if (!canHandle(offering)) { |
| return null; |
| } |
| |
| if (offering.getTrafficType() == TrafficType.Public) { |
| NetworkVO ntwk = |
| new NetworkVO(offering.getTrafficType(), Mode.Static, network.getBroadcastDomainType(), offering.getId(), State.Setup, plan.getDataCenterId(), |
| plan.getPhysicalNetworkId(), offering.isRedundantRouter()); |
| return ntwk; |
| } else { |
| return null; |
| } |
| } |
| |
| protected PublicNetworkGuru() { |
| super(); |
| } |
| |
| protected void getIp(NicProfile nic, DataCenter dc, VirtualMachineProfile vm, Network network) throws InsufficientVirtualNetworkCapacityException, |
| InsufficientAddressCapacityException, ConcurrentOperationException { |
| if (nic.getIPv4Address() == null) { |
| boolean forSystemVms = false; |
| if (vm.getType().equals(VirtualMachine.Type.ConsoleProxy) || vm.getType().equals(VirtualMachine.Type.SecondaryStorageVm)) { |
| forSystemVms = true; |
| } |
| PublicIp ip = _ipAddrMgr.assignPublicIpAddress(dc.getId(), null, vm.getOwner(), VlanType.VirtualNetwork, null, null, false, forSystemVms); |
| nic.setIPv4Address(ip.getAddress().toString()); |
| nic.setIPv4Gateway(ip.getGateway()); |
| nic.setIPv4Netmask(ip.getNetmask()); |
| if (network.getBroadcastDomainType() == BroadcastDomainType.Vxlan) { |
| nic.setIsolationUri(BroadcastDomainType.Vxlan.toUri(ip.getVlanTag())); |
| nic.setBroadcastUri(BroadcastDomainType.Vxlan.toUri(ip.getVlanTag())); |
| nic.setBroadcastType(BroadcastDomainType.Vxlan); |
| } else { |
| nic.setIsolationUri(IsolationType.Vlan.toUri(ip.getVlanTag())); |
| nic.setBroadcastUri(BroadcastDomainType.Vlan.toUri(ip.getVlanTag())); |
| nic.setBroadcastType(BroadcastDomainType.Vlan); |
| } |
| nic.setFormat(AddressFormat.Ip4); |
| nic.setReservationId(String.valueOf(ip.getVlanTag())); |
| nic.setMacAddress(ip.getMacAddress()); |
| } |
| |
| nic.setIPv4Dns1(dc.getDns1()); |
| nic.setIPv4Dns2(dc.getDns2()); |
| } |
| |
| @Override |
| public void updateNicProfile(NicProfile profile, Network network) { |
| DataCenter dc = _dcDao.findById(network.getDataCenterId()); |
| if (profile != null) { |
| profile.setIPv4Dns1(dc.getDns1()); |
| profile.setIPv4Dns2(dc.getDns2()); |
| } |
| } |
| |
| @Override |
| public NicProfile allocate(Network network, NicProfile nic, VirtualMachineProfile vm) throws InsufficientVirtualNetworkCapacityException, |
| InsufficientAddressCapacityException, ConcurrentOperationException { |
| |
| DataCenter dc = _dcDao.findById(network.getDataCenterId()); |
| |
| if (nic != null && nic.getRequestedIPv4() != null) { |
| throw new CloudRuntimeException("Does not support custom ip allocation at this time: " + nic); |
| } |
| |
| if (nic == null) { |
| nic = new NicProfile(ReservationStrategy.Create, null, null, null, null); |
| } |
| |
| getIp(nic, dc, vm, network); |
| |
| if (nic.getIPv4Address() == null) { |
| nic.setReservationStrategy(ReservationStrategy.Start); |
| } else if (vm.getVirtualMachine().getType() == VirtualMachine.Type.DomainRouter) { |
| nic.setReservationStrategy(ReservationStrategy.Managed); |
| } else { |
| nic.setReservationStrategy(ReservationStrategy.Create); |
| } |
| |
| return nic; |
| } |
| |
| @Override |
| public void reserve(NicProfile nic, Network network, VirtualMachineProfile vm, DeployDestination dest, ReservationContext context) |
| throws InsufficientVirtualNetworkCapacityException, InsufficientAddressCapacityException, ConcurrentOperationException { |
| if (nic.getIPv4Address() == null) { |
| getIp(nic, dest.getDataCenter(), vm, network); |
| } |
| } |
| |
| @Override |
| public boolean release(NicProfile nic, VirtualMachineProfile vm, String reservationId) { |
| return true; |
| } |
| |
| @Override |
| public Network implement(Network network, NetworkOffering offering, DeployDestination destination, ReservationContext context) |
| throws InsufficientVirtualNetworkCapacityException { |
| return network; |
| } |
| |
| @Override |
| @DB |
| public void deallocate(Network network, NicProfile nic, VirtualMachineProfile vm) { |
| if (s_logger.isDebugEnabled()) { |
| s_logger.debug("public network deallocate network: networkId: " + nic.getNetworkId() + ", ip: " + nic.getIPv4Address()); |
| } |
| |
| final IPAddressVO ip = _ipAddressDao.findByIpAndSourceNetworkId(nic.getNetworkId(), nic.getIPv4Address()); |
| if (ip != null && nic.getReservationStrategy() != ReservationStrategy.Managed) { |
| Transaction.execute(new TransactionCallbackNoReturn() { |
| @Override |
| public void doInTransactionWithoutResult(TransactionStatus status) { |
| _ipAddrMgr.markIpAsUnavailable(ip.getId()); |
| _ipAddressDao.unassignIpAddress(ip.getId()); |
| } |
| }); |
| } |
| nic.deallocate(); |
| |
| if (s_logger.isDebugEnabled()) { |
| s_logger.debug("Deallocated nic: " + nic); |
| } |
| } |
| |
| @Override |
| public void shutdown(NetworkProfile network, NetworkOffering offering) { |
| } |
| |
| @Override |
| public boolean trash(Network network, NetworkOffering offering) { |
| return true; |
| } |
| |
| @Override |
| public void updateNetworkProfile(NetworkProfile networkProfile) { |
| DataCenter dc = _dcDao.findById(networkProfile.getDataCenterId()); |
| networkProfile.setDns1(dc.getDns1()); |
| networkProfile.setDns2(dc.getDns2()); |
| } |
| |
| } |
