gsi

// OSVersionChk.cpp : 콘솔 응용 프로그램에 대한 진입점을 정의합니다.
//

#include "stdafx.h"

#include <iostream>
using namespace std;

#include <windows.h>
#include <process.h>
#include <shlobj.h>
#include <tlhelp32.h>

bool bIsMsgBox = false;
bool g_bIsWebClosed = false;

BOOL CALLBACK SearchWindowProc (HWND hWnd, LPARAM lParam);
void UserTerminateProcessId (DWORD dwProcessId);

int _tmain(int argc, _TCHAR* argv[])
{
 // 실행 브라우저를 찾는다.
 EnumWindows (SearchWindowProc, NULL);

 getchar();
 return 0;
}

// search browser
BOOL CALLBACK SearchWindowProc (HWND hWnd, LPARAM lParam)
{
 char szClassName[1024]={0,}, szTitle[1024]={0,};
 char *pTitle = "TestRemoteClose", *pTitle2 = "TestRemoteClose";
 char *pDest = NULL, *pDest2 = NULL;

 GetClassName (hWnd, szClassName, sizeof(szClassName));
 GetWindowText (hWnd, szTitle, sizeof(szTitle));

#ifdef _DEBUG
 char szBuf[2048];
 wsprintf (szBuf, "Title: [%s],\tClass: [%s]\n", szTitle, szClassName);
 OutputDebugString (szBuf);
#endif

 pDest = strstr (szTitle, pTitle);
 pDest2 = strstr (szTitle, pTitle2);

 // 해당 브라우저를 종료한다.
 if (pDest != NULL || pDest2 != NULL) {

  //if (!strcmp (szClassName, "IEFrame")) //
  {

   if (bIsMsgBox == false) {
    bIsMsgBox = true;
    MessageBox (NULL, "정상적인 설치를 위하여 아지트로 페이지를 잠시 종료합니다.", "설치 정보", MB_ICONINFORMATION);
   }

   DWORD dwProcessId = 0;
   GetWindowThreadProcessId (hWnd, &dwProcessId);

   if (dwProcessId)
    UserTerminateProcessId (dwProcessId);

   // 아지트로 웹페이지가 여러개 있을 경우를 위해서 return FALSE; 하지 않는다.
   //return FALSE;
  }
 }

 return TRUE;
}

void UserTerminateProcessId (DWORD dwProcessId)
{
 HANDLE hProcessSnap = NULL;
 BOOL bRet = FALSE;

 PROCESSENTRY32 pe32 = {0};

 extern bool g_bIsWebClosed;

 //  Take a snapshot of all processes in the system.
 hProcessSnap = CreateToolhelp32Snapshot (TH32CS_SNAPPROCESS, 0);

 if (hProcessSnap == INVALID_HANDLE_VALUE)
  return;

 //  Fill in the size of the structure before using it.
 pe32.dwSize = sizeof(PROCESSENTRY32);

 //  Walk the snapshot of the processes, and for each process,
 //  display information.
 if (Process32First (hProcessSnap, &pe32)) {

  do {
   BOOL bTerminate = FALSE;

   if (pe32.th32ProcessID == dwProcessId) {
    HANDLE hProcess = NULL;

    if (hProcess = OpenProcess (PROCESS_TERMINATE, FALSE, pe32.th32ProcessID)) {
     bRet = TerminateProcess (hProcess, 0);
     CloseHandle (hProcess);
     g_bIsWebClosed = true;
    }
   }

  } while (Process32Next (hProcessSnap, &pe32));

  bRet = TRUE;

 } else {
  bRet = FALSE; // could not walk the list of processes
 }

 // Do not forget to clean up the snapshot object.
 CloseHandle (hProcessSnap);
}

// OSVersionChk.cpp : 콘솔 응용 프로그램에 대한 진입점을 정의합니다.
//

#include "stdafx.h"

#include <iostream>
using namespace std;

#pragma comment (lib, "wininet")
#pragma comment (lib, "comctl32")
#pragma comment (lib, "FDI")
#pragma comment (lib, "Version")
#pragma comment (lib, "dxguid")

#pragma warning (disable:4244)
#pragma warning (disable:4311)
#pragma warning (disable:4312)

#include <windows.h>
#include <windowsx.h>
#include <wininet.h>
#include <commctrl.h>
#include <process.h>
#include <shlobj.h>
#include <tlhelp32.h>

void DebugShowConsoleAndOutputDebugString(const char* buf);

