extras/vom/vom/nat_binding.cpp

   1 /*
   2  * Copyright (c) 2017 Cisco and/or its affiliates.
   3  * Licensed under the Apache License, Version 2.0 (the "License");
   4  * you may not use this file except in compliance with the License.
   5  * You may obtain a copy of the License at:
   6  *
   7  *     http://www.apache.org/licenses/LICENSE-2.0
   8  *
   9  * Unless required by applicable law or agreed to in writing, software
  10  * distributed under the License is distributed on an "AS IS" BASIS,
  11  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12  * See the License for the specific language governing permissions and
  13  * limitations under the License.
  14  */
  15
  16 #include "vom/nat_binding.hpp"
  17 #include "vom/cmd.hpp"
  18 #include "vom/nat_binding_cmds.hpp"
  19 #include "vom/singular_db_funcs.hpp"
  20
  21 namespace VOM {
  22 singular_db<const nat_binding::key_t, nat_binding> nat_binding::m_db;
  23
  24 nat_binding::event_handler nat_binding::m_evh;
  25
  26 const nat_binding::zone_t nat_binding::zone_t::INSIDE(0, "inside");
  27 const nat_binding::zone_t nat_binding::zone_t::OUTSIDE(0, "outside");
  28
  29 nat_binding::zone_t::zone_t(int v, const std::string s)
  30   : enum_base(v, s)
  31 {
  32 }
  33 const nat_binding::zone_t&
  34 nat_binding::zone_t::from_vpp(u8 is_inside)
  35 {
  36   if (is_inside)
  37     return zone_t::INSIDE;
  38   return zone_t::OUTSIDE;
  39 }
  40
  41 /**
  42  * Construct a new object matching the desried state
  43  */
  44 nat_binding::nat_binding(const interface& itf,
  45                          const direction_t& dir,
  46                          const l3_proto_t& proto,
  47                          const zone_t& zone)
  48   : m_binding(false)
  49   , m_itf(itf.singular())
  50   , m_dir(dir)
  51   , m_proto(proto)
  52   , m_zone(zone)
  53 {
  54 }
  55
  56 nat_binding::nat_binding(const nat_binding& o)
  57   : m_binding(o.m_binding)
  58   , m_itf(o.m_itf)
  59   , m_dir(o.m_dir)
  60   , m_proto(o.m_proto)
  61   , m_zone(o.m_zone)
  62 {
  63 }
  64
  65 nat_binding::~nat_binding()
  66 {
  67   sweep();
  68   m_db.release(key(), this);
  69 }
  70
  71 const nat_binding::key_t
  72 nat_binding::key() const
  73 {
  74   return (make_tuple(m_itf->key(), m_dir, m_proto));
  75 }
  76
  77 bool
  78 nat_binding::operator==(const nat_binding& n) const
  79 {
  80   return ((key() == n.key()) && (m_zone == n.m_zone));
  81 }
  82
  83 void
  84 nat_binding::sweep()
  85 {
  86   if (m_binding) {
  87     if (direction_t::INPUT == m_dir) {
  88       if (l3_proto_t::IPV4 == m_proto) {
  89         HW::enqueue(new nat_binding_cmds::unbind_44_input_cmd(
  90           m_binding, m_itf->handle(), m_zone));
  91       } else {
  92         HW::enqueue(new nat_binding_cmds::unbind_66_input_cmd(
  93           m_binding, m_itf->handle(), m_zone));
  94       }
  95     } else {
  96       if (l3_proto_t::IPV4 == m_proto) {
  97         HW::enqueue(new nat_binding_cmds::unbind_44_output_cmd(
  98           m_binding, m_itf->handle(), m_zone));
  99       } else {
 100         VOM_LOG(log_level_t::ERROR) << "NAT 66 output feature not supported";
 101       }
 102     }
 103   }
 104   HW::write();
 105 }
 106
 107 void
 108 nat_binding::replay()
 109 {
 110   if (m_binding) {
 111     if (direction_t::INPUT == m_dir) {
 112       if (l3_proto_t::IPV4 == m_proto) {
 113         HW::enqueue(new nat_binding_cmds::bind_44_input_cmd(
 114           m_binding, m_itf->handle(), m_zone));
 115       } else {
 116         HW::enqueue(new nat_binding_cmds::bind_66_input_cmd(
 117           m_binding, m_itf->handle(), m_zone));
 118       }
 119     } else {
 120       if (l3_proto_t::IPV4 == m_proto) {
 121         HW::enqueue(new nat_binding_cmds::bind_44_output_cmd(
 122           m_binding, m_itf->handle(), m_zone));
 123       } else {
 124         VOM_LOG(log_level_t::ERROR) << "NAT 66 output feature not supported";
 125       }
 126     }
 127   }
 128 }
 129
 130 void
 131 nat_binding::update(const nat_binding& desired)
 132 {
 133   /*
 134  * the desired state is always that the interface should be created
 135  */
 136   if (!m_binding) {
 137     if (direction_t::INPUT == m_dir) {
 138       if (l3_proto_t::IPV4 == m_proto) {
 139         HW::enqueue(new nat_binding_cmds::bind_44_input_cmd(
 140           m_binding, m_itf->handle(), m_zone));
 141       } else {
 142         HW::enqueue(new nat_binding_cmds::bind_66_input_cmd(
 143           m_binding, m_itf->handle(), m_zone));
 144       }
 145     } else {
 146       if (l3_proto_t::IPV4 == m_proto) {
 147         HW::enqueue(new nat_binding_cmds::bind_44_output_cmd(
 148           m_binding, m_itf->handle(), m_zone));
 149       } else {
 150         VOM_LOG(log_level_t::ERROR) << "NAT 66 output feature not supported";
 151       }
 152     }
 153   }
 154 }
 155
 156 std::string
 157 nat_binding::to_string() const
 158 {
 159   std::ostringstream s;
 160   s << "nat-binding:[" << m_itf->to_string()
 161     << " direction:" << m_dir.to_string() << " proto:" << m_proto.to_string()
 162     << " zone:" << m_zone.to_string() << "]";
 163
 164   return (s.str());
 165 }
 166
 167 std::shared_ptr<nat_binding>
 168 nat_binding::find_or_add(const nat_binding& temp)
 169 {
 170   return (m_db.find_or_add(temp.key(), temp));
 171 }
 172
 173 std::shared_ptr<nat_binding>
 174 nat_binding::find(const key_t& key)
 175 {
 176   return (m_db.find(key));
 177 }
 178
 179 std::shared_ptr<nat_binding>
 180 nat_binding::singular() const
 181 {
 182   return find_or_add(*this);
 183 }
 184
 185 void
 186 nat_binding::dump(std::ostream& os)
 187 {
 188   db_dump(m_db, os);
 189 }
 190
 191 std::ostream&
 192 operator<<(std::ostream& os, const nat_binding::key_t& key)
 193 {
 194   os << "[" << std::get<0>(key) << ", " << std::get<1>(key) << ", "
 195      << std::get<2>(key) << "]";
 196
 197   return (os);
 198 }
 199
 200 nat_binding::event_handler::event_handler()
 201 {
 202   OM::register_listener(this);
 203   inspect::register_handler({ "nat-binding" }, "NAT bindings", this);
 204 }
 205
 206 void
 207 nat_binding::event_handler::handle_replay()
 208 {
 209   m_db.replay();
 210 }
 211
 212 void
 213 nat_binding::event_handler::handle_populate(const client_db::key_t& key)
 214 {
 215   std::shared_ptr<nat_binding_cmds::dump_input_44_cmd> icmd =
 216     std::make_shared<nat_binding_cmds::dump_input_44_cmd>();
 217
 218   HW::enqueue(icmd);
 219   HW::write();
 220
 221   for (auto& record : *icmd) {
 222     auto& payload = record.get_payload();
 223
 224     std::shared_ptr<interface> itf = interface::find(payload.sw_if_index);
 225     nat_binding nb(*itf, direction_t::INPUT, l3_proto_t::IPV4,
 226                    zone_t::from_vpp(payload.is_inside));
 227     OM::commit(key, nb);
 228   }
 229
 230   std::shared_ptr<nat_binding_cmds::dump_output_44_cmd> ocmd =
 231     std::make_shared<nat_binding_cmds::dump_output_44_cmd>();
 232
 233   HW::enqueue(ocmd);
 234   HW::write();
 235
 236   for (auto& record : *ocmd) {
 237     auto& payload = record.get_payload();
 238
 239     std::shared_ptr<interface> itf = interface::find(payload.sw_if_index);
 240     nat_binding nb(*itf, direction_t::OUTPUT, l3_proto_t::IPV4,
 241                    zone_t::from_vpp(payload.is_inside));
 242     OM::commit(key, nb);
 243   }
 244
 245   std::shared_ptr<nat_binding_cmds::dump_input_66_cmd> i6cmd =
 246     std::make_shared<nat_binding_cmds::dump_input_66_cmd>();
 247
 248   HW::enqueue(i6cmd);
 249   HW::write();
 250
 251   for (auto& record : *i6cmd) {
 252     auto& payload = record.get_payload();
 253
 254     std::shared_ptr<interface> itf = interface::find(payload.sw_if_index);
 255     nat_binding nb(*itf, direction_t::INPUT, l3_proto_t::IPV6,
 256                    zone_t::from_vpp(payload.is_inside));
 257     OM::commit(key, nb);
 258   }
 259 }
 260
 261 dependency_t
 262 nat_binding::event_handler::order() const
 263 {
 264   return (dependency_t::BINDING);
 265 }
 266
 267 void
 268 nat_binding::event_handler::show(std::ostream& os)
 269 {
 270   db_dump(m_db, os);
 271 }
 272 }
 273
 274 /*
 275  * fd.io coding-style-patch-verification: ON
 276  *
 277  * Local Variables:
 278  * eval: (c-set-style "mozilla")
 279  * End:
 280  */