/
register32.go
53 lines (46 loc) · 972 Bytes
/
register32.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
package sha
import (
"bytes"
"encoding/binary"
"fmt"
"github.com/OhYee/goutils/functional"
"strings"
)
type register32 struct {
v []uint32
l int
}
func newRegister32(iv ...uint32) *register32 {
v := make([]uint32, len(iv))
copy(v, iv)
return ®ister32{
v: v,
l: len(iv),
}
}
func (reg *register32) toString() string {
s := make([]string, reg.l)
for i, v := range reg.v {
s[i] = fmt.Sprintf("%08x", v)
}
return strings.Join(s, " ")
}
func (reg *register32) op(reg2 *register32, f func(uint32, uint32) uint32) {
for i := range reg.v {
reg.v[i] = f(reg.v[i], reg2.v[i])
}
}
func (reg register32) copy() *register32 {
return newRegister32(reg.v...)
}
func (reg register32) toBytes() []byte {
// transfer from []uint32 to []byte
buf := bytes.NewBuffer([]byte{})
temp := make([]byte, 4)
fp.MapUint32(func(data uint32, idx int) uint32 {
binary.BigEndian.PutUint32(temp, data)
buf.Write(temp)
return 0
}, reg.v)
return buf.Bytes()
}