@sharelist/node-smb-server
Version:
Sharelist SMB Server Implementation Base On node-smb-server
203 lines (181 loc) • 7.26 kB
JavaScript
/*
* Copyright 2015 Adobe Systems Incorporated. All rights reserved.
* This file is licensed to you under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License. You may obtain a copy
* of the License at http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software distributed under
* the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR REPRESENTATIONS
* OF ANY KIND, either express or implied. See the License for the specific language
* governing permissions and limitations under the License.
*/
'use strict';
var path = require('path');
var fs = require('fs');
var put = require('put');
var binary = require('binary');
var logger = require('winston').loggers.get('smb');
var _ = require('lodash');
var ntstatus = require('../../ntstatus');
var SMB = require('../constants');
var utils = require('../../utils');
var subCmdHandlers = {};
function loadSubCmdHandlers() {
var p = path.join(__dirname, 'trans');
var files = fs.readdirSync(p);
for (var i = 0; i < files.length; i++) {
var f = files[i];
var stat = fs.statSync(path.resolve(p, f));
if (stat.isDirectory()) {
continue;
}
if (f.substr(-3) === '.js') {
f = f.slice(0, -3);
subCmdHandlers[f] = require(path.resolve(p, f));
}
}
}
loadSubCmdHandlers();
/**
* SMB_COM_TRANSACTION (0x25):
* This command serves as the transport for the Transaction Subprotocol Commands.
* These commands operate on mailslots and named pipes, which are interprocess
* communication endpoints within the CIFS file system.
*
* @param {Object} msg - an SMB message object
* @param {Number} commandId - the command id
* @param {Buffer} commandParams - the command parameters
* @param {Buffer} commandData - the command data
* @param {Number} commandParamsOffset - the command parameters offset within the SMB
* @param {Number} commandDataOffset - the command data offset within the SMB
* @param {Object} connection - an SMBConnection instance
* @param {Object} server - an SMBServer instance
* @param {Function} cb callback called with the command's result
* @param {Object} cb.result - an object with the command's result params and data
* or null if the handler already sent the response and
* no further processing is required by the caller
* @param {Number} cb.result.status
* @param {Buffer} cb.result.params
* @param {Buffer} cb.result.data
*/
function handle(msg, commandId, commandParams, commandData, commandParamsOffset, commandDataOffset, connection, server, cb) {
// decode params
var parser = binary.parse(commandParams);
var paramsObj = parser.word16le('totalParameterCount') // bytes (not words)
.word16le('totalDataCount')
.word16le('maxParameterCount') // bytes (not words)
.word16le('maxDataCount')
.word8le('maxSetupCount')
.skip(1) // reserved1
.word16le('flags')
.word32le('timeout')
.skip(2) // reserved2
.word16le('parameterCount') // bytes (not words)
.word16le('parameterOffset')
.word16le('dataCount')
.word16le('dataOffset')
.word8le('setupCount')
.skip(1) // reserved3
.buffer('setup', 2 * parser.vars['setupCount'])
.vars;
_.assign(msg, paramsObj);
var result;
msg.subCommandId = msg.setup.readUInt16LE(0);
var subCommand = SMB.TRANS_SUBCOMMAND_TO_STRING[msg.subCommandId];
if (!subCommand) {
logger.error('encountered invalid subcommand 0x%s', msg.subCommandId.toString(16));
result = {
status: ntstatus.STATUS_SMB_BAD_COMMAND,
params: utils.EMPTY_BUFFER,
data: utils.EMPTY_BUFFER
};
process.nextTick(function () { cb(result); });
return;
}
// decode data
var off = 0;
msg.name = utils.extractUnicodeBytes(commandData, off); // not used in transaction
off += 1;
var subParams = msg.buf.slice(msg.parameterOffset, msg.parameterOffset + msg.parameterCount);
var subData = msg.buf.slice(msg.dataOffset, msg.dataOffset + msg.dataCount);
logger.debug('[%s][%s] totalParameterCount: %d, parameterCount: %d, totalDataCount: %d, dataCount: %d, flags: %s', SMB.COMMAND_TO_STRING[commandId].toUpperCase(), subCommand.toUpperCase(), msg.totalParameterCount, msg.parameterCount, msg.totalDataCount, msg.dataCount, msg.flags.toString(2));
if (msg.parameterCount < msg.totalParameterCount || msg.dataCount < msg.totalDataCount) {
// todo support transaction_secondary messages (reassembling chunked transaction messages)
logger.error('chunked transaction messages are not yet supported.');
result = {
status: ntstatus.STATUS_NOT_IMPLEMENTED,
params: utils.EMPTY_BUFFER,
data: utils.EMPTY_BUFFER
};
process.nextTick(function () { cb(result); });
return;
}
// invoke subcommand handler
var handler = subCmdHandlers[subCommand];
if (handler) {
handler(msg, msg.subCommandId, subParams, subData, msg.parameterOffset, msg.dataOffset, connection, server, function (result) {
if (result.status !== ntstatus.STATUS_SUCCESS) {
result.params = commandParams;
result.data = commandData;
cb(result);
return;
}
// build response
var subParams = result.params;
var subData = result.data;
var setup = result.setup || utils.EMPTY_BUFFER;
// transaction response have a params length of 20 (10 words) plus length of setup
var paramsLength = 20 + setup.length;
var dataOffset = SMB.SMB_MIN_LENGTH + paramsLength;
// calculate offsets and paddings
var subParamsOffset, subDataOffset, pad1, pad2;
var off = dataOffset;
pad1 = utils.calculatePadLength(off, 4);
off += pad1;
subParamsOffset = off;
off += subParams.length;
pad2 = utils.calculatePadLength(off, 4);
off += pad2;
subDataOffset = off;
off += subData.length;
// params
var out = put();
out.word16le(subParams.length) // TotalParameterCount
.word16le(subData.length) // TotalDataCount
.word16le(0) // Reserved1
.word16le(subParams.length) // ParameterCount
.word16le(subParamsOffset) // ParameterOffset
.word16le(0) // ParameterDisplacement
.word16le(subData.length) // DataCount
.word16le(subDataOffset) // DataOffset
.word16le(0) // DataDisplacement
.word8(setup.length ? setup.length / 2 : 0) // SetupCount
.word8(0) // Reserved2
.put(setup); // Setup
var params = out.buffer();
// data
out = put();
out.pad(pad1)
.put(subParams)
.pad(pad2)
.put(subData);
var data = out.buffer();
// return result
result = {
status: ntstatus.STATUS_SUCCESS,
params: params,
data: data
};
cb(result);
});
} else {
logger.error('encountered unsupported subcommand 0x%s \'%s\'', msg.subCommandId.toString(16), subCommand.toUpperCase());
result = {
status: ntstatus.STATUS_NOT_IMPLEMENTED,
params: utils.EMPTY_BUFFER,
data: utils.EMPTY_BUFFER
};
cb(result);
}
}
module.exports = handle;