399 lines
8.6 KiB
C++
399 lines
8.6 KiB
C++
/*++
|
||
|
||
Copyright (c) 1997 Microsoft Corporation
|
||
|
||
Module Name:
|
||
|
||
write.cxx
|
||
|
||
Abstract:
|
||
|
||
This file contains the implementation of the HttpWriteData API.
|
||
|
||
Contents:
|
||
HttpWriteData
|
||
HTTP_REQUEST_HANDLE_OBJECT::WriteData
|
||
|
||
Author:
|
||
|
||
Arthur Bierer (arthurbi) 07-Apr-1997
|
||
|
||
Revision History:
|
||
|
||
|
||
|
||
--*/
|
||
|
||
#include <wininetp.h>
|
||
#include "httpp.h"
|
||
|
||
|
||
|
||
//
|
||
// functions
|
||
//
|
||
|
||
#if !defined(THREAD_POOL)
|
||
|
||
|
||
INTERNETAPI_(BOOL) HttpWriteData(
|
||
IN HINTERNET hRequest,
|
||
IN LPVOID lpBuffer,
|
||
IN DWORD dwNumberOfBytesToWrite,
|
||
OUT LPDWORD lpdwNumberOfBytesWritten
|
||
)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
Writes a block of data for an outstanding HTTP request
|
||
|
||
Assumes: 1. this function can only be called from InternetWriteFile() which
|
||
globally validates parameters for all Internet data write
|
||
functions
|
||
|
||
2. That we the caller has called HttpBeginSendRequest but not HttpEndSendRequest
|
||
|
||
Arguments:
|
||
|
||
hRequest - an open HTTP request handle returned by
|
||
HttpOpenRequest()
|
||
|
||
lpBuffer - pointer to the buffer to receive the data
|
||
|
||
dwNumberOfBytesToWrite - number of bytes to write from user's buffer
|
||
|
||
lpdwNumberOfBytesWritten - number of bytes actually written
|
||
|
||
Return Value:
|
||
|
||
TRUE - The data was written successfully. lpdwNumberOfBytesRead points to the
|
||
number of BYTEs actually read. This value will be set to zero
|
||
when the transfer has completed.
|
||
|
||
FALSE - The operation failed. Error status is available by calling
|
||
GetLastError().
|
||
|
||
--*/
|
||
|
||
{
|
||
|
||
DEBUG_ENTER((DBG_HTTP,
|
||
Dword,
|
||
"HttpWriteData",
|
||
"%#x, %#x, %d, %#x",
|
||
hRequest,
|
||
lpBuffer,
|
||
dwNumberOfBytesToWrite,
|
||
lpdwNumberOfBytesWritten
|
||
));
|
||
|
||
DWORD error;
|
||
|
||
//
|
||
// find path from internet handle
|
||
//
|
||
|
||
BOOL isLocal;
|
||
BOOL isAsync;
|
||
|
||
error = RIsHandleLocal(hRequest,
|
||
&isLocal,
|
||
&isAsync,
|
||
TypeHttpRequestHandle
|
||
);
|
||
if (error != ERROR_SUCCESS) {
|
||
goto quit;
|
||
}
|
||
|
||
//
|
||
// need to set these for both local and remote paths, in case we're in async
|
||
// mode
|
||
//
|
||
|
||
DWORD context;
|
||
|
||
error = RGetContext(hRequest, &context);
|
||
if (error != ERROR_SUCCESS) {
|
||
goto quit;
|
||
}
|
||
|
||
//InternetSetObjectHandle(hRequest);
|
||
//InternetSetContext(context);
|
||
|
||
//
|
||
// Cast it to the object that we know. We are going to do call
|
||
// into the method to do the work.
|
||
//
|
||
|
||
HTTP_REQUEST_HANDLE_OBJECT *pRequest;
|
||
|
||
pRequest = (HTTP_REQUEST_HANDLE_OBJECT *)hRequest;
|
||
|
||
if (!IS_VALID_HTTP_STATE(pRequest, WRITE, TRUE)) {
|
||
error = ERROR_INTERNET_INCORRECT_HANDLE_STATE;
|
||
} else {
|
||
// This request needs a special CR-LF added to the end at EndRequest
|
||
pRequest->SetAddCRLF(TRUE);
|
||
error = pRequest->WriteData(lpBuffer,
|
||
dwNumberOfBytesToWrite,
|
||
lpdwNumberOfBytesWritten
|
||
);
|
||
}
|
||
|
||
|
||
|
||
quit:
|
||
|
||
|
||
DEBUG_LEAVE(error);
|
||
|
||
return error;
|
||
}
|
||
|
||
|
||
|
||
|
||
DWORD
|
||
HTTP_REQUEST_HANDLE_OBJECT::WriteData(
|
||
OUT LPVOID lpBuffer,
|
||
IN DWORD dwNumberOfBytesToWrite,
|
||
OUT LPDWORD lpdwNumberOfBytesWritten
|
||
)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
HTTP_REQUEST_HANDLE_OBJECT WriteData method
|
||
|
||
Writes data from users buffer. Writes the data to the currently open
|
||
socket.
|
||
|
||
Arguments:
|
||
|
||
lpBuffer - pointer to users buffer
|
||
|
||
dwNumberOfBytesToWrite - number of bytes to write from user's buffer
|
||
|
||
lpdwNumberOfBytesWritten - number of bytes actually written
|
||
|
||
|
||
Return Value:
|
||
|
||
DWORD
|
||
Success - ERROR_SUCCESS
|
||
|
||
Failure - WSA error
|
||
|
||
--*/
|
||
|
||
{
|
||
DEBUG_ENTER((DBG_HTTP,
|
||
Dword,
|
||
"HTTP_REQUEST_HANDLE_OBJECT::WriteData",
|
||
"%#x, %d, %#x",
|
||
lpBuffer,
|
||
dwNumberOfBytesToWrite,
|
||
lpdwNumberOfBytesWritten
|
||
));
|
||
|
||
DWORD error = ERROR_SUCCESS;
|
||
|
||
if ( _Socket == NULL )
|
||
{
|
||
error = ERROR_INTERNET_INTERNAL_ERROR;
|
||
goto quit;
|
||
}
|
||
|
||
error = _Socket->Send(lpBuffer, dwNumberOfBytesToWrite, SF_INDICATE);
|
||
if (error == ERROR_SUCCESS)
|
||
{
|
||
*lpdwNumberOfBytesWritten = dwNumberOfBytesToWrite;
|
||
}
|
||
else
|
||
{
|
||
goto quit;
|
||
}
|
||
|
||
quit:
|
||
|
||
DEBUG_LEAVE(error);
|
||
|
||
return error;
|
||
}
|
||
|
||
|
||
#else
|
||
|
||
|
||
|
||
|
||
DWORD
|
||
HttpWriteData(
|
||
IN HINTERNET hRequest,
|
||
IN LPVOID lpBuffer,
|
||
IN DWORD dwNumberOfBytesToWrite,
|
||
OUT LPDWORD lpdwNumberOfBytesWritten,
|
||
IN DWORD dwSocketFlags
|
||
)
|
||
|
||
/*++
|
||
|
||
Routine Description:
|
||
|
||
Writes a block of data for an outstanding HTTP request
|
||
|
||
Assumes: 1. this function can only be called from InternetWriteFile() which
|
||
globally validates parameters for all Internet data write
|
||
functions
|
||
|
||
2. That we the caller has called HttpBeginSendRequest but not HttpEndSendRequest
|
||
|
||
Arguments:
|
||
|
||
hRequest - an open HTTP request handle returned by
|
||
HttpOpenRequest()
|
||
|
||
lpBuffer - pointer to the buffer to receive the data
|
||
|
||
dwNumberOfBytesToWrite - number of bytes to write from user's buffer
|
||
|
||
lpdwNumberOfBytesWritten - number of bytes actually written
|
||
|
||
dwSocketFlags - controlling socket operation
|
||
|
||
|
||
Return Value:
|
||
|
||
TRUE - The data was written successfully. lpdwNumberOfBytesRead points to the
|
||
number of BYTEs actually read. This value will be set to zero
|
||
when the transfer has completed.
|
||
|
||
FALSE - The operation failed. Error status is available by calling
|
||
GetLastError().
|
||
|
||
--*/
|
||
|
||
|
||
{
|
||
DEBUG_ENTER((DBG_HTTP,
|
||
Dword,
|
||
"HttpWriteData",
|
||
"%#x, %#x, %d, %#x, %#x",
|
||
hRequest,
|
||
lpBuffer,
|
||
dwNumberOfBytesToWrite,
|
||
lpdwNumberOfBytesWritten,
|
||
dwSocketFlags
|
||
));
|
||
|
||
DWORD error = DoFsm(new CFsm_HttpWriteData(lpBuffer,
|
||
dwNumberOfBytesToWrite,
|
||
lpdwNumberOfBytesWritten,
|
||
dwSocketFlags,
|
||
(HTTP_REQUEST_HANDLE_OBJECT *)hRequest
|
||
));
|
||
|
||
DEBUG_LEAVE(error);
|
||
|
||
return error;
|
||
}
|
||
|
||
|
||
DWORD
|
||
CFsm_HttpWriteData::RunSM(
|
||
IN CFsm * Fsm
|
||
)
|
||
{
|
||
DEBUG_ENTER((DBG_HTTP,
|
||
Dword,
|
||
"CFsm_HttpWriteData::RunSM",
|
||
"%#x",
|
||
Fsm
|
||
));
|
||
|
||
DWORD error;
|
||
HTTP_REQUEST_HANDLE_OBJECT * pRequest;
|
||
CFsm_HttpWriteData * stateMachine = (CFsm_HttpWriteData *)Fsm;
|
||
|
||
pRequest = (HTTP_REQUEST_HANDLE_OBJECT *)Fsm->GetContext();
|
||
switch (Fsm->GetState()) {
|
||
case FSM_STATE_INIT:
|
||
|
||
pRequest->SetAddCRLF(TRUE);
|
||
|
||
//
|
||
// Fall through
|
||
//
|
||
|
||
case FSM_STATE_CONTINUE:
|
||
error = pRequest->HttpWriteData_Fsm(stateMachine);
|
||
break;
|
||
|
||
default:
|
||
error = ERROR_INTERNET_INTERNAL_ERROR;
|
||
Fsm->SetDone(ERROR_INTERNET_INTERNAL_ERROR);
|
||
|
||
INET_ASSERT(FALSE);
|
||
|
||
break;
|
||
}
|
||
|
||
DEBUG_LEAVE(error);
|
||
|
||
return error;
|
||
}
|
||
|
||
|
||
DWORD
|
||
HTTP_REQUEST_HANDLE_OBJECT::HttpWriteData_Fsm(
|
||
IN CFsm_HttpWriteData * Fsm
|
||
)
|
||
{
|
||
DEBUG_ENTER((DBG_HTTP,
|
||
Dword,
|
||
"HTTP_REQUEST_HANDLE_OBJECT::HttpWriteData_Fsm",
|
||
"%#x",
|
||
Fsm
|
||
));
|
||
|
||
CFsm_HttpWriteData & fsm = *Fsm;
|
||
DWORD error = fsm.GetError();
|
||
|
||
if (fsm.GetState() == FSM_STATE_INIT) {
|
||
if (!IsValidHttpState(WRITE)) {
|
||
error = ERROR_INTERNET_INCORRECT_HANDLE_STATE;
|
||
goto quit;
|
||
}
|
||
|
||
error = _Socket->Send(
|
||
fsm.m_lpBuffer,
|
||
fsm.m_dwNumberOfBytesToWrite,
|
||
SF_INDICATE
|
||
);
|
||
|
||
}
|
||
|
||
if (error == ERROR_SUCCESS)
|
||
{
|
||
*fsm.m_lpdwNumberOfBytesWritten = fsm.m_dwNumberOfBytesToWrite;
|
||
}
|
||
|
||
|
||
quit:
|
||
|
||
if (error != ERROR_IO_PENDING) {
|
||
fsm.SetDone();
|
||
}
|
||
|
||
DEBUG_LEAVE(error);
|
||
|
||
return error;
|
||
}
|
||
|
||
|
||
#endif // defined(THREAD_POOL)
|
||
|