-
Notifications
You must be signed in to change notification settings - Fork 95
/
canonical_text.go
111 lines (92 loc) · 2.3 KB
/
canonical_text.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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
// Copyright 2011 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 openpgp
import (
"bytes"
"hash"
"io"
)
// NewCanonicalTextHash reformats text written to it into the canonical
// form and then applies the hash h. See RFC 4880, section 5.2.1.
func NewCanonicalTextHash(h hash.Hash) hash.Hash {
return &canonicalTextHash{h, 0}
}
// NewCanonicalTextWriteCloser reformats text written to it into the canonical
// form. See RFC 4880, section 5.2.1.
func NewCanonicalTextWriteCloser(w io.WriteCloser) io.WriteCloser {
return &canonicalTextWriteCloser{w, 0}
}
// NewCanonicalTextReader reformats text read from it into the canonical
// form. See RFC 4880, section 5.2.1.
func NewCanonicalTextReader(r io.Reader) io.Reader {
return &canonicalTextReader{r, bytes.NewBuffer(nil), 0}
}
type canonicalTextHash struct {
h hash.Hash
s int
}
var newline = []byte{'\r', '\n'}
func writeCanonical(cw io.Writer, buf []byte, s *int) (int, error) {
start := 0
for i, c := range buf {
switch *s {
case 0:
if c == '\r' {
*s = 1
} else if c == '\n' {
cw.Write(buf[start:i])
cw.Write(newline)
start = i + 1
}
case 1:
*s = 0
}
}
cw.Write(buf[start:])
return len(buf), nil
}
func (cth *canonicalTextHash) Write(buf []byte) (int, error) {
return writeCanonical(cth.h, buf, &cth.s)
}
func (cth *canonicalTextHash) Sum(in []byte) []byte {
return cth.h.Sum(in)
}
func (cth *canonicalTextHash) Reset() {
cth.h.Reset()
cth.s = 0
}
func (cth *canonicalTextHash) Size() int {
return cth.h.Size()
}
func (cth *canonicalTextHash) BlockSize() int {
return cth.h.BlockSize()
}
type canonicalTextWriteCloser struct {
w io.WriteCloser
s int
}
func (tw *canonicalTextWriteCloser) Write(buf []byte) (int, error) {
return writeCanonical(tw.w, buf, &tw.s)
}
func (tw *canonicalTextWriteCloser) Close() error {
return tw.w.Close()
}
type canonicalTextReader struct {
r io.Reader
buffer *bytes.Buffer
s int
}
func (tr *canonicalTextReader) Read(buf []byte) (int, error) {
if tr.buffer.Len() > 0 {
return tr.buffer.Read(buf)
}
n, err := tr.r.Read(buf)
if err != nil {
return n, err
}
if _, err = writeCanonical(tr.buffer, buf[:n], &tr.s); err != nil {
return 0, err
}
return tr.buffer.Read(buf)
}