2016년 2월 20일 토요일

[VS2010 VC++] OSVersion.h 파일


////////////////////////////////////////////////////////////////////////////////////
//
// < COSVersion >
//
// Retrieve the Windows OS version.
//
// * AUTHOR
// Nam, in hyouk (GreenB)
//
// * ACKNOWLEDGEMENT
// Hans Dietrich(hdietrich@gmail.com)
// I found it by reference to his "CXWinVer" class.
// Please see his article at www.codeproject.com
//
// * HISTORY
// 2009-04-11 Initial release.
// 2010-12-06 IsWin7() added.
//
// * LICENCE
//     This software is released into the public domain. You are free to use
//     it in any way you like, except that you may not sell this source code.
//
//     This software is provided "as is" with no expressed or implied warranty.
//     I accept no liability for any damage or loss of business that this
//     software may cause.
//
//
// copyright(c) 2009-2010. GreenB. All right reserved.
// http://greenmaru.com
//
//
////////////////////////////////////////////////////////////////////////////////////
#ifndef _GREENB_OSVERSION_
#define _GREENB_OSVERSION_
#pragma once

 #include <windows.h>

 ////////////////////////////////////////////////////////////////////////////////////
//
// COSVersion
//
////////////////////////////////////////////////////////////////////////////////////
class COSVersion
{
////////////////////////////////////////////////////////////////////////////////
// MEMBER VARS
////////////////////////////////////////////////////////////////////////////////
private:
OSVERSIONINFOEX m_osver;

 ////////////////////////////////////////////////////////////////////////////////
// CONSTRUCTION
////////////////////////////////////////////////////////////////////////////////
public:
COSVersion( void )
{
memset( &m_osver, NULL, sizeof( OSVERSIONINFOEX ) );
m_osver.dwOSVersionInfoSize = sizeof( OSVERSIONINFO );

 if ( GetVersionEx( reinterpret_cast<LPOSVERSIONINFO>(&m_osver) ) )
{
//
// Is NT4 SP6 or later?
//
if ( ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT ) &&  (m_osver.dwMajorVersion >= 5 ) )
{
memset( &m_osver, NULL, sizeof( OSVERSIONINFOEX ) );
m_osver.dwOSVersionInfoSize = sizeof( OSVERSIONINFOEX );
GetVersionEx( reinterpret_cast<LPOSVERSIONINFO>(&m_osver) );
}
}
}

 ~COSVersion( void )
{
}

 ////////////////////////////////////////////////////////////////////////////////
// PROPERTY
////////////////////////////////////////////////////////////////////////////////
public:
DWORD MajorVersion( void ) const
{
return m_osver.dwMajorVersion;
}

 DWORD MinorVersion( void ) const
{
return m_osver.dwMinorVersion;
}

 DWORD BuildNumber( void ) const
{
return ( m_osver.dwBuildNumber & 0xFFFF );
}

 DWORD ServicePack( void ) const
{
int nServicePack = 0;
for ( int i = 0 ; ( m_osver.szCSDVersion[i] != _T('\0') ) && ( i < (sizeof(m_osver.szCSDVersion) / sizeof(TCHAR) ) ) ; i++ )
{
if ( _istdigit( m_osver.szCSDVersion[i] ) )
{
nServicePack = _ttoi( &m_osver.szCSDVersion[i] );
break;
}
}
return nServicePack;
}

 //
// Windows 9x
//
BOOL IsWin9x( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_WINDOWS && m_osver.dwMajorVersion == 4 )
return TRUE;
return FALSE;
}

 //
// Windows NT
//
BOOL IsWinNT4( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT && m_osver.dwMajorVersion == 4 && m_osver.dwMinorVersion == 0 )
return TRUE;
return FALSE;
}
BOOL IsWinNT4SP6( void ) const
{
if ( IsWinNT4() && ServicePack() == 6 )
return TRUE;
return FALSE;
}

 //
