Ejemplo n.º 1
0
func main() {
	flag.Parse()

	file := flag.Arg(0)

	if file != `` {
		hashutil.WriteChecksum(file)
	} else {
		log.Fatalln("No file provided")
	}
}
Ejemplo n.º 2
0
func checkSum(filePath string, checksumFileUrl string) bool {
	hashutil.WriteChecksum(filePath)
	downloadChecksum(checksumFileUrl)

	return hashutil.CompareChecksum(filePath)
}
Ejemplo n.º 3
0
func Chunk(path string) {
	fileinfo, err := os.Stat(path)
	if err != nil {
		panic(err)
	}

	if fileinfo.Size() < 100000000 {
		panic(`File must be 100MB or more`)
	}

	// Make sha1
	hashutil.WriteChecksum(path)

	// open input file
	fi, err := os.Open(path)
	if err != nil {
		panic(err)
	}
	// close fi on exit and check for its returned error
	defer func() {
		if err := fi.Close(); err != nil {
			panic(err)
		}
	}()

	readIndex := 0
	i := 0
	for {
		// open output file
		fo, err := os.Create(path + `.part` + strconv.Itoa(i))
		if err != nil {
			panic(err)
		}
		// close fo on exit and check for its returned error
		defer func() {
			finfo, err := fo.Stat()
			if err != nil {
				panic(err)
			}

			if err := fo.Close(); err != nil {
				panic(err)
			}

			if finfo.Size() == 0 {
				os.Remove(fo.Name())
			} else {
				hashutil.WriteChecksum(fo.Name())
			}
		}()

		// make a buffer to keep chunks that are read
		buf := make([]byte, 50000000)
		// read a chunk
		log.Println(`readIndex =`, readIndex)
		byteRead, err := fi.ReadAt(buf, int64(readIndex))
		log.Println(`Read`, byteRead, `bytes`)
		// Somehow ReadAt() returns EOF != io.EOF so we have to manually compare like this
		if err != nil && err.Error() != `read `+fi.Name()+`: Reached the end of the file.` {
			panic(err)
		}
		if byteRead == 0 {
			break
		}

		// write a chunk
		log.Println(`Begin writing`)
		if _, err := fo.Write(buf[:byteRead]); err != nil {
			panic(err)
		}

		readIndex += byteRead
		i++
	}
}