briar/test/net/sf/briar/transport/NullOutgoingEncryptionLayer.java

package net.sf.briar.transport;

import static net.sf.briar.api.transport.TransportConstants.HEADER_LENGTH;
import static net.sf.briar.api.transport.TransportConstants.MAC_LENGTH;

import java.io.IOException;
import java.io.OutputStream;

/** An encryption layer that performs no encryption. */
class NullOutgoingEncryptionLayer implements FrameWriter {

	private final OutputStream out;

	private long capacity;

	NullOutgoingEncryptionLayer(OutputStream out) {
		this.out = out;
		capacity = Long.MAX_VALUE;
	}

	NullOutgoingEncryptionLayer(OutputStream out, long capacity) {
		this.out = out;
		this.capacity = capacity;
	}

	public void writeFrame(byte[] frame) throws IOException {
		int payload = HeaderEncoder.getPayloadLength(frame);
		int padding = HeaderEncoder.getPaddingLength(frame);
		int length = HEADER_LENGTH + payload + padding + MAC_LENGTH;
		out.write(frame, 0, length);
		capacity -= length;
	}

	public void flush() throws IOException {
		out.flush();
	}

	public long getRemainingCapacity() {
		return capacity;
	}
}