deviceset.go 44 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678
  1. // +build linux
  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. "strings"
  15. "sync"
  16. "syscall"
  17. "time"
  18. log "github.com/Sirupsen/logrus"
  19. "github.com/docker/docker/daemon/graphdriver"
  20. "github.com/docker/docker/pkg/devicemapper"
  21. "github.com/docker/docker/pkg/parsers"
  22. "github.com/docker/docker/pkg/units"
  23. "github.com/docker/libcontainer/label"
  24. )
  25. var (
  26. DefaultDataLoopbackSize int64 = 100 * 1024 * 1024 * 1024
  27. DefaultMetaDataLoopbackSize int64 = 2 * 1024 * 1024 * 1024
  28. DefaultBaseFsSize uint64 = 10 * 1024 * 1024 * 1024
  29. DefaultThinpBlockSize uint32 = 128 // 64K = 128 512b sectors
  30. MaxDeviceId int = 0xffffff // 24 bit, pool limit
  31. DeviceIdMapSz int = (MaxDeviceId + 1) / 8
  32. )
  33. const deviceSetMetaFile string = "deviceset-metadata"
  34. const transactionMetaFile string = "transaction-metadata"
  35. type Transaction struct {
  36. OpenTransactionId uint64 `json:"open_transaction_id"`
  37. DeviceIdHash string `json:"device_hash"`
  38. DeviceId int `json:"device_id"`
  39. }
  40. type DevInfo struct {
  41. Hash string `json:"-"`
  42. DeviceId int `json:"device_id"`
  43. Size uint64 `json:"size"`
  44. TransactionId uint64 `json:"transaction_id"`
  45. Initialized bool `json:"initialized"`
  46. devices *DeviceSet
  47. mountCount int
  48. mountPath string
  49. // The global DeviceSet lock guarantees that we serialize all
  50. // the calls to libdevmapper (which is not threadsafe), but we
  51. // sometimes release that lock while sleeping. In that case
  52. // this per-device lock is still held, protecting against
  53. // other accesses to the device that we're doing the wait on.
  54. //
  55. // WARNING: In order to avoid AB-BA deadlocks when releasing
  56. // the global lock while holding the per-device locks all
  57. // device locks must be aquired *before* the device lock, and
  58. // multiple device locks should be aquired parent before child.
  59. lock sync.Mutex
  60. }
  61. type MetaData struct {
  62. Devices map[string]*DevInfo `json:"Devices"`
  63. devicesLock sync.Mutex // Protects all read/writes to Devices map
  64. }
  65. type DeviceSet struct {
  66. MetaData `json:"-"`
  67. sync.Mutex `json:"-"` // Protects Devices map and serializes calls into libdevmapper
  68. root string
  69. devicePrefix string
  70. TransactionId uint64 `json:"-"`
  71. NextDeviceId int `json:"next_device_id"`
  72. deviceIdMap []byte
  73. // Options
  74. dataLoopbackSize int64
  75. metaDataLoopbackSize int64
  76. baseFsSize uint64
  77. filesystem string
  78. mountOptions string
  79. mkfsArgs []string
  80. dataDevice string // block or loop dev
  81. dataLoopFile string // loopback file, if used
  82. metadataDevice string // block or loop dev
  83. metadataLoopFile string // loopback file, if used
  84. doBlkDiscard bool
  85. thinpBlockSize uint32
  86. thinPoolDevice string
  87. Transaction `json:"-"`
  88. }
  89. type DiskUsage struct {
  90. Used uint64
  91. Total uint64
  92. }
  93. type Status struct {
  94. PoolName string
  95. DataFile string // actual block device for data
  96. DataLoopback string // loopback file, if used
  97. MetadataFile string // actual block device for metadata
  98. MetadataLoopback string // loopback file, if used
  99. Data DiskUsage
  100. Metadata DiskUsage
  101. SectorSize uint64
  102. }
  103. type DevStatus struct {
  104. DeviceId int
  105. Size uint64
  106. TransactionId uint64
  107. SizeInSectors uint64
  108. MappedSectors uint64
  109. HighestMappedSector uint64
  110. }
  111. func getDevName(name string) string {
  112. return "/dev/mapper/" + name
  113. }
  114. func (info *DevInfo) Name() string {
  115. hash := info.Hash
  116. if hash == "" {
  117. hash = "base"
  118. }
  119. return fmt.Sprintf("%s-%s", info.devices.devicePrefix, hash)
  120. }
  121. func (info *DevInfo) DevName() string {
  122. return getDevName(info.Name())
  123. }
  124. func (devices *DeviceSet) loopbackDir() string {
  125. return path.Join(devices.root, "devicemapper")
  126. }
  127. func (devices *DeviceSet) metadataDir() string {
  128. return path.Join(devices.root, "metadata")
  129. }
  130. func (devices *DeviceSet) metadataFile(info *DevInfo) string {
  131. file := info.Hash
  132. if file == "" {
  133. file = "base"
  134. }
  135. return path.Join(devices.metadataDir(), file)
  136. }
  137. func (devices *DeviceSet) transactionMetaFile() string {
  138. return path.Join(devices.metadataDir(), transactionMetaFile)
  139. }
  140. func (devices *DeviceSet) deviceSetMetaFile() string {
  141. return path.Join(devices.metadataDir(), deviceSetMetaFile)
  142. }
  143. func (devices *DeviceSet) oldMetadataFile() string {
  144. return path.Join(devices.loopbackDir(), "json")
  145. }
  146. func (devices *DeviceSet) getPoolName() string {
  147. if devices.thinPoolDevice == "" {
  148. return devices.devicePrefix + "-pool"
  149. }
  150. return devices.thinPoolDevice
  151. }
  152. func (devices *DeviceSet) getPoolDevName() string {
  153. return getDevName(devices.getPoolName())
  154. }
  155. func (devices *DeviceSet) hasImage(name string) bool {
  156. dirname := devices.loopbackDir()
  157. filename := path.Join(dirname, name)
  158. _, err := os.Stat(filename)
  159. return err == nil
  160. }
  161. // ensureImage creates a sparse file of <size> bytes at the path
  162. // <root>/devicemapper/<name>.
  163. // If the file already exists, it does nothing.
  164. // Either way it returns the full path.
  165. func (devices *DeviceSet) ensureImage(name string, size int64) (string, error) {
  166. dirname := devices.loopbackDir()
  167. filename := path.Join(dirname, name)
  168. if err := os.MkdirAll(dirname, 0700); err != nil && !os.IsExist(err) {
  169. return "", err
  170. }
  171. if _, err := os.Stat(filename); err != nil {
  172. if !os.IsNotExist(err) {
  173. return "", err
  174. }
  175. log.Debugf("Creating loopback file %s for device-manage use", filename)
  176. file, err := os.OpenFile(filename, os.O_RDWR|os.O_CREATE, 0600)
  177. if err != nil {
  178. return "", err
  179. }
  180. defer file.Close()
  181. if err = file.Truncate(size); err != nil {
  182. return "", err
  183. }
  184. }
  185. return filename, nil
  186. }
  187. func (devices *DeviceSet) allocateTransactionId() uint64 {
  188. devices.OpenTransactionId = devices.TransactionId + 1
  189. return devices.OpenTransactionId
  190. }
  191. func (devices *DeviceSet) updatePoolTransactionId() error {
  192. if err := devicemapper.SetTransactionId(devices.getPoolDevName(), devices.TransactionId, devices.OpenTransactionId); err != nil {
  193. return fmt.Errorf("Error setting devmapper transaction ID: %s", err)
  194. }
  195. devices.TransactionId = devices.OpenTransactionId
  196. return nil
  197. }
  198. func (devices *DeviceSet) removeMetadata(info *DevInfo) error {
  199. if err := os.RemoveAll(devices.metadataFile(info)); err != nil {
  200. return fmt.Errorf("Error removing metadata file %s: %s", devices.metadataFile(info), err)
  201. }
  202. return nil
  203. }
  204. // Given json data and file path, write it to disk
  205. func (devices *DeviceSet) writeMetaFile(jsonData []byte, filePath string) error {
  206. tmpFile, err := ioutil.TempFile(devices.metadataDir(), ".tmp")
  207. if err != nil {
  208. return fmt.Errorf("Error creating metadata file: %s", err)
  209. }
  210. n, err := tmpFile.Write(jsonData)
  211. if err != nil {
  212. return fmt.Errorf("Error writing metadata to %s: %s", tmpFile.Name(), err)
  213. }
  214. if n < len(jsonData) {
  215. return io.ErrShortWrite
  216. }
  217. if err := tmpFile.Sync(); err != nil {
  218. return fmt.Errorf("Error syncing metadata file %s: %s", tmpFile.Name(), err)
  219. }
  220. if err := tmpFile.Close(); err != nil {
  221. return fmt.Errorf("Error closing metadata file %s: %s", tmpFile.Name(), err)
  222. }
  223. if err := os.Rename(tmpFile.Name(), filePath); err != nil {
  224. return fmt.Errorf("Error committing metadata file %s: %s", tmpFile.Name(), err)
  225. }
  226. return nil
  227. }
  228. func (devices *DeviceSet) saveMetadata(info *DevInfo) error {
  229. jsonData, err := json.Marshal(info)
  230. if err != nil {
  231. return fmt.Errorf("Error encoding metadata to json: %s", err)
  232. }
  233. if err := devices.writeMetaFile(jsonData, devices.metadataFile(info)); err != nil {
  234. return err
  235. }
  236. return nil
  237. }
  238. func (devices *DeviceSet) markDeviceIdUsed(deviceId int) {
  239. var mask byte
  240. i := deviceId % 8
  241. mask = 1 << uint(i)
  242. devices.deviceIdMap[deviceId/8] = devices.deviceIdMap[deviceId/8] | mask
  243. }
  244. func (devices *DeviceSet) markDeviceIdFree(deviceId int) {
  245. var mask byte
  246. i := deviceId % 8
  247. mask = ^(1 << uint(i))
  248. devices.deviceIdMap[deviceId/8] = devices.deviceIdMap[deviceId/8] & mask
  249. }
  250. func (devices *DeviceSet) isDeviceIdFree(deviceId int) bool {
  251. var mask byte
  252. i := deviceId % 8
  253. mask = (1 << uint(i))
  254. if (devices.deviceIdMap[deviceId/8] & mask) != 0 {
  255. return false
  256. }
  257. return true
  258. }
  259. func (devices *DeviceSet) lookupDevice(hash string) (*DevInfo, error) {
  260. devices.devicesLock.Lock()
  261. defer devices.devicesLock.Unlock()
  262. info := devices.Devices[hash]
  263. if info == nil {
  264. info = devices.loadMetadata(hash)
  265. if info == nil {
  266. return nil, fmt.Errorf("Unknown device %s", hash)
  267. }
  268. devices.Devices[hash] = info
  269. }
  270. return info, nil
  271. }
  272. func (devices *DeviceSet) deviceFileWalkFunction(path string, finfo os.FileInfo) error {
  273. // Skip some of the meta files which are not device files.
  274. if strings.HasSuffix(finfo.Name(), ".migrated") {
  275. log.Debugf("Skipping file %s", path)
  276. return nil
  277. }
  278. if finfo.Name() == deviceSetMetaFile {
  279. log.Debugf("Skipping file %s", path)
  280. return nil
  281. }
  282. log.Debugf("Loading data for file %s", path)
  283. hash := finfo.Name()
  284. if hash == "base" {
  285. hash = ""
  286. }
  287. dinfo := devices.loadMetadata(hash)
  288. if dinfo == nil {
  289. return fmt.Errorf("Error loading device metadata file %s", hash)
  290. }
  291. if dinfo.DeviceId > MaxDeviceId {
  292. log.Errorf("Warning: Ignoring Invalid DeviceId=%d", dinfo.DeviceId)
  293. return nil
  294. }
  295. devices.Lock()
  296. devices.markDeviceIdUsed(dinfo.DeviceId)
  297. devices.Unlock()
  298. log.Debugf("Added deviceId=%d to DeviceIdMap", dinfo.DeviceId)
  299. return nil
  300. }
  301. func (devices *DeviceSet) constructDeviceIdMap() error {
  302. log.Debugf("[deviceset] constructDeviceIdMap()")
  303. defer log.Debugf("[deviceset] constructDeviceIdMap() END")
  304. var scan = func(path string, info os.FileInfo, err error) error {
  305. if err != nil {
  306. log.Debugf("Can't walk the file %s", path)
  307. return nil
  308. }
  309. // Skip any directories
  310. if info.IsDir() {
  311. return nil
  312. }
  313. return devices.deviceFileWalkFunction(path, info)
  314. }
  315. return filepath.Walk(devices.metadataDir(), scan)
  316. }
  317. func (devices *DeviceSet) unregisterDevice(id int, hash string) error {
  318. log.Debugf("unregisterDevice(%v, %v)", id, hash)
  319. info := &DevInfo{
  320. Hash: hash,
  321. DeviceId: id,
  322. }
  323. devices.devicesLock.Lock()
  324. delete(devices.Devices, hash)
  325. devices.devicesLock.Unlock()
  326. if err := devices.removeMetadata(info); err != nil {
  327. log.Debugf("Error removing meta data: %s", err)
  328. return err
  329. }
  330. return nil
  331. }
  332. func (devices *DeviceSet) registerDevice(id int, hash string, size uint64, transactionId uint64) (*DevInfo, error) {
  333. log.Debugf("registerDevice(%v, %v)", id, hash)
  334. info := &DevInfo{
  335. Hash: hash,
  336. DeviceId: id,
  337. Size: size,
  338. TransactionId: transactionId,
  339. Initialized: false,
  340. devices: devices,
  341. }
  342. devices.devicesLock.Lock()
  343. devices.Devices[hash] = info
  344. devices.devicesLock.Unlock()
  345. if err := devices.saveMetadata(info); err != nil {
  346. // Try to remove unused device
  347. devices.devicesLock.Lock()
  348. delete(devices.Devices, hash)
  349. devices.devicesLock.Unlock()
  350. return nil, err
  351. }
  352. return info, nil
  353. }
  354. func (devices *DeviceSet) activateDeviceIfNeeded(info *DevInfo) error {
  355. log.Debugf("activateDeviceIfNeeded(%v)", info.Hash)
  356. if devinfo, _ := devicemapper.GetInfo(info.Name()); devinfo != nil && devinfo.Exists != 0 {
  357. return nil
  358. }
  359. return devicemapper.ActivateDevice(devices.getPoolDevName(), info.Name(), info.DeviceId, info.Size)
  360. }
  361. func (devices *DeviceSet) createFilesystem(info *DevInfo) error {
  362. devname := info.DevName()
  363. args := []string{}
  364. for _, arg := range devices.mkfsArgs {
  365. args = append(args, arg)
  366. }
  367. args = append(args, devname)
  368. var err error
  369. switch devices.filesystem {
  370. case "xfs":
  371. err = exec.Command("mkfs.xfs", args...).Run()
  372. case "ext4":
  373. err = exec.Command("mkfs.ext4", append([]string{"-E", "nodiscard,lazy_itable_init=0,lazy_journal_init=0"}, args...)...).Run()
  374. if err != nil {
  375. err = exec.Command("mkfs.ext4", append([]string{"-E", "nodiscard,lazy_itable_init=0"}, args...)...).Run()
  376. }
  377. if err != nil {
  378. return err
  379. }
  380. err = exec.Command("tune2fs", append([]string{"-c", "-1", "-i", "0"}, devname)...).Run()
  381. default:
  382. err = fmt.Errorf("Unsupported filesystem type %s", devices.filesystem)
  383. }
  384. if err != nil {
  385. return err
  386. }
  387. return nil
  388. }
  389. func (devices *DeviceSet) migrateOldMetaData() error {
  390. // Migrate old metadatafile
  391. jsonData, err := ioutil.ReadFile(devices.oldMetadataFile())
  392. if err != nil && !os.IsNotExist(err) {
  393. return err
  394. }
  395. if jsonData != nil {
  396. m := MetaData{Devices: make(map[string]*DevInfo)}
  397. if err := json.Unmarshal(jsonData, &m); err != nil {
  398. return err
  399. }
  400. for hash, info := range m.Devices {
  401. info.Hash = hash
  402. devices.saveMetadata(info)
  403. }
  404. if err := os.Rename(devices.oldMetadataFile(), devices.oldMetadataFile()+".migrated"); err != nil {
  405. return err
  406. }
  407. }
  408. return nil
  409. }
  410. func (devices *DeviceSet) initMetaData() error {
  411. if err := devices.migrateOldMetaData(); err != nil {
  412. return err
  413. }
  414. _, transactionId, _, _, _, _, err := devices.poolStatus()
  415. if err != nil {
  416. return err
  417. }
  418. devices.TransactionId = transactionId
  419. if err := devices.constructDeviceIdMap(); err != nil {
  420. return err
  421. }
  422. if err := devices.processPendingTransaction(); err != nil {
  423. return err
  424. }
  425. return nil
  426. }
  427. func (devices *DeviceSet) incNextDeviceId() {
  428. // Ids are 24bit, so wrap around
  429. devices.NextDeviceId = (devices.NextDeviceId + 1) & MaxDeviceId
  430. }
  431. func (devices *DeviceSet) getNextFreeDeviceId() (int, error) {
  432. devices.incNextDeviceId()
  433. for i := 0; i <= MaxDeviceId; i++ {
  434. if devices.isDeviceIdFree(devices.NextDeviceId) {
  435. devices.markDeviceIdUsed(devices.NextDeviceId)
  436. return devices.NextDeviceId, nil
  437. }
  438. devices.incNextDeviceId()
  439. }
  440. return 0, fmt.Errorf("Unable to find a free device Id")
  441. }
  442. func (devices *DeviceSet) createRegisterDevice(hash string) (*DevInfo, error) {
  443. deviceId, err := devices.getNextFreeDeviceId()
  444. if err != nil {
  445. return nil, err
  446. }
  447. if err := devices.openTransaction(hash, deviceId); err != nil {
  448. log.Debugf("Error opening transaction hash = %s deviceId = %d", hash, deviceId)
  449. devices.markDeviceIdFree(deviceId)
  450. return nil, err
  451. }
  452. for {
  453. if err := devicemapper.CreateDevice(devices.getPoolDevName(), deviceId); err != nil {
  454. if devicemapper.DeviceIdExists(err) {
  455. // Device Id already exists. This should not
  456. // happen. Now we have a mechianism to find
  457. // a free device Id. So something is not right.
  458. // Give a warning and continue.
  459. log.Errorf("Warning: Device Id %d exists in pool but it is supposed to be unused", deviceId)
  460. deviceId, err = devices.getNextFreeDeviceId()
  461. if err != nil {
  462. return nil, err
  463. }
  464. // Save new device id into transaction
  465. devices.refreshTransaction(deviceId)
  466. continue
  467. }
  468. log.Debugf("Error creating device: %s", err)
  469. devices.markDeviceIdFree(deviceId)
  470. return nil, err
  471. }
  472. break
  473. }
  474. log.Debugf("Registering device (id %v) with FS size %v", deviceId, devices.baseFsSize)
  475. info, err := devices.registerDevice(deviceId, hash, devices.baseFsSize, devices.OpenTransactionId)
  476. if err != nil {
  477. _ = devicemapper.DeleteDevice(devices.getPoolDevName(), deviceId)
  478. devices.markDeviceIdFree(deviceId)
  479. return nil, err
  480. }
  481. if err := devices.closeTransaction(); err != nil {
  482. devices.unregisterDevice(deviceId, hash)
  483. devicemapper.DeleteDevice(devices.getPoolDevName(), deviceId)
  484. devices.markDeviceIdFree(deviceId)
  485. return nil, err
  486. }
  487. return info, nil
  488. }
  489. func (devices *DeviceSet) createRegisterSnapDevice(hash string, baseInfo *DevInfo) error {
  490. deviceId, err := devices.getNextFreeDeviceId()
  491. if err != nil {
  492. return err
  493. }
  494. if err := devices.openTransaction(hash, deviceId); err != nil {
  495. log.Debugf("Error opening transaction hash = %s deviceId = %d", hash, deviceId)
  496. devices.markDeviceIdFree(deviceId)
  497. return err
  498. }
  499. for {
  500. if err := devicemapper.CreateSnapDevice(devices.getPoolDevName(), deviceId, baseInfo.Name(), baseInfo.DeviceId); err != nil {
  501. if devicemapper.DeviceIdExists(err) {
  502. // Device Id already exists. This should not
  503. // happen. Now we have a mechianism to find
  504. // a free device Id. So something is not right.
  505. // Give a warning and continue.
  506. log.Errorf("Warning: Device Id %d exists in pool but it is supposed to be unused", deviceId)
  507. deviceId, err = devices.getNextFreeDeviceId()
  508. if err != nil {
  509. return err
  510. }
  511. // Save new device id into transaction
  512. devices.refreshTransaction(deviceId)
  513. continue
  514. }
  515. log.Debugf("Error creating snap device: %s", err)
  516. devices.markDeviceIdFree(deviceId)
  517. return err
  518. }
  519. break
  520. }
  521. if _, err := devices.registerDevice(deviceId, hash, baseInfo.Size, devices.OpenTransactionId); err != nil {
  522. devicemapper.DeleteDevice(devices.getPoolDevName(), deviceId)
  523. devices.markDeviceIdFree(deviceId)
  524. log.Debugf("Error registering device: %s", err)
  525. return err
  526. }
  527. if err := devices.closeTransaction(); err != nil {
  528. devices.unregisterDevice(deviceId, hash)
  529. devicemapper.DeleteDevice(devices.getPoolDevName(), deviceId)
  530. devices.markDeviceIdFree(deviceId)
  531. return err
  532. }
  533. return nil
  534. }
  535. func (devices *DeviceSet) loadMetadata(hash string) *DevInfo {
  536. info := &DevInfo{Hash: hash, devices: devices}
  537. jsonData, err := ioutil.ReadFile(devices.metadataFile(info))
  538. if err != nil {
  539. return nil
  540. }
  541. if err := json.Unmarshal(jsonData, &info); err != nil {
  542. return nil
  543. }
  544. return info
  545. }
  546. func (devices *DeviceSet) setupBaseImage() error {
  547. oldInfo, _ := devices.lookupDevice("")
  548. if oldInfo != nil && oldInfo.Initialized {
  549. return nil
  550. }
  551. if oldInfo != nil && !oldInfo.Initialized {
  552. log.Debugf("Removing uninitialized base image")
  553. if err := devices.DeleteDevice(""); err != nil {
  554. return err
  555. }
  556. }
  557. if devices.thinPoolDevice != "" && oldInfo == nil {
  558. _, transactionId, dataUsed, _, _, _, err := devices.poolStatus()
  559. if err != nil {
  560. return err
  561. }
  562. if dataUsed != 0 {
  563. return fmt.Errorf("Unable to take ownership of thin-pool (%s) that already has used data blocks",
  564. devices.thinPoolDevice)
  565. }
  566. if transactionId != 0 {
  567. return fmt.Errorf("Unable to take ownership of thin-pool (%s) with non-zero transaction Id",
  568. devices.thinPoolDevice)
  569. }
  570. }
  571. log.Debugf("Initializing base device-mapper thin volume")
  572. // Create initial device
  573. info, err := devices.createRegisterDevice("")
  574. if err != nil {
  575. return err
  576. }
  577. log.Debugf("Creating filesystem on base device-mapper thin volume")
  578. if err = devices.activateDeviceIfNeeded(info); err != nil {
  579. return err
  580. }
  581. if err := devices.createFilesystem(info); err != nil {
  582. return err
  583. }
  584. info.Initialized = true
  585. if err = devices.saveMetadata(info); err != nil {
  586. info.Initialized = false
  587. return err
  588. }
  589. return nil
  590. }
  591. func setCloseOnExec(name string) {
  592. if fileInfos, _ := ioutil.ReadDir("/proc/self/fd"); fileInfos != nil {
  593. for _, i := range fileInfos {
  594. link, _ := os.Readlink(filepath.Join("/proc/self/fd", i.Name()))
  595. if link == name {
  596. fd, err := strconv.Atoi(i.Name())
  597. if err == nil {
  598. syscall.CloseOnExec(fd)
  599. }
  600. }
  601. }
  602. }
  603. }
  604. func (devices *DeviceSet) DMLog(level int, file string, line int, dmError int, message string) {
  605. if level >= 7 {
  606. return // Ignore _LOG_DEBUG
  607. }
  608. // FIXME(vbatts) push this back into ./pkg/devicemapper/
  609. log.Debugf("libdevmapper(%d): %s:%d (%d) %s", level, file, line, dmError, message)
  610. }
  611. func major(device uint64) uint64 {
  612. return (device >> 8) & 0xfff
  613. }
  614. func minor(device uint64) uint64 {
  615. return (device & 0xff) | ((device >> 12) & 0xfff00)
  616. }
  617. func (devices *DeviceSet) ResizePool(size int64) error {
  618. dirname := devices.loopbackDir()
  619. datafilename := path.Join(dirname, "data")
  620. if len(devices.dataDevice) > 0 {
  621. datafilename = devices.dataDevice
  622. }
  623. metadatafilename := path.Join(dirname, "metadata")
  624. if len(devices.metadataDevice) > 0 {
  625. metadatafilename = devices.metadataDevice
  626. }
  627. datafile, err := os.OpenFile(datafilename, os.O_RDWR, 0)
  628. if datafile == nil {
  629. return err
  630. }
  631. defer datafile.Close()
  632. fi, err := datafile.Stat()
  633. if fi == nil {
  634. return err
  635. }
  636. if fi.Size() > size {
  637. return fmt.Errorf("Can't shrink file")
  638. }
  639. dataloopback := devicemapper.FindLoopDeviceFor(datafile)
  640. if dataloopback == nil {
  641. return fmt.Errorf("Unable to find loopback mount for: %s", datafilename)
  642. }
  643. defer dataloopback.Close()
  644. metadatafile, err := os.OpenFile(metadatafilename, os.O_RDWR, 0)
  645. if metadatafile == nil {
  646. return err
  647. }
  648. defer metadatafile.Close()
  649. metadataloopback := devicemapper.FindLoopDeviceFor(metadatafile)
  650. if metadataloopback == nil {
  651. return fmt.Errorf("Unable to find loopback mount for: %s", metadatafilename)
  652. }
  653. defer metadataloopback.Close()
  654. // Grow loopback file
  655. if err := datafile.Truncate(size); err != nil {
  656. return fmt.Errorf("Unable to grow loopback file: %s", err)
  657. }
  658. // Reload size for loopback device
  659. if err := devicemapper.LoopbackSetCapacity(dataloopback); err != nil {
  660. return fmt.Errorf("Unable to update loopback capacity: %s", err)
  661. }
  662. // Suspend the pool
  663. if err := devicemapper.SuspendDevice(devices.getPoolName()); err != nil {
  664. return fmt.Errorf("Unable to suspend pool: %s", err)
  665. }
  666. // Reload with the new block sizes
  667. if err := devicemapper.ReloadPool(devices.getPoolName(), dataloopback, metadataloopback, devices.thinpBlockSize); err != nil {
  668. return fmt.Errorf("Unable to reload pool: %s", err)
  669. }
  670. // Resume the pool
  671. if err := devicemapper.ResumeDevice(devices.getPoolName()); err != nil {
  672. return fmt.Errorf("Unable to resume pool: %s", err)
  673. }
  674. return nil
  675. }
  676. func (devices *DeviceSet) loadTransactionMetaData() error {
  677. jsonData, err := ioutil.ReadFile(devices.transactionMetaFile())
  678. if err != nil {
  679. // There is no active transaction. This will be the case
  680. // during upgrade.
  681. if os.IsNotExist(err) {
  682. devices.OpenTransactionId = devices.TransactionId
  683. return nil
  684. }
  685. return err
  686. }
  687. json.Unmarshal(jsonData, &devices.Transaction)
  688. return nil
  689. }
  690. func (devices *DeviceSet) saveTransactionMetaData() error {
  691. jsonData, err := json.Marshal(&devices.Transaction)
  692. if err != nil {
  693. return fmt.Errorf("Error encoding metadata to json: %s", err)
  694. }
  695. return devices.writeMetaFile(jsonData, devices.transactionMetaFile())
  696. }
  697. func (devices *DeviceSet) removeTransactionMetaData() error {
  698. if err := os.RemoveAll(devices.transactionMetaFile()); err != nil {
  699. return err
  700. }
  701. return nil
  702. }
  703. func (devices *DeviceSet) rollbackTransaction() error {
  704. log.Debugf("Rolling back open transaction: TransactionId=%d hash=%s device_id=%d", devices.OpenTransactionId, devices.DeviceIdHash, devices.DeviceId)
  705. // A device id might have already been deleted before transaction
  706. // closed. In that case this call will fail. Just leave a message
  707. // in case of failure.
  708. if err := devicemapper.DeleteDevice(devices.getPoolDevName(), devices.DeviceId); err != nil {
  709. log.Errorf("Warning: Unable to delete device: %s", err)
  710. }
  711. dinfo := &DevInfo{Hash: devices.DeviceIdHash}
  712. if err := devices.removeMetadata(dinfo); err != nil {
  713. log.Errorf("Warning: Unable to remove meta data: %s", err)
  714. } else {
  715. devices.markDeviceIdFree(devices.DeviceId)
  716. }
  717. if err := devices.removeTransactionMetaData(); err != nil {
  718. log.Errorf("Warning: Unable to remove transaction meta file %s: %s", devices.transactionMetaFile(), err)
  719. }
  720. return nil
  721. }
  722. func (devices *DeviceSet) processPendingTransaction() error {
  723. if err := devices.loadTransactionMetaData(); err != nil {
  724. return err
  725. }
  726. // If there was open transaction but pool transaction Id is same
  727. // as open transaction Id, nothing to roll back.
  728. if devices.TransactionId == devices.OpenTransactionId {
  729. return nil
  730. }
  731. // If open transaction Id is less than pool transaction Id, something
  732. // is wrong. Bail out.
  733. if devices.OpenTransactionId < devices.TransactionId {
  734. log.Errorf("Warning: Open Transaction id %d is less than pool transaction id %d", devices.OpenTransactionId, devices.TransactionId)
  735. return nil
  736. }
  737. // Pool transaction Id is not same as open transaction. There is
  738. // a transaction which was not completed.
  739. if err := devices.rollbackTransaction(); err != nil {
  740. return fmt.Errorf("Rolling back open transaction failed: %s", err)
  741. }
  742. devices.OpenTransactionId = devices.TransactionId
  743. return nil
  744. }
  745. func (devices *DeviceSet) loadDeviceSetMetaData() error {
  746. jsonData, err := ioutil.ReadFile(devices.deviceSetMetaFile())
  747. if err != nil {
  748. // For backward compatibility return success if file does
  749. // not exist.
  750. if os.IsNotExist(err) {
  751. return nil
  752. }
  753. return err
  754. }
  755. return json.Unmarshal(jsonData, devices)
  756. }
  757. func (devices *DeviceSet) saveDeviceSetMetaData() error {
  758. jsonData, err := json.Marshal(devices)
  759. if err != nil {
  760. return fmt.Errorf("Error encoding metadata to json: %s", err)
  761. }
  762. return devices.writeMetaFile(jsonData, devices.deviceSetMetaFile())
  763. }
  764. func (devices *DeviceSet) openTransaction(hash string, DeviceId int) error {
  765. devices.allocateTransactionId()
  766. devices.DeviceIdHash = hash
  767. devices.DeviceId = DeviceId
  768. if err := devices.saveTransactionMetaData(); err != nil {
  769. return fmt.Errorf("Error saving transaction meta data: %s", err)
  770. }
  771. return nil
  772. }
  773. func (devices *DeviceSet) refreshTransaction(DeviceId int) error {
  774. devices.DeviceId = DeviceId
  775. if err := devices.saveTransactionMetaData(); err != nil {
  776. return fmt.Errorf("Error saving transaction meta data: %s", err)
  777. }
  778. return nil
  779. }
  780. func (devices *DeviceSet) closeTransaction() error {
  781. if err := devices.updatePoolTransactionId(); err != nil {
  782. log.Debugf("Failed to close Transaction")
  783. return err
  784. }
  785. return nil
  786. }
  787. func (devices *DeviceSet) initDevmapper(doInit bool) error {
  788. // give ourselves to libdm as a log handler
  789. devicemapper.LogInit(devices)
  790. _, err := devicemapper.GetDriverVersion()
  791. if err != nil {
  792. // Can't even get driver version, assume not supported
  793. return graphdriver.ErrNotSupported
  794. }
  795. if err := os.MkdirAll(devices.metadataDir(), 0700); err != nil && !os.IsExist(err) {
  796. return err
  797. }
  798. // Set the device prefix from the device id and inode of the docker root dir
  799. st, err := os.Stat(devices.root)
  800. if err != nil {
  801. return fmt.Errorf("Error looking up dir %s: %s", devices.root, err)
  802. }
  803. sysSt := st.Sys().(*syscall.Stat_t)
  804. // "reg-" stands for "regular file".
  805. // In the future we might use "dev-" for "device file", etc.
  806. // docker-maj,min[-inode] stands for:
  807. // - Managed by docker
  808. // - The target of this device is at major <maj> and minor <min>
  809. // - If <inode> is defined, use that file inside the device as a loopback image. Otherwise use the device itself.
  810. devices.devicePrefix = fmt.Sprintf("docker-%d:%d-%d", major(sysSt.Dev), minor(sysSt.Dev), sysSt.Ino)
  811. log.Debugf("Generated prefix: %s", devices.devicePrefix)
  812. // Check for the existence of the thin-pool device
  813. log.Debugf("Checking for existence of the pool '%s'", devices.getPoolName())
  814. info, err := devicemapper.GetInfo(devices.getPoolName())
  815. if info == nil {
  816. log.Debugf("Error device devicemapper.GetInfo: %s", err)
  817. return err
  818. }
  819. // It seems libdevmapper opens this without O_CLOEXEC, and go exec will not close files
  820. // that are not Close-on-exec, and lxc-start will die if it inherits any unexpected files,
  821. // so we add this badhack to make sure it closes itself
  822. setCloseOnExec("/dev/mapper/control")
  823. // Make sure the sparse images exist in <root>/devicemapper/data and
  824. // <root>/devicemapper/metadata
  825. createdLoopback := false
  826. // If the pool doesn't exist, create it
  827. if info.Exists == 0 && devices.thinPoolDevice == "" {
  828. log.Debugf("Pool doesn't exist. Creating it.")
  829. var (
  830. dataFile *os.File
  831. metadataFile *os.File
  832. )
  833. if devices.dataDevice == "" {
  834. // Make sure the sparse images exist in <root>/devicemapper/data
  835. hasData := devices.hasImage("data")
  836. if !doInit && !hasData {
  837. return errors.New("Loopback data file not found")
  838. }
  839. if !hasData {
  840. createdLoopback = true
  841. }
  842. data, err := devices.ensureImage("data", devices.dataLoopbackSize)
  843. if err != nil {
  844. log.Debugf("Error device ensureImage (data): %s", err)
  845. return err
  846. }
  847. dataFile, err = devicemapper.AttachLoopDevice(data)
  848. if err != nil {
  849. return err
  850. }
  851. devices.dataLoopFile = data
  852. devices.dataDevice = dataFile.Name()
  853. } else {
  854. dataFile, err = os.OpenFile(devices.dataDevice, os.O_RDWR, 0600)
  855. if err != nil {
  856. return err
  857. }
  858. }
  859. defer dataFile.Close()
  860. if devices.metadataDevice == "" {
  861. // Make sure the sparse images exist in <root>/devicemapper/metadata
  862. hasMetadata := devices.hasImage("metadata")
  863. if !doInit && !hasMetadata {
  864. return errors.New("Loopback metadata file not found")
  865. }
  866. if !hasMetadata {
  867. createdLoopback = true
  868. }
  869. metadata, err := devices.ensureImage("metadata", devices.metaDataLoopbackSize)
  870. if err != nil {
  871. log.Debugf("Error device ensureImage (metadata): %s", err)
  872. return err
  873. }
  874. metadataFile, err = devicemapper.AttachLoopDevice(metadata)
  875. if err != nil {
  876. return err
  877. }
  878. devices.metadataLoopFile = metadata
  879. devices.metadataDevice = metadataFile.Name()
  880. } else {
  881. metadataFile, err = os.OpenFile(devices.metadataDevice, os.O_RDWR, 0600)
  882. if err != nil {
  883. return err
  884. }
  885. }
  886. defer metadataFile.Close()
  887. if err := devicemapper.CreatePool(devices.getPoolName(), dataFile, metadataFile, devices.thinpBlockSize); err != nil {
  888. return err
  889. }
  890. }
  891. // If we didn't just create the data or metadata image, we need to
  892. // load the transaction id and migrate old metadata
  893. if !createdLoopback {
  894. if err = devices.initMetaData(); err != nil {
  895. return err
  896. }
  897. }
  898. // Right now this loads only NextDeviceId. If there is more metatadata
  899. // down the line, we might have to move it earlier.
  900. if err = devices.loadDeviceSetMetaData(); err != nil {
  901. return err
  902. }
  903. // Setup the base image
  904. if doInit {
  905. if err := devices.setupBaseImage(); err != nil {
  906. log.Debugf("Error device setupBaseImage: %s", err)
  907. return err
  908. }
  909. }
  910. return nil
  911. }
  912. func (devices *DeviceSet) AddDevice(hash, baseHash string) error {
  913. log.Debugf("[deviceset] AddDevice() hash=%s basehash=%s", hash, baseHash)
  914. defer log.Debugf("[deviceset] AddDevice END")
  915. baseInfo, err := devices.lookupDevice(baseHash)
  916. if err != nil {
  917. return err
  918. }
  919. baseInfo.lock.Lock()
  920. defer baseInfo.lock.Unlock()
  921. devices.Lock()
  922. defer devices.Unlock()
  923. if info, _ := devices.lookupDevice(hash); info != nil {
  924. return fmt.Errorf("device %s already exists", hash)
  925. }
  926. if err := devices.createRegisterSnapDevice(hash, baseInfo); err != nil {
  927. return err
  928. }
  929. return nil
  930. }
  931. func (devices *DeviceSet) deleteDevice(info *DevInfo) error {
  932. if devices.doBlkDiscard {
  933. // This is a workaround for the kernel not discarding block so
  934. // on the thin pool when we remove a thinp device, so we do it
  935. // manually
  936. if err := devices.activateDeviceIfNeeded(info); err == nil {
  937. if err := devicemapper.BlockDeviceDiscard(info.DevName()); err != nil {
  938. log.Debugf("Error discarding block on device: %s (ignoring)", err)
  939. }
  940. }
  941. }
  942. devinfo, _ := devicemapper.GetInfo(info.Name())
  943. if devinfo != nil && devinfo.Exists != 0 {
  944. if err := devices.removeDeviceAndWait(info.Name()); err != nil {
  945. log.Debugf("Error removing device: %s", err)
  946. return err
  947. }
  948. }
  949. if err := devices.openTransaction(info.Hash, info.DeviceId); err != nil {
  950. log.Debugf("Error opening transaction hash = %s deviceId = %d", "", info.DeviceId)
  951. return err
  952. }
  953. if err := devicemapper.DeleteDevice(devices.getPoolDevName(), info.DeviceId); err != nil {
  954. log.Debugf("Error deleting device: %s", err)
  955. return err
  956. }
  957. if err := devices.unregisterDevice(info.DeviceId, info.Hash); err != nil {
  958. return err
  959. }
  960. if err := devices.closeTransaction(); err != nil {
  961. return err
  962. }
  963. devices.markDeviceIdFree(info.DeviceId)
  964. return nil
  965. }
  966. func (devices *DeviceSet) DeleteDevice(hash string) error {
  967. info, err := devices.lookupDevice(hash)
  968. if err != nil {
  969. return err
  970. }
  971. info.lock.Lock()
  972. defer info.lock.Unlock()
  973. devices.Lock()
  974. defer devices.Unlock()
  975. return devices.deleteDevice(info)
  976. }
  977. func (devices *DeviceSet) deactivatePool() error {
  978. log.Debugf("[devmapper] deactivatePool()")
  979. defer log.Debugf("[devmapper] deactivatePool END")
  980. devname := devices.getPoolDevName()
  981. devinfo, err := devicemapper.GetInfo(devname)
  982. if err != nil {
  983. return err
  984. }
  985. if d, err := devicemapper.GetDeps(devname); err == nil {
  986. // Access to more Debug output
  987. log.Debugf("[devmapper] devicemapper.GetDeps() %s: %#v", devname, d)
  988. }
  989. if devinfo.Exists != 0 {
  990. return devicemapper.RemoveDevice(devname)
  991. }
  992. return nil
  993. }
  994. func (devices *DeviceSet) deactivateDevice(info *DevInfo) error {
  995. log.Debugf("[devmapper] deactivateDevice(%s)", info.Hash)
  996. defer log.Debugf("[devmapper] deactivateDevice END")
  997. // Wait for the unmount to be effective,
  998. // by watching the value of Info.OpenCount for the device
  999. if err := devices.waitClose(info); err != nil {
  1000. log.Errorf("Warning: error waiting for device %s to close: %s", info.Hash, err)
  1001. }
  1002. devinfo, err := devicemapper.GetInfo(info.Name())
  1003. if err != nil {
  1004. return err
  1005. }
  1006. if devinfo.Exists != 0 {
  1007. if err := devices.removeDeviceAndWait(info.Name()); err != nil {
  1008. return err
  1009. }
  1010. }
  1011. return nil
  1012. }
  1013. // Issues the underlying dm remove operation and then waits
  1014. // for it to finish.
  1015. func (devices *DeviceSet) removeDeviceAndWait(devname string) error {
  1016. var err error
  1017. for i := 0; i < 1000; i++ {
  1018. err = devicemapper.RemoveDevice(devname)
  1019. if err == nil {
  1020. break
  1021. }
  1022. if err != devicemapper.ErrBusy {
  1023. return err
  1024. }
  1025. // If we see EBUSY it may be a transient error,
  1026. // sleep a bit a retry a few times.
  1027. devices.Unlock()
  1028. time.Sleep(10 * time.Millisecond)
  1029. devices.Lock()
  1030. }
  1031. if err != nil {
  1032. return err
  1033. }
  1034. if err := devices.waitRemove(devname); err != nil {
  1035. return err
  1036. }
  1037. return nil
  1038. }
  1039. // waitRemove blocks until either:
  1040. // a) the device registered at <device_set_prefix>-<hash> is removed,
  1041. // or b) the 10 second timeout expires.
  1042. func (devices *DeviceSet) waitRemove(devname string) error {
  1043. log.Debugf("[deviceset %s] waitRemove(%s)", devices.devicePrefix, devname)
  1044. defer log.Debugf("[deviceset %s] waitRemove(%s) END", devices.devicePrefix, devname)
  1045. i := 0
  1046. for ; i < 1000; i++ {
  1047. devinfo, err := devicemapper.GetInfo(devname)
  1048. if err != nil {
  1049. // If there is an error we assume the device doesn't exist.
  1050. // The error might actually be something else, but we can't differentiate.
  1051. return nil
  1052. }
  1053. if i%100 == 0 {
  1054. log.Debugf("Waiting for removal of %s: exists=%d", devname, devinfo.Exists)
  1055. }
  1056. if devinfo.Exists == 0 {
  1057. break
  1058. }
  1059. devices.Unlock()
  1060. time.Sleep(10 * time.Millisecond)
  1061. devices.Lock()
  1062. }
  1063. if i == 1000 {
  1064. return fmt.Errorf("Timeout while waiting for device %s to be removed", devname)
  1065. }
  1066. return nil
  1067. }
  1068. // waitClose blocks until either:
  1069. // a) the device registered at <device_set_prefix>-<hash> is closed,
  1070. // or b) the 10 second timeout expires.
  1071. func (devices *DeviceSet) waitClose(info *DevInfo) error {
  1072. i := 0
  1073. for ; i < 1000; i++ {
  1074. devinfo, err := devicemapper.GetInfo(info.Name())
  1075. if err != nil {
  1076. return err
  1077. }
  1078. if i%100 == 0 {
  1079. log.Debugf("Waiting for unmount of %s: opencount=%d", info.Hash, devinfo.OpenCount)
  1080. }
  1081. if devinfo.OpenCount == 0 {
  1082. break
  1083. }
  1084. devices.Unlock()
  1085. time.Sleep(10 * time.Millisecond)
  1086. devices.Lock()
  1087. }
  1088. if i == 1000 {
  1089. return fmt.Errorf("Timeout while waiting for device %s to close", info.Hash)
  1090. }
  1091. return nil
  1092. }
  1093. func (devices *DeviceSet) Shutdown() error {
  1094. log.Debugf("[deviceset %s] shutdown()", devices.devicePrefix)
  1095. log.Debugf("[devmapper] Shutting down DeviceSet: %s", devices.root)
  1096. defer log.Debugf("[deviceset %s] shutdown END", devices.devicePrefix)
  1097. var devs []*DevInfo
  1098. devices.devicesLock.Lock()
  1099. for _, info := range devices.Devices {
  1100. devs = append(devs, info)
  1101. }
  1102. devices.devicesLock.Unlock()
  1103. for _, info := range devs {
  1104. info.lock.Lock()
  1105. if info.mountCount > 0 {
  1106. // We use MNT_DETACH here in case it is still busy in some running
  1107. // container. This means it'll go away from the global scope directly,
  1108. // and the device will be released when that container dies.
  1109. if err := syscall.Unmount(info.mountPath, syscall.MNT_DETACH); err != nil {
  1110. log.Debugf("Shutdown unmounting %s, error: %s", info.mountPath, err)
  1111. }
  1112. devices.Lock()
  1113. if err := devices.deactivateDevice(info); err != nil {
  1114. log.Debugf("Shutdown deactivate %s , error: %s", info.Hash, err)
  1115. }
  1116. devices.Unlock()
  1117. }
  1118. info.lock.Unlock()
  1119. }
  1120. info, _ := devices.lookupDevice("")
  1121. if info != nil {
  1122. info.lock.Lock()
  1123. devices.Lock()
  1124. if err := devices.deactivateDevice(info); err != nil {
  1125. log.Debugf("Shutdown deactivate base , error: %s", err)
  1126. }
  1127. devices.Unlock()
  1128. info.lock.Unlock()
  1129. }
  1130. devices.Lock()
  1131. if devices.thinPoolDevice == "" {
  1132. if err := devices.deactivatePool(); err != nil {
  1133. log.Debugf("Shutdown deactivate pool , error: %s", err)
  1134. }
  1135. }
  1136. devices.saveDeviceSetMetaData()
  1137. devices.Unlock()
  1138. return nil
  1139. }
  1140. func (devices *DeviceSet) MountDevice(hash, path, mountLabel string) error {
  1141. info, err := devices.lookupDevice(hash)
  1142. if err != nil {
  1143. return err
  1144. }
  1145. info.lock.Lock()
  1146. defer info.lock.Unlock()
  1147. devices.Lock()
  1148. defer devices.Unlock()
  1149. if info.mountCount > 0 {
  1150. if path != info.mountPath {
  1151. return fmt.Errorf("Trying to mount devmapper device in multple places (%s, %s)", info.mountPath, path)
  1152. }
  1153. info.mountCount++
  1154. return nil
  1155. }
  1156. if err := devices.activateDeviceIfNeeded(info); err != nil {
  1157. return fmt.Errorf("Error activating devmapper device for '%s': %s", hash, err)
  1158. }
  1159. var flags uintptr = syscall.MS_MGC_VAL
  1160. fstype, err := ProbeFsType(info.DevName())
  1161. if err != nil {
  1162. return err
  1163. }
  1164. options := ""
  1165. if fstype == "xfs" {
  1166. // XFS needs nouuid or it can't mount filesystems with the same fs
  1167. options = joinMountOptions(options, "nouuid")
  1168. }
  1169. options = joinMountOptions(options, devices.mountOptions)
  1170. options = joinMountOptions(options, label.FormatMountLabel("", mountLabel))
  1171. err = syscall.Mount(info.DevName(), path, fstype, flags, joinMountOptions("discard", options))
  1172. if err != nil && err == syscall.EINVAL {
  1173. err = syscall.Mount(info.DevName(), path, fstype, flags, options)
  1174. }
  1175. if err != nil {
  1176. return fmt.Errorf("Error mounting '%s' on '%s': %s", info.DevName(), path, err)
  1177. }
  1178. info.mountCount = 1
  1179. info.mountPath = path
  1180. return nil
  1181. }
  1182. func (devices *DeviceSet) UnmountDevice(hash string) error {
  1183. log.Debugf("[devmapper] UnmountDevice(hash=%s)", hash)
  1184. defer log.Debugf("[devmapper] UnmountDevice END")
  1185. info, err := devices.lookupDevice(hash)
  1186. if err != nil {
  1187. return err
  1188. }
  1189. info.lock.Lock()
  1190. defer info.lock.Unlock()
  1191. devices.Lock()
  1192. defer devices.Unlock()
  1193. if info.mountCount == 0 {
  1194. return fmt.Errorf("UnmountDevice: device not-mounted id %s", hash)
  1195. }
  1196. info.mountCount--
  1197. if info.mountCount > 0 {
  1198. return nil
  1199. }
  1200. log.Debugf("[devmapper] Unmount(%s)", info.mountPath)
  1201. if err := syscall.Unmount(info.mountPath, syscall.MNT_DETACH); err != nil {
  1202. return err
  1203. }
  1204. log.Debugf("[devmapper] Unmount done")
  1205. if err := devices.deactivateDevice(info); err != nil {
  1206. return err
  1207. }
  1208. info.mountPath = ""
  1209. return nil
  1210. }
  1211. func (devices *DeviceSet) HasDevice(hash string) bool {
  1212. devices.Lock()
  1213. defer devices.Unlock()
  1214. info, _ := devices.lookupDevice(hash)
  1215. return info != nil
  1216. }
  1217. func (devices *DeviceSet) HasActivatedDevice(hash string) bool {
  1218. info, _ := devices.lookupDevice(hash)
  1219. if info == nil {
  1220. return false
  1221. }
  1222. info.lock.Lock()
  1223. defer info.lock.Unlock()
  1224. devices.Lock()
  1225. defer devices.Unlock()
  1226. devinfo, _ := devicemapper.GetInfo(info.Name())
  1227. return devinfo != nil && devinfo.Exists != 0
  1228. }
  1229. func (devices *DeviceSet) List() []string {
  1230. devices.Lock()
  1231. defer devices.Unlock()
  1232. devices.devicesLock.Lock()
  1233. ids := make([]string, len(devices.Devices))
  1234. i := 0
  1235. for k := range devices.Devices {
  1236. ids[i] = k
  1237. i++
  1238. }
  1239. devices.devicesLock.Unlock()
  1240. return ids
  1241. }
  1242. func (devices *DeviceSet) deviceStatus(devName string) (sizeInSectors, mappedSectors, highestMappedSector uint64, err error) {
  1243. var params string
  1244. _, sizeInSectors, _, params, err = devicemapper.GetStatus(devName)
  1245. if err != nil {
  1246. return
  1247. }
  1248. if _, err = fmt.Sscanf(params, "%d %d", &mappedSectors, &highestMappedSector); err == nil {
  1249. return
  1250. }
  1251. return
  1252. }
  1253. func (devices *DeviceSet) GetDeviceStatus(hash string) (*DevStatus, error) {
  1254. info, err := devices.lookupDevice(hash)
  1255. if err != nil {
  1256. return nil, err
  1257. }
  1258. info.lock.Lock()
  1259. defer info.lock.Unlock()
  1260. devices.Lock()
  1261. defer devices.Unlock()
  1262. status := &DevStatus{
  1263. DeviceId: info.DeviceId,
  1264. Size: info.Size,
  1265. TransactionId: info.TransactionId,
  1266. }
  1267. if err := devices.activateDeviceIfNeeded(info); err != nil {
  1268. return nil, fmt.Errorf("Error activating devmapper device for '%s': %s", hash, err)
  1269. }
  1270. if sizeInSectors, mappedSectors, highestMappedSector, err := devices.deviceStatus(info.DevName()); err != nil {
  1271. return nil, err
  1272. } else {
  1273. status.SizeInSectors = sizeInSectors
  1274. status.MappedSectors = mappedSectors
  1275. status.HighestMappedSector = highestMappedSector
  1276. }
  1277. return status, nil
  1278. }
  1279. func (devices *DeviceSet) poolStatus() (totalSizeInSectors, transactionId, dataUsed, dataTotal, metadataUsed, metadataTotal uint64, err error) {
  1280. var params string
  1281. if _, totalSizeInSectors, _, params, err = devicemapper.GetStatus(devices.getPoolName()); err == nil {
  1282. _, err = fmt.Sscanf(params, "%d %d/%d %d/%d", &transactionId, &metadataUsed, &metadataTotal, &dataUsed, &dataTotal)
  1283. }
  1284. return
  1285. }
  1286. // MetadataDevicePath returns the path to the metadata storage for this deviceset,
  1287. // regardless of loopback or block device
  1288. func (devices *DeviceSet) DataDevicePath() string {
  1289. return devices.dataDevice
  1290. }
  1291. // MetadataDevicePath returns the path to the metadata storage for this deviceset,
  1292. // regardless of loopback or block device
  1293. func (devices *DeviceSet) MetadataDevicePath() string {
  1294. return devices.metadataDevice
  1295. }
  1296. // Status returns the current status of this deviceset
  1297. func (devices *DeviceSet) Status() *Status {
  1298. devices.Lock()
  1299. defer devices.Unlock()
  1300. status := &Status{}
  1301. status.PoolName = devices.getPoolName()
  1302. status.DataFile = devices.DataDevicePath()
  1303. status.DataLoopback = devices.dataLoopFile
  1304. status.MetadataFile = devices.MetadataDevicePath()
  1305. status.MetadataLoopback = devices.metadataLoopFile
  1306. totalSizeInSectors, _, dataUsed, dataTotal, metadataUsed, metadataTotal, err := devices.poolStatus()
  1307. if err == nil {
  1308. // Convert from blocks to bytes
  1309. blockSizeInSectors := totalSizeInSectors / dataTotal
  1310. status.Data.Used = dataUsed * blockSizeInSectors * 512
  1311. status.Data.Total = dataTotal * blockSizeInSectors * 512
  1312. // metadata blocks are always 4k
  1313. status.Metadata.Used = metadataUsed * 4096
  1314. status.Metadata.Total = metadataTotal * 4096
  1315. status.SectorSize = blockSizeInSectors * 512
  1316. }
  1317. return status
  1318. }
  1319. func NewDeviceSet(root string, doInit bool, options []string) (*DeviceSet, error) {
  1320. devicemapper.SetDevDir("/dev")
  1321. devices := &DeviceSet{
  1322. root: root,
  1323. MetaData: MetaData{Devices: make(map[string]*DevInfo)},
  1324. dataLoopbackSize: DefaultDataLoopbackSize,
  1325. metaDataLoopbackSize: DefaultMetaDataLoopbackSize,
  1326. baseFsSize: DefaultBaseFsSize,
  1327. filesystem: "ext4",
  1328. doBlkDiscard: true,
  1329. thinpBlockSize: DefaultThinpBlockSize,
  1330. deviceIdMap: make([]byte, DeviceIdMapSz),
  1331. }
  1332. foundBlkDiscard := false
  1333. for _, option := range options {
  1334. key, val, err := parsers.ParseKeyValueOpt(option)
  1335. if err != nil {
  1336. return nil, err
  1337. }
  1338. key = strings.ToLower(key)
  1339. switch key {
  1340. case "dm.basesize":
  1341. size, err := units.RAMInBytes(val)
  1342. if err != nil {
  1343. return nil, err
  1344. }
  1345. devices.baseFsSize = uint64(size)
  1346. case "dm.loopdatasize":
  1347. size, err := units.RAMInBytes(val)
  1348. if err != nil {
  1349. return nil, err
  1350. }
  1351. devices.dataLoopbackSize = size
  1352. case "dm.loopmetadatasize":
  1353. size, err := units.RAMInBytes(val)
  1354. if err != nil {
  1355. return nil, err
  1356. }
  1357. devices.metaDataLoopbackSize = size
  1358. case "dm.fs":
  1359. if val != "ext4" && val != "xfs" {
  1360. return nil, fmt.Errorf("Unsupported filesystem %s\n", val)
  1361. }
  1362. devices.filesystem = val
  1363. case "dm.mkfsarg":
  1364. devices.mkfsArgs = append(devices.mkfsArgs, val)
  1365. case "dm.mountopt":
  1366. devices.mountOptions = joinMountOptions(devices.mountOptions, val)
  1367. case "dm.metadatadev":
  1368. devices.metadataDevice = val
  1369. case "dm.datadev":
  1370. devices.dataDevice = val
  1371. case "dm.thinpooldev":
  1372. devices.thinPoolDevice = strings.TrimPrefix(val, "/dev/mapper/")
  1373. case "dm.blkdiscard":
  1374. foundBlkDiscard = true
  1375. devices.doBlkDiscard, err = strconv.ParseBool(val)
  1376. if err != nil {
  1377. return nil, err
  1378. }
  1379. case "dm.blocksize":
  1380. size, err := units.RAMInBytes(val)
  1381. if err != nil {
  1382. return nil, err
  1383. }
  1384. // convert to 512b sectors
  1385. devices.thinpBlockSize = uint32(size) >> 9
  1386. default:
  1387. return nil, fmt.Errorf("Unknown option %s\n", key)
  1388. }
  1389. }
  1390. // By default, don't do blk discard hack on raw devices, its rarely useful and is expensive
  1391. if !foundBlkDiscard && (devices.dataDevice != "" || devices.thinPoolDevice != "") {
  1392. devices.doBlkDiscard = false
  1393. }
  1394. if err := devices.initDevmapper(doInit); err != nil {
  1395. return nil, err
  1396. }
  1397. return devices, nil
  1398. }