From 9897c641867880417ff5971b1c9e44843f52d879 Mon Sep 17 00:00:00 2001
From: zelig <viktor.tron@gmail.com>
Date: Tue, 13 Jan 2015 01:25:42 +0000
Subject: [PATCH] fix pos update in recursive join, put size instead of
 treeSize into the non-leaf chunks

---
 bzz/chunker.go      | 13 ++++++++-----
 bzz/chunker_test.go |  2 +-
 2 files changed, 9 insertions(+), 6 deletions(-)

diff --git a/bzz/chunker.go b/bzz/chunker.go
index a160747202..5b38e16a18 100644
--- a/bzz/chunker.go
+++ b/bzz/chunker.go
@@ -252,11 +252,11 @@ func (self *TreeChunker) split(depth int, treeSize int64, key Key, data SectionR
 		childrenWg.Wait()
 		// now we got the hashes in the chunk, then hash the chunk
 		chunkReader := NewChunkReaderFromBytes(chunk) // bytes.Reader almost implements SectionReader
-		hash = self.Hash(treeSize, chunkReader)
+		hash = self.Hash(size, chunkReader)
 		newChunk = &Chunk{
 			Key:  hash,
 			Data: chunkReader,
-			Size: treeSize,
+			Size: size,
 		}
 	}
 	// send off new chunk to storage
@@ -329,7 +329,7 @@ func (self *TreeChunker) Join(key Key, data SectionWriter) (chunkC chan *Chunk,
 			depth++
 		}
 		// launch recursive call on root chunk
-		self.join(depth, treeSize, chunk, data, chunkC, rerrC, wg, quitC)
+		self.join(depth, treeSize/self.Branches, chunk, data, chunkC, rerrC, wg, quitC)
 	}()
 
 	// waits for all the processes to finish and signals by closing internal rerrc
@@ -365,7 +365,10 @@ func (self *TreeChunker) join(depth int, treeSize int64, chunk *Chunk, data Sect
 	case <-chunk.C: // bells are ringing, data have been delivered
 		dpaLogger.Debugf("received chunk data: %v", chunk)
 		switch {
-		case chunk.Size <= treeSize && depth == 0:
+		case depth == 0:
+			if chunk.Size > self.chunkSize {
+				panic("oops")
+			}
 			dpaLogger.Debugf("reading into data")
 			// we received a chunk for a leaf node representing actual content
 			if _, err := data.ReadFrom(chunk.Data); err != nil {
@@ -394,7 +397,7 @@ func (self *TreeChunker) join(depth int, treeSize int64, chunk *Chunk, data Sect
 				// submit request
 				chunkC <- subtree
 				i++
-				pos += subtree.Size
+				pos += treeSize
 			}
 
 		}
diff --git a/bzz/chunker_test.go b/bzz/chunker_test.go
index 3dee423d63..ec350dd516 100644
--- a/bzz/chunker_test.go
+++ b/bzz/chunker_test.go
@@ -198,7 +198,7 @@ func TestChunker1(t *testing.T) {
 	chunker.Init()
 	tester := &chunkerTester{}
 	key, input := tester.Split(chunker, 70)
-	tester.checkChunks(t, 2)
+	tester.checkChunks(t, 3)
 	t.Logf("chunks: %v", tester.chunks)
 	output := tester.Join(t, chunker, key)
 	if bytes.Compare(output, input) != 0 {