VpnGateway.java

package org.batfish.representation.aws_vpcs;

import java.io.Serializable;
import java.util.LinkedList;
import java.util.List;

import org.batfish.common.BatfishLogger;
import org.batfish.datamodel.Configuration;
import org.batfish.datamodel.Interface;
import org.batfish.datamodel.Ip;
import org.batfish.datamodel.Prefix;
import org.batfish.datamodel.StaticRoute;
import org.codehaus.jettison.json.JSONArray;
import org.codehaus.jettison.json.JSONException;
import org.codehaus.jettison.json.JSONObject;

public class VpnGateway implements AwsVpcEntity, Serializable {

   private static final long serialVersionUID = 1L;

   private List<String> _attachmentVpcIds = new LinkedList<>();

   private String _vpnGatewayId;

   public VpnGateway(JSONObject jObj, BatfishLogger logger)
         throws JSONException {
      _vpnGatewayId = jObj.getString(JSON_KEY_VPN_GATEWAY_ID);

      JSONArray attachments = jObj.getJSONArray(JSON_KEY_VPC_ATTACHMENTS);
      for (int index = 0; index < attachments.length(); index++) {
         JSONObject childObject = attachments.getJSONObject(index);
         _attachmentVpcIds.add(childObject.getString(JSON_KEY_VPC_ID));
      }

   }

   public List<String> getAttachmentVpcIds() {
      return _attachmentVpcIds;
   }

   @Override
   public String getId() {
      return _vpnGatewayId;
   }

   public Configuration toConfigurationNode(
         AwsVpcConfiguration awsVpcConfiguration) {
      Configuration cfgNode = new Configuration(_vpnGatewayId);

      for (String vpcId : _attachmentVpcIds) {

         String vgwIfaceName = vpcId;
         Interface vgwIface = new Interface(vgwIfaceName, cfgNode);
         Prefix vgwIfacePrefix = awsVpcConfiguration
               .getNextGeneratedLinkSubnet();
         vgwIface.setPrefix(vgwIfacePrefix);
         cfgNode.getInterfaces().put(vgwIfaceName, vgwIface);
         cfgNode.getDefaultVrf().getInterfaces().put(vgwIfaceName, vgwIface);

         // add the interface to the vpc router
         Configuration vpcConfigNode = awsVpcConfiguration
               .getConfigurationNodes().get(vpcId);
         String vpcIfaceName = _vpnGatewayId;
         Interface vpcIface = new Interface(vpcIfaceName, vpcConfigNode);
         Ip vpcIfaceIp = vgwIfacePrefix.getEndAddress();
         Prefix vpcIfacePrefix = new Prefix(vpcIfaceIp,
               vgwIfacePrefix.getPrefixLength());
         vpcIface.setPrefix(vpcIfacePrefix);
         vpcConfigNode.getInterfaces().put(vpcIfaceName, vpcIface);
         vpcConfigNode.getDefaultVrf().getInterfaces().put(vpcIfaceName,
               vpcIface);

         // associate this gateway with the vpc
         awsVpcConfiguration.getVpcs().get(vpcId)
               .setVpnGatewayId(_vpnGatewayId);

         // add a route on the gateway to the vpc
         Vpc vpc = awsVpcConfiguration.getVpcs().get(vpcId);
         StaticRoute vgwVpcRoute = new StaticRoute(vpc.getCidrBlock(),
               vpcIfaceIp, null, Route.DEFAULT_STATIC_ROUTE_ADMIN,
               Route.DEFAULT_STATIC_ROUTE_COST);
         cfgNode.getDefaultVrf().getStaticRoutes().add(vgwVpcRoute);
      }

      return cfgNode;
   }
}