forked from keybase/kbfs
-
Notifications
You must be signed in to change notification settings - Fork 0
/
updates_file.go
70 lines (63 loc) · 1.92 KB
/
updates_file.go
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
// Copyright 2016 Keybase Inc. All rights reserved.
// Use of this source code is governed by a BSD
// license that can be found in the LICENSE file.
package libdokan
import (
"errors"
"github.com/keybase/kbfs/dokan"
"github.com/keybase/kbfs/libkbfs"
"golang.org/x/net/context"
)
// UpdatesFile represents a write-only file where any write of at
// least one byte triggers either disabling remote folder updates and
// conflict resolution, or re-enabling both. It is mainly useful for
// testing.
type UpdatesFile struct {
folder *Folder
enable bool
specialWriteFile
}
// WriteFile performs writes for dokan.
func (f *UpdatesFile) WriteFile(ctx context.Context, fi *dokan.FileInfo, bs []byte, offset int64) (n int, err error) {
f.folder.fs.logEnter(ctx, "UpdatesFile WriteFile")
defer func() { f.folder.fs.reportErr(ctx, libkbfs.WriteMode, err) }()
f.folder.fs.log.CDebugf(ctx, "UpdatesFile (enable: %t) Write", f.enable)
if len(bs) == 0 {
return 0, nil
}
f.folder.updateMu.Lock()
defer f.folder.updateMu.Unlock()
if f.enable {
if f.folder.updateChan == nil {
return 0, errors.New("Updates are already enabled")
}
err = libkbfs.RestartCRForTesting(
libkbfs.BackgroundContextWithCancellationDelayer(),
f.folder.fs.config, f.folder.getFolderBranch())
if err != nil {
return 0, err
}
f.folder.updateChan <- struct{}{}
close(f.folder.updateChan)
f.folder.updateChan = nil
} else {
if f.folder.updateChan != nil {
return 0, errors.New("Updates are already disabled")
}
f.folder.updateChan, err =
libkbfs.DisableUpdatesForTesting(f.folder.fs.config,
f.folder.getFolderBranch())
if err != nil {
return 0, err
}
err = libkbfs.DisableCRForTesting(f.folder.fs.config,
f.folder.getFolderBranch())
if err != nil {
return 0, err
}
}
// Because we store state in the folder it must not be forgotten
// even if it appears empty and unused.
f.folder.noForget = true
return len(bs), err
}