leveldb.go 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026
  1. // Copyright (C) 2014 The Syncthing Authors.
  2. //
  3. // This program is free software: you can redistribute it and/or modify it
  4. // under the terms of the GNU General Public License as published by the Free
  5. // Software Foundation, either version 3 of the License, or (at your option)
  6. // any later version.
  7. //
  8. // This program is distributed in the hope that it will be useful, but WITHOUT
  9. // ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  10. // FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
  11. // more details.
  12. //
  13. // You should have received a copy of the GNU General Public License along
  14. // with this program. If not, see <http://www.gnu.org/licenses/>.
  15. //go:generate -command genxdr go run ../../Godeps/_workspace/src/github.com/calmh/xdr/cmd/genxdr/main.go
  16. //go:generate genxdr -o leveldb_xdr.go leveldb.go
  17. package db
  18. import (
  19. "bytes"
  20. "fmt"
  21. "runtime"
  22. "sort"
  23. "sync"
  24. "github.com/syncthing/protocol"
  25. "github.com/syncthing/syncthing/internal/lamport"
  26. "github.com/syndtr/goleveldb/leveldb"
  27. "github.com/syndtr/goleveldb/leveldb/iterator"
  28. "github.com/syndtr/goleveldb/leveldb/opt"
  29. "github.com/syndtr/goleveldb/leveldb/util"
  30. )
  31. var (
  32. clockTick int64
  33. clockMut sync.Mutex
  34. )
  35. func clock(v int64) int64 {
  36. clockMut.Lock()
  37. defer clockMut.Unlock()
  38. if v > clockTick {
  39. clockTick = v + 1
  40. } else {
  41. clockTick++
  42. }
  43. return clockTick
  44. }
  45. const (
  46. keyTypeDevice = iota
  47. keyTypeGlobal
  48. keyTypeBlock
  49. )
  50. type fileVersion struct {
  51. version int64
  52. device []byte
  53. }
  54. type versionList struct {
  55. versions []fileVersion
  56. }
  57. func (l versionList) String() string {
  58. var b bytes.Buffer
  59. var id protocol.DeviceID
  60. b.WriteString("{")
  61. for i, v := range l.versions {
  62. if i > 0 {
  63. b.WriteString(", ")
  64. }
  65. copy(id[:], v.device)
  66. fmt.Fprintf(&b, "{%d, %v}", v.version, id)
  67. }
  68. b.WriteString("}")
  69. return b.String()
  70. }
  71. type fileList []protocol.FileInfo
  72. func (l fileList) Len() int {
  73. return len(l)
  74. }
  75. func (l fileList) Swap(a, b int) {
  76. l[a], l[b] = l[b], l[a]
  77. }
  78. func (l fileList) Less(a, b int) bool {
  79. return l[a].Name < l[b].Name
  80. }
  81. type dbReader interface {
  82. Get([]byte, *opt.ReadOptions) ([]byte, error)
  83. }
  84. type dbWriter interface {
  85. Put([]byte, []byte)
  86. Delete([]byte)
  87. }
  88. // deviceKey returns a byte slice encoding the following information:
  89. // keyTypeDevice (1 byte)
  90. // folder (64 bytes)
  91. // device (32 bytes)
  92. // name (variable size)
  93. func deviceKey(folder, device, file []byte) []byte {
  94. k := make([]byte, 1+64+32+len(file))
  95. k[0] = keyTypeDevice
  96. if len(folder) > 64 {
  97. panic("folder name too long")
  98. }
  99. copy(k[1:], []byte(folder))
  100. copy(k[1+64:], device[:])
  101. copy(k[1+64+32:], []byte(file))
  102. return k
  103. }
  104. func deviceKeyName(key []byte) []byte {
  105. return key[1+64+32:]
  106. }
  107. func deviceKeyFolder(key []byte) []byte {
  108. folder := key[1 : 1+64]
  109. izero := bytes.IndexByte(folder, 0)
  110. if izero < 0 {
  111. return folder
  112. }
  113. return folder[:izero]
  114. }
  115. func deviceKeyDevice(key []byte) []byte {
  116. return key[1+64 : 1+64+32]
  117. }
  118. // globalKey returns a byte slice encoding the following information:
  119. // keyTypeGlobal (1 byte)
  120. // folder (64 bytes)
  121. // name (variable size)
  122. func globalKey(folder, file []byte) []byte {
  123. k := make([]byte, 1+64+len(file))
  124. k[0] = keyTypeGlobal
  125. if len(folder) > 64 {
  126. panic("folder name too long")
  127. }
  128. copy(k[1:], []byte(folder))
  129. copy(k[1+64:], []byte(file))
  130. return k
  131. }
  132. func globalKeyName(key []byte) []byte {
  133. return key[1+64:]
  134. }
  135. func globalKeyFolder(key []byte) []byte {
  136. folder := key[1 : 1+64]
  137. izero := bytes.IndexByte(folder, 0)
  138. if izero < 0 {
  139. return folder
  140. }
  141. return folder[:izero]
  142. }
  143. type deletionHandler func(db dbReader, batch dbWriter, folder, device, name []byte, dbi iterator.Iterator) int64
  144. func ldbGenericReplace(db *leveldb.DB, folder, device []byte, fs []protocol.FileInfo, deleteFn deletionHandler) int64 {
  145. runtime.GC()
  146. sort.Sort(fileList(fs)) // sort list on name, same as in the database
  147. start := deviceKey(folder, device, nil) // before all folder/device files
  148. limit := deviceKey(folder, device, []byte{0xff, 0xff, 0xff, 0xff}) // after all folder/device files
  149. batch := new(leveldb.Batch)
  150. if debugDB {
  151. l.Debugf("new batch %p", batch)
  152. }
  153. snap, err := db.GetSnapshot()
  154. if err != nil {
  155. panic(err)
  156. }
  157. if debugDB {
  158. l.Debugf("created snapshot %p", snap)
  159. }
  160. defer func() {
  161. if debugDB {
  162. l.Debugf("close snapshot %p", snap)
  163. }
  164. snap.Release()
  165. }()
  166. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  167. defer dbi.Release()
  168. moreDb := dbi.Next()
  169. fsi := 0
  170. var maxLocalVer int64
  171. for {
  172. var newName, oldName []byte
  173. moreFs := fsi < len(fs)
  174. if !moreDb && !moreFs {
  175. break
  176. }
  177. if moreFs {
  178. newName = []byte(fs[fsi].Name)
  179. }
  180. if moreDb {
  181. oldName = deviceKeyName(dbi.Key())
  182. }
  183. cmp := bytes.Compare(newName, oldName)
  184. if debugDB {
  185. l.Debugf("generic replace; folder=%q device=%v moreFs=%v moreDb=%v cmp=%d newName=%q oldName=%q", folder, protocol.DeviceIDFromBytes(device), moreFs, moreDb, cmp, newName, oldName)
  186. }
  187. switch {
  188. case moreFs && (!moreDb || cmp == -1):
  189. if debugDB {
  190. l.Debugln("generic replace; missing - insert")
  191. }
  192. // Database is missing this file. Insert it.
  193. if lv := ldbInsert(batch, folder, device, fs[fsi]); lv > maxLocalVer {
  194. maxLocalVer = lv
  195. }
  196. if fs[fsi].IsInvalid() {
  197. ldbRemoveFromGlobal(snap, batch, folder, device, newName)
  198. } else {
  199. ldbUpdateGlobal(snap, batch, folder, device, newName, fs[fsi].Version)
  200. }
  201. fsi++
  202. case moreFs && moreDb && cmp == 0:
  203. // File exists on both sides - compare versions. We might get an
  204. // update with the same version and different flags if a device has
  205. // marked a file as invalid, so handle that too.
  206. if debugDB {
  207. l.Debugln("generic replace; exists - compare")
  208. }
  209. var ef FileInfoTruncated
  210. ef.UnmarshalXDR(dbi.Value())
  211. if fs[fsi].Version > ef.Version ||
  212. (fs[fsi].Version == ef.Version && fs[fsi].Flags != ef.Flags) {
  213. if debugDB {
  214. l.Debugln("generic replace; differs - insert")
  215. }
  216. if lv := ldbInsert(batch, folder, device, fs[fsi]); lv > maxLocalVer {
  217. maxLocalVer = lv
  218. }
  219. if fs[fsi].IsInvalid() {
  220. ldbRemoveFromGlobal(snap, batch, folder, device, newName)
  221. } else {
  222. ldbUpdateGlobal(snap, batch, folder, device, newName, fs[fsi].Version)
  223. }
  224. } else if debugDB {
  225. l.Debugln("generic replace; equal - ignore")
  226. }
  227. fsi++
  228. moreDb = dbi.Next()
  229. case moreDb && (!moreFs || cmp == 1):
  230. if debugDB {
  231. l.Debugln("generic replace; exists - remove")
  232. }
  233. if lv := deleteFn(snap, batch, folder, device, oldName, dbi); lv > maxLocalVer {
  234. maxLocalVer = lv
  235. }
  236. moreDb = dbi.Next()
  237. }
  238. }
  239. if debugDB {
  240. l.Debugf("db.Write %p", batch)
  241. }
  242. err = db.Write(batch, nil)
  243. if err != nil {
  244. panic(err)
  245. }
  246. return maxLocalVer
  247. }
  248. func ldbReplace(db *leveldb.DB, folder, device []byte, fs []protocol.FileInfo) int64 {
  249. // TODO: Return the remaining maxLocalVer?
  250. return ldbGenericReplace(db, folder, device, fs, func(db dbReader, batch dbWriter, folder, device, name []byte, dbi iterator.Iterator) int64 {
  251. // Database has a file that we are missing. Remove it.
  252. if debugDB {
  253. l.Debugf("delete; folder=%q device=%v name=%q", folder, protocol.DeviceIDFromBytes(device), name)
  254. }
  255. ldbRemoveFromGlobal(db, batch, folder, device, name)
  256. if debugDB {
  257. l.Debugf("batch.Delete %p %x", batch, dbi.Key())
  258. }
  259. batch.Delete(dbi.Key())
  260. return 0
  261. })
  262. }
  263. func ldbReplaceWithDelete(db *leveldb.DB, folder, device []byte, fs []protocol.FileInfo) int64 {
  264. return ldbGenericReplace(db, folder, device, fs, func(db dbReader, batch dbWriter, folder, device, name []byte, dbi iterator.Iterator) int64 {
  265. var tf FileInfoTruncated
  266. err := tf.UnmarshalXDR(dbi.Value())
  267. if err != nil {
  268. panic(err)
  269. }
  270. if !tf.IsDeleted() {
  271. if debugDB {
  272. l.Debugf("mark deleted; folder=%q device=%v name=%q", folder, protocol.DeviceIDFromBytes(device), name)
  273. }
  274. ts := clock(tf.LocalVersion)
  275. f := protocol.FileInfo{
  276. Name: tf.Name,
  277. Version: lamport.Default.Tick(tf.Version),
  278. LocalVersion: ts,
  279. Flags: tf.Flags | protocol.FlagDeleted,
  280. Modified: tf.Modified,
  281. }
  282. bs, _ := f.MarshalXDR()
  283. if debugDB {
  284. l.Debugf("batch.Put %p %x", batch, dbi.Key())
  285. }
  286. batch.Put(dbi.Key(), bs)
  287. ldbUpdateGlobal(db, batch, folder, device, deviceKeyName(dbi.Key()), f.Version)
  288. return ts
  289. }
  290. return 0
  291. })
  292. }
  293. func ldbUpdate(db *leveldb.DB, folder, device []byte, fs []protocol.FileInfo) int64 {
  294. runtime.GC()
  295. batch := new(leveldb.Batch)
  296. if debugDB {
  297. l.Debugf("new batch %p", batch)
  298. }
  299. snap, err := db.GetSnapshot()
  300. if err != nil {
  301. panic(err)
  302. }
  303. if debugDB {
  304. l.Debugf("created snapshot %p", snap)
  305. }
  306. defer func() {
  307. if debugDB {
  308. l.Debugf("close snapshot %p", snap)
  309. }
  310. snap.Release()
  311. }()
  312. var maxLocalVer int64
  313. for _, f := range fs {
  314. name := []byte(f.Name)
  315. fk := deviceKey(folder, device, name)
  316. if debugDB {
  317. l.Debugf("snap.Get %p %x", snap, fk)
  318. }
  319. bs, err := snap.Get(fk, nil)
  320. if err == leveldb.ErrNotFound {
  321. if lv := ldbInsert(batch, folder, device, f); lv > maxLocalVer {
  322. maxLocalVer = lv
  323. }
  324. if f.IsInvalid() {
  325. ldbRemoveFromGlobal(snap, batch, folder, device, name)
  326. } else {
  327. ldbUpdateGlobal(snap, batch, folder, device, name, f.Version)
  328. }
  329. continue
  330. }
  331. var ef FileInfoTruncated
  332. err = ef.UnmarshalXDR(bs)
  333. if err != nil {
  334. panic(err)
  335. }
  336. // Flags might change without the version being bumped when we set the
  337. // invalid flag on an existing file.
  338. if ef.Version != f.Version || ef.Flags != f.Flags {
  339. if lv := ldbInsert(batch, folder, device, f); lv > maxLocalVer {
  340. maxLocalVer = lv
  341. }
  342. if f.IsInvalid() {
  343. ldbRemoveFromGlobal(snap, batch, folder, device, name)
  344. } else {
  345. ldbUpdateGlobal(snap, batch, folder, device, name, f.Version)
  346. }
  347. }
  348. }
  349. if debugDB {
  350. l.Debugf("db.Write %p", batch)
  351. }
  352. err = db.Write(batch, nil)
  353. if err != nil {
  354. panic(err)
  355. }
  356. return maxLocalVer
  357. }
  358. func ldbInsert(batch dbWriter, folder, device []byte, file protocol.FileInfo) int64 {
  359. if debugDB {
  360. l.Debugf("insert; folder=%q device=%v %v", folder, protocol.DeviceIDFromBytes(device), file)
  361. }
  362. if file.LocalVersion == 0 {
  363. file.LocalVersion = clock(0)
  364. }
  365. name := []byte(file.Name)
  366. nk := deviceKey(folder, device, name)
  367. if debugDB {
  368. l.Debugf("batch.Put %p %x", batch, nk)
  369. }
  370. batch.Put(nk, file.MustMarshalXDR())
  371. return file.LocalVersion
  372. }
  373. // ldbUpdateGlobal adds this device+version to the version list for the given
  374. // file. If the device is already present in the list, the version is updated.
  375. // If the file does not have an entry in the global list, it is created.
  376. func ldbUpdateGlobal(db dbReader, batch dbWriter, folder, device, file []byte, version int64) bool {
  377. if debugDB {
  378. l.Debugf("update global; folder=%q device=%v file=%q version=%d", folder, protocol.DeviceIDFromBytes(device), file, version)
  379. }
  380. gk := globalKey(folder, file)
  381. svl, err := db.Get(gk, nil)
  382. if err != nil && err != leveldb.ErrNotFound {
  383. panic(err)
  384. }
  385. var fl versionList
  386. nv := fileVersion{
  387. device: device,
  388. version: version,
  389. }
  390. if svl != nil {
  391. err = fl.UnmarshalXDR(svl)
  392. if err != nil {
  393. panic(err)
  394. }
  395. for i := range fl.versions {
  396. if bytes.Compare(fl.versions[i].device, device) == 0 {
  397. if fl.versions[i].version == version {
  398. // No need to do anything
  399. return false
  400. }
  401. fl.versions = append(fl.versions[:i], fl.versions[i+1:]...)
  402. break
  403. }
  404. }
  405. }
  406. for i := range fl.versions {
  407. if fl.versions[i].version <= version {
  408. t := append(fl.versions, fileVersion{})
  409. copy(t[i+1:], t[i:])
  410. t[i] = nv
  411. fl.versions = t
  412. goto done
  413. }
  414. }
  415. fl.versions = append(fl.versions, nv)
  416. done:
  417. if debugDB {
  418. l.Debugf("batch.Put %p %x", batch, gk)
  419. l.Debugf("new global after update: %v", fl)
  420. }
  421. batch.Put(gk, fl.MustMarshalXDR())
  422. return true
  423. }
  424. // ldbRemoveFromGlobal removes the device from the global version list for the
  425. // given file. If the version list is empty after this, the file entry is
  426. // removed entirely.
  427. func ldbRemoveFromGlobal(db dbReader, batch dbWriter, folder, device, file []byte) {
  428. if debugDB {
  429. l.Debugf("remove from global; folder=%q device=%v file=%q", folder, protocol.DeviceIDFromBytes(device), file)
  430. }
  431. gk := globalKey(folder, file)
  432. svl, err := db.Get(gk, nil)
  433. if err != nil {
  434. // We might be called to "remove" a global version that doesn't exist
  435. // if the first update for the file is already marked invalid.
  436. return
  437. }
  438. var fl versionList
  439. err = fl.UnmarshalXDR(svl)
  440. if err != nil {
  441. panic(err)
  442. }
  443. for i := range fl.versions {
  444. if bytes.Compare(fl.versions[i].device, device) == 0 {
  445. fl.versions = append(fl.versions[:i], fl.versions[i+1:]...)
  446. break
  447. }
  448. }
  449. if len(fl.versions) == 0 {
  450. if debugDB {
  451. l.Debugf("batch.Delete %p %x", batch, gk)
  452. }
  453. batch.Delete(gk)
  454. } else {
  455. if debugDB {
  456. l.Debugf("batch.Put %p %x", batch, gk)
  457. l.Debugf("new global after remove: %v", fl)
  458. }
  459. batch.Put(gk, fl.MustMarshalXDR())
  460. }
  461. }
  462. func ldbWithHave(db *leveldb.DB, folder, device []byte, truncate bool, fn Iterator) {
  463. start := deviceKey(folder, device, nil) // before all folder/device files
  464. limit := deviceKey(folder, device, []byte{0xff, 0xff, 0xff, 0xff}) // after all folder/device files
  465. snap, err := db.GetSnapshot()
  466. if err != nil {
  467. panic(err)
  468. }
  469. if debugDB {
  470. l.Debugf("created snapshot %p", snap)
  471. }
  472. defer func() {
  473. if debugDB {
  474. l.Debugf("close snapshot %p", snap)
  475. }
  476. snap.Release()
  477. }()
  478. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  479. defer dbi.Release()
  480. for dbi.Next() {
  481. f, err := unmarshalTrunc(dbi.Value(), truncate)
  482. if err != nil {
  483. panic(err)
  484. }
  485. if cont := fn(f); !cont {
  486. return
  487. }
  488. }
  489. }
  490. func ldbWithAllFolderTruncated(db *leveldb.DB, folder []byte, fn func(device []byte, f FileInfoTruncated) bool) {
  491. runtime.GC()
  492. start := deviceKey(folder, nil, nil) // before all folder/device files
  493. limit := deviceKey(folder, protocol.LocalDeviceID[:], []byte{0xff, 0xff, 0xff, 0xff}) // after all folder/device files
  494. snap, err := db.GetSnapshot()
  495. if err != nil {
  496. panic(err)
  497. }
  498. if debugDB {
  499. l.Debugf("created snapshot %p", snap)
  500. }
  501. defer func() {
  502. if debugDB {
  503. l.Debugf("close snapshot %p", snap)
  504. }
  505. snap.Release()
  506. }()
  507. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  508. defer dbi.Release()
  509. for dbi.Next() {
  510. device := deviceKeyDevice(dbi.Key())
  511. var f FileInfoTruncated
  512. err := f.UnmarshalXDR(dbi.Value())
  513. if err != nil {
  514. panic(err)
  515. }
  516. switch f.Name {
  517. case "", ".", "..", "/": // A few obviously invalid filenames
  518. l.Infof("Dropping invalid filename %q from database", f.Name)
  519. batch := new(leveldb.Batch)
  520. ldbRemoveFromGlobal(db, batch, folder, device, nil)
  521. batch.Delete(dbi.Key())
  522. db.Write(batch, nil)
  523. continue
  524. }
  525. if cont := fn(device, f); !cont {
  526. return
  527. }
  528. }
  529. }
  530. func ldbGet(db *leveldb.DB, folder, device, file []byte) (protocol.FileInfo, bool) {
  531. nk := deviceKey(folder, device, file)
  532. bs, err := db.Get(nk, nil)
  533. if err == leveldb.ErrNotFound {
  534. return protocol.FileInfo{}, false
  535. }
  536. if err != nil {
  537. panic(err)
  538. }
  539. var f protocol.FileInfo
  540. err = f.UnmarshalXDR(bs)
  541. if err != nil {
  542. panic(err)
  543. }
  544. return f, true
  545. }
  546. func ldbGetGlobal(db *leveldb.DB, folder, file []byte, truncate bool) (FileIntf, bool) {
  547. k := globalKey(folder, file)
  548. snap, err := db.GetSnapshot()
  549. if err != nil {
  550. panic(err)
  551. }
  552. if debugDB {
  553. l.Debugf("created snapshot %p", snap)
  554. }
  555. defer func() {
  556. if debugDB {
  557. l.Debugf("close snapshot %p", snap)
  558. }
  559. snap.Release()
  560. }()
  561. if debugDB {
  562. l.Debugf("snap.Get %p %x", snap, k)
  563. }
  564. bs, err := snap.Get(k, nil)
  565. if err == leveldb.ErrNotFound {
  566. return nil, false
  567. }
  568. if err != nil {
  569. panic(err)
  570. }
  571. var vl versionList
  572. err = vl.UnmarshalXDR(bs)
  573. if err != nil {
  574. panic(err)
  575. }
  576. if len(vl.versions) == 0 {
  577. l.Debugln(k)
  578. panic("no versions?")
  579. }
  580. k = deviceKey(folder, vl.versions[0].device, file)
  581. if debugDB {
  582. l.Debugf("snap.Get %p %x", snap, k)
  583. }
  584. bs, err = snap.Get(k, nil)
  585. if err != nil {
  586. panic(err)
  587. }
  588. fi, err := unmarshalTrunc(bs, truncate)
  589. if err != nil {
  590. panic(err)
  591. }
  592. return fi, true
  593. }
  594. func ldbWithGlobal(db *leveldb.DB, folder []byte, truncate bool, fn Iterator) {
  595. runtime.GC()
  596. start := globalKey(folder, nil)
  597. limit := globalKey(folder, []byte{0xff, 0xff, 0xff, 0xff})
  598. snap, err := db.GetSnapshot()
  599. if err != nil {
  600. panic(err)
  601. }
  602. if debugDB {
  603. l.Debugf("created snapshot %p", snap)
  604. }
  605. defer func() {
  606. if debugDB {
  607. l.Debugf("close snapshot %p", snap)
  608. }
  609. snap.Release()
  610. }()
  611. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  612. defer dbi.Release()
  613. for dbi.Next() {
  614. var vl versionList
  615. err := vl.UnmarshalXDR(dbi.Value())
  616. if err != nil {
  617. panic(err)
  618. }
  619. if len(vl.versions) == 0 {
  620. l.Debugln(dbi.Key())
  621. panic("no versions?")
  622. }
  623. name := globalKeyName(dbi.Key())
  624. fk := deviceKey(folder, vl.versions[0].device, name)
  625. if debugDB {
  626. l.Debugf("snap.Get %p %x", snap, fk)
  627. }
  628. bs, err := snap.Get(fk, nil)
  629. if err != nil {
  630. l.Debugf("folder: %q (%x)", folder, folder)
  631. l.Debugf("key: %q (%x)", dbi.Key(), dbi.Key())
  632. l.Debugf("vl: %v", vl)
  633. l.Debugf("vl.versions[0].device: %x", vl.versions[0].device)
  634. l.Debugf("name: %q (%x)", name, name)
  635. l.Debugf("fk: %q", fk)
  636. l.Debugf("fk: %x %x %x", fk[1:1+64], fk[1+64:1+64+32], fk[1+64+32:])
  637. panic(err)
  638. }
  639. f, err := unmarshalTrunc(bs, truncate)
  640. if err != nil {
  641. panic(err)
  642. }
  643. if cont := fn(f); !cont {
  644. return
  645. }
  646. }
  647. }
  648. func ldbAvailability(db *leveldb.DB, folder, file []byte) []protocol.DeviceID {
  649. k := globalKey(folder, file)
  650. bs, err := db.Get(k, nil)
  651. if err == leveldb.ErrNotFound {
  652. return nil
  653. }
  654. if err != nil {
  655. panic(err)
  656. }
  657. var vl versionList
  658. err = vl.UnmarshalXDR(bs)
  659. if err != nil {
  660. panic(err)
  661. }
  662. var devices []protocol.DeviceID
  663. for _, v := range vl.versions {
  664. if v.version != vl.versions[0].version {
  665. break
  666. }
  667. n := protocol.DeviceIDFromBytes(v.device)
  668. devices = append(devices, n)
  669. }
  670. return devices
  671. }
  672. func ldbWithNeed(db *leveldb.DB, folder, device []byte, truncate bool, fn Iterator) {
  673. runtime.GC()
  674. start := globalKey(folder, nil)
  675. limit := globalKey(folder, []byte{0xff, 0xff, 0xff, 0xff})
  676. snap, err := db.GetSnapshot()
  677. if err != nil {
  678. panic(err)
  679. }
  680. if debugDB {
  681. l.Debugf("created snapshot %p", snap)
  682. }
  683. defer func() {
  684. if debugDB {
  685. l.Debugf("close snapshot %p", snap)
  686. }
  687. snap.Release()
  688. }()
  689. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  690. defer dbi.Release()
  691. outer:
  692. for dbi.Next() {
  693. var vl versionList
  694. err := vl.UnmarshalXDR(dbi.Value())
  695. if err != nil {
  696. panic(err)
  697. }
  698. if len(vl.versions) == 0 {
  699. l.Debugln(dbi.Key())
  700. panic("no versions?")
  701. }
  702. have := false // If we have the file, any version
  703. need := false // If we have a lower version of the file
  704. var haveVersion int64
  705. for _, v := range vl.versions {
  706. if bytes.Compare(v.device, device) == 0 {
  707. have = true
  708. haveVersion = v.version
  709. need = v.version < vl.versions[0].version
  710. break
  711. }
  712. }
  713. if need || !have {
  714. name := globalKeyName(dbi.Key())
  715. needVersion := vl.versions[0].version
  716. inner:
  717. for i := range vl.versions {
  718. if vl.versions[i].version != needVersion {
  719. // We haven't found a valid copy of the file with the needed version.
  720. continue outer
  721. }
  722. fk := deviceKey(folder, vl.versions[i].device, name)
  723. if debugDB {
  724. l.Debugf("snap.Get %p %x", snap, fk)
  725. }
  726. bs, err := snap.Get(fk, nil)
  727. if err != nil {
  728. var id protocol.DeviceID
  729. copy(id[:], device)
  730. l.Debugf("device: %v", id)
  731. l.Debugf("need: %v, have: %v", need, have)
  732. l.Debugf("key: %q (%x)", dbi.Key(), dbi.Key())
  733. l.Debugf("vl: %v", vl)
  734. l.Debugf("i: %v", i)
  735. l.Debugf("fk: %q (%x)", fk, fk)
  736. l.Debugf("name: %q (%x)", name, name)
  737. panic(err)
  738. }
  739. gf, err := unmarshalTrunc(bs, truncate)
  740. if err != nil {
  741. panic(err)
  742. }
  743. if gf.IsInvalid() {
  744. // The file is marked invalid for whatever reason, don't use it.
  745. continue inner
  746. }
  747. if gf.IsDeleted() && !have {
  748. // We don't need deleted files that we don't have
  749. continue outer
  750. }
  751. if debugDB {
  752. l.Debugf("need folder=%q device=%v name=%q need=%v have=%v haveV=%d globalV=%d", folder, protocol.DeviceIDFromBytes(device), name, need, have, haveVersion, vl.versions[0].version)
  753. }
  754. if cont := fn(gf); !cont {
  755. return
  756. }
  757. // This file is handled, no need to look further in the version list
  758. continue outer
  759. }
  760. }
  761. }
  762. }
  763. func ldbListFolders(db *leveldb.DB) []string {
  764. runtime.GC()
  765. start := []byte{keyTypeGlobal}
  766. limit := []byte{keyTypeGlobal + 1}
  767. snap, err := db.GetSnapshot()
  768. if err != nil {
  769. panic(err)
  770. }
  771. if debugDB {
  772. l.Debugf("created snapshot %p", snap)
  773. }
  774. defer func() {
  775. if debugDB {
  776. l.Debugf("close snapshot %p", snap)
  777. }
  778. snap.Release()
  779. }()
  780. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  781. defer dbi.Release()
  782. folderExists := make(map[string]bool)
  783. for dbi.Next() {
  784. folder := string(globalKeyFolder(dbi.Key()))
  785. if !folderExists[folder] {
  786. folderExists[folder] = true
  787. }
  788. }
  789. folders := make([]string, 0, len(folderExists))
  790. for k := range folderExists {
  791. folders = append(folders, k)
  792. }
  793. sort.Strings(folders)
  794. return folders
  795. }
  796. func ldbDropFolder(db *leveldb.DB, folder []byte) {
  797. runtime.GC()
  798. snap, err := db.GetSnapshot()
  799. if err != nil {
  800. panic(err)
  801. }
  802. if debugDB {
  803. l.Debugf("created snapshot %p", snap)
  804. }
  805. defer func() {
  806. if debugDB {
  807. l.Debugf("close snapshot %p", snap)
  808. }
  809. snap.Release()
  810. }()
  811. // Remove all items related to the given folder from the device->file bucket
  812. start := []byte{keyTypeDevice}
  813. limit := []byte{keyTypeDevice + 1}
  814. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  815. for dbi.Next() {
  816. itemFolder := deviceKeyFolder(dbi.Key())
  817. if bytes.Compare(folder, itemFolder) == 0 {
  818. db.Delete(dbi.Key(), nil)
  819. }
  820. }
  821. dbi.Release()
  822. // Remove all items related to the given folder from the global bucket
  823. start = []byte{keyTypeGlobal}
  824. limit = []byte{keyTypeGlobal + 1}
  825. dbi = snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  826. for dbi.Next() {
  827. itemFolder := globalKeyFolder(dbi.Key())
  828. if bytes.Compare(folder, itemFolder) == 0 {
  829. db.Delete(dbi.Key(), nil)
  830. }
  831. }
  832. dbi.Release()
  833. }
  834. func unmarshalTrunc(bs []byte, truncate bool) (FileIntf, error) {
  835. if truncate {
  836. var tf FileInfoTruncated
  837. err := tf.UnmarshalXDR(bs)
  838. return tf, err
  839. } else {
  840. var tf protocol.FileInfo
  841. err := tf.UnmarshalXDR(bs)
  842. return tf, err
  843. }
  844. }
  845. func ldbCheckGlobals(db *leveldb.DB, folder []byte) {
  846. defer runtime.GC()
  847. snap, err := db.GetSnapshot()
  848. if err != nil {
  849. panic(err)
  850. }
  851. if debugDB {
  852. l.Debugf("created snapshot %p", snap)
  853. }
  854. defer func() {
  855. if debugDB {
  856. l.Debugf("close snapshot %p", snap)
  857. }
  858. snap.Release()
  859. }()
  860. start := globalKey(folder, nil)
  861. limit := globalKey(folder, []byte{0xff, 0xff, 0xff, 0xff})
  862. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  863. defer dbi.Release()
  864. batch := new(leveldb.Batch)
  865. if debugDB {
  866. l.Debugf("new batch %p", batch)
  867. }
  868. for dbi.Next() {
  869. gk := dbi.Key()
  870. var vl versionList
  871. err := vl.UnmarshalXDR(dbi.Value())
  872. if err != nil {
  873. panic(err)
  874. }
  875. // Check the global version list for consistency. An issue in previous
  876. // versions of goleveldb could result in reordered writes so that
  877. // there are global entries pointing to no longer existing files. Here
  878. // we find those and clear them out.
  879. name := globalKeyName(gk)
  880. var newVL versionList
  881. for _, version := range vl.versions {
  882. fk := deviceKey(folder, version.device, name)
  883. if debugDB {
  884. l.Debugf("snap.Get %p %x", snap, fk)
  885. }
  886. _, err := snap.Get(fk, nil)
  887. if err == leveldb.ErrNotFound {
  888. continue
  889. }
  890. if err != nil {
  891. panic(err)
  892. }
  893. newVL.versions = append(newVL.versions, version)
  894. }
  895. if len(newVL.versions) != len(vl.versions) {
  896. l.Infof("db repair: rewriting global version list for %x %x", gk[1:1+64], gk[1+64:])
  897. batch.Put(dbi.Key(), newVL.MustMarshalXDR())
  898. }
  899. }
  900. if debugDB {
  901. l.Infoln("db check completed for %q", folder)
  902. }
  903. db.Write(batch, nil)
  904. }