1 /* 2 * Copyright (c) 2012 The WebM 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 #ifndef TEST_MD5_HELPER_H_ 12 #define TEST_MD5_HELPER_H_ 13 14 #include "./md5_utils.h" 15 #include "vpx/vpx_decoder.h" 16 17 namespace libvpx_test { 18 class MD5 { 19 public: MD5()20 MD5() { MD5Init(&md5_); } 21 Add(const vpx_image_t * img)22 void Add(const vpx_image_t *img) { 23 for (int plane = 0; plane < 3; ++plane) { 24 const uint8_t *buf = img->planes[plane]; 25 // Calculate the width and height to do the md5 check. For the chroma 26 // plane, we never want to round down and thus skip a pixel so if 27 // we are shifting by 1 (chroma_shift) we add 1 before doing the shift. 28 // This works only for chroma_shift of 0 and 1. 29 const int bytes_per_sample = 30 (img->fmt & VPX_IMG_FMT_HIGHBITDEPTH) ? 2 : 1; 31 const int h = 32 plane ? (img->d_h + img->y_chroma_shift) >> img->y_chroma_shift 33 : img->d_h; 34 const int w = 35 (plane ? (img->d_w + img->x_chroma_shift) >> img->x_chroma_shift 36 : img->d_w) * 37 bytes_per_sample; 38 39 for (int y = 0; y < h; ++y) { 40 MD5Update(&md5_, buf, w); 41 buf += img->stride[plane]; 42 } 43 } 44 } 45 Add(const uint8_t * data,size_t size)46 void Add(const uint8_t *data, size_t size) { 47 MD5Update(&md5_, data, static_cast<uint32_t>(size)); 48 } 49 Get(void)50 const char *Get(void) { 51 static const char hex[16] = { 52 '0', '1', '2', '3', '4', '5', '6', '7', 53 '8', '9', 'a', 'b', 'c', 'd', 'e', 'f', 54 }; 55 uint8_t tmp[16]; 56 MD5Context ctx_tmp = md5_; 57 58 MD5Final(tmp, &ctx_tmp); 59 for (int i = 0; i < 16; i++) { 60 res_[i * 2 + 0] = hex[tmp[i] >> 4]; 61 res_[i * 2 + 1] = hex[tmp[i] & 0xf]; 62 } 63 res_[32] = 0; 64 65 return res_; 66 } 67 68 protected: 69 char res_[33]; 70 MD5Context md5_; 71 }; 72 73 } // namespace libvpx_test 74 75 #endif // TEST_MD5_HELPER_H_ 76