Initial commit
This commit is contained in:
70
src/nodash/core/NoConfig.java
Normal file
70
src/nodash/core/NoConfig.java
Normal file
@@ -0,0 +1,70 @@
|
||||
package nodash.core;
|
||||
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
import java.io.ObjectInputStream;
|
||||
import java.io.ObjectOutputStream;
|
||||
import java.io.Serializable;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.StandardOpenOption;
|
||||
import java.security.NoSuchAlgorithmException;
|
||||
|
||||
import javax.crypto.KeyGenerator;
|
||||
import javax.crypto.SecretKey;
|
||||
|
||||
import nodash.exceptions.NoDashFatalException;
|
||||
|
||||
public class NoConfig implements Serializable {
|
||||
private static final long serialVersionUID = -8498303909736017075L;
|
||||
|
||||
public static final String CONFIG_FILENAME = "noconfig.cfg";
|
||||
|
||||
public SecretKey secretKey;
|
||||
|
||||
public boolean saveDatabase = true;
|
||||
public String databaseFilename = "nodatabase.hash";
|
||||
public boolean saveByteSets = false;
|
||||
public String byteSetFilename = "";
|
||||
|
||||
public NoConfig() {
|
||||
try {
|
||||
KeyGenerator keyGenerator = KeyGenerator.getInstance(NoUtil.CIPHER_KEY_SPEC);
|
||||
keyGenerator.init(NoUtil.AES_STRENGTH);
|
||||
this.secretKey = keyGenerator.generateKey();
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
throw new NoDashFatalException("Value for CIPHER_KEY_SPEC not valid.");
|
||||
}
|
||||
}
|
||||
|
||||
public void saveNoConfigToFile(File file) {
|
||||
try {
|
||||
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
||||
ObjectOutputStream oos = new ObjectOutputStream(baos);
|
||||
oos.writeObject(this);
|
||||
byte[] data = baos.toByteArray();
|
||||
|
||||
Files.write(file.toPath(), data, StandardOpenOption.CREATE_NEW);
|
||||
} catch (IOException e) {
|
||||
throw new NoDashFatalException("Unable to save config, including generated secret key.");
|
||||
}
|
||||
}
|
||||
|
||||
public static NoConfig getNoConfigFromFile(File file) {
|
||||
try {
|
||||
byte[] data = Files.readAllBytes(file.toPath());
|
||||
ByteArrayInputStream bais = new ByteArrayInputStream(data);
|
||||
ObjectInputStream ois = new ObjectInputStream(bais);
|
||||
NoConfig noConfig;
|
||||
try {
|
||||
noConfig = (NoConfig) ois.readObject();
|
||||
} catch (ClassNotFoundException e) {
|
||||
throw new NoDashFatalException("Given bytestream does not compile into a configuration object.");
|
||||
}
|
||||
return noConfig;
|
||||
} catch (IOException e) {
|
||||
throw new NoDashFatalException("Instructed to read config from file but unable to do so.");
|
||||
}
|
||||
}
|
||||
}
|
||||
82
src/nodash/core/NoCore.java
Normal file
82
src/nodash/core/NoCore.java
Normal file
@@ -0,0 +1,82 @@
|
||||
package nodash.core;
|
||||
|
||||
import java.io.File;
|
||||
import java.security.PublicKey;
|
||||
|
||||
import nodash.core.spheres.NoByteSetSphere;
|
||||
import nodash.core.spheres.NoHashSphere;
|
||||
import nodash.core.spheres.NoSessionSphere;
|
||||
import nodash.exceptions.NoDashSessionBadUUID;
|
||||
import nodash.exceptions.NoSessionAlreadyAwaitingConfirmationException;
|
||||
import nodash.exceptions.NoSessionConfirmedException;
|
||||
import nodash.exceptions.NoSessionExpiredException;
|
||||
import nodash.exceptions.NoSessionNotAwaitingConfirmationException;
|
||||
import nodash.exceptions.NoSessionNotChangedException;
|
||||
import nodash.exceptions.NoUserAlreadyOnlineException;
|
||||
import nodash.exceptions.NoUserNotValidException;
|
||||
import nodash.models.NoByteSet;
|
||||
import nodash.models.NoUser;
|
||||
import nodash.models.NoSession.NoState;
|
||||
|
||||
public final class NoCore {
|
||||
public static NoConfig config;
|
||||
|
||||
public static void setup() {
|
||||
File configFile = new File(NoConfig.CONFIG_FILENAME);
|
||||
if (configFile.exists()) {
|
||||
config = NoConfig.getNoConfigFromFile(configFile);
|
||||
} else {
|
||||
config = new NoConfig();
|
||||
config.saveNoConfigToFile(configFile);
|
||||
}
|
||||
|
||||
NoHashSphere.setup();
|
||||
}
|
||||
|
||||
public static byte[] login(byte[] data, char[] password) throws NoUserNotValidException, NoUserAlreadyOnlineException, NoSessionExpiredException {
|
||||
/* steps 1 through to pre-3 */
|
||||
return NoSessionSphere.login(data, password);
|
||||
}
|
||||
|
||||
public static NoRegister register(NoUser user, char[] password) {
|
||||
/* Straight to step 4 */
|
||||
return NoSessionSphere.registerUser(user, password);
|
||||
}
|
||||
|
||||
public static NoUser getUser(byte[] cookie) throws NoSessionExpiredException, NoSessionConfirmedException, NoDashSessionBadUUID {
|
||||
/* Facilitates step 3
|
||||
* allow website-side modifications to the NoUser or NoUser inheritant */
|
||||
return NoSessionSphere.getUser(cookie);
|
||||
}
|
||||
|
||||
public static NoState getSessionState(byte[] cookie) throws NoSessionExpiredException, NoSessionConfirmedException, NoDashSessionBadUUID {
|
||||
/* Facilitates step 3
|
||||
* allow front-side to keep track of session state */
|
||||
return NoSessionSphere.getState(cookie);
|
||||
}
|
||||
|
||||
public static byte[] requestSave(byte[] cookie, char[] password) throws NoSessionExpiredException, NoSessionConfirmedException, NoSessionNotChangedException, NoSessionAlreadyAwaitingConfirmationException, NoDashSessionBadUUID {
|
||||
/* Step 4. Provides a user with the new binary file */
|
||||
return NoSessionSphere.save(cookie, password);
|
||||
}
|
||||
|
||||
public static void confirm(byte[] cookie, char[] password, byte[] data) throws NoSessionExpiredException, NoSessionConfirmedException, NoSessionNotAwaitingConfirmationException, NoUserNotValidException, NoDashSessionBadUUID {
|
||||
/* Step 5. Assumes the user has re-uploaded the file along with providing the same password.
|
||||
* Further attempts of getUser or getSessionState will fail with a NoSessionExpiredException*/
|
||||
NoSessionSphere.confirm(cookie, password, data);
|
||||
}
|
||||
|
||||
public static void addByteSet(NoByteSet byteSet, PublicKey publicKey) {
|
||||
NoByteSetSphere.add(byteSet, publicKey);
|
||||
}
|
||||
|
||||
public static void shred(byte[] cookie) {
|
||||
/* 3.2 Hot pull */
|
||||
NoSessionSphere.shred(cookie);
|
||||
}
|
||||
|
||||
public static void triggerPrune() {
|
||||
NoSessionSphere.prune();
|
||||
}
|
||||
|
||||
}
|
||||
6
src/nodash/core/NoRegister.java
Normal file
6
src/nodash/core/NoRegister.java
Normal file
@@ -0,0 +1,6 @@
|
||||
package nodash.core;
|
||||
|
||||
public final class NoRegister {
|
||||
public byte[] cookie;
|
||||
public byte[] data;
|
||||
}
|
||||
191
src/nodash/core/NoUtil.java
Normal file
191
src/nodash/core/NoUtil.java
Normal file
@@ -0,0 +1,191 @@
|
||||
package nodash.core;
|
||||
|
||||
import java.security.InvalidKeyException;
|
||||
import java.security.MessageDigest;
|
||||
import java.security.NoSuchAlgorithmException;
|
||||
import java.security.PrivateKey;
|
||||
import java.security.PublicKey;
|
||||
import java.security.spec.InvalidKeySpecException;
|
||||
import java.security.spec.KeySpec;
|
||||
|
||||
import javax.crypto.BadPaddingException;
|
||||
import javax.crypto.Cipher;
|
||||
import javax.crypto.IllegalBlockSizeException;
|
||||
import javax.crypto.NoSuchPaddingException;
|
||||
import javax.crypto.SecretKey;
|
||||
import javax.crypto.SecretKeyFactory;
|
||||
import javax.crypto.spec.PBEKeySpec;
|
||||
import javax.crypto.spec.SecretKeySpec;
|
||||
|
||||
import nodash.exceptions.NoDashFatalException;
|
||||
|
||||
public final class NoUtil {
|
||||
public static final String CIPHER_TYPE = "AES/ECB/PKCS5PADDING";
|
||||
public static final String CIPHER_KEY_SPEC = "AES";
|
||||
public static final String DIGEST_TYPE = "SHA-512";
|
||||
public static final String PBE_TYPE = "PBKDF2WithHmacSHA1";
|
||||
public static final String CIPHER_RSA_TYPE = "RSA/ECB/PKCS1PADDING";
|
||||
public static final String KEYPAIR_ALGORITHM = "RSA";
|
||||
public static final String SECURERANDOM_ALGORITHM = "SHA1PRNG";
|
||||
public static final String SECURERANDOM_PROVIDER = "SUN";
|
||||
public static final int RSA_STRENGTH = 4096;
|
||||
public static final int AES_STRENGTH = 256;
|
||||
public static final byte BLANK_BYTE = 'A';
|
||||
|
||||
public static char[] bytesToChars(byte[] array) {
|
||||
char[] result = new char[array.length];
|
||||
for (int x=0; x<array.length; x++) {
|
||||
result[x] = (char) array[x];
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
public static byte[] charToBytes(char[] array) {
|
||||
byte[] result = new byte[array.length];
|
||||
for (int x=0; x<array.length; x++) {
|
||||
result[x] = (byte) array[x];
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
public static void wipeBytes(byte[] array) {
|
||||
for (int x=0; x<array.length; x++) {
|
||||
array[x] = NoUtil.BLANK_BYTE;
|
||||
}
|
||||
}
|
||||
|
||||
public static void wipeChars(char[] array) {
|
||||
for (int x=0; x<array.length; x++) {
|
||||
array[x] = NoUtil.BLANK_BYTE;
|
||||
}
|
||||
}
|
||||
|
||||
public static byte[] getPBEKeyFromPassword(char[] password) {
|
||||
SecretKeyFactory skf;
|
||||
try {
|
||||
skf = SecretKeyFactory.getInstance(NoUtil.PBE_TYPE);
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
throw new NoDashFatalException("Value for PBE_TYPE is not valid.");
|
||||
}
|
||||
KeySpec spec = new PBEKeySpec(password, NoCore.config.secretKey.getEncoded(), 65536, 256);
|
||||
SecretKey key;
|
||||
try {
|
||||
key = skf.generateSecret(spec);
|
||||
} catch (InvalidKeySpecException e) {
|
||||
throw new NoDashFatalException("PBE manager unable to derive key from password.");
|
||||
}
|
||||
NoUtil.wipeChars(password);
|
||||
return key.getEncoded();
|
||||
}
|
||||
|
||||
public static byte[] getHashFromByteArray(byte[] bytes) {
|
||||
try {
|
||||
MessageDigest messageDigest = MessageDigest.getInstance(NoUtil.DIGEST_TYPE);
|
||||
return messageDigest.digest(bytes);
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
public static byte[] decryptByteArray(byte[] data, char[] password) throws IllegalBlockSizeException, BadPaddingException {
|
||||
byte[] passwordByte = NoUtil.getPBEKeyFromPassword(password);
|
||||
byte[] response = NoUtil.decrypt(NoUtil.decrypt(data), passwordByte);
|
||||
NoUtil.wipeBytes(passwordByte);
|
||||
return response;
|
||||
}
|
||||
|
||||
public static byte[] encryptByteArray(byte[] data, char[] password) {
|
||||
byte[] passwordByte = NoUtil.getPBEKeyFromPassword(password);
|
||||
byte[] response = NoUtil.encrypt(NoUtil.encrypt(data, passwordByte));
|
||||
NoUtil.wipeBytes(passwordByte);
|
||||
return response;
|
||||
}
|
||||
|
||||
public static byte[] encrypt(byte[] data, byte[] key) {
|
||||
Cipher cipher;
|
||||
try {
|
||||
cipher = Cipher.getInstance(NoUtil.CIPHER_TYPE);
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
throw new NoDashFatalException("Value for CIPHER_TYPE is not valid (no such algorithm).");
|
||||
} catch (NoSuchPaddingException e) {
|
||||
throw new NoDashFatalException("Value for CIPHER_TYPE is not valid (no such padding).");
|
||||
}
|
||||
SecretKeySpec secretKey = new SecretKeySpec(key, NoUtil.CIPHER_KEY_SPEC);
|
||||
try {
|
||||
cipher.init(Cipher.ENCRYPT_MODE, secretKey);
|
||||
} catch (InvalidKeyException e) {
|
||||
throw new NoDashFatalException("Secret key is invalid.");
|
||||
}
|
||||
|
||||
try {
|
||||
return cipher.doFinal(data);
|
||||
} catch (IllegalBlockSizeException e) {
|
||||
throw new NoDashFatalException("Block size exception encountered during encryption.");
|
||||
} catch (BadPaddingException e) {
|
||||
throw new NoDashFatalException("Bad padding exception encountered during encryption.");
|
||||
}
|
||||
}
|
||||
|
||||
public static byte[] encrypt(byte[] data) {
|
||||
return NoUtil.encrypt(data, NoCore.config.secretKey.getEncoded());
|
||||
}
|
||||
|
||||
public static byte[] decrypt(byte[] data, byte[] key) throws IllegalBlockSizeException, BadPaddingException {
|
||||
Cipher cipher;
|
||||
try {
|
||||
cipher = Cipher.getInstance(NoUtil.CIPHER_TYPE);
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
throw new NoDashFatalException("Value for CIPHER_TYPE is not valid (no such algorithm).");
|
||||
} catch (NoSuchPaddingException e) {
|
||||
throw new NoDashFatalException("Value for CIPHER_TYPE is not valid (no such padding).");
|
||||
}
|
||||
SecretKeySpec secretKey = new SecretKeySpec(key, NoUtil.CIPHER_KEY_SPEC);
|
||||
try {
|
||||
cipher.init(Cipher.DECRYPT_MODE, secretKey);
|
||||
} catch (InvalidKeyException e) {
|
||||
throw new NoDashFatalException("Secret key is invalid.");
|
||||
}
|
||||
|
||||
return cipher.doFinal(data);
|
||||
}
|
||||
|
||||
public static byte[] decrypt(byte[] data) throws IllegalBlockSizeException, BadPaddingException {
|
||||
return NoUtil.decrypt(data, NoCore.config.secretKey.getEncoded());
|
||||
}
|
||||
|
||||
public static byte[] encryptRSA(byte[] data, PublicKey publicKey) {
|
||||
Cipher cipher;
|
||||
try {
|
||||
cipher = Cipher.getInstance(NoUtil.CIPHER_RSA_TYPE);
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
throw new NoDashFatalException("Value for CIPHER_RSA_TYPE is not valid (no such algorithm).");
|
||||
} catch (NoSuchPaddingException e) {
|
||||
throw new NoDashFatalException("Value for CIPHER_RSA_TYPE is not valid (no such padding).");
|
||||
}
|
||||
try {
|
||||
cipher.init(Cipher.ENCRYPT_MODE, publicKey);
|
||||
return cipher.doFinal(data);
|
||||
} catch (InvalidKeyException e){
|
||||
throw new NoDashFatalException("Public key invalid.");
|
||||
} catch (IllegalBlockSizeException e) {
|
||||
throw new NoDashFatalException("Unable to encrypt data stream with public key.");
|
||||
} catch (BadPaddingException e) {
|
||||
throw new NoDashFatalException("Unable to encrypt data stream with public key.");
|
||||
}
|
||||
}
|
||||
|
||||
public static byte[] decryptRSA(byte[] data, PrivateKey privateKey) throws InvalidKeyException, IllegalBlockSizeException, BadPaddingException {
|
||||
Cipher cipher;
|
||||
try {
|
||||
cipher = Cipher.getInstance(NoUtil.CIPHER_RSA_TYPE);
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
throw new NoDashFatalException("Value for CIPHER_RSA_TYPE is not valid (no such algorithm).");
|
||||
} catch (NoSuchPaddingException e) {
|
||||
throw new NoDashFatalException("Value for CIPHER_RSA_TYPE is not valid (no such padding).");
|
||||
}
|
||||
cipher.init(Cipher.DECRYPT_MODE, privateKey);
|
||||
return cipher.doFinal(data);
|
||||
}
|
||||
|
||||
}
|
||||
38
src/nodash/core/spheres/NoByteSetSphere.java
Normal file
38
src/nodash/core/spheres/NoByteSetSphere.java
Normal file
@@ -0,0 +1,38 @@
|
||||
package nodash.core.spheres;
|
||||
|
||||
import java.security.PublicKey;
|
||||
import java.util.ArrayList;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
|
||||
import nodash.models.NoByteSet;
|
||||
import nodash.models.NoUser;
|
||||
|
||||
public final class NoByteSetSphere {
|
||||
private static final ArrayList<NoByteSet> EMPTY_BYTESET_LIST = new ArrayList<NoByteSet>(0);
|
||||
|
||||
private static ConcurrentHashMap<PublicKey, ArrayList<NoByteSet>> byteSets = new ConcurrentHashMap<PublicKey, ArrayList<NoByteSet>>();
|
||||
|
||||
public static void add(NoByteSet byteSet, PublicKey publicKey) {
|
||||
if (!NoByteSetSphere.byteSets.containsKey(publicKey)) {
|
||||
NoByteSetSphere.byteSets.put(publicKey, new ArrayList<NoByteSet>());
|
||||
}
|
||||
NoByteSetSphere.byteSets.get(publicKey).add(byteSet);
|
||||
}
|
||||
|
||||
public static void addList(ArrayList<NoByteSet> byteSetList, PublicKey publicKey) {
|
||||
if (!NoByteSetSphere.byteSets.containsKey(publicKey)) {
|
||||
NoByteSetSphere.byteSets.put(publicKey, new ArrayList<NoByteSet>());
|
||||
}
|
||||
NoByteSetSphere.byteSets.get(publicKey).addAll(byteSetList);
|
||||
}
|
||||
|
||||
public static ArrayList<NoByteSet> consume(NoUser user) {
|
||||
if (NoByteSetSphere.byteSets.containsKey(user.getRSAPublicKey())) {
|
||||
ArrayList<NoByteSet> result = NoByteSetSphere.byteSets.get(user.getRSAPublicKey());
|
||||
NoByteSetSphere.byteSets.remove(user.getRSAPublicKey());
|
||||
return result;
|
||||
} else {
|
||||
return NoByteSetSphere.EMPTY_BYTESET_LIST;
|
||||
}
|
||||
}
|
||||
}
|
||||
77
src/nodash/core/spheres/NoHashSphere.java
Normal file
77
src/nodash/core/spheres/NoHashSphere.java
Normal file
@@ -0,0 +1,77 @@
|
||||
package nodash.core.spheres;
|
||||
|
||||
import java.io.ByteArrayInputStream;
|
||||
import java.io.ByteArrayOutputStream;
|
||||
import java.io.File;
|
||||
import java.io.IOException;
|
||||
import java.io.ObjectInputStream;
|
||||
import java.io.ObjectOutputStream;
|
||||
import java.nio.file.Files;
|
||||
import java.nio.file.StandardOpenOption;
|
||||
import java.util.Collections;
|
||||
import java.util.Set;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
|
||||
import nodash.core.NoCore;
|
||||
import nodash.exceptions.NoDashFatalException;
|
||||
import nodash.models.NoUser;
|
||||
|
||||
public final class NoHashSphere {
|
||||
private static Set<String> database = Collections.newSetFromMap(new ConcurrentHashMap<String, Boolean>());
|
||||
|
||||
@SuppressWarnings("unchecked")
|
||||
public static void setup() {
|
||||
if (NoCore.config.saveDatabase) {
|
||||
File file = new File(NoCore.config.databaseFilename);
|
||||
if (file.exists()) {
|
||||
try {
|
||||
byte[] data = Files.readAllBytes(file.toPath());
|
||||
ByteArrayInputStream bais = new ByteArrayInputStream(data);
|
||||
ObjectInputStream ois = new ObjectInputStream(bais);
|
||||
NoHashSphere.database = (Set<String>) ois.readObject();
|
||||
ois.close();
|
||||
bais.close();
|
||||
} catch (IOException e){
|
||||
throw new NoDashFatalException("Unable to load up given database file.");
|
||||
} catch (ClassNotFoundException e) {
|
||||
throw new NoDashFatalException("Database file not in a verifiable format.");
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
public static synchronized void saveToFile() throws IOException {
|
||||
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
||||
ObjectOutputStream oos = new ObjectOutputStream(baos);
|
||||
oos.writeObject(NoHashSphere.database);
|
||||
byte[] data = baos.toByteArray();
|
||||
oos.close();
|
||||
baos.close();
|
||||
File file = new File(NoCore.config.databaseFilename);
|
||||
Files.write(file.toPath(), data, StandardOpenOption.CREATE);
|
||||
}
|
||||
|
||||
public static synchronized void addNewNoUser(NoUser user) throws IOException {
|
||||
String hash = user.createHashString();
|
||||
NoHashSphere.database.add(hash);
|
||||
NoHashSphere.saveToFile();
|
||||
}
|
||||
|
||||
public static synchronized void insertHash(String hash) throws IOException {
|
||||
NoHashSphere.database.add(hash);
|
||||
NoHashSphere.saveToFile();
|
||||
}
|
||||
|
||||
public static synchronized void removeHash(String hash) throws IOException {
|
||||
NoHashSphere.database.remove(hash);
|
||||
NoHashSphere.saveToFile();
|
||||
}
|
||||
|
||||
public static synchronized boolean checkHash(String hash) {
|
||||
return NoHashSphere.database.contains(hash);
|
||||
}
|
||||
|
||||
public static synchronized int size() {
|
||||
return NoHashSphere.database.size();
|
||||
}
|
||||
}
|
||||
165
src/nodash/core/spheres/NoSessionSphere.java
Normal file
165
src/nodash/core/spheres/NoSessionSphere.java
Normal file
@@ -0,0 +1,165 @@
|
||||
package nodash.core.spheres;
|
||||
|
||||
import java.util.Collections;
|
||||
import java.util.Set;
|
||||
import java.util.UUID;
|
||||
import java.util.concurrent.ConcurrentHashMap;
|
||||
|
||||
import nodash.core.NoRegister;
|
||||
import nodash.exceptions.NoByteSetBadDecryptionException;
|
||||
import nodash.exceptions.NoDashFatalException;
|
||||
import nodash.exceptions.NoDashSessionBadUUID;
|
||||
import nodash.exceptions.NoSessionAlreadyAwaitingConfirmationException;
|
||||
import nodash.exceptions.NoSessionConfirmedException;
|
||||
import nodash.exceptions.NoSessionExpiredException;
|
||||
import nodash.exceptions.NoSessionNotAwaitingConfirmationException;
|
||||
import nodash.exceptions.NoSessionNotChangedException;
|
||||
import nodash.exceptions.NoUserAlreadyOnlineException;
|
||||
import nodash.exceptions.NoUserNotValidException;
|
||||
import nodash.models.NoByteSet;
|
||||
import nodash.models.NoSession;
|
||||
import nodash.models.NoUser;
|
||||
import nodash.models.NoSession.NoState;
|
||||
|
||||
public final class NoSessionSphere {
|
||||
private static ConcurrentHashMap<UUID, NoSession> sessions = new ConcurrentHashMap<UUID, NoSession>();
|
||||
private static Set<byte[]> originalHashesOnline = Collections.newSetFromMap(new ConcurrentHashMap<byte[], Boolean>());
|
||||
|
||||
public static synchronized void prune() {
|
||||
for (UUID uuid : NoSessionSphere.sessions.keySet()) {
|
||||
pruneSingle(uuid);
|
||||
}
|
||||
}
|
||||
|
||||
public static void shred(byte[] encryptedUUID) {
|
||||
try {
|
||||
UUID uuid = NoSession.decryptUUID(encryptedUUID);
|
||||
if (NoSessionSphere.sessions.containsKey(uuid)) {
|
||||
NoSession session = NoSessionSphere.sessions.get(uuid);
|
||||
NoByteSetSphere.addList(session.incoming, session.current.getRSAPublicKey());
|
||||
NoSessionSphere.originalHashesOnline.remove(session.getOriginalHash());
|
||||
NoSessionSphere.sessions.remove(uuid);
|
||||
session = null;
|
||||
}
|
||||
} catch (NoDashSessionBadUUID e) {
|
||||
// Suppress, doesn't matter
|
||||
}
|
||||
}
|
||||
|
||||
public static synchronized void pruneSingle(UUID uuid) {
|
||||
NoSession session = NoSessionSphere.sessions.get(uuid);
|
||||
try {
|
||||
session.check();
|
||||
} catch (NoSessionExpiredException e) {
|
||||
/* Resultant from 3.1 and 3.2 */
|
||||
NoByteSetSphere.addList(session.incoming, session.current.getRSAPublicKey());
|
||||
NoSessionSphere.originalHashesOnline.remove(session.getOriginalHash());
|
||||
NoSessionSphere.sessions.remove(uuid);
|
||||
session = null;
|
||||
} catch (NoSessionConfirmedException e) {
|
||||
/* Should be cleaned up at 5.2 */
|
||||
}
|
||||
}
|
||||
|
||||
public static synchronized byte[] login(byte[] data, char[] password) throws NoUserNotValidException, NoUserAlreadyOnlineException, NoSessionExpiredException {
|
||||
/* 1. Login with byte[] data and byte[] password */
|
||||
NoSession session = new NoSession(data, password);
|
||||
/* 1.1. User currently has an online session, must wait for it to expire. */
|
||||
if (originalHashesOnline.contains(session.getOriginalHash())) {
|
||||
throw new NoUserAlreadyOnlineException();
|
||||
}
|
||||
/* 1.2. User successfully logged in: set up session records. */
|
||||
NoSessionSphere.originalHashesOnline.add(session.getOriginalHash());
|
||||
NoSessionSphere.sessions.put(session.uuid, session);
|
||||
|
||||
/* 2. Check NoByteSetSphere for incoming Influences */
|
||||
session.incoming = NoByteSetSphere.consume(session.current);
|
||||
for (NoByteSet nbs : session.incoming) {
|
||||
/* 2.1 Decrypt NoInfluence from NoByteSet, let the current user consume them */
|
||||
try {
|
||||
session.consume(nbs);
|
||||
} catch (NoByteSetBadDecryptionException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
} /* 2.2 Alternatively, no NoByteSets to consume */
|
||||
|
||||
try {
|
||||
session.check();
|
||||
} catch (NoSessionConfirmedException e) {
|
||||
/* Should be impossible to reach */
|
||||
throw new NoDashFatalException(e);
|
||||
}
|
||||
|
||||
/* Will set to 2.1[MODIFIED] or 2.2[IDLE] */
|
||||
|
||||
/* Precursor to 3.; allow website to associate user session with a cookie. */
|
||||
return session.getEncryptedUUID();
|
||||
}
|
||||
|
||||
public static NoUser getUser(byte[] encryptedUUID) throws NoDashSessionBadUUID, NoSessionExpiredException, NoSessionConfirmedException {
|
||||
UUID uuid = NoSession.decryptUUID(encryptedUUID);
|
||||
if (NoSessionSphere.sessions.containsKey(uuid)) {
|
||||
NoSessionSphere.pruneSingle(uuid);
|
||||
return NoSessionSphere.sessions.get(uuid).getNoUser();
|
||||
}
|
||||
throw new NoSessionExpiredException();
|
||||
}
|
||||
|
||||
public static NoState getState(byte[] encryptedUUID) throws NoDashSessionBadUUID, NoSessionExpiredException, NoSessionConfirmedException {
|
||||
UUID uuid = NoSession.decryptUUID(encryptedUUID);
|
||||
if (NoSessionSphere.sessions.containsKey(uuid)) {
|
||||
NoSessionSphere.pruneSingle(uuid);
|
||||
NoSession session = NoSessionSphere.sessions.get(uuid);
|
||||
return session.getNoState();
|
||||
}
|
||||
throw new NoSessionExpiredException();
|
||||
}
|
||||
|
||||
public static synchronized byte[] save(byte[] encryptedUUID, char[] password) throws NoDashSessionBadUUID, NoSessionExpiredException, NoSessionConfirmedException, NoSessionNotChangedException, NoSessionAlreadyAwaitingConfirmationException {
|
||||
UUID uuid = NoSession.decryptUUID(encryptedUUID);
|
||||
if (NoSessionSphere.sessions.containsKey(uuid)) {
|
||||
NoSessionSphere.pruneSingle(uuid);
|
||||
NoSession session = NoSessionSphere.sessions.get(uuid);
|
||||
|
||||
if (session.getNoState().equals(NoState.IDLE)) {
|
||||
throw new NoSessionNotChangedException();
|
||||
} else if (session.getNoState().equals(NoState.AWAITING_CONFIRMATION)) {
|
||||
throw new NoSessionAlreadyAwaitingConfirmationException();
|
||||
}
|
||||
return session.initiateSaveAttempt(password);
|
||||
}
|
||||
throw new NoSessionExpiredException();
|
||||
}
|
||||
|
||||
public static synchronized void confirm(byte[] encryptedUUID, char[] password, byte[] data) throws NoDashSessionBadUUID, NoSessionExpiredException, NoSessionConfirmedException, NoSessionNotAwaitingConfirmationException, NoUserNotValidException {
|
||||
UUID uuid = NoSession.decryptUUID(encryptedUUID);
|
||||
if (NoSessionSphere.sessions.containsKey(uuid)) {
|
||||
NoSessionSphere.pruneSingle(uuid);
|
||||
NoSession session = NoSessionSphere.sessions.get(uuid);
|
||||
session.confirmSave(data, password);
|
||||
return;
|
||||
}
|
||||
throw new NoSessionExpiredException();
|
||||
}
|
||||
|
||||
public static synchronized NoRegister registerUser(NoUser user, char[] password) {
|
||||
NoRegister result = new NoRegister();
|
||||
NoSession session = new NoSession(user);
|
||||
NoSessionSphere.sessions.put(session.uuid, session);
|
||||
result.cookie = session.getEncryptedUUID();
|
||||
try {
|
||||
result.data = NoSessionSphere.save(result.cookie, password);
|
||||
} catch (NoDashSessionBadUUID e) {
|
||||
throw new NoDashFatalException("Immediately generated cookie throwing bad cookie error.");
|
||||
} catch (NoSessionExpiredException e) {
|
||||
throw new NoDashFatalException("Session expired before it was even returned to client.");
|
||||
} catch (NoSessionConfirmedException e) {
|
||||
throw new NoDashFatalException("Session is in confirmed state before it was returned to client.");
|
||||
} catch (NoSessionNotChangedException e) {
|
||||
throw new NoDashFatalException("Session claims to be unchanged but user is newly registered.");
|
||||
} catch (NoSessionAlreadyAwaitingConfirmationException e) {
|
||||
throw new NoDashFatalException("Session claims to be awaiting confirmation before returning data to the user.");
|
||||
}
|
||||
return result;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user