// TODO: when no rewind - stop thread
package kg.apc.jmeter.modifiers;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.nio.ByteBuffer;
import java.nio.channels.FileChannel;
import java.nio.charset.Charset;
import kg.apc.jmeter.EndOfFileException;
import kg.apc.jmeter.RuntimeEOFException;
import org.apache.commons.io.IOExceptionWithCause;
import org.apache.jmeter.engine.util.NoThreadClone;
import org.apache.jmeter.processor.PreProcessor;
import org.apache.jmeter.testelement.AbstractTestElement;
import org.apache.jmeter.testelement.TestStateListener;
import org.apache.jmeter.testelement.property.BooleanProperty;
import org.apache.jmeter.threads.JMeterContextService;
import org.apache.jmeter.threads.JMeterVariables;
import org.apache.jorphan.logging.LoggingManager;
import org.apache.jorphan.util.JOrphanUtils;
import org.apache.log.Logger;
public class RawRequestSourcePreProcessor
extends AbstractTestElement
implements PreProcessor, NoThreadClone, TestStateListener {
public static final String regexp = "\\s";
private static final Logger log = LoggingManager.getLoggerForClass();
public static final String VARIABLE_NAME = "variable_name";
public static final String FILENAME = "filename";
public static final String REWIND = "rewind";
public static final String ENCODE_HEX = "isHex";
private FileChannel file;
private ByteBuffer metaBuf = null;
private ByteBuffer oneByte = null;
public static final Charset binaryCharset = Charset.forName("UTF8");
public RawRequestSourcePreProcessor() {
super();
}
@Override
public synchronized void process() {
if (file == null) {
log.info("Creating file object: " + getFileName());
try {
file = new FileInputStream(getFileName()).getChannel();
} catch (FileNotFoundException ex) {
log.error(getFileName(), ex);
return;
}
}
String rawData;
try {
rawData = readNextChunk(getNextChunkSize());
} catch (EndOfFileException ex) {
if (getRewindOnEOF()) {
if (log.isDebugEnabled()) {
log.debug("Rewind file");
}
try {
file.position(0);
} catch (IOException ex1) {
log.error("Cannot rewind", ex1);
}
process();
return;
} else {
log.info("End of file reached: " + getFileName());
if (JMeterContextService.getContext().getThread() != null) {
JMeterContextService.getContext().getThread().stop();
}
throw new RuntimeEOFException("End of file reached", ex);
}
} catch (IOException ex) {
log.error("Error reading next chunk", ex);
throw new RuntimeException("Error reading next chunk", ex);
}
final JMeterVariables vars = JMeterContextService.getContext().getVariables();
if (vars != null) {
vars.put(getVarName(), rawData);
}
}
private synchronized String readNextChunk(int capacity)
throws IOException {
if (capacity == 0) {
throw new EndOfFileException("Zero chunk size, possibly end of file reached.");
}
ByteBuffer buf = ByteBuffer.allocateDirect(capacity);
byte[] dst = new byte[capacity];
int cnt = file.read(buf);
//log.debug("Read " + cnt);
if (cnt != capacity) {
throw new IOException("Expected chunk size (" + capacity + ") differs from read bytes count (" + cnt + ")");
}
buf.flip();
buf.get(dst);
if (log.isDebugEnabled()) {
log.debug("Chunk : " + new String(dst));
}
if (isHexEncode()) {
return JOrphanUtils.baToHexString(dst);
} else {
return new String(dst, binaryCharset);
}
}
private int getNextChunkSize() throws IOException {
metaBuf.clear();
while (true) {
byte b = getOneByte();
if (b == 10 || b == 13) {
// if we have \r\n then skip \n
byte b2 = getOneByte();
if (b2 != 10) {
file.position(file.position() - 1);
}
// ignore newlines before length marker
if (metaBuf.position() > 0) {
break;
}
} else {
//if (log.isDebugEnabled()) log.debug("Read byte: "+b);
metaBuf.put(b);
}
}
//if (log.isDebugEnabled()) log.debug("Meta line: "+JMeterPluginsUtils.byteBufferToString(metaBuf));
byte[] bLine = new byte[metaBuf.position()];
metaBuf.rewind();
metaBuf.get(bLine);
String sLine = new String(bLine).trim();
String[] ar = sLine.split(regexp);
if (log.isDebugEnabled()) {
log.debug("Chunk size: " + ar[0]);
}
int res = 0;
try {
res = Integer.parseInt(ar[0]);
} catch (NumberFormatException ex) {
log.error("Error reading chunk size near: " + sLine, ex);
throw new IOExceptionWithCause(ex);
}
return res;
}
private byte getOneByte() throws IOException {
oneByte.rewind();
if (file.read(oneByte) < 1) {
throw new EndOfFileException(getFileName());
}
return oneByte.get(0);
}
public String getVarName() {
return getPropertyAsString(VARIABLE_NAME);
}
public void setVarName(String name) {
setProperty(VARIABLE_NAME, name);
}
public String getFileName() {
return getPropertyAsString(FILENAME);
}
public void setFileName(String filename) {
setProperty(FILENAME, filename);
file = null;
}
public void setRewindOnEOF(boolean isRew) {
setProperty(new BooleanProperty(REWIND, isRew));
}
public boolean getRewindOnEOF() {
return getPropertyAsBoolean(REWIND);
}
public boolean isHexEncode() {
return getPropertyAsBoolean(ENCODE_HEX);
}
public void setEncodeHex(boolean b) {
setProperty(ENCODE_HEX, b);
}
@Override
public void testStarted() {
testStarted("");
}
@Override
public void testStarted(String host) {
// performance concerns and copying ability
metaBuf=ByteBuffer.allocateDirect(1024);
oneByte=ByteBuffer.allocateDirect(1);
}
@Override
public void testEnded() {
}
@Override
public void testEnded(String host) {
}
}