@awo00/smb2
Version:
A SMB2 implementation in TypeScript
84 lines (83 loc) • 3.1 kB
JavaScript
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || (function () {
var ownKeys = function(o) {
ownKeys = Object.getOwnPropertyNames || function (o) {
var ar = [];
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
return ar;
};
return ownKeys(o);
};
return function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
__setModuleDefault(result, mod);
return result;
};
})();
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const events_1 = require("events");
const SmbRequest_1 = __importDefault(require("./SmbRequest"));
const Smb2Request_1 = __importDefault(require("./Smb2Request"));
const Packet_1 = __importDefault(require("../protocol/Packet"));
const protocolIds = __importStar(require("../protocol/protocolIds"));
class Client extends events_1.EventEmitter {
server;
socket;
restChunk;
targetDialect;
targetDialectName;
constructor(server, socket) {
super();
this.server = server;
this.socket = socket;
}
setup() {
this.socket.setNoDelay(true);
this.socket.addListener("data", this.onData);
}
onData = (buffer) => {
if (this.restChunk) {
buffer = Buffer.concat([this.restChunk, buffer]);
this.restChunk = undefined;
}
const { chunks, restChunk } = Packet_1.default.getChunks(buffer);
this.restChunk = restChunk;
for (const chunk of chunks) {
const protocolId = Packet_1.default.parseProtocolId(chunk);
if (protocolId === protocolIds.smb) {
const request = SmbRequest_1.default.parse(chunk);
this.emit("request", request);
}
else {
const request = Smb2Request_1.default.parse(chunk);
this.emit("request", request);
}
}
};
send(response) {
const buffer = response.serialize();
this.socket.write(buffer);
}
}
exports.default = Client;
;