mirror of
https://github.com/moparisthebest/mailiverse
synced 2024-11-21 00:15:01 -05:00
adds a set of files
This commit is contained in:
parent
b0c64f9d08
commit
10028cd6e8
10
License.txt
Normal file
10
License.txt
Normal file
@ -0,0 +1,10 @@
|
||||
Portions of this code are release GPL and portions are released BSD.
|
||||
|
||||
Anything to do with the web is GPLv3 Affero + keep my name in the code.
|
||||
Anything to do with the iPhone is BSD + keep my name in the code.
|
||||
Anything to do with the Android is GPLv3 Affero + keep my name in the code.
|
||||
|
||||
License documentation will be placed in each file accordingly.
|
||||
|
||||
|
||||
If you need a different license, please contact me.
|
40
java/core/src/core/crypt/Cryptor.java
Normal file
40
java/core/src/core/crypt/Cryptor.java
Normal file
@ -0,0 +1,40 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import core.callback.Callback;
|
||||
import core.callback.CallbackDefault;
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
public abstract class Cryptor
|
||||
{
|
||||
static final boolean TEST_ENCRYPTION = false;
|
||||
|
||||
abstract public byte[] encrypt (byte[] bytes) throws CryptoException;
|
||||
abstract public byte[] decrypt (byte[] bytes) throws CryptoException;
|
||||
|
||||
public Callback encrypt_()
|
||||
{
|
||||
return new CallbackDefault() {
|
||||
|
||||
@Override
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
next(Cryptor.this.encrypt((byte[])arguments[0]));
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public Callback decrypt_()
|
||||
{
|
||||
return new CallbackDefault() {
|
||||
|
||||
@Override
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
next(Cryptor.this.decrypt((byte[])arguments[0]));
|
||||
}
|
||||
};
|
||||
}
|
||||
}
|
105
java/core/src/core/crypt/CryptorAES.java
Normal file
105
java/core/src/core/crypt/CryptorAES.java
Normal file
@ -0,0 +1,105 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import core.callback.Callback;
|
||||
import core.callback.CallbackDefault;
|
||||
import core.exceptions.CryptoException;
|
||||
import core.util.Arrays;
|
||||
import core.util.LogNull;
|
||||
import core.util.LogOut;
|
||||
import core.util.SecureRandom;
|
||||
|
||||
public class CryptorAES extends Cryptor
|
||||
{
|
||||
static LogNull log = new LogNull(CryptorAES.class);
|
||||
|
||||
public static final int AES_KEYSIZE_BYTES = 32;
|
||||
public static final int AES_IVSIZE_BYTES = 16;
|
||||
static SecureRandom random = new SecureRandom();
|
||||
|
||||
byte[] key;
|
||||
|
||||
public CryptorAES (byte[] key)
|
||||
{
|
||||
this.key = key;
|
||||
}
|
||||
|
||||
public static final byte[] NullIV = {
|
||||
0,0,0,0,0,0,0,0,
|
||||
0,0,0,0,0,0,0,0
|
||||
} ;
|
||||
|
||||
public static byte[] newIV()
|
||||
{
|
||||
byte[] iv = new byte[AES_IVSIZE_BYTES];
|
||||
random.nextBytes(iv);
|
||||
|
||||
return iv;
|
||||
}
|
||||
|
||||
public static byte[] newKey()
|
||||
{
|
||||
byte[] key = new byte[AES_KEYSIZE_BYTES];
|
||||
random.nextBytes(key);
|
||||
|
||||
return key;
|
||||
}
|
||||
|
||||
@Override
|
||||
public byte[] encrypt(byte[] bytes) throws CryptoException
|
||||
{
|
||||
byte[] iv = newIV();
|
||||
Cryptor cryptor = new CryptorAESIV(key, iv);
|
||||
return Arrays.concat(iv, cryptor.encrypt(bytes));
|
||||
}
|
||||
|
||||
@Override
|
||||
public byte[] decrypt(byte[] bytes) throws CryptoException
|
||||
{
|
||||
byte[] iv = Arrays.copyOf(bytes, 0, AES_IVSIZE_BYTES);
|
||||
byte[] data = Arrays.copyOf(bytes, AES_IVSIZE_BYTES, bytes.length - AES_IVSIZE_BYTES);
|
||||
|
||||
Cryptor cryptor = new CryptorAESIV(key, iv);
|
||||
return cryptor.decrypt(data);
|
||||
}
|
||||
|
||||
public Callback decrypt_()
|
||||
{
|
||||
return new CallbackDefault() {
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
byte[] bytes = (byte[]) arguments[0];
|
||||
byte[] iv = Arrays.copyOf(bytes, 0, AES_IVSIZE_BYTES);
|
||||
byte[] data = Arrays.copyOf(bytes, AES_IVSIZE_BYTES, bytes.length - AES_IVSIZE_BYTES);
|
||||
|
||||
log.debug("decrypt_", bytes.length, iv.length, data.length);
|
||||
|
||||
Cryptor cryptor = new CryptorAESIV(key, iv);
|
||||
call(cryptor.decrypt_(), data);
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public Callback encrypt_()
|
||||
{
|
||||
byte[] iv = newIV();
|
||||
Cryptor cryptor = new CryptorAESIV(key, iv);
|
||||
|
||||
return
|
||||
cryptor.encrypt_()
|
||||
.addCallback(new CallbackDefault(iv) {
|
||||
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
byte[] iv = V(0);
|
||||
byte[] data = (byte[]) arguments[0];
|
||||
log.debug("encrypt_", iv.length, data.length);
|
||||
|
||||
next(Arrays.concat(iv, data));
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
}
|
93
java/core/src/core/crypt/CryptorAESIV.java
Normal file
93
java/core/src/core/crypt/CryptorAESIV.java
Normal file
@ -0,0 +1,93 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import core.util.Arrays;
|
||||
import core.util.SecureRandom;
|
||||
|
||||
import org.bc.crypto.BufferedBlockCipher;
|
||||
import org.bc.crypto.engines.AESEngine;
|
||||
import org.bc.crypto.modes.CBCBlockCipher;
|
||||
import org.bc.crypto.paddings.BlockCipherPadding;
|
||||
import org.bc.crypto.paddings.PKCS7Padding;
|
||||
import org.bc.crypto.paddings.PaddedBufferedBlockCipher;
|
||||
import org.bc.crypto.params.KeyParameter;
|
||||
import org.bc.crypto.params.ParametersWithIV;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
|
||||
public class CryptorAESIV extends Cryptor
|
||||
{
|
||||
static SecureRandom random = new SecureRandom();
|
||||
|
||||
byte[] key;
|
||||
byte[] iv;
|
||||
|
||||
public CryptorAESIV (byte[] key, byte[] iv)
|
||||
{
|
||||
// setup cipher parameters with key and IV
|
||||
this.key = key;
|
||||
this.iv = iv;
|
||||
}
|
||||
|
||||
@Override
|
||||
public byte[] decrypt(byte[] bytes) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
BlockCipherPadding padding = new PKCS7Padding();
|
||||
BufferedBlockCipher cipher = new PaddedBufferedBlockCipher(
|
||||
new CBCBlockCipher(new AESEngine()), padding);
|
||||
|
||||
ParametersWithIV params = new ParametersWithIV(new KeyParameter(key), iv);
|
||||
cipher.reset();
|
||||
cipher.init(false, params);
|
||||
|
||||
byte[] out = new byte[cipher.getOutputSize(bytes.length)];
|
||||
int len = cipher.processBytes(bytes, 0, bytes.length, out, 0);
|
||||
len += cipher.doFinal(out, len);
|
||||
|
||||
if (len == out.length)
|
||||
return out;
|
||||
|
||||
return Arrays.copyOf(out, len);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public byte[] encrypt(byte[] bytes) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
BlockCipherPadding padding = new PKCS7Padding();
|
||||
BufferedBlockCipher cipher = new PaddedBufferedBlockCipher(
|
||||
new CBCBlockCipher(new AESEngine()), padding);
|
||||
|
||||
ParametersWithIV params = new ParametersWithIV(new KeyParameter(key), iv);
|
||||
cipher.reset();
|
||||
cipher.init(true, params);
|
||||
|
||||
byte[] out = new byte[cipher.getOutputSize(bytes.length)];
|
||||
int len = cipher.processBytes(bytes, 0, bytes.length, out, 0);
|
||||
len += cipher.doFinal(out, len);
|
||||
|
||||
if (len == out.length)
|
||||
return out;
|
||||
|
||||
return Arrays.copyOf(out, len);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
}
|
53
java/core/src/core/crypt/CryptorAESJCE.java
Normal file
53
java/core/src/core/crypt/CryptorAESJCE.java
Normal file
@ -0,0 +1,53 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import javax.crypto.Cipher;
|
||||
import javax.crypto.spec.IvParameterSpec;
|
||||
import javax.crypto.spec.SecretKeySpec;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
|
||||
public class CryptorAESJCE extends Cryptor
|
||||
{
|
||||
SecretKeySpec secretKey;
|
||||
IvParameterSpec iv;
|
||||
|
||||
public CryptorAESJCE (SecretKeySpec secretKey, IvParameterSpec iv)
|
||||
{
|
||||
this.secretKey = secretKey;
|
||||
this.iv = iv;
|
||||
}
|
||||
|
||||
public byte[] encrypt(byte[] clearText) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
|
||||
cipher.init(Cipher.ENCRYPT_MODE, secretKey, iv);
|
||||
return cipher.doFinal(clearText);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public byte[] decrypt(byte[] encrypted) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
Cipher cipher = Cipher.getInstance("AES/CBC/PKCS5Padding");
|
||||
cipher.init(Cipher.DECRYPT_MODE, secretKey, iv);
|
||||
return cipher.doFinal(encrypted);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException (e);
|
||||
}
|
||||
}
|
||||
}
|
25
java/core/src/core/crypt/CryptorNone.java
Normal file
25
java/core/src/core/crypt/CryptorNone.java
Normal file
@ -0,0 +1,25 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
public class CryptorNone extends Cryptor
|
||||
{
|
||||
|
||||
@Override
|
||||
public byte[] encrypt(byte[] bytes) throws CryptoException
|
||||
{
|
||||
return bytes;
|
||||
}
|
||||
|
||||
@Override
|
||||
public byte[] decrypt(byte[] bytes) throws CryptoException
|
||||
{
|
||||
return bytes;
|
||||
}
|
||||
|
||||
}
|
27
java/core/src/core/crypt/CryptorRSA.java
Normal file
27
java/core/src/core/crypt/CryptorRSA.java
Normal file
@ -0,0 +1,27 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
public abstract class CryptorRSA extends Cryptor
|
||||
{
|
||||
byte[] publicKeyBytes, privateKeyBytes;
|
||||
|
||||
public byte[] getPublicKey ()
|
||||
{
|
||||
return publicKeyBytes;
|
||||
}
|
||||
|
||||
public byte[] getPrivateKey ()
|
||||
{
|
||||
return privateKeyBytes;
|
||||
}
|
||||
|
||||
protected void initialize (byte[] publicKeyBytes, byte[] privateKeyBytes)
|
||||
{
|
||||
this.privateKeyBytes = privateKeyBytes;
|
||||
this.publicKeyBytes = publicKeyBytes;
|
||||
}
|
||||
}
|
216
java/core/src/core/crypt/CryptorRSAAES.java
Normal file
216
java/core/src/core/crypt/CryptorRSAAES.java
Normal file
@ -0,0 +1,216 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.OutputStream;
|
||||
|
||||
import core.util.LogNull;
|
||||
import core.util.LogOut;
|
||||
import core.util.Streams;
|
||||
import core.callback.Callback;
|
||||
import core.callback.CallbackDefault;
|
||||
import core.callback.CallbackSync;
|
||||
import core.callbacks.Memory;
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
public class CryptorRSAAES extends Cryptor
|
||||
{
|
||||
static LogNull log = new LogNull(CryptorRSAAES.class);
|
||||
|
||||
CryptorRSA rsa;
|
||||
|
||||
public static class Version {
|
||||
public static final int
|
||||
R2012 = 0x00,
|
||||
R201303 = 0x01,
|
||||
CURRENT = R201303;
|
||||
};
|
||||
|
||||
public final byte INITIAL = 0x00;
|
||||
public final byte VERSION = 0x01;
|
||||
public final int MAX_RSA_BLOCK_SIZE = 117;
|
||||
|
||||
private final static String M_VERSION = "version";
|
||||
|
||||
public CryptorRSAAES(CryptorRSA rsa)
|
||||
{
|
||||
this.rsa = rsa;
|
||||
}
|
||||
|
||||
void readVersion (InputStream is, Memory memory) throws Exception
|
||||
{
|
||||
int version = is.read()&0xFF;
|
||||
if (version < 0 || version > Version.CURRENT)
|
||||
throw new Exception("Unknown version");
|
||||
|
||||
memory.put(M_VERSION, version);
|
||||
|
||||
log.debug("readVersion", version);
|
||||
}
|
||||
|
||||
void writeVersion (OutputStream os) throws IOException
|
||||
{
|
||||
os.write(Version.CURRENT);
|
||||
|
||||
log.debug("writeVersion", Version.CURRENT);
|
||||
}
|
||||
|
||||
Callback readEncryptedEmbeddedKey_ (InputStream is, Memory memory)
|
||||
{
|
||||
return new CallbackDefault(memory, is) {
|
||||
public void onSuccess(Object... arguments) throws Exception
|
||||
{
|
||||
Memory memory = V(0);
|
||||
InputStream is = V(1);
|
||||
|
||||
int version = (Integer)memory.get(M_VERSION);
|
||||
switch (version) {
|
||||
|
||||
case Version.R2012:
|
||||
Streams.readInt3(is); // ignore the wrapper
|
||||
next(Streams.readBoundedArray(is));
|
||||
break;
|
||||
case Version.R201303:
|
||||
next(Streams.readBoundedArray(is));
|
||||
break;
|
||||
};
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
Callback writeEncryptedEmbeddedKey_ (OutputStream os)
|
||||
{
|
||||
return new CallbackDefault(os) {
|
||||
public void onSuccess(Object... arguments) throws Exception
|
||||
{
|
||||
OutputStream os = V(0);
|
||||
byte[] bytes = (byte[]) arguments[0];
|
||||
Streams.writeBoundedArray(os, bytes);
|
||||
next(arguments);
|
||||
}
|
||||
} ;
|
||||
}
|
||||
|
||||
Callback decryptMainBlock_(InputStream is, Memory memory)
|
||||
{
|
||||
return new CallbackDefault (is, memory) {
|
||||
|
||||
@Override
|
||||
public void onSuccess(Object... arguments) throws Exception
|
||||
{
|
||||
InputStream is = V(0);
|
||||
Memory memory = V(1);
|
||||
|
||||
byte[] key = (byte[]) arguments[0];
|
||||
int version = (Integer)memory.get(M_VERSION);
|
||||
|
||||
Cryptor cryptor = null;
|
||||
|
||||
switch (version)
|
||||
{
|
||||
case Version.R2012:
|
||||
cryptor = new CryptorAESIV(key, CryptorAES.NullIV);
|
||||
break;
|
||||
case Version.R201303:
|
||||
cryptor = new CryptorAES(key);
|
||||
break;
|
||||
}
|
||||
|
||||
call(cryptor.decrypt_(), Streams.readFullyBytes(is));
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
Callback encryptMainBlock_ (OutputStream os, Cryptor cryptor, byte[] bytes)
|
||||
{
|
||||
return new CallbackDefault (os, cryptor, bytes) {
|
||||
public void onSuccess(Object... arguments) throws Exception
|
||||
{
|
||||
OutputStream os = V(0);
|
||||
Cryptor cryptor = V(1);
|
||||
byte[] bytes = V(2);
|
||||
|
||||
call(cryptor.encrypt_().addCallback(Streams.writeBytes_(os)), bytes);
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public Callback decrypt_()
|
||||
{
|
||||
return new CallbackDefault () {
|
||||
|
||||
@Override
|
||||
public void onSuccess(Object... arguments) throws Exception
|
||||
{
|
||||
ByteArrayInputStream is = new ByteArrayInputStream((byte[])arguments[0]);
|
||||
Memory memory = new Memory();
|
||||
readVersion(is, memory);
|
||||
|
||||
call(
|
||||
readEncryptedEmbeddedKey_(is, memory)
|
||||
.addCallback(rsa.decrypt_())
|
||||
.addCallback(decryptMainBlock_(is, memory))
|
||||
);
|
||||
}
|
||||
|
||||
};
|
||||
}
|
||||
|
||||
public Callback encrypt_ ()
|
||||
{
|
||||
return new CallbackDefault () {
|
||||
|
||||
@Override
|
||||
public void onSuccess(Object... arguments) throws Exception
|
||||
{
|
||||
ByteArrayOutputStream os = new ByteArrayOutputStream();
|
||||
writeVersion(os);
|
||||
|
||||
byte[] key = CryptorAES.newKey();
|
||||
CryptorAES aes = new CryptorAES(key);
|
||||
|
||||
call(
|
||||
rsa.encrypt_()
|
||||
.addCallback(writeEncryptedEmbeddedKey_(os))
|
||||
.addCallback(encryptMainBlock_(os, aes, (byte[])arguments[0]))
|
||||
.addCallback(Streams.toByteArray_(os)),
|
||||
key
|
||||
);
|
||||
}
|
||||
|
||||
};
|
||||
|
||||
}
|
||||
|
||||
public byte[] encrypt (byte[] bytes) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
return new CallbackSync<byte[]>(encrypt_()).invoke(bytes).export();
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public byte[] decrypt (byte[] bytes) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
return new CallbackSync<byte[]>(decrypt_()).invoke(bytes).export();
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
}
|
172
java/core/src/core/crypt/CryptorRSABC.java
Normal file
172
java/core/src/core/crypt/CryptorRSABC.java
Normal file
@ -0,0 +1,172 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.IOException;
|
||||
import java.security.SecureRandom;
|
||||
|
||||
import org.bc.asn1.ASN1InputStream;
|
||||
import org.bc.asn1.ASN1Integer;
|
||||
import org.bc.asn1.ASN1ObjectIdentifier;
|
||||
import org.bc.asn1.ASN1OctetString;
|
||||
import org.bc.asn1.ASN1Primitive;
|
||||
import org.bc.asn1.ASN1Sequence;
|
||||
import org.bc.asn1.DERBitString;
|
||||
import org.bc.asn1.pkcs.RSAPrivateKey;
|
||||
import org.bc.asn1.pkcs.RSAPublicKey;
|
||||
import org.bc.crypto.encodings.PKCS1Encoding;
|
||||
import org.bc.crypto.engines.RSAEngine;
|
||||
import org.bc.crypto.params.RSAKeyParameters;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
import core.util.Arrays;
|
||||
|
||||
|
||||
public class CryptorRSABC extends CryptorRSA
|
||||
{
|
||||
SecureRandom random = new SecureRandom();
|
||||
|
||||
public final int MAX_RSA_BLOCK_SIZE = 117;
|
||||
|
||||
RSAPublicKey publicKey;
|
||||
RSAPrivateKey privateKey;
|
||||
|
||||
public static final byte[] iv = Arrays.generate(16, 0);
|
||||
|
||||
public CryptorRSABC (byte[] publicKey, byte[] privateKey) throws CryptoException
|
||||
{
|
||||
initialize(publicKey, privateKey);
|
||||
|
||||
try
|
||||
{
|
||||
if (privateKey != null)
|
||||
{
|
||||
ASN1InputStream asn1InputStream = new ASN1InputStream(new ByteArrayInputStream(privateKey));
|
||||
ASN1Primitive keyObject = asn1InputStream.readObject();
|
||||
asn1InputStream.close();
|
||||
|
||||
ASN1Sequence keySequence = (ASN1Sequence)keyObject;
|
||||
ASN1Integer pkcs8Version = (ASN1Integer) keySequence.getObjectAt(0);
|
||||
ASN1Sequence algorithmSequence = (ASN1Sequence) keySequence.getObjectAt(1);
|
||||
ASN1ObjectIdentifier algorithmIdentifier = (ASN1ObjectIdentifier) algorithmSequence.getObjectAt(0);
|
||||
|
||||
String algorithm = algorithmIdentifier.getId();
|
||||
if (!algorithm.equals("1.2.840.113549.1.1.1"))
|
||||
throw new CryptoException("Unknown RSA algorithm");
|
||||
|
||||
ASN1OctetString privateKeyOctets = (ASN1OctetString) keySequence.getObjectAt(2);
|
||||
|
||||
ASN1InputStream asn1PrivateKeyStream = new ASN1InputStream(privateKeyOctets.getOctetStream());
|
||||
ASN1Primitive privateKeyObject = asn1PrivateKeyStream.readObject();
|
||||
asn1PrivateKeyStream.close();
|
||||
ASN1Sequence privateKeySequence = (ASN1Sequence)privateKeyObject;
|
||||
|
||||
int I=0;
|
||||
ASN1Integer
|
||||
s = (ASN1Integer)privateKeySequence.getObjectAt(I++),
|
||||
n = (ASN1Integer)privateKeySequence.getObjectAt(I++),
|
||||
e = (ASN1Integer)privateKeySequence.getObjectAt(I++),
|
||||
d = (ASN1Integer)privateKeySequence.getObjectAt(I++),
|
||||
p = (ASN1Integer)privateKeySequence.getObjectAt(I++),
|
||||
q = (ASN1Integer)privateKeySequence.getObjectAt(I++),
|
||||
d1 = (ASN1Integer)privateKeySequence.getObjectAt(I++),
|
||||
d2 = (ASN1Integer)privateKeySequence.getObjectAt(I++),
|
||||
c = (ASN1Integer)privateKeySequence.getObjectAt(I++)
|
||||
;
|
||||
|
||||
/*
|
||||
org.bouncycastle.asn1.pkcs.RSAPrivateKey.RSAPrivateKey(
|
||||
BigInteger modulus,
|
||||
BigInteger publicExponent,
|
||||
BigInteger privateExponent,
|
||||
BigInteger prime1,
|
||||
BigInteger prime2,
|
||||
BigInteger exponent1,
|
||||
BigInteger exponent2,
|
||||
BigInteger coefficient
|
||||
)
|
||||
*/
|
||||
this.privateKey =
|
||||
new RSAPrivateKey(
|
||||
n.getValue(),
|
||||
e.getValue(),
|
||||
d.getValue(),
|
||||
p.getValue(),
|
||||
q.getValue(),
|
||||
d1.getValue(),
|
||||
d2.getValue(),
|
||||
c.getValue()
|
||||
);
|
||||
}
|
||||
|
||||
if (publicKey != null)
|
||||
{
|
||||
ASN1InputStream asn1InputStream = new ASN1InputStream(new ByteArrayInputStream(publicKey));
|
||||
ASN1Primitive keyObject = asn1InputStream.readObject();
|
||||
asn1InputStream.close();
|
||||
|
||||
ASN1Sequence keySequence = (ASN1Sequence)keyObject;
|
||||
ASN1Sequence algorithmSequence = (ASN1Sequence) keySequence.getObjectAt(0);
|
||||
ASN1ObjectIdentifier algorithmIdentifier = (ASN1ObjectIdentifier) algorithmSequence.getObjectAt(0);
|
||||
|
||||
String algorithm = algorithmIdentifier.getId();
|
||||
if (!algorithm.equals("1.2.840.113549.1.1.1"))
|
||||
throw new CryptoException("Unknown RSA algorithm");
|
||||
|
||||
DERBitString keyOctets = (DERBitString) keySequence.getObjectAt(1);
|
||||
ASN1InputStream asn1PublicKeyStream = new ASN1InputStream(new ByteArrayInputStream(keyOctets.getBytes()));
|
||||
ASN1Primitive publicKeyObject = asn1PublicKeyStream.readObject();
|
||||
asn1PublicKeyStream.close();
|
||||
|
||||
ASN1Sequence publicKeySequence = (ASN1Sequence)publicKeyObject;
|
||||
|
||||
int I=0;
|
||||
ASN1Integer
|
||||
n = (ASN1Integer)publicKeySequence.getObjectAt(I++),
|
||||
e = (ASN1Integer)publicKeySequence.getObjectAt(I++);
|
||||
|
||||
this.publicKey = new RSAPublicKey(n.getValue(), e.getValue());
|
||||
}
|
||||
}
|
||||
catch (IOException e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public byte[] encrypt (byte[] block) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
PKCS1Encoding e = new PKCS1Encoding(new RSAEngine());
|
||||
RSAKeyParameters key = new RSAKeyParameters(false, publicKey.getModulus(), publicKey.getPublicExponent());
|
||||
e.init(true, key);
|
||||
|
||||
return e.processBlock(block, 0, block.length);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public byte[] decrypt (byte[] block) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
PKCS1Encoding e = new PKCS1Encoding(new RSAEngine());
|
||||
RSAKeyParameters key = new RSAKeyParameters(false, privateKey.getModulus(), privateKey.getPrivateExponent());
|
||||
e.init(false, key);
|
||||
|
||||
return e.processBlock(block, 0, block.length);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
}
|
78
java/core/src/core/crypt/CryptorRSAFactory.java
Normal file
78
java/core/src/core/crypt/CryptorRSAFactory.java
Normal file
@ -0,0 +1,78 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import java.io.InputStream;
|
||||
import java.security.KeyFactory;
|
||||
import java.security.KeyPair;
|
||||
import java.security.KeyPairGenerator;
|
||||
import java.security.PublicKey;
|
||||
import java.security.spec.PKCS8EncodedKeySpec;
|
||||
import java.security.spec.X509EncodedKeySpec;
|
||||
|
||||
import org.bc.crypto.util.PrivateKeyFactory;
|
||||
import org.bc.crypto.util.PublicKeyFactory;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
import core.callback.Callback;
|
||||
import core.util.Base64;
|
||||
import core.util.Pair;
|
||||
|
||||
public class CryptorRSAFactory
|
||||
{
|
||||
public Pair<byte[], byte[]> generate (int bits) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("RSA");
|
||||
keyPairGenerator.initialize(bits);
|
||||
KeyPair keyPair = keyPairGenerator.genKeyPair();
|
||||
byte[] privateKey = keyPair.getPrivate().getEncoded();
|
||||
byte[] publicKey = keyPair.getPublic().getEncoded();
|
||||
|
||||
return new Pair<byte[], byte[]>(publicKey, privateKey);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public void generate (int bits, Callback callback)
|
||||
{
|
||||
try
|
||||
{
|
||||
callback.invoke(generate(bits));
|
||||
}
|
||||
catch (CryptoException e)
|
||||
{
|
||||
callback.invoke(e);
|
||||
}
|
||||
}
|
||||
|
||||
public static CryptorRSA fromResources(InputStream publicKey, InputStream privateKey) throws Exception
|
||||
{
|
||||
return new CryptorRSAJCE(publicKey, privateKey);
|
||||
}
|
||||
|
||||
public static CryptorRSA fromString(String publicKey, String privateKey) throws Exception
|
||||
{
|
||||
return
|
||||
new CryptorRSAJCE(
|
||||
publicKey != null ?
|
||||
KeyFactory.getInstance("RSA").generatePublic(new X509EncodedKeySpec(
|
||||
Base64.decode(publicKey))
|
||||
) :
|
||||
null,
|
||||
privateKey != null ?
|
||||
KeyFactory.getInstance("RSA").generatePrivate(new PKCS8EncodedKeySpec(
|
||||
Base64.decode(privateKey))
|
||||
) :
|
||||
null
|
||||
);
|
||||
}
|
||||
|
||||
}
|
83
java/core/src/core/crypt/CryptorRSAFactoryEnvironment.java
Normal file
83
java/core/src/core/crypt/CryptorRSAFactoryEnvironment.java
Normal file
@ -0,0 +1,83 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import java.security.KeyFactory;
|
||||
import java.security.PrivateKey;
|
||||
import java.security.PublicKey;
|
||||
import java.security.spec.PKCS8EncodedKeySpec;
|
||||
import java.security.spec.RSAPublicKeySpec;
|
||||
import java.security.spec.X509EncodedKeySpec;
|
||||
|
||||
import org.bc.util.encoders.Base64;
|
||||
|
||||
import core.constants.ConstantsEnvironmentKeys;
|
||||
import core.exceptions.CryptoException;
|
||||
import core.util.Environment;
|
||||
|
||||
public class CryptorRSAFactoryEnvironment
|
||||
{
|
||||
public static CryptorRSA createJCE (Environment env) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
PublicKey publicKey = null;
|
||||
PrivateKey privateKey = null;
|
||||
|
||||
String publicKeyString = env.get(ConstantsEnvironmentKeys.PUBLIC_ENCRYPTION_KEY);
|
||||
if (publicKeyString != null)
|
||||
{
|
||||
publicKey =
|
||||
KeyFactory.getInstance("RSA").generatePublic(
|
||||
new X509EncodedKeySpec((byte[])Base64.decode(publicKeyString))
|
||||
);
|
||||
}
|
||||
String privateKeyString = env.get(ConstantsEnvironmentKeys.PRIVATE_DECRYPTION_KEY);
|
||||
if (privateKeyString != null)
|
||||
{
|
||||
privateKey =
|
||||
KeyFactory.getInstance("RSA").generatePrivate(
|
||||
new PKCS8EncodedKeySpec((byte[])Base64.decode(privateKeyString))
|
||||
);
|
||||
}
|
||||
|
||||
return new CryptorRSAJCE (publicKey, privateKey);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
/*
|
||||
public static CryptorRSA createBC (Environment env) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
Base64 base64 = new Base64();
|
||||
|
||||
String publicKeyString = env.get(ConstantsEnvironmentKeys.PUBLIC_ENCRYPTION_KEY);
|
||||
byte[] publicKeyBytes = null;
|
||||
if (publicKeyString != null)
|
||||
publicKeyBytes = (byte[])base64.decode(publicKeyString);
|
||||
|
||||
String privateKeyString = env.get(ConstantsEnvironmentKeys.PRIVATE_DECRYPTION_KEY);
|
||||
byte[] privateKeyBytes = null;
|
||||
if (privateKeyString != null)
|
||||
privateKeyBytes = base64.decode(privateKeyString);
|
||||
|
||||
return new CryptorRSABC (publicKeyBytes, privateKeyBytes);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
*/
|
||||
public static CryptorRSA create (Environment env) throws CryptoException
|
||||
{
|
||||
return createJCE(env);
|
||||
}
|
||||
}
|
91
java/core/src/core/crypt/CryptorRSAJCE.java
Normal file
91
java/core/src/core/crypt/CryptorRSAJCE.java
Normal file
@ -0,0 +1,91 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import java.io.InputStream;
|
||||
import java.security.KeyStore;
|
||||
import java.security.PrivateKey;
|
||||
import java.security.PublicKey;
|
||||
import java.security.SecureRandom;
|
||||
import javax.crypto.Cipher;
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
public class CryptorRSAJCE extends CryptorRSA
|
||||
{
|
||||
SecureRandom random = new SecureRandom();
|
||||
|
||||
public final int MAX_RSA_BLOCK_SIZE = 117;
|
||||
|
||||
PublicKey publicKey;
|
||||
PrivateKey privateKey;
|
||||
|
||||
public CryptorRSAJCE (InputStream keyStore, InputStream trustStore) throws Exception
|
||||
{
|
||||
if (keyStore != null)
|
||||
{
|
||||
KeyStore kks = KeyStore.getInstance("JKS");
|
||||
kks.load(keyStore, "password".toCharArray());
|
||||
privateKey = (PrivateKey)kks.getKey(kks.aliases().nextElement(), "password".toCharArray());
|
||||
|
||||
if (privateKey == null)
|
||||
throw new CryptoException(new Exception("RSA Private Key not found in KeyStore."));
|
||||
}
|
||||
|
||||
if (trustStore != null)
|
||||
{
|
||||
KeyStore tks = KeyStore.getInstance("JKS");
|
||||
tks.load(trustStore, "password".toCharArray());
|
||||
publicKey = tks.getCertificate(tks.aliases().nextElement()).getPublicKey();
|
||||
|
||||
if (publicKey == null)
|
||||
throw new CryptoException(new Exception("RSA Public Key not found in KeyStore."));
|
||||
}
|
||||
|
||||
initialize(
|
||||
publicKey!=null ? publicKey.getEncoded() : null,
|
||||
privateKey!=null ? privateKey.getEncoded() : null
|
||||
);
|
||||
}
|
||||
|
||||
public CryptorRSAJCE (PublicKey publicKey, PrivateKey privateKey)
|
||||
{
|
||||
this.publicKey = publicKey;
|
||||
this.privateKey = privateKey;
|
||||
|
||||
initialize(
|
||||
publicKey!=null ? publicKey.getEncoded() : null,
|
||||
privateKey!=null ? privateKey.getEncoded() : null
|
||||
);
|
||||
}
|
||||
|
||||
public byte[] encrypt (byte[] block) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
Cipher cipher = Cipher.getInstance("RSA");
|
||||
cipher.init(Cipher.ENCRYPT_MODE, publicKey);
|
||||
return cipher.doFinal(block, 0, block.length);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public byte[] decrypt (byte[] block) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
Cipher cipher = Cipher.getInstance("RSA");
|
||||
cipher.init(Cipher.DECRYPT_MODE, privateKey);
|
||||
return cipher.doFinal(block, 0, block.length);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
}
|
16
java/core/src/core/crypt/CryptorSeed.java
Normal file
16
java/core/src/core/crypt/CryptorSeed.java
Normal file
@ -0,0 +1,16 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
public class CryptorSeed
|
||||
{
|
||||
public byte[] seed;
|
||||
|
||||
public CryptorSeed(byte[] seed)
|
||||
{
|
||||
this.seed = seed;
|
||||
}
|
||||
}
|
26
java/core/src/core/crypt/HashMd5.java
Normal file
26
java/core/src/core/crypt/HashMd5.java
Normal file
@ -0,0 +1,26 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import org.bc.crypto.digests.MD5Digest;
|
||||
|
||||
public class HashMd5
|
||||
{
|
||||
public HashMd5 ()
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
public byte[] hash (byte[] bytes)
|
||||
{
|
||||
MD5Digest digest = new MD5Digest();
|
||||
|
||||
byte[] out = new byte[digest.getDigestSize()];
|
||||
digest.update(bytes, 0, bytes.length);
|
||||
digest.doFinal(out, 0);
|
||||
return out;
|
||||
}
|
||||
}
|
26
java/core/src/core/crypt/HashSha256.java
Normal file
26
java/core/src/core/crypt/HashSha256.java
Normal file
@ -0,0 +1,26 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import org.bc.crypto.digests.SHA256Digest;
|
||||
|
||||
public class HashSha256
|
||||
{
|
||||
public HashSha256 ()
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
public byte[] hash (byte[] bytes)
|
||||
{
|
||||
SHA256Digest digest = new SHA256Digest();
|
||||
|
||||
byte[] out = new byte[digest.getDigestSize()];
|
||||
digest.update(bytes, 0, bytes.length);
|
||||
digest.doFinal(out, 0);
|
||||
return out;
|
||||
}
|
||||
}
|
32
java/core/src/core/crypt/HmacSha1.java
Normal file
32
java/core/src/core/crypt/HmacSha1.java
Normal file
@ -0,0 +1,32 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import org.bc.crypto.digests.SHA1Digest;
|
||||
import org.bc.crypto.macs.HMac;
|
||||
import org.bc.crypto.params.KeyParameter;
|
||||
|
||||
public class HmacSha1
|
||||
{
|
||||
HMac mac;
|
||||
byte[] key;
|
||||
|
||||
public HmacSha1(byte[] key)
|
||||
{
|
||||
mac = new HMac(new SHA1Digest());
|
||||
mac.init(new KeyParameter(key));
|
||||
}
|
||||
|
||||
public byte[] mac(byte[] bytes)
|
||||
{
|
||||
mac.update(bytes, 0, bytes.length);
|
||||
byte[] out = new byte[mac.getMacSize()];
|
||||
mac.doFinal(out, 0);
|
||||
|
||||
return out;
|
||||
}
|
||||
|
||||
}
|
75
java/core/src/core/crypt/KeyPairFromPassword.java
Normal file
75
java/core/src/core/crypt/KeyPairFromPassword.java
Normal file
@ -0,0 +1,75 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import core.callback.Callback;
|
||||
import core.callback.CallbackChain;
|
||||
import core.callback.CallbackDefault;
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
public class KeyPairFromPassword
|
||||
{
|
||||
String password;
|
||||
public PBE[] pbe = { new PBE(), new PBE() };
|
||||
|
||||
static final int CRYPTOR = 0, VERIFIER = 1;
|
||||
|
||||
public KeyPairFromPassword (String password)
|
||||
{
|
||||
this.password = password;
|
||||
}
|
||||
|
||||
public Callback generate_()
|
||||
{
|
||||
CallbackChain chain = new CallbackChain();
|
||||
|
||||
return chain
|
||||
.addCallback(
|
||||
pbe[CRYPTOR].generate_(
|
||||
password,
|
||||
PBE.DEFAULT_SALT[CRYPTOR],
|
||||
PBE.DEFAULT_ITERATIONS,
|
||||
PBE.DEFAULT_KEYLENGTH
|
||||
)
|
||||
)
|
||||
.addCallback(
|
||||
pbe[VERIFIER].generate_(
|
||||
password,
|
||||
PBE.DEFAULT_SALT[VERIFIER],
|
||||
PBE.DEFAULT_ITERATIONS,
|
||||
PBE.DEFAULT_KEYLENGTH
|
||||
)
|
||||
)
|
||||
.addCallback(
|
||||
new CallbackDefault() {
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
next(KeyPairFromPassword.this);
|
||||
}
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
public void generate() throws CryptoException
|
||||
{
|
||||
for (int i=0; i<2; ++i)
|
||||
pbe[i].generate(
|
||||
password,
|
||||
PBE.DEFAULT_SALT[i],
|
||||
PBE.DEFAULT_ITERATIONS,
|
||||
PBE.DEFAULT_KEYLENGTH
|
||||
);
|
||||
}
|
||||
|
||||
public byte[] getVerifier ()
|
||||
{
|
||||
return pbe[VERIFIER].key;
|
||||
}
|
||||
|
||||
public PBE getCryptor ()
|
||||
{
|
||||
return pbe[CRYPTOR];
|
||||
}
|
||||
}
|
41
java/core/src/core/crypt/KeyPairFromPasswordCryptor.java
Normal file
41
java/core/src/core/crypt/KeyPairFromPasswordCryptor.java
Normal file
@ -0,0 +1,41 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import core.callback.Callback;
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
public class KeyPairFromPasswordCryptor extends Cryptor
|
||||
{
|
||||
KeyPairFromPassword keyPair;
|
||||
|
||||
public KeyPairFromPasswordCryptor (KeyPairFromPassword keyPair)
|
||||
{
|
||||
this.keyPair = keyPair;
|
||||
}
|
||||
|
||||
public byte[] encrypt (byte[] clearBlock) throws CryptoException
|
||||
{
|
||||
return keyPair.getCryptor().encrypt(clearBlock);
|
||||
}
|
||||
|
||||
public byte[] decrypt (byte[] encryptedBlock) throws CryptoException
|
||||
{
|
||||
return keyPair.getCryptor().decrypt(encryptedBlock);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Callback encrypt_()
|
||||
{
|
||||
return keyPair.getCryptor().encrypt_();
|
||||
}
|
||||
|
||||
@Override
|
||||
public Callback decrypt_()
|
||||
{
|
||||
return keyPair.getCryptor().decrypt_();
|
||||
}
|
||||
}
|
105
java/core/src/core/crypt/PBE.java
Normal file
105
java/core/src/core/crypt/PBE.java
Normal file
@ -0,0 +1,105 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import org.bc.crypto.PBEParametersGenerator;
|
||||
import org.bc.crypto.digests.SHA256Digest;
|
||||
import org.bc.crypto.generators.PKCS5S2ParametersGenerator;
|
||||
import org.bc.crypto.params.KeyParameter;
|
||||
|
||||
import core.callback.Callback;
|
||||
import core.callback.CallbackDefault;
|
||||
import core.crypt.Cryptor;
|
||||
import core.crypt.CryptorAES;
|
||||
import core.exceptions.CryptoException;
|
||||
import core.util.Arrays;
|
||||
import core.util.Base64;
|
||||
import core.util.LogNull;
|
||||
import core.util.LogOut;
|
||||
|
||||
/**
|
||||
* http://stackoverflow.com/questions/992019/java-256-bit-aes-password-based-encryption
|
||||
*/
|
||||
public class PBE extends Cryptor
|
||||
{
|
||||
LogNull log = new LogNull(PBE.class);
|
||||
|
||||
public byte[] key = null;
|
||||
Cryptor cryptorAES;
|
||||
|
||||
public static final int DEFAULT_ITERATIONS = 131072;
|
||||
public static final int SHORT_ITERATIONS = 32768;
|
||||
public static final int DEFAULT_KEYLENGTH = 256;
|
||||
|
||||
/**
|
||||
* bytes used to salt the key (set before making an instance)
|
||||
*/
|
||||
public static final byte[] DEFAULT_SALT_0 = {
|
||||
(byte)0xc8, (byte)0x73, (byte)0x41, (byte)0x8c,
|
||||
(byte)0x7e, (byte)0xd8, (byte)0xee, (byte)0x89
|
||||
};
|
||||
|
||||
/**
|
||||
* bytes used to salt the key (set before making an instance)
|
||||
*/
|
||||
public static final byte[] DEFAULT_SALT_1 = {
|
||||
(byte)0x12, (byte)0x53, (byte)0x14, (byte)0xbb,
|
||||
(byte)0x7e, (byte)0x97, (byte)0xce, (byte)0x55
|
||||
};
|
||||
|
||||
/**
|
||||
* bytes used to salt the key (set before making an instance)
|
||||
*/
|
||||
public static final byte[] DEFAULT_SALT_2 = {
|
||||
(byte)0x0a, (byte)0x48, (byte)0x33, (byte)0xfe,
|
||||
(byte)0xa7, (byte)0xc2, (byte)0x2c, (byte)0x71
|
||||
};
|
||||
|
||||
public static final byte[][] DEFAULT_SALT = { DEFAULT_SALT_0, DEFAULT_SALT_1, DEFAULT_SALT_2 };
|
||||
|
||||
public PBE ()
|
||||
{
|
||||
|
||||
}
|
||||
public PBE (String password, byte[] salt, int iterationCount, int keyLength) throws CryptoException
|
||||
{
|
||||
generate(password,salt, iterationCount, keyLength);
|
||||
}
|
||||
|
||||
public void generate (String password, byte[] salt, int iterationCount, int keyLength)
|
||||
throws CryptoException
|
||||
{
|
||||
key = PBEPlatform.generate(password, salt, iterationCount, keyLength);
|
||||
cryptorAES = new CryptorAES (key);
|
||||
|
||||
log.debug("computed", new String(Base64.encode(key)), " using ", new String(password),"salt",new String(Base64.encode(salt)), iterationCount, keyLength);
|
||||
}
|
||||
|
||||
public Callback generate_ (String password, byte[] salt, int iterationCount, int keyLength)
|
||||
{
|
||||
return
|
||||
new CallbackDefault(password, salt, iterationCount, keyLength) {
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
String password = V(0);
|
||||
byte[] salt = V(1);
|
||||
int iterationCount = (Integer)V(2);
|
||||
int keyLength = (Integer)V(3);
|
||||
|
||||
generate(password, salt, iterationCount, keyLength);
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public byte[] encrypt(byte[] clearText) throws CryptoException
|
||||
{
|
||||
return cryptorAES.encrypt(clearText);
|
||||
}
|
||||
|
||||
public byte[] decrypt(byte[] encrypted) throws CryptoException
|
||||
{
|
||||
return cryptorAES.decrypt(encrypted);
|
||||
}
|
||||
}
|
16
java/core/src/core/crypt/PBEPlatform.java
Normal file
16
java/core/src/core/crypt/PBEPlatform.java
Normal file
@ -0,0 +1,16 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
public class PBEPlatform
|
||||
{
|
||||
static byte[] generate (String password, byte[] salt, int iterationCount, int keyLength) throws CryptoException
|
||||
{
|
||||
return PBEPlatformBC.generate(password, salt, iterationCount, keyLength);
|
||||
}
|
||||
}
|
29
java/core/src/core/crypt/PBEPlatformBC.java
Normal file
29
java/core/src/core/crypt/PBEPlatformBC.java
Normal file
@ -0,0 +1,29 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import org.bc.crypto.PBEParametersGenerator;
|
||||
import org.bc.crypto.digests.SHA256Digest;
|
||||
import org.bc.crypto.generators.PKCS5S2ParametersGenerator;
|
||||
import org.bc.crypto.params.KeyParameter;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
public class PBEPlatformBC
|
||||
{
|
||||
public static byte[] generate (String password, byte[] salt, int iterationCount, int keyLength)
|
||||
throws CryptoException
|
||||
{
|
||||
PBEParametersGenerator generator = new PKCS5S2ParametersGenerator(new SHA256Digest());
|
||||
generator.init(
|
||||
PBEParametersGenerator.PKCS5PasswordToBytes(password.toCharArray()),
|
||||
salt,
|
||||
iterationCount
|
||||
);
|
||||
|
||||
return ((KeyParameter)generator.generateDerivedParameters(keyLength)).getKey();
|
||||
}
|
||||
}
|
13
java/core/src/core/crypt/PBEPlatformNative.java
Normal file
13
java/core/src/core/crypt/PBEPlatformNative.java
Normal file
@ -0,0 +1,13 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
public class PBEPlatformNative
|
||||
{
|
||||
public native static byte[] generate (String password, byte[] salt, int iterationCount, int keyLength);
|
||||
}
|
30
java/core/src/core/crypt/PasswordValidator.java
Normal file
30
java/core/src/core/crypt/PasswordValidator.java
Normal file
@ -0,0 +1,30 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.crypt;
|
||||
|
||||
import java.io.Serializable;
|
||||
|
||||
|
||||
public class PasswordValidator implements Serializable
|
||||
{
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
||||
public String user;
|
||||
public String password;
|
||||
byte[] payload;
|
||||
|
||||
public PasswordValidator (String user, String password, byte[] payload)
|
||||
{
|
||||
this.user = user;
|
||||
this.password = password;
|
||||
this.payload = payload;
|
||||
}
|
||||
|
||||
public PasswordValidator (String user, String password)
|
||||
{
|
||||
this(user, password, null);
|
||||
}
|
||||
}
|
19
java/core/src/core/exceptions/CryptoException.java
Normal file
19
java/core/src/core/exceptions/CryptoException.java
Normal file
@ -0,0 +1,19 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.exceptions;
|
||||
|
||||
public class CryptoException extends Exception
|
||||
{
|
||||
public CryptoException (Exception e)
|
||||
{
|
||||
super(e);
|
||||
}
|
||||
|
||||
public CryptoException(String s)
|
||||
{
|
||||
super(s);
|
||||
}
|
||||
}
|
23
java/core/src/core/exceptions/InternalException.java
Normal file
23
java/core/src/core/exceptions/InternalException.java
Normal file
@ -0,0 +1,23 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.exceptions;
|
||||
|
||||
import org.timepedia.exporter.client.Export;
|
||||
import org.timepedia.exporter.client.Exportable;
|
||||
|
||||
@Export
|
||||
public class InternalException extends PublicMessageException implements Exportable
|
||||
{
|
||||
public InternalException (Exception e)
|
||||
{
|
||||
super("An internal error has occurred with " + e.getClass().getName(), e);
|
||||
}
|
||||
|
||||
public InternalException (String with, Exception e)
|
||||
{
|
||||
super("An internal error has occurred with " + with, e);
|
||||
}
|
||||
}
|
32
java/core/src/core/exceptions/PublicMessageException.java
Normal file
32
java/core/src/core/exceptions/PublicMessageException.java
Normal file
@ -0,0 +1,32 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.exceptions;
|
||||
|
||||
import org.timepedia.exporter.client.Export;
|
||||
import org.timepedia.exporter.client.Exportable;
|
||||
|
||||
@Export
|
||||
public class PublicMessageException extends RuntimeException implements Exportable
|
||||
{
|
||||
public String message;
|
||||
|
||||
public PublicMessageException (String message)
|
||||
{
|
||||
this.message = message;
|
||||
}
|
||||
|
||||
public PublicMessageException (String message, Exception e)
|
||||
{
|
||||
super(e);
|
||||
this.message = message;
|
||||
}
|
||||
|
||||
@Override
|
||||
public String getMessage ()
|
||||
{
|
||||
return message;
|
||||
}
|
||||
}
|
16
java/core/src/core/exceptions/UserExistsException.java
Normal file
16
java/core/src/core/exceptions/UserExistsException.java
Normal file
@ -0,0 +1,16 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.exceptions;
|
||||
|
||||
import java.io.IOException;
|
||||
|
||||
public class UserExistsException extends PublicMessageException
|
||||
{
|
||||
public UserExistsException ()
|
||||
{
|
||||
super("User already exists");
|
||||
}
|
||||
}
|
98
java/core/src/core/io/IoChain.java
Normal file
98
java/core/src/core/io/IoChain.java
Normal file
@ -0,0 +1,98 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
public abstract class IoChain
|
||||
{
|
||||
protected boolean finished = false;
|
||||
protected IoChain sender, receiver;
|
||||
|
||||
public IoChain (IoChain sender)
|
||||
{
|
||||
if (sender != null)
|
||||
setSender(sender);
|
||||
}
|
||||
|
||||
public void setSender (IoChain sender)
|
||||
{
|
||||
this.sender = sender;
|
||||
sender.setReceiver (this);
|
||||
}
|
||||
|
||||
public void setReceiver (IoChain receiver)
|
||||
{
|
||||
this.receiver = receiver;
|
||||
}
|
||||
|
||||
public void send (byte[] packet) throws Exception
|
||||
{
|
||||
this.sender.send(packet);
|
||||
}
|
||||
|
||||
protected void onReceive (byte[] object) throws Exception
|
||||
{
|
||||
receiver.onReceive(object);
|
||||
}
|
||||
|
||||
public void receive(byte[] object)
|
||||
{
|
||||
try
|
||||
{
|
||||
onReceive(object);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
onException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public void onException (Exception e)
|
||||
{
|
||||
if (receiver != null)
|
||||
receiver.onException(e);
|
||||
}
|
||||
|
||||
public IoChain getFinalSender ()
|
||||
{
|
||||
if (sender != null)
|
||||
return sender.getFinalSender();
|
||||
|
||||
return this;
|
||||
}
|
||||
|
||||
public void open () throws Exception
|
||||
{
|
||||
if (receiver != null)
|
||||
receiver.open();
|
||||
}
|
||||
|
||||
protected void close () throws Exception
|
||||
{
|
||||
if (sender != null)
|
||||
sender.close();
|
||||
}
|
||||
|
||||
public void stop () throws Exception
|
||||
{
|
||||
if (receiver != null)
|
||||
receiver.stop();
|
||||
else
|
||||
close();
|
||||
}
|
||||
|
||||
public void run () throws Exception
|
||||
{
|
||||
if (sender != null)
|
||||
sender.run();
|
||||
else
|
||||
open();
|
||||
}
|
||||
|
||||
public boolean isFinished ()
|
||||
{
|
||||
return finished;
|
||||
}
|
||||
}
|
83
java/core/src/core/io/IoChainAccumulator.java
Normal file
83
java/core/src/core/io/IoChainAccumulator.java
Normal file
@ -0,0 +1,83 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
import core.util.LogNull;
|
||||
import core.util.LogOut;
|
||||
|
||||
public class IoChainAccumulator extends IoChain
|
||||
{
|
||||
LogNull log = new LogNull(IoChainAccumulator.class);
|
||||
List<byte[]> out;
|
||||
boolean opened, closed;
|
||||
Exception exception;
|
||||
|
||||
public IoChainAccumulator()
|
||||
{
|
||||
super(null);
|
||||
out = new ArrayList<byte[]>();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send(byte[] packet) throws Exception
|
||||
{
|
||||
log.debug("accumulating packet");
|
||||
out.add(packet);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void onReceive(byte[] object) throws Exception
|
||||
{
|
||||
super.onReceive(object);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void open () throws Exception
|
||||
{
|
||||
opened = true;
|
||||
super.open();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void close () throws Exception
|
||||
{
|
||||
closed = true;
|
||||
super.close();
|
||||
}
|
||||
|
||||
public void onException(Exception e)
|
||||
{
|
||||
this.exception = e;
|
||||
super.onException(e);
|
||||
}
|
||||
|
||||
public List<byte[]> getAndClearPackets()
|
||||
{
|
||||
List<byte[]> result = out;
|
||||
out = new ArrayList<byte[]>();
|
||||
return result;
|
||||
}
|
||||
|
||||
public Exception getAndClearException ()
|
||||
{
|
||||
Exception e = this.exception;
|
||||
this.exception = null;
|
||||
return e;
|
||||
}
|
||||
|
||||
public boolean isOpened ()
|
||||
{
|
||||
return opened;
|
||||
}
|
||||
|
||||
public boolean isClosed ()
|
||||
{
|
||||
return closed;
|
||||
}
|
||||
}
|
41
java/core/src/core/io/IoChainBase64.java
Normal file
41
java/core/src/core/io/IoChainBase64.java
Normal file
@ -0,0 +1,41 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
import java.io.IOException;
|
||||
|
||||
import core.util.Base64;
|
||||
|
||||
import core.util.LogNull;
|
||||
import core.util.LogOut;
|
||||
|
||||
public class IoChainBase64 extends IoChain
|
||||
{
|
||||
static LogNull log = new LogNull(IoChainBase64.class);
|
||||
|
||||
public IoChainBase64(IoChain sender)
|
||||
{
|
||||
super(sender);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send(byte[] packet) throws Exception
|
||||
{
|
||||
byte[] encoded = Base64.encodeBytes(packet);
|
||||
// log.debug("send base64 check:", Strings.toString(Base64.decode(encoded)));
|
||||
|
||||
super.send(encoded);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void onReceive (byte[] packet) throws Exception
|
||||
{
|
||||
byte[] decoded = Base64.decodeBytes(packet);
|
||||
// log.debug("onReceive:",new String(packet));
|
||||
|
||||
super.onReceive(decoded);
|
||||
}
|
||||
}
|
10
java/core/src/core/io/IoChainFinishedException.java
Normal file
10
java/core/src/core/io/IoChainFinishedException.java
Normal file
@ -0,0 +1,10 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
public class IoChainFinishedException extends Exception {
|
||||
|
||||
}
|
65
java/core/src/core/io/IoChainNewLinePackets.java
Normal file
65
java/core/src/core/io/IoChainNewLinePackets.java
Normal file
@ -0,0 +1,65 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
import core.util.LogNull;
|
||||
import core.util.Strings;
|
||||
|
||||
public class IoChainNewLinePackets extends IoChain
|
||||
{
|
||||
LogNull log = new LogNull(IoChainNewLinePackets.class);
|
||||
List<byte[]> packets = new ArrayList<byte[]>();
|
||||
ByteArrayOutputStream unfinished = new ByteArrayOutputStream();
|
||||
|
||||
public IoChainNewLinePackets(IoChain sender)
|
||||
{
|
||||
super(sender);
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void onReceive(byte[] bytes) throws Exception
|
||||
{
|
||||
for (byte b: bytes)
|
||||
{
|
||||
if (b == '\n' || b == '\r')
|
||||
{
|
||||
log.debug("found newline");
|
||||
if (unfinished.size()>0)
|
||||
{
|
||||
packets.add(unfinished.toByteArray());
|
||||
unfinished = new ByteArrayOutputStream();
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
unfinished.write(b);
|
||||
}
|
||||
}
|
||||
|
||||
while (!packets.isEmpty())
|
||||
{
|
||||
log.debug("recieving packet");
|
||||
|
||||
byte[] packet = packets.get(0);
|
||||
packets.remove(0);
|
||||
super.onReceive(packet);
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send (byte[] bytes) throws Exception
|
||||
{
|
||||
ByteArrayOutputStream bos = new ByteArrayOutputStream();
|
||||
bos.write(bytes);
|
||||
bos.write(Strings.toBytes("\n"));
|
||||
|
||||
super.send(bos.toByteArray());
|
||||
}
|
||||
}
|
32
java/core/src/core/io/IoChainSSLSocket.java
Normal file
32
java/core/src/core/io/IoChainSSLSocket.java
Normal file
@ -0,0 +1,32 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.net.Socket;
|
||||
import java.net.UnknownHostException;
|
||||
|
||||
import javax.net.ssl.SSLContext;
|
||||
import javax.net.ssl.SSLSocket;
|
||||
import javax.net.ssl.SSLSocketFactory;
|
||||
|
||||
public class IoChainSSLSocket extends IoChainSocket
|
||||
{
|
||||
public IoChainSSLSocket(SSLContext context, String host, int port) throws Exception
|
||||
{
|
||||
useSocket(createSSLSocket(context, host, port));
|
||||
}
|
||||
|
||||
public Socket createSSLSocket (SSLContext context, String host, int port) throws UnknownHostException, IOException
|
||||
{
|
||||
SSLSocketFactory sslsocketfactory = context.getSocketFactory();
|
||||
SSLSocket socket = (SSLSocket) sslsocketfactory.createSocket(host, port);
|
||||
socket.setUseClientMode(true);
|
||||
|
||||
return socket;
|
||||
}
|
||||
|
||||
}
|
87
java/core/src/core/io/IoChainSizedPackets.java
Normal file
87
java/core/src/core/io/IoChainSizedPackets.java
Normal file
@ -0,0 +1,87 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.DataInputStream;
|
||||
import java.io.DataOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.PipedInputStream;
|
||||
import java.io.PipedOutputStream;
|
||||
|
||||
import core.util.CircularByteBuffer;
|
||||
|
||||
public class IoChainSizedPackets extends IoChain
|
||||
{
|
||||
DataInputStream in;
|
||||
CircularByteBuffer internalIn;
|
||||
|
||||
Integer nextPacket;
|
||||
|
||||
public IoChainSizedPackets (IoChain sender) throws IOException
|
||||
{
|
||||
super(sender);
|
||||
|
||||
internalIn = new CircularByteBuffer(CircularByteBuffer.INFINITE_SIZE);
|
||||
this.in = new DataInputStream(internalIn.getInputStream());
|
||||
}
|
||||
|
||||
public boolean ready () throws IOException
|
||||
{
|
||||
if (nextPacket == null)
|
||||
{
|
||||
if (in.available() >= Integer.SIZE/8)
|
||||
nextPacket = in.readInt();
|
||||
}
|
||||
|
||||
if (nextPacket != null)
|
||||
return in.available() >= nextPacket;
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void onReceive(byte[] bytes) throws Exception
|
||||
{
|
||||
internalIn.getOutputStream().write(bytes);
|
||||
|
||||
while (ready())
|
||||
read();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send(byte[] bytes) throws Exception
|
||||
{
|
||||
ByteArrayOutputStream bos = new ByteArrayOutputStream();
|
||||
DataOutputStream dos = new DataOutputStream(bos);
|
||||
int size = bytes.length;
|
||||
dos.writeInt(size);
|
||||
dos.write(bytes);
|
||||
dos.flush();
|
||||
|
||||
super.send(bos.toByteArray());
|
||||
}
|
||||
|
||||
public void read () throws Exception
|
||||
{
|
||||
if (!ready())
|
||||
throw new IOException();
|
||||
|
||||
byte[] buf = new byte[nextPacket];
|
||||
if (in.read(buf, 0, nextPacket) != nextPacket)
|
||||
throw new IOException ();
|
||||
|
||||
nextPacket = null;
|
||||
|
||||
super.onReceive(buf);
|
||||
}
|
||||
|
||||
public boolean eof () throws IOException
|
||||
{
|
||||
return false;
|
||||
}
|
||||
}
|
111
java/core/src/core/io/IoChainSocket.java
Normal file
111
java/core/src/core/io/IoChainSocket.java
Normal file
@ -0,0 +1,111 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.io.InputStream;
|
||||
import java.io.OutputStream;
|
||||
import java.net.Socket;
|
||||
|
||||
import core.util.CircularByteBuffer;
|
||||
|
||||
public class IoChainSocket extends IoChain implements Runnable
|
||||
{
|
||||
String host;
|
||||
int port;
|
||||
Socket socket;
|
||||
|
||||
InputStream in;
|
||||
OutputStream out;
|
||||
|
||||
protected IoChainSocket() throws Exception
|
||||
{
|
||||
super(null);
|
||||
}
|
||||
|
||||
protected void useSocket (Socket socket) throws Exception
|
||||
{
|
||||
this.socket = socket;
|
||||
|
||||
in = socket.getInputStream();
|
||||
out = socket.getOutputStream();
|
||||
}
|
||||
|
||||
public IoChainSocket(String host, int port)
|
||||
{
|
||||
super(null);
|
||||
|
||||
this.host = host;
|
||||
this.port = port;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void open () throws Exception
|
||||
{
|
||||
if (socket == null)
|
||||
useSocket(new Socket(host, port));
|
||||
|
||||
super.open();
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void close () throws Exception
|
||||
{
|
||||
if (socket != null)
|
||||
{
|
||||
socket.close();
|
||||
socket = null;
|
||||
}
|
||||
}
|
||||
|
||||
public boolean pumpBlock () throws Exception
|
||||
{
|
||||
CircularByteBuffer buffer = new CircularByteBuffer(CircularByteBuffer.INFINITE_SIZE);
|
||||
|
||||
boolean first = true;
|
||||
while (socket!= null && (first || in.available()>0))
|
||||
{
|
||||
first = false;
|
||||
int c = in.read();
|
||||
|
||||
if (c == -1)
|
||||
throw new IoChainFinishedException();
|
||||
|
||||
buffer.getOutputStream().write(c);
|
||||
}
|
||||
|
||||
if (buffer.getAvailable() > 0)
|
||||
{
|
||||
byte[] b = new byte[buffer.getAvailable()];
|
||||
buffer.getInputStream().read(b);
|
||||
|
||||
receive(b);
|
||||
}
|
||||
|
||||
return socket != null;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send(byte[] bytes) throws IOException
|
||||
{
|
||||
out.write(bytes);
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run ()
|
||||
{
|
||||
try
|
||||
{
|
||||
open();
|
||||
|
||||
while (pumpBlock()) ;
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
onException(e);
|
||||
}
|
||||
}
|
||||
}
|
29
java/core/src/core/io/IoChainThread.java
Normal file
29
java/core/src/core/io/IoChainThread.java
Normal file
@ -0,0 +1,29 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
public class IoChainThread extends Thread
|
||||
{
|
||||
IoChain session;
|
||||
|
||||
public IoChainThread (IoChain session)
|
||||
{
|
||||
this.session = session;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void run()
|
||||
{
|
||||
try
|
||||
{
|
||||
((IoChain)session).run();
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
}
|
47
java/core/src/core/io/SSLContextGenerator.java
Normal file
47
java/core/src/core/io/SSLContextGenerator.java
Normal file
@ -0,0 +1,47 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.io;
|
||||
|
||||
import java.io.InputStream;
|
||||
import java.security.KeyStore;
|
||||
|
||||
import javax.net.ssl.KeyManagerFactory;
|
||||
import javax.net.ssl.SSLContext;
|
||||
import javax.net.ssl.TrustManagerFactory;
|
||||
|
||||
public class SSLContextGenerator
|
||||
{
|
||||
static public SSLContext getSslContext(InputStream clientKeyStore, InputStream serverKeyStore)
|
||||
{
|
||||
SSLContext sslContext = null;
|
||||
|
||||
try
|
||||
{
|
||||
KeyStore cks = KeyStore.getInstance("JKS");
|
||||
cks.load(clientKeyStore, "password".toCharArray());
|
||||
|
||||
KeyStore sks = KeyStore.getInstance("JKS");
|
||||
sks.load(serverKeyStore, "password".toCharArray());
|
||||
|
||||
KeyManagerFactory kmf = KeyManagerFactory.getInstance("SunX509");
|
||||
kmf.init(cks, "password".toCharArray());
|
||||
|
||||
TrustManagerFactory tmf = TrustManagerFactory.getInstance("SunX509", "SunJSSE");
|
||||
tmf.init(sks);
|
||||
|
||||
// this might need to be SSL not TLS
|
||||
sslContext = SSLContext.getInstance("TLS");
|
||||
sslContext.init(kmf.getKeyManagers(), tmf.getTrustManagers(), null);
|
||||
|
||||
return sslContext;
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
ex.printStackTrace();
|
||||
}
|
||||
return sslContext;
|
||||
}
|
||||
}
|
146
java/core/src/core/server/captcha/Captcha.java
Normal file
146
java/core/src/core/server/captcha/Captcha.java
Normal file
@ -0,0 +1,146 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.server.captcha;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.math.BigInteger;
|
||||
import java.security.SecureRandom;
|
||||
import java.sql.Connection;
|
||||
import java.sql.DriverManager;
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
import java.sql.Statement;
|
||||
import java.util.Random;
|
||||
|
||||
import core.server.captcha.sql.Catalog;
|
||||
import core.util.Passwords;
|
||||
|
||||
|
||||
public class Captcha
|
||||
{
|
||||
Random random = new SecureRandom();
|
||||
Catalog catalog = new Catalog();
|
||||
|
||||
public static final String
|
||||
SignUp = "SignUp",
|
||||
CreateBucket = "CreateBucket";
|
||||
|
||||
public void prune () throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(Catalog.PRUNE_TOKENS));
|
||||
log(statement);
|
||||
|
||||
statement.executeUpdate();
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
protected void insertCaptchToken (String token, String use) throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(Catalog.ADD_TOKEN));
|
||||
statement.setString(1, token + "_" + use);
|
||||
log(statement);
|
||||
statement.executeUpdate();
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public String captchaSucceeded () throws SQLException, IOException
|
||||
{
|
||||
String token = BigInteger.valueOf(Math.abs(random.nextLong())).toString(32);
|
||||
insertCaptchToken (token, SignUp);
|
||||
insertCaptchToken (token, CreateBucket);
|
||||
|
||||
return token;
|
||||
}
|
||||
|
||||
public void useToken (String token, String use) throws SQLException, IOException
|
||||
{
|
||||
prune();
|
||||
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(Catalog.CHECK_TOKEN));
|
||||
statement.setString(1, token + "_" + use);
|
||||
log(statement);
|
||||
|
||||
ResultSet rs = statement.executeQuery();
|
||||
if (!rs.next())
|
||||
throw new IOException ("Null captcha");
|
||||
rs.close();
|
||||
|
||||
statement = connection.prepareStatement (catalog.getSingle(Catalog.USE_TOKEN));
|
||||
statement.setString(1, token + "_" + use);
|
||||
log(statement);
|
||||
|
||||
statement.executeUpdate();
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public void ensureTables() throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
for (String sql : catalog.getMulti(Catalog.CREATE_TABLES))
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (sql);
|
||||
log(statement);
|
||||
statement.executeUpdate();
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public Connection openConnection () throws SQLException, IOException
|
||||
{
|
||||
return DriverManager.getConnection(Catalog.CONNECTION_STRING, Catalog.USER, Passwords.getPasswordFor(Catalog.USER));
|
||||
}
|
||||
|
||||
public void closeConnection (Connection connection)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (connection != null)
|
||||
connection.close();
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public void log (Statement sql)
|
||||
{
|
||||
System.out.println (sql);
|
||||
}
|
||||
}
|
22
java/core/src/core/server/captcha/sql/Catalog.java
Normal file
22
java/core/src/core/server/captcha/sql/Catalog.java
Normal file
@ -0,0 +1,22 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.server.captcha.sql;
|
||||
|
||||
import core.constants.ConstantsServer;
|
||||
import core.util.SqlCatalog;
|
||||
|
||||
|
||||
public final class Catalog extends SqlCatalog
|
||||
{
|
||||
static public final String CONNECTION_STRING = ConstantsServer.DBCONNECTION_PREFIX + "captcha";
|
||||
public static final String USER = "captcha";
|
||||
|
||||
static public final String CREATE_TABLES = "create_tables";
|
||||
static public final String USE_TOKEN = "use_token";
|
||||
static public final String CHECK_TOKEN = "check_token";
|
||||
static public final String ADD_TOKEN = "add_token";
|
||||
static public final String PRUNE_TOKENS = "prune_tokens";
|
||||
}
|
1
java/core/src/core/server/captcha/sql/add_token
Normal file
1
java/core/src/core/server/captcha/sql/add_token
Normal file
@ -0,0 +1 @@
|
||||
INSERT INTO captcha (token) values (?)
|
1
java/core/src/core/server/captcha/sql/check_token
Normal file
1
java/core/src/core/server/captcha/sql/check_token
Normal file
@ -0,0 +1 @@
|
||||
SELECT true FROM captcha WHERE token=?
|
5
java/core/src/core/server/captcha/sql/create_tables
Normal file
5
java/core/src/core/server/captcha/sql/create_tables
Normal file
@ -0,0 +1,5 @@
|
||||
CREATE TABLE IF NOT EXISTS captcha (
|
||||
token VARCHAR(255),
|
||||
mark TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
PRIMARY KEY (token)
|
||||
)
|
1
java/core/src/core/server/captcha/sql/prune_tokens
Normal file
1
java/core/src/core/server/captcha/sql/prune_tokens
Normal file
@ -0,0 +1 @@
|
||||
DELETE FROM captcha WHERE mark < DATE_SUB(now(), INTERVAL 2 HOUR);
|
1
java/core/src/core/server/captcha/sql/use_token
Normal file
1
java/core/src/core/server/captcha/sql/use_token
Normal file
@ -0,0 +1 @@
|
||||
DELETE FROM captcha WHERE token=?
|
479
java/core/src/core/server/db/UserDb.java
Normal file
479
java/core/src/core/server/db/UserDb.java
Normal file
@ -0,0 +1,479 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.server.srp.db;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.math.BigInteger;
|
||||
import java.security.SecureRandom;
|
||||
import java.sql.Connection;
|
||||
import java.sql.DriverManager;
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
import java.sql.Statement;
|
||||
import java.sql.Timestamp;
|
||||
import java.util.Date;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
import core.exceptions.PublicMessageException;
|
||||
import core.exceptions.UserExistsException;
|
||||
import core.server.srp.db.sql.Catalog;
|
||||
import core.util.LogOut;
|
||||
import core.util.Pair;
|
||||
import core.util.Passwords;
|
||||
import core.util.Strings;
|
||||
import core.util.Triple;
|
||||
import core.util.Base64;
|
||||
|
||||
public abstract class UserDb
|
||||
{
|
||||
static LogOut log = new LogOut(UserDb.class);
|
||||
|
||||
SecureRandom random = new SecureRandom();
|
||||
Catalog catalog;
|
||||
|
||||
protected UserDb (Catalog catalog)
|
||||
{
|
||||
this.catalog = catalog;
|
||||
}
|
||||
|
||||
public void testCreateUser (String version, String userName) throws Exception
|
||||
{
|
||||
checkRoomForNewUser();
|
||||
testIllegalUserName(userName);
|
||||
|
||||
if (getUser(userName)!=null)
|
||||
throw new UserExistsException();
|
||||
}
|
||||
|
||||
public void checkRoomForNewUser () throws Exception
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.ROOM_FOR_NEW_USER));
|
||||
ResultSet results = statement.executeQuery();
|
||||
if (results.next())
|
||||
{
|
||||
boolean hasRoom = results.getBoolean("room");
|
||||
if (hasRoom)
|
||||
return;
|
||||
}
|
||||
|
||||
throw new PublicMessageException("No room for new users");
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public Integer getUserId (String userName) throws IOException, SQLException
|
||||
{
|
||||
return getUser(userName).first;
|
||||
}
|
||||
|
||||
public void createUser(String version, String userName, byte[] v, byte[] s) throws Exception
|
||||
{
|
||||
checkRoomForNewUser();
|
||||
testIllegalUserName(userName);
|
||||
|
||||
if (getUser(userName)!=null)
|
||||
throw new UserExistsException();
|
||||
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.CREATE_USER));
|
||||
statement.setString(1, version);
|
||||
statement.setString(2, userName);
|
||||
statement.setString(3, Base64.encode(v));
|
||||
statement.setString(4, Base64.encode(s));
|
||||
log(statement);
|
||||
|
||||
statement.executeUpdate();
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public Pair<Integer, Triple<String, byte[], byte[]> > getUser (String userName) throws IOException, SQLException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
Pair<Integer, Triple<String, byte[], byte[]>> result = null;
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.GET_USER));
|
||||
statement.setString(1, userName);
|
||||
log(statement);
|
||||
|
||||
ResultSet results = statement.executeQuery();
|
||||
if (results.next())
|
||||
{
|
||||
result =
|
||||
new Pair<Integer, Triple<String, byte[], byte[] > >(
|
||||
results.getInt("id"),
|
||||
new Triple<String, byte[], byte[]> (
|
||||
results.getString("version"),
|
||||
Base64.decode(results.getString("v")),
|
||||
Base64.decode(results.getString("s"))
|
||||
)
|
||||
);
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public Triple<String, BigInteger, BigInteger> getVVS (String userName) throws IOException, SQLException
|
||||
{
|
||||
Triple<String, byte[], byte[]> vvs = getUser(userName).second;
|
||||
return
|
||||
new Triple<String, BigInteger, BigInteger>(
|
||||
vvs.first,
|
||||
new BigInteger (vvs.second),
|
||||
new BigInteger (vvs.third)
|
||||
);
|
||||
}
|
||||
|
||||
protected byte[] setMailBlock (String userName, byte[] block) throws IOException, SQLException
|
||||
{
|
||||
Integer id = getUser(userName).first;
|
||||
|
||||
Connection connection = openConnection();
|
||||
|
||||
byte[] result = null;
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.SET_USER_MAIL_BLOCK));
|
||||
statement.setInt(1, id);
|
||||
statement.setString (2, Base64.encode(block));
|
||||
log(statement);
|
||||
|
||||
statement.executeUpdate();
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
abstract public byte[] setBlock (String userName, byte[] block) throws IOException, SQLException, CryptoException;
|
||||
abstract public byte[] getBlock (String userName) throws IOException, SQLException, CryptoException;
|
||||
|
||||
protected byte[] setKeyBlock (String userName, byte[] block) throws IOException, SQLException
|
||||
{
|
||||
Integer id = getUser(userName).first;
|
||||
|
||||
Connection connection = openConnection();
|
||||
|
||||
byte[] result = null;
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.SET_USER_KEY_BLOCK));
|
||||
statement.setInt(1, id);
|
||||
statement.setString (2, Base64.encode(block));
|
||||
log(statement);
|
||||
|
||||
statement.executeUpdate();
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
protected byte[] getMailBlock (String userName) throws IOException, SQLException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
byte[] result = null;
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.GET_USER_MAIL_BLOCK));
|
||||
statement.setString(1, userName);
|
||||
log(statement);
|
||||
|
||||
ResultSet results = statement.executeQuery();
|
||||
if (results.next())
|
||||
result = Base64.decode(results.getString("block"));
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public byte[] getDeletedMailBlock(String userName) throws IOException, SQLException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
byte[] result = null;
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.GET_DELETED_USER_MAIL_BLOCK));
|
||||
statement.setString(1, userName);
|
||||
log(statement);
|
||||
|
||||
ResultSet results = statement.executeQuery();
|
||||
if (results.next())
|
||||
result = Base64.decode(results.getString("block"));
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public String getDeletedUser() throws IOException, SQLException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
String result = null;
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.GET_DELETED_USER));
|
||||
log(statement);
|
||||
|
||||
ResultSet results = statement.executeQuery();
|
||||
if (results.next())
|
||||
result = results.getString("name");
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
protected byte[] getKeyBlock (String userName) throws IOException, SQLException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
byte[] result = null;
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.GET_USER_KEY_BLOCK));
|
||||
statement.setString(1, userName);
|
||||
log(statement);
|
||||
|
||||
ResultSet results = statement.executeQuery();
|
||||
if (results.next())
|
||||
result = Base64.decode(results.getString("block"));
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public void ensureTables() throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
try
|
||||
{
|
||||
for (String sql : catalog.getMulti(catalog.CREATE_TABLES))
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (sql);
|
||||
log(statement);
|
||||
statement.executeUpdate();
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public void rateLimitFailure (String userName) throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.GET_LAST_FAILURE));
|
||||
statement.setString(1, userName);
|
||||
log(statement);
|
||||
|
||||
ResultSet rs = statement.executeQuery();
|
||||
if (rs.next())
|
||||
{
|
||||
Timestamp timeStamp = rs.getTimestamp("mark");
|
||||
Date now = new Date();
|
||||
|
||||
if (now.getTime() - timeStamp.getTime() < catalog.FAILURE_TIMEOUT_SECONDS * 1000)
|
||||
throw new PublicMessageException ("Too many failures, try again later.");
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public void markFailure (String userName) throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(catalog.MARK_FAILURE));
|
||||
statement.setString(1, userName);
|
||||
statement.executeUpdate();
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public void deleteUser(String userName) throws IOException, SQLException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
try
|
||||
{
|
||||
String[] texts = catalog.getMulti(catalog.DELETE);
|
||||
for (String text : texts)
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (text);
|
||||
statement.setString(1, userName);
|
||||
|
||||
log(statement);
|
||||
statement.executeUpdate();
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public void expungeUser(String userName) throws IOException, SQLException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
try
|
||||
{
|
||||
String[] texts = catalog.getMulti(catalog.EXPUNGE);
|
||||
for (String text : texts)
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (text);
|
||||
statement.setString(1, userName);
|
||||
statement.executeUpdate();
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public Connection openConnection () throws IOException, SQLException
|
||||
{
|
||||
log.debug("Connecting to", catalog.CONNECTION_STRING);
|
||||
return DriverManager.getConnection(catalog.CONNECTION_STRING, catalog.USER, Passwords.getPasswordFor(catalog.USER));
|
||||
}
|
||||
|
||||
public void closeConnection (Connection connection)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (connection != null)
|
||||
connection.close();
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
}
|
||||
public void log (Statement sql)
|
||||
{
|
||||
System.out.println (sql);
|
||||
}
|
||||
|
||||
protected void testIllegalUserName(String userName) throws Exception
|
||||
{
|
||||
// http://www.ietf.org/rfc/rfc2142.txt
|
||||
final String[] illegalStartsWith = {
|
||||
"info",
|
||||
"marketing",
|
||||
"sales",
|
||||
"support",
|
||||
|
||||
"abuse",
|
||||
"noc",
|
||||
"security",
|
||||
|
||||
"postmaster",
|
||||
"hostmaster",
|
||||
"usenet",
|
||||
"news",
|
||||
"webmaster",
|
||||
"www",
|
||||
"uucp",
|
||||
"ftp",
|
||||
|
||||
"admin",
|
||||
"system",
|
||||
"root",
|
||||
"test",
|
||||
"root",
|
||||
"hostma",
|
||||
"web",
|
||||
"post",
|
||||
"mail",
|
||||
};
|
||||
|
||||
final String[] illegalParts = {
|
||||
"postmaster",
|
||||
"webmaster",
|
||||
"root",
|
||||
"admin",
|
||||
"system",
|
||||
};
|
||||
|
||||
String username = userName.toLowerCase();
|
||||
for (String illegal : illegalParts)
|
||||
{
|
||||
if (username.indexOf(illegal) != -1)
|
||||
throw new Exception("Illegal username");
|
||||
}
|
||||
|
||||
for (String illegal : illegalStartsWith)
|
||||
{
|
||||
if (username.startsWith(illegal))
|
||||
throw new Exception("Illegal username");
|
||||
}
|
||||
}
|
||||
}
|
51
java/core/src/core/server/db/sql/Catalog.java
Normal file
51
java/core/src/core/server/db/sql/Catalog.java
Normal file
@ -0,0 +1,51 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.server.srp.db.sql;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.sql.SQLException;
|
||||
|
||||
import core.constants.ConstantsServer;
|
||||
import core.util.Passwords;
|
||||
import core.util.SqlCatalog;
|
||||
import core.util.Streams;
|
||||
|
||||
|
||||
public final class Catalog extends SqlCatalog
|
||||
{
|
||||
public String CONNECTION_STRING = ConstantsServer.DBCONNECTION_PREFIX + "mail";
|
||||
public String USER = "mail";
|
||||
|
||||
public int FAILURE_TIMEOUT_SECONDS = 60;
|
||||
|
||||
public final String
|
||||
CREATE_TABLES = "create_tables",
|
||||
CREATE_USER = "create_user",
|
||||
GET_USER = "get_user",
|
||||
GET_USER_MAIL_BLOCK = "get_user_mail_block",
|
||||
SET_USER_MAIL_BLOCK = "set_user_mail_block",
|
||||
GET_USER_KEY_BLOCK = "get_user_key_block",
|
||||
SET_USER_KEY_BLOCK = "set_user_key_block",
|
||||
GET_LAST_FAILURE = "get_last_failure",
|
||||
MARK_FAILURE = "mark_failure",
|
||||
|
||||
DELETE = "delete_user.sql",
|
||||
EXPUNGE = "expunge_deleted_user.sql",
|
||||
GET_DELETED_USER = "get_deleted_user.sql",
|
||||
GET_DELETED_USER_MAIL_BLOCK = "get_deleted_user_mail_block.sql",
|
||||
|
||||
ROOM_FOR_NEW_USER = "room_for_new_user";
|
||||
|
||||
public Catalog ()
|
||||
{
|
||||
|
||||
}
|
||||
|
||||
public String getPassword () throws IOException
|
||||
{
|
||||
return Passwords.getPasswordFor(USER);
|
||||
}
|
||||
}
|
55
java/core/src/core/server/db/sql/create_tables
Normal file
55
java/core/src/core/server/db/sql/create_tables
Normal file
@ -0,0 +1,55 @@
|
||||
CREATE TABLE IF NOT EXISTS registry (
|
||||
k VARCHAR(50),
|
||||
v VARCHAR(255),
|
||||
mark TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
PRIMARY KEY (k)
|
||||
);
|
||||
|
||||
INSERT IGNORE INTO registry (k,v) VALUES ("max_users", "1000");
|
||||
|
||||
CREATE TABLE IF NOT EXISTS user (
|
||||
version VARCHAR(50),
|
||||
id INTEGER AUTO_INCREMENT,
|
||||
name VARCHAR(255) COLLATE utf8_general_ci NOT NULL DEFAULT '',
|
||||
v TEXT,
|
||||
s TEXT,
|
||||
mark TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
PRIMARY KEY (name),
|
||||
UNIQUE KEY (id)
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS mail_block (
|
||||
user_id INTEGER,
|
||||
block TEXT,
|
||||
mark TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
PRIMARY KEY (user_id)
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS key_block (
|
||||
user_id INTEGER,
|
||||
block TEXT,
|
||||
mark TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
PRIMARY KEY (user_id)
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS failure (
|
||||
user_id INTEGER,
|
||||
mark TIMESTAMP,
|
||||
PRIMARY KEY (user_id)
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS deleted_user (
|
||||
version VARCHAR(50),
|
||||
id INTEGER AUTO_INCREMENT,
|
||||
name VARCHAR(255) COLLATE utf8_general_ci NOT NULL DEFAULT '',
|
||||
v TEXT,
|
||||
s TEXT,
|
||||
mark TIMESTAMP NULL DEFAULT NULL,
|
||||
deleted TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
|
||||
PRIMARY KEY (id)
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS deleted_mail_block LIKE mail_block;
|
||||
CREATE TABLE IF NOT EXISTS deleted_key_block LIKE key_block;
|
||||
DROP TABLE IF EXISTS metadata;
|
||||
|
1
java/core/src/core/server/db/sql/create_user
Normal file
1
java/core/src/core/server/db/sql/create_user
Normal file
@ -0,0 +1 @@
|
||||
INSERT INTO user (version, name, v, s) VALUES (?, ?, ?, ?)
|
24
java/core/src/core/server/db/sql/delete_user.sql
Normal file
24
java/core/src/core/server/db/sql/delete_user.sql
Normal file
@ -0,0 +1,24 @@
|
||||
REPLACE INTO deleted_mail_block
|
||||
SELECT mail_block.*
|
||||
FROM mail_block, user
|
||||
WHERE user.name=? AND mail_block.user_id = user.id;
|
||||
|
||||
REPLACE INTO deleted_key_block
|
||||
SELECT key_block.*
|
||||
FROM key_block, user
|
||||
WHERE user.name=? AND key_block.user_id = user.id;
|
||||
|
||||
REPLACE INTO deleted_user(version, id, name, v, s, mark)
|
||||
SELECT version, id, name, v, s, mark
|
||||
FROM user
|
||||
WHERE user.name=?;
|
||||
|
||||
DELETE mail_block.*
|
||||
FROM mail_block, user
|
||||
WHERE user.name=? AND mail_block.user_id = user.id;
|
||||
|
||||
DELETE key_block.*
|
||||
FROM key_block, user
|
||||
WHERE user.name=? AND key_block.user_id = user.id;
|
||||
|
||||
DELETE FROM user WHERE user.name=?
|
@ -0,0 +1,9 @@
|
||||
DELETE deleted_mail_block.*
|
||||
FROM deleted_mail_block, deleted_user
|
||||
WHERE deleted_user.name=? AND deleted_mail_block.user_id = deleted_user.id;
|
||||
|
||||
DELETE deleted_key_block.*
|
||||
FROM deleted_key_block, deleted_user
|
||||
WHERE deleted_user.name=? AND deleted_key_block.user_id = deleted_user.id;
|
||||
|
||||
DELETE FROM deleted_user WHERE deleted_user.name=?
|
7
java/core/src/core/server/db/sql/get_deleted_user.sql
Normal file
7
java/core/src/core/server/db/sql/get_deleted_user.sql
Normal file
@ -0,0 +1,7 @@
|
||||
select
|
||||
*
|
||||
from
|
||||
deleted_user
|
||||
order by
|
||||
deleted
|
||||
limit 1
|
@ -0,0 +1,8 @@
|
||||
SELECT
|
||||
deleted_mail_block.*
|
||||
FROM
|
||||
deleted_user,
|
||||
deleted_mail_block
|
||||
WHERE
|
||||
deleted_mail_block.user_id = deleted_user.id AND
|
||||
deleted_user.name = ?
|
8
java/core/src/core/server/db/sql/get_last_failure
Normal file
8
java/core/src/core/server/db/sql/get_last_failure
Normal file
@ -0,0 +1,8 @@
|
||||
SELECT
|
||||
failure.mark
|
||||
FROM
|
||||
user,
|
||||
failure
|
||||
WHERE
|
||||
user.name = ? AND
|
||||
failure.user_id = user.id
|
6
java/core/src/core/server/db/sql/get_user
Normal file
6
java/core/src/core/server/db/sql/get_user
Normal file
@ -0,0 +1,6 @@
|
||||
SELECT
|
||||
*
|
||||
FROM
|
||||
user
|
||||
WHERE
|
||||
user.name = ?
|
8
java/core/src/core/server/db/sql/get_user_key_block
Normal file
8
java/core/src/core/server/db/sql/get_user_key_block
Normal file
@ -0,0 +1,8 @@
|
||||
SELECT
|
||||
key_block.*
|
||||
FROM
|
||||
user,
|
||||
key_block
|
||||
WHERE
|
||||
key_block.user_id = user.id AND
|
||||
user.name = ?
|
8
java/core/src/core/server/db/sql/get_user_mail_block
Normal file
8
java/core/src/core/server/db/sql/get_user_mail_block
Normal file
@ -0,0 +1,8 @@
|
||||
SELECT
|
||||
mail_block.*
|
||||
FROM
|
||||
user,
|
||||
mail_block
|
||||
WHERE
|
||||
mail_block.user_id = user.id AND
|
||||
user.name = ?
|
8
java/core/src/core/server/db/sql/mark_failure
Normal file
8
java/core/src/core/server/db/sql/mark_failure
Normal file
@ -0,0 +1,8 @@
|
||||
REPLACE INTO failure (user_id, mark)
|
||||
SELECT
|
||||
id as user_id,
|
||||
now() as mark
|
||||
FROM
|
||||
user
|
||||
WHERE
|
||||
name = ?
|
4
java/core/src/core/server/db/sql/room_for_new_user
Normal file
4
java/core/src/core/server/db/sql/room_for_new_user
Normal file
@ -0,0 +1,4 @@
|
||||
select
|
||||
count(*) < (select convert(v, unsigned integer) from registry where k="max_users") as room
|
||||
from
|
||||
user
|
1
java/core/src/core/server/db/sql/set_user_key_block
Normal file
1
java/core/src/core/server/db/sql/set_user_key_block
Normal file
@ -0,0 +1 @@
|
||||
REPLACE INTO key_block (user_id, block) VALUES (?, ?)
|
1
java/core/src/core/server/db/sql/set_user_mail_block
Normal file
1
java/core/src/core/server/db/sql/set_user_mail_block
Normal file
@ -0,0 +1 @@
|
||||
REPLACE INTO mail_block (user_id, block) VALUES (?, ?)
|
51
java/core/src/core/server/log/Tail.java
Normal file
51
java/core/src/core/server/log/Tail.java
Normal file
@ -0,0 +1,51 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.server.log;
|
||||
|
||||
import java.io.File;
|
||||
import java.io.FileNotFoundException;
|
||||
import java.io.IOException;
|
||||
import java.io.RandomAccessFile;
|
||||
import java.util.Date;
|
||||
|
||||
public class Tail
|
||||
{
|
||||
static final int TAIL_INTERVAL = 1000;
|
||||
|
||||
String fileName;
|
||||
String lastTail;
|
||||
Date lastTailTime;
|
||||
|
||||
public Tail (String fileName)
|
||||
{
|
||||
this.fileName = fileName;
|
||||
}
|
||||
|
||||
public String getTail () throws IOException
|
||||
{
|
||||
Date now = new Date();
|
||||
if (lastTailTime != null &&
|
||||
now.getTime() < lastTailTime.getTime() + TAIL_INTERVAL)
|
||||
return lastTail;
|
||||
|
||||
lastTailTime = now;
|
||||
lastTail = null;
|
||||
|
||||
File file = new File(fileName);
|
||||
RandomAccessFile fileHandler = new RandomAccessFile( file, "r" );
|
||||
long fileLength = file.length() - 1;
|
||||
long blockLength = Math.min(fileLength, 4096);
|
||||
|
||||
fileHandler.seek(fileLength - blockLength);
|
||||
|
||||
byte[] bytes = new byte[(int) blockLength];
|
||||
fileHandler.readFully(bytes);
|
||||
fileHandler.close();
|
||||
|
||||
lastTail = new String(bytes, "UTF-8");
|
||||
return lastTail;
|
||||
}
|
||||
}
|
220
java/core/src/core/server/mailextra/MailExtraDb.java
Normal file
220
java/core/src/core/server/mailextra/MailExtraDb.java
Normal file
@ -0,0 +1,220 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.server.mailextra;
|
||||
|
||||
import java.io.IOException;
|
||||
import java.sql.Connection;
|
||||
import java.sql.DriverManager;
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
|
||||
import org.bc.util.Strings;
|
||||
import org.json.JSONException;
|
||||
import org.json.JSONObject;
|
||||
|
||||
import core.util.Base64;
|
||||
import core.constants.ConstantsPushNotifications;
|
||||
import core.crypt.HashSha256;
|
||||
import core.server.mailextra.sql.Catalog;
|
||||
import core.util.LogOut;
|
||||
import core.util.Pair;
|
||||
import core.util.Passwords;
|
||||
import core.util.Triple;
|
||||
|
||||
public class MailExtraDb
|
||||
{
|
||||
LogOut log = new LogOut(MailExtraDb.class);
|
||||
Catalog catalog = new Catalog();
|
||||
|
||||
public String getUserHash (String email)
|
||||
{
|
||||
HashSha256 hasher = new HashSha256();
|
||||
return Base64.encode(hasher.hash(Strings.toByteArray(email.toLowerCase())));
|
||||
}
|
||||
|
||||
public Triple<String,String,String> getDeviceFor (String email) throws SQLException, IOException
|
||||
{
|
||||
return getDeviceForHash(getUserHash(email));
|
||||
}
|
||||
|
||||
public Triple<String,String,String> getDeviceForHash (String hash) throws SQLException, IOException
|
||||
{
|
||||
|
||||
Connection connection = openConnection();
|
||||
Triple<String,String,String> result;
|
||||
|
||||
try
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(Catalog.GET_NOTIFICATIONS_FOR));
|
||||
statement.setString(1, hash);
|
||||
log.debug(statement);
|
||||
|
||||
ResultSet rs = statement.executeQuery();
|
||||
if (!rs.next())
|
||||
return null;
|
||||
|
||||
result = new Triple<String,String,String>(rs.getString("notification_type"), rs.getString("device_type"), rs.getString("device_id"));
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public void setNotificationsFor (String email, String notificationType, String deviceType, String deviceId) throws SQLException, IOException
|
||||
{
|
||||
setNotificationsFor (getUserHash(email), deviceType, deviceId, notificationType);
|
||||
}
|
||||
|
||||
public void setNotificationsForHash (String hash, String notificationType, String deviceType, String deviceId) throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
String statements[] = catalog.getMulti(Catalog.SET_NOTIFICATIONS_FOR);
|
||||
|
||||
boolean first = true;
|
||||
for (String s : statements)
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (s);
|
||||
|
||||
if (first)
|
||||
{
|
||||
statement.setString(1, hash);
|
||||
statement.setString(2, notificationType);
|
||||
statement.setString(3, deviceType);
|
||||
statement.setString(4, deviceId);
|
||||
}
|
||||
|
||||
log.debug(statement);
|
||||
statement.executeUpdate();
|
||||
first = false;
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public void addDaysTo (String email, int days) throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
String statements[] = catalog.getMulti(Catalog.ADD_DAYS_TO);
|
||||
|
||||
boolean first = true;
|
||||
for (String s : statements)
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (s);
|
||||
|
||||
if (first)
|
||||
{
|
||||
statement.setString(1, email);
|
||||
statement.setInt(2, days);
|
||||
}
|
||||
|
||||
|
||||
log.debug(statement);
|
||||
statement.executeUpdate();
|
||||
first = false;
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public int getDaysLeft (String email) throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
log.debug(catalog.getSingle(Catalog.GET_DAYS_LEFT));
|
||||
PreparedStatement statement = connection.prepareStatement (catalog.getSingle(Catalog.GET_DAYS_LEFT));
|
||||
statement.setString(1, email);
|
||||
log.debug(statement);
|
||||
|
||||
ResultSet rs = statement.executeQuery();
|
||||
if (!rs.next())
|
||||
throw new IOException ("No results");
|
||||
|
||||
int days = rs.getInt("days");
|
||||
rs.close();
|
||||
|
||||
return days;
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public void ensureTables() throws SQLException, IOException
|
||||
{
|
||||
Connection connection = openConnection();
|
||||
|
||||
try
|
||||
{
|
||||
for (String sql : catalog.getMulti(Catalog.CREATE_TABLES))
|
||||
{
|
||||
PreparedStatement statement = connection.prepareStatement (sql);
|
||||
log.debug(statement);
|
||||
statement.executeUpdate();
|
||||
}
|
||||
}
|
||||
finally
|
||||
{
|
||||
closeConnection(connection);
|
||||
}
|
||||
}
|
||||
|
||||
public Connection openConnection () throws SQLException, IOException
|
||||
{
|
||||
return DriverManager.getConnection(Catalog.CONNECTION_STRING, Catalog.USER, Passwords.getPasswordFor(Catalog.USER));
|
||||
}
|
||||
|
||||
public void closeConnection (Connection connection)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (connection != null)
|
||||
connection.close();
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public void handlePushNotificationsJson(JSONObject json) throws JSONException, SQLException, IOException
|
||||
{
|
||||
String user = json.getString(ConstantsPushNotifications.USER);
|
||||
|
||||
setNotificationsForHash(
|
||||
user,
|
||||
json.getString(ConstantsPushNotifications.NOTIFICATION_TYPE),
|
||||
json.has(ConstantsPushNotifications.DEVICE_TYPE) ?
|
||||
json.getString(ConstantsPushNotifications.DEVICE_TYPE) : null,
|
||||
json.has(ConstantsPushNotifications.DEVICE_ID) ?
|
||||
json.getString(ConstantsPushNotifications.DEVICE_ID) : null
|
||||
);
|
||||
}
|
||||
|
||||
public void removeDevice(String deviceType, String deviceId)
|
||||
{
|
||||
}
|
||||
|
||||
}
|
25
java/core/src/core/server/mailextra/sql/Catalog.java
Normal file
25
java/core/src/core/server/mailextra/sql/Catalog.java
Normal file
@ -0,0 +1,25 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.server.mailextra.sql;
|
||||
|
||||
import java.io.IOException;
|
||||
|
||||
import core.util.SqlCatalog;
|
||||
import core.util.Streams;
|
||||
|
||||
public class Catalog extends SqlCatalog
|
||||
{
|
||||
static public final String CONNECTION_STRING = "jdbc:mysql://localhost/mail_extra";
|
||||
public static final String USER = "mail_extra";
|
||||
|
||||
static public final String GET_NOTIFICATIONS_FOR = "get_notifications_for.sql";
|
||||
static public final String SET_NOTIFICATIONS_FOR = "set_notifications_for.sql";
|
||||
static public final String PRUNE_DEVICES = "prune_devices.sql";
|
||||
|
||||
static public final String CREATE_TABLES = "create_tables";
|
||||
static public final String ADD_DAYS_TO = "add_days_to";
|
||||
static public final String GET_DAYS_LEFT = "get_days_left";
|
||||
}
|
10
java/core/src/core/server/mailextra/sql/add_days_to
Normal file
10
java/core/src/core/server/mailextra/sql/add_days_to
Normal file
@ -0,0 +1,10 @@
|
||||
SET @email = ?, @days = ?;
|
||||
|
||||
INSERT INTO expirations (email, expiration) VALUES(@email, DATE_ADD(CURDATE(), INTERVAL @days DAY))
|
||||
ON DUPLICATE KEY UPDATE
|
||||
expiration =
|
||||
IF(
|
||||
DATEDIFF(expiration, CURDATE())>0,
|
||||
DATE_ADD(expiration, INTERVAL @days DAY),
|
||||
VALUES(expiration)
|
||||
)
|
15
java/core/src/core/server/mailextra/sql/create_tables
Normal file
15
java/core/src/core/server/mailextra/sql/create_tables
Normal file
@ -0,0 +1,15 @@
|
||||
CREATE TABLE IF NOT EXISTS expirations (
|
||||
email VARCHAR(255) NOT NULL,
|
||||
expiration DATE NOT NULL,
|
||||
PRIMARY KEY (email)
|
||||
);
|
||||
|
||||
CREATE TABLE IF NOT EXISTS `device` (
|
||||
`email` VARCHAR(255) NOT NULL,
|
||||
`notification_type` ENUM('NONE','SHORT','LONG') NOT NULL,
|
||||
`device_type` VARCHAR(255) DEFAULT NULL,
|
||||
`device_id` VARCHAR(255) DEFAULT NULL,
|
||||
`mark` date NOT NULL,
|
||||
PRIMARY KEY (`email`),
|
||||
KEY `device` (`device_type`,`device_id`)
|
||||
);
|
6
java/core/src/core/server/mailextra/sql/get_days_left
Normal file
6
java/core/src/core/server/mailextra/sql/get_days_left
Normal file
@ -0,0 +1,6 @@
|
||||
SELECT
|
||||
DATEDIFF(expiration, CURDATE()) as days
|
||||
FROM
|
||||
expirations
|
||||
WHERE
|
||||
expirations.email = ?
|
@ -0,0 +1,10 @@
|
||||
SELECT
|
||||
device.notification_type,
|
||||
device.device_type,
|
||||
device.device_id,
|
||||
device.mark
|
||||
FROM
|
||||
device
|
||||
WHERE
|
||||
device.email = ?
|
||||
|
@ -0,0 +1,4 @@
|
||||
SET @email = ?, @notification_type = ?, @device_type = ?, @device_id = ?;
|
||||
|
||||
DELETE FROM device WHERE device_type = @device_type AND device_id = @device_id;
|
||||
REPLACE INTO device (email, device_type, device_id, notification_type, mark) VALUES(@email, @device_type, @device_id, @notification_type, now())
|
301
java/core/src/core/srp/SRPPacketSerializerJSON.java
Normal file
301
java/core/src/core/srp/SRPPacketSerializerJSON.java
Normal file
@ -0,0 +1,301 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp;
|
||||
|
||||
import java.math.BigInteger;
|
||||
|
||||
import core.util.Base64;
|
||||
import org.json.JSONObject;
|
||||
|
||||
import core.util.CallSingle;
|
||||
import core.util.JSONRegistry;
|
||||
import core.util.Strings;
|
||||
|
||||
|
||||
public class SRPPacketSerializerJSON
|
||||
{
|
||||
static public void register () {}
|
||||
|
||||
static {
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$Packet1_ClientSendsHello",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.Packet1_ClientSendsHello p = (SRPPackets.Packet1_ClientSendsHello)v;
|
||||
object.put("user", p.user);
|
||||
object.put("version", p.version);
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
return new SRPPackets.Packet1_ClientSendsHello(
|
||||
v.get("user").toString(),
|
||||
v.get("version").toString()
|
||||
);
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$Packet2_ServerSendSalt",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.Packet2_ServerSendSalt p = (SRPPackets.Packet2_ServerSendSalt)v;
|
||||
object.put("salt", Base64.encode(p.salt));
|
||||
object.put("version", p.version);
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
return new SRPPackets.Packet2_ServerSendSalt(
|
||||
v.get("version").toString(),
|
||||
Base64.decode(v.get("salt").toString())
|
||||
);
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$Packet3_ClientSendPublicKey",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.Packet3_ClientSendPublicKey p = (SRPPackets.Packet3_ClientSendPublicKey)v;
|
||||
object.put("publicKey", new String(Base64.encode(p.publicKey)));
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
return new SRPPackets.Packet3_ClientSendPublicKey(
|
||||
Base64.decode(v.get("publicKey").toString())
|
||||
);
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$Packet4_ServerSendPublicKey",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.Packet4_ServerSendPublicKey p = (SRPPackets.Packet4_ServerSendPublicKey)v;
|
||||
object.put("publicKey", new String(Base64.encode(p.publicKey)));
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
return new SRPPackets.Packet4_ServerSendPublicKey(
|
||||
new BigInteger(Base64.decode(v.get("publicKey").toString()))
|
||||
);
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$Packet5_ClientSendEvidence",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.Packet5_ClientSendEvidence p = (SRPPackets.Packet5_ClientSendEvidence)v;
|
||||
object.put("evidence", new String(Base64.encode(p.evidence)));
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
return new SRPPackets.Packet5_ClientSendEvidence(
|
||||
Base64.decode(v.get("evidence").toString())
|
||||
);
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$Packet6_ServerSendEvidenceAndPayload",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.Packet6_ServerSendEvidenceAndPayload p = (SRPPackets.Packet6_ServerSendEvidenceAndPayload)v;
|
||||
object.put("evidence", new String(Base64.encode(p.evidence)));
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
return new SRPPackets.Packet6_ServerSendEvidenceAndPayload(
|
||||
new BigInteger(Base64.decode(v.get("evidence").toString()))
|
||||
);
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$PacketInit_ClientTestCreate",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.PacketInit_ClientTestCreate p = (SRPPackets.PacketInit_ClientTestCreate)v;
|
||||
object.put("user", p.user);
|
||||
object.put("version", p.version);
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
return new
|
||||
SRPPackets.PacketInit_ClientTestCreate(
|
||||
v.getString("version"),
|
||||
v.getString("user")
|
||||
);
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$PacketInit_EncryptedPacket",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.PacketInit_EncryptedPacket p = (SRPPackets.PacketInit_EncryptedPacket)v;
|
||||
object.put("encryptedBlock", new String(Base64.encode(p.encryptedBlock)));
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
return new
|
||||
SRPPackets.PacketInit_EncryptedPacket(
|
||||
Base64.decode(v.getString("encryptedBlock"))
|
||||
);
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$PacketInit_ServerResponse",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.PacketInit_ServerResponse p = (SRPPackets.PacketInit_ServerResponse)v;
|
||||
object.put("succeeded", p.succeeded);
|
||||
object.put("reason", p.reason);
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
return new
|
||||
SRPPackets.PacketInit_ServerResponse(
|
||||
v.getBoolean("succeeded"),
|
||||
v.has("reason") ? v.getString("reason") : null
|
||||
);
|
||||
}
|
||||
}
|
||||
);
|
||||
|
||||
|
||||
JSONRegistry.register (
|
||||
"core.srp.SRPPackets$PacketInit_ClientPreAutheticationInitialization",
|
||||
new CallSingle<JSONObject,Object> () {
|
||||
@Override
|
||||
public JSONObject invoke(Object v) throws Exception
|
||||
{
|
||||
JSONObject object = new JSONObject();
|
||||
|
||||
SRPPackets.PacketInit_ClientPreAutheticationInitialization p = (SRPPackets.PacketInit_ClientPreAutheticationInitialization)v;
|
||||
object.put("version", p.version);
|
||||
object.put("user", p.user);
|
||||
object.put("s", new String(Base64.encode(p.s)));
|
||||
object.put("v", new String(Base64.encode(p.v)));
|
||||
|
||||
if (p.extra != null)
|
||||
object.put("extra", new String(Base64.encode(p.extra)));
|
||||
|
||||
return object;
|
||||
}
|
||||
},
|
||||
new CallSingle<Object,JSONObject> () {
|
||||
@Override
|
||||
public Object invoke(JSONObject v) throws Exception
|
||||
{
|
||||
SRPPackets.PacketInit_ClientPreAutheticationInitialization p =
|
||||
new SRPPackets.PacketInit_ClientPreAutheticationInitialization(
|
||||
v.getString("version"),
|
||||
v.getString("user"),
|
||||
Base64.decode(v.getString("v")),
|
||||
Base64.decode(v.getString("s"))
|
||||
);
|
||||
|
||||
if (v.has("extra"))
|
||||
p.extra = Base64.decode(v.getString("extra"));
|
||||
|
||||
return p;
|
||||
}
|
||||
}
|
||||
);
|
||||
}
|
||||
}
|
182
java/core/src/core/srp/SRPPackets.java
Normal file
182
java/core/src/core/srp/SRPPackets.java
Normal file
@ -0,0 +1,182 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp;
|
||||
|
||||
import java.math.BigInteger;
|
||||
|
||||
import core.callback.Callback;
|
||||
import core.callback.CallbackDefault;
|
||||
|
||||
|
||||
public class SRPPackets
|
||||
{
|
||||
public static void register () { }
|
||||
static {
|
||||
SRPPacketSerializerJSON.register();
|
||||
}
|
||||
|
||||
public static class PacketInit_ClientTestCreate
|
||||
{
|
||||
public String version;
|
||||
public String user;
|
||||
|
||||
public PacketInit_ClientTestCreate(String version, String user)
|
||||
{
|
||||
this.version = version;
|
||||
this.user = user;
|
||||
}
|
||||
}
|
||||
|
||||
public static class PacketInit_ClientPreAutheticationInitialization
|
||||
{
|
||||
public String user;
|
||||
public byte[] v;
|
||||
public byte[] s;
|
||||
public byte[] extra;
|
||||
public String version;
|
||||
|
||||
public PacketInit_ClientPreAutheticationInitialization(String version, String user, byte[] v, byte[] s)
|
||||
{
|
||||
this.version = version;
|
||||
this.user = user;
|
||||
this.v = v;
|
||||
this.s = s;
|
||||
extra = null;
|
||||
}
|
||||
}
|
||||
|
||||
public static class PacketInit_EncryptedPacket
|
||||
{
|
||||
public byte[] encryptedBlock;
|
||||
|
||||
public PacketInit_EncryptedPacket (byte[] encryptedBlock)
|
||||
{
|
||||
this.encryptedBlock = encryptedBlock;
|
||||
}
|
||||
|
||||
public static Callback wrap_()
|
||||
{
|
||||
return new CallbackDefault() {
|
||||
@Override
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
callback.invoke(new PacketInit_EncryptedPacket((byte[])(arguments[0])));
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public static Callback unwrap_()
|
||||
{
|
||||
return new CallbackDefault() {
|
||||
@Override
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
PacketInit_EncryptedPacket packet = (PacketInit_EncryptedPacket)(arguments[0]);
|
||||
callback.invoke(packet.encryptedBlock);
|
||||
}
|
||||
};
|
||||
}
|
||||
}
|
||||
|
||||
public static class PacketInit_ServerResponse
|
||||
{
|
||||
public boolean succeeded;
|
||||
public String reason;
|
||||
|
||||
public PacketInit_ServerResponse(boolean succeeded, String reason)
|
||||
{
|
||||
this.succeeded = succeeded;
|
||||
this.reason = reason;
|
||||
}
|
||||
}
|
||||
|
||||
public static class Packet1_ClientSendsHello
|
||||
{
|
||||
public String user;
|
||||
public String version;
|
||||
|
||||
public Packet1_ClientSendsHello (String user, String version)
|
||||
{
|
||||
this.user = user;
|
||||
this.version = version;
|
||||
}
|
||||
}
|
||||
|
||||
public static class Packet2_ServerSendSalt
|
||||
{
|
||||
public byte[] salt;
|
||||
public String version;
|
||||
|
||||
public Packet2_ServerSendSalt (String version, byte[] salt)
|
||||
{
|
||||
this.salt = salt;
|
||||
this.version = version;
|
||||
}
|
||||
|
||||
public byte[] getSalt ()
|
||||
{
|
||||
return salt;
|
||||
}
|
||||
}
|
||||
|
||||
public static class Packet3_ClientSendPublicKey
|
||||
{
|
||||
public byte[] publicKey;
|
||||
|
||||
public Packet3_ClientSendPublicKey (byte[] publicKey)
|
||||
{
|
||||
this.publicKey = publicKey;
|
||||
}
|
||||
|
||||
public byte[] getPublicKey ()
|
||||
{
|
||||
return publicKey;
|
||||
}
|
||||
}
|
||||
|
||||
public static class Packet4_ServerSendPublicKey
|
||||
{
|
||||
public byte[] publicKey;
|
||||
|
||||
public Packet4_ServerSendPublicKey (BigInteger publicKey)
|
||||
{
|
||||
this.publicKey = publicKey.toByteArray();
|
||||
}
|
||||
|
||||
public byte[] getPublicKey ()
|
||||
{
|
||||
return publicKey;
|
||||
}
|
||||
}
|
||||
|
||||
public static class Packet5_ClientSendEvidence
|
||||
{
|
||||
public byte[] evidence;
|
||||
|
||||
public Packet5_ClientSendEvidence (byte[] evidence)
|
||||
{
|
||||
this.evidence = evidence;
|
||||
}
|
||||
|
||||
public byte[] getEvidence ()
|
||||
{
|
||||
return evidence;
|
||||
}
|
||||
}
|
||||
|
||||
public static class Packet6_ServerSendEvidenceAndPayload
|
||||
{
|
||||
public byte[] evidence;
|
||||
|
||||
public Packet6_ServerSendEvidenceAndPayload (BigInteger evidence)
|
||||
{
|
||||
this.evidence = evidence.toByteArray();
|
||||
}
|
||||
|
||||
public byte[] getEvidence ()
|
||||
{
|
||||
return evidence;
|
||||
}
|
||||
}
|
||||
}
|
43
java/core/src/core/srp/SRPSession.java
Normal file
43
java/core/src/core/srp/SRPSession.java
Normal file
@ -0,0 +1,43 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp;
|
||||
|
||||
import core.crypt.Cryptor;
|
||||
import core.crypt.CryptorAES;
|
||||
import core.exceptions.CryptoException;
|
||||
|
||||
|
||||
public class SRPSession
|
||||
{
|
||||
protected byte[] sessionKey;
|
||||
|
||||
public byte[] streamDecrypt (byte[] bytes) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
Cryptor cryptor = new CryptorAES(sessionKey);
|
||||
return cryptor.decrypt(bytes);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
e.printStackTrace();
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public byte[] streamEncrypt (byte[] bytes) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
Cryptor cryptor = new CryptorAES(sessionKey);
|
||||
return cryptor.encrypt(bytes);
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
}
|
91
java/core/src/core/srp/client/SRPClient.java
Normal file
91
java/core/src/core/srp/client/SRPClient.java
Normal file
@ -0,0 +1,91 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp.client;
|
||||
|
||||
import core.callback.Callback;
|
||||
import com.jordanzimmerman.SRPClientSession;
|
||||
import com.jordanzimmerman.SRPFactory;
|
||||
import com.jordanzimmerman.SRPVerifier;
|
||||
|
||||
import core.callback.CallbackDefault;
|
||||
import core.callbacks.JSONSerialize;
|
||||
import core.callbacks.IoSend;
|
||||
import core.crypt.KeyPairFromPassword;
|
||||
import core.exceptions.CryptoException;
|
||||
import core.io.IoChain;
|
||||
import core.srp.SRPPackets;
|
||||
import core.srp.SRPSession;
|
||||
|
||||
public class SRPClient extends SRPSession
|
||||
{
|
||||
static { SRPPackets.register(); }
|
||||
|
||||
SRPFactory factory;
|
||||
SRPClientAsync session;
|
||||
|
||||
public SRPClient (byte[] password)
|
||||
{
|
||||
session = new SRPClientAsync(password);
|
||||
}
|
||||
|
||||
public static SRPVerifier createVerifierFromKeyPair (KeyPairFromPassword keyPair) throws CryptoException
|
||||
{
|
||||
SRPFactory factory = SRPFactory.getInstance();
|
||||
SRPVerifier verifier = factory.makeVerifier(keyPair.getVerifier());
|
||||
|
||||
return verifier;
|
||||
}
|
||||
|
||||
public SRPPackets.Packet1_ClientSendsHello step1_generateHello (String user, String version)
|
||||
{
|
||||
return new SRPPackets.Packet1_ClientSendsHello(user, version);
|
||||
}
|
||||
|
||||
public Callback step2_generatePublicKey_send_ (SRPPackets.Packet2_ServerSendSalt packet, IoChain sender)
|
||||
{
|
||||
return
|
||||
session.setSalt_(packet.getSalt())
|
||||
.addCallback(
|
||||
new CallbackDefault() {
|
||||
public void onSuccess(Object...arguments) {
|
||||
next(new SRPPackets.Packet3_ClientSendPublicKey(session.getPublicKey()));
|
||||
}
|
||||
}
|
||||
)
|
||||
.addCallback(new JSONSerialize())
|
||||
.addCallback(new IoSend(sender));
|
||||
}
|
||||
|
||||
public Callback step3_generateEvidenceUsingServerProvidedInputs_send (SRPPackets.Packet4_ServerSendPublicKey packet, IoChain sender) throws CryptoException
|
||||
{
|
||||
return
|
||||
session.setServerPublicKey_(packet.getPublicKey())
|
||||
.addCallback(
|
||||
new CallbackDefault() {
|
||||
public void onSuccess(Object...arguments) {
|
||||
next(new SRPPackets.Packet5_ClientSendEvidence(session.getEvidenceValue()));
|
||||
}
|
||||
}
|
||||
)
|
||||
.addCallback(new JSONSerialize())
|
||||
.addCallback(new IoSend(sender));
|
||||
}
|
||||
|
||||
public Callback step4_validateServerEvidence (SRPPackets.Packet6_ServerSendEvidenceAndPayload packet) throws CryptoException
|
||||
{
|
||||
return
|
||||
session.validateServerEvidenceValue_M2_(packet.getEvidence())
|
||||
.addCallback(
|
||||
new CallbackDefault() {
|
||||
public void onSuccess(Object...arguments) {
|
||||
sessionKey = session.getSessionKey();
|
||||
next(sessionKey);
|
||||
}
|
||||
}
|
||||
);
|
||||
}
|
||||
|
||||
}
|
71
java/core/src/core/srp/client/SRPClientAsync.java
Normal file
71
java/core/src/core/srp/client/SRPClientAsync.java
Normal file
@ -0,0 +1,71 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp.client;
|
||||
|
||||
import com.jordanzimmerman.SRPClientSession;
|
||||
import com.jordanzimmerman.SRPFactory;
|
||||
|
||||
import core.util.Base64;
|
||||
import core.callback.Callback;
|
||||
import core.callback.CallbackDefault;
|
||||
|
||||
public class SRPClientAsync
|
||||
{
|
||||
SRPClientSession key;
|
||||
|
||||
public SRPClientAsync (byte[] password)
|
||||
{
|
||||
key = SRPFactory.getInstance().newClientSession(password);
|
||||
}
|
||||
|
||||
public Callback setSalt_(byte[] bs)
|
||||
{
|
||||
return new CallbackDefault(bs)
|
||||
{
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
key.setSalt_s((byte[])V(0));
|
||||
callback.invoke();
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public Callback setServerPublicKey_(byte[] publicKey)
|
||||
{
|
||||
return new CallbackDefault(publicKey)
|
||||
{
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
key.setServerPublicKey_B((byte[])V(0));
|
||||
callback.invoke();
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public Callback validateServerEvidenceValue_M2_(byte[] evidence)
|
||||
{
|
||||
return new CallbackDefault(evidence)
|
||||
{
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
key.validateServerEvidenceValue_M2((byte[])V(0));
|
||||
callback.invoke();
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public byte[] getSessionKey()
|
||||
{
|
||||
return key.getSessionKey_K();
|
||||
}
|
||||
|
||||
public byte[] getPublicKey()
|
||||
{
|
||||
return key.getPublicKey_A_();
|
||||
}
|
||||
|
||||
public byte[] getEvidenceValue()
|
||||
{
|
||||
return key.getEvidenceValue_M1_();
|
||||
}
|
||||
}
|
10
java/core/src/core/srp/client/SRPClientListener.java
Normal file
10
java/core/src/core/srp/client/SRPClientListener.java
Normal file
@ -0,0 +1,10 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp.client;
|
||||
|
||||
public interface SRPClientListener {
|
||||
public void onSRPStep (String stepName);
|
||||
}
|
93
java/core/src/core/srp/client/SRPClientUserSession.java
Normal file
93
java/core/src/core/srp/client/SRPClientUserSession.java
Normal file
@ -0,0 +1,93 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp.client;
|
||||
|
||||
import core.callbacks.IoOpen;
|
||||
import core.constants.ConstantsVersion;
|
||||
import core.crypt.KeyPairFromPassword;
|
||||
import core.exceptions.CryptoException;
|
||||
import core.io.IoChain;
|
||||
import core.srp.SRPPackets;
|
||||
import core.util.SimpleSerializer;
|
||||
|
||||
public class SRPClientUserSession extends IoChain
|
||||
{
|
||||
String user;
|
||||
KeyPairFromPassword keyPair;
|
||||
SRPClient srp;
|
||||
SRPClientListener listener;
|
||||
int step;
|
||||
|
||||
public SRPClientUserSession (String user, KeyPairFromPassword keyPair, IoChain sender, SRPClientListener listener) throws CryptoException
|
||||
{
|
||||
super(sender);
|
||||
|
||||
this.user = user;
|
||||
this.keyPair = keyPair;
|
||||
this.listener = listener;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void open () throws Exception
|
||||
{
|
||||
if (listener != null)
|
||||
listener.onSRPStep("H");
|
||||
|
||||
step = 0;
|
||||
this.srp = new SRPClient(keyPair.getVerifier());
|
||||
sender.send(SimpleSerializer.serialize(new SRPPackets.Packet1_ClientSendsHello(user, ConstantsVersion.LOGIN)));
|
||||
}
|
||||
|
||||
@Override
|
||||
protected void onReceive(byte[] bytes) throws Exception
|
||||
{
|
||||
switch (step++)
|
||||
{
|
||||
case 0:
|
||||
{
|
||||
if (listener != null)
|
||||
listener.onSRPStep("" + step);
|
||||
|
||||
SRPPackets.Packet2_ServerSendSalt in = SimpleSerializer.deserialize(bytes);
|
||||
srp.step2_generatePublicKey_send_(in, sender).invoke();
|
||||
} break;
|
||||
|
||||
case 1:
|
||||
{
|
||||
if (listener != null)
|
||||
listener.onSRPStep("" + step);
|
||||
|
||||
SRPPackets.Packet4_ServerSendPublicKey in = SimpleSerializer.deserialize(bytes);
|
||||
srp.step3_generateEvidenceUsingServerProvidedInputs_send(in, sender).invoke();
|
||||
} break;
|
||||
case 2:
|
||||
{
|
||||
if (listener != null)
|
||||
listener.onSRPStep("" + step);
|
||||
|
||||
SRPPackets.Packet6_ServerSendEvidenceAndPayload in = SimpleSerializer.deserialize(bytes);
|
||||
srp.step4_validateServerEvidence(in).addCallback(new IoOpen(receiver)).invoke();
|
||||
} break;
|
||||
|
||||
default:
|
||||
|
||||
if (listener != null)
|
||||
listener.onSRPStep("R");
|
||||
|
||||
super.onReceive(srp.streamDecrypt(bytes));
|
||||
}
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send(byte[] packet) throws Exception
|
||||
{
|
||||
if (listener != null)
|
||||
listener.onSRPStep("S");
|
||||
|
||||
sender.send(srp.streamEncrypt(packet));
|
||||
}
|
||||
|
||||
}
|
77
java/core/src/core/srp/server/SRPServer.java
Normal file
77
java/core/src/core/srp/server/SRPServer.java
Normal file
@ -0,0 +1,77 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp.server;
|
||||
|
||||
import java.math.BigInteger;
|
||||
import java.security.InvalidKeyException;
|
||||
|
||||
import javax.crypto.Cipher;
|
||||
import javax.crypto.spec.SecretKeySpec;
|
||||
|
||||
import org.bc.crypto.params.KeyParameter;
|
||||
|
||||
|
||||
import com.jordanzimmerman.SRPFactory;
|
||||
import com.jordanzimmerman.SRPServerSession;
|
||||
import com.jordanzimmerman.SRPVerifier;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
import core.srp.SRPPackets;
|
||||
import core.srp.SRPSession;
|
||||
import core.srp.SRPPackets.Packet2_ServerSendSalt;
|
||||
import core.srp.SRPPackets.Packet3_ClientSendPublicKey;
|
||||
import core.srp.SRPPackets.Packet4_ServerSendPublicKey;
|
||||
import core.srp.SRPPackets.Packet5_ClientSendEvidence;
|
||||
import core.srp.SRPPackets.Packet6_ServerSendEvidenceAndPayload;
|
||||
|
||||
public class SRPServer extends SRPSession
|
||||
{
|
||||
static { SRPPackets.register(); }
|
||||
|
||||
SRPFactory factory;
|
||||
SRPServerSession session;
|
||||
|
||||
public SRPServer ()
|
||||
{
|
||||
factory = SRPFactory.getInstance();
|
||||
}
|
||||
|
||||
public SRPPackets.Packet2_ServerSendSalt step1_getSalt_send (String version, BigInteger verifier, BigInteger salt)
|
||||
{
|
||||
session = factory.newServerSession(new SRPVerifier (verifier, salt));
|
||||
return new SRPPackets.Packet2_ServerSendSalt(version, session.getVerifier().salt_s.toByteArray());
|
||||
}
|
||||
|
||||
public SRPPackets.Packet4_ServerSendPublicKey step2_receivePublicKey_generatePublicKey_send (SRPPackets.Packet3_ClientSendPublicKey packet) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
session.setClientPublicKey_A(packet.getPublicKey());
|
||||
return new SRPPackets.Packet4_ServerSendPublicKey(session.getPublicKey_B());
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
|
||||
public SRPPackets.Packet6_ServerSendEvidenceAndPayload step3_validateClientEvidence_generateEvidence_send (SRPPackets.Packet5_ClientSendEvidence packet) throws CryptoException
|
||||
{
|
||||
try
|
||||
{
|
||||
session.computeCommonValue_S();
|
||||
session.validateClientEvidenceValue_M1(packet.getEvidence());
|
||||
|
||||
sessionKey = session.getSessionKey_K();
|
||||
|
||||
return new SRPPackets.Packet6_ServerSendEvidenceAndPayload(session.getEvidenceValue_M2());
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
throw new CryptoException(e);
|
||||
}
|
||||
}
|
||||
}
|
177
java/core/src/core/srp/server/SRPServerUserSession.java
Normal file
177
java/core/src/core/srp/server/SRPServerUserSession.java
Normal file
@ -0,0 +1,177 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp.server;
|
||||
|
||||
import java.math.BigInteger;
|
||||
|
||||
import core.client.messages.ClientMessagesSerializerJSON;
|
||||
import core.crypt.CryptorRSAAES;
|
||||
import core.exceptions.CryptoException;
|
||||
import core.exceptions.PublicMessageException;
|
||||
import core.io.IoChain;
|
||||
import core.srp.SRPPackets;
|
||||
import core.srp.SRPPackets.PacketInit_ClientTestCreate;
|
||||
import core.util.LogOut;
|
||||
import core.util.SimpleSerializer;
|
||||
import core.util.Strings;
|
||||
import core.util.Triple;
|
||||
|
||||
|
||||
public class SRPServerUserSession extends IoChain
|
||||
{
|
||||
static { SRPPackets.register(); ClientMessagesSerializerJSON.register(); }
|
||||
|
||||
LogOut log = new LogOut(SRPServerUserSession.class);
|
||||
|
||||
CryptorRSAAES cryptorRSA;
|
||||
boolean noMoreRoomForUsers = false;
|
||||
String userName;
|
||||
SRPServer srp = new SRPServer();
|
||||
int step = 0;
|
||||
SRPServerUserSessionDb db;
|
||||
|
||||
public SRPServerUserSession (CryptorRSAAES cryptorRSA, SRPServerUserSessionDb db, IoChain session)
|
||||
{
|
||||
super(session);
|
||||
|
||||
this.db = db;
|
||||
this.cryptorRSA = cryptorRSA;
|
||||
}
|
||||
|
||||
public String getUserName ()
|
||||
{
|
||||
return userName;
|
||||
}
|
||||
|
||||
@Override
|
||||
public void onReceive (byte[] bytes) throws Exception
|
||||
{
|
||||
try
|
||||
{
|
||||
if (step == 0)
|
||||
{
|
||||
Object object = SimpleSerializer.deserialize(bytes);
|
||||
if (object instanceof SRPPackets.PacketInit_EncryptedPacket)
|
||||
{
|
||||
step = -1;
|
||||
|
||||
SRPPackets.PacketInit_EncryptedPacket
|
||||
block = (SRPPackets.PacketInit_EncryptedPacket)object;
|
||||
|
||||
Object packet = SimpleSerializer.deserialize(cryptorRSA.decrypt(block.encryptedBlock));
|
||||
|
||||
if (packet instanceof SRPPackets.PacketInit_ClientPreAutheticationInitialization)
|
||||
{
|
||||
step_init((SRPPackets.PacketInit_ClientPreAutheticationInitialization)packet);
|
||||
return;
|
||||
}
|
||||
else
|
||||
if (packet instanceof SRPPackets.PacketInit_ClientTestCreate)
|
||||
{
|
||||
step_testcreate((SRPPackets.PacketInit_ClientTestCreate)packet);
|
||||
return;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
switch (step++)
|
||||
{
|
||||
case 0:
|
||||
{
|
||||
SRPPackets.Packet1_ClientSendsHello packet = SimpleSerializer.deserialize(bytes);
|
||||
this.userName = packet.user;
|
||||
db.rateLimitFailure(userName);
|
||||
|
||||
Triple<String, BigInteger, BigInteger> vvs = db.getUserVVS(packet.user);
|
||||
sender.send(SimpleSerializer.serialize(
|
||||
srp.step1_getSalt_send(vvs.first, vvs.second, vvs.third)
|
||||
));
|
||||
}
|
||||
break;
|
||||
|
||||
case 1:
|
||||
{
|
||||
SRPPackets.Packet3_ClientSendPublicKey packet = SimpleSerializer.deserialize(bytes);
|
||||
sender.send(SimpleSerializer.serialize(srp.step2_receivePublicKey_generatePublicKey_send(packet)));
|
||||
}
|
||||
break;
|
||||
|
||||
case 2:
|
||||
{
|
||||
SRPPackets.Packet5_ClientSendEvidence packet = SimpleSerializer.deserialize(bytes);
|
||||
SRPPackets.Packet6_ServerSendEvidenceAndPayload out =
|
||||
srp.step3_validateClientEvidence_generateEvidence_send(packet);
|
||||
|
||||
sender.send(SimpleSerializer.serialize(out));
|
||||
super.open();
|
||||
}
|
||||
break;
|
||||
|
||||
default:
|
||||
super.onReceive(srp.streamDecrypt(bytes));
|
||||
|
||||
}
|
||||
}
|
||||
catch (CryptoException e)
|
||||
{
|
||||
db.markFailure(userName);
|
||||
throw e;
|
||||
}
|
||||
catch (PublicMessageException e)
|
||||
{
|
||||
db.markFailure(userName);
|
||||
throw e;
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
db.markFailure(userName);
|
||||
throw new CryptoException (e);
|
||||
}
|
||||
}
|
||||
|
||||
public void step_testcreate(PacketInit_ClientTestCreate packet) throws Exception
|
||||
{
|
||||
try
|
||||
{
|
||||
db.testCreate (packet.version, packet.user);
|
||||
sender.send (SimpleSerializer.serialize(new SRPPackets.PacketInit_ServerResponse(true, null)));
|
||||
}
|
||||
catch (PublicMessageException e)
|
||||
{
|
||||
sender.send (SimpleSerializer.serialize(new SRPPackets.PacketInit_ServerResponse(false, e.message)));
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
sender.send (SimpleSerializer.serialize(new SRPPackets.PacketInit_ServerResponse(false, null)));
|
||||
}
|
||||
stop();
|
||||
}
|
||||
|
||||
public void step_init (SRPPackets.PacketInit_ClientPreAutheticationInitialization packet) throws Exception
|
||||
{
|
||||
try
|
||||
{
|
||||
db.createUser (packet.version, packet.user, new BigInteger(packet.v), new BigInteger(packet.s), packet.extra);
|
||||
sender.send (SimpleSerializer.serialize(new SRPPackets.PacketInit_ServerResponse(true, null)));
|
||||
}
|
||||
catch (PublicMessageException e)
|
||||
{
|
||||
sender.send (SimpleSerializer.serialize(new SRPPackets.PacketInit_ServerResponse(false, e.message)));
|
||||
}
|
||||
catch (Exception e)
|
||||
{
|
||||
sender.send (SimpleSerializer.serialize(new SRPPackets.PacketInit_ServerResponse(false, null)));
|
||||
}
|
||||
stop();
|
||||
}
|
||||
|
||||
@Override
|
||||
public void send(byte[] packet) throws Exception
|
||||
{
|
||||
sender.send(srp.streamEncrypt(packet));
|
||||
}
|
||||
|
||||
}
|
23
java/core/src/core/srp/server/SRPServerUserSessionDb.java
Normal file
23
java/core/src/core/srp/server/SRPServerUserSessionDb.java
Normal file
@ -0,0 +1,23 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp.server;
|
||||
|
||||
import java.math.BigInteger;
|
||||
|
||||
import core.util.Pair;
|
||||
import core.util.Triple;
|
||||
|
||||
|
||||
public interface SRPServerUserSessionDb
|
||||
{
|
||||
public void rateLimitFailure (String userName) throws Exception;
|
||||
public void markFailure (String userName) throws Exception;
|
||||
|
||||
public Triple<String, BigInteger, BigInteger> getUserVVS (String userName) throws Exception;
|
||||
|
||||
public void testCreate (String version, String userName) throws Exception;
|
||||
public void createUser (String version, String userName, BigInteger v, BigInteger s, byte[] extra) throws Exception;
|
||||
}
|
57
java/core/src/core/srp/test/PBETest.java
Normal file
57
java/core/src/core/srp/test/PBETest.java
Normal file
@ -0,0 +1,57 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.srp.test;
|
||||
|
||||
import java.io.FileInputStream;
|
||||
import java.io.FileOutputStream;
|
||||
import java.io.IOException;
|
||||
import java.math.BigInteger;
|
||||
import java.util.Random;
|
||||
|
||||
import core.exceptions.CryptoException;
|
||||
import core.crypt.KeyPairFromPassword;
|
||||
import core.crypt.KeyPairFromPasswordCryptor;
|
||||
import core.crypt.PBE;
|
||||
import core.util.Streams;
|
||||
|
||||
|
||||
public class PBETest
|
||||
{
|
||||
public static void main (String[] args) throws CryptoException, IOException
|
||||
{
|
||||
Random random = new Random();
|
||||
|
||||
String password = "TESTPASSWORD";
|
||||
|
||||
String s = "I am a string which is not to short not to long. I really should be much much larger. " +
|
||||
"but what can I do. I wonder, in reality this string would be a very long random number. ";
|
||||
|
||||
for (int i=0; i<50; ++i)
|
||||
s += BigInteger.valueOf(Math.abs(random.nextLong())).toString(32);
|
||||
|
||||
KeyPairFromPassword keyPair = new KeyPairFromPassword(password);
|
||||
KeyPairFromPasswordCryptor cryptor = new KeyPairFromPasswordCryptor(keyPair);
|
||||
byte[] verifier = keyPair.getVerifier();
|
||||
|
||||
System.out.println(new BigInteger(verifier));
|
||||
|
||||
byte[] e = cryptor.encrypt(s.getBytes());
|
||||
|
||||
FileOutputStream fe = new FileOutputStream("PBETest.enc");
|
||||
fe.write(e);
|
||||
fe.flush();
|
||||
|
||||
FileOutputStream fp = new FileOutputStream("PBETest.plain");
|
||||
fp.write(e);
|
||||
fp.flush();
|
||||
|
||||
byte[] in = Streams.readFullyBytes(new FileInputStream("PBETest.enc"));
|
||||
String match = new String(cryptor.decrypt(in));
|
||||
|
||||
if (match.equals(s))
|
||||
System.out.println("decryption succeeded");
|
||||
}
|
||||
}
|
31
java/core/src/core/swing/CheckListener.java
Normal file
31
java/core/src/core/swing/CheckListener.java
Normal file
@ -0,0 +1,31 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.swing;
|
||||
|
||||
import javax.swing.SwingUtilities;
|
||||
import javax.swing.event.DocumentEvent;
|
||||
import javax.swing.event.DocumentListener;
|
||||
|
||||
public class CheckListener implements DocumentListener
|
||||
{
|
||||
Checker checker;
|
||||
|
||||
public CheckListener (Checker checker)
|
||||
{
|
||||
this.checker = checker;
|
||||
}
|
||||
|
||||
public void onChange ()
|
||||
{
|
||||
SwingUtilities.invokeLater(new Runnable() {
|
||||
public void run() { checker.onCheck(); }
|
||||
});
|
||||
}
|
||||
|
||||
public void changedUpdate(DocumentEvent e) { onChange(); }
|
||||
public void insertUpdate(DocumentEvent e) { onChange(); }
|
||||
public void removeUpdate(DocumentEvent e) { onChange(); }
|
||||
}
|
11
java/core/src/core/swing/Checker.java
Normal file
11
java/core/src/core/swing/Checker.java
Normal file
@ -0,0 +1,11 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.swing;
|
||||
|
||||
public interface Checker
|
||||
{
|
||||
public void onCheck ();
|
||||
}
|
16
java/core/src/core/swing/DimensionCalculator.java
Normal file
16
java/core/src/core/swing/DimensionCalculator.java
Normal file
@ -0,0 +1,16 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.swing;
|
||||
|
||||
import java.awt.Dimension;
|
||||
|
||||
public class DimensionCalculator
|
||||
{
|
||||
public Dimension calculate (Dimension dimension)
|
||||
{
|
||||
return dimension;
|
||||
}
|
||||
}
|
38
java/core/src/core/swing/FileFilterStandard.java
Normal file
38
java/core/src/core/swing/FileFilterStandard.java
Normal file
@ -0,0 +1,38 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.swing;
|
||||
|
||||
import java.io.File;
|
||||
|
||||
import javax.swing.filechooser.FileFilter;
|
||||
|
||||
public class FileFilterStandard extends FileFilter
|
||||
{
|
||||
|
||||
String name, extension;
|
||||
|
||||
public FileFilterStandard(String name, String extension)
|
||||
{
|
||||
this.name = name;
|
||||
this.extension = extension;
|
||||
}
|
||||
|
||||
public boolean accept(File f)
|
||||
{
|
||||
if (f.isDirectory())
|
||||
{
|
||||
return true;
|
||||
}
|
||||
|
||||
return (f.getPath().toLowerCase().endsWith(extension));
|
||||
}
|
||||
|
||||
// The description of this filter
|
||||
public String getDescription()
|
||||
{
|
||||
return name;
|
||||
}
|
||||
}
|
81
java/core/src/core/swing/RedirectStreams.java
Normal file
81
java/core/src/core/swing/RedirectStreams.java
Normal file
@ -0,0 +1,81 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.swing;
|
||||
|
||||
import java.awt.TextArea;
|
||||
import java.io.IOException;
|
||||
import java.io.OutputStream;
|
||||
import java.io.PrintStream;
|
||||
|
||||
import javax.swing.JTextArea;
|
||||
import javax.swing.SwingUtilities;
|
||||
|
||||
public class RedirectStreams
|
||||
{
|
||||
JTextArea textArea;
|
||||
|
||||
PrintStream savedOutput;
|
||||
PrintStream savedError;
|
||||
PrintStream out;
|
||||
|
||||
public RedirectStreams(JTextArea textArea)
|
||||
{
|
||||
this.textArea = textArea;
|
||||
OutputStream os =
|
||||
new OutputStream() {
|
||||
@Override
|
||||
public void write(int b) throws IOException
|
||||
{
|
||||
updateTextArea(String.valueOf((char) b));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void write(byte[] b, int off, int len) throws IOException
|
||||
{
|
||||
updateTextArea(new String(b, off, len));
|
||||
}
|
||||
|
||||
@Override
|
||||
public void write(byte[] b) throws IOException
|
||||
{
|
||||
write(b, 0, b.length);
|
||||
}
|
||||
};
|
||||
|
||||
out = new PrintStream(os, true);
|
||||
}
|
||||
|
||||
private void updateTextArea(final String text)
|
||||
{
|
||||
SwingUtilities.invokeLater(new Runnable() {
|
||||
public void run()
|
||||
{
|
||||
textArea.append(text);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
public void redirectSystem()
|
||||
{
|
||||
savedOutput = System.out;
|
||||
savedError = System.err;
|
||||
|
||||
System.setOut(out);
|
||||
System.setErr(out);
|
||||
}
|
||||
|
||||
public PrintStream getOut ()
|
||||
{
|
||||
return out;
|
||||
}
|
||||
|
||||
public void restoreSystem ()
|
||||
{
|
||||
System.setOut(savedOutput);
|
||||
System.setErr(savedError);
|
||||
}
|
||||
|
||||
}
|
72
java/core/src/core/swing/SpringLayoutSize.java
Normal file
72
java/core/src/core/swing/SpringLayoutSize.java
Normal file
@ -0,0 +1,72 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.swing;
|
||||
|
||||
import java.awt.Container;
|
||||
import java.awt.Dimension;
|
||||
|
||||
import javax.swing.SpringLayout;
|
||||
|
||||
public class SpringLayoutSize extends SpringLayout
|
||||
{
|
||||
Dimension dimension;
|
||||
DimensionCalculator dimensionCalculator;
|
||||
int depth;
|
||||
|
||||
public SpringLayoutSize (int depth, Dimension dimension, DimensionCalculator dimensionCalculator)
|
||||
{
|
||||
this.dimension = dimension;
|
||||
this.depth = depth;
|
||||
this.dimensionCalculator = dimensionCalculator;
|
||||
}
|
||||
|
||||
public SpringLayoutSize (int depth, Dimension dimension)
|
||||
{
|
||||
this(depth, dimension, null);
|
||||
}
|
||||
|
||||
public Container getNestedParent(Container parent)
|
||||
{
|
||||
for (int i=1; i<depth; ++i)
|
||||
parent = parent.getParent();
|
||||
|
||||
return parent;
|
||||
}
|
||||
|
||||
public Dimension calculateDimension (Container parent)
|
||||
{
|
||||
Container nestedParent = getNestedParent (parent);
|
||||
|
||||
Dimension result =
|
||||
new Dimension (
|
||||
dimension.width > 0 ? dimension.width : nestedParent.getSize().width,
|
||||
dimension.height > 0 ? dimension.height : nestedParent.getSize().height
|
||||
);
|
||||
|
||||
if (dimensionCalculator != null)
|
||||
result = dimensionCalculator.calculate(result);
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Dimension maximumLayoutSize(Container parent)
|
||||
{
|
||||
return calculateDimension(parent);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Dimension minimumLayoutSize(Container parent)
|
||||
{
|
||||
return calculateDimension(parent);
|
||||
}
|
||||
|
||||
@Override
|
||||
public Dimension preferredLayoutSize(Container parent)
|
||||
{
|
||||
return calculateDimension(parent);
|
||||
}
|
||||
}
|
73
java/core/src/core/util/Arrays.java
Normal file
73
java/core/src/core/util/Arrays.java
Normal file
@ -0,0 +1,73 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.List;
|
||||
|
||||
public class Arrays
|
||||
{
|
||||
public static <T> T firstOrNull (T[] t)
|
||||
{
|
||||
if (t != null && t.length > 0)
|
||||
return t[0];
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
public static byte[] generate (int length, int j)
|
||||
{
|
||||
byte[] v = new byte[length];
|
||||
for (int i=0; i<length; ++i)
|
||||
v[i]=(byte)j;
|
||||
|
||||
return v;
|
||||
}
|
||||
|
||||
public static byte[] concat (byte[]... arrays)
|
||||
{
|
||||
int length = 0;
|
||||
for (byte[] array : arrays)
|
||||
length += array.length;
|
||||
|
||||
byte[] result = new byte[length];
|
||||
|
||||
int i=0;
|
||||
for (byte[] array : arrays)
|
||||
{
|
||||
for (byte b : array)
|
||||
result[i++] = b;
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
|
||||
public static byte[] copyOf(byte[] src, int l)
|
||||
{
|
||||
return copyOf(src, 0, l);
|
||||
}
|
||||
|
||||
public static void copyFromTo(byte[] src, int srcOff, byte[] dst, int dstOff, int l)
|
||||
{
|
||||
for (int i=0; i<l; ++i)
|
||||
dst[i+dstOff] = src[i+srcOff];
|
||||
}
|
||||
|
||||
public static void copyFromTo(byte[] src, byte[] dst, int l)
|
||||
{
|
||||
copyFromTo(src,0, dst,0, l);
|
||||
}
|
||||
|
||||
public static byte[] copyOf(byte[] src, int offset, int l)
|
||||
{
|
||||
byte[] result = new byte[l];
|
||||
|
||||
for (int i=0; i<l; ++i)
|
||||
result[i] = src[offset + i];
|
||||
|
||||
return result;
|
||||
}
|
||||
}
|
58
java/core/src/core/util/Base16.java
Normal file
58
java/core/src/core/util/Base16.java
Normal file
@ -0,0 +1,58 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
public class Base16
|
||||
{
|
||||
static String decoder = "0123456789abcdef";
|
||||
static byte[] encoder = null;
|
||||
|
||||
static void initialize ()
|
||||
{
|
||||
if (encoder != null)
|
||||
return;
|
||||
|
||||
encoder = new byte[0xFF];
|
||||
for (int i=0; i<decoder.length(); ++i)
|
||||
encoder[decoder.charAt(i)] = (byte)i;
|
||||
}
|
||||
|
||||
public static byte[] decode (String value)
|
||||
{
|
||||
initialize();
|
||||
|
||||
byte[] bytes = new byte[value.length() / 2];
|
||||
|
||||
int i=0, j=0;
|
||||
while (i<value.length())
|
||||
{
|
||||
int nibbleHi = (int)encoder[(int)value.charAt(i++)];
|
||||
int nibbleLo = (int)encoder[(int)value.charAt(i++)];
|
||||
|
||||
bytes[j++] = (byte) ((nibbleHi << 4) | (nibbleLo));
|
||||
}
|
||||
|
||||
return bytes;
|
||||
}
|
||||
|
||||
public static String encode (byte[] bytes)
|
||||
{
|
||||
String result = "";
|
||||
|
||||
int i=0;
|
||||
while (i<bytes.length)
|
||||
{
|
||||
int b = bytes[i++] & 0xFF;
|
||||
int nibbleLo = b & 0x0F;
|
||||
int nibbleHi = b >> 4;
|
||||
|
||||
result += decoder.charAt(nibbleHi);
|
||||
result += decoder.charAt(nibbleLo);
|
||||
}
|
||||
|
||||
return result;
|
||||
}
|
||||
}
|
68
java/core/src/core/util/Base64.java
Normal file
68
java/core/src/core/util/Base64.java
Normal file
@ -0,0 +1,68 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import core.callback.Callback;
|
||||
import core.callback.CallbackDefault;
|
||||
|
||||
public class Base64
|
||||
{
|
||||
public static String encode(byte[] bytes)
|
||||
{
|
||||
return Strings.toString(org.bc.util.encoders.Base64.encode(bytes));
|
||||
}
|
||||
|
||||
public static byte[] decode(String b64)
|
||||
{
|
||||
return org.bc.util.encoders.Base64.decode(Strings.toBytes(b64));
|
||||
}
|
||||
|
||||
public static byte[] encodeBytes(byte[] bytes)
|
||||
{
|
||||
return org.bc.util.encoders.Base64.encode(bytes);
|
||||
}
|
||||
|
||||
public static byte[] decodeBytes(byte[] b64)
|
||||
{
|
||||
return org.bc.util.encoders.Base64.decode(b64);
|
||||
}
|
||||
|
||||
public static Callback decode_()
|
||||
{
|
||||
return new CallbackDefault() {
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
next(Base64.decode((String)arguments[0]));
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public static Callback encode_()
|
||||
{
|
||||
return new CallbackDefault() {
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
next(Base64.encode((byte[])arguments[0]));
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public static Callback decodeBytes_()
|
||||
{
|
||||
return new CallbackDefault() {
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
next(Base64.decodeBytes((byte[])arguments[0]));
|
||||
}
|
||||
};
|
||||
}
|
||||
|
||||
public static Callback encodeBytes_()
|
||||
{
|
||||
return new CallbackDefault() {
|
||||
public void onSuccess(Object... arguments) throws Exception {
|
||||
next(Base64.encodeBytes((byte[])arguments[0]));
|
||||
}
|
||||
};
|
||||
}
|
||||
}
|
20
java/core/src/core/util/Block.java
Normal file
20
java/core/src/core/util/Block.java
Normal file
@ -0,0 +1,20 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import java.io.Serializable;
|
||||
|
||||
public class Block implements Serializable
|
||||
{
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
||||
public byte[] bytes;
|
||||
|
||||
public Block (byte[] bytes)
|
||||
{
|
||||
this.bytes = bytes;
|
||||
}
|
||||
}
|
15
java/core/src/core/util/CallSingle.java
Normal file
15
java/core/src/core/util/CallSingle.java
Normal file
@ -0,0 +1,15 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import org.timepedia.exporter.client.Export;
|
||||
import org.timepedia.exporter.client.Exportable;
|
||||
|
||||
@Export
|
||||
public interface CallSingle<T, V> extends Exportable
|
||||
{
|
||||
public T invoke(V v) throws Exception;
|
||||
}
|
24
java/core/src/core/util/CallSingleWithVariables.java
Normal file
24
java/core/src/core/util/CallSingleWithVariables.java
Normal file
@ -0,0 +1,24 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import core.util.CallSingle;
|
||||
|
||||
abstract class CallSingleWithVariables<T,V> implements CallSingle<T,V>
|
||||
{
|
||||
protected Object[] v;
|
||||
|
||||
public CallSingleWithVariables(Object... v)
|
||||
{
|
||||
this.v = v;
|
||||
}
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
public <T> T V(int i)
|
||||
{
|
||||
return (T)v[i];
|
||||
}
|
||||
}
|
20
java/core/src/core/util/Characters.java
Normal file
20
java/core/src/core/util/Characters.java
Normal file
@ -0,0 +1,20 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
public class Characters {
|
||||
|
||||
public static boolean isWhitespace(char c)
|
||||
{
|
||||
return " \t\n\r".indexOf(c)!=-1;
|
||||
}
|
||||
|
||||
public static boolean isNumber(char c)
|
||||
{
|
||||
return "0123456789".indexOf(c)!=-1;
|
||||
}
|
||||
|
||||
}
|
809
java/core/src/core/util/CircularByteBuffer.java
Normal file
809
java/core/src/core/util/CircularByteBuffer.java
Normal file
@ -0,0 +1,809 @@
|
||||
/*
|
||||
* Circular Byte Buffer
|
||||
* Copyright (C) 2002-2010 Stephen Ostermiller
|
||||
* http://ostermiller.org/contact.pl?regarding=Java+Utilities
|
||||
*
|
||||
* This program is free software; you can redistribute it and/or modify
|
||||
* it under the terms of the GNU General Public License as published by
|
||||
* the Free Software Foundation; either version 2 of the License, or
|
||||
* (at your option) any later version.
|
||||
*
|
||||
* This program is distributed in the hope that it will be useful,
|
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
* GNU General Public License for more details.
|
||||
*
|
||||
* See LICENSE.txt for details.
|
||||
*/
|
||||
package core.util;
|
||||
|
||||
import java.io.*;
|
||||
|
||||
/**
|
||||
* Implements the Circular Buffer producer/consumer model for bytes.
|
||||
* More information about this class is available from <a target="_top" href=
|
||||
* "http://ostermiller.org/utils/CircularByteBuffer.html">ostermiller.org</a>.
|
||||
* <p>
|
||||
* Using this class is a simpler alternative to using a PipedInputStream
|
||||
* and a PipedOutputStream. PipedInputStreams and PipedOutputStreams don't support the
|
||||
* mark operation, don't allow you to control buffer sizes that they use,
|
||||
* and have a more complicated API that requires instantiating two
|
||||
* classes and connecting them.
|
||||
* <p>
|
||||
* This class is thread safe.
|
||||
*
|
||||
* @see CircularCharBuffer
|
||||
* @see CircularObjectBuffer
|
||||
*
|
||||
* @author Stephen Ostermiller http://ostermiller.org/contact.pl?regarding=Java+Utilities
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public class CircularByteBuffer {
|
||||
|
||||
/**
|
||||
* The default size for a circular byte buffer.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
private final static int DEFAULT_SIZE = 1024;
|
||||
|
||||
/**
|
||||
* A buffer that will grow as things are added.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public final static int INFINITE_SIZE = -1;
|
||||
|
||||
/**
|
||||
* The circular buffer.
|
||||
* <p>
|
||||
* The actual capacity of the buffer is one less than the actual length
|
||||
* of the buffer so that an empty and a full buffer can be
|
||||
* distinguished. An empty buffer will have the markPostion and the
|
||||
* writePosition equal to each other. A full buffer will have
|
||||
* the writePosition one less than the markPostion.
|
||||
* <p>
|
||||
* There are three important indexes into the buffer:
|
||||
* The readPosition, the writePosition, and the markPosition.
|
||||
* If the InputStream has never been marked, the readPosition and
|
||||
* the markPosition should always be the same. The bytes
|
||||
* available to be read go from the readPosition to the writePosition,
|
||||
* wrapping around the end of the buffer. The space available for writing
|
||||
* goes from the write position to one less than the markPosition,
|
||||
* wrapping around the end of the buffer. The bytes that have
|
||||
* been saved to support a reset() of the InputStream go from markPosition
|
||||
* to readPosition, wrapping around the end of the buffer.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected byte[] buffer;
|
||||
/**
|
||||
* Index of the first byte available to be read.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected volatile int readPosition = 0;
|
||||
/**
|
||||
* Index of the first byte available to be written.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected volatile int writePosition = 0;
|
||||
/**
|
||||
* Index of the first saved byte. (To support stream marking.)
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected volatile int markPosition = 0;
|
||||
/**
|
||||
* Number of bytes that have to be saved
|
||||
* to support mark() and reset() on the InputStream.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected volatile int markSize = 0;
|
||||
/**
|
||||
* If this buffer is infinite (should resize itself when full)
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected volatile boolean infinite = false;
|
||||
/**
|
||||
* True if a write to a full buffer should block until the buffer
|
||||
* has room, false if the write method should throw an IOException
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected boolean blockingWrite = true;
|
||||
/**
|
||||
* The InputStream that can empty this buffer.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected InputStream in = new CircularByteBufferInputStream();
|
||||
/**
|
||||
* true if the close() method has been called on the InputStream
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected boolean inputStreamClosed = false;
|
||||
/**
|
||||
* The OutputStream that can fill this buffer.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected OutputStream out = new CircularByteBufferOutputStream();
|
||||
/**
|
||||
* true if the close() method has been called on the OutputStream
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected boolean outputStreamClosed = false;
|
||||
|
||||
/**
|
||||
* Make this buffer ready for reuse. The contents of the buffer
|
||||
* will be cleared and the streams associated with this buffer
|
||||
* will be reopened if they had been closed.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public void clear(){
|
||||
synchronized (this){
|
||||
readPosition = 0;
|
||||
writePosition = 0;
|
||||
markPosition = 0;
|
||||
outputStreamClosed = false;
|
||||
inputStreamClosed = false;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve a OutputStream that can be used to fill
|
||||
* this buffer.
|
||||
* <p>
|
||||
* Write methods may throw a BufferOverflowException if
|
||||
* the buffer is not large enough. A large enough buffer
|
||||
* size must be chosen so that this does not happen or
|
||||
* the caller must be prepared to catch the exception and
|
||||
* try again once part of the buffer has been consumed.
|
||||
*
|
||||
*
|
||||
* @return the producer for this buffer.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public OutputStream getOutputStream(){
|
||||
return out;
|
||||
}
|
||||
|
||||
/**
|
||||
* Retrieve a InputStream that can be used to empty
|
||||
* this buffer.
|
||||
* <p>
|
||||
* This InputStream supports marks at the expense
|
||||
* of the buffer size.
|
||||
*
|
||||
* @return the consumer for this buffer.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public InputStream getInputStream(){
|
||||
return in;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get number of bytes that are available to be read.
|
||||
* <p>
|
||||
* Note that the number of bytes available plus
|
||||
* the number of bytes free may not add up to the
|
||||
* capacity of this buffer, as the buffer may reserve some
|
||||
* space for other purposes.
|
||||
*
|
||||
* @return the size in bytes of this buffer
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public int getAvailable(){
|
||||
synchronized (this){
|
||||
return available();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of bytes this buffer has free for
|
||||
* writing.
|
||||
* <p>
|
||||
* Note that the number of bytes available plus
|
||||
* the number of bytes free may not add up to the
|
||||
* capacity of this buffer, as the buffer may reserve some
|
||||
* space for other purposes.
|
||||
*
|
||||
* @return the available space in bytes of this buffer
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public int getSpaceLeft(){
|
||||
synchronized (this){
|
||||
return spaceLeft();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the capacity of this buffer.
|
||||
* <p>
|
||||
* Note that the number of bytes available plus
|
||||
* the number of bytes free may not add up to the
|
||||
* capacity of this buffer, as the buffer may reserve some
|
||||
* space for other purposes.
|
||||
*
|
||||
* @return the size in bytes of this buffer
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public int getSize(){
|
||||
synchronized (this){
|
||||
return buffer.length;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* double the size of the buffer
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
private void resize(){
|
||||
byte[] newBuffer = new byte[buffer.length * 2];
|
||||
int marked = marked();
|
||||
int available = available();
|
||||
if (markPosition <= writePosition){
|
||||
// any space between the mark and
|
||||
// the first write needs to be saved.
|
||||
// In this case it is all in one piece.
|
||||
int length = writePosition - markPosition;
|
||||
System.arraycopy(buffer, markPosition, newBuffer, 0, length);
|
||||
} else {
|
||||
int length1 = buffer.length - markPosition;
|
||||
System.arraycopy(buffer, markPosition, newBuffer, 0, length1);
|
||||
int length2 = writePosition;
|
||||
System.arraycopy(buffer, 0, newBuffer, length1, length2);
|
||||
}
|
||||
buffer = newBuffer;
|
||||
markPosition = 0;
|
||||
readPosition = marked;
|
||||
writePosition = marked + available;
|
||||
}
|
||||
|
||||
/**
|
||||
* Space available in the buffer which can be written.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
private int spaceLeft(){
|
||||
if (writePosition < markPosition){
|
||||
// any space between the first write and
|
||||
// the mark except one byte is available.
|
||||
// In this case it is all in one piece.
|
||||
return (markPosition - writePosition - 1);
|
||||
}
|
||||
// space at the beginning and end.
|
||||
return ((buffer.length - 1) - (writePosition - markPosition));
|
||||
}
|
||||
|
||||
/**
|
||||
* Bytes available for reading.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
private int available(){
|
||||
if (readPosition <= writePosition){
|
||||
// any space between the first read and
|
||||
// the first write is available. In this case i
|
||||
// is all in one piece.
|
||||
return (writePosition - readPosition);
|
||||
}
|
||||
// space at the beginning and end.
|
||||
return (buffer.length - (readPosition - writePosition));
|
||||
}
|
||||
|
||||
/**
|
||||
* Bytes saved for supporting marks.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
private int marked(){
|
||||
if (markPosition <= readPosition){
|
||||
// any space between the markPosition and
|
||||
// the first write is marked. In this case i
|
||||
// is all in one piece.
|
||||
return (readPosition - markPosition);
|
||||
}
|
||||
// space at the beginning and end.
|
||||
return (buffer.length - (markPosition - readPosition));
|
||||
}
|
||||
|
||||
/**
|
||||
* If we have passed the markSize reset the
|
||||
* mark so that the space can be used.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
private void ensureMark(){
|
||||
if (marked() > markSize){
|
||||
markPosition = readPosition;
|
||||
markSize = 0;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new buffer with a default capacity.
|
||||
* Writing to a full buffer will block until space
|
||||
* is available rather than throw an exception.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public CircularByteBuffer(){
|
||||
this (DEFAULT_SIZE, true);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new buffer with given capacity.
|
||||
* Writing to a full buffer will block until space
|
||||
* is available rather than throw an exception.
|
||||
* <p>
|
||||
* Note that the buffer may reserve some bytes for
|
||||
* special purposes and capacity number of bytes may
|
||||
* not be able to be written to the buffer.
|
||||
* <p>
|
||||
* Note that if the buffer is of INFINITE_SIZE it will
|
||||
* neither block or throw exceptions, but rather grow
|
||||
* without bound.
|
||||
*
|
||||
* @param size desired capacity of the buffer in bytes or CircularByteBuffer.INFINITE_SIZE.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public CircularByteBuffer(int size){
|
||||
this (size, true);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new buffer with a default capacity and
|
||||
* given blocking behavior.
|
||||
*
|
||||
* @param blockingWrite true writing to a full buffer should block
|
||||
* until space is available, false if an exception should
|
||||
* be thrown instead.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public CircularByteBuffer(boolean blockingWrite){
|
||||
this (DEFAULT_SIZE, blockingWrite);
|
||||
}
|
||||
|
||||
/**
|
||||
* Create a new buffer with the given capacity and
|
||||
* blocking behavior.
|
||||
* <p>
|
||||
* Note that the buffer may reserve some bytes for
|
||||
* special purposes and capacity number of bytes may
|
||||
* not be able to be written to the buffer.
|
||||
* <p>
|
||||
* Note that if the buffer is of INFINITE_SIZE it will
|
||||
* neither block or throw exceptions, but rather grow
|
||||
* without bound.
|
||||
*
|
||||
* @param size desired capacity of the buffer in bytes or CircularByteBuffer.INFINITE_SIZE.
|
||||
* @param blockingWrite true writing to a full buffer should block
|
||||
* until space is available, false if an exception should
|
||||
* be thrown instead.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
public CircularByteBuffer(int size, boolean blockingWrite){
|
||||
if (size == INFINITE_SIZE){
|
||||
buffer = new byte[DEFAULT_SIZE];
|
||||
infinite = true;
|
||||
} else {
|
||||
buffer = new byte[size];
|
||||
infinite = false;
|
||||
}
|
||||
this.blockingWrite = blockingWrite;
|
||||
}
|
||||
|
||||
/**
|
||||
* Class for reading from a circular byte buffer.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected class CircularByteBufferInputStream extends InputStream {
|
||||
|
||||
/**
|
||||
* Returns the number of bytes that can be read (or skipped over) from this
|
||||
* input stream without blocking by the next caller of a method for this input
|
||||
* stream. The next caller might be the same thread or or another thread.
|
||||
*
|
||||
* @return the number of bytes that can be read from this input stream without blocking.
|
||||
* @throws IOException if the stream is closed.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public int available() throws IOException {
|
||||
synchronized (CircularByteBuffer.this){
|
||||
if (inputStreamClosed) throw new IOException("InputStream has been closed, it is not ready.");
|
||||
return (CircularByteBuffer.this.available());
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Close the stream. Once a stream has been closed, further read(), available(),
|
||||
* mark(), or reset() invocations will throw an IOException. Closing a
|
||||
* previously-closed stream, however, has no effect.
|
||||
*
|
||||
* @throws IOException never.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public void close() throws IOException {
|
||||
synchronized (CircularByteBuffer.this){
|
||||
inputStreamClosed = true;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Mark the present position in the stream. Subsequent calls to reset() will
|
||||
* attempt to reposition the stream to this point.
|
||||
* <p>
|
||||
* The readAheadLimit must be less than the size of circular buffer, otherwise
|
||||
* this method has no effect.
|
||||
*
|
||||
* @param readAheadLimit Limit on the number of bytes that may be read while
|
||||
* still preserving the mark. After reading this many bytes, attempting to
|
||||
* reset the stream will fail.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public void mark(int readAheadLimit) {
|
||||
synchronized (CircularByteBuffer.this){
|
||||
//if (inputStreamClosed) throw new IOException("InputStream has been closed; cannot mark a closed InputStream.");
|
||||
if (buffer.length - 1 > readAheadLimit) {
|
||||
markSize = readAheadLimit;
|
||||
markPosition = readPosition;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Tell whether this stream supports the mark() operation.
|
||||
*
|
||||
* @return true, mark is supported.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public boolean markSupported() {
|
||||
return true;
|
||||
}
|
||||
|
||||
/**
|
||||
* Read a single byte.
|
||||
* This method will block until a byte is available, an I/O error occurs,
|
||||
* or the end of the stream is reached.
|
||||
*
|
||||
* @return The byte read, as an integer in the range 0 to 255 (0x00-0xff),
|
||||
* or -1 if the end of the stream has been reached
|
||||
* @throws IOException if the stream is closed.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public int read() throws IOException {
|
||||
while (true){
|
||||
synchronized (CircularByteBuffer.this){
|
||||
if (inputStreamClosed) throw new IOException("InputStream has been closed; cannot read from a closed InputStream.");
|
||||
int available = CircularByteBuffer.this.available();
|
||||
if (available > 0){
|
||||
int result = buffer[readPosition] & 0xff;
|
||||
readPosition++;
|
||||
if (readPosition == buffer.length){
|
||||
readPosition = 0;
|
||||
}
|
||||
ensureMark();
|
||||
return result;
|
||||
} else if (outputStreamClosed){
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
try {
|
||||
Thread.sleep(100);
|
||||
} catch(Exception x){
|
||||
throw new IOException("Blocking read operation interrupted.");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Read bytes into an array.
|
||||
* This method will block until some input is available,
|
||||
* an I/O error occurs, or the end of the stream is reached.
|
||||
*
|
||||
* @param cbuf Destination buffer.
|
||||
* @return The number of bytes read, or -1 if the end of
|
||||
* the stream has been reached
|
||||
* @throws IOException if the stream is closed.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public int read(byte[] cbuf) throws IOException {
|
||||
return read(cbuf, 0, cbuf.length);
|
||||
}
|
||||
|
||||
/**
|
||||
* Read bytes into a portion of an array.
|
||||
* This method will block until some input is available,
|
||||
* an I/O error occurs, or the end of the stream is reached.
|
||||
*
|
||||
* @param cbuf Destination buffer.
|
||||
* @param off Offset at which to start storing bytes.
|
||||
* @param len Maximum number of bytes to read.
|
||||
* @return The number of bytes read, or -1 if the end of
|
||||
* the stream has been reached
|
||||
* @throws IOException if the stream is closed.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public int read(byte[] cbuf, int off, int len) throws IOException {
|
||||
while (true){
|
||||
synchronized (CircularByteBuffer.this){
|
||||
if (inputStreamClosed) throw new IOException("InputStream has been closed; cannot read from a closed InputStream.");
|
||||
int available = CircularByteBuffer.this.available();
|
||||
if (available > 0){
|
||||
int length = Math.min(len, available);
|
||||
int firstLen = Math.min(length, buffer.length - readPosition);
|
||||
int secondLen = length - firstLen;
|
||||
System.arraycopy(buffer, readPosition, cbuf, off, firstLen);
|
||||
if (secondLen > 0){
|
||||
System.arraycopy(buffer, 0, cbuf, off+firstLen, secondLen);
|
||||
readPosition = secondLen;
|
||||
} else {
|
||||
readPosition += length;
|
||||
}
|
||||
if (readPosition == buffer.length) {
|
||||
readPosition = 0;
|
||||
}
|
||||
ensureMark();
|
||||
return length;
|
||||
} else if (outputStreamClosed){
|
||||
return -1;
|
||||
}
|
||||
}
|
||||
try {
|
||||
Thread.sleep(100);
|
||||
} catch(Exception x){
|
||||
throw new IOException("Blocking read operation interrupted.");
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Reset the stream.
|
||||
* If the stream has been marked, then attempt to reposition i
|
||||
* at the mark. If the stream has not been marked, or more bytes
|
||||
* than the readAheadLimit have been read, this method has no effect.
|
||||
*
|
||||
* @throws IOException if the stream is closed.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public void reset() throws IOException {
|
||||
synchronized (CircularByteBuffer.this){
|
||||
if (inputStreamClosed) throw new IOException("InputStream has been closed; cannot reset a closed InputStream.");
|
||||
readPosition = markPosition;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Skip bytes.
|
||||
* This method will block until some bytes are available,
|
||||
* an I/O error occurs, or the end of the stream is reached.
|
||||
*
|
||||
* @param n The number of bytes to skip
|
||||
* @return The number of bytes actually skipped
|
||||
* @throws IllegalArgumentException if n is negative.
|
||||
* @throws IOException if the stream is closed.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public long skip(long n) throws IOException, IllegalArgumentException {
|
||||
while (true){
|
||||
synchronized (CircularByteBuffer.this){
|
||||
if (inputStreamClosed) throw new IOException("InputStream has been closed; cannot skip bytes on a closed InputStream.");
|
||||
int available = CircularByteBuffer.this.available();
|
||||
if (available > 0){
|
||||
int length = Math.min((int)n, available);
|
||||
int firstLen = Math.min(length, buffer.length - readPosition);
|
||||
int secondLen = length - firstLen;
|
||||
if (secondLen > 0){
|
||||
readPosition = secondLen;
|
||||
} else {
|
||||
readPosition += length;
|
||||
}
|
||||
if (readPosition == buffer.length) {
|
||||
readPosition = 0;
|
||||
}
|
||||
ensureMark();
|
||||
return length;
|
||||
} else if (outputStreamClosed){
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
try {
|
||||
Thread.sleep(100);
|
||||
} catch(Exception x){
|
||||
throw new IOException("Blocking read operation interrupted.");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Class for writing to a circular byte buffer.
|
||||
* If the buffer is full, the writes will either block
|
||||
* until there is some space available or throw an IOException
|
||||
* based on the CircularByteBuffer's preference.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
protected class CircularByteBufferOutputStream extends OutputStream {
|
||||
|
||||
/**
|
||||
* Close the stream, flushing it first.
|
||||
* This will cause the InputStream associated with this circular buffer
|
||||
* to read its last bytes once it empties the buffer.
|
||||
* Once a stream has been closed, further write() or flush() invocations
|
||||
* will cause an IOException to be thrown. Closing a previously-closed stream,
|
||||
* however, has no effect.
|
||||
*
|
||||
* @throws IOException never.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public void close() throws IOException {
|
||||
synchronized (CircularByteBuffer.this){
|
||||
if (!outputStreamClosed){
|
||||
flush();
|
||||
}
|
||||
outputStreamClosed = true;
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Flush the stream.
|
||||
*
|
||||
* @throws IOException if the stream is closed.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public void flush() throws IOException {
|
||||
synchronized (CircularByteBuffer.this){
|
||||
if (outputStreamClosed) throw new IOException("OutputStream has been closed; cannot flush a closed OutputStream.");
|
||||
if (inputStreamClosed) throw new IOException("Buffer closed by inputStream; cannot flush.");
|
||||
}
|
||||
// this method needs to do nothing
|
||||
}
|
||||
|
||||
/**
|
||||
* Write an array of bytes.
|
||||
* If the buffer allows blocking writes, this method will block until
|
||||
* all the data has been written rather than throw an IOException.
|
||||
*
|
||||
* @param cbuf Array of bytes to be written
|
||||
* @throws BufferOverflowException if buffer does not allow blocking writes
|
||||
* and the buffer is full. If the exception is thrown, no data
|
||||
* will have been written since the buffer was set to be non-blocking.
|
||||
* @throws IOException if the stream is closed, or the write is interrupted.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public void write(byte[] cbuf) throws IOException {
|
||||
write(cbuf, 0, cbuf.length);
|
||||
}
|
||||
|
||||
/**
|
||||
* Write a portion of an array of bytes.
|
||||
* If the buffer allows blocking writes, this method will block until
|
||||
* all the data has been written rather than throw an IOException.
|
||||
*
|
||||
* @param cbuf Array of bytes
|
||||
* @param off Offset from which to start writing bytes
|
||||
* @param len - Number of bytes to write
|
||||
* @throws BufferOverflowException if buffer does not allow blocking writes
|
||||
* and the buffer is full. If the exception is thrown, no data
|
||||
* will have been written since the buffer was set to be non-blocking.
|
||||
* @throws IOException if the stream is closed, or the write is interrupted.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public void write(byte[] cbuf, int off, int len) throws IOException {
|
||||
while (len > 0){
|
||||
synchronized (CircularByteBuffer.this){
|
||||
if (outputStreamClosed) throw new IOException("OutputStream has been closed; cannot write to a closed OutputStream.");
|
||||
if (inputStreamClosed) throw new IOException("Buffer closed by InputStream; cannot write to a closed buffer.");
|
||||
int spaceLeft = spaceLeft();
|
||||
while (infinite && spaceLeft < len){
|
||||
resize();
|
||||
spaceLeft = spaceLeft();
|
||||
}
|
||||
if (!blockingWrite && spaceLeft < len) throw new IOException("CircularByteBuffer is full; cannot write " + len + " bytes");
|
||||
int realLen = Math.min(len, spaceLeft);
|
||||
int firstLen = Math.min(realLen, buffer.length - writePosition);
|
||||
int secondLen = Math.min(realLen - firstLen, buffer.length - markPosition - 1);
|
||||
int written = firstLen + secondLen;
|
||||
if (firstLen > 0){
|
||||
System.arraycopy(cbuf, off, buffer, writePosition, firstLen);
|
||||
}
|
||||
if (secondLen > 0){
|
||||
System.arraycopy(cbuf, off+firstLen, buffer, 0, secondLen);
|
||||
writePosition = secondLen;
|
||||
} else {
|
||||
writePosition += written;
|
||||
}
|
||||
if (writePosition == buffer.length) {
|
||||
writePosition = 0;
|
||||
}
|
||||
off += written;
|
||||
len -= written;
|
||||
}
|
||||
if (len > 0){
|
||||
try {
|
||||
Thread.sleep(100);
|
||||
} catch(Exception x){
|
||||
throw new IOException("Waiting for available space in buffer interrupted.");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Write a single byte.
|
||||
* The byte to be written is contained in the 8 low-order bits of the
|
||||
* given integer value; the 24 high-order bits are ignored.
|
||||
* If the buffer allows blocking writes, this method will block until
|
||||
* all the data has been written rather than throw an IOException.
|
||||
*
|
||||
* @param c number of bytes to be written
|
||||
* @throws BufferOverflowException if buffer does not allow blocking writes
|
||||
* and the buffer is full.
|
||||
* @throws IOException if the stream is closed, or the write is interrupted.
|
||||
*
|
||||
* @since ostermillerutils 1.00.00
|
||||
*/
|
||||
@Override public void write(int c) throws IOException {
|
||||
boolean written = false;
|
||||
while (!written){
|
||||
synchronized (CircularByteBuffer.this){
|
||||
if (outputStreamClosed) throw new IOException("OutputStream has been closed; cannot write to a closed OutputStream.");
|
||||
if (inputStreamClosed) throw new IOException("Buffer closed by InputStream; cannot write to a closed buffer.");
|
||||
int spaceLeft = spaceLeft();
|
||||
while (infinite && spaceLeft < 1){
|
||||
resize();
|
||||
spaceLeft = spaceLeft();
|
||||
}
|
||||
if (!blockingWrite && spaceLeft < 1) throw new IOException("CircularByteBuffer is full; cannot write 1 byte");
|
||||
if (spaceLeft > 0){
|
||||
buffer[writePosition] = (byte)(c & 0xff);
|
||||
writePosition++;
|
||||
if (writePosition == buffer.length) {
|
||||
writePosition = 0;
|
||||
}
|
||||
written = true;
|
||||
}
|
||||
}
|
||||
if (!written){
|
||||
try {
|
||||
Thread.sleep(100);
|
||||
} catch(Exception x){
|
||||
throw new IOException("Waiting for available space in buffer interrupted.");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
98
java/core/src/core/util/Collectionz.java
Normal file
98
java/core/src/core/util/Collectionz.java
Normal file
@ -0,0 +1,98 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import java.util.ArrayList;
|
||||
import java.util.Collection;
|
||||
import java.util.List;
|
||||
|
||||
public class Collectionz
|
||||
{
|
||||
public static <T> List<T> toMutableList(T[] a)
|
||||
{
|
||||
ArrayList<T> l = new ArrayList<T>();
|
||||
for (T t : a)
|
||||
l.add(t);
|
||||
|
||||
return l;
|
||||
}
|
||||
|
||||
public static <T> Collection<T> filterNull (Collection<T> items)
|
||||
{
|
||||
List<T> l = new ArrayList<T>();
|
||||
|
||||
for (T i : items)
|
||||
if (i != null)
|
||||
l.add(i);
|
||||
|
||||
return l;
|
||||
}
|
||||
|
||||
public static <T> Collection<T> filterNull (T... items)
|
||||
{
|
||||
List<T> l = new ArrayList<T>();
|
||||
|
||||
for (T i : items)
|
||||
if (i != null)
|
||||
l.add(i);
|
||||
|
||||
return l;
|
||||
}
|
||||
|
||||
static public void removeByFirst (Collection<? extends Pair> items, Object remove)
|
||||
{
|
||||
Object found = null;
|
||||
for (Pair x : items)
|
||||
{
|
||||
if (x.first.equals(remove))
|
||||
{
|
||||
found = x;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (found != null)
|
||||
items.remove(found);
|
||||
}
|
||||
|
||||
static public void removeBySecond (Collection<? extends Pair> items, Object remove)
|
||||
{
|
||||
Object found = null;
|
||||
for (Pair x : items)
|
||||
{
|
||||
if (x.second.equals(remove))
|
||||
{
|
||||
found = x;
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (found != null)
|
||||
items.remove(found);
|
||||
}
|
||||
|
||||
static public boolean containsByFirst (Collection<? extends Pair> items, Object remove)
|
||||
{
|
||||
for (Pair x : items)
|
||||
{
|
||||
if (x.first.equals(remove))
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
static public boolean containsBySecond (Collection<? extends Pair> items, Object remove)
|
||||
{
|
||||
for (Pair x : items)
|
||||
{
|
||||
if (x.second.equals(remove))
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
96
java/core/src/core/util/Comparators.java
Normal file
96
java/core/src/core/util/Comparators.java
Normal file
@ -0,0 +1,96 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import java.util.Comparator;
|
||||
|
||||
public class Comparators
|
||||
{
|
||||
public static class SortBySecond<S> implements Comparator<Pair<?,S>>
|
||||
{
|
||||
Comparator<S> c;
|
||||
|
||||
public SortBySecond(Comparator<S> c)
|
||||
{
|
||||
this.c = c;
|
||||
}
|
||||
|
||||
@Override
|
||||
public int compare(Pair<?,S> p1, Pair<?,S> p2)
|
||||
{
|
||||
return c.compare(p1.second, p2.second);
|
||||
}
|
||||
}
|
||||
|
||||
public static class SortByFirst<S> implements Comparator<Pair<S,?>>
|
||||
{
|
||||
Comparator<S> c;
|
||||
|
||||
public SortByFirst(Comparator<S> c)
|
||||
{
|
||||
this.c = c;
|
||||
}
|
||||
|
||||
@Override
|
||||
public int compare(Pair<S,?> p1, Pair<S,?> p2)
|
||||
{
|
||||
return c.compare(p1.first, p2.first);
|
||||
}
|
||||
}
|
||||
|
||||
public static class SortByFirstReverse<S> implements Comparator<Pair<S,?>>
|
||||
{
|
||||
Comparator<S> c;
|
||||
|
||||
public SortByFirstReverse(Comparator<S> c)
|
||||
{
|
||||
this.c = c;
|
||||
}
|
||||
|
||||
@Override
|
||||
public int compare(Pair<S,?> p1, Pair<S,?> p2)
|
||||
{
|
||||
return c.compare(p2.first, p1.first);
|
||||
}
|
||||
}
|
||||
|
||||
public static class SortBySecondNatural<S extends Comparable<S>> implements Comparator<Pair<?,S>>
|
||||
{
|
||||
|
||||
@Override
|
||||
public int compare(Pair<?, S> o1, Pair<?, S> o2)
|
||||
{
|
||||
return o1.second.compareTo(o2.second);
|
||||
}
|
||||
}
|
||||
|
||||
public static class SortByFirstNatural<S extends Comparable<S>> implements Comparator<Pair<S,?>>
|
||||
{
|
||||
@Override
|
||||
public int compare(Pair<S,?> o1, Pair<S,?> o2)
|
||||
{
|
||||
return o1.first.compareTo(o2.first);
|
||||
}
|
||||
}
|
||||
|
||||
public static class SortBySecondNaturalOpposite<S extends Comparable<S>> implements Comparator<Pair<?,S>>
|
||||
{
|
||||
@Override
|
||||
public int compare(Pair<?, S> o1, Pair<?, S> o2)
|
||||
{
|
||||
return o2.second.compareTo(o1.second);
|
||||
}
|
||||
}
|
||||
|
||||
public static class SortByFirstNaturalOpposite<S extends Comparable<S>> implements Comparator<Pair<S,?>>
|
||||
{
|
||||
@Override
|
||||
public int compare(Pair<S,?> o1, Pair<S,?> o2)
|
||||
{
|
||||
return o2.first.compareTo(o1.first);
|
||||
}
|
||||
}
|
||||
}
|
38
java/core/src/core/util/DateFormat.java
Normal file
38
java/core/src/core/util/DateFormat.java
Normal file
@ -0,0 +1,38 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import java.text.ParseException;
|
||||
import java.text.SimpleDateFormat;
|
||||
import java.util.Date;
|
||||
import java.util.TimeZone;
|
||||
|
||||
public class DateFormat
|
||||
{
|
||||
SimpleDateFormat sdf;
|
||||
|
||||
public DateFormat(String format)
|
||||
{
|
||||
sdf = new SimpleDateFormat(format);
|
||||
}
|
||||
|
||||
public String format(Date date, int i)
|
||||
{
|
||||
sdf.setTimeZone(TimeZone.getTimeZone("GMT-"+i+":00"));
|
||||
return sdf.format(date);
|
||||
}
|
||||
|
||||
public String format(Date date)
|
||||
{
|
||||
return format(date, 0);
|
||||
}
|
||||
|
||||
public Date parse(String time) throws ParseException
|
||||
{
|
||||
return sdf.parse(time);
|
||||
}
|
||||
|
||||
}
|
111
java/core/src/core/util/Environment.java
Normal file
111
java/core/src/core/util/Environment.java
Normal file
@ -0,0 +1,111 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
|
||||
import core.connector.FileInfo;
|
||||
//import core.connector.sync.StoreConnector;
|
||||
|
||||
|
||||
public class Environment extends HashMap<String,String>
|
||||
{
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
||||
public Environment ()
|
||||
{
|
||||
}
|
||||
|
||||
/*
|
||||
public static Environment fromStore (StoreConnector connector) throws Exception
|
||||
{
|
||||
Environment e = new Environment ();
|
||||
connector.open();
|
||||
e.readFromStore(connector);
|
||||
connector.close();
|
||||
return e;
|
||||
}
|
||||
|
||||
public void readFromStore (StoreConnector connector) throws Exception
|
||||
{
|
||||
List<FileInfo> files = connector.listDirectory("");
|
||||
for (FileInfo file : files)
|
||||
{
|
||||
put(file.relativePath, new String(connector.get(file.relativePath), "UTF-8"));
|
||||
}
|
||||
}
|
||||
|
||||
public static void toStore (StoreConnector connector, Environment e) throws Exception
|
||||
{
|
||||
connector.open();
|
||||
e.writeToStore(connector);
|
||||
connector.close();
|
||||
}
|
||||
|
||||
public void writeToStore (StoreConnector connector) throws Exception
|
||||
{
|
||||
for (String key : keySet())
|
||||
connector.put(key, Strings.toBytes(this.get(key)));
|
||||
}
|
||||
*/
|
||||
|
||||
public String checkGet (String key)
|
||||
{
|
||||
// find a better exception
|
||||
if (!containsKey(key))
|
||||
throw new NullPointerException("Unknown key: " + key);
|
||||
|
||||
return get(key);
|
||||
}
|
||||
|
||||
public Environment childEnvironment (String key)
|
||||
{
|
||||
Environment e = new Environment();
|
||||
String prefix = key + "/";
|
||||
int prefixLength = prefix.length();
|
||||
|
||||
for (Map.Entry<String, String> i : entrySet())
|
||||
{
|
||||
String k = i.getKey();
|
||||
if (k.startsWith(prefix))
|
||||
{
|
||||
e.put(k.substring(prefixLength), i.getValue());
|
||||
}
|
||||
}
|
||||
|
||||
return e;
|
||||
}
|
||||
|
||||
public void addChildEnvironment (String key, Environment e)
|
||||
{
|
||||
String prefix = key + "/";
|
||||
|
||||
for (Map.Entry<String, String> i : e.entrySet())
|
||||
{
|
||||
String k = i.getKey();
|
||||
this.put(prefix + k, i.getValue());
|
||||
}
|
||||
}
|
||||
|
||||
public boolean hasChildEnvironment (String key)
|
||||
{
|
||||
String prefix = key + "/";
|
||||
int prefixLength = prefix.length();
|
||||
|
||||
for (Map.Entry<String, String> i : entrySet())
|
||||
{
|
||||
String k = i.getKey();
|
||||
if (k.startsWith(prefix))
|
||||
{
|
||||
return true;
|
||||
}
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
}
|
11
java/core/src/core/util/ExceptionHandler.java
Normal file
11
java/core/src/core/util/ExceptionHandler.java
Normal file
@ -0,0 +1,11 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
public interface ExceptionHandler
|
||||
{
|
||||
public void exception (Object ... arguments);
|
||||
}
|
39
java/core/src/core/util/ExternalResource.java
Normal file
39
java/core/src/core/util/ExternalResource.java
Normal file
@ -0,0 +1,39 @@
|
||||
/**
|
||||
* Author: Timothy Prepscius
|
||||
* License: GPLv3 Affero + keep my name in the code!
|
||||
*/
|
||||
|
||||
package core.util;
|
||||
|
||||
import java.io.FileInputStream;
|
||||
import java.io.InputStream;
|
||||
|
||||
public class ExternalResource
|
||||
{
|
||||
static LogOut log = new LogOut(ExternalResource.class);
|
||||
|
||||
static protected String prefix;
|
||||
|
||||
static {
|
||||
prefix = System.getProperty("user.home") + "/resources/";
|
||||
}
|
||||
|
||||
public static byte[] get(String key) throws Exception
|
||||
{
|
||||
String path = prefix + key;
|
||||
log.debug("get", path);
|
||||
return Streams.readFullyBytes(new FileInputStream(path));
|
||||
}
|
||||
|
||||
public static InputStream getResourceAsStream(Class<?> c, String key) throws Exception
|
||||
{
|
||||
String path = prefix + c.getPackage().getName() + "/" + key;
|
||||
log.debug("getResourceAsStream", path);
|
||||
return new FileInputStream(path);
|
||||
}
|
||||
|
||||
public static String getTrimmedString(String key) throws Exception
|
||||
{
|
||||
return Strings.toString(get(key)).trim();
|
||||
}
|
||||
}
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue
Block a user