diff -r 000000000000 -r 6474c204b198 widget/windows/LSPAnnotator.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/widget/windows/LSPAnnotator.cpp Wed Dec 31 06:09:35 2014 +0100 @@ -0,0 +1,112 @@ +/* This Source Code Form is subject to the terms of the Mozilla Public + * License, v. 2.0. If a copy of the MPL was not distributed with this + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */ + +/** + * LSPs are evil little bits of code that are allowed to inject into our + * networking stack by Windows. Once they have wormed into our process + * they gnaw at our innards until we crash. Here we force the buggers + * into the light by recording them in our crash information. + * We do the enumeration on a thread because I'm concerned about startup perf + * on machines with several LSPs. + */ + +#include "nsICrashReporter.h" +#include "nsISupportsImpl.h" +#include "nsServiceManagerUtils.h" +#include "nsThreadUtils.h" +#include +#include + +namespace mozilla { +namespace crashreporter { + +class LSPAnnotationGatherer : public nsRunnable +{ +public: + NS_DECL_THREADSAFE_ISUPPORTS + NS_DECL_NSIRUNNABLE + + void Annotate(); + nsCString mString; + nsCOMPtr mThread; +}; + +NS_IMPL_ISUPPORTS(LSPAnnotationGatherer, nsIRunnable) + +void +LSPAnnotationGatherer::Annotate() +{ + nsCOMPtr cr = + do_GetService("@mozilla.org/toolkit/crash-reporter;1"); + bool enabled; + if (cr && NS_SUCCEEDED(cr->GetEnabled(&enabled)) && enabled) { + cr->AnnotateCrashReport(NS_LITERAL_CSTRING("Winsock_LSP"), mString); + } + mThread->Shutdown(); +} + +NS_IMETHODIMP +LSPAnnotationGatherer::Run() +{ + PR_SetCurrentThreadName("LSP Annotator"); + + mThread = NS_GetCurrentThread(); + + DWORD size = 0; + int err; + // Get the size of the buffer we need + if (SOCKET_ERROR != WSCEnumProtocols(nullptr, nullptr, &size, &err) || + err != WSAENOBUFS) { + // Er, what? + NS_NOTREACHED("WSCEnumProtocols suceeded when it should have failed ..."); + return NS_ERROR_FAILURE; + } + + nsAutoArrayPtr byteArray(new char[size]); + WSAPROTOCOL_INFOW* providers = + reinterpret_cast(byteArray.get()); + + int n = WSCEnumProtocols(nullptr, providers, &size, &err); + if (n == SOCKET_ERROR) { + // Lame. We provided the right size buffer; we'll just give up now. + NS_WARNING("Could not get LSP list"); + return NS_ERROR_FAILURE; + } + + nsCString str; + for (int i = 0; i < n; i++) { + AppendUTF16toUTF8(nsDependentString(providers[i].szProtocol), str); + str.AppendLiteral(" : "); + str.AppendInt(providers[i].iVersion); + str.AppendLiteral(" : "); + str.AppendInt(providers[i].iSocketType); + str.AppendLiteral(" : "); + + wchar_t path[MAX_PATH]; + int dummy; + if (!WSCGetProviderPath(&providers[i].ProviderId, path, + &dummy, &err)) { + AppendUTF16toUTF8(nsDependentString(path), str); + } + + if (i + 1 != n) { + str.AppendLiteral(" \n "); + } + } + + mString = str; + NS_DispatchToMainThread(NS_NewRunnableMethod(this, &LSPAnnotationGatherer::Annotate)); + return NS_OK; +} + +void LSPAnnotate() +{ + nsCOMPtr thread; + nsCOMPtr runnable = + do_QueryObject(new LSPAnnotationGatherer()); + NS_NewThread(getter_AddRefs(thread), runnable); +} + +} // namespace crashreporter +} // namespace mozilla