InputChannel_test.cpp revision 5c225b1680e696ae8bbf505a1997d6f720672f74 
1//
2// Copyright 2010 The Android Open Source Project
3//
4
5#include <ui/InputTransport.h>
6#include <utils/Timers.h>
7#include <utils/StopWatch.h>
8#include <gtest/gtest.h>
9#include <unistd.h>
10#include <time.h>
11#include <sys/mman.h>
12#include <cutils/ashmem.h>
13
14#include "../../utils/tests/TestHelpers.h"
15
16namespace android {
17
18class InputChannelTest : public testing::Test {
19protected:
20    virtual void SetUp() { }
21    virtual void TearDown() { }
22};
23
24
25TEST_F(InputChannelTest, ConstructorAndDestructor_TakesOwnershipOfFileDescriptors) {
26    // Our purpose here is to verify that the input channel destructor closes the
27    // file descriptors provided to it.  One easy way is to provide it with one end
28    // of a pipe and to check for EPIPE on the other end after the channel is destroyed.
29    Pipe fakeAshmem, sendPipe, receivePipe;
30
31    sp<InputChannel> inputChannel = new InputChannel(String8("channel name"),
32            fakeAshmem.sendFd, receivePipe.receiveFd, sendPipe.sendFd);
33
34    EXPECT_STREQ("channel name", inputChannel->getName().string())
35            << "channel should have provided name";
36    EXPECT_EQ(fakeAshmem.sendFd, inputChannel->getAshmemFd())
37            << "channel should have provided ashmem fd";
38    EXPECT_EQ(receivePipe.receiveFd, inputChannel->getReceivePipeFd())
39            << "channel should have provided receive pipe fd";
40    EXPECT_EQ(sendPipe.sendFd, inputChannel->getSendPipeFd())
41            << "channel should have provided send pipe fd";
42
43    inputChannel.clear(); // destroys input channel
44
45    EXPECT_EQ(-EPIPE, fakeAshmem.readSignal())
46            << "channel should have closed ashmem fd when destroyed";
47    EXPECT_EQ(-EPIPE, receivePipe.writeSignal())
48            << "channel should have closed receive pipe fd when destroyed";
49    EXPECT_EQ(-EPIPE, sendPipe.readSignal())
50            << "channel should have closed send pipe fd when destroyed";
51
52    // clean up fds of Pipe endpoints that were closed so we don't try to close them again
53    fakeAshmem.sendFd = -1;
54    receivePipe.receiveFd = -1;
55    sendPipe.sendFd = -1;
56}
57
58TEST_F(InputChannelTest, OpenInputChannelPair_ReturnsAPairOfConnectedChannels) {
59    sp<InputChannel> serverChannel, clientChannel;
60
61    status_t result = InputChannel::openInputChannelPair(String8("channel name"),
62            serverChannel, clientChannel);
63
64    ASSERT_EQ(OK, result)
65            << "should have successfully opened a channel pair";
66
67    // Name
68    EXPECT_STREQ("channel name (server)", serverChannel->getName().string())
69            << "server channel should have suffixed name";
70    EXPECT_STREQ("channel name (client)", clientChannel->getName().string())
71            << "client channel should have suffixed name";
72
73    // Ashmem uniqueness
74    EXPECT_NE(serverChannel->getAshmemFd(), clientChannel->getAshmemFd())
75            << "server and client channel should have different ashmem fds because it was dup'd";
76
77    // Ashmem usability
78    ssize_t serverAshmemSize = ashmem_get_size_region(serverChannel->getAshmemFd());
79    ssize_t clientAshmemSize = ashmem_get_size_region(clientChannel->getAshmemFd());
80    uint32_t* serverAshmem = static_cast<uint32_t*>(mmap(NULL, serverAshmemSize,
81            PROT_READ | PROT_WRITE, MAP_SHARED, serverChannel->getAshmemFd(), 0));
82    uint32_t* clientAshmem = static_cast<uint32_t*>(mmap(NULL, clientAshmemSize,
83            PROT_READ | PROT_WRITE, MAP_SHARED, clientChannel->getAshmemFd(), 0));
84    ASSERT_TRUE(serverAshmem != NULL)
85            << "server channel ashmem should be mappable";
86    ASSERT_TRUE(clientAshmem != NULL)
87            << "client channel ashmem should be mappable";
88    *serverAshmem = 0xf00dd00d;
89    EXPECT_EQ(0xf00dd00d, *clientAshmem)
90            << "ashmem buffer should be shared by client and server";
91    munmap(serverAshmem, serverAshmemSize);
92    munmap(clientAshmem, clientAshmemSize);
93
94    // Server->Client communication
95    EXPECT_EQ(OK, serverChannel->sendSignal('S'))
96            << "server channel should be able to send signal to client channel";
97    char signal;
98    EXPECT_EQ(OK, clientChannel->receiveSignal(& signal))
99            << "client channel should be able to receive signal from server channel";
100    EXPECT_EQ('S', signal)
101            << "client channel should receive the correct signal from server channel";
102
103    // Client->Server communication
104    EXPECT_EQ(OK, clientChannel->sendSignal('c'))
105            << "client channel should be able to send signal to server channel";
106    EXPECT_EQ(OK, serverChannel->receiveSignal(& signal))
107            << "server channel should be able to receive signal from client channel";
108    EXPECT_EQ('c', signal)
109            << "server channel should receive the correct signal from client channel";
110}
111
112TEST_F(InputChannelTest, ReceiveSignal_WhenNoSignalPresent_ReturnsAnError) {
113    sp<InputChannel> serverChannel, clientChannel;
114
115    status_t result = InputChannel::openInputChannelPair(String8("channel name"),
116            serverChannel, clientChannel);
117
118    ASSERT_EQ(OK, result)
119            << "should have successfully opened a channel pair";
120
121    char signal;
122    EXPECT_EQ(WOULD_BLOCK, clientChannel->receiveSignal(& signal))
123            << "receiveSignal should have returned WOULD_BLOCK";
124}
125
126TEST_F(InputChannelTest, ReceiveSignal_WhenPeerClosed_ReturnsAnError) {
127    sp<InputChannel> serverChannel, clientChannel;
128
129    status_t result = InputChannel::openInputChannelPair(String8("channel name"),
130            serverChannel, clientChannel);
131
132    ASSERT_EQ(OK, result)
133            << "should have successfully opened a channel pair";
134
135    serverChannel.clear(); // close server channel
136
137    char signal;
138    EXPECT_EQ(DEAD_OBJECT, clientChannel->receiveSignal(& signal))
139            << "receiveSignal should have returned DEAD_OBJECT";
140}
141
142TEST_F(InputChannelTest, SendSignal_WhenPeerClosed_ReturnsAnError) {
143    sp<InputChannel> serverChannel, clientChannel;
144
145    status_t result = InputChannel::openInputChannelPair(String8("channel name"),
146            serverChannel, clientChannel);
147
148    ASSERT_EQ(OK, result)
149            << "should have successfully opened a channel pair";
150
151    serverChannel.clear(); // close server channel
152
153    EXPECT_EQ(DEAD_OBJECT, clientChannel->sendSignal('S'))
154            << "sendSignal should have returned DEAD_OBJECT";
155}
156
157
158} // namespace android
159