// Windows XP / 2000 / 2003
//
BOOL IsWinXp( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT && m_osver.dwMajorVersion == 5 && m_osver.dwMinorVersion == 1 )
return TRUE;
return FALSE;
}
BOOL IsWinXpSP2( void ) const
{
if ( IsWinXp() && ServicePack() == 2 )
return TRUE;
return FALSE;
}
BOOL IsWinXpHome( void ) const
{
if ( IsWinXp() && m_osver.wSuiteMask & VER_SUITE_PERSONAL )
return TRUE;
return FALSE;
}
BOOL IsWinXpPro( void ) const
{
if ( IsWinXp() && m_osver.wProductType == VER_NT_WORKSTATION && !IsWinXpHome() )
return TRUE;
return FALSE;
}
BOOL IsWinXpOrLater( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT && m_osver.dwMajorVersion >= 5 )
return TRUE;
return FALSE;
}

 BOOL IsWin2000( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT && m_osver.dwMajorVersion == 5 && m_osver.dwMinorVersion == 0 )
return TRUE;
return FALSE;
}

 BOOL IsWin2003( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT && m_osver.dwMajorVersion == 5 && m_osver.dwMinorVersion == 2 )
return TRUE;
return FALSE;
}

 //
// Windows Vista / 2008 / 7
//
BOOL IsVista( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT && m_osver.dwMajorVersion == 6 && m_osver.dwMinorVersion == 0 && m_osver.wProductType == VER_NT_WORKSTATION )
return TRUE;
return FALSE;
}
BOOL IsVistaOrLater( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT && m_osver.dwMajorVersion >= 6 )
return TRUE;
return FALSE;
}

 BOOL IsWin2008( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT && m_osver.dwMajorVersion == 6 && m_osver.dwMinorVersion == 0 && m_osver.wProductType != VER_NT_WORKSTATION )
return TRUE;
return FALSE;
}

 BOOL IsWin7( void ) const
{
if ( m_osver.dwPlatformId == VER_PLATFORM_WIN32_NT && m_osver.dwMajorVersion == 6 && m_osver.dwMinorVersion == 1 && m_osver.wProductType == VER_NT_WORKSTATION )
return TRUE;
return FALSE;
}

 //
// Retrieve the server or workstation
//
BOOL IsServerOS( void ) const
{
if ( m_osver.wProductType > VER_NT_WORKSTATION )
return TRUE;
return FALSE;
}

 BOOL IsDomainController( void )
{
return ( m_osver.wProductType && VER_NT_DOMAIN_CONTROLLER );
}
};

작성자: 댓글 없음:

[VS2010 VC++] 운영체제 버전을 확인하기

요즘 Windows Vista 때문에 코드를 뜯어고쳐야 하는 경우가 간혹 있다.
한편으론 짜증나지만 예쁘니까 계속 쓰게된다 -_-;
아무튼. APP가 실행중인 윈도우 플랫폼을 식별할 일이 있을때 마다, 상수값을 찾느라 MSDN 뒤적거리는것도 귀찮아서 이참에 클래스로 만들었다. 코드자체는 단순하지만, 혹시나 필요하신 분이 있을까 싶어 올려둔다.
별도로 사용법을 설명하지 않아도 될 정도로 간단하다.
UPDATE: 2010-12-06 Window7 판별 함수 추가
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#include "OSVersion.h"
 
COSVersion osver;
 
if ( osver.IsVista() )
{
    MessageBox( NULL, _T("이 운영체제는 Vista임"), _T(""), MB_OK );
}
if ( ! osver.IsServerOS() )
{
    MessageBox( NULL, _T("서버운영체제가 아님"), _T(""), MB_OK );
}
if ( osver.IsVistaOrLater() )
{   
    MessageBox( NULL, _T("Vista 이상에서 사용가능한 API 쓸 수 있음"), _T(""), MB_OK );
}
작성자: 댓글 없음:
라벨: , , ,

[VS2010 VC++] ATL 작성하기 - (4) 권한상승시키기

오늘은 연재의 마지막으로, Windows Vista 이상의 운영체제에서 동작하기 위한 작업을 진행하겠다.

