lowlevel.go 30 KB

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