deviceset.go 28 KB

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