blob: 5b65289393706545e221d7f05b2df9f615b88068 (
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
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
|
// Copyright 2015 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef IPC_ATTACHMENT_BROKER_H_
#define IPC_ATTACHMENT_BROKER_H_
#include "base/gtest_prod_util.h"
#include "base/macros.h"
#include "base/process/process_handle.h"
#include "ipc/brokerable_attachment.h"
#include "ipc/ipc_export.h"
#include "ipc/ipc_listener.h"
// If the platform has no attachments that need brokering, then it shouldn't
// compile any code that calls member functions of AttachmentBroker. This
// prevents symbols only used by AttachmentBroker and its subclasses from
// making it into the binary.
#if defined(OS_WIN)
#define USE_ATTACHMENT_BROKER 1
#else
#define USE_ATTACHMENT_BROKER 0
#endif // defined(OS_WIN)
namespace IPC {
class AttachmentBroker;
// Classes that inherit from this abstract base class are capable of
// communicating with a broker to send and receive attachments to Chrome IPC
// messages.
class IPC_EXPORT SupportsAttachmentBrokering {
public:
// Returns an AttachmentBroker used to broker attachments of IPC messages to
// other processes. There must be exactly one AttachmentBroker per process.
virtual AttachmentBroker* GetAttachmentBroker() = 0;
};
// Responsible for brokering attachments to Chrome IPC messages. On platforms
// that support attachment brokering, every IPC channel should have a reference
// to a AttachmentBroker.
// This class is not thread safe. The implementation of this class assumes that
// it is only ever used on the same thread as its consumers.
class IPC_EXPORT AttachmentBroker : public Listener {
public:
// A standard observer interface that allows consumers of the AttachmentBroker
// to be notified when a new attachment has been received.
class Observer {
public:
virtual void ReceivedBrokerableAttachmentWithId(
const BrokerableAttachment::AttachmentId& id) = 0;
};
AttachmentBroker();
~AttachmentBroker() override;
// Sends |attachment| to |destination_process|. The implementation uses an
// IPC::Channel to communicate with the broker process. This may be the same
// IPC::Channel that is requesting the brokering of an attachment.
// Returns true on success and false otherwise.
virtual bool SendAttachmentToProcess(const BrokerableAttachment* attachment,
base::ProcessId destination_process) = 0;
// Returns whether the attachment was available. If the attachment was
// available, populates the output parameter |attachment|.
bool GetAttachmentWithId(BrokerableAttachment::AttachmentId id,
scoped_refptr<BrokerableAttachment>* attachment);
// Any given observer should only ever add itself once to the observer list.
void AddObserver(Observer* observer);
void RemoveObserver(Observer* observer);
protected:
using AttachmentVector = std::vector<scoped_refptr<BrokerableAttachment>>;
// Adds |attachment| to |attachments_|, and notifies the observers.
void HandleReceivedAttachment(
const scoped_refptr<BrokerableAttachment>& attachment);
// Informs the observers that a new BrokerableAttachment has been received.
void NotifyObservers(const BrokerableAttachment::AttachmentId& id);
// This method is exposed for testing only.
AttachmentVector* get_attachments() { return &attachments_; }
private:
#if defined(OS_WIN)
FRIEND_TEST_ALL_PREFIXES(AttachmentBrokerUnprivilegedWinTest,
ReceiveValidMessage);
FRIEND_TEST_ALL_PREFIXES(AttachmentBrokerUnprivilegedWinTest,
ReceiveInvalidMessage);
#endif // defined(OS_WIN)
// A vector of BrokerableAttachments that have been received, but not yet
// consumed.
// A std::vector is used instead of a std::map because this container is
// expected to have few elements, for which a std::vector is expected to have
// better performance.
AttachmentVector attachments_;
std::vector<Observer*> observers_;
DISALLOW_COPY_AND_ASSIGN(AttachmentBroker);
};
} // namespace IPC
#endif // IPC_ATTACHMENT_BROKER_H_
|
