마눌사장님의 컴철수 명령이 떨어져.....

소스 공개로 일단락 짓겠습니다(v 9.2)

오토핫키 잘 다루시는 분들이 완전체로 만들어주세요.

(제 컴에선 몇번을 제거 설치해도 잘됩니다)

 

1. 구글드라이브에 공유한 원본파일에

2. DISM_Onekey_BR배포용9.zip에 들어있는 배치파일 2개 교체  

3. 아래 소스로 만든 파일을 실행파일로 교체하셔야 합니다.

    DISM_Onekey_BR배포용9.2.ahk

 

저도 보완하여 공유하도록 하겠습니다.

그리고 오토핫키 설치하시면 얻는 것이 많을겁니다. 컴파일 않아도 되고...

DISM_Onekey_BR_Maker.ahk(UTF8로 저장하시고 실행하시면 백신 걱정없습니다.)

;-------------------------------------------------------------------------------------------------------------------------

#SingleInstance Force

#NoTrayIcon 

SetWorkingDir %A_ScriptDir%

if not A_IsAdmin

{

   Run *RunAs "%A_ScriptFullPath%"  

   ExitApp

}

UorB = % IsUEFIorBIOS()

FileDelete,%A_Temp%\DrvList.txt

DriveGet, DriveList, list

StringTrimLeft, DriveList, DriveList, 1

DriveGet, cdVar, List, CDROM

DriveList := trim(DriveList,cdVar)

Loop, Parse, DriveList

{

DriveGet, lbl, Label, %A_LoopField%:

DriveGet, fsys, FS, %A_LoopField%:

DriveGet, typ, Type, %A_LoopField%:

Sps := ""

DriveGet, Cap, Capacity, %A_LoopField%:

DriveSpaceFree, Free_Space, %A_LoopField%:

Sps =% Cap/1000

If (Free_Space > 1000)

{

Free_S =% Free_Space/1000

Unit = GB

StringTrimRight, Sps, Sps, 7

StringTrimRight, Free_S, Free_S, 7

}

Else

{

Unit = MB

}

FileAppend, %A_LoopField%: %Sps%%Unit%. .(%Free_S%%Unit%여유). .%typ% [%fsys%]. .%lbl%|, %A_Temp%\DrvList.txt

}

FileReadLine, DrvVar, %A_Temp%\DrvList.txt, 1

StringTrimRight, DrvVar, DrvVar, 1

For Prt in (ComObjGet("winmgmts:").ExecQuery("Select * FROM Win32_LogicalDiskToPartition"),DP:=[])

    DP.Insert(RegExReplace(Prt.Antecedent " " Prt.Dependent,"^.*?""|"".*"))

For Prp in (ComObjGet("winmgmts:").ExecQuery("Select * FROM Win32_LogicalDisk"),DINF:=[],i:=1)

    (f:=Prp.FileSystem) ? DINF.Insert(DP[i++] " " Prp.Name):

For i in (DINF,Disp:="")

    Disp.=DINF[i] "`r`n"

DetectHiddenWindows On

Run %ComSpec%,, Hide, pid

WinWait ahk_pid %pid%

DllCall("AttachConsole", "UInt", pid)

shell := ComObjCreate("WScript.Shell")

cmd := shell.Exec("cmd.exe /q /c diskpart")

cmd.StdIn.WriteLine("list disk`nexit")

result := isGpt(cmd.StdOut.ReadAll())

DllCall("FreeConsole")

Process Close, %pid%

HH := WMI_HDD_HEALTH()

HT := WMI_HDD_TEMP()

Gui, Add, GroupBox, x30 y5 w310 h206 ,

Gui, Add, GroupBox, x160 y13 w160 h188 ,

Gui, Add, GroupBox, x110 y333 w248 h30 ,

Gui, Add, Text, x111 y344 w246 vCopiedStatus,

Gui, Add, Text, x42 y24 w112 h20 , ▶〔 디스크 타입 〕: 

Gui, Add, Text, x168 y24 w150 h175 , %result%`n%Disp%

Gui, Add, Text, x42 y64 w110 h20 , ▼〔 디스크 건강 〕    

Gui, Add, Text, x42 y94 w110 h70 , `        ☞  %HH% `n`n        ☞  %HT% °C

