summaryrefslogtreecommitdiff
path: root/xdelta3/go/src/xdelta/rstream.go
blob: 99c3d1783cbc8189833d2a647347779ecf129be5 (plain)
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
package xdelta


import (
	"io"
	"math/rand"
)

const (
	blocksize = 1<<17
)

func (t *TestGroup) WriteRstreams(desc string, seed, offset, len int64,
	src, tgt io.WriteCloser) {
	t.Go("src-write:"+desc, func (g *Goroutine) {
		writeOne(g, seed, 0, len, tgt, false)
	})
	t.Go("tgt-write:"+desc, func (g *Goroutine) {
		writeOne(g, seed, offset, len, src, true)
	})
}

func writeOne(g *Goroutine, seed, offset, len int64, stream io.WriteCloser, readall bool) {
	if !readall {
		// Allow the source-read to fail or block until the process terminates.
		// This behavior is reserved for the decoder, which is not required to
		// read the entire source.
		g.OK()
	}
	if offset != 0 {
		// Fill with other random data until the offset
		if err := writeRand(g, rand.New(rand.NewSource(^seed)), offset, stream); err != nil {
			g.Panic(err)
		}
	}
	if err := writeRand(g, rand.New(rand.NewSource(seed)),
		len - offset, stream); err != nil {
		g.Panic(err)
	}
	if err := stream.Close(); err != nil {
		g.Panic(err)
	}
	g.OK()
}

func writeRand(g *Goroutine, r *rand.Rand, len int64, s io.Writer) error {
	blk := make([]byte, blocksize)
	for len > 0 {
		fillRand(r, blk)
		c := blocksize
		if len < blocksize {
			c = int(len)
		}
		if _, err := s.Write(blk[0:c]); err != nil {
			return err
		}
		len -= int64(c)
	}
	return nil
}

func fillRand(r *rand.Rand, blk []byte) {
	for p := 0; p < len(blk); {
		v := r.Int63()
		for i := 7; i != 0 && p < len(blk); i-- {
			blk[p] = byte(v)
			p++
			v >>= 8
		}
	}
}