Newer
Older
/* Copyright or (C) or Copr. GET / ENST, Telecom-Paris, Ludovic Apvrille
*
* ludovic.apvrille AT enst.fr
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
*
* This software is a computer program whose purpose is to allow the
* edition of TURTLE analysis, design and deployment diagrams, to
* allow the generation of RT-LOTOS or Java code from this diagram,
* and at last to allow the analysis of formal validation traces
* obtained from external tools, e.g. RTL from LAAS-CNRS and CADP
* from INRIA Rhone-Alpes.
*
* This software is governed by the CeCILL license under French law and
* abiding by the rules of distribution of free software. You can use,
* modify and/ or redistribute the software under the terms of the CeCILL
* license as circulated by CEA, CNRS and INRIA at the following URL
* "http://www.cecill.info".
*
* As a counterpart to the access to the source code and rights to copy,
* modify and redistribute granted by the license, users are provided only
* with a limited warranty and the software's author, the holder of the
* economic rights, and the successive licensors have only limited
* liability.
*
* In this respect, the user's attention is drawn to the risks associated
* with loading, using, modifying and/or developing or reproducing the
* software by the user in light of its specific status of free software,
* that may mean that it is complicated to manipulate, and that also
* therefore means that it is reserved for developers and experienced
* professionals having in-depth computer knowledge. Users are therefore
* encouraged to load and test the software's suitability as regards their
* requirements in conditions enabling the security of their systems and/or
* data to be ensured and, more generally, to use and operate it in the
* same conditions as regards security.
*
* The fact that you are presently reading this means that you have had
* knowledge of the CeCILL license and that you accept its terms.
*/
import ui.atd.*;
//import translator.*;
/**
* Class AvatarTreePanelTranslator
* Creation: 13/04/2015
* @author Ludovic APVRILLE
*/
public class AttackTreePanelTranslator {
protected AttackTreePanel atp;
Florian Lugou
committed
protected LinkedList<CheckingError> checkingErrors, warnings;
protected CorrespondanceTGElement listE; // usual list
//protected CorrespondanceTGElement listB; // list for particular element -> first element of group of blocks
protected LinkedList <TDiagramPanel> panels;
public AttackTreePanelTranslator(AttackTreePanel _atp) {
atp = _atp;
reinit();
}
public void reinit() {
Florian Lugou
committed
checkingErrors = new LinkedList<CheckingError> ();
warnings = new LinkedList<CheckingError> ();
listE = new CorrespondanceTGElement();
panels = new LinkedList <TDiagramPanel>();
}
Florian Lugou
committed
public LinkedList<CheckingError> getCheckingErrors() {
return checkingErrors;
}
Florian Lugou
committed
public LinkedList<CheckingError> getWarnings() {
return warnings;
}
public CorrespondanceTGElement getCorrespondanceTGElement() {
return listE;
}
public AttackTree translateToAttackTreeDataStructure() {
at = new AttackTree("AttackTree", atp);
for(TDiagramPanel panel: atp.panels) {
if (panel instanceof AttackTreeDiagramPanel) {
translate((AttackTreeDiagramPanel)panel);
}
}
//TraceManager.addDev("AT=" + at.toString());
return at;
}
public void translate(AttackTreeDiagramPanel atdp) {
LinkedList<TGComponent> allComponents = atdp.getAllComponentList();
TGComponent father;
//Create attacks, nodes
for(TGComponent comp: allComponents) {
if (comp instanceof ATDAttack) {
ATDAttack atdatt = (ATDAttack)comp;
Attack att;
String value = atdatt.getValue();
father = atdatt.getFather();
if ((father != null) && (father instanceof ATDBlock)) {
value = ((ATDBlock)father).getNodeName() + "__" + value;
}
att = new Attack(value, atdatt);
att.setRoot(atdatt.isRootAttack());
att.setEnabled(atdatt.isEnabled());
at.addAttack(att);
listE.addCor(att, comp);
}
if (comp instanceof ATDConstraint) {
ATDConstraint cons = (ATDConstraint)comp;
nodeID ++;
//OR
if (cons.isOR()) {
ORNode ornode = new ORNode("OR__" + nodeID, cons);
at.addNode(ornode);
listE.addCor(ornode, comp);
//AND
} else if (cons.isAND()) {
ANDNode andnode = new ANDNode("AND__" + nodeID, cons);
at.addNode(andnode);
listE.addCor(andnode, comp);
//XOR
} else if (cons.isXOR()) {
XORNode xornode = new XORNode("XOR__" + nodeID, cons);
at.addNode(xornode);
listE.addCor(xornode, comp);
//SEQUENCE
} else if (cons.isSequence()) {
SequenceNode seqnode = new SequenceNode("SEQUENCE__" + nodeID, cons);
at.addNode(seqnode);
listE.addCor(seqnode, comp);
//BEFORE
} else if (cons.isBefore()) {
String eq = cons.getEquation();
int time;
try {
time = Integer.decode(eq).intValue();
BeforeNode befnode = new BeforeNode("BEFORE__" + nodeID, cons, time);
at.addNode(befnode);
listE.addCor(befnode, comp);
} catch (Exception e) {
UICheckingError ce = new UICheckingError(CheckingError.STRUCTURE_ERROR, "Invalid time in before node");
ce.setTGComponent(comp);
ce.setTDiagramPanel(atdp);
addCheckingError(ce);
}
//AFTER
} else if (cons.isAfter()) {
String eq = cons.getEquation();
int time;
try {
time = Integer.decode(eq).intValue();
AfterNode aftnode = new AfterNode("AFTER__" + nodeID, cons, time);
at.addNode(aftnode);
listE.addCor(aftnode, comp);
UICheckingError ce = new UICheckingError(CheckingError.STRUCTURE_ERROR, "Invalid time in after node");
ce.setTGComponent(comp);
ce.setTDiagramPanel(atdp);
addCheckingError(ce);
}
UICheckingError ce = new UICheckingError(CheckingError.STRUCTURE_ERROR, "Invalid attack node");
ce.setTGComponent(comp);
ce.setTDiagramPanel(atdp);
addCheckingError(ce);
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
// Making connections between nodes&attacks
TGComponent tgc1, tgc2;
for(TGComponent comp: allComponents) {
if (comp instanceof ATDAttackConnector) {
ATDAttackConnector con = (ATDAttackConnector)(comp);
tgc1 = atdp.getComponentToWhichBelongs(con.getTGConnectingPointP1());
tgc2 = atdp.getComponentToWhichBelongs(con.getTGConnectingPointP2());
if ( ((tgc1 instanceof ATDAttack) || (tgc1 instanceof ATDConstraint)) &&
((tgc2 instanceof ATDAttack) || (tgc2 instanceof ATDConstraint)) ) {
try {
// We must transpose this into attack -> node or node -> attack
// Attack -> attack
if ((tgc1 instanceof ATDAttack) && (tgc2 instanceof ATDAttack)) {
// We link the two attacks with an "and" node
Attack at1 = (Attack)(listE.getObject(tgc1));
Attack at2 = (Attack)(listE.getObject(tgc2));
nodeID ++;
ANDNode andnode = new ANDNode("ANDBetweenAttacks__" + nodeID + "__" + at1.getName() + "__" + at2.getName(), tgc1);
at.addNode(andnode);
listE.addCor(andnode, comp);
at1.addDestinationNode(andnode);
at2.setOriginNode(andnode);
andnode.addInputAttack(at1, new Integer("0"));
andnode.setResultingAttack(at2);
// Attack -> node
} else if ((tgc1 instanceof ATDAttack) && (tgc2 instanceof ATDConstraint)) {
Attack at1 = (Attack)(listE.getObject(tgc1));
AttackNode node1 = (AttackNode)(listE.getObject(tgc2));
at1.addDestinationNode(node1);
String val = comp.getValue().trim();
if (val.length() == 0) {
val = "0";
}
node1.addInputAttack(at1, new Integer(val));
// Node -> attack
} else if ((tgc1 instanceof ATDConstraint) && (tgc2 instanceof ATDAttack)) {
Attack at1 = (Attack)(listE.getObject(tgc2));
AttackNode node1 = (AttackNode)(listE.getObject(tgc1));
at1.setOriginNode(node1);
if (node1.getResultingAttack() != null) {
// Already a resulting attack -> error
UICheckingError ce = new UICheckingError(CheckingError.STRUCTURE_ERROR, "Too many resulting attacks");
ce.setTGComponent(tgc1);
ce.setTDiagramPanel(atdp);
addCheckingError(ce);
} else {
node1.setResultingAttack(at1);
}
// Node -> Node
} else if ((tgc1 instanceof ATDConstraint) && (tgc2 instanceof ATDConstraint)) {
AttackNode node1 = (AttackNode)(listE.getObject(tgc1));
AttackNode node2 = (AttackNode)(listE.getObject(tgc2));
// Make fake attack
Attack att = new Attack("Attack__from_" + node1.getName() + "_to_" + node2.getName(), tgc1);
att.setRoot(false);
at.addAttack(att);
listE.addCor(att, comp);
att.setOriginNode(node1);
att.addDestinationNode(node2);
if (node1.getResultingAttack() != null) {
// Already a resulting attack -> error
UICheckingError ce = new UICheckingError(CheckingError.STRUCTURE_ERROR, "Too many resulting attacks");
ce.setTGComponent(tgc1);
ce.setTDiagramPanel(atdp);
addCheckingError(ce);
} else {
node1.setResultingAttack(att);
}
String val = comp.getValue().trim();
if (val.length() == 0) {
val = "0";
}
node2.addInputAttack(att, new Integer(val));
}
} catch (Exception e) {
UICheckingError ce = new UICheckingError(CheckingError.STRUCTURE_ERROR, "Badly formed connector");
ce.setTGComponent(comp);
ce.setTDiagramPanel(atdp);
addCheckingError(ce);
}
}
public AvatarSpecification generateAvatarSpec() {
AvatarSpecification as = new AvatarSpecification("spec from attack trees", atp);
// One block per attacknode to receive the attack
// One block per attack -> syncho
// One mast block with all channels declared at that level
AvatarBlock mainBlock = new AvatarBlock("MainBlock", as, null);
AvatarStartState ass = new AvatarStartState("StartStateOfMainBlock", null);
mainBlock.getStateMachine().setStartState(ass);
mainBlock.getStateMachine().addElement(ass);
// Declare all attacks
declareAllAttacks(as, mainBlock);
// Make blocks for nodes
makeAttackNodeBlocks(as, mainBlock);
private void declareAllAttacks(AvatarSpecification _as, AvatarBlock _main) {
AvatarRelation ar = new AvatarRelation("MainRelation", _main, _main, null);
ar.setAsynchronous(false);
ar.setPrivate(true);
ar.setBroadcast(false);
_as.addRelation(ar);
for(Attack attack: at.getAttacks()) {
avatartranslator.AvatarSignal makeAttack = new avatartranslator.AvatarSignal("make__" + attack.getName(), AvatarSignal.OUT, listE.getTG(attack));
_main.addSignal(makeAttack);
avatartranslator.AvatarSignal stopMakeAttack = new avatartranslator.AvatarSignal("makeStop__" + attack.getName(), AvatarSignal.IN, listE.getTG(attack));
_main.addSignal(stopMakeAttack);
avatartranslator.AvatarSignal acceptAttack = new avatartranslator.AvatarSignal("accept__" + attack.getName(), AvatarSignal.IN, listE.getTG(attack));
_main.addSignal(acceptAttack);
avatartranslator.AvatarSignal stopAcceptAttack = new avatartranslator.AvatarSignal("acceptStop__" + attack.getName(), AvatarSignal.OUT, listE.getTG(attack));
_main.addSignal(stopAcceptAttack);
ar.addSignals(makeAttack, acceptAttack);
ar.addSignals(stopMakeAttack, stopAcceptAttack);
// If attack is not leaf: add the intermediate action to activate the intermediate leaf
if (!attack.isLeaf()) {
avatartranslator.AvatarSignal nodeDone = new avatartranslator.AvatarSignal("nodeDone__" + attack.getName(), AvatarSignal.OUT, listE.getTG(attack));
_main.addSignal(nodeDone);
avatartranslator.AvatarSignal activateAttack = new avatartranslator.AvatarSignal("activate__" + attack.getName(), AvatarSignal.IN, listE.getTG(attack));
_main.addSignal(activateAttack);
ar.addSignals(nodeDone, activateAttack);
}
private void makeAttackBlocks(AvatarSpecification _as, AvatarBlock _main) {
for(Attack attack: at.getAttacks()) {
if (attack.isLeaf()) {
// Make the block
AvatarBlock ab = new AvatarBlock(attack.getName(), _as, listE.getTG(attack));
_as.addBlock(ab);
ab.setFather(_main);
avatartranslator.AvatarSignal sigAttack = _main.getAvatarSignalWithName("make__" + attack.getName());
avatartranslator.AvatarSignal stopAttack = _main.getAvatarSignalWithName("makeStop__" + attack.getName());
if ((sigAttack != null) && (stopAttack != null)) {
makeAttackBlockSMD(ab, sigAttack, stopAttack, attack.isEnabled(), listE.getTG(attack));
// Make the block
AvatarBlock ab = new AvatarBlock(attack.getName(), _as, listE.getTG(attack));
_as.addBlock(ab);
ab.setFather(_main);
avatartranslator.AvatarSignal sigAttack = _main.getAvatarSignalWithName("make__" + attack.getName());
avatartranslator.AvatarSignal stopAttack = _main.getAvatarSignalWithName("makeStop__" + attack.getName());
avatartranslator.AvatarSignal activateAttack = _main.getAvatarSignalWithName("activate__" + attack.getName());
makeIntermediateAttackBlockSMD(ab, sigAttack, stopAttack, activateAttack, attack.isEnabled(), listE.getTG(attack));
// Intermediate attack
}
private void makeAttackBlockSMD(AvatarBlock _ab, avatartranslator.AvatarSignal _sigAttack, avatartranslator.AvatarSignal _sigStop, boolean isEnabled, Object _ref) {
AvatarStateMachine asm = _ab.getStateMachine();
if (_ref1 instanceof TGComponent) {
isCheckable = ((TGComponent)(_ref1)).hasCheckableAccessibility();
}
AvatarStartState start = new AvatarStartState("start", _ref);
AvatarState mainState = new AvatarState("main", _ref, false);
AvatarState performedState = new AvatarState("main", _ref1, isCheckable);
performedState.setAsVerifiable(true);
AvatarState mainStop = new AvatarState("stop", _ref, false);
AvatarActionOnSignal getMake = new AvatarActionOnSignal("GettingAttack", _sigAttack, _ref1);
AvatarActionOnSignal getStop = new AvatarActionOnSignal("GettingStop", _sigStop, _ref);
asm.addElement(start);
asm.setStartState(start);
asm.addElement(mainState);
asm.addElement(performedState);
asm.addElement(getMake);
asm.addElement(getStop);
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at1", _ref);
asm.addElement(at);
start.addNext(at);
at.addNext(mainState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at2", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(getMake);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at3", _ref);
asm.addElement(at);
getMake.addNext(at);
at.addNext(performedState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "backToMain", _ref);
asm.addElement(at);
performedState.addNext(at);
at.addNext(mainState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at4", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(getStop);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at5", _ref);
asm.addElement(at);
getStop.addNext(at);
at.addNext(mainStop);
} else {
AvatarStartState start = new AvatarStartState("start", _ref);
AvatarState mainState = new AvatarState("main", _ref, false);
AvatarState mainStop = new AvatarState("stop", _ref, false);
AvatarActionOnSignal getStop = new AvatarActionOnSignal("GettingStop", _sigStop, _ref);
asm.addElement(start);
asm.setStartState(start);
asm.addElement(mainState);
asm.addElement(getStop);
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at1", _ref);
asm.addElement(at);
start.addNext(at);
at.addNext(mainState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at4", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(getStop);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at5", _ref);
asm.addElement(at);
getStop.addNext(at);
at.addNext(mainStop);
}
private void makeIntermediateAttackBlockSMD(AvatarBlock _ab, avatartranslator.AvatarSignal _sigAttack, avatartranslator.AvatarSignal _sigStop, avatartranslator.AvatarSignal _sigActivate, boolean isEnabled, Object _ref) {
Object _ref1 = _ref;
_ref = null;
AvatarStateMachine asm = _ab.getStateMachine();
if (_ref1 instanceof TGComponent) {
isCheckable = ((TGComponent)(_ref1)).hasCheckableAccessibility();
}
AvatarStartState start = new AvatarStartState("start", _ref);
AvatarState activateState = new AvatarState("activate", _ref, false);
AvatarState mainState = new AvatarState("main", _ref, false);
AvatarState activatedState = new AvatarState("activated", _ref1, isCheckable);
if (_ref1 instanceof ATDAttack) {
activatedState.setAsVerifiable(true);
}
AvatarState performedState = new AvatarState("performed", _ref, false);
AvatarState mainStop = new AvatarState("stop", _ref, false);
AvatarState stopBeforeActivate = new AvatarState("stopBeforeActivate", _ref, false);
AvatarActionOnSignal getMake = new AvatarActionOnSignal("GettingAttack", _sigAttack, _ref1);
AvatarActionOnSignal getStop = new AvatarActionOnSignal("GettingStop", _sigStop, _ref);
AvatarActionOnSignal getStopInitial = new AvatarActionOnSignal("GettingInitialStop", _sigStop, _ref);
AvatarActionOnSignal getActivate = new AvatarActionOnSignal("GettingActivate", _sigActivate, _ref1);
AvatarActionOnSignal getActivateAfterStop = new AvatarActionOnSignal("GettingActivateAfterStop", _sigActivate, _ref1);
asm.addElement(start);
asm.setStartState(start);
asm.addElement(activateState);
asm.addElement(activatedState);
asm.addElement(mainState);
asm.addElement(stopBeforeActivate);
asm.addElement(performedState);
asm.addElement(getMake);
asm.addElement(getStop);
asm.addElement(getStopInitial);
asm.addElement(getActivate);
asm.addElement(getActivateAfterStop);
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at1", _ref);
asm.addElement(at);
start.addNext(at);
at.addNext(activateState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at1_act", _ref);
asm.addElement(at);
activateState.addNext(at);
at.addNext(getActivate);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at1_performed", _ref);
asm.addElement(at);
getActivate.addNext(at);
at.addNext(activatedState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at2_main", _ref);
asm.addElement(at);
activatedState.addNext(at);
at.addNext(mainState);
at.setHidden(true);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at2", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(getMake);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at3", _ref);
asm.addElement(at);
getMake.addNext(at);
at.addNext(performedState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "backToMain", _ref);
asm.addElement(at);
performedState.addNext(at);
at.addNext(mainState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at4", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(getStop);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at5", _ref);
asm.addElement(at);
getStop.addNext(at);
at.addNext(mainStop);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at6", _ref);
asm.addElement(at);
activateState.addNext(at);
at.addNext(getStopInitial);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at7", _ref);
asm.addElement(at);
getStopInitial.addNext(at);
at.addNext(stopBeforeActivate);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at8", _ref);
asm.addElement(at);
stopBeforeActivate.addNext(at);
at.addNext(getActivateAfterStop);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at9", _ref);
asm.addElement(at);
getActivateAfterStop.addNext(at);
at.addNext(mainStop);
} else {
AvatarStartState start = new AvatarStartState("start", _ref);
AvatarState activateState = new AvatarState("activate", _ref, false);
AvatarState mainState = new AvatarState("main", _ref, false);
AvatarState activatedState = new AvatarState("main", _ref1, isCheckable);
AvatarState mainStop = new AvatarState("stop", _ref, false);
AvatarState stopBeforeActivate = new AvatarState("stopBeforeActivate", _ref, false);
AvatarActionOnSignal getStop = new AvatarActionOnSignal("GettingStop", _sigStop, _ref);
AvatarActionOnSignal getStopInitial = new AvatarActionOnSignal("GettingInitialStop", _sigStop, _ref);
AvatarActionOnSignal getActivate = new AvatarActionOnSignal("GettingActivate", _sigActivate, _ref1);
AvatarActionOnSignal getActivateAfterStop = new AvatarActionOnSignal("GettingActivateAfterStop", _sigActivate, _ref1);
asm.addElement(start);
asm.setStartState(start);
asm.addElement(activateState);
asm.addElement(activatedState);
asm.addElement(mainState);
asm.addElement(stopBeforeActivate);
asm.addElement(getStop);
asm.addElement(getStopInitial);
asm.addElement(getActivate);
asm.addElement(getActivateAfterStop);
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at1", _ref);
asm.addElement(at);
start.addNext(at);
at.addNext(activateState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at1_act", _ref);
asm.addElement(at);
activateState.addNext(at);
at.addNext(getActivate);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at1_performed", _ref);
asm.addElement(at);
getActivate.addNext(at);
at.addNext(activatedState);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at2_main", _ref);
asm.addElement(at);
activatedState.addNext(at);
at.addNext(mainState);
at.setHidden(true);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at4", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(getStop);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at5", _ref);
asm.addElement(at);
getStop.addNext(at);
at.addNext(mainStop);
// Stop before activate
Florian Lugou
committed
at = new AvatarTransition(_ab, "at6", _ref);
asm.addElement(at);
activateState.addNext(at);
at.addNext(getStopInitial);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at7", _ref);
asm.addElement(at);
getStopInitial.addNext(at);
at.addNext(stopBeforeActivate);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at8", _ref);
asm.addElement(at);
stopBeforeActivate.addNext(at);
at.addNext(getActivateAfterStop);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at9", _ref);
asm.addElement(at);
getActivateAfterStop.addNext(at);
at.addNext(mainStop);
}
}
private void makeAttackNodeBlocks(AvatarSpecification _as, AvatarBlock _main) {
Attack att;
for(AttackNode node: at.getAttackNodes()) {
if (node.isWellFormed()) {
// Make the block
AvatarBlock ab = new AvatarBlock(node.getName(), _as, listE.getTG(node));
_as.addBlock(ab);
ab.setFather(_main);
if (node instanceof ANDNode) {
makeANDNode(_as, _main, ab, (ANDNode)node, listE.getTG(node));
} else if (node instanceof ORNode) {
makeORNode(_as, _main, ab, (ORNode)node, listE.getTG(node));
} else if (node instanceof XORNode) {
makeXORNode(_as, _main, ab, (XORNode)node, listE.getTG(node));
} else if (node instanceof SequenceNode) {
makeSequenceNode(_as, _main, ab, (SequenceNode)node, listE.getTG(node));
} else if (node instanceof AfterNode) {
makeAfterNode(_as, _main, ab, (AfterNode)node, listE.getTG(node));
} else if (node instanceof BeforeNode) {
makeBeforeNode(_as, _main, ab, (BeforeNode)node, listE.getTG(node));
}
}
}
private void makeANDNode(AvatarSpecification _as, AvatarBlock _main, AvatarBlock _ab, ANDNode _node, Object _ref) {
AvatarStateMachine asm = _ab.getStateMachine();
// Basic machine
AvatarStartState start = new AvatarStartState("start", _ref);
AvatarState mainState = new AvatarState("main", _ref, false);
AvatarState endState = new AvatarState("end", _ref, false);
AvatarState overallState = new AvatarState("overall", _ref, false);
asm.addElement(start);
asm.setStartState(start);
asm.addElement(mainState);
asm.addElement(endState);
asm.addElement(overallState);
Florian Lugou
committed
AvatarTransition atF = new AvatarTransition(_ab, "at1", _ref);
asm.addElement(atF);
start.addNext(atF);
atF.addNext(mainState);
String finalGuard = "";
for(Attack att: _node.getInputAttacks()) {
AvatarAttribute aa = new AvatarAttribute(att.getName() + "__performed", AvatarType.BOOLEAN, _ab, _ref);
if (finalGuard.length() ==0) {
finalGuard += "(" + att.getName() + "__performed == true)";
} else {
finalGuard += " && (" + att.getName() + "__performed == true)";
}
_ab.addAttribute(aa);
atF.addAction(att.getName() + "__performed = false");
avatartranslator.AvatarSignal sigAtt = _main.getAvatarSignalWithName("accept__" + att.getName());
AvatarActionOnSignal acceptAttack = new AvatarActionOnSignal("AcceptAttack", sigAtt, _ref1);
asm.addElement(acceptAttack);
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at_toInputAttack", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(acceptAttack);
at.setGuard(new AvatarSimpleGuardDuo (aa, AvatarConstant.FALSE, "=="));
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_fromInputAttack", _ref);
at.addAction(att.getName() + "__performed = true");
asm.addElement(at);
acceptAttack.addNext(at);
at.addNext(mainState);
}
// Adding resulting attack
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at_toEnd", _ref);
mainState.addNext(at);
at.addNext(endState);
at.setGuard("[" + finalGuard + "]");
Attack resulting = _node.getResultingAttack();
avatartranslator.AvatarSignal sigAttack = _main.getAvatarSignalWithName("nodeDone__" + resulting.getName());
AvatarActionOnSignal resultingAttack = new AvatarActionOnSignal("ResultingAttack", sigAttack, _ref1);
asm.addElement(resultingAttack);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_toResultingAttack", _ref);
asm.addElement(at);
endState.addNext(at);
at.addNext(resultingAttack);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_Overall", _ref);
asm.addElement(at);
resultingAttack.addNext(at);
at.addNext(overallState);
}
private void makeORNode(AvatarSpecification _as, AvatarBlock _main, AvatarBlock _ab, ORNode _node, Object _ref) {
AvatarStateMachine asm = _ab.getStateMachine();
// Basic machine
AvatarStartState start = new AvatarStartState("start", _ref);
AvatarState mainState = new AvatarState("main", _ref, false);
AvatarState endState = new AvatarState("end", _ref, false);
AvatarState overallState = new AvatarState("overall", _ref, false);
asm.addElement(start);
asm.setStartState(start);
asm.addElement(mainState);
asm.addElement(endState);
asm.addElement(overallState);
Florian Lugou
committed
AvatarTransition atF = new AvatarTransition(_ab, "at1", _ref);
asm.addElement(atF);
start.addNext(atF);
atF.addNext(mainState);
for(Attack att: _node.getInputAttacks()) {
AvatarAttribute aa = new AvatarAttribute(att.getName() + "__performed", AvatarType.BOOLEAN, _ab, _ref);
if (finalGuard == null)
finalGuard = new AvatarSimpleGuardDuo (aa, AvatarConstant.TRUE, "==");
else
finalGuard = AvatarGuard.addGuard (finalGuard, new AvatarSimpleGuardDuo (aa, AvatarConstant.TRUE, "=="), "||");
_ab.addAttribute(aa);
atF.addAction(new AvatarActionAssignment (aa, AvatarConstant.FALSE));
// From Main
avatartranslator.AvatarSignal sigAtt = _main.getAvatarSignalWithName("accept__" + att.getName());
AvatarActionOnSignal acceptAttack = new AvatarActionOnSignal("AcceptAttack", sigAtt, _ref1);
asm.addElement(acceptAttack);
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at_toInputAttack", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(acceptAttack);
at.setGuard(new AvatarSimpleGuardDuo (aa, AvatarConstant.FALSE, "=="));
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_fromInputAttack", _ref);
at.addAction(new AvatarActionAssignment (aa, AvatarConstant.TRUE));
asm.addElement(at);
acceptAttack.addNext(at);
at.addNext(mainState);
// Link from End
acceptAttack = new AvatarActionOnSignal("AcceptAttack", sigAtt, _ref1);
asm.addElement(acceptAttack);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_toInputAttack", _ref);
asm.addElement(at);
endState.addNext(at);
at.addNext(acceptAttack);
at.setGuard(new AvatarSimpleGuardDuo (aa, AvatarConstant.FALSE, "=="));
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_fromInputAttack", _ref);
at.addAction(new AvatarActionAssignment (aa, AvatarConstant.TRUE));
asm.addElement(at);
acceptAttack.addNext(at);
at.addNext(endState);
// Link from Overall
acceptAttack = new AvatarActionOnSignal("AcceptAttack", sigAtt, _ref1);
asm.addElement(acceptAttack);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_toInputAttack", _ref);
asm.addElement(at);
overallState.addNext(at);
at.addNext(acceptAttack);
at.setGuard(new AvatarSimpleGuardDuo (aa, AvatarConstant.FALSE, "=="));
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_fromInputAttack", _ref);
at.addAction(new AvatarActionAssignment (aa, AvatarConstant.TRUE));
asm.addElement(at);
acceptAttack.addNext(at);
at.addNext(overallState);
}
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at_toEnd", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(endState);
Attack resulting = _node.getResultingAttack();
avatartranslator.AvatarSignal sigAttack = _main.getAvatarSignalWithName("nodeDone__" + resulting.getName());
AvatarActionOnSignal resultingAttack = new AvatarActionOnSignal("ResultingAttack", sigAttack, _ref1);
asm.addElement(resultingAttack);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_toResultingAttack", _ref);
asm.addElement(at);
endState.addNext(at);
at.addNext(resultingAttack);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_Overall", _ref);
asm.addElement(at);
resultingAttack.addNext(at);
at.addNext(overallState);
private void makeXORNode(AvatarSpecification _as, AvatarBlock _main, AvatarBlock _ab, XORNode _node, Object _ref) {
AvatarStateMachine asm = _ab.getStateMachine();
// Basic machine
AvatarStartState start = new AvatarStartState("start", _ref);
AvatarState mainState = new AvatarState("main", _ref, false);
AvatarState stoppingAll = new AvatarState("stoppingAll", _ref, false);
AvatarState endState = new AvatarState("end", _ref, false);
AvatarState overallState = new AvatarState("overall", _ref, false);
asm.addElement(start);
asm.setStartState(start);
asm.addElement(mainState);
asm.addElement(endState);
asm.addElement(overallState);
Florian Lugou
committed
AvatarTransition atF = new AvatarTransition(_ab, "at1", _ref);
asm.addElement(atF);
start.addNext(atF);
atF.addNext(mainState);
AvatarAttribute oneDone = new AvatarAttribute("oneDone", AvatarType.BOOLEAN, _ab, _ref);
AvatarGuard finalGuard = new AvatarSimpleGuardDuo (oneDone, AvatarConstant.TRUE, "==");
AvatarGuard toEndGuard = null;
_ab.addAttribute(oneDone);
atF.addAction("oneDone = false");
for(Attack att: _node.getInputAttacks()) {
AvatarAttribute aa = new AvatarAttribute(att.getName() + "__performed", AvatarType.BOOLEAN, _ab, _ref);
_ab.addAttribute(aa);
atF.addAction(new AvatarActionAssignment (aa, AvatarConstant.FALSE));
if (toEndGuard == null)
toEndGuard = new AvatarSimpleGuardDuo (aa, AvatarConstant.TRUE, "==");
else
toEndGuard = AvatarGuard.addGuard (toEndGuard, new AvatarSimpleGuardDuo (aa, AvatarConstant.TRUE, "=="), "&&");
// From Main
avatartranslator.AvatarSignal sigAtt = _main.getAvatarSignalWithName("accept__" + att.getName());
AvatarActionOnSignal acceptAttack = new AvatarActionOnSignal("AcceptAttack", sigAtt, _ref1);
asm.addElement(acceptAttack);
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at_toInputAttack", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(acceptAttack);
at.setGuard(new AvatarBinaryGuard (
new AvatarSimpleGuardDuo (aa, AvatarConstant.FALSE, "=="),
new AvatarSimpleGuardDuo (oneDone, AvatarConstant.FALSE, "=="),
"&&"));
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_fromInputAttack", _ref);
at.addAction(new AvatarActionAssignment (aa, AvatarConstant.TRUE));
at.addAction(new AvatarActionAssignment (oneDone, AvatarConstant.TRUE));
asm.addElement(at);
acceptAttack.addNext(at);
at.addNext(mainState);
// Link from stoppingAll
// if (att.isLeaf()) {
// Leaf attack -> must make a stop
sigAtt = _main.getAvatarSignalWithName("acceptStop__" + att.getName());
acceptAttack = new AvatarActionOnSignal("StopAttack", sigAtt, _ref1);
asm.addElement(acceptAttack);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_toInputAttack_leaf", _ref);
asm.addElement(at);
stoppingAll.addNext(at);
at.addNext(acceptAttack);
at.setGuard(new AvatarSimpleGuardDuo (aa, AvatarConstant.FALSE, "=="));
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_fromInputAttack", _ref);
at.addAction(new AvatarActionAssignment (aa, AvatarConstant.TRUE));
at.setHidden(true);
asm.addElement(at);
acceptAttack.addNext(at);
at.addNext(stoppingAll);
// } else {
// Generated attack-> must set performed to true.
/* at = new AvatarTransition("at_toInputAttack", _ref);
stoppingAll.addNext(at);
asm.addElement(at);
at.addNext(stoppingAll);
at.setGuard("["+att.getName() + "__performed == false]");
at.addAction(att.getName() + "__performed = true");
at.setHidden(true);*/
// }
Florian Lugou
committed
AvatarTransition at = new AvatarTransition(_ab, "at_toStoppingAll", _ref);
asm.addElement(at);
mainState.addNext(at);
at.addNext(stoppingAll);
// Adding resulting attack
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_toEnd", _ref);
asm.addElement(at);
stoppingAll.addNext(at);
at.addNext(endState);
Attack resulting = _node.getResultingAttack();
avatartranslator.AvatarSignal sigAttack = _main.getAvatarSignalWithName("nodeDone__" + resulting.getName());
AvatarActionOnSignal resultingAttack = new AvatarActionOnSignal("ResultingAttack", sigAttack, _ref1);
asm.addElement(resultingAttack);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_toResultingAttack", _ref);
asm.addElement(at);
endState.addNext(at);
at.addNext(resultingAttack);
Florian Lugou
committed
at = new AvatarTransition(_ab, "at_Overall", _ref);
asm.addElement(at);
resultingAttack.addNext(at);
at.addNext(overallState);
}
private void makeSequenceNode(AvatarSpecification _as, AvatarBlock _main, AvatarBlock _ab, SequenceNode _node, Object _ref) {
AvatarStateMachine asm = _ab.getStateMachine();
_node.orderAttacks();
// Basic machine
AvatarStartState start = new AvatarStartState("start", _ref);
AvatarState mainState = new AvatarState("main", _ref, false);
AvatarState endState = new AvatarState("end", _ref, false);
AvatarState overallState = new AvatarState("overall", _ref, false);
asm.addElement(start);
asm.setStartState(start);
asm.addElement(mainState);
asm.addElement(endState);
asm.addElement(overallState);