Updated Branches:
  refs/heads/master 3d1989c10 -> 06acd9f0e

IPv6: One network can have more than one vlan


Project: http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/repo
Commit: 
http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/commit/06acd9f0
Tree: http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/tree/06acd9f0
Diff: http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/diff/06acd9f0

Branch: refs/heads/master
Commit: 06acd9f0eac69d3380b54c470b4cbe889415cae9
Parents: 3d1989c
Author: Sheng Yang <[email protected]>
Authored: Wed Feb 6 19:46:58 2013 -0800
Committer: Sheng Yang <[email protected]>
Committed: Wed Feb 6 20:07:07 2013 -0800

----------------------------------------------------------------------
 api/src/com/cloud/network/NetworkModel.java        |    4 +-
 .../com/cloud/network/Ipv6AddressManagerImpl.java  |   49 ++++++++++-----
 .../src/com/cloud/network/NetworkManagerImpl.java  |    6 +--
 server/src/com/cloud/network/NetworkModelImpl.java |   29 +++++----
 .../com/cloud/network/dao/UserIpv6AddressDao.java  |    2 +
 .../cloud/network/dao/UserIpv6AddressDaoImpl.java  |    8 +++
 .../com/cloud/network/MockNetworkModelImpl.java    |    6 +-
 .../test/com/cloud/vpc/MockNetworkModelImpl.java   |    6 +-
 8 files changed, 67 insertions(+), 43 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/blob/06acd9f0/api/src/com/cloud/network/NetworkModel.java
----------------------------------------------------------------------
diff --git a/api/src/com/cloud/network/NetworkModel.java 
b/api/src/com/cloud/network/NetworkModel.java
index 57eeb29..0432e4f 100644
--- a/api/src/com/cloud/network/NetworkModel.java
+++ b/api/src/com/cloud/network/NetworkModel.java
@@ -249,7 +249,7 @@ public interface NetworkModel {
     
     boolean isNetworkInlineMode(Network network);
 
-       Vlan getVlanForNetwork(long networkId);
+       boolean isIP6AddressAvailableInNetwork(long networkId);
 
-       boolean isIP6AddressAvailable(long networkId);
+       boolean isIP6AddressAvailableInVlan(long vlanId);
 }
\ No newline at end of file

http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/blob/06acd9f0/server/src/com/cloud/network/Ipv6AddressManagerImpl.java
----------------------------------------------------------------------
diff --git a/server/src/com/cloud/network/Ipv6AddressManagerImpl.java 
b/server/src/com/cloud/network/Ipv6AddressManagerImpl.java
index f5ad7da..ecef5a2 100644
--- a/server/src/com/cloud/network/Ipv6AddressManagerImpl.java
+++ b/server/src/com/cloud/network/Ipv6AddressManagerImpl.java
@@ -17,6 +17,7 @@
 
 package com.cloud.network;
 
+import java.util.List;
 import java.util.Map;
 
 import javax.ejb.Local;
@@ -30,6 +31,7 @@ import com.cloud.configuration.dao.ConfigurationDao;
 import com.cloud.dc.DataCenter;
 import com.cloud.dc.DataCenterVO;
 import com.cloud.dc.Vlan;
+import com.cloud.dc.VlanVO;
 import com.cloud.dc.dao.DataCenterDao;
 import com.cloud.dc.dao.VlanDao;
 import com.cloud.exception.InsufficientAddressCapacityException;
@@ -76,25 +78,34 @@ public class Ipv6AddressManagerImpl extends ManagerBase 
implements Ipv6AddressMa
                if (network == null) {
                        return null;
                }
