You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
1.3 KiB
60 lines
1.3 KiB
10 years ago
|
// +build amd64
|
||
|
|
||
|
package sha256
|
||
|
|
||
|
// #include <stdint.h>
|
||
|
// void sha256_transform_avx (uint8_t *input_data, uint32_t digest[8], uint64_t num_blks);
|
||
|
// void sha256_transform_ssse3 (uint8_t *input_data, uint32_t digest[8], uint64_t num_blks);
|
||
|
// void sha256_transform_rorx (uint8_t *input_data, uint32_t digest[8], uint64_t num_blks);
|
||
|
// #define SHA256_DIGEST_SIZE 32
|
||
|
// #define SHA256_BLOCK_SIZE 64
|
||
|
// #define SHA256_H0 0x6a09e667UL
|
||
|
// #define SHA256_H1 0xbb67ae85UL
|
||
|
// #define SHA256_H2 0x3c6ef372UL
|
||
|
// #define SHA256_H3 0xa54ff53aUL
|
||
|
// #define SHA256_H4 0x510e527fUL
|
||
|
// #define SHA256_H5 0x9b05688cUL
|
||
|
// #define SHA256_H6 0x1f83d9abUL
|
||
|
// #define SHA256_H7 0x5be0cd19UL
|
||
|
import "C"
|
||
10 years ago
|
import (
|
||
|
gosha256 "crypto/sha256"
|
||
|
"io"
|
||
|
)
|
||
10 years ago
|
|
||
|
/*
|
||
|
func Sha256(buffer []byte) ([]uint32, error) {
|
||
|
|
||
|
if cpu.HasSSE41() {
|
||
|
C.sha256_transform_ssse3()
|
||
|
return 0, nil
|
||
|
}
|
||
|
|
||
|
if cpu.HasAVX() {
|
||
|
C.sha256_transform_avx()
|
||
|
return 0, nil
|
||
|
}
|
||
|
|
||
|
if cpu.HasAVX2() {
|
||
|
C.sha256_transform_rorx()
|
||
|
return 0, nil
|
||
|
}
|
||
|
}
|
||
|
*/
|
||
10 years ago
|
|
||
|
func Sum(reader io.Reader) ([]byte, error) {
|
||
|
hash := gosha256.New()
|
||
|
var err error
|
||
|
for err == nil {
|
||
|
length := 0
|
||
|
byteBuffer := make([]byte, 1024*1024)
|
||
|
length, err = reader.Read(byteBuffer)
|
||
|
byteBuffer = byteBuffer[0:length]
|
||
|
hash.Write(byteBuffer)
|
||
|
}
|
||
|
if err != io.EOF {
|
||
|
return nil, err
|
||
|
}
|
||
|
return hash.Sum(nil), nil
|
||
|
}
|