diff options
Diffstat (limited to 'protocols/Weather/src/weather_http.cpp')
| -rw-r--r-- | protocols/Weather/src/weather_http.cpp | 157 |
1 files changed, 157 insertions, 0 deletions
diff --git a/protocols/Weather/src/weather_http.cpp b/protocols/Weather/src/weather_http.cpp new file mode 100644 index 0000000000..319663bfe7 --- /dev/null +++ b/protocols/Weather/src/weather_http.cpp @@ -0,0 +1,157 @@ +/* +Weather Protocol plugin for Miranda IM +Copyright (c) 2012 Miranda NG team +Copyright (c) 2005-2011 Boris Krasnovskiy All Rights Reserved +Copyright (c) 2002-2005 Calvin Che + +This program is free software; you can redistribute it and/or +modify it under the terms of the GNU General Public License +as published by the Free Software Foundation; version 2 +of the License. + +This program is distributed in the hope that it will be useful, +but WITHOUT ANY WARRANTY; without even the implied warranty of +MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +GNU General Public License for more details. + +You should have received a copy of the GNU General Public License +along with this program. If not, see <http://www.gnu.org/licenses/>. +*/ + +/* +This file contain the source related to downloading weather info +from the web using netlib +*/ + +#include "stdafx.h" + +HNETLIBUSER hNetlibUser; + +static int findHeader(const NETLIBHTTPREQUEST *nlhrReply, const char *hdr) +{ + for (int i = 0; i < nlhrReply->headersCount; i++) { + if (_stricmp(nlhrReply->headers[i].szName, hdr) == 0) { + return i; + } + } + return -1; +} + +//============ DOWNLOAD NEW WEATHER ============ +// +// function to download webpage from the internet +// szUrl = URL of the webpage to be retrieved +// return value = 0 for success, 1 or HTTP error code for failure +// global var used: szData, szInfo = containing the retrieved data +// +int InternetDownloadFile(char *szUrl, char *cookie, char *userAgent, wchar_t **szData) +{ + if (userAgent == nullptr || userAgent[0] == 0) + userAgent = NETLIB_USER_AGENT; + + NETLIBHTTPHEADER headers[5]; + headers[0].szName = "User-Agent"; + headers[0].szValue = userAgent; + headers[1].szName = "Cache-Control"; + headers[1].szValue = "no-cache"; + headers[2].szName = "Pragma"; + headers[2].szValue = "no-cache"; + headers[3].szName = "Connection"; + headers[3].szValue = "close"; + headers[4].szName = "Cookie"; + headers[4].szValue = cookie; + + // initialize the netlib request + NETLIBHTTPREQUEST nlhr = { sizeof(nlhr) }; + nlhr.requestType = REQUEST_GET; + nlhr.flags = NLHRF_DUMPASTEXT | NLHRF_HTTP11 | NLHRF_REDIRECT; + nlhr.szUrl = szUrl; + nlhr.headers = headers; + nlhr.headersCount = _countof(headers); + + if (cookie == nullptr || cookie[0] == 0) + --nlhr.headersCount; + + // download the page + NETLIBHTTPREQUEST *nlhrReply = Netlib_HttpTransaction(hNetlibUser, &nlhr); + if (nlhrReply == nullptr) { + // if the data does not downloaded successfully (ie. disconnected), then return 1000 as error code + *szData = (wchar_t*)mir_alloc(512); + // store the error code in szData + mir_wstrcpy(*szData, L"NetLib error occurred!!"); + return NLHRF_REDIRECT; + } + + // if the recieved code is 200 OK + int result; + if (nlhrReply->resultCode == 200) { + if (nlhrReply->dataLength) { + bool bIsUtf = false; + result = 0; + + // allocate memory and save the retrieved data + int i = findHeader(nlhrReply, "Content-Type"); + // look for Content-Type=utf-8 in header + if (i != -1 && strstr(_strlwr(nlhrReply->headers[i].szValue), "utf-8")) + bIsUtf = true; + else { + char *end = nlhrReply->pData; + while (end) { + // look for + // <meta http-equiv="Content-Type" content="utf-8" /> + char* beg = strstr(end, "<meta"); + if (beg) + { + end = strchr(beg, '>'); + if (end) + { + char tmp = *end; + *end = 0; + + char *method = strstr(beg, "http-equiv=\""); + if (method && _strnicmp(method + 12, "Content-Type", 12) == 0 && strstr(method, "utf-8")) { + bIsUtf = true; + *end = tmp; + break; + } + else *end = tmp; + } + } + else + break; + } + } + + wchar_t *retVal = nullptr; + if (bIsUtf) + retVal = mir_utf8decodeW(nlhrReply->pData); + if (retVal == nullptr) + retVal = mir_a2u(nlhrReply->pData); + *szData = retVal; + } + else result = DATA_EMPTY; + } + // return error code if the recieved code is neither 200 OK nor 302 Moved + else { + // store the error code in szData + CMStringW wszError(FORMAT, L"Error occured! HTTP Error: %i\n", nlhrReply->resultCode); + *szData = wszError.Detach(); + result = nlhrReply->resultCode; + } + + // make a copy of the retrieved data, then free the memory of the http reply + Netlib_FreeHttpRequest(nlhrReply); + return result; +} + +//============ NETLIB INITIALIZATION ============ +// +// initialize netlib support for weather protocol +void NetlibInit(void) +{ + NETLIBUSER nlu = {}; + nlu.flags = NUF_OUTGOING | NUF_HTTPCONNS | NUF_NOHTTPSOPTION | NUF_UNICODE; + nlu.szSettingsModule = MODULENAME; + nlu.szDescriptiveName.w = TranslateT("Weather HTTP connections"); + hNetlibUser = Netlib_RegisterUser(&nlu); +} |