| OLD | NEW |
| 1 /* | 1 /* |
| 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2015 The WebRTC project authors. All Rights Reserved. |
| 3 * | 3 * |
| 4 * Use of this source code is governed by a BSD-style license | 4 * Use of this source code is governed by a BSD-style license |
| 5 * that can be found in the LICENSE file in the root of the source | 5 * that can be found in the LICENSE file in the root of the source |
| 6 * tree. An additional intellectual property rights grant can be found | 6 * tree. An additional intellectual property rights grant can be found |
| 7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
| 8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
| 9 */ | 9 */ |
| 10 | 10 |
| 11 #include <string> | 11 #include <string> |
| 12 | 12 |
| 13 #include "webrtc/common_video/include/i420_buffer_pool.h" | 13 #include "webrtc/common_video/include/i420_buffer_pool.h" |
| 14 #include "webrtc/test/gtest.h" | 14 #include "webrtc/test/gtest.h" |
| 15 | 15 |
| 16 namespace webrtc { | 16 namespace webrtc { |
| 17 | 17 |
| 18 TEST(TestI420BufferPool, SimpleFrameReuse) { | 18 TEST(TestI420BufferPool, SimpleFrameReuse) { |
| 19 I420BufferPool pool; | 19 I420BufferPool pool; |
| 20 rtc::scoped_refptr<VideoFrameBuffer> buffer = pool.CreateBuffer(16, 16); | 20 rtc::scoped_refptr<I420BufferInterface> buffer = pool.CreateBuffer(16, 16); |
| 21 EXPECT_EQ(16, buffer->width()); | 21 EXPECT_EQ(16, buffer->width()); |
| 22 EXPECT_EQ(16, buffer->height()); | 22 EXPECT_EQ(16, buffer->height()); |
| 23 // Extract non-refcounted pointers for testing. | 23 // Extract non-refcounted pointers for testing. |
| 24 const uint8_t* y_ptr = buffer->DataY(); | 24 const uint8_t* y_ptr = buffer->DataY(); |
| 25 const uint8_t* u_ptr = buffer->DataU(); | 25 const uint8_t* u_ptr = buffer->DataU(); |
| 26 const uint8_t* v_ptr = buffer->DataV(); | 26 const uint8_t* v_ptr = buffer->DataV(); |
| 27 // Release buffer so that it is returned to the pool. | 27 // Release buffer so that it is returned to the pool. |
| 28 buffer = nullptr; | 28 buffer = nullptr; |
| 29 // Check that the memory is resued. | 29 // Check that the memory is resued. |
| 30 buffer = pool.CreateBuffer(16, 16); | 30 buffer = pool.CreateBuffer(16, 16); |
| 31 EXPECT_EQ(y_ptr, buffer->DataY()); | 31 EXPECT_EQ(y_ptr, buffer->DataY()); |
| 32 EXPECT_EQ(u_ptr, buffer->DataU()); | 32 EXPECT_EQ(u_ptr, buffer->DataU()); |
| 33 EXPECT_EQ(v_ptr, buffer->DataV()); | 33 EXPECT_EQ(v_ptr, buffer->DataV()); |
| 34 EXPECT_EQ(16, buffer->width()); | 34 EXPECT_EQ(16, buffer->width()); |
| 35 EXPECT_EQ(16, buffer->height()); | 35 EXPECT_EQ(16, buffer->height()); |
| 36 } | 36 } |
| 37 | 37 |
| 38 TEST(TestI420BufferPool, FailToReuse) { | 38 TEST(TestI420BufferPool, FailToReuse) { |
| 39 I420BufferPool pool; | 39 I420BufferPool pool; |
| 40 rtc::scoped_refptr<VideoFrameBuffer> buffer = pool.CreateBuffer(16, 16); | 40 rtc::scoped_refptr<I420BufferInterface> buffer = pool.CreateBuffer(16, 16); |
| 41 // Extract non-refcounted pointers for testing. | 41 // Extract non-refcounted pointers for testing. |
| 42 const uint8_t* u_ptr = buffer->DataU(); | 42 const uint8_t* u_ptr = buffer->DataU(); |
| 43 const uint8_t* v_ptr = buffer->DataV(); | 43 const uint8_t* v_ptr = buffer->DataV(); |
| 44 // Release buffer so that it is returned to the pool. | 44 // Release buffer so that it is returned to the pool. |
| 45 buffer = nullptr; | 45 buffer = nullptr; |
| 46 // Check that the pool doesn't try to reuse buffers of incorrect size. | 46 // Check that the pool doesn't try to reuse buffers of incorrect size. |
| 47 buffer = pool.CreateBuffer(32, 16); | 47 buffer = pool.CreateBuffer(32, 16); |
| 48 EXPECT_EQ(32, buffer->width()); | 48 EXPECT_EQ(32, buffer->width()); |
| 49 EXPECT_EQ(16, buffer->height()); | 49 EXPECT_EQ(16, buffer->height()); |
| 50 EXPECT_NE(u_ptr, buffer->DataU()); | 50 EXPECT_NE(u_ptr, buffer->DataU()); |
| 51 EXPECT_NE(v_ptr, buffer->DataV()); | 51 EXPECT_NE(v_ptr, buffer->DataV()); |
| 52 } | 52 } |
| 53 | 53 |
| 54 TEST(TestI420BufferPool, FrameValidAfterPoolDestruction) { | 54 TEST(TestI420BufferPool, FrameValidAfterPoolDestruction) { |
| 55 rtc::scoped_refptr<I420Buffer> buffer; | 55 rtc::scoped_refptr<I420Buffer> buffer; |
| 56 { | 56 { |
| 57 I420BufferPool pool; | 57 I420BufferPool pool; |
| 58 buffer = pool.CreateBuffer(16, 16); | 58 buffer = pool.CreateBuffer(16, 16); |
| 59 } | 59 } |
| 60 EXPECT_EQ(16, buffer->width()); | 60 EXPECT_EQ(16, buffer->width()); |
| 61 EXPECT_EQ(16, buffer->height()); | 61 EXPECT_EQ(16, buffer->height()); |
| 62 // Try to trigger use-after-free errors by writing to y-plane. | 62 // Try to trigger use-after-free errors by writing to y-plane. |
| 63 memset(buffer->MutableDataY(), 0xA5, 16 * buffer->StrideY()); | 63 memset(buffer->MutableDataY(), 0xA5, 16 * buffer->StrideY()); |
| 64 } | 64 } |
| 65 | 65 |
| 66 TEST(TestI420BufferPool, MaxNumberOfBuffers) { | 66 TEST(TestI420BufferPool, MaxNumberOfBuffers) { |
| 67 I420BufferPool pool(false, 1); | 67 I420BufferPool pool(false, 1); |
| 68 rtc::scoped_refptr<VideoFrameBuffer> buffer1 = pool.CreateBuffer(16, 16); | 68 rtc::scoped_refptr<I420BufferInterface> buffer1 = pool.CreateBuffer(16, 16); |
| 69 EXPECT_NE(nullptr, buffer1.get()); | 69 EXPECT_NE(nullptr, buffer1.get()); |
| 70 EXPECT_EQ(nullptr, pool.CreateBuffer(16, 16).get()); | 70 EXPECT_EQ(nullptr, pool.CreateBuffer(16, 16).get()); |
| 71 } | 71 } |
| 72 | 72 |
| 73 } // namespace webrtc | 73 } // namespace webrtc |
| OLD | NEW |