189 lines
7.3 KiB
PowerShell
189 lines
7.3 KiB
PowerShell
# $Global:TestDomain = "www.google.com" #测试代理是否可用
|
|
# $Global:TestIP = "223.5.5.5" #测试是否连接网络
|
|
# $Global:originalDns = "223.5.5.5" #默认DNS
|
|
# $Global:originalGateway = "192.168.6.1" #默认网关
|
|
# $Global:CustomGateway = "192.168.6.2" #代理网关
|
|
# $Global:CustomIP = "192.168.6.4" #默认ip
|
|
# $Global:TargetNetworkAdapter = {} #网络适配器实例
|
|
# $Global:TargetNetworkAdapterKeyword = "Ethernet" #网络适配器名字通配符
|
|
# $Global:isCustomGateway = $false #是否为自定义网关
|
|
|
|
. $(-Join($(Get-Location).Path, '\config.ps1'))
|
|
. $(-Join($(Get-Location).Path, '\proxy-test.ps1'))
|
|
. $(-Join($(Get-Location).Path, '\logger.ps1'))
|
|
|
|
|
|
function isDhcpEnable(){
|
|
return Get-NetIPConfiguration | select-object InterfaceAlias, IPv4Address, @{n="dhcp";e={$_.NetIPv4Interface.DHCP}} | Select-Object -ExpandProperty dhcp
|
|
}
|
|
|
|
function isForeignConnected {
|
|
# check foreign network availability
|
|
$result = Test-NetConnection -ComputerName $Global:TestDomain -Port 443
|
|
|
|
if ($result.TcpTestSucceeded) {
|
|
Write-Log -Message "Successfully connected to $($Global:TestDomain) via HTTPS. isCustomGateway:$($Global:isCustomGateway)"
|
|
return $true
|
|
} else {
|
|
Write-Log -Message "Failed to connect to $($Global:TestDomain) via HTTPS. isCustomGateway:$($Global:isCustomGateway)"
|
|
return $false
|
|
}
|
|
}
|
|
|
|
function isNetworkConnected {
|
|
# check network availability
|
|
$result = $(Test-Connection -ComputerName $Global:TestIP -Count 1).Status -eq 'Success'
|
|
if ($result) {
|
|
Write-Log -Message "Successfully connected to $($Global:TestIP) via ICMP. isCustomGateway:$($Global:isCustomGateway)"
|
|
return $true
|
|
} else {
|
|
Write-Log -Message "Failed to connect to $($Global:TestIP) via ICMP. isCustomGateway:$($Global:isCustomGateway)"
|
|
return $false
|
|
}
|
|
}
|
|
|
|
function setDhcp($enable){
|
|
if($enable){
|
|
Set-NetIPInterface -InterfaceAlias $Global:TargetNetworkAdapter.name -Dhcp Enabled
|
|
Write-Log -Message "enable dhcp"
|
|
} else {
|
|
Set-NetIPInterface -InterfaceAlias $Global:TargetNetworkAdapter.name -Dhcp Disabled
|
|
Write-Log -Message "disable dhcp"
|
|
}
|
|
}
|
|
|
|
function changeGateway{
|
|
param(
|
|
$adapter,$isCustomGateway
|
|
)
|
|
Write-Log -Message "Changing gateway for adapter: $($adapter.Name) isCustomGateway $($isCustomGateway)"
|
|
|
|
|
|
# check gateway availability
|
|
if($isCustomGateway){
|
|
$currentGateway = $(Get-NetIPConfiguration).IPv4DefaultGateway.NextHop
|
|
if($currentGateway -eq $Global:CustomGateway){
|
|
Write-Log -Message "current gateway is $currentGateway same as targetGateway $Global:TargetGateway, change gateway canceled"
|
|
$Global:isCustomGateway = $true
|
|
return
|
|
}
|
|
$result = $(Test-Connection $Global:CustomGateway -Count 1).Status
|
|
if (! ($result -eq 'Success')){
|
|
Write-Log -Message "custom gateway $Global:CustomGateway is $result, refuse to change"
|
|
return
|
|
}
|
|
}
|
|
|
|
# check current ip configuration
|
|
$existingIP = Get-NetIPAddress -InterfaceAlias $adapter.Name -AddressFamily IPv4
|
|
|
|
# if have ip configuration, delete it
|
|
if ($existingIP) {
|
|
Write-Log -Message "Removing existing IP address: $($existingIP.IPAddress)"
|
|
Remove-NetIPAddress -InterfaceAlias $adapter.Name -Confirm:$false
|
|
}
|
|
|
|
# delete current route(gateway)
|
|
$existingRoute = Get-NetRoute -InterfaceAlias $adapter.Name -DestinationPrefix "0.0.0.0/0"
|
|
if ($existingRoute) {
|
|
Write-Log -Message "Removing existing default route..."
|
|
Remove-NetRoute -InterfaceAlias $adapter.Name -Confirm:$false
|
|
}
|
|
|
|
if ($isCustomGateway){ #setting gateway
|
|
|
|
# config customGateway
|
|
$Global:isCustomGateway = $true
|
|
|
|
# setting dns server
|
|
Set-DnsClientServerAddress -InterfaceAlias $adapter.name -ServerAddresses ($Global:CustomGateway, $Global:CustomGateway)
|
|
# disable dhcp
|
|
setDhcp($false)
|
|
# setting gateway to customGateway
|
|
New-NetIPAddress -InterfaceAlias $adapter.Name -AddressFamily IPv4 -IPAddress $Global:CustomIP -PrefixLength 24 -DefaultGateway $Global:CustomGateway
|
|
} else {
|
|
# config originalGateway
|
|
$Global:isCustomGateway = $false
|
|
|
|
# setting dns server
|
|
Set-DnsClientServerAddress -InterfaceAlias $adapter.name -ServerAddresses ($Global:originalDns, $Global:originalDns)
|
|
# disable dhcp
|
|
setDhcp($false)
|
|
# setting gateway to originalGateway
|
|
New-NetIPAddress -InterfaceAlias $adapter.Name -AddressFamily IPv4 -IPAddress $Global:CustomIP -PrefixLength 24 -DefaultGateway $Global:originalGateway
|
|
}
|
|
Start-Sleep 3
|
|
}
|
|
|
|
|
|
function getIPAddress(){
|
|
$IP = Get-NetIPConfiguration | Select-Object -ExpandProperty IPV4Address | Select-Object -ExpandProperty IPAddress
|
|
return $IP
|
|
}
|
|
|
|
function getNetworkAdapter(){
|
|
$networkAdapters = Get-NetAdapter | ForEach-Object {
|
|
[PSCustomObject]@{
|
|
Name = $_.Name
|
|
Interface = $_.InterfaceDescription
|
|
Status = $_.Status
|
|
LinkSpeed = $_.LinkSpeed
|
|
MacAddress = $_.MacAddress
|
|
}
|
|
}
|
|
|
|
foreach ($networkAdapter in $networkAdapters){
|
|
if($networkAdapter.name.Contains($TargetNetworkAdapterKeyword)){
|
|
$Global:TargetNetworkAdapter = $networkAdapter
|
|
break;
|
|
}
|
|
}
|
|
Write-Log -Message "getNetworkAdapter: $Global:TargetNetworkAdapter"
|
|
}
|
|
|
|
function init {
|
|
Write-Log -Message "GateWay-KeepAliveForPowershell is start" -ToEventLog
|
|
# check permission
|
|
$IsAdmin = [Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()
|
|
if (-not $IsAdmin.IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)) {
|
|
# if not admin permission, restart it with admin permission
|
|
Write-Log -Message "Requesting Administrator privileges..."
|
|
Start-Process pwsh -ArgumentList "-NoExit", "-File", $PSCommandPath -Verb RunAs
|
|
exit
|
|
}
|
|
|
|
Write-Log -Message "Running with Administrator privileges!"
|
|
|
|
# get netAdapter instance
|
|
getNetworkAdapter
|
|
|
|
Write-Log -Message "current adapter: $($Global:TargetNetworkAdapter.name)"
|
|
|
|
# try to change to customGateway at start
|
|
changeGateway $Global:TargetNetworkAdapter $true
|
|
}
|
|
|
|
function loop(){
|
|
while($true){
|
|
# if network offline and using customGateway, try to restore to originalGateway
|
|
if((!$(isNetworkConnected)) -and $Global:isCustomGateway){
|
|
Write-Log -Message "Network is down, change to original gateway" -ToEventLog
|
|
changeGateway $Global:TargetNetworkAdapter $false
|
|
}
|
|
|
|
# if customGateway proxy is working and using original gateway, try to change to customGateway
|
|
if($(Test-SocksProxy -ProxyHost $Global:CustomGateway -ProxyPort 1070) -and (!$Global:isCustomGateway)){
|
|
Write-Log -Message "isCustomGateway $Global:isCustomGateway"
|
|
Write-Log -Message "proxy host is up, change to custom gateway" -ToEventLog
|
|
changeGateway $Global:TargetNetworkAdapter $true
|
|
}
|
|
|
|
# if customGateway proxy is not working but network still online, do nothing
|
|
|
|
Start-Sleep 1
|
|
}
|
|
}
|
|
|
|
init
|
|
loop
|