-
Notifications
You must be signed in to change notification settings - Fork 0
/
check_msinfo32.psm1
107 lines (82 loc) · 3.17 KB
/
check_msinfo32.psm1
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
function check_msinfo32 ([string]$para1,[string]$para2){
Set-ExecutionPolicy -Scope CurrentUser -ExecutionPolicy Bypass -Force;
#$wshell=New-Object -ComObject wscript.shell
Add-Type -AssemblyName Microsoft.VisualBasic
Add-Type -AssemblyName System.Windows.Forms
Add-Type -AssemblyName System.Windows.Forms,System.Drawing
if($PSScriptRoot.length -eq 0){
$scriptRoot="C:\testing_AI\modules"
}
else{
$scriptRoot=$PSScriptRoot
}
$outlogtype=$para1
$nonlog_flag=$para2
$action="check_msinfo32"
$tcpath=(Split-Path -Parent $scriptRoot)+"\currentjob\TC.txt"
$tcnumber=((get-content $tcpath).split(","))[0]
$tcstep=((get-content $tcpath).split(","))[1]
$picpath=(Split-Path -Parent $scriptRoot)+"\logs\$($tcnumber)\"
if(-not(test-path $picpath)){new-item -ItemType directory -path $picpath |out-null}
$mslog1=$picpath+"step$($tcstep)_msinfo.txt"
$mslog2=$picpath+"step$($tcstep)_msinfo.nfo"
$results="NG"
$index="fail to open msinfo32"
(Get-Process -name msinfo32 -ea SilentlyContinue).CloseMainWindow()
#get report #
if($outlogtype -match "txt"){
Start-Process 'C:\Windows\System32\msinfo32.exe' -ArgumentList '/report', $mslog1 -Wait
}
if($outlogtype -match "nfo"){
Start-Process 'C:\Windows\System32\msinfo32.exe' -ArgumentList '/nfo', $mslog2 -Wait
}
if($outlogtype.Length -eq 0){
Start-Process 'C:\Windows\System32\msinfo32.exe' -ArgumentList '/nfo', $mslog1 -Wait
Start-Process 'C:\Windows\System32\msinfo32.exe' -ArgumentList '/report', $mslog2 -Wait
}
$actionss="screenshot"
Get-Module -name $actionss|remove-module
$mdpath=(Get-ChildItem -path $scriptRoot -r -file |Where-object{$_.name -match "^$actionss\b" -and $_.name -match "psm1"}).fullname
Import-Module $mdpath -WarningAction SilentlyContinue -Global
#screenshots
start-process msinfo32
Start-Sleep -s 10
$process = Get-Process -name msinfo32
if ($process) {
$results="OK"
$index="check screenshots"
# Define the ShowWindow function from the user32.dll
Add-Type @"
using System;
using System.Runtime.InteropServices;
public class Win32ShowWindow {
[DllImport("user32.dll")]
[return: MarshalAs(UnmanagedType.Bool)]
public static extern bool ShowWindow(IntPtr hWnd, int nCmdShow);
}
"@
# Maximize the window
[Win32ShowWindow]::ShowWindow($process.MainWindowHandle, 3) # 3 is for SW_MAXIMIZE
Start-Sleep -s 2
[System.Windows.Forms.SendKeys]::SendWait("{tab}")
Start-Sleep -s 2
## screen capture ##
&$actionss -para3 nonlog -para5 "1"
[System.Windows.Forms.SendKeys]::SendWait("{PGDN}")
Start-Sleep -s 5
&$actionss -para3 nonlog -para5 "2"
[System.Windows.Forms.SendKeys]::SendWait("{PGDN}")
Start-Sleep -s 5
&$actionss -para3 nonlog -para5 "3"
(Get-Process -name msinfo32).CloseMainWindow()
}
######### write log #######
if($nonlog_flag.Length -eq 0){
Get-Module -name "outlog"|remove-module
$mdpath=(Get-ChildItem -path "C:\testing_AI\modules\" -r -file |Where-object{$_.name -match "outlog" -and $_.name -match "psm1"}).fullname
Import-Module $mdpath -WarningAction SilentlyContinue -Global
#write-host "Do $action!"
outlog $action $results $tcnumber $tcstep $index
}
}
export-modulemember -Function check_msinfo32