-       Vlan vlan = _networkModel.getVlanForNetwork(networkId);
-       if (vlan == null) {
+       List<VlanVO> vlans = _vlanDao.listVlansByNetworkId(networkId);
+       if (vlans == null) {
                s_logger.debug("Cannot find related vlan or too many vlan 
attached to network " + networkId);
                return null;
        }
        String ip = null; 
+       Vlan ipVlan = null;
        if (requestedIp6 == null) {
-               if (!_networkModel.isIP6AddressAvailable(networkId)) {
+               if (!_networkModel.isIP6AddressAvailableInNetwork(networkId)) {
                        throw new InsufficientAddressCapacityException("There 
is no more address available in the network " + network.getName(), 
DataCenter.class, network.getDataCenterId());
                }
-               ip = NetUtils.getIp6FromRange(vlan.getIp6Range());
-               int count = 0;
-               while (_ipv6Dao.findByNetworkIdAndIp(networkId, ip) != null) {
-                       ip = NetUtils.getNextIp6InRange(ip, vlan.getIp6Range());
-                       count ++;
-                       // It's an arbitrate number to prevent the infinite 
loop 
-                       if (count > _ipv6RetryMax) {
-                               ip = null;
-                               break;
+               for (Vlan vlan : vlans) {
+                       if 
(!_networkModel.isIP6AddressAvailableInVlan(vlan.getId())) {
+                               continue;
+                       }
+                       ip = NetUtils.getIp6FromRange(vlan.getIp6Range());
+                       int count = 0;
+                       while (_ipv6Dao.findByNetworkIdAndIp(networkId, ip) != 
null) {
+                               ip = NetUtils.getNextIp6InRange(ip, 
vlan.getIp6Range());
+                               count ++;
+                               // It's an arbitrate number to prevent the 
infinite loop 
+                               if (count > _ipv6RetryMax) {
+                                       ip = null;
+                                       break;
+                               }
+                       }
+                       if (ip != null) {
+                               ipVlan = vlan;
                        }
                }
                if (ip == null) {
@@ -102,7 +113,13 @@ public class Ipv6AddressManagerImpl extends ManagerBase 
implements Ipv6AddressMa
                                                DataCenter.class, 
network.getDataCenterId());
                }
        } else {
-               if (!NetUtils.isIp6InRange(requestedIp6, vlan.getIp6Range())) {
+               for (Vlan vlan : vlans) {
+                       if (NetUtils.isIp6InRange(requestedIp6, 
vlan.getIp6Range())) {
+                               ipVlan = vlan;
+                               break;
+                       }
+               }
+               if (ipVlan == null) {
                        throw new CloudRuntimeException("Requested IPv6 is not 
in the predefined range!");
                }
                ip = requestedIp6;
@@ -117,9 +134,9 @@ public class Ipv6AddressManagerImpl extends ManagerBase 
implements Ipv6AddressMa
         _dcDao.update(dc.getId(), dc);
         
        String macAddress = 
NetUtils.long2Mac(NetUtils.createSequenceBasedMacAddress(mac));
-       UserIpv6AddressVO ipVO = new UserIpv6AddressVO(ip, dcId, macAddress, 
vlan.getId());
-       ipVO.setPhysicalNetworkId(vlan.getPhysicalNetworkId());
-       ipVO.setSourceNetworkId(vlan.getNetworkId());
+       UserIpv6AddressVO ipVO = new UserIpv6AddressVO(ip, dcId, macAddress, 
ipVlan.getId());
+       ipVO.setPhysicalNetworkId(network.getPhysicalNetworkId());
+       ipVO.setSourceNetworkId(networkId);
        ipVO.setState(UserIpv6Address.State.Allocated);
        ipVO.setDomainId(owner.getDomainId());
        ipVO.setAccountId(owner.getAccountId());

http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/blob/06acd9f0/server/src/com/cloud/network/NetworkManagerImpl.java
----------------------------------------------------------------------
diff --git a/server/src/com/cloud/network/NetworkManagerImpl.java 
b/server/src/com/cloud/network/NetworkManagerImpl.java
index 65b4b1a..f24623e 100755
--- a/server/src/com/cloud/network/NetworkManagerImpl.java
+++ b/server/src/com/cloud/network/NetworkManagerImpl.java
@@ -3405,11 +3405,7 @@ public class NetworkManagerImpl extends ManagerBase 
implements NetworkManager, L
                if (nic.getIp6Address() == null) {
                        ipv6 = true;
                        UserIpv6Address ip = 
_ipv6Mgr.assignDirectIp6Address(dc.getId(), vm.getOwner(), network.getId(), 
requestedIpv6);
-                       Vlan vlan = 
_networkModel.getVlanForNetwork(network.getId());
-                       if (vlan == null) {
-                               s_logger.debug("Cannot find related vlan or too 
many vlan attached to network " + network.getId());
-                               return;
-                       }
+                       Vlan vlan = _vlanDao.findById(ip.getVlanId());
                        nic.setIp6Address(ip.getAddress().toString());
                        nic.setIp6Gateway(vlan.getIp6Gateway());
                        nic.setIp6Cidr(vlan.getIp6Cidr());

http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/blob/06acd9f0/server/src/com/cloud/network/NetworkModelImpl.java
----------------------------------------------------------------------
diff --git a/server/src/com/cloud/network/NetworkModelImpl.java 
b/server/src/com/cloud/network/NetworkModelImpl.java
index f8d0add..ce8fac7 100644
--- a/server/src/com/cloud/network/NetworkModelImpl.java
+++ b/server/src/com/cloud/network/NetworkModelImpl.java
@@ -527,7 +527,7 @@ public class NetworkModelImpl extends ManagerBase 
implements NetworkModel {
                        return false;
                }
                if (network.getIp6Gateway() != null) {
-                       hasFreeIps = isIP6AddressAvailable(network.getId());
+                       hasFreeIps = 
isIP6AddressAvailableInNetwork(network.getId());
                }
         } else {
             hasFreeIps = (getAvailableIps(network, null)).size() > 0;
@@ -537,17 +537,7 @@ public class NetworkModelImpl extends ManagerBase 
implements NetworkModel {
     }
 
     @Override
-    public Vlan getVlanForNetwork(long networkId) {
-       List<VlanVO> vlans = _vlanDao.listVlansByNetworkId(networkId);
-       if (vlans == null || vlans.size() > 1) {
-               s_logger.debug("Cannot find related vlan or too many vlan 
attached to network " + networkId);
-               return null;
-       }
-       return vlans.get(0);
-    }
-
-    @Override
-    public boolean isIP6AddressAvailable(long networkId) {
+    public boolean isIP6AddressAvailableInNetwork(long networkId) {
        Network network = _networksDao.findById(networkId);
        if (network == null) {
                return false;
@@ -555,8 +545,19 @@ public class NetworkModelImpl extends ManagerBase 
implements NetworkModel {
        if (network.getIp6Gateway() == null) {
                return false;
        }
-       Vlan vlan = getVlanForNetwork(network.getId());
-       long existedCount = _ipv6Dao.countExistedIpsInNetwork(network.getId());
+       List<VlanVO> vlans = _vlanDao.listVlansByNetworkId(networkId);
+       for (Vlan vlan : vlans) {
+               if (isIP6AddressAvailableInVlan(vlan.getId())) {
+                       return true;
+               }
+       }
+               return false;
+       }
+
+    @Override
+    public boolean isIP6AddressAvailableInVlan(long vlanId) {
+       VlanVO vlan = _vlanDao.findById(vlanId);
+       long existedCount = _ipv6Dao.countExistedIpsInVlan(vlanId);
        BigInteger existedInt = BigInteger.valueOf(existedCount);
        BigInteger rangeInt = NetUtils.countIp6InRange(vlan.getIp6Range());
                return (existedInt.compareTo(rangeInt) < 0);

http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/blob/06acd9f0/server/src/com/cloud/network/dao/UserIpv6AddressDao.java
----------------------------------------------------------------------
diff --git a/server/src/com/cloud/network/dao/UserIpv6AddressDao.java 
b/server/src/com/cloud/network/dao/UserIpv6AddressDao.java
index 81e0da8..555b7f6 100644
--- a/server/src/com/cloud/network/dao/UserIpv6AddressDao.java
+++ b/server/src/com/cloud/network/dao/UserIpv6AddressDao.java
@@ -36,4 +36,6 @@ public interface UserIpv6AddressDao extends 
GenericDao<UserIpv6AddressVO, Long>
        List<UserIpv6AddressVO> listByPhysicalNetworkId(long physicalNetworkId);
 
        long countExistedIpsInNetwork(long networkId);
+
+       long countExistedIpsInVlan(long vlanId);
 }

http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/blob/06acd9f0/server/src/com/cloud/network/dao/UserIpv6AddressDaoImpl.java
----------------------------------------------------------------------
diff --git a/server/src/com/cloud/network/dao/UserIpv6AddressDaoImpl.java 
b/server/src/com/cloud/network/dao/UserIpv6AddressDaoImpl.java
index 8a1115d..587193c 100644
--- a/server/src/com/cloud/network/dao/UserIpv6AddressDaoImpl.java
+++ b/server/src/com/cloud/network/dao/UserIpv6AddressDaoImpl.java
@@ -59,6 +59,7 @@ public class UserIpv6AddressDaoImpl extends 
GenericDaoBase<UserIpv6AddressVO, Lo
         CountFreePublicIps = createSearchBuilder(Long.class);
         CountFreePublicIps.select(null, Func.COUNT, null);
         CountFreePublicIps.and("networkId", 
CountFreePublicIps.entity().getSourceNetworkId(), SearchCriteria.Op.EQ);
+        CountFreePublicIps.and("vlanId", 
CountFreePublicIps.entity().getSourceNetworkId(), SearchCriteria.Op.EQ);
         CountFreePublicIps.done();
     }
 
@@ -111,4 +112,11 @@ public class UserIpv6AddressDaoImpl extends 
GenericDaoBase<UserIpv6AddressVO, Lo
         sc.setParameters("networkId", networkId);
         return customSearch(sc, null).get(0);       
        }
+
+       @Override
+       public long countExistedIpsInVlan(long vlanId) {
+        SearchCriteria<Long> sc = CountFreePublicIps.create();
+        sc.setParameters("vlanId", vlanId);
+        return customSearch(sc, null).get(0);       
+       }
 }

http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/blob/06acd9f0/server/test/com/cloud/network/MockNetworkModelImpl.java
----------------------------------------------------------------------
diff --git a/server/test/com/cloud/network/MockNetworkModelImpl.java 
b/server/test/com/cloud/network/MockNetworkModelImpl.java
index 502354e..1088321 100644
--- a/server/test/com/cloud/network/MockNetworkModelImpl.java
+++ b/server/test/com/cloud/network/MockNetworkModelImpl.java
@@ -810,13 +810,13 @@ public class MockNetworkModelImpl extends ManagerBase 
implements NetworkModel {
     }
 
        @Override
-       public Vlan getVlanForNetwork(long networkId) {
+       public boolean isIP6AddressAvailableInNetwork(long networkId) {
                // TODO Auto-generated method stub
-               return null;
+               return false;
        }
 
        @Override
-       public boolean isIP6AddressAvailable(long networkId) {
+       public boolean isIP6AddressAvailableInVlan(long vlanId) {
                // TODO Auto-generated method stub
                return false;
        }

http://git-wip-us.apache.org/repos/asf/incubator-cloudstack/blob/06acd9f0/server/test/com/cloud/vpc/MockNetworkModelImpl.java
----------------------------------------------------------------------
diff --git a/server/test/com/cloud/vpc/MockNetworkModelImpl.java 
b/server/test/com/cloud/vpc/MockNetworkModelImpl.java
index 1080d48..aed54c7 100644
--- a/server/test/com/cloud/vpc/MockNetworkModelImpl.java
+++ b/server/test/com/cloud/vpc/MockNetworkModelImpl.java
@@ -824,13 +824,13 @@ public class MockNetworkModelImpl extends ManagerBase 
implements NetworkModel {
     }
 
        @Override
-       public Vlan getVlanForNetwork(long networkId) {
+       public boolean isIP6AddressAvailableInNetwork(long networkId) {
                // TODO Auto-generated method stub
-               return null;
+               return false;
        }
 
        @Override
-       public boolean isIP6AddressAvailable(long networkId) {
+       public boolean isIP6AddressAvailableInVlan(long vlanId) {
                // TODO Auto-generated method stub
                return false;
        }

Reply via email to