관리자 권한: UAC(User Access Control)
아시다시피, Windows Vista부터 UAC(User Access Control)라는 개념이 도입 되었다.
UAC는 한마디로, "시스템에 중대한 영향을 끼치는 작업을 하려면 적절한 권한을 가지고 있을 것"이라고 할 수있다. 보통때는 "일반 사용자 권한"으로 사용하다가, 새로운 어플리케이션의 설치나 레지스트리 편집처럼 시스템에 중대한 변경을 가하는 작업을 하기 위해서는 "관리자 권한"이 있어야 한다.


여기서 한단계 더 나아가, 인터넷 익스플로러는 "보호모드(Low IL)"에서 동작하게 되었고, 이 위에서 동작하는 ActiveX도 덩달아 많은 제약을 받게 되었다. 이러한 변화는 최근 대부분의 바이러스가 ActiveX형태로 전파되는 점을 감안하면 반드시 필요한 조치이긴 하다.
아무튼, 이제 사용자 PC에 있는 파일을 읽고 쓰거나 하는 저수준 작업은 원칙적으로 ActiveX에서 허용되지 않는다. 이러한 작업을 하고 싶다면, ActiveX가 관리자 권한을 획득해야 한다.

프로그램 흐름
최종 결과물 Javascript를 통해 전체 프로그램의 흐름을 파악해 보자.
여러가지 방법이 있지만, 나는 Javascript에서 프로그램의 전체 흐름을 제어하고, ActiveX는 필요한 기능만을 제공하는 방법을 선호한다.
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
function runInUAC() {
 
    var helloCtrl = null;
 
    try {
 
        //
        // 1. ActiveX 컨트롤의 인스턴스를 생성.
        //
        helloCtrl = new ActiveXObject("GreenmaruX.HelloCtrl");
        if (helloCtrl == null) {
            // HelloCtrl이 시스템에 설치되지 않은 경우의 처리.
            alert("GreenmaruX HelloCtrl was not installed.");
        }
 
        //
        // 2. UAC가 필요한 OS인지를 판단.
        //
        if (helloCtrl.IsNeedElevation) {
 
            //
            // 3. 관리자 권한이 적용된 개체를 구함.
            //
            helloCtrl = helloCtrl.GetElevationObject();
        }
 
        //
        // 4. 관리자 권한이 필요한 저수준 기능을 사용.
        //
        helloCtrl.SomeFunction();
    }
    catch (ex) {
        var em = "ERROR:" + ex.message + "(" + ex.number + ")";
        alert(em);
    }
}
대충 감을 잡으셨는가? 그러면 하나씩 구현해 보도록 하자. 다만, 관리자 권한을 획득하는 과정까지가 중요하기 때문에 SomeFunction은 실제로 구현하지 않겠다.

UAC 필요여부 판단하기
아직도 Windows XP는 많이 사용되고 있기 때문에, 무조건 UAC를 진행할 필요는 없다. Windows OS를 판별해서 UAC가 필요한 운영체제인지를 판단하도록 하자. OS버전을 알아내기 위해서 COSVersion이라는 클래스를 사용했다. 이 클래스에 대한 설명은 다음 포스트를 참조하시기 바란다.
1
2
3
4
5
6
7
HRESULT FinalConstruct()
{
    // Vista이상의 운영체제는 UAC elevation이 필요합니다.
    Greenmaru::COSVersion osv;
    _isNeedElevation = osv.IsWindowsVistaOrLater();
    return S_OK;
}
_isNeedElevation이라는 BOOL형 멤버변수를 추가하고, FinalConstruct에서 UAC의 필요 여부에 따라 값을 설정하도록 했다. 참고로, FinalConstruct는 이름처럼 개체 생성이 완료되는 시점에서 호출되는 함수다.
그리고 이 값을 개체 외부로 알리기 위한 IsNeedElevation COM Property를 추가해 준다. 이 값은 개체 외부에서 설정할 필요는 없으므로(해서도 안되므로) Get함수만 구현하도록 하겠다.
1
2
3
4
5
STDMETHODIMP CHelloCtrl::get_IsNeedElevation(VARIANT_BOOL* pVal)   
{   
    *pVal = m_isNeedElevation;   
    return S_OK;   
} 