Gui, Add, Text, x40 y168 w112 h20 , ` ♣ %UorB% ♣

Gui, Add, Text, x16 y224 w335 h20 , 1.〔 설치 〕✔〔 PE 〕를 넣을 드라이브 선택 ✔

Gui, Add, DDL, gDirSelectBt vDirDrv Choose1 x19 y250 w346, %DrvVar%

Gui, Add, Button, x42 y280 w96 h20 gPChange, ▶ 파티션 수정

Gui, Add, Button, x170 y280 w100 h20 gOpenfolder, ✔드라이브 확인

Gui, Add, Text, x282 y284 w27 h20 ,  ☜

Gui, Add, Button, x307 y280 w50 h20 gDirCopyBt, 넣 기

Gui, Add, Text, x80 y314 w280 h20 , * 바탕화면 우메뉴에 등록까지 진행됩니다.   ↑↑↑

Gui, Add, Text, x16 y374 w286 h20 , 2.  우메뉴에서〔원키_백업〕만 제거 . . . . . . . .  ☞

Gui, Add, Button, x307 y370 w50 h20 gAfterBackupBt, 제 거

Gui, Add, Text, x16 y404 w284 h60 , 3.〔DISM_원키〕레지확인 . . . . . . . . . . . . . .   ☞

Gui, Add, Button, x307 y400 w50 h20 gDetectDismBt, 확 인

Gui, Add, Text, x16 y434 w284 h60 , 4.〔DISM_원키〕제거. . . . . . . . . . . . . . . . . .  ☞

Gui, Add, Button, x307 y430 w50 h20 gRemoveDismBt, 제 거

Gui, Add, Text, x16 y464 w284 h60 , 5.  백업 중〔비정상 종료〕했다면. . . . . . . . . .  ☞

Gui, Add, Button, x307 y460 w50 h20 gCareBt, 치 유

Gui, Show, x450 y224 h512 w386, `     〔 DISM_원키_복원 〕_ Maker

return 

GuiClose:

FileDelete,dir.ini

ExitApp

PChange:

gui,submit,nohide

runwait Partitionwizard10.2.2.exe

WinWaitclose, MiniTool Partition Wizard Technician 10.2.2

reload

return

DirSelectBT:

gui,submit,nohide

StringLeft,DirSelect,DirDrv,1

IniWrite, %DirSelect%:, dir.ini, dir, dir 

PDir = %DirSelect%:

Return

Openfolder:

gui,submit,nohide

IfExist, dir.ini

{

iniread, dir, dir.ini, dir, dir

IfExist, %dir%\Dism_Backup

run, %dir%\Dism_Backup

else

{

run, %dir%

}

return

}

GuiControl,, CopiedStatus, 먼저〔드라이브〕를 선택하세요. . . . 

Sleep, 800

GuiControl,, CopiedStatus,

Sleep, 500

return

DirCopyBt:

gui,submit,nohide

IfNotExist, dir.ini

{

GuiControl,, CopiedStatus, 먼저〔드라이브〕를 선택하세요. . . . 

Sleep, 1800

GuiControl,, CopiedStatus,

Sleep, 500

return

}

FileCreateDir, %PDir%\DISM_Backup   

runwait nircmd shellcopy "DISM_PE\*.*" "%PDir%\DISM_Backup" yestoall noerrorui

Sleep, 500

GuiControl,, CopiedStatus, 〔필수파일〕복사 완료. . . . 

Sleep, 800

GuiControl,, CopiedStatus,

Sleep, 500

FileAppend, , %A_WinDir%\Chk_C.wim

GuiControl,, CopiedStatus, 〔C:\ 체크파일〕 생성 완료. . . . 

Sleep, 800

GuiControl,, CopiedStatus,

Sleep, 500

fileread, text_list, %PDir%\DISM_Backup\_.bat

RegExMatch(text_list, ".+?\v", oVar)                                   

text_list := RegExReplace(text_list, ".+?\v", "", "", 1)

Sleep, 300         

filedelete, %PDir%\DISM_Backup\_.bat

fileappend,

(

set b="_.bat"&set wim=%PDir%\DISM_Backup\boot.wim&set pdrv=%PDir%

), %PDir%\DISM_Backup\_.bat

fileappend, %text_list%, %PDir%\DISM_Backup\_.bat   

Sleep, 500

RunWait, nircmd elevatecmd exec hide %PDir%\DISM_Backup\_.bat

Sleep, 1500

GuiControl,, CopiedStatus, 〔PE 부팅 등록〕완료. . . . 

Sleep, 800

GuiControl,, CopiedStatus,

Sleep, 500

