@Override public boolean apply(Allocation allocInfo) throws MetadataException { if (allocInfo.getRequest().getKeyName() == null || "".equals(allocInfo.getRequest().getKeyName())) { if (ImageMetadata.Platform.windows.equals( allocInfo.getBootSet().getMachine().getPlatform())) { throw new InvalidMetadataException( "You must specify a keypair when running a windows vm: " + allocInfo.getRequest().getImageId()); } else { allocInfo.setSshKeyPair(KeyPairs.noKey()); return true; } } Context ctx = allocInfo.getContext(); RunInstancesType request = allocInfo.getRequest(); String keyName = request.getKeyName(); SshKeyPair key = KeyPairs.lookup(ctx.getUserFullName().asAccountFullName(), keyName); if (!ctx.hasAdministrativePrivileges() && !RestrictedTypes.filterPrivileged().apply(key)) { throw new IllegalMetadataAccessException( "Not authorized to use keypair " + keyName + " by " + ctx.getUser().getName()); } allocInfo.setSshKeyPair(key); return true; }
@Override public void allocate(Allocation allocInfo) throws Exception { Partition reqPartition = allocInfo.getPartition(); String zoneName = reqPartition.getName(); String vmTypeName = allocInfo.getVmType().getName(); /* Validate min and max amount */ final int minAmount = allocInfo.getMinCount(); final int maxAmount = allocInfo.getMaxCount(); if (minAmount > maxAmount) throw new RuntimeException( "Maximum instance count must not be smaller than minimum instance count"); /* Retrieve our context and list of clusters associated with this zone */ List<Cluster> authorizedClusters = this.doPrivilegedLookup(zoneName, vmTypeName); int remaining = maxAmount; int allocated = 0; int available; LOG.info( "Found authorized clusters: " + Iterables.transform(authorizedClusters, HasName.GET_NAME)); /* Do we have any VM available throughout our clusters? */ if ((available = checkAvailability(vmTypeName, authorizedClusters)) < minAmount) { throw new NotEnoughResourcesException( "Not enough resources (" + available + " in " + zoneName + " < " + minAmount + "): vm instances."); } else { for (Cluster cluster : authorizedClusters) { if (remaining <= 0) { break; } else { ResourceState state = cluster.getNodeState(); Partition partition = cluster.getConfiguration().lookupPartition(); /* Has a partition been set if the AZ was not specified? */ if (allocInfo.getPartition().equals(Partition.DEFAULT)) { /* * Ok, do we have enough slots in this partition to support our request? We should have at least * the minimum. The list is sorted in order of resource availability from the cluster with the most * available to the cluster with the least amount available. This is why we don't check against the * maxAmount value since its a best effort at this point. If we select the partition here and we * can't fit maxAmount, based on the sorting order, the next partition will not fit maxAmount anyway. */ int zoneAvailable = checkZoneAvailability(vmTypeName, partition, authorizedClusters); if (zoneAvailable < minAmount) continue; /* Lets use this partition */ allocInfo.setPartition(partition); } else if (!allocInfo.getPartition().equals(partition)) { /* We should only pick clusters that are part of the selected AZ */ continue; } if (allocInfo.getBootSet().getMachine() instanceof BlockStorageImageInfo) { try { Topology.lookup(Storage.class, partition); } catch (Exception ex) { allocInfo.abort(); allocInfo.setPartition(reqPartition); throw new NotEnoughResourcesException( "Not enough resources: Cannot run EBS instances in partition w/o a storage controller: " + ex.getMessage(), ex); } } try { int tryAmount = (remaining > state.getAvailability(vmTypeName).getAvailable()) ? state.getAvailability(vmTypeName).getAvailable() : remaining; List<ResourceToken> tokens = this.requestResourceToken(allocInfo, tryAmount, maxAmount); remaining -= tokens.size(); allocated += tokens.size(); } catch (Exception t) { LOG.error(t); Logs.extreme().error(t, t); allocInfo.abort(); allocInfo.setPartition(reqPartition); /* if we still have some allocation remaining AND no more resources are available */ if (((available = checkZoneAvailability(vmTypeName, partition, authorizedClusters)) < remaining) && (remaining > 0)) { throw new NotEnoughResourcesException( "Not enough resources (" + available + " in " + zoneName + " < " + minAmount + "): vm instances.", t); } else { throw new NotEnoughResourcesException(t.getMessage(), t); } } } } /* Were we able to meet our minimum requirements? */ if ((allocated < minAmount) && (remaining > 0)) { allocInfo.abort(); allocInfo.setPartition(reqPartition); if (reqPartition.equals(Partition.DEFAULT)) { throw new NotEnoughResourcesException( "Not enough resources available in all zone for " + minAmount + "): vm instances."); } else { available = checkZoneAvailability(vmTypeName, reqPartition, authorizedClusters); throw new NotEnoughResourcesException( "Not enough resources (" + available + " in " + zoneName + " < " + minAmount + "): vm instances."); } } } }
@Override public boolean apply(Allocation allocInfo) throws MetadataException { BootableImageInfo imageInfo = allocInfo.getBootSet().getMachine(); final ArrayList<BlockDeviceMappingItemType> instanceMappings = allocInfo.getRequest().getBlockDeviceMapping() != null ? allocInfo.getRequest().getBlockDeviceMapping() : new ArrayList<BlockDeviceMappingItemType>(); List<DeviceMapping> imageMappings = new ArrayList<DeviceMapping>(((ImageInfo) imageInfo).getDeviceMappings()); // Figure out the final set of mappings for this instance and add it to the request before // verifying the mappings. // for ( DeviceMapping imageMapping : imageMappings ) { // if( !Iterables.any( instanceMappings, Images.findBlockDeviceMappingItempType( // imageMapping.getDeviceName() ) ) ) { // instanceMappings.add(Images.DeviceMappingDetails.INSTANCE.apply(imageMapping)); // } // } // Is this an overkill?? Should I rather go with the above logic. Damn complexity seems // same... // probably m * n where m is the number of image mappings and n is the number of instance // mappings instanceMappings.addAll( Lists.transform( Lists.newArrayList( Iterables.filter( imageMappings, new Predicate<DeviceMapping>() { @Override public boolean apply(DeviceMapping arg0) { return !Iterables.any( instanceMappings, Images.findBlockDeviceMappingItempType(arg0.getDeviceName())); } })), Images.DeviceMappingDetails.INSTANCE)); if (imageInfo instanceof BlockStorageImageInfo) { // bfebs image if (!instanceMappings.isEmpty()) { // Verify all block device mappings. Dont fuss if both snapshot id and volume size are // left blank Images.isDeviceMappingListValid(instanceMappings, Boolean.TRUE, Boolean.TRUE); BlockStorageImageInfo bfebsImage = (BlockStorageImageInfo) imageInfo; Integer imageSizeGB = (int) (bfebsImage.getImageSizeBytes() / BYTES_PER_GB); Integer userRequestedSizeGB = null; // Find the root block device mapping in the run instance request. Validate it BlockDeviceMappingItemType rootBlockDevice = Iterables.find( instanceMappings, Images.findEbsRootOptionalSnapshot(bfebsImage.getRootDeviceName()), null); if (rootBlockDevice != null) { // Ensure that root device is not mapped to a different snapshot, logical device or // suppressed. // Verify that the root device size is not smaller than the image size if (StringUtils.isNotBlank(rootBlockDevice.getEbs().getSnapshotId()) && !StringUtils.equals( rootBlockDevice.getEbs().getSnapshotId(), bfebsImage.getSnapshotId())) { throw new InvalidMetadataException( "Snapshot ID cannot be modified for the root device. " + "Source snapshot from the image registration will be used for creating the root device"); } else if (StringUtils.isNotBlank(rootBlockDevice.getVirtualName())) { throw new InvalidMetadataException( "Logical type cannot be modified for the root device. " + "Source snapshot from the image registration will be used for creating the root device"); } else if (rootBlockDevice.getNoDevice() != null && rootBlockDevice.getNoDevice()) { throw new InvalidMetadataException( "Root device cannot be suppressed. " + "Source snapshot from the image registration will be used for creating the root device"); } else if ((userRequestedSizeGB = rootBlockDevice.getEbs().getVolumeSize()) != null && userRequestedSizeGB < imageSizeGB) { throw new InvalidMetadataException( "Root device volume cannot be smaller than the image size"); } // Gather all the information for the root device mapping and populate it in the run // instance request if (rootBlockDevice.getEbs().getSnapshotId() == null) { rootBlockDevice.getEbs().setSnapshotId(bfebsImage.getSnapshotId()); } if (rootBlockDevice.getEbs().getVolumeSize() == null) { rootBlockDevice.getEbs().setVolumeSize(imageSizeGB); } if (rootBlockDevice.getEbs().getDeleteOnTermination() == null) { rootBlockDevice.getEbs().setDeleteOnTermination(bfebsImage.getDeleteOnTerminate()); } } else { // This should never happen. Root device mapping will always exist in the block storage // image and or run instance request throw new InvalidMetadataException("Root block device mapping not found\n"); } } } else { // Instance store image // Verify all block device mappings. EBS mappings must be considered invalid since AWS // doesn't support it Images.isDeviceMappingListValid(instanceMappings, Boolean.TRUE, Boolean.FALSE); } // Set the final list of block device mappings in the run instance request (necessary if the // instance mappings were null). Checked with grze that its okay allocInfo.getRequest().setBlockDeviceMapping(instanceMappings); return true; }