blob: dfd963763cd49d9d740c3c1c0dc7e704981abc07 (
plain)
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
// Package openfiletable maintains a table of currently opened files, identified
// by the device number + inode number pair. This table is used by fusefrontend
// to centrally store the current file ID and to lock files against concurrent
// writes.
package openfiletable
import (
"sync"
"sync/atomic"
"github.com/rfjakob/gocryptfs/internal/inomap"
)
// wlock - serializes write accesses to each file (identified by inode number)
// Writing partial blocks means we have to do read-modify-write cycles. We
// really don't want concurrent writes there.
// Concurrent full-block writes could actually be allowed, but are not to
// keep the locking simple.
var t table
func init() {
t.entries = make(map[inomap.QIno]*Entry)
}
type table struct {
// writeOpCount counts entry.ContentLock.Lock() calls. As every operation that
// modifies a file should
// call it, this effectively serves as a write-operation counter.
// The variable is accessed without holding any locks so atomic operations
// must be used. It must be the first element of the struct to guarantee
// 64-bit alignment.
writeOpCount uint64
// Protects map access
sync.Mutex
// Table entries
entries map[inomap.QIno]*Entry
}
// Entry is an entry in the open file table
type Entry struct {
// Reference count. Protected by the table lock.
refCount int
// ContentLock protects on-disk content from concurrent writes. Every writer
// must take this lock before modifying the file content.
ContentLock countingMutex
// ID is the file ID in the file header.
ID []byte
// IDLock must be taken before reading or writing the ID field in this struct,
// unless you have an exclusive lock on ContentLock.
IDLock sync.Mutex
}
// Register creates an open file table entry for "qi" (or incrementes the
// reference count if the entry already exists) and returns the entry.
func Register(qi inomap.QIno) *Entry {
t.Lock()
defer t.Unlock()
e := t.entries[qi]
if e == nil {
e = &Entry{}
t.entries[qi] = e
}
e.refCount++
return e
}
// Unregister decrements the reference count for "qi" and deletes the entry from
// the open file table if the reference count reaches 0.
func Unregister(qi inomap.QIno) {
t.Lock()
defer t.Unlock()
e := t.entries[qi]
e.refCount--
if e.refCount == 0 {
delete(t.entries, qi)
}
}
// countingMutex incrementes t.writeLockCount on each Lock() call.
type countingMutex struct {
sync.RWMutex
}
func (c *countingMutex) Lock() {
c.RWMutex.Lock()
atomic.AddUint64(&t.writeOpCount, 1)
}
// WriteOpCount returns the write lock counter value. This value is incremented
// each time writeLock.Lock() on a file table entry is called.
func WriteOpCount() uint64 {
return atomic.LoadUint64(&t.writeOpCount)
}
// CountOpenFiles returns how many entries are currently in the table
// in a threadsafe manner.
func CountOpenFiles() int {
t.Lock()
defer t.Unlock()
return len(t.entries)
}
|
