如何在ISAPI过滤器的HttpFilterProc函数中关闭客户端连接？

我之前刚好碰到过一模一样的需求——要在ISAPI过滤器里提前断开客户端连接，避免下载完整数据，折腾了一阵终于找到可行的方法，核心是利用Win32套接字API结合ISAPI的上下文机制来实现，具体步骤如下：

1. 提前捕获客户端套接字句柄 #

ISAPI在SF_NOTIFY_CONNECT通知阶段会把客户端的TCP套接字句柄放到HTTP_FILTER_CONTEXT的EventKey字段里，我们需要把这个句柄保存到自定义的过滤器上下文中，方便后续调用：

首先定义一个上下文结构体来存储套接字：

typedef struct _FilterContext {
    SOCKET clientSocket;
    // 可以添加其他需要的上下文数据
} FilterContext, *PFilterContext;

然后在HttpFilterProc的SF_NOTIFY_CONNECT分支里分配并初始化上下文：

DWORD WINAPI HttpFilterProc(
    PHTTP_FILTER_CONTEXT pfc,
    DWORD notificationType,
    LPVOID pvNotification
) {
    switch (notificationType) {
        case SF_NOTIFY_CONNECT:
            // 分配上下文内存
            PFilterContext pContext = (PFilterContext)LocalAlloc(LPTR, sizeof(FilterContext));
            if (pContext != NULL) {
                // 保存客户端套接字句柄
                pContext->clientSocket = (SOCKET)pfc->EventKey;
                pfc->pFilterContext = pContext;
            }
            break;
        // 其他通知处理逻辑...
    }
    return SF_STATUS_REQ_NEXT_NOTIFICATION;
}

2. 在需要的时机强制断开连接 #

当你判断到需要终止下载（比如在SF_NOTIFY_SEND_RAW_DATA阶段检测到不符合要求的请求），就可以从上下文中取出套接字，调用closesocket()强制关闭连接，同时返回特定状态告诉IIS停止处理当前请求：

case SF_NOTIFY_SEND_RAW_DATA: {
    PHTTP_FILTER_RAW_DATA pRawData = (PHTTP_FILTER_RAW_DATA)pvNotification;
    PFilterContext pContext = (PFilterContext)pfc->pFilterContext;
    
    // 这里替换成你的判断逻辑：比如检测到非法下载请求
    if (pContext != NULL && 需要断开连接的条件) {
        // 可选：先发送一个错误响应头，让客户端收到明确的反馈
        char responseHeaders[] = "HTTP/1.1 403 Forbidden\r\nContent-Length: 0\r\n\r\n";
        HttpSendResponseHeaders(pfc, responseHeaders, strlen(responseHeaders), 0, 0);
        
        // 强制关闭客户端套接字
        closesocket(pContext->clientSocket);
        
        // 返回这个状态告诉IIS不再处理该请求
        return SF_STATUS_REQ_FINISHED;
    }
    break;
}

3. 清理上下文内存避免泄漏 #

一定要在SF_NOTIFY_DISCONNECT通知里释放之前分配的上下文内存，不然会造成内存泄漏：

case SF_NOTIFY_DISCONNECT:
    PFilterContext pContext = (PFilterContext)pfc->pFilterContext;
    if (pContext != NULL) {
        LocalFree(pContext);
        pfc->pFilterContext = NULL;
    }
    break;

关键注意点 #

• 调用closesocket()后必须返回SF_STATUS_REQ_FINISHED，否则IIS可能还会继续尝试向客户端发送数据，导致异常。
• 如果不需要返回错误响应，可以直接关闭套接字，但这样客户端会收到连接重置的错误，体验不如明确的HTTP状态码友好。
• 确保在SF_NOTIFY_CONNECT阶段成功分配上下文，否则后续无法获取套接字句柄。

内容的提问来源于stack exchange，提问作者zeus