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