Upload files to "host_scripts"

host_scripts/grabhandshakes.ps1

@@ -0,0 +1,109 @@
+# Define the path to the config file (adjust the path as needed)
+$configFilePath = "$PSScriptRoot\config.ini"
+
+# Function to create the config file if it doesn't exist
+function Create-ConfigFile {
+    param (
+        [string]$filePath
+    )
+    
+    Write-Host "Configuration file not found. Let's create one."
+
+    $ip = Read-Host "Enter the IP address of the remote machine"
+    $username = Read-Host "Enter the username"
+    $password = Read-Host "Enter the password"
+
+    $configContent = @"
+[Connection]
+ip = $ip
+username = $username
+password = $password
+"@
+
+    $configContent | Out-File -FilePath $filePath
+    Write-Host "Configuration file created at $filePath"
+}
+
+# Function to read the config file and get values
+function Get-ConfigValue {
+    param (
+        [string]$section,
+        [string]$key
+    )
+    $ini = Get-Content $configFilePath | Out-String
+    $ini = $ini -replace '^\s*;.*$', '' # Remove comments
+    $ini = $ini -replace '^\s*#.*$', '' # Remove comments
+    $ini = $ini -replace '\s*=\s*', '=' # Remove spaces around equal sign
+
+    $sectionData = ($ini -split "\r?\n" -replace '^\[([^\]]+)\]\s*$', '$1' -ne $section) -split "\r?\n"
+    foreach ($line in $sectionData) {
+        if ($line -match "^\s*$key\s*=\s*(.*)$") {
+            return $matches[1]
+        }
+    }
+}
+
+# Check if the config file exists, if not, prompt user to create it
+if (-not (Test-Path -Path $configFilePath)) {
+    Create-ConfigFile -filePath $configFilePath
+}
+
+# Read config values
+$ip = Get-ConfigValue -section "Connection" -key "ip"
+$username = Get-ConfigValue -section "Connection" -key "username"
+$password = Get-ConfigValue -section "Connection" -key "password"
+
+# Convert the plain password to a secure string for further use
+$securePassword = ConvertTo-SecureString $password -AsPlainText -Force
+
+# Create the local directory if it doesn't exist
+$remotePath = "/root/handshakes"  # You can make this dynamic as well
+$localFolder = "$([Environment]::GetFolderPath('Desktop'))\$([System.IO.Path]::GetFileName($remotePath))"
+
+if (-not (Test-Path -Path $localFolder)) {
+    New-Item -ItemType Directory -Path $localFolder | Out-Null
+}
+
+# Construct the SCP command using the password and SSH command to avoid password prompt
+$scpCommand = "scp -o PreferredAuthentications=password -o PubkeyAuthentication=no ${username}@${ip}:${remotePath} ${localFolder}"
+$sshCommand = "ssh -o PreferredAuthentications=password -o PubkeyAuthentication=no ${username}@${ip} sudo rm -rf ${remotePath}"
+
+# Execute the SCP command by passing the password through a pipeline
+$processInfo = New-Object System.Diagnostics.ProcessStartInfo
+$processInfo.FileName = "cmd.exe"
+$processInfo.RedirectStandardInput = $true
+$processInfo.RedirectStandardOutput = $true
+$processInfo.RedirectStandardError = $true
+$processInfo.UseShellExecute = $false
+$processInfo.Arguments = "/c $scpCommand"
+$process = New-Object System.Diagnostics.Process
+$process.StartInfo = $processInfo
+$process.Start() | Out-Null
+
+# Pass the password to the command
+$process.StandardInput.WriteLine($password)
+$process.StandardInput.Close()
+
+$process.WaitForExit()
+
+# Check if the SCP transfer was successful
+if ($process.ExitCode -eq 0) {
+    # Execute the SSH command by passing the password through a pipeline
+    $processInfo.Arguments = "/c $sshCommand"
+    $process = New-Object System.Diagnostics.Process
+    $process.StartInfo = $processInfo
+    $process.Start() | Out-Null
+
+    # Pass the password to the command
+    $process.StandardInput.WriteLine($password)
+    $process.StandardInput.Close()
+
+    $process.WaitForExit()
+    
+    Write-Host "Files transferred and removed from the remote machine successfully."
+} else {
+    Write-Host "SCP transfer failed. Files were not removed from the remote machine."
+}
+
+# Clear the plain password from memory
+$securePassword = $null

