syncthing/integration/genfiles.go

// Copyright (C) 2014 Jakob Borg and Contributors (see the CONTRIBUTORS file).
// All rights reserved. Use of this source code is governed by an MIT-style
// license that can be found in the LICENSE file.

// +build ignore

package main

import (
	"crypto/rand"
	"flag"
	"fmt"
	"io/ioutil"
	mr "math/rand"
	"os"
	"path/filepath"
	"time"
)

func name() string {
	var b [16]byte
	rand.Reader.Read(b[:])
	return fmt.Sprintf("%x", b[:])
}

func main() {
	var files int
	var maxexp int

	flag.IntVar(&files, "files", 1000, "Number of files")
	flag.IntVar(&maxexp, "maxexp", 20, "Maximum file size (max = 2^n + 128*1024 B)")
	flag.Parse()

	for i := 0; i < files; i++ {
		n := name()
		p0 := filepath.Join(string(n[0]), n[0:2])
		os.MkdirAll(p0, 0755)
		s := 1 << uint(mr.Intn(maxexp))
		a := 128 * 1024
		if a > s {
			a = s
		}
		s += mr.Intn(a)
		b := make([]byte, s)
		rand.Reader.Read(b)
		p1 := filepath.Join(p0, n)
		ioutil.WriteFile(p1, b, 0644)

		os.Chmod(p1, os.FileMode(mr.Intn(0777)|0400))

		t := time.Now().Add(-time.Duration(mr.Intn(30*86400)) * time.Second)
		os.Chtimes(p1, t, t)
	}
}