在上篇文章中,我们在PSNet包中创建了Test-TCPPort函数用于探测指定IP的指定端口是否开放,检测端口之后大多数人想到的可能就是需要通过PowerShell收发TCP消息包了,这篇文章里将会描述如何在PSNet包中创建针对TCP消息包的函数Receive-和TCPMessageSend-TCPMessage。为了承接上篇中我们创建的PSNet工具集的思路,在确定了函数的命名之后,创建上述两个函数对应的.ps1文件放置在$env:PSSpace/PSNet/TCPOp/中。
接下来在$env:PSSpace/PSNet/PSNet.psm1中添加以下两条语句用于在工具集中引入上述两个函数文件:
. $env:PSSpace/PSNet/TCPOp/Receive-TCPMessage.ps1
. $env:PSSpace/PSNet/TCPOp/Send-TCPMessage.ps1然后分别在创建的.ps1文件中添加以下代码:
=====文件名:Receive-TCPMessage.ps1===== Receive-TCPMessage{ ( [ValidateNotNullOrEmpty()] [int] $Port ) try { $EndPoint = System.Net.IPEndPoint([System.Net.IPAddress]::Loopback,$Port) $Socket = System.Net.Sockets.TCPListener($EndPoint) $Socket.Start() $Socket = $Socket.AcceptTCPClient() $EncodedText = System.Text.ASCIIEncoding $Stream = $Socket.GetStream() $Buffer = System.Byte[] $Socket.ReceiveBufferSize ( $Bytes = $Stream.Read($Buffer,0,$Buffer.Length) ) { $Stream.($Buffer,0,$Bytes) $EncodedText.GetString($Buffer,0,$Bytes) } $Socket.Close() $Socket.Stop() } catch{}}
=====文件名:Send-TCPMessage.ps1===== Send-TCPMessage{ ( [ValidateNotNullOrEmpty()] [string] $EndPoint, [int] $Port, [string] $Message ) $IP = [System.Net.Dns]::GetHostAddresses($EndPoint) $Address = [System.Net.IPAddress]::Parse($IP) $Socket = System.Net.Sockets.TCPClient($Address,$Port) $Stream = $Socket.GetStream() $Writer = System.IO.StreamWriter($Stream) $Writer.AutoFlush = $true $Writer.NewLine = $true $Writer.($Message) $Socket.Close()}
将代码保存到指定目录的对应文件后,分别启动两个PowerShell进程,分别导入PSNet Module:
Import-Module $env:PSSpace\PSNet
或者在启动PowerShell时指定参数,或者将下列语句创建批处理脚本启动加载指定Module的PowerShell进程。
start %windir%\System32\WindowsPowerShell\v1.0\powershell.exe -noExit -Command "Import-Module '%PSSpace%\PSNet' "
首先在其中一个PowerShell窗口中,指定端口用Receive-TCPMessage函数在制定端口监听,等待接受数据
Receive-TCPMessage 8080
在另外一个窗口中向上面的端口发送数据:
Send-TCPMessage 127.0.0.1 8080 “This a Message Send from PSNet!”
会发现这个进程发送消息后,前一个进程能收到相应的消息,消息中如果有空格需要用双引号包括确保PowerShell解释引擎知道这是一个完整的参数。这两个小函数虽然功能简单,而且其中的消息使用ASCII的方式发送的,这是为了给大家演示发送和接收的效果,在真正的实际环境中还是有问题的,但是这两个函数将会在后续的功能中起到很重要的作用,使用PowerShell发送TCP和接收TCP报文将会用到,也将会在后续的文章中进行改进。
作者: 付海军
出处: 版权:本文版权归作者和博客园共有 转载:欢迎转载,为了保存作者的创作热情,请按要求【转载】,谢谢 要求:未经作者同意,必须保留此段声明;必须在文章中给出原文连接且保证内容完整!否则必究法律责任! 个人网站: