zqz/platform-external-webrtc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add checks for buffer size in MediaCodecVideoEncoder.

Browse Source 
This should help users of the library to more easily debug issues.

Bug: None
Change-Id: I85d8101d3b26ccbc34c8beded069461252e61293
Reviewed-on: https://webrtc-review.googlesource.com/4663
Reviewed-by: Magnus Jedvert <magjed@webrtc.org>
Reviewed-by: Sami Kalliomäki <sakal@webrtc.org>
Commit-Queue: Sami Kalliomäki <sakal@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#20073}
This commit is contained in:
Sami Kalliomäki
2017-10-02 09:41:55 +02:00
committed by Commit Bot
parent a58f231018
commit e3044fe12e
3 changed files with 37 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
sdk/android/api/org/webrtc/MediaCodecVideoEncoder.java
View File

@ -621,9 +621,24 @@ public class MediaCodecVideoEncoder {
eglBase.swapBuffers(frame.getTimestampNs());
} else {
VideoFrame.I420Buffer i420Buffer = buffer.toI420();
nativeFillBuffer(nativeEncoder, bufferIndex, i420Buffer.getDataY(), i420Buffer.getStrideY(),
i420Buffer.getDataU(), i420Buffer.getStrideU(), i420Buffer.getDataV(),
i420Buffer.getStrideV());
final int chromaHeight = (height + 1) / 2;
final ByteBuffer dataY = i420Buffer.getDataY();
final ByteBuffer dataU = i420Buffer.getDataU();
final ByteBuffer dataV = i420Buffer.getDataV();
final int strideY = i420Buffer.getStrideY();
final int strideU = i420Buffer.getStrideU();
final int strideV = i420Buffer.getStrideV();
if (dataY.capacity() < strideY * height) {
throw new RuntimeException("Y-plane buffer size too small.");
}
if (dataU.capacity() < strideU * chromaHeight) {
throw new RuntimeException("U-plane buffer size too small.");
}
if (dataV.capacity() < strideV * chromaHeight) {
throw new RuntimeException("V-plane buffer size too small.");
}
nativeFillBuffer(
nativeEncoder, bufferIndex, dataY, strideY, dataU, strideU, dataV, strideV);
i420Buffer.release();
// I420 consists of one full-resolution and two half-resolution planes.
// 1 + 1 / 4 + 1 / 4 = 3 / 2