PROGRAM PARK

안정성을 보장하는 ActiveX 컨트롤 제작



  일반적인 방식으로(ActiveX 컨트롤 제작에서 설명한 방식) ActiveX 컨트롤을 작성하면, 웹에서 ActiveX 사용의 마지막 부분에 ActiveX 컨트롤과 Html 객체와의 연동 시에 보안 문제가 있다는 것을 설명했다. 이제 그 해결법을 설명하고자 한다.

간단히 설명하면 ActiveX 컨트롤 루틴에 안정성을 보장하는 루틴(안정성을 보장하는 clsid를 레지스트리에 등록하는 루틴)을 추가해야 한다. 먼저 레지스트리에 clsid를 등록하는 함수를 작성하고 실제 ActiveX 컨트롤 제작에서 작성한 컨트롤에 추가를 해보기로 한다.



함수 작성

다음과 같은 내용의 cathelp.h라는 파일을 작성한다.
  

#if !defined(__CATHELP_H)
#define __CATHELP_H

#include "comcat.h"

// Helper function to create a component category and associated
// description
HRESULT CreateComponentCategory(CATID catid, WCHAR* catDescription);

// Helper function to register a CLSID as belonging to a component
// category
HRESULT RegisterCLSIDInCategory(REFCLSID clsid, CATID catid);

#endif



다음과 같은 내용의 cathelp.cpp라는 파일을 작성한다.

#include "stdafx.h"
#include "comcat.h"

// Helper function to create a component category and associated
// description
HRESULT CreateComponentCategory(CATID catid, WCHAR* catDescription)
{
        ICatRegister* pcr = NULL ;
        HRESULT hr = S_OK ;

        hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr,
                                           NULL,
                                           CLSCTX_INPROC_SERVER,
                                           IID_ICatRegister,
                                           (void**)&pcr);
        if (FAILED(hr))
                return hr;

        // Make sure the HKCR\Component Categories\{..catid...}
        // key is registered
        CATEGORYINFO catinfo;
        catinfo.catid = catid;
        catinfo.lcid = 0x0409 ; // english

        // Make sure the provided description is not too long.
        // Only copy the first 127 characters if it is
        int len = wcslen(catDescription);
        if (len>127)
                len = 127;
        wcsncpy(catinfo.szDescription, catDescription, len);
        // Make sure the description is null terminated
        catinfo.szDescription[len] = '\0';

        hr = pcr->RegisterCategories(1, &catinfo);
        pcr->Release();

        return hr;
}

// Helper function to register a CLSID as belonging to a component
// category
HRESULT RegisterCLSIDInCategory(REFCLSID clsid, CATID catid)
{
// Register your component categories information.
        ICatRegister* pcr = NULL ;
        HRESULT hr = S_OK ;
        hr = CoCreateInstance(CLSID_StdComponentCategoriesMgr,
                                           NULL,
                                           CLSCTX_INPROC_SERVER,
                                           IID_ICatRegister,
                                           (void**)&pcr);
        if (SUCCEEDED(hr))
        {
                // Register this category as being "implemented" by
                // the class.
                CATID rgcatid[1] ;
                rgcatid[0] = catid;
                hr = pcr->RegisterClassImplCategories(clsid, 1, rgcatid);
        }

        if (pcr != NULL)
                pcr->Release();

        return hr;
}



  위의 루틴의 내용을 이해하기 위해서 노력할 필요는 없다. 단지 "레지스트리에 clsid를 등록하는 함수구나"라고만 이해하고 넘어가기 바란다.



Test Control에 추가

App 파일(여기서는 Test.cpp)를 열어 다음과 같이 위에서 작성한 cathelp.h를 include한다.

#include "cathelp.h"

조금 아래에 다음과 같이 _tlid가 선언되어 있는 것을 볼 수 있을 것이다(_tlid에 할당된 값은 프로그램 마다 다르다).

const GUID CDECL BASED_CODE _tlid =
                { 0x9b548709, 0xc3df, 0x4956, { 0x9f, 0x65, 0x29, 0x28, 0xca, 0xbb, 0x6e, 0xc8 } };

바로 아래에 비슷한 3개를 더 등록해야 한다.

두 개는 다음과 같이 안정성을 보장하기 위해 예약된(고정된) clsid이고,

const CATID CATID_SafeForScripting     =
                {0x7dd95801,0x9882,0x11cf,{0x9f,0xa9,0x00,0xaa,0x00,0x6c,0x42,0xc4}};
const CATID CATID_SafeForInitializing  =
                {0x7dd95802,0x9882,0x11cf,{0x9f,0xa9,0x00,0xaa,0x00,0x6c,0x42,0xc4}};

나머지 하나는 ctrl 파일(여기서는 testctrl.cpp)에 있는 다음과 같은 내용을 복사해 와서,

IMPLEMENT_OLECREATE_EX(CTestCtrl, "TEST.TestCtrl.1",
        0xd886696, 0xc7ce, 0x11d3, 0xa1, 0x75, 0x8, 0, 0x2b, 0xf1, 0x75, 0x7)

다음과 같이 변경(_ctlid 선언)하면 된다.

const GUID CDECL BASED_CODE _ctlid =
        {0xd886696, 0xc7ce, 0x11d3, 0xa1, 0x75, 0x8, 0, 0x2b, 0xf1, 0x75, 0x7}



이제 App 파일에 있는 다음과 같은 함수에 레지스트리 등록 루틴을 추가하면된다.
  

STDAPI DllRegisterServer(void)
{
        AFX_MANAGE_STATE(_afxModuleAddrThis);

        if (!AfxOleRegisterTypeLib(AfxGetInstanceHandle(), _tlid))
                return ResultFromScode(SELFREG_E_TYPELIB);

        if (!COleObjectFactoryEx::UpdateRegistryAll(TRUE))
                return ResultFromScode(SELFREG_E_CLASS);

        return NOERROR;
}




추가한 루틴은 다음과 같다.


  

STDAPI DllRegisterServer(void)
{
        AFX_MANAGE_STATE(_afxModuleAddrThis);

        if (!AfxOleRegisterTypeLib(AfxGetInstanceHandle(), _tlid))
                return ResultFromScode(SELFREG_E_TYPELIB);

        if (!COleObjectFactoryEx::UpdateRegistryAll(TRUE))
                return ResultFromScode(SELFREG_E_CLASS);

        if (FAILED( CreateComponentCategory(CATID_SafeForScripting, L"Controls that are safely scriptable") ))
                return ResultFromScode(SELFREG_E_CLASS);

        if (FAILED( CreateComponentCategory(CATID_SafeForInitializing, L"Controls safely initializable from persistent data") ))
                return ResultFromScode(SELFREG_E_CLASS);

        if (FAILED( RegisterCLSIDInCategory(_ctlid, CATID_SafeForScripting) ))
                return ResultFromScode(SELFREG_E_CLASS);

        if (FAILED( RegisterCLSIDInCategory(_ctlid, CATID_SafeForInitializing) ))
                return ResultFromScode(SELFREG_E_CLASS);

        return NOERROR;
}




위 내용 또한 그대로 사용하면 되므로 이해하기 위해서 너무 많은 노력을 들이지는 말기 바란다.

컴파일을 하고 실행을 하면 보안 대화 상자가 생성되지 않고 잘 동작된다는 것을 알 수 있다.

고수닷넷에 있는 내용입니다.(회원가입을 해야 해서...)

http://www.gosu.net/GosuWeb/Article-detail.aspx?ArticleCode=368

1. 트레이 통지와 관련된 함수와 파라미터
BOOL WINAPI Shell_NontifyIcon(DWORD dwMessage, PNOTIFYICONDATA pnid);

dwMessage로는 다음과 같은 값을 입력할 수 있습니다.
NIM_ADD : 트레이에 새로운 아이콘 추가
NIM_DELETE : 트레이 영역의 아이콘 제거
NIM_MODIFY : 트레이 영역에 있는 아이콘 수정
typedef struct _NOTIFYICONDATA
{
    DWORD    cbSize;
    HWND    hWnd;
    UINT    uID;
    UINT    uFlags;
    UINT    uCallbackMessage;
    HICON    hIcon;
    char    szTip[64];
} NOTIFYICONDATA, *PNOTIFYICONDATA;

cbSize: 구조체의 크기
hWnd: 윈도우 핸들
uID: 아이콘 식별자, 호출한 애플리케이션의 아이콘을 다른것과 구ㅕㄹ해서 식별할수 있게 해주는 사용자 정의값
uFlags: NIF_MESSAGE : uCallbackMessage 사용 NIF_ICON : hIcon 사용 NIF_TIP : szTip 사용
uCallbackMessage: 아이콘이 hWnd윈도우와 통신하기 위해서 사용할 메시지 ID. 메시지는 WM_APP의 오프셋으로 선언되는 사용자 정의 메시지이다.
hIcon: 화면에 그릴 아이콘의 핸들.
szTip: 아이콘의 툴팁을 위한 텍스트.
2. 트레이에 아이콘 추가
NOTIFYICONDATA    nid;

