Home | History | Annotate | Download | only in sha256 
      1 // Copyright 2009 The Go Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style
      3 // license that can be found in the LICENSE file.
      4 
      5 // SHA256 hash algorithm. See FIPS 180-2.
      6 
      7 package sha256
      8 
      9 import (
     10 	"crypto/rand"
     11 	"fmt"
     12 	"io"
     13 	"testing"
     14 )
     15 
     16 type sha256Test struct {
     17 	out string
     18 	in  string
     19 }
     20 
     21 var golden = []sha256Test{
     22 	{"e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855", ""},
     23 	{"ca978112ca1bbdcafac231b39a23dc4da786eff8147c4e72b9807785afee48bb", "a"},
     24 	{"fb8e20fc2e4c3f248c60c39bd652f3c1347298bb977b8b4d5903b85055620603", "ab"},
     25 	{"ba7816bf8f01cfea414140de5dae2223b00361a396177a9cb410ff61f20015ad", "abc"},
     26 	{"88d4266fd4e6338d13b845fcf289579d209c897823b9217da3e161936f031589", "abcd"},
     27 	{"36bbe50ed96841d10443bcb670d6554f0a34b761be67ec9c4a8ad2c0c44ca42c", "abcde"},
     28 	{"bef57ec7f53a6d40beb640a780a639c83bc29ac8a9816f1fc6c5c6dcd93c4721", "abcdef"},
     29 	{"7d1a54127b222502f5b79b5fb0803061152a44f92b37e23c6527baf665d4da9a", "abcdefg"},
     30 	{"9c56cc51b374c3ba189210d5b6d4bf57790d351c96c47c02190ecf1e430635ab", "abcdefgh"},
     31 	{"19cc02f26df43cc571bc9ed7b0c4d29224a3ec229529221725ef76d021c8326f", "abcdefghi"},
     32 	{"72399361da6a7754fec986dca5b7cbaf1c810a28ded4abaf56b2106d06cb78b0", "abcdefghij"},
     33 	{"a144061c271f152da4d151034508fed1c138b8c976339de229c3bb6d4bbb4fce", "Discard medicine more than two years old."},
     34 	{"6dae5caa713a10ad04b46028bf6dad68837c581616a1589a265a11288d4bb5c4", "He who has a shady past knows that nice guys finish last."},
     35 	{"ae7a702a9509039ddbf29f0765e70d0001177914b86459284dab8b348c2dce3f", "I wouldn't marry him with a ten foot pole."},
     36 	{"6748450b01c568586715291dfa3ee018da07d36bb7ea6f180c1af6270215c64f", "Free! Free!/A trip/to Mars/for 900/empty jars/Burma Shave"},
     37 	{"14b82014ad2b11f661b5ae6a99b75105c2ffac278cd071cd6c05832793635774", "The days of the digital watch are numbered.  -Tom Stoppard"},
     38 	{"7102cfd76e2e324889eece5d6c41921b1e142a4ac5a2692be78803097f6a48d8", "Nepal premier won't resign."},
     39 	{"23b1018cd81db1d67983c5f7417c44da9deb582459e378d7a068552ea649dc9f", "For every action there is an equal and opposite government program."},
     40 	{"8001f190dfb527261c4cfcab70c98e8097a7a1922129bc4096950e57c7999a5a", "His money is twice tainted: 'taint yours and 'taint mine."},
     41 	{"8c87deb65505c3993eb24b7a150c4155e82eee6960cf0c3a8114ff736d69cad5", "There is no reason for any individual to have a computer in their home. -Ken Olsen, 1977"},
     42 	{"bfb0a67a19cdec3646498b2e0f751bddc41bba4b7f30081b0b932aad214d16d7", "It's a tiny change to the code and not completely disgusting. - Bob Manchek"},
     43 	{"7f9a0b9bf56332e19f5a0ec1ad9c1425a153da1c624868fda44561d6b74daf36", "size:  a.out:  bad magic"},
     44 	{"b13f81b8aad9e3666879af19886140904f7f429ef083286195982a7588858cfc", "The major problem is with sendmail.  -Mark Horton"},
     45 	{"b26c38d61519e894480c70c8374ea35aa0ad05b2ae3d6674eec5f52a69305ed4", "Give me a rock, paper and scissors and I will move the world.  CCFestoon"},
     46 	{"049d5e26d4f10222cd841a119e38bd8d2e0d1129728688449575d4ff42b842c1", "If the enemy is within range, then so are you."},
     47 	{"0e116838e3cc1c1a14cd045397e29b4d087aa11b0853fc69ec82e90330d60949", "It's well we cannot hear the screams/That we create in others' dreams."},
     48 	{"4f7d8eb5bcf11de2a56b971021a444aa4eafd6ecd0f307b5109e4e776cd0fe46", "You remind me of a TV show, but that's all right: I watch it anyway."},
     49 	{"61c0cc4c4bd8406d5120b3fb4ebc31ce87667c162f29468b3c779675a85aebce", "C is as portable as Stonehedge!!"},
     50 	{"1fb2eb3688093c4a3f80cd87a5547e2ce940a4f923243a79a2a1e242220693ac", "Even if I could be Shakespeare, I think I should still choose to be Faraday. - A. Huxley"},
     51 	{"395585ce30617b62c80b93e8208ce866d4edc811a177fdb4b82d3911d8696423", "The fugacity of a constituent in a mixture of gases at a given temperature is proportional to its mole fraction.  Lewis-Randall Rule"},
     52 	{"4f9b189a13d030838269dce846b16a1ce9ce81fe63e65de2f636863336a98fe6", "How can you write a big system without C++?  -Paul Glick"},
     53 }
     54 
     55 var golden224 = []sha256Test{
     56 	{"d14a028c2a3a2bc9476102bb288234c415a2b01f828ea62ac5b3e42f", ""},
     57 	{"abd37534c7d9a2efb9465de931cd7055ffdb8879563ae98078d6d6d5", "a"},
     58 	{"db3cda86d4429a1d39c148989566b38f7bda0156296bd364ba2f878b", "ab"},
     59 	{"23097d223405d8228642a477bda255b32aadbce4bda0b3f7e36c9da7", "abc"},
     60 	{"a76654d8e3550e9a2d67a0eeb6c67b220e5885eddd3fde135806e601", "abcd"},
     61 	{"bdd03d560993e675516ba5a50638b6531ac2ac3d5847c61916cfced6", "abcde"},
     62 	{"7043631cb415556a275a4ebecb802c74ee9f6153908e1792a90b6a98", "abcdef"},
     63 	{"d1884e711701ad81abe0c77a3b0ea12e19ba9af64077286c72fc602d", "abcdefg"},
     64 	{"17eb7d40f0356f8598e89eafad5f6c759b1f822975d9c9b737c8a517", "abcdefgh"},
     65 	{"aeb35915346c584db820d2de7af3929ffafef9222a9bcb26516c7334", "abcdefghi"},
     66 	{"d35e1e5af29ddb0d7e154357df4ad9842afee527c689ee547f753188", "abcdefghij"},
     67 	{"19297f1cef7ddc8a7e947f5c5a341e10f7245045e425db67043988d7", "Discard medicine more than two years old."},
     68 	{"0f10c2eb436251f777fbbd125e260d36aecf180411726c7c885f599a", "He who has a shady past knows that nice guys finish last."},
     69 	{"4d1842104919f314cad8a3cd20b3cba7e8ed3e7abed62b57441358f6", "I wouldn't marry him with a ten foot pole."},
     70 	{"a8ba85c6fe0c48fbffc72bbb2f03fcdbc87ae2dc7a56804d1590fb3b", "Free! Free!/A trip/to Mars/for 900/empty jars/Burma Shave"},
     71 	{"5543fbab26e67e8885b1a852d567d1cb8b9bfe42e0899584c50449a9", "The days of the digital watch are numbered.  -Tom Stoppard"},
     72 	{"65ca107390f5da9efa05d28e57b221657edc7e43a9a18fb15b053ddb", "Nepal premier won't resign."},
     73 	{"84953962be366305a9cc9b5cd16ed019edc37ac96c0deb3e12cca116", "For every action there is an equal and opposite government program."},
     74 	{"35a189ce987151dfd00b3577583cc6a74b9869eecf894459cb52038d", "His money is twice tainted: 'taint yours and 'taint mine."},
     75 	{"2fc333713983edfd4ef2c0da6fb6d6415afb94987c91e4069eb063e6", "There is no reason for any individual to have a computer in their home. -Ken Olsen, 1977"},
     76 	{"cbe32d38d577a1b355960a4bc3c659c2dc4670859a19777a875842c4", "It's a tiny change to the code and not completely disgusting. - Bob Manchek"},
     77 	{"a2dc118ce959e027576413a7b440c875cdc8d40df9141d6ef78a57e1", "size:  a.out:  bad magic"},
     78 	{"d10787e24052bcff26dc484787a54ed819e4e4511c54890ee977bf81", "The major problem is with sendmail.  -Mark Horton"},
     79 	{"62efcf16ab8a893acdf2f348aaf06b63039ff1bf55508c830532c9fb", "Give me a rock, paper and scissors and I will move the world.  CCFestoon"},
     80 	{"3e9b7e4613c59f58665104c5fa86c272db5d3a2ff30df5bb194a5c99", "If the enemy is within range, then so are you."},
     81 	{"5999c208b8bdf6d471bb7c359ac5b829e73a8211dff686143a4e7f18", "It's well we cannot hear the screams/That we create in others' dreams."},
     82 	{"3b2d67ff54eabc4ef737b14edf87c64280ef582bcdf2a6d56908b405", "You remind me of a TV show, but that's all right: I watch it anyway."},
     83 	{"d0733595d20e4d3d6b5c565a445814d1bbb2fd08b9a3b8ffb97930c6", "C is as portable as Stonehedge!!"},
     84 	{"43fb8aeed8a833175c9295c1165415f98c866ef08a4922959d673507", "Even if I could be Shakespeare, I think I should still choose to be Faraday. - A. Huxley"},
     85 	{"ec18e66e93afc4fb1604bc2baedbfd20b44c43d76e65c0996d7851c6", "The fugacity of a constituent in a mixture of gases at a given temperature is proportional to its mole fraction.  Lewis-Randall Rule"},
     86 	{"86ed2eaa9c75ba98396e5c9fb2f679ecf0ea2ed1e0ee9ceecb4a9332", "How can you write a big system without C++?  -Paul Glick"},
     87 }
     88 
     89 func TestGolden(t *testing.T) {
     90 	for i := 0; i < len(golden); i++ {
     91 		g := golden[i]
     92 		s := fmt.Sprintf("%x", Sum256([]byte(g.in)))
     93 		if s != g.out {
     94 			t.Fatalf("Sum256 function: sha256(%s) = %s want %s", g.in, s, g.out)
     95 		}
     96 		c := New()
     97 		for j := 0; j < 3; j++ {
     98 			if j < 2 {
     99 				io.WriteString(c, g.in)
    100 			} else {
    101 				io.WriteString(c, g.in[0:len(g.in)/2])
    102 				c.Sum(nil)
    103 				io.WriteString(c, g.in[len(g.in)/2:])
    104 			}
    105 			s := fmt.Sprintf("%x", c.Sum(nil))
    106 			if s != g.out {
    107 				t.Fatalf("sha256[%d](%s) = %s want %s", j, g.in, s, g.out)
    108 			}
    109 			c.Reset()
    110 		}
    111 	}
    112 	for i := 0; i < len(golden224); i++ {
    113 		g := golden224[i]
    114 		s := fmt.Sprintf("%x", Sum224([]byte(g.in)))
    115 		if s != g.out {
    116 			t.Fatalf("Sum224 function: sha224(%s) = %s want %s", g.in, s, g.out)
    117 		}
    118 		c := New224()
    119 		for j := 0; j < 3; j++ {
    120 			if j < 2 {
    121 				io.WriteString(c, g.in)
    122 			} else {
    123 				io.WriteString(c, g.in[0:len(g.in)/2])
    124 				c.Sum(nil)
    125 				io.WriteString(c, g.in[len(g.in)/2:])
    126 			}
    127 			s := fmt.Sprintf("%x", c.Sum(nil))
    128 			if s != g.out {
    129 				t.Fatalf("sha224[%d](%s) = %s want %s", j, g.in, s, g.out)
    130 			}
    131 			c.Reset()
    132 		}
    133 	}
    134 }
    135 
    136 func TestSize(t *testing.T) {
    137 	c := New()
    138 	if got := c.Size(); got != Size {
    139 		t.Errorf("Size = %d; want %d", got, Size)
    140 	}
    141 	c = New224()
    142 	if got := c.Size(); got != Size224 {
    143 		t.Errorf("New224.Size = %d; want %d", got, Size224)
    144 	}
    145 }
    146 
    147 func TestBlockSize(t *testing.T) {
    148 	c := New()
    149 	if got := c.BlockSize(); got != BlockSize {
    150 		t.Errorf("BlockSize = %d want %d", got, BlockSize)
    151 	}
    152 }
    153 
    154 // Tests that blockGeneric (pure Go) and block (in assembly for some architectures) match.
    155 func TestBlockGeneric(t *testing.T) {
    156 	gen, asm := New().(*digest), New().(*digest)
    157 	buf := make([]byte, BlockSize*20) // arbitrary factor
    158 	rand.Read(buf)
    159 	blockGeneric(gen, buf)
    160 	block(asm, buf)
    161 	if *gen != *asm {
    162 		t.Error("block and blockGeneric resulted in different states")
    163 	}
    164 }
    165 
    166 var bench = New()
    167 var buf = make([]byte, 8192)
    168 
    169 func benchmarkSize(b *testing.B, size int) {
    170 	b.SetBytes(int64(size))
    171 	sum := make([]byte, bench.Size())
    172 	for i := 0; i < b.N; i++ {
    173 		bench.Reset()
    174 		bench.Write(buf[:size])
    175 		bench.Sum(sum[:0])
    176 	}
    177 }
    178 
    179 func BenchmarkHash8Bytes(b *testing.B) {
    180 	benchmarkSize(b, 8)
    181 }
    182 
    183 func BenchmarkHash1K(b *testing.B) {
    184 	benchmarkSize(b, 1024)
    185 }
    186 
    187 func BenchmarkHash8K(b *testing.B) {
    188 	benchmarkSize(b, 8192)
    189 }
    190