| OLD | NEW |
| 1 // Copyright 2013 The Chromium Authors. All rights reserved. | 1 // Copyright 2013 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_message.h" | 5 #include "ipc/ipc_message.h" |
| 6 #include "ppapi/proxy/nacl_message_scanner.h" | 6 #include "ppapi/proxy/nacl_message_scanner.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 #include "ppapi/proxy/serialized_handle.h" | 9 #include "ppapi/proxy/serialized_handle.h" |
| 10 #include "ppapi/shared_impl/host_resource.h" | 10 #include "ppapi/shared_impl/host_resource.h" |
| 11 | 11 |
| 12 namespace ppapi { | 12 namespace ppapi { |
| 13 namespace proxy { | 13 namespace proxy { |
| 14 | 14 |
| 15 namespace { | 15 namespace { |
| 16 const PP_Resource kInvalidResource = 0; | 16 const PP_Resource kInvalidResource = 0; |
| 17 const PP_Resource kFileSystem = 1; | 17 const PP_Resource kFileSystem = 1; |
| 18 const PP_Resource kFileIO = 2; | 18 const PP_Resource kFileIO = 2; |
| 19 const int64_t kQuotaReservationAmount = 100; | 19 const int64_t kQuotaReservationAmount = 100; |
| 20 } | 20 } |
| 21 | 21 |
| 22 class NaClMessageScannerTest : public PluginProxyTest { | 22 class NaClMessageScannerTest : public PluginProxyTest { |
| 23 public: | 23 public: |
| 24 NaClMessageScannerTest() {} | 24 NaClMessageScannerTest() {} |
| 25 | 25 |
| 26 uint32 FindPendingSyncMessage( | |
| 27 const NaClMessageScanner& scanner, | |
| 28 const IPC::Message& msg) { | |
| 29 int msg_id = IPC::SyncMessage::GetMessageId(msg); | |
| 30 std::map<int, uint32>::const_iterator it = | |
| 31 scanner.pending_sync_msgs_.find(msg_id); | |
| 32 // O can signal that no message was found. | |
| 33 return (it != scanner.pending_sync_msgs_.end()) ? it->second : 0; | |
| 34 } | |
| 35 | |
| 36 NaClMessageScanner::FileSystem* FindFileSystem( | 26 NaClMessageScanner::FileSystem* FindFileSystem( |
| 37 const NaClMessageScanner& scanner, | 27 const NaClMessageScanner& scanner, |
| 38 PP_Resource file_system) { | 28 PP_Resource file_system) { |
| 39 NaClMessageScanner::FileSystemMap::const_iterator it = | 29 NaClMessageScanner::FileSystemMap::const_iterator it = |
| 40 scanner.file_systems_.find(file_system); | 30 scanner.file_systems_.find(file_system); |
| 41 return (it != scanner.file_systems_.end()) ? it->second : NULL; | 31 return (it != scanner.file_systems_.end()) ? it->second : NULL; |
| 42 } | 32 } |
| 43 | 33 |
| 44 NaClMessageScanner::FileIO* FindFileIO( | 34 NaClMessageScanner::FileIO* FindFileIO( |
| 45 const NaClMessageScanner& scanner, | 35 const NaClMessageScanner& scanner, |
| 46 PP_Resource file_io) { | 36 PP_Resource file_io) { |
| 47 NaClMessageScanner::FileIOMap::const_iterator it = | 37 NaClMessageScanner::FileIOMap::const_iterator it = |
| 48 scanner.files_.find(file_io); | 38 scanner.files_.find(file_io); |
| 49 return (it != scanner.files_.end()) ? it->second : NULL; | 39 return (it != scanner.files_.end()) ? it->second : NULL; |
| 50 } | 40 } |
| 51 | 41 |
| 52 void OpenQuotaFile(NaClMessageScanner* scanner, | 42 void OpenQuotaFile(NaClMessageScanner* scanner, |
| 53 PP_Resource file_io, | 43 PP_Resource file_io, |
| 54 PP_Resource file_system) { | 44 PP_Resource file_system) { |
| 55 std::vector<SerializedHandle> unused_handles; | 45 std::vector<SerializedHandle> unused_handles; |
| 56 ResourceMessageReplyParams fio_reply_params(file_io, 0); | 46 ResourceMessageReplyParams fio_reply_params(file_io, 0); |
| 57 scoped_ptr<IPC::Message> new_msg_ptr; | 47 scoped_ptr<IPC::Message> new_msg_ptr; |
| 58 scanner->ScanMessage( | 48 scanner->ScanMessage( |
| 59 PpapiPluginMsg_ResourceReply( | 49 PpapiPluginMsg_ResourceReply( |
| 60 fio_reply_params, | 50 fio_reply_params, |
| 61 PpapiPluginMsg_FileIO_OpenReply(file_system, 0)), | 51 PpapiPluginMsg_FileIO_OpenReply(file_system, 0)), |
| 62 &unused_handles, | 52 PpapiPluginMsg_ResourceReply::ID, |
| 63 &new_msg_ptr); | 53 &unused_handles, |
| 54 &new_msg_ptr); |
| 64 EXPECT_FALSE(new_msg_ptr); | 55 EXPECT_FALSE(new_msg_ptr); |
| 65 } | 56 } |
| 66 }; | 57 }; |
| 67 | 58 |
| 68 TEST_F(NaClMessageScannerTest, SyncMessageAndReply) { | |
| 69 NaClMessageScanner test; | |
| 70 ppapi::proxy::SerializedHandle handle( | |
| 71 ppapi::proxy::SerializedHandle::SHARED_MEMORY); | |
| 72 int id = -1; | |
| 73 IPC::Message msg = | |
| 74 PpapiHostMsg_PPBGraphics3D_CreateTransferBuffer( | |
| 75 ppapi::API_ID_PPB_GRAPHICS_3D, | |
| 76 HostResource(), | |
| 77 4096, // size | |
| 78 &id, | |
| 79 &handle); | |
| 80 scoped_ptr<IPC::Message> new_msg_ptr; | |
| 81 EXPECT_NE(msg.type(), FindPendingSyncMessage(test, msg)); | |
| 82 test.ScanUntrustedMessage(msg, &new_msg_ptr); | |
| 83 EXPECT_FALSE(new_msg_ptr); | |
| 84 EXPECT_EQ(msg.type(), FindPendingSyncMessage(test, msg)); | |
| 85 | |
| 86 // TODO(bbudge) Figure out how to put together a sync reply message. | |
| 87 } | |
| 88 | |
| 89 TEST_F(NaClMessageScannerTest, FileOpenClose) { | 59 TEST_F(NaClMessageScannerTest, FileOpenClose) { |
| 90 NaClMessageScanner test; | 60 NaClMessageScanner test; |
| 91 std::vector<SerializedHandle> unused_handles; | 61 std::vector<SerializedHandle> unused_handles; |
| 92 ResourceMessageCallParams fio_call_params(kFileIO, 0); | 62 ResourceMessageCallParams fio_call_params(kFileIO, 0); |
| 93 ResourceMessageCallParams fs_call_params(kFileSystem, 0); | 63 ResourceMessageCallParams fs_call_params(kFileSystem, 0); |
| 94 ResourceMessageReplyParams fio_reply_params(kFileIO, 0); | 64 ResourceMessageReplyParams fio_reply_params(kFileIO, 0); |
| 95 ResourceMessageReplyParams fs_reply_params(kFileSystem, 0); | 65 ResourceMessageReplyParams fs_reply_params(kFileSystem, 0); |
| 96 scoped_ptr<IPC::Message> new_msg_ptr; | 66 scoped_ptr<IPC::Message> new_msg_ptr; |
| 97 | 67 |
| 98 EXPECT_EQ(NULL, FindFileSystem(test, kFileSystem)); | 68 EXPECT_EQ(NULL, FindFileSystem(test, kFileSystem)); |
| 99 EXPECT_EQ(NULL, FindFileIO(test, kFileIO)); | 69 EXPECT_EQ(NULL, FindFileIO(test, kFileIO)); |
| 100 | 70 |
| 101 // Open a file, not in a quota file system. | 71 // Open a file, not in a quota file system. |
| 102 test.ScanMessage( | 72 test.ScanMessage( |
| 103 PpapiPluginMsg_ResourceReply( | 73 PpapiPluginMsg_ResourceReply( |
| 104 fio_reply_params, | 74 fio_reply_params, |
| 105 PpapiPluginMsg_FileIO_OpenReply(kInvalidResource, 0)), | 75 PpapiPluginMsg_FileIO_OpenReply(kInvalidResource, 0)), |
| 76 PpapiPluginMsg_ResourceReply::ID, |
| 106 &unused_handles, | 77 &unused_handles, |
| 107 &new_msg_ptr); | 78 &new_msg_ptr); |
| 108 EXPECT_FALSE(new_msg_ptr); | 79 EXPECT_FALSE(new_msg_ptr); |
| 109 EXPECT_FALSE(FindFileSystem(test, kFileSystem)); | 80 EXPECT_FALSE(FindFileSystem(test, kFileSystem)); |
| 110 EXPECT_FALSE(FindFileIO(test, kFileIO)); | 81 EXPECT_FALSE(FindFileIO(test, kFileIO)); |
| 111 | 82 |
| 112 // Open a file in a quota file system; info objects for it and its file system | 83 // Open a file in a quota file system; info objects for it and its file system |
| 113 // should be created. | 84 // should be created. |
| 114 OpenQuotaFile(&test, kFileIO, kFileSystem); | 85 OpenQuotaFile(&test, kFileIO, kFileSystem); |
| 115 NaClMessageScanner::FileSystem* fs = FindFileSystem(test, kFileSystem); | 86 NaClMessageScanner::FileSystem* fs = FindFileSystem(test, kFileSystem); |
| (...skipping 72 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 188 // Receive reserved quota, and updated file sizes. | 159 // Receive reserved quota, and updated file sizes. |
| 189 const int64_t kNewFileSize = 10; | 160 const int64_t kNewFileSize = 10; |
| 190 FileSizeMap file_sizes; | 161 FileSizeMap file_sizes; |
| 191 file_sizes[kFileIO] = kNewFileSize; | 162 file_sizes[kFileIO] = kNewFileSize; |
| 192 test.ScanMessage( | 163 test.ScanMessage( |
| 193 PpapiPluginMsg_ResourceReply( | 164 PpapiPluginMsg_ResourceReply( |
| 194 fs_reply_params, | 165 fs_reply_params, |
| 195 PpapiPluginMsg_FileSystem_ReserveQuotaReply( | 166 PpapiPluginMsg_FileSystem_ReserveQuotaReply( |
| 196 kQuotaReservationAmount, | 167 kQuotaReservationAmount, |
| 197 file_sizes)), | 168 file_sizes)), |
| 169 PpapiPluginMsg_ResourceReply::ID, |
| 198 &unused_handles, | 170 &unused_handles, |
| 199 &new_msg_ptr); | 171 &new_msg_ptr); |
| 200 EXPECT_FALSE(new_msg_ptr); | 172 EXPECT_FALSE(new_msg_ptr); |
| 201 EXPECT_EQ(kQuotaReservationAmount, fs->reserved_quota()); | 173 EXPECT_EQ(kQuotaReservationAmount, fs->reserved_quota()); |
| 202 EXPECT_EQ(kNewFileSize, fio->max_written_offset()); | 174 EXPECT_EQ(kNewFileSize, fio->max_written_offset()); |
| 203 | 175 |
| 204 // We should be able to grow the file within quota. | 176 // We should be able to grow the file within quota. |
| 205 EXPECT_TRUE(fio->Grow(1)); | 177 EXPECT_TRUE(fio->Grow(1)); |
| 206 EXPECT_EQ(kQuotaReservationAmount - 1, fs->reserved_quota()); | 178 EXPECT_EQ(kQuotaReservationAmount - 1, fs->reserved_quota()); |
| 207 EXPECT_EQ(kNewFileSize + 1, fio->max_written_offset()); | 179 EXPECT_EQ(kNewFileSize + 1, fio->max_written_offset()); |
| (...skipping 42 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 250 // Receive reserved quota, and updated file sizes. | 222 // Receive reserved quota, and updated file sizes. |
| 251 const int64_t kNewFileSize = 10; | 223 const int64_t kNewFileSize = 10; |
| 252 FileSizeMap file_sizes; | 224 FileSizeMap file_sizes; |
| 253 file_sizes[kFileIO] = 0; | 225 file_sizes[kFileIO] = 0; |
| 254 test.ScanMessage( | 226 test.ScanMessage( |
| 255 PpapiPluginMsg_ResourceReply( | 227 PpapiPluginMsg_ResourceReply( |
| 256 fs_reply_params, | 228 fs_reply_params, |
| 257 PpapiPluginMsg_FileSystem_ReserveQuotaReply( | 229 PpapiPluginMsg_FileSystem_ReserveQuotaReply( |
| 258 kQuotaReservationAmount, | 230 kQuotaReservationAmount, |
| 259 file_sizes)), | 231 file_sizes)), |
| 232 PpapiPluginMsg_ResourceReply::ID, |
| 260 &unused_handles, | 233 &unused_handles, |
| 261 &new_msg_ptr); | 234 &new_msg_ptr); |
| 262 | 235 |
| 263 // We should be able to SetLength within quota. | 236 // We should be able to SetLength within quota. |
| 264 test.ScanUntrustedMessage( | 237 test.ScanUntrustedMessage( |
| 265 PpapiHostMsg_ResourceCall( | 238 PpapiHostMsg_ResourceCall( |
| 266 fio_call_params, | 239 fio_call_params, |
| 267 PpapiHostMsg_FileIO_SetLength(kNewFileSize)), | 240 PpapiHostMsg_FileIO_SetLength(kNewFileSize)), |
| 268 &new_msg_ptr); | 241 &new_msg_ptr); |
| 269 EXPECT_FALSE(new_msg_ptr); | 242 EXPECT_FALSE(new_msg_ptr); |
| (...skipping 16 matching lines...) Expand all Loading... |
| 286 UnpackMessage<PpapiHostMsg_FileIO_SetLength>( | 259 UnpackMessage<PpapiHostMsg_FileIO_SetLength>( |
| 287 nested_msg, &length)); | 260 nested_msg, &length)); |
| 288 new_msg_ptr.reset(); | 261 new_msg_ptr.reset(); |
| 289 EXPECT_EQ(-1, length); | 262 EXPECT_EQ(-1, length); |
| 290 EXPECT_EQ(kQuotaReservationAmount - kNewFileSize, fs->reserved_quota()); | 263 EXPECT_EQ(kQuotaReservationAmount - kNewFileSize, fs->reserved_quota()); |
| 291 EXPECT_EQ(kNewFileSize, fio->max_written_offset()); | 264 EXPECT_EQ(kNewFileSize, fio->max_written_offset()); |
| 292 } | 265 } |
| 293 | 266 |
| 294 } // namespace proxy | 267 } // namespace proxy |
| 295 } // namespace ppapi | 268 } // namespace ppapi |
| OLD | NEW |