1a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)/*
2a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) *  Copyright 2012 The WebRTC Project Authors. All rights reserved.
3a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) *
4a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) *  Use of this source code is governed by a BSD-style license
5a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) *  that can be found in the LICENSE file in the root of the source
6a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) *  tree. An additional intellectual property rights grant can be found
7a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) *  in the file PATENTS.  All contributing project authors may
8a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) *  be found in the AUTHORS file in the root of the source tree.
9a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) */
10a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)
11a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#ifndef WEBRTC_BASE_MD5DIGEST_H_
12a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#define WEBRTC_BASE_MD5DIGEST_H_
13a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)
14a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "webrtc/base/md5.h"
15a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#include "webrtc/base/messagedigest.h"
16a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)
17a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)namespace rtc {
18a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)
19a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)// A simple wrapper for our MD5 implementation.
20a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)class Md5Digest : public MessageDigest {
21a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) public:
22a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  enum { kSize = 16 };
23a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  Md5Digest() {
24a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    MD5Init(&ctx_);
25a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  }
26a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  virtual size_t Size() const {
27a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    return kSize;
28a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  }
29a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  virtual void Update(const void* buf, size_t len) {
30a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    MD5Update(&ctx_, static_cast<const uint8*>(buf), len);
31a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  }
32a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  virtual size_t Finish(void* buf, size_t len) {
33a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    if (len < kSize) {
34a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)      return 0;
35a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    }
36a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    MD5Final(&ctx_, static_cast<uint8*>(buf));
37a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    MD5Init(&ctx_);  // Reset for next use.
38a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)    return kSize;
39a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  }
40a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) private:
41a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)  MD5_CTX ctx_;
42a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)};
43a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)
44a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)}  // namespace rtc
45a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)
46a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)#endif  // WEBRTC_BASE_MD5DIGEST_H_
47a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles)