mirror of
https://github.com/PowerShellMafia/PowerSploit
synced 2026-06-08 12:13:33 +00:00
95 lines
2.6 KiB
PowerShell
95 lines
2.6 KiB
PowerShell
function Get-PEArchitecture
|
|
{
|
|
<#
|
|
.SYNOPSIS
|
|
|
|
Outputs the architecture for which a binary was compiled.
|
|
|
|
PowerSploit Function: Get-PEArchitecture
|
|
Author: Matthew Graeber (@mattifestation)
|
|
License: BSD 3-Clause
|
|
Required Dependencies: None
|
|
Optional Dependencies: None
|
|
|
|
.DESCRIPTION
|
|
|
|
Get-PEArchitecture returns the architecture for which a Windows portable executable was compiled.
|
|
|
|
.PARAMETER Path
|
|
|
|
Path to the executable.
|
|
|
|
.EXAMPLE
|
|
|
|
C:\PS> Get-PEArchitecture C:\Windows\SysWOW64\calc.exe
|
|
X86
|
|
|
|
.EXAMPLE
|
|
|
|
C:\PS> Get-PEArchitecture C:\Windows\System32\cmd.exe
|
|
X64
|
|
|
|
.LINK
|
|
|
|
http://www.exploit-monday.com
|
|
#>
|
|
|
|
Param (
|
|
[Parameter(Position = 0, Mandatory = $True)]
|
|
[String]
|
|
$Path
|
|
)
|
|
|
|
if (!(Test-Path $Path)) {
|
|
Write-Warning 'Invalid path or file does not exist.'
|
|
return
|
|
}
|
|
|
|
# Parse PE header to see if binary was compiled 32 or 64-bit
|
|
$FileStream = New-Object System.IO.FileStream($Path, [System.IO.FileMode]::Open, [System.IO.FileAccess]::Read)
|
|
|
|
[Byte[]] $MZHeader = New-Object Byte[](2)
|
|
$FileStream.Read($MZHeader,0,2) | Out-Null
|
|
|
|
$Header = [System.Text.AsciiEncoding]::ASCII.GetString($MZHeader)
|
|
if ($Header -ne 'MZ') {
|
|
Write-Warning 'Invalid PE header.'
|
|
$FileStream.Close()
|
|
return
|
|
}
|
|
|
|
# Seek to 0x3c - IMAGE_DOS_HEADER.e_lfanew (i.e. Offset to PE Header)
|
|
$FileStream.Seek(0x3c, [System.IO.SeekOrigin]::Begin) | Out-Null
|
|
|
|
[Byte[]] $lfanew = New-Object Byte[](4)
|
|
|
|
# Read offset to the PE Header (will be read in reverse)
|
|
$FileStream.Read($lfanew,0,4) | Out-Null
|
|
$PEOffset = [Int] ('0x{0}' -f (( $lfanew[-1..-4] | % { $_.ToString('X2') } ) -join ''))
|
|
|
|
# Seek to IMAGE_FILE_HEADER.IMAGE_FILE_MACHINE
|
|
$FileStream.Seek($PEOffset + 4, [System.IO.SeekOrigin]::Begin) | Out-Null
|
|
[Byte[]] $IMAGE_FILE_MACHINE = New-Object Byte[](2)
|
|
|
|
# Read compiled architecture
|
|
$FileStream.Read($IMAGE_FILE_MACHINE,0,2) | Out-Null
|
|
$Architecture = '{0}' -f (( $IMAGE_FILE_MACHINE[-1..-2] | % { $_.ToString('X2') } ) -join '')
|
|
$FileStream.Close()
|
|
|
|
if (($Architecture -ne '014C') -and ($Architecture -ne '8664') -and ($Architecture -ne '01C4')) {
|
|
Write-Warning 'Invalid PE header or unsupported architecture.'
|
|
return
|
|
}
|
|
|
|
if ($Architecture -eq '014C') {
|
|
return 'X86'
|
|
} elseif ($Architecture -eq '8664') {
|
|
return 'X64'
|
|
} elseif ($Architecture -eq '01C4') {
|
|
return 'ARM'
|
|
} else {
|
|
return 'OTHER'
|
|
}
|
|
|
|
}
|