Index: webrtc/media/base/videoframe.cc |
diff --git a/webrtc/media/base/videoframe.cc b/webrtc/media/base/videoframe.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..4ce0dfd90cf4dd2a340b407701418aacaeff731b |
--- /dev/null |
+++ b/webrtc/media/base/videoframe.cc |
@@ -0,0 +1,171 @@ |
+/* |
+ * Copyright (c) 2011 The WebRTC project authors. All Rights Reserved. |
+ * |
+ * Use of this source code is governed by a BSD-style license |
+ * that can be found in the LICENSE file in the root of the source |
+ * tree. An additional intellectual property rights grant can be found |
+ * in the file PATENTS. All contributing project authors may |
+ * be found in the AUTHORS file in the root of the source tree. |
+ */ |
+ |
+#include "webrtc/media/base/videoframe.h" |
+ |
+#include <string.h> |
+ |
+#include "webrtc/base/arraysize.h" |
+#include "webrtc/base/checks.h" |
+#include "webrtc/base/logging.h" |
+#include "webrtc/media/base/videocommon.h" |
+ |
+namespace cricket { |
+ |
+static const size_t kMaxSampleSize = 1000000000u; |
+// Returns whether a sample is valid. |
+bool VideoFrame::Validate(uint32_t fourcc, |
+ int w, |
+ int h, |
+ const uint8_t* sample, |
+ size_t sample_size) { |
+ if (h < 0) { |
+ h = -h; |
+ } |
+ // 16384 is maximum resolution for VP8 codec. |
+ if (w < 1 || w > 16384 || h < 1 || h > 16384) { |
+ LOG(LS_ERROR) << "Invalid dimensions: " << w << "x" << h; |
+ return false; |
+ } |
+ uint32_t format = CanonicalFourCC(fourcc); |
+ int expected_bpp = 8; |
+ switch (format) { |
+ case FOURCC_I400: |
+ case FOURCC_RGGB: |
+ case FOURCC_BGGR: |
+ case FOURCC_GRBG: |
+ case FOURCC_GBRG: |
+ expected_bpp = 8; |
+ break; |
+ case FOURCC_I420: |
+ case FOURCC_I411: |
+ case FOURCC_YU12: |
+ case FOURCC_YV12: |
+ case FOURCC_M420: |
+ case FOURCC_NV21: |
+ case FOURCC_NV12: |
+ expected_bpp = 12; |
+ break; |
+ case FOURCC_I422: |
+ case FOURCC_YV16: |
+ case FOURCC_YUY2: |
+ case FOURCC_UYVY: |
+ case FOURCC_RGBP: |
+ case FOURCC_RGBO: |
+ case FOURCC_R444: |
+ expected_bpp = 16; |
+ break; |
+ case FOURCC_I444: |
+ case FOURCC_YV24: |
+ case FOURCC_24BG: |
+ case FOURCC_RAW: |
+ expected_bpp = 24; |
+ break; |
+ |
+ case FOURCC_ABGR: |
+ case FOURCC_BGRA: |
+ case FOURCC_ARGB: |
+ expected_bpp = 32; |
+ break; |
+ |
+ case FOURCC_MJPG: |
+ case FOURCC_H264: |
+ expected_bpp = 0; |
+ break; |
+ default: |
+ expected_bpp = 8; // Expect format is at least 8 bits per pixel. |
+ break; |
+ } |
+ size_t expected_size = (w * expected_bpp + 7) / 8 * h; |
+ // For compressed formats, expect 4 bits per 16 x 16 macro. I420 would be |
+ // 6 bits, but grey can be 4 bits. |
+ if (expected_bpp == 0) { |
+ expected_size = ((w + 15) / 16) * ((h + 15) / 16) * 4 / 8; |
+ } |
+ if (sample == NULL) { |
+ LOG(LS_ERROR) << "NULL sample pointer." |
+ << " format: " << GetFourccName(format) |
+ << " bpp: " << expected_bpp |
+ << " size: " << w << "x" << h |
+ << " expected: " << expected_size |
+ << " " << sample_size; |
+ return false; |
+ } |
+ // TODO(fbarchard): Make function to dump information about frames. |
+ uint8_t four_samples[4] = {0, 0, 0, 0}; |
+ for (size_t i = 0; i < arraysize(four_samples) && i < sample_size; ++i) { |
+ four_samples[i] = sample[i]; |
+ } |
+ if (sample_size < expected_size) { |
+ LOG(LS_ERROR) << "Size field is too small." |
+ << " format: " << GetFourccName(format) |
+ << " bpp: " << expected_bpp |
+ << " size: " << w << "x" << h |
+ << " " << sample_size |
+ << " expected: " << expected_size |
+ << " sample[0..3]: " << static_cast<int>(four_samples[0]) |
+ << ", " << static_cast<int>(four_samples[1]) |
+ << ", " << static_cast<int>(four_samples[2]) |
+ << ", " << static_cast<int>(four_samples[3]); |
+ return false; |
+ } |
+ if (sample_size > kMaxSampleSize) { |
+ LOG(LS_WARNING) << "Size field is invalid." |
+ << " format: " << GetFourccName(format) |
+ << " bpp: " << expected_bpp |
+ << " size: " << w << "x" << h |
+ << " " << sample_size |
+ << " expected: " << 2 * expected_size |
+ << " sample[0..3]: " << static_cast<int>(four_samples[0]) |
+ << ", " << static_cast<int>(four_samples[1]) |
+ << ", " << static_cast<int>(four_samples[2]) |
+ << ", " << static_cast<int>(four_samples[3]); |
+ return false; |
+ } |
+ // Show large size warning once every 100 frames. |
+ // TODO(fbarchard): Make frame counter atomic for thread safety. |
+ static int large_warn100 = 0; |
+ size_t large_expected_size = expected_size * 2; |
+ if (expected_bpp >= 8 && |
+ (sample_size > large_expected_size || sample_size > kMaxSampleSize) && |
+ large_warn100 % 100 == 0) { |
+ ++large_warn100; |
+ LOG(LS_WARNING) << "Size field is too large." |
+ << " format: " << GetFourccName(format) |
+ << " bpp: " << expected_bpp |
+ << " size: " << w << "x" << h |
+ << " bytes: " << sample_size |
+ << " expected: " << large_expected_size |
+ << " sample[0..3]: " << static_cast<int>(four_samples[0]) |
+ << ", " << static_cast<int>(four_samples[1]) |
+ << ", " << static_cast<int>(four_samples[2]) |
+ << ", " << static_cast<int>(four_samples[3]); |
+ } |
+ |
+ // TODO(fbarchard): Add duplicate pixel check. |
+ // TODO(fbarchard): Use frame counter atomic for thread safety. |
+ static bool valid_once = true; |
+ if (valid_once) { |
+ valid_once = false; |
+ LOG(LS_INFO) << "Validate frame passed." |
+ << " format: " << GetFourccName(format) |
+ << " bpp: " << expected_bpp |
+ << " size: " << w << "x" << h |
+ << " bytes: " << sample_size |
+ << " expected: " << expected_size |
+ << " sample[0..3]: " << static_cast<int>(four_samples[0]) |
+ << ", " << static_cast<int>(four_samples[1]) |
+ << ", " << static_cast<int>(four_samples[2]) |
+ << ", " << static_cast<int>(four_samples[3]); |
+ } |
+ return true; |
+} |
+ |
+} // namespace cricket |