找回密码
 加入
搜索
查看: 4787|回复: 1

[AU3基础] 最近实现的一个定时器的小程序,作为经验积累吧

[复制链接]
发表于 2014-10-29 17:07:26 | 显示全部楼层 |阅读模式
简单的实现了一个定时器的小程序,类似于多线程并发操作时,超时后的程序会主动停止运行的进程,之后做了一些清理。

作为积累吧,发来大家瞧瞧。
#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
发表于 2016-2-25 22:45:20 | 显示全部楼层
支持一下。。
您需要登录后才可以回帖 登录 | 加入

本版积分规则

QQ|手机版|小黑屋|AUTOIT CN ( 鲁ICP备19019924号-1 )谷歌 百度

GMT+8, 2024-11-17 03:28 , Processed in 0.077892 second(s), 19 queries .

Powered by Discuz! X3.5 Licensed

© 2001-2024 Discuz! Team.

快速回复 返回顶部 返回列表