Skip to content

Commit

Permalink
Close nodejs#2166 Close the fd in lchmod
Browse files Browse the repository at this point in the history
  • Loading branch information
isaacs authored and alexkwolfe committed Dec 23, 2011
1 parent 58124c3 commit dd99b83
Show file tree
Hide file tree
Showing 2 changed files with 86 additions and 3 deletions.
25 changes: 23 additions & 2 deletions lib/fs.js
Original file line number Diff line number Diff line change
Expand Up @@ -459,13 +459,34 @@ if (constants.hasOwnProperty('O_SYMLINK')) {
callback(err);
return;
}
fs.fchmod(fd, mode, callback);
// prefer to return the chmod error, if one occurs,
// but still try to close, and report closing errors if they occur.
fs.fchmod(fd, mode, function(err) {
fs.close(fd, function(err2) {
callback(err || err2);
});
});
});
};

fs.lchmodSync = function(path, mode) {
var fd = fs.openSync(path, constants.O_WRONLY | constants.O_SYMLINK);
return fs.fchmodSync(fd, mode);

// prefer to return the chmod error, if one occurs,
// but still try to close, and report closing errors if they occur.
var err, err2;
try {
var ret = fs.fchmodSync(fd, mode);
} catch (er) {
err = er;
}
try {
fs.closeSync(fd);
} catch (er) {
err2 = er;
}
if (err || err2) throw (err || err2);
return ret;
};
}

Expand Down
64 changes: 63 additions & 1 deletion test/simple/test-fs-chmod.js
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,40 @@ var mode_async;
var mode_sync;
var is_windows = process.platform === 'win32';

// Need to hijack fs.open/close to make sure that things
// get closed once they're opened.
fs._open = fs.open;
fs._openSync = fs.openSync;
fs.open = open;
fs.openSync = openSync;
fs._close = fs.close;
fs._closeSync = fs.closeSync;
fs.close = close;
fs.closeSync = closeSync;

var openCount = 0;

function open() {
openCount++;
return fs._open.apply(fs, arguments);
}

function openSync() {
openCount++;
return fs._openSync.apply(fs, arguments);
}

function close() {
openCount--;
return fs._close.apply(fs, arguments);
}

function closeSync() {
openCount--;
return fs._closeSync.apply(fs, arguments);
}


// On Windows chmod is only able to manipulate read-only bit
if (is_windows) {
mode_async = 0600; // read-write
Expand Down Expand Up @@ -87,12 +121,40 @@ fs.open(file, 'a', function(err, fd) {
assert.equal(mode_sync, fs.fstatSync(fd).mode & 0777);
}
success_count++;
fs.close(fd);
}
});
});

// lchmod
if (!is_windows) {
var link = path.join(common.tmpDir, 'symbolic-link');

try {
fs.unlinkSync(link);
} catch (er) {}
fs.symlinkSync(file, link);

fs.lchmod(link, mode_async, function(err) {
if (err) {
got_error = true;
} else {
console.log(fs.lstatSync(link).mode);
assert.equal(mode_async, fs.lstatSync(link).mode & 0777);

fs.lchmodSync(link, mode_sync);
assert.equal(mode_sync, fs.lstatSync(link).mode & 0777);
success_count++;
}
});
} else {
success_count++;
}


process.on('exit', function() {
assert.equal(2, success_count);
assert.equal(3, success_count);
assert.equal(0, openCount);
assert.equal(false, got_error);
});

0 comments on commit dd99b83

Please sign in to comment.