|  | // Copyright 2009 The Go Authors. All rights reserved. | 
|  | // Use of this source code is governed by a BSD-style | 
|  | // license that can be found in the LICENSE file. | 
|  |  | 
|  | // Package sha1 implements the SHA-1 hash algorithm as defined in RFC 3174. | 
|  | // | 
|  | // SHA-1 is cryptographically broken and should not be used for secure | 
|  | // applications. | 
|  | package sha1 | 
|  |  | 
|  | import ( | 
|  | "crypto" | 
|  | "errors" | 
|  | "hash" | 
|  | ) | 
|  |  | 
|  | func init() { | 
|  | crypto.RegisterHash(crypto.SHA1, New) | 
|  | } | 
|  |  | 
|  | // The size of a SHA-1 checksum in bytes. | 
|  | const Size = 20 | 
|  |  | 
|  | // The blocksize of SHA-1 in bytes. | 
|  | const BlockSize = 64 | 
|  |  | 
|  | const ( | 
|  | chunk = 64 | 
|  | init0 = 0x67452301 | 
|  | init1 = 0xEFCDAB89 | 
|  | init2 = 0x98BADCFE | 
|  | init3 = 0x10325476 | 
|  | init4 = 0xC3D2E1F0 | 
|  | ) | 
|  |  | 
|  | // digest represents the partial evaluation of a checksum. | 
|  | type digest struct { | 
|  | h   [5]uint32 | 
|  | x   [chunk]byte | 
|  | nx  int | 
|  | len uint64 | 
|  | } | 
|  |  | 
|  | const ( | 
|  | magic         = "sha\x01" | 
|  | marshaledSize = len(magic) + 5*4 + chunk + 8 | 
|  | ) | 
|  |  | 
|  | func (d *digest) MarshalBinary() ([]byte, error) { | 
|  | b := make([]byte, 0, marshaledSize) | 
|  | b = append(b, magic...) | 
|  | b = appendUint32(b, d.h[0]) | 
|  | b = appendUint32(b, d.h[1]) | 
|  | b = appendUint32(b, d.h[2]) | 
|  | b = appendUint32(b, d.h[3]) | 
|  | b = appendUint32(b, d.h[4]) | 
|  | b = append(b, d.x[:d.nx]...) | 
|  | b = b[:len(b)+len(d.x)-int(d.nx)] // already zero | 
|  | b = appendUint64(b, d.len) | 
|  | return b, nil | 
|  | } | 
|  |  | 
|  | func (d *digest) UnmarshalBinary(b []byte) error { | 
|  | if len(b) < len(magic) || string(b[:len(magic)]) != magic { | 
|  | return errors.New("crypto/sha1: invalid hash state identifier") | 
|  | } | 
|  | if len(b) != marshaledSize { | 
|  | return errors.New("crypto/sha1: invalid hash state size") | 
|  | } | 
|  | b = b[len(magic):] | 
|  | b, d.h[0] = consumeUint32(b) | 
|  | b, d.h[1] = consumeUint32(b) | 
|  | b, d.h[2] = consumeUint32(b) | 
|  | b, d.h[3] = consumeUint32(b) | 
|  | b, d.h[4] = consumeUint32(b) | 
|  | b = b[copy(d.x[:], b):] | 
|  | b, d.len = consumeUint64(b) | 
|  | d.nx = int(d.len) % chunk | 
|  | return nil | 
|  | } | 
|  |  | 
|  | func appendUint64(b []byte, x uint64) []byte { | 
|  | var a [8]byte | 
|  | putUint64(a[:], x) | 
|  | return append(b, a[:]...) | 
|  | } | 
|  |  | 
|  | func appendUint32(b []byte, x uint32) []byte { | 
|  | var a [4]byte | 
|  | putUint32(a[:], x) | 
|  | return append(b, a[:]...) | 
|  | } | 
|  |  | 
|  | func consumeUint64(b []byte) ([]byte, uint64) { | 
|  | _ = b[7] | 
|  | x := uint64(b[7]) | uint64(b[6])<<8 | uint64(b[5])<<16 | uint64(b[4])<<24 | | 
|  | uint64(b[3])<<32 | uint64(b[2])<<40 | uint64(b[1])<<48 | uint64(b[0])<<56 | 
|  | return b[8:], x | 
|  | } | 
|  |  | 
|  | func consumeUint32(b []byte) ([]byte, uint32) { | 
|  | _ = b[3] | 
|  | x := uint32(b[3]) | uint32(b[2])<<8 | uint32(b[1])<<16 | uint32(b[0])<<24 | 
|  | return b[4:], x | 
|  | } | 
|  |  | 
|  | func (d *digest) Reset() { | 
|  | d.h[0] = init0 | 
|  | d.h[1] = init1 | 
|  | d.h[2] = init2 | 
|  | d.h[3] = init3 | 
|  | d.h[4] = init4 | 
|  | d.nx = 0 | 
|  | d.len = 0 | 
|  | } | 
|  |  | 
|  | // New returns a new hash.Hash computing the SHA1 checksum. The Hash also | 
|  | // implements encoding.BinaryMarshaler and encoding.BinaryUnmarshaler to | 
|  | // marshal and unmarshal the internal state of the hash. | 
|  | func New() hash.Hash { | 
|  | d := new(digest) | 
|  | d.Reset() | 
|  | return d | 
|  | } | 
|  |  | 
|  | func (d *digest) Size() int { return Size } | 
|  |  | 
|  | func (d *digest) BlockSize() int { return BlockSize } | 
|  |  | 
|  | func (d *digest) Write(p []byte) (nn int, err error) { | 
|  | nn = len(p) | 
|  | d.len += uint64(nn) | 
|  | if d.nx > 0 { | 
|  | n := copy(d.x[d.nx:], p) | 
|  | d.nx += n | 
|  | if d.nx == chunk { | 
|  | block(d, d.x[:]) | 
|  | d.nx = 0 | 
|  | } | 
|  | p = p[n:] | 
|  | } | 
|  | if len(p) >= chunk { | 
|  | n := len(p) &^ (chunk - 1) | 
|  | block(d, p[:n]) | 
|  | p = p[n:] | 
|  | } | 
|  | if len(p) > 0 { | 
|  | d.nx = copy(d.x[:], p) | 
|  | } | 
|  | return | 
|  | } | 
|  |  | 
|  | func (d *digest) Sum(in []byte) []byte { | 
|  | // Make a copy of d so that caller can keep writing and summing. | 
|  | d0 := *d | 
|  | hash := d0.checkSum() | 
|  | return append(in, hash[:]...) | 
|  | } | 
|  |  | 
|  | func (d *digest) checkSum() [Size]byte { | 
|  | len := d.len | 
|  | // Padding.  Add a 1 bit and 0 bits until 56 bytes mod 64. | 
|  | var tmp [64]byte | 
|  | tmp[0] = 0x80 | 
|  | if len%64 < 56 { | 
|  | d.Write(tmp[0 : 56-len%64]) | 
|  | } else { | 
|  | d.Write(tmp[0 : 64+56-len%64]) | 
|  | } | 
|  |  | 
|  | // Length in bits. | 
|  | len <<= 3 | 
|  | putUint64(tmp[:], len) | 
|  | d.Write(tmp[0:8]) | 
|  |  | 
|  | if d.nx != 0 { | 
|  | panic("d.nx != 0") | 
|  | } | 
|  |  | 
|  | var digest [Size]byte | 
|  |  | 
|  | putUint32(digest[0:], d.h[0]) | 
|  | putUint32(digest[4:], d.h[1]) | 
|  | putUint32(digest[8:], d.h[2]) | 
|  | putUint32(digest[12:], d.h[3]) | 
|  | putUint32(digest[16:], d.h[4]) | 
|  |  | 
|  | return digest | 
|  | } | 
|  |  | 
|  | // ConstantTimeSum computes the same result of Sum() but in constant time | 
|  | func (d *digest) ConstantTimeSum(in []byte) []byte { | 
|  | d0 := *d | 
|  | hash := d0.constSum() | 
|  | return append(in, hash[:]...) | 
|  | } | 
|  |  | 
|  | func (d *digest) constSum() [Size]byte { | 
|  | var length [8]byte | 
|  | l := d.len << 3 | 
|  | for i := uint(0); i < 8; i++ { | 
|  | length[i] = byte(l >> (56 - 8*i)) | 
|  | } | 
|  |  | 
|  | nx := byte(d.nx) | 
|  | t := nx - 56                 // if nx < 56 then the MSB of t is one | 
|  | mask1b := byte(int8(t) >> 7) // mask1b is 0xFF iff one block is enough | 
|  |  | 
|  | separator := byte(0x80) // gets reset to 0x00 once used | 
|  | for i := byte(0); i < chunk; i++ { | 
|  | mask := byte(int8(i-nx) >> 7) // 0x00 after the end of data | 
|  |  | 
|  | // if we reached the end of the data, replace with 0x80 or 0x00 | 
|  | d.x[i] = (^mask & separator) | (mask & d.x[i]) | 
|  |  | 
|  | // zero the separator once used | 
|  | separator &= mask | 
|  |  | 
|  | if i >= 56 { | 
|  | // we might have to write the length here if all fit in one block | 
|  | d.x[i] |= mask1b & length[i-56] | 
|  | } | 
|  | } | 
|  |  | 
|  | // compress, and only keep the digest if all fit in one block | 
|  | block(d, d.x[:]) | 
|  |  | 
|  | var digest [Size]byte | 
|  | for i, s := range d.h { | 
|  | digest[i*4] = mask1b & byte(s>>24) | 
|  | digest[i*4+1] = mask1b & byte(s>>16) | 
|  | digest[i*4+2] = mask1b & byte(s>>8) | 
|  | digest[i*4+3] = mask1b & byte(s) | 
|  | } | 
|  |  | 
|  | for i := byte(0); i < chunk; i++ { | 
|  | // second block, it's always past the end of data, might start with 0x80 | 
|  | if i < 56 { | 
|  | d.x[i] = separator | 
|  | separator = 0 | 
|  | } else { | 
|  | d.x[i] = length[i-56] | 
|  | } | 
|  | } | 
|  |  | 
|  | // compress, and only keep the digest if we actually needed the second block | 
|  | block(d, d.x[:]) | 
|  |  | 
|  | for i, s := range d.h { | 
|  | digest[i*4] |= ^mask1b & byte(s>>24) | 
|  | digest[i*4+1] |= ^mask1b & byte(s>>16) | 
|  | digest[i*4+2] |= ^mask1b & byte(s>>8) | 
|  | digest[i*4+3] |= ^mask1b & byte(s) | 
|  | } | 
|  |  | 
|  | return digest | 
|  | } | 
|  |  | 
|  | // Sum returns the SHA-1 checksum of the data. | 
|  | func Sum(data []byte) [Size]byte { | 
|  | var d digest | 
|  | d.Reset() | 
|  | d.Write(data) | 
|  | return d.checkSum() | 
|  | } | 
|  |  | 
|  | func putUint64(x []byte, s uint64) { | 
|  | _ = x[7] | 
|  | x[0] = byte(s >> 56) | 
|  | x[1] = byte(s >> 48) | 
|  | x[2] = byte(s >> 40) | 
|  | x[3] = byte(s >> 32) | 
|  | x[4] = byte(s >> 24) | 
|  | x[5] = byte(s >> 16) | 
|  | x[6] = byte(s >> 8) | 
|  | x[7] = byte(s) | 
|  | } | 
|  |  | 
|  | func putUint32(x []byte, s uint32) { | 
|  | _ = x[3] | 
|  | x[0] = byte(s >> 24) | 
|  | x[1] = byte(s >> 16) | 
|  | x[2] = byte(s >> 8) | 
|  | x[3] = byte(s) | 
|  | } |