#cs ____________________________________
Au3 版本: 3.3.6.1
脚本作者: wozijisun
Email: wozijisun@sina.com
QQ/TM: Test Dept.
脚本版本: v-1.0
脚本功能:
#ce _______________脚本开始_________________
#include <Timers.au3>
#include <GUIConstantsEx.au3>
Opt('MustDeclareVars', 1)
Global $os,$config,$user,$execute_time,$mark=0,$pid="",$file,$line,$arr,$timereset
Local $time_mark = 0
Global $num,$arr_cmd[1000],$soc_cli,$dst_cli,$is_client=0
If $CmdLine[0] == 2 Then
$config = $CmdLine[1]
$user = $CmdLine[2]
Local $arr = StringSplit($config,".")
Local $brr = StringSplit($arr[1],"_")
_log("================================================")
_log("用户:" & $user & "执行场景:" & $brr[$brr[0]] & @CRLF & "START->")
_log("================================================")
Else
_log("给出参数个数为:" & $CmdLine[0] & " 出错,请检查。","error")
MsgBox(0,"INFO","执行完成!",3)
Exit
EndIf
$file = FileOpen($config, 0)
If $file = -1 Then
_log("不能打开 " & $config & " 文件,请检查。","error")
Exit
EndIf
While 1
$line = FileReadLine($file)
If @error = -1 Then ExitLoop
$arr = StringSplit($line,"=>",1)
If IsArray($arr) Then
If StringCompare($arr[1],"run_time") == 0 Then
$execute_time = $arr[2] * 1000
$mark = 1
ExitLoop
EndIf
EndIf
Wend
FileClose($file)
If $mark <> 1 Then
_log("没有在 " & $config & " 文件,找到run_time配置。","error")
Exit
EndIf
;~ 设置定时器
Local $myGui = GUICreate("")
Global $over_time = 61000
Local $timeset = _Timer_SetTimer($myGui, $over_time, "_over_time")
GUISetState(@SW_HIDE)
_getcmd_process("cmd.exe")
While 1
_execute_now($config,$user)
_log("按照正常流程执行完成此次用例。" & @CRLF & "END<-")
_Timer_KillAllTimers($myGui)
;~ _Timer_KillTimer($myGui,$timereset)
GUIDelete($myGui)
MsgBox(0,"INFO","执行完成!",3)
Exit
WEnd
;~ 作者:wozijisun
;~ 创建时间:2014-10-28
;~ 功能:设置超时定时器
;~ 参数:继承参数无实际意义
;~ 返回值:无
;~ 修改人:
;~ 修改内容:
;~ 修改时间:
Func _over_time($hWnd, $Msg, $iIDTimer, $dwTime)
#forceref $hWnd, $Msg, $iIDTimer, $dwTime
Local $count=0
If $is_client == 0 Then
While 1
$count = $count + 1
Sleep(20000)
$pid = _checkcmd_process("cmd.exe")
If StringLen(StringStripWS($pid,8)) <> 0 Then
ExitLoop
Else
If $count == 3 Then
_log("一分钟内程序异常,检测不到运行迹象,请查看执行过程中的相关问题!O(∩_∩)O~)","error")
;~ _Timer_KillTimer($myGui,$timeset)
_Timer_KillAllTimers($myGui)
GUIDelete($myGui)
Exit
EndIf
ContinueLoop
EndIf
WEnd
EndIf
_Timer_KillTimer($myGui,$timeset)
$timereset = _Timer_SetTimer($myGui,$execute_time + 300000,"_exit_app") ;在原有的时间基础上延时5分钟
;~ $timereset = _Timer_SetTimer($myGui,3000,"_exit_app") ;测试使用
EndFunc
;~ 作者:wozijisun
;~ 创建时间:2014-10-29
;~ 功能:设置超时定时器
;~ 参数:继承参数无实际意义
;~ 返回值:无
;~ 修改人:
;~ 修改内容:
;~ 修改时间:
Func _exit_app($hWnd, $Msg, $iIDTimer, $dwTime)
#forceref $hWnd, $Msg, $iIDTimer, $dwTime
If $is_client == 0 Then
ProcessClose($pid)
EndIf
If $is_client == 1 Then
Run(@ComSpec & " /c " & $dst_cli & "cmd[""TASKKILL /F /IM meter_client.exe""]", "", Default)
Run(@ComSpec & " /c " & $dst_cli & "cmd[""rd /S /Q C:\__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
EndIf
If $is_client == 2 Then
Run(@ComSpec & " /c " & $dst_cli & "cmd[""kill `ps -ef | grep -v grep | grep meter_client | awk '{print $2}'`""]", "", Default)
Run(@ComSpec & " /c " & $dst_cli & "cmd[""rm -rf /home/__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
EndIf
If StringCompare($os,"Linux") == 0 Then
_log("执行程序超时,原因:可能meter-server主机系统的内存已被占满。","error")
Run(@ComSpec & " /c " & $soc_cli & "cmd[""kill `ps -ef | grep meter_server | grep -v grep | awk '{print $2}'`""]", "", Default)
_log("8、发包结束后,杀掉meter_server进程。")
Sleep(10000)
Run(@ComSpec & " /c " & $soc_cli & "cmd[""rm -rf /home/__" & $user & "_" & @AutoItPID & "_meter_server""]", "", Default)
_log("9、删除__" & $user & "_" & @AutoItPID & "_meter_server目录。")
EndIf
If StringCompare($os,"Windows") == 0 Then
Run(@ComSpec & " /c " & $soc_cli & "cmd[""TASKKILL /F /IM meter_server.exe""]", "", Default)
_log("7、发包结束后,杀掉meter_server进程。")
Sleep(10000)
Run(@ComSpec & " /c " & $soc_cli & "cmd[""rd /S /Q C:\__" & $user & "_" & @AutoItPID & "_meter_server""]", "", Default)
_log("8、删除__" & $user & "_" & @AutoItPID & "_meter_server目录。")
EndIf
_log("程序超时退出,请关注是不是由于meter-server端内存占用较多导致。","error")
_Timer_KillTimer($myGui,$timereset)
GUIDelete($myGui)
MsgBox(0,"INFO","执行完成!但是,超时退出",3)
Exit
EndFunc
;~ 作者:wozijisun
;~ 创建时间:2014-10-29
;~ 功能:获取当前的cmd进程id
;~ 参数:检查进程的名字
;~ 返回值:无
;~ 修改人:
;~ 修改内容:
;~ 修改时间:
Func _getcmd_process($name)
Dim $arr_process = ProcessList($name)
Local $i
For $i = 0 To $arr_process[0][0]
$arr_cmd[$i] = $arr_process[$i][1]
$num=$num+1
Next
EndFunc
;~ 作者:wozijisun
;~ 创建时间:2014-10-29
;~ 功能:检查两次cmd进程id的结果
;~ 参数:检查进程的名字
;~ 返回值:返回pid的值
;~ 修改人:
;~ 修改内容:
;~ 修改时间:
Func _checkcmd_process($name)
Dim $arr_process = ProcessList($name)
For $i = 0 To $arr_process[0][0]
Local $mark_cmd = 0
For $j = 0 To $num
If $arr_cmd[$j] == $arr_process[$i][1] Then
$mark = 1
ExitLoop
EndIf
Next
If $mark_cmd == 0 And StringLen(StringStripWS($arr_process[$i][1],8)) <> 0 Then
$pid = $arr_process[$i][1]
ExitLoop
EndIf
Next
Return $pid
EndFunc
;~ 作者:wozijisun
;~ 创建时间:2014-10-10
;~ 功能:立刻执行Meter程序,进行收发判断
;~ 参数:$config是给定的执行配置文件,$user是给定的执行用户名称
;~ 返回值:无
;~ 修改人:
;~ 修改内容:
;~ 修改时间:
Func _execute_now($config,$user)
Local $conf_num = 1,$c[25],$main_socket,$var
If FileExists($config) And StringLen(StringStripWS($user,8)) <> 0 Then
$file = FileOpen($config, 0)
; 检查打开的文件是否可读
If $file = -1 Then
_log("不能打开 " & $config & " 文件,请检查。","error")
Return
EndIf
; 每次读取一行文本,直到文件结束.
While 1
$line = FileReadLine($file)
If @error = -1 Then ExitLoop
If StringInStr($line,"[start]",1) == 1 Then
_log("读取首行[start]")
ContinueLoop
EndIf
If StringInStr($line,"[end]",1) == 1 Then
;~ MsgBox(0,"xx",$conf_num)
If $conf_num == 25 Then
_log("读取最后行[end]共" & $conf_num-1 & "行")
ExitLoop
Else
_log($config & @CRLF & "配置文件参数没有达到24个,请检查。","error")
Return
EndIf
EndIf
$arr = StringSplit($line,"=>",1)
Switch $arr[1]
Case "meter_server"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("meter_server为空值,不能执行!")
Return
EndIf
If StringCompare(_check_file($arr[2]),"no") == 0 Then
_log("meter_server文件不是一个正常的可执行文件!"&$arr[2])
Return
EndIf
$c[0]=StringStripWS($arr[2],1+2)
$conf_num = $conf_num + 1
_log("meter_server=" & $arr[2])
Case "meter_client"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("meter_client为空值,不能执行!")
Return
EndIf
If StringCompare(_check_file($arr[2]),"no") == 0 Then
_log("meter_client文件不是一个正常的可执行文件!"&$arr[2])
Return
EndIf
$c[1]=StringStripWS($arr[2],1+2)
$conf_num = $conf_num + 1
_log("meter_client=" & $arr[2])
Case "socket_client"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("socket_client为空值,不能执行!")
Return
EndIf
If StringCompare(_check_file($arr[2]),"no") == 0 Then
_log("socket_client文件不是一个正常的可执行文件!"&$arr[2])
Return
EndIf
$c[2]=StringStripWS($arr[2],1+2)
$conf_num = $conf_num + 1
_log("socket_client=" & $arr[2])
Case "global_port"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
$c[3]="6677"
_log("global_port为空值,设置默认值为6677!")
ContinueLoop
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^[1-9][0-9]{0,}",0) Or $arr[2] * 1 > 65535 Then
_log("global_port输入不合法的数字。"&$arr[2])
Return
EndIf
$c[3]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("global_port=" & $arr[2])
Case "is_os"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("is_os为空值,不能执行!")
Return
EndIf
If StringCompare(StringStripWS($arr[2],8),"Windows") <> 0 And StringCompare(StringStripWS($arr[2],8),"Linux") <> 0 Then
_log("is_os值为Windows/Linux!不为"&$arr[2])
Return
EndIf
$c[4]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("is_os=" & $arr[2])
Case "server_ip"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("server_ip为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])$",0) Then
_log("server_ip输入了非IP地址表达式。"&$arr[2])
Return "ng"
EndIf
$c[5]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("server_ip=" & $arr[2])
Case "listen_port"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("listen_port为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^[1-9][0-9]{0,}",0) Or $arr[2] * 1 > 65535 Then
_log("listen_port输入不合法的数字。"&$arr[2])
Return
EndIf
$c[6]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("listen_port=" & $arr[2])
Case "dbx_ip"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("dbx_ip为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])$",0) Then
_log("dbx_ip输入了非IP地址表达式。"&$arr[2])
Return
EndIf
$c[7]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("dbx_ip=" & $arr[2])
Case "is_agent"
If Number(StringStripWS($arr[2],8)) <> 0 And Number(StringStripWS($arr[2],8)) <> 1 Then
_log("is_agent值不在执行0(无代理) / 1(有代理)范围内!")
Return
EndIf
$c[8]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("is_agent=" & $arr[2])
Case "agent_ip"
$c[9]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("agent_ip=" & $arr[2])
Case "agent_port"
$c[10]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("agent_port=" & $arr[2])
Case "is_client"
If $arr[2] * 1 <> 0 And $arr[2] * 1 <> 1 And $arr[2] * 1 <> 2 Then
_log("is_client值不在执行 0(无) / 1(windows) / 2(linux) 范围内!")
Return
EndIf
$c[11]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("is_client=" & $arr[2])
Case "client_ip"
$c[12]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("client_ip=" & $arr[2])
Case "simulate_client_ip"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("simulate_client_ip为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])$",0) Then
_log("simulate_client_ip输入了非IP地址表达式。"&$arr[2])
Return
EndIf
$c[13]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("simulate_client_ip=" & $arr[2])
Case "simulate_client_port"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("simulate_client_port为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^[1-9][0-9]{0,}",0) Or $arr[2] * 1 > 65535 Then
_log("simulate_client_port输入不合法的数字。"&$arr[2])
Return
EndIf
$c[14]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("simulate_client_port=" & $arr[2])
Case "simulate_server_ip"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("simulate_server_ip为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])$",0) Then
_log("simulate_server_ip输入了非IP地址表达式。"&$arr[2])
Return
EndIf
$c[15]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("simulate_server_ip=" & $arr[2])
Case "simulate_server_port"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("simulate_server_port为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^[1-9][0-9]{0,}",0) Or $arr[2] * 1 > 65535 Then
_log("simulate_server_port输入不合法的数字。"&$arr[2])
Return
EndIf
$c[16]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("simulate_server_port=" & $arr[2])
Case "sequence"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("sequence为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"[0-9]{0,}",0) Then
_log("sequence输入不合法的数字。"&$arr[2])
Return
EndIf
$c[17]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("sequence=" & $arr[2])
Case "pcapfile"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("pcapfile为空值,不能执行!")
Return
EndIf
If StringCompare(_check_file($arr[2]),"no") == 0 Then
_log("pcapfile不是一个正常的PCAP文件!"&$arr[2])
Return
EndIf
$c[18]=StringStripWS($arr[2],1+2)
$conf_num = $conf_num + 1
_log("pcapfile=" & $arr[2])
Case "simulate_client_count"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("simulate_client_count为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^[1-9][0-9]{0,}",0) Then
_log("simulate_client_count输入不合法的数字。"&$arr[2])
Return
EndIf
$c[19]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("simulate_client_count=" & $arr[2])
Case "connect_per_sec"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("connect_per_sec为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^[1-9][0-9]{0,}",0) Then
_log("connect_per_sec输入不合法的数字。"&$arr[2])
Return
EndIf
$c[20]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("connect_per_sec=" & $arr[2])
Case "loop_count"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("loop_count为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^[1-9][0-9]{0,}",0) Then
_log("loop_count输入不合法的数字。"&$arr[2])
Return
EndIf
$c[21]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("loop_count=" & $arr[2])
Case "think_time"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("think_time为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^[0-9]{0,}",0) Then
_log("think_time输入不合法的数字。"&$arr[2])
Return
EndIf
$c[22]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("think_time=" & $arr[2])
Case "run_time"
If StringLen(StringStripWS($arr[2],8)) == 0 Then
_log("run_time为空值,不能执行!")
Return
EndIf
If Not StringRegExp(StringStripWS($arr[2],8),"^[0-9]{0,}",0) Then
_log("run_time输入不合法的数字。"&$arr[2])
Return
EndIf
$c[23]=StringStripWS($arr[2],8)
$conf_num = $conf_num + 1
_log("run_time=" & $arr[2])
Case Else
_log("无法识别的字符:" & $arr[1] & "=" & $arr[2],"warn")
EndSwitch
Wend
FileClose($file)
;~ 判断meter-client是否需要放置在远程机器上
If $c[11] * 1 <> 0 Then
$is_client = $c[11] * 1
TCPStartup()
$main_socket = TCPConnect($c[12], $c[3])
If $main_socket = -1 Then
TCPCloseSocket($main_socket)
TCPShutdown()
_log("服务器端的Socket-Server服务器没有开启,请开启IP:" & $c[5] & "<=>Port:" & $c[3] &" 后执行。","warn")
Return
EndIf
$dst_cli = $c[2] & " "& $c[12] & "#" & $c[3] & "#"
;~ windows OS
If $c[11] * 1 == 1 Then
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""echo yes > C:\yes.log""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "copyfile[C:\yes.log,"& @ScriptDir &"\yes.log]", "", Default)
If Not FileExists(@ScriptDir & "\yes.log") Then
_log("指定Meter客户端的操作系统可能不是Windows,请假查。")
Return
EndIf
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""del /F /Q C:\yes.log""]", "", Default)
FileDelete(@ScriptDir & "\yes.log")
;~ Linux OS
ElseIf $c[11] * 1 == 2 Then
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""echo yes > /home/yes.log""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "copyfile[/home/yes.log,"& @ScriptDir &"\yes.log]", "", Default)
If Not FileExists(@ScriptDir & "\yes.log") Then
_log("指定Meter客户端的操作系统可能不是Linux,请假查。")
Return
EndIf
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""rm -rf /home/yes.log""]", "", Default)
FileDelete(@ScriptDir & "\yes.log")
;~ Other OS
Else
_log("操作系统可能不是Windows / Linux,请假查。","warn")
Return
EndIf
EndIf
$soc_cli = $c[2] & " "& $c[5] & "#" & $c[3] & "#"
;~ 对Server_IP进行处理,因为Server_IP是Meter-Server所在的主机,执行时需要将meter工具上传到该主机上
TCPStartup()
$main_socket = TCPConnect($c[5], $c[3])
If $main_socket = -1 Then
TCPCloseSocket($main_socket)
TCPShutdown()
_log("服务器端的Socket-Server服务器没有开启,请开启IP:" & $c[5] & "<=>Port:" & $c[3] &" 后执行。","warn")
Return
EndIf
;~ 对DBx_IP进行处理,因为要到DBx_IP上获取tracelog文件
TCPStartup()
$main_socket = TCPConnect($c[7], $c[3])
If $main_socket = -1 Then
TCPCloseSocket($main_socket)
TCPShutdown()
_log("服务器端的Socket-Server服务器没有开启,请开启IP:" & $c[7] & "<=>Port:" & $c[3] &" 后执行。","warn")
Return
EndIf
;~ 设置代理模式后,对代理进行判断,MS可以不写
;~ If $c[8] * 1 == 1 Then
;~ TCPStartup()
;~ $main_socket = TCPConnect($c[9], $c[10])
;~ If $main_socket = -1 Then
;~ TCPCloseSocket($main_socket)
;~ TCPShutdown()
;~ _log("服务器端的Socket-Server服务器没有开启,请开启IP:" & $c[5] & "<=>Port:" & $c[3] &" 后执行。","warn")
;~ Return
;~ EndIf
;~ EndIf
;~ 判断meter-server所在主机的类型
If StringCompare(StringStripWS($c[4],8), "Windows") == 0 Then
$os = "Windows"
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""echo yes > C:\yes.log""]", "", Default)
RunWait(@ComSpec & " /c " & $soc_cli & "copyfile[C:\yes.log,"& @ScriptDir &"\yes.log]", "", Default)
If Not FileExists(@ScriptDir & "\yes.log") Then
_log("服务器端的操作系统可能不是Windows,请假查。")
Return
EndIf
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""del /F /Q C:\yes.log""]", "", Default)
FileDelete(@ScriptDir & "\yes.log")
_log("执行Windows系统,共八步。")
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""rd /S /Q C:\__" & $user & "_" & @AutoItPID & "_meter_server""]", "", Default)
_log("1、先执行删除__meter_server目录的操作,防止影响Meter执行。")
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""mkdir C:\__" & $user & "_" & @AutoItPID & "_meter_server""]", "", Default)
_log("2、执行创建__meter_server目录的操作。")
RunWait(@ComSpec & " /c " & $soc_cli & "sendfile["& $c[0] &",C:\__" & $user & "_" & @AutoItPID & "_meter_server\meter_server.exe]", "", Default)
_log("3、向__meter_server目录发送meter_server程序。")
RunWait(@ComSpec & " /c " & $soc_cli & "sendfile["& $c[18] &",C:\__" & $user & "_" & @AutoItPID & "_meter_server\meter_need.pcap]", "", Default)
_log("4、向__meter_server目录发送PCAP包。")
Run(@ComSpec & " /c " & $soc_cli & "cmd[""C:\__" & $user & "_" & @AutoItPID & "_meter_server\meter_server.exe " & $c[6] & " " & $c[13] & " " & $c[14] & " " & $c[15] & " " & $c[16] & " " & $c[17] & "" & " C:\__" & $user & "_" & @AutoItPID & "_meter_server\meter_need.pcap""]", "", Default)
_log("5、启动meter_server程序。")
If $c[11] * 1 == 0 Then
RunWait(@ComSpec & " /c " & $c[1] & " " & $c[5] & " " & $c[6] & " " & $c[13] & " " & $c[14] & " " & $c[15] & " " & $c[16] & " " & $c[19] & " " & $c[20] & " " & $c[21] & " " & $c[17] & " " & $c[22] & " " & $c[23] & " " & $c[18], "", @SW_MAXIMIZE)
ElseIf $c[11] * 1 == 1 Then
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""mkdir C:\__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "sendfile["& $c[1] &",C:\__" & $user & "_" & @AutoItPID & "_meter_client\meter_client.exe]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""C:\__" & $user & "_" & @AutoItPID & "_meter_client\meter_client.exe" & " " & $c[2] & " " & $c[4] & " " & $c[5] & " " & $c[6] & " " & $c[7] & " " & $c[8] & " " & $c[9] & " " & $c[10] & " " & $c[11] & " " & $c[12] & " " & $c[13] & " " & $c[14] & " " & $c[15] & """]", "", @SW_MAXIMIZE)
;~ RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""TASKKILL /F /IM meter_server.exe""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""rd /S /Q C:\__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
ElseIf $c[11] * 1 == 2 Then
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""mkdir /home/__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "sendfile["& $c[1] &",/home/__" & $user & "_" & @AutoItPID & "_meter_client\meter_client]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""/home/__" & $user & "_" & @AutoItPID & "_meter_client\meter_client" & " " & $c[2] & " " & $c[4] & " " & $c[5] & " " & $c[6] & " " & $c[7] & " " & $c[8] & " " & $c[9] & " " & $c[10] & " " & $c[11] & " " & $c[12] & " " & $c[13] & " " & $c[14] & " " & $c[15] & """]", "", @SW_MAXIMIZE)
;~ RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""TASKKILL /F /IM meter_server.exe""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""rm -rf /home/__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
EndIf
_log("6、启动meter_client程序,向meter-server端发包。")
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""TASKKILL /F /IM meter_server.exe""]", "", Default)
_log("7、发包结束后,杀掉meter_server进程。")
Sleep(10000)
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""rd /S /Q C:\__" & $user & "_" & @AutoItPID & "_meter_server""]", "", Default)
_log("8、删除__" & $user & "_" & @AutoItPID & "_meter_server目录。")
ElseIf StringCompare(StringStripWS($c[4],8), "Linux") == 0 Then
$os = "Linux"
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""echo yes > /home/yes.log""]", "", Default)
RunWait(@ComSpec & " /c " & $soc_cli & "copyfile[/home/yes.log,"& @ScriptDir &"\yes.log]", "", Default)
If Not FileExists(@ScriptDir & "\yes.log") Then
_log("服务器端的操作系统可能不是Linux,请假查。")
Return
EndIf
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""rm -rf /home/yes.log""]", "", Default)
FileDelete(@ScriptDir & "\yes.log")
_log("执行Linux系统,共九步。")
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""rm -rf /home/__" & $user & "_" & @AutoItPID & "_meter_server""]", "", Default)
_log("1、先执行删除__meter_server目录的操作,防止影响Meter执行。")
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""mkdir /home/__" & $user & "_" & @AutoItPID & "_meter_server""]", "", Default)
_log("2、执行创建__meter_server目录的操作。")
RunWait(@ComSpec & " /c " & $soc_cli & "sendfile["& $c[0] &",/home/__" & $user & "_" & @AutoItPID & "_meter_server/meter_server]", "", Default)
_log("3、向__meter_server目录发送meter_server程序。")
RunWait(@ComSpec & " /c " & $soc_cli & "sendfile["& $c[7] &",/home/__" & $user & "_" & @AutoItPID & "_meter_server/meter_need.pcap]", "", Default)
_log("4、向__meter_server目录发送PCAP包。")
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""chmod 777 /home/__" & $user & "_" & @AutoItPID & "_meter_server/*""]", "", Default)
_log("5、向__meter_server目录赋予可执行权限。")
Run(@ComSpec & " /c " & $soc_cli & "cmd[""cd /home/__" & $user & "_" & @AutoItPID & "_meter_server;./meter_server "& $c[1] & " " & $c[2] & " " & $c[3] & " " & $c[4] & " " & $c[5] & " " & $c[6] & "" & " /home/__" & $user & "_" & @AutoItPID & "_meter_server/meter_need.pcap""]", "", Default)
_log("6、启动meter_server程序。")
If $c[11] * 1 == 0 Then
RunWait(@ComSpec & " /c " & $c[1] & " " & $c[5] & " " & $c[6] & " " & $c[13] & " " & $c[14] & " " & $c[15] & " " & $c[16] & " " & $c[19] & " " & $c[20] & " " & $c[21] & " " & $c[17] & " " & $c[22] & " " & $c[23] & " " & $c[18], "", @SW_MAXIMIZE)
ElseIf $c[11] * 1 == 1 Then
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""mkdir C:\__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "sendfile["& $c[1] &",C:\__" & $user & "_" & @AutoItPID & "_meter_client\meter_client.exe]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""C:\__" & $user & "_" & @AutoItPID & "_meter_client\meter_client.exe" & " " & $c[2] & " " & $c[4] & " " & $c[5] & " " & $c[6] & " " & $c[7] & " " & $c[8] & " " & $c[9] & " " & $c[10] & " " & $c[11] & " " & $c[12] & " " & $c[13] & " " & $c[14] & " " & $c[15] & """]", "", @SW_MAXIMIZE)
;~ RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""TASKKILL /F /IM meter_server.exe""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""rd /S /Q C:\__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
ElseIf $c[11] * 1 == 2 Then
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""mkdir /home/__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "sendfile["& $c[1] &",/home/__" & $user & "_" & @AutoItPID & "_meter_client\meter_client]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""/home/__" & $user & "_" & @AutoItPID & "_meter_client\meter_client" & " " & $c[2] & " " & $c[4] & " " & $c[5] & " " & $c[6] & " " & $c[7] & " " & $c[8] & " " & $c[9] & " " & $c[10] & " " & $c[11] & " " & $c[12] & " " & $c[13] & " " & $c[14] & " " & $c[15] & """]", "", @SW_MAXIMIZE)
;~ RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""TASKKILL /F /IM meter_server.exe""]", "", Default)
RunWait(@ComSpec & " /c " & $dst_cli & "cmd[""rm -rf /home/__" & $user & "_" & @AutoItPID & "_meter_client""]", "", Default)
EndIf
_log("7、启动meter_client程序,向meter-server端发包。")
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""kill `ps -ef | grep meter_server | grep -v grep | awk '{print $2}'`""]", "", Default)
_log("8、发包结束后,杀掉meter_server进程。")
Sleep(10000)
RunWait(@ComSpec & " /c " & $soc_cli & "cmd[""rm -rf /home/__" & $user & "_" & @AutoItPID & "_meter_server""]", "", Default)
_log("9、删除__" & $user & "_" & @AutoItPID & "_meter_server目录。")
Else
_log("有病毒入侵了吗?遇到了无法解析的操作系统os=|" & $c[4] & "|","warn")
Return
EndIf
Else
_log("当前执行的PCAP包没有传入配置文件Config或者执行者名称,程序无法正常执行。" & _
"请先配置后再执行。","warn")
Return
EndIf
EndFunc
;~ 作者:wozijisun
;~ 创建时间:2014-10-17
;~ 功能:生成Log日志
;~ 参数:message记录信息,level出错的级别(info:正常信息打印,warn:提示性错误,error:严重错误导致程序无法运行)
;~ 返回值:无
;~ 修改人:
;~ 修改内容:
;~ 修改时间:
Func _log($message,$level="info")
Local $date,$time
$date = @YEAR & "-" & @MON & "-" & @MDAY
$time = "[" & @YEAR & "-" & @MON & "-" & @MDAY & "_" & @HOUR & ":" & @MIN & ":" & @SEC & "." & @MSEC & "-" & $level _
& "]"
If Not FileExists(@ScriptDir & "\Log") Then
DirCreate(@ScriptDir & "\Log")
EndIf
FileWriteLine(@ScriptDir & "\Log" & $date & ".log",@AutoItPID & $time & $message)
EndFunc
;~ 作者:wozijisun
;~ 创建时间:2014-09-28
;~ 功能:检查文件是否存在于固定路径下
;~ 修改人:
;~ 修改内容:
;~ 修改时间:
Func _check_file($path)
Local $attr
If FileExists($path) Then
$attr = FileGetAttrib($path)
If StringInStr($attr,"N") Or StringInStr($attr,"C") Or StringInStr($attr,"O") Or StringInStr($attr,"A") Then
Return "yes"
Else
Return "no"
EndIf
Else
Return "no"
EndIf
EndFunc
;~ 作者:wozijisun
;~ 创建时间:2014-10-22
;~ 功能:检查IP地址是否为可用的IP
;~ 参数:$ip输入的IP地址
;~ 返回值:ng=Not go;ok=go
;~ 修改人:
;~ 修改内容:
;~ 修改时间:
;~ Func _check_ip($ip)
;~ Local $var
;~ If Not StringRegExp(StringStripWS($ip,8),"^(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])\.(\d{1,2}|1\d\d|2[0-4]\d|25[0-5])$",0) Then
;~ _log("simulate_client_ip输入了非IP地址表达式。"&$ip)
;~ Return "ng"
;~ EndIf
;~ $var = Ping(StringStripWS($ip,8))
;~ If $var == 0 Then
;~ Switch @error
;~ Case 1
;~ _log("simulate_client_ip所在的目标主机离线。" & $ip)
;~ Return "ng"
;~ Case 2
;~ _log("simulate_client_ip所在的目标主机无法到达。" & $ip)
;~ Return "ng"
;~ Case 3
;~ _log("simulate_client_ip所在的目标主机错误。" & $ip)
;~ Return "ng"
;~ Case 4
;~ _log("simulate_client_ip所在的目标其它错误。" & $ip)
;~ Return "ng"
;~ EndSwitch
;~ Else
;~ Return "ok"
;~ EndIf
;~ EndFunc