host_scripts/win_connection_share.ps1

@@ -0,0 +1,290 @@
+<#
+    .SYNOPSIS
+        A script that setups Internet Connection Sharing for Pwnagotchi.
+    
+    .DESCRIPTION
+        A script that setups Internet Connection Sharing for Pwnagotchi. 
+
+        Note: Internet Connection Sharing on Windows can be a bit unstable on between reboots.
+              You might need to run this script occasionally to disable and re-enable Internet Connection Sharing.
+    
+    .PARAMETER EnableInternetConnectionSharing
+        Enable Internet Connection Sharing
+    
+    .PARAMETER DisableInternetConnectionSharing
+        Disable Internet Connection Sharing
+    
+    .PARAMETER SetPwnagotchiSubnet
+        Change the Internet Connection Sharing subnet to the Pwnagotchi subnet. The USB Gadget Interface IP will default to 10.0.0.1.
+    
+    .PARAMETER ScopeAddress
+        Custom ScopeAddress (The IP Address of the USB Gadget Interface.)
+    
+    .EXAMPLE
+        # Enable Internet Connection Sharing
+        PS C:\> .\win_connection_share -EnableInternetConnectionSharing
+
+    .EXAMPLE
+        # Disable Internet Connection Sharing
+        PS C:\> .\win_connection_share -DisableInternetConnectionSharing
+
+    .EXAMPLE
+        # Change the regkeys of Internet Connection Sharing to the Pwnagotchi Subnet
+        PS C:\> .\win_connection_share -SetPwnagotchiSubnet
+
+    .EXAMPLE
+        # Change the regkeys of Internet Connection Sharing to the Pwnagotchi Subnet with a custom ScopeAddress (The IP Address of the USB Gadget Interface.)
+        PS C:\> .\win_connection_share -SetPwnagotchiSubnet -ScopeAddress 10.0.0.10
+#>
+
+#Requires -Version 5
+#Requires -RunAsAdministrator
+[Cmdletbinding()]
+Param (
+    [switch]$EnableInternetConnectionSharing,
+    [switch]$DisableInternetConnectionSharing,
+    [switch]$SetPwnagotchiSubnet,
+    [ipaddress]$ScopeAddress = '10.0.0.1'
+)
+
+# Load helper functions
+Function Create-HNetObjects {
+    <#
+    .SYNOPSIS
+        A helper function that does the heavy lifting with NetCfg.HNetShare
+    
+    .DESCRIPTION
+        A helper function that does the heavy lifting with NetCfg.HNetShare. This returns a PSObject containing the `INetSharingConfigurationForINetConnection` info of 2 Adapters.
+    
+    .PARAMETER InternetAdaptor
+        The output of Get-NetAdaptor filtered down to the 'main' uplink interface.
+    
+    .PARAMETER RNDISGadget
+        The output of Get-NetAdaptor filtered down to the 'USB Ethernet/RNDIS Gadget' interface.
+    
+    .EXAMPLE
+        PS> $HNetObject = Create-HNetObjects
+        PS> $HNetObject
+            RNDISIntConfig     InternetIntConfig
+            --------------     -----------------
+            System.__ComObject System.__ComObject
+    #>
+    [Cmdletbinding()]
+    Param (
+        $InternetAdaptor = $(Select-NetAdaptor -Message "Please select your main a ethernet adaptor with internet access that will be used for internet sharing."),
+        $RNDISGadget = $(Select-NetAdaptor -Message "Please select your 'USB Ethernet/RNDIS Gadget' adaptor")
+    )
+    Begin {
+        regsvr32.exe /s hnetcfg.dll
+        $HNetShare = New-Object -ComObject HNetCfg.HNetShare
+    }
+    Process {
+        if ($HNetShare.EnumEveryConnection -ne $null) {
+            $InternetInt       = $HNetShare.EnumEveryConnection | Where-Object { $HNetShare.NetConnectionProps.Invoke($_).Name -eq ($InternetAdaptor).Name }
+            $InternetIntConfig = $HNetShare.INetSharingConfigurationForINetConnection.Invoke($InternetInt)
+            $RNDISInt          = $HNetShare.EnumEveryConnection | Where-Object { $HNetShare.NetConnectionProps.Invoke($_).Name -eq ($RNDISGadget).Name }
+            $RNDISIntConfig    = $HNetShare.INetSharingConfigurationForINetConnection.Invoke($RNDISInt)
+        }
+    }
+    End {
+        Return $(New-Object -TypeName PSObject -Property @{InternetIntConfig=$InternetIntConfig;RNDISIntConfig=$RNDISIntConfig;})
+    }
+}
+Function Enable-InternetConnectionSharing {
+    <#
+    .SYNOPSIS
+        Enables internet connection sharing between the 'main' uplink interface and the 'USB Ethernet/RNDIS Gadget' interface.
+    
+    .DESCRIPTION
+        Enables internet connection sharing between the 'main' uplink interface and the 'USB Ethernet/RNDIS Gadget' interface.
+    
+    .EXAMPLE
+        PS> Enable-InternetConnectionSharing
+    
+    #>
+    [Cmdletbinding()]
+    $HNetObject = Create-HNetObjects
+    $HNetObject.InternetIntConfig.EnableSharing(0)
+    $HNetObject.RNDISIntConfig.EnableSharing(1)
+    Write-Output "[x] Enabled Internet Connection Sharing."
+}
+Function Disable-InternetConnectionSharing {
+    <#
+    .SYNOPSIS
+        Disables internet connection sharing between the 'main' uplink interface and the 'USB Ethernet/RNDIS Gadget' interface.
+    
+    .DESCRIPTION
+        Disables internet connection sharing between the 'main' uplink interface and the 'USB Ethernet/RNDIS Gadget' interface.
+
+    .EXAMPLE
+        PS> Disable-InternetConnectionSharing
+    
+    #>
+    [Cmdletbinding()]
+    $HNetObject = $(Create-HNetObjects)
+    $HNetObject.InternetIntConfig.DisableSharing()
+    $HNetObject.RNDISIntConfig.DisableSharing()
+    Write-Output "[x] Disabled Internet Connection Sharing."
+}
+Function Test-PwnagotchiSubnet {
+    <#
+    .SYNOPSIS
+        Tests the registry for the correct ScopeAddress.
+    
+    .DESCRIPTION
+        Tests the registry for the correct ScopeAddress. By default windows uses a 192.168.137.x subnet for Internet Connection Sharing. This value can be changed
+        in the registry.
+    
+    .EXAMPLE
+        PS> Test-PwnagotchiSubnet
+        [!] By default Internet Connection Sharing uses a 192.168.137.x subnet. Run Set-PwnagotchiSubnet to ensure you and your little friend are on the same subnet.
+    #>
+    [Cmdletbinding()]
+    $RegKeys = Get-ItemProperty HKLM:\SYSTEM\CurrentControlSet\Services\SharedAccess\Parameters -ErrorAction Stop
+    If ($RegKeys.ScopeAddress -notmatch '10.0.0.') {
+        Write-Error "By default Internet Connection Sharing uses a 192.168.137.x subnet. Run Set-PwnagotchiSubnet to ensure you and your little friend are on the same subnet." -ErrorAction Stop
+    }
+    If ($RegKeys.ScopeAddressBackup -notmatch '10.0.0.') {
+        Write-Error "By default Internet Connection Sharing uses a 192.168.137.x subnet. Run Set-PwnagotchiSubnet to ensure you and your little friend are on the same subnet." -ErrorAction Stop
+    } 
+}
+Function Set-PwnagotchiSubnet {
+    <#
+    .SYNOPSIS
+        Set the registry for the correct ScopeAddress.
+    
+    .DESCRIPTION
+        Set the registry for the correct ScopeAddress. By default windows uses a 192.168.137.x subnet for Internet Connection Sharing. This value can be changed
+        in the registry. By default it will be changed to 10.0.0.1
+    
+    .PARAMETER ScopeAddress
+        The IP address the USB Gadget interface should use.
+    
+    .EXAMPLE
+    Set-PwnagotchiSubnet
+    
+    #>
+    [Cmdletbinding()]
+    Param (
+        $ScopeAddress = '10.0.0.1'
+    )
+    Try {
+        [void]([ipaddress]$ScopeAddress)
+        [void]([byte[]] $ScopeAddress.split('.'))
+    } Catch {
+        Write-Error "$ScopeAddress is not a valid IP."
+    }
+    Try {
+        Set-ItemProperty -Name ScopeAddress       -Path "HKLM:\SYSTEM\CurrentControlSet\Services\SharedAccess\Parameters\" -Value $ScopeAddress -ErrorAction Stop
+        Set-ItemProperty -Name ScopeAddressBackup -Path "HKLM:\SYSTEM\CurrentControlSet\Services\SharedAccess\Parameters\" -Value $ScopeAddress -ErrorAction Stop
+        Write-Warning "The Internet Connection Sharing subnet has been updated. A reboot of windows is required !"
+    } Catch {
+        $PSCmdlet.ThrowTerminatingError($PSItem)
+    }
+ 
+}
+
+# Main Function
+Function Setup-PwnagotchiNetwork {
+    <#
+    .SYNOPSIS
+        Function to setup networking.
+    
+    .DESCRIPTION
+        Function to setup networking. Main function calls helpers functions.
+    
+    .PARAMETER EnableInternetConnectionSharing
+        Enable Internet Connection Sharing
+    
+    .PARAMETER DisableInternetConnectionSharing
+        Disable Internet Connection Sharing
+    
+    .PARAMETER SetPwnagotchiSubnet
+        Change the Internet Connection Sharing subnet to the Pwnagotchi. Defaults to 10.0.0.1.
+    
+    .PARAMETER ScopeAddress
+        Custom ScopeAddress (the ICS ip address)
+    
+    .EXAMPLE
+        PS> Setup-PwnagotchiNetwork -EnableInternetConnectionSharing
+    
+    #>
+    
+    Param (
+        [switch]$EnableInternetConnectionSharing,
+        [switch]$DisableInternetConnectionSharing,
+        [switch]$SetPwnagotchiSubnet,
+        $ScopeAddress = '10.0.0.1'
+    )
+    Begin {
+        Try {
+            Write-Debug "Begin"
+            $ErrorSplat=@{ErrorAction="stop"}
+            Write-Debug "Testing subnet"
+            Try {
+                Test-PwnagotchiSubnet @ErrorSplat
+            } Catch {
+                If ($SetPwnagotchiSubnet) {
+                    Write-Debug "Setting subnet"
+                    Set-PwnagotchiSubnet -ScopeAddress $ScopeAddress @ErrorSplat
+                } Else {
+                    Write-Error "By default Internet Connection Sharing uses a 192.168.137.x subnet. Run this script with the -SetPwnagotchiSubnet to setup the network." -ErrorAction Stop
+                }
+            }
+        } Catch {
+            $PSCmdlet.ThrowTerminatingError($PSItem)
+        }
+    }
+    Process {
+        Write-Debug "Process"
+        Try {
+            If ($EnableInternetConnectionSharing) {
+                Write-Debug "Enable network Sharing"
+                Enable-InternetConnectionSharing @ErrorSplat
+            } ElseIf ($DisableInternetConnectionSharing) {
+                Write-Debug "Disable network Sharing"
+                Disable-InternetConnectionSharing @ErrorSplat
+            }
+        } Catch {
+            $PSCmdlet.ThrowTerminatingError($PSItem)
+        }
+    }
+    End {
+        Write-Debug "End"
+        Try {
+            # Nothing to return.
+        } Catch {
+            $PSCmdlet.ThrowTerminatingError($PSItem)
+        }
+    }
+}
+Function Select-NetAdaptor {
+    <#
+    .SYNOPSIS
+        A menu function to select the correct network adaptors.
+    
+    .DESCRIPTION
+        A menu function to select the correct network adaptors.
+    
+    .PARAMETER Message
+        Message that will be displayed during the question.
+    
+    #>
+    
+    Param (
+        $Message
+    )
+    $Adaptors = Get-NetAdapter | Where-Object {$_.MediaConnectionState -eq 'Connected'} | Sort-Object LinkSpeed -Descending
+    do { 
+        Write-Host $Message
+        $index = 1
+        foreach ($Adaptor in $Adaptors) {
+            Write-Host "[$index] $($Adaptor.Name), $($Adaptor.InterfaceDescription)"
+            $index++
+        }   
+        $Selection = Read-Host "Number"
+    } until ($Adaptors[$selection-1])
+     Return $Adaptors[$selection-1]
+}
+# Dynamically create params for Setup-PwnagotchiNetwork function based of param input of script.
+Setup-PwnagotchiNetwork @psBoundParameters