echelon/i2p.i2p
1
0
Fork 0
forked from I2P_Developers/i2p.i2p
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Data: Check sooner for unknown sig type;

Browse Source 
minor efficiency improvements in parsing
This commit is contained in:
zzz
2018-08-04 13:41:01 +00:00
parent c0be83fe08
commit f1df49606d
4 changed files with 29 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
core/java/src/net/i2p/data/Certificate.java
View File

@ -89,7 +89,8 @@ public class Certificate extends DataStructureImpl {
* @since 0.8.3
*/
public static Certificate create(InputStream in) throws DataFormatException, IOException {
int type = (int) DataHelper.readLong(in, 1);
// EOF will be thrown in next read
int type = in.read();
int length = (int) DataHelper.readLong(in, 2);
if (type == 0 && length == 0)
return NULL_CERT;
@ -161,7 +162,8 @@ public class Certificate extends DataStructureImpl {
public void readBytes(InputStream in) throws DataFormatException, IOException {
if (_type != 0 || _payload != null)
throw new IllegalStateException("already set");
_type = (int) DataHelper.readLong(in, 1);
// EOF will be thrown in next read
_type = in.read();
int length = (int) DataHelper.readLong(in, 2);
if (length > 0) {
_payload = new byte[length];

13
core/java/src/net/i2p/data/LeaseSet.java
View File

@ -19,6 +19,7 @@ import java.util.List;
import net.i2p.I2PAppContext;
import net.i2p.crypto.DSAEngine;
import net.i2p.crypto.SigType;
import net.i2p.util.Clock;
import net.i2p.util.Log;
import net.i2p.util.RandomSource;
@ -327,9 +328,15 @@ public class LeaseSet extends DatabaseEntry {
_destination = Destination.create(in);
_encryptionKey = PublicKey.create(in);
// revocation signing key must be same type as the destination signing key
_signingKey = new SigningPublicKey(_destination.getSigningPublicKey().getType());
SigType type = _destination.getSigningPublicKey().getType();
// Even if not verifying, we have to construct a Signature object
// below, which will fail for null type.
if (type == null)
throw new DataFormatException("unknown sig type");
_signingKey = new SigningPublicKey(type);
// EOF will be thrown in signature read below
_signingKey.readBytes(in);
int numLeases = (int) DataHelper.readLong(in, 1);
int numLeases = in.read();
if (numLeases > MAX_LEASES)
throw new DataFormatException("Too many leases - max is " + MAX_LEASES);
//_version = DataHelper.readLong(in, 4);
@ -339,7 +346,7 @@ public class LeaseSet extends DatabaseEntry {
addLease(lease);
}
// signature must be same type as the destination signing key
_signature = new Signature(_destination.getSigningPublicKey().getType());
_signature = new Signature(type);
_signature.readBytes(in);
}