809 lines
23 KiB
C++
809 lines
23 KiB
C++
// ===========================================================================
|
|
// THIS CODE AND INFORMATION IS PROVIDED "AS IS" WITHOUT WARRANTY OF
|
|
// ANY KIND, EITHER EXPRESSED OR IMPLIED, INCLUDING BUT NOT LIMITED TO
|
|
// THE IMPLIED WARRANTIES OF MERCHANTABILITY AND/OR FITNESS FOR A
|
|
// PARTICULAR PURPOSE.
|
|
//
|
|
// Copyright 1996 Microsoft Corporation. All Rights Reserved.
|
|
// ===========================================================================
|
|
#include <windows.h>
|
|
#include <stdlib.h>
|
|
#include <stdio.h>
|
|
#include <fcntl.h>
|
|
#include <stdlib.h>
|
|
#include <io.h>
|
|
#include <winhttp.h>
|
|
#include <wincrypt.h>
|
|
#define _UNICODE
|
|
#include <tchar.h>
|
|
|
|
//==============================================================================
|
|
//#define GLOBAL_SESSION 1
|
|
|
|
#ifdef GLOBAL_SESSION
|
|
HINTERNET g_hInternet = NULL;
|
|
#endif
|
|
|
|
BOOL g_fCloseOnSSLFailure = FALSE;
|
|
|
|
#ifndef ASSERT
|
|
#define ASSERT( exp ) \
|
|
((!(exp)) ? \
|
|
DebugBreak() : \
|
|
((void)0))
|
|
#endif
|
|
|
|
typedef enum
|
|
{
|
|
HTTP_INIT = 0,
|
|
HTTP_OPEN = 1,
|
|
HTTP_SEND = 2,
|
|
HTTP_QDA = 3,
|
|
HTTP_READ = 4,
|
|
HTTP_READ_PASS_SYNC = 5,
|
|
HTTP_READ_PASS_ASYNC = 6,
|
|
HTTP_READ_FAIL_SYNC = 7,
|
|
HTTP_READ_FAIL_ASYNC = 8,
|
|
HTTP_SEND_FAIL_SYNC = 9,
|
|
HTTP_SEND_FAIL_ASYNC = 10
|
|
} HTTP_STATE;
|
|
|
|
typedef struct
|
|
{
|
|
HTTP_STATE state;
|
|
BOOL bQDA;
|
|
DWORD dwTotal;
|
|
DWORD_PTR dwResult;
|
|
DWORD dwError;
|
|
HANDLE hEvent;
|
|
DWORD dwSignature;
|
|
BOOL bCallbackDelete;
|
|
BOOL fCallbackClose;
|
|
} TestContext;
|
|
|
|
VOID CALLBACK Callback(IN HINTERNET hInternet, IN DWORD_PTR dwContext,
|
|
IN DWORD dwStatus, IN LPVOID pvInfo, IN DWORD dwStatusLen)
|
|
{
|
|
#if 0
|
|
fprintf(stderr, "callback!!!!!! dwStatus %d\n", dwStatus);
|
|
#endif
|
|
|
|
TestContext* pContext = (TestContext*)dwContext;
|
|
if ((dwStatus == WINHTTP_CALLBACK_STATUS_SECURE_FAILURE) && pContext)
|
|
{
|
|
fprintf(stderr, "\n\tWINHTTP_CALLBACK_STATUS_SECURE_FAILURE, StatusFlags=%x\n", *((LPDWORD) pvInfo));
|
|
if (g_fCloseOnSSLFailure)
|
|
{
|
|
WinHttpCloseHandle(hInternet);
|
|
pContext->fCallbackClose = TRUE;
|
|
}
|
|
}
|
|
else if ((dwStatus == WINHTTP_CALLBACK_STATUS_REQUEST_COMPLETE) && pContext)
|
|
{
|
|
ASSERT (pContext->dwSignature == 0);
|
|
if (pContext->bCallbackDelete)
|
|
{
|
|
delete pContext;
|
|
goto end;
|
|
}
|
|
|
|
DWORD_PTR dwResult = ((LPWINHTTP_ASYNC_RESULT)(pvInfo))->dwResult;
|
|
DWORD dwError = ((LPWINHTTP_ASYNC_RESULT) (pvInfo) )->dwError;
|
|
|
|
pContext->dwResult = dwResult;
|
|
pContext->dwError = dwError;
|
|
SetEvent(pContext->hEvent);
|
|
#if 0
|
|
OutputDebugString("\n\tWINHTTP_STATUS_REQUEST_COMPLETE for QDA\n");
|
|
fprintf(stderr, "\n\tWINHTTP_STATUS_REQUEST_COMPLETE for QDA with %d result and %d error.\n", dwResult, dwBytes);
|
|
#endif
|
|
}
|
|
else if ((dwStatus == WINHTTP_CALLBACK_STATUS_HANDLE_CLOSING) && pContext)
|
|
{
|
|
// We aborted the connection due to an SSL failure.
|
|
// Time to wake up.
|
|
SetEvent(pContext->hEvent);
|
|
}
|
|
|
|
end:
|
|
return;
|
|
}
|
|
|
|
#define CHECK_ERROR(cond, err) if (!(cond)) {pszErr=(err); goto done;}
|
|
|
|
//==============================================================================
|
|
int RequestLoop (int argc, char **argv)
|
|
{
|
|
#ifndef GLOBAL_SESSION
|
|
HINTERNET g_hInternet = NULL;
|
|
#endif
|
|
HINTERNET hConnect = NULL;
|
|
HINTERNET hRequest = NULL;
|
|
|
|
DWORD dwAccessType = WINHTTP_ACCESS_TYPE_DEFAULT_PROXY;
|
|
WCHAR szProxyServer[256];
|
|
WCHAR szProxyBypass[256];
|
|
|
|
DWORD dwConnectFlags = 0;
|
|
BOOL fPreAuth = FALSE;
|
|
BOOL fEnableRevocation = FALSE;
|
|
DWORD dwEnableFlags = WINHTTP_ENABLE_SSL_REVOCATION;
|
|
|
|
PSTR pPostData = NULL;
|
|
DWORD cbPostData = 0;
|
|
|
|
PSTR pszErr = NULL;
|
|
BOOL fRet;
|
|
DWORD dwErr = ERROR_SUCCESS;
|
|
|
|
DWORD dwTarget = WINHTTP_AUTH_TARGET_SERVER;
|
|
LPVOID pAuthParams = NULL;
|
|
|
|
DWORD option = WINHTTP_OPTION_RESOLVE_TIMEOUT;
|
|
DWORD dwTimeout = 1000;
|
|
HANDLE hEvent = NULL;
|
|
|
|
PSTR pszObject = NULL;
|
|
PSTR pszPostFile = NULL;
|
|
char pszHost[128];
|
|
PSTR pszColon = NULL;
|
|
BOOL bDeleteContext = TRUE;
|
|
|
|
HCERTSTORE hMyStore = NULL;
|
|
PCCERT_CONTEXT pCertContext = NULL;
|
|
WCHAR szCertName[256];
|
|
DWORD dwErrorMask;
|
|
|
|
szCertName[0] = L'\0';
|
|
|
|
TestContext* pContext = new TestContext();
|
|
CHECK_ERROR(pContext, "new TestContext");
|
|
|
|
memset(pContext, 0, sizeof(TestContext));
|
|
|
|
g_fCloseOnSSLFailure = FALSE;
|
|
|
|
// Parse options.
|
|
while (argc && argv[0][0] == '-')
|
|
{
|
|
switch (tolower(argv[0][1]))
|
|
{
|
|
case 'p':
|
|
dwAccessType = WINHTTP_ACCESS_TYPE_NAMED_PROXY;
|
|
dwTarget = WINHTTP_AUTH_TARGET_PROXY;
|
|
::MultiByteToWideChar(CP_ACP, 0, argv[1], -1, &szProxyServer[0], 256);
|
|
// pszProxyServer = argv[1];
|
|
::MultiByteToWideChar(CP_ACP, 0, "<local>", -1, &szProxyBypass[0], 256);
|
|
// pszProxyBypass = "<local>";
|
|
argv++;
|
|
argc--;
|
|
break;
|
|
|
|
case 'c':
|
|
::MultiByteToWideChar(CP_ACP, 0, argv[1], -1, &szCertName[0], 256);
|
|
argv++;
|
|
argc--;
|
|
break;
|
|
|
|
case 's':
|
|
dwConnectFlags = WINHTTP_FLAG_SECURE;
|
|
break;
|
|
|
|
case 'r':
|
|
fEnableRevocation = TRUE;
|
|
break;
|
|
|
|
case 'd':
|
|
g_fCloseOnSSLFailure = TRUE;
|
|
break;
|
|
|
|
case 'a':
|
|
fPreAuth = TRUE;
|
|
break;
|
|
|
|
default:
|
|
fprintf (stderr, "\nUsage: [-p <proxy>] [-s] [-c CERT_FIND_SUBJECT_STR] <host>[:port] [<object> [<POST-file>]]");
|
|
fprintf (stderr, "\n -s: use secure sockets layer");
|
|
fprintf (stderr, "\n -p: specify proxy server. (\"<local>\" assumed for bypass.)");
|
|
fprintf (stderr, "\n -c: Use CERT_FIND_SUBJECT_STR to find cert context when client auth needed");
|
|
fprintf (stderr, "\n -r: Enable certificate revocation checking");
|
|
fprintf (stderr, "\n -d: CloseHandle on SSL failure");
|
|
exit (1);
|
|
}
|
|
|
|
argv++;
|
|
argc--;
|
|
}
|
|
|
|
// Parse host:port
|
|
{
|
|
int nlen = strlen(argv[0]);
|
|
if (nlen > sizeof(pszHost))
|
|
goto done;
|
|
strcpy(pszHost, argv[0]);
|
|
}
|
|
|
|
DWORD dwPort;
|
|
pszColon = strchr(pszHost, ':');
|
|
if (!pszColon)
|
|
dwPort = INTERNET_DEFAULT_PORT;
|
|
else
|
|
{
|
|
*pszColon++ = 0;
|
|
dwPort = atol (pszColon);
|
|
}
|
|
|
|
pszObject = argc >= 2 ? argv[1] : "/";
|
|
pszPostFile = argc >= 3 ? argv[2] : NULL;
|
|
|
|
// Read any POST data into a buffer.
|
|
if (pszPostFile)
|
|
{
|
|
HANDLE hf =
|
|
CreateFile
|
|
(
|
|
pszPostFile,
|
|
GENERIC_READ,
|
|
FILE_SHARE_READ,
|
|
NULL,
|
|
OPEN_EXISTING,
|
|
FILE_ATTRIBUTE_NORMAL | FILE_FLAG_SEQUENTIAL_SCAN,
|
|
NULL
|
|
);
|
|
if (hf != INVALID_HANDLE_VALUE)
|
|
{
|
|
cbPostData = GetFileSize (hf, NULL);
|
|
pPostData = (PSTR) LocalAlloc (LMEM_FIXED, cbPostData + 1);
|
|
if (pPostData)
|
|
ReadFile (hf, pPostData, cbPostData, &cbPostData, NULL);
|
|
pPostData[cbPostData] = 0;
|
|
CloseHandle (hf);
|
|
}
|
|
}
|
|
|
|
// Initialize wininet.
|
|
#ifdef GLOBAL_SESSION
|
|
if (!g_hInternet)
|
|
#endif
|
|
g_hInternet = WinHttpOpen
|
|
(
|
|
_T("HttpAuth Sample"), // user agent
|
|
// "Mozilla/4.0 (compatible; MSIE 4.0b2; Windows 95",
|
|
dwAccessType, // access type
|
|
dwAccessType == WINHTTP_ACCESS_TYPE_NAMED_PROXY ? szProxyServer : NULL, // proxy server
|
|
dwAccessType == WINHTTP_ACCESS_TYPE_NAMED_PROXY ? szProxyBypass : NULL, // proxy bypass
|
|
WINHTTP_FLAG_ASYNC // flags
|
|
);
|
|
CHECK_ERROR (g_hInternet, "WinHttpOpen");
|
|
|
|
|
|
WCHAR szHost[256];
|
|
::MultiByteToWideChar(CP_ACP, 0, pszHost, -1, &szHost[0], 256);
|
|
|
|
WinHttpSetStatusCallback(g_hInternet, Callback, WINHTTP_CALLBACK_FLAG_ALL_NOTIFICATIONS, NULL);
|
|
|
|
dwTimeout = 1000;
|
|
WinHttpSetOption(
|
|
g_hInternet,
|
|
WINHTTP_OPTION_RESOLVE_TIMEOUT,
|
|
(LPVOID)&dwTimeout,
|
|
sizeof(DWORD)
|
|
);
|
|
|
|
dwTimeout = 1000;
|
|
WinHttpSetOption(
|
|
g_hInternet,
|
|
WINHTTP_OPTION_CONNECT_TIMEOUT,
|
|
(LPVOID)&dwTimeout,
|
|
sizeof(DWORD)
|
|
);
|
|
|
|
dwTimeout = 100;
|
|
WinHttpSetOption(
|
|
g_hInternet,
|
|
WINHTTP_OPTION_SEND_TIMEOUT,
|
|
(LPVOID)&dwTimeout,
|
|
sizeof(DWORD)
|
|
);
|
|
|
|
dwTimeout = 10000;
|
|
WinHttpSetOption(
|
|
g_hInternet,
|
|
WINHTTP_OPTION_RECEIVE_TIMEOUT,
|
|
(LPVOID)&dwTimeout,
|
|
sizeof(DWORD)
|
|
);
|
|
|
|
hEvent = CreateEvent(NULL, FALSE /*want auto-reset*/, FALSE /*non-signaled init.*/, NULL);
|
|
|
|
CHECK_ERROR(hEvent, "CreateEvent");
|
|
|
|
pContext->hEvent = hEvent;
|
|
|
|
// Connect to host.
|
|
hConnect = WinHttpConnect
|
|
(
|
|
g_hInternet, // session handle,
|
|
szHost, // host
|
|
(INTERNET_PORT) dwPort, // port
|
|
0 // flags
|
|
);
|
|
CHECK_ERROR (hConnect, "WinHttpConnect");
|
|
|
|
WCHAR szObject[256];
|
|
::MultiByteToWideChar(CP_ACP, 0, pszObject, -1, &szObject[0], 256);
|
|
|
|
// Create request.
|
|
|
|
pContext->state = HTTP_OPEN;
|
|
hRequest = WinHttpOpenRequest
|
|
(
|
|
hConnect, // connect handle
|
|
pszPostFile? _T("POST") : _T("GET"), // request method
|
|
szObject, // object name
|
|
NULL, // version
|
|
NULL, // referer
|
|
NULL, // accept types
|
|
dwConnectFlags // flags
|
|
);
|
|
CHECK_ERROR (hRequest, "WinHttpOpenRequest");
|
|
|
|
if (fEnableRevocation)
|
|
{
|
|
WinHttpSetOption(hRequest,
|
|
WINHTTP_OPTION_ENABLE_FEATURE,
|
|
(LPVOID)&dwEnableFlags,
|
|
sizeof(dwEnableFlags));
|
|
}
|
|
|
|
try_again:
|
|
// Send request.
|
|
pContext->state = HTTP_SEND;
|
|
fRet = WinHttpSendRequest
|
|
(
|
|
hRequest, // request handle
|
|
_T(""), // header string
|
|
0, // header length
|
|
pPostData, // post data
|
|
cbPostData, // post data length
|
|
cbPostData, // total post length
|
|
(DWORD_PTR) pContext // flags
|
|
);
|
|
|
|
if (!fRet)
|
|
{
|
|
dwErr = GetLastError();
|
|
switch (dwErr)
|
|
{
|
|
case ERROR_WINHTTP_CLIENT_AUTH_CERT_NEEDED:
|
|
break;
|
|
|
|
case ERROR_IO_PENDING:
|
|
fRet = WinHttpReceiveResponse(hRequest, NULL);
|
|
if (!fRet)
|
|
{
|
|
dwErr = GetLastError();
|
|
if (dwErr != ERROR_IO_PENDING)
|
|
{
|
|
fprintf (stderr, "WinHttpReceiveResponse failed err=%d\n", dwErr);
|
|
break;
|
|
}
|
|
}
|
|
goto async;
|
|
|
|
default:
|
|
fprintf (stderr, "HttpSendRequest failed err=%d\n", dwErr);
|
|
}
|
|
pContext->state = HTTP_SEND_FAIL_SYNC;
|
|
goto done;
|
|
}
|
|
else
|
|
{
|
|
fRet = WinHttpReceiveResponse(hRequest, NULL);
|
|
if (!fRet)
|
|
{
|
|
dwErr = GetLastError();
|
|
fprintf (stderr, "WinHttpReceiveResponse failed err=%d\n", dwErr);
|
|
pContext->state = HTTP_SEND_FAIL_SYNC;
|
|
}
|
|
goto sync;
|
|
}
|
|
|
|
async:
|
|
fRet = WinHttpReceiveResponse(hRequest, NULL);
|
|
|
|
if (!fRet)
|
|
{
|
|
DWORD dwErr = GetLastError();
|
|
|
|
if (dwErr != ERROR_IO_PENDING)
|
|
{
|
|
fprintf(stderr, "WinHttpReceiveResponse failed err=%d\n", dwErr);
|
|
pContext->state = HTTP_SEND_FAIL_SYNC;
|
|
goto done;
|
|
}
|
|
}
|
|
|
|
#if 0
|
|
fprintf(stderr, "\nERROR_IO_PENDING on HSR...\n");
|
|
#endif
|
|
WaitForSingleObject(hEvent, INFINITE);
|
|
|
|
ASSERT( pContext->state == HTTP_SEND );
|
|
|
|
if (pContext->dwError == ERROR_WINHTTP_CLIENT_AUTH_CERT_NEEDED)
|
|
{
|
|
if (szCertName[0])
|
|
{
|
|
hMyStore = CertOpenSystemStore(0, TEXT("MY"));
|
|
if (hMyStore)
|
|
{
|
|
pCertContext = CertFindCertificateInStore(hMyStore,
|
|
X509_ASN_ENCODING | PKCS_7_ASN_ENCODING,
|
|
0,
|
|
CERT_FIND_SUBJECT_STR,
|
|
(LPVOID) szCertName,
|
|
NULL);
|
|
if (pCertContext)
|
|
{
|
|
WinHttpSetOption(hRequest,
|
|
WINHTTP_OPTION_CLIENT_CERT_CONTEXT,
|
|
(LPVOID) pCertContext,
|
|
sizeof(CERT_CONTEXT));
|
|
CertFreeCertificateContext(pCertContext);
|
|
}
|
|
CertCloseStore(hMyStore, 0);
|
|
goto try_again;
|
|
}
|
|
}
|
|
}
|
|
|
|
if (! (pContext->dwResult) )
|
|
{
|
|
pContext->state = HTTP_SEND_FAIL_ASYNC;
|
|
SetLastError (pContext->dwError);
|
|
goto done;
|
|
}
|
|
|
|
sync:
|
|
|
|
fRet = WinHttpReceiveResponse(hRequest, NULL);
|
|
|
|
if (!fRet)
|
|
{
|
|
DWORD dwErr = GetLastError();
|
|
|
|
if (dwErr != ERROR_IO_PENDING)
|
|
{
|
|
fprintf(stderr, "WinHttpReceiveResponse failed err=%d\n", dwErr);
|
|
pContext->state = HTTP_SEND_FAIL_SYNC;
|
|
goto done;
|
|
}
|
|
}
|
|
// Get status code.
|
|
DWORD dwStatus, cbStatus;
|
|
cbStatus = sizeof(dwStatus);
|
|
WinHttpQueryHeaders
|
|
(
|
|
hRequest,
|
|
WINHTTP_QUERY_FLAG_NUMBER | WINHTTP_QUERY_STATUS_CODE,
|
|
NULL,
|
|
&dwStatus,
|
|
&cbStatus,
|
|
NULL
|
|
);
|
|
fprintf (stderr, "Status: %d\n", dwStatus);
|
|
|
|
// Dump some bytes.
|
|
BYTE bBuf[1024];
|
|
DWORD cbBuf;
|
|
|
|
#define QDA 1
|
|
#ifdef QDA
|
|
while (TRUE)
|
|
{
|
|
BOOL bAsyncRead = FALSE;
|
|
pContext->state = HTTP_QDA;
|
|
fRet = WinHttpQueryDataAvailable (hRequest, &(pContext->dwError));
|
|
|
|
if (!fRet)
|
|
{
|
|
if (GetLastError() == ERROR_IO_PENDING)
|
|
{
|
|
#if 0
|
|
OutputDebugString("\nERROR_IO_PENDING on QDA\n");
|
|
fprintf(stderr, "\nERROR_IO_PENDING on QDA...\n");
|
|
#endif
|
|
WaitForSingleObject(hEvent, INFINITE);
|
|
|
|
ASSERT (pContext->state = HTTP_QDA);
|
|
if (!(pContext->dwResult))
|
|
{
|
|
// Done
|
|
pContext->state = HTTP_READ_FAIL_ASYNC;
|
|
SetLastError(pContext->dwError);
|
|
goto done;
|
|
}
|
|
else if (!(pContext->dwError))
|
|
{
|
|
pContext->state = HTTP_READ_PASS_ASYNC;
|
|
goto done;
|
|
}
|
|
bAsyncRead = TRUE;
|
|
}
|
|
else
|
|
{
|
|
pContext->state = HTTP_READ_FAIL_SYNC;
|
|
goto done;
|
|
}
|
|
}
|
|
else if (!(pContext->dwError))
|
|
{
|
|
//Done sync
|
|
pContext->state = HTTP_READ_PASS_SYNC;
|
|
goto done;
|
|
}
|
|
|
|
DWORD dwRead = pContext->dwError;
|
|
DWORD dwActuallyRead = 0;
|
|
while (dwRead > 0)
|
|
{
|
|
cbBuf = (dwRead > sizeof(bBuf)) ? sizeof(bBuf) : dwRead;
|
|
pContext->state = HTTP_READ;
|
|
fRet = WinHttpReadData(hRequest, bBuf, cbBuf, &dwActuallyRead);
|
|
|
|
if (!fRet)
|
|
{
|
|
ASSERT ( GetLastError() != ERROR_IO_PENDING );
|
|
fprintf(stderr, "\nError in WinHttpReadData = %d\n", GetLastError());
|
|
goto done;
|
|
}
|
|
ASSERT((cbBuf == dwActuallyRead));
|
|
#if 0
|
|
ASSERT(fRet);
|
|
fwrite (bBuf, 1, dwActuallyRead, stdout);
|
|
#endif
|
|
pContext->dwTotal += dwActuallyRead;
|
|
dwRead -= dwActuallyRead;
|
|
}
|
|
}
|
|
#else
|
|
cbBuf = sizeof(bBuf);
|
|
while (WinHttpReadData (hRequest, bBuf, cbBuf, &(pContext->dwRead)) && pContext->dwRead)
|
|
fwrite (bBuf, 1, pContext->dwRead, stdout);
|
|
#endif
|
|
|
|
done: // Clean up.
|
|
|
|
if (pszErr)
|
|
fprintf (stderr, "Failed on %s, last error %d\n", pszErr, GetLastError());
|
|
if (hRequest)
|
|
WinHttpCloseHandle (hRequest);
|
|
if (hConnect)
|
|
WinHttpCloseHandle (hConnect);
|
|
|
|
#ifndef GLOBAL_SESSION
|
|
if (g_hInternet)
|
|
WinHttpCloseHandle (g_hInternet);
|
|
#endif
|
|
|
|
if (pPostData)
|
|
LocalFree (pPostData);
|
|
if (hEvent)
|
|
CloseHandle(hEvent);
|
|
if (pContext && bDeleteContext)
|
|
{
|
|
pContext->dwSignature = 0x41414141;
|
|
delete pContext;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
//==============================================================================
|
|
void ParseArguments
|
|
(
|
|
LPSTR InBuffer,
|
|
LPSTR* CArgv,
|
|
DWORD* CArgc
|
|
)
|
|
{
|
|
LPSTR CurrentPtr = InBuffer;
|
|
DWORD i = 0;
|
|
DWORD Cnt = 0;
|
|
|
|
for ( ;; ) {
|
|
|
|
//
|
|
// skip blanks.
|
|
//
|
|
|
|
while( *CurrentPtr == ' ' ) {
|
|
CurrentPtr++;
|
|
}
|
|
|
|
if( *CurrentPtr == '\0' ) {
|
|
break;
|
|
}
|
|
|
|
CArgv[i++] = CurrentPtr;
|
|
|
|
//
|
|
// go to next space.
|
|
//
|
|
|
|
while( (*CurrentPtr != '\0') &&
|
|
(*CurrentPtr != '\n') ) {
|
|
if( *CurrentPtr == '"' ) { // Deal with simple quoted args
|
|
if( Cnt == 0 )
|
|
CArgv[i-1] = ++CurrentPtr; // Set arg to after quote
|
|
else
|
|
*CurrentPtr = '\0'; // Remove end quote
|
|
Cnt = !Cnt;
|
|
}
|
|
if( (Cnt == 0) && (*CurrentPtr == ' ') || // If we hit a space and no quotes yet we are done with this arg
|
|
(*CurrentPtr == '\0') )
|
|
break;
|
|
CurrentPtr++;
|
|
}
|
|
|
|
if( *CurrentPtr == '\0' ) {
|
|
break;
|
|
}
|
|
|
|
*CurrentPtr++ = '\0';
|
|
}
|
|
|
|
*CArgc = i;
|
|
return;
|
|
}
|
|
|
|
|
|
DWORD WINAPI WorkThread1(LPVOID lpParameter)
|
|
{
|
|
int nCount = 0;
|
|
char* pargv[] = {"dennisch", "venkatk/large.html", 0};
|
|
|
|
while (nCount++ < 500000)
|
|
{
|
|
fprintf (stderr, "\nLARGE Iteration #%d\n", nCount);
|
|
RequestLoop( 2, pargv);
|
|
}
|
|
|
|
fprintf (stderr, "\nLARGE DONE!\n");
|
|
|
|
return 0;
|
|
}
|
|
DWORD WINAPI WorkThread2(LPVOID lpParameter)
|
|
{
|
|
int nCount = 0;
|
|
char* pargv[] = {"dennisch", "venkatk/small.html", 0};
|
|
|
|
while (nCount++ < 1000000)
|
|
{
|
|
fprintf (stderr, "\nSMALL Iteration #%d\n", nCount);
|
|
RequestLoop( 2, pargv);
|
|
}
|
|
|
|
fprintf (stderr, "\nSMALL DONE!\n");
|
|
|
|
return 0;
|
|
}
|
|
DWORD WINAPI WorkThread3(LPVOID lpParameter)
|
|
{
|
|
int nCount = 0;
|
|
char* pargv[] = {"venkatk:180", 0};
|
|
|
|
while (nCount++ < 1000)
|
|
{
|
|
fprintf (stderr, "\n180 Iteration #%d\n", nCount);
|
|
RequestLoop( 1, pargv);
|
|
}
|
|
|
|
fprintf (stderr, "\n180 DONE!\n");
|
|
|
|
return 0;
|
|
}
|
|
|
|
//==============================================================================
|
|
int __cdecl main (int argc, char **argv)
|
|
{
|
|
char * port;
|
|
int nCount = 0;
|
|
// Discard program arg.
|
|
argv++;
|
|
argc--;
|
|
|
|
|
|
/*
|
|
#if 1
|
|
DWORD dwThreadId;
|
|
HANDLE hThread1 = CreateThread(NULL, 0, &WorkThread1,
|
|
NULL, 0, &dwThreadId);
|
|
Sleep(1000);
|
|
HANDLE hThread2 = CreateThread(NULL, 0, &WorkThread2,
|
|
NULL, 0, &dwThreadId);
|
|
HANDLE hThread3 = CreateThread(NULL, 0, &WorkThread3,
|
|
NULL, 0, &dwThreadId);
|
|
|
|
{
|
|
char* pargv[] = {"pmidge", 0};
|
|
|
|
while (nCount++ < 1000000)
|
|
{
|
|
fprintf (stderr, "\nPMIDGE Iteration #%d\n", nCount);
|
|
RequestLoop( 1, pargv);
|
|
}
|
|
|
|
fprintf (stderr, "\n180 DONE!\n");
|
|
}
|
|
|
|
WaitForSingleObject( hThread1, INFINITE );
|
|
WaitForSingleObject( hThread2, INFINITE );
|
|
WaitForSingleObject( hThread3, INFINITE );
|
|
#endif
|
|
|
|
#if 0
|
|
char* argv_large[] = {"dennisch", "venkatk/large.html", 0};
|
|
char* argv_small[] = {"dennisch", "venkatk/small.html", 0};
|
|
char* argv_delay[] = {"venkatk:180", 0};
|
|
|
|
while(nCount++ < 30)
|
|
{
|
|
RequestLoop( 1, argv_delay );
|
|
}
|
|
|
|
while(nCount++ < 100)
|
|
{
|
|
#if 0
|
|
fprintf (stderr, "\nIteration #%d\n", nCount);
|
|
#endif
|
|
RequestLoop( 2, argv_large);
|
|
RequestLoop( 2, argv_small);
|
|
RequestLoop( 1, argv_delay);
|
|
}
|
|
#endif
|
|
fprintf (stderr, "\nIteration #%d\n", nCount);
|
|
|
|
*/
|
|
if (argc)
|
|
RequestLoop (argc, argv);
|
|
|
|
else // Enter command prompt loop
|
|
{
|
|
fprintf (stderr, "\nUsage: [-p <proxy>] [-s] [-c CERT_FIND_SUBJECT_STR] <host>[:port] [<object> [<POST-file>]]");
|
|
fprintf (stderr, "\n -s: use secure sockets layer");
|
|
fprintf (stderr, "\n -p: specify proxy server. (\"<local>\" assumed for bypass.)");
|
|
fprintf (stderr, "\n -c: Use CERT_FIND_SUBJECT_STR to find cert context when client auth needed");
|
|
fprintf (stderr, "\n -r: Enable certificate revocation checking");
|
|
fprintf (stderr, "\n -d: CloseHandle on SSL failure");
|
|
fprintf (stderr, "\nTo exit input loop, enter no params");
|
|
|
|
while (1)
|
|
{
|
|
char szIn[1024];
|
|
DWORD argcIn;
|
|
LPSTR argvIn[10];
|
|
|
|
fprintf (stderr, "\nhttpauth> ");
|
|
gets (szIn);
|
|
|
|
argcIn = 0;
|
|
ParseArguments (szIn, argvIn, &argcIn);
|
|
if (!argcIn)
|
|
break;
|
|
RequestLoop (argcIn, argvIn);
|
|
//g_cbRead=0;
|
|
}
|
|
}
|
|
|
|
#ifdef GLOBAL_SESSION
|
|
if (g_hInternet)
|
|
WinHttpCloseHandle (g_hInternet);
|
|
g_hInternet = NULL;
|
|
#endif
|
|
|
|
|
|
}
|
|
|