crypto/crypto.go

   1 // Copyright 2014 The go-ethereum Authors
   2 // This file is part of the go-ethereum library.
   3 //
   4 // The go-ethereum library is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU Lesser General Public License as published by
   6 // the Free Software Foundation, either version 3 of the License, or
   7 // (at your option) any later version.
   8 //
   9 // The go-ethereum library is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12 // GNU Lesser General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Lesser General Public License
  15 // along with the go-ethereum library. If not, see <http://www.gnu.org/licenses/>.
  16
  17 package crypto
  18
  19 import (
  20         "fmt"
  21         //"io"
  22         //"io/ioutil"
  23         //"math/big"
  24         //"os"
  25
  26         //"encoding/hex"
  27         //"errors"
  28
  29         "github.com/bytom/common"
  30         "golang.org/x/crypto/ripemd160"
  31         "golang.org/x/crypto/sha3"
  32 )
  33
  34 func Sha256(data ...[]byte) []byte {
  35         d := sha3.New256()
  36         for _, b := range data {
  37                 d.Write(b)
  38         }
  39         return d.Sum(nil)
  40 }
  41
  42 func Sha256Hash(data ...[]byte) (h common.Hash) {
  43         d := sha3.New256()
  44         for _, b := range data {
  45                 d.Write(b)
  46         }
  47         d.Sum(h[:0])
  48         return h
  49 }
  50
  51 func Sha3(data ...[]byte) []byte          { return Sha256(data...) }
  52 func Sha3Hash(data ...[]byte) common.Hash { return Sha256Hash(data...) }
  53
  54 func Ripemd160(data []byte) []byte {
  55         ripemd := ripemd160.New()
  56         ripemd.Write(data)
  57
  58         return ripemd.Sum(nil)
  59 }
  60
  61 func PubkeyToAddress(pubBytes []byte) common.Address {
  62         address, _ := common.AddressEncode("bm", 1, toInt(Ripemd160(Sha3(pubBytes))))
  63         fmt.Printf(address)
  64         return common.StringToAddress(address)
  65 }
  66
  67 func AddressToPubkey(addr common.Address) (int, []byte, error) {
  68         ver, data, err := common.AddressDecode("bm", addr.Str())
  69         return ver, toBytes(data), err
  70 }
  71
  72 func zeroBytes(bytes []byte) {
  73         for i := range bytes {
  74                 bytes[i] = 0
  75         }
  76 }
  77
  78 func toInt(bytes []byte) []int {
  79         ints := make([]int, len(bytes))
  80         for i := range bytes {
  81                 ints[i] = int(bytes[i])
  82         }
  83         return ints
  84 }
  85
  86 func toBytes(ints []int) []byte {
  87         bytes := make([]byte, len(ints))
  88         for i := range ints {
  89                 bytes[i] = byte(ints[i])
  90         }
  91         return bytes
  92 }