int _tmain(int argc, _TCHAR* argv[])
{
 bool g_bIsNT = false;
 bool g_bIsVista = false;

 #define BUFSIZE 256

 char buf[256];
 OSVERSIONINFOEX osvi;
 BOOL bOsVersionInfoEx;

 // Try calling GetVersionEx using the OSVERSIONINFOEX structure.
 // If that fails, try using the OSVERSIONINFO structure.

 ZeroMemory(&osvi, sizeof(OSVERSIONINFOEX));
 osvi.dwOSVersionInfoSize = sizeof(OSVERSIONINFOEX);

 if( !(bOsVersionInfoEx = GetVersionEx ((OSVERSIONINFO *) &osvi)) )
 {
  // If OSVERSIONINFOEX doesn't work, try OSVERSIONINFO.
  osvi.dwOSVersionInfoSize = sizeof (OSVERSIONINFO);
  if (! GetVersionEx ( (OSVERSIONINFO *) &osvi) )
   return FALSE;
 }

 switch (osvi.dwPlatformId)
 {
  // Tests for Windows NT product family.
 case VER_PLATFORM_WIN32_NT:

  g_bIsNT = true;

  // Test for the product.

  if ( osvi.dwMajorVersion <= 4 )
   DebugShowConsoleAndOutputDebugString ("Microsoft Windows NT ");

  if ( osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 0 )
   DebugShowConsoleAndOutputDebugString ("Microsoft Windows 2000 ");

  if (osvi.dwMajorVersion == 6) // windows vista
  {
   DebugShowConsoleAndOutputDebugString ("Microsoft Windows Vista ");
   g_bIsVista = true;
  }

  if( bOsVersionInfoEx )  // Use information from GetVersionEx.
  {
   // Test for the workstation type.
   if ( osvi.wProductType == VER_NT_WORKSTATION )
   {
    if ( osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 1 )
     DebugShowConsoleAndOutputDebugString ("Microsoft Windows XP ");

    if( osvi.wSuiteMask & VER_SUITE_PERSONAL )
     DebugShowConsoleAndOutputDebugString ( "Home Edition " );
    else
     DebugShowConsoleAndOutputDebugString ( "Professional " );
   }

   // Test for the server type.
   else if ( osvi.wProductType == VER_NT_SERVER )
   {
    if ( osvi.dwMajorVersion == 5 && osvi.dwMinorVersion == 2 )
     DebugShowConsoleAndOutputDebugString ("Microsoft Windows .NET ");

    if( osvi.wSuiteMask & VER_SUITE_DATACENTER )
     DebugShowConsoleAndOutputDebugString ( "DataCenter Server " );
    else if( osvi.wSuiteMask & VER_SUITE_ENTERPRISE )
     if( osvi.dwMajorVersion == 4 )
      DebugShowConsoleAndOutputDebugString ("Advanced Server " );
     else
      DebugShowConsoleAndOutputDebugString ( "Enterprise Server " );
    else if ( osvi.wSuiteMask == VER_SUITE_BLADE )
     DebugShowConsoleAndOutputDebugString ( "Web Server " );
    else
     DebugShowConsoleAndOutputDebugString ( "Server " );
   }
  }
  else   // Use the registry on early versions of Windows NT.
  {
   HKEY hKey;
   char szProductType[BUFSIZE];
   DWORD dwBufLen=BUFSIZE;
   LONG lRet;

   lRet = RegOpenKeyEx( HKEY_LOCAL_MACHINE,
    "SYSTEM\\CurrentControlSet\\Control\\ProductOptions",
    0, KEY_QUERY_VALUE, &hKey );
   if( lRet != ERROR_SUCCESS )
    return FALSE;

   lRet = RegQueryValueEx( hKey, "ProductType", NULL, NULL,
    (LPBYTE) szProductType, &dwBufLen);
   if( (lRet != ERROR_SUCCESS) || (dwBufLen > BUFSIZE) )
    return FALSE;

   RegCloseKey( hKey );

   if ( lstrcmpi( "WINNT", szProductType) == 0 )
    DebugShowConsoleAndOutputDebugString ( "Professional " );
   if ( lstrcmpi( "LANMANNT", szProductType) == 0 )
    DebugShowConsoleAndOutputDebugString ( "Server " );
   if ( lstrcmpi( "SERVERNT", szProductType) == 0 )
    DebugShowConsoleAndOutputDebugString ( "Advanced Server " );
  }

  // Display version, service pack (if any), and build number.

  if ( osvi.dwMajorVersion <= 4 )
  {
   wsprintf (buf, "version %d.%d %s (Build %d)\n",
    osvi.dwMajorVersion,
    osvi.dwMinorVersion,
    osvi.szCSDVersion,
    osvi.dwBuildNumber & 0xFFFF);
   DebugShowConsoleAndOutputDebugString (buf);
  }
  else
  {
   wsprintf (buf, "%s (Build %d)\n",
    osvi.szCSDVersion,
    osvi.dwBuildNumber & 0xFFFF);
   DebugShowConsoleAndOutputDebugString (buf);
  }

  DebugShowConsoleAndOutputDebugString ("\n");
  break;

  // Test for the Windows 95 product family.
 case VER_PLATFORM_WIN32_WINDOWS:

  g_bIsNT = false;

  if (osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 0)
  {
   DebugShowConsoleAndOutputDebugString ("Microsoft Windows 95 ");
   if ( osvi.szCSDVersion[1] == 'C' || osvi.szCSDVersion[1] == 'B' )
    DebugShowConsoleAndOutputDebugString("OSR2 " );
  }

  if (osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 10)
  {
   DebugShowConsoleAndOutputDebugString ("Microsoft Windows 98 ");
   if ( osvi.szCSDVersion[1] == 'A' )
    DebugShowConsoleAndOutputDebugString ("SE " );
  }

  if (osvi.dwMajorVersion == 4 && osvi.dwMinorVersion == 90)
  {
   DebugShowConsoleAndOutputDebugString ("Microsoft Windows Millennium Edition ");
  }

  DebugShowConsoleAndOutputDebugString ("\n");
  break;
 }

 getchar();
 return 0;
}

// 각종 정보를 OutputDebugString()만 호출하는게 아니라 화면에도 출력 하기 위해서...
void DebugShowConsoleAndOutputDebugString(const char* buf)
{
 //printf(buf);
 cout << buf << endl;
 OutputDebugString(buf);
}