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
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/apps/app_shim/mach_bootstrap_acceptor.h"
#include <bsm/libbsm.h>
#include "base/process/process_handle.h"
#include "base/run_loop.h"
#include "base/test/task_environment.h"
#include "chrome/app_shim/app_shim_controller.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace apps {
class MachBootstrapAcceptorTest : public testing::Test {
public:
// Friend accessors:
static mojo::PlatformChannelEndpoint ConnectToBrowser(
const mojo::NamedPlatformChannel::ServerName& server_name) {
return AppShimController::ConnectToBrowser(server_name);
}
static mach_port_t GetAcceptorPort(MachBootstrapAcceptor* acceptor) {
return acceptor->port();
}
static const mojo::NamedPlatformChannel::ServerName& GetAcceptorServerName(
MachBootstrapAcceptor* acceptor) {
return acceptor->server_name_;
}
private:
base::test::TaskEnvironment task_environment_;
};
class TestMachBootstrapAcceptorDelegate
: public MachBootstrapAcceptor::Delegate {
public:
explicit TestMachBootstrapAcceptorDelegate(base::OnceClosure quit_closure)
: quit_closure_(std::move(quit_closure)) {}
void OnClientConnected(mojo::PlatformChannelEndpoint endpoint,
audit_token_t audit_token) override {
endpoint_ = std::move(endpoint);
pid_ = audit_token_to_pid(audit_token);
std::move(quit_closure_).Run();
}
void OnServerChannelCreateError() override {
error_ = true;
std::move(quit_closure_).Run();
}
base::ProcessId pid() const { return pid_; }
const mojo::PlatformChannelEndpoint& endpoint() const { return endpoint_; }
bool error() const { return error_; }
private:
base::ProcessId pid_ = base::kNullProcessId;
mojo::PlatformChannelEndpoint endpoint_;
bool error_ = false;
base::OnceClosure quit_closure_;
};
TEST_F(MachBootstrapAcceptorTest, SingleRequest) {
base::RunLoop run_loop;
TestMachBootstrapAcceptorDelegate delegate(run_loop.QuitClosure());
MachBootstrapAcceptor acceptor("simplereq", &delegate);
acceptor.Start();
EXPECT_TRUE(GetAcceptorPort(&acceptor) != MACH_PORT_NULL);
mojo::PlatformChannelEndpoint endpoint =
ConnectToBrowser(GetAcceptorServerName(&acceptor));
run_loop.Run();
EXPECT_FALSE(delegate.error());
EXPECT_EQ(base::GetCurrentProcId(), delegate.pid());
// In the same process, the send and receive rights are known by the same
// Mach port name.
EXPECT_EQ(endpoint.platform_handle().GetMachReceiveRight().get(),
delegate.endpoint().platform_handle().GetMachSendRight().get());
}
TEST_F(MachBootstrapAcceptorTest, FailToRegister) {
base::RunLoop run_loop;
TestMachBootstrapAcceptorDelegate delegate(run_loop.QuitClosure());
MachBootstrapAcceptor acceptor("failtoreg", &delegate);
mojo::NamedPlatformChannel::ServerName server_name =
GetAcceptorServerName(&acceptor);
// Squat on the server name in the bootstrap server.
mojo::NamedPlatformChannel::Options server_options;
server_options.server_name = server_name;
mojo::NamedPlatformChannel server_endpoint =
mojo::NamedPlatformChannel(server_options);
// The acceptor will fail to start and reports an error.
acceptor.Start();
run_loop.Run();
EXPECT_TRUE(delegate.error());
}
} // namespace apps
|
