verigraph/src/it/polito/verigraph/mcnet/netobjs/PolitoMailServer.java

   1 /*******************************************************************************
   2  * Copyright (c) 2017 Politecnico di Torino and others.
   3  *
   4  * All rights reserved. This program and the accompanying materials
   5  * are made available under the terms of the Apache License, Version 2.0
   6  * which accompanies this distribution, and is available at
   7  * http://www.apache.org/licenses/LICENSE-2.0
   8  *******************************************************************************/
   9 package it.polito.verigraph.mcnet.netobjs;
  10
  11 import java.util.ArrayList;
  12 import java.util.List;
  13
  14 import com.microsoft.z3.BoolExpr;
  15 import com.microsoft.z3.Context;
  16 import com.microsoft.z3.DatatypeExpr;
  17 import com.microsoft.z3.Expr;
  18 import com.microsoft.z3.FuncDecl;
  19 import com.microsoft.z3.IntExpr;
  20 import com.microsoft.z3.Solver;
  21
  22 import it.polito.verigraph.mcnet.components.NetContext;
  23 import it.polito.verigraph.mcnet.components.Network;
  24 import it.polito.verigraph.mcnet.components.NetworkObject;
  25 /** Mail server objects
  26  *
  27  *
  28  */
  29 public class PolitoMailServer extends NetworkObject{
  30
  31     List<BoolExpr> constraints;
  32     Context ctx;
  33     DatatypeExpr politoMailServer;
  34     Network net;
  35     NetContext nctx;
  36     FuncDecl isInBlacklist;
  37
  38     public PolitoMailServer(Context ctx, Object[]... args) {
  39         super(ctx, args);
  40     }
  41
  42     @Override
  43     protected void init(Context ctx, Object[]... args) {
  44         this.ctx = ctx;
  45         isEndHost=false;
  46         constraints = new ArrayList<BoolExpr>();
  47         z3Node = ((NetworkObject)args[0][0]).getZ3Node();
  48         politoMailServer = z3Node;
  49         net = (Network)args[0][1];
  50         nctx = (NetContext)args[0][2];
  51         mailServerRules();
  52         net.saneSend(this);
  53     }
  54
  55     @Override
  56     public DatatypeExpr getZ3Node() {
  57         return politoMailServer;
  58     }
  59
  60     @Override
  61     protected void addConstraints(Solver solver) {
  62         BoolExpr[] constr = new BoolExpr[constraints.size()];
  63         solver.add(constraints.toArray(constr));
  64     }
  65
  66     private void mailServerRules (){
  67         Expr n_0 = ctx.mkConst(politoMailServer+"_n_0", nctx.node);
  68         Expr p_0 = ctx.mkConst(politoMailServer+"_p_0", nctx.packet);
  69         Expr p_1 = ctx.mkConst(politoMailServer+"_p_1", nctx.packet);
  70         IntExpr t_0 = ctx.mkIntConst(politoMailServer+"_t_0");
  71         IntExpr t_1 = ctx.mkIntConst(politoMailServer+"_t_1");
  72
  73         //Constraint1 send(politoMailServer, n_0, p, t_0) -> nodeHasAddr(politoMailServer,p.src)
  74         constraints.add( ctx.mkForall(new Expr[]{n_0, p_0, t_0},
  75                 ctx.mkImplies((BoolExpr)nctx.send.apply(politoMailServer, n_0, p_0, t_0),
  76                         (BoolExpr)nctx.nodeHasAddr.apply(politoMailServer,nctx.pf.get("src").apply(p_0))),1,null,null,null,null));
  77
  78         //Constraint2 send(politoMailServer, n_0, p, t_0) -> p.origin == politoMailServer
  79         constraints.add( ctx.mkForall(new Expr[]{n_0, p_0, t_0},
  80                 ctx.mkImplies((BoolExpr)nctx.send.apply(politoMailServer, n_0, p_0, t_0),
  81                         ctx.mkEq(nctx.pf.get("origin").apply(p_0),politoMailServer)),1,null,null,null,null));
  82
  83         //Constraint3 send(politoMailServer, n_0, p, t_0) -> p.orig_body == p.body
  84         constraints.add( ctx.mkForall(new Expr[]{n_0, p_0, t_0},
  85                 ctx.mkImplies((BoolExpr)nctx.send.apply(politoMailServer, n_0, p_0, t_0),
  86                         ctx.mkEq(nctx.pf.get("orig_body").apply(p_0),nctx.pf.get("body").apply(p_0))),1,null,null,null,null));
  87
  88         //Constraint4 recv(n_0, politoMailServer, p, t_0) -> nodeHasAddr(politoMailServer,p.dest)
  89         constraints.add( ctx.mkForall(new Expr[]{n_0, p_0, t_0},
  90                 ctx.mkImplies((BoolExpr)nctx.recv.apply(n_0,politoMailServer, p_0, t_0),
  91                         (BoolExpr)nctx.nodeHasAddr.apply(politoMailServer,nctx.pf.get("dest").apply(p_0))),1,null,null,null,null));
  92
  93         //Constraint5 send(politoMailServer, n_0, p, t_0) -> p.proto == POP3_RESP && p.emailFrom == 1
  94         constraints.add( ctx.mkForall(new Expr[]{n_0, p_0, t_0},
  95                 ctx.mkImplies((BoolExpr)nctx.send.apply(politoMailServer, n_0, p_0, t_0),
  96                         ctx.mkAnd(ctx.mkEq(nctx.pf.get("proto").apply(p_0), ctx.mkInt(nctx.POP3_RESPONSE)),
  97                                 ctx.mkEq(nctx.pf.get("emailFrom").apply(p_0), ctx.mkInt(1)))),1,null,null,null,null));
  98
  99         //Constraint6 send(politoMailServer, n_0, p, t_0)  ->
 100         //    (exist p_1, t_1 : (t_1 < t_0 && recv(n_0, politoMailServer, p_1, t_1) &&
 101         //    p_0.proto == POP3_RESP &&  p_1.proto == POP3_REQ && p_0.dest == p_1.src )
 102
 103         constraints.add( ctx.mkForall(new Expr[]{n_0, p_0, t_0},
 104                 ctx.mkImplies((BoolExpr)nctx.send.apply(politoMailServer, n_0, p_0, t_0),
 105                         ctx.mkExists(new Expr[]{p_1, t_1},
 106                                 ctx.mkAnd(ctx.mkLt(t_1, t_0),
 107                                         (BoolExpr)nctx.recv.apply(n_0, politoMailServer, p_1, t_1),
 108                                         ctx.mkEq(nctx.pf.get("proto").apply(p_0), ctx.mkInt(nctx.POP3_RESPONSE)),
 109                                         ctx.mkEq(nctx.pf.get("proto").apply(p_1), ctx.mkInt(nctx.POP3_REQUEST)),
 110                                         ctx.mkEq(nctx.pf.get("dest").apply(p_0), nctx.pf.get("src").apply(p_1))),1,null,null,null,null)),1,null,null,null,null));
 111
 112         //    constraints.add( ctx.mkForall(new Expr[]{n_0, p_0, t_0},
 113         //                    ctx.mkImplies((BoolExpr)nctx.send.apply(politoMailServer, n_0, p_0, t_0),
 114         //                    ctx.mkEq(nctx.pf.get("emailFrom").apply(p_0),ctx.mkInt(2))),1,null,null,null,null));
 115     }
 116 }