// 구조체 초기화
ZeroMemory(&nid, sizeof nid);

// 구조체 설정
nid.cbSize = sizeof nid;
nid.hWnd = hWnd;
nid.uID = ICON_ID;
nid.uFlags = NIF_TIP | NIF_ICON | NIF_MESSAGE;
nid.uCallbackMessage = WM_MESSAGE;
nid.hIcon = hSmallIcon;
lstrcpy(nid.szTip, "Sample");

// 트레이 영역에 추가
Shell_NotifyIcon(NIM_ADD, &nid);

3. 트레이 아이콘 제거
제거시에는 hWnd와 uID멤버만 셋팅하면 됩니다.

NOTIFYICONDATA    nid;

ZeroMemory(&nid, sizeof nid);

nid.cbSize = sizeof nid;
nid.hWnd = hWnd;
nid.uID = ICON_ID;

Shell_NotifyIcon(NIM_DELETE, &nid);

4. 트레이로 부터의 메시지 수신
마우스와 관련된 메시지만 통지합니다.

wParam : 등록시에 설정한 ID(nid.uID).
lParam : 메시지 종류 (WM_LBUTTONUP, ...).
case WM_MESSAGE:
    if(wParam == ICON_ID)
    {
        switch(lParam)
        {
            case WM_RBUTTONUP:
                // 처리
                break;
        }
    }

    break;

5. Context메뉴
일반적으로 트레이에서 오른쪽 버튼을 누르면 context 메뉴를 화면에 표시합니다. 아래는 그러한 일을 하는 코드의 일부입니다.

POINT pt;
GetCursorPos(&pt);

SetForegroundWindow(hWnd);
TrackPopupMenu(menu, TPM_LEFTALIGN, pt.x, pt.y, 0, hWnd, NULL);
PostMessage(hWnd, WM_NULL, 0, 0);

주의할 점은 TrackPopupMenu 앞에 SetForegroundWindow API 를 호출해 주고, 후에 WM_NULL 메시지를 포스트 해주어야 합니다. 그렇지 않을 경우 메뉴가 화면에서 사라지지 않는 버그 현상이 생기게 됩니다.

6. For MFC.
아래는 MFC에서 일반적으로 하게 되는 트레이 작업 영역에 추가 및 삭제를 위한 코드의 일부입니다.

6.1 추가
#define WM_TRAYNOTIFY WM_APP + 1

NOTIFYICONDATA    nid;

::ZeroMemory(&nid, sizeof nid);

nid.cbSize        = sizeof nid;
nid.hIcon        = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
nid.hWnd        = this->m_hWnd;
nid.uID            = IDR_MAINFRAME;
nid.uFlags        = NIF_TIP | NIF_ICON | NIF_MESSAGE;
nid.uCallbackMessage    = WM_TRAYNOTIFY;

lstrcpy(nid.szTip, "Test");

::Shell_NotifyIcon(NIM_ADD, &nid);

6.2 제거
NOTIFYICONDATA    nid;

::ZeroMemory(&nid, sizeof nid);

nid.cbSize    = sizeof nid;
nid.uID        = IDR_MAINFRAME;
nid.hWnd    = this->m_hWnd;

::Shell_NotifyIcon(NIM_DELETE, &nid);

6.3 메시지 처리
ON_MESSAGE(WM_TRAYNOTIFY, OnTrayNotify)

LONG TestDlg::OnTrayNotify(WPARAM wParam, LPARAM lParam)
{
    CMenu    menu, *pTrayMenu;
    CPoint    pt;

    if(wParam == IDR_MAINFRAME)
    {
        switch(lParam)
        {
        case WM_RBUTTONUP:
            menu.LoadMenu(IDR_TRAYMENU);
            pTrayMenu = menu.GetSubMenu(0);

            ::GetCursorPos(&pt);

            SetForegroundWindow();
            pTrayMenu->TrackPopupMenu(TPM_LEFTALIGN, pt.x, pt.y, this, NULL);
            SetForegroundWindow();

            pTrayMenu->DestroyMenu();
            menu.DestroyMenu();
            break;

        case WM_LBUTTONDBLCLK:
            if(!IsWindowVisible())
                ShowWindow(SW_SHOW);

            SetForegroundWindow();
            break;
        }
    }

    return 0;
}