blob: 4ad4b7282951e984cead91600415d2e9333dd17e (
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
|
// Copyright (c) 2012 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 CHROME_SERVICE_SERVICE_IPC_SERVER_H_
#define CHROME_SERVICE_SERVICE_IPC_SERVER_H_
#include <vector>
#include "base/memory/scoped_ptr.h"
#include "base/memory/scoped_vector.h"
#include "ipc/ipc_channel_handle.h"
#include "ipc/ipc_listener.h"
#include "ipc/ipc_sync_channel.h"
#include "ipc/ipc_sender.h"
namespace base {
class HistogramDeltaSerialization;
class WaitableEvent;
} // namespace base
// This class handles IPC commands for the service process.
class ServiceIPCServer : public IPC::Listener, public IPC::Sender {
public:
class MessageHandler {
public:
virtual ~MessageHandler() {}
// Must return true if the message is handled.
virtual bool HandleMessage(const IPC::Message& message) = 0;
};
class Client {
public:
virtual ~Client() {}
// Called when the service process must shut down.
virtual void OnShutdown() = 0;
// Called when a product update is available.
virtual void OnUpdateAvailable() = 0;
// Called when the IPC channel is closed. A return value of true indicates
// that the IPC server should continue listening for new connections.
virtual bool OnIPCClientDisconnect() = 0;
};
ServiceIPCServer(
Client* client,
const scoped_refptr<base::SingleThreadTaskRunner>& io_task_runner,
const IPC::ChannelHandle& handle,
base::WaitableEvent* shutdown_event);
~ServiceIPCServer() override;
bool Init();
// IPC::Sender implementation.
bool Send(IPC::Message* msg) override;
// Registers a MessageHandler with the ServiceIPCServer. When an IPC message
// is received that is not handled by the ServiceIPCServer itself, the
// handlers will be called to handle the message in first-add first-call order
// until it is handled or there are no more handlers.
void AddMessageHandler(scoped_ptr<MessageHandler> handler);
bool is_ipc_client_connected() const { return ipc_client_connected_; }
private:
friend class ServiceIPCServerTest;
friend class MockServiceIPCServer;
// IPC::Listener implementation.
bool OnMessageReceived(const IPC::Message& msg) override;
void OnChannelConnected(int32 peer_pid) override;
void OnChannelError() override;
// IPC message handlers.
void OnGetHistograms();
void OnShutdown();
void OnUpdateAvailable();
// Helper method to create the sync channel.
void CreateChannel();
Client* client_;
scoped_refptr<base::SingleThreadTaskRunner> io_task_runner_;
IPC::ChannelHandle channel_handle_;
scoped_ptr<IPC::SyncChannel> channel_;
base::WaitableEvent* shutdown_event_;
ScopedVector<MessageHandler> message_handlers_;
// Indicates whether an IPC client is currently connected to the channel.
bool ipc_client_connected_;
// Calculates histograms deltas.
scoped_ptr<base::HistogramDeltaSerialization> histogram_delta_serializer_;
DISALLOW_COPY_AND_ASSIGN(ServiceIPCServer);
};
#endif // CHROME_SERVICE_SERVICE_IPC_SERVER_H_
|
