From 76e166730f07043330eab1c111d4cfce0ba9de9a Mon Sep 17 00:00:00 2001
From: "didier.amyot" <didier.amyot@gmail.com>
Date: Thu, 16 Oct 2014 08:03:07 -0700
Subject: [PATCH] change computer

---
 payload/payload.go      | 37 ++++++++++++++++++++++++++++++++-----
 payload/payload_test.go | 23 +++++++++++++++++++++++
 2 files changed, 55 insertions(+), 5 deletions(-)

diff --git a/payload/payload.go b/payload/payload.go
index 7421504..eb7f15c 100644
--- a/payload/payload.go
+++ b/payload/payload.go
@@ -3,11 +3,13 @@ package payload
 import (
 	"crypto"
 	"crypto/cipher"
+	"fmt"
 	"io"
 )
 
-/*
+var _ = fmt.Print
 
+/*
 	type Payload interface {
 		GetHash() []byte
 		GetKey() []byte
@@ -16,14 +18,19 @@ import (
 		GetHashMethod() string
 		GetPayloadData() io.Reader
 	}
-
 */
 
+type BlockMode string
+
+const OFB BlockMode = "ofb"
+
+type BlockModeStream func(block cipher.Block, iv []byte) cipher.Stream
+
 type Payload struct {
 	DataSource io.Reader
 	Hash       crypto.Hash
-	Block      cipher.Block
-	BlockMode  cipher.Stream
+	Block      Block
+	BlockMode  BlockMode
 }
 
 func (this *Payload) GetHashMethod() string {
@@ -52,6 +59,26 @@ func (this *Payload) GetHashMethod() string {
 }
 
 func (this *Payload) GetPayloadData() io.Reader {
-	//  this.BlockMode.
+	//  zip data source
+
+	// create metadata
+	// zip the zipped data and metadata
+
+	// encryptp everything togeter
 	return nil
 }
+
+func (this *Payload) getStream() cipher.Stream {
+	return nil //this.BlockMode
+}
+
+func (this *Payload) encrypt(reader io.Reader) io.Reader {
+	ret, writer := io.Pipe()
+	go func() {
+		cipherWriter := &cipher.StreamWriter{S: this.getStream(), W: writer}
+		io.Copy(cipherWriter, reader)
+		cipherWriter.Close()
+	}()
+	return ret
+
+}
diff --git a/payload/payload_test.go b/payload/payload_test.go
index 0378e3d..885a907 100644
--- a/payload/payload_test.go
+++ b/payload/payload_test.go
@@ -1,7 +1,9 @@
 package payload
 
 import (
+	"bytes"
 	"crypto"
+	"io/ioutil"
 	"testing"
 )
 
@@ -23,5 +25,26 @@ func TestPayloadGetHashMethod(t *testing.T) {
 			t.Error("bad hash name.  supposed to be ", name, " but received", returnedName)
 		}
 	}
+}
+
+func TestGetStream(t *testing.T) {
+	p := new(Payload)
+	if s := p.getStream(); s != nil {
+		t.Fail()
+	}
+}
+
+func TestEncrypt(t *testing.T) {
+	if true {
+		return
+	}
+	p := new(Payload)
+	data := bytes.NewBufferString("nice string")
+	reader := p.encrypt(data)
+	if outputdata, err := ioutil.ReadAll(reader); err != nil {
+		t.Error(err)
+	} else if string(outputdata) != "nice string" {
+		t.Fail()
+	}
 
 }