deviceset.go 46 KB

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