deviceset.go 25 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956
  1. package devmapper
  2. import (
  3. "encoding/json"
  4. "fmt"
  5. "github.com/dotcloud/docker/utils"
  6. "io"
  7. "io/ioutil"
  8. "path"
  9. "path/filepath"
  10. "strconv"
  11. "sync"
  12. "time"
  13. )
  14. var (
  15. DefaultDataLoopbackSize int64 = 100 * 1024 * 1024 * 1024
  16. DefaultMetaDataLoopbackSize int64 = 2 * 1024 * 1024 * 1024
  17. DefaultBaseFsSize uint64 = 10 * 1024 * 1024 * 1024
  18. )
  19. type DevInfo struct {
  20. Hash string `json:"-"`
  21. DeviceId int `json:"device_id"`
  22. Size uint64 `json:"size"`
  23. TransactionId uint64 `json:"transaction_id"`
  24. Initialized bool `json:"initialized"`
  25. devices *DeviceSet `json:"-"`
  26. }
  27. type MetaData struct {
  28. Devices map[string]*DevInfo `json:devices`
  29. }
  30. type DeviceSet struct {
  31. MetaData
  32. sync.Mutex
  33. root string
  34. devicePrefix string
  35. TransactionId uint64
  36. NewTransactionId uint64
  37. nextFreeDevice int
  38. activeMounts map[string]int
  39. }
  40. type DiskUsage struct {
  41. Used uint64
  42. Total uint64
  43. }
  44. type Status struct {
  45. PoolName string
  46. DataLoopback string
  47. MetadataLoopback string
  48. Data DiskUsage
  49. Metadata DiskUsage
  50. SectorSize uint64
  51. }
  52. type DevStatus struct {
  53. DeviceId int
  54. Size uint64
  55. TransactionId uint64
  56. SizeInSectors uint64
  57. MappedSectors uint64
  58. HighestMappedSector uint64
  59. }
  60. func getDevName(name string) string {
  61. return "/dev/mapper/" + name
  62. }
  63. func (info *DevInfo) Name() string {
  64. hash := info.Hash
  65. if hash == "" {
  66. hash = "base"
  67. }
  68. return fmt.Sprintf("%s-%s", info.devices.devicePrefix, hash)
  69. }
  70. func (info *DevInfo) DevName() string {
  71. return getDevName(info.Name())
  72. }
  73. func (devices *DeviceSet) loopbackDir() string {
  74. return path.Join(devices.root, "devicemapper")
  75. }
  76. func (devices *DeviceSet) jsonFile() string {
  77. return path.Join(devices.loopbackDir(), "json")
  78. }
  79. func (devices *DeviceSet) getPoolName() string {
  80. return devices.devicePrefix + "-pool"
  81. }
  82. func (devices *DeviceSet) getPoolDevName() string {
  83. return getDevName(devices.getPoolName())
  84. }
  85. func (devices *DeviceSet) hasImage(name string) bool {
  86. dirname := devices.loopbackDir()
  87. filename := path.Join(dirname, name)
  88. _, err := osStat(filename)
  89. return err == nil
  90. }
  91. // ensureImage creates a sparse file of <size> bytes at the path
  92. // <root>/devicemapper/<name>.
  93. // If the file already exists, it does nothing.
  94. // Either way it returns the full path.
  95. func (devices *DeviceSet) ensureImage(name string, size int64) (string, error) {
  96. dirname := devices.loopbackDir()
  97. filename := path.Join(dirname, name)
  98. if err := osMkdirAll(dirname, 0700); err != nil && !osIsExist(err) {
  99. return "", err
  100. }
  101. if _, err := osStat(filename); err != nil {
  102. if !osIsNotExist(err) {
  103. return "", err
  104. }
  105. utils.Debugf("Creating loopback file %s for device-manage use", filename)
  106. file, err := osOpenFile(filename, osORdWr|osOCreate, 0600)
  107. if err != nil {
  108. return "", err
  109. }
  110. defer file.Close()
  111. if err = file.Truncate(size); err != nil {
  112. return "", err
  113. }
  114. }
  115. return filename, nil
  116. }
  117. func (devices *DeviceSet) allocateDeviceId() int {
  118. // TODO: Add smarter reuse of deleted devices
  119. id := devices.nextFreeDevice
  120. devices.nextFreeDevice = devices.nextFreeDevice + 1
  121. return id
  122. }
  123. func (devices *DeviceSet) allocateTransactionId() uint64 {
  124. devices.NewTransactionId = devices.NewTransactionId + 1
  125. return devices.NewTransactionId
  126. }
  127. func (devices *DeviceSet) saveMetadata() error {
  128. jsonData, err := json.Marshal(devices.MetaData)
  129. if err != nil {
  130. return fmt.Errorf("Error encoding metaadata to json: %s", err)
  131. }
  132. tmpFile, err := ioutil.TempFile(filepath.Dir(devices.jsonFile()), ".json")
  133. if err != nil {
  134. return fmt.Errorf("Error creating metadata file: %s", err)
  135. }
  136. n, err := tmpFile.Write(jsonData)
  137. if err != nil {
  138. return fmt.Errorf("Error writing metadata to %s: %s", tmpFile.Name(), err)
  139. }
  140. if n < len(jsonData) {
  141. return io.ErrShortWrite
  142. }
  143. if err := tmpFile.Sync(); err != nil {
  144. return fmt.Errorf("Error syncing metadata file %s: %s", tmpFile.Name(), err)
  145. }
  146. if err := tmpFile.Close(); err != nil {
  147. return fmt.Errorf("Error closing metadata file %s: %s", tmpFile.Name(), err)
  148. }
  149. if err := osRename(tmpFile.Name(), devices.jsonFile()); err != nil {
  150. return fmt.Errorf("Error committing metadata file", err)
  151. }
  152. if devices.NewTransactionId != devices.TransactionId {
  153. if err = setTransactionId(devices.getPoolDevName(), devices.TransactionId, devices.NewTransactionId); err != nil {
  154. return fmt.Errorf("Error setting devmapper transition ID: %s", err)
  155. }
  156. devices.TransactionId = devices.NewTransactionId
  157. }
  158. return nil
  159. }
  160. func (devices *DeviceSet) registerDevice(id int, hash string, size uint64) (*DevInfo, error) {
  161. utils.Debugf("registerDevice(%v, %v)", id, hash)
  162. info := &DevInfo{
  163. Hash: hash,
  164. DeviceId: id,
  165. Size: size,
  166. TransactionId: devices.allocateTransactionId(),
  167. Initialized: false,
  168. devices: devices,
  169. }
  170. devices.Devices[hash] = info
  171. if err := devices.saveMetadata(); err != nil {
  172. // Try to remove unused device
  173. delete(devices.Devices, hash)
  174. return nil, err
  175. }
  176. return info, nil
  177. }
  178. func (devices *DeviceSet) activateDeviceIfNeeded(hash string) error {
  179. utils.Debugf("activateDeviceIfNeeded(%v)", hash)
  180. info := devices.Devices[hash]
  181. if info == nil {
  182. return fmt.Errorf("Unknown device %s", hash)
  183. }
  184. if devinfo, _ := getInfo(info.Name()); devinfo != nil && devinfo.Exists != 0 {
  185. return nil
  186. }
  187. return activateDevice(devices.getPoolDevName(), info.Name(), info.DeviceId, info.Size)
  188. }
  189. func (devices *DeviceSet) createFilesystem(info *DevInfo) error {
  190. devname := info.DevName()
  191. err := execRun("mkfs.ext4", "-E", "discard,lazy_itable_init=0,lazy_journal_init=0", devname)
  192. if err != nil {
  193. err = execRun("mkfs.ext4", "-E", "discard,lazy_itable_init=0", devname)
  194. }
  195. if err != nil {
  196. utils.Debugf("\n--->Err: %s\n", err)
  197. return err
  198. }
  199. return nil
  200. }
  201. func (devices *DeviceSet) loadMetaData() error {
  202. utils.Debugf("loadMetadata()")
  203. defer utils.Debugf("loadMetadata END")
  204. _, _, _, params, err := getStatus(devices.getPoolName())
  205. if err != nil {
  206. utils.Debugf("\n--->Err: %s\n", err)
  207. return err
  208. }
  209. if _, err := fmt.Sscanf(params, "%d", &devices.TransactionId); err != nil {
  210. utils.Debugf("\n--->Err: %s\n", err)
  211. return err
  212. }
  213. devices.NewTransactionId = devices.TransactionId
  214. jsonData, err := ioutil.ReadFile(devices.jsonFile())
  215. if err != nil && !osIsNotExist(err) {
  216. utils.Debugf("\n--->Err: %s\n", err)
  217. return err
  218. }
  219. devices.MetaData.Devices = make(map[string]*DevInfo)
  220. if jsonData != nil {
  221. if err := json.Unmarshal(jsonData, &devices.MetaData); err != nil {
  222. utils.Debugf("\n--->Err: %s\n", err)
  223. return err
  224. }
  225. }
  226. for hash, d := range devices.Devices {
  227. d.Hash = hash
  228. d.devices = devices
  229. if d.DeviceId >= devices.nextFreeDevice {
  230. devices.nextFreeDevice = d.DeviceId + 1
  231. }
  232. // If the transaction id is larger than the actual one we lost the device due to some crash
  233. if d.TransactionId > devices.TransactionId {
  234. utils.Debugf("Removing lost device %s with id %d", hash, d.TransactionId)
  235. delete(devices.Devices, hash)
  236. }
  237. }
  238. return nil
  239. }
  240. func (devices *DeviceSet) setupBaseImage() error {
  241. oldInfo := devices.Devices[""]
  242. if oldInfo != nil && oldInfo.Initialized {
  243. return nil
  244. }
  245. if oldInfo != nil && !oldInfo.Initialized {
  246. utils.Debugf("Removing uninitialized base image")
  247. if err := devices.removeDevice(""); err != nil {
  248. utils.Debugf("\n--->Err: %s\n", err)
  249. return err
  250. }
  251. }
  252. utils.Debugf("Initializing base device-manager snapshot")
  253. id := devices.allocateDeviceId()
  254. // Create initial device
  255. if err := createDevice(devices.getPoolDevName(), id); err != nil {
  256. utils.Debugf("\n--->Err: %s\n", err)
  257. return err
  258. }
  259. utils.Debugf("Registering base device (id %v) with FS size %v", id, DefaultBaseFsSize)
  260. info, err := devices.registerDevice(id, "", DefaultBaseFsSize)
  261. if err != nil {
  262. _ = deleteDevice(devices.getPoolDevName(), id)
  263. utils.Debugf("\n--->Err: %s\n", err)
  264. return err
  265. }
  266. utils.Debugf("Creating filesystem on base device-manager snapshot")
  267. if err = devices.activateDeviceIfNeeded(""); err != nil {
  268. utils.Debugf("\n--->Err: %s\n", err)
  269. return err
  270. }
  271. if err := devices.createFilesystem(info); err != nil {
  272. utils.Debugf("\n--->Err: %s\n", err)
  273. return err
  274. }
  275. info.Initialized = true
  276. if err = devices.saveMetadata(); err != nil {
  277. info.Initialized = false
  278. utils.Debugf("\n--->Err: %s\n", err)
  279. return err
  280. }
  281. return nil
  282. }
  283. func setCloseOnExec(name string) {
  284. if fileInfos, _ := ioutil.ReadDir("/proc/self/fd"); fileInfos != nil {
  285. for _, i := range fileInfos {
  286. link, _ := osReadlink(filepath.Join("/proc/self/fd", i.Name()))
  287. if link == name {
  288. fd, err := strconv.Atoi(i.Name())
  289. if err == nil {
  290. sysCloseOnExec(fd)
  291. }
  292. }
  293. }
  294. }
  295. }
  296. func (devices *DeviceSet) log(level int, file string, line int, dmError int, message string) {
  297. if level >= 7 {
  298. return // Ignore _LOG_DEBUG
  299. }
  300. utils.Debugf("libdevmapper(%d): %s:%d (%d) %s", level, file, line, dmError, message)
  301. }
  302. func major(device uint64) uint64 {
  303. return (device >> 8) & 0xfff
  304. }
  305. func minor(device uint64) uint64 {
  306. return (device & 0xff) | ((device >> 12) & 0xfff00)
  307. }
  308. func (devices *DeviceSet) ResizePool(size int64) error {
  309. dirname := devices.loopbackDir()
  310. datafilename := path.Join(dirname, "data")
  311. metadatafilename := path.Join(dirname, "metadata")
  312. datafile, err := osOpenFile(datafilename, osORdWr, 0)
  313. if datafile == nil {
  314. return err
  315. }
  316. defer datafile.Close()
  317. fi, err := datafile.Stat()
  318. if fi == nil {
  319. return err
  320. }
  321. if fi.Size() > size {
  322. return fmt.Errorf("Can't shrink file")
  323. }
  324. dataloopback := FindLoopDeviceFor(&osFile{File: datafile})
  325. if dataloopback == nil {
  326. return fmt.Errorf("Unable to find loopback mount for: %s", datafilename)
  327. }
  328. defer dataloopback.Close()
  329. metadatafile, err := osOpenFile(metadatafilename, osORdWr, 0)
  330. if metadatafile == nil {
  331. return err
  332. }
  333. defer metadatafile.Close()
  334. metadataloopback := FindLoopDeviceFor(&osFile{File: metadatafile})
  335. if metadataloopback == nil {
  336. return fmt.Errorf("Unable to find loopback mount for: %s", metadatafilename)
  337. }
  338. defer metadataloopback.Close()
  339. // Grow loopback file
  340. if err := datafile.Truncate(size); err != nil {
  341. return fmt.Errorf("Unable to grow loopback file: %s", err)
  342. }
  343. // Reload size for loopback device
  344. if err := LoopbackSetCapacity(dataloopback); err != nil {
  345. return fmt.Errorf("Unable to update loopback capacity: %s", err)
  346. }
  347. // Suspend the pool
  348. if err := suspendDevice(devices.getPoolName()); err != nil {
  349. return fmt.Errorf("Unable to suspend pool: %s", err)
  350. }
  351. // Reload with the new block sizes
  352. if err := reloadPool(devices.getPoolName(), dataloopback, metadataloopback); err != nil {
  353. return fmt.Errorf("Unable to reload pool: %s", err)
  354. }
  355. // Resume the pool
  356. if err := resumeDevice(devices.getPoolName()); err != nil {
  357. return fmt.Errorf("Unable to resume pool: %s", err)
  358. }
  359. return nil
  360. }
  361. func (devices *DeviceSet) initDevmapper(doInit bool) error {
  362. logInit(devices)
  363. // Make sure the sparse images exist in <root>/devicemapper/data and
  364. // <root>/devicemapper/metadata
  365. hasData := devices.hasImage("data")
  366. hasMetadata := devices.hasImage("metadata")
  367. if !doInit && !hasData {
  368. return fmt.Errorf("Looback data file not found %s")
  369. }
  370. if !doInit && !hasMetadata {
  371. return fmt.Errorf("Looback metadata file not found %s")
  372. }
  373. createdLoopback := !hasData || !hasMetadata
  374. data, err := devices.ensureImage("data", DefaultDataLoopbackSize)
  375. if err != nil {
  376. utils.Debugf("Error device ensureImage (data): %s\n", err)
  377. return err
  378. }
  379. metadata, err := devices.ensureImage("metadata", DefaultMetaDataLoopbackSize)
  380. if err != nil {
  381. utils.Debugf("Error device ensureImage (metadata): %s\n", err)
  382. return err
  383. }
  384. // Set the device prefix from the device id and inode of the docker root dir
  385. st, err := osStat(devices.root)
  386. if err != nil {
  387. return fmt.Errorf("Error looking up dir %s: %s", devices.root, err)
  388. }
  389. sysSt := toSysStatT(st.Sys())
  390. // "reg-" stands for "regular file".
  391. // In the future we might use "dev-" for "device file", etc.
  392. // docker-maj,min[-inode] stands for:
  393. // - Managed by docker
  394. // - The target of this device is at major <maj> and minor <min>
  395. // - If <inode> is defined, use that file inside the device as a loopback image. Otherwise use the device itself.
  396. devices.devicePrefix = fmt.Sprintf("docker-%d:%d-%d", major(sysSt.Dev), minor(sysSt.Dev), sysSt.Ino)
  397. utils.Debugf("Generated prefix: %s", devices.devicePrefix)
  398. // Check for the existence of the device <prefix>-pool
  399. utils.Debugf("Checking for existence of the pool '%s'", devices.getPoolName())
  400. info, err := getInfo(devices.getPoolName())
  401. if info == nil {
  402. utils.Debugf("Error device getInfo: %s", err)
  403. return err
  404. }
  405. // It seems libdevmapper opens this without O_CLOEXEC, and go exec will not close files
  406. // that are not Close-on-exec, and lxc-start will die if it inherits any unexpected files,
  407. // so we add this badhack to make sure it closes itself
  408. setCloseOnExec("/dev/mapper/control")
  409. // If the pool doesn't exist, create it
  410. if info.Exists == 0 {
  411. utils.Debugf("Pool doesn't exist. Creating it.")
  412. dataFile, err := AttachLoopDevice(data)
  413. if err != nil {
  414. utils.Debugf("\n--->Err: %s\n", err)
  415. return err
  416. }
  417. defer dataFile.Close()
  418. metadataFile, err := AttachLoopDevice(metadata)
  419. if err != nil {
  420. utils.Debugf("\n--->Err: %s\n", err)
  421. return err
  422. }
  423. defer metadataFile.Close()
  424. if err := createPool(devices.getPoolName(), dataFile, metadataFile); err != nil {
  425. utils.Debugf("\n--->Err: %s\n", err)
  426. return err
  427. }
  428. }
  429. // If we didn't just create the data or metadata image, we need to
  430. // load the metadata from the existing file.
  431. if !createdLoopback {
  432. if err = devices.loadMetaData(); err != nil {
  433. utils.Debugf("\n--->Err: %s\n", err)
  434. return err
  435. }
  436. }
  437. // Setup the base image
  438. if doInit {
  439. if err := devices.setupBaseImage(); err != nil {
  440. utils.Debugf("Error device setupBaseImage: %s\n", err)
  441. return err
  442. }
  443. }
  444. return nil
  445. }
  446. func (devices *DeviceSet) AddDevice(hash, baseHash string) error {
  447. devices.Lock()
  448. defer devices.Unlock()
  449. if devices.Devices[hash] != nil {
  450. return fmt.Errorf("hash %s already exists", hash)
  451. }
  452. baseInfo := devices.Devices[baseHash]
  453. if baseInfo == nil {
  454. return fmt.Errorf("Error adding device for '%s': can't find device for parent '%s'", hash, baseHash)
  455. }
  456. deviceId := devices.allocateDeviceId()
  457. if err := devices.createSnapDevice(devices.getPoolDevName(), deviceId, baseInfo.Name(), baseInfo.DeviceId); err != nil {
  458. utils.Debugf("Error creating snap device: %s\n", err)
  459. return err
  460. }
  461. if _, err := devices.registerDevice(deviceId, hash, baseInfo.Size); err != nil {
  462. deleteDevice(devices.getPoolDevName(), deviceId)
  463. utils.Debugf("Error registering device: %s\n", err)
  464. return err
  465. }
  466. return nil
  467. }
  468. func (devices *DeviceSet) removeDevice(hash string) error {
  469. info := devices.Devices[hash]
  470. if info == nil {
  471. return fmt.Errorf("hash %s doesn't exists", hash)
  472. }
  473. devinfo, _ := getInfo(info.Name())
  474. if devinfo != nil && devinfo.Exists != 0 {
  475. if err := removeDevice(info.Name()); err != nil {
  476. utils.Debugf("Error removing device: %s\n", err)
  477. return err
  478. }
  479. }
  480. if info.Initialized {
  481. info.Initialized = false
  482. if err := devices.saveMetadata(); err != nil {
  483. utils.Debugf("Error saving meta data: %s\n", err)
  484. return err
  485. }
  486. }
  487. if err := deleteDevice(devices.getPoolDevName(), info.DeviceId); err != nil {
  488. utils.Debugf("Error deleting device: %s\n", err)
  489. return err
  490. }
  491. devices.allocateTransactionId()
  492. delete(devices.Devices, info.Hash)
  493. if err := devices.saveMetadata(); err != nil {
  494. devices.Devices[info.Hash] = info
  495. utils.Debugf("Error saving meta data: %s\n", err)
  496. return err
  497. }
  498. return nil
  499. }
  500. func (devices *DeviceSet) RemoveDevice(hash string) error {
  501. devices.Lock()
  502. defer devices.Unlock()
  503. return devices.removeDevice(hash)
  504. }
  505. func (devices *DeviceSet) deactivateDevice(hash string) error {
  506. utils.Debugf("[devmapper] deactivateDevice(%s)", hash)
  507. defer utils.Debugf("[devmapper] deactivateDevice END")
  508. var devname string
  509. // FIXME: shouldn't we just register the pool into devices?
  510. devname, err := devices.byHash(hash)
  511. if err != nil {
  512. return err
  513. }
  514. devinfo, err := getInfo(devname)
  515. if err != nil {
  516. utils.Debugf("\n--->Err: %s\n", err)
  517. return err
  518. }
  519. if devinfo.Exists != 0 {
  520. if err := removeDevice(devname); err != nil {
  521. utils.Debugf("\n--->Err: %s\n", err)
  522. return err
  523. }
  524. if err := devices.waitRemove(hash); err != nil {
  525. return err
  526. }
  527. }
  528. return nil
  529. }
  530. // waitRemove blocks until either:
  531. // a) the device registered at <device_set_prefix>-<hash> is removed,
  532. // or b) the 1 second timeout expires.
  533. func (devices *DeviceSet) waitRemove(hash string) error {
  534. utils.Debugf("[deviceset %s] waitRemove(%s)", devices.devicePrefix, hash)
  535. defer utils.Debugf("[deviceset %s] waitRemove END", devices.devicePrefix, hash)
  536. devname, err := devices.byHash(hash)
  537. if err != nil {
  538. return err
  539. }
  540. i := 0
  541. for ; i < 1000; i += 1 {
  542. devinfo, err := getInfo(devname)
  543. if err != nil {
  544. // If there is an error we assume the device doesn't exist.
  545. // The error might actually be something else, but we can't differentiate.
  546. return nil
  547. }
  548. if i%100 == 0 {
  549. utils.Debugf("Waiting for removal of %s: exists=%d", devname, devinfo.Exists)
  550. }
  551. if devinfo.Exists == 0 {
  552. break
  553. }
  554. time.Sleep(1 * time.Millisecond)
  555. }
  556. if i == 1000 {
  557. return fmt.Errorf("Timeout while waiting for device %s to be removed", devname)
  558. }
  559. return nil
  560. }
  561. // waitClose blocks until either:
  562. // a) the device registered at <device_set_prefix>-<hash> is closed,
  563. // or b) the 1 second timeout expires.
  564. func (devices *DeviceSet) waitClose(hash string) error {
  565. devname, err := devices.byHash(hash)
  566. if err != nil {
  567. return err
  568. }
  569. i := 0
  570. for ; i < 1000; i += 1 {
  571. devinfo, err := getInfo(devname)
  572. if err != nil {
  573. return err
  574. }
  575. if i%100 == 0 {
  576. utils.Debugf("Waiting for unmount of %s: opencount=%d", devname, devinfo.OpenCount)
  577. }
  578. if devinfo.OpenCount == 0 {
  579. break
  580. }
  581. time.Sleep(1 * time.Millisecond)
  582. }
  583. if i == 1000 {
  584. return fmt.Errorf("Timeout while waiting for device %s to close", devname)
  585. }
  586. return nil
  587. }
  588. // byHash is a hack to allow looking up the deviceset's pool by the hash "pool".
  589. // FIXME: it seems probably cleaner to register the pool in devices.Devices,
  590. // but I am afraid of arcane implications deep in the devicemapper code,
  591. // so this will do.
  592. func (devices *DeviceSet) byHash(hash string) (devname string, err error) {
  593. if hash == "pool" {
  594. return devices.getPoolDevName(), nil
  595. }
  596. info := devices.Devices[hash]
  597. if info == nil {
  598. return "", fmt.Errorf("hash %s doesn't exists", hash)
  599. }
  600. return info.Name(), nil
  601. }
  602. func (devices *DeviceSet) Shutdown() error {
  603. devices.Lock()
  604. defer devices.Unlock()
  605. utils.Debugf("[deviceset %s] shutdown()", devices.devicePrefix)
  606. utils.Debugf("[devmapper] Shutting down DeviceSet: %s", devices.root)
  607. defer utils.Debugf("[deviceset %s] shutdown END", devices.devicePrefix)
  608. for path, count := range devices.activeMounts {
  609. for i := count; i > 0; i-- {
  610. if err := sysUnmount(path, 0); err != nil {
  611. utils.Debugf("Shutdown unmounting %s, error: %s\n", path, err)
  612. }
  613. }
  614. delete(devices.activeMounts, path)
  615. }
  616. for _, d := range devices.Devices {
  617. if err := devices.waitClose(d.Hash); err != nil {
  618. utils.Errorf("Warning: error waiting for device %s to unmount: %s\n", d.Hash, err)
  619. }
  620. if err := devices.deactivateDevice(d.Hash); err != nil {
  621. utils.Debugf("Shutdown deactivate %s , error: %s\n", d.Hash, err)
  622. }
  623. }
  624. pool := devices.getPoolDevName()
  625. if devinfo, err := getInfo(pool); err == nil && devinfo.Exists != 0 {
  626. if err := devices.deactivateDevice("pool"); err != nil {
  627. utils.Debugf("Shutdown deactivate %s , error: %s\n", pool, err)
  628. }
  629. }
  630. return nil
  631. }
  632. func (devices *DeviceSet) MountDevice(hash, path string, readOnly bool) error {
  633. devices.Lock()
  634. defer devices.Unlock()
  635. if err := devices.activateDeviceIfNeeded(hash); err != nil {
  636. return fmt.Errorf("Error activating devmapper device for '%s': %s", hash, err)
  637. }
  638. info := devices.Devices[hash]
  639. var flags uintptr = sysMsMgcVal
  640. if readOnly {
  641. flags = flags | sysMsRdOnly
  642. }
  643. err := sysMount(info.DevName(), path, "ext4", flags, "discard")
  644. if err != nil && err == sysEInval {
  645. err = sysMount(info.DevName(), path, "ext4", flags, "")
  646. }
  647. if err != nil {
  648. return fmt.Errorf("Error mounting '%s' on '%s': %s", info.DevName(), path, err)
  649. }
  650. count := devices.activeMounts[path]
  651. devices.activeMounts[path] = count + 1
  652. return devices.setInitialized(hash)
  653. }
  654. func (devices *DeviceSet) UnmountDevice(hash, path string, deactivate bool) error {
  655. utils.Debugf("[devmapper] UnmountDevice(hash=%s path=%s)", hash, path)
  656. defer utils.Debugf("[devmapper] UnmountDevice END")
  657. devices.Lock()
  658. defer devices.Unlock()
  659. utils.Debugf("[devmapper] Unmount(%s)", path)
  660. if err := sysUnmount(path, 0); err != nil {
  661. utils.Debugf("\n--->Err: %s\n", err)
  662. return err
  663. }
  664. utils.Debugf("[devmapper] Unmount done")
  665. // Wait for the unmount to be effective,
  666. // by watching the value of Info.OpenCount for the device
  667. if err := devices.waitClose(hash); err != nil {
  668. return err
  669. }
  670. if count := devices.activeMounts[path]; count > 1 {
  671. devices.activeMounts[path] = count - 1
  672. } else {
  673. delete(devices.activeMounts, path)
  674. }
  675. if deactivate {
  676. devices.deactivateDevice(hash)
  677. }
  678. return nil
  679. }
  680. func (devices *DeviceSet) HasDevice(hash string) bool {
  681. devices.Lock()
  682. defer devices.Unlock()
  683. return devices.Devices[hash] != nil
  684. }
  685. func (devices *DeviceSet) HasInitializedDevice(hash string) bool {
  686. devices.Lock()
  687. defer devices.Unlock()
  688. info := devices.Devices[hash]
  689. return info != nil && info.Initialized
  690. }
  691. func (devices *DeviceSet) HasActivatedDevice(hash string) bool {
  692. devices.Lock()
  693. defer devices.Unlock()
  694. info := devices.Devices[hash]
  695. if info == nil {
  696. return false
  697. }
  698. devinfo, _ := getInfo(info.Name())
  699. return devinfo != nil && devinfo.Exists != 0
  700. }
  701. func (devices *DeviceSet) setInitialized(hash string) error {
  702. info := devices.Devices[hash]
  703. if info == nil {
  704. return fmt.Errorf("Unknown device %s", hash)
  705. }
  706. info.Initialized = true
  707. if err := devices.saveMetadata(); err != nil {
  708. info.Initialized = false
  709. utils.Debugf("\n--->Err: %s\n", err)
  710. return err
  711. }
  712. return nil
  713. }
  714. func (devices *DeviceSet) List() []string {
  715. devices.Lock()
  716. defer devices.Unlock()
  717. ids := make([]string, len(devices.Devices))
  718. i := 0
  719. for k := range devices.Devices {
  720. ids[i] = k
  721. i++
  722. }
  723. return ids
  724. }
  725. func (devices *DeviceSet) deviceStatus(devName string) (sizeInSectors, mappedSectors, highestMappedSector uint64, err error) {
  726. var params string
  727. _, sizeInSectors, _, params, err = getStatus(devName)
  728. if err != nil {
  729. return
  730. }
  731. if _, err = fmt.Sscanf(params, "%d %d", &mappedSectors, &highestMappedSector); err == nil {
  732. return
  733. }
  734. return
  735. }
  736. func (devices *DeviceSet) GetDeviceStatus(hash string) (*DevStatus, error) {
  737. devices.Lock()
  738. defer devices.Unlock()
  739. info := devices.Devices[hash]
  740. if info == nil {
  741. return nil, fmt.Errorf("No device %s", hash)
  742. }
  743. status := &DevStatus{
  744. DeviceId: info.DeviceId,
  745. Size: info.Size,
  746. TransactionId: info.TransactionId,
  747. }
  748. if err := devices.activateDeviceIfNeeded(hash); err != nil {
  749. return nil, fmt.Errorf("Error activating devmapper device for '%s': %s", hash, err)
  750. }
  751. if sizeInSectors, mappedSectors, highestMappedSector, err := devices.deviceStatus(info.DevName()); err != nil {
  752. return nil, err
  753. } else {
  754. status.SizeInSectors = sizeInSectors
  755. status.MappedSectors = mappedSectors
  756. status.HighestMappedSector = highestMappedSector
  757. }
  758. return status, nil
  759. }
  760. func (devices *DeviceSet) poolStatus() (totalSizeInSectors, transactionId, dataUsed, dataTotal, metadataUsed, metadataTotal uint64, err error) {
  761. var params string
  762. if _, totalSizeInSectors, _, params, err = getStatus(devices.getPoolName()); err == nil {
  763. _, err = fmt.Sscanf(params, "%d %d/%d %d/%d", &transactionId, &metadataUsed, &metadataTotal, &dataUsed, &dataTotal)
  764. }
  765. return
  766. }
  767. func (devices *DeviceSet) Status() *Status {
  768. devices.Lock()
  769. defer devices.Unlock()
  770. status := &Status{}
  771. status.PoolName = devices.getPoolName()
  772. status.DataLoopback = path.Join(devices.loopbackDir(), "data")
  773. status.MetadataLoopback = path.Join(devices.loopbackDir(), "metadata")
  774. totalSizeInSectors, _, dataUsed, dataTotal, metadataUsed, metadataTotal, err := devices.poolStatus()
  775. if err == nil {
  776. // Convert from blocks to bytes
  777. blockSizeInSectors := totalSizeInSectors / dataTotal
  778. status.Data.Used = dataUsed * blockSizeInSectors * 512
  779. status.Data.Total = dataTotal * blockSizeInSectors * 512
  780. // metadata blocks are always 4k
  781. status.Metadata.Used = metadataUsed * 4096
  782. status.Metadata.Total = metadataTotal * 4096
  783. status.SectorSize = blockSizeInSectors * 512
  784. }
  785. return status
  786. }
  787. func NewDeviceSet(root string, doInit bool) (*DeviceSet, error) {
  788. SetDevDir("/dev")
  789. devices := &DeviceSet{
  790. root: root,
  791. MetaData: MetaData{Devices: make(map[string]*DevInfo)},
  792. activeMounts: make(map[string]int),
  793. }
  794. if err := devices.initDevmapper(doInit); err != nil {
  795. return nil, err
  796. }
  797. return devices, nil
  798. }