Java tutorial
/* Copyright 2015 Pablo Arrighi, Sarah Boukris, Mehdi Chtiwi, Michael Dubuis, Kevin Perrot, Julien Prudhomme. This file is part of SXP. SXP is free software: you can redistribute it and/or modify it under the terms of the GNU Lesser General Public License as published by the Free Software Foundation, version 3. SXP 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 Lesser General Public License for more details. You should have received a copy of the GNU Lesser General Public License along with SXP. If not, see <http://www.gnu.org/licenses/>. */ package model.data.contrat; import java.util.ArrayList; import java.util.HashMap; import java.util.Map.Entry; import model.advertisement.AbstractAdvertisement; import model.data.item.Item; import model.data.user.User; import org.jdom2.Element; import util.Printer; import util.StringToElement; import util.secure.ElGamalSign; import util.secure.AVProtocol.Delta; /** * This class can be instantiated for contains an agreement. * This class extends AbstractAdvertisement and can be used like an advertisement. * @author Michael Dubuis */ public class Contrat extends AbstractAdvertisement { private static final String[] stringState = { "draft", "waiting", "signed", "to sign" }; private String title; // Title of deal private int state; // State of deal (draft at start) private ArrayList<String> signatories; // All participants publicKey private ArrayList<Item> items; // All items of this deal private HashMap<String, String> rules; // All exchange rule private ArrayList<Clause> clauses; // all clauses private HashMap<String, Delta> proofs; // All proof of signature (null at start) private HashMap<String, ElGamalSign> signatures; // All signature of participants ///////////////////////////////////////////////// CONSTRUCTORS \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ /** * Constructor for new Deal * @param title * @param user */ public Contrat(String title, User user) { super(); setState(0); setTitle(title); setKeys(user.getKeys()); //TODO verify ! addSignatory(user); } public Contrat(String XML) { super(XML); } public Contrat(Element i) { super(i); } @SuppressWarnings("rawtypes") public Contrat(net.jxta.document.Element e) { super(e); } //////////////////////////////////////////////////// GETTERS \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ /** * For 2 party protocol * @param curUserPublicKey * @return */ public String getOtherSignatorie(String curUserPublicKey) { for (String s : signatories) { if (!s.equals(curUserPublicKey)) { return s; } } return null; } public String getTitle() { return title; } public ArrayList<String> getSignatories() { return signatories; } public ArrayList<Item> getItems() { return items; } public String getRecipientOf(Item item) { String itemKey = item.getItemKey(); return rules.containsKey(itemKey) ? rules.get(itemKey) : null; } public ArrayList<Item> getItemsReceivedBy(String publicKey) { ArrayList<Item> is = new ArrayList<Item>(); for (Item item : items) { String itemKey = item.getItemKey(); if (rules.containsKey(itemKey) && rules.get(itemKey).equals(publicKey)) is.add(item); } return is; } public String getStateStringFormat() { if (state < 0 || state > stringState.length - 1) { Printer.printError(this, "getStateStringFormat", "Unknown state !"); return "unknown state"; } return stringState[state]; } public int getState() { if (state < 0 || state >= stringState.length) { Printer.printError(this, "getState", "Unknown state !"); return -1; } return state; } public boolean isDraft() { return state == 0; } public boolean isWaiting() { return state == 1; } public boolean isSigned() { return state == 2; } public boolean isToSign() { return state == 3; } public ArrayList<Clause> getClauses() { return clauses; } public Clause getClause(String id) { for (Clause c : clauses) { if (c.getId().equals(id)) return c; } return null; } ///////////////////////////////////////////////// STATIC GETTERS \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ public static String[] getPossiblesStates() { return stringState; } public static String getPossibleState(int i) { if (i < 0 || i > stringState.length - 1) { Printer.printError(Contrat.class, "getPossibleState", "Unknown state !"); return "unknown state"; } return stringState[i]; } //////////////////////////////////////////////////// SETTERS \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ public void setTitle(String title) { if (!isDraft()) { Printer.printError(this, "setTitle", "This Deal isn't a draft"); return; } this.title = title; } private void setState(int state) { if (state < 0 || state >= stringState.length) { Printer.printError(this, "setState", "Unknown state !"); return; } this.state = state; } public void draft() { this.state = 0; // TODO Maybe delete proofs received } public boolean addSignatory(String publicKey) { if (!isDraft()) return Printer.printError(this, "addSignatory", "This Deal isn't a draft"); if (publicKey == null || publicKey.isEmpty()) return Printer.printError(this, "addSignatory", "publicKey Empty !"); if (signatories.contains(publicKey)) return false; return signatories.add(publicKey); } public boolean addSignatory(User user) { if (!isDraft()) return Printer.printError(this, "addSignatory", "This Deal isn't a draft"); if (user == null) return Printer.printError(this, "addSignatory", "User empty"); if (user.getKeys() == null || user.getKeys().getPublicKey() == null) return Printer.printError(this, "addSignatory", "User have to got PublicKey !"); return addSignatory(user.getKeys().getPublicKey().toString(16)); } public boolean addItem(Item item) { if (!isDraft()) return Printer.printError(this, "addItem", "This Deal isn't a draft"); if (item == null) return Printer.printError(this, "addItem", "Item empty"); if (item.getOwner() == null || item.getOwner().isEmpty()) return Printer.printError(this, "addItem", "Item " + item.getTitle() + " doesn't have Owner"); String owner = item.getOwner(); if (!signatories.contains(owner)) return signatories.add(owner) && items.add(item); if (items.contains(item)) return false; return items.add(item); } public boolean addTransferRule(String itemKey, String publicKey) { if (!isDraft()) return Printer.printError(this, "addTransferRule", "This Deal isn't a draft"); if (itemKey == null || itemKey.isEmpty()) return Printer.printError(this, "addTransferRule", "itemKey is null or empty"); if (publicKey == null || publicKey.isEmpty()) return Printer.printError(this, "", "publicKey is null or empty"); if (!signatories.contains(publicKey)) if (!addSignatory(publicKey)) return Printer.printError(this, "addTransferRule", "Impossible to add publicKey"); boolean ok = false; for (Item item : items) if (item.getItemKey().equals(itemKey)) { ok = true; } if (!ok) return Printer.printError(this, "addTransferRule", "item not found !"); if (rules.containsKey(itemKey)) { Printer.printInfo(this, "addTransferRule", "Rule for Item deleted"); rules.remove(itemKey); } rules.put(itemKey, publicKey); return true; } public boolean addTransferRule(Item item, String publicKey) { if (!isDraft()) return Printer.printError(this, "addTransferRule", "This Deal isn't a draft"); if (item == null || publicKey == null || publicKey.isEmpty()) return Printer.printError(this, "addTransferRule", "Item and PublicKey haven't to be empty"); if (!signatories.contains(publicKey)) if (!addSignatory(publicKey)) return Printer.printError(this, "addTransferRule", "Impossible to add publicKey"); if (!items.contains(item)) if (!addItem(item)) return Printer.printError(this, "addTransferRule", "Impossible to add Item"); String itemKey = item.getItemKey(); if (rules.containsKey(itemKey)) { Printer.printInfo(this, "addTransferRule", "Rule for Item " + item.getTitle() + " deleted"); rules.remove(itemKey); } rules.put(itemKey, publicKey); return true; } public boolean addTransferRule(Item item, User user) { if (!isDraft()) return Printer.printError(this, "addTransferRule", "This Deal isn't a draft"); if (user == null) return Printer.printError(this, "addTransferRule", "User empty"); if (user.getKeys() == null || user.getKeys().getPublicKey() == null) return Printer.printError(this, "addTransferRule", "User haven't publicKey"); return addTransferRule(item, user.getKeys().getPublicKey().toString(16)); } public boolean addClaus(Clause layout) { if (layout == null) return Printer.printError(this, "addLayout", "Layout empty"); return clauses.add(layout); } //////////////////////////////////////////////////// REMOVER \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ public boolean removeSignatory(String publicKey) { if (publicKey == null || publicKey.isEmpty()) return Printer.printError(this, "removeSignatory", "publicKey empty"); for (Item item : items) { if (item.getOwner().equals(publicKey)) { String itemKey = item.getItemKey(); rules.remove(itemKey); items.remove(item); } } for (Entry<String, String> entry : rules.entrySet()) { String itemKey = entry.getKey(); if (rules.get(itemKey).equals(publicKey)) rules.remove(itemKey); } signatories.remove(publicKey); return true; } public boolean removeSignatory(User user) { if (user == null) return Printer.printError(this, "removeSignatory", "User Empty"); if (user.getKeys() == null || user.getKeys().getPublicKey() == null) return Printer.printError(this, "removeSignatory", "User have to got a public Key"); String publicKey = user.getKeys().getPublicKey().toString(16); return removeSignatory(publicKey); } public boolean removeItem(Item item) { if (item == null) return Printer.printError(this, "removeItem", "Item empty"); if (item.getOwner() == null || item.getOwner().isEmpty()) return Printer.printError(this, "removeItem", "Item haven't owner"); if (item.getTitle() == null || item.getTitle().isEmpty()) return Printer.printError(this, "removeItem", "Item haven't title"); String itemKey = item.getItemKey(); rules.put(itemKey, ""); return items.remove(item) && rules.remove(itemKey) != null; } public boolean removeItem(String itemKey) { if (itemKey == null || itemKey.isEmpty()) return Printer.printError(this, "removeItem", "ItemKey null or empty"); if (rules.containsKey(itemKey)) rules.remove(itemKey); for (Item item : items) { if (item.getItemKey().equals(itemKey)) { rules.put(itemKey, ""); return items.remove(item) && rules.remove(itemKey) != null; } } return false; } public boolean removeRule(Item item, String publicKey) { if (item == null) return Printer.printError(this, "removeRule", "Item empty"); if (item.getOwner() == null || item.getOwner().isEmpty()) return Printer.printError(this, "removeRule", "Item haven't owner"); if (item.getTitle() == null || item.getTitle().isEmpty()) return Printer.printError(this, "removeRule", "Item haven't title"); if (publicKey == null || publicKey.isEmpty()) return Printer.printError(this, "removeRule", "publicKey null or empty"); String itemKey = item.getItemKey(); return rules.remove(itemKey) != null; } public boolean removeRule(Item item, User user) { if (user == null) return Printer.printError(this, "removeRule", "User empty"); if (user.getKeys() == null || user.getKeys().getPublicKey() == null) return Printer.printError(this, "removeRule", "User haven't public Key"); String publicKey = user.getKeys().getPublicKey().toString(16); return removeRule(item, publicKey); } public boolean removeClaus(Clause clause) { if (clause == null) return Printer.printError(this, "removeClaus", "Claus empty"); return clauses.remove(clause); } public boolean removeClause(String id) { if (id == null || id.isEmpty()) return Printer.printError(this, "removeClause", "id is null or empty"); for (Clause c : clauses) { if (c.getId().equals(id)) return clauses.remove(c); } return false; } //////////////////////////////////////////////////// OTHERS \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ @Override public Contrat clone() { return new Contrat(this.toString()); } //////////////////////////////////////////////////// PRINTER \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ public String toPrint() { StringBuffer s = new StringBuffer(); s.append("Signatories (" + signatories.size() + ") :\n"); for (String signatorie : signatories) { s.append("\t- " + signatorie + "\n"); } s.append("Transfer :\n"); for (Item item : items) { s.append("\t- " + item.getTitle() + " (" + item.getItemKey() + ")" + "\n\t\tfrom "); s.append(item.getOwner()); s.append(" \n\t\tto "); s.append(rules.get(item.getItemKey())); s.append("\n"); } return s.toString(); } /////////////////////////////////////////////////// AVPROTOCOL \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ /*public void lauchProtocol(AsymKeysImpl dealerKeys){ // TODO thread maybe Dealer dealer = new Dealer(dealerKeys); TTP ttp = new TTP(VARIABLES.AVProtocolN, VARIABLES.AVProtocolK); ArrayList<BigInteger> aj = new ArrayList<BigInteger>(); for (int i = 0 ; i < VARIABLES.AVProtocolK; i++) { // Add to aj new random BigInteger } for (int i = 0; i < VARIABLES.AVProtocolN; i++) { // Search TTP in network; ParticipantEx p = new ParticipantEx(dealerKeys, i, null); // TODO change null with random BigInteger p.setMi(dealer.createMi(new BigInteger(this.toString()), p, aj).toByteArray()); ttp.addParticipant(p); } this.state = 1; }*/ ////////////////////////////////////////////////////// XML \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ private String signatoriesXML() { StringBuffer s = new StringBuffer(); for (String string : signatories) { s.append("<signatory>"); s.append(string); s.append("</signatory>"); } return s.toString(); } private String itemsXML() { StringBuffer s = new StringBuffer(); for (Item i : items) { s.append(i); } return s.toString(); } private String rulesXML() { StringBuffer s = new StringBuffer(); for (Entry<String, String> entry : rules.entrySet()) { String itemKey = entry.getKey(); String rule = entry.getKey(); s.append("<rule>"); s.append("<itemKey>"); s.append(itemKey); s.append("</itemKey>"); s.append("<receiver>"); s.append(rule); s.append("</receiver>"); s.append("</rule>"); } return s.toString(); } private String clausesXML() { StringBuffer s = new StringBuffer(); for (Clause l : clauses) { s.append(l); } return s.toString(); } private void loadSignatories(Element e) { Element root = StringToElement.getElementFromString(e.getValue(), e.getName()); for (Element s : root.getChildren("signatory")) { addSignatory(s.getText()); } } private void loadItems(Element e) { Element root = StringToElement.getElementFromString(e.getValue(), e.getName()); for (Element i : root.getChildren()) { addItem(new Item(i)); } } private void loadRules(Element e) { Element root = StringToElement.getElementFromString(e.getValue(), e.getName()); for (Element r : root.getChildren("rule")) { Element itemKeyElement = r.getChild("itemKey"); Element receiverElement = r.getChild("receiver"); if (itemKeyElement == null || receiverElement == null) continue; String itemKey = itemKeyElement.getText(); String receiver = receiverElement.getText(); if (!rules.containsKey(itemKey)) rules.put(itemKey, receiver); } } private void loadClauses(Element e) { Element root = StringToElement.getElementFromString(e.getValue(), e.getName()); for (Element i : root.getChildren()) { addClaus(new Clause(i)); } } ///////////////////////////////////////////////// ADVERTISEMENT \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ @Override protected String getAdvertisementName() { return Contrat.class.getName(); } @Override protected void setKeys() { state = 0; signatories = new ArrayList<String>(); items = new ArrayList<Item>(); rules = new HashMap<String, String>(); clauses = new ArrayList<Clause>(); proofs = new HashMap<String, Delta>(); this.addKey("title", false, true); this.addKey("state", false, true); this.addKey("signatories", false, true); this.addKey("items", false, true); this.addKey("rules", false, true); this.addKey("clauses", false, true); } @Override protected void putValues() { this.addValue("title", getTitle()); this.addValue("state", String.valueOf(this.getState())); this.addValue("signatories", this.signatoriesXML()); this.addValue("items", this.itemsXML()); this.addValue("rules", this.rulesXML()); this.addValue("clauses", this.clausesXML()); } @Override protected boolean handleElement(Element e) { String val = e.getText(); switch (e.getName()) { case "title": setTitle(val); break; case "state": setState(Integer.parseInt(val)); break; case "signatories": loadSignatories(e); break; case "items": loadItems(e); break; case "rules": loadRules(e); break; case "clauses": loadClauses(e); break; default: return false; } return true; } @Override public String getSimpleName() { return getClass().getSimpleName(); } }