static int apply_default_credentials(HINTERNET request, int mechanisms)
{
/* Either the caller explicitly requested that default credentials be passed,
* or our fallback credential callback was invoked and checked that the target
* URI was in the appropriate Internet Explorer security zone. By setting this
* flag, we guarantee that the credentials are delivered by WinHTTP. The default
* is "medium" which applies to the intranet and sounds like it would correspond
* to Internet Explorer security zones, but in fact does not. */
DWORD data = WINHTTP_AUTOLOGON_SECURITY_LEVEL_LOW;
DWORD native_scheme = 0;
if ((mechanisms & GIT_WINHTTP_AUTH_NTLM) != 0)
native_scheme |= WINHTTP_AUTH_SCHEME_NTLM;
if ((mechanisms & GIT_WINHTTP_AUTH_NEGOTIATE) != 0)
native_scheme |= WINHTTP_AUTH_SCHEME_NEGOTIATE;
if (!native_scheme) {
giterr_set(GITERR_NET, "invalid authentication scheme");
return -1;
}
if (!WinHttpSetOption(request, WINHTTP_OPTION_AUTOLOGON_POLICY, &data, sizeof(DWORD)))
return -1;
if (!WinHttpSetCredentials(request, WINHTTP_AUTH_TARGET_SERVER, native_scheme, NULL, NULL, NULL))
return -1;
return 0;
}
// POST request to URL
void WinHttpIO::post(HttpReq* req, const char* data, unsigned len)
{
LOG_debug << "POST target URL: " << req->posturl << " chunked: " << req->chunked;
if (req->binary)
{
LOG_debug << "[sending " << (data ? len : req->out->size()) << " bytes of raw data]";
}
else
{
LOG_debug << "Sending: " << *req->out;
}
WinHttpContext* httpctx;
WCHAR szURL[8192];
WCHAR szHost[256];
URL_COMPONENTS urlComp = { sizeof urlComp };
urlComp.lpszHostName = szHost;
urlComp.dwHostNameLength = sizeof szHost / sizeof *szHost;
urlComp.dwUrlPathLength = (DWORD)-1;
urlComp.dwSchemeLength = (DWORD)-1;
httpctx = new WinHttpContext;
httpctx->httpio = this;
httpctx->req = req;
httpctx->gzip = false;
req->httpiohandle = (void*)httpctx;
if (MultiByteToWideChar(CP_UTF8, 0, req->posturl.c_str(), -1, szURL,
sizeof szURL / sizeof *szURL)
&& WinHttpCrackUrl(szURL, 0, 0, &urlComp))
{
if ((httpctx->hConnect = WinHttpConnect(hSession, szHost, urlComp.nPort, 0)))
{
httpctx->hRequest = WinHttpOpenRequest(httpctx->hConnect, L"POST",
urlComp.lpszUrlPath, NULL,
WINHTTP_NO_REFERER,
WINHTTP_DEFAULT_ACCEPT_TYPES,
(urlComp.nScheme == INTERNET_SCHEME_HTTPS)
? WINHTTP_FLAG_SECURE
: 0);
if (httpctx->hRequest)
{
if (proxyUsername.size())
{
LOG_verbose << "Setting proxy credentials";
WinHttpSetCredentials(httpctx->hRequest, WINHTTP_AUTH_TARGET_PROXY,
WINHTTP_AUTH_SCHEME_BASIC,
(LPWSTR)proxyUsername.data(), (LPWSTR)proxyPassword.data(), NULL);
}
WinHttpSetTimeouts(httpctx->hRequest, 58000, 58000, 0, 0);
WinHttpSetStatusCallback(httpctx->hRequest, asynccallback,
WINHTTP_CALLBACK_FLAG_DATA_AVAILABLE
| WINHTTP_CALLBACK_FLAG_READ_COMPLETE
| WINHTTP_CALLBACK_FLAG_HEADERS_AVAILABLE
| WINHTTP_CALLBACK_FLAG_REQUEST_ERROR
| WINHTTP_CALLBACK_FLAG_SECURE_FAILURE
| WINHTTP_CALLBACK_FLAG_SENDREQUEST_COMPLETE
| WINHTTP_CALLBACK_FLAG_SEND_REQUEST
| WINHTTP_CALLBACK_FLAG_WRITE_COMPLETE
| WINHTTP_CALLBACK_FLAG_HANDLES,
0);
LPCWSTR pwszHeaders = req->type == REQ_JSON || !req->buf
? L"Content-Type: application/json\r\nAccept-Encoding: gzip"
: L"Content-Type: application/octet-stream";
// data is sent in HTTP_POST_CHUNK_SIZE instalments to ensure
// semi-smooth UI progress info
if (req->chunkedout.size())
{
req->outbuf.append(req->chunkedout);
req->chunkedout.clear();
}
req->chunked = 0;
httpctx->postlen = data ? len : req->out->size();
httpctx->postdata = data ? data : req->out->data();
if (urlComp.nPort == 80)
{
LOG_verbose << "HTTP connection";
// HTTP connection: send a chunk of data immediately
httpctx->postpos = (httpctx->postlen < HTTP_POST_CHUNK_SIZE)
? httpctx->postlen
: HTTP_POST_CHUNK_SIZE;
}
else
{
LOG_verbose << "HTTPS connection";
// HTTPS connection: ignore certificate errors, send no data yet
DWORD flags = SECURITY_FLAG_IGNORE_CERT_CN_INVALID
| SECURITY_FLAG_IGNORE_CERT_DATE_INVALID
| SECURITY_FLAG_IGNORE_UNKNOWN_CA;
WinHttpSetOption(httpctx->hRequest, WINHTTP_OPTION_SECURITY_FLAGS, &flags, sizeof flags);
httpctx->postpos = 0;
}
if (WinHttpSendRequest(httpctx->hRequest, pwszHeaders,
wcslen(pwszHeaders),
(LPVOID)httpctx->postdata,
httpctx->postpos,
httpctx->postlen,
(DWORD_PTR)httpctx))
{
LOG_verbose << "Request sent";
req->status = REQ_INFLIGHT;
return;
}
LOG_err << "Error sending request: " << req->posturl << " Code: " << GetLastError();
}
else
{
LOG_err << "Error opening request: " << req->posturl << " Code: " << GetLastError();
}
}
else
{
LOG_err << "Error connecting to " << req->posturl << " Code: " << GetLastError();
httpctx->hRequest = NULL;
}
}
else
{
LOG_err << "Error parsing POST URL: " << req->posturl << " Code: " << GetLastError();
httpctx->hRequest = NULL;
httpctx->hConnect = NULL;
}
LOG_err << "Request failed";
req->status = REQ_FAILURE;
}
static int _apply_userpass_credential(HINTERNET request, DWORD target, DWORD scheme, git_cred *cred)
{
git_cred_userpass_plaintext *c = (git_cred_userpass_plaintext *)cred;
wchar_t *user, *pass;
int user_len = 0, pass_len = 0, error = 0;
if ((error = user_len = git__utf8_to_16_alloc(&user, c->username)) < 0)
goto done;
if ((error = pass_len = git__utf8_to_16_alloc(&pass, c->password)) < 0)
goto done;
if (!WinHttpSetCredentials(request, target, scheme, user, pass, NULL)) {
giterr_set(GITERR_OS, "failed to set credentials");
error = -1;
}
done:
if (user_len > 0)
git__memzero(user, user_len * sizeof(wchar_t));
if (pass_len > 0)
git__memzero(pass, pass_len * sizeof(wchar_t));
git__free(user);
git__free(pass);
return error;
}
static BOOL set_request_credentials(HINTERNET req, BackgroundCopyJobImpl *job)
{
UINT i, j;
for (i = 0; i < BG_AUTH_TARGET_PROXY; i++)
{
UINT target = target_from_index(i);
for (j = 0; j < BG_AUTH_SCHEME_PASSPORT; j++)
{
UINT scheme = scheme_from_index(j);
const WCHAR *username = job->http_options.creds[i][j].Credentials.Basic.UserName;
const WCHAR *password = job->http_options.creds[i][j].Credentials.Basic.Password;
if (!username) continue;
if (!WinHttpSetCredentials(req, target, scheme, username, password, NULL)) return FALSE;
}
}
return TRUE;
}
static int apply_basic_credential_proxy(HINTERNET request, git_cred *cred)
{
git_cred_userpass_plaintext *c = (git_cred_userpass_plaintext *)cred;
wchar_t *user, *pass;
int error;
if ((error = git__utf8_to_16_alloc(&user, c->username)) < 0)
return error;
if ((error = git__utf8_to_16_alloc(&pass, c->password)) < 0)
return error;
if (!WinHttpSetCredentials(request, WINHTTP_AUTH_TARGET_PROXY, WINHTTP_AUTH_SCHEME_BASIC,
user, pass, NULL)) {
giterr_set(GITERR_OS, "failed to set proxy auth");
error = -1;
}
git__free(user);
git__free(pass);
return error;
}
void HttpRequest::SetCredentials(IN DWORD AuthTargets, IN DWORD AuthScheme, IN LPCWSTR pwszUserName, IN LPCWSTR pwszPassword, IN LPVOID pAuthParams) const
{
if (!WinHttpSetCredentials(_handle, AuthTargets, AuthScheme, pwszUserName, pwszPassword, pAuthParams))
throw WinHttpException("WinHttpSetCredentials error", GetLastError());
}
BOOL CSyoboiCalUtil::SendReserve(const vector<RESERVE_DATA>* reserveList, const vector<TUNER_RESERVE_INFO>* tunerList)
{
if( reserveList == NULL || tunerList == NULL ){
return FALSE;
}
if( reserveList->size() == 0 ){
return FALSE;
}
wstring iniAppPath = L"";
GetModuleIniPath(iniAppPath);
if( GetPrivateProfileInt(L"SYOBOI", L"use", 0, iniAppPath.c_str()) == 0 ){
return FALSE;
}
_OutputDebugString(L"★SyoboiCalUtil:SendReserve");
wstring textPath;
GetModuleFolderPath(textPath);
textPath += L"\\SyoboiCh.txt";
CParseServiceChgText srvChg;
srvChg.ParseText(textPath.c_str());
wstring proxyServerName;
wstring proxyUserName;
wstring proxyPassword;
if( GetPrivateProfileInt(L"SYOBOI", L"useProxy", 0, iniAppPath.c_str()) != 0 ){
proxyServerName = GetPrivateProfileToString(L"SYOBOI", L"ProxyServer", L"", iniAppPath.c_str());
proxyUserName = GetPrivateProfileToString(L"SYOBOI", L"ProxyID", L"", iniAppPath.c_str());
proxyPassword = GetPrivateProfileToString(L"SYOBOI", L"ProxyPWD", L"", iniAppPath.c_str());
}
wstring id=GetPrivateProfileToString(L"SYOBOI", L"userID", L"", iniAppPath.c_str());
wstring pass=GetPrivateProfileToString(L"SYOBOI", L"PWD", L"", iniAppPath.c_str());
int slot = GetPrivateProfileInt(L"SYOBOI", L"slot", 0, iniAppPath.c_str());
wstring devcolors=GetPrivateProfileToString(L"SYOBOI", L"devcolors", L"", iniAppPath.c_str());
wstring epgurl=GetPrivateProfileToString(L"SYOBOI", L"epgurl", L"", iniAppPath.c_str());
if( id.size() == 0 ){
_OutputDebugString(L"★SyoboiCalUtil:NoUserID");
return FALSE;
}
//Authorization
wstring auth = L"";
auth = id;
auth += L":";
auth += pass;
string authA;
WtoA(auth, authA);
DWORD destSize = 0;
Base64Enc(authA.c_str(), (DWORD)authA.size(), NULL, &destSize);
vector<WCHAR> base64(destSize + 1, L'\0');
Base64Enc(authA.c_str(), (DWORD)authA.size(), &base64.front(), &destSize);
//無駄なCRLFが混じることがあるため
std::replace(base64.begin(), base64.end(), L'\r', L'\0');
std::replace(base64.begin(), base64.end(), L'\n', L'\0');
wstring authHead = L"";
Format(authHead, L"Authorization: Basic %s\r\nContent-type: application/x-www-form-urlencoded\r\n", &base64.front());
//data
wstring dataParam;
wstring param;
map<DWORD, wstring> tunerMap;
for( size_t i=0; i<tunerList->size(); i++ ){
for( size_t j=0; j<(*tunerList)[i].reserveList.size(); j++ ){
tunerMap.insert(pair<DWORD, wstring>((*tunerList)[i].reserveList[j], (*tunerList)[i].tunerName));
}
}
map<DWORD, wstring>::iterator itrTuner;
DWORD dataCount = 0;
for(size_t i=0; i<reserveList->size(); i++ ){
if( dataCount>=200 ){
break;
}
const RESERVE_DATA* info = &(*reserveList)[i];
if( info->recSetting.recMode == RECMODE_NO || info->recSetting.recMode == RECMODE_VIEW ){
continue;
}
wstring device=L"";
itrTuner = tunerMap.find(info->reserveID);
if( itrTuner != tunerMap.end() ){
device = itrTuner->second;
}
wstring stationName = info->stationName;
srvChg.ChgText(stationName);
__int64 startTime = GetTimeStamp(info->startTime);
Format(param, L"%I64d\t%I64d\t%s\t%s\t%s\t\t0\t%d\n", startTime, startTime+info->durationSecond, device.c_str(), info->title.c_str(), stationName.c_str(), info->reserveID );
dataParam+=param;
}
if(dataParam.size() == 0 ){
_OutputDebugString(L"★SyoboiCalUtil:NoReserve");
return FALSE;
}
string utf8;
UrlEncodeUTF8(dataParam.c_str(), (DWORD)dataParam.size(), utf8);
string data;
Format(data, "slot=%d&data=%s",slot, utf8.c_str());
if( devcolors.size() > 0){
utf8 = "";
UrlEncodeUTF8(devcolors.c_str(), (DWORD)devcolors.size(), utf8);
data += "&devcolors=";
data += utf8;
}
if( epgurl.size() > 0){
utf8 = "";
UrlEncodeUTF8(epgurl.c_str(), (DWORD)epgurl.size(), utf8);
data += "&epgurl=";
data += utf8;
}
vector<char> dataBuff(data.begin(), data.end());
//URLの分解
URL_COMPONENTS stURL = {};
stURL.dwStructSize = sizeof(stURL);
stURL.dwSchemeLength = (DWORD)-1;
stURL.dwHostNameLength = (DWORD)-1;
stURL.dwUrlPathLength = (DWORD)-1;
stURL.dwExtraInfoLength = (DWORD)-1;
if( WinHttpCrackUrl(SYOBOI_UP_URL, 0, 0, &stURL) == FALSE || stURL.dwHostNameLength == 0 ){
return FALSE;
}
wstring host(stURL.lpszHostName, stURL.dwHostNameLength);
wstring sendUrl(stURL.lpszUrlPath, stURL.dwUrlPathLength + stURL.dwExtraInfoLength);
HINTERNET session;
if( proxyServerName.empty() ){
session = WinHttpOpen(L"EpgTimerSrv", WINHTTP_ACCESS_TYPE_NO_PROXY, WINHTTP_NO_PROXY_NAME, WINHTTP_NO_PROXY_BYPASS, 0);
}else{
session = WinHttpOpen(L"EpgTimerSrv", WINHTTP_ACCESS_TYPE_NAMED_PROXY, proxyServerName.c_str(), WINHTTP_NO_PROXY_BYPASS, 0);
}
if( session == NULL ){
return FALSE;
}
LPCWSTR result = L"1";
HINTERNET connect = NULL;
HINTERNET request = NULL;
if( WinHttpSetTimeouts(session, 15000, 15000, 15000, 15000) == FALSE ){
result = L"0 SetTimeouts";
goto EXIT;
}
//コネクションオープン
connect = WinHttpConnect(session, host.c_str(), stURL.nPort, 0);
if( connect == NULL ){
result = L"0 Connect";
goto EXIT;
}
//リクエストオープン
request = WinHttpOpenRequest(connect, L"POST", sendUrl.c_str(), NULL, WINHTTP_NO_REFERER, WINHTTP_DEFAULT_ACCEPT_TYPES,
stURL.nPort == INTERNET_DEFAULT_HTTPS_PORT ? WINHTTP_FLAG_SECURE : 0);
if( request == NULL ){
result = L"0 OpenRequest";
goto EXIT;
}
if( proxyServerName.empty() == false ){
//ProxyのIDかパスワードがあったらセット
if( proxyUserName.empty() == false || proxyPassword.empty() == false ){
if( WinHttpSetCredentials(request, WINHTTP_AUTH_TARGET_PROXY, WINHTTP_AUTH_SCHEME_BASIC,
proxyUserName.c_str(), proxyPassword.c_str(), NULL) == FALSE ){
result = L"0 SetCredentials";
goto EXIT;
}
}
}
if( WinHttpSendRequest(request, authHead.c_str(), (DWORD)-1, &dataBuff.front(), (DWORD)dataBuff.size(), (DWORD)dataBuff.size(), 0) == FALSE ){
result = L"0 SendRequest";
goto EXIT;
}
if( WinHttpReceiveResponse(request, NULL) == FALSE ){
result = L"0 ReceiveResponse";
goto EXIT;
}
//HTTPのステータスコード確認
DWORD statusCode;
DWORD statusCodeSize = sizeof(statusCode);
if( WinHttpQueryHeaders(request, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER, WINHTTP_HEADER_NAME_BY_INDEX,
&statusCode, &statusCodeSize, WINHTTP_NO_HEADER_INDEX) == FALSE ){
statusCode = 0;
}
if( statusCode != 200 && statusCode != 201 ){
result = L"0 StatusNotOK";
goto EXIT;
}
EXIT:
if( request != NULL ){
WinHttpCloseHandle(request);
}
if( connect != NULL ){
WinHttpCloseHandle(connect);
}
if( session != NULL ){
WinHttpCloseHandle(session);
}
_OutputDebugString(L"★SyoboiCalUtil:SendRequest res:%s", result);
if( result[0] != L'1' ){
return FALSE;
}
return TRUE;
}
//リクエストを送る
//内部メモリバッファにDLする場合はファイル容量に要注意
//戻り値：エラーコード
DWORD CWinHTTPUtil::SendRequest(
LPCWSTR url, //[IN] アクセスするURL
NW_VERB_MODE verbMode, //[IN] VERBの種類
LPCWSTR addHttpHeader, //[IN] Httpヘッダに追加するものあるなら指定
LPCWSTR saveFilePath, //[IN] DLファイル名、NULL時は内部メモリバッファにDL
UPLOAD_DATA_LIST* upList //[IN] サーバーに送信するデータ(PUT or POST)
)
{
if( url == NULL ){
return ERR_INVALID_ARG;
}
if( this->session == NULL ){
return ERR_NW_NO_SESSION;
}
//リクエストクローズ
if( this->request != NULL ){
WinHttpSetStatusCallback( this->request, NULL, NULL, NULL );
WinHttpCloseHandle(this->request);
this->request = NULL;
}
ClearUpList();
for( size_t i=0; i<this->dlBuffList.size(); i++ ){
SAFE_DELETE(this->dlBuffList[i]);
}
this->dlBuffList.clear();
//URLの分解
URL_COMPONENTS stURL;
ZeroMemory(&stURL, sizeof(URL_COMPONENTS));
stURL.dwStructSize = sizeof(URL_COMPONENTS);
stURL.dwSchemeLength = -1;
stURL.dwHostNameLength = -1;
stURL.dwUrlPathLength = -1;
stURL.dwExtraInfoLength = -1;
if( WinHttpCrackUrl( url, (DWORD)wcslen(url), 0, &stURL) == NULL ){
return ERR_FALSE;
}
if( stURL.dwHostNameLength <= 0 ){
return ERR_FALSE;
}
wstring host = L"";
host.insert(0,stURL.lpszHostName, stURL.dwHostNameLength );
host += L"\0";
if( CompareNoCase(host, stURL.lpszHostName) != 0 || this->lastPort != stURL.nPort){
//前回と違うコネクションなのでやりなおし
if( this->connect != NULL ){
WinHttpCloseHandle(this->connect);
this->connect = NULL;
}
}
//コネクションオープン
if( this->connect == NULL ){
this->connect = WinHttpConnect( this->session, host.c_str(), stURL.nPort, 0 );
if( this->connect == NULL ){
return ERR_NW_OPEN_CONNECT;
}
}
//Verbの設定
wstring verb = L"GET";
if( verbMode == NW_VERB_GET ){
verb = L"GET";
}else if( verbMode == NW_VERB_POST ){
verb = L"POST";
}else if( verbMode == NW_VERB_PUT ){
verb = L"PUT";
}
wstring sendUrl = L"";
sendUrl.insert(0, stURL.lpszUrlPath, stURL.dwUrlPathLength+stURL.dwExtraInfoLength);
sendUrl += L"\0";
//リクエストオープン
if( stURL.nPort == INTERNET_DEFAULT_HTTPS_PORT ){
this->request = WinHttpOpenRequest( this->connect, verb.c_str(), sendUrl.c_str(),
NULL, WINHTTP_NO_REFERER,
WINHTTP_DEFAULT_ACCEPT_TYPES,
WINHTTP_FLAG_SECURE);
}else{
this->request = WinHttpOpenRequest( this->connect, verb.c_str(), sendUrl.c_str(),
NULL, WINHTTP_NO_REFERER,
WINHTTP_DEFAULT_ACCEPT_TYPES,
0 );
}
if( this->request == NULL ){
return ERR_NW_OPEN_REQUEST;
}
if( this->useProxy == TRUE ){
//ProxyのIDとパスワードがあったらセット
if( this->proxyID.size() > 0 || this->proxyPWD.size() > 0 ){
if( WinHttpSetCredentials( this->request,
WINHTTP_AUTH_TARGET_PROXY,
WINHTTP_AUTH_SCHEME_BASIC,
this->proxyID.c_str(),
this->proxyPWD.c_str(),
NULL ) == FALSE ){
return ERR_NW_PROXY_LOGIN;
}
}
}
//HTTPヘッダを設定
if( addHttpHeader != NULL ){
WinHttpAddRequestHeaders( this->request, addHttpHeader, -1, WINHTTP_ADDREQ_FLAG_ADD );
}
//コールバック設定
WinHttpSetStatusCallback(this->request, StatusCallback, WINHTTP_CALLBACK_FLAG_ALL_COMPLETIONS, NULL );
//アップロードデータのコピー
this->totalUpSize = 0;
if( upList != NULL ){
for( DWORD i=0; i<upList->listCount; i++ ){
UPLOAD_DATA* item = new UPLOAD_DATA;
item->filePathFlag = upList->list[i].filePathFlag;
item->buffSize = upList->list[i].buffSize;
if( item->buffSize > 0 ){
item->buff = new BYTE[item->buffSize];
memcpy(item->buff, upList->list[i].buff, item->buffSize);
}
upDataList.push_back(item);
if( item->filePathFlag == TRUE ){
HANDLE file = CreateFileW( (WCHAR*)item->buff, GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL );
if( file == INVALID_HANDLE_VALUE ){
//ファイルにアクセスできないのでエラー
return ERR_NW_OPEN_FILE;
}
DWORD HSize=0;
DWORD LSize=0;
LSize = GetFileSize(file, &HSize);
CloseHandle(file);
if( HSize > 0 ){
//DWORD以上は未サポート
return ERR_SIZE;
}
this->totalUpSize += LSize;
}else{
this->totalUpSize += item->buffSize;
}
}
}
this->lastHost = host;
this->lastPort = stURL.nPort;
if( saveFilePath != NULL ){
this->saveFilePath = saveFilePath;
}else{
this->saveFilePath = L"";
}
ResetEvent(this->upStopEvent);
ResetEvent(this->responseCompEvent);
this->errEndCode = NO_ERR;
if( WinHttpSendRequest( this->request, WINHTTP_NO_ADDITIONAL_HEADERS, 0, WINHTTP_NO_REQUEST_DATA, 0, totalUpSize, (DWORD_PTR)this) == FALSE ){
return ERR_NW_SEND_REQUEST;
}
if( this->asyncMode == FALSE ){
//同期なので完了するまで待つ
WaitForSingleObject(this->responseCompEvent, INFINITE);
return this->errEndCode;
}
return NO_ERR;
}
bool HttpQuery(const HttpRequest& request, HttpResponse& response) {
// initialize
response.statusCode = -1;
HINTERNET internet = NULL;
HINTERNET connectedInternet = NULL;
HINTERNET requestInternet = NULL;
BOOL httpResult = FALSE;
DWORD error = 0;
std::deque<LPCWSTR> acceptTypes;
std::deque<BufferPair> availableBuffers;
// access http
internet = WinHttpOpen(L"Raven", WINHTTP_ACCESS_TYPE_NO_PROXY, NULL, NULL, 0);
error = GetLastError();
if (!internet) goto CLEANUP;
// connect
connectedInternet = WinHttpConnect(internet, request.server.c_str(), (int)request.port, 0);
error = GetLastError();
if (!connectedInternet) goto CLEANUP;
// open request
for (int i = 0; i<(int)request.acceptTypes.size(); i++) {
acceptTypes.push_front(request.acceptTypes[i].c_str());
}
acceptTypes.push_front(0);
requestInternet = WinHttpOpenRequest(connectedInternet, request.method.c_str(), request.query.c_str(), NULL, WINHTTP_NO_REFERER, &acceptTypes[0], (request.secure ? WINHTTP_FLAG_SECURE : 0));
error = GetLastError();
if (!requestInternet) goto CLEANUP;
// authentication, cookie and request
if (request.username != L"" && request.password != L"") {
WinHttpSetCredentials(requestInternet, WINHTTP_AUTH_TARGET_SERVER, WINHTTP_AUTH_SCHEME_BASIC, request.username.c_str(), request.password.c_str(), NULL);
}
if (request.contentType != L"") {
httpResult = WinHttpAddRequestHeaders(requestInternet, (L"Content-type:" + request.contentType).c_str(), -1, WINHTTP_ADDREQ_FLAG_REPLACE | WINHTTP_ADDREQ_FLAG_ADD);
}
if (request.cookie != L"") {
WinHttpAddRequestHeaders(requestInternet, (L"Cookie:" + request.cookie).c_str(), -1, WINHTTP_ADDREQ_FLAG_REPLACE | WINHTTP_ADDREQ_FLAG_ADD);
}
// extra headers
for (auto it = request.extraHeaders.begin(); it != request.extraHeaders.end(); it++) {
std::wstring key = it->first;
std::wstring value = it->second;
WinHttpAddRequestHeaders(requestInternet, (key + L":" + value).c_str(), -1, WINHTTP_ADDREQ_FLAG_REPLACE | WINHTTP_ADDREQ_FLAG_ADD);
}
if (request.body.size()>0) {
httpResult = WinHttpSendRequest(requestInternet, WINHTTP_NO_ADDITIONAL_HEADERS, 0, (LPVOID)&request.body[0], (int)request.body.size(), (int)request.body.size(), NULL);
}
else {
httpResult = WinHttpSendRequest(requestInternet, WINHTTP_NO_ADDITIONAL_HEADERS, 0, WINHTTP_NO_REQUEST_DATA, 0, 0, NULL);
}
error = GetLastError();
if (httpResult == FALSE) goto CLEANUP;
// receive response
httpResult = WinHttpReceiveResponse(requestInternet, NULL);
error = GetLastError();
if (httpResult != TRUE) goto CLEANUP;
DWORD headerLength = sizeof(DWORD);
// read response status code
DWORD statusCode = 0;
httpResult = WinHttpQueryHeaders(requestInternet, WINHTTP_QUERY_STATUS_CODE | WINHTTP_QUERY_FLAG_NUMBER, WINHTTP_HEADER_NAME_BY_INDEX, &statusCode, &headerLength, WINHTTP_NO_HEADER_INDEX);
error = GetLastError();
if (httpResult == FALSE) goto CLEANUP;
response.statusCode = statusCode;
// read respons cookie
httpResult = WinHttpQueryHeaders(requestInternet, WINHTTP_QUERY_RAW_HEADERS_CRLF, WINHTTP_HEADER_NAME_BY_INDEX, NULL, &headerLength, WINHTTP_NO_HEADER_INDEX);
error = GetLastError();
if (error == ERROR_INSUFFICIENT_BUFFER) {
wchar_t* rawHeader = new wchar_t[headerLength / sizeof(wchar_t)];
ZeroMemory(rawHeader, headerLength);
httpResult = WinHttpQueryHeaders(requestInternet, WINHTTP_QUERY_RAW_HEADERS_CRLF, WINHTTP_HEADER_NAME_BY_INDEX, rawHeader, &headerLength, WINHTTP_NO_HEADER_INDEX);
const wchar_t* cookieStart = wcsstr(rawHeader, L"Cookie:");
if (cookieStart) {
const wchar_t* cookieEnd = wcsstr(cookieStart, L";");
if (cookieEnd) {
response.cookie = std::wstring(cookieStart + 7, cookieEnd - cookieStart - 7);
}
}
delete[] rawHeader;
}
// read response body
while (true) {
DWORD bytesAvailable = 0;
BOOL queryDataAvailableResult = WinHttpQueryDataAvailable(requestInternet, &bytesAvailable);
error = GetLastError();
if (queryDataAvailableResult == TRUE && bytesAvailable != 0) {
char* utf8 = new char[bytesAvailable];
DWORD bytesRead = 0;
BOOL readDataResult = WinHttpReadData(requestInternet, utf8, bytesAvailable, &bytesRead);
error = GetLastError();
if (readDataResult == TRUE) {
availableBuffers.push_front(BufferPair(utf8, bytesRead));
}
else {
delete[] utf8;
}
}
else {
break;
}
}
// concatincate response body
int totalSize = 0;
for each (BufferPair p in availableBuffers) {
totalSize += p.length;
}
response.body.resize(totalSize);
if (totalSize>0) {
char* utf8 = new char[totalSize];
{
char* temp = utf8;
for each (BufferPair p in availableBuffers) {
memcpy(temp, p.buffer, p.length);
temp += p.length;
}
}
int
__cdecl
wmain(
int argc,
WCHAR **argv
)
{
DWORD dwError = ERROR_SUCCESS;
HINTERNET hSession = NULL;
HINTERNET hConnect = NULL;
HINTERNET hRequest = NULL;
DWORD dwStatusCode = 0;
DWORD dwSize = sizeof(dwStatusCode);
DWORD dwAutoLogonPolicy = WINHTTP_AUTOLOGON_SECURITY_LEVEL_LOW;
PWSTR pwszServerName = NULL;
if (argc != 2)
{
wprintf(L"Usage: %s <servername>\n", argv[0]);
goto Exit;
}
pwszServerName = argv[1];
// Use WinHttpOpen to obtain a session handle and specify no proxy
hSession = WinHttpOpen(USER_AGENT,
WINHTTP_ACCESS_TYPE_NO_PROXY,
WINHTTP_NO_PROXY_NAME,
WINHTTP_NO_PROXY_BYPASS,
0);
if (hSession == NULL)
{
dwError = GetLastError();
wprintf(L"WinHttpOpen failed with error %d\n", dwError);
goto Exit;
}
// Use WinHttpConnect to specify target server and port
hConnect = WinHttpConnect(hSession,
pwszServerName,
INTERNET_DEFAULT_HTTP_PORT,
0);
if (hConnect == NULL)
{
dwError = GetLastError();
wprintf(L"WinHttpConnect failed with error %d\n", dwError);
goto Exit;
}
// Use WinHttpOpenRequest to open a GET request and specify taget path
hRequest = WinHttpOpenRequest(hConnect,
L"GET",
TARGET_PATH,
NULL,
WINHTTP_NO_REFERER,
WINHTTP_DEFAULT_ACCEPT_TYPES,
0);
if (hRequest == NULL)
{
dwError = GetLastError();
wprintf(L"WinHttpOpenRequest failed with error %d\n", dwError);
goto Exit;
}
// Use WinHttpSetOption to set autologon policy to low level
// to include the default credentials in a request
if (!WinHttpSetOption(hRequest,
WINHTTP_OPTION_AUTOLOGON_POLICY,
&dwAutoLogonPolicy,
sizeof(dwAutoLogonPolicy)))
{
dwError = GetLastError();
wprintf(L"WinHttpSetOption failed with error %d\n", dwError);
goto Exit;
}
// Use WinHttpSetCredentials with NULL username and password
// to use default credentials
if (!WinHttpSetCredentials(hRequest,
WINHTTP_AUTH_TARGET_SERVER,
WINHTTP_AUTH_SCHEME_NTLM,
NULL,
NULL,
NULL))
{
dwError = GetLastError();
wprintf(L"WinHttpSetCredentials failed with error %d\n", dwError);
goto Exit;
}
// Use WinHttpSendRequest to send the request and
// specify no additional headers and request data
if (!WinHttpSendRequest(hRequest,
WINHTTP_NO_ADDITIONAL_HEADERS,
0,
WINHTTP_NO_REQUEST_DATA,
0,
0,
0))
{
dwError = GetLastError();
wprintf(L"WinHttpSendRequest failed with error %d\n", dwError);
goto Exit;
}
// Use WinHttpReceiveResponse to receive the response
if (!WinHttpReceiveResponse(hRequest,
NULL))
{
dwError = GetLastError();
wprintf(L"WinHttpReceiveResponse failed with error %d\n", dwError);
goto Exit;
}
// Use WinHttpQueryHeaders to retrieve the status code
if (!WinHttpQueryHeaders(hRequest,
WINHTTP_QUERY_FLAG_NUMBER |
WINHTTP_QUERY_STATUS_CODE,
NULL,
&dwStatusCode,
&dwSize,
NULL))
{
dwError = GetLastError();
wprintf(L"WinHttpQueryHeaders failed with error %d\n", dwError);
goto Exit;
}
// Expect to get status code 200
if(dwStatusCode == 200)
{
wprintf(L"Got expected status code=200\n");
}
else
{
wprintf(L"Unexpected status code=%d\n", dwStatusCode);
}
Exit:
if (hRequest != NULL)
{
WinHttpCloseHandle(hRequest);
}
if (hConnect != NULL)
{
WinHttpCloseHandle(hConnect);
}
if (hSession != NULL)
{
WinHttpCloseHandle(hSession);
}
return dwError;
}
