leveldb_dbinstance.go 27 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033
  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. "encoding/binary"
  10. "os"
  11. "sort"
  12. "strings"
  13. "sync/atomic"
  14. "github.com/syncthing/syncthing/lib/protocol"
  15. "github.com/syncthing/syncthing/lib/sync"
  16. "github.com/syndtr/goleveldb/leveldb"
  17. "github.com/syndtr/goleveldb/leveldb/errors"
  18. "github.com/syndtr/goleveldb/leveldb/iterator"
  19. "github.com/syndtr/goleveldb/leveldb/opt"
  20. "github.com/syndtr/goleveldb/leveldb/storage"
  21. "github.com/syndtr/goleveldb/leveldb/util"
  22. )
  23. type deletionHandler func(t readWriteTransaction, folder, device, name []byte, dbi iterator.Iterator)
  24. type Instance struct {
  25. committed int64 // this must be the first attribute in the struct to ensure 64 bit alignment on 32 bit plaforms
  26. *leveldb.DB
  27. location string
  28. folderIdx *smallIndex
  29. deviceIdx *smallIndex
  30. }
  31. const (
  32. keyPrefixLen = 1
  33. keyFolderLen = 4 // indexed
  34. keyDeviceLen = 4 // indexed
  35. keySequenceLen = 8
  36. keyHashLen = 32
  37. maxInt64 int64 = 1<<63 - 1
  38. )
  39. func Open(file string) (*Instance, error) {
  40. opts := &opt.Options{
  41. OpenFilesCacheCapacity: 100,
  42. WriteBuffer: 4 << 20,
  43. }
  44. db, err := leveldb.OpenFile(file, opts)
  45. if leveldbIsCorrupted(err) {
  46. db, err = leveldb.RecoverFile(file, opts)
  47. }
  48. if leveldbIsCorrupted(err) {
  49. // The database is corrupted, and we've tried to recover it but it
  50. // didn't work. At this point there isn't much to do beyond dropping
  51. // the database and reindexing...
  52. l.Infoln("Database corruption detected, unable to recover. Reinitializing...")
  53. if err := os.RemoveAll(file); err != nil {
  54. return nil, err
  55. }
  56. db, err = leveldb.OpenFile(file, opts)
  57. }
  58. if err != nil {
  59. return nil, err
  60. }
  61. return newDBInstance(db, file), nil
  62. }
  63. func OpenMemory() *Instance {
  64. db, _ := leveldb.Open(storage.NewMemStorage(), nil)
  65. return newDBInstance(db, "<memory>")
  66. }
  67. func newDBInstance(db *leveldb.DB, location string) *Instance {
  68. i := &Instance{
  69. DB: db,
  70. location: location,
  71. }
  72. i.folderIdx = newSmallIndex(i, []byte{KeyTypeFolderIdx})
  73. i.deviceIdx = newSmallIndex(i, []byte{KeyTypeDeviceIdx})
  74. return i
  75. }
  76. // UpdateSchema does transitions to the current db version if necessary
  77. func (db *Instance) UpdateSchema() {
  78. miscDB := NewNamespacedKV(db, string(KeyTypeMiscData))
  79. prevVersion, _ := miscDB.Int64("dbVersion")
  80. if prevVersion >= dbVersion {
  81. return
  82. }
  83. l.Infof("Updating database schema version from %v to %v...", prevVersion, dbVersion)
  84. if prevVersion == 0 {
  85. db.updateSchema0to1()
  86. }
  87. if prevVersion <= 1 {
  88. db.updateSchema1to2()
  89. }
  90. l.Infof("Finished updating database schema version from %v to %v", prevVersion, dbVersion)
  91. miscDB.PutInt64("dbVersion", dbVersion)
  92. }
  93. // Committed returns the number of items committed to the database since startup
  94. func (db *Instance) Committed() int64 {
  95. return atomic.LoadInt64(&db.committed)
  96. }
  97. // Location returns the filesystem path where the database is stored
  98. func (db *Instance) Location() string {
  99. return db.location
  100. }
  101. func (db *Instance) updateFiles(folder, device []byte, fs []protocol.FileInfo, meta *metadataTracker) {
  102. t := db.newReadWriteTransaction()
  103. defer t.close()
  104. var fk []byte
  105. var gk []byte
  106. for _, f := range fs {
  107. name := []byte(f.Name)
  108. fk = db.deviceKeyInto(fk, folder, device, name)
  109. // Get and unmarshal the file entry. If it doesn't exist or can't be
  110. // unmarshalled we'll add it as a new entry.
  111. bs, err := t.Get(fk, nil)
  112. var ef FileInfoTruncated
  113. if err == nil {
  114. err = ef.Unmarshal(bs)
  115. }
  116. // The Invalid flag might change without the version being bumped.
  117. if err == nil && ef.Version.Equal(f.Version) && ef.Invalid == f.Invalid {
  118. continue
  119. }
  120. devID := protocol.DeviceIDFromBytes(device)
  121. if err == nil {
  122. meta.removeFile(devID, ef)
  123. }
  124. meta.addFile(devID, f)
  125. t.insertFile(fk, folder, device, f)
  126. gk = db.globalKeyInto(gk, folder, name)
  127. t.updateGlobal(gk, folder, device, f, meta)
  128. // Write out and reuse the batch every few records, to avoid the batch
  129. // growing too large and thus allocating unnecessarily much memory.
  130. t.checkFlush()
  131. }
  132. }
  133. func (db *Instance) addSequences(folder []byte, fs []protocol.FileInfo) {
  134. t := db.newReadWriteTransaction()
  135. defer t.close()
  136. var sk []byte
  137. var dk []byte
  138. for _, f := range fs {
  139. sk = db.sequenceKeyInto(sk, folder, f.Sequence)
  140. dk = db.deviceKeyInto(dk, folder, protocol.LocalDeviceID[:], []byte(f.Name))
  141. t.Put(sk, dk)
  142. l.Debugf("adding sequence; folder=%q sequence=%v %v", folder, f.Sequence, f.Name)
  143. t.checkFlush()
  144. }
  145. }
  146. func (db *Instance) removeSequences(folder []byte, fs []protocol.FileInfo) {
  147. t := db.newReadWriteTransaction()
  148. defer t.close()
  149. var sk []byte
  150. for _, f := range fs {
  151. t.Delete(db.sequenceKeyInto(sk, folder, f.Sequence))
  152. l.Debugf("removing sequence; folder=%q sequence=%v %v", folder, f.Sequence, f.Name)
  153. t.checkFlush()
  154. }
  155. }
  156. func (db *Instance) withHave(folder, device, prefix []byte, truncate bool, fn Iterator) {
  157. t := db.newReadOnlyTransaction()
  158. defer t.close()
  159. dbi := t.NewIterator(util.BytesPrefix(db.deviceKey(folder, device, prefix)[:keyPrefixLen+keyFolderLen+keyDeviceLen+len(prefix)]), nil)
  160. defer dbi.Release()
  161. slashedPrefix := prefix
  162. if !bytes.HasSuffix(prefix, []byte{'/'}) {
  163. slashedPrefix = append(slashedPrefix, '/')
  164. }
  165. for dbi.Next() {
  166. name := db.deviceKeyName(dbi.Key())
  167. if len(prefix) > 0 && !bytes.Equal(name, prefix) && !bytes.HasPrefix(name, slashedPrefix) {
  168. return
  169. }
  170. // The iterator function may keep a reference to the unmarshalled
  171. // struct, which in turn references the buffer it was unmarshalled
  172. // from. dbi.Value() just returns an internal slice that it reuses, so
  173. // we need to copy it.
  174. f, err := unmarshalTrunc(append([]byte{}, dbi.Value()...), truncate)
  175. if err != nil {
  176. l.Debugln("unmarshal error:", err)
  177. continue
  178. }
  179. if !fn(f) {
  180. return
  181. }
  182. }
  183. }
  184. func (db *Instance) withHaveSequence(folder []byte, startSeq int64, fn Iterator) {
  185. t := db.newReadOnlyTransaction()
  186. defer t.close()
  187. dbi := t.NewIterator(&util.Range{Start: db.sequenceKey(folder, startSeq), Limit: db.sequenceKey(folder, maxInt64)}, nil)
  188. defer dbi.Release()
  189. for dbi.Next() {
  190. f, ok := db.getFile(dbi.Value())
  191. if !ok {
  192. l.Debugln("missing file for sequence number", db.sequenceKeySequence(dbi.Key()))
  193. continue
  194. }
  195. if !fn(f) {
  196. return
  197. }
  198. }
  199. }
  200. func (db *Instance) withAllFolderTruncated(folder []byte, fn func(device []byte, f FileInfoTruncated) bool) {
  201. t := db.newReadWriteTransaction()
  202. defer t.close()
  203. dbi := t.NewIterator(util.BytesPrefix(db.deviceKey(folder, nil, nil)[:keyPrefixLen+keyFolderLen]), nil)
  204. defer dbi.Release()
  205. var gk []byte
  206. for dbi.Next() {
  207. device := db.deviceKeyDevice(dbi.Key())
  208. var f FileInfoTruncated
  209. // The iterator function may keep a reference to the unmarshalled
  210. // struct, which in turn references the buffer it was unmarshalled
  211. // from. dbi.Value() just returns an internal slice that it reuses, so
  212. // we need to copy it.
  213. err := f.Unmarshal(append([]byte{}, dbi.Value()...))
  214. if err != nil {
  215. l.Debugln("unmarshal error:", err)
  216. continue
  217. }
  218. switch f.Name {
  219. case "", ".", "..", "/": // A few obviously invalid filenames
  220. l.Infof("Dropping invalid filename %q from database", f.Name)
  221. name := []byte(f.Name)
  222. gk = db.globalKeyInto(gk, folder, name)
  223. t.removeFromGlobal(gk, folder, device, name, nil)
  224. t.Delete(dbi.Key())
  225. t.checkFlush()
  226. continue
  227. }
  228. if !fn(device, f) {
  229. return
  230. }
  231. }
  232. }
  233. func (db *Instance) getFile(key []byte) (protocol.FileInfo, bool) {
  234. bs, err := db.Get(key, nil)
  235. if err == leveldb.ErrNotFound {
  236. return protocol.FileInfo{}, false
  237. }
  238. if err != nil {
  239. l.Debugln("surprise error:", err)
  240. return protocol.FileInfo{}, false
  241. }
  242. var f protocol.FileInfo
  243. err = f.Unmarshal(bs)
  244. if err != nil {
  245. l.Debugln("unmarshal error:", err)
  246. return protocol.FileInfo{}, false
  247. }
  248. return f, true
  249. }
  250. func (db *Instance) getGlobal(folder, file []byte, truncate bool) (FileIntf, bool) {
  251. k := db.globalKey(folder, file)
  252. t := db.newReadOnlyTransaction()
  253. defer t.close()
  254. bs, err := t.Get(k, nil)
  255. if err != nil {
  256. return nil, false
  257. }
  258. var vl VersionList
  259. err = vl.Unmarshal(bs)
  260. if err == leveldb.ErrNotFound {
  261. return nil, false
  262. }
  263. if err != nil {
  264. l.Debugln("unmarshal error:", k, err)
  265. return nil, false
  266. }
  267. if len(vl.Versions) == 0 {
  268. l.Debugln("no versions:", k)
  269. return nil, false
  270. }
  271. k = db.deviceKey(folder, vl.Versions[0].Device, file)
  272. bs, err = t.Get(k, nil)
  273. if err != nil {
  274. l.Debugln("surprise error:", k, err)
  275. return nil, false
  276. }
  277. fi, err := unmarshalTrunc(bs, truncate)
  278. if err != nil {
  279. l.Debugln("unmarshal error:", k, err)
  280. return nil, false
  281. }
  282. return fi, true
  283. }
  284. func (db *Instance) withGlobal(folder, prefix []byte, truncate bool, fn Iterator) {
  285. t := db.newReadOnlyTransaction()
  286. defer t.close()
  287. dbi := t.NewIterator(util.BytesPrefix(db.globalKey(folder, prefix)), nil)
  288. defer dbi.Release()
  289. slashedPrefix := prefix
  290. if !bytes.HasSuffix(prefix, []byte{'/'}) {
  291. slashedPrefix = append(slashedPrefix, '/')
  292. }
  293. var fk []byte
  294. for dbi.Next() {
  295. var vl VersionList
  296. err := vl.Unmarshal(dbi.Value())
  297. if err != nil {
  298. l.Debugln("unmarshal error:", err)
  299. continue
  300. }
  301. if len(vl.Versions) == 0 {
  302. l.Debugln("no versions:", dbi.Key())
  303. continue
  304. }
  305. name := db.globalKeyName(dbi.Key())
  306. if len(prefix) > 0 && !bytes.Equal(name, prefix) && !bytes.HasPrefix(name, slashedPrefix) {
  307. return
  308. }
  309. fk = db.deviceKeyInto(fk, folder, vl.Versions[0].Device, name)
  310. bs, err := t.Get(fk, nil)
  311. if err != nil {
  312. l.Debugln("surprise error:", err)
  313. continue
  314. }
  315. f, err := unmarshalTrunc(bs, truncate)
  316. if err != nil {
  317. l.Debugln("unmarshal error:", err)
  318. continue
  319. }
  320. if !fn(f) {
  321. return
  322. }
  323. }
  324. }
  325. func (db *Instance) availability(folder, file []byte) []protocol.DeviceID {
  326. k := db.globalKey(folder, file)
  327. bs, err := db.Get(k, nil)
  328. if err == leveldb.ErrNotFound {
  329. return nil
  330. }
  331. if err != nil {
  332. l.Debugln("surprise error:", err)
  333. return nil
  334. }
  335. var vl VersionList
  336. err = vl.Unmarshal(bs)
  337. if err != nil {
  338. l.Debugln("unmarshal error:", err)
  339. return nil
  340. }
  341. var devices []protocol.DeviceID
  342. for _, v := range vl.Versions {
  343. if !v.Version.Equal(vl.Versions[0].Version) {
  344. break
  345. }
  346. if v.Invalid {
  347. continue
  348. }
  349. n := protocol.DeviceIDFromBytes(v.Device)
  350. devices = append(devices, n)
  351. }
  352. return devices
  353. }
  354. func (db *Instance) withNeed(folder, device []byte, truncate bool, fn Iterator) {
  355. t := db.newReadOnlyTransaction()
  356. defer t.close()
  357. dbi := t.NewIterator(util.BytesPrefix(db.globalKey(folder, nil)[:keyPrefixLen+keyFolderLen]), nil)
  358. defer dbi.Release()
  359. var fk []byte
  360. for dbi.Next() {
  361. var vl VersionList
  362. err := vl.Unmarshal(dbi.Value())
  363. if err != nil {
  364. l.Debugln("unmarshal error:", err)
  365. continue
  366. }
  367. if len(vl.Versions) == 0 {
  368. l.Debugln("no versions:", dbi.Key())
  369. continue
  370. }
  371. have := false // If we have the file, any version
  372. need := false // If we have a lower version of the file
  373. var haveFileVersion FileVersion
  374. for _, v := range vl.Versions {
  375. if bytes.Equal(v.Device, device) {
  376. have = true
  377. haveFileVersion = v
  378. // XXX: This marks Concurrent (i.e. conflicting) changes as
  379. // needs. Maybe we should do that, but it needs special
  380. // handling in the puller.
  381. need = !v.Version.GreaterEqual(vl.Versions[0].Version)
  382. break
  383. }
  384. }
  385. if have && !need {
  386. continue
  387. }
  388. name := db.globalKeyName(dbi.Key())
  389. needVersion := vl.Versions[0].Version
  390. needDevice := protocol.DeviceIDFromBytes(vl.Versions[0].Device)
  391. for i := range vl.Versions {
  392. if !vl.Versions[i].Version.Equal(needVersion) {
  393. // We haven't found a valid copy of the file with the needed version.
  394. break
  395. }
  396. if vl.Versions[i].Invalid {
  397. // The file is marked invalid, don't use it.
  398. continue
  399. }
  400. fk = db.deviceKeyInto(fk, folder, vl.Versions[i].Device, name)
  401. bs, err := t.Get(fk, nil)
  402. if err != nil {
  403. l.Debugln("surprise error:", err)
  404. continue
  405. }
  406. gf, err := unmarshalTrunc(bs, truncate)
  407. if err != nil {
  408. l.Debugln("unmarshal error:", err)
  409. continue
  410. }
  411. if gf.IsDeleted() && !have {
  412. // We don't need deleted files that we don't have
  413. break
  414. }
  415. l.Debugf("need folder=%q device=%v name=%q need=%v have=%v invalid=%v haveV=%v globalV=%v globalDev=%v", folder, protocol.DeviceIDFromBytes(device), name, need, have, haveFileVersion.Invalid, haveFileVersion.Version, needVersion, needDevice)
  416. if !fn(gf) {
  417. return
  418. }
  419. // This file is handled, no need to look further in the version list
  420. break
  421. }
  422. }
  423. }
  424. func (db *Instance) ListFolders() []string {
  425. t := db.newReadOnlyTransaction()
  426. defer t.close()
  427. dbi := t.NewIterator(util.BytesPrefix([]byte{KeyTypeGlobal}), nil)
  428. defer dbi.Release()
  429. folderExists := make(map[string]bool)
  430. for dbi.Next() {
  431. folder, ok := db.globalKeyFolder(dbi.Key())
  432. if ok && !folderExists[string(folder)] {
  433. folderExists[string(folder)] = true
  434. }
  435. }
  436. folders := make([]string, 0, len(folderExists))
  437. for k := range folderExists {
  438. folders = append(folders, k)
  439. }
  440. sort.Strings(folders)
  441. return folders
  442. }
  443. func (db *Instance) dropFolder(folder []byte) {
  444. t := db.newReadOnlyTransaction()
  445. defer t.close()
  446. // Remove all items related to the given folder from the device->file bucket
  447. dbi := t.NewIterator(util.BytesPrefix([]byte{KeyTypeDevice}), nil)
  448. for dbi.Next() {
  449. itemFolder := db.deviceKeyFolder(dbi.Key())
  450. if bytes.Equal(folder, itemFolder) {
  451. db.Delete(dbi.Key(), nil)
  452. }
  453. }
  454. dbi.Release()
  455. // Remove all sequences related to the folder
  456. sequenceKey := db.sequenceKey([]byte(folder), 0)
  457. dbi = t.NewIterator(util.BytesPrefix(sequenceKey[:4]), nil)
  458. for dbi.Next() {
  459. db.Delete(dbi.Key(), nil)
  460. }
  461. dbi.Release()
  462. // Remove all items related to the given folder from the global bucket
  463. dbi = t.NewIterator(util.BytesPrefix([]byte{KeyTypeGlobal}), nil)
  464. for dbi.Next() {
  465. itemFolder, ok := db.globalKeyFolder(dbi.Key())
  466. if ok && bytes.Equal(folder, itemFolder) {
  467. db.Delete(dbi.Key(), nil)
  468. }
  469. }
  470. dbi.Release()
  471. }
  472. func (db *Instance) dropDeviceFolder(device, folder []byte, meta *metadataTracker) {
  473. t := db.newReadWriteTransaction()
  474. defer t.close()
  475. dbi := t.NewIterator(util.BytesPrefix(db.deviceKey(folder, device, nil)), nil)
  476. defer dbi.Release()
  477. var gk []byte
  478. for dbi.Next() {
  479. key := dbi.Key()
  480. name := db.deviceKeyName(key)
  481. gk = db.globalKeyInto(gk, folder, name)
  482. t.removeFromGlobal(gk, folder, device, name, meta)
  483. t.Delete(key)
  484. t.checkFlush()
  485. }
  486. }
  487. func (db *Instance) checkGlobals(folder []byte, meta *metadataTracker) {
  488. t := db.newReadWriteTransaction()
  489. defer t.close()
  490. dbi := t.NewIterator(util.BytesPrefix(db.globalKey(folder, nil)[:keyPrefixLen+keyFolderLen]), nil)
  491. defer dbi.Release()
  492. var fk []byte
  493. for dbi.Next() {
  494. gk := dbi.Key()
  495. var vl VersionList
  496. err := vl.Unmarshal(dbi.Value())
  497. if err != nil {
  498. l.Debugln("unmarshal error:", err)
  499. continue
  500. }
  501. // Check the global version list for consistency. An issue in previous
  502. // versions of goleveldb could result in reordered writes so that
  503. // there are global entries pointing to no longer existing files. Here
  504. // we find those and clear them out.
  505. name := db.globalKeyName(gk)
  506. var newVL VersionList
  507. for i, version := range vl.Versions {
  508. fk = db.deviceKeyInto(fk, folder, version.Device, name)
  509. _, err := t.Get(fk, nil)
  510. if err == leveldb.ErrNotFound {
  511. continue
  512. }
  513. if err != nil {
  514. l.Debugln("surprise error:", err)
  515. return
  516. }
  517. newVL.Versions = append(newVL.Versions, version)
  518. if i == 0 {
  519. if fi, ok := db.getFile(fk); ok {
  520. meta.addFile(globalDeviceID, fi)
  521. }
  522. }
  523. }
  524. if len(newVL.Versions) != len(vl.Versions) {
  525. t.Put(dbi.Key(), mustMarshal(&newVL))
  526. t.checkFlush()
  527. }
  528. }
  529. l.Debugf("db check completed for %q", folder)
  530. }
  531. func (db *Instance) updateSchema0to1() {
  532. t := db.newReadWriteTransaction()
  533. defer t.close()
  534. dbi := t.NewIterator(util.BytesPrefix([]byte{KeyTypeDevice}), nil)
  535. defer dbi.Release()
  536. symlinkConv := 0
  537. changedFolders := make(map[string]struct{})
  538. ignAdded := 0
  539. meta := newMetadataTracker() // dummy metadata tracker
  540. var gk []byte
  541. for dbi.Next() {
  542. folder := db.deviceKeyFolder(dbi.Key())
  543. device := db.deviceKeyDevice(dbi.Key())
  544. name := db.deviceKeyName(dbi.Key())
  545. // Remove files with absolute path (see #4799)
  546. if strings.HasPrefix(string(name), "/") {
  547. if _, ok := changedFolders[string(folder)]; !ok {
  548. changedFolders[string(folder)] = struct{}{}
  549. }
  550. gk = db.globalKeyInto(gk, folder, name)
  551. t.removeFromGlobal(gk, folder, device, nil, nil)
  552. t.Delete(dbi.Key())
  553. t.checkFlush()
  554. continue
  555. }
  556. // Change SYMLINK_FILE and SYMLINK_DIRECTORY types to the current SYMLINK
  557. // type (previously SYMLINK_UNKNOWN). It does this for all devices, both
  558. // local and remote, and does not reset delta indexes. It shouldn't really
  559. // matter what the symlink type is, but this cleans it up for a possible
  560. // future when SYMLINK_FILE and SYMLINK_DIRECTORY are no longer understood.
  561. var f protocol.FileInfo
  562. if err := f.Unmarshal(dbi.Value()); err != nil {
  563. // probably can't happen
  564. continue
  565. }
  566. if f.Type == protocol.FileInfoTypeDeprecatedSymlinkDirectory || f.Type == protocol.FileInfoTypeDeprecatedSymlinkFile {
  567. f.Type = protocol.FileInfoTypeSymlink
  568. bs, err := f.Marshal()
  569. if err != nil {
  570. panic("can't happen: " + err.Error())
  571. }
  572. t.Put(dbi.Key(), bs)
  573. t.checkFlush()
  574. symlinkConv++
  575. }
  576. // Add invalid files to global list
  577. if f.Invalid {
  578. gk = db.globalKeyInto(gk, folder, name)
  579. if t.updateGlobal(gk, folder, device, f, meta) {
  580. if _, ok := changedFolders[string(folder)]; !ok {
  581. changedFolders[string(folder)] = struct{}{}
  582. }
  583. ignAdded++
  584. }
  585. }
  586. }
  587. for folder := range changedFolders {
  588. db.dropFolderMeta([]byte(folder))
  589. }
  590. l.Infof("Updated symlink type for %d index entries and added %d invalid files to global list", symlinkConv, ignAdded)
  591. }
  592. func (db *Instance) updateSchema1to2() {
  593. t := db.newReadWriteTransaction()
  594. defer t.close()
  595. var sk []byte
  596. var dk []byte
  597. for _, folderStr := range db.ListFolders() {
  598. folder := []byte(folderStr)
  599. db.withHave(folder, protocol.LocalDeviceID[:], nil, true, func(f FileIntf) bool {
  600. sk = db.sequenceKeyInto(sk, folder, f.SequenceNo())
  601. dk = db.deviceKeyInto(dk, folder, protocol.LocalDeviceID[:], []byte(f.FileName()))
  602. t.Put(sk, dk)
  603. t.checkFlush()
  604. return true
  605. })
  606. }
  607. }
  608. // deviceKey returns a byte slice encoding the following information:
  609. // keyTypeDevice (1 byte)
  610. // folder (4 bytes)
  611. // device (4 bytes)
  612. // name (variable size)
  613. func (db *Instance) deviceKey(folder, device, file []byte) []byte {
  614. return db.deviceKeyInto(nil, folder, device, file)
  615. }
  616. func (db *Instance) deviceKeyInto(k, folder, device, file []byte) []byte {
  617. reqLen := keyPrefixLen + keyFolderLen + keyDeviceLen + len(file)
  618. k = resize(k, reqLen)
  619. k[0] = KeyTypeDevice
  620. binary.BigEndian.PutUint32(k[keyPrefixLen:], db.folderIdx.ID(folder))
  621. binary.BigEndian.PutUint32(k[keyPrefixLen+keyFolderLen:], db.deviceIdx.ID(device))
  622. copy(k[keyPrefixLen+keyFolderLen+keyDeviceLen:], file)
  623. return k
  624. }
  625. // deviceKeyName returns the device ID from the key
  626. func (db *Instance) deviceKeyName(key []byte) []byte {
  627. return key[keyPrefixLen+keyFolderLen+keyDeviceLen:]
  628. }
  629. // deviceKeyFolder returns the folder name from the key
  630. func (db *Instance) deviceKeyFolder(key []byte) []byte {
  631. folder, ok := db.folderIdx.Val(binary.BigEndian.Uint32(key[keyPrefixLen:]))
  632. if !ok {
  633. panic("bug: lookup of nonexistent folder ID")
  634. }
  635. return folder
  636. }
  637. // deviceKeyDevice returns the device ID from the key
  638. func (db *Instance) deviceKeyDevice(key []byte) []byte {
  639. device, ok := db.deviceIdx.Val(binary.BigEndian.Uint32(key[keyPrefixLen+keyFolderLen:]))
  640. if !ok {
  641. panic("bug: lookup of nonexistent device ID")
  642. }
  643. return device
  644. }
  645. // globalKey returns a byte slice encoding the following information:
  646. // keyTypeGlobal (1 byte)
  647. // folder (4 bytes)
  648. // name (variable size)
  649. func (db *Instance) globalKey(folder, file []byte) []byte {
  650. return db.globalKeyInto(nil, folder, file)
  651. }
  652. func (db *Instance) globalKeyInto(gk, folder, file []byte) []byte {
  653. reqLen := keyPrefixLen + keyFolderLen + len(file)
  654. gk = resize(gk, reqLen)
  655. gk[0] = KeyTypeGlobal
  656. binary.BigEndian.PutUint32(gk[keyPrefixLen:], db.folderIdx.ID(folder))
  657. copy(gk[keyPrefixLen+keyFolderLen:], file)
  658. return gk[:reqLen]
  659. }
  660. // globalKeyName returns the filename from the key
  661. func (db *Instance) globalKeyName(key []byte) []byte {
  662. return key[keyPrefixLen+keyFolderLen:]
  663. }
  664. // globalKeyFolder returns the folder name from the key
  665. func (db *Instance) globalKeyFolder(key []byte) ([]byte, bool) {
  666. return db.folderIdx.Val(binary.BigEndian.Uint32(key[keyPrefixLen:]))
  667. }
  668. // sequenceKey returns a byte slice encoding the following information:
  669. // KeyTypeSequence (1 byte)
  670. // folder (4 bytes)
  671. // sequence number (8 bytes)
  672. func (db *Instance) sequenceKey(folder []byte, seq int64) []byte {
  673. return db.sequenceKeyInto(nil, folder, seq)
  674. }
  675. func (db *Instance) sequenceKeyInto(k []byte, folder []byte, seq int64) []byte {
  676. reqLen := keyPrefixLen + keyFolderLen + keySequenceLen
  677. k = resize(k, reqLen)
  678. k[0] = KeyTypeSequence
  679. binary.BigEndian.PutUint32(k[keyPrefixLen:], db.folderIdx.ID(folder))
  680. binary.BigEndian.PutUint64(k[keyPrefixLen+keyFolderLen:], uint64(seq))
  681. return k[:reqLen]
  682. }
  683. // sequenceKeySequence returns the sequence number from the key
  684. func (db *Instance) sequenceKeySequence(key []byte) int64 {
  685. return int64(binary.BigEndian.Uint64(key[keyPrefixLen+keyFolderLen:]))
  686. }
  687. func (db *Instance) getIndexID(device, folder []byte) protocol.IndexID {
  688. key := db.indexIDKey(device, folder)
  689. cur, err := db.Get(key, nil)
  690. if err != nil {
  691. return 0
  692. }
  693. var id protocol.IndexID
  694. if err := id.Unmarshal(cur); err != nil {
  695. return 0
  696. }
  697. return id
  698. }
  699. func (db *Instance) setIndexID(device, folder []byte, id protocol.IndexID) {
  700. key := db.indexIDKey(device, folder)
  701. bs, _ := id.Marshal() // marshalling can't fail
  702. if err := db.Put(key, bs, nil); err != nil {
  703. panic("storing index ID: " + err.Error())
  704. }
  705. }
  706. func (db *Instance) indexIDKey(device, folder []byte) []byte {
  707. k := make([]byte, keyPrefixLen+keyDeviceLen+keyFolderLen)
  708. k[0] = KeyTypeIndexID
  709. binary.BigEndian.PutUint32(k[keyPrefixLen:], db.deviceIdx.ID(device))
  710. binary.BigEndian.PutUint32(k[keyPrefixLen+keyDeviceLen:], db.folderIdx.ID(folder))
  711. return k
  712. }
  713. func (db *Instance) indexIDDevice(key []byte) []byte {
  714. device, ok := db.deviceIdx.Val(binary.BigEndian.Uint32(key[keyPrefixLen:]))
  715. if !ok {
  716. // uuh ...
  717. return nil
  718. }
  719. return device
  720. }
  721. func (db *Instance) mtimesKey(folder []byte) []byte {
  722. prefix := make([]byte, 5) // key type + 4 bytes folder idx number
  723. prefix[0] = KeyTypeVirtualMtime
  724. binary.BigEndian.PutUint32(prefix[1:], db.folderIdx.ID(folder))
  725. return prefix
  726. }
  727. func (db *Instance) folderMetaKey(folder []byte) []byte {
  728. prefix := make([]byte, 5) // key type + 4 bytes folder idx number
  729. prefix[0] = KeyTypeFolderMeta
  730. binary.BigEndian.PutUint32(prefix[1:], db.folderIdx.ID(folder))
  731. return prefix
  732. }
  733. // DropLocalDeltaIndexIDs removes all index IDs for the local device ID from
  734. // the database. This will cause a full index transmission on the next
  735. // connection.
  736. func (db *Instance) DropLocalDeltaIndexIDs() {
  737. db.dropDeltaIndexIDs(true)
  738. }
  739. // DropRemoteDeltaIndexIDs removes all index IDs for the other devices than
  740. // the local one from the database. This will cause them to send us a full
  741. // index on the next connection.
  742. func (db *Instance) DropRemoteDeltaIndexIDs() {
  743. db.dropDeltaIndexIDs(false)
  744. }
  745. func (db *Instance) dropDeltaIndexIDs(local bool) {
  746. t := db.newReadWriteTransaction()
  747. defer t.close()
  748. dbi := t.NewIterator(util.BytesPrefix([]byte{KeyTypeIndexID}), nil)
  749. defer dbi.Release()
  750. for dbi.Next() {
  751. device := db.indexIDDevice(dbi.Key())
  752. if bytes.Equal(device, protocol.LocalDeviceID[:]) == local {
  753. t.Delete(dbi.Key())
  754. }
  755. }
  756. }
  757. func (db *Instance) dropMtimes(folder []byte) {
  758. db.dropPrefix(db.mtimesKey(folder))
  759. }
  760. func (db *Instance) dropFolderMeta(folder []byte) {
  761. db.dropPrefix(db.folderMetaKey(folder))
  762. }
  763. func (db *Instance) dropPrefix(prefix []byte) {
  764. t := db.newReadWriteTransaction()
  765. defer t.close()
  766. dbi := t.NewIterator(util.BytesPrefix(prefix), nil)
  767. defer dbi.Release()
  768. for dbi.Next() {
  769. t.Delete(dbi.Key())
  770. }
  771. }
  772. func unmarshalTrunc(bs []byte, truncate bool) (FileIntf, error) {
  773. if truncate {
  774. var tf FileInfoTruncated
  775. err := tf.Unmarshal(bs)
  776. return tf, err
  777. }
  778. var tf protocol.FileInfo
  779. err := tf.Unmarshal(bs)
  780. return tf, err
  781. }
  782. // A "better" version of leveldb's errors.IsCorrupted.
  783. func leveldbIsCorrupted(err error) bool {
  784. switch {
  785. case err == nil:
  786. return false
  787. case errors.IsCorrupted(err):
  788. return true
  789. case strings.Contains(err.Error(), "corrupted"):
  790. return true
  791. }
  792. return false
  793. }
  794. // A smallIndex is an in memory bidirectional []byte to uint32 map. It gives
  795. // fast lookups in both directions and persists to the database. Don't use for
  796. // storing more items than fit comfortably in RAM.
  797. type smallIndex struct {
  798. db *Instance
  799. prefix []byte
  800. id2val map[uint32]string
  801. val2id map[string]uint32
  802. nextID uint32
  803. mut sync.Mutex
  804. }
  805. func newSmallIndex(db *Instance, prefix []byte) *smallIndex {
  806. idx := &smallIndex{
  807. db: db,
  808. prefix: prefix,
  809. id2val: make(map[uint32]string),
  810. val2id: make(map[string]uint32),
  811. mut: sync.NewMutex(),
  812. }
  813. idx.load()
  814. return idx
  815. }
  816. // load iterates over the prefix space in the database and populates the in
  817. // memory maps.
  818. func (i *smallIndex) load() {
  819. tr := i.db.newReadOnlyTransaction()
  820. it := tr.NewIterator(util.BytesPrefix(i.prefix), nil)
  821. for it.Next() {
  822. val := string(it.Value())
  823. id := binary.BigEndian.Uint32(it.Key()[len(i.prefix):])
  824. i.id2val[id] = val
  825. i.val2id[val] = id
  826. if id >= i.nextID {
  827. i.nextID = id + 1
  828. }
  829. }
  830. it.Release()
  831. tr.close()
  832. }
  833. // ID returns the index number for the given byte slice, allocating a new one
  834. // and persisting this to the database if necessary.
  835. func (i *smallIndex) ID(val []byte) uint32 {
  836. i.mut.Lock()
  837. // intentionally avoiding defer here as we want this call to be as fast as
  838. // possible in the general case (folder ID already exists). The map lookup
  839. // with the conversion of []byte to string is compiler optimized to not
  840. // copy the []byte, which is why we don't assign it to a temp variable
  841. // here.
  842. if id, ok := i.val2id[string(val)]; ok {
  843. i.mut.Unlock()
  844. return id
  845. }
  846. id := i.nextID
  847. i.nextID++
  848. valStr := string(val)
  849. i.val2id[valStr] = id
  850. i.id2val[id] = valStr
  851. key := make([]byte, len(i.prefix)+8) // prefix plus uint32 id
  852. copy(key, i.prefix)
  853. binary.BigEndian.PutUint32(key[len(i.prefix):], id)
  854. i.db.Put(key, val, nil)
  855. i.mut.Unlock()
  856. return id
  857. }
  858. // Val returns the value for the given index number, or (nil, false) if there
  859. // is no such index number.
  860. func (i *smallIndex) Val(id uint32) ([]byte, bool) {
  861. i.mut.Lock()
  862. val, ok := i.id2val[id]
  863. i.mut.Unlock()
  864. if !ok {
  865. return nil, false
  866. }
  867. return []byte(val), true
  868. }
  869. // resize returns a byte array of length reqLen, reusing k if possible
  870. func resize(k []byte, reqLen int) []byte {
  871. if cap(k) < reqLen {
  872. return make([]byte, reqLen)
  873. }
  874. return k[:reqLen]
  875. }