powershell 作服务端 响应网络(socket tcp)连接 提供文件夹大小查询服务 V1.1
包含：
- 端口占用检测
- 心跳包网络断线检测
- 传入的数据是否为合法有效的目录路径检测
- 读取计算文件夹大小（不含软链接|symlink）
- 传回查询到的文件夹大小
Powershell Service
# Service.ps1
[cmdletbinding()] Param($Port = 8888)
$VerbosePreference = "Continue"
# 值或取`SilentlyContinue`，此时需调用脚本时传入`-Verbose`才等效`Continue`。
# `Continue`输出`Write-Verbose`的内容。
Write-Verbose("Check port {0} busy/available" -f $Port)
$Connection = Get-NetTCPConnection -LocalPort $Port -ErrorAction SilentlyContinue
if($Connection){
if($Connection.OwningProcess.GetType().Name -eq "UInt32"){
$Process_ID = $Connection.OwningProcess
}
else{
$Process_ID = $Connection.OwningProcess[1]
# `$Connection.OwningProcess`为数组对象(System.Object)
}
if($Process_ID){
Write-Error("Port {0} is busy, using by process ID {1}." -f $Port, $Process_ID)
Exit
}
else{
Write-Verbose("Port {0} is available" -f $Port)
}
}
else{
Write-Verbose("Port {0} is available" -f $Port)
}
$Connection = $Null
$Listener = New-Object System.Net.Sockets.TcpListener -ArgumentList $Port
try{
$Listener.Start()
# 可能的异常：
# # "通常每个套接字地址(协议/网络地址/端口)只允许使用一次。"
# # CategoryInfo : NotSpecified: (:) [], ParentContainsErrorRecordException
# # FullyQualifiedErrorId : SocketException
Write-Verbose("Start linsten.")
}
catch {
Write-Error("Linsten failed.")
throw
}
Write-Verbose "Server started."
While ($True) {
Write-Host("Waiting connection from port {0}" -f $Port)
$Socket_TCP_Client = $Listener.AcceptTcpClient()
# incoming socket tcp client endpoint.
$Client_Address = $Socket_TCP_Client.client.RemoteEndPoint.ToString()
Write-Verbose ("New connection from {0}" -f $Client_Address)
# Start-Sleep -Milliseconds 1000
$Stream = $Socket_TCP_Client.GetStream()
$Stream.ReadTimeout = 1000 # ms
# [NetworkStream.ReadTimeout Property](https://learn.microsoft.com/en-us/dotnet/api/system.net.sockets.networkstream.readtimeout?view=net-8.0)
$Stream_Reader = [System.IO.StreamReader]::new($Stream)
$Stream_Writer = New-Object System.IO.StreamWriter($Stream)
$Previous_Communication_Time = Get-Date -UFormat %s
$string = $Null
While ($Socket_TCP_Client.Connected) {
Write-Verbose "Try to read line."
try{
$string = $Stream_Reader.ReadLine()
}
catch{
# 可能的异常：
# # 使用“0”个参数调用“ReadLine”时发生异常:“无法从传输连接中读取数据: 由于连接方在一段时间后没有正确答复或连接的主机没有反应，连接尝试失败。。”
# # 使用“0”个参数调用“ReadLine”时发生异常:“无法从传输连接中读取数据: 你的主机中的软件中止了一个已建立的连接。。”
Write-Verbose "Timeout, waiting data."
}
if($string -eq "exit"){
Write-Verbose "Exit."
$Stream_Reader.Dispose()
Write-Verbose "Stream_Reader.Dispose done."
$Stream_Reader.Close()
Write-Verbose "Stream_Reader.Close done."
$Stream_Writer.Dispose()
Write-Verbose "Stream_Writer.Dispose done."
$Stream_Writer.Close()
Write-Verbose "Stream_Writer.Close done."
$Stream.Dispose()
Write-Verbose "Stream.Dispose done."
$Stream.Close()
Write-Verbose "Stream.Close done."
$Socket_TCP_Client.Dispose()
Write-Verbose "Socket_TCP_Client.Dispose done."
$Socket_TCP_Client.Close()
Write-Verbose "Socket_TCP_Client.Close done."
break
}
if ($string) {
Write-Verbose "Data available, line read."
Write-Host("Message received from {0}:`n {1}" -f $Client_Address, $string)
$Previous_Communication_Time = Get-Date -UFormat %s
$Path = $string
$string = $Null
$Is_Path_Exist_Folder = $Null
try{
$Is_Path_Exist_Folder = Test-Path -Path $Path -PathType Container
# 可能的异常：
# # "Test-Path : 路径中具有非法字符。"
}catch{}
if($Is_Path_Exist_Folder){
Write-Verbose "Path valid - Path is a folder."
}
else{
Write-Verbose "Path invalid - Path is not a folder, skip."
}
if($Is_Path_Exist_Folder){
Write-Verbose "Measure size (length) of folder.."
$Command = 'Get-ChildItem -Path "'+ $Path + '" -Recurse | Measure-Object -Property Length -Sum'
$GenericMeasureInfo = Invoke-Expression $Command
$Size = $GenericMeasureInfo.Sum
Write-Verbose("Folder size (length) is {0}." -f $Size)
Write-Verbose "Send size to client."
$Stream_Writer.WriteLine($Size)
$Stream_Writer.Flush()
}
else{
Write-Verbose "Send error code to client."
$Stream_Writer.WriteLine("")
$Stream_Writer.Flush()
}
}
ElseIf((($Current_Time = (Get-Date -UFormat %s)) - $Previous_Communication_Time) -gt 3){ #等待数据超时，检查断线
$bytes = 0
try{ #heart-beat
Write-Verbose "Heart-beat test."
# 参见：[(16)Powershell中的转义字符 - yang-leo - 博客园](https://www.cnblogs.com/leoyang63/articles/12060596.html)
$Stream_Writer.WriteLine("")
$Stream_Writer.Flush()
$Previous_Communication_Time = $Current_Time
Write-Verbose("Connection {0} alive." -f $Client_Address)
}catch{
Write-Verbose("Connection {0} failed." -f $Client_Address)
break
}
}
}
if($string -eq "exit"){
break
}else{
Write-Host "Client disconnect."
}
}
$Listener.Stop()
Write-Verbose "Listener.Stop done."
Write-Verbose "Server stopped."
Write-Host "Exit."
Exit
Lua Client
-- Client.lua
local Socket = require("socket")
local function Query_Size(self,Path)
local Client = self._Client
Client:send(Path..'\n')
local result,response = ''
repeat
response = Client:receive"*l"
result = result .. (response or '')
until not response or result~=''
return result
end
local function Close(self)
local Client = self._Client
-- Client:shutdown()
Client:close()
end
local Metatable = {__gc = Close,}
local Master = {
Connect = function(Host_Name_or_IP,Port)
local Client = assert(Socket.connect(Host_Name_or_IP or "localhost", Port or 8888))
-- "connection refused"
local Client_Agent = setmetatable(
{
Query_Size = Query_Size,
Query = Query_Size,
Close_Client = Close,
Close = Close,
_Client = Client,
Close_Service = function()
Client:send"Exit\n"
Client:close()
end,
},Metatable
)
return Client_Agent
end,
}
if not ... then --test
local Client = Master.Connect()
print(Client:Query_Size[[H:\Ghost]])
print(Client:Query[[H:\土豆]])
Client.Close_Service()
Client:Close()
end
return Master
