lowlevel.go 32 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258
  1. // Copyright (C) 2014 The Syncthing Authors.
  2. //
  3. // This Source Code Form is subject to the terms of the Mozilla Public
  4. // License, v. 2.0. If a copy of the MPL was not distributed with this file,
  5. // You can obtain one at https://mozilla.org/MPL/2.0/.
  6. package db
  7. import (
  8. "bytes"
  9. "context"
  10. "encoding/binary"
  11. "errors"
  12. "fmt"
  13. "io"
  14. "os"
  15. "regexp"
  16. "time"
  17. "github.com/dchest/siphash"
  18. "github.com/greatroar/blobloom"
  19. "github.com/syncthing/syncthing/lib/db/backend"
  20. "github.com/syncthing/syncthing/lib/events"
  21. "github.com/syncthing/syncthing/lib/fs"
  22. "github.com/syncthing/syncthing/lib/protocol"
  23. "github.com/syncthing/syncthing/lib/rand"
  24. "github.com/syncthing/syncthing/lib/sha256"
  25. "github.com/syncthing/syncthing/lib/svcutil"
  26. "github.com/syncthing/syncthing/lib/sync"
  27. "github.com/syncthing/syncthing/lib/util"
  28. "github.com/thejerf/suture/v4"
  29. )
  30. const (
  31. // We set the bloom filter capacity to handle 100k individual items with
  32. // a false positive probability of 1% for the first pass. Once we know
  33. // how many items we have we will use that number instead, if it's more
  34. // than 100k. For fewer than 100k items we will just get better false
  35. // positive rate instead.
  36. indirectGCBloomCapacity = 100000
  37. indirectGCBloomFalsePositiveRate = 0.01 // 1%
  38. indirectGCBloomMaxBytes = 32 << 20 // Use at most 32MiB memory, which covers our desired FP rate at 27 M items
  39. indirectGCDefaultInterval = 13 * time.Hour
  40. indirectGCTimeKey = "lastIndirectGCTime"
  41. // Use indirection for the block list when it exceeds this many entries
  42. blocksIndirectionCutoff = 3
  43. // Use indirection for the version vector when it exceeds this many entries
  44. versionIndirectionCutoff = 10
  45. recheckDefaultInterval = 30 * 24 * time.Hour
  46. needsRepairSuffix = ".needsrepair"
  47. )
  48. // Lowlevel is the lowest level database interface. It has a very simple
  49. // purpose: hold the actual backend database, and the in-memory state
  50. // that belong to that database. In the same way that a single on disk
  51. // database can only be opened once, there should be only one Lowlevel for
  52. // any given backend.
  53. type Lowlevel struct {
  54. *suture.Supervisor
  55. backend.Backend
  56. folderIdx *smallIndex
  57. deviceIdx *smallIndex
  58. keyer keyer
  59. gcMut sync.RWMutex
  60. gcKeyCount int
  61. indirectGCInterval time.Duration
  62. recheckInterval time.Duration
  63. oneFileSetCreated chan struct{}
  64. evLogger events.Logger
  65. }
  66. func NewLowlevel(backend backend.Backend, evLogger events.Logger, opts ...Option) (*Lowlevel, error) {
  67. // Only log restarts in debug mode.
  68. spec := svcutil.SpecWithDebugLogger(l)
  69. db := &Lowlevel{
  70. Supervisor: suture.New("db.Lowlevel", spec),
  71. Backend: backend,
  72. folderIdx: newSmallIndex(backend, []byte{KeyTypeFolderIdx}),
  73. deviceIdx: newSmallIndex(backend, []byte{KeyTypeDeviceIdx}),
  74. gcMut: sync.NewRWMutex(),
  75. indirectGCInterval: indirectGCDefaultInterval,
  76. recheckInterval: recheckDefaultInterval,
  77. oneFileSetCreated: make(chan struct{}),
  78. evLogger: evLogger,
  79. }
  80. for _, opt := range opts {
  81. opt(db)
  82. }
  83. db.keyer = newDefaultKeyer(db.folderIdx, db.deviceIdx)
  84. db.Add(svcutil.AsService(db.gcRunner, "db.Lowlevel/gcRunner"))
  85. if path := db.needsRepairPath(); path != "" {
  86. if _, err := os.Lstat(path); err == nil {
  87. l.Infoln("Database was marked for repair - this may take a while")
  88. if err := db.checkRepair(); err != nil {
  89. db.handleFailure(err)
  90. return nil, err
  91. }
  92. os.Remove(path)
  93. }
  94. }
  95. return db, nil
  96. }
  97. type Option func(*Lowlevel)
  98. // WithRecheckInterval sets the time interval in between metadata recalculations
  99. // and consistency checks.
  100. func WithRecheckInterval(dur time.Duration) Option {
  101. return func(db *Lowlevel) {
  102. if dur > 0 {
  103. db.recheckInterval = dur
  104. }
  105. }
  106. }
  107. // WithIndirectGCInterval sets the time interval in between GC runs.
  108. func WithIndirectGCInterval(dur time.Duration) Option {
  109. return func(db *Lowlevel) {
  110. if dur > 0 {
  111. db.indirectGCInterval = dur
  112. }
  113. }
  114. }
  115. // ListFolders returns the list of folders currently in the database
  116. func (db *Lowlevel) ListFolders() []string {
  117. return db.folderIdx.Values()
  118. }
  119. // updateRemoteFiles adds a list of fileinfos to the database and updates the
  120. // global versionlist and metadata.
  121. func (db *Lowlevel) updateRemoteFiles(folder, device []byte, fs []protocol.FileInfo, meta *metadataTracker) error {
  122. db.gcMut.RLock()
  123. defer db.gcMut.RUnlock()
  124. t, err := db.newReadWriteTransaction(meta.CommitHook(folder))
  125. if err != nil {
  126. return err
  127. }
  128. defer t.close()
  129. var dk, gk, keyBuf []byte
  130. devID, err := protocol.DeviceIDFromBytes(device)
  131. if err != nil {
  132. return err
  133. }
  134. for _, f := range fs {
  135. name := []byte(f.Name)
  136. dk, err = db.keyer.GenerateDeviceFileKey(dk, folder, device, name)
  137. if err != nil {
  138. return err
  139. }
  140. ef, ok, err := t.getFileTrunc(dk, true)
  141. if err != nil {
  142. return err
  143. }
  144. if ok && unchanged(f, ef) {
  145. l.Debugf("not inserting unchanged (remote); folder=%q device=%v %v", folder, devID, f)
  146. continue
  147. }
  148. if ok {
  149. meta.removeFile(devID, ef)
  150. }
  151. meta.addFile(devID, f)
  152. l.Debugf("insert (remote); folder=%q device=%v %v", folder, devID, f)
  153. if err := t.putFile(dk, f); err != nil {
  154. return err
  155. }
  156. gk, err = db.keyer.GenerateGlobalVersionKey(gk, folder, name)
  157. if err != nil {
  158. return err
  159. }
  160. keyBuf, _, err = t.updateGlobal(gk, keyBuf, folder, device, f, meta)
  161. if err != nil {
  162. return err
  163. }
  164. if err := t.Checkpoint(); err != nil {
  165. return err
  166. }
  167. }
  168. return t.Commit()
  169. }
  170. // updateLocalFiles adds fileinfos to the db, and updates the global versionlist,
  171. // metadata, sequence and blockmap buckets.
  172. func (db *Lowlevel) updateLocalFiles(folder []byte, fs []protocol.FileInfo, meta *metadataTracker) error {
  173. db.gcMut.RLock()
  174. defer db.gcMut.RUnlock()
  175. t, err := db.newReadWriteTransaction(meta.CommitHook(folder))
  176. if err != nil {
  177. return err
  178. }
  179. defer t.close()
  180. var dk, gk, keyBuf []byte
  181. blockBuf := make([]byte, 4)
  182. for _, f := range fs {
  183. name := []byte(f.Name)
  184. dk, err = db.keyer.GenerateDeviceFileKey(dk, folder, protocol.LocalDeviceID[:], name)
  185. if err != nil {
  186. return err
  187. }
  188. ef, ok, err := t.getFileByKey(dk)
  189. if err != nil {
  190. return err
  191. }
  192. if ok && unchanged(f, ef) {
  193. l.Debugf("not inserting unchanged (local); folder=%q %v", folder, f)
  194. continue
  195. }
  196. blocksHashSame := ok && bytes.Equal(ef.BlocksHash, f.BlocksHash)
  197. if ok {
  198. if len(ef.Blocks) != 0 && !ef.IsInvalid() && ef.Size > 0 {
  199. for _, block := range ef.Blocks {
  200. keyBuf, err = db.keyer.GenerateBlockMapKey(keyBuf, folder, block.Hash, name)
  201. if err != nil {
  202. return err
  203. }
  204. if err := t.Delete(keyBuf); err != nil {
  205. return err
  206. }
  207. }
  208. if !blocksHashSame {
  209. keyBuf, err := db.keyer.GenerateBlockListMapKey(keyBuf, folder, ef.BlocksHash, name)
  210. if err != nil {
  211. return err
  212. }
  213. if err = t.Delete(keyBuf); err != nil {
  214. return err
  215. }
  216. }
  217. }
  218. keyBuf, err = db.keyer.GenerateSequenceKey(keyBuf, folder, ef.SequenceNo())
  219. if err != nil {
  220. return err
  221. }
  222. if err := t.Delete(keyBuf); err != nil {
  223. return err
  224. }
  225. l.Debugf("removing sequence; folder=%q sequence=%v %v", folder, ef.SequenceNo(), ef.FileName())
  226. }
  227. f.Sequence = meta.nextLocalSeq()
  228. if ok {
  229. meta.removeFile(protocol.LocalDeviceID, ef)
  230. }
  231. meta.addFile(protocol.LocalDeviceID, f)
  232. l.Debugf("insert (local); folder=%q %v", folder, f)
  233. if err := t.putFile(dk, f); err != nil {
  234. return err
  235. }
  236. gk, err = db.keyer.GenerateGlobalVersionKey(gk, folder, []byte(f.Name))
  237. if err != nil {
  238. return err
  239. }
  240. keyBuf, _, err = t.updateGlobal(gk, keyBuf, folder, protocol.LocalDeviceID[:], f, meta)
  241. if err != nil {
  242. return err
  243. }
  244. keyBuf, err = db.keyer.GenerateSequenceKey(keyBuf, folder, f.Sequence)
  245. if err != nil {
  246. return err
  247. }
  248. if err := t.Put(keyBuf, dk); err != nil {
  249. return err
  250. }
  251. l.Debugf("adding sequence; folder=%q sequence=%v %v", folder, f.Sequence, f.Name)
  252. if len(f.Blocks) != 0 && !f.IsInvalid() && f.Size > 0 {
  253. for i, block := range f.Blocks {
  254. binary.BigEndian.PutUint32(blockBuf, uint32(i))
  255. keyBuf, err = db.keyer.GenerateBlockMapKey(keyBuf, folder, block.Hash, name)
  256. if err != nil {
  257. return err
  258. }
  259. if err := t.Put(keyBuf, blockBuf); err != nil {
  260. return err
  261. }
  262. }
  263. if !blocksHashSame {
  264. keyBuf, err := db.keyer.GenerateBlockListMapKey(keyBuf, folder, f.BlocksHash, name)
  265. if err != nil {
  266. return err
  267. }
  268. if err = t.Put(keyBuf, nil); err != nil {
  269. return err
  270. }
  271. }
  272. }
  273. if err := t.Checkpoint(); err != nil {
  274. return err
  275. }
  276. }
  277. return t.Commit()
  278. }
  279. func (db *Lowlevel) dropFolder(folder []byte) error {
  280. db.gcMut.RLock()
  281. defer db.gcMut.RUnlock()
  282. t, err := db.newReadWriteTransaction()
  283. if err != nil {
  284. return err
  285. }
  286. defer t.close()
  287. // Remove all items related to the given folder from the device->file bucket
  288. k0, err := db.keyer.GenerateDeviceFileKey(nil, folder, nil, nil)
  289. if err != nil {
  290. return err
  291. }
  292. if err := t.deleteKeyPrefix(k0.WithoutNameAndDevice()); err != nil {
  293. return err
  294. }
  295. // Remove all sequences related to the folder
  296. k1, err := db.keyer.GenerateSequenceKey(k0, folder, 0)
  297. if err != nil {
  298. return err
  299. }
  300. if err := t.deleteKeyPrefix(k1.WithoutSequence()); err != nil {
  301. return err
  302. }
  303. // Remove all items related to the given folder from the global bucket
  304. k2, err := db.keyer.GenerateGlobalVersionKey(k1, folder, nil)
  305. if err != nil {
  306. return err
  307. }
  308. if err := t.deleteKeyPrefix(k2.WithoutName()); err != nil {
  309. return err
  310. }
  311. // Remove all needs related to the folder
  312. k3, err := db.keyer.GenerateNeedFileKey(k2, folder, nil)
  313. if err != nil {
  314. return err
  315. }
  316. if err := t.deleteKeyPrefix(k3.WithoutName()); err != nil {
  317. return err
  318. }
  319. // Remove the blockmap of the folder
  320. k4, err := db.keyer.GenerateBlockMapKey(k3, folder, nil, nil)
  321. if err != nil {
  322. return err
  323. }
  324. if err := t.deleteKeyPrefix(k4.WithoutHashAndName()); err != nil {
  325. return err
  326. }
  327. k5, err := db.keyer.GenerateBlockListMapKey(k4, folder, nil, nil)
  328. if err != nil {
  329. return err
  330. }
  331. if err := t.deleteKeyPrefix(k5.WithoutHashAndName()); err != nil {
  332. return err
  333. }
  334. return t.Commit()
  335. }
  336. func (db *Lowlevel) dropDeviceFolder(device, folder []byte, meta *metadataTracker) error {
  337. db.gcMut.RLock()
  338. defer db.gcMut.RUnlock()
  339. t, err := db.newReadWriteTransaction(meta.CommitHook(folder))
  340. if err != nil {
  341. return err
  342. }
  343. defer t.close()
  344. key, err := db.keyer.GenerateDeviceFileKey(nil, folder, device, nil)
  345. if err != nil {
  346. return err
  347. }
  348. dbi, err := t.NewPrefixIterator(key)
  349. if err != nil {
  350. return err
  351. }
  352. defer dbi.Release()
  353. var gk, keyBuf []byte
  354. for dbi.Next() {
  355. name := db.keyer.NameFromDeviceFileKey(dbi.Key())
  356. gk, err = db.keyer.GenerateGlobalVersionKey(gk, folder, name)
  357. if err != nil {
  358. return err
  359. }
  360. keyBuf, err = t.removeFromGlobal(gk, keyBuf, folder, device, name, meta)
  361. if err != nil {
  362. return err
  363. }
  364. if err := t.Delete(dbi.Key()); err != nil {
  365. return err
  366. }
  367. if err := t.Checkpoint(); err != nil {
  368. return err
  369. }
  370. }
  371. dbi.Release()
  372. if err := dbi.Error(); err != nil {
  373. return err
  374. }
  375. if bytes.Equal(device, protocol.LocalDeviceID[:]) {
  376. key, err := db.keyer.GenerateBlockMapKey(nil, folder, nil, nil)
  377. if err != nil {
  378. return err
  379. }
  380. if err := t.deleteKeyPrefix(key.WithoutHashAndName()); err != nil {
  381. return err
  382. }
  383. key2, err := db.keyer.GenerateBlockListMapKey(key, folder, nil, nil)
  384. if err != nil {
  385. return err
  386. }
  387. if err := t.deleteKeyPrefix(key2.WithoutHashAndName()); err != nil {
  388. return err
  389. }
  390. }
  391. return t.Commit()
  392. }
  393. func (db *Lowlevel) checkGlobals(folderStr string) (int, error) {
  394. t, err := db.newReadWriteTransaction()
  395. if err != nil {
  396. return 0, err
  397. }
  398. defer t.close()
  399. folder := []byte(folderStr)
  400. key, err := db.keyer.GenerateGlobalVersionKey(nil, folder, nil)
  401. if err != nil {
  402. return 0, err
  403. }
  404. dbi, err := t.NewPrefixIterator(key.WithoutName())
  405. if err != nil {
  406. return 0, err
  407. }
  408. defer dbi.Release()
  409. fixed := 0
  410. var dk []byte
  411. ro := t.readOnlyTransaction
  412. for dbi.Next() {
  413. var vl VersionList
  414. if err := vl.Unmarshal(dbi.Value()); err != nil || vl.Empty() {
  415. if err := t.Delete(dbi.Key()); err != nil && !backend.IsNotFound(err) {
  416. return 0, err
  417. }
  418. continue
  419. }
  420. // Check the global version list for consistency. An issue in previous
  421. // versions of goleveldb could result in reordered writes so that
  422. // there are global entries pointing to no longer existing files. Here
  423. // we find those and clear them out.
  424. name := db.keyer.NameFromGlobalVersionKey(dbi.Key())
  425. newVL := &VersionList{}
  426. var changed, changedHere bool
  427. for _, fv := range vl.RawVersions {
  428. changedHere, err = checkGlobalsFilterDevices(dk, folder, name, fv.Devices, newVL, ro)
  429. if err != nil {
  430. return 0, err
  431. }
  432. changed = changed || changedHere
  433. changedHere, err = checkGlobalsFilterDevices(dk, folder, name, fv.InvalidDevices, newVL, ro)
  434. if err != nil {
  435. return 0, err
  436. }
  437. changed = changed || changedHere
  438. }
  439. if newVL.Empty() {
  440. if err := t.Delete(dbi.Key()); err != nil && !backend.IsNotFound(err) {
  441. return 0, err
  442. }
  443. fixed++
  444. } else if changed {
  445. if err := t.Put(dbi.Key(), mustMarshal(newVL)); err != nil {
  446. return 0, err
  447. }
  448. fixed++
  449. }
  450. }
  451. dbi.Release()
  452. if err := dbi.Error(); err != nil {
  453. return 0, err
  454. }
  455. l.Debugf("global db check completed for %v", folder)
  456. return fixed, t.Commit()
  457. }
  458. func checkGlobalsFilterDevices(dk, folder, name []byte, devices [][]byte, vl *VersionList, t readOnlyTransaction) (bool, error) {
  459. var changed bool
  460. var err error
  461. for _, device := range devices {
  462. dk, err = t.keyer.GenerateDeviceFileKey(dk, folder, device, name)
  463. if err != nil {
  464. return false, err
  465. }
  466. f, ok, err := t.getFileTrunc(dk, false)
  467. if err != nil {
  468. return false, err
  469. }
  470. if !ok {
  471. changed = true
  472. continue
  473. }
  474. _, _, _, _, _, _, err = vl.update(folder, device, f, t)
  475. if err != nil {
  476. return false, err
  477. }
  478. }
  479. return changed, nil
  480. }
  481. func (db *Lowlevel) getIndexID(device, folder []byte) (protocol.IndexID, error) {
  482. key, err := db.keyer.GenerateIndexIDKey(nil, device, folder)
  483. if err != nil {
  484. return 0, err
  485. }
  486. cur, err := db.Get(key)
  487. if backend.IsNotFound(err) {
  488. return 0, nil
  489. } else if err != nil {
  490. return 0, err
  491. }
  492. var id protocol.IndexID
  493. if err := id.Unmarshal(cur); err != nil {
  494. return 0, nil
  495. }
  496. return id, nil
  497. }
  498. func (db *Lowlevel) setIndexID(device, folder []byte, id protocol.IndexID) error {
  499. bs, _ := id.Marshal() // marshalling can't fail
  500. key, err := db.keyer.GenerateIndexIDKey(nil, device, folder)
  501. if err != nil {
  502. return err
  503. }
  504. return db.Put(key, bs)
  505. }
  506. func (db *Lowlevel) dropFolderIndexIDs(folder []byte) error {
  507. t, err := db.newReadWriteTransaction()
  508. if err != nil {
  509. return err
  510. }
  511. defer t.close()
  512. if err := t.deleteKeyPrefixMatching([]byte{KeyTypeIndexID}, func(key []byte) bool {
  513. keyFolder, ok := t.keyer.FolderFromIndexIDKey(key)
  514. if !ok {
  515. l.Debugf("Deleting IndexID with missing FolderIdx: %v", key)
  516. return true
  517. }
  518. return bytes.Equal(keyFolder, folder)
  519. }); err != nil {
  520. return err
  521. }
  522. return t.Commit()
  523. }
  524. func (db *Lowlevel) dropMtimes(folder []byte) error {
  525. key, err := db.keyer.GenerateMtimesKey(nil, folder)
  526. if err != nil {
  527. return err
  528. }
  529. return db.dropPrefix(key)
  530. }
  531. func (db *Lowlevel) dropFolderMeta(folder []byte) error {
  532. key, err := db.keyer.GenerateFolderMetaKey(nil, folder)
  533. if err != nil {
  534. return err
  535. }
  536. return db.dropPrefix(key)
  537. }
  538. func (db *Lowlevel) dropPrefix(prefix []byte) error {
  539. t, err := db.newReadWriteTransaction()
  540. if err != nil {
  541. return err
  542. }
  543. defer t.close()
  544. if err := t.deleteKeyPrefix(prefix); err != nil {
  545. return err
  546. }
  547. return t.Commit()
  548. }
  549. func (db *Lowlevel) gcRunner(ctx context.Context) error {
  550. // Calculate the time for the next GC run. Even if we should run GC
  551. // directly, give the system a while to get up and running and do other
  552. // stuff first. (We might have migrations and stuff which would be
  553. // better off running before GC.)
  554. next := db.timeUntil(indirectGCTimeKey, db.indirectGCInterval)
  555. if next < time.Minute {
  556. next = time.Minute
  557. }
  558. t := time.NewTimer(next)
  559. defer t.Stop()
  560. for {
  561. select {
  562. case <-ctx.Done():
  563. return ctx.Err()
  564. case <-t.C:
  565. if err := db.gcIndirect(ctx); err != nil {
  566. l.Warnln("Database indirection GC failed:", err)
  567. }
  568. db.recordTime(indirectGCTimeKey)
  569. t.Reset(db.timeUntil(indirectGCTimeKey, db.indirectGCInterval))
  570. }
  571. }
  572. }
  573. // recordTime records the current time under the given key, affecting the
  574. // next call to timeUntil with the same key.
  575. func (db *Lowlevel) recordTime(key string) {
  576. miscDB := NewMiscDataNamespace(db)
  577. _ = miscDB.PutInt64(key, time.Now().Unix()) // error wilfully ignored
  578. }
  579. // timeUntil returns how long we should wait until the next interval, or
  580. // zero if it should happen directly.
  581. func (db *Lowlevel) timeUntil(key string, every time.Duration) time.Duration {
  582. miscDB := NewMiscDataNamespace(db)
  583. lastTime, _, _ := miscDB.Int64(key) // error wilfully ignored
  584. nextTime := time.Unix(lastTime, 0).Add(every)
  585. sleepTime := time.Until(nextTime)
  586. if sleepTime < 0 {
  587. sleepTime = 0
  588. }
  589. return sleepTime
  590. }
  591. func (db *Lowlevel) gcIndirect(ctx context.Context) error {
  592. // The indirection GC uses bloom filters to track used block lists and
  593. // versions. This means iterating over all items, adding their hashes to
  594. // the filter, then iterating over the indirected items and removing
  595. // those that don't match the filter. The filter will give false
  596. // positives so we will keep around one percent of things that we don't
  597. // really need (at most).
  598. //
  599. // Indirection GC needs to run when there are no modifications to the
  600. // FileInfos or indirected items.
  601. db.gcMut.Lock()
  602. defer db.gcMut.Unlock()
  603. t, err := db.newReadWriteTransaction()
  604. if err != nil {
  605. return err
  606. }
  607. defer t.Release()
  608. // Set up the bloom filters with the initial capacity and false positive
  609. // rate, or higher capacity if we've done this before and seen lots of
  610. // items. For simplicity's sake we track just one count, which is the
  611. // highest of the various indirected items.
  612. capacity := indirectGCBloomCapacity
  613. if db.gcKeyCount > capacity {
  614. capacity = db.gcKeyCount
  615. }
  616. blockFilter := newBloomFilter(capacity)
  617. versionFilter := newBloomFilter(capacity)
  618. // Iterate the FileInfos, unmarshal the block and version hashes and
  619. // add them to the filter.
  620. it, err := t.NewPrefixIterator([]byte{KeyTypeDevice})
  621. if err != nil {
  622. return err
  623. }
  624. defer it.Release()
  625. for it.Next() {
  626. select {
  627. case <-ctx.Done():
  628. return ctx.Err()
  629. default:
  630. }
  631. var hashes IndirectionHashesOnly
  632. if err := hashes.Unmarshal(it.Value()); err != nil {
  633. return err
  634. }
  635. if len(hashes.BlocksHash) > 0 {
  636. blockFilter.add(hashes.BlocksHash)
  637. }
  638. if len(hashes.VersionHash) > 0 {
  639. versionFilter.add(hashes.VersionHash)
  640. }
  641. }
  642. it.Release()
  643. if err := it.Error(); err != nil {
  644. return err
  645. }
  646. // Iterate over block lists, removing keys with hashes that don't match
  647. // the filter.
  648. it, err = t.NewPrefixIterator([]byte{KeyTypeBlockList})
  649. if err != nil {
  650. return err
  651. }
  652. defer it.Release()
  653. matchedBlocks := 0
  654. for it.Next() {
  655. select {
  656. case <-ctx.Done():
  657. return ctx.Err()
  658. default:
  659. }
  660. key := blockListKey(it.Key())
  661. if blockFilter.has(key.Hash()) {
  662. matchedBlocks++
  663. continue
  664. }
  665. if err := t.Delete(key); err != nil {
  666. return err
  667. }
  668. }
  669. it.Release()
  670. if err := it.Error(); err != nil {
  671. return err
  672. }
  673. // Iterate over version lists, removing keys with hashes that don't match
  674. // the filter.
  675. it, err = db.NewPrefixIterator([]byte{KeyTypeVersion})
  676. if err != nil {
  677. return err
  678. }
  679. matchedVersions := 0
  680. for it.Next() {
  681. select {
  682. case <-ctx.Done():
  683. return ctx.Err()
  684. default:
  685. }
  686. key := versionKey(it.Key())
  687. if versionFilter.has(key.Hash()) {
  688. matchedVersions++
  689. continue
  690. }
  691. if err := t.Delete(key); err != nil {
  692. return err
  693. }
  694. }
  695. it.Release()
  696. if err := it.Error(); err != nil {
  697. return err
  698. }
  699. // Remember the number of unique keys we kept until the next pass.
  700. db.gcKeyCount = matchedBlocks
  701. if matchedVersions > matchedBlocks {
  702. db.gcKeyCount = matchedVersions
  703. }
  704. if err := t.Commit(); err != nil {
  705. return err
  706. }
  707. return db.Compact()
  708. }
  709. func newBloomFilter(capacity int) bloomFilter {
  710. var buf [16]byte
  711. io.ReadFull(rand.Reader, buf[:])
  712. return bloomFilter{
  713. f: blobloom.NewOptimized(blobloom.Config{
  714. Capacity: uint64(capacity),
  715. FPRate: indirectGCBloomFalsePositiveRate,
  716. MaxBits: 8 * indirectGCBloomMaxBytes,
  717. }),
  718. k0: binary.LittleEndian.Uint64(buf[:8]),
  719. k1: binary.LittleEndian.Uint64(buf[8:]),
  720. }
  721. }
  722. type bloomFilter struct {
  723. f *blobloom.Filter
  724. k0, k1 uint64 // Random key for SipHash.
  725. }
  726. func (b *bloomFilter) add(id []byte) { b.f.Add(b.hash(id)) }
  727. func (b *bloomFilter) has(id []byte) bool { return b.f.Has(b.hash(id)) }
  728. // Hash function for the bloomfilter: SipHash of the SHA-256.
  729. //
  730. // The randomization in SipHash means we get different collisions across
  731. // runs and colliding keys are not kept indefinitely.
  732. func (b *bloomFilter) hash(id []byte) uint64 {
  733. if len(id) != sha256.Size {
  734. panic("bug: bloomFilter.hash passed something not a SHA256 hash")
  735. }
  736. return siphash.Hash(b.k0, b.k1, id)
  737. }
  738. // checkRepair checks folder metadata and sequences for miscellaneous errors.
  739. func (db *Lowlevel) checkRepair() error {
  740. db.gcMut.RLock()
  741. defer db.gcMut.RUnlock()
  742. for _, folder := range db.ListFolders() {
  743. if _, err := db.getMetaAndCheckGCLocked(folder); err != nil {
  744. return err
  745. }
  746. }
  747. return nil
  748. }
  749. func (db *Lowlevel) getMetaAndCheck(folder string) (*metadataTracker, error) {
  750. db.gcMut.RLock()
  751. defer db.gcMut.RUnlock()
  752. return db.getMetaAndCheckGCLocked(folder)
  753. }
  754. func (db *Lowlevel) getMetaAndCheckGCLocked(folder string) (*metadataTracker, error) {
  755. fixed, err := db.checkLocalNeed([]byte(folder))
  756. if err != nil {
  757. return nil, fmt.Errorf("checking local need: %w", err)
  758. }
  759. if fixed != 0 {
  760. l.Infof("Repaired %d local need entries for folder %v in database", fixed, folder)
  761. }
  762. fixed, err = db.checkGlobals(folder)
  763. if err != nil {
  764. return nil, fmt.Errorf("checking globals: %w", err)
  765. }
  766. if fixed != 0 {
  767. l.Infof("Repaired %d global entries for folder %v in database", fixed, folder)
  768. }
  769. meta, err := db.recalcMeta(folder)
  770. if err != nil {
  771. return nil, fmt.Errorf("recalculating metadata: %w", err)
  772. }
  773. fixed, err = db.repairSequenceGCLocked(folder, meta)
  774. if err != nil {
  775. return nil, fmt.Errorf("repairing sequences: %w", err)
  776. }
  777. if fixed != 0 {
  778. l.Infof("Repaired %d sequence entries for folder %v in database", fixed, folder)
  779. meta, err = db.recalcMeta(folder)
  780. if err != nil {
  781. return nil, fmt.Errorf("recalculating metadata: %w", err)
  782. }
  783. }
  784. return meta, nil
  785. }
  786. func (db *Lowlevel) loadMetadataTracker(folder string) (*metadataTracker, error) {
  787. meta := newMetadataTracker(db.keyer, db.evLogger)
  788. if err := meta.fromDB(db, []byte(folder)); err != nil {
  789. if err == errMetaInconsistent {
  790. l.Infof("Stored folder metadata for %q is inconsistent; recalculating", folder)
  791. } else {
  792. l.Infof("No stored folder metadata for %q; recalculating", folder)
  793. }
  794. return db.getMetaAndCheck(folder)
  795. }
  796. curSeq := meta.Sequence(protocol.LocalDeviceID)
  797. if metaOK, err := db.verifyLocalSequence(curSeq, folder); err != nil {
  798. return nil, fmt.Errorf("verifying sequences: %w", err)
  799. } else if !metaOK {
  800. l.Infof("Stored folder metadata for %q is out of date after crash; recalculating", folder)
  801. return db.getMetaAndCheck(folder)
  802. }
  803. if age := time.Since(meta.Created()); age > db.recheckInterval {
  804. l.Infof("Stored folder metadata for %q is %v old; recalculating", folder, util.NiceDurationString(age))
  805. return db.getMetaAndCheck(folder)
  806. }
  807. return meta, nil
  808. }
  809. func (db *Lowlevel) recalcMeta(folderStr string) (*metadataTracker, error) {
  810. folder := []byte(folderStr)
  811. meta := newMetadataTracker(db.keyer, db.evLogger)
  812. t, err := db.newReadWriteTransaction(meta.CommitHook(folder))
  813. if err != nil {
  814. return nil, err
  815. }
  816. defer t.close()
  817. var deviceID protocol.DeviceID
  818. err = t.withAllFolderTruncated(folder, func(device []byte, f FileInfoTruncated) bool {
  819. copy(deviceID[:], device)
  820. meta.addFile(deviceID, f)
  821. return true
  822. })
  823. if err != nil {
  824. return nil, err
  825. }
  826. err = t.withGlobal(folder, nil, true, func(f protocol.FileIntf) bool {
  827. meta.addFile(protocol.GlobalDeviceID, f)
  828. return true
  829. })
  830. if err != nil {
  831. return nil, err
  832. }
  833. meta.emptyNeeded(protocol.LocalDeviceID)
  834. err = t.withNeed(folder, protocol.LocalDeviceID[:], true, func(f protocol.FileIntf) bool {
  835. meta.addNeeded(protocol.LocalDeviceID, f)
  836. return true
  837. })
  838. if err != nil {
  839. return nil, err
  840. }
  841. for _, device := range meta.devices() {
  842. meta.emptyNeeded(device)
  843. err = t.withNeed(folder, device[:], true, func(f protocol.FileIntf) bool {
  844. meta.addNeeded(device, f)
  845. return true
  846. })
  847. if err != nil {
  848. return nil, err
  849. }
  850. }
  851. meta.SetCreated()
  852. if err := t.Commit(); err != nil {
  853. return nil, err
  854. }
  855. return meta, nil
  856. }
  857. // Verify the local sequence number from actual sequence entries. Returns
  858. // true if it was all good, or false if a fixup was necessary.
  859. func (db *Lowlevel) verifyLocalSequence(curSeq int64, folder string) (bool, error) {
  860. // Walk the sequence index from the current (supposedly) highest
  861. // sequence number and raise the alarm if we get anything. This recovers
  862. // from the occasion where we have written sequence entries to disk but
  863. // not yet written new metadata to disk.
  864. //
  865. // Note that we can have the same thing happen for remote devices but
  866. // there it's not a problem -- we'll simply advertise a lower sequence
  867. // number than we've actually seen and receive some duplicate updates
  868. // and then be in sync again.
  869. t, err := db.newReadOnlyTransaction()
  870. if err != nil {
  871. return false, err
  872. }
  873. ok := true
  874. if err := t.withHaveSequence([]byte(folder), curSeq+1, func(fi protocol.FileIntf) bool {
  875. ok = false // we got something, which we should not have
  876. return false
  877. }); err != nil {
  878. return false, err
  879. }
  880. t.close()
  881. return ok, nil
  882. }
  883. // repairSequenceGCLocked makes sure the sequence numbers in the sequence keys
  884. // match those in the corresponding file entries. It returns the amount of fixed
  885. // entries.
  886. func (db *Lowlevel) repairSequenceGCLocked(folderStr string, meta *metadataTracker) (int, error) {
  887. t, err := db.newReadWriteTransaction(meta.CommitHook([]byte(folderStr)))
  888. if err != nil {
  889. return 0, err
  890. }
  891. defer t.close()
  892. fixed := 0
  893. folder := []byte(folderStr)
  894. // First check that every file entry has a matching sequence entry
  895. // (this was previously db schema upgrade to 9).
  896. dk, err := t.keyer.GenerateDeviceFileKey(nil, folder, protocol.LocalDeviceID[:], nil)
  897. if err != nil {
  898. return 0, err
  899. }
  900. it, err := t.NewPrefixIterator(dk.WithoutName())
  901. if err != nil {
  902. return 0, err
  903. }
  904. defer it.Release()
  905. var sk sequenceKey
  906. for it.Next() {
  907. intf, err := t.unmarshalTrunc(it.Value(), false)
  908. if err != nil {
  909. // Delete local items with invalid indirected blocks/versions.
  910. // They will be rescanned.
  911. var ierr *blocksIndirectionError
  912. if ok := errors.As(err, &ierr); ok && backend.IsNotFound(err) {
  913. intf, err = t.unmarshalTrunc(it.Value(), true)
  914. if err != nil {
  915. return 0, err
  916. }
  917. name := []byte(intf.FileName())
  918. gk, err := t.keyer.GenerateGlobalVersionKey(nil, folder, name)
  919. if err != nil {
  920. return 0, err
  921. }
  922. _, err = t.removeFromGlobal(gk, nil, folder, protocol.LocalDeviceID[:], name, nil)
  923. if err != nil {
  924. return 0, err
  925. }
  926. sk, err = db.keyer.GenerateSequenceKey(sk, folder, intf.SequenceNo())
  927. if err != nil {
  928. return 0, err
  929. }
  930. if err := t.Delete(sk); err != nil {
  931. return 0, err
  932. }
  933. if err := t.Delete(it.Key()); err != nil {
  934. return 0, err
  935. }
  936. }
  937. return 0, err
  938. }
  939. fi := intf.(protocol.FileInfo)
  940. if sk, err = t.keyer.GenerateSequenceKey(sk, folder, fi.Sequence); err != nil {
  941. return 0, err
  942. }
  943. switch dk, err = t.Get(sk); {
  944. case err != nil:
  945. if !backend.IsNotFound(err) {
  946. return 0, err
  947. }
  948. fallthrough
  949. case !bytes.Equal(it.Key(), dk):
  950. fixed++
  951. fi.Sequence = meta.nextLocalSeq()
  952. if sk, err = t.keyer.GenerateSequenceKey(sk, folder, fi.Sequence); err != nil {
  953. return 0, err
  954. }
  955. if err := t.Put(sk, it.Key()); err != nil {
  956. return 0, err
  957. }
  958. if err := t.putFile(it.Key(), fi); err != nil {
  959. return 0, err
  960. }
  961. }
  962. if err := t.Checkpoint(); err != nil {
  963. return 0, err
  964. }
  965. }
  966. if err := it.Error(); err != nil {
  967. return 0, err
  968. }
  969. it.Release()
  970. // Secondly check there's no sequence entries pointing at incorrect things.
  971. sk, err = t.keyer.GenerateSequenceKey(sk, folder, 0)
  972. if err != nil {
  973. return 0, err
  974. }
  975. it, err = t.NewPrefixIterator(sk.WithoutSequence())
  976. if err != nil {
  977. return 0, err
  978. }
  979. defer it.Release()
  980. for it.Next() {
  981. // Check that the sequence from the key matches the
  982. // sequence in the file.
  983. fi, ok, err := t.getFileTrunc(it.Value(), true)
  984. if err != nil {
  985. return 0, err
  986. }
  987. if ok {
  988. if seq := t.keyer.SequenceFromSequenceKey(it.Key()); seq == fi.SequenceNo() {
  989. continue
  990. }
  991. }
  992. // Either the file is missing or has a different sequence number
  993. fixed++
  994. if err := t.Delete(it.Key()); err != nil {
  995. return 0, err
  996. }
  997. }
  998. if err := it.Error(); err != nil {
  999. return 0, err
  1000. }
  1001. it.Release()
  1002. return fixed, t.Commit()
  1003. }
  1004. // Does not take care of metadata - if anything is repaired, the need count
  1005. // needs to be recalculated.
  1006. func (db *Lowlevel) checkLocalNeed(folder []byte) (int, error) {
  1007. repaired := 0
  1008. t, err := db.newReadWriteTransaction()
  1009. if err != nil {
  1010. return 0, err
  1011. }
  1012. defer t.close()
  1013. key, err := t.keyer.GenerateNeedFileKey(nil, folder, nil)
  1014. if err != nil {
  1015. return 0, err
  1016. }
  1017. dbi, err := t.NewPrefixIterator(key.WithoutName())
  1018. if err != nil {
  1019. return 0, err
  1020. }
  1021. defer dbi.Release()
  1022. var needName string
  1023. var needDone bool
  1024. next := func() {
  1025. needDone = !dbi.Next()
  1026. if !needDone {
  1027. needName = string(t.keyer.NameFromGlobalVersionKey(dbi.Key()))
  1028. }
  1029. }
  1030. next()
  1031. t.withNeedIteratingGlobal(folder, protocol.LocalDeviceID[:], true, func(fi protocol.FileIntf) bool {
  1032. f := fi.(FileInfoTruncated)
  1033. for !needDone && needName < f.Name {
  1034. repaired++
  1035. if err = t.Delete(dbi.Key()); err != nil && !backend.IsNotFound(err) {
  1036. return false
  1037. }
  1038. l.Debugln("check local need: removing", needName)
  1039. next()
  1040. }
  1041. if needName == f.Name {
  1042. next()
  1043. } else {
  1044. repaired++
  1045. key, err = t.keyer.GenerateNeedFileKey(key, folder, []byte(f.Name))
  1046. if err != nil {
  1047. return false
  1048. }
  1049. if err = t.Put(key, nil); err != nil {
  1050. return false
  1051. }
  1052. l.Debugln("check local need: adding", f.Name)
  1053. }
  1054. return true
  1055. })
  1056. if err != nil {
  1057. return 0, err
  1058. }
  1059. for !needDone {
  1060. repaired++
  1061. if err := t.Delete(dbi.Key()); err != nil && !backend.IsNotFound(err) {
  1062. return 0, err
  1063. }
  1064. l.Debugln("check local need: removing", needName)
  1065. next()
  1066. }
  1067. if err := dbi.Error(); err != nil {
  1068. return 0, err
  1069. }
  1070. dbi.Release()
  1071. if err = t.Commit(); err != nil {
  1072. return 0, err
  1073. }
  1074. return repaired, nil
  1075. }
  1076. func (db *Lowlevel) needsRepairPath() string {
  1077. path := db.Location()
  1078. if path == "" {
  1079. return ""
  1080. }
  1081. if path[len(path)-1] == fs.PathSeparator {
  1082. path = path[:len(path)-1]
  1083. }
  1084. return path + needsRepairSuffix
  1085. }
  1086. func (db *Lowlevel) checkErrorForRepair(err error) {
  1087. if errors.Is(err, errEntryFromGlobalMissing) || errors.Is(err, errEmptyGlobal) {
  1088. // Inconsistency error, mark db for repair on next start.
  1089. if path := db.needsRepairPath(); path != "" {
  1090. if fd, err := os.Create(path); err == nil {
  1091. fd.Close()
  1092. }
  1093. }
  1094. }
  1095. }
  1096. // unchanged checks if two files are the same and thus don't need to be updated.
  1097. // Local flags or the invalid bit might change without the version
  1098. // being bumped.
  1099. func unchanged(nf, ef protocol.FileIntf) bool {
  1100. return ef.FileVersion().Equal(nf.FileVersion()) && ef.IsInvalid() == nf.IsInvalid() && ef.FileLocalFlags() == nf.FileLocalFlags()
  1101. }
  1102. func (db *Lowlevel) handleFailure(err error) {
  1103. db.checkErrorForRepair(err)
  1104. if shouldReportFailure(err) {
  1105. db.evLogger.Log(events.Failure, err.Error())
  1106. }
  1107. }
  1108. var ldbPathRe = regexp.MustCompile(`(open|write|read) .+[\\/].+[\\/]index[^\\/]+[\\/][^\\/]+: `)
  1109. func shouldReportFailure(err error) bool {
  1110. return !ldbPathRe.MatchString(err.Error())
  1111. }