SHA1.h 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. /*
  2. * Copyright (C) 2011 Google Inc. All rights reserved.
  3. *
  4. * Redistribution and use in source and binary forms, with or without
  5. * modification, are permitted provided that the following conditions are
  6. * met:
  7. *
  8. * * Redistributions of source code must retain the above copyright
  9. * notice, this list of conditions and the following disclaimer.
  10. * * Redistributions in binary form must reproduce the above
  11. * copyright notice, this list of conditions and the following disclaimer
  12. * in the documentation and/or other materials provided with the
  13. * distribution.
  14. * * Neither the name of Google Inc. nor the names of its
  15. * contributors may be used to endorse or promote products derived from
  16. * this software without specific prior written permission.
  17. *
  18. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  19. * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  20. * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  21. * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  22. * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  23. * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  24. * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  25. * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  26. * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  27. * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  28. * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  29. */
  30. #ifndef WTF_SHA1_h
  31. #define WTF_SHA1_h
  32. #include <wtf/Vector.h>
  33. #include <wtf/text/CString.h>
  34. namespace WTF {
  35. class SHA1 {
  36. public:
  37. WTF_EXPORT_PRIVATE SHA1();
  38. void addBytes(const Vector<uint8_t>& input)
  39. {
  40. addBytes(input.data(), input.size());
  41. }
  42. void addBytes(const CString& input)
  43. {
  44. const char* string = input.data();
  45. // Make sure that the creator of the CString didn't make the mistake
  46. // of forcing length() to be the size of the buffer used to create the
  47. // string, prior to inserting the null terminator earlier in the
  48. // sequence.
  49. ASSERT(input.length() == strlen(string));
  50. addBytes(reinterpret_cast<const uint8_t*>(string), input.length());
  51. }
  52. WTF_EXPORT_PRIVATE void addBytes(const uint8_t* input, size_t length);
  53. // computeHash has a side effect of resetting the state of the object.
  54. WTF_EXPORT_PRIVATE void computeHash(Vector<uint8_t, 20>&);
  55. // Get a hex hash from the digest. Pass a limit less than 40 if you want a shorter digest.
  56. WTF_EXPORT_PRIVATE static CString hexDigest(const Vector<uint8_t, 20>&);
  57. // Compute the hex digest directly. Pass a limit less than 40 if you want a shorter digest.
  58. WTF_EXPORT_PRIVATE CString computeHexDigest();
  59. private:
  60. void finalize();
  61. void processBlock();
  62. void reset();
  63. uint8_t m_buffer[64];
  64. size_t m_cursor; // Number of bytes filled in m_buffer (0-64).
  65. uint64_t m_totalBytes; // Number of bytes added so far.
  66. uint32_t m_hash[5];
  67. };
  68. } // namespace WTF
  69. using WTF::SHA1;
  70. #endif // WTF_SHA1_h