EncryptionUpdate.java

package client;

import remote.IDrawingUpdate;
import remote.IEncryptionUpdate;

import javax.crypto.*;
import javax.crypto.spec.SecretKeySpec;
import java.io.Serializable;
import java.rmi.RemoteException;
import java.rmi.server.UnicastRemoteObject;
import java.security.*;

public class EncryptionUpdate extends UnicastRemoteObject implements IEncryptionUpdate, Serializable {

    private PrivateKey priv;
    private PublicKey pub;
    private SecretKey sharedSecretKey;

    public PublicKey getPub() {
        return pub;
    }

    public void setSharedKey(byte[] encryptedSharedSecretKey){
        try {
            Cipher cipher = Cipher.getInstance("RSA", "SunJCE");
            cipher.init(Cipher.DECRYPT_MODE, priv);
            sharedSecretKey = new SecretKeySpec(cipher.doFinal(encryptedSharedSecretKey), "AES");
            System.out.println("Client holds this shared key: "+sharedSecretKey);
        } catch (NoSuchAlgorithmException e) {
            e.printStackTrace();
        } catch (NoSuchProviderException e) {
            e.printStackTrace();
        } catch (NoSuchPaddingException | InvalidKeyException e) {
            e.printStackTrace();
        } catch (BadPaddingException e) {
            e.printStackTrace();
        } catch (IllegalBlockSizeException e) {
            e.printStackTrace();
        }


    }

    EncryptionUpdate() throws RemoteException {
        try {
            KeyPairGenerator keyGen = KeyPairGenerator.getInstance("RSA", "SunJSSE");
            keyGen.initialize(2048);
            KeyPair pair = keyGen.generateKeyPair();
            this.priv = pair.getPrivate();
            this.pub = pair.getPublic();

        }
        catch (Exception e){
            e.printStackTrace();
        }

    }
}