92efa10eef
Currently the windows client sends the current process command line which is potentially an issue since some of those arguments may have been consumed already. It will also not support sending custom command lines after startup. This also switches to sending the UTF-8 command line we already have since that matches other platforms better and makes some arguments const. Differential Revision: https://phabricator.services.mozilla.com/D216031
131 lines
4.1 KiB
C++
131 lines
4.1 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* 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/. */
|
|
|
|
#include "nsCommandLine.h"
|
|
#include "mozilla/CmdLineAndEnvUtils.h"
|
|
#include "WinRemoteMessage.h"
|
|
|
|
using namespace mozilla;
|
|
|
|
#define MOZ_MAGIC_COPYDATA_PREFIX "🔥🦊"
|
|
|
|
WinRemoteMessageSender::WinRemoteMessageSender(int32_t aArgc,
|
|
const char** aArgv,
|
|
const nsAString& aWorkingDir)
|
|
: mData({static_cast<DWORD>(
|
|
WinRemoteMessageVersion::NullSeparatedArguments)}) {
|
|
mCmdLineBuffer.AppendLiteral(MOZ_MAGIC_COPYDATA_PREFIX "\0");
|
|
AppendUTF16toUTF8(aWorkingDir, mCmdLineBuffer);
|
|
mCmdLineBuffer.Append('\0');
|
|
|
|
for (int32_t i = 0; i < aArgc; i++) {
|
|
mCmdLineBuffer.Append(aArgv[i]);
|
|
mCmdLineBuffer.Append('\0');
|
|
}
|
|
|
|
char* mutableBuffer;
|
|
mData.cbData = mCmdLineBuffer.GetMutableData(&mutableBuffer);
|
|
mData.lpData = mutableBuffer;
|
|
}
|
|
|
|
COPYDATASTRUCT* WinRemoteMessageSender::CopyData() { return &mData; }
|
|
|
|
nsresult WinRemoteMessageReceiver::ParseV2(const nsAString& aBuffer) {
|
|
CommandLineParserWin<char16_t> parser;
|
|
size_t cch = parser.HandleCommandLine(aBuffer);
|
|
++cch; // skip a null char
|
|
|
|
nsCOMPtr<nsIFile> workingDir;
|
|
if (cch < aBuffer.Length()) {
|
|
NS_NewLocalFile(Substring(aBuffer, cch), false, getter_AddRefs(workingDir));
|
|
}
|
|
|
|
int argc = parser.Argc();
|
|
Vector<nsAutoCString> utf8args;
|
|
if (!utf8args.reserve(argc)) {
|
|
return NS_ERROR_OUT_OF_MEMORY;
|
|
}
|
|
|
|
UniquePtr<const char*[]> argv(new const char*[argc]);
|
|
for (int i = 0; i < argc; ++i) {
|
|
utf8args.infallibleAppend(NS_ConvertUTF16toUTF8(parser.Argv()[i]));
|
|
argv[i] = utf8args[i].get();
|
|
}
|
|
|
|
mCommandLine = new nsCommandLine();
|
|
return mCommandLine->Init(argc, argv.get(), workingDir,
|
|
nsICommandLine::STATE_REMOTE_AUTO);
|
|
}
|
|
|
|
nsresult WinRemoteMessageReceiver::ParseV3(const nsACString& aBuffer) {
|
|
nsCOMPtr<nsIFile> workingDir;
|
|
|
|
// String should start with the magic sequence followed by null
|
|
int32_t nextNul = aBuffer.FindChar('\0');
|
|
if (nextNul < 0) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
if (!Substring(aBuffer, 0, nextNul).Equals(MOZ_MAGIC_COPYDATA_PREFIX)) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
int32_t pos = nextNul + 1;
|
|
nextNul = aBuffer.FindChar('\0', pos);
|
|
if (nextNul < 0) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
nsresult rv = NS_NewLocalFile(
|
|
NS_ConvertUTF8toUTF16(Substring(aBuffer, pos, nextNul - pos)), false,
|
|
getter_AddRefs(workingDir));
|
|
NS_ENSURE_SUCCESS(rv, rv);
|
|
|
|
pos = nextNul + 1;
|
|
nsTArray<const char*> argv;
|
|
|
|
while (true) {
|
|
nextNul = aBuffer.FindChar('\0', pos);
|
|
if (nextNul < 0) {
|
|
break;
|
|
}
|
|
|
|
// Because each argument is null terminated we can just add the pointer to
|
|
// the array directly.
|
|
argv.AppendElement(aBuffer.BeginReading() + pos);
|
|
pos = nextNul + 1;
|
|
}
|
|
|
|
// There should always be at least one argument, the path to the binary.
|
|
if (argv.IsEmpty()) {
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
|
|
mCommandLine = new nsCommandLine();
|
|
return mCommandLine->Init(argv.Length(), argv.Elements(), workingDir,
|
|
nsICommandLine::STATE_REMOTE_AUTO);
|
|
}
|
|
|
|
nsresult WinRemoteMessageReceiver::Parse(const COPYDATASTRUCT* aMessageData) {
|
|
switch (static_cast<WinRemoteMessageVersion>(aMessageData->dwData)) {
|
|
case WinRemoteMessageVersion::CommandLineAndWorkingDirInUtf16:
|
|
return ParseV2(
|
|
nsDependentSubstring(reinterpret_cast<wchar_t*>(aMessageData->lpData),
|
|
aMessageData->cbData / sizeof(char16_t)));
|
|
case WinRemoteMessageVersion::NullSeparatedArguments:
|
|
return ParseV3(nsDependentCSubstring(
|
|
reinterpret_cast<char*>(aMessageData->lpData), aMessageData->cbData));
|
|
default:
|
|
MOZ_ASSERT_UNREACHABLE("Unsupported message version");
|
|
return NS_ERROR_FAILURE;
|
|
}
|
|
}
|
|
|
|
nsICommandLineRunner* WinRemoteMessageReceiver::CommandLineRunner() {
|
|
return mCommandLine;
|
|
}
|
|
|
|
#undef MOZ_MAGIC_COPYDATA_PREFIX
|