UAC Elevation
UAC 권한상승(Elevation)의 원리는 사실 간단하다. ActiveX객체가 자신의 새로운 인스턴스를 권한상승을 거쳐 만들도록 하면 된다.
GetElevationObject라는 COM Method를 IHelloCtrl에 추가하자. 반환형은 VARIANT*형으로 지정한다. 이 함수는 권한 상승된 HelloCtrl의 새로운 인스턴스를 반환해 준다.
HelloCtrl.cpp에서 strsafe.h를 Include해 준 다음, 함수를 다음과 같이 구현하록 한다.
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
STDMETHODIMP CHelloCtrl::GetElevationObject(VARIANT* lpvObject)
{
    HRESULT hr;
    BIND_OPTS3 bo;
    WCHAR wszCLSID[64];
    WCHAR wszMonikerName[512];
    IHelloCtrl * objElevator = NULL;
 
    if ( ! m_isNeedElevation )
    {
        lpvObject->vt = VT_DISPATCH;
        lpvObject->pdispVal = this;
        return S_OK;
    }
 
    // 권한 상승을 시도합니다.
    StringFromGUID2( CLSID_HelloCtrl, wszCLSID, sizeof(wszCLSID)/sizeof(wszCLSID[0]) );
    hr = StringCchPrintfW( wszMonikerName,
            sizeof(wszMonikerName)/sizeof(wszMonikerName[0]),
            L"Elevation:Administrator!new:%s", wszCLSID );
    if ( FAILED(hr) )
    {
        lpvObject->vt = VT_NULL;
        return S_FALSE;
    }
 
    memset( &bo, NULL, sizeof(bo) );
    bo.cbStruct = sizeof(bo);
    bo.hwnd = NULL;
    bo.dwClassContext = CLSCTX_LOCAL_SERVER;
 
    hr =  CoGetObject( wszMonikerName, &bo, IID_IHelloCtrl, (void**)&objElevator );
    if ( FAILED(hr) )
    {
        lpvObject->vt = VT_NULL;
        return S_FALSE;
    }
 
    lpvObject->vt = VT_DISPATCH;
    lpvObject->pdispVal = objElevator;
    lpvObject->pdispVal->AddRef();
 
    return S_OK;
}
천천히 살펴보면 눈치챌 수 있겠지만, 핵심은 CoGetObject를 통해 COM개체 자신의 새로운 인스턴스를 만들어 내는 것이다. 다만 일반적인 CoCreateObject등의 방식으로는 Elevation처리를 거칠 수 없기 때문에 조금 더 복잡해 졌다.

그 다음, 문자열 리소스를 하나 추가한다. 나는 IDS_ELEVATION이라고 이름지었는데, 이 문자열은 Elevation과정에서 사용자에게 컨트롤의 이름으로 표시된다.
그리고, DLL이름.rgs파일(Greenmaru.rgs)을 다음과 같이 만든다.
아시다시피, rgs파일의 내용은 COM개체를 시스템에 등록할 때(regsvr32.exe) 레지스트리에 기록된다. 다른 컨트롤을 만들때는 대충 어디를 어떻게 바꿔야 할지를 짐작할 수 있을 것이다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
HKCR
{
    NoRemove AppID
    {
        '%APPID%' = s 'GreenmaruX'
        {
            val DllSurrogate = s ''
        }
 
        'GreenmaruX.DLL'
        {
            val AppID = s '%APPID%'
        }
    }
}