FileRead, DDD, %PDir%\DISM_Backup\guid.txt

{

if ErrorLevel = 1

{

GuiControl,, CopiedStatus,  〔우메뉴에 등록〕실패. . . . 

Sleep, 1800

GuiControl,, CopiedStatus, 〔DISM 원키 BR〕설치 미완료. . . .

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

return

}

RegWrite, REG_SZ, HKCR,DesktopBackground\shell\DISM_원키복원, Icon, shell32.dll,146

RegWrite, REG_SZ, HKCR,DesktopBackground\shell\DISM_원키복원, Position, Bottom

RegWrite, REG_SZ, HKCR,DesktopBackground\shell\DISM_원키복원, SubCommands,

RegWrite, REG_SZ, HKCR,DesktopBackground\shell\DISM_원키복원\shell\런원스_PE, Icon, shell32.dll,146

RegWrite, REG_SZ, HKCR,DesktopBackground\shell\DISM_원키복원\shell\런원스_PE\command, , mshta vbscript:Execute("With CreateObject(""WScript.Shell""):.Run ""nircmd execmd del """"%PDir%\DISM_Backup\Onekey_*.cmd"""""", 0, True:.Run ""bcdedit /bootsequence %DDD%"", 0, True:.Run ""nircmd wait 500"", 0, True:.Run ""shutdown.exe -r -t 0"", 0, True:End With:close")

RegWrite, REG_SZ, HKCR,DesktopBackground\shell\DISM_원키복원\shell\원키_백업, Icon, shell32.dll,146

RegWrite, REG_SZ, HKCR,DesktopBackground\shell\DISM_원키복원\shell\원키_백업\command, , mshta vbscript:Execute("With CreateObject(""WScript.Shell""):.Run ""nircmd execmd del """"%PDir%\DISM_Backup\Onekey_*.cmd"""""", 0, True:.Run ""nircmd shellcopy """"%PDir%\DISM_Backup\_Backup.cmd"""" """"%PDir%\DISM_Backup\Onekey_Backup.cmd"""" yestoall noerrorui"", 0, True:.Run ""bcdedit /bootsequence %DDD%"", 0, True:.Run ""nircmd wait 500"", 0, True:.Run ""shutdown.exe -r -t 0"", 0, True:End With:close")

RegWrite, REG_SZ, HKCR,DesktopBackground\shell\DISM_원키복원\shell\원키_복원, Icon, shell32.dll,146

RegWrite, REG_SZ, HKCR,DesktopBackground\shell\DISM_원키복원\shell\원키_복원\command, , mshta vbscript:Execute("With CreateObject(""WScript.Shell""):.Run ""nircmd execmd del """"%PDir%\DISM_Backup\Onekey_*.cmd"""""", 0, True:.Run ""nircmd shellcopy """"%PDir%\DISM_Backup\_Restore.cmd"""" """"%PDir%\DISM_Backup\Onekey_Restore.cmd"""" yestoall noerrorui"", 0, True:.Run ""bcdedit /bootsequence %DDD%"", 0, True:.Run ""nircmd wait 500"", 0, True:.Run ""shutdown.exe -r -t 0"", 0, True:End With:close")

GuiControl,, CopiedStatus,  〔우메뉴에 등록〕완료. . . . 

Sleep, 800

GuiControl,, CopiedStatus, 〔DISM 원키복원〕설치 완료. . . .

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

return

}

AfterBackupBt:

gui,submit,nohide

count=0

Loop, HKCR, DesktopBackground\shell\DISM_원키복원\shell\원키_백업, 1

  count:=A_INDEX

if count=0

{ 

GuiControl,, CopiedStatus, 〔원키_백업〕레지 존재 안함. . . . 

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

} 

else

{

RegDelete, HKCR,DesktopBackground\shell\DISM_원키복원\shell\원키_백업

GuiControl,, CopiedStatus, 우메뉴에서 〔원키_백업〕제거 완료. . . . 

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

} 

return

DetectDismBt:

gui,submit,nohide

count=0

Loop, HKCR, DesktopBackground\shell\DISM_원키복원, 1

  count:=A_INDEX

if count=0

{ 

GuiControl,, CopiedStatus, 〔DISM_원키복원〕레지 존재 안함. . . . 

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

} 

else

{

Loop, HKCR, DesktopBackground\shell\DISM_원키복원\shell\원키_백업\command, 1, 1

{

if a_LoopRegType = key

value =

else

{

RegRead, value

if ErrorLevel <> 0

value = *error*

}

MsgBox, %a_LoopRegName% = %value% (%a_LoopRegType%)

}

GuiControl,, CopiedStatus, 〔DISM_원키백업〕레지 확인 . . . . 

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

}

return

RemoveDismBt:

gui,submit,nohide

DriveGet, DrivegList, list

Loop, Parse, DrivegList

{

IfExist, %A_LoopField%:\Dism_Backup

pedrv =%A_LoopField%:

}

IfExist, %pedrv%\DISM_Backup\guid.txt

{

msgbox, 4, 〔_DISM_원키_제거_〕, `  〔DISM_원키〕를 제거합니다.!`n `n레지와 DISM_Backup폴더가 제거됩니다.`n `nBCD에서도 제거됩니다`n `n〔아니오〕 누르면 제거되지 않습니다.

IfMsgBox Yes

{

RegDelete, HKCR,DesktopBackground\shell\DISM_원키복원

GuiControl,, CopiedStatus, 〔DISM_원키복원〕레지 제거 완료. . . . 

Sleep, 800

FileRead, OOD, %pedrv%\DISM_Backup\guid.txt

Runwait, %comspec% /c "bcdedit /delete "%OOD%" /f", , hide

GuiControl,, CopiedStatus, 〔PE〕BCD에서 제거 완료. . . . 

Sleep, 800

FileRemoveDir, %pedrv%\DISM_Backup, 1

GuiControl,, CopiedStatus, 〔DISM_Backup〕폴더 삭제. . . . 

Sleep, 800

GuiControl,, CopiedStatus, 〔DISM_원키복원〕제거 완료. . . . 

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

return

}

}

GuiControl,, CopiedStatus, 〔DISM_원키〕존재 안함. . . . 

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

return

CareBt:

gui,submit,nohide

DriveGet, DrivecList, list

Loop, Parse, DrivecList

{

IfExist, %A_LoopField%:\Dism_Backup

Cpedrv =%A_LoopField%:

}

IfExist, %Cpedrv%\DISM_Backup\windows*.wim

{

FileDelete,%Cpedrv%\DISM_Backup\windows*.wim

FileDelete,%Cpedrv%\DISM_Backup\Onkey_*.cmd

GuiControl,, CopiedStatus, 〔백업이미지〕삭제 완료. . . . 

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

return

}

GuiControl,, CopiedStatus, 〔백업이미지〕존재 안함. . . . 

Sleep, 3000

GuiControl,, CopiedStatus,

Sleep, 500

return

WMI_HDD_HEALTH()

{

    for objItem in ComObjGet("winmgmts:\\.\root\WMI").ExecQuery("SELECT * FROM MSStorageDriver_FailurePredictStatus")

        return, % (objItem.PredictFailure = "0") ? "양호 ^^" : (objItem.PredictFailure = "1") ? "Critical" : "불량 ㅠㅠ"

}

WMI_HDD_TEMP()

{

    for objItem in ComObjGet("winmgmts:\\.\root\WMI").ExecQuery("SELECT * FROM MSStorageDriver_ATAPISmartData")

    {

        loop, 361

        {

            i := A_Index - 1

            if (objItem.VendorSpecific[i] = "194")

            {

                j := i + 5

                return, % objItem.VendorSpecific[j]

            }

        }

    }

}

IsUEFIorBIOS()

{

static DUMMY_GUID := "{00000000-0000-0000-0000-000000000000}"

static ERROR_INVALID_FUNCTION := 0x1

DllCall("GetFirmwareEnvironmentVariable", "str", "", "str", DUMMY_GUID, "ptr", 0, "uint", 0, "uint")

return (A_LastError = ERROR_INVALID_FUNCTION) ? "Legacy BIOS" : "UEFI Mode"

}

isGpt(string)

{

Loop, Parse, string, `n

{

if (A_index > 9)

{

Disk := SubStr(A_LoopField, 3, 6)

if (flag != "")

if (!InStr(Disk, "디스크"))

break

RegExReplace(A_LoopField, "\*","fgh",count)

if (count = 0)

flag := "MBR No-Dyn"

else if (count = 1)

flag := StrLen(RegExReplace(A_LoopField, "(.*)\*")) = 1 ? "Gpt No-Dyn" : "MBR Dyn"

else if (count = 2)

flag := "Gpt Dyn"

result .= Disk ", " flag "`n"

}

}

return result

}