Пример #1
0
func (p *puller) closeFile(f scanner.File) {
	if debug {
		l.Debugf("pull: closing %q / %q", p.repo, f.Name)
	}

	of := p.openFiles[f.Name]
	of.file.Close()
	defer os.Remove(of.temp)

	delete(p.openFiles, f.Name)

	fd, err := os.Open(of.temp)
	if err != nil {
		if debug {
			l.Debugf("pull: error: %q / %q: %v", p.repo, f.Name, err)
		}
		return
	}
	hb, _ := scanner.Blocks(fd, scanner.StandardBlockSize)
	fd.Close()

	if l0, l1 := len(hb), len(f.Blocks); l0 != l1 {
		if debug {
			l.Debugf("pull: %q / %q: nblocks %d != %d", p.repo, f.Name, l0, l1)
		}
		return
	}

	for i := range hb {
		if bytes.Compare(hb[i].Hash, f.Blocks[i].Hash) != 0 {
			l.Debugf("pull: %q / %q: block %d hash mismatch", p.repo, f.Name, i)
			return
		}
	}

	t := time.Unix(f.Modified, 0)
	os.Chtimes(of.temp, t, t)
	os.Chmod(of.temp, os.FileMode(f.Flags&0777))
	defTempNamer.Show(of.temp)
	if debug {
		l.Debugf("pull: rename %q / %q: %q", p.repo, f.Name, of.filepath)
	}
	if err := Rename(of.temp, of.filepath); err == nil {
		p.model.updateLocal(p.repo, f)
	} else {
		l.Debugf("pull: error: %q / %q: %v", p.repo, f.Name, err)
	}
}
Пример #2
0
func (p *puller) closeFile(f protocol.FileInfo) {
	if debug {
		l.Debugf("pull: closing %q / %q", p.repoCfg.ID, f.Name)
	}

	of := p.openFiles[f.Name]
	of.file.Close()
	defer os.Remove(of.temp)

	delete(p.openFiles, f.Name)

	fd, err := os.Open(of.temp)
	if err != nil {
		if debug {
			l.Debugf("pull: error: %q / %q: %v", p.repoCfg.ID, f.Name, err)
		}
		return
	}
	hb, _ := scanner.Blocks(fd, scanner.StandardBlockSize)
	fd.Close()

	if l0, l1 := len(hb), len(f.Blocks); l0 != l1 {
		if debug {
			l.Debugf("pull: %q / %q: nblocks %d != %d", p.repoCfg.ID, f.Name, l0, l1)
		}
		return
	}

	for i := range hb {
		if bytes.Compare(hb[i].Hash, f.Blocks[i].Hash) != 0 {
			l.Debugf("pull: %q / %q: block %d hash mismatch\n\thave: %x\n\twant: %x", p.repoCfg.ID, f.Name, i, hb[i].Hash, f.Blocks[i].Hash)
			return
		}
	}

	t := time.Unix(f.Modified, 0)
	err = os.Chtimes(of.temp, t, t)
	if debug && err != nil {
		l.Debugf("pull: error: %q / %q: %v", p.repoCfg.ID, f.Name, err)
	}
	if !p.repoCfg.IgnorePerms && protocol.HasPermissionBits(f.Flags) {
		err = os.Chmod(of.temp, os.FileMode(f.Flags&0777))
		if debug && err != nil {
			l.Debugf("pull: error: %q / %q: %v", p.repoCfg.ID, f.Name, err)
		}
	}

	osutil.ShowFile(of.temp)

	if p.versioner != nil {
		err := p.versioner.Archive(p.repoCfg.Directory, of.filepath)
		if err != nil {
			if debug {
				l.Debugf("pull: error: %q / %q: %v", p.repoCfg.ID, f.Name, err)
			}
			return
		}
	}

	if debug {
		l.Debugf("pull: rename %q / %q: %q", p.repoCfg.ID, f.Name, of.filepath)
	}
	if err := osutil.Rename(of.temp, of.filepath); err == nil {
		p.model.updateLocal(p.repoCfg.ID, f)
	} else {
		l.Debugf("pull: error: %q / %q: %v", p.repoCfg.ID, f.Name, err)
	}
}