2014-11-16 21:13:20 +01:00
|
|
|
// Copyright (C) 2014 The Syncthing Authors.
|
2014-09-29 21:43:32 +02:00
|
|
|
//
|
2015-03-07 21:36:35 +01:00
|
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
2017-02-09 07:52:18 +01:00
|
|
|
// You can obtain one at https://mozilla.org/MPL/2.0/.
|
2014-06-01 22:50:14 +02:00
|
|
|
|
2014-03-08 23:02:01 +01:00
|
|
|
package scanner
|
|
|
|
|
|
|
|
import (
|
2017-04-26 02:15:23 +02:00
|
|
|
"context"
|
2018-05-14 09:47:23 +02:00
|
|
|
"path/filepath"
|
2014-05-19 23:42:08 +02:00
|
|
|
"runtime"
|
2018-05-14 09:47:23 +02:00
|
|
|
"strings"
|
2015-08-27 00:49:06 +02:00
|
|
|
"sync/atomic"
|
2014-12-10 00:58:58 +01:00
|
|
|
"time"
|
2015-03-18 23:54:50 +01:00
|
|
|
"unicode/utf8"
|
2014-08-16 18:33:01 +02:00
|
|
|
|
2015-11-17 21:08:36 +01:00
|
|
|
"github.com/rcrowley/go-metrics"
|
2015-08-27 00:49:06 +02:00
|
|
|
"github.com/syncthing/syncthing/lib/events"
|
2017-04-01 11:04:11 +02:00
|
|
|
"github.com/syncthing/syncthing/lib/fs"
|
2016-04-07 11:34:07 +02:00
|
|
|
"github.com/syncthing/syncthing/lib/ignore"
|
2015-08-06 11:29:25 +02:00
|
|
|
"github.com/syncthing/syncthing/lib/osutil"
|
2015-09-22 19:38:46 +02:00
|
|
|
"github.com/syncthing/syncthing/lib/protocol"
|
2014-11-30 00:17:00 +01:00
|
|
|
"golang.org/x/text/unicode/norm"
|
2014-03-08 23:02:01 +01:00
|
|
|
)
|
|
|
|
|
2017-04-01 11:04:11 +02:00
|
|
|
var maskModePerm fs.FileMode
|
2015-03-15 14:14:44 +01:00
|
|
|
|
|
|
|
func init() {
|
|
|
|
if runtime.GOOS == "windows" {
|
|
|
|
// There is no user/group/others in Windows' read-only
|
2017-04-01 11:04:11 +02:00
|
|
|
// attribute, and all "w" bits are set in fs.FileMode
|
2015-03-15 14:14:44 +01:00
|
|
|
// if the file is not read-only. Do not send these
|
|
|
|
// group/others-writable bits to other devices in order to
|
|
|
|
// avoid unexpected world-writable files on other platforms.
|
2017-04-01 11:04:11 +02:00
|
|
|
maskModePerm = fs.ModePerm & 0755
|
2015-03-15 14:14:44 +01:00
|
|
|
} else {
|
2017-04-01 11:04:11 +02:00
|
|
|
maskModePerm = fs.ModePerm
|
2015-03-15 14:14:44 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-05-09 20:25:39 +02:00
|
|
|
type Config struct {
|
2015-08-27 00:49:06 +02:00
|
|
|
// Folder for which the walker has been created
|
|
|
|
Folder string
|
2015-03-27 09:51:18 +01:00
|
|
|
// Limit walking to these paths within Dir, or no limit if Sub is empty
|
|
|
|
Subs []string
|
2014-10-14 22:30:00 +02:00
|
|
|
// If Matcher is not nil, it is used to identify files to ignore which were specified by the user.
|
2016-04-07 11:34:07 +02:00
|
|
|
Matcher *ignore.Matcher
|
2014-12-10 00:58:58 +01:00
|
|
|
// Number of hours to keep temporary files for
|
|
|
|
TempLifetime time.Duration
|
2018-02-14 08:59:46 +01:00
|
|
|
// If CurrentFiler is not nil, it is queried for the current file before rescanning.
|
|
|
|
CurrentFiler CurrentFiler
|
2017-04-01 11:04:11 +02:00
|
|
|
// The Filesystem provides an abstraction on top of the actual filesystem.
|
|
|
|
Filesystem fs.Filesystem
|
2014-05-23 14:31:16 +02:00
|
|
|
// If IgnorePerms is true, changes to permission bits will not be
|
2014-05-23 13:10:26 +02:00
|
|
|
// detected. Scanned files will get zero permission bits and the
|
|
|
|
// NoPermissionBits flag set.
|
2014-05-23 14:31:16 +02:00
|
|
|
IgnorePerms bool
|
2015-03-18 23:54:50 +01:00
|
|
|
// When AutoNormalize is set, file names that are in UTF8 but incorrect
|
|
|
|
// normalization form will be corrected.
|
|
|
|
AutoNormalize bool
|
2014-12-25 00:12:12 +01:00
|
|
|
// Number of routines to use for hashing
|
|
|
|
Hashers int
|
2015-03-25 22:37:35 +01:00
|
|
|
// Our vector clock id
|
2016-01-20 20:10:22 +01:00
|
|
|
ShortID protocol.ShortID
|
2015-08-27 00:49:06 +02:00
|
|
|
// Optional progress tick interval which defines how often FolderScanProgress
|
|
|
|
// events are emitted. Negative number means disabled.
|
|
|
|
ProgressTickIntervalS int
|
2018-04-16 20:08:50 +02:00
|
|
|
// Whether to use large blocks for large files or the old standard of 128KiB for everything.
|
|
|
|
UseLargeBlocks bool
|
2014-03-08 23:02:01 +01:00
|
|
|
}
|
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
type CurrentFiler interface {
|
|
|
|
// CurrentFile returns the file as seen at last scan.
|
|
|
|
CurrentFile(name string) (protocol.FileInfo, bool)
|
2014-03-16 08:14:55 +01:00
|
|
|
}
|
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
func Walk(ctx context.Context, cfg Config) chan protocol.FileInfo {
|
2016-05-09 20:25:39 +02:00
|
|
|
w := walker{cfg}
|
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
if w.CurrentFiler == nil {
|
|
|
|
w.CurrentFiler = noCurrentFiler{}
|
2016-05-09 20:25:39 +02:00
|
|
|
}
|
2017-04-01 11:04:11 +02:00
|
|
|
if w.Filesystem == nil {
|
2017-08-19 16:36:56 +02:00
|
|
|
panic("no filesystem specified")
|
2016-05-09 20:25:39 +02:00
|
|
|
}
|
2017-09-06 08:39:18 +02:00
|
|
|
if w.Matcher == nil {
|
|
|
|
w.Matcher = ignore.New(w.Filesystem)
|
|
|
|
}
|
2016-05-09 20:25:39 +02:00
|
|
|
|
2017-04-26 02:15:23 +02:00
|
|
|
return w.walk(ctx)
|
2016-05-09 20:25:39 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
type walker struct {
|
|
|
|
Config
|
|
|
|
}
|
|
|
|
|
2014-09-28 13:00:38 +02:00
|
|
|
// Walk returns the list of files found in the local folder by scanning the
|
2014-03-08 23:02:01 +01:00
|
|
|
// file system. Files are blockwise hashed.
|
2018-02-14 08:59:46 +01:00
|
|
|
func (w *walker) walk(ctx context.Context) chan protocol.FileInfo {
|
2018-04-16 20:08:50 +02:00
|
|
|
l.Debugln("Walk", w.Subs, w.Matcher)
|
2014-05-04 18:20:25 +02:00
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
toHashChan := make(chan protocol.FileInfo)
|
|
|
|
finishedChan := make(chan protocol.FileInfo)
|
2014-03-08 23:02:01 +01:00
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
// A routine which walks the filesystem tree, and sends files which have
|
|
|
|
// been modified to the counter routine.
|
|
|
|
go func() {
|
|
|
|
hashFiles := w.walkAndHashFiles(ctx, toHashChan, finishedChan)
|
|
|
|
if len(w.Subs) == 0 {
|
|
|
|
w.Filesystem.Walk(".", hashFiles)
|
|
|
|
} else {
|
|
|
|
for _, sub := range w.Subs {
|
|
|
|
w.Filesystem.Walk(sub, hashFiles)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
close(toHashChan)
|
|
|
|
}()
|
2014-03-08 23:02:01 +01:00
|
|
|
|
2015-08-27 00:49:06 +02:00
|
|
|
// We're not required to emit scan progress events, just kick off hashers,
|
|
|
|
// and feed inputs directly from the walker.
|
|
|
|
if w.ProgressTickIntervalS < 0 {
|
2018-05-05 10:24:44 +02:00
|
|
|
newParallelHasher(ctx, w.Filesystem, w.Hashers, finishedChan, toHashChan, nil, nil)
|
2018-01-24 01:05:47 +01:00
|
|
|
return finishedChan
|
2015-08-27 00:49:06 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
// Defaults to every 2 seconds.
|
|
|
|
if w.ProgressTickIntervalS == 0 {
|
|
|
|
w.ProgressTickIntervalS = 2
|
|
|
|
}
|
|
|
|
|
|
|
|
ticker := time.NewTicker(time.Duration(w.ProgressTickIntervalS) * time.Second)
|
|
|
|
|
|
|
|
// We need to emit progress events, hence we create a routine which buffers
|
|
|
|
// the list of files to be hashed, counts the total number of
|
|
|
|
// bytes to hash, and once no more files need to be hashed (chan gets closed),
|
|
|
|
// start a routine which periodically emits FolderScanProgress events,
|
|
|
|
// until a stop signal is sent by the parallel hasher.
|
|
|
|
// Parallel hasher is stopped by this routine when we close the channel over
|
|
|
|
// which it receives the files we ask it to hash.
|
|
|
|
go func() {
|
2018-02-14 08:59:46 +01:00
|
|
|
var filesToHash []protocol.FileInfo
|
2015-11-17 21:08:36 +01:00
|
|
|
var total int64 = 1
|
|
|
|
|
2015-08-27 00:49:06 +02:00
|
|
|
for file := range toHashChan {
|
|
|
|
filesToHash = append(filesToHash, file)
|
2018-02-14 08:59:46 +01:00
|
|
|
total += file.Size
|
2015-08-27 00:49:06 +02:00
|
|
|
}
|
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
realToHashChan := make(chan protocol.FileInfo)
|
2015-08-27 00:49:06 +02:00
|
|
|
done := make(chan struct{})
|
2015-11-18 10:57:11 +01:00
|
|
|
progress := newByteCounter()
|
|
|
|
|
2018-05-05 10:24:44 +02:00
|
|
|
newParallelHasher(ctx, w.Filesystem, w.Hashers, finishedChan, realToHashChan, progress, done)
|
2015-08-27 00:49:06 +02:00
|
|
|
|
|
|
|
// A routine which actually emits the FolderScanProgress events
|
|
|
|
// every w.ProgressTicker ticks, until the hasher routines terminate.
|
|
|
|
go func() {
|
2017-01-03 15:03:18 +01:00
|
|
|
defer progress.Close()
|
|
|
|
|
2015-08-27 00:49:06 +02:00
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case <-done:
|
2018-04-16 20:08:50 +02:00
|
|
|
l.Debugln("Walk progress done", w.Folder, w.Subs, w.Matcher)
|
2015-08-27 00:49:06 +02:00
|
|
|
ticker.Stop()
|
|
|
|
return
|
|
|
|
case <-ticker.C:
|
2015-11-17 21:08:36 +01:00
|
|
|
current := progress.Total()
|
|
|
|
rate := progress.Rate()
|
2017-08-19 16:36:56 +02:00
|
|
|
l.Debugf("Walk %s %s current progress %d/%d at %.01f MiB/s (%d%%)", w.Folder, w.Subs, current, total, rate/1024/1024, current*100/total)
|
2015-08-27 00:49:06 +02:00
|
|
|
events.Default.Log(events.FolderScanProgress, map[string]interface{}{
|
|
|
|
"folder": w.Folder,
|
|
|
|
"current": current,
|
|
|
|
"total": total,
|
2015-11-17 21:08:36 +01:00
|
|
|
"rate": rate, // bytes per second
|
2015-08-27 00:49:06 +02:00
|
|
|
})
|
2017-04-26 02:15:23 +02:00
|
|
|
case <-ctx.Done():
|
2015-11-13 15:00:32 +01:00
|
|
|
ticker.Stop()
|
|
|
|
return
|
2015-08-27 00:49:06 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}()
|
|
|
|
|
2015-11-13 15:00:32 +01:00
|
|
|
loop:
|
2015-08-27 00:49:06 +02:00
|
|
|
for _, file := range filesToHash {
|
2018-02-14 08:59:46 +01:00
|
|
|
l.Debugln("real to hash:", file.Name)
|
2015-11-13 15:00:32 +01:00
|
|
|
select {
|
|
|
|
case realToHashChan <- file:
|
2017-04-26 02:15:23 +02:00
|
|
|
case <-ctx.Done():
|
2015-11-13 15:00:32 +01:00
|
|
|
break loop
|
|
|
|
}
|
2015-08-27 00:49:06 +02:00
|
|
|
}
|
|
|
|
close(realToHashChan)
|
|
|
|
}()
|
|
|
|
|
2018-01-24 01:05:47 +01:00
|
|
|
return finishedChan
|
2014-03-08 23:02:01 +01:00
|
|
|
}
|
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
func (w *walker) walkAndHashFiles(ctx context.Context, fchan, dchan chan protocol.FileInfo) fs.WalkFunc {
|
|
|
|
now := time.Now()
|
2018-05-14 09:47:23 +02:00
|
|
|
ignoredParent := ""
|
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
return func(path string, info fs.FileInfo, err error) error {
|
2017-04-26 02:15:23 +02:00
|
|
|
select {
|
|
|
|
case <-ctx.Done():
|
2018-02-14 08:59:46 +01:00
|
|
|
return ctx.Err()
|
2017-04-26 02:15:23 +02:00
|
|
|
default:
|
|
|
|
}
|
|
|
|
|
2015-03-19 07:44:43 +01:00
|
|
|
// Return value used when we are returning early and don't want to
|
|
|
|
// process the item. For directories, this means do-not-descend.
|
|
|
|
var skip error // nil
|
2015-03-22 15:06:29 +01:00
|
|
|
// info nil when error is not nil
|
|
|
|
if info != nil && info.IsDir() {
|
2017-04-01 11:04:11 +02:00
|
|
|
skip = fs.SkipDir
|
2015-03-19 07:44:43 +01:00
|
|
|
}
|
|
|
|
|
2018-02-13 11:02:07 +01:00
|
|
|
if err != nil {
|
2018-02-14 08:59:46 +01:00
|
|
|
l.Debugln("error:", path, info, err)
|
2018-02-13 11:02:07 +01:00
|
|
|
return skip
|
|
|
|
}
|
|
|
|
|
|
|
|
if path == "." {
|
2014-04-01 23:18:32 +02:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-09-02 07:52:38 +02:00
|
|
|
if fs.IsTemporary(path) {
|
2017-08-19 16:36:56 +02:00
|
|
|
l.Debugln("temporary:", path)
|
2017-04-01 11:04:11 +02:00
|
|
|
if info.IsRegular() && info.ModTime().Add(w.TempLifetime).Before(now) {
|
2017-08-19 16:36:56 +02:00
|
|
|
w.Filesystem.Remove(path)
|
|
|
|
l.Debugln("removing temporary:", path, info.ModTime())
|
2014-12-10 00:58:58 +01:00
|
|
|
}
|
2014-03-08 23:02:01 +01:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
2017-09-02 07:52:38 +02:00
|
|
|
if fs.IsInternal(path) {
|
2018-02-14 08:59:46 +01:00
|
|
|
l.Debugln("ignored (internal):", path)
|
2016-12-01 15:00:11 +01:00
|
|
|
return skip
|
|
|
|
}
|
|
|
|
|
2018-05-14 09:47:23 +02:00
|
|
|
if !utf8.ValidString(path) {
|
|
|
|
l.Warnf("File name %q is not in UTF8 encoding; skipping.", path)
|
|
|
|
return skip
|
|
|
|
}
|
|
|
|
|
2017-08-19 16:36:56 +02:00
|
|
|
if w.Matcher.Match(path).IsIgnored() {
|
2018-02-14 08:59:46 +01:00
|
|
|
l.Debugln("ignored (patterns):", path)
|
2018-05-14 09:47:23 +02:00
|
|
|
// Only descend if matcher says so and the current file is not a symlink.
|
|
|
|
if w.Matcher.SkipIgnoredDirs() || (info.IsSymlink() && info.IsDir()) {
|
|
|
|
return fs.SkipDir
|
|
|
|
}
|
|
|
|
// If the parent wasn't ignored already, set this path as the "highest" ignored parent
|
|
|
|
if info.IsDir() && (ignoredParent == "" || !strings.HasPrefix(path, ignoredParent+string(fs.PathSeparator))) {
|
|
|
|
ignoredParent = path
|
|
|
|
}
|
|
|
|
return nil
|
2018-02-10 16:56:53 +01:00
|
|
|
}
|
|
|
|
|
2018-05-14 09:47:23 +02:00
|
|
|
if ignoredParent == "" {
|
|
|
|
// parent isn't ignored, nothing special
|
|
|
|
return w.handleItem(ctx, path, fchan, dchan, skip)
|
2014-06-11 17:51:31 +02:00
|
|
|
}
|
|
|
|
|
2018-05-14 09:47:23 +02:00
|
|
|
// Part of current path below the ignored (potential) parent
|
|
|
|
rel := strings.TrimPrefix(path, ignoredParent+string(fs.PathSeparator))
|
|
|
|
|
|
|
|
// ignored path isn't actually a parent of the current path
|
|
|
|
if rel == path {
|
|
|
|
ignoredParent = ""
|
|
|
|
return w.handleItem(ctx, path, fchan, dchan, skip)
|
2018-02-10 16:56:53 +01:00
|
|
|
}
|
|
|
|
|
2018-05-14 09:47:23 +02:00
|
|
|
// The previously ignored parent directories of the current, not
|
|
|
|
// ignored path need to be handled as well.
|
|
|
|
if err = w.handleItem(ctx, ignoredParent, fchan, dchan, skip); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
for _, name := range strings.Split(rel, string(fs.PathSeparator)) {
|
|
|
|
ignoredParent = filepath.Join(ignoredParent, name)
|
|
|
|
if err = w.handleItem(ctx, ignoredParent, fchan, dchan, skip); err != nil {
|
2018-02-14 08:59:46 +01:00
|
|
|
return err
|
2017-01-05 15:05:49 +01:00
|
|
|
}
|
2018-05-14 09:47:23 +02:00
|
|
|
}
|
|
|
|
ignoredParent = ""
|
2015-05-05 21:19:59 +02:00
|
|
|
|
2018-05-14 09:47:23 +02:00
|
|
|
return nil
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (w *walker) handleItem(ctx context.Context, path string, fchan, dchan chan protocol.FileInfo, skip error) error {
|
|
|
|
info, err := w.Filesystem.Lstat(path)
|
|
|
|
// An error here would be weird as we've already gotten to this point, but act on it nonetheless
|
|
|
|
if err != nil {
|
|
|
|
return skip
|
|
|
|
}
|
|
|
|
|
|
|
|
path, shouldSkip := w.normalizePath(path, info)
|
|
|
|
if shouldSkip {
|
|
|
|
return skip
|
|
|
|
}
|
2018-02-10 16:56:53 +01:00
|
|
|
|
2018-05-14 09:47:23 +02:00
|
|
|
switch {
|
|
|
|
case info.IsSymlink():
|
|
|
|
if err := w.walkSymlink(ctx, path, dchan); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if info.IsDir() {
|
|
|
|
// under no circumstances shall we descend into a symlink
|
|
|
|
return fs.SkipDir
|
2018-02-10 16:56:53 +01:00
|
|
|
}
|
2018-05-14 09:47:23 +02:00
|
|
|
return nil
|
2018-02-10 16:56:53 +01:00
|
|
|
|
2018-05-14 09:47:23 +02:00
|
|
|
case info.IsDir():
|
|
|
|
err = w.walkDir(ctx, path, info, dchan)
|
|
|
|
|
|
|
|
case info.IsRegular():
|
|
|
|
err = w.walkRegular(ctx, path, info, fchan)
|
2018-02-10 16:56:53 +01:00
|
|
|
}
|
2018-05-14 09:47:23 +02:00
|
|
|
|
|
|
|
return err
|
2018-02-10 16:56:53 +01:00
|
|
|
}
|
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
func (w *walker) walkRegular(ctx context.Context, relPath string, info fs.FileInfo, fchan chan protocol.FileInfo) error {
|
2018-05-29 08:01:23 +02:00
|
|
|
curFile, hasCurFile := w.CurrentFiler.CurrentFile(relPath)
|
|
|
|
|
|
|
|
newMode := uint32(info.Mode())
|
|
|
|
if runtime.GOOS == "windows" {
|
|
|
|
if osutil.IsWindowsExecutable(relPath) {
|
|
|
|
// Set executable bits on files with executable extenions (.exe,
|
|
|
|
// .bat, etc).
|
|
|
|
newMode |= 0111
|
|
|
|
} else if hasCurFile {
|
|
|
|
// If we have an existing index entry, copy the executable bits
|
|
|
|
// from there.
|
|
|
|
newMode |= (curFile.Permissions & 0111)
|
|
|
|
}
|
2015-11-20 10:32:16 +01:00
|
|
|
}
|
2014-03-08 23:02:01 +01:00
|
|
|
|
2018-04-16 20:08:50 +02:00
|
|
|
blockSize := protocol.MinBlockSize
|
|
|
|
|
|
|
|
if w.UseLargeBlocks {
|
|
|
|
blockSize = protocol.BlockSize(info.Size())
|
|
|
|
|
|
|
|
if hasCurFile {
|
|
|
|
// Check if we should retain current block size.
|
|
|
|
curBlockSize := curFile.BlockSize()
|
|
|
|
if blockSize > curBlockSize && blockSize/curBlockSize <= 2 {
|
|
|
|
// New block size is larger, but not more than twice larger.
|
|
|
|
// Retain.
|
|
|
|
blockSize = curBlockSize
|
|
|
|
} else if curBlockSize > blockSize && curBlockSize/blockSize <= 2 {
|
|
|
|
// Old block size is larger, but not more than twice larger.
|
|
|
|
// Retain.
|
|
|
|
blockSize = curBlockSize
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2016-05-09 20:25:39 +02:00
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
f := protocol.FileInfo{
|
|
|
|
Name: relPath,
|
|
|
|
Type: protocol.FileInfoTypeFile,
|
2018-04-16 20:08:50 +02:00
|
|
|
Version: curFile.Version.Update(w.ShortID),
|
2018-05-29 08:01:23 +02:00
|
|
|
Permissions: newMode & uint32(maskModePerm),
|
2018-02-14 08:59:46 +01:00
|
|
|
NoPermissions: w.IgnorePerms,
|
|
|
|
ModifiedS: info.ModTime().Unix(),
|
|
|
|
ModifiedNs: int32(info.ModTime().Nanosecond()),
|
|
|
|
ModifiedBy: w.ShortID,
|
|
|
|
Size: info.Size(),
|
2018-04-16 20:08:50 +02:00
|
|
|
RawBlockSize: int32(blockSize),
|
2014-03-08 23:02:01 +01:00
|
|
|
}
|
2018-02-25 09:39:00 +01:00
|
|
|
|
2018-04-16 20:08:50 +02:00
|
|
|
if hasCurFile {
|
|
|
|
if curFile.IsEquivalent(f, w.IgnorePerms, true) {
|
2018-02-25 09:39:00 +01:00
|
|
|
return nil
|
|
|
|
}
|
2018-04-16 20:08:50 +02:00
|
|
|
if curFile.Invalid {
|
2018-02-25 09:39:00 +01:00
|
|
|
// We do not want to override the global version with the file we
|
|
|
|
// currently have. Keeping only our local counter makes sure we are in
|
|
|
|
// conflict with any other existing versions, which will be resolved by
|
|
|
|
// the normal pulling mechanisms.
|
2018-03-27 22:24:20 +02:00
|
|
|
f.Version = f.Version.DropOthers(w.ShortID)
|
2018-02-25 09:39:00 +01:00
|
|
|
}
|
2018-04-16 20:08:50 +02:00
|
|
|
l.Debugln("rescan:", curFile, info.ModTime().Unix(), info.Mode()&fs.ModePerm)
|
2018-02-25 09:39:00 +01:00
|
|
|
}
|
|
|
|
|
2015-11-20 10:32:16 +01:00
|
|
|
l.Debugln("to hash:", relPath, f)
|
|
|
|
|
|
|
|
select {
|
2018-02-14 08:59:46 +01:00
|
|
|
case fchan <- f:
|
2017-04-26 02:15:23 +02:00
|
|
|
case <-ctx.Done():
|
2018-02-14 08:59:46 +01:00
|
|
|
return ctx.Err()
|
2015-11-20 10:32:16 +01:00
|
|
|
}
|
2018-02-14 08:59:46 +01:00
|
|
|
|
|
|
|
return nil
|
2014-03-08 23:02:01 +01:00
|
|
|
}
|
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
func (w *walker) walkDir(ctx context.Context, relPath string, info fs.FileInfo, dchan chan protocol.FileInfo) error {
|
|
|
|
cf, ok := w.CurrentFiler.CurrentFile(relPath)
|
|
|
|
|
|
|
|
f := protocol.FileInfo{
|
|
|
|
Name: relPath,
|
|
|
|
Type: protocol.FileInfoTypeDirectory,
|
|
|
|
Version: cf.Version.Update(w.ShortID),
|
|
|
|
Permissions: uint32(info.Mode() & maskModePerm),
|
|
|
|
NoPermissions: w.IgnorePerms,
|
|
|
|
ModifiedS: info.ModTime().Unix(),
|
|
|
|
ModifiedNs: int32(info.ModTime().Nanosecond()),
|
|
|
|
ModifiedBy: w.ShortID,
|
2015-11-20 09:54:12 +01:00
|
|
|
}
|
2018-02-25 09:39:00 +01:00
|
|
|
|
|
|
|
if ok {
|
|
|
|
if cf.IsEquivalent(f, w.IgnorePerms, true) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
if cf.Invalid {
|
|
|
|
// We do not want to override the global version with the file we
|
|
|
|
// currently have. Keeping only our local counter makes sure we are in
|
|
|
|
// conflict with any other existing versions, which will be resolved by
|
|
|
|
// the normal pulling mechanisms.
|
2018-03-27 22:24:20 +02:00
|
|
|
f.Version = f.Version.DropOthers(w.ShortID)
|
2018-02-25 09:39:00 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2015-11-20 09:54:12 +01:00
|
|
|
l.Debugln("dir:", relPath, f)
|
|
|
|
|
|
|
|
select {
|
2018-02-14 08:59:46 +01:00
|
|
|
case dchan <- f:
|
2017-04-26 02:15:23 +02:00
|
|
|
case <-ctx.Done():
|
2018-02-14 08:59:46 +01:00
|
|
|
return ctx.Err()
|
2015-11-20 09:54:12 +01:00
|
|
|
}
|
2018-02-14 08:59:46 +01:00
|
|
|
|
|
|
|
return nil
|
2015-11-20 09:54:12 +01:00
|
|
|
}
|
|
|
|
|
2017-01-05 14:26:29 +01:00
|
|
|
// walkSymlink returns nil or an error, if the error is of the nature that
|
|
|
|
// it should stop the entire walk.
|
2018-02-14 08:59:46 +01:00
|
|
|
func (w *walker) walkSymlink(ctx context.Context, relPath string, dchan chan protocol.FileInfo) error {
|
2017-03-18 01:25:47 +01:00
|
|
|
// Symlinks are not supported on Windows. We ignore instead of returning
|
|
|
|
// an error.
|
|
|
|
if runtime.GOOS == "windows" {
|
2018-02-14 08:59:46 +01:00
|
|
|
return nil
|
2017-03-18 01:25:47 +01:00
|
|
|
}
|
|
|
|
|
2015-11-20 09:50:46 +01:00
|
|
|
// We always rehash symlinks as they have no modtime or
|
|
|
|
// permissions. We check if they point to the old target by
|
|
|
|
// checking that their existing blocks match with the blocks in
|
|
|
|
// the index.
|
|
|
|
|
2017-08-19 16:36:56 +02:00
|
|
|
target, err := w.Filesystem.ReadSymlink(relPath)
|
2015-11-20 09:50:46 +01:00
|
|
|
if err != nil {
|
2017-08-19 16:36:56 +02:00
|
|
|
l.Debugln("readlink error:", relPath, err)
|
2018-02-14 08:59:46 +01:00
|
|
|
return nil
|
2015-11-20 09:50:46 +01:00
|
|
|
}
|
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
cf, ok := w.CurrentFiler.CurrentFile(relPath)
|
2015-11-20 09:50:46 +01:00
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
f := protocol.FileInfo{
|
|
|
|
Name: relPath,
|
|
|
|
Type: protocol.FileInfoTypeSymlink,
|
|
|
|
Version: cf.Version.Update(w.ShortID),
|
|
|
|
NoPermissions: true, // Symlinks don't have permissions of their own
|
|
|
|
SymlinkTarget: target,
|
2018-02-25 14:51:37 +01:00
|
|
|
ModifiedBy: w.ShortID,
|
2015-11-20 09:50:46 +01:00
|
|
|
}
|
|
|
|
|
2018-02-25 09:39:00 +01:00
|
|
|
if ok {
|
|
|
|
if cf.IsEquivalent(f, w.IgnorePerms, true) {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
if cf.Invalid {
|
|
|
|
// We do not want to override the global version with the file we
|
|
|
|
// currently have. Keeping only our local counter makes sure we are in
|
|
|
|
// conflict with any other existing versions, which will be resolved by
|
|
|
|
// the normal pulling mechanisms.
|
2018-03-27 22:24:20 +02:00
|
|
|
f.Version = f.Version.DropOthers(w.ShortID)
|
2018-02-25 09:39:00 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-08-19 16:36:56 +02:00
|
|
|
l.Debugln("symlink changedb:", relPath, f)
|
2015-11-20 09:50:46 +01:00
|
|
|
|
|
|
|
select {
|
2018-02-14 08:59:46 +01:00
|
|
|
case dchan <- f:
|
2017-04-26 02:15:23 +02:00
|
|
|
case <-ctx.Done():
|
2018-02-14 08:59:46 +01:00
|
|
|
return ctx.Err()
|
2015-11-20 09:50:46 +01:00
|
|
|
}
|
2018-02-14 08:59:46 +01:00
|
|
|
|
|
|
|
return nil
|
2015-11-20 09:50:46 +01:00
|
|
|
}
|
|
|
|
|
2015-11-20 09:41:44 +01:00
|
|
|
// normalizePath returns the normalized relative path (possibly after fixing
|
|
|
|
// it on disk), or skip is true.
|
lib/scanner: Fix UTF-8 normalization on ZFS (fixes #4649)
It turns out that ZFS doesn't do any normalization when storing files,
but does do normalization "as part of any comparison process".
In practice, this seems to mean that if you LStat a normalized filename,
ZFS will return the FileInfo for the un-normalized version of that
filename.
This meant that our test to see whether a separate file with a
normalized version of the filename already exists was failing, as we
were detecting the same file.
The fix is to use os.SameFile, to see whether we're getting the same
FileInfo from the normalized and un-normalized versions of the same
filename.
One complication is that ZFS also seems to apply its magic to os.Rename,
meaning that we can't use it to rename an un-normalized file to its
normalized filename. Instead we have to move via a temporary object. If
the move to the temporary object fails, that's OK, we can skip it and
move on. If the move from the temporary object fails however, I'm not
sure of the best approach: the current one is to leave the temporary
file name as-is, and get Syncthing to syncronize it, so at least we
don't lose the file. I'm not sure if there are any implications of this
however.
As part of reworking normalizePath, I spotted that it appeared to be
returning the wrong thing: the doc and the surrounding code expecting it
to return the normalized filename, but it was returning the
un-normalized one. I fixed this, but it seems suspicious that, if the
previous behaviour was incorrect, noone ever ran afoul of it. Maybe all
filesystems will do some searching and give you a normalized filename if
you request an unnormalized one.
As part of this, I found that TestNormalization was broken: it was
passing, when in fact one of the files it should have verified was
present was missing. Maybe this was related to the above issue with
normalizePath's return value, I'm not sure. Fixed en route.
Kindly tested by @khinsen on the forum, and it appears to work.
GitHub-Pull-Request: https://github.com/syncthing/syncthing/pull/4646
2018-01-05 19:11:09 +01:00
|
|
|
func (w *walker) normalizePath(path string, info fs.FileInfo) (normPath string, skip bool) {
|
2015-11-20 09:41:44 +01:00
|
|
|
if runtime.GOOS == "darwin" {
|
|
|
|
// Mac OS X file names should always be NFD normalized.
|
2017-08-19 16:36:56 +02:00
|
|
|
normPath = norm.NFD.String(path)
|
2015-11-20 09:41:44 +01:00
|
|
|
} else {
|
|
|
|
// Every other OS in the known universe uses NFC or just plain
|
|
|
|
// doesn't bother to define an encoding. In our case *we* do care,
|
|
|
|
// so we enforce NFC regardless.
|
2017-08-19 16:36:56 +02:00
|
|
|
normPath = norm.NFC.String(path)
|
2015-11-20 09:41:44 +01:00
|
|
|
}
|
|
|
|
|
lib/scanner: Fix UTF-8 normalization on ZFS (fixes #4649)
It turns out that ZFS doesn't do any normalization when storing files,
but does do normalization "as part of any comparison process".
In practice, this seems to mean that if you LStat a normalized filename,
ZFS will return the FileInfo for the un-normalized version of that
filename.
This meant that our test to see whether a separate file with a
normalized version of the filename already exists was failing, as we
were detecting the same file.
The fix is to use os.SameFile, to see whether we're getting the same
FileInfo from the normalized and un-normalized versions of the same
filename.
One complication is that ZFS also seems to apply its magic to os.Rename,
meaning that we can't use it to rename an un-normalized file to its
normalized filename. Instead we have to move via a temporary object. If
the move to the temporary object fails, that's OK, we can skip it and
move on. If the move from the temporary object fails however, I'm not
sure of the best approach: the current one is to leave the temporary
file name as-is, and get Syncthing to syncronize it, so at least we
don't lose the file. I'm not sure if there are any implications of this
however.
As part of reworking normalizePath, I spotted that it appeared to be
returning the wrong thing: the doc and the surrounding code expecting it
to return the normalized filename, but it was returning the
un-normalized one. I fixed this, but it seems suspicious that, if the
previous behaviour was incorrect, noone ever ran afoul of it. Maybe all
filesystems will do some searching and give you a normalized filename if
you request an unnormalized one.
As part of this, I found that TestNormalization was broken: it was
passing, when in fact one of the files it should have verified was
present was missing. Maybe this was related to the above issue with
normalizePath's return value, I'm not sure. Fixed en route.
Kindly tested by @khinsen on the forum, and it appears to work.
GitHub-Pull-Request: https://github.com/syncthing/syncthing/pull/4646
2018-01-05 19:11:09 +01:00
|
|
|
if path == normPath {
|
|
|
|
// The file name is already normalized: nothing to do
|
|
|
|
return path, false
|
|
|
|
}
|
|
|
|
|
|
|
|
if !w.AutoNormalize {
|
|
|
|
// We're not authorized to do anything about it, so complain and skip.
|
2015-11-20 09:41:44 +01:00
|
|
|
|
lib/scanner: Fix UTF-8 normalization on ZFS (fixes #4649)
It turns out that ZFS doesn't do any normalization when storing files,
but does do normalization "as part of any comparison process".
In practice, this seems to mean that if you LStat a normalized filename,
ZFS will return the FileInfo for the un-normalized version of that
filename.
This meant that our test to see whether a separate file with a
normalized version of the filename already exists was failing, as we
were detecting the same file.
The fix is to use os.SameFile, to see whether we're getting the same
FileInfo from the normalized and un-normalized versions of the same
filename.
One complication is that ZFS also seems to apply its magic to os.Rename,
meaning that we can't use it to rename an un-normalized file to its
normalized filename. Instead we have to move via a temporary object. If
the move to the temporary object fails, that's OK, we can skip it and
move on. If the move from the temporary object fails however, I'm not
sure of the best approach: the current one is to leave the temporary
file name as-is, and get Syncthing to syncronize it, so at least we
don't lose the file. I'm not sure if there are any implications of this
however.
As part of reworking normalizePath, I spotted that it appeared to be
returning the wrong thing: the doc and the surrounding code expecting it
to return the normalized filename, but it was returning the
un-normalized one. I fixed this, but it seems suspicious that, if the
previous behaviour was incorrect, noone ever ran afoul of it. Maybe all
filesystems will do some searching and give you a normalized filename if
you request an unnormalized one.
As part of this, I found that TestNormalization was broken: it was
passing, when in fact one of the files it should have verified was
present was missing. Maybe this was related to the above issue with
normalizePath's return value, I'm not sure. Fixed en route.
Kindly tested by @khinsen on the forum, and it appears to work.
GitHub-Pull-Request: https://github.com/syncthing/syncthing/pull/4646
2018-01-05 19:11:09 +01:00
|
|
|
l.Warnf("File name %q is not in the correct UTF8 normalization form; skipping.", path)
|
|
|
|
return "", true
|
|
|
|
}
|
2015-11-20 09:41:44 +01:00
|
|
|
|
lib/scanner: Fix UTF-8 normalization on ZFS (fixes #4649)
It turns out that ZFS doesn't do any normalization when storing files,
but does do normalization "as part of any comparison process".
In practice, this seems to mean that if you LStat a normalized filename,
ZFS will return the FileInfo for the un-normalized version of that
filename.
This meant that our test to see whether a separate file with a
normalized version of the filename already exists was failing, as we
were detecting the same file.
The fix is to use os.SameFile, to see whether we're getting the same
FileInfo from the normalized and un-normalized versions of the same
filename.
One complication is that ZFS also seems to apply its magic to os.Rename,
meaning that we can't use it to rename an un-normalized file to its
normalized filename. Instead we have to move via a temporary object. If
the move to the temporary object fails, that's OK, we can skip it and
move on. If the move from the temporary object fails however, I'm not
sure of the best approach: the current one is to leave the temporary
file name as-is, and get Syncthing to syncronize it, so at least we
don't lose the file. I'm not sure if there are any implications of this
however.
As part of reworking normalizePath, I spotted that it appeared to be
returning the wrong thing: the doc and the surrounding code expecting it
to return the normalized filename, but it was returning the
un-normalized one. I fixed this, but it seems suspicious that, if the
previous behaviour was incorrect, noone ever ran afoul of it. Maybe all
filesystems will do some searching and give you a normalized filename if
you request an unnormalized one.
As part of this, I found that TestNormalization was broken: it was
passing, when in fact one of the files it should have verified was
present was missing. Maybe this was related to the above issue with
normalizePath's return value, I'm not sure. Fixed en route.
Kindly tested by @khinsen on the forum, and it appears to work.
GitHub-Pull-Request: https://github.com/syncthing/syncthing/pull/4646
2018-01-05 19:11:09 +01:00
|
|
|
// We will attempt to normalize it.
|
|
|
|
normInfo, err := w.Filesystem.Lstat(normPath)
|
|
|
|
if fs.IsNotExist(err) {
|
|
|
|
// Nothing exists with the normalized filename. Good.
|
|
|
|
if err = w.Filesystem.Rename(path, normPath); err != nil {
|
|
|
|
l.Infof(`Error normalizing UTF8 encoding of file "%s": %v`, path, err)
|
2015-11-20 09:41:44 +01:00
|
|
|
return "", true
|
|
|
|
}
|
lib/scanner: Fix UTF-8 normalization on ZFS (fixes #4649)
It turns out that ZFS doesn't do any normalization when storing files,
but does do normalization "as part of any comparison process".
In practice, this seems to mean that if you LStat a normalized filename,
ZFS will return the FileInfo for the un-normalized version of that
filename.
This meant that our test to see whether a separate file with a
normalized version of the filename already exists was failing, as we
were detecting the same file.
The fix is to use os.SameFile, to see whether we're getting the same
FileInfo from the normalized and un-normalized versions of the same
filename.
One complication is that ZFS also seems to apply its magic to os.Rename,
meaning that we can't use it to rename an un-normalized file to its
normalized filename. Instead we have to move via a temporary object. If
the move to the temporary object fails, that's OK, we can skip it and
move on. If the move from the temporary object fails however, I'm not
sure of the best approach: the current one is to leave the temporary
file name as-is, and get Syncthing to syncronize it, so at least we
don't lose the file. I'm not sure if there are any implications of this
however.
As part of reworking normalizePath, I spotted that it appeared to be
returning the wrong thing: the doc and the surrounding code expecting it
to return the normalized filename, but it was returning the
un-normalized one. I fixed this, but it seems suspicious that, if the
previous behaviour was incorrect, noone ever ran afoul of it. Maybe all
filesystems will do some searching and give you a normalized filename if
you request an unnormalized one.
As part of this, I found that TestNormalization was broken: it was
passing, when in fact one of the files it should have verified was
present was missing. Maybe this was related to the above issue with
normalizePath's return value, I'm not sure. Fixed en route.
Kindly tested by @khinsen on the forum, and it appears to work.
GitHub-Pull-Request: https://github.com/syncthing/syncthing/pull/4646
2018-01-05 19:11:09 +01:00
|
|
|
l.Infof(`Normalized UTF8 encoding of file name "%s".`, path)
|
|
|
|
} else if w.Filesystem.SameFile(info, normInfo) {
|
|
|
|
// With some filesystems (ZFS), if there is an un-normalized path and you ask whether the normalized
|
|
|
|
// version exists, it responds with true. Therefore we need to check fs.SameFile as well.
|
|
|
|
// In this case, a call to Rename won't do anything, so we have to rename via a temp file.
|
|
|
|
|
|
|
|
// We don't want to use the standard syncthing prefix here, as that will result in the file being ignored
|
|
|
|
// and eventually deleted by Syncthing if the rename back fails.
|
|
|
|
|
|
|
|
tempPath := fs.TempNameWithPrefix(normPath, "")
|
|
|
|
if err = w.Filesystem.Rename(path, tempPath); err != nil {
|
|
|
|
l.Infof(`Error during normalizing UTF8 encoding of file "%s" (renamed to "%s"): %v`, path, tempPath, err)
|
2015-11-20 09:41:44 +01:00
|
|
|
return "", true
|
|
|
|
}
|
lib/scanner: Fix UTF-8 normalization on ZFS (fixes #4649)
It turns out that ZFS doesn't do any normalization when storing files,
but does do normalization "as part of any comparison process".
In practice, this seems to mean that if you LStat a normalized filename,
ZFS will return the FileInfo for the un-normalized version of that
filename.
This meant that our test to see whether a separate file with a
normalized version of the filename already exists was failing, as we
were detecting the same file.
The fix is to use os.SameFile, to see whether we're getting the same
FileInfo from the normalized and un-normalized versions of the same
filename.
One complication is that ZFS also seems to apply its magic to os.Rename,
meaning that we can't use it to rename an un-normalized file to its
normalized filename. Instead we have to move via a temporary object. If
the move to the temporary object fails, that's OK, we can skip it and
move on. If the move from the temporary object fails however, I'm not
sure of the best approach: the current one is to leave the temporary
file name as-is, and get Syncthing to syncronize it, so at least we
don't lose the file. I'm not sure if there are any implications of this
however.
As part of reworking normalizePath, I spotted that it appeared to be
returning the wrong thing: the doc and the surrounding code expecting it
to return the normalized filename, but it was returning the
un-normalized one. I fixed this, but it seems suspicious that, if the
previous behaviour was incorrect, noone ever ran afoul of it. Maybe all
filesystems will do some searching and give you a normalized filename if
you request an unnormalized one.
As part of this, I found that TestNormalization was broken: it was
passing, when in fact one of the files it should have verified was
present was missing. Maybe this was related to the above issue with
normalizePath's return value, I'm not sure. Fixed en route.
Kindly tested by @khinsen on the forum, and it appears to work.
GitHub-Pull-Request: https://github.com/syncthing/syncthing/pull/4646
2018-01-05 19:11:09 +01:00
|
|
|
if err = w.Filesystem.Rename(tempPath, normPath); err != nil {
|
|
|
|
// I don't ever expect this to happen, but if it does, we should probably tell our caller that the normalized
|
|
|
|
// path is the temp path: that way at least the user's data still gets synced.
|
|
|
|
l.Warnf(`Error renaming "%s" to "%s" while normalizating UTF8 encoding: %v. You will want to rename this file back manually`, tempPath, normPath, err)
|
|
|
|
return tempPath, false
|
|
|
|
}
|
|
|
|
} else {
|
|
|
|
// There is something already in the way at the normalized
|
|
|
|
// file name.
|
|
|
|
l.Infof(`File "%s" path has UTF8 encoding conflict with another file; ignoring.`, path)
|
|
|
|
return "", true
|
2015-11-20 09:41:44 +01:00
|
|
|
}
|
|
|
|
|
lib/scanner: Fix UTF-8 normalization on ZFS (fixes #4649)
It turns out that ZFS doesn't do any normalization when storing files,
but does do normalization "as part of any comparison process".
In practice, this seems to mean that if you LStat a normalized filename,
ZFS will return the FileInfo for the un-normalized version of that
filename.
This meant that our test to see whether a separate file with a
normalized version of the filename already exists was failing, as we
were detecting the same file.
The fix is to use os.SameFile, to see whether we're getting the same
FileInfo from the normalized and un-normalized versions of the same
filename.
One complication is that ZFS also seems to apply its magic to os.Rename,
meaning that we can't use it to rename an un-normalized file to its
normalized filename. Instead we have to move via a temporary object. If
the move to the temporary object fails, that's OK, we can skip it and
move on. If the move from the temporary object fails however, I'm not
sure of the best approach: the current one is to leave the temporary
file name as-is, and get Syncthing to syncronize it, so at least we
don't lose the file. I'm not sure if there are any implications of this
however.
As part of reworking normalizePath, I spotted that it appeared to be
returning the wrong thing: the doc and the surrounding code expecting it
to return the normalized filename, but it was returning the
un-normalized one. I fixed this, but it seems suspicious that, if the
previous behaviour was incorrect, noone ever ran afoul of it. Maybe all
filesystems will do some searching and give you a normalized filename if
you request an unnormalized one.
As part of this, I found that TestNormalization was broken: it was
passing, when in fact one of the files it should have verified was
present was missing. Maybe this was related to the above issue with
normalizePath's return value, I'm not sure. Fixed en route.
Kindly tested by @khinsen on the forum, and it appears to work.
GitHub-Pull-Request: https://github.com/syncthing/syncthing/pull/4646
2018-01-05 19:11:09 +01:00
|
|
|
return normPath, false
|
2015-11-20 09:41:44 +01:00
|
|
|
}
|
|
|
|
|
2015-11-17 21:08:36 +01:00
|
|
|
// A byteCounter gets bytes added to it via Update() and then provides the
|
|
|
|
// Total() and one minute moving average Rate() in bytes per second.
|
|
|
|
type byteCounter struct {
|
|
|
|
total int64
|
|
|
|
metrics.EWMA
|
|
|
|
stop chan struct{}
|
|
|
|
}
|
|
|
|
|
|
|
|
func newByteCounter() *byteCounter {
|
|
|
|
c := &byteCounter{
|
|
|
|
EWMA: metrics.NewEWMA1(), // a one minute exponentially weighted moving average
|
|
|
|
stop: make(chan struct{}),
|
|
|
|
}
|
|
|
|
go c.ticker()
|
|
|
|
return c
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *byteCounter) ticker() {
|
|
|
|
// The metrics.EWMA expects clock ticks every five seconds in order to
|
|
|
|
// decay the average properly.
|
|
|
|
t := time.NewTicker(5 * time.Second)
|
|
|
|
for {
|
|
|
|
select {
|
|
|
|
case <-t.C:
|
|
|
|
c.Tick()
|
|
|
|
case <-c.stop:
|
|
|
|
t.Stop()
|
|
|
|
return
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *byteCounter) Update(bytes int64) {
|
|
|
|
atomic.AddInt64(&c.total, bytes)
|
|
|
|
c.EWMA.Update(bytes)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *byteCounter) Total() int64 {
|
|
|
|
return atomic.LoadInt64(&c.total)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (c *byteCounter) Close() {
|
|
|
|
close(c.stop)
|
|
|
|
}
|
2016-05-09 20:25:39 +02:00
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
// A no-op CurrentFiler
|
|
|
|
|
|
|
|
type noCurrentFiler struct{}
|
2016-05-09 20:25:39 +02:00
|
|
|
|
2018-02-14 08:59:46 +01:00
|
|
|
func (noCurrentFiler) CurrentFile(name string) (protocol.FileInfo, bool) {
|
|
|
|
return protocol.FileInfo{}, false
|
|
|
|
}
|