| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067 |
- // Copyright (C) 2014 The Syncthing Authors.
- //
- // 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,
- // You can obtain one at http://mozilla.org/MPL/2.0/.
- //go:generate -command genxdr go run ../../Godeps/_workspace/src/github.com/calmh/xdr/cmd/genxdr/main.go
- //go:generate genxdr -o leveldb_xdr.go leveldb.go
- package db
- import (
- "bytes"
- "fmt"
- "runtime"
- "sort"
- "github.com/syncthing/protocol"
- "github.com/syncthing/syncthing/internal/sync"
- "github.com/syndtr/goleveldb/leveldb"
- "github.com/syndtr/goleveldb/leveldb/iterator"
- "github.com/syndtr/goleveldb/leveldb/opt"
- "github.com/syndtr/goleveldb/leveldb/util"
- )
- var (
- clockTick int64
- clockMut = sync.NewMutex()
- )
- func clock(v int64) int64 {
- clockMut.Lock()
- defer clockMut.Unlock()
- if v > clockTick {
- clockTick = v + 1
- } else {
- clockTick++
- }
- return clockTick
- }
- const (
- KeyTypeDevice = iota
- KeyTypeGlobal
- KeyTypeBlock
- KeyTypeDeviceStatistic
- KeyTypeFolderStatistic
- KeyTypeVirtualMtime
- )
- type fileVersion struct {
- version protocol.Vector
- device []byte
- }
- type versionList struct {
- versions []fileVersion
- }
- func (l versionList) String() string {
- var b bytes.Buffer
- var id protocol.DeviceID
- b.WriteString("{")
- for i, v := range l.versions {
- if i > 0 {
- b.WriteString(", ")
- }
- copy(id[:], v.device)
- fmt.Fprintf(&b, "{%d, %v}", v.version, id)
- }
- b.WriteString("}")
- return b.String()
- }
- type fileList []protocol.FileInfo
- func (l fileList) Len() int {
- return len(l)
- }
- func (l fileList) Swap(a, b int) {
- l[a], l[b] = l[b], l[a]
- }
- func (l fileList) Less(a, b int) bool {
- return l[a].Name < l[b].Name
- }
- type dbReader interface {
- Get([]byte, *opt.ReadOptions) ([]byte, error)
- }
- type dbWriter interface {
- Put([]byte, []byte)
- Delete([]byte)
- }
- // Flush batches to disk when they contain this many records.
- const batchFlushSize = 64
- // deviceKey returns a byte slice encoding the following information:
- // keyTypeDevice (1 byte)
- // folder (64 bytes)
- // device (32 bytes)
- // name (variable size)
- func deviceKey(folder, device, file []byte) []byte {
- return deviceKeyInto(nil, folder, device, file)
- }
- func deviceKeyInto(k []byte, folder, device, file []byte) []byte {
- reqLen := 1 + 64 + 32 + len(file)
- if len(k) < reqLen {
- k = make([]byte, reqLen)
- }
- k[0] = KeyTypeDevice
- if len(folder) > 64 {
- panic("folder name too long")
- }
- copy(k[1:], []byte(folder))
- copy(k[1+64:], device[:])
- copy(k[1+64+32:], []byte(file))
- return k[:reqLen]
- }
- func deviceKeyName(key []byte) []byte {
- return key[1+64+32:]
- }
- func deviceKeyFolder(key []byte) []byte {
- folder := key[1 : 1+64]
- izero := bytes.IndexByte(folder, 0)
- if izero < 0 {
- return folder
- }
- return folder[:izero]
- }
- func deviceKeyDevice(key []byte) []byte {
- return key[1+64 : 1+64+32]
- }
- // globalKey returns a byte slice encoding the following information:
- // keyTypeGlobal (1 byte)
- // folder (64 bytes)
- // name (variable size)
- func globalKey(folder, file []byte) []byte {
- k := make([]byte, 1+64+len(file))
- k[0] = KeyTypeGlobal
- if len(folder) > 64 {
- panic("folder name too long")
- }
- copy(k[1:], []byte(folder))
- copy(k[1+64:], []byte(file))
- return k
- }
- func globalKeyName(key []byte) []byte {
- return key[1+64:]
- }
- func globalKeyFolder(key []byte) []byte {
- folder := key[1 : 1+64]
- izero := bytes.IndexByte(folder, 0)
- if izero < 0 {
- return folder
- }
- return folder[:izero]
- }
- type deletionHandler func(db dbReader, batch dbWriter, folder, device, name []byte, dbi iterator.Iterator) int64
- func ldbGenericReplace(db *leveldb.DB, folder, device []byte, fs []protocol.FileInfo, deleteFn deletionHandler) int64 {
- runtime.GC()
- sort.Sort(fileList(fs)) // sort list on name, same as in the database
- start := deviceKey(folder, device, nil) // before all folder/device files
- limit := deviceKey(folder, device, []byte{0xff, 0xff, 0xff, 0xff}) // after all folder/device files
- batch := new(leveldb.Batch)
- if debugDB {
- l.Debugf("new batch %p", batch)
- }
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
- defer dbi.Release()
- moreDb := dbi.Next()
- fsi := 0
- var maxLocalVer int64
- for {
- var newName, oldName []byte
- moreFs := fsi < len(fs)
- if !moreDb && !moreFs {
- break
- }
- if moreFs {
- newName = []byte(fs[fsi].Name)
- }
- if moreDb {
- oldName = deviceKeyName(dbi.Key())
- }
- cmp := bytes.Compare(newName, oldName)
- if debugDB {
- l.Debugf("generic replace; folder=%q device=%v moreFs=%v moreDb=%v cmp=%d newName=%q oldName=%q", folder, protocol.DeviceIDFromBytes(device), moreFs, moreDb, cmp, newName, oldName)
- }
- switch {
- case moreFs && (!moreDb || cmp == -1):
- if debugDB {
- l.Debugln("generic replace; missing - insert")
- }
- // Database is missing this file. Insert it.
- if lv := ldbInsert(batch, folder, device, fs[fsi]); lv > maxLocalVer {
- maxLocalVer = lv
- }
- if fs[fsi].IsInvalid() {
- ldbRemoveFromGlobal(snap, batch, folder, device, newName)
- } else {
- ldbUpdateGlobal(snap, batch, folder, device, newName, fs[fsi].Version)
- }
- fsi++
- case moreFs && moreDb && cmp == 0:
- // File exists on both sides - compare versions. We might get an
- // update with the same version and different flags if a device has
- // marked a file as invalid, so handle that too.
- if debugDB {
- l.Debugln("generic replace; exists - compare")
- }
- var ef FileInfoTruncated
- ef.UnmarshalXDR(dbi.Value())
- if !fs[fsi].Version.Equal(ef.Version) || fs[fsi].Flags != ef.Flags {
- if debugDB {
- l.Debugln("generic replace; differs - insert")
- }
- if lv := ldbInsert(batch, folder, device, fs[fsi]); lv > maxLocalVer {
- maxLocalVer = lv
- }
- if fs[fsi].IsInvalid() {
- ldbRemoveFromGlobal(snap, batch, folder, device, newName)
- } else {
- ldbUpdateGlobal(snap, batch, folder, device, newName, fs[fsi].Version)
- }
- } else if debugDB {
- l.Debugln("generic replace; equal - ignore")
- }
- fsi++
- moreDb = dbi.Next()
- case moreDb && (!moreFs || cmp == 1):
- if debugDB {
- l.Debugln("generic replace; exists - remove")
- }
- if lv := deleteFn(snap, batch, folder, device, oldName, dbi); lv > maxLocalVer {
- maxLocalVer = lv
- }
- moreDb = dbi.Next()
- }
- // Write out and reuse the batch every few records, to avoid the batch
- // growing too large and thus allocating unnecessarily much memory.
- if batch.Len() > batchFlushSize {
- if debugDB {
- l.Debugf("db.Write %p", batch)
- }
- err = db.Write(batch, nil)
- if err != nil {
- panic(err)
- }
- batch.Reset()
- }
- }
- if debugDB {
- l.Debugf("db.Write %p", batch)
- }
- err = db.Write(batch, nil)
- if err != nil {
- panic(err)
- }
- return maxLocalVer
- }
- func ldbReplace(db *leveldb.DB, folder, device []byte, fs []protocol.FileInfo) int64 {
- // TODO: Return the remaining maxLocalVer?
- return ldbGenericReplace(db, folder, device, fs, func(db dbReader, batch dbWriter, folder, device, name []byte, dbi iterator.Iterator) int64 {
- // Database has a file that we are missing. Remove it.
- if debugDB {
- l.Debugf("delete; folder=%q device=%v name=%q", folder, protocol.DeviceIDFromBytes(device), name)
- }
- ldbRemoveFromGlobal(db, batch, folder, device, name)
- if debugDB {
- l.Debugf("batch.Delete %p %x", batch, dbi.Key())
- }
- batch.Delete(dbi.Key())
- return 0
- })
- }
- func ldbReplaceWithDelete(db *leveldb.DB, folder, device []byte, fs []protocol.FileInfo, myID uint64) int64 {
- mtimeRepo := NewVirtualMtimeRepo(db, string(folder))
- return ldbGenericReplace(db, folder, device, fs, func(db dbReader, batch dbWriter, folder, device, name []byte, dbi iterator.Iterator) int64 {
- var tf FileInfoTruncated
- err := tf.UnmarshalXDR(dbi.Value())
- if err != nil {
- panic(err)
- }
- if !tf.IsDeleted() {
- if debugDB {
- l.Debugf("mark deleted; folder=%q device=%v name=%q", folder, protocol.DeviceIDFromBytes(device), name)
- }
- ts := clock(tf.LocalVersion)
- f := protocol.FileInfo{
- Name: tf.Name,
- Version: tf.Version.Update(myID),
- LocalVersion: ts,
- Flags: tf.Flags | protocol.FlagDeleted,
- Modified: tf.Modified,
- }
- bs, _ := f.MarshalXDR()
- if debugDB {
- l.Debugf("batch.Put %p %x", batch, dbi.Key())
- }
- batch.Put(dbi.Key(), bs)
- mtimeRepo.DeleteMtime(tf.Name)
- ldbUpdateGlobal(db, batch, folder, device, deviceKeyName(dbi.Key()), f.Version)
- return ts
- }
- return 0
- })
- }
- func ldbUpdate(db *leveldb.DB, folder, device []byte, fs []protocol.FileInfo) int64 {
- runtime.GC()
- batch := new(leveldb.Batch)
- if debugDB {
- l.Debugf("new batch %p", batch)
- }
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- var maxLocalVer int64
- var fk []byte
- for _, f := range fs {
- name := []byte(f.Name)
- fk = deviceKeyInto(fk[:cap(fk)], folder, device, name)
- if debugDB {
- l.Debugf("snap.Get %p %x", snap, fk)
- }
- bs, err := snap.Get(fk, nil)
- if err == leveldb.ErrNotFound {
- if lv := ldbInsert(batch, folder, device, f); lv > maxLocalVer {
- maxLocalVer = lv
- }
- if f.IsInvalid() {
- ldbRemoveFromGlobal(snap, batch, folder, device, name)
- } else {
- ldbUpdateGlobal(snap, batch, folder, device, name, f.Version)
- }
- continue
- }
- var ef FileInfoTruncated
- err = ef.UnmarshalXDR(bs)
- if err != nil {
- panic(err)
- }
- // Flags might change without the version being bumped when we set the
- // invalid flag on an existing file.
- if !ef.Version.Equal(f.Version) || ef.Flags != f.Flags {
- if lv := ldbInsert(batch, folder, device, f); lv > maxLocalVer {
- maxLocalVer = lv
- }
- if f.IsInvalid() {
- ldbRemoveFromGlobal(snap, batch, folder, device, name)
- } else {
- ldbUpdateGlobal(snap, batch, folder, device, name, f.Version)
- }
- }
- // Write out and reuse the batch every few records, to avoid the batch
- // growing too large and thus allocating unnecessarily much memory.
- if batch.Len() > batchFlushSize {
- if debugDB {
- l.Debugf("db.Write %p", batch)
- }
- err = db.Write(batch, nil)
- if err != nil {
- panic(err)
- }
- batch.Reset()
- }
- }
- if debugDB {
- l.Debugf("db.Write %p", batch)
- }
- err = db.Write(batch, nil)
- if err != nil {
- panic(err)
- }
- return maxLocalVer
- }
- func ldbInsert(batch dbWriter, folder, device []byte, file protocol.FileInfo) int64 {
- if debugDB {
- l.Debugf("insert; folder=%q device=%v %v", folder, protocol.DeviceIDFromBytes(device), file)
- }
- if file.LocalVersion == 0 {
- file.LocalVersion = clock(0)
- }
- name := []byte(file.Name)
- nk := deviceKey(folder, device, name)
- if debugDB {
- l.Debugf("batch.Put %p %x", batch, nk)
- }
- batch.Put(nk, file.MustMarshalXDR())
- return file.LocalVersion
- }
- // ldbUpdateGlobal adds this device+version to the version list for the given
- // file. If the device is already present in the list, the version is updated.
- // If the file does not have an entry in the global list, it is created.
- func ldbUpdateGlobal(db dbReader, batch dbWriter, folder, device, file []byte, version protocol.Vector) bool {
- if debugDB {
- l.Debugf("update global; folder=%q device=%v file=%q version=%d", folder, protocol.DeviceIDFromBytes(device), file, version)
- }
- gk := globalKey(folder, file)
- svl, err := db.Get(gk, nil)
- if err != nil && err != leveldb.ErrNotFound {
- panic(err)
- }
- var fl versionList
- // Remove the device from the current version list
- if svl != nil {
- err = fl.UnmarshalXDR(svl)
- if err != nil {
- panic(err)
- }
- for i := range fl.versions {
- if bytes.Compare(fl.versions[i].device, device) == 0 {
- if fl.versions[i].version.Equal(version) {
- // No need to do anything
- return false
- }
- fl.versions = append(fl.versions[:i], fl.versions[i+1:]...)
- break
- }
- }
- }
- nv := fileVersion{
- device: device,
- version: version,
- }
- for i := range fl.versions {
- // We compare against ConcurrentLesser as well here because we need
- // to enforce a consistent ordering of versions even in the case of
- // conflicts.
- if comp := fl.versions[i].version.Compare(version); comp == protocol.Equal || comp == protocol.Lesser || comp == protocol.ConcurrentLesser {
- t := append(fl.versions, fileVersion{})
- copy(t[i+1:], t[i:])
- t[i] = nv
- fl.versions = t
- goto done
- }
- }
- fl.versions = append(fl.versions, nv)
- done:
- if debugDB {
- l.Debugf("batch.Put %p %x", batch, gk)
- l.Debugf("new global after update: %v", fl)
- }
- batch.Put(gk, fl.MustMarshalXDR())
- return true
- }
- // ldbRemoveFromGlobal removes the device from the global version list for the
- // given file. If the version list is empty after this, the file entry is
- // removed entirely.
- func ldbRemoveFromGlobal(db dbReader, batch dbWriter, folder, device, file []byte) {
- if debugDB {
- l.Debugf("remove from global; folder=%q device=%v file=%q", folder, protocol.DeviceIDFromBytes(device), file)
- }
- gk := globalKey(folder, file)
- svl, err := db.Get(gk, nil)
- if err != nil {
- // We might be called to "remove" a global version that doesn't exist
- // if the first update for the file is already marked invalid.
- return
- }
- var fl versionList
- err = fl.UnmarshalXDR(svl)
- if err != nil {
- panic(err)
- }
- for i := range fl.versions {
- if bytes.Compare(fl.versions[i].device, device) == 0 {
- fl.versions = append(fl.versions[:i], fl.versions[i+1:]...)
- break
- }
- }
- if len(fl.versions) == 0 {
- if debugDB {
- l.Debugf("batch.Delete %p %x", batch, gk)
- }
- batch.Delete(gk)
- } else {
- if debugDB {
- l.Debugf("batch.Put %p %x", batch, gk)
- l.Debugf("new global after remove: %v", fl)
- }
- batch.Put(gk, fl.MustMarshalXDR())
- }
- }
- func ldbWithHave(db *leveldb.DB, folder, device []byte, truncate bool, fn Iterator) {
- start := deviceKey(folder, device, nil) // before all folder/device files
- limit := deviceKey(folder, device, []byte{0xff, 0xff, 0xff, 0xff}) // after all folder/device files
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
- defer dbi.Release()
- for dbi.Next() {
- f, err := unmarshalTrunc(dbi.Value(), truncate)
- if err != nil {
- panic(err)
- }
- if cont := fn(f); !cont {
- return
- }
- }
- }
- func ldbWithAllFolderTruncated(db *leveldb.DB, folder []byte, fn func(device []byte, f FileInfoTruncated) bool) {
- runtime.GC()
- start := deviceKey(folder, nil, nil) // before all folder/device files
- limit := deviceKey(folder, protocol.LocalDeviceID[:], []byte{0xff, 0xff, 0xff, 0xff}) // after all folder/device files
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
- defer dbi.Release()
- for dbi.Next() {
- device := deviceKeyDevice(dbi.Key())
- var f FileInfoTruncated
- err := f.UnmarshalXDR(dbi.Value())
- if err != nil {
- panic(err)
- }
- switch f.Name {
- case "", ".", "..", "/": // A few obviously invalid filenames
- l.Infof("Dropping invalid filename %q from database", f.Name)
- batch := new(leveldb.Batch)
- ldbRemoveFromGlobal(db, batch, folder, device, nil)
- batch.Delete(dbi.Key())
- db.Write(batch, nil)
- continue
- }
- if cont := fn(device, f); !cont {
- return
- }
- }
- }
- func ldbGet(db *leveldb.DB, folder, device, file []byte) (protocol.FileInfo, bool) {
- nk := deviceKey(folder, device, file)
- bs, err := db.Get(nk, nil)
- if err == leveldb.ErrNotFound {
- return protocol.FileInfo{}, false
- }
- if err != nil {
- panic(err)
- }
- var f protocol.FileInfo
- err = f.UnmarshalXDR(bs)
- if err != nil {
- panic(err)
- }
- return f, true
- }
- func ldbGetGlobal(db *leveldb.DB, folder, file []byte, truncate bool) (FileIntf, bool) {
- k := globalKey(folder, file)
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- if debugDB {
- l.Debugf("snap.Get %p %x", snap, k)
- }
- bs, err := snap.Get(k, nil)
- if err == leveldb.ErrNotFound {
- return nil, false
- }
- if err != nil {
- panic(err)
- }
- var vl versionList
- err = vl.UnmarshalXDR(bs)
- if err != nil {
- panic(err)
- }
- if len(vl.versions) == 0 {
- l.Debugln(k)
- panic("no versions?")
- }
- k = deviceKey(folder, vl.versions[0].device, file)
- if debugDB {
- l.Debugf("snap.Get %p %x", snap, k)
- }
- bs, err = snap.Get(k, nil)
- if err != nil {
- panic(err)
- }
- fi, err := unmarshalTrunc(bs, truncate)
- if err != nil {
- panic(err)
- }
- return fi, true
- }
- func ldbWithGlobal(db *leveldb.DB, folder, prefix []byte, truncate bool, fn Iterator) {
- runtime.GC()
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- dbi := snap.NewIterator(util.BytesPrefix(globalKey(folder, prefix)), nil)
- defer dbi.Release()
- var fk []byte
- for dbi.Next() {
- var vl versionList
- err := vl.UnmarshalXDR(dbi.Value())
- if err != nil {
- panic(err)
- }
- if len(vl.versions) == 0 {
- l.Debugln(dbi.Key())
- panic("no versions?")
- }
- name := globalKeyName(dbi.Key())
- fk = deviceKeyInto(fk[:cap(fk)], folder, vl.versions[0].device, name)
- if debugDB {
- l.Debugf("snap.Get %p %x", snap, fk)
- }
- bs, err := snap.Get(fk, nil)
- if err != nil {
- l.Debugf("folder: %q (%x)", folder, folder)
- l.Debugf("key: %q (%x)", dbi.Key(), dbi.Key())
- l.Debugf("vl: %v", vl)
- l.Debugf("vl.versions[0].device: %x", vl.versions[0].device)
- l.Debugf("name: %q (%x)", name, name)
- l.Debugf("fk: %q", fk)
- l.Debugf("fk: %x %x %x", fk[1:1+64], fk[1+64:1+64+32], fk[1+64+32:])
- panic(err)
- }
- f, err := unmarshalTrunc(bs, truncate)
- if err != nil {
- panic(err)
- }
- if cont := fn(f); !cont {
- return
- }
- }
- }
- func ldbAvailability(db *leveldb.DB, folder, file []byte) []protocol.DeviceID {
- k := globalKey(folder, file)
- bs, err := db.Get(k, nil)
- if err == leveldb.ErrNotFound {
- return nil
- }
- if err != nil {
- panic(err)
- }
- var vl versionList
- err = vl.UnmarshalXDR(bs)
- if err != nil {
- panic(err)
- }
- var devices []protocol.DeviceID
- for _, v := range vl.versions {
- if !v.version.Equal(vl.versions[0].version) {
- break
- }
- n := protocol.DeviceIDFromBytes(v.device)
- devices = append(devices, n)
- }
- return devices
- }
- func ldbWithNeed(db *leveldb.DB, folder, device []byte, truncate bool, fn Iterator) {
- runtime.GC()
- start := globalKey(folder, nil)
- limit := globalKey(folder, []byte{0xff, 0xff, 0xff, 0xff})
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
- defer dbi.Release()
- var fk []byte
- nextFile:
- for dbi.Next() {
- var vl versionList
- err := vl.UnmarshalXDR(dbi.Value())
- if err != nil {
- panic(err)
- }
- if len(vl.versions) == 0 {
- l.Debugln(dbi.Key())
- panic("no versions?")
- }
- have := false // If we have the file, any version
- need := false // If we have a lower version of the file
- var haveVersion protocol.Vector
- for _, v := range vl.versions {
- if bytes.Compare(v.device, device) == 0 {
- have = true
- haveVersion = v.version
- // XXX: This marks Concurrent (i.e. conflicting) changes as
- // needs. Maybe we should do that, but it needs special
- // handling in the puller.
- need = !v.version.GreaterEqual(vl.versions[0].version)
- break
- }
- }
- if need || !have {
- name := globalKeyName(dbi.Key())
- needVersion := vl.versions[0].version
- nextVersion:
- for i := range vl.versions {
- if !vl.versions[i].version.Equal(needVersion) {
- // We haven't found a valid copy of the file with the needed version.
- continue nextFile
- }
- fk = deviceKeyInto(fk[:cap(fk)], folder, vl.versions[i].device, name)
- if debugDB {
- l.Debugf("snap.Get %p %x", snap, fk)
- }
- bs, err := snap.Get(fk, nil)
- if err != nil {
- var id protocol.DeviceID
- copy(id[:], device)
- l.Debugf("device: %v", id)
- l.Debugf("need: %v, have: %v", need, have)
- l.Debugf("key: %q (%x)", dbi.Key(), dbi.Key())
- l.Debugf("vl: %v", vl)
- l.Debugf("i: %v", i)
- l.Debugf("fk: %q (%x)", fk, fk)
- l.Debugf("name: %q (%x)", name, name)
- panic(err)
- }
- gf, err := unmarshalTrunc(bs, truncate)
- if err != nil {
- panic(err)
- }
- if gf.IsInvalid() {
- // The file is marked invalid for whatever reason, don't use it.
- continue nextVersion
- }
- if gf.IsDeleted() && !have {
- // We don't need deleted files that we don't have
- continue nextFile
- }
- if debugDB {
- l.Debugf("need folder=%q device=%v name=%q need=%v have=%v haveV=%d globalV=%d", folder, protocol.DeviceIDFromBytes(device), name, need, have, haveVersion, vl.versions[0].version)
- }
- if cont := fn(gf); !cont {
- return
- }
- // This file is handled, no need to look further in the version list
- continue nextFile
- }
- }
- }
- }
- func ldbListFolders(db *leveldb.DB) []string {
- runtime.GC()
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- dbi := snap.NewIterator(util.BytesPrefix([]byte{KeyTypeGlobal}), nil)
- defer dbi.Release()
- folderExists := make(map[string]bool)
- for dbi.Next() {
- folder := string(globalKeyFolder(dbi.Key()))
- if !folderExists[folder] {
- folderExists[folder] = true
- }
- }
- folders := make([]string, 0, len(folderExists))
- for k := range folderExists {
- folders = append(folders, k)
- }
- sort.Strings(folders)
- return folders
- }
- func ldbDropFolder(db *leveldb.DB, folder []byte) {
- runtime.GC()
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- // Remove all items related to the given folder from the device->file bucket
- dbi := snap.NewIterator(util.BytesPrefix([]byte{KeyTypeDevice}), nil)
- for dbi.Next() {
- itemFolder := deviceKeyFolder(dbi.Key())
- if bytes.Compare(folder, itemFolder) == 0 {
- db.Delete(dbi.Key(), nil)
- }
- }
- dbi.Release()
- // Remove all items related to the given folder from the global bucket
- dbi = snap.NewIterator(util.BytesPrefix([]byte{KeyTypeGlobal}), nil)
- for dbi.Next() {
- itemFolder := globalKeyFolder(dbi.Key())
- if bytes.Compare(folder, itemFolder) == 0 {
- db.Delete(dbi.Key(), nil)
- }
- }
- dbi.Release()
- }
- func unmarshalTrunc(bs []byte, truncate bool) (FileIntf, error) {
- if truncate {
- var tf FileInfoTruncated
- err := tf.UnmarshalXDR(bs)
- return tf, err
- }
- var tf protocol.FileInfo
- err := tf.UnmarshalXDR(bs)
- return tf, err
- }
- func ldbCheckGlobals(db *leveldb.DB, folder []byte) {
- defer runtime.GC()
- snap, err := db.GetSnapshot()
- if err != nil {
- panic(err)
- }
- if debugDB {
- l.Debugf("created snapshot %p", snap)
- }
- defer func() {
- if debugDB {
- l.Debugf("close snapshot %p", snap)
- }
- snap.Release()
- }()
- start := globalKey(folder, nil)
- limit := globalKey(folder, []byte{0xff, 0xff, 0xff, 0xff})
- dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
- defer dbi.Release()
- batch := new(leveldb.Batch)
- if debugDB {
- l.Debugf("new batch %p", batch)
- }
- var fk []byte
- for dbi.Next() {
- gk := dbi.Key()
- var vl versionList
- err := vl.UnmarshalXDR(dbi.Value())
- if err != nil {
- panic(err)
- }
- // Check the global version list for consistency. An issue in previous
- // versions of goleveldb could result in reordered writes so that
- // there are global entries pointing to no longer existing files. Here
- // we find those and clear them out.
- name := globalKeyName(gk)
- var newVL versionList
- for _, version := range vl.versions {
- fk = deviceKeyInto(fk[:cap(fk)], folder, version.device, name)
- if debugDB {
- l.Debugf("snap.Get %p %x", snap, fk)
- }
- _, err := snap.Get(fk, nil)
- if err == leveldb.ErrNotFound {
- continue
- }
- if err != nil {
- panic(err)
- }
- newVL.versions = append(newVL.versions, version)
- }
- if len(newVL.versions) != len(vl.versions) {
- l.Infof("db repair: rewriting global version list for %x %x", gk[1:1+64], gk[1+64:])
- batch.Put(dbi.Key(), newVL.MustMarshalXDR())
- }
- }
- if debugDB {
- l.Infoln("db check completed for %q", folder)
- }
- db.Write(batch, nil)
- }
|