github.com/zuoyebang/bitalosdb@v1.1.1-0.20240516111551-79a8c4d8ce20/internal/hash/md5_test.go (about)

     1  // Copyright 2021 The Bitalosdb author(hustxrb@163.com) and other contributors.
     2  //
     3  // Licensed under the Apache License, Version 2.0 (the "License");
     4  // you may not use this file except in compliance with the License.
     5  // You may obtain a copy of the License at
     6  //
     7  //     http://www.apache.org/licenses/LICENSE-2.0
     8  //
     9  // Unless required by applicable law or agreed to in writing, software
    10  // distributed under the License is distributed on an "AS IS" BASIS,
    11  // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    12  // See the License for the specific language governing permissions and
    13  // limitations under the License.
    14  
    15  package hash
    16  
    17  import (
    18  	"bytes"
    19  	"crypto/rand"
    20  	"encoding"
    21  	"fmt"
    22  	"hash"
    23  	"io"
    24  	"testing"
    25  	"unsafe"
    26  )
    27  
    28  type md5Test struct {
    29  	out       string
    30  	in        string
    31  	halfState string // marshaled hash state after first half of in written, used by TestGoldenMarshal
    32  }
    33  
    34  var golden = []md5Test{
    35  	{"d41d8cd98f00b204e9800998ecf8427e", "", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tv\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"},
    36  	{"0cc175b9c0f1b6a831c399e269772661", "a", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tv\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00"},
    37  	{"187ef4436122d1cc2f40dc2b92f0eba0", "ab", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tva\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01"},
    38  	{"900150983cd24fb0d6963f7d28e17f72", "abc", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tva\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01"},
    39  	{"e2fc714c4727ee9395f324cd2e7f331f", "abcd", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tvab\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x02"},
    40  	{"ab56b4d92b40713acc5af89985d4b786", "abcde", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tvab\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x02"},
    41  	{"e80b5017098950fc58aad83c8c14978e", "abcdef", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tvabc\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03"},
    42  	{"7ac66c0f148de9519b8bd264312c4d64", "abcdefg", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tvabc\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x03"},
    43  	{"e8dc4081b13434b45189a720b77b6818", "abcdefgh", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tvabcd\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04"},
    44  	{"8aa99b1f439ff71293e95357bac6fd94", "abcdefghi", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tvabcd\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x04"},
    45  	{"a925576942e94b2ef57a066101b48876", "abcdefghij", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tvabcde\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x05"},
    46  	{"d747fc1719c7eacb84058196cfe56d57", "Discard medicine more than two years old.", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvDiscard medicine mor\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x14"},
    47  	{"bff2dcb37ef3a44ba43ab144768ca837", "He who has a shady past knows that nice guys finish last.", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvHe who has a shady past know\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x1c"},
    48  	{"0441015ecb54a7342d017ed1bcfdbea5", "I wouldn't marry him with a ten foot pole.", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvI wouldn't marry him \x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x15"},
    49  	{"9e3cac8e9e9757a60c3ea391130d3689", "Free! Free!/A trip/to Mars/for 900/empty jars/Burma Shave", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvFree! Free!/A trip/to Mars/f\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x1c"},
    50  	{"a0f04459b031f916a59a35cc482dc039", "The days of the digital watch are numbered.  -Tom Stoppard", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvThe days of the digital watch\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x1d"},
    51  	{"e7a48e0fe884faf31475d2a04b1362cc", "Nepal premier won't resign.", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvNepal premier\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\r"},
    52  	{"637d2fe925c07c113800509964fb0e06", "For every action there is an equal and opposite government program.", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvFor every action there is an equa\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00!"},
    53  	{"834a8d18d5c6562119cf4c7f5086cb71", "His money is twice tainted: 'taint yours and 'taint mine.", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvHis money is twice tainted: \x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x1c"},
    54  	{"de3a4d2fd6c73ec2db2abad23b444281", "There is no reason for any individual to have a computer in their home. -Ken Olsen, 1977", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvThere is no reason for any individual to hav\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00,"},
    55  	{"acf203f997e2cf74ea3aff86985aefaf", "It's a tiny change to the code and not completely disgusting. - Bob Manchek", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvIt's a tiny change to the code and no\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00%"},
    56  	{"e1c1384cb4d2221dfdd7c795a4222c9a", "size:  a.out:  bad magic", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102Tvsize:  a.out\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\f"},
    57  	{"c90f3ddecc54f34228c063d7525bf644", "The major problem is with sendmail.  -Mark Horton", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvThe major problem is wit\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x18"},
    58  	{"cdf7ab6c1fd49bd9933c43f3ea5af185", "Give me a rock, paper and scissors and I will move the world.  CCFestoon", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvGive me a rock, paper and scissors a\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00$"},
    59  	{"83bc85234942fc883c063cbd7f0ad5d0", "If the enemy is within range, then so are you.", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvIf the enemy is within \x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x17"},
    60  	{"277cbe255686b48dd7e8f389394d9299", "It's well we cannot hear the screams/That we create in others' dreams.", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvIt's well we cannot hear the scream\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00#"},
    61  	{"fd3fb0a7ffb8af16603f3d3af98f8e1f", "You remind me of a TV show, but that's all right: I watch it anyway.", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvYou remind me of a TV show, but th\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\""},
    62  	{"469b13a78ebf297ecda64d4723655154", "C is as portable as Stonehedge!!", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvC is as portable\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x10"},
    63  	{"63eb3a2f466410104731c4b037600110", "Even if I could be Shakespeare, I think I should still choose to be Faraday. - A. Huxley", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvEven if I could be Shakespeare, I think I sh\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00,"},
    64  	{"72c2ed7592debca1c90fc0100f931a2f", "The fugacity of a constituent in a mixture of gases at a given temperature is proportional to its mole fraction.  Lewis-Randall Rule", "md5\x01\xa7\xc9\x18\x9b\xc3E\x18\xf2\x82\xfd\xf3$\x9d_\v\nem\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00B"},
    65  	{"132f7619d33b523b1d9e5bd8e0928355", "How can you write a big system without C++?  -Paul Glick", "md5\x01gE#\x01\xefͫ\x89\x98\xba\xdc\xfe\x102TvHow can you write a big syst\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x1c"},
    66  }
    67  
    68  func TestGolden(t *testing.T) {
    69  	for i := 0; i < len(golden); i++ {
    70  		g := golden[i]
    71  		s := fmt.Sprintf("%x", Sum([]byte(g.in)))
    72  		if s != g.out {
    73  			t.Fatalf("Sum function: md5(%s) = %s want %s", g.in, s, g.out)
    74  		}
    75  		c := New()
    76  		buf := make([]byte, len(g.in)+4)
    77  		for j := 0; j < 3+4; j++ {
    78  			if j < 2 {
    79  				io.WriteString(c, g.in)
    80  			} else if j == 2 {
    81  				io.WriteString(c, g.in[0:len(g.in)/2])
    82  				c.Sum(nil)
    83  				io.WriteString(c, g.in[len(g.in)/2:])
    84  			} else if j > 2 {
    85  				// test unaligned write
    86  				buf = buf[1:]
    87  				copy(buf, g.in)
    88  				c.Write(buf[:len(g.in)])
    89  			}
    90  			s := fmt.Sprintf("%x", c.Sum(nil))
    91  			if s != g.out {
    92  				t.Fatalf("md5[%d](%s) = %s want %s", j, g.in, s, g.out)
    93  			}
    94  			c.Reset()
    95  		}
    96  	}
    97  }
    98  
    99  func TestGoldenMarshal(t *testing.T) {
   100  	for _, g := range golden {
   101  		h := New()
   102  		h2 := New()
   103  
   104  		io.WriteString(h, g.in[:len(g.in)/2])
   105  
   106  		state, err := h.(encoding.BinaryMarshaler).MarshalBinary()
   107  		if err != nil {
   108  			t.Errorf("could not marshal: %v", err)
   109  			continue
   110  		}
   111  
   112  		if string(state) != g.halfState {
   113  			t.Errorf("md5(%q) state = %q, want %q", g.in, state, g.halfState)
   114  			continue
   115  		}
   116  
   117  		if err := h2.(encoding.BinaryUnmarshaler).UnmarshalBinary(state); err != nil {
   118  			t.Errorf("could not unmarshal: %v", err)
   119  			continue
   120  		}
   121  
   122  		io.WriteString(h, g.in[len(g.in)/2:])
   123  		io.WriteString(h2, g.in[len(g.in)/2:])
   124  
   125  		if actual, actual2 := h.Sum(nil), h2.Sum(nil); !bytes.Equal(actual, actual2) {
   126  			t.Errorf("md5(%q) = 0x%x != marshaled 0x%x", g.in, actual, actual2)
   127  		}
   128  	}
   129  }
   130  
   131  func TestLarge(t *testing.T) {
   132  	const N = 10000
   133  	ok := "2bb571599a4180e1d542f76904adc3df" // md5sum of "0123456789" * 1000
   134  	block := make([]byte, 10004)
   135  	c := New()
   136  	for offset := 0; offset < 4; offset++ {
   137  		for i := 0; i < N; i++ {
   138  			block[offset+i] = '0' + byte(i%10)
   139  		}
   140  		for blockSize := 10; blockSize <= N; blockSize *= 10 {
   141  			blocks := N / blockSize
   142  			b := block[offset : offset+blockSize]
   143  			c.Reset()
   144  			for i := 0; i < blocks; i++ {
   145  				c.Write(b)
   146  			}
   147  			s := fmt.Sprintf("%x", c.Sum(nil))
   148  			if s != ok {
   149  				t.Fatalf("md5 TestLarge offset=%d, blockSize=%d = %s want %s", offset, blockSize, s, ok)
   150  			}
   151  		}
   152  	}
   153  }
   154  
   155  // Tests that blockGeneric (pure Go) and block (in assembly for amd64, 386, arm) match.
   156  func TestBlockGeneric(t *testing.T) {
   157  	gen, asm := New().(*digest), New().(*digest)
   158  	buf := make([]byte, BlockSize*20) // arbitrary factor
   159  	rand.Read(buf)
   160  	blockGeneric(gen, buf)
   161  	block(asm, buf)
   162  	if *gen != *asm {
   163  		t.Error("block and blockGeneric resulted in different states")
   164  	}
   165  }
   166  
   167  // Tests for unmarshaling hashes that have hashed a large amount of data
   168  // The initial hash generation is omitted from the test, because it takes a long time.
   169  // The test contains some already-generated states, and their expected sums
   170  // Tests a problem that is outlined in GitHub issue #29541
   171  // The problem is triggered when an amount of data has been hashed for which
   172  // the data length has a 1 in the 32nd bit. When casted to int, this changes
   173  // the sign of the value, and causes the modulus operation to return a
   174  // different result.
   175  type unmarshalTest struct {
   176  	state string
   177  	sum   string
   178  }
   179  
   180  var largeUnmarshalTests = []unmarshalTest{
   181  	// Data length: 7_102_415_735
   182  	{
   183  		state: "md5\x01\xa5\xf7\xf0=\xd6S\x85\xd9M\n}\xc3\u0601\x89\xe7@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuv\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\xa7VCw",
   184  		sum:   "cddefcf74ffec709a0b45a6a987564d5",
   185  	},
   186  	// Data length: 6_565_544_823
   187  	{
   188  		state: "md5\x01{\xda\x1a\xc7\xc9'?\x83EX\xe0\x88q\xfeG\x18@ABCDEFGHIJKLMNOPQRSTUVWXYZ[\\]^_`abcdefghijklmnopqrstuv\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x01\x87VCw",
   189  		sum:   "fd9f41874ab240698e7bc9c3ae70c8e4",
   190  	},
   191  }
   192  
   193  func safeSum(h hash.Hash) (sum []byte, err error) {
   194  	defer func() {
   195  		if r := recover(); r != nil {
   196  			err = fmt.Errorf("sum panic: %v", r)
   197  		}
   198  	}()
   199  
   200  	return h.Sum(nil), nil
   201  }
   202  
   203  func TestLargeHashes(t *testing.T) {
   204  	for i, test := range largeUnmarshalTests {
   205  
   206  		h := New()
   207  		if err := h.(encoding.BinaryUnmarshaler).UnmarshalBinary([]byte(test.state)); err != nil {
   208  			t.Errorf("test %d could not unmarshal: %v", i, err)
   209  			continue
   210  		}
   211  
   212  		sum, err := safeSum(h)
   213  		if err != nil {
   214  			t.Errorf("test %d could not sum: %v", i, err)
   215  			continue
   216  		}
   217  
   218  		if fmt.Sprintf("%x", sum) != test.sum {
   219  			t.Errorf("test %d sum mismatch: expect %s got %x", i, test.sum, sum)
   220  		}
   221  	}
   222  }
   223  
   224  func TestAllocations(t *testing.T) {
   225  	in := []byte("hello, world!")
   226  	out := make([]byte, 0, Size)
   227  	h := New()
   228  	n := int(testing.AllocsPerRun(10, func() {
   229  		h.Reset()
   230  		h.Write(in)
   231  		out = h.Sum(out[:0])
   232  	}))
   233  	if n > 0 {
   234  		t.Errorf("allocs = %d, want 0", n)
   235  	}
   236  }
   237  
   238  func TestMD5Hash(t *testing.T) {
   239  	in := []byte("hello, world!")
   240  	h, hi, lo := MD5Hash(in)
   241  	h2, hi2, lo2 := MD5Sum(in)
   242  
   243  	if !bytes.Equal(h, h2) {
   244  		t.Errorf("MD5Hash(in) = %x, want %x", h, h2)
   245  	}
   246  	if hi != hi2 {
   247  		t.Errorf("MD5Hash(in) = %x, want %x", hi, hi2)
   248  	}
   249  	if lo != lo2 {
   250  		t.Errorf("MD5Hash(in) = %x, want %x", lo, lo2)
   251  	}
   252  }
   253  
   254  var bench = New()
   255  var buf = make([]byte, 1024*1024*8+1)
   256  var sum = make([]byte, bench.Size())
   257  
   258  func benchmarkSize(b *testing.B, size int, unaligned bool) {
   259  	b.SetBytes(int64(size))
   260  	buf := buf
   261  	if unaligned {
   262  		if uintptr(unsafe.Pointer(&buf[0]))&(unsafe.Alignof(uint32(0))-1) == 0 {
   263  			buf = buf[1:]
   264  		}
   265  	}
   266  	b.ResetTimer()
   267  	for i := 0; i < b.N; i++ {
   268  		bench.Reset()
   269  		bench.Write(buf[:size])
   270  		bench.Sum(sum[:0])
   271  	}
   272  }
   273  
   274  func BenchmarkHash8Bytes(b *testing.B) {
   275  	benchmarkSize(b, 8, false)
   276  }
   277  
   278  func BenchmarkHash64(b *testing.B) {
   279  	benchmarkSize(b, 64, false)
   280  }
   281  
   282  func BenchmarkHash128(b *testing.B) {
   283  	benchmarkSize(b, 128, false)
   284  }
   285  
   286  func BenchmarkHash256(b *testing.B) {
   287  	benchmarkSize(b, 256, false)
   288  }
   289  
   290  func BenchmarkHash512(b *testing.B) {
   291  	benchmarkSize(b, 512, false)
   292  }
   293  
   294  func BenchmarkHash1K(b *testing.B) {
   295  	benchmarkSize(b, 1024, false)
   296  }
   297  
   298  func BenchmarkHash8K(b *testing.B) {
   299  	benchmarkSize(b, 8192, false)
   300  }
   301  
   302  func BenchmarkHash1M(b *testing.B) {
   303  	benchmarkSize(b, 1024*1024, false)
   304  }
   305  
   306  func BenchmarkHash8M(b *testing.B) {
   307  	benchmarkSize(b, 8*1024*1024, false)
   308  }
   309  
   310  func BenchmarkHash8BytesUnaligned(b *testing.B) {
   311  	benchmarkSize(b, 8, true)
   312  }
   313  
   314  func BenchmarkHash1KUnaligned(b *testing.B) {
   315  	benchmarkSize(b, 1024, true)
   316  }
   317  
   318  func BenchmarkHash8KUnaligned(b *testing.B) {
   319  	benchmarkSize(b, 8192, true)
   320  }