// The MIT License (MIT)

//

// Copyright (c) 2019 Robert C. Seacord

//

// Permission is hereby granted, free of charge, to any person obtaining a copy

// of this software and associated documentation files (the "Software"), to deal

// in the Software without restriction, including without limitation the rights

// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell

// copies of the Software, and to permit persons to whom the Software is

// furnished to do so, subject to the following conditions:

//

// The above copyright notice and this permission notice shall be included in all

// copies or substantial portions of the Software.

//

// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR

// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,

// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE

// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER

// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,

// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE

// SOFTWARE.

package ser02j;

import java.io.IOException;

import java.security.InvalidKeyException;

import java.security.Key;

import java.security.KeyPair;

import java.security.KeyPairGenerator;

import java.security.NoSuchAlgorithmException;

import java.security.SecureRandom;

import java.security.Signature;

import java.security.SignatureException;

import java.security.SignedObject;

import javax.crypto.BadPaddingException;

import javax.crypto.Cipher;

import javax.crypto.IllegalBlockSizeException;

import javax.crypto.KeyGenerator;

import javax.crypto.NoSuchPaddingException;

import javax.crypto.SealedObject;

import serial.Serial;

class MapSerializer {

private static SerializableMap<String, String> buildMap() {

final SerializableMap<String, String> map = new SerializableMap<>();

map.setData("John Doe", "012-34-5678");

map.setData("Jane Doe", "987-65-4321");

return map;

}

private static void InspectMap(final SerializableMap<String, String> map) {

System.out.println("John Doe CC#: " + map.getData("John Doe"));

System.out.println("Jane Doe CC#: " + map.getData("Jane Doe"));

}

@SuppressWarnings("unchecked")

public static void main(final String[] args) {

// Build map

SerializableMap<String, String> map = buildMap();

try {

// Generate signing public/private key pair & sign map

final KeyPairGenerator kpg = KeyPairGenerator.getInstance("DSA");

final KeyPair kp = kpg.generateKeyPair();

final Signature sig = Signature.getInstance("SHA1withDSA");

SignedObject signedMap = new SignedObject(map, kp.getPrivate(), sig);

// Generate sealing key & seal map

KeyGenerator generator;

generator = KeyGenerator.getInstance("AES");

generator.init(new SecureRandom());

final Key key = generator.generateKey();

Cipher cipher = Cipher.getInstance("AES");

cipher.init(Cipher.ENCRYPT_MODE, key);

SealedObject sealedMap = new SealedObject(signedMap, cipher);

sealedMap = (SealedObject)Serial.deserialize(Serial.serialize(sealedMap));

// Unseal map

cipher = Cipher.getInstance("AES");

cipher.init(Cipher.DECRYPT_MODE, key);

signedMap = (SignedObject) sealedMap.getObject(cipher);

// Verify signature and retrieve map

if (!signedMap.verify(kp.getPublic(), sig)) {

System.err.println("Map failed verification");

}

map = (SerializableMap<String, String>) signedMap.getObject();

} catch (InvalidKeyException | NoSuchAlgorithmException | ClassNotFoundException | IOException | BadPaddingException

| IllegalBlockSizeException | NoSuchPaddingException | SignatureException e) {

e.printStackTrace();

}

// Inspect map

InspectMap(map);

}

}