vSphereAutoLab/Build/Automate/PSFunctions.ps1

212 lines
7.0 KiB
PowerShell

# Functions used in AutoLab PowerShell scripts
#
#
# Version 2.6
#
# This Function from the PowerCLI Blog
# http://blogs.vmware.com/vipowershell/2009/08/how-to-list-datastores-that-are-on-shared-storage.html
#
function Write-BuildLog {
param([string]$message)
if (!(Test-Path C:\Buildlog.txt)) {
New-Item -Path C:\Buildlog.txt -ItemType File
}
Write-Host $message
$message | Out-File C:\BuildLog.txt -Encoding Default -append
}
function Get-ShareableDatastore {
# Get all datastores.
$datastores = Get-Datastore
# Load the HostStorageSystems of all hosts.
$hosts = Get-VMHost | Get-View -property ConfigManager
$storageSystems = @()
foreach ($h in $hosts) {
$sdi = Get-View $h.ConfigManager.StorageSystem -Property StorageDeviceInfo
Write-Debug ("GSD: SDI for host $h is " + $sdi)
$storageSystems += $sdi
}
foreach ($dso in $datastores) {
$ds = $dso | Get-View -Property Info
# Check if this datastore is NFS.
$dsInfo = $ds.Info
Write-Debug ("GSD: Is it NFS? " + $dsInfo.getType())
if ($dsInfo -is [VMware.Vim.NasDatastoreInfo]) {
Write-Output $dso
continue
}
# Get the first extent of the datastore.
$firstExtent = $dsInfo.Vmfs.Extent[0]
Write-Debug ("GSD: first extent: " + $firstExtent.DiskName)
# Find a host that maps this LUN.
foreach ($hss in $storageSystems) {
$lun = $hss.StorageDeviceInfo.ScsiLun | Where {$_.CanonicalName -eq $firstExtent.DiskName }
if ($lun) {
Write-Debug ("GSD: found " + $lun.DeviceName + " on " + $hss.MoRef.Value)
Write-Debug ("GSD: LUN details: Name:" + $lun.DisplayName + ", Type:" + $lun.DeviceType + ", Vendor:" + $lun.Vendor + ", Model:" + $lun.Model)
# Search the adapter topology of this host, looking for the LUN.
$adapterTopology = $hss.StorageDeviceInfo.ScsiTopology.Adapter |
Where {$_.Target |
Where {$_.Lun |
Where {$_.ScsiLun -eq $lun.key }
}
} | Select -First 1
# We've found a host that has this LUN. Find how it maps to an adapter.
$adapter = $hss.StorageDeviceInfo.HostBusAdapter | Where {$_.Key -eq $adapterTopology.Adapter }
Write-Debug ("GSD: HBA type is: " + $adapter.getType())
# It's shared if it's Fibre Channel or iSCSI (we checked for NFS earlier)
if ($adapter -is [VMware.Vim.HostFibreChannelHba] -or $adapter -is [VMware.Vim.HostInternetScsiHba]) {
Write-Debug "GSO: $dso is sharable"
Write-Output $dso
}
# Otherwise it's not shared and we quit walking through hosts.
break
}
}
}
}
Function Check-File ($a, $b) {
if (test-Path $a) {
Write-Host ($b + " found.") -foregroundcolor "green"
} else {
Write-Host ($b + " missing.") -foregroundcolor "red"
$Global:Pass = $False
}
}
Function Check-OptionalFile ($a, $b) {
if (test-Path $a) {
Write-Host ($b + " found.") -foregroundcolor "green"
} else {
Write-Host ($b + " missing.") -foregroundcolor "Yellow"
}
}
Function Check-ServiceRunning ($a) {
$SVC = Get-Service -name $a -ErrorAction "SilentlyContinue"
if ($SVC.Status -eq "Running") {
Write-Host ("Service " + $a + " running.") -foregroundcolor "green"
} else {
Write-Host ("Service " + $a + " not running.") -foregroundcolor "red"
$Global:Pass = $False
}
}
Function Check-DNSRecord ($a) {
$FWDIP = ""
$ErrorActionPreference = "silentlycontinue"
$FWDIP = [System.Net.Dns]::GetHostAddresses($a)
$ErrorActionPreference = "continue"
if ($FWDIP -eq "") {
Write-Host ("No DNS for " + $a ) -foregroundcolor "red"
$Global:Pass = $False
} else {
Write-Host ("DNS OK for " + $a) -foregroundcolor "green"
}
}
function Test-Administrator {
$user = [Security.Principal.WindowsIdentity]::GetCurrent();
(New-Object Security.Principal.WindowsPrincipal $user).IsInRole([Security.Principal.WindowsBuiltinRole]::Administrator)
}
#
# This section from Luc Dekens http://www.lucd.info/2011/08/11/vmx-raiders-revisited/
#
Function Register-VMs ($a){
# Collect .vmx paths of registered VMs on the datastore
$registered = @{}
$Datastore = Get-Datastore -name $a
Get-VM -Datastore $Datastore | %{$_.Extensiondata.LayoutEx.File | where {$_.Name -like "*.vmx"} | %{$registered.Add($_.Name,$true)}}
# Set up Search for .VMX Files in Datastore
New-PSDrive -Name TgtDS -Location $Datastore -PSProvider VimDatastore -Root '\' | Out-Null
$unregistered = @(Get-ChildItem -Path TgtDS: -Recurse | where {$_.FolderPath -notmatch ".snapshot" -and $_.Name -like "*.vmx" -and !$registered.ContainsKey($_.DatastoreFullPath)})
Remove-PSDrive -Name TgtDS
#Register all .vmx Files as VMs on the datastore
foreach($VMXFile in $unregistered) {$null= New-VM -VMFilePath $VMXFile.DatastoreFullPath -VMHost $vmhost -Location "vm" -RunAsync}
}
#
# This section from dEPLOYMENT gUYS BLOG http://blogs.technet.com/b/deploymentguys/archive/2010/07/15/reading-and-modifying-ini-files-with-scripts.aspx
#
function Convert-IniFile ($file) {
$REGEX_INI_COMMENT_STRING = ";"
$REGEX_INI_SECTION_HEADER = "^\s*(?!$($REGEX_INI_COMMENT_STRING))\s*\[\s*(.*[^\s*])\s*]\s*$"
$REGEX_INI_KEY_VALUE_LINE = "^\s*(?!$($REGEX_INI_COMMENT_STRING))\s*([^=]*)\s*=\s*(.*)\s*$"
$ini = @{}
switch -regex -file $file {
"$($REGEX_INI_SECTION_HEADER)" {
$section = $matches[1]
$ini[$section] = @{}
}
"$($REGEX_INI_KEY_VALUE_LINE)" {
$name,$value = $matches[1..2]
if ($name -ne $null -and $section -ne $null)
{$ini[$section][$name] = $value }
}
}
$ini
}
#Menu selection from http://poshtips.com/2011/09/03/howto-make-menus-in-powershell/
function isNumeric ($x) {
$x2 = 0
$isNum = [System.Int32]::TryParse($x, [ref]$x2)
return $isNum
}
function SelectFromList {
param([string[]]$List,[string]$Title="Choices",[switch]$verbose=$false)
Write-Host $Title.padright(80) -back green -fore black
$digits = ([string]$List.length).length
$fmt = "{0,$digits}"
#display selection list
for ($LN=0; $LN -lt $List.length) {
Write-Host (" $fmt $($List[$LN])" -f ++$ln)
Write-Host " "
}
#query user until valid selection is made
do {
Write-Host (" Please select from list (1 to {0}) or `"q`" to quit" -f ($list.length)) -back black -fore green -nonewline
$sel = read-host " "
if ($sel -eq "q") {
Write-Host " quitting selection per user request..." -back black -fore yellow
Exit
} elseif (isNumeric $sel) {
if (([int]$sel -gt 0) -and ([int]$sel -le $list.length)) {
if ($verbose) {
Write-Host (" You selected item #{0} ({1})" -f $sel,$List[$sel-1]) -back black -fore green}
} else {
$sel = $null
}
}
else {$sel = $null}
} until ($sel)
if (isNumeric $sel) {
$sel -1
} else {
$null
}
}
Function Get-VMPlatform ($VMType) {
If ((Get-WmiObject Win32_BIOS).Manufacturer.Tolower() -like "bochs*") {$VMType = "Ravello"}
Elseif ((Get-WmiObject Win32_BIOS).Serialnumber.Tolower() -like "vmware*") {$VMType = "VMware"}
Else {$VMType = "Unknown"}
return $VMType
}