leveldb.go 24 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055
  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 []byte, truncate bool, fn Iterator) {
  623. runtime.GC()
  624. start := globalKey(folder, nil)
  625. limit := globalKey(folder, []byte{0xff, 0xff, 0xff, 0xff})
  626. snap, err := db.GetSnapshot()
  627. if err != nil {
  628. panic(err)
  629. }
  630. if debugDB {
  631. l.Debugf("created snapshot %p", snap)
  632. }
  633. defer func() {
  634. if debugDB {
  635. l.Debugf("close snapshot %p", snap)
  636. }
  637. snap.Release()
  638. }()
  639. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  640. defer dbi.Release()
  641. for dbi.Next() {
  642. var vl versionList
  643. err := vl.UnmarshalXDR(dbi.Value())
  644. if err != nil {
  645. panic(err)
  646. }
  647. if len(vl.versions) == 0 {
  648. l.Debugln(dbi.Key())
  649. panic("no versions?")
  650. }
  651. name := globalKeyName(dbi.Key())
  652. fk := deviceKey(folder, vl.versions[0].device, name)
  653. if debugDB {
  654. l.Debugf("snap.Get %p %x", snap, fk)
  655. }
  656. bs, err := snap.Get(fk, nil)
  657. if err != nil {
  658. l.Debugf("folder: %q (%x)", folder, folder)
  659. l.Debugf("key: %q (%x)", dbi.Key(), dbi.Key())
  660. l.Debugf("vl: %v", vl)
  661. l.Debugf("vl.versions[0].device: %x", vl.versions[0].device)
  662. l.Debugf("name: %q (%x)", name, name)
  663. l.Debugf("fk: %q", fk)
  664. l.Debugf("fk: %x %x %x", fk[1:1+64], fk[1+64:1+64+32], fk[1+64+32:])
  665. panic(err)
  666. }
  667. f, err := unmarshalTrunc(bs, truncate)
  668. if err != nil {
  669. panic(err)
  670. }
  671. if cont := fn(f); !cont {
  672. return
  673. }
  674. }
  675. }
  676. func ldbAvailability(db *leveldb.DB, folder, file []byte) []protocol.DeviceID {
  677. k := globalKey(folder, file)
  678. bs, err := db.Get(k, nil)
  679. if err == leveldb.ErrNotFound {
  680. return nil
  681. }
  682. if err != nil {
  683. panic(err)
  684. }
  685. var vl versionList
  686. err = vl.UnmarshalXDR(bs)
  687. if err != nil {
  688. panic(err)
  689. }
  690. var devices []protocol.DeviceID
  691. for _, v := range vl.versions {
  692. if v.version != vl.versions[0].version {
  693. break
  694. }
  695. n := protocol.DeviceIDFromBytes(v.device)
  696. devices = append(devices, n)
  697. }
  698. return devices
  699. }
  700. func ldbWithNeed(db *leveldb.DB, folder, device []byte, truncate bool, fn Iterator) {
  701. runtime.GC()
  702. start := globalKey(folder, nil)
  703. limit := globalKey(folder, []byte{0xff, 0xff, 0xff, 0xff})
  704. snap, err := db.GetSnapshot()
  705. if err != nil {
  706. panic(err)
  707. }
  708. if debugDB {
  709. l.Debugf("created snapshot %p", snap)
  710. }
  711. defer func() {
  712. if debugDB {
  713. l.Debugf("close snapshot %p", snap)
  714. }
  715. snap.Release()
  716. }()
  717. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  718. defer dbi.Release()
  719. outer:
  720. for dbi.Next() {
  721. var vl versionList
  722. err := vl.UnmarshalXDR(dbi.Value())
  723. if err != nil {
  724. panic(err)
  725. }
  726. if len(vl.versions) == 0 {
  727. l.Debugln(dbi.Key())
  728. panic("no versions?")
  729. }
  730. have := false // If we have the file, any version
  731. need := false // If we have a lower version of the file
  732. var haveVersion int64
  733. for _, v := range vl.versions {
  734. if bytes.Compare(v.device, device) == 0 {
  735. have = true
  736. haveVersion = v.version
  737. need = v.version < vl.versions[0].version
  738. break
  739. }
  740. }
  741. if need || !have {
  742. name := globalKeyName(dbi.Key())
  743. needVersion := vl.versions[0].version
  744. inner:
  745. for i := range vl.versions {
  746. if vl.versions[i].version != needVersion {
  747. // We haven't found a valid copy of the file with the needed version.
  748. continue outer
  749. }
  750. fk := deviceKey(folder, vl.versions[i].device, name)
  751. if debugDB {
  752. l.Debugf("snap.Get %p %x", snap, fk)
  753. }
  754. bs, err := snap.Get(fk, nil)
  755. if err != nil {
  756. var id protocol.DeviceID
  757. copy(id[:], device)
  758. l.Debugf("device: %v", id)
  759. l.Debugf("need: %v, have: %v", need, have)
  760. l.Debugf("key: %q (%x)", dbi.Key(), dbi.Key())
  761. l.Debugf("vl: %v", vl)
  762. l.Debugf("i: %v", i)
  763. l.Debugf("fk: %q (%x)", fk, fk)
  764. l.Debugf("name: %q (%x)", name, name)
  765. panic(err)
  766. }
  767. gf, err := unmarshalTrunc(bs, truncate)
  768. if err != nil {
  769. panic(err)
  770. }
  771. if gf.IsInvalid() {
  772. // The file is marked invalid for whatever reason, don't use it.
  773. continue inner
  774. }
  775. if gf.IsDeleted() && !have {
  776. // We don't need deleted files that we don't have
  777. continue outer
  778. }
  779. if debugDB {
  780. 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)
  781. }
  782. if cont := fn(gf); !cont {
  783. return
  784. }
  785. // This file is handled, no need to look further in the version list
  786. continue outer
  787. }
  788. }
  789. }
  790. }
  791. func ldbListFolders(db *leveldb.DB) []string {
  792. runtime.GC()
  793. snap, err := db.GetSnapshot()
  794. if err != nil {
  795. panic(err)
  796. }
  797. if debugDB {
  798. l.Debugf("created snapshot %p", snap)
  799. }
  800. defer func() {
  801. if debugDB {
  802. l.Debugf("close snapshot %p", snap)
  803. }
  804. snap.Release()
  805. }()
  806. dbi := snap.NewIterator(util.BytesPrefix([]byte{KeyTypeGlobal}), nil)
  807. defer dbi.Release()
  808. folderExists := make(map[string]bool)
  809. for dbi.Next() {
  810. folder := string(globalKeyFolder(dbi.Key()))
  811. if !folderExists[folder] {
  812. folderExists[folder] = true
  813. }
  814. }
  815. folders := make([]string, 0, len(folderExists))
  816. for k := range folderExists {
  817. folders = append(folders, k)
  818. }
  819. sort.Strings(folders)
  820. return folders
  821. }
  822. func ldbDropFolder(db *leveldb.DB, folder []byte) {
  823. runtime.GC()
  824. snap, err := db.GetSnapshot()
  825. if err != nil {
  826. panic(err)
  827. }
  828. if debugDB {
  829. l.Debugf("created snapshot %p", snap)
  830. }
  831. defer func() {
  832. if debugDB {
  833. l.Debugf("close snapshot %p", snap)
  834. }
  835. snap.Release()
  836. }()
  837. // Remove all items related to the given folder from the device->file bucket
  838. dbi := snap.NewIterator(util.BytesPrefix([]byte{KeyTypeDevice}), nil)
  839. for dbi.Next() {
  840. itemFolder := deviceKeyFolder(dbi.Key())
  841. if bytes.Compare(folder, itemFolder) == 0 {
  842. db.Delete(dbi.Key(), nil)
  843. }
  844. }
  845. dbi.Release()
  846. // Remove all items related to the given folder from the global bucket
  847. dbi = snap.NewIterator(util.BytesPrefix([]byte{KeyTypeGlobal}), nil)
  848. for dbi.Next() {
  849. itemFolder := globalKeyFolder(dbi.Key())
  850. if bytes.Compare(folder, itemFolder) == 0 {
  851. db.Delete(dbi.Key(), nil)
  852. }
  853. }
  854. dbi.Release()
  855. }
  856. func unmarshalTrunc(bs []byte, truncate bool) (FileIntf, error) {
  857. if truncate {
  858. var tf FileInfoTruncated
  859. err := tf.UnmarshalXDR(bs)
  860. return tf, err
  861. } else {
  862. var tf protocol.FileInfo
  863. err := tf.UnmarshalXDR(bs)
  864. return tf, err
  865. }
  866. }
  867. func ldbCheckGlobals(db *leveldb.DB, folder []byte) {
  868. defer runtime.GC()
  869. snap, err := db.GetSnapshot()
  870. if err != nil {
  871. panic(err)
  872. }
  873. if debugDB {
  874. l.Debugf("created snapshot %p", snap)
  875. }
  876. defer func() {
  877. if debugDB {
  878. l.Debugf("close snapshot %p", snap)
  879. }
  880. snap.Release()
  881. }()
  882. start := globalKey(folder, nil)
  883. limit := globalKey(folder, []byte{0xff, 0xff, 0xff, 0xff})
  884. dbi := snap.NewIterator(&util.Range{Start: start, Limit: limit}, nil)
  885. defer dbi.Release()
  886. batch := new(leveldb.Batch)
  887. if debugDB {
  888. l.Debugf("new batch %p", batch)
  889. }
  890. for dbi.Next() {
  891. gk := dbi.Key()
  892. var vl versionList
  893. err := vl.UnmarshalXDR(dbi.Value())
  894. if err != nil {
  895. panic(err)
  896. }
  897. // Check the global version list for consistency. An issue in previous
  898. // versions of goleveldb could result in reordered writes so that
  899. // there are global entries pointing to no longer existing files. Here
  900. // we find those and clear them out.
  901. name := globalKeyName(gk)
  902. var newVL versionList
  903. for _, version := range vl.versions {
  904. fk := deviceKey(folder, version.device, name)
  905. if debugDB {
  906. l.Debugf("snap.Get %p %x", snap, fk)
  907. }
  908. _, err := snap.Get(fk, nil)
  909. if err == leveldb.ErrNotFound {
  910. continue
  911. }
  912. if err != nil {
  913. panic(err)
  914. }
  915. newVL.versions = append(newVL.versions, version)
  916. }
  917. if len(newVL.versions) != len(vl.versions) {
  918. l.Infof("db repair: rewriting global version list for %x %x", gk[1:1+64], gk[1+64:])
  919. batch.Put(dbi.Key(), newVL.MustMarshalXDR())
  920. }
  921. }
  922. if debugDB {
  923. l.Infoln("db check completed for %q", folder)
  924. }
  925. db.Write(batch, nil)
  926. }