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
|
/* -*- 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;
WinRemoteMessageSender::WinRemoteMessageSender(const char* aCommandLine)
: mData({static_cast<DWORD>(WinRemoteMessageVersion::CommandLineOnly)}),
mUtf8Buffer(aCommandLine) {
mUtf8Buffer.Append('\0');
char* mutableBuffer;
mData.cbData = mUtf8Buffer.GetMutableData(&mutableBuffer);
mData.lpData = mutableBuffer;
}
WinRemoteMessageSender::WinRemoteMessageSender(const char* aCommandLine,
const char* aWorkingDir)
: mData({static_cast<DWORD>(
WinRemoteMessageVersion::CommandLineAndWorkingDir)}),
mUtf8Buffer(aCommandLine) {
mUtf8Buffer.Append('\0');
mUtf8Buffer.Append(aWorkingDir);
mUtf8Buffer.Append('\0');
char* mutableBuffer;
mData.cbData = mUtf8Buffer.GetMutableData(&mutableBuffer);
mData.lpData = mutableBuffer;
}
WinRemoteMessageSender::WinRemoteMessageSender(const wchar_t* aCommandLine,
const wchar_t* aWorkingDir)
: mData({static_cast<DWORD>(
WinRemoteMessageVersion::CommandLineAndWorkingDirInUtf16)}),
mUtf16Buffer(aCommandLine) {
mUtf16Buffer.Append(u'\0');
mUtf16Buffer.Append(aWorkingDir);
mUtf16Buffer.Append(u'\0');
char16_t* mutableBuffer;
mData.cbData = mUtf16Buffer.GetMutableData(&mutableBuffer) * sizeof(char16_t);
mData.lpData = mutableBuffer;
}
COPYDATASTRUCT* WinRemoteMessageSender::CopyData() { return &mData; }
nsresult WinRemoteMessageReceiver::ParseV0(const nsACString& aBuffer) {
CommandLineParserWin<char> parser;
parser.HandleCommandLine(aBuffer);
mCommandLine = new nsCommandLine();
return mCommandLine->Init(parser.Argc(), parser.Argv(), nullptr,
nsICommandLine::STATE_REMOTE_AUTO);
}
nsresult WinRemoteMessageReceiver::ParseV1(const nsACString& aBuffer) {
CommandLineParserWin<char> parser;
size_t cch = parser.HandleCommandLine(aBuffer);
++cch; // skip a null char
nsCOMPtr<nsIFile> workingDir;
if (cch < aBuffer.Length()) {
NS_NewLocalFile(NS_ConvertUTF8toUTF16(Substring(aBuffer, cch)), false,
getter_AddRefs(workingDir));
}
mCommandLine = new nsCommandLine();
return mCommandLine->Init(parser.Argc(), parser.Argv(), workingDir,
nsICommandLine::STATE_REMOTE_AUTO);
}
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::Parse(const COPYDATASTRUCT* aMessageData) {
switch (static_cast<WinRemoteMessageVersion>(aMessageData->dwData)) {
case WinRemoteMessageVersion::CommandLineOnly:
return ParseV0(nsDependentCSubstring(
reinterpret_cast<char*>(aMessageData->lpData), aMessageData->cbData));
case WinRemoteMessageVersion::CommandLineAndWorkingDir:
return ParseV1(nsDependentCSubstring(
reinterpret_cast<char*>(aMessageData->lpData), aMessageData->cbData));
case WinRemoteMessageVersion::CommandLineAndWorkingDirInUtf16:
return ParseV2(nsDependentSubstring(
reinterpret_cast<char16_t*>(aMessageData->lpData),
aMessageData->cbData / sizeof(char16_t)));
default:
MOZ_ASSERT_UNREACHABLE("Unsupported message version");
return NS_ERROR_FAILURE;
}
}
nsICommandLineRunner* WinRemoteMessageReceiver::CommandLineRunner() {
return mCommandLine;
}
|