다음으로, 컨트롤이름.rgs파일(HelloCtrl.rgs)의 내용은 다음과 같다.
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
HKCR
{
    GreenmaruX.HelloCtrl.1 = s 'Greenmaru Example ActiveX Control'
    {
        CLSID = s '{4C80B2EE-6D19-4F77-9946-DF7AD17EEE67}'
    }
    GreenmaruX.HelloCtrl = s 'Greenmaru Example ActiveX Control'
    {
        CLSID = s '{4C80B2EE-6D19-4F77-9946-DF7AD17EEE67}'
        CurVer = s 'GreenmaruX.HelloCtrl.1'
    }
 
    NoRemove CLSID
    {
        ForceRemove {4C80B2EE-6D19-4F77-9946-DF7AD17EEE67}
            = s 'Greenmaru Example ActiveX Control'
        {
            ProgID = s 'GreenmaruX.HelloCtrl.1'
            VersionIndependentProgID = s 'GreenmaruX.HelloCtrl'
 
            ForceRemove Programmable
            InprocServer32 = s '%MODULE%'
            {
                val ThreadingModel = s 'Apartment'
            }
 
            val AppID = s '%APPID%'
 
            Elevation
            {
                val Enabled = d 1
            }
            val LocalizedString = s '@%MODULE%,-101'
 
            ForceRemove Control
            ForceRemove 'ToolboxBitmap32' = s '%MODULE%, 106'
            MiscStatus = s '0'
            {
                '1' = s '%OLEMISC%'
            }
            TypeLib = s '{C5133EA0-1CE5-4FFC-9C7D-79AE48AD2233}'
            Version = s '1.0'
        }
    }
}
Visual Studio의 버전에 따라 기본적으로 생성되는 rgs파일의 내용이 조금씩 차이가 있으므로, 위의 내용을 참고해서 적절히 편집하도록 한다.
주의할 부분은 32번째 라인의 LocalizedString이다. LocalizedString은 UAC화면에서 표시할 문자열 리소스의 숫자형 ID를 의미한다. 여기서 101은 리소스에 추가한 IDS_ELEVATION을 의미한다.

디버그
F5를 누르고 디버거를 실행해 보면, UAC Elevation화면이 표시되지 않는다. Visual Studio내부에서 컨트롤을 실행하는 경우에는 보호모드가 작동하지 않기 때문이다. 윈도우 탐색기를 통해 HelloCtrl.htm을 직접 실행해야 예상된 결과(UAC 화면)를 볼 수 있다.
여기서 중요한 문제!
SomeFunction의 구현부에 아무런 처리나 집어넣고 중단점(Break point)을 걸어보자. 이 상태에서 실행하면 디버거에서 중단점을 잡지 못한다.
이유는 두가지다.
첫째로, UAC를 거치면서 디버거가 실행한 컨트롤의 인스턴스가 아니라, (GetElevationObject함수에서) Elevation을 거쳐서 새롭게 생성된 컨트롤의 인스턴스를 통해 Logic이 진행되기 때문이다.
또한, Javascript에서 CreateObject를 통해 ActiveX인스턴스를 만들어내도록 했으므로 디버거가 더욱 추적할 수 없게 되버린다.
이런 상태로 디버그를 한다는건 무척 짜증스러운 일이다. Visual Studio의 강력한 디버그 기능을 사용하지 못하게 되므로, 전통적인 디버그 방식 - 실행하고 출력해서 확인하고 수정하는- 밖에는 쓸 수 없는 것이다.
따라서, 고전적인(?) object 태그를 통해 개체를 생성하고, 디버그를 위한 Javascript를 별도로 작성하는 쪽을 추천하겠다. 조금 귀찮기는 해도, 디버거 없이 테스트 하는 쪽보다는 훨씬 효과적인 방법이다.
1
2
3
4
5
6
7
8
9
10
11
12
13
<script type="text/javascript">
    function doDebug() {
 
        var helloCtrl = document.getElementById("HelloCtrl");
 
        //
        // UAC를 통한 개체생성 부분을 제거하고, 필요한 함수들을 테스트한다.
        //
        helloCtrl.SomeFunction();
    }
        </script>
 
<object id="HelloCtrl" classid="CLSID:4C80B2EE-6D19-4F77-9946-DF7AD17EEE67"></object>

마무리
이제 남은 작업은 컨트롤에 서명(Signing)하고 CAB으로 만들어서 배포하는 것이다. 이 과정에 관한 정보는 참고할만한 문서를 찾기 쉬운 편이므로 여기서는 다루지 않겠다.
소스코드를 다운로드 하게 되면, Build한 다음 프로젝트를 닫았다가 다시 로드해서 보시기 바란다.
작성자: 댓글 없음:
라벨: , , ,
피드 구독하기: 글 (Atom)