summaryrefslogtreecommitdiffstats
path: root/toolkit/components/remote/WinRemoteMessage.h
blob: 948c6db310501ab8cee7253646c0033fd44a485c (plain)
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
/* -*- 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/. */

#ifndef __WinRemoteMessage_h__
#define __WinRemoteMessage_h__

#include <windows.h>

#include "nsICommandLineRunner.h"
#include "nsCOMPtr.h"
#include "nsString.h"

// This version defines the format of COPYDATASTRUCT::lpData in a message of
// WM_COPYDATA.
// Always use the latest version for production use because the older versions
// have a bug that a non-ascii character in a utf-8 message cannot be parsed
// correctly (bug 1650637).  We keep the older versions for backward
// compatibility and the testing purpose only.
enum class WinRemoteMessageVersion : uint32_t {
  // "<CommandLine>\0" in utf8
  CommandLineOnly = 0,
  // "<CommandLine>\0<WorkingDir>\0" in utf8
  CommandLineAndWorkingDir,
  // L"<CommandLine>\0<WorkingDir>\0" in utf16
  CommandLineAndWorkingDirInUtf16,
};

class WinRemoteMessageSender final {
  COPYDATASTRUCT mData;
  nsAutoString mUtf16Buffer;
  nsAutoCString mUtf8Buffer;

 public:
  WinRemoteMessageSender(const wchar_t* aCommandLine,
                         const wchar_t* aWorkingDir);

  WinRemoteMessageSender(const WinRemoteMessageSender&) = delete;
  WinRemoteMessageSender(WinRemoteMessageSender&&) = delete;
  WinRemoteMessageSender& operator=(const WinRemoteMessageSender&) = delete;
  WinRemoteMessageSender& operator=(WinRemoteMessageSender&&) = delete;

  COPYDATASTRUCT* CopyData();

  // Constructors for the old formats.  Testing purpose only.
  explicit WinRemoteMessageSender(const char* aCommandLine);
  WinRemoteMessageSender(const char* aCommandLine, const char* aWorkingDir);
};

class WinRemoteMessageReceiver final {
  nsCOMPtr<nsICommandLineRunner> mCommandLine;

  nsresult ParseV0(const nsACString& aBuffer);
  nsresult ParseV1(const nsACString& aBuffer);
  nsresult ParseV2(const nsAString& aBuffer);

 public:
  WinRemoteMessageReceiver() = default;
  WinRemoteMessageReceiver(const WinRemoteMessageReceiver&) = delete;
  WinRemoteMessageReceiver(WinRemoteMessageReceiver&&) = delete;
  WinRemoteMessageReceiver& operator=(const WinRemoteMessageReceiver&) = delete;
  WinRemoteMessageReceiver& operator=(WinRemoteMessageReceiver&&) = delete;

  nsresult Parse(const COPYDATASTRUCT* aMessageData);
  nsICommandLineRunner* CommandLineRunner();
};

#endif  // __WinRemoteMessage_h__