/**
* Copyright (c) Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See License.txt in the project root for
* license information.
*/
package com.microsoft.azure.management.compute.samples;
import com.jcraft.jsch.JSchException;
import com.microsoft.azure.management.Azure;
import com.microsoft.azure.management.compute.CachingTypes;
import com.microsoft.azure.management.compute.Disk;
import com.microsoft.azure.management.compute.KnownLinuxVirtualMachineImage;
import com.microsoft.azure.management.compute.OperatingSystemStateTypes;
import com.microsoft.azure.management.compute.VirtualMachine;
import com.microsoft.azure.management.compute.VirtualMachineCustomImage;
import com.microsoft.azure.management.compute.VirtualMachineDataDisk;
import com.microsoft.azure.management.compute.VirtualMachineSizeTypes;
import com.microsoft.azure.management.resources.fluentcore.arm.Region;
import com.microsoft.azure.management.resources.fluentcore.utils.SdkContext;
import com.microsoft.azure.management.samples.SSHShell;
import com.microsoft.azure.management.samples.Utils;
import com.microsoft.rest.LogLevel;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;
/**
* Azure Compute sample for managing virtual machines -
* - Create an un-managed virtual machine from PIR image with data disks
* - Deallocate the virtual machine
* - Generalize the virtual machine
* - Create a virtual machine custom image from OS and Data disk VHDs of virtual machine
* - Create a second virtual machine using the custom image
* - Create a second virtual machine using the custom image and configure the data disks
* - Deletes the custom image
* - Get SAS Uri to the virtual machine's managed disks.
*/
public final class CreateVirtualMachineUsingCustomImageFromVHD {
/**
* Main function which runs the actual sample.
* @param azure instance of the azure client
* @return true if sample runs successfully
*/
public static boolean runSample(Azure azure) {
final String linuxVMName1 = SdkContext.randomResourceName("VM1", 10);
final String linuxVMName2 = SdkContext.randomResourceName("VM2", 10);
final String linuxVMName3 = SdkContext.randomResourceName("VM3", 10);
final String customImageName = SdkContext.randomResourceName("img", 10);
final String rgName = SdkContext.randomResourceName("rgCOMV", 15);
final String publicIPDnsLabel = SdkContext.randomResourceName("pip", 10);
final String userName = "tirekicker";
final String password = "12NewPA$$w0rd!";
final Region region = Region.US_WEST_CENTRAL;
final String apacheInstallScript = "https://raw.githubusercontent.com/Azure/azure-sdk-for-java/master/azure-samples/src/main/resources/install_apache.sh";
final String apacheInstallCommand = "bash install_apache.sh";
List<String> apacheInstallScriptUris = new ArrayList<>();
apacheInstallScriptUris.add(apacheInstallScript);
try {
//=============================================================
// Create a Linux VM using a PIR image with un-managed OS and data disks and customize virtual
// machine using custom script extension
System.out.println("Creating a un-managed Linux VM");
VirtualMachine linuxVM = azure.virtualMachines().define(linuxVMName1)
.withRegion(region)
.withNewResourceGroup(rgName)
.withNewPrimaryNetwork("10.0.0.0/28")
.withPrimaryPrivateIPAddressDynamic()
.withNewPrimaryPublicIPAddress(publicIPDnsLabel)
.withPopularLinuxImage(KnownLinuxVirtualMachineImage.UBUNTU_SERVER_16_04_LTS)
.withRootUsername(userName)
.withRootPassword(password)
.withUnmanagedDisks()
.defineUnmanagedDataDisk("disk-1")
.withNewVhd(100)
.withLun(1)
.attach()
.defineUnmanagedDataDisk("disk-2")
.withNewVhd(50)
.withLun(2)
.attach()
.defineUnmanagedDataDisk("disk-3")
.withNewVhd(60)
.withLun(3)
.attach()
.defineNewExtension("CustomScriptForLinux")
.withPublisher("Microsoft.OSTCExtensions")
.withType("CustomScriptForLinux")
.withVersion("1.4")
.withMinorVersionAutoUpgrade()
.withPublicSetting("fileUris", apacheInstallScriptUris)
.withPublicSetting("commandToExecute", apacheInstallCommand)
.attach()
.withSize(VirtualMachineSizeTypes.STANDARD_D3_V2)
.create();
System.out.println("Created a Linux VM with un-managed OS and data disks: " + linuxVM.id());
Utils.print(linuxVM);
// De-provision the virtual machine
deprovisionAgentInLinuxVM(linuxVM.getPrimaryPublicIPAddress().fqdn(), 22, userName, password);
//=============================================================
// Deallocate the virtual machine
System.out.println("Deallocate VM: " + linuxVM.id());
linuxVM.deallocate();
System.out.println("De-allocated VM: " + linuxVM.id() + "; state = " + linuxVM.powerState());
//=============================================================
// Generalize the virtual machine
System.out.println("Generalize VM: " + linuxVM.id());
linuxVM.generalize();
System.out.println("Generalized VM: " + linuxVM.id());
//=============================================================
// Create a virtual machine custom image from OS and data disks of the virtual machine
System.out.println("Creating virtual machine custom image from un-managed disk VHDs: " + linuxVM.id());
VirtualMachineCustomImage virtualMachineCustomImage = azure.virtualMachineCustomImages()
.define(customImageName)
.withRegion(region)
.withExistingResourceGroup(rgName)
.withLinuxFromVhd(linuxVM.osUnmanagedDiskVhdUri(), OperatingSystemStateTypes.GENERALIZED)
.defineDataDiskImage()
.withLun(linuxVM.unmanagedDataDisks().get(1).lun())
.fromVhd(linuxVM.unmanagedDataDisks().get(1).vhdUri())
.attach()
.defineDataDiskImage()
.withLun(linuxVM.unmanagedDataDisks().get(2).lun())
.fromVhd(linuxVM.unmanagedDataDisks().get(2).vhdUri())
.attach()
.defineDataDiskImage()
.withLun(linuxVM.unmanagedDataDisks().get(3).lun())
.fromVhd(linuxVM.unmanagedDataDisks().get(3).vhdUri())
.withDiskCaching(CachingTypes.READ_ONLY)
.attach()
.create();
System.out.println("Created custom image");
Utils.print(virtualMachineCustomImage);
//=============================================================
// Create a Linux VM using custom image
System.out.println("Creating a Linux VM using custom image: " + virtualMachineCustomImage.id());
VirtualMachine linuxVM2 = azure.virtualMachines().define(linuxVMName2)
.withRegion(region)
.withExistingResourceGroup(rgName)
.withNewPrimaryNetwork("10.0.0.0/28")
.withPrimaryPrivateIPAddressDynamic()
.withoutPrimaryPublicIPAddress()
.withLinuxCustomImage(virtualMachineCustomImage.id())
.withRootUsername(userName)
.withRootPassword(password)
.withSize(VirtualMachineSizeTypes.STANDARD_D3_V2)
.create();
System.out.println("Created Linux VM");
Utils.print(linuxVM2);
//=============================================================
// Create another Linux VM using custom image and configure the data disks from image and
// add another data disk
VirtualMachine linuxVM3 = azure.virtualMachines().define(linuxVMName3)
.withRegion(region)
.withExistingResourceGroup(rgName)
.withNewPrimaryNetwork("10.0.0.0/28")
.withPrimaryPrivateIPAddressDynamic()
.withoutPrimaryPublicIPAddress()
.withLinuxCustomImage(virtualMachineCustomImage.id())
.withRootUsername(userName)
.withRootPassword(password)
.withNewDataDiskFromImage(1, 200, CachingTypes.READ_WRITE)
.withNewDataDiskFromImage(2, 100, CachingTypes.READ_ONLY)
.withNewDataDiskFromImage(3, 100, CachingTypes.READ_WRITE)
.withNewDataDisk(50)
.withSize(VirtualMachineSizeTypes.STANDARD_D3_V2)
.create();
Utils.print(linuxVM3);
// Getting the SAS URIs requires virtual machines to be de-allocated
// [Access not permitted because'disk' is currently attached to running VM]
//
System.out.println("De-allocating the virtual machine - " + linuxVM3.id());
linuxVM3.deallocate();
//=============================================================
// Get the readonly SAS URI to the OS and data disks
System.out.println("Getting OS and data disks SAS Uris");
// OS Disk SAS Uri
Disk osDisk = azure.disks().getById(linuxVM3.osDiskId());
String osDiskSasUri = osDisk.grantAccess(24 * 60);
System.out.println("OS disk SAS Uri: " + osDiskSasUri);
// Data Disks SAS Uri
for (VirtualMachineDataDisk disk : linuxVM3.dataDisks().values()) {
Disk dataDisk = azure.disks().getById(disk.id());
String dataDiskSasUri = dataDisk.grantAccess(24 * 60);
System.out.println(String.format("Data disk (lun: %d) SAS Uri: %s", disk.lun(), dataDiskSasUri));
}
//=============================================================
// Deleting the custom image
System.out.println("Deleting custom Image: " + virtualMachineCustomImage.id());
azure.virtualMachineCustomImages().deleteById(virtualMachineCustomImage.id());
System.out.println("Deleted custom image");
return true;
} catch (Exception f) {
System.out.println(f.getMessage());
f.printStackTrace();
} finally {
try {
System.out.println("Deleting Resource Group: " + rgName);
azure.resourceGroups().beginDeleteByName(rgName);
System.out.println("Deleted Resource Group: " + rgName);
} catch (NullPointerException npe) {
System.out.println("Did not create any resources in Azure. No clean up is necessary");
} catch (Exception g) {
g.printStackTrace();
}
}
return false;
}
/**
* Main entry point.
* @param args the parameters
*/
public static void main(String[] args) {
try {
//=============================================================
// Authenticate
final File credFile = new File(System.getenv("AZURE_AUTH_LOCATION"));
Azure azure = Azure.configure()
.withLogLevel(LogLevel.BODY)
.authenticate(credFile)
.withDefaultSubscription();
// Print selected subscription
System.out.println("Selected subscription: " + azure.subscriptionId());
runSample(azure);
} catch (Exception e) {
System.out.println(e.getMessage());
e.printStackTrace();
}
}
/**
* De-provision an Azure linux virtual machine.
*
* @param host the public host name
* @param port the ssh port
* @param userName the ssh user name
* @param password the ssh user password
*/
protected static void deprovisionAgentInLinuxVM(String host, int port, String userName, String password) {
SSHShell shell = null;
try {
System.out.println("Trying to de-provision: " + host);
shell = SSHShell.open(host, port, userName, password);
List<String> deprovisionCommand = new ArrayList<>();
deprovisionCommand.add("sudo waagent -deprovision+user --force");
String output = shell.runCommands(deprovisionCommand);
System.out.println(output);
} catch (JSchException jSchException) {
System.out.println(jSchException.getMessage());
} catch (IOException ioException) {
System.out.println(ioException.getMessage());
} catch (Exception exception) {
System.out.println(exception.getMessage());
} finally {
if (shell != null) {
shell.close();
}
}
}
private CreateVirtualMachineUsingCustomImageFromVHD() {
}
}