Add a streaming api to KeyRing and SessionKey (#131)
* barebone streaming functionality * encryption needs to return a writecloser * added eof check * workaround for reader problem with copies * separate mobile wrappers from main api * add a clone in the read result to avoid memory corruption * refactor to reuse code, and fix verification * have to give the verify key at the start of the decryption * enfore readAll before signature verification * streaming api for SessionKey * add split message stream apis * name interface params * fix streaming api so it's supported by go-mobile * hide internal writeCloser * fix nil access * added detached sigs methods * started unit testing * unit testing and fixed a bug where key and data packets where inverted * remove unecessary error wrapping * figured out closing order and error handling * add GC calls to mobile writer and reader * remove debugging values and arrays * writer with builtin sha256 * unit testing the mobile helpers * comments and linting * Typo in error Co-authored-by: wussler <aron@wussler.it> * Add GetKeyPacket doc Co-authored-by: wussler <aron@wussler.it> * Add rfc reference in comments Co-authored-by: wussler <aron@wussler.it> * small improvements * add compatibility tests with normal methods * remove unecessary copies in the tests * update go-crypto to the merged changes commit * update comments of core internal functions * remove unused nolint comment * group message metadata in a struct * fix comments * change default values for metadata * change the mobile reader wrapper to fit the behavior of java * remove gc calls in the wrappers to avoid performance penalties * bring back the former Go2MobileReader to be used for ios * Update crypto/keyring_streaming.go Co-authored-by: wussler <aron@wussler.it> * return an error when verifying an embedded sig with no keyring * Update crypto/sessionkey_streaming.go Co-authored-by: wussler <aron@wussler.it> * linter error * update changelog * update changelog Co-authored-by: wussler <aron@wussler.it>
This commit is contained in:
marinthiercelin
GitHub
parent
7380f7391f
commit
c46ed8ed9e
11 changed files with 1718 additions and 97 deletions
182
helper/mobile_stream_test.go
Normal file
182
helper/mobile_stream_test.go
Normal file
|
|
@ -0,0 +1,182 @@
|
|||
package helper
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/sha256"
|
||||
"errors"
|
||||
"io"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func cloneTestData() (a, b []byte) {
|
||||
a = []byte("Hello World!")
|
||||
b = clone(a)
|
||||
return a, b
|
||||
}
|
||||
func Test_clone(t *testing.T) {
|
||||
if a, b := cloneTestData(); !bytes.Equal(a, b) {
|
||||
t.Fatalf("expected %x, got %x", a, b)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMobile2GoWriter(t *testing.T) {
|
||||
testData := []byte("Hello World!")
|
||||
outBuf := &bytes.Buffer{}
|
||||
reader := bytes.NewReader(testData)
|
||||
writer := NewMobile2GoWriter(outBuf)
|
||||
bufSize := 2
|
||||
writeBuf := make([]byte, bufSize)
|
||||
reachedEnd := false
|
||||
for !reachedEnd {
|
||||
n, err := reader.Read(writeBuf)
|
||||
if err != nil {
|
||||
if errors.Is(err, io.EOF) {
|
||||
reachedEnd = true
|
||||
} else {
|
||||
t.Fatal("Expected no error while reading, got:", err)
|
||||
}
|
||||
}
|
||||
writtenTotal := 0
|
||||
for writtenTotal < n {
|
||||
written, err := writer.Write(writeBuf[writtenTotal:n])
|
||||
if err != nil {
|
||||
t.Fatal("Expected no error while writing, got:", err)
|
||||
}
|
||||
writtenTotal += written
|
||||
}
|
||||
}
|
||||
if writtenData := outBuf.Bytes(); !bytes.Equal(testData, writtenData) {
|
||||
t.Fatalf("expected %x, got %x", testData, writtenData)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMobile2GoWriterWithSHA256(t *testing.T) {
|
||||
testData := []byte("Hello World!")
|
||||
testHash := sha256.Sum256(testData)
|
||||
outBuf := &bytes.Buffer{}
|
||||
reader := bytes.NewReader(testData)
|
||||
writer := NewMobile2GoWriterWithSHA256(outBuf)
|
||||
bufSize := 2
|
||||
writeBuf := make([]byte, bufSize)
|
||||
reachedEnd := false
|
||||
for !reachedEnd {
|
||||
n, err := reader.Read(writeBuf)
|
||||
if err != nil {
|
||||
if errors.Is(err, io.EOF) {
|
||||
reachedEnd = true
|
||||
} else {
|
||||
t.Fatal("Expected no error while reading, got:", err)
|
||||
}
|
||||
}
|
||||
writtenTotal := 0
|
||||
for writtenTotal < n {
|
||||
written, err := writer.Write(writeBuf[writtenTotal:n])
|
||||
if err != nil {
|
||||
t.Fatal("Expected no error while writing, got:", err)
|
||||
}
|
||||
writtenTotal += written
|
||||
}
|
||||
}
|
||||
if writtenData := outBuf.Bytes(); !bytes.Equal(testData, writtenData) {
|
||||
t.Fatalf("expected data to be %x, got %x", testData, writtenData)
|
||||
}
|
||||
|
||||
if writtenHash := writer.GetSHA256(); !bytes.Equal(testHash[:], writtenHash) {
|
||||
t.Fatalf("expected has to be %x, got %x", testHash, writtenHash)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGo2AndroidReader(t *testing.T) {
|
||||
testData := []byte("Hello World!")
|
||||
reader := NewGo2AndroidReader(bytes.NewReader(testData))
|
||||
var readData []byte
|
||||
bufSize := 2
|
||||
buffer := make([]byte, bufSize)
|
||||
reachedEnd := false
|
||||
for !reachedEnd {
|
||||
n, err := reader.Read(buffer)
|
||||
if err != nil {
|
||||
t.Fatal("Expected no error while reading, got:", err)
|
||||
}
|
||||
reachedEnd = n < 0
|
||||
if n > 0 {
|
||||
readData = append(readData, buffer[:n]...)
|
||||
}
|
||||
}
|
||||
if !bytes.Equal(testData, readData) {
|
||||
t.Fatalf("expected data to be %x, got %x", testData, readData)
|
||||
}
|
||||
}
|
||||
|
||||
func TestGo2IOSReader(t *testing.T) {
|
||||
testData := []byte("Hello World!")
|
||||
reader := NewGo2IOSReader(bytes.NewReader(testData))
|
||||
var readData []byte
|
||||
bufSize := 2
|
||||
reachedEnd := false
|
||||
for !reachedEnd {
|
||||
res, err := reader.Read(bufSize)
|
||||
if err != nil {
|
||||
t.Fatal("Expected no error while reading, got:", err)
|
||||
}
|
||||
n := res.N
|
||||
reachedEnd = res.IsEOF
|
||||
if n > 0 {
|
||||
readData = append(readData, res.Data[:n]...)
|
||||
}
|
||||
}
|
||||
if !bytes.Equal(testData, readData) {
|
||||
t.Fatalf("expected data to be %x, got %x", testData, readData)
|
||||
}
|
||||
}
|
||||
|
||||
type testMobileReader struct {
|
||||
reader io.Reader
|
||||
returnError bool
|
||||
}
|
||||
|
||||
func (r *testMobileReader) Read(max int) (*MobileReadResult, error) {
|
||||
if r.returnError {
|
||||
return nil, errors.New("gopenpgp: test - forced error while reading")
|
||||
}
|
||||
buf := make([]byte, max)
|
||||
n, err := r.reader.Read(buf)
|
||||
eof := false
|
||||
if err != nil {
|
||||
if errors.Is(err, io.EOF) {
|
||||
eof = true
|
||||
} else {
|
||||
return nil, errors.New("gopenpgp: test - error while reading")
|
||||
}
|
||||
}
|
||||
return NewMobileReadResult(n, eof, buf[:n]), nil
|
||||
}
|
||||
|
||||
func TestMobile2GoReader(t *testing.T) {
|
||||
testData := []byte("Hello World!")
|
||||
reader := NewMobile2GoReader(&testMobileReader{bytes.NewReader(testData), false})
|
||||
var readData []byte
|
||||
bufSize := 2
|
||||
readBuf := make([]byte, bufSize)
|
||||
reachedEnd := false
|
||||
for !reachedEnd {
|
||||
n, err := reader.Read(readBuf)
|
||||
if err != nil {
|
||||
if errors.Is(err, io.EOF) {
|
||||
reachedEnd = true
|
||||
} else {
|
||||
t.Fatal("Expected no error while reading, got:", err)
|
||||
}
|
||||
}
|
||||
if n > 0 {
|
||||
readData = append(readData, readBuf[:n]...)
|
||||
}
|
||||
}
|
||||
if !bytes.Equal(testData, readData) {
|
||||
t.Fatalf("expected data to be %x, got %x", testData, readData)
|
||||
}
|
||||
readerErr := NewMobile2GoReader(&testMobileReader{bytes.NewReader(testData), true})
|
||||
if _, err := readerErr.Read(readBuf); err == nil {
|
||||
t.Fatal("expected an error while reading, got nil")
|
||||
}
|
||||
}
|
||||
Loading…
Add table
Add a link
Reference in a new issue