leveldb_dbinstance.go 24 KB

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