public static int sum(TcpPacket s) {
ByteBuffer buf = build(s);
Buffer data = s.getData();
int length = buf.limit();
if (data != null) length += data.readableBytes();
short[] checksumBytes = new short[length / 2];
int i = 0;
for (; i < checksumBytes.length; i++) {
checksumBytes[i] = buf.getShort();
}
/* add payload to checksum */
if (data != null) {
while (data.isEOB()) {
// TODO: buffer underflow handling (padding)
checksumBytes[i++] = data.getShort();
}
}
if (data != null) data.rewind();
return Checksum.sum(checksumBytes);
}
@Override
public SmbData parseResponse(SmbHeader h, Buffer b, SmbSession session) {
WritePrintFileResponse data = new WritePrintFileResponse();
data.setWordCount(b.get());
data.setByteCount(ByteOrderConverter.swap(b.getShort()));
return data;
}
@Override
public SmbData parseRequest(SmbHeader h, Buffer b, SmbSession session) {
WritePrintFileRequest data = new WritePrintFileRequest();
byte[] buff;
data.setWordCount(b.get());
data.setFid(ByteOrderConverter.swap(b.getShort()));
data.setByteCount(ByteOrderConverter.swap(b.getShort()));
if (b.readableBytes() != data.getByteCount()) {
data.setMalformed(true);
return data;
}
data.setBufferFormat(b.get());
data.setDataLength(ByteOrderConverter.swap(b.getShort()));
buff = new byte[data.getDataLength()];
b.gets(buff);
data.setData(buff);
return data;
}
@Override
public SmbData parseResponse(SmbHeader h, Buffer b, SmbSession session) {
Transaction2Response data = new Transaction2Response();
Trans2SubcommandMapper mapper = new Trans2SubcommandMapper();
byte[] setup;
byte[] pad1;
byte[] pad2;
byte[] parameters;
byte[] buff;
data.setWordCount(b.get());
if (data.getWordCount() != 0x00) {
data.setTotalParameterCount(ByteOrderConverter.swap(b.getShort()));
data.setTotalDataCount(ByteOrderConverter.swap(b.getShort()));
data.setReserved1(ByteOrderConverter.swap(b.getShort()));
data.setParameterCount(ByteOrderConverter.swap(b.getShort()));
data.setParameterOffset(ByteOrderConverter.swap(b.getShort()));
data.setParameterDisplacement(ByteOrderConverter.swap(b.getShort()));
data.setDataCount(ByteOrderConverter.swap(b.getShort()));
data.setDataOffset(ByteOrderConverter.swap(b.getShort()));
data.setDataDisplacement(ByteOrderConverter.swap(b.getShort()));
data.setSetupCount(b.get());
data.setResreved2(b.get());
setup = new byte[data.getSetupCount() * 2];
b.gets(setup);
data.setSetup(setup);
} // it final response
data.setByteCount(ByteOrderConverter.swap(b.getShort()));
if (b.readableBytes() != data.getByteCount()) {
data.setMalformed(true);
return data;
}
if (data.getByteCount() != 0x00) {
if (data.getParameterCount() != 0) {
pad1 = new byte[data.getParameterOffset() - b.position()];
if (pad1.length != 0) {
b.gets(pad1);
data.setPad1(pad1);
}
parameters = new byte[data.getParameterCount()];
b.gets(parameters);
data.setTrans2Parameter(parameters);
}
if (data.getDataCount() != 0) {
pad2 = new byte[data.getDataOffset() - b.position()];
if (pad2.length != 0) {
b.gets(pad2);
data.setPad2(pad2);
}
buff = new byte[data.getDataCount()];
b.gets(buff);
data.setTrans2Data(buff);
}
}
Buffer setupBuffer = new ChainBuffer();
Buffer parameterBuffer = new ChainBuffer();
Buffer dataBuffer = new ChainBuffer();
setupBuffer.addLast(data.getSetup());
parameterBuffer.addLast(data.getTrans2Parameter());
dataBuffer.addLast(data.getTrans2Data());
if (setupBuffer.readableBytes() + parameterBuffer.readableBytes() + dataBuffer.readableBytes()
== 0) {
return data;
}
// setupBuffer.mark();
TransParser parser = mapper.getParser(session.getSessionTrans2Command());
if (parser == null) {
return data;
}
// add buffer;
if (data.getTotalDataCount() != dataBuffer.readableBytes()) {
if (session.getTransBuffer().readableBytes() == 0) {
session.setParameterBuffer(parameterBuffer);
session.setTransBuffer(dataBuffer);
} else if (session.getTransBuffer().readableBytes() != 0) {
session.addTransBuffer(dataBuffer);
if (data.getTotalDataCount() == session.getTransBuffer().readableBytes()) {
data.setTransaction2Data(
parser.parseResponse(
setupBuffer, session.getParameterBuffer(), session.getTransBuffer(), session));
session.getTransBuffer().skip(session.getTransBuffer().readableBytes());
session.getParameterBuffer().discardReadBytes();
}
}
return data;
} else {
data.setTransaction2Data(
parser.parseResponse(setupBuffer, parameterBuffer, dataBuffer, session));
}
return data;
}
@Override
public SmbData parseRequest(SmbHeader h, Buffer b, SmbSession session) {
Transaction2Request data = new Transaction2Request();
Trans2SubcommandMapper mapper = new Trans2SubcommandMapper();
byte[] setup;
byte[] pad1;
byte[] pad2;
byte[] parameters;
byte[] buff;
data.setWordCount(b.get());
if (data.getWordCount() >= 0x0E) {
data.setTotalParameterCount(ByteOrderConverter.swap(b.getShort()));
data.setTotalDataCount(ByteOrderConverter.swap(b.getShort()));
data.setMaxParameterCount(ByteOrderConverter.swap(b.getShort()));
data.setMaxDataCount(ByteOrderConverter.swap(b.getShort()));
data.setMaxSetupCount(b.get());
data.setReserved1(b.get());
data.setFlags(ByteOrderConverter.swap(b.getShort()));
data.setTimeout(ByteOrderConverter.swap(b.getInt()));
data.setReserved2(ByteOrderConverter.swap(b.getShort()));
data.setParameterCount(ByteOrderConverter.swap(b.getShort()));
data.setParameterOffset(ByteOrderConverter.swap(b.getShort()));
data.setDataCount(ByteOrderConverter.swap(b.getShort()));
data.setDataOffset(ByteOrderConverter.swap(b.getShort()));
data.setSetupCount(b.get());
data.setReserved3(b.get());
setup = new byte[data.getSetupCount() * 2];
b.gets(setup);
data.setSetup(setup);
} else {
data.setMalformed(true);
b.skip(data.getWordCount() * 2);
}
data.setByteCount(ByteOrderConverter.swap(b.getShort()));
if (b.readableBytes() != data.getByteCount()) {
data.setMalformed(true);
return data;
} else if (data.isMalformed()) {
return data;
}
data.setName(NetBiosNameCodec.readOemName(b));
if (data.getParameterCount() != 0) {
pad1 = new byte[data.getParameterOffset() - b.position()];
if (pad1.length != 0) {
b.gets(pad1);
data.setPad1(pad1);
}
parameters = new byte[data.getParameterCount()];
b.gets(parameters);
data.setTrans2Parameters(parameters);
}
if (data.getDataCount() != 0) {
pad2 = new byte[data.getDataOffset() - b.position()];
if (pad2.length != 0) {
b.gets(pad2);
data.setPad2(pad2);
}
buff = new byte[data.getDataCount()];
b.gets(buff);
data.setTrans2Data(buff);
}
Buffer setupBuffer = new ChainBuffer();
Buffer parameterBuffer = new ChainBuffer();
Buffer dataBuffer = new ChainBuffer();
setupBuffer.addLast(data.getSetup());
parameterBuffer.addLast(data.getTrans2Parameters());
dataBuffer.addLast(data.getTrans2Data());
setupBuffer.mark();
TransParser parser =
mapper.getParser(
Transaction2Command.parse(ByteOrderConverter.swap(setupBuffer.getShort())));
setupBuffer.reset();
data.setTransaction2Data(parser.parseRequest(setupBuffer, parameterBuffer, dataBuffer));
return data;
}