ajouté commentaires descriptifs
This commit is contained in:
parent
d3adc16146
commit
a2a091206b
7 changed files with 70 additions and 741 deletions
|
@ -39,13 +39,8 @@ public class CommunicationUDP extends Thread {
|
|||
return tmp;
|
||||
}
|
||||
|
||||
public void setObserver (Observer obs) {
|
||||
this.observer=obs;
|
||||
}
|
||||
|
||||
public Observer getObserver () {
|
||||
return this.observer;
|
||||
}
|
||||
// ----- CHECKERS ----- //
|
||||
|
||||
protected boolean containsUserFromID(String id) {
|
||||
for(Utilisateur u : users) {
|
||||
|
@ -67,6 +62,9 @@ public class CommunicationUDP extends Thread {
|
|||
return false;
|
||||
}
|
||||
|
||||
|
||||
// ----- GETTERS ----- //
|
||||
|
||||
public int getPortFromPseudo(String pseudo) {
|
||||
for(int i=0; i < users.size() ; i++) {
|
||||
|
||||
|
@ -95,13 +93,30 @@ public class CommunicationUDP extends Thread {
|
|||
return null;
|
||||
}
|
||||
|
||||
public Observer getObserver () {
|
||||
return this.observer;
|
||||
}
|
||||
|
||||
|
||||
// ----- SETTERS ----- //
|
||||
|
||||
public void setObserver (Observer obs) {
|
||||
this.observer=obs;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
// ----- USER LIST MANAGEMENT ----- //
|
||||
|
||||
//Prints a html table containing the pseudo of all active local users
|
||||
public void printActiveUsersUDP(PrintWriter out) {
|
||||
for (Utilisateur uIn : users) {
|
||||
out.println("<TH> " + uIn.getPseudo() + ",</TH>");
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
//Add an user to the list of active local users
|
||||
protected synchronized void addUser(String idClient, String pseudoClient, InetAddress ipClient, int port) throws IOException {
|
||||
users.add(new Utilisateur(idClient, pseudoClient, ipClient, port));
|
||||
try {
|
||||
|
@ -111,6 +126,7 @@ public class CommunicationUDP extends Thread {
|
|||
}
|
||||
}
|
||||
|
||||
//Change the pseudo of an user already in the active local users list
|
||||
protected synchronized void changePseudoUser(String idClient, String pseudoClient, InetAddress ipClient, int port) {
|
||||
int index = getIndexFromID(idClient);
|
||||
users.get(index).setPseudo(pseudoClient);
|
||||
|
@ -121,7 +137,7 @@ public class CommunicationUDP extends Thread {
|
|||
}
|
||||
}
|
||||
|
||||
|
||||
//Remove an user from the active local users list
|
||||
protected synchronized void removeUser(String idClient, String pseudoClient,InetAddress ipClient, int port) {
|
||||
int index = getIndexFromID(idClient);
|
||||
if( index != -1) {
|
||||
|
@ -134,6 +150,7 @@ public class CommunicationUDP extends Thread {
|
|||
}
|
||||
}
|
||||
|
||||
//Remove all users from the active local users list
|
||||
public void removeAll(){
|
||||
int oSize = users.size();
|
||||
for(int i=0; i<oSize;i++) {
|
||||
|
@ -142,6 +159,9 @@ public class CommunicationUDP extends Thread {
|
|||
}
|
||||
|
||||
|
||||
// ----- SENDING MESSAGES ----- //
|
||||
|
||||
|
||||
// Send the message "add,id,pseudo" to localhost on all the ports in
|
||||
// "portOthers"
|
||||
// This allows the receivers' agent (portOthers) to create or modify an entry with the
|
||||
|
@ -177,7 +197,7 @@ public class CommunicationUDP extends Thread {
|
|||
}
|
||||
|
||||
|
||||
//Broadcast a given message on the local network (here podelized by ports)
|
||||
//Broadcast a given message on the local network (here modelized by ports)
|
||||
public void sendMessage(Message m) {
|
||||
try {
|
||||
for(int port : this.portOthers) {
|
||||
|
|
|
@ -15,16 +15,28 @@ public class UDPClient {
|
|||
private DatagramSocket sockUDP;
|
||||
//private InetAddress broadcast;
|
||||
|
||||
|
||||
/**
|
||||
* Create an UDP client on the specified port. It will be used to notify the
|
||||
* other users of this application's user state (Connected/Disconnected/Pseudo
|
||||
* changed).
|
||||
*
|
||||
* @param port
|
||||
* @throws SocketException
|
||||
* @throws UnknownHostException
|
||||
*/
|
||||
public UDPClient(int port) throws SocketException, UnknownHostException {
|
||||
this.sockUDP = new DatagramSocket(port);
|
||||
|
||||
//InetAddress localHost = InetAddress.getLocalHost();
|
||||
//NetworkInterface networkInterface = NetworkInterface.getByInetAddress(localHost);
|
||||
//this.broadcast = networkInterface.getInterfaceAddresses().get(0).getBroadcast();
|
||||
}
|
||||
|
||||
|
||||
//Send a message casted as string to the specified port on localhost
|
||||
/**
|
||||
* Send a message to the specified port on localhost.
|
||||
*
|
||||
* @param message
|
||||
* @param port
|
||||
* @throws IOException
|
||||
*/
|
||||
protected void sendMessageUDP_local(Message message, int port, InetAddress clientAddress) throws IOException {
|
||||
String messageString= message.toString();
|
||||
DatagramPacket outpacket = new DatagramPacket(messageString.getBytes(), messageString.length(), clientAddress, port);
|
||||
|
@ -32,10 +44,19 @@ public class UDPClient {
|
|||
|
||||
}
|
||||
|
||||
// protected void sendMessageUDP_broadcast(String message, int port) throws IOException{
|
||||
// String messageString=message.toString();
|
||||
// DatagramPacket outpacket = new DatagramPacket(messageString.getBytes(), messageString.length(), this.broadcast, port);
|
||||
// this.sockUDP.send(outpacket);
|
||||
// }
|
||||
/**
|
||||
* Send a message to the given address on the specified port.
|
||||
*
|
||||
* @param message
|
||||
* @param port
|
||||
* @param clientAddress
|
||||
* @throws IOException
|
||||
*/
|
||||
private void sendMessageUDP(Message message, int port, InetAddress clientAddress) throws IOException {
|
||||
String messageString = message.toString();
|
||||
DatagramPacket outpacket = new DatagramPacket(messageString.getBytes(), messageString.length(), clientAddress,
|
||||
port);
|
||||
this.sockUDP.send(outpacket);
|
||||
}
|
||||
|
||||
}
|
||||
|
|
|
@ -16,6 +16,15 @@ public class UDPServer extends Thread {
|
|||
private CommunicationUDP commUDP;
|
||||
private byte[] buffer;
|
||||
|
||||
|
||||
/**
|
||||
* Create an UDP Server on the specified port. It will be used to read the
|
||||
* other users states (Connected/Disconnected/Pseudo).
|
||||
*
|
||||
* @param port
|
||||
* @param commUDP
|
||||
* @throws SocketException
|
||||
*/
|
||||
public UDPServer(int port, CommunicationUDP commUDP) throws SocketException {
|
||||
this.commUDP = commUDP;
|
||||
this.sockUDP = new DatagramSocket(port);
|
||||
|
|
|
@ -1,73 +0,0 @@
|
|||
package database;
|
||||
|
||||
import java.sql.Connection;
|
||||
import java.sql.SQLException;
|
||||
import java.sql.Statement;
|
||||
|
||||
class SQLiteCreateTables {
|
||||
|
||||
|
||||
protected static void createTableUser(Connection connec) throws SQLException {
|
||||
String createTableUser = "CREATE TABLE IF NOT EXISTS user (\r\n"
|
||||
+ " id INTEGER PRIMARY KEY AUTOINCREMENT,\r\n"
|
||||
+ " username VARCHAR (50) NOT NULL\r\n"
|
||||
+ " UNIQUE ON CONFLICT ROLLBACK,\r\n"
|
||||
+ " pwd_salt BLOB,\r\n"
|
||||
+ " db_datakey_salt BLOB,\r\n"
|
||||
+ " encrypted_pwd_hashsalt BLOB,\r\n"
|
||||
+ " encrypted_db_datakey BLOB,\r\n"
|
||||
+ " iv_datakey BLOB\r\n"
|
||||
+ ");";
|
||||
|
||||
Statement stmt = connec.createStatement();
|
||||
stmt.execute(createTableUser);
|
||||
|
||||
}
|
||||
|
||||
protected static void createTableConversation(Connection connec) throws SQLException {
|
||||
String createTableConversation = "CREATE TABLE IF NOT EXISTS conversation (\r\n"
|
||||
+ " id_conversation INTEGER PRIMARY KEY AUTOINCREMENT,\r\n"
|
||||
+ " id_emetteur INTEGER REFERENCES user (id) \r\n"
|
||||
+ " NOT NULL,\r\n"
|
||||
+ " id_recepteur INTEGER REFERENCES user (id) \r\n"
|
||||
+ " NOT NULL,\r\n"
|
||||
+" iv_conversation BLOB NOT NULL"
|
||||
+ ");";
|
||||
|
||||
Statement stmt = connec.createStatement();
|
||||
stmt.execute(createTableConversation);
|
||||
|
||||
}
|
||||
|
||||
protected static void createTableMessage(Connection connec) throws SQLException {
|
||||
String createTableMessage = "CREATE TABLE IF NOT EXISTS message (\r\n"
|
||||
+ " id_message INTEGER PRIMARY KEY AUTOINCREMENT,\r\n"
|
||||
+ " id_conversation INTEGER REFERENCES conversation (id_conversation) \r\n"
|
||||
+ " NOT NULL,\r\n"
|
||||
+ " id_type INTEGER REFERENCES type (id_type) \r\n"
|
||||
+ " NOT NULL,\r\n"
|
||||
+ " content BLOB,\r\n"
|
||||
+ " date INTEGER NOT NULL,\r\n"
|
||||
+ " extension VARCHAR (20) \r\n"
|
||||
+ ");\r\n";
|
||||
|
||||
Statement stmt = connec.createStatement();
|
||||
stmt.execute(createTableMessage);
|
||||
}
|
||||
|
||||
protected static void createTableType(Connection connec) throws SQLException {
|
||||
String createTableType = "CREATE TABLE IF NOT EXISTS type (\r\n" + " id_type INTEGER PRIMARY KEY,\r\n"
|
||||
+ " label VARCHAR (20) NOT NULL\r\n" + ");";
|
||||
|
||||
Statement stmt = connec.createStatement();
|
||||
stmt.execute(createTableType);
|
||||
|
||||
String typeText = "INSERT OR IGNORE INTO type (id_type, label) " + "VALUES (0, 'text');";
|
||||
String typeFile = "INSERT OR IGNORE INTO type (id_type, label) " + "VALUES (1, 'file');";
|
||||
String typeImage = "INSERT OR IGNORE INTO type (id_type, label) " + "VALUES (2, 'image');";
|
||||
|
||||
stmt.execute(typeText);
|
||||
stmt.execute(typeFile);
|
||||
stmt.execute(typeImage);
|
||||
}
|
||||
}
|
|
@ -1,84 +0,0 @@
|
|||
package database;
|
||||
|
||||
|
||||
import java.util.Base64;
|
||||
import java.util.Random;
|
||||
|
||||
import javax.crypto.*;
|
||||
import javax.crypto.spec.*;
|
||||
import java.security.*;
|
||||
import java.security.spec.*;
|
||||
|
||||
class SQLiteEncprytion {
|
||||
|
||||
private static final Random RANDOM = new SecureRandom();
|
||||
private static final int ITERATIONS = 10000;
|
||||
private static final int KEY_LENGTH = 256;
|
||||
protected static final String encryptAlgorithm = "AES/CBC/PKCS5Padding";
|
||||
|
||||
protected static byte[] getNextSalt() {
|
||||
byte[] salt = new byte[24];
|
||||
RANDOM.nextBytes(salt);
|
||||
return salt;
|
||||
}
|
||||
|
||||
|
||||
protected static byte[] hash(char[] password, byte[] salt) {
|
||||
return SQLiteEncprytion.getKey(password, salt).getEncoded();
|
||||
}
|
||||
|
||||
|
||||
protected static SecretKey getKey(char[] password, byte[] salt) {
|
||||
PBEKeySpec saltpwd = new PBEKeySpec(password, salt, ITERATIONS, KEY_LENGTH);
|
||||
|
||||
try {
|
||||
SecretKeyFactory skf = SecretKeyFactory.getInstance("PBKDF2WithHmacSHA1");
|
||||
SecretKey tmp = skf.generateSecret(saltpwd);
|
||||
SecretKey key = new SecretKeySpec(tmp.getEncoded(), "AES");
|
||||
return key;
|
||||
} catch (InvalidKeySpecException | NoSuchAlgorithmException e) {
|
||||
e.printStackTrace();
|
||||
} finally {
|
||||
saltpwd.clearPassword();
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
|
||||
public static IvParameterSpec generateIv() {
|
||||
byte[] iv = new byte[16];
|
||||
new SecureRandom().nextBytes(iv);
|
||||
return new IvParameterSpec(iv);
|
||||
}
|
||||
|
||||
|
||||
protected static byte[] encrypt(String algorithm, byte[] input, SecretKey key, IvParameterSpec iv) throws NoSuchAlgorithmException, NoSuchPaddingException, InvalidKeyException, InvalidAlgorithmParameterException, IllegalBlockSizeException, BadPaddingException{
|
||||
Cipher cipher = Cipher.getInstance(algorithm);
|
||||
cipher.init(Cipher.ENCRYPT_MODE, key, iv);
|
||||
byte[] cipherText = cipher.doFinal(input);
|
||||
return Base64.getEncoder().encode(cipherText);
|
||||
}
|
||||
|
||||
|
||||
protected static byte[] decryptByte(String algorithm, byte[] cipherText, SecretKey key, IvParameterSpec iv) throws NoSuchAlgorithmException, NoSuchPaddingException, InvalidKeyException, InvalidAlgorithmParameterException, IllegalBlockSizeException, BadPaddingException{
|
||||
Cipher cipher = Cipher.getInstance(algorithm);
|
||||
cipher.init(Cipher.DECRYPT_MODE, key, iv);
|
||||
byte[] plainText = cipher.doFinal(Base64.getDecoder().decode(cipherText));
|
||||
return plainText;
|
||||
}
|
||||
|
||||
protected static String decryptString(String algorithm, byte[] cipherText, SecretKey key, IvParameterSpec iv) throws InvalidKeyException, NoSuchAlgorithmException, NoSuchPaddingException, InvalidAlgorithmParameterException, IllegalBlockSizeException, BadPaddingException {
|
||||
return new String(SQLiteEncprytion.decryptByte(algorithm, cipherText, key, iv) );
|
||||
}
|
||||
|
||||
|
||||
public static byte[] keyToByte(SecretKey key) {
|
||||
return Base64.getEncoder().encode(key.getEncoded());
|
||||
}
|
||||
|
||||
|
||||
public static SecretKey byteToKey(byte[] encodedKey) {
|
||||
byte[] decodedKey = Base64.getDecoder().decode(encodedKey);
|
||||
return new SecretKeySpec(decodedKey, 0, decodedKey.length, "AES");
|
||||
}
|
||||
}
|
|
@ -1,563 +0,0 @@
|
|||
package database;
|
||||
|
||||
import java.io.File;
|
||||
import java.security.InvalidAlgorithmParameterException;
|
||||
import java.security.InvalidKeyException;
|
||||
import java.security.NoSuchAlgorithmException;
|
||||
|
||||
import java.sql.Connection;
|
||||
import java.sql.DriverManager;
|
||||
import java.sql.PreparedStatement;
|
||||
import java.sql.ResultSet;
|
||||
import java.sql.SQLException;
|
||||
import java.util.ArrayList;
|
||||
|
||||
import javax.crypto.BadPaddingException;
|
||||
import javax.crypto.IllegalBlockSizeException;
|
||||
import javax.crypto.KeyGenerator;
|
||||
import javax.crypto.NoSuchPaddingException;
|
||||
import javax.crypto.SecretKey;
|
||||
import javax.crypto.spec.IvParameterSpec;
|
||||
|
||||
|
||||
import main.Utilisateur;
|
||||
import messages.MauvaisTypeMessageException;
|
||||
import messages.Message;
|
||||
import messages.MessageTexte;
|
||||
import messages.Message.TypeMessage;
|
||||
import messages.MessageFichier;
|
||||
|
||||
public class SQLiteManager {
|
||||
|
||||
private static final String DATABASE_RELATIVE_PATH = "../database";
|
||||
|
||||
public static String[] hardcodedNames = {"Olivia","Liam","Benjamin","Sophia","Charlotte","Noah","Elijah","Isabella",
|
||||
"Oliver","Emma","William","Amelia","Evelyn","James","Mia","Ava","Lucas","Mason","Ethan","Harper"};
|
||||
|
||||
private Connection connec;
|
||||
private int numDatabase;
|
||||
private SecretKey dbDataKey;
|
||||
|
||||
public SQLiteManager(int numDatabase) {
|
||||
this.numDatabase = numDatabase;
|
||||
|
||||
this.openConnection();
|
||||
|
||||
try {
|
||||
|
||||
SQLiteCreateTables.createTableUser(this.connec);
|
||||
SQLiteCreateTables.createTableConversation(this.connec);
|
||||
SQLiteCreateTables.createTableType(this.connec);
|
||||
SQLiteCreateTables.createTableMessage(this.connec);
|
||||
|
||||
} catch (SQLException e) {
|
||||
this.closeConnection();
|
||||
File db = new File("../database"+this.numDatabase+".db");
|
||||
if(db.delete()) {
|
||||
System.out.println("supp");
|
||||
}else {
|
||||
System.out.println("no supp");
|
||||
}
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
this.closeConnection();
|
||||
}
|
||||
|
||||
private void openConnection() {
|
||||
String url = "jdbc:sqlite:"+ DATABASE_RELATIVE_PATH + this.numDatabase + ".db";
|
||||
try {
|
||||
//Class.forName("org.sqlite.JDBC");
|
||||
this.connec = DriverManager.getConnection(url);
|
||||
System.out.println("Connection to bdd established");
|
||||
} catch (SQLException /*| ClassNotFoundException*/ e) {
|
||||
System.out.println(e.getMessage());
|
||||
}
|
||||
}
|
||||
|
||||
private void closeConnection() {
|
||||
try {
|
||||
if (this.connec != null) {
|
||||
this.connec.close();
|
||||
}
|
||||
} catch (SQLException e) {
|
||||
System.out.println(e.getMessage());
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
public int insertAllMessages(ArrayList<Message> messages, String usernameSender, String usernameReceiver) throws SQLException{
|
||||
int nbRows = 0;
|
||||
this.openConnection();
|
||||
|
||||
int idSender = this.getIDUser(usernameSender);
|
||||
int idReceiver = this.getIDUser(usernameReceiver);
|
||||
|
||||
if(idSender == -1) {
|
||||
this.insertUser(usernameSender);
|
||||
idSender = this.getIDUser(usernameSender);
|
||||
}
|
||||
|
||||
if(idReceiver == -1) {
|
||||
this.insertUser(usernameReceiver);
|
||||
idReceiver = this.getIDUser(usernameReceiver);
|
||||
}
|
||||
|
||||
int idConversation = getIDConversation(idSender, idReceiver);
|
||||
|
||||
if(idConversation == -1) {
|
||||
this.insertConversation(idSender, idReceiver);
|
||||
idConversation = getIDConversation(idSender, idReceiver);
|
||||
}
|
||||
|
||||
IvParameterSpec ivConversation = this.getIvConversation(idConversation);
|
||||
|
||||
this.connec.setAutoCommit(false);
|
||||
|
||||
for(Message m : messages) {
|
||||
try {
|
||||
nbRows += this.insertMessage(idConversation, m, ivConversation);
|
||||
} catch (SQLException e) {
|
||||
e.printStackTrace();
|
||||
this.connec.rollback();
|
||||
}
|
||||
}
|
||||
|
||||
this.connec.commit();
|
||||
|
||||
this.closeConnection();
|
||||
|
||||
//System.out.println("Nombre de message(s) insérée(s) : " + nbRows);
|
||||
|
||||
return nbRows;
|
||||
}
|
||||
|
||||
|
||||
public ArrayList<Message> getHistoriquesMessages(String usernameOther, String pseudoOther) throws SQLException {
|
||||
|
||||
this.openConnection();
|
||||
|
||||
|
||||
ArrayList<Message> messages = new ArrayList<Message>();
|
||||
|
||||
String usernameSelf = Utilisateur.getSelf().getId();
|
||||
|
||||
int idSelf = this.getIDUser(usernameSelf);
|
||||
int idOther = this.getIDUser(usernameOther);
|
||||
|
||||
int idConversationSelf = this.getIDConversation(idSelf, idOther);
|
||||
int idConversationOther = this.getIDConversation(idOther, idSelf);
|
||||
IvParameterSpec ivConversation = this.getIvConversation(idConversationSelf);
|
||||
// String str = "datetime(d1,'unixepoch','localtime')";
|
||||
|
||||
|
||||
String getHistoriqueRequest = "SELECT id_conversation, id_type, content, date, extension "
|
||||
+ "FROM message "
|
||||
+ "WHERE id_conversation IN (?,?) "
|
||||
+ "ORDER by date";
|
||||
|
||||
PreparedStatement prepStmt = this.connec.prepareStatement(getHistoriqueRequest);
|
||||
prepStmt.setInt(1, idConversationSelf);
|
||||
prepStmt.setInt(2, idConversationOther);
|
||||
ResultSet res = prepStmt.executeQuery();
|
||||
|
||||
//Retrieve the messages one by one
|
||||
//Create the appropriate message object depending on the type and sender/receiver
|
||||
//and add the message in the list
|
||||
while(res.next()) {
|
||||
int idType = res.getInt("id_type");
|
||||
String type = this.getType(idType);
|
||||
|
||||
String content = null;
|
||||
try {
|
||||
content = this.bytesToStringContent(res.getBytes("content"), ivConversation);
|
||||
} catch (InvalidKeyException | NoSuchAlgorithmException | NoSuchPaddingException
|
||||
| InvalidAlgorithmParameterException | IllegalBlockSizeException | BadPaddingException
|
||||
| SQLException e1) {
|
||||
//System.out.println("erreur déchiffrement");
|
||||
}
|
||||
|
||||
Message message = null;
|
||||
String extension;
|
||||
|
||||
if(!type.equals("")) {
|
||||
try {
|
||||
switch(type) {
|
||||
case "text":
|
||||
message = new MessageTexte(TypeMessage.TEXTE, content);
|
||||
break;
|
||||
case "file":
|
||||
extension = res.getString("extension");
|
||||
message = new MessageFichier(TypeMessage.FICHIER, content, extension);
|
||||
break;
|
||||
default:
|
||||
extension = res.getString("extension");
|
||||
message = new MessageFichier(TypeMessage.IMAGE, content, extension);
|
||||
}
|
||||
|
||||
} catch (MauvaisTypeMessageException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
}
|
||||
|
||||
if(res.getInt("id_conversation") == idConversationSelf) {
|
||||
message.setSender("Moi");
|
||||
}else{
|
||||
message.setSender(pseudoOther);
|
||||
}
|
||||
message.setDateMessage(res.getString("date"));
|
||||
if(content != null) {
|
||||
messages.add(message);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
|
||||
this.closeConnection();
|
||||
|
||||
return messages;
|
||||
}
|
||||
|
||||
|
||||
private void insertUser(String username) throws SQLException {
|
||||
String insertUserRequest = "INSERT INTO user (username) " + "VALUES (?);";
|
||||
|
||||
PreparedStatement prepStmt = this.connec.prepareStatement(insertUserRequest);
|
||||
prepStmt.setString(1, username);
|
||||
prepStmt.executeUpdate();
|
||||
}
|
||||
|
||||
|
||||
public int getIDUser(String username) throws SQLException {
|
||||
String getIDRequest = "SELECT id " + " FROM user" + " WHERE username = ? ;";
|
||||
|
||||
PreparedStatement prepStmt = this.connec.prepareStatement(getIDRequest);
|
||||
prepStmt.setString(1, username);
|
||||
ResultSet res = prepStmt.executeQuery();
|
||||
|
||||
if (res.next()) {
|
||||
return res.getInt("id");
|
||||
}
|
||||
return -1;
|
||||
|
||||
}
|
||||
|
||||
|
||||
private void insertConversation(int idSender, int idReceiver) throws SQLException {
|
||||
String insertConversationRequest = "INSERT INTO conversation (id_emetteur, id_recepteur, iv_conversation) " + "VALUES "
|
||||
+ "(?, ?, ?),"
|
||||
+ "(?, ?, ?);";
|
||||
|
||||
byte[] ivConversation = SQLiteEncprytion.generateIv().getIV();
|
||||
|
||||
PreparedStatement prepStmt = this.connec.prepareStatement(insertConversationRequest);
|
||||
prepStmt.setInt(1, idSender);
|
||||
prepStmt.setInt(2, idReceiver);
|
||||
prepStmt.setBytes(3, ivConversation);
|
||||
prepStmt.setInt(4, idReceiver);
|
||||
prepStmt.setInt(5, idSender);
|
||||
prepStmt.setBytes(6, ivConversation);
|
||||
|
||||
prepStmt.executeUpdate();
|
||||
}
|
||||
|
||||
|
||||
private int getIDConversation(int idSender, int idReceiver) throws SQLException {
|
||||
String getIDRequest = "SELECT id_conversation " + "FROM conversation " + "WHERE id_emetteur = ? "
|
||||
+ "AND id_recepteur = ? ;";
|
||||
|
||||
PreparedStatement prepStmt = this.connec.prepareStatement(getIDRequest);
|
||||
prepStmt.setInt(1, idSender);
|
||||
prepStmt.setInt(2, idReceiver);
|
||||
ResultSet res = prepStmt.executeQuery();
|
||||
|
||||
if (res.next()) {
|
||||
return res.getInt("id_conversation");
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
private IvParameterSpec getIvConversation(int idConversation) throws SQLException {
|
||||
String getIvRequest = "SELECT iv_conversation " + "FROM conversation " + "WHERE id_conversation = ?;";
|
||||
|
||||
PreparedStatement prepStmt = this.connec.prepareStatement(getIvRequest);
|
||||
prepStmt.setInt(1, idConversation);
|
||||
ResultSet res = prepStmt.executeQuery();
|
||||
|
||||
if (res.next()) {
|
||||
return new IvParameterSpec(res.getBytes("iv_conversation"));
|
||||
}
|
||||
|
||||
return null;
|
||||
}
|
||||
|
||||
|
||||
|
||||
|
||||
private int getIDType(String label) throws SQLException {
|
||||
String getIDRequest = "SELECT id_type FROM type WHERE label = ?;";
|
||||
|
||||
PreparedStatement prepStmt = this.connec.prepareStatement(getIDRequest);
|
||||
prepStmt.setString(1, label);
|
||||
|
||||
ResultSet res = prepStmt.executeQuery();
|
||||
|
||||
if (res.next()) {
|
||||
return res.getInt("id_type");
|
||||
}
|
||||
return -1;
|
||||
}
|
||||
|
||||
|
||||
private String getType(int idType) throws SQLException {
|
||||
String getTypeRequest = "SELECT label FROM type WHERE id_type = ?;";
|
||||
|
||||
PreparedStatement prepStmt = this.connec.prepareStatement(getTypeRequest);
|
||||
prepStmt.setInt(1, idType);
|
||||
|
||||
ResultSet res = prepStmt.executeQuery();
|
||||
|
||||
if(res.next()) {
|
||||
return res.getString("label");
|
||||
}
|
||||
|
||||
return "";
|
||||
|
||||
}
|
||||
|
||||
|
||||
private byte[] stringToBytesContent(Message m, IvParameterSpec iv) throws InvalidKeyException, NoSuchAlgorithmException, NoSuchPaddingException, InvalidAlgorithmParameterException, IllegalBlockSizeException, BadPaddingException {
|
||||
String content;
|
||||
if (m.getTypeMessage() == TypeMessage.TEXTE) {
|
||||
MessageTexte messageTxt = (MessageTexte) m;
|
||||
content = messageTxt.getContenu();
|
||||
}else {
|
||||
MessageFichier messageFichier = (MessageFichier) m;
|
||||
content = messageFichier.getContenu();
|
||||
}
|
||||
byte[] encryptedContent = SQLiteEncprytion.encrypt(SQLiteEncprytion.encryptAlgorithm, content.getBytes(), this.dbDataKey, iv);
|
||||
return encryptedContent;
|
||||
|
||||
}
|
||||
|
||||
private String bytesToStringContent(byte[] encryptedContent, IvParameterSpec iv) throws InvalidKeyException, NoSuchAlgorithmException, NoSuchPaddingException, InvalidAlgorithmParameterException, IllegalBlockSizeException, BadPaddingException {
|
||||
return SQLiteEncprytion.decryptString(SQLiteEncprytion.encryptAlgorithm, encryptedContent, this.dbDataKey, iv);
|
||||
}
|
||||
|
||||
|
||||
private String processMessageType(Message m) {
|
||||
switch (m.getTypeMessage()) {
|
||||
case TEXTE: return "text";
|
||||
case FICHIER: return "file";
|
||||
case IMAGE: return "image";
|
||||
default: return "";
|
||||
}
|
||||
}
|
||||
|
||||
private String processExtension(Message m) {
|
||||
if(m.getTypeMessage() == TypeMessage.TEXTE) {
|
||||
return null;
|
||||
}else {
|
||||
MessageFichier mFile = (MessageFichier) m;
|
||||
return mFile.getExtension();
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
private int insertMessage(int idConversation, Message m, IvParameterSpec iv) throws SQLException {
|
||||
|
||||
|
||||
String dateMessage = m.getDateMessage();
|
||||
String extension = this.processExtension(m);
|
||||
String type = this.processMessageType(m);
|
||||
int idType = this.getIDType(type);
|
||||
|
||||
byte[] content = null;
|
||||
|
||||
try {
|
||||
content = this.stringToBytesContent(m, iv);
|
||||
} catch (InvalidKeyException | NoSuchAlgorithmException | NoSuchPaddingException
|
||||
| InvalidAlgorithmParameterException | IllegalBlockSizeException | BadPaddingException e) {
|
||||
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
String insertMessageRequest = "INSERT INTO message(id_conversation, id_type, content, date, extension) "
|
||||
+ "VALUES (?, ?, ?, ?, ?);";
|
||||
|
||||
PreparedStatement prepStmt = this.connec.prepareStatement(insertMessageRequest);
|
||||
prepStmt.setInt(1, idConversation);
|
||||
prepStmt.setInt(2, idType);
|
||||
prepStmt.setBytes(3, content);
|
||||
prepStmt.setString(4, dateMessage);
|
||||
prepStmt.setString(5, extension);
|
||||
|
||||
int nbRows = prepStmt.executeUpdate();
|
||||
|
||||
return nbRows;
|
||||
}
|
||||
|
||||
public void createNewUserEncrypt(String username, String password) {
|
||||
|
||||
|
||||
String algo = SQLiteEncprytion.encryptAlgorithm;
|
||||
|
||||
KeyGenerator keyGen = null;
|
||||
try {
|
||||
keyGen = KeyGenerator.getInstance("AES");
|
||||
keyGen.init(256);
|
||||
} catch (NoSuchAlgorithmException e) {
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
byte[] passwordSalt = SQLiteEncprytion.getNextSalt();
|
||||
byte[] dbDataKeySalt = SQLiteEncprytion.getNextSalt();
|
||||
|
||||
SecretKey dbDataKey = keyGen.generateKey();
|
||||
|
||||
SecretKey dbDataEncryptKey = SQLiteEncprytion.getKey(password.toCharArray(), dbDataKeySalt);
|
||||
IvParameterSpec ivDbDataKey = SQLiteEncprytion.generateIv();
|
||||
|
||||
byte[] passwordHash = SQLiteEncprytion.hash(password.toCharArray(), passwordSalt);
|
||||
|
||||
byte[] dbDataKeyEncrypted = null;
|
||||
byte[] encryptedPasswordHash = null;
|
||||
|
||||
try {
|
||||
dbDataKeyEncrypted = SQLiteEncprytion.encrypt(
|
||||
algo, SQLiteEncprytion.keyToByte(dbDataKey), dbDataEncryptKey, ivDbDataKey);
|
||||
encryptedPasswordHash = SQLiteEncprytion.encrypt(
|
||||
algo, passwordHash , dbDataKey, ivDbDataKey);
|
||||
|
||||
} catch (InvalidKeyException | NoSuchAlgorithmException | NoSuchPaddingException
|
||||
| InvalidAlgorithmParameterException | IllegalBlockSizeException | BadPaddingException e) {
|
||||
// TODO Auto-generated catch block
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
this.openConnection();
|
||||
|
||||
String createUserRequest = "INSERT INTO user(username, pwd_salt, db_datakey_salt, encrypted_pwd_hashsalt, encrypted_db_datakey, iv_datakey) "
|
||||
+ "VALUES (?, ?, ?, ?, ?, ?); ";
|
||||
|
||||
PreparedStatement prepStmt = null;
|
||||
try {
|
||||
prepStmt = this.connec.prepareStatement(createUserRequest);
|
||||
prepStmt.setString(1, username);
|
||||
prepStmt.setBytes(2, passwordSalt);
|
||||
prepStmt.setBytes(3, dbDataKeySalt);
|
||||
prepStmt.setBytes(4, encryptedPasswordHash);
|
||||
prepStmt.setBytes(5, dbDataKeyEncrypted);
|
||||
prepStmt.setBytes(6, ivDbDataKey.getIV());
|
||||
|
||||
} catch (SQLException e) {
|
||||
// TODO Auto-generated catch block
|
||||
e.printStackTrace();
|
||||
}
|
||||
|
||||
try {
|
||||
prepStmt.executeUpdate();
|
||||
System.out.println("Utilisateur crée");
|
||||
} catch (SQLException e) {
|
||||
System.out.println("Nom d'utilisateur déjà pris");
|
||||
}
|
||||
|
||||
this.closeConnection();
|
||||
|
||||
}
|
||||
|
||||
/**
|
||||
*
|
||||
* @param username
|
||||
* @param password
|
||||
* @return -1 if user do not exists,
|
||||
* 0 if password incorrect,
|
||||
* 1 if password correct
|
||||
* @throws SQLException
|
||||
*/
|
||||
public int checkPwd(String username, char[] password) throws SQLException {
|
||||
|
||||
this.openConnection();
|
||||
|
||||
String selectUserDataRequest = "SELECT pwd_salt, db_datakey_salt, encrypted_pwd_hashsalt, encrypted_db_datakey, iv_datakey "
|
||||
+ "FROM user "
|
||||
+ "WHERE username = ?;";
|
||||
PreparedStatement prepStmt;
|
||||
prepStmt = this.connec.prepareStatement(selectUserDataRequest);
|
||||
prepStmt.setString(1, username);
|
||||
|
||||
ResultSet res = prepStmt.executeQuery();
|
||||
if(!res.next()) {
|
||||
return -1;
|
||||
}
|
||||
|
||||
byte[] passwordSalt = res.getBytes("pwd_salt");
|
||||
byte[] dbDataKeySalt = res.getBytes("db_datakey_salt");
|
||||
|
||||
SecretKey dbDataEncryptKey = SQLiteEncprytion.getKey(password, dbDataKeySalt);
|
||||
IvParameterSpec iv = new IvParameterSpec(res.getBytes("iv_datakey"));
|
||||
|
||||
byte[] encryptedDbDataKey = res.getBytes("encrypted_db_datakey");
|
||||
|
||||
|
||||
SecretKey dbDataKey = null;
|
||||
try {
|
||||
dbDataKey = SQLiteEncprytion.byteToKey(
|
||||
SQLiteEncprytion.decryptByte(SQLiteEncprytion.encryptAlgorithm, encryptedDbDataKey, dbDataEncryptKey, iv)
|
||||
);
|
||||
} catch (InvalidKeyException | NoSuchAlgorithmException | NoSuchPaddingException
|
||||
| InvalidAlgorithmParameterException | IllegalBlockSizeException | BadPaddingException e) {
|
||||
//System.out.println("Problème déchiffrement clé db");
|
||||
}
|
||||
|
||||
this.dbDataKey = dbDataKey;
|
||||
|
||||
byte[] encryptedPasswordHash = res.getBytes("encrypted_pwd_hashsalt");
|
||||
|
||||
|
||||
byte[] passwordHash = SQLiteEncprytion.hash(password, passwordSalt);
|
||||
|
||||
this.closeConnection();
|
||||
|
||||
boolean checkHash = this.checkHashPwd(passwordHash ,encryptedPasswordHash, dbDataKey, iv);
|
||||
if(checkHash) {
|
||||
return 1;
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
||||
private boolean checkHashPwd(byte[] passwordHash, byte[] encryptedPasswordHash, SecretKey dbDataKey, IvParameterSpec iv) {
|
||||
|
||||
byte[] expectedHash = "".getBytes();
|
||||
try {
|
||||
expectedHash = SQLiteEncprytion.decryptByte(SQLiteEncprytion.encryptAlgorithm, encryptedPasswordHash, dbDataKey, iv);
|
||||
} catch (InvalidKeyException | NoSuchAlgorithmException | NoSuchPaddingException
|
||||
| InvalidAlgorithmParameterException | IllegalBlockSizeException | BadPaddingException e) {
|
||||
}
|
||||
|
||||
|
||||
|
||||
if (passwordHash.length != expectedHash.length) return false;
|
||||
for (int i = 0; i < passwordHash.length; i++) {
|
||||
if (passwordHash[i] != expectedHash[i]) return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
|
||||
public static void main(String[] args) {
|
||||
String[] hardcodedNames = {"Olivia","Liam","Benjamin","Sophia","Charlotte","Noah","Elijah","Isabella",
|
||||
"Oliver","Emma","William","Amelia","Evelyn","James","Mia","Ava","Lucas","Mason","Ethan","Harper"};
|
||||
|
||||
String pwdPrefix = "aze1$";
|
||||
|
||||
SQLiteManager sqlManager = new SQLiteManager(0);
|
||||
|
||||
for(int i=0; i<hardcodedNames.length; i++) {
|
||||
sqlManager.createNewUserEncrypt(hardcodedNames[i]+i, pwdPrefix+hardcodedNames[i].charAt(0)+i);
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
}
|
|
@ -21,7 +21,6 @@ import messages.*;
|
|||
*/
|
||||
@WebServlet("/ServletPresence")
|
||||
|
||||
//Faire un publish (get) séparé en utilisant les cookies pour stocker les modifications : pose problème au niveau de la synchro des pseudos
|
||||
|
||||
public class ServletPresence extends HttpServlet implements Observer {
|
||||
private static final long serialVersionUID = 1L;
|
||||
|
@ -135,7 +134,7 @@ public class ServletPresence extends HttpServlet implements Observer {
|
|||
out.println( "</HTML>" );
|
||||
}
|
||||
|
||||
//Affiche un message d'erreur en cas de requête invalide
|
||||
//Affiche un message d'erreur en cas d'id inconnue
|
||||
private void printErrorUnkwownUser(PrintWriter out) {
|
||||
out.println( "<HTML>" );
|
||||
out.println( "<HEAD>");
|
||||
|
|
Loading…
Reference in a new issue