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 |