在go语言中实现生成文件的MD5校验和、生成文件的校验和

Go  /  管理员 发布于 2年前   672

每天都有数十亿个文件在互联网上下载和传输。生成校验和是当今程序开发的重要组成部分，以确保文件不被篡改。

如何用 Go 为示例文件生成 MD5 校验和。

 package main
   import (
       "fmt"
       "io"
       "os"
       "math"
       "crypto/md5"
   )
   const filechunk = 8192    // we settle for 8KB
   func main() {
       file, err := os.Open("utf8.txt")
       if err != nil {
          panic(err.Error())
       }
      defer file.Close()
      // calculate the file size  
      info, _ := file.Stat()
      filesize := info.Size()  
      blocks := uint64(math.Ceil(float64(filesize) / float64(filechunk)))
      hash := md5.New()
      for i := uint64(0); i < blocks; i++ {
          blocksize := int(math.Min(filechunk, float64(filesize-int64(i*filechunk))))
          buf := make([] byte, blocksize)
          file.Read(buf)
          io.WriteString(hash, string(buf))   // append into the hash
      }
      fmt.Printf("%s checksum is %x\n",file.Name(), hash.Sum(nil)) 
 }

输出 : 

>go run checksum.go
utf8.txt checksum is c3882b3e45e88ba87617f44ce1e7611d

以上是md5加密，当然你也可以通过更改加密包，用 SHA1 或 SHA256 轻松替换它。

使用了 io.Copy 函数，而不是将文件分割成块。

io.Copy 函数会读取文件内容，直到到达 EOF，

它可以用来将文件数据复制到 MD5 哈希值中。

package main
 import (
 "crypto/md5"
 "fmt"
 "io"
 "os"
 )
 func main() {
 file, err := os.Open("utf8.txt")
 if err != nil {
 panic(err)
 }
 defer file.Close()
 hash := md5.New()
 _, err = io.Copy(hash, file)
 if err != nil {
 panic(err)
 }
 fmt.Printf("%s MD5 checksum is %x \n", file.Name(), hash.Sum(nil))
 }

输出:

utf8.txt MD5 checksum is 5af1ca1d92340d72a29c194d3f4096e0