OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "ipc/ipc_test_sink.h" | 5 #include "ipc/ipc_test_sink.h" |
6 #include "ppapi/proxy/plugin_var_tracker.h" | 6 #include "ppapi/proxy/plugin_var_tracker.h" |
7 #include "ppapi/proxy/ppapi_messages.h" | 7 #include "ppapi/proxy/ppapi_messages.h" |
8 #include "ppapi/proxy/ppapi_proxy_test.h" | 8 #include "ppapi/proxy/ppapi_proxy_test.h" |
9 | 9 |
10 namespace pp { | 10 namespace pp { |
11 namespace proxy { | 11 namespace proxy { |
12 | 12 |
13 namespace { | 13 namespace { |
14 | 14 |
15 PP_Var MakeObject(PluginVarTracker::VarID object_id) { | 15 PP_Var MakeObject(int32 object_id) { |
16 PP_Var ret; | 16 PP_Var ret; |
17 ret.type = PP_VARTYPE_OBJECT; | 17 ret.type = PP_VARTYPE_OBJECT; |
18 ret.value.as_id = object_id; | 18 ret.value.as_id = object_id; |
19 return ret; | 19 return ret; |
20 } | 20 } |
21 | 21 |
22 // Creates a PP_Var from the given string ID. | |
23 PP_Var MakeString(PluginVarTracker::VarID string_id) { | |
24 PP_Var ret; | |
25 ret.type = PP_VARTYPE_STRING; | |
26 ret.value.as_id = string_id; | |
27 return ret; | |
28 } | |
29 | |
30 } // namespace | 22 } // namespace |
31 | 23 |
32 class PluginVarTrackerTest : public PluginProxyTest { | 24 class PluginVarTrackerTest : public PluginProxyTest { |
33 public: | 25 public: |
34 PluginVarTrackerTest() {} | 26 PluginVarTrackerTest() {} |
35 | 27 |
36 protected: | 28 protected: |
37 // Asserts that there is a unique "release object" IPC message in the test | 29 // Asserts that there is a unique "release object" IPC message in the test |
38 // sink. This will return the var ID from the message or -1 if none found. | 30 // sink. This will return the var ID from the message or -1 if none found. |
39 PluginVarTracker::VarID GetObjectIDForUniqueReleaseObject() { | 31 int32 GetObjectIDForUniqueReleaseObject() { |
40 const IPC::Message* release_msg = sink().GetUniqueMessageMatching( | 32 const IPC::Message* release_msg = sink().GetUniqueMessageMatching( |
41 PpapiHostMsg_PPBVar_ReleaseObject::ID); | 33 PpapiHostMsg_PPBVar_ReleaseObject::ID); |
42 if (!release_msg) | 34 if (!release_msg) |
43 return -1; | 35 return -1; |
44 | 36 |
45 Tuple1<int64> id; | 37 Tuple1<int64> id; |
46 PpapiHostMsg_PPBVar_ReleaseObject::Read(release_msg, &id); | 38 PpapiHostMsg_PPBVar_ReleaseObject::Read(release_msg, &id); |
47 return id.a; | 39 return id.a; |
48 } | 40 } |
49 }; | 41 }; |
50 | 42 |
51 TEST_F(PluginVarTrackerTest, Strings) { | |
52 std::string str("Hello"); | |
53 PluginVarTracker::VarID str_id1 = var_tracker().MakeString(str); | |
54 EXPECT_NE(0, str_id1); | |
55 | |
56 PluginVarTracker::VarID str_id2 = var_tracker().MakeString( | |
57 str.c_str(), static_cast<uint32_t>(str.size())); | |
58 EXPECT_NE(0, str_id2); | |
59 | |
60 // Make sure the strings come out the other end. | |
61 const std::string* result = | |
62 var_tracker().GetExistingString(MakeString(str_id1)); | |
63 EXPECT_EQ(str, *result); | |
64 result = var_tracker().GetExistingString(MakeString(str_id2)); | |
65 EXPECT_EQ(str, *result); | |
66 } | |
67 | |
68 TEST_F(PluginVarTrackerTest, GetHostObject) { | 43 TEST_F(PluginVarTrackerTest, GetHostObject) { |
69 PP_Var host_object = MakeObject(12345); | 44 PP_Var host_object = MakeObject(12345); |
70 | 45 |
71 // Round-trip through the tracker to make sure the host object comes out the | 46 // Round-trip through the tracker to make sure the host object comes out the |
72 // other end. | 47 // other end. |
73 PP_Var plugin_object = var_tracker().ReceiveObjectPassRef( | 48 PP_Var plugin_object = var_tracker().ReceiveObjectPassRef( |
74 host_object, plugin_dispatcher()); | 49 host_object, plugin_dispatcher()); |
75 PP_Var host_object2 = var_tracker().GetHostObject(plugin_object); | 50 PP_Var host_object2 = var_tracker().GetHostObject(plugin_object); |
76 EXPECT_EQ(PP_VARTYPE_OBJECT, host_object2.type); | 51 EXPECT_EQ(PP_VARTYPE_OBJECT, host_object2.type); |
77 EXPECT_EQ(host_object.value.as_id, host_object2.value.as_id); | 52 EXPECT_EQ(host_object.value.as_id, host_object2.value.as_id); |
78 | 53 |
79 var_tracker().Release(plugin_object); | 54 var_tracker().ReleaseVar(plugin_object); |
80 } | 55 } |
81 | 56 |
82 TEST_F(PluginVarTrackerTest, ReceiveObjectPassRef) { | 57 TEST_F(PluginVarTrackerTest, ReceiveObjectPassRef) { |
83 PP_Var host_object = MakeObject(12345); | 58 PP_Var host_object = MakeObject(12345); |
84 | 59 |
85 // Receive the object, we should have one ref and no messages. | 60 // Receive the object, we should have one ref and no messages. |
86 PP_Var plugin_object = var_tracker().ReceiveObjectPassRef( | 61 PP_Var plugin_object = var_tracker().ReceiveObjectPassRef( |
87 host_object, plugin_dispatcher()); | 62 host_object, plugin_dispatcher()); |
88 EXPECT_EQ(0u, sink().message_count()); | 63 EXPECT_EQ(0u, sink().message_count()); |
89 EXPECT_EQ(1, var_tracker().GetRefCountForObject(plugin_object)); | 64 EXPECT_EQ(1, var_tracker().GetRefCountForObject(plugin_object)); |
90 EXPECT_EQ(0, | 65 EXPECT_EQ(0, |
91 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_object)); | 66 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_object)); |
92 | 67 |
93 // Receive the same object again, we should get the same plugin ID out. | 68 // Receive the same object again, we should get the same plugin ID out. |
94 PP_Var plugin_object2 = var_tracker().ReceiveObjectPassRef( | 69 PP_Var plugin_object2 = var_tracker().ReceiveObjectPassRef( |
95 host_object, plugin_dispatcher()); | 70 host_object, plugin_dispatcher()); |
96 EXPECT_EQ(plugin_object.value.as_id, plugin_object2.value.as_id); | 71 EXPECT_EQ(plugin_object.value.as_id, plugin_object2.value.as_id); |
97 EXPECT_EQ(2, var_tracker().GetRefCountForObject(plugin_object)); | 72 EXPECT_EQ(2, var_tracker().GetRefCountForObject(plugin_object)); |
98 EXPECT_EQ(0, | 73 EXPECT_EQ(0, |
99 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_object)); | 74 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_object)); |
100 | 75 |
101 // It should have sent one message to decerment the refcount in the host. | 76 // It should have sent one message to decerment the refcount in the host. |
102 // This is because it only maintains one host refcount for all references | 77 // This is because it only maintains one host refcount for all references |
103 // in the plugin, but the host just sent the second one. | 78 // in the plugin, but the host just sent the second one. |
104 EXPECT_EQ(host_object.value.as_id, GetObjectIDForUniqueReleaseObject()); | 79 EXPECT_EQ(host_object.value.as_id, GetObjectIDForUniqueReleaseObject()); |
105 sink().ClearMessages(); | 80 sink().ClearMessages(); |
106 | 81 |
107 // Release the object, one ref at a time. The second release should free | 82 // Release the object, one ref at a time. The second release should free |
108 // the tracking data and send a release message to the browser. | 83 // the tracking data and send a release message to the browser. |
109 var_tracker().Release(plugin_object); | 84 var_tracker().ReleaseVar(plugin_object); |
110 EXPECT_EQ(1, var_tracker().GetRefCountForObject(plugin_object)); | 85 EXPECT_EQ(1, var_tracker().GetRefCountForObject(plugin_object)); |
111 var_tracker().Release(plugin_object); | 86 var_tracker().ReleaseVar(plugin_object); |
112 EXPECT_EQ(-1, var_tracker().GetRefCountForObject(plugin_object)); | 87 EXPECT_EQ(-1, var_tracker().GetRefCountForObject(plugin_object)); |
113 EXPECT_EQ(host_object.value.as_id, GetObjectIDForUniqueReleaseObject()); | 88 EXPECT_EQ(host_object.value.as_id, GetObjectIDForUniqueReleaseObject()); |
114 } | 89 } |
115 | 90 |
116 // Tests freeing objects that have both refcounts and "tracked with no ref". | 91 // Tests freeing objects that have both refcounts and "tracked with no ref". |
117 TEST_F(PluginVarTrackerTest, FreeTrackedAndReferencedObject) { | 92 TEST_F(PluginVarTrackerTest, FreeTrackedAndReferencedObject) { |
118 PP_Var host_object = MakeObject(12345); | 93 PP_Var host_object = MakeObject(12345); |
119 | 94 |
120 // Phase one: First receive via a "pass ref", then a tracked with no ref. | 95 // Phase one: First receive via a "pass ref", then a tracked with no ref. |
121 PP_Var plugin_var = var_tracker().ReceiveObjectPassRef( | 96 PP_Var plugin_var = var_tracker().ReceiveObjectPassRef( |
122 host_object, plugin_dispatcher()); | 97 host_object, plugin_dispatcher()); |
123 PP_Var plugin_var2 = var_tracker().TrackObjectWithNoReference( | 98 PP_Var plugin_var2 = var_tracker().TrackObjectWithNoReference( |
124 host_object, plugin_dispatcher()); | 99 host_object, plugin_dispatcher()); |
125 EXPECT_EQ(plugin_var.value.as_id, plugin_var2.value.as_id); | 100 EXPECT_EQ(plugin_var.value.as_id, plugin_var2.value.as_id); |
126 EXPECT_EQ(1, var_tracker().GetRefCountForObject(plugin_var)); | 101 EXPECT_EQ(1, var_tracker().GetRefCountForObject(plugin_var)); |
127 EXPECT_EQ(1, | 102 EXPECT_EQ(1, |
128 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); | 103 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); |
129 | 104 |
130 // Free via the refcount, this should release the object to the browser but | 105 // Free via the refcount, this should release the object to the browser but |
131 // maintain the tracked object. | 106 // maintain the tracked object. |
132 var_tracker().Release(plugin_var); | 107 var_tracker().ReleaseVar(plugin_var); |
133 EXPECT_EQ(0, var_tracker().GetRefCountForObject(plugin_var)); | 108 EXPECT_EQ(0, var_tracker().GetRefCountForObject(plugin_var)); |
134 EXPECT_EQ(1u, sink().message_count()); | 109 EXPECT_EQ(1u, sink().message_count()); |
135 EXPECT_EQ(host_object.value.as_id, GetObjectIDForUniqueReleaseObject()); | 110 EXPECT_EQ(host_object.value.as_id, GetObjectIDForUniqueReleaseObject()); |
136 | 111 |
137 // Now free via the tracked object, this should free it. | 112 // Now free via the tracked object, this should free it. |
138 var_tracker().StopTrackingObjectWithNoReference(plugin_var); | 113 var_tracker().StopTrackingObjectWithNoReference(plugin_var); |
139 EXPECT_EQ(-1, var_tracker().GetRefCountForObject(plugin_var)); | 114 EXPECT_EQ(-1, var_tracker().GetRefCountForObject(plugin_var)); |
140 | 115 |
141 // Phase two: Receive via a tracked, then get an addref. | 116 // Phase two: Receive via a tracked, then get an addref. |
142 sink().ClearMessages(); | 117 sink().ClearMessages(); |
143 plugin_var = var_tracker().TrackObjectWithNoReference( | 118 plugin_var = var_tracker().TrackObjectWithNoReference( |
144 host_object, plugin_dispatcher()); | 119 host_object, plugin_dispatcher()); |
145 plugin_var2 = var_tracker().ReceiveObjectPassRef( | 120 plugin_var2 = var_tracker().ReceiveObjectPassRef( |
146 host_object, plugin_dispatcher()); | 121 host_object, plugin_dispatcher()); |
147 EXPECT_EQ(plugin_var.value.as_id, plugin_var2.value.as_id); | 122 EXPECT_EQ(plugin_var.value.as_id, plugin_var2.value.as_id); |
148 EXPECT_EQ(1, var_tracker().GetRefCountForObject(plugin_var)); | 123 EXPECT_EQ(1, var_tracker().GetRefCountForObject(plugin_var)); |
149 EXPECT_EQ(1, | 124 EXPECT_EQ(1, |
150 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); | 125 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); |
151 | 126 |
152 // Free via the tracked object, this should have no effect. | 127 // Free via the tracked object, this should have no effect. |
153 var_tracker().StopTrackingObjectWithNoReference(plugin_var); | 128 var_tracker().StopTrackingObjectWithNoReference(plugin_var); |
154 EXPECT_EQ(0, | 129 EXPECT_EQ(0, |
155 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); | 130 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); |
156 EXPECT_EQ(0u, sink().message_count()); | 131 EXPECT_EQ(0u, sink().message_count()); |
157 | 132 |
158 // Now free via the refcount, this should delete it. | 133 // Now free via the refcount, this should delete it. |
159 var_tracker().Release(plugin_var); | 134 var_tracker().ReleaseVar(plugin_var); |
160 EXPECT_EQ(-1, var_tracker().GetRefCountForObject(plugin_var)); | 135 EXPECT_EQ(-1, var_tracker().GetRefCountForObject(plugin_var)); |
161 EXPECT_EQ(host_object.value.as_id, GetObjectIDForUniqueReleaseObject()); | 136 EXPECT_EQ(host_object.value.as_id, GetObjectIDForUniqueReleaseObject()); |
162 } | 137 } |
163 | 138 |
164 TEST_F(PluginVarTrackerTest, RecursiveTrackWithNoRef) { | 139 TEST_F(PluginVarTrackerTest, RecursiveTrackWithNoRef) { |
165 PP_Var host_object = MakeObject(12345); | 140 PP_Var host_object = MakeObject(12345); |
166 | 141 |
167 // Receive a tracked object twice. | 142 // Receive a tracked object twice. |
168 PP_Var plugin_var = var_tracker().TrackObjectWithNoReference( | 143 PP_Var plugin_var = var_tracker().TrackObjectWithNoReference( |
169 host_object, plugin_dispatcher()); | 144 host_object, plugin_dispatcher()); |
(...skipping 10 matching lines...) Expand all Loading... |
180 var_tracker().StopTrackingObjectWithNoReference(plugin_var); | 155 var_tracker().StopTrackingObjectWithNoReference(plugin_var); |
181 EXPECT_EQ(1, | 156 EXPECT_EQ(1, |
182 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); | 157 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); |
183 var_tracker().StopTrackingObjectWithNoReference(plugin_var); | 158 var_tracker().StopTrackingObjectWithNoReference(plugin_var); |
184 EXPECT_EQ(-1, | 159 EXPECT_EQ(-1, |
185 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); | 160 var_tracker().GetTrackedWithNoReferenceCountForObject(plugin_var)); |
186 } | 161 } |
187 | 162 |
188 } // namespace proxy | 163 } // namespace proxy |
189 } // namespace pp | 164 } // namespace pp |
OLD | NEW |