ExternalVendorCode/balena-cli
1
0
Fork 0
mirror of https://github.com/balena-io/balena-cli.git synced 2024-12-24 07:46:39 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Erase MBR before rescanning

Browse Source
This commit is contained in:
Juan Cruz Viotti 2015-01-28 14:51:11 -04:00
parent 8b0ce36955
commit 320981770e
1 changed files with 29 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
lib/drive/drive.coffee
View File

@ -13,6 +13,32 @@ exports.rescanDrives = (callback) ->
console.log("DISKPART RESULT: #{arguments}") console.log("DISKPART RESULT: #{arguments}")
return callback(error) return callback(error)
exports.eraseMBR = (devicePath, callback) ->
return callback() if not IS_WINDOWS
bufferSize = 512
async.waterfall([
(callback) ->
fs.open(devicePath, 'rs+', null, callback)
(fd, callback) ->
buffer = new Buffer(bufferSize)
buffer.fill(0)
fs.write fd, buffer, 0, bufferSize, 0, (error, bytesWritten) ->
return callback(error) if error?
return callback(null, bytesWritten, fd)
(bytesWritten, fd, callback) ->
if bytesWritten isnt bufferSize
error = "Bytes written: #{bytesWritten}, expected #{bufferSize}"
return callback(error)
fs.close(fd, callback)
], callback)
exports.writeImage = (devicePath, imagePath, options = {}, callback = _.noop) -> exports.writeImage = (devicePath, imagePath, options = {}, callback = _.noop) ->
if not fs.existsSync(imagePath) if not fs.existsSync(imagePath)
@ -39,6 +65,9 @@ exports.writeImage = (devicePath, imagePath, options = {}, callback = _.noop) ->
async.waterfall([ async.waterfall([
(callback) ->
exports.eraseMBR(devicePath, callback)
(callback) -> (callback) ->
exports.rescanDrives(callback) exports.rescanDrives(callback)