/* Miranda IM: the free IM client for Microsoft* Windows* Copyright 2000-12 Miranda IM, 2012-13 Miranda NG project, all portions of this codebase are copyrighted to the people listed in contributors.txt. 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; either version 2 of the License, or (at your option) any later version. 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, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ #include "..\..\core\commonheaders.h" #include "netlib.h" static BOOL bModuleInitialized = FALSE; HANDLE hConnectionHeaderMutex, hConnectionOpenMutex; DWORD g_LastConnectionTick; int connectionTimeout; HANDLE hSendEvent = NULL, hRecvEvent = NULL; typedef BOOL (WINAPI *tGetProductInfo)(DWORD, DWORD, DWORD, DWORD, PDWORD); static int CompareNetlibUser(const NetlibUser* p1, const NetlibUser* p2) { return strcmp(p1->user.szSettingsModule, p2->user.szSettingsModule); } LIST netlibUser(5, CompareNetlibUser); CRITICAL_SECTION csNetlibUser; SSL_API si; void NetlibFreeUserSettingsStruct(NETLIBUSERSETTINGS *settings) { mir_free(settings->szIncomingPorts); mir_free(settings->szOutgoingPorts); mir_free(settings->szProxyAuthPassword); mir_free(settings->szProxyAuthUser); mir_free(settings->szProxyServer); } void NetlibInitializeNestedCS(struct NetlibNestedCriticalSection *nlncs) { nlncs->dwOwningThreadId = 0; nlncs->lockCount = 0; nlncs->hMutex = CreateMutex(NULL, FALSE, NULL); } void NetlibDeleteNestedCS(struct NetlibNestedCriticalSection *nlncs) { CloseHandle(nlncs->hMutex); } int NetlibEnterNestedCS(struct NetlibConnection *nlc, int which) { struct NetlibNestedCriticalSection *nlncs; DWORD dwCurrentThreadId = GetCurrentThreadId(); WaitForSingleObject(hConnectionHeaderMutex, INFINITE); if (nlc == NULL || nlc->handleType != NLH_CONNECTION) { ReleaseMutex(hConnectionHeaderMutex); SetLastError(ERROR_INVALID_PARAMETER); return 0; } nlncs = (which == NLNCS_SEND) ? &nlc->ncsSend : &nlc->ncsRecv; if (nlncs->lockCount && nlncs->dwOwningThreadId == dwCurrentThreadId) { nlncs->lockCount++; ReleaseMutex(hConnectionHeaderMutex); return 1; } InterlockedIncrement(&nlc->dontCloseNow); ResetEvent(nlc->hOkToCloseEvent); ReleaseMutex(hConnectionHeaderMutex); WaitForSingleObject(nlncs->hMutex, INFINITE); nlncs->dwOwningThreadId = dwCurrentThreadId; nlncs->lockCount = 1; if (InterlockedDecrement(&nlc->dontCloseNow) == 0) SetEvent(nlc->hOkToCloseEvent); return 1; } void NetlibLeaveNestedCS(struct NetlibNestedCriticalSection *nlncs) { if (--nlncs->lockCount == 0) { nlncs->dwOwningThreadId = 0; ReleaseMutex(nlncs->hMutex); } } static INT_PTR GetNetlibUserSettingInt(const char *szUserModule, const char *szSetting, int defValue) { DBVARIANT dbv; if (db_get(NULL, szUserModule, szSetting, &dbv) && db_get(NULL, "Netlib", szSetting, &dbv)) return defValue; if (dbv.type == DBVT_BYTE) return dbv.bVal; if (dbv.type == DBVT_WORD) return dbv.wVal; return dbv.dVal; } static char *GetNetlibUserSettingString(const char *szUserModule, const char *szSetting, int decode) { DBVARIANT dbv; if (db_get_s(NULL, szUserModule, szSetting, &dbv) && db_get_s(NULL, "Netlib", szSetting, &dbv)) { return NULL; } else { char *szRet; if (decode) CallService(MS_DB_CRYPT_DECODESTRING, strlen(dbv.pszVal) + 1, (LPARAM)dbv.pszVal); szRet = mir_strdup(dbv.pszVal); db_free(&dbv); if (szRet == NULL) SetLastError(ERROR_OUTOFMEMORY); return szRet; } } static INT_PTR NetlibRegisterUser(WPARAM, LPARAM lParam) { NETLIBUSER *nlu = (NETLIBUSER*)lParam; if (nlu == NULL || nlu->cbSize != sizeof(NETLIBUSER) || nlu->szSettingsModule == NULL || ( !(nlu->flags&NUF_NOOPTIONS) && nlu->szDescriptiveName == NULL) || (nlu->flags&NUF_HTTPGATEWAY && (nlu->pfnHttpGatewayInit == NULL))) { SetLastError(ERROR_INVALID_PARAMETER); return 0; } NetlibUser *thisUser = (struct NetlibUser*)mir_calloc(sizeof(struct NetlibUser)); thisUser->handleType = NLH_USER; thisUser->user = *nlu; int idx; { mir_cslock lck(csNetlibUser); idx = netlibUser.getIndex(thisUser); } if (idx != -1) { mir_free(thisUser); SetLastError(ERROR_DUP_NAME); return 0; } if (nlu->szDescriptiveName) { thisUser->user.ptszDescriptiveName = (thisUser->user.flags&NUF_UNICODE ? mir_u2t((WCHAR*)nlu->ptszDescriptiveName) : mir_a2t(nlu->szDescriptiveName)); } if ((thisUser->user.szSettingsModule = mir_strdup(nlu->szSettingsModule)) == NULL || (nlu->szDescriptiveName && thisUser->user.ptszDescriptiveName == NULL) || (nlu->szHttpGatewayUserAgent && (thisUser->user.szHttpGatewayUserAgent = mir_strdup(nlu->szHttpGatewayUserAgent)) == NULL)) { mir_free(thisUser); SetLastError(ERROR_OUTOFMEMORY); return 0; } if (nlu->szHttpGatewayHello) thisUser->user.szHttpGatewayHello = mir_strdup(nlu->szHttpGatewayHello); else thisUser->user.szHttpGatewayHello = NULL; thisUser->settings.cbSize = sizeof(NETLIBUSERSETTINGS); thisUser->settings.useProxy = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLUseProxy", 0); thisUser->settings.proxyType = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLProxyType", PROXYTYPE_SOCKS5); if (thisUser->user.flags&NUF_NOHTTPSOPTION && thisUser->settings.proxyType == PROXYTYPE_HTTPS) thisUser->settings.proxyType = PROXYTYPE_HTTP; if ( !(thisUser->user.flags&(NUF_HTTPCONNS|NUF_HTTPGATEWAY)) && thisUser->settings.proxyType == PROXYTYPE_HTTP) { thisUser->settings.useProxy = 0; thisUser->settings.proxyType = PROXYTYPE_SOCKS5; } thisUser->settings.szProxyServer = GetNetlibUserSettingString(thisUser->user.szSettingsModule, "NLProxyServer", 0); thisUser->settings.wProxyPort = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLProxyPort", 1080); thisUser->settings.useProxyAuth = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLUseProxyAuth", 0); thisUser->settings.szProxyAuthUser = GetNetlibUserSettingString(thisUser->user.szSettingsModule, "NLProxyAuthUser", 0); thisUser->settings.szProxyAuthPassword = GetNetlibUserSettingString(thisUser->user.szSettingsModule, "NLProxyAuthPassword", 1); thisUser->settings.dnsThroughProxy = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLDnsThroughProxy", 1); thisUser->settings.specifyIncomingPorts = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLSpecifyIncomingPorts", 0); thisUser->settings.szIncomingPorts = GetNetlibUserSettingString(thisUser->user.szSettingsModule, "NLIncomingPorts", 0); thisUser->settings.specifyOutgoingPorts = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLSpecifyOutgoingPorts", 0); thisUser->settings.szOutgoingPorts = GetNetlibUserSettingString(thisUser->user.szSettingsModule, "NLOutgoingPorts", 0); thisUser->settings.enableUPnP = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLEnableUPnP", 1); //default to on thisUser->settings.validateSSL = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLValidateSSL", 0); thisUser->toLog = GetNetlibUserSettingInt(thisUser->user.szSettingsModule, "NLlog", 1); mir_cslock lck(csNetlibUser); netlibUser.insert(thisUser); return (INT_PTR)thisUser; } static INT_PTR NetlibGetUserSettings(WPARAM wParam, LPARAM lParam) { NETLIBUSERSETTINGS *nlus = (NETLIBUSERSETTINGS*)lParam; struct NetlibUser *nlu = (struct NetlibUser*)wParam; if (GetNetlibHandleType(nlu) != NLH_USER || nlus == NULL || nlus->cbSize != sizeof(NETLIBUSERSETTINGS)) { SetLastError(ERROR_INVALID_PARAMETER); return 0; } *nlus = nlu->settings; return 1; } static INT_PTR NetlibSetUserSettings(WPARAM wParam, LPARAM lParam) { NETLIBUSERSETTINGS *nlus = (NETLIBUSERSETTINGS*)lParam; struct NetlibUser *nlu = (struct NetlibUser*)wParam; if (GetNetlibHandleType(nlu) != NLH_USER || nlus == NULL || nlus->cbSize != sizeof(NETLIBUSERSETTINGS)) { SetLastError(ERROR_INVALID_PARAMETER); return 0; } NetlibSaveUserSettingsStruct(nlu->user.szSettingsModule, nlus); return 1; } void NetlibDoClose(NetlibConnection *nlc, bool noShutdown) { if (nlc->s == INVALID_SOCKET) return; NetlibLogf(nlc->nlu, "(%p:%u) Connection closed internal", nlc, nlc->s); if (nlc->hSsl) { if ( !noShutdown) si.shutdown(nlc->hSsl); si.sfree(nlc->hSsl); nlc->hSsl = NULL; } closesocket(nlc->s); nlc->s = INVALID_SOCKET; } INT_PTR NetlibCloseHandle(WPARAM wParam, LPARAM) { switch(GetNetlibHandleType(wParam)) { case NLH_USER: { struct NetlibUser *nlu = (struct NetlibUser*)wParam; { mir_cslock lck(csNetlibUser); int i = netlibUser.getIndex(nlu); if (i >= 0) netlibUser.remove(i); } NetlibFreeUserSettingsStruct(&nlu->settings); mir_free(nlu->user.szSettingsModule); mir_free(nlu->user.szDescriptiveName); mir_free(nlu->user.szHttpGatewayHello); mir_free(nlu->user.szHttpGatewayUserAgent); mir_free(nlu->szStickyHeaders); break; } case NLH_CONNECTION: { struct NetlibConnection *nlc = (struct NetlibConnection*)wParam; HANDLE waitHandles[4]; DWORD waitResult; WaitForSingleObject(hConnectionHeaderMutex, INFINITE); if (nlc->usingHttpGateway) { HttpGatewayRemovePacket(nlc, -1); } else { if (nlc->s != INVALID_SOCKET) { NetlibDoClose(nlc); } if (nlc->s2 != INVALID_SOCKET) closesocket(nlc->s2); nlc->s2 = INVALID_SOCKET; } ReleaseMutex(hConnectionHeaderMutex); waitHandles[0] = hConnectionHeaderMutex; waitHandles[1] = nlc->hOkToCloseEvent; waitHandles[2] = nlc->ncsRecv.hMutex; waitHandles[3] = nlc->ncsSend.hMutex; waitResult = WaitForMultipleObjects(SIZEOF(waitHandles), waitHandles, TRUE, INFINITE); if (waitResult= WAIT_OBJECT_0 + SIZEOF(waitHandles)) { ReleaseMutex(hConnectionHeaderMutex); SetLastError(ERROR_INVALID_PARAMETER); //already been closed return 0; } nlc->handleType = 0; mir_free(nlc->nlhpi.szHttpPostUrl); mir_free(nlc->nlhpi.szHttpGetUrl); mir_free(nlc->dataBuffer); mir_free((char*)nlc->nloc.szHost); mir_free(nlc->szNewUrl); mir_free(nlc->szProxyServer); NetlibDeleteNestedCS(&nlc->ncsRecv); NetlibDeleteNestedCS(&nlc->ncsSend); CloseHandle(nlc->hOkToCloseEvent); DeleteCriticalSection(&nlc->csHttpSequenceNums); ReleaseMutex(hConnectionHeaderMutex); NetlibLogf(nlc->nlu, "(%p:%u) Connection closed", nlc, nlc->s); break; } case NLH_BOUNDPORT: return NetlibFreeBoundPort((struct NetlibBoundPort*)wParam); case NLH_PACKETRECVER: { struct NetlibPacketRecver *nlpr = (struct NetlibPacketRecver*)wParam; mir_free(nlpr->packetRecver.buffer); break; } default: SetLastError(ERROR_INVALID_PARAMETER); return 0; } mir_free((void*)wParam); return 1; } static INT_PTR NetlibGetSocket(WPARAM wParam, LPARAM) { SOCKET s; if (wParam == 0) { s = INVALID_SOCKET; SetLastError(ERROR_INVALID_PARAMETER); } else { WaitForSingleObject(hConnectionHeaderMutex, INFINITE); switch (GetNetlibHandleType(wParam)) { case NLH_CONNECTION: s = ((struct NetlibConnection*)wParam)->s; break; case NLH_BOUNDPORT: s = ((struct NetlibBoundPort*)wParam)->s; break; default: s = INVALID_SOCKET; SetLastError(ERROR_INVALID_PARAMETER); break; } ReleaseMutex(hConnectionHeaderMutex); } return s; } INT_PTR NetlibStringToAddressSrv(WPARAM wParam, LPARAM lParam) { return (INT_PTR)!NetlibStringToAddress((char*)wParam, (SOCKADDR_INET_M*)lParam); } INT_PTR NetlibAddressToStringSrv(WPARAM wParam, LPARAM lParam) { if (wParam) { SOCKADDR_INET_M iaddr = {0}; iaddr.Ipv4.sin_family = AF_INET; iaddr.Ipv4.sin_addr.s_addr = htonl((unsigned)lParam); return (INT_PTR)NetlibAddressToString(&iaddr); } else return (INT_PTR)NetlibAddressToString((SOCKADDR_INET_M*)lParam); } INT_PTR NetlibGetConnectionInfoSrv(WPARAM wParam, LPARAM lParam) { NetlibGetConnectionInfo((NetlibConnection*)wParam, (NETLIBCONNINFO*)lParam); return 0; } INT_PTR NetlibGetMyIp(WPARAM wParam, LPARAM) { return (INT_PTR)GetMyIp((unsigned)wParam); } INT_PTR NetlibShutdown(WPARAM wParam, LPARAM) { if (wParam) { WaitForSingleObject(hConnectionHeaderMutex, INFINITE); switch(GetNetlibHandleType(wParam)) { case NLH_CONNECTION: { struct NetlibConnection* nlc = (struct NetlibConnection*)wParam; if (nlc->hSsl) si.shutdown(nlc->hSsl); if (nlc->s != INVALID_SOCKET) shutdown(nlc->s, 2); if (nlc->s2 != INVALID_SOCKET) shutdown(nlc->s2, 2); nlc->termRequested = true; } break; case NLH_BOUNDPORT: { struct NetlibBoundPort* nlb = (struct NetlibBoundPort*)wParam; if (nlb->s != INVALID_SOCKET) shutdown(nlb->s, 2); } break; } ReleaseMutex(hConnectionHeaderMutex); } return 0; } INT_PTR NetlibHttpUrlEncode(WPARAM, LPARAM lParam) { if (lParam == NULL) { SetLastError(ERROR_INVALID_PARAMETER); return NULL; } char *p = mir_urlEncode((LPCSTR)lParam); if (p == NULL) { SetLastError(ERROR_OUTOFMEMORY); return NULL; } return (INT_PTR)p; } void UnloadNetlibModule(void) { if ( !bModuleInitialized) return; if (hConnectionHeaderMutex != NULL) { int i; NetlibUnloadIeProxy(); NetlibSecurityDestroy(); NetlibUPnPDestroy(); NetlibLogShutdown(); DestroyHookableEvent(hRecvEvent); hRecvEvent = NULL; DestroyHookableEvent(hSendEvent); hSendEvent = NULL; for (i = netlibUser.getCount(); i > 0; i--) NetlibCloseHandle((WPARAM)netlibUser[i-1], 0); netlibUser.destroy(); CloseHandle(hConnectionHeaderMutex); if (hConnectionOpenMutex) CloseHandle(hConnectionOpenMutex); DeleteCriticalSection(&csNetlibUser); WSACleanup(); } } int LoadNetlibModule(void) { WSADATA wsadata; bModuleInitialized = TRUE; WSAStartup(MAKEWORD(2, 2), &wsadata); HookEvent(ME_OPT_INITIALISE, NetlibOptInitialise); InitializeCriticalSection(&csNetlibUser); hConnectionHeaderMutex = CreateMutex(NULL, FALSE, NULL); NetlibLogInit(); connectionTimeout = 0; OSVERSIONINFOEX osvi = {0}; osvi.dwOSVersionInfoSize = sizeof(osvi); if (GetVersionEx((LPOSVERSIONINFO)&osvi)) { // Connection limiting was introduced in Windows XP SP2 and later and set to 10 / sec if (osvi.dwMajorVersion == 5 && ((osvi.dwMinorVersion == 1 && osvi.wServicePackMajor >= 2) || osvi.dwMinorVersion > 1)) connectionTimeout = 150; // Connection limiting has limits based on addition Windows Vista pre SP2 else if (osvi.dwMajorVersion == 6 && osvi.wServicePackMajor < 2) { DWORD dwType = 0; tGetProductInfo pGetProductInfo = (tGetProductInfo) GetProcAddress(GetModuleHandleA("kernel32"), "GetProductInfo"); if (pGetProductInfo != NULL) pGetProductInfo(6, 0, 0, 0, &dwType); switch(dwType) { case 0x01: // Vista Ultimate edition have connection limit of 25 / sec - plenty for Miranda case 0x1c: break; case 0x02: // Vista Home Basic edition have connection limit of 2 / sec case 0x05: connectionTimeout = 1000; break; default: // all other editions have connection limit of 10 / sec connectionTimeout = 150; break; } } // Connection limiting is disabled by default and is controlled by registry setting in Windows Vista SP2 and later else if (osvi.dwMajorVersion >= 6) { static const char keyn[] = "SYSTEM\\CurrentControlSet\\Services\\Tcpip\\Parameters"; static const char valn[] = "EnableConnectionRateLimiting"; HKEY hSettings; if (RegOpenKeyExA(HKEY_LOCAL_MACHINE, keyn, 0, KEY_QUERY_VALUE, &hSettings) == ERROR_SUCCESS) { DWORD tValueLen, enabled; tValueLen = sizeof(enabled); if (RegQueryValueExA(hSettings, valn, NULL, NULL, (BYTE*)&enabled, &tValueLen) == ERROR_SUCCESS && enabled) connectionTimeout = 150; // if enabled limit is set to 10 / sec RegCloseKey(hSettings); } } } hConnectionOpenMutex = connectionTimeout ? CreateMutex(NULL, FALSE, NULL) : NULL; g_LastConnectionTick = GetTickCount(); CreateServiceFunction(MS_NETLIB_REGISTERUSER, NetlibRegisterUser); CreateServiceFunction(MS_NETLIB_GETUSERSETTINGS, NetlibGetUserSettings); CreateServiceFunction(MS_NETLIB_SETUSERSETTINGS, NetlibSetUserSettings); CreateServiceFunction(MS_NETLIB_CLOSEHANDLE, NetlibCloseHandle); CreateServiceFunction(MS_NETLIB_BINDPORT, NetlibBindPort); CreateServiceFunction(MS_NETLIB_OPENCONNECTION, NetlibOpenConnection); CreateServiceFunction(MS_NETLIB_SETHTTPPROXYINFO, NetlibHttpGatewaySetInfo); CreateServiceFunction(MS_NETLIB_SETSTICKYHEADERS, NetlibHttpSetSticky); CreateServiceFunction(MS_NETLIB_GETSOCKET, NetlibGetSocket); CreateServiceFunction(MS_NETLIB_URLENCODE, NetlibHttpUrlEncode); CreateServiceFunction(MS_NETLIB_SENDHTTPREQUEST, NetlibHttpSendRequest); CreateServiceFunction(MS_NETLIB_RECVHTTPHEADERS, NetlibHttpRecvHeaders); CreateServiceFunction(MS_NETLIB_FREEHTTPREQUESTSTRUCT, NetlibHttpFreeRequestStruct); CreateServiceFunction(MS_NETLIB_HTTPTRANSACTION, NetlibHttpTransaction); CreateServiceFunction(MS_NETLIB_SEND, NetlibSend); CreateServiceFunction(MS_NETLIB_RECV, NetlibRecv); CreateServiceFunction(MS_NETLIB_SELECT, NetlibSelect); CreateServiceFunction(MS_NETLIB_SELECTEX, NetlibSelectEx); CreateServiceFunction(MS_NETLIB_SHUTDOWN, NetlibShutdown); CreateServiceFunction(MS_NETLIB_CREATEPACKETRECVER, NetlibPacketRecverCreate); CreateServiceFunction(MS_NETLIB_GETMOREPACKETS, NetlibPacketRecverGetMore); CreateServiceFunction(MS_NETLIB_SETPOLLINGTIMEOUT, NetlibHttpSetPollingTimeout); CreateServiceFunction(MS_NETLIB_STARTSSL, NetlibStartSsl); CreateServiceFunction(MS_NETLIB_STARINGTOADDRESS, NetlibStringToAddressSrv); CreateServiceFunction(MS_NETLIB_ADDRESSTOSTRING, NetlibAddressToStringSrv); CreateServiceFunction(MS_NETLIB_GETCONNECTIONINFO, NetlibGetConnectionInfoSrv); CreateServiceFunction(MS_NETLIB_GETMYIP, NetlibGetMyIp); hRecvEvent = CreateHookableEvent(ME_NETLIB_FASTRECV); hSendEvent = CreateHookableEvent(ME_NETLIB_FASTSEND); NetlibUPnPInit(); NetlibSecurityInit(); NetlibLoadIeProxy(); return 0; } void NetlibInitSsl(void) { mir_getSI(&si); }