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