leveldb.go 24 KB

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