linkpair/collect/agent/ovs_agent.py

   1 # vim: tabstop=4 shiftwidth=4 softtabstop=4
   2 # -*- coding: utf-8 -*-
   3 #
   4
   5 __version__ = '1.1'
   6
   7 import re
   8 from linkpair.device import Device
   9 from linkpair.port import Port
  10 from linkpair.collect.agent.base_agent import BaseAgent
  11
  12
  13 class OVSAgent(BaseAgent):
  14     '''
  15     Open vSwitch Collector Agent
  16     '''
  17
  18     def run(self):
  19         self.get_configuration()
  20
  21     def get_bridge_metadata(self, br):
  22         br_meta = {}
  23         br_meta["dpid"] = ""
  24         br_meta["controller"] = ""
  25         br_meta["manager"] = ""
  26
  27         res_mgr = self._runner.exec_cmd("ovs-vsctl get-manager")
  28         for mgr in res_mgr:
  29             br_meta["manager"] = self._u.str_join(
  30                 br_meta["manager"], mgr.strip(), ", ")
  31
  32         res_ctl = self._runner.exec_cmd("ovs-vsctl get-controller " + br)
  33         for ctl in res_ctl:
  34             br_meta["controller"] = self._u.str_join(
  35                 br_meta["controller"], ctl.strip(), ", ")
  36
  37         res_ofsw = self._runner.exec_cmd("ovs-ofctl show " + br)
  38         for ofsw_spec in res_ofsw:
  39             ofsw_spec = ofsw_spec.strip()
  40             if self._u.d_push(re.search(r': dpid:(.*)', ofsw_spec)) is not None:
  41                 br_meta["dpid"] = self._u.d_pop().group(1).strip()
  42             elif self._u.d_push(re.search(r'n_tables:(.*), n_buffers:(.*)', ofsw_spec)) is not None:
  43                 match = self._u.d_pop()
  44                 br_meta["tables"] = match.group(1).strip()
  45                 br_meta["buffers"] = match.group(2).strip()
  46             elif self._u.d_push(re.search(r'capabilities:(.*)', ofsw_spec)) is not None:
  47                 br_meta["capabilities"] = self._u.d_pop().group(1).strip()
  48             elif self._u.d_push(re.search(r'actions:(.*)', ofsw_spec)) is not None:
  49                 br_meta["actions"] = self._u.d_pop().group(1).strip()
  50
  51         return br_meta
  52
  53     def get_port_metadata(self, port_name, br_name=""):
  54         port_meta = {}
  55         exist_port = self._cu.get_port(port_name)
  56         if exist_port != self._cu.PORT_NOT_FOUND:
  57             port_meta = exist_port.metadata
  58
  59         res_ovsport = self._runner.exec_cmd("ovs-vsctl show")
  60         for i in range(0, len(res_ovsport)):
  61             port_start = res_ovsport[i].strip()
  62             if self._u.d_push(re.search(r'Port "?%s"?' % port_name, port_start)) is not None:
  63                 for j in range(i + 1, len(res_ovsport)):
  64                     port_spec = res_ovsport[j].strip()
  65                     if self._u.d_push(re.search(r'tag: (.*)', port_spec)) is not None:
  66                         port_meta["tag"] = self._u.d_pop().group(1).strip()
  67                     elif self._u.d_push(re.search(r'type: (.*)', port_spec)) is not None:
  68                         port_meta["type"] = self._u.d_pop().group(1).strip()
  69                     elif self._u.d_push(re.search(r'options: {(.*)}', port_spec)) is not None:
  70                         port_options = self._u.d_pop().group(
  71                             1).strip().split(",")
  72                         for port_option in port_options:
  73                             (opt_name, opt_value) = port_option.split("=")
  74                             opt_name = opt_name.strip()
  75                             opt_value = opt_value.replace("\""," ")
  76                             opt_value = opt_value.strip()
  77                             port_meta[opt_name] = opt_value
  78                     elif self._u.d_push(re.search(r'Port "*', port_spec)) is not None:
  79                         break
  80
  81         if br_name != "":
  82             res_ofctl = self._runner.exec_cmd("ovs-ofctl show " + br_name)
  83             for i in range(0, len(res_ofctl)):
  84                 port_start = res_ofctl[i].strip()
  85                 if self._u.d_push(re.search(r'(\d+)\(%s\): addr:(.*)' % port_name, port_start)) is not None:
  86                     match = self._u.d_pop()
  87                     port_meta["port_no"] = match.group(1).strip()
  88                     port_meta["mac_addr"] = match.group(2).strip()
  89                     break
  90
  91             res_ofctl = self._runner.exec_cmd("ovs-ofctl dump-ports " + br_name)
  92             if "port_no" in port_meta:
  93                 found_port = False
  94                 for i in range(0, len(res_ofctl)):
  95                     port_start = res_ofctl[i].strip()
  96                     if self._u.d_push(
  97                         re.search(r'port\s*%s:\s*rx pkts=(\d+), bytes=(\d+), drop=(\d+), errs=(\d+), frame=(\d+), over=(\d+), crc=(\d+)'
  98                             % port_meta["port_no"], port_start)) is not None:
  99                         match = self._u.d_pop()
 100                         port_meta["rx_pkts"] = match.group(1).strip()
 101                         port_meta["rx_bytes"] = match.group(2).strip()
 102                         port_meta["rx_drop"] = match.group(3).strip()
 103                         port_meta["rx_errs"] = match.group(4).strip()
 104                         port_meta["rx_frame"] = match.group(5).strip()
 105                         port_meta["rx_over"] = match.group(6).strip()
 106                         port_meta["rx_crc"] = match.group(7).strip()
 107                         found_port = True
 108                     elif self._u.d_push(
 109                         re.search(r'\s*tx pkts=(\d+), bytes=(\d+), drop=(\d+), errs=(\d+), coll=(\d+)',
 110                             port_start)) is not None and found_port:
 111                         match = self._u.d_pop()
 112                         port_meta["tx_pkts"] = match.group(1).strip()
 113                         port_meta["tx_bytes"] = match.group(2).strip()
 114                         port_meta["tx_drop"] = match.group(3).strip()
 115                         port_meta["tx_errs"] = match.group(4).strip()
 116                         port_meta["tx_coll"] = match.group(5).strip()
 117                         break
 118
 119         return port_meta
 120
 121     def get_configuration(self):
 122         patch_peers = {}
 123         veth_peers = {}
 124
 125         result = self._runner.exec_cmd("ovs-vsctl list-br")
 126         for br_src in result:
 127             br_src = br_src.rstrip()
 128             br_src_meta = self.get_bridge_metadata(br_src)
 129             port_names = self._runner.exec_cmd(
 130                 "ovs-vsctl list-ports " + br_src)
 131             for port_name in port_names:
 132                 port_name = port_name.rstrip()
 133                 port_meta = self.get_port_metadata(port_name, br_src)
 134
 135                 self._port_to_br[port_name] = br_src
 136                 # patch port
 137                 if self._cu.get_port_type(port_meta) == "patch":
 138                     patch_src = port_name
 139                     patch_dst = self._cu.get_port_peer(port_meta)
 140                     patch_peers[patch_src + ":" + patch_dst] = self.PEER_FOUND
 141
 142                     if patch_dst + ":" + patch_src in patch_peers:
 143                         continue
 144
 145                     patch_src_meta = port_meta
 146                     result3 = self._runner.exec_cmd(
 147                         "ovs-vsctl port-to-br " + patch_dst)
 148                     if result3 is not None and len(result3) > 0:
 149                         br_dst = result3[0].rstrip()
 150                         br_dst_meta = self.get_bridge_metadata(br_dst)
 151                         patch_dst_meta = self.get_port_metadata(patch_dst, br_dst)
 152                         self._cu.add_linkpair(
 153                             Device(br_src, Device.BR_TYPE, br_src_meta),
 154                             Device(br_dst, Device.BR_TYPE, br_dst_meta),
 155                             Port(patch_src, Port.DEFAULT_TYPE, patch_src_meta),
 156                             Port(patch_dst, Port.DEFAULT_TYPE, patch_dst_meta),
 157                             self._formatter.PATCH_FORMAT)
 158                     else:
 159                         patch_dst_meta = self.get_port_metadata(patch_dst, br_dst)
 160                         self._cu.add_linkpair(
 161                             Device(br_src, Device.BR_TYPE, br_src_meta),
 162                             Device("NOT CONNECTED", Device.NOT_CONNECTED_TYPE),
 163                             Port(patch_src, Port.DEFAULT_TYPE, patch_src_meta),
 164                             Port(patch_dst, Port.DEFAULT_TYPE, patch_dst_meta),
 165                             self._formatter.PATCH_FORMAT)
 166
 167                 # veth port
 168                 elif self._cu.get_port_veth_peer(port_name) != self._cu.PORT_PEER_UNSPECIFIED:
 169                     peer_src = port_name
 170                     peer_dst = self._cu.get_port_veth_peer(port_name)
 171                     veth_peers[peer_src + ":" + peer_dst] = self.PEER_FOUND
 172
 173                     if peer_dst + ":" + peer_src in veth_peers:
 174                         continue
 175
 176                     peer_src_meta = port_meta
 177                     result3 = self._runner.exec_cmd(
 178                         "ovs-vsctl port-to-br " + peer_dst)
 179                     if result3 is not None and len(result3) > 0:
 180                         br_dst = result3[0].rstrip()
 181                         br_dst_meta = self.get_bridge_metadata(br_dst)
 182                         peer_dst_meta = self.get_port_metadata(peer_dst, br_dst)
 183                         self._cu.add_linkpair(
 184                             Device(br_src, Device.BR_TYPE, br_src_meta),
 185                             Device(br_dst, Device.BR_TYPE, br_dst_meta),
 186                             Port(peer_src, Port.DEFAULT_TYPE, peer_src_meta),
 187                             Port(peer_dst, Port.DEFAULT_TYPE, peer_dst_meta),
 188                             self._formatter.VETH_FORMAT)
 189                     else:
 190                         peer_dst_meta = self.get_port_metadata(peer_dst, br_dst)
 191                         self._cu.add_linkpair(
 192                             Device(br_src, Device.BR_TYPE, br_src_meta),
 193                             Device("NOT CONNECTED", Device.NOT_CONNECTED_TYPE),
 194                             Port(peer_src, Port.DEFAULT_TYPE, peer_src_meta),
 195                             Port(peer_dst, Port.DEFAULT_TYPE, peer_dst_meta),
 196                             self._formatter.VETH_FORMAT)
 197
 198                 # none patch/veth port
 199                 else:
 200                     # NOT Internal Bridge Port.
 201                     if self._cu.get_port_type(port_meta) != "internal":
 202
 203                         # physical interface
 204                         if self._u.d_push(re.match(r'^eth\d+$', port_name)) \
 205                             or self._u.d_push(re.match(r'^em\d+$', port_name)) \
 206                                 or self._u.d_push(re.match(r'^igb\d+$', port_name)) \
 207                                 or self._u.d_push(re.match(r'^bond\d+$', port_name)):
 208                             self._cu.add_linkpair(
 209                                 Device(br_src, Device.BR_TYPE, br_src_meta),
 210                                 Device("Physical NW", Device.PHYNET_TYPE),
 211                                 Port(port_name, Port.DEFAULT_TYPE, port_meta),
 212                                 Port(""))
 213
 214                         # tunnel port
 215                         elif self._u.d_push(re.match(r'(vxlan\d+)', port_name)) \
 216                             or self._cu.get_port_type(port_meta) == "gre" \
 217                                 or self._cu.get_port_type(port_meta) == "stt":
 218                             self._cu.add_linkpair(
 219                                 Device(br_src, Device.BR_TYPE, br_src_meta),
 220                                 Device("OS Routing", Device.OS_ROUTE_TYPE),
 221                                 Port(port_name, Port.DEFAULT_TYPE, port_meta),
 222                                 Port(""),
 223                                 self._formatter.OS_ROUTING_FORMAT)
 224
 225                         # Other OVSPort
 226                         else:
 227                             #### VLAN interface
 228                             if self._cu.get_port_vlan_master_if(port_name) != self._cu.PORT_VLAN_UNSPECIFIED:
 229                                 master_if = self._cu.get_port_vlan_master_if(port_name)
 230                                 if master_if in self._port_to_br:
 231                                     br_dst = self._port_to_br[master_if]
 232                                     br_dst_obj = self._cu.get_device(br_dst)
 233                                     port_src = port_name
 234                                     port_src_meta = port_meta
 235                                     port_dst = master_if
 236                                     #port_dst_obj = self._cu.get_port(master_if)
 237                                     port_dst_meta = self.get_port_metadata(port_dst, br_dst)
 238                                     self._cu.add_linkpair(
 239                                         Device(br_src, Device.BR_TYPE, br_src_meta),
 240                                         br_dst_obj,
 241                                         Port(port_src, Port.DEFAULT_TYPE, port_src_meta),
 242                                         #port_dst_obj,
 243                                         Port(port_dst, Port.DEFAULT_TYPE, port_dst_meta),
 244                                         self._formatter.VLAN_DIST_FORMAT)
 245
 246                                 elif self._u.d_push(re.match(r'^eth\d+$', master_if)) \
 247                                     or self._u.d_push(re.match(r'^em\d+$', master_if)) \
 248                                         or self._u.d_push(re.match(r'^igb\d+$', master_if)) \
 249                                         or self._u.d_push(re.match(r'^bond\d+$', master_if)):
 250                                     self._cu.add_linkpair(
 251                                         Device(br_src, Device.BR_TYPE, br_src_meta),
 252                                         Device("Physical NW", Device.PHYNET_TYPE),
 253                                         Port(port_name, Port.DEFAULT_TYPE, port_meta),
 254                                         Port(""))
 255                             else:
 256                                 self._cu.add_linkpair(
 257                                     Device(br_src, Device.BR_TYPE, br_src_meta),
 258                                     Device("NOT CONNECTED", Device.NOT_CONNECTED_TYPE),
 259                                     Port(port_name, Port.DEFAULT_TYPE, port_meta),
 260                                     Port(""))
 261
 262                     # internal
 263                     elif self._cu.get_port_type(port_meta) == "internal":
 264                         if port_name in self._iface_to_nss:
 265                             self._cu.add_linkpair(
 266                                 Device(br_src, Device.BR_TYPE, br_src_meta),
 267                                 Device(self._iface_to_nss[
 268                                        port], Device.NAMESPACE_TYPE),
 269                                 Port(port_name, Port.DEFAULT_TYPE, port_meta),
 270                                 Port(""),
 271                                 self._formatter.NAMESPACE_FORMAT)
 272                         else:
 273                             self._cu.add_linkpair(
 274                                 Device(br_src, Device.BR_TYPE, br_src_meta),
 275                                 Device("INTERNAL", Device.OS_ROUTE_TYPE),
 276                                 Port(port_name, Port.DEFAULT_TYPE, port_meta),
 277                                 Port(""))
 278                     # else:
 279                     #        ## Other OVSPort
 280                     #        self._port_to_br[port_name] = br_src
 281                     else:
 282                         continue