用ISAPI实现URL转向

 

工程名：TestISAPI

开发环境：Windows XP SP2 + Visual Studio.Net 2003 + IIS 5.1

TestISAPI.h

#pragma once

// TestISAPI.h – Internet Information Server 的头文件

//    TestISAPI Filter

#include "resource.h"

class CTestISAPIFilter : public CHttpFilter

{

public:

    CTestISAPIFilter();

    ~CTestISAPIFilter();

// 重写

    public:

    virtual BOOL GetFilterVersion(PHTTP_FILTER_VERSION pVer);

    virtual DWORD OnUrlMap(CHttpFilterContext* pCtxt, PHTTP_FILTER_URL_MAP pMapInfo);

};

TestISAPI.cpp

// TestISAPI.cpp – ISAPI 的实现文件

//    TestISAPI Filter

#include "resource.h"

#include "stdafx.h"

#include "TestISAPI.h"

// 唯一的 CTestISAPIFilter 对象

CTestISAPIFilter theFilter;

// CTestISAPIFilter 实现

CTestISAPIFilter::CTestISAPIFilter()

{

}

CTestISAPIFilter::~CTestISAPIFilter()

{

}

BOOL CTestISAPIFilter::GetFilterVersion(PHTTP_FILTER_VERSION pVer)

{

    // 调用初始化的默认实现

    CHttpFilter::GetFilterVersion(pVer);

    // 清除由基类设置的标志

    pVer->dwFlags &= ~SF_NOTIFY_ORDER_MASK;

    // 设置我们感兴趣的标志

    pVer->dwFlags |= SF_NOTIFY_SECURE_PORT | SF_NOTIFY_NONSECURE_PORT | SF_NOTIFY_URL_MAP;

    // 设置优先级

    pVer->dwFlags |= SF_NOTIFY_ORDER_MEDIUM;

    // 加载描述字符串

    TCHAR sz[SF_MAX_FILTER_DESC_LEN+1];

    ISAPIVERIFY(::LoadString(AfxGetResourceHandle(),

            IDS_FILTER, sz, SF_MAX_FILTER_DESC_LEN));

    _tcscpy(pVer->lpszFilterDesc, sz);

    return TRUE;

}

DWORD CTestISAPIFilter::OnUrlMap(CHttpFilterContext* pCtxt,

    PHTTP_FILTER_URL_MAP pMapInfo)

{

    CHAR szHostName[256] = {0};

    DWORD dw = 256;

    // 这里为了尽量简化问题，突出重点，我把网址写死了

    // 如果要扩展，可以用各种方法从文件或注册表中读取网址

    static const CHAR szHostToRedirect[] = "blog.realdodo.com";

    static const CHAR szRedirectToHost[] = "www.realdodo.com";

    static const CHAR szRedirectPrefix[] = "/blog";

    // 获得客户端访问的网址

    pCtxt->GetServerVariable(_T("SERVER_NAME"), (LPVOID)szHostName, &dw);

    if (!strnicmp(szHostName, szHostToRedirect, dw))

    {

        CHAR szRedirect [512];  // 这个数组并不安全，可能溢出，请大家注意

                                // 推荐大家用std::string，这里是MSDN例子中的写法

        // 替换地址，做域名转向

        sprintf(szRedirect, "Location: http://%s/r/n/r/n", szRedirectToHost, szRedirectPrefix, pMapInfo->pszURL);

        pCtxt->ServerSupportFunction ( SF_REQ_SEND_RESPONSE_HEADER,

            (LPVOID) "302 Redirect",

            (DWORD *) szRedirect,

            0 );

        return SF_STATUS_REQ_FINISHED_KEEP_CONN;

    }

    // 主机名不是blog.realdodo.com，则不用跳转

    return SF_STATUS_REQ_NEXT_NOTIFICATION;

}

// 如果您的扩展不使用 MFC，您将需要此代码，以确保

//扩展对象可以找到模块的资源句柄。

//如果将扩展转换为不依赖于 MFC，

//请移除以下 AfxGetResourceHandle()

// 和 DllMain() 函数周围的注释，以及 g_hInstance 全局变量。

#ifndef _AFXDLL

static HINSTANCE g_hInstance;

HINSTANCE AFXISAPI AfxGetResourceHandle()

{

    return g_hInstance;

}

BOOL WINAPI DllMain(HINSTANCE hInst, ULONG ulReason,

                    LPVOID lpReserved)

{

    if (ulReason == DLL_PROCESS_ATTACH)

    {

        g_hInstance = hInst;

    }

    return TRUE;

}

#endif