1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
/* -*- 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 "mozilla/Try.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()) {
MOZ_TRY(
NS_NewLocalFile(Substring(aBuffer, cch), 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_NewUTF8LocalFile(Substring(aBuffer, pos, nextNul - pos),
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
|
