ff8674bca9
* Reimplement the I2NP reading with less temporary memory allocation.
There is still significant GC churn, especially under load, but this
should help.
* Catch some oddball errors in the transport (message timeout while
establishing).
109 lines
4.1 KiB
Java
109 lines
4.1 KiB
Java
package net.i2p.data.i2np;
|
|
/*
|
|
* free (adj.): unencumbered; not under the control of others
|
|
* Written by jrandom in 2003 and released into the public domain
|
|
* with no warranty of any kind, either expressed or implied.
|
|
* It probably won't make your computer catch on fire, or eat
|
|
* your children, but it might. Use at your own risk.
|
|
*
|
|
*/
|
|
|
|
import java.io.FileInputStream;
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
|
|
import net.i2p.I2PAppContext;
|
|
import net.i2p.data.DataFormatException;
|
|
import net.i2p.data.DataHelper;
|
|
import net.i2p.util.Log;
|
|
|
|
/**
|
|
* Handle messages from router to router
|
|
*
|
|
*/
|
|
public class I2NPMessageHandler {
|
|
private Log _log;
|
|
private I2PAppContext _context;
|
|
private long _lastReadBegin;
|
|
private long _lastReadEnd;
|
|
private byte _messageBuffer[];
|
|
public I2NPMessageHandler(I2PAppContext context) {
|
|
_context = context;
|
|
_log = context.logManager().getLog(I2NPMessageHandler.class);
|
|
_messageBuffer = null;
|
|
}
|
|
|
|
/**
|
|
* Read an I2NPMessage from the stream and return the fully populated object.
|
|
*
|
|
* @throws IOException if there is an IO problem reading from the stream
|
|
* @throws I2NPMessageException if there is a problem handling the particular
|
|
* message - if it is an unknown type or has improper formatting, etc.
|
|
*/
|
|
public I2NPMessage readMessage(InputStream in) throws IOException, I2NPMessageException {
|
|
if (_messageBuffer == null) _messageBuffer = new byte[38*1024]; // more than necessary
|
|
try {
|
|
int type = (int)DataHelper.readLong(in, 1);
|
|
_lastReadBegin = System.currentTimeMillis();
|
|
I2NPMessage msg = createMessage(type);
|
|
if (msg == null)
|
|
throw new I2NPMessageException("The type "+ type + " is an unknown I2NP message");
|
|
try {
|
|
msg.readBytes(in, type, _messageBuffer);
|
|
} catch (IOException ioe) {
|
|
throw ioe;
|
|
} catch (I2NPMessageException ime) {
|
|
throw ime;
|
|
} catch (Exception e) {
|
|
if (_log.shouldLog(Log.WARN))
|
|
_log.warn("Error reading the stream", e);
|
|
throw new IOException("Unknown error reading the " + msg.getClass().getName()
|
|
+ ": " + e.getMessage());
|
|
}
|
|
_lastReadEnd = System.currentTimeMillis();
|
|
return msg;
|
|
} catch (DataFormatException dfe) {
|
|
throw new I2NPMessageException("Error reading the message", dfe);
|
|
}
|
|
}
|
|
public long getLastReadTime() { return _lastReadEnd - _lastReadBegin; }
|
|
|
|
/**
|
|
* Yes, this is fairly ugly, but its the only place it ever happens.
|
|
*
|
|
*/
|
|
private I2NPMessage createMessage(int type) throws I2NPMessageException {
|
|
switch (type) {
|
|
case DatabaseStoreMessage.MESSAGE_TYPE:
|
|
return new DatabaseStoreMessage(_context);
|
|
case DatabaseLookupMessage.MESSAGE_TYPE:
|
|
return new DatabaseLookupMessage(_context);
|
|
case DatabaseSearchReplyMessage.MESSAGE_TYPE:
|
|
return new DatabaseSearchReplyMessage(_context);
|
|
case DeliveryStatusMessage.MESSAGE_TYPE:
|
|
return new DeliveryStatusMessage(_context);
|
|
case GarlicMessage.MESSAGE_TYPE:
|
|
return new GarlicMessage(_context);
|
|
case TunnelMessage.MESSAGE_TYPE:
|
|
return new TunnelMessage(_context);
|
|
case DataMessage.MESSAGE_TYPE:
|
|
return new DataMessage(_context);
|
|
case TunnelCreateMessage.MESSAGE_TYPE:
|
|
return new TunnelCreateMessage(_context);
|
|
case TunnelCreateStatusMessage.MESSAGE_TYPE:
|
|
return new TunnelCreateStatusMessage(_context);
|
|
default:
|
|
return null;
|
|
}
|
|
}
|
|
|
|
public static void main(String args[]) {
|
|
try {
|
|
I2NPMessage msg = new I2NPMessageHandler(I2PAppContext.getGlobalContext()).readMessage(new FileInputStream(args[0]));
|
|
System.out.println(msg);
|
|
} catch (Exception e) {
|
|
e.printStackTrace();
|
|
}
|
|
}
|
|
}
|