Skip to content

M
Marsaud Smb2
Commit 4b7d8773 authored by Julien Fontanet's avatar Julien Fontanet
Browse files
Options

chore: format with Prettier

parent 0d4574ab
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 21498 additions and 19725 deletions
.prettierrc 0 → 100644
View file @ 4b7d8773
{
"singleQuote": true,
"trailingComma": "es5"
}
lib/api/close.js
View file @ 4b7d8773
var SMB2Connection = require('../tools/smb2-connection'); var SMB2Connection = require('../tools/smb2-connection');
/* /*
* close * close
* ===== * =====
...@@ -13,8 +9,6 @@ var SMB2Connection = require('../tools/smb2-connection'); ...@@ -13,8 +9,6 @@ var SMB2Connection = require('../tools/smb2-connection');
* - close TCP connection * - close TCP connection
* *
*/ */
module.exports = function(){ module.exports = function() {
SMB2Connection.close(this); SMB2Connection.close(this);
} };
lib/api/exists.js
View file @ 4b7d8773
var SMB2Forge = require('../tools/smb2-forge'),
SMB2Request = SMB2Forge.request;
var SMB2Forge = require('../tools/smb2-forge')
, SMB2Request = SMB2Forge.request
;
/* /*
* exists * exists
...@@ -15,15 +12,14 @@ var SMB2Forge = require('../tools/smb2-forge') ...@@ -15,15 +12,14 @@ var SMB2Forge = require('../tools/smb2-forge')
* - close the file * - close the file
* *
*/ */
module.exports = function(path, cb){ module.exports = function(path, cb) {
var connection = this; var connection = this;
SMB2Request('open', {path:path}, connection, function(err, file){ SMB2Request('open', { path: path }, connection, function(err, file) {
if(err) cb && cb(null, false); if (err) cb && cb(null, false);
else SMB2Request('close', file, connection, function(err){ else
cb && cb(null, true); SMB2Request('close', file, connection, function(err) {
}); cb && cb(null, true);
});
}); });
};
}
lib/api/mkdir.js
View file @ 4b7d8773
var SMB2Forge = require('../tools/smb2-forge'),
SMB2Request = SMB2Forge.request;
var SMB2Forge = require('../tools/smb2-forge')
, SMB2Request = SMB2Forge.request
;
/* /*
* mkdir * mkdir
...@@ -15,37 +12,31 @@ var SMB2Forge = require('../tools/smb2-forge') ...@@ -15,37 +12,31 @@ var SMB2Forge = require('../tools/smb2-forge')
* - close the folder * - close the folder
* *
*/ */
module.exports = function(path, mode, cb){ module.exports = function(path, mode, cb) {
if (typeof mode == 'function') {
if(typeof mode == 'function'){
cb = mode; cb = mode;
mode = '0777'; mode = '0777';
} }
var connection = this; var connection = this;
connection.exists(path, function(err, exists){ connection.exists(path, function(err, exists) {
if (err) cb && cb(err);
if(err) cb && cb(err); else if (!exists) {
else if(!exists){
// SMB2 open file // SMB2 open file
SMB2Request('create_folder', {path:path}, connection, function(err, file){ SMB2Request('create_folder', { path: path }, connection, function(
if(err) cb && cb(err); err,
file
) {
if (err) cb && cb(err);
// SMB2 query directory // SMB2 query directory
else SMB2Request('close', file, connection, function(err){ else
cb && cb(null); SMB2Request('close', file, connection, function(err) {
}); cb && cb(null);
});
}); });
} else { } else {
cb(new Error('File/Folder already exists')); cb(new Error('File/Folder already exists'));
} }
}); });
};
}
lib/api/readdir.js
View file @ 4b7d8773
var SMB2Forge = require('../tools/smb2-forge'),
SMB2Request = SMB2Forge.request;
var SMB2Forge = require('../tools/smb2-forge')
, SMB2Request = SMB2Forge.request
;
/* /*
* readdir * readdir
...@@ -17,27 +14,31 @@ var SMB2Forge = require('../tools/smb2-forge') ...@@ -17,27 +14,31 @@ var SMB2Forge = require('../tools/smb2-forge')
* - close the directory * - close the directory
* *
*/ */
module.exports = function(path, cb){ module.exports = function(path, cb) {
var connection = this; var connection = this;
// SMB2 open directory // SMB2 open directory
SMB2Request('open', {path:path}, connection, function(err, file){ SMB2Request('open', { path: path }, connection, function(err, file) {
if(err) cb && cb(err); if (err) cb && cb(err);
// SMB2 query directory // SMB2 query directory
else SMB2Request('query_directory', file, connection, function(err, files){ else
if(err) cb && cb(err); SMB2Request('query_directory', file, connection, function(err, files) {
// SMB2 close directory if (err) cb && cb(err);
else SMB2Request('close', file, connection, function(err){ // SMB2 close directory
cb && cb( else
null SMB2Request('close', file, connection, function(err) {
, files cb &&
.map(function(v){ return v.Filename }) // get the filename only cb(
.filter(function(v){ return v!='.' && v!='..' }) // remove '.' and '..' values null,
); files
.map(function(v) {
return v.Filename;
}) // get the filename only
.filter(function(v) {
return v != '.' && v != '..';
}) // remove '.' and '..' values
);
});
}); });
});
}); });
};
}
lib/api/readfile.js
View file @ 4b7d8773
var SMB2Forge = require('../tools/smb2-forge'),
SMB2Request = SMB2Forge.request,
var SMB2Forge = require('../tools/smb2-forge') bigint = require('../tools/bigint');
, SMB2Request = SMB2Forge.request
, bigint = require('../tools/bigint')
;
/* /*
* readFile * readFile
...@@ -18,34 +15,33 @@ var SMB2Forge = require('../tools/smb2-forge') ...@@ -18,34 +15,33 @@ var SMB2Forge = require('../tools/smb2-forge')
* - close the file * - close the file
* *
*/ */
module.exports = function(filename, options, cb){ module.exports = function(filename, options, cb) {
var connection = this; var connection = this;
if(typeof options == 'function'){ if (typeof options == 'function') {
cb = options; cb = options;
options = {}; options = {};
} }
SMB2Request('open', {path:filename}, connection, function(err, file){ SMB2Request('open', { path: filename }, connection, function(err, file) {
if(err) cb && cb(err); if (err) cb && cb(err);
// SMB2 read file content // SMB2 read file content
else { else {
var fileLength = 0 var fileLength = 0,
, offset = new bigint(8) offset = new bigint(8),
, stop = false stop = false,
, nbRemainingPackets = 0 nbRemainingPackets = 0,
, maxPacketSize = 0x00010000 maxPacketSize = 0x00010000;
;
// get file length // get file length
for(var i=0;i<file.EndofFile.length;i++){ for (var i = 0; i < file.EndofFile.length; i++) {
fileLength += file.EndofFile[i] * Math.pow(2, i * 8); fileLength += file.EndofFile[i] * Math.pow(2, i * 8);
} }
var result = new Buffer(fileLength); var result = new Buffer(fileLength);
// callback manager // callback manager
function callback(offset){ function callback(offset) {
return function(err, content){ return function(err, content) {
if(stop) return; if (stop) return;
if(err) { if (err) {
cb && cb(err); cb && cb(err);
stop = true; stop = true;
} else { } else {
...@@ -53,33 +49,43 @@ module.exports = function(filename, options, cb){ ...@@ -53,33 +49,43 @@ module.exports = function(filename, options, cb){
nbRemainingPackets--; nbRemainingPackets--;
checkDone(); checkDone();
} }
} };
} }
// callback manager // callback manager
function checkDone(){ function checkDone() {
if(stop) return; if (stop) return;
createPackets(); createPackets();
if(nbRemainingPackets==0 && offset.ge(fileLength)) { if (nbRemainingPackets == 0 && offset.ge(fileLength)) {
SMB2Request('close', file, connection, function(err){ SMB2Request('close', file, connection, function(err) {
if(options.encoding){ if (options.encoding) {
result = result.toString(options.encoding); result = result.toString(options.encoding);
} }
cb && cb(null, result); cb && cb(null, result);
}) });
} }
} }
// create packets // create packets
function createPackets(){ function createPackets() {
while(nbRemainingPackets<connection.packetConcurrency && offset.lt(fileLength)){ while (
nbRemainingPackets < connection.packetConcurrency &&
offset.lt(fileLength)
) {
// process packet size // process packet size
var rest = offset.sub(fileLength).neg(); var rest = offset.sub(fileLength).neg();
var packetSize = rest.gt(maxPacketSize) ? maxPacketSize : rest.toNumber(); var packetSize = rest.gt(maxPacketSize)
? maxPacketSize
: rest.toNumber();
// generate buffer // generate buffer
SMB2Request('read', { SMB2Request(
'FileId':file.FileId 'read',
, 'Length':packetSize {
, 'Offset':offset.toBuffer() FileId: file.FileId,
}, connection, callback(offset)); Length: packetSize,
Offset: offset.toBuffer(),
},
connection,
callback(offset)
);
offset = offset.add(packetSize); offset = offset.add(packetSize);
nbRemainingPackets++; nbRemainingPackets++;
} }
...@@ -87,4 +93,4 @@ module.exports = function(filename, options, cb){ ...@@ -87,4 +93,4 @@ module.exports = function(filename, options, cb){
checkDone(); checkDone();
} }
}); });
} };
lib/api/rename.js
View file @ 4b7d8773
var bigint = require('../tools/bigint') var bigint = require('../tools/bigint');
var SMB2Request = require('../tools/smb2-forge').request var SMB2Request = require('../tools/smb2-forge').request;
var FILE_OPEN_IF = require('../structures/constants').FILE_OPEN_IF var FILE_OPEN_IF = require('../structures/constants').FILE_OPEN_IF;
/* /*
* rename * rename
...@@ -15,55 +15,66 @@ var FILE_OPEN_IF = require('../structures/constants').FILE_OPEN_IF ...@@ -15,55 +15,66 @@ var FILE_OPEN_IF = require('../structures/constants').FILE_OPEN_IF
* - close the file * - close the file
* *
*/ */
module.exports = function(oldPath, newPath, cb){ module.exports = function(oldPath, newPath, cb) {
var connection = this; var connection = this;
// SMB2 open the folder / file // SMB2 open the folder / file
SMB2Request('open_folder', {path:oldPath}, connection, function(err, file){ SMB2Request('open_folder', { path: oldPath }, connection, function(
if (err) SMB2Request('create', { path: oldPath, createDisposition: FILE_OPEN_IF }, connection, function (err, file) { err,
if(err) cb && cb(err); file
else rename(connection, file, newPath, cb); ) {
}); if (err)
SMB2Request(
'create',
{ path: oldPath, createDisposition: FILE_OPEN_IF },
connection,
function(err, file) {
if (err) cb && cb(err);
else rename(connection, file, newPath, cb);
}
);
else rename(connection, file, newPath, cb); else rename(connection, file, newPath, cb);
}); });
};
} function rename(connection, file, newPath, cb) {
function rename (connection, file, newPath, cb) {
// SMB2 rename // SMB2 rename
SMB2Request('set_info', {FileId:file.FileId, FileInfoClass:'FileRenameInformation',Buffer:renameBuffer(newPath)}, connection, function(err){ SMB2Request(
if(err) cb && cb(err); 'set_info',
// SMB2 close file {
else SMB2Request('close', file, connection, function(err){ FileId: file.FileId,
cb && cb(null); FileInfoClass: 'FileRenameInformation',
}); Buffer: renameBuffer(newPath),
}); },
connection,
function(err) {
if (err) cb && cb(err);
// SMB2 close file
else
SMB2Request('close', file, connection, function(err) {
cb && cb(null);
});
}
);
} }
function renameBuffer(newPath) {
function renameBuffer (newPath) {
var filename = new Buffer(newPath, 'ucs2'); var filename = new Buffer(newPath, 'ucs2');
return Buffer.concat([ return Buffer.concat([
// ReplaceIfExists 1 byte // ReplaceIfExists 1 byte
(new bigint(1,0)).toBuffer() new bigint(1, 0).toBuffer(),
// Reserved 7 bytes // Reserved 7 bytes
, (new bigint(7,0)).toBuffer() new bigint(7, 0).toBuffer(),
// RootDirectory 8 bytes // RootDirectory 8 bytes
, (new bigint(8,0)).toBuffer() new bigint(8, 0).toBuffer(),
// FileNameLength 4 bytes // FileNameLength 4 bytes
, (new bigint(4,filename.length)).toBuffer() new bigint(4, filename.length).toBuffer(),
// FileName // FileName
, filename filename,
]); ]);
} }
lib/api/rmdir.js
View file @ 4b7d8773
var SMB2Forge = require('../tools/smb2-forge'),
SMB2Request = SMB2Forge.request,
var SMB2Forge = require('../tools/smb2-forge') bigint = require('../tools/bigint');
, SMB2Request = SMB2Forge.request
, bigint = require('../tools/bigint')
;
/* /*
* rmdir * rmdir
...@@ -18,34 +15,40 @@ var SMB2Forge = require('../tools/smb2-forge') ...@@ -18,34 +15,40 @@ var SMB2Forge = require('../tools/smb2-forge')
* - close the folder * - close the folder
* *
*/ */
module.exports = function(path, cb){ module.exports = function(path, cb) {
var connection = this; var connection = this;
connection.exists(path, function(err, exists){ connection.exists(path, function(err, exists) {
if (err) cb && cb(err);
if(err) cb && cb(err); else if (exists) {
else if(exists){
// SMB2 open file // SMB2 open file
SMB2Request('open_folder', {path:path}, connection, function(err, file){ SMB2Request('open_folder', { path: path }, connection, function(
if(err) cb && cb(err); err,
file
) {
if (err) cb && cb(err);
// SMB2 query directory // SMB2 query directory
else SMB2Request('set_info', {FileId:file.FileId, FileInfoClass:'FileDispositionInformation',Buffer:(new bigint(1,1)).toBuffer()}, connection, function(err, files){ else
if(err) cb && cb(err); SMB2Request(
// SMB2 close directory 'set_info',
else SMB2Request('close', file, connection, function(err){ {
cb && cb(null, files); FileId: file.FileId,
}); FileInfoClass: 'FileDispositionInformation',
}); Buffer: new bigint(1, 1).toBuffer(),
},
connection,
function(err, files) {
if (err) cb && cb(err);
// SMB2 close directory
else
SMB2Request('close', file, connection, function(err) {
cb && cb(null, files);
});
}
);
}); });
} else { } else {
cb(new Error('Folder does not exists')); cb(new Error('Folder does not exists'));
} }
}); });
};
}
lib/api/src/createReadStream.js
View file @ 4b7d8773
import Bigint from '../tools/bigint' import Bigint from '../tools/bigint';
import Bluebird from 'bluebird' import Bluebird from 'bluebird';
import {Readable} from 'stream' import { Readable } from 'stream';
import {request} from '../tools/smb2-forge' import { request } from '../tools/smb2-forge';
const requestAsync = Bluebird.promisify(request) const requestAsync = Bluebird.promisify(request);
const maxPacketSize = 0x00010000 const maxPacketSize = 0x00010000;
class SmbReadableStream extends Readable { class SmbReadableStream extends Readable {
constructor (connection, file, options = {}) { constructor(connection, file, options = {}) {
super(options) super(options);
const { const { start = 0, end, encoding } = options;
start = 0,
end,
encoding
} = options
this.connection = connection this.connection = connection;
this.encoding = encoding this.encoding = encoding;
this.file = file this.file = file;
this.offset = new Bigint(8, start) this.offset = new Bigint(8, start);
let fileLength = 0 let fileLength = 0;
for (let i = 0; i < file.EndofFile.length; i++) { for (let i = 0; i < file.EndofFile.length; i++) {
fileLength += file.EndofFile[i] * Math.pow(2, i * 8) fileLength += file.EndofFile[i] * Math.pow(2, i * 8);
} }
this.fileLength = fileLength this.fileLength = fileLength;
this.wait = false this.wait = false;
if (end >= 0 && end < fileLength) { if (end >= 0 && end < fileLength) {
this.fileLength = end + 1 this.fileLength = end + 1;
} }
} }
async _read (size) { async _read(size) {
while (this.offset.lt(this.fileLength)/* && size > 0*/) { while (this.offset.lt(this.fileLength) /* && size > 0*/) {
if (this.wait) { if (this.wait) {
return return;
} }
const rest = this.offset.sub(this.fileLength).neg() const rest = this.offset.sub(this.fileLength).neg();
const packetSize = Math.min(maxPacketSize, rest.toNumber()/*, size*/) const packetSize = Math.min(maxPacketSize, rest.toNumber() /*, size*/);
const offset = new Bigint(this.offset) const offset = new Bigint(this.offset);
this.wait = true this.wait = true;
let content = await requestAsync('read', { let content = await requestAsync(
FileId: this.file.FileId, 'read',
Length: packetSize, {
Offset: offset.toBuffer() FileId: this.file.FileId,
}, this.connection) Length: packetSize,
this.wait = false Offset: offset.toBuffer(),
},
this.connection
);
this.wait = false;
if (this.encoding) { if (this.encoding) {
content = content.toString(this.encoding) content = content.toString(this.encoding);
} }
this.offset = this.offset.add(packetSize) this.offset = this.offset.add(packetSize);
// size -= packetSize // size -= packetSize
if (!this.push(content)) { if (!this.push(content)) {
return return;
} }
} }
if (this.offset.ge(this.fileLength)) { if (this.offset.ge(this.fileLength)) {
this.push(null) this.push(null);
await requestAsync('close', this.file, this.connection) await requestAsync('close', this.file, this.connection);
} }
} }
} }
export default function (path, options, cb) { export default function(path, options, cb) {
if (typeof options === 'function') { if (typeof options === 'function') {
cb = options cb = options;
options = {} options = {};
} }
request('open', {path}, this, (err, file) => { request('open', { path }, this, (err, file) => {
if (err) { if (err) {
if (err.code === 'STATUS_OBJECT_NAME_NOT_FOUND') { if (err.code === 'STATUS_OBJECT_NAME_NOT_FOUND') {
err.code = 'ENOENT' err.code = 'ENOENT';
} }
cb(err) cb(err);
} else { } else {
cb(null, new SmbReadableStream(this, file, options)) cb(null, new SmbReadableStream(this, file, options));
} }
}) });
} }
lib/api/src/createWriteStream.js
View file @ 4b7d8773
import Bigint from '../tools/bigint' import Bigint from '../tools/bigint';
import Bluebird from 'bluebird' import Bluebird from 'bluebird';
import {request} from '../tools/smb2-forge' import { request } from '../tools/smb2-forge';
import {Writable} from 'stream' import { Writable } from 'stream';
import { import {
FILE_OPEN, FILE_OPEN,
FILE_OPEN_IF, FILE_OPEN_IF,
FILE_OVERWRITE_IF, FILE_OVERWRITE_IF,
FILE_CREATE FILE_CREATE,
} from '../structures/constants' } from '../structures/constants';
const requestAsync = Bluebird.promisify(request) const requestAsync = Bluebird.promisify(request);
const maxPacketSize = new Bigint(8, 0x00010000 - 0x71) const maxPacketSize = new Bigint(8, 0x00010000 - 0x71);
function * fibonacci () { function* fibonacci() {
let a = 1 let a = 1;
let b = 2 let b = 2;
for (;;) { for (;;) {
const c = a const c = a;
a = b a = b;
b = c + a b = c + a;
yield c yield c;
} }
} }
class SmbWritableStream extends Writable { class SmbWritableStream extends Writable {
constructor (connection, file, options = {}) { constructor(connection, file, options = {}) {
super(options) super(options);
const { const { encoding = 'utf8', start = 0 } = options;
encoding = 'utf8',
start = 0 this.connection = connection;
} = options this.encoding = encoding;
this.file = file;
this.connection = connection this.offset = new Bigint(8, start);
this.encoding = encoding
this.file = file
this.offset = new Bigint(8, start)
} }
async _write (chunk, encoding, next) { async _write(chunk, encoding, next) {
encoding = this.encoding || encoding encoding = this.encoding || encoding;
chunk = Buffer.isBuffer(chunk) ? chunk : new Buffer(chunk, encoding) chunk = Buffer.isBuffer(chunk) ? chunk : new Buffer(chunk, encoding);
while (chunk.length > 0) { while (chunk.length > 0) {
const packetSize = Math.min(maxPacketSize.toNumber(), chunk.length) const packetSize = Math.min(maxPacketSize.toNumber(), chunk.length);
const packet = chunk.slice(0, packetSize) const packet = chunk.slice(0, packetSize);
chunk = chunk.slice(packetSize) chunk = chunk.slice(packetSize);
const offset = new Bigint(this.offset) const offset = new Bigint(this.offset);
this.offset = this.offset.add(packetSize) this.offset = this.offset.add(packetSize);
const retryInterval = fibonacci() const retryInterval = fibonacci();
let pending = true let pending = true;
while (pending) { while (pending) {
try { try {
await requestAsync('write', { await requestAsync(
FileId: this.file.FileId, 'write',
Offset: offset.toBuffer(), {
Buffer: packet FileId: this.file.FileId,
}, this.connection) Offset: offset.toBuffer(),
Buffer: packet,
pending = false },
this.connection
);
pending = false;
} catch (error) { } catch (error) {
if (error.code === 'STATUS_PENDING') { if (error.code === 'STATUS_PENDING') {
await new Promise((resolve, reject) => { await new Promise((resolve, reject) => {
setTimeout(resolve, retryInterval.next().value) setTimeout(resolve, retryInterval.next().value);
}) });
} else { } else {
throw error throw error;
} }
} }
} }
} }
next() next();
} }
async end (...args) { async end(...args) {
try { try {
super.end(...args) super.end(...args);
} finally { } finally {
await requestAsync('close', this.file, this.connection) await requestAsync('close', this.file, this.connection);
} }
} }
} }
export default function (path, options, cb) { export default function(path, options, cb) {
if (typeof options === 'function') { if (typeof options === 'function') {
cb = options cb = options;
options = {} options = {};
} }
let createDisposition let createDisposition;
const flags = options && options.flags const flags = options && options.flags;
if (flags === 'r') { if (flags === 'r') {
createDisposition = FILE_OPEN createDisposition = FILE_OPEN;
} else if (flags === 'r+') { } else if (flags === 'r+') {
createDisposition = FILE_OPEN_IF createDisposition = FILE_OPEN_IF;
} else if (flags === 'w' || flags === 'w+') { } else if (flags === 'w' || flags === 'w+') {
createDisposition = FILE_OVERWRITE_IF createDisposition = FILE_OVERWRITE_IF;
} else if (flags === 'wx' || flags === 'w+x') { } else if (flags === 'wx' || flags === 'w+x') {
createDisposition = FILE_CREATE createDisposition = FILE_CREATE;
} }
request('create', { path, createDisposition }, this, (err, file) => { request('create', { path, createDisposition }, this, (err, file) => {
if (err) { if (err) {
cb(err) cb(err);
} else { } else {
cb(null, new SmbWritableStream(this, file, options)) cb(null, new SmbWritableStream(this, file, options));
} }
}) });
} }
lib/api/src/ensureDir.js
View file @ 4b7d8773
import {request} from '../tools/smb2-forge' import { request } from '../tools/smb2-forge';
import Bluebird from 'bluebird' import Bluebird from 'bluebird';
const requestAsync = Bluebird.promisify(request) const requestAsync = Bluebird.promisify(request);
const ensureOneDir = async function (path, connection) { const ensureOneDir = async function(path, connection) {
try { try {
const fileOrDir = await requestAsync('open', {path}, connection) const fileOrDir = await requestAsync('open', { path }, connection);
if (fileOrDir.FileAttributes.readIntBE(0, 1) === 0x00000010) { // See http://download.microsoft.com/DOWNLOAD/9/5/E/95EF66AF-9026-4BB0-A41D-A4F81802D92C/[MS-FSCC].pdf Section 2.6 if (fileOrDir.FileAttributes.readIntBE(0, 1) === 0x00000010) {
await requestAsync('close', fileOrDir, connection) // See http://download.microsoft.com/DOWNLOAD/9/5/E/95EF66AF-9026-4BB0-A41D-A4F81802D92C/[MS-FSCC].pdf Section 2.6
await requestAsync('close', fileOrDir, connection);
} else { } else {
throw new Error(`${path} exists but is not a directory`) throw new Error(`${path} exists but is not a directory`);
} }
} catch (err) { } catch (err) {
if (err.code === 'STATUS_OBJECT_NAME_NOT_FOUND') { if (err.code === 'STATUS_OBJECT_NAME_NOT_FOUND') {
try { try {
await requestAsync('create_folder', {path}, connection) await requestAsync('create_folder', { path }, connection);
} catch (err) { } catch (err) {
if (err.code !== 'STATUS_OBJECT_NAME_COLLISION') { if (err.code !== 'STATUS_OBJECT_NAME_COLLISION') {
throw err throw err;
} }
} }
} else { } else {
throw err throw err;
} }
} }
} };
export default async function (path, cb) { export default async function(path, cb) {
const structure = path.split('\\') const structure = path.split('\\');
const base = [] const base = [];
try { try {
while (structure.length) { while (structure.length) {
base.push(structure.shift()) base.push(structure.shift());
const basePath = base.join('\\') const basePath = base.join('\\');
if (!basePath.length) { if (!basePath.length) {
continue continue;
} }
await ensureOneDir(basePath, this) await ensureOneDir(basePath, this);
} }
cb(null) cb(null);
} catch (error) { } catch (error) {
cb(error) cb(error);
} }
} }
lib/api/src/getSize.js
View file @ 4b7d8773
import {request} from '../tools/smb2-forge' import { request } from '../tools/smb2-forge';
export default function (path, cb) { export default function(path, cb) {
request('open', {path}, this, (err, file) => { request('open', { path }, this, (err, file) => {
if (err) { if (err) {
if (err.code === 'STATUS_OBJECT_NAME_NOT_FOUND') { if (err.code === 'STATUS_OBJECT_NAME_NOT_FOUND') {
err.code = 'ENOENT' err.code = 'ENOENT';
} }
cb(err) cb(err);
} else { } else {
let fileLength = 0 let fileLength = 0;
for (let i = 0; i < file.EndofFile.length; i++) { for (let i = 0; i < file.EndofFile.length; i++) {
fileLength += file.EndofFile[i] * Math.pow(2, i * 8) fileLength += file.EndofFile[i] * Math.pow(2, i * 8);
} }
cb(null, fileLength) cb(null, fileLength);
} }
}) });
} }
lib/api/unlink.js
View file @ 4b7d8773
var SMB2Forge = require('../tools/smb2-forge'),
SMB2Request = SMB2Forge.request,
var SMB2Forge = require('../tools/smb2-forge') bigint = require('../tools/bigint');
, SMB2Request = SMB2Forge.request
, bigint = require('../tools/bigint')
;
/* /*
* unlink * unlink
...@@ -18,34 +15,37 @@ var SMB2Forge = require('../tools/smb2-forge') ...@@ -18,34 +15,37 @@ var SMB2Forge = require('../tools/smb2-forge')
* - close the file * - close the file
* *
*/ */
module.exports = function(path, cb){ module.exports = function(path, cb) {
var connection = this; var connection = this;
connection.exists(path, function(err, exists){ connection.exists(path, function(err, exists) {
if (err) cb && cb(err);
if(err) cb && cb(err); else if (exists) {
else if(exists){
// SMB2 open file // SMB2 open file
SMB2Request('create', {path:path}, connection, function(err, file){ SMB2Request('create', { path: path }, connection, function(err, file) {
if(err) cb && cb(err); if (err) cb && cb(err);
// SMB2 query directory // SMB2 query directory
else SMB2Request('set_info', {FileId:file.FileId, FileInfoClass:'FileDispositionInformation',Buffer:(new bigint(1,1)).toBuffer()}, connection, function(err, files){ else
if(err) cb && cb(err); SMB2Request(
// SMB2 close directory 'set_info',
else SMB2Request('close', file, connection, function(err){ {
cb && cb(null, files); FileId: file.FileId,
}); FileInfoClass: 'FileDispositionInformation',
}); Buffer: new bigint(1, 1).toBuffer(),
},
connection,
function(err, files) {
if (err) cb && cb(err);
// SMB2 close directory
else
SMB2Request('close', file, connection, function(err) {
cb && cb(null, files);
});
}
);
}); });
} else { } else {
cb(new Error('File does not exists')); cb(new Error('File does not exists'));
} }
}); });
};
}
\ No newline at end of file
lib/api/writefile.js
View file @ 4b7d8773
var SMB2Forge = require('../tools/smb2-forge'),
SMB2Request = SMB2Forge.request,
var SMB2Forge = require('../tools/smb2-forge') bigint = require('../tools/bigint');
, SMB2Request = SMB2Forge.request
, bigint = require('../tools/bigint')
;
/* /*
* writeFile * writeFile
...@@ -20,24 +17,24 @@ var SMB2Forge = require('../tools/smb2-forge') ...@@ -20,24 +17,24 @@ var SMB2Forge = require('../tools/smb2-forge')
* - close the file * - close the file
* *
*/ */
module.exports = function(filename, data, options, cb){ module.exports = function(filename, data, options, cb) {
if (typeof options == 'function') {
if(typeof options == 'function'){
cb = options; cb = options;
options = {}; options = {};
} }
options.encoding = options.encoding || 'utf8'; options.encoding = options.encoding || 'utf8';
var connection = this var connection = this,
, file file,
, fileContent = Buffer.isBuffer(data) ? data : new Buffer(data, options.encoding) fileContent = Buffer.isBuffer(data)
, fileLength = new bigint(8, fileContent.length) ? data
; : new Buffer(data, options.encoding),
fileLength = new bigint(8, fileContent.length);
function createFile(fileCreated){ function createFile(fileCreated) {
SMB2Request('create', {path:filename}, connection, function(err, f){ SMB2Request('create', { path: filename }, connection, function(err, f) {
if(err) cb && cb(err); if (err) cb && cb(err);
// SMB2 set file size // SMB2 set file size
else { else {
file = f; file = f;
...@@ -46,9 +43,9 @@ module.exports = function(filename, data, options, cb){ ...@@ -46,9 +43,9 @@ module.exports = function(filename, data, options, cb){
}); });
} }
function closeFile(fileClosed){ function closeFile(fileClosed) {
SMB2Request('close', file, connection, function(err){ SMB2Request('close', file, connection, function(err) {
if(err) cb && cb(err); if (err) cb && cb(err);
else { else {
file = null; file = null;
fileClosed(); fileClosed();
...@@ -56,52 +53,71 @@ module.exports = function(filename, data, options, cb){ ...@@ -56,52 +53,71 @@ module.exports = function(filename, data, options, cb){
}); });
} }
function setFileSize(fileSizeSetted){ function setFileSize(fileSizeSetted) {
SMB2Request('set_info', {FileId:file.FileId, FileInfoClass:'FileEndOfFileInformation', Buffer:fileLength.toBuffer()}, connection, function(err){ SMB2Request(
if(err) cb && cb(err); 'set_info',
else fileSizeSetted(); {
}); FileId: file.FileId,
FileInfoClass: 'FileEndOfFileInformation',
Buffer: fileLength.toBuffer(),
},
connection,
function(err) {
if (err) cb && cb(err);
else fileSizeSetted();
}
);
} }
function writeFile(fileWritten){ function writeFile(fileWritten) {
var offset = new bigint(8) var offset = new bigint(8),
, stop = false stop = false,
, nbRemainingPackets = 0 nbRemainingPackets = 0,
, maxPacketSize = new bigint(8, 0x00010000 - 0x71) maxPacketSize = new bigint(8, 0x00010000 - 0x71);
;
// callback manager // callback manager
function callback(offset){ function callback(offset) {
return function(err){ return function(err) {
if(stop) return; if (stop) return;
if(err) { if (err) {
cb && cb(err); cb && cb(err);
stop = true; stop = true;
} else { } else {
nbRemainingPackets--; nbRemainingPackets--;
checkDone(); checkDone();
} }
} };
} }
// callback manager // callback manager
function checkDone(){ function checkDone() {
if(stop) return; if (stop) return;
createPackets(); createPackets();
if(nbRemainingPackets==0 && offset.ge(fileLength)) { if (nbRemainingPackets == 0 && offset.ge(fileLength)) {
fileWritten(); fileWritten();
} }
} }
// create packets // create packets
function createPackets(){ function createPackets() {
while(nbRemainingPackets<connection.packetConcurrency && offset.lt(fileLength)){ while (
nbRemainingPackets < connection.packetConcurrency &&
offset.lt(fileLength)
) {
// process packet size // process packet size
var rest = fileLength.sub(offset); var rest = fileLength.sub(offset);
var packetSize = rest.gt(maxPacketSize) ? maxPacketSize : rest; var packetSize = rest.gt(maxPacketSize) ? maxPacketSize : rest;
// generate buffer // generate buffer
SMB2Request('write', { SMB2Request(
'FileId':file.FileId 'write',
, 'Offset':offset.toBuffer() {
, 'Buffer':fileContent.slice(offset.toNumber(), offset.add(packetSize).toNumber()) FileId: file.FileId,
}, connection, callback(offset)); Offset: offset.toBuffer(),
Buffer: fileContent.slice(
offset.toNumber(),
offset.add(packetSize).toNumber()
),
},
connection,
callback(offset)
);
offset = offset.add(packetSize); offset = offset.add(packetSize);
nbRemainingPackets++; nbRemainingPackets++;
} }
...@@ -109,13 +125,11 @@ module.exports = function(filename, data, options, cb){ ...@@ -109,13 +125,11 @@ module.exports = function(filename, data, options, cb){
checkDone(); checkDone();
} }
createFile(function() {
createFile(function(){ setFileSize(function() {
setFileSize(function(){ writeFile(function() {
writeFile(function(){
closeFile(cb); closeFile(cb);
}); });
}); });
}); });
};
}
lib/messages/close.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
module.exports = message({ module.exports = message({
generate: function(connection, params) {
generate:function(connection, params){
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'CLOSE' Command: 'CLOSE',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'TreeId':connection.TreeId TreeId: connection.TreeId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'FileId':params.FileId FileId: params.FileId,
} },
}); });
},
}
}); });
lib/messages/create.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message') var SMB2Message = require('../tools/smb2-message');
var message = require('../tools/message') var message = require('../tools/message');
var FILE_OVERWRITE_IF = require('../structures/constants').FILE_OVERWRITE_IF var FILE_OVERWRITE_IF = require('../structures/constants').FILE_OVERWRITE_IF;
module.exports = message({ module.exports = message({
generate: function(connection, params) {
generate:function(connection, params){
var buffer = new Buffer(params.path, 'ucs2'); var buffer = new Buffer(params.path, 'ucs2');
var createDisposition = params.createDisposition var createDisposition = params.createDisposition;
/* See: https://msdn.microsoft.com/en-us/library/cc246502.aspx /* See: https://msdn.microsoft.com/en-us/library/cc246502.aspx
6 values for CreateDisposition. */ 6 values for CreateDisposition. */
if (!(createDisposition >= 0 && createDisposition <= 5)) { if (!(createDisposition >= 0 && createDisposition <= 5)) {
createDisposition = FILE_OVERWRITE_IF createDisposition = FILE_OVERWRITE_IF;
} }
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'CREATE' Command: 'CREATE',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'TreeId':connection.TreeId TreeId: connection.TreeId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'Buffer':buffer Buffer: buffer,
, 'DesiredAccess':0x001701DF DesiredAccess: 0x001701df,
, 'FileAttributes':0x00000080 FileAttributes: 0x00000080,
, 'ShareAccess':0x00000000 ShareAccess: 0x00000000,
, 'CreateDisposition': createDisposition CreateDisposition: createDisposition,
, 'CreateOptions':0x00000044 CreateOptions: 0x00000044,
, 'NameOffset':0x0078 NameOffset: 0x0078,
, 'CreateContextsOffset':0x007A+buffer.length CreateContextsOffset: 0x007a + buffer.length,
} },
}); });
},
}
}); });
lib/messages/create_folder.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
module.exports = message({ module.exports = message({
generate: function(connection, params) {
generate:function(connection, params){
var buffer = new Buffer(params.path, 'ucs2'); var buffer = new Buffer(params.path, 'ucs2');
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'CREATE' Command: 'CREATE',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'TreeId':connection.TreeId TreeId: connection.TreeId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'Buffer':buffer Buffer: buffer,
, 'DesiredAccess':0x001701DF DesiredAccess: 0x001701df,
, 'FileAttributes':0x00000000 FileAttributes: 0x00000000,
, 'ShareAccess':0x00000000 ShareAccess: 0x00000000,
, 'CreateDisposition':0x00000002 CreateDisposition: 0x00000002,
, 'CreateOptions':0x00000021 CreateOptions: 0x00000021,
, 'NameOffset':0x0078 NameOffset: 0x0078,
, 'CreateContextsOffset':0x007A+buffer.length CreateContextsOffset: 0x007a + buffer.length,
} },
}); });
},
}
}); });
lib/messages/negotiate.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
module.exports = message({ module.exports = message({
generate: function(connection) {
generate:function(connection){
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'NEGOTIATE' Command: 'NEGOTIATE',
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
}); });
},
}
}); });
lib/messages/open.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
module.exports = message({ module.exports = message({
generate: function(connection, params) {
generate:function(connection, params){
var buffer = new Buffer(params.path, 'ucs2'); var buffer = new Buffer(params.path, 'ucs2');
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'CREATE' Command: 'CREATE',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'TreeId':connection.TreeId TreeId: connection.TreeId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'Buffer':buffer Buffer: buffer,
, 'NameOffset':0x0078 NameOffset: 0x0078,
, 'CreateContextsOffset':0x007A+buffer.length CreateContextsOffset: 0x007a + buffer.length,
} },
}); });
},
}
}); });
lib/messages/open_folder.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
module.exports = message({ module.exports = message({
generate: function(connection, params) {
generate:function(connection, params){
var buffer = new Buffer(params.path, 'ucs2'); var buffer = new Buffer(params.path, 'ucs2');
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'CREATE' Command: 'CREATE',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'TreeId':connection.TreeId TreeId: connection.TreeId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'Buffer':buffer Buffer: buffer,
, 'DesiredAccess':0x001701DF DesiredAccess: 0x001701df,
, 'FileAttributes':0x00000000 FileAttributes: 0x00000000,
, 'ShareAccess':0x00000007 ShareAccess: 0x00000007,
, 'CreateDisposition':0x00000001 CreateDisposition: 0x00000001,
, 'CreateOptions':0x00200021 CreateOptions: 0x00200021,
, 'NameOffset':0x0078 NameOffset: 0x0078,
, 'CreateContextsOffset':0x007A+buffer.length CreateContextsOffset: 0x007a + buffer.length,
} },
}); });
},
}
}); });
lib/messages/query_directory.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
module.exports = message({ module.exports = message({
generate: function(connection, params) {
generate:function(connection, params){
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'QUERY_DIRECTORY' Command: 'QUERY_DIRECTORY',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'TreeId':connection.TreeId TreeId: connection.TreeId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'FileId':params.FileId FileId: params.FileId,
, 'Buffer':new Buffer('*', 'ucs2') Buffer: new Buffer('*', 'ucs2'),
} },
}); });
},
} parseResponse: function(response) {
, parseResponse:function(response){
return parseFiles(response.getResponse().Buffer); return parseFiles(response.getResponse().Buffer);
} },
}); });
/* /*
* HELPERS * HELPERS
*/ */
function parseFiles(buffer){ function parseFiles(buffer) {
var files = [] var files = [],
, offset = 0 offset = 0,
, nextFileOffset = -1 nextFileOffset = -1;
; while (nextFileOffset != 0) {
while(nextFileOffset!=0){
// extract next file offset // extract next file offset
nextFileOffset = buffer.readUInt32LE(offset); nextFileOffset = buffer.readUInt32LE(offset);
// extract the file // extract the file
files.push( files.push(
parseFile( parseFile(
buffer.slice(offset+4, nextFileOffset ? offset+nextFileOffset : buffer.length) buffer.slice(
offset + 4,
nextFileOffset ? offset + nextFileOffset : buffer.length
)
) )
); );
// move to nex file // move to nex file
...@@ -58,65 +47,66 @@ function parseFiles(buffer){ ...@@ -58,65 +47,66 @@ function parseFiles(buffer){
return files; return files;
} }
function parseFile(buffer){ function parseFile(buffer) {
var file = {} var file = {},
, offset = 0 offset = 0;
;
// index // index
file.Index = buffer.readUInt32LE(offset); file.Index = buffer.readUInt32LE(offset);
offset+=4; offset += 4;
// CreationTime // CreationTime
file.CreationTime = buffer.slice(offset, offset+8); file.CreationTime = buffer.slice(offset, offset + 8);
offset+=8; offset += 8;
// LastAccessTime // LastAccessTime
file.LastAccessTime = buffer.slice(offset, offset+8); file.LastAccessTime = buffer.slice(offset, offset + 8);
offset+=8; offset += 8;
// LastWriteTime // LastWriteTime
file.LastWriteTime = buffer.slice(offset, offset+8); file.LastWriteTime = buffer.slice(offset, offset + 8);
offset+=8; offset += 8;
// ChangeTime // ChangeTime
file.ChangeTime = buffer.slice(offset, offset+8); file.ChangeTime = buffer.slice(offset, offset + 8);
offset+=8; offset += 8;
// EndofFile // EndofFile
file.EndofFile = buffer.slice(offset, offset+8); file.EndofFile = buffer.slice(offset, offset + 8);
offset+=8; offset += 8;
// AllocationSize // AllocationSize
file.AllocationSize = buffer.slice(offset, offset+8); file.AllocationSize = buffer.slice(offset, offset + 8);
offset+=8; offset += 8;
// FileAttributes // FileAttributes
file.FileAttributes = buffer.readUInt32LE(offset); file.FileAttributes = buffer.readUInt32LE(offset);
offset+=4; offset += 4;
// FilenameLength // FilenameLength
file.FilenameLength = buffer.readUInt32LE(offset); file.FilenameLength = buffer.readUInt32LE(offset);
offset+=4; offset += 4;
// EASize // EASize
file.EASize = buffer.readUInt32LE(offset); file.EASize = buffer.readUInt32LE(offset);
offset+=4; offset += 4;
// ShortNameLength // ShortNameLength
file.ShortNameLength = buffer.readUInt8(offset); file.ShortNameLength = buffer.readUInt8(offset);
offset+=1; offset += 1;
// FileId // FileId
file.FileId = buffer.slice(offset, offset+8); file.FileId = buffer.slice(offset, offset + 8);
offset+=8; offset += 8;
// Reserved // Reserved
offset+=27; offset += 27;
// Filename // Filename
file.Filename = buffer.slice(offset, offset+file.FilenameLength).toString('ucs2'); file.Filename = buffer
offset+=file.FilenameLength; .slice(offset, offset + file.FilenameLength)
.toString('ucs2');
offset += file.FilenameLength;
return file; return file;
} }
\ No newline at end of file
lib/messages/read.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
module.exports = message({ module.exports = message({
generate: function(connection, file) {
generate:function(connection, file){
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'READ' Command: 'READ',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'TreeId':connection.TreeId TreeId: connection.TreeId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'FileId':file.FileId FileId: file.FileId,
, 'Length':file.Length Length: file.Length,
, 'Offset':file.Offset Offset: file.Offset,
} },
}); });
},
} parseResponse: function(response) {
, parseResponse:function(response){
return response.getResponse().Buffer; return response.getResponse().Buffer;
} },
}); });
lib/messages/session_setup_step1.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message'),
var SMB2Message = require('../tools/smb2-message') ntlm = require('ntlm');
, message = require('../tools/message')
, ntlm = require('ntlm')
;
module.exports = message({ module.exports = message({
generate: function(connection) {
generate:function(connection){
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'SESSION_SETUP' Command: 'SESSION_SETUP',
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'Buffer':ntlm.encodeType1( Buffer: ntlm.encodeType1(connection.ip, connection.domain),
connection.ip },
, connection.domain
)
}
}); });
},
} successCode: 'STATUS_MORE_PROCESSING_REQUIRED',
, successCode: 'STATUS_MORE_PROCESSING_REQUIRED'
, onSuccess:function(connection, response){ onSuccess: function(connection, response) {
var h = response.getHeaders(); var h = response.getHeaders();
connection.SessionId = h.SessionId; connection.SessionId = h.SessionId;
connection.nonce = ntlm.decodeType2(response.getResponse().Buffer); connection.nonce = ntlm.decodeType2(response.getResponse().Buffer);
} },
}); });
lib/messages/session_setup_step2.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message'),
var SMB2Message = require('../tools/smb2-message') ntlm = require('ntlm');
, message = require('../tools/message')
, ntlm = require('ntlm')
;
module.exports = message({ module.exports = message({
generate: function(connection) {
generate:function(connection){
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'SESSION_SETUP' Command: 'SESSION_SETUP',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'Buffer':ntlm.encodeType3( Buffer: ntlm.encodeType3(
connection.username connection.username,
, connection.ip connection.ip,
, connection.domain connection.domain,
, connection.nonce connection.nonce,
, connection.password connection.password
) ),
} },
}); });
},
}
}); });
lib/messages/set_info.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
var fileInfoClasses = { var fileInfoClasses = {
'FileAllocationInformation': 19 FileAllocationInformation: 19,
, 'FileBasicInformation': 4 FileBasicInformation: 4,
, 'FileDispositionInformation': 13 FileDispositionInformation: 13,
, 'FileEndOfFileInformation': 20 FileEndOfFileInformation: 20,
, 'FileFullEaInformation': 15 FileFullEaInformation: 15,
, 'FileLinkInformation': 11 FileLinkInformation: 11,
, 'FileModeInformation': 16 FileModeInformation: 16,
, 'FilePipeInformation': 23 FilePipeInformation: 23,
, 'FilePositionInformation': 14 FilePositionInformation: 14,
, 'FileRenameInformation': 10 FileRenameInformation: 10,
, 'FileShortNameInformation': 40 FileShortNameInformation: 40,
, 'FileValidDataLengthInformation': 39 FileValidDataLengthInformation: 39,
}; };
module.exports = message({ module.exports = message({
generate: function(connection, params) {
generate:function(connection, params){
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'SET_INFO' Command: 'SET_INFO',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'TreeId':connection.TreeId TreeId: connection.TreeId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'FileInfoClass':fileInfoClasses[params.FileInfoClass] FileInfoClass: fileInfoClasses[params.FileInfoClass],
, 'FileId':params.FileId FileId: params.FileId,
, 'Buffer':params.Buffer Buffer: params.Buffer,
} },
}); });
},
}
}); });
lib/messages/tree_connect.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
module.exports = message({ module.exports = message({
generate: function(connection) {
generate:function(connection){
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'TREE_CONNECT' Command: 'TREE_CONNECT',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'Buffer':new Buffer(connection.fullPath, 'ucs2') Buffer: new Buffer(connection.fullPath, 'ucs2'),
} },
}); });
},
} onSuccess: function(connection, response) {
, onSuccess:function(connection, response){
var h = response.getHeaders(); var h = response.getHeaders();
connection.TreeId = h.TreeId; connection.TreeId = h.TreeId;
} },
}); });
lib/messages/write.js
View file @ 4b7d8773
var SMB2Message = require('../tools/smb2-message'),
message = require('../tools/message');
var SMB2Message = require('../tools/smb2-message')
, message = require('../tools/message')
;
module.exports = message({ module.exports = message({
generate: function(connection, params) {
generate:function(connection, params){
return new SMB2Message({ return new SMB2Message({
headers:{ headers: {
'Command':'WRITE' Command: 'WRITE',
, 'SessionId':connection.SessionId SessionId: connection.SessionId,
, 'TreeId':connection.TreeId TreeId: connection.TreeId,
, 'ProcessId':connection.ProcessId ProcessId: connection.ProcessId,
} },
, request:{ request: {
'FileId':params.FileId FileId: params.FileId,
, 'Offset':params.Offset Offset: params.Offset,
, 'Buffer':params.Buffer Buffer: params.Buffer,
} },
}); });
},
} parseResponse: function(response) {
, parseResponse:function(response){
return response.getResponse().Buffer; return response.getResponse().Buffer;
} },
}); });
lib/smb2.js
View file @ 4b7d8773
/* /*
* CONSTANTS * CONSTANTS
*/ */
var shareRegExp = /\\\\([^\\]*)\\([^\\]*)\\?/ var shareRegExp = /\\\\([^\\]*)\\([^\\]*)\\?/,
, port = 445 port = 445,
, packetConcurrency = 20 packetConcurrency = 20,
, autoCloseTimeout = 10000 autoCloseTimeout = 10000;
;
/* /*
* DEPENDENCIES * DEPENDENCIES
*/ */
var SMB2Connection = require('./tools/smb2-connection'); var SMB2Connection = require('./tools/smb2-connection');
/* /*
* CONSTRUCTOR * CONSTRUCTOR
*/ */
var SMB = module.exports = function(opt){ var SMB = (module.exports = function(opt) {
opt = opt || {}; opt = opt || {};
// Parse share-string // Parse share-string
var matches; var matches;
if(!opt.share || !(matches = opt.share.match(shareRegExp))){ if (!opt.share || !(matches = opt.share.match(shareRegExp))) {
throw new Error('the share is not valid'); throw new Error('the share is not valid');
} }
...@@ -50,26 +44,26 @@ var SMB = module.exports = function(opt){ ...@@ -50,26 +44,26 @@ var SMB = module.exports = function(opt){
// close timeout default 10s // close timeout default 10s
if (opt.autoCloseTimeout !== undefined) { if (opt.autoCloseTimeout !== undefined) {
this.autoCloseTimeout = opt.autoCloseTimeout this.autoCloseTimeout = opt.autoCloseTimeout;
} else { } else {
this.autoCloseTimeout = autoCloseTimeout this.autoCloseTimeout = autoCloseTimeout;
} }
// store authentification // store authentification
this.domain = opt.domain; this.domain = opt.domain;
this.username = opt.username; this.username = opt.username;
this.password = opt.password; this.password = opt.password;
// set session id // set session id
this.SessionId = Math.floor(Math.random()*256) & 0xFF; this.SessionId = Math.floor(Math.random() * 256) & 0xff;
// set the process id // set the process id
// https://msdn.microsoft.com/en-us/library/ff470100.aspx // https://msdn.microsoft.com/en-us/library/ff470100.aspx
this.ProcessId = new Buffer([ this.ProcessId = new Buffer([
Math.floor(Math.random()*256) & 0xFF, Math.floor(Math.random() * 256) & 0xff,
Math.floor(Math.random()*256) & 0xFF, Math.floor(Math.random() * 256) & 0xff,
Math.floor(Math.random()*256) & 0xFF, Math.floor(Math.random() * 256) & 0xff,
Math.floor(Math.random()*256) & 0xFE Math.floor(Math.random() * 256) & 0xfe,
]); ]);
// activate debug mode // activate debug mode
...@@ -77,29 +71,32 @@ var SMB = module.exports = function(opt){ ...@@ -77,29 +71,32 @@ var SMB = module.exports = function(opt){
// init connection (socket) // init connection (socket)
SMB2Connection.init(this); SMB2Connection.init(this);
}; });
/* /*
* PROTOTYPE * PROTOTYPE
*/ */
var proto = SMB.prototype = {}; var proto = (SMB.prototype = {});
proto.close = require('./api/close'); proto.close = require('./api/close');
proto.exists = SMB2Connection.requireConnect(require('./api/exists')); proto.exists = SMB2Connection.requireConnect(require('./api/exists'));
proto.rename = SMB2Connection.requireConnect(require('./api/rename')); proto.rename = SMB2Connection.requireConnect(require('./api/rename'));
proto.readFile = SMB2Connection.requireConnect(require('./api/readfile')); proto.readFile = SMB2Connection.requireConnect(require('./api/readfile'));
proto.createReadStream = SMB2Connection.requireConnect(require('./api/createReadStream')); proto.createReadStream = SMB2Connection.requireConnect(
proto.createWriteStream = SMB2Connection.requireConnect(require('./api/createWriteStream')); require('./api/createReadStream')
);
proto.createWriteStream = SMB2Connection.requireConnect(
require('./api/createWriteStream')
);
proto.writeFile = SMB2Connection.requireConnect(require('./api/writefile')); proto.writeFile = SMB2Connection.requireConnect(require('./api/writefile'));
proto.unlink = SMB2Connection.requireConnect(require('./api/unlink')); proto.unlink = SMB2Connection.requireConnect(require('./api/unlink'));
proto.readdir = SMB2Connection.requireConnect(require('./api/readdir')); proto.readdir = SMB2Connection.requireConnect(require('./api/readdir'));
proto.rmdir = SMB2Connection.requireConnect(require('./api/rmdir')); proto.rmdir = SMB2Connection.requireConnect(require('./api/rmdir'));
proto.mkdir = SMB2Connection.requireConnect(require('./api/mkdir')); proto.mkdir = SMB2Connection.requireConnect(require('./api/mkdir'));
proto.ensureDir = SMB2Connection.requireConnect(require('./api/ensureDir')); proto.ensureDir = SMB2Connection.requireConnect(require('./api/ensureDir'));
proto.getSize = SMB2Connection.requireConnect(require('./api/getSize')); proto.getSize = SMB2Connection.requireConnect(require('./api/getSize'));
lib/structures/close.js
View file @ 4b7d8773
module.exports = { module.exports = {
request: [
['StructureSize', 2, 24],
['Flags', 2, 0],
['Reserved', 4, 0],
['FileId', 16],
],
request:[ response: [
['StructureSize', 2, 24] ['StructureSize', 2],
, ['Flags', 2, 0] ['Flags', 2],
, ['Reserved', 4, 0] ['Reserved', 4],
, ['FileId', 16] ['CreationTime', 8],
] ['LastAccessTime', 8],
['LastWriteTime', 8],
, response:[ ['ChangeTime', 8],
['StructureSize', 2] ['AllocationSize', 8],
, ['Flags', 2] ['EndofFile', 8],
, ['Reserved', 4] ['FileAttributes', 4],
, ['CreationTime', 8] ],
, ['LastAccessTime', 8] };
, ['LastWriteTime', 8]
, ['ChangeTime', 8]
, ['AllocationSize', 8]
, ['EndofFile', 8]
, ['FileAttributes', 4]
]
}
lib/structures/constants.js
View file @ 4b7d8773
...@@ -8,5 +8,5 @@ module.exports = { ...@@ -8,5 +8,5 @@ module.exports = {
FILE_CREATE: 0x00000002, FILE_CREATE: 0x00000002,
FILE_OPEN_IF: 0x00000003, FILE_OPEN_IF: 0x00000003,
FILE_OVERWRITE: 0x00000004, FILE_OVERWRITE: 0x00000004,
FILE_OVERWRITE_IF: 0x00000005 FILE_OVERWRITE_IF: 0x00000005,
} };
lib/structures/create.js
View file @ 4b7d8773
module.exports = { module.exports = {
request: [
request:[ ['StructureSize', 2, 57],
['SecurityFlags', 1, 0],
['StructureSize', 2, 57] ['RequestedOplockLevel', 1, 0],
, ['SecurityFlags', 1, 0] ['ImpersonationLevel', 4, 0x00000002],
, ['RequestedOplockLevel', 1, 0] ['SmbCreateFlags', 8, 0],
, ['ImpersonationLevel', 4, 0x00000002] ['Reserved', 8, 0],
, ['SmbCreateFlags', 8, 0] ['DesiredAccess', 4, 0x00100081],
, ['Reserved', 8, 0] ['FileAttributes', 4, 0x00000000],
, ['DesiredAccess', 4, 0x00100081] ['ShareAccess', 4, 0x00000007],
, ['FileAttributes', 4, 0x00000000] ['CreateDisposition', 4, 0x00000001],
, ['ShareAccess', 4, 0x00000007] ['CreateOptions', 4, 0x00000020],
, ['CreateDisposition', 4, 0x00000001] ['NameOffset', 2],
, ['CreateOptions', 4, 0x00000020] ['NameLength', 2],
, ['NameOffset', 2] ['CreateContextsOffset', 4],
, ['NameLength', 2] ['CreateContextsLength', 4],
, ['CreateContextsOffset', 4] ['Buffer', 'NameLength'],
, ['CreateContextsLength', 4] ['Reserved2', 2, 0x4200],
, ['Buffer', 'NameLength'] ['CreateContexts', 'CreateContextsLength', ''],
, ['Reserved2', 2, 0x4200] ],
, ['CreateContexts', 'CreateContextsLength', '']
response: [
] ['StructureSize', 2],
['OplockLevel', 1],
, response:[ ['Flags', 1],
['CreateAction', 4],
['StructureSize', 2] ['CreationTime', 8],
, ['OplockLevel', 1] ['LastAccessTime', 8],
, ['Flags', 1] ['LastWriteTime', 8],
, ['CreateAction', 4] ['ChangeTime', 8],
, ['CreationTime', 8] ['AllocationSize', 8],
, ['LastAccessTime', 8] ['EndofFile', 8],
, ['LastWriteTime', 8] ['FileAttributes', 4],
, ['ChangeTime', 8] ['Reserved2', 4],
, ['AllocationSize', 8] ['FileId', 16],
, ['EndofFile', 8] ['CreateContextsOffset', 4],
, ['FileAttributes', 4] ['CreateContextsLength', 4],
, ['Reserved2', 4] ['Buffer', 'CreateContextsLength'],
, ['FileId', 16] ],
, ['CreateContextsOffset', 4] };
, ['CreateContextsLength', 4]
, ['Buffer', 'CreateContextsLength']
]
}
lib/structures/negotiate.js
View file @ 4b7d8773
module.exports = { module.exports = {
request: [
['StructureSize', 2, 36],
['DialectCount', 2, 2],
['SecurityMode', 2, 1],
['Reserved', 2, 0],
['Capabilities', 4, 0],
['ClientGuid', 16, 0],
['ClientStartTime', 8, 0],
['Dialects', 4, new Buffer([0x02, 0x02, 0x10, 0x02])],
],
request:[ response: [
['StructureSize', 2, 36] ['StructureSize', 2],
, ['DialectCount', 2, 2] ['SecurityMode', 2],
, ['SecurityMode', 2, 1] ['DialectRevision', 2],
, ['Reserved', 2, 0] ['Reserved', 2],
, ['Capabilities', 4, 0] ['ServerGuid', 16],
, ['ClientGuid', 16, 0] ['Capabilities', 4],
, ['ClientStartTime', 8, 0] ['MaxTransactSize', 4],
, ['Dialects', 4, new Buffer([0x02,0x02,0x10,0x02])] ['MaxReadSize', 4],
] ['MaxWriteSize', 4],
['SystemTime', 8],
, response:[ ['ServerStartTime', 8],
['StructureSize', 2] ['SecurityBufferOffset', 2],
, ['SecurityMode', 2] ['SecurityBufferLength', 2],
, ['DialectRevision', 2] ['Reserved2', 4],
, ['Reserved', 2] ['Buffer', 'SecurityBufferLength'],
, ['ServerGuid', 16] ],
, ['Capabilities', 4] };
, ['MaxTransactSize', 4]
, ['MaxReadSize', 4]
, ['MaxWriteSize', 4]
, ['SystemTime', 8]
, ['ServerStartTime', 8]
, ['SecurityBufferOffset', 2]
, ['SecurityBufferLength', 2]
, ['Reserved2', 4]
, ['Buffer', 'SecurityBufferLength']
]
}
lib/structures/query_directory.js
View file @ 4b7d8773
module.exports = { module.exports = {
request: [
['StructureSize', 2, 33],
['FileInformationClass', 1, 0x25], // FileBothDirectoryInformation plus volume file ID about a file or directory.
['Flags', 1, 0],
['FileIndex', 4, 0],
['FileId', 16],
['FileNameOffset', 2, 96],
['FileNameLength', 2],
['OutputBufferLength', 4, 0x00010000],
['Buffer', 'FileNameLength'],
],
request:[ response: [
['StructureSize', 2, 33] ['StructureSize', 2],
, ['FileInformationClass', 1, 0x25] // FileBothDirectoryInformation plus volume file ID about a file or directory. ['OutputBufferOffset', 2],
, ['Flags', 1, 0] ['OutputBufferLength', 4],
, ['FileIndex', 4, 0] ['Buffer', 'OutputBufferLength'],
, ['FileId', 16] ],
, ['FileNameOffset', 2, 96] };
, ['FileNameLength', 2]
, ['OutputBufferLength', 4, 0x00010000]
, ['Buffer', 'FileNameLength']
]
, response:[
['StructureSize', 2]
, ['OutputBufferOffset', 2]
, ['OutputBufferLength', 4]
, ['Buffer', 'OutputBufferLength']
]
}
lib/structures/read.js
View file @ 4b7d8773
module.exports = { module.exports = {
request: [
['StructureSize', 2, 49],
['Padding', 1, 0x50],
['Flags', 1, 0],
['Length', 4],
['Offset', 8],
['FileId', 16],
['MinimumCount', 4, 0],
['Channel', 4, 0],
['RemainingBytes', 4, 0],
['ReadChannelInfoOffset', 2, 0],
['ReadChannelInfoLength', 2, 0],
['Buffer', 1, 0],
],
request:[ response: [
['StructureSize', 2, 49] ['StructureSize', 2],
, ['Padding', 1, 0x50] ['DataOffset', 1],
, ['Flags', 1, 0] ['Reserved', 1],
, ['Length', 4] ['DataLength', 4],
, ['Offset', 8] ['DataRemaining', 4],
, ['FileId', 16] ['Reserved2', 4],
, ['MinimumCount', 4, 0] ['Buffer', 'DataLength'],
, ['Channel', 4, 0] ],
, ['RemainingBytes', 4, 0] };
, ['ReadChannelInfoOffset', 2, 0]
, ['ReadChannelInfoLength', 2, 0]
, ['Buffer', 1, 0]
]
, response:[
['StructureSize', 2]
, ['DataOffset', 1]
, ['Reserved', 1]
, ['DataLength', 4]
, ['DataRemaining', 4]
, ['Reserved2', 4]
, ['Buffer', 'DataLength']
]
}
lib/structures/session_setup.js
View file @ 4b7d8773
module.exports = { module.exports = {
request: [
['StructureSize', 2, 25],
['Flags', 1, 0],
['SecurityMode', 1, 1],
['Capabilities', 4, 1],
['Channel', 4, 0],
['SecurityBufferOffset', 2, 88],
['SecurityBufferLength', 2],
['PreviousSessionId', 8, 0],
['Buffer', 'SecurityBufferLength'],
],
request:[ response: [
['StructureSize', 2, 25] ['StructureSize', 2],
, ['Flags', 1, 0] ['SessionFlags', 2],
, ['SecurityMode', 1, 1] ['SecurityBufferOffset', 2],
, ['Capabilities', 4, 1] ['SecurityBufferLength', 2],
, ['Channel', 4, 0] ['Buffer', 'SecurityBufferLength'],
, ['SecurityBufferOffset', 2, 88] ],
, ['SecurityBufferLength', 2] };
, ['PreviousSessionId', 8, 0]
, ['Buffer','SecurityBufferLength']
]
, response:[
['StructureSize', 2]
, ['SessionFlags', 2]
, ['SecurityBufferOffset', 2]
, ['SecurityBufferLength', 2]
, ['Buffer', 'SecurityBufferLength']
]
}
lib/structures/set_info.js
View file @ 4b7d8773
module.exports = { module.exports = {
request: [
request:[ ['StructureSize', 2, 33],
['StructureSize', 2, 33] ['InfoType', 1, 1],
, ['InfoType', 1, 1] ['FileInfoClass', 1],
, ['FileInfoClass', 1] ['BufferLength', 4],
, ['BufferLength', 4] ['BufferOffset', 2, 0x0060],
, ['BufferOffset', 2, 0x0060] ['Reserved', 2, 0],
, ['Reserved', 2, 0] ['AdditionalInformation', 4, 0],
, ['AdditionalInformation', 4, 0] ['FileId', 16],
, ['FileId', 16] ['Buffer', 'BufferLength'],
, ['Buffer','BufferLength'] ],
]
response: [['StructureSize', 2]],
, response:[ };
['StructureSize', 2]
]
}
lib/structures/tree_connect.js
View file @ 4b7d8773
module.exports = { module.exports = {
request: [
['StructureSize', 2, 9],
['Reserved', 2, 0],
['PathOffset', 2, 72],
['PathLength', 2],
['Buffer', 'PathLength'],
],
request:[ response: [
['StructureSize', 2, 9] ['StructureSize', 2],
, ['Reserved', 2, 0] ['ShareType', 1],
, ['PathOffset', 2, 72] ['Reserved', 1],
, ['PathLength', 2] ['ShareFlags', 4],
, ['Buffer', 'PathLength'] ['Capabilities', 4],
] ['MaximalAccess', 4],
],
, response:[ };
['StructureSize', 2]
, ['ShareType', 1]
, ['Reserved', 1]
, ['ShareFlags', 4]
, ['Capabilities', 4]
, ['MaximalAccess', 4]
]
}
lib/structures/write.js
View file @ 4b7d8773
module.exports = { module.exports = {
request: [
['StructureSize', 2, 49],
['DataOffset', 2, 0x70],
['Length', 4, 0],
['Offset', 8],
['FileId', 16],
['Channel', 4, 0],
['RemainingBytes', 4, 0],
['WriteChannelInfoOffset', 2, 0],
['WriteChannelInfoLength', 2, 0],
['Flags', 4, 0],
['Buffer', 'Length'],
],
request:[ response: [
['StructureSize', 2, 49] ['StructureSize', 2],
, ['DataOffset', 2, 0x70] ['Reserved', 2],
, ['Length', 4, 0] ['Count', 4],
, ['Offset', 8] ['Remaining', 4],
, ['FileId', 16] ['WriteChannelInfoOffset', 2],
, ['Channel', 4, 0] ['WriteChannelInfoLength', 2],
, ['RemainingBytes', 4, 0] ],
, ['WriteChannelInfoOffset', 2, 0] };
, ['WriteChannelInfoLength', 2, 0]
, ['Flags', 4, 0]
, ['Buffer', 'Length']
]
, response:[
['StructureSize', 2]
, ['Reserved', 2]
, ['Count', 4]
, ['Remaining', 4]
, ['WriteChannelInfoOffset', 2]
, ['WriteChannelInfoLength', 2]
]
}
lib/tools/bigint.js
View file @ 4b7d8773
var BigInt = (module.exports = function(n, v) {
var BigInt = module.exports = function(n, v){ if (BigInt.isBigInt(n)) {
if(BigInt.isBigInt(n)){
this.buffer = new Buffer(n.buffer.length); this.buffer = new Buffer(n.buffer.length);
n.buffer.copy(this.buffer, 0); n.buffer.copy(this.buffer, 0);
this.sign = n.sign; this.sign = n.sign;
} else { } else {
this.buffer = new Buffer(n); this.buffer = new Buffer(n);
this.buffer.fill(0); this.buffer.fill(0);
this.sign = 1; this.sign = 1;
v = v || 0; v = v || 0;
if(v != 0){ if (v != 0) {
if (v < 0) {
if(v<0) {
this.sign = -1; this.sign = -1;
v = -v; v = -v;
} }
v = v.toString(16); v = v.toString(16);
var size = Math.ceil(v.length/2) var size = Math.ceil(v.length / 2),
, carry = size*2 - v.length carry = size * 2 - v.length;
;
for(var i=0; i<size; i++) { for (var i = 0; i < size; i++) {
var start = (size - i - 1) * 2 - carry; var start = (size - i - 1) * 2 - carry;
this.buffer.writeUInt8( this.buffer.writeUInt8(
parseInt( start == -1 ? v.substr(0,1) : v.substr(start, 2), 16) parseInt(start == -1 ? v.substr(0, 1) : v.substr(start, 2), 16),
, i i
); );
} }
} }
} }
});
}
/* /*
* STATICS * STATICS
*/ */
BigInt.isBigInt = function(v){ BigInt.isBigInt = function(v) {
return v && v.buffer && Buffer.isBuffer(v.buffer); return v && v.buffer && Buffer.isBuffer(v.buffer);
} };
BigInt.toBigInt = function(n, v){ BigInt.toBigInt = function(n, v) {
return new BigInt(n, v); return new BigInt(n, v);
} };
BigInt.fromBuffer = function(b, sign){ BigInt.fromBuffer = function(b, sign) {
sign = typeof sign == 'undefined' ? 1 : sign; sign = typeof sign == 'undefined' ? 1 : sign;
var bi = new BigInt(0); var bi = new BigInt(0);
bi.sign = sign; bi.sign = sign;
bi.buffer = b; bi.buffer = b;
return bi; return bi;
} };
/* /*
* ADD / SUB * ADD / SUB
*/ */
BigInt.prototype.add = function(v){ BigInt.prototype.add = function(v) {
if (!BigInt.isBigInt(v)) {
if(!BigInt.isBigInt(v)) {
v = BigInt.toBigInt(this.buffer.length, v); v = BigInt.toBigInt(this.buffer.length, v);
} }
if(this.sign != v.sign){ if (this.sign != v.sign) {
return this.neg().sub(v); return this.neg().sub(v);
} }
var carry = 0 var carry = 0,
, n = Math.max(v.buffer.length, this.buffer.length) n = Math.max(v.buffer.length, this.buffer.length),
, result = new BigInt(n); result = new BigInt(n);
; for (var i = 0; i < n; i++) {
var r =
for(var i=0;i<n;i++){ (i < this.buffer.length ? this.buffer.readUInt8(i) : 0) +
var r = ( i<this.buffer.length ? this.buffer.readUInt8(i) : 0 ) (i < v.buffer.length ? v.buffer.readUInt8(i) : 0) +
+ ( i<v.buffer.length ? v.buffer.readUInt8(i) : 0 ) carry;
+ carry result.buffer.writeUInt8(r & 0xff, i);
;
result.buffer.writeUInt8(r & 0xFF, i);
carry = r >> 8; carry = r >> 8;
} }
result.sign = this.sign; result.sign = this.sign;
return result;
} return result;
};
BigInt.prototype.neg = function(v){ BigInt.prototype.neg = function(v) {
var result = new BigInt(this); var result = new BigInt(this);
result.sign *= -1; result.sign *= -1;
return result; return result;
} };
BigInt.prototype.abs = function(v){ BigInt.prototype.abs = function(v) {
var result = new BigInt(this); var result = new BigInt(this);
result.sign = 1; result.sign = 1;
return result; return result;
} };
BigInt.prototype.sub = function(v){
if(!BigInt.isBigInt(v)) { BigInt.prototype.sub = function(v) {
if (!BigInt.isBigInt(v)) {
v = BigInt.toBigInt(this.buffer.length, v); v = BigInt.toBigInt(this.buffer.length, v);
} }
if(this.sign != v.sign) { if (this.sign != v.sign) {
return this.add(v.neg()); return this.add(v.neg());
} }
var carry = 0 var carry = 0,
, a = new BigInt(this) a = new BigInt(this),
, b = new BigInt(v) b = new BigInt(v),
, n = Math.max(a.buffer.length, b.buffer.length) n = Math.max(a.buffer.length, b.buffer.length),
, result = new BigInt(n) result = new BigInt(n),
, sign = this.sign sign = this.sign;
;
if(a.abs().lt(b.abs())){ if (a.abs().lt(b.abs())) {
var t = a; var t = a;
a = b; a = b;
b = t; b = t;
sign *= -1; sign *= -1;
} }
for(var i=0; i<n; i++){ for (var i = 0; i < n; i++) {
var va = ( i<a.buffer.length ? a.buffer.readUInt8(i) : 0 ) var va = i < a.buffer.length ? a.buffer.readUInt8(i) : 0,
, vb = ( i<b.buffer.length ? b.buffer.readUInt8(i) : 0 ) vb = i < b.buffer.length ? b.buffer.readUInt8(i) : 0,
, c = 0 c = 0,
, r = va-vb-carry r = va - vb - carry;
;
while(r<0){ while (r < 0) {
r+=0xFF; r += 0xff;
c--; c--;
} }
result.buffer.writeUInt8(r & 0xFF, i); result.buffer.writeUInt8(r & 0xff, i);
carry = (r >> 8) + c; carry = (r >> 8) + c;
} }
result.sign = sign; result.sign = sign;
return result; return result;
};
}
/* /*
* EXPORTS * EXPORTS
*/ */
BigInt.prototype.toBuffer = function(){ return this.buffer } BigInt.prototype.toBuffer = function() {
return this.buffer;
BigInt.prototype.toNumber = function(){ };
BigInt.prototype.toNumber = function() {
var b = new Buffer(this.buffer.length); var b = new Buffer(this.buffer.length);
for(var i=0; i<this.buffer.length; i++){ for (var i = 0; i < this.buffer.length; i++) {
b.writeUInt8(this.buffer.readUInt8(this.buffer.length - i - 1), i); b.writeUInt8(this.buffer.readUInt8(this.buffer.length - i - 1), i);
} }
return parseInt(b.toString('hex'), 16); return parseInt(b.toString('hex'), 16);
};
}
/* /*
* COMPARE * COMPARE
*/ */
BigInt.prototype.compare = function(v){ BigInt.prototype.compare = function(v) {
if (!BigInt.isBigInt(v)) {
if(!BigInt.isBigInt(v)) {
v = BigInt.toBigInt(this.buffer.length, v); v = BigInt.toBigInt(this.buffer.length, v);
} }
var n = Math.max(v.buffer.length, this.buffer.length); var n = Math.max(v.buffer.length, this.buffer.length);
if(this.sign > v.sign) return 1; if (this.sign > v.sign) return 1;
if(this.sign < v.sign) return -1; if (this.sign < v.sign) return -1;
for(var i=n-1;i>=0;i--){ for (var i = n - 1; i >= 0; i--) {
var a = ( i<this.buffer.length ? this.buffer.readUInt8(i) : 0 ) var a = i < this.buffer.length ? this.buffer.readUInt8(i) : 0,
, b = ( i<v.buffer.length ? v.buffer.readUInt8(i) : 0 ) b = i < v.buffer.length ? v.buffer.readUInt8(i) : 0;
; if (a != b) {
if(a!=b){
return a > b ? this.sign : -this.sign; return a > b ? this.sign : -this.sign;
} }
} }
return 0; return 0;
};
} BigInt.prototype.lt = function(v) {
BigInt.prototype.lt = function(v){
return this.compare(v) < 0; return this.compare(v) < 0;
} };
BigInt.prototype.le = function(v){ BigInt.prototype.le = function(v) {
return this.compare(v) <= 0; return this.compare(v) <= 0;
} };
BigInt.prototype.gt = function(v){ BigInt.prototype.gt = function(v) {
return this.compare(v) > 0; return this.compare(v) > 0;
} };
BigInt.prototype.ge = function(v){ BigInt.prototype.ge = function(v) {
return this.compare(v) >= 0; return this.compare(v) >= 0;
} };
lib/tools/message.js
View file @ 4b7d8773
var MsErref = require('./ms_erref'),
var MsErref = require('./ms_erref') bigint = require('./bigint');
, bigint = require('./bigint')
;
var defaults = { var defaults = {
successCode: 'STATUS_SUCCESS',
successCode: 'STATUS_SUCCESS' parse: function(connection, cb) {
, parse:function(connection, cb){
var self = this; var self = this;
return function(response){ return function(response) {
var h = response.getHeaders() var h = response.getHeaders(),
, err = MsErref.getStatus(bigint.fromBuffer(h.Status).toNumber()) err = MsErref.getStatus(bigint.fromBuffer(h.Status).toNumber());
; if (err.code == self.successCode) {
if(err.code == self.successCode){
self.onSuccess && self.onSuccess(connection, response); self.onSuccess && self.onSuccess(connection, response);
cb && cb( cb && cb(null, self.parseResponse && self.parseResponse(response));
null
, self.parseResponse && self.parseResponse(response)
);
} else { } else {
var error = new Error(MsErref.getErrorMessage(err)) var error = new Error(MsErref.getErrorMessage(err));
error.code = err.code error.code = err.code;
cb && cb(error); cb && cb(error);
} }
}; };
},
} parseResponse: function(response) {
, parseResponse:function(response){
return response.getResponse(); return response.getResponse();
} },
}; };
module.exports = function(obj){ module.exports = function(obj) {
for (var key in defaults) {
for ( var key in defaults ) {
obj[key] = obj[key] || defaults[key]; obj[key] = obj[key] || defaults[key];
} }
return obj; return obj;
}; };
lib/tools/ms_erref.js
View file @ 4b7d8773
This diff is collapsed.
lib/tools/smb2-connection.js
View file @ 4b7d8773
/* /*
* DEPENDENCIES * DEPENDENCIES
*/ */
var net = require('net') var net = require('net'),
, SMB2Forge = require('./smb2-forge') SMB2Forge = require('./smb2-forge'),
, SMB2Request = SMB2Forge.request SMB2Request = SMB2Forge.request;
;
/* /*
* CONNECTION MANAGER * CONNECTION MANAGER
*/ */
var SMB2Connection = module.exports = {}; var SMB2Connection = (module.exports = {});
/* /*
* CLOSE CONNECTION * CLOSE CONNECTION
*/ */
SMB2Connection.close = function(connection){ SMB2Connection.close = function(connection) {
clearAutoCloseTimeout(connection); clearAutoCloseTimeout(connection);
if(connection.connected){ if (connection.connected) {
connection.connected = false; connection.connected = false;
connection.socket.end(); connection.socket.end();
} }
} };
/* /*
* OPEN CONNECTION * OPEN CONNECTION
*/ */
SMB2Connection.requireConnect = function(method){ SMB2Connection.requireConnect = function(method) {
return function(){ return function() {
var connection = this; var connection = this;
var args = Array.prototype.slice.call(arguments); var args = Array.prototype.slice.call(arguments);
connect(connection, function(err){ connect(
// process the cb connection,
var cb = args.pop(); function(err) {
if (typeof cb !== 'function') { // process the cb
args.push(cb) var cb = args.pop();
cb = function (err) { if (typeof cb !== 'function') {
if (err) { args.push(cb);
if (!err instanceof Error) { cb = function(err) {
err = new Error(String(err)) if (err) {
if (!err instanceof Error) {
err = new Error(String(err));
}
throw err;
} }
throw err };
}
} }
} cb = scheduleAutoClose(connection, cb);
cb = scheduleAutoClose(connection, cb); args.push(cb);
args.push(cb);
// manage the connection error
if(err) cb(err);
else method.apply(connection, args);
});
}
}
// manage the connection error
if (err) cb(err);
else method.apply(connection, args);
}
);
};
};
/* /*
* INIT CONNECTION * INIT CONNECTION
*/ */
SMB2Connection.init = function(connection){ SMB2Connection.init = function(connection) {
// create a socket // create a socket
connection.connected = false; connection.connected = false;
connection.socket = new net.Socket({ connection.socket = new net.Socket({
allowHalfOpen:true allowHalfOpen: true,
}); });
// attach data events to socket // attach data events to socket
connection.socket.on('data', SMB2Forge.response(connection)); connection.socket.on('data', SMB2Forge.response(connection));
connection.errorHandler = []; connection.errorHandler = [];
connection.socket.on('error', function(err){ connection.socket.on('error', function(err) {
if(connection.errorHandler.length > 0){ if (connection.errorHandler.length > 0) {
connection.errorHandler[0].call(null, err) connection.errorHandler[0].call(null, err);
} }
if(connection.debug){ if (connection.debug) {
console.log('-- error'); console.log('-- error');
console.log(arguments); console.log(arguments);
} }
}); });
} };
/* /*
* PRIVATE FUNCTION TO HANDLE CONNECTION * PRIVATE FUNCTION TO HANDLE CONNECTION
*/ */
function connect(connection, cb){ function connect(connection, cb) {
if (connection.connected) {
if(connection.connected){
cb && cb(null); cb && cb(null);
return; return;
} }
...@@ -100,67 +91,69 @@ function connect(connection, cb){ ...@@ -100,67 +91,69 @@ function connect(connection, cb){
cb = scheduleAutoClose(connection, cb); cb = scheduleAutoClose(connection, cb);
// open TCP socket // open TCP socket
connection.socket.connect(connection.port, connection.ip); connection.socket.connect(
connection.port,
connection.ip
);
// SMB2 negotiate connection // SMB2 negotiate connection
SMB2Request('negotiate', {}, connection, function(err){ SMB2Request('negotiate', {}, connection, function(err) {
if(err) cb && cb(err); if (err) cb && cb(err);
// SMB2 setup session / negotiate ntlm // SMB2 setup session / negotiate ntlm
else SMB2Request('session_setup_step1', {}, connection, function(err){ else
if(err) cb && cb(err); SMB2Request('session_setup_step1', {}, connection, function(err) {
// SMB2 setup session / autheticate with ntlm if (err) cb && cb(err);
else SMB2Request('session_setup_step2', {}, connection, function(err){ // SMB2 setup session / autheticate with ntlm
if(err) cb && cb(err); else
// SMB2 tree connect SMB2Request('session_setup_step2', {}, connection, function(err) {
else SMB2Request('tree_connect', {}, connection, function(err){ if (err) cb && cb(err);
if(err) cb && cb(err); // SMB2 tree connect
else { else
connection.connected = true; SMB2Request('tree_connect', {}, connection, function(err) {
cb && cb(null); if (err) cb && cb(err);
} else {
}); connection.connected = true;
cb && cb(null);
}
});
});
}); });
});
}); });
} }
/* /*
* PRIVATE FUNCTION TO HANDLE CLOSING THE CONNECTION * PRIVATE FUNCTION TO HANDLE CLOSING THE CONNECTION
*/ */
function clearAutoCloseTimeout(connection){ function clearAutoCloseTimeout(connection) {
if(connection.scheduledAutoClose){ if (connection.scheduledAutoClose) {
clearTimeout(connection.scheduledAutoClose); clearTimeout(connection.scheduledAutoClose);
connection.scheduledAutoClose = null; connection.scheduledAutoClose = null;
} }
} }
function setAutoCloseTimeout(connection){ function setAutoCloseTimeout(connection) {
clearAutoCloseTimeout(connection); clearAutoCloseTimeout(connection);
if(connection.autoCloseTimeout != 0){ if (connection.autoCloseTimeout != 0) {
connection.scheduledAutoClose = setTimeout(function(){ connection.scheduledAutoClose = setTimeout(function() {
connection.close(); connection.close();
}, connection.autoCloseTimeout); }, connection.autoCloseTimeout);
} }
} }
function scheduleAutoClose(connection, cb){ function scheduleAutoClose(connection, cb) {
addErrorListener(connection, cb); addErrorListener(connection, cb);
clearAutoCloseTimeout(connection); clearAutoCloseTimeout(connection);
return function(){ return function() {
removeErrorListener(connection); removeErrorListener(connection);
setAutoCloseTimeout(connection); setAutoCloseTimeout(connection);
cb.apply(null, arguments); cb.apply(null, arguments);
} };
} }
/* /*
* PRIVATE FUNCTIONS TO HANDLE ERRORS * PRIVATE FUNCTIONS TO HANDLE ERRORS
*/ */
function addErrorListener(connection, callback){ function addErrorListener(connection, callback) {
connection.errorHandler.unshift(callback); connection.errorHandler.unshift(callback);
} }
function removeErrorListener(connection){ function removeErrorListener(connection) {
connection.errorHandler.shift(); connection.errorHandler.shift();
} }
lib/tools/smb2-forge.js
View file @ 4b7d8773
...@@ -3,64 +3,56 @@ ...@@ -3,64 +3,56 @@
*/ */
var SMB2Message = require('./smb2-message'); var SMB2Message = require('./smb2-message');
/* /*
* SMB2 MESSAGE FORGE * SMB2 MESSAGE FORGE
*/ */
var SMB2Forge = module.exports = {}; var SMB2Forge = (module.exports = {});
/* /*
* SMB2 MESSAGE FORGE * SMB2 MESSAGE FORGE
*/ */
SMB2Forge.request = function(messageName, params, connection, cb){ SMB2Forge.request = function(messageName, params, connection, cb) {
var msg = require('../messages/'+messageName) var msg = require('../messages/' + messageName),
, smbMessage = msg.generate(connection, params) smbMessage = msg.generate(connection, params);
;
// send // send
sendNetBiosMessage( sendNetBiosMessage(connection, smbMessage);
connection
, smbMessage
);
// wait for the response // wait for the response
getResponse( getResponse(
connection connection,
, smbMessage.getHeaders().MessageId smbMessage.getHeaders().MessageId,
, msg.parse(connection, cb) msg.parse(connection, cb)
); );
} };
/* /*
* SMB2 RESPONSE MESSAGE PARSER * SMB2 RESPONSE MESSAGE PARSER
*/ */
SMB2Forge.response = function(c){ SMB2Forge.response = function(c) {
c.responses = {}; c.responses = {};
c.responsesCB = {}; c.responsesCB = {};
c.responseBuffer = new Buffer(0); c.responseBuffer = new Buffer(0);
return function(response){ return function(response) {
// concat new response // concat new response
c.responseBuffer = Buffer.concat([c.responseBuffer, response]); c.responseBuffer = Buffer.concat([c.responseBuffer, response]);
// extract complete messages // extract complete messages
var extract = true; var extract = true;
while(extract){ while (extract) {
extract = false; extract = false;
// has a message header // has a message header
if(c.responseBuffer.length >= 4) { if (c.responseBuffer.length >= 4) {
// message is complete // message is complete
var msgLength = (c.responseBuffer.readUInt8(1) << 16) + c.responseBuffer.readUInt16BE(2); var msgLength =
if(c.responseBuffer.length >= msgLength + 4) { (c.responseBuffer.readUInt8(1) << 16) +
c.responseBuffer.readUInt16BE(2);
if (c.responseBuffer.length >= msgLength + 4) {
// set the flags // set the flags
extract = true; extract = true;
// parse message // parse message
var r = c.responseBuffer.slice(4, msgLength+4) var r = c.responseBuffer.slice(4, msgLength + 4),
, message = new SMB2Message() message = new SMB2Message();
;
message.parseBuffer(r); message.parseBuffer(r);
//debug //debug
if(c.debug){ if (c.debug) {
console.log('--response'); console.log('--response');
console.log(r.toString('hex')); console.log(r.toString('hex'));
} }
...@@ -68,22 +60,19 @@ SMB2Forge.response = function(c){ ...@@ -68,22 +60,19 @@ SMB2Forge.response = function(c){
var mId = message.getHeaders().MessageId.toString('hex'); var mId = message.getHeaders().MessageId.toString('hex');
// check if the message can be dispatched // check if the message can be dispatched
// or store it // or store it
if(c.responsesCB[mId]) { if (c.responsesCB[mId]) {
c.responsesCB[mId](message); c.responsesCB[mId](message);
delete c.responsesCB[mId]; delete c.responsesCB[mId];
} else { } else {
c.responses[mId] = message; c.responses[mId] = message;
} }
// remove from response buffer // remove from response buffer
c.responseBuffer = c.responseBuffer.slice(msgLength+4); c.responseBuffer = c.responseBuffer.slice(msgLength + 4);
} }
} }
} }
} };
} };
/* /*
* HELPERS * HELPERS
...@@ -91,39 +80,36 @@ SMB2Forge.response = function(c){ ...@@ -91,39 +80,36 @@ SMB2Forge.response = function(c){
function sendNetBiosMessage(connection, message) { function sendNetBiosMessage(connection, message) {
var smbRequest = message.getBuffer(connection); var smbRequest = message.getBuffer(connection);
if(connection.debug){ if (connection.debug) {
console.log('--request'); console.log('--request');
console.log(smbRequest.toString('hex')); console.log(smbRequest.toString('hex'));
} }
// create NetBios package // create NetBios package
var buffer = new Buffer(smbRequest.length+4); var buffer = new Buffer(smbRequest.length + 4);
// write NetBios cmd // write NetBios cmd
buffer.writeUInt8(0x00, 0); buffer.writeUInt8(0x00, 0);
// write message length // write message length
buffer.writeUInt8((0xFF0000 & smbRequest.length) >> 16, 1); buffer.writeUInt8((0xff0000 & smbRequest.length) >> 16, 1);
buffer.writeUInt16BE(0xFFFF & smbRequest.length, 2); buffer.writeUInt16BE(0xffff & smbRequest.length, 2);
// write message content // write message content
smbRequest.copy(buffer, 4, 0, smbRequest.length); smbRequest.copy(buffer, 4, 0, smbRequest.length);
// Send it !!! // Send it !!!
connection.newResponse = false; connection.newResponse = false;
connection.socket.write(buffer); connection.socket.write(buffer);
return true; return true;
} }
function getResponse(c, mId, cb) { function getResponse(c, mId, cb) {
var messageId = new Buffer(4); var messageId = new Buffer(4);
messageId.writeUInt32LE(mId, 0); messageId.writeUInt32LE(mId, 0);
messageId = messageId.toString('hex'); messageId = messageId.toString('hex');
if(c.responses[messageId]) { if (c.responses[messageId]) {
cb(c.responses[messageId]); cb(c.responses[messageId]);
delete c.responses[messageId]; delete c.responses[messageId];
} else { } else {
......
lib/tools/smb2-message.js
View file @ 4b7d8773
This diff is collapsed.
package.json
View file @ 4b7d8773
...@@ -39,6 +39,7 @@ ...@@ -39,6 +39,7 @@
"devDependencies": { "devDependencies": {
"babel": "^5.8.34", "babel": "^5.8.34",
"babel-eslint": "^4.1.6", "babel-eslint": "^4.1.6",
"prettier": "^1.13.7",
"source-map-support": "^0.4.0", "source-map-support": "^0.4.0",
"standard": "^5.4.1" "standard": "^5.4.1"
}, },
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment