212 lines
7.0 KiB
PowerShell
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
|
|
} |