您可以使用HSE_REQ_VECTOR_SEND - 这比HSE_REQ_TRANSMIT_FILE 更通用。如何查看HSE_REQ_VECTOR_SEND 让发送任意数量的HSE_VECTOR_ELEMENT 结构。我们如何查看HSE_VECTOR_ELEMENT 让发送文件或内存缓冲区。如果文件 - cbOffset(文件中的偏移量)和 cbSize(来自 cbOffset 的文件数据的字节数) - 两者都是 ULONGLONG - 所以没有大小限制
演示(纯粹是为了清晰)我们如何将HSE_REQ_VECTOR_SEND 转换为更通用的HSE_REQ_VECTOR_SEND
BOOL Repack(LPEXTENSION_CONTROL_BLOCK lpECB, LPHSE_TF_INFO ti)
{
// code have same effect as call
// lpECB->ServerSupportFunction(lpECB->ConnID, HSE_REQ_TRANSMIT_FILE, ti, 0, 0);
if (lpECB->ServerSupportFunction(lpECB->ConnID, HSE_REQ_IO_COMPLETION, ti->pfnHseIO, 0, (DWORD*)ti->pContext))
{
HSE_VECTOR_ELEMENT Element[3] = {
{ HSE_VECTOR_ELEMENT_TYPE_MEMORY_BUFFER, ti->pHead, 0, ti->HeadLength },
{ HSE_VECTOR_ELEMENT_TYPE_FILE_HANDLE, ti->hFile, ti->Offset, ti->BytesToWrite },
{ HSE_VECTOR_ELEMENT_TYPE_MEMORY_BUFFER, ti->pTail, 0, ti->TailLength }
};
HSE_RESPONSE_VECTOR rv = {
HSE_IO_ASYNC|HSE_IO_DISCONNECT_AFTER_SEND|HSE_IO_FINAL_SEND,
0,
0,
RTL_NUMBER_OF(Element),
Element
};
if (!Element[2].cbSize) rv.nElementCount--;
if (!Element[0].cbSize) rv.lpElementArray++, rv.nElementCount--;
if (lpECB->ServerSupportFunction(lpECB->ConnID, HSE_REQ_VECTOR_SEND, &rv, 0, 0))
{
return TRUE;
}
}
return FALSE;
}
因此,如果您需要完整的 64 位大小和偏移量,您通常需要下一个代码:
struct TRANSMIT_FILE_CONTEXT
{
ULONGLONG _BytesToWrite;
ULONGLONG _cbOffset;
HANDLE _hFile;
char _szHeaders[];
TRANSMIT_FILE_CONTEXT(HANDLE hFile, ULONGLONG BytesToWrite, ULONGLONG _cbOffset);
~TRANSMIT_FILE_CONTEXT();
void* operator new(size_t cb, ULONG cbHeaders)
{
return ::operator new(cb + cbHeaders);
}
void operator delete(void* p)
{
::operator delete(p);
}
static VOID WINAPI OnIoComplete (
LPEXTENSION_CONTROL_BLOCK lpECB,
PVOID pContext,
DWORD /*cbIO*/,
DWORD dwError
)
{
DbgPrint("OnIoComplete(%p, %u)\n", lpECB->ConnID, dwError);
dwError = (dwError == NOERROR ? HSE_STATUS_SUCCESS : HSE_STATUS_ERROR);
lpECB->ServerSupportFunction(lpECB->ConnID, HSE_REQ_DONE_WITH_SESSION, &dwError, 0, 0);
delete reinterpret_cast<TRANSMIT_FILE_CONTEXT*>(pContext);
}
DWORD Send(LPEXTENSION_CONTROL_BLOCK lpECB)
{
if (lpECB->ServerSupportFunction(lpECB->ConnID, HSE_REQ_IO_COMPLETION, OnIoComplete, 0, (DWORD*)this))
{
HSE_VECTOR_ELEMENT Element[2] = {
{ HSE_VECTOR_ELEMENT_TYPE_MEMORY_BUFFER, _szHeaders, 0, strlen(_szHeaders) },
{ HSE_VECTOR_ELEMENT_TYPE_FILE_HANDLE, _hFile, _cbOffset, _BytesToWrite }
};
HSE_RESPONSE_VECTOR rv = {
HSE_IO_ASYNC|HSE_IO_DISCONNECT_AFTER_SEND|HSE_IO_FINAL_SEND,
0,
0,
RTL_NUMBER_OF(Element),
Element
};
if (lpECB->ServerSupportFunction(lpECB->ConnID, HSE_REQ_VECTOR_SEND, &rv, 0, 0))
{
return HSE_STATUS_PENDING;
}
}
delete this;
return HSE_STATUS_ERROR;
}
};
DWORD WINAPI HttpExtensionProc(LPEXTENSION_CONTROL_BLOCK lpECB)
{
//...
if (TRANSMIT_FILE_CONTEXT* p = new(cb + 1) TRANSMIT_FILE_CONTEXT(*))
{
//...
return p->Send(lpECB);
}
//...
return HSE_STATUS_ERROR;
}
还要注意HSE_REQ_VECTOR_SEND 和HSE_REQ_TRANSMIT_FILE 内部调用HttpSendResponseEntityBody