LockAndReadParser.java

package org.krakenapps.pcap.decoder.smb.comparser;
import org.krakenapps.pcap.decoder.smb.SmbSession;
import org.krakenapps.pcap.decoder.smb.request.LockAndReadRequest;
import org.krakenapps.pcap.decoder.smb.response.LockAndReadResponse;
import org.krakenapps.pcap.decoder.smb.structure.SmbData;
import org.krakenapps.pcap.decoder.smb.structure.SmbHeader;
import org.krakenapps.pcap.util.Buffer;
import org.krakenapps.pcap.util.ByteOrderConverter;

public class LockAndReadParser implements SmbDataParser{

	@Override
	public SmbData parseRequest(SmbHeader h ,Buffer b , SmbSession session) {
		LockAndReadRequest data = new LockAndReadRequest();
		data.setWordCount(b.get());
		if(data.getWordCount() == 0x05){
			data.setFid(ByteOrderConverter.swap(b.getShort()));
			data.setCountOfBytesToRead(ByteOrderConverter.swap(b.getShort()));
			data.setReadOffsetInBytes(ByteOrderConverter.swap(b.getInt()));
			data.setEstimateOfRemainingBytesToBeRead(ByteOrderConverter.swap(b.getShort()));
		}
		else{
			data.setMalformed(true);
		}
		data.setByteCount(ByteOrderConverter.swap(b.getShort()));
		return data;
	}
	@Override
	public SmbData parseResponse(SmbHeader h,Buffer b ,SmbSession session) {
		LockAndReadResponse data = new LockAndReadResponse();
		byte []reserved = new byte[8];
		byte []bytes;
		data.setWordCount(b.get());
		if(data.getWordCount() == 0x05){
			data.setCountofBytesReturned(ByteOrderConverter.swap(b.getShort()));
			b.gets(reserved);
			data.setReserved(reserved);
		}
		else{
			data.setMalformed(true);
		}
		data.setByteCount(ByteOrderConverter.swap(b.getShort()));
		if(b.readableBytes() != data.getByteCount()){
			data.setMalformed(true);
			return data;
		}
		else if(b.readableBytes() == 0){
			data.setMalformed(true);
			return data;
		}
		data.setBufferType(b.get());
		data.setCountOfBytesRead(ByteOrderConverter.swap(b.getShort()));
		bytes = new byte[data.getCountOfBytesRead()];
		b.gets(bytes);
		data.setBytes(bytes);
		return data;
	}
}