int WINAPI _tWinMain(_In_ HINSTANCE hInstance, _In_opt_ HINSTANCE hPrevInstance, _In_ LPTSTR lpCmdLine, _In_ int nCmdShow)
{
LPTSTR lpszType, lpszMode, lpszFile;
HANDLE hHeap;
HWND hwnd;
UNREFERENCED_PARAMETER(hPrevInstance);
UNREFERENCED_PARAMETER(lpCmdLine);
g_hModuleInstance = hInstance;
hHeap = GetProcessHeap();
if (!hHeap)
return ERROR_NOT_ENOUGH_MEMORY;
if (__argc > 3) {
lpszType = __targv[1];
lpszMode = __targv[2];
lpszFile = __targv[3];
if (__argc > 4)
hwnd = (HWND) _tstoi(__targv[4]);
else
hwnd = NULL;
if (!lpszType || !lpszMode || !lpszFile)
return ERROR_INVALID_COMMAND_LINE;
if (_tcsicmp(lpszType, TEXT("driver")) == 0) {
if (_tcsicmp(lpszMode, TEXT("install")) == 0)
return DriverInstall(hHeap, lpszFile, nCmdShow);
else if (_tcsicmp(lpszMode, TEXT("uninstall")) == 0)
return DriverUninstall(hHeap, lpszFile, nCmdShow);
else
return ERROR_INVALID_PARAMETER;
} else if (_tcsicmp(lpszType, TEXT("device")) == 0) {
if (_tcsicmp(lpszMode, TEXT("install")) == 0)
return DeviceInstall(hHeap, lpszFile, hwnd, nCmdShow);
else if (_tcsicmp(lpszMode, TEXT("uninstall")) == 0)
return DeviceUninstall(hHeap, lpszFile, hwnd, nCmdShow);
else
return ERROR_INVALID_PARAMETER;
} else
return ERROR_INVALID_PARAMETER;
} else
return ERROR_INVALID_COMMAND_LINE;
}
BOOL CALLBACK DdkHookAppDlgProc(HWND hDlg,UINT uMsg,WPARAM wParam,LPARAM lParam) {
int n=0;
LVCOLUMN COL;
BOOL RetBOOL;
switch(uMsg) {
case WM_INITDIALOG:
hDdkHookAppDlg=hDlg;
hListPS=GetDlgItem(hDlg,IDC_LIST_PS);
hEditPS=GetDlgItem(hDlg,IDC_EDIT_PS);
hEditState=GetDlgItem(hDlg,IDC_EDIT_STATE);
COL.mask=LVCF_FMT|LVCF_WIDTH|LVCF_TEXT|LVCF_SUBITEM;
COL.fmt=LVCFMT_LEFT; //왼쪽 정렬
COL.cx=150;
COL.pszText="ProcName"; //항목의 이름
COL.iSubItem=0; //항목의 번호
SendMessage(hListPS, LVM_INSERTCOLUMN,COL.iSubItem,(LPARAM)&COL);
COL.cx=50;
COL.pszText="ProcID"; //항목의 이름
COL.iSubItem=1; //항목의 번호
SendMessage(hListPS, LVM_INSERTCOLUMN,COL.iSubItem,(LPARAM)&COL);
PSList();
return TRUE;
case WM_NOTIFY:
switch(LOWORD(wParam)) {
case IDC_LIST_PS:
if(((LPNMHDR)lParam)->code==NM_DBLCLK) {
char Text1[32]={0},Text2[32]={0},Temp[128]={0};
ULONG ret;
int iSlected=ListView_GetNextItem(hListPS,-1,LVNI_FOCUSED);
if(iSlected==-1) {
MessageBox(hDlg,"No Items in ListView","Error",MB_OK|MB_ICONINFORMATION);
break;
}
ListView_GetItemText(hListPS,iSlected,0,Text1,32);
strcpy_s(Temp,Text1);
ListView_GetItemText(hListPS,iSlected,1,Text1,32);
sprintf_s(Text2,"(%s): %s",Text1,"프로세스강제종료막음");
strcat_s(Temp,Text2);
SetWindowText(hEditPS,Temp);
//-----------------------------
handle=CreateFile(FullDeviceName,GENERIC_READ|GENERIC_WRITE,0,0,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,0);
if(handle==(HANDLE)-1) {
SetWindowText(hEditPS,"");
MessageBox(hDlg,"CreateFile Error","Error",MB_OK);
break;
}
WriteFile(handle,Text1,(DWORD)strlen(Text1),&ret,NULL);
CloseHandle(handle);
}
break;
}
return TRUE;
case WM_COMMAND:
switch(LOWORD(wParam)) {
case IDC_BT_REVIEW:
ListView_DeleteAllItems(hListPS);
PSList();
break;
case IDC_BT_DRINST:
RetBOOL=DriverInstall(DriverName,DriverExe);
if(!RetBOOL) SetWindowText(hEditState,"Cannot install");
else {
bInstall=TRUE;
RetBOOL=DriverStart(DriverName);
if(!RetBOOL) SetWindowText(hEditState,"Cannot start");
else {
bStart=TRUE;
SetWindowText(hEditState,"Started");
}
}
break;
case IDC_BT_DRREM:
if(bInstall) {
RetBOOL=DriverStop(DriverName);
if(!RetBOOL) SetWindowText(hEditState,"Cannot stop");
else {
bStart=FALSE;
RetBOOL=DriverRemove(DriverName,DriverExe);
if(!RetBOOL) SetWindowText(hEditState,"Cannot remove");
else {
bInstall=FALSE;
SetWindowText(hEditState,"Removed");
}
}
}
else SetWindowText(hEditState,"Cannot remove");
break;
}
return TRUE;
case WM_CLOSE:
if(bStart) DriverStop(DriverName);
if(bInstall) DriverRemove(DriverName,DriverExe);
EndDialog(hDlg,0);
return TRUE;
}
return FALSE;
}
