| #!/usr/bin/env python3 |
| # SPDX-License-Identifier: GPL-2.0 |
| |
| # Controls the openvswitch module. Part of the kselftest suite, but |
| # can be used for some diagnostic purpose as well. |
| |
| import argparse |
| import errno |
| import ipaddress |
| import logging |
| import multiprocessing |
| import re |
| import struct |
| import sys |
| import time |
| import types |
| import uuid |
| |
| try: |
| from pyroute2 import NDB |
| |
| from pyroute2.netlink import NLA_F_NESTED |
| from pyroute2.netlink import NLM_F_ACK |
| from pyroute2.netlink import NLM_F_DUMP |
| from pyroute2.netlink import NLM_F_REQUEST |
| from pyroute2.netlink import genlmsg |
| from pyroute2.netlink import nla |
| from pyroute2.netlink import nlmsg_atoms |
| from pyroute2.netlink.exceptions import NetlinkError |
| from pyroute2.netlink.generic import GenericNetlinkSocket |
| import pyroute2 |
| |
| except ModuleNotFoundError: |
| print("Need to install the python pyroute2 package >= 0.6.") |
| sys.exit(0) |
| |
| |
| OVS_DATAPATH_FAMILY = "ovs_datapath" |
| OVS_VPORT_FAMILY = "ovs_vport" |
| OVS_FLOW_FAMILY = "ovs_flow" |
| OVS_PACKET_FAMILY = "ovs_packet" |
| OVS_METER_FAMILY = "ovs_meter" |
| OVS_CT_LIMIT_FAMILY = "ovs_ct_limit" |
| |
| OVS_DATAPATH_VERSION = 2 |
| OVS_DP_CMD_NEW = 1 |
| OVS_DP_CMD_DEL = 2 |
| OVS_DP_CMD_GET = 3 |
| OVS_DP_CMD_SET = 4 |
| |
| OVS_VPORT_CMD_NEW = 1 |
| OVS_VPORT_CMD_DEL = 2 |
| OVS_VPORT_CMD_GET = 3 |
| OVS_VPORT_CMD_SET = 4 |
| |
| OVS_FLOW_CMD_NEW = 1 |
| OVS_FLOW_CMD_DEL = 2 |
| OVS_FLOW_CMD_GET = 3 |
| OVS_FLOW_CMD_SET = 4 |
| |
| |
| def macstr(mac): |
| outstr = ":".join(["%02X" % i for i in mac]) |
| return outstr |
| |
| |
| def strcspn(str1, str2): |
| tot = 0 |
| for char in str1: |
| if str2.find(char) != -1: |
| return tot |
| tot += 1 |
| return tot |
| |
| |
| def strspn(str1, str2): |
| tot = 0 |
| for char in str1: |
| if str2.find(char) == -1: |
| return tot |
| tot += 1 |
| return tot |
| |
| |
| def intparse(statestr, defmask="0xffffffff"): |
| totalparse = strspn(statestr, "0123456789abcdefABCDEFx/") |
| # scan until "/" |
| count = strspn(statestr, "x0123456789abcdefABCDEF") |
| |
| firstnum = statestr[:count] |
| if firstnum[-1] == "/": |
| firstnum = firstnum[:-1] |
| k = int(firstnum, 0) |
| |
| m = None |
| if defmask is not None: |
| secondnum = defmask |
| if statestr[count] == "/": |
| secondnum = statestr[count + 1 :] # this is wrong... |
| m = int(secondnum, 0) |
| |
| return statestr[totalparse + 1 :], k, m |
| |
| |
| def parse_flags(flag_str, flag_vals): |
| bitResult = 0 |
| maskResult = 0 |
| |
| if len(flag_str) == 0: |
| return flag_str, bitResult, maskResult |
| |
| if flag_str[0].isdigit(): |
| idx = 0 |
| while flag_str[idx].isdigit() or flag_str[idx] == "x": |
| idx += 1 |
| digits = flag_str[:idx] |
| flag_str = flag_str[idx:] |
| |
| bitResult = int(digits, 0) |
| maskResult = int(digits, 0) |
| |
| while len(flag_str) > 0 and (flag_str[0] == "+" or flag_str[0] == "-"): |
| if flag_str[0] == "+": |
| setFlag = True |
| elif flag_str[0] == "-": |
| setFlag = False |
| |
| flag_str = flag_str[1:] |
| |
| flag_len = 0 |
| while ( |
| flag_str[flag_len] != "+" |
| and flag_str[flag_len] != "-" |
| and flag_str[flag_len] != "," |
| and flag_str[flag_len] != ")" |
| ): |
| flag_len += 1 |
| |
| flag = flag_str[0:flag_len] |
| |
| if flag in flag_vals: |
| if maskResult & flag_vals[flag]: |
| raise KeyError( |
| "Flag %s set once, cannot be set in multiples" % flag |
| ) |
| |
| if setFlag: |
| bitResult |= flag_vals[flag] |
| |
| maskResult |= flag_vals[flag] |
| else: |
| raise KeyError("Missing flag value: %s" % flag) |
| |
| flag_str = flag_str[flag_len:] |
| |
| return flag_str, bitResult, maskResult |
| |
| |
| def parse_ct_state(statestr): |
| ct_flags = { |
| "new": 1 << 0, |
| "est": 1 << 1, |
| "rel": 1 << 2, |
| "rpl": 1 << 3, |
| "inv": 1 << 4, |
| "trk": 1 << 5, |
| "snat": 1 << 6, |
| "dnat": 1 << 7, |
| } |
| |
| return parse_flags(statestr, ct_flags) |
| |
| |
| def convert_mac(data): |
| def to_bytes(mac): |
| mac_split = mac.split(":") |
| ret = bytearray([int(i, 16) for i in mac_split]) |
| return bytes(ret) |
| |
| mac_str, _, mask_str = data.partition('/') |
| |
| if not mac_str: |
| mac_str = mask_str = "00:00:00:00:00:00" |
| elif not mask_str: |
| mask_str = "FF:FF:FF:FF:FF:FF" |
| |
| return to_bytes(mac_str), to_bytes(mask_str) |
| |
| def convert_ipv4(data): |
| ip, _, mask = data.partition('/') |
| |
| if not ip: |
| ip = mask = 0 |
| elif not mask: |
| mask = 0xFFFFFFFF |
| elif mask.isdigit(): |
| mask = (0xFFFFFFFF << (32 - int(mask))) & 0xFFFFFFFF |
| |
| return int(ipaddress.IPv4Address(ip)), int(ipaddress.IPv4Address(mask)) |
| |
| def convert_int(size): |
| def convert_int_sized(data): |
| value, _, mask = data.partition('/') |
| |
| if not value: |
| return 0, 0 |
| elif not mask: |
| return int(value, 0), pow(2, size) - 1 |
| else: |
| return int(value, 0), int(mask, 0) |
| |
| return convert_int_sized |
| |
| def parse_starts_block(block_str, scanstr, returnskipped, scanregex=False): |
| if scanregex: |
| m = re.search(scanstr, block_str) |
| if m is None: |
| if returnskipped: |
| return block_str |
| return False |
| if returnskipped: |
| block_str = block_str[len(m.group(0)) :] |
| return block_str |
| return True |
| |
| if block_str.startswith(scanstr): |
| if returnskipped: |
| block_str = block_str[len(scanstr) :] |
| else: |
| return True |
| |
| if returnskipped: |
| return block_str |
| |
| return False |
| |
| |
| def parse_extract_field( |
| block_str, fieldstr, scanfmt, convert, masked=False, defval=None |
| ): |
| if fieldstr and not block_str.startswith(fieldstr): |
| return block_str, defval |
| |
| if fieldstr: |
| str_skiplen = len(fieldstr) |
| str_skipped = block_str[str_skiplen:] |
| if str_skiplen == 0: |
| return str_skipped, defval |
| else: |
| str_skiplen = 0 |
| str_skipped = block_str |
| |
| m = re.search(scanfmt, str_skipped) |
| if m is None: |
| raise ValueError("Bad fmt string") |
| |
| data = m.group(0) |
| if convert: |
| data = convert(m.group(0)) |
| |
| str_skipped = str_skipped[len(m.group(0)) :] |
| if masked: |
| if str_skipped[0] == "/": |
| raise ValueError("Masking support TBD...") |
| |
| str_skipped = str_skipped[strspn(str_skipped, ", ") :] |
| return str_skipped, data |
| |
| |
| class ovs_dp_msg(genlmsg): |
| # include the OVS version |
| # We need a custom header rather than just being able to rely on |
| # genlmsg because fields ends up not expressing everything correctly |
| # if we use the canonical example of setting fields = (('customfield',),) |
| fields = genlmsg.fields + (("dpifindex", "I"),) |
| |
| |
| class ovsactions(nla): |
| nla_flags = NLA_F_NESTED |
| |
| nla_map = ( |
| ("OVS_ACTION_ATTR_UNSPEC", "none"), |
| ("OVS_ACTION_ATTR_OUTPUT", "uint32"), |
| ("OVS_ACTION_ATTR_USERSPACE", "userspace"), |
| ("OVS_ACTION_ATTR_SET", "none"), |
| ("OVS_ACTION_ATTR_PUSH_VLAN", "none"), |
| ("OVS_ACTION_ATTR_POP_VLAN", "flag"), |
| ("OVS_ACTION_ATTR_SAMPLE", "none"), |
| ("OVS_ACTION_ATTR_RECIRC", "uint32"), |
| ("OVS_ACTION_ATTR_HASH", "none"), |
| ("OVS_ACTION_ATTR_PUSH_MPLS", "none"), |
| ("OVS_ACTION_ATTR_POP_MPLS", "flag"), |
| ("OVS_ACTION_ATTR_SET_MASKED", "none"), |
| ("OVS_ACTION_ATTR_CT", "ctact"), |
| ("OVS_ACTION_ATTR_TRUNC", "uint32"), |
| ("OVS_ACTION_ATTR_PUSH_ETH", "none"), |
| ("OVS_ACTION_ATTR_POP_ETH", "flag"), |
| ("OVS_ACTION_ATTR_CT_CLEAR", "flag"), |
| ("OVS_ACTION_ATTR_PUSH_NSH", "none"), |
| ("OVS_ACTION_ATTR_POP_NSH", "flag"), |
| ("OVS_ACTION_ATTR_METER", "none"), |
| ("OVS_ACTION_ATTR_CLONE", "recursive"), |
| ("OVS_ACTION_ATTR_CHECK_PKT_LEN", "none"), |
| ("OVS_ACTION_ATTR_ADD_MPLS", "none"), |
| ("OVS_ACTION_ATTR_DEC_TTL", "none"), |
| ("OVS_ACTION_ATTR_DROP", "uint32"), |
| ) |
| |
| class ctact(nla): |
| nla_flags = NLA_F_NESTED |
| |
| nla_map = ( |
| ("OVS_CT_ATTR_NONE", "none"), |
| ("OVS_CT_ATTR_COMMIT", "flag"), |
| ("OVS_CT_ATTR_ZONE", "uint16"), |
| ("OVS_CT_ATTR_MARK", "none"), |
| ("OVS_CT_ATTR_LABELS", "none"), |
| ("OVS_CT_ATTR_HELPER", "asciiz"), |
| ("OVS_CT_ATTR_NAT", "natattr"), |
| ("OVS_CT_ATTR_FORCE_COMMIT", "flag"), |
| ("OVS_CT_ATTR_EVENTMASK", "uint32"), |
| ("OVS_CT_ATTR_TIMEOUT", "asciiz"), |
| ) |
| |
| class natattr(nla): |
| nla_flags = NLA_F_NESTED |
| |
| nla_map = ( |
| ("OVS_NAT_ATTR_NONE", "none"), |
| ("OVS_NAT_ATTR_SRC", "flag"), |
| ("OVS_NAT_ATTR_DST", "flag"), |
| ("OVS_NAT_ATTR_IP_MIN", "ipaddr"), |
| ("OVS_NAT_ATTR_IP_MAX", "ipaddr"), |
| ("OVS_NAT_ATTR_PROTO_MIN", "uint16"), |
| ("OVS_NAT_ATTR_PROTO_MAX", "uint16"), |
| ("OVS_NAT_ATTR_PERSISTENT", "flag"), |
| ("OVS_NAT_ATTR_PROTO_HASH", "flag"), |
| ("OVS_NAT_ATTR_PROTO_RANDOM", "flag"), |
| ) |
| |
| def dpstr(self, more=False): |
| print_str = "nat(" |
| |
| if self.get_attr("OVS_NAT_ATTR_SRC"): |
| print_str += "src" |
| elif self.get_attr("OVS_NAT_ATTR_DST"): |
| print_str += "dst" |
| else: |
| print_str += "XXX-unknown-nat" |
| |
| if self.get_attr("OVS_NAT_ATTR_IP_MIN") or self.get_attr( |
| "OVS_NAT_ATTR_IP_MAX" |
| ): |
| if self.get_attr("OVS_NAT_ATTR_IP_MIN"): |
| print_str += "=%s," % str( |
| self.get_attr("OVS_NAT_ATTR_IP_MIN") |
| ) |
| |
| if self.get_attr("OVS_NAT_ATTR_IP_MAX"): |
| print_str += "-%s," % str( |
| self.get_attr("OVS_NAT_ATTR_IP_MAX") |
| ) |
| else: |
| print_str += "," |
| |
| if self.get_attr("OVS_NAT_ATTR_PROTO_MIN"): |
| print_str += "proto_min=%d," % self.get_attr( |
| "OVS_NAT_ATTR_PROTO_MIN" |
| ) |
| |
| if self.get_attr("OVS_NAT_ATTR_PROTO_MAX"): |
| print_str += "proto_max=%d," % self.get_attr( |
| "OVS_NAT_ATTR_PROTO_MAX" |
| ) |
| |
| if self.get_attr("OVS_NAT_ATTR_PERSISTENT"): |
| print_str += "persistent," |
| if self.get_attr("OVS_NAT_ATTR_HASH"): |
| print_str += "hash," |
| if self.get_attr("OVS_NAT_ATTR_RANDOM"): |
| print_str += "random" |
| print_str += ")" |
| return print_str |
| |
| def dpstr(self, more=False): |
| print_str = "ct(" |
| |
| if self.get_attr("OVS_CT_ATTR_COMMIT") is not None: |
| print_str += "commit," |
| if self.get_attr("OVS_CT_ATTR_ZONE") is not None: |
| print_str += "zone=%d," % self.get_attr("OVS_CT_ATTR_ZONE") |
| if self.get_attr("OVS_CT_ATTR_HELPER") is not None: |
| print_str += "helper=%s," % self.get_attr("OVS_CT_ATTR_HELPER") |
| if self.get_attr("OVS_CT_ATTR_NAT") is not None: |
| print_str += self.get_attr("OVS_CT_ATTR_NAT").dpstr(more) |
| print_str += "," |
| if self.get_attr("OVS_CT_ATTR_FORCE_COMMIT") is not None: |
| print_str += "force," |
| if self.get_attr("OVS_CT_ATTR_EVENTMASK") is not None: |
| print_str += "emask=0x%X," % self.get_attr( |
| "OVS_CT_ATTR_EVENTMASK" |
| ) |
| if self.get_attr("OVS_CT_ATTR_TIMEOUT") is not None: |
| print_str += "timeout=%s" % self.get_attr( |
| "OVS_CT_ATTR_TIMEOUT" |
| ) |
| print_str += ")" |
| return print_str |
| |
| class userspace(nla): |
| nla_flags = NLA_F_NESTED |
| |
| nla_map = ( |
| ("OVS_USERSPACE_ATTR_UNUSED", "none"), |
| ("OVS_USERSPACE_ATTR_PID", "uint32"), |
| ("OVS_USERSPACE_ATTR_USERDATA", "array(uint8)"), |
| ("OVS_USERSPACE_ATTR_EGRESS_TUN_PORT", "uint32"), |
| ) |
| |
| def dpstr(self, more=False): |
| print_str = "userspace(" |
| if self.get_attr("OVS_USERSPACE_ATTR_PID") is not None: |
| print_str += "pid=%d," % self.get_attr( |
| "OVS_USERSPACE_ATTR_PID" |
| ) |
| if self.get_attr("OVS_USERSPACE_ATTR_USERDATA") is not None: |
| print_str += "userdata=" |
| for f in self.get_attr("OVS_USERSPACE_ATTR_USERDATA"): |
| print_str += "%x." % f |
| if self.get_attr("OVS_USERSPACE_ATTR_TUN_PORT") is not None: |
| print_str += "egress_tun_port=%d" % self.get_attr( |
| "OVS_USERSPACE_ATTR_TUN_PORT" |
| ) |
| print_str += ")" |
| return print_str |
| |
| def dpstr(self, more=False): |
| print_str = "" |
| |
| for field in self.nla_map: |
| if field[1] == "none" or self.get_attr(field[0]) is None: |
| continue |
| if print_str != "": |
| print_str += "," |
| |
| if field[1] == "uint32": |
| if field[0] == "OVS_ACTION_ATTR_OUTPUT": |
| print_str += "%d" % int(self.get_attr(field[0])) |
| elif field[0] == "OVS_ACTION_ATTR_RECIRC": |
| print_str += "recirc(0x%x)" % int(self.get_attr(field[0])) |
| elif field[0] == "OVS_ACTION_ATTR_TRUNC": |
| print_str += "trunc(%d)" % int(self.get_attr(field[0])) |
| elif field[0] == "OVS_ACTION_ATTR_DROP": |
| print_str += "drop(%d)" % int(self.get_attr(field[0])) |
| elif field[1] == "flag": |
| if field[0] == "OVS_ACTION_ATTR_CT_CLEAR": |
| print_str += "ct_clear" |
| elif field[0] == "OVS_ACTION_ATTR_POP_VLAN": |
| print_str += "pop_vlan" |
| elif field[0] == "OVS_ACTION_ATTR_POP_ETH": |
| print_str += "pop_eth" |
| elif field[0] == "OVS_ACTION_ATTR_POP_NSH": |
| print_str += "pop_nsh" |
| elif field[0] == "OVS_ACTION_ATTR_POP_MPLS": |
| print_str += "pop_mpls" |
| else: |
| datum = self.get_attr(field[0]) |
| if field[0] == "OVS_ACTION_ATTR_CLONE": |
| print_str += "clone(" |
| print_str += datum.dpstr(more) |
| print_str += ")" |
| else: |
| print_str += datum.dpstr(more) |
| |
| return print_str |
| |
| def parse(self, actstr): |
| totallen = len(actstr) |
| while len(actstr) != 0: |
| parsed = False |
| parencount = 0 |
| if actstr.startswith("drop"): |
| # If no reason is provided, the implicit drop is used (i.e no |
| # action). If some reason is given, an explicit action is used. |
| reason = None |
| if actstr.startswith("drop("): |
| parencount += 1 |
| |
| actstr, reason = parse_extract_field( |
| actstr, |
| "drop(", |
| "([0-9]+)", |
| lambda x: int(x, 0), |
| False, |
| None, |
| ) |
| |
| if reason is not None: |
| self["attrs"].append(["OVS_ACTION_ATTR_DROP", reason]) |
| parsed = True |
| else: |
| actstr = actstr[len("drop"): ] |
| return (totallen - len(actstr)) |
| |
| elif parse_starts_block(actstr, "^(\d+)", False, True): |
| actstr, output = parse_extract_field( |
| actstr, None, "(\d+)", lambda x: int(x), False, "0" |
| ) |
| self["attrs"].append(["OVS_ACTION_ATTR_OUTPUT", output]) |
| parsed = True |
| elif parse_starts_block(actstr, "recirc(", False): |
| actstr, recircid = parse_extract_field( |
| actstr, |
| "recirc(", |
| "([0-9a-fA-Fx]+)", |
| lambda x: int(x, 0), |
| False, |
| 0, |
| ) |
| parencount += 1 |
| self["attrs"].append(["OVS_ACTION_ATTR_RECIRC", recircid]) |
| parsed = True |
| |
| parse_flat_map = ( |
| ("ct_clear", "OVS_ACTION_ATTR_CT_CLEAR"), |
| ("pop_vlan", "OVS_ACTION_ATTR_POP_VLAN"), |
| ("pop_eth", "OVS_ACTION_ATTR_POP_ETH"), |
| ("pop_nsh", "OVS_ACTION_ATTR_POP_NSH"), |
| ) |
| |
| for flat_act in parse_flat_map: |
| if parse_starts_block(actstr, flat_act[0], False): |
| actstr = actstr[len(flat_act[0]):] |
| self["attrs"].append([flat_act[1]]) |
| actstr = actstr[strspn(actstr, ", ") :] |
| parsed = True |
| |
| if parse_starts_block(actstr, "clone(", False): |
| parencount += 1 |
| subacts = ovsactions() |
| actstr = actstr[len("clone("):] |
| parsedLen = subacts.parse(actstr) |
| lst = [] |
| self["attrs"].append(("OVS_ACTION_ATTR_CLONE", subacts)) |
| actstr = actstr[parsedLen:] |
| parsed = True |
| elif parse_starts_block(actstr, "ct(", False): |
| parencount += 1 |
| actstr = actstr[len("ct(") :] |
| ctact = ovsactions.ctact() |
| |
| for scan in ( |
| ("commit", "OVS_CT_ATTR_COMMIT", None), |
| ("force_commit", "OVS_CT_ATTR_FORCE_COMMIT", None), |
| ("zone", "OVS_CT_ATTR_ZONE", int), |
| ("mark", "OVS_CT_ATTR_MARK", int), |
| ("helper", "OVS_CT_ATTR_HELPER", lambda x, y: str(x)), |
| ("timeout", "OVS_CT_ATTR_TIMEOUT", lambda x, y: str(x)), |
| ): |
| if actstr.startswith(scan[0]): |
| actstr = actstr[len(scan[0]) :] |
| if scan[2] is not None: |
| if actstr[0] != "=": |
| raise ValueError("Invalid ct attr") |
| actstr = actstr[1:] |
| pos = strcspn(actstr, ",)") |
| datum = scan[2](actstr[:pos], 0) |
| ctact["attrs"].append([scan[1], datum]) |
| actstr = actstr[pos:] |
| else: |
| ctact["attrs"].append([scan[1], None]) |
| actstr = actstr[strspn(actstr, ", ") :] |
| # it seems strange to put this here, but nat() is a complex |
| # sub-action and this lets it sit anywhere in the ct() action |
| if actstr.startswith("nat"): |
| actstr = actstr[3:] |
| natact = ovsactions.ctact.natattr() |
| |
| if actstr.startswith("("): |
| parencount += 1 |
| t = None |
| actstr = actstr[1:] |
| if actstr.startswith("src"): |
| t = "OVS_NAT_ATTR_SRC" |
| actstr = actstr[3:] |
| elif actstr.startswith("dst"): |
| t = "OVS_NAT_ATTR_DST" |
| actstr = actstr[3:] |
| |
| actstr, ip_block_min = parse_extract_field( |
| actstr, "=", "([0-9a-fA-F\.]+)", str, False |
| ) |
| actstr, ip_block_max = parse_extract_field( |
| actstr, "-", "([0-9a-fA-F\.]+)", str, False |
| ) |
| |
| actstr, proto_min = parse_extract_field( |
| actstr, ":", "(\d+)", int, False |
| ) |
| actstr, proto_max = parse_extract_field( |
| actstr, "-", "(\d+)", int, False |
| ) |
| |
| if t is not None: |
| natact["attrs"].append([t, None]) |
| |
| if ip_block_min is not None: |
| natact["attrs"].append( |
| ["OVS_NAT_ATTR_IP_MIN", ip_block_min] |
| ) |
| if ip_block_max is not None: |
| natact["attrs"].append( |
| ["OVS_NAT_ATTR_IP_MAX", ip_block_max] |
| ) |
| if proto_min is not None: |
| natact["attrs"].append( |
| ["OVS_NAT_ATTR_PROTO_MIN", proto_min] |
| ) |
| if proto_max is not None: |
| natact["attrs"].append( |
| ["OVS_NAT_ATTR_PROTO_MAX", proto_max] |
| ) |
| |
| for natscan in ( |
| ("persistent", "OVS_NAT_ATTR_PERSISTENT"), |
| ("hash", "OVS_NAT_ATTR_PROTO_HASH"), |
| ("random", "OVS_NAT_ATTR_PROTO_RANDOM"), |
| ): |
| if actstr.startswith(natscan[0]): |
| actstr = actstr[len(natscan[0]) :] |
| natact["attrs"].append([natscan[1], None]) |
| actstr = actstr[strspn(actstr, ", ") :] |
| |
| ctact["attrs"].append(["OVS_CT_ATTR_NAT", natact]) |
| actstr = actstr[strspn(actstr, ", ") :] |
| |
| self["attrs"].append(["OVS_ACTION_ATTR_CT", ctact]) |
| parsed = True |
| |
| actstr = actstr[strspn(actstr, ", ") :] |
| while parencount > 0: |
| parencount -= 1 |
| actstr = actstr[strspn(actstr, " "):] |
| if len(actstr) and actstr[0] != ")": |
| raise ValueError("Action str: '%s' unbalanced" % actstr) |
| actstr = actstr[1:] |
| |
| if len(actstr) and actstr[0] == ")": |
| return (totallen - len(actstr)) |
| |
| actstr = actstr[strspn(actstr, ", ") :] |
| |
| if not parsed: |
| raise ValueError("Action str: '%s' not supported" % actstr) |
| |
| return (totallen - len(actstr)) |
| |
| |
| class ovskey(nla): |
| nla_flags = NLA_F_NESTED |
| nla_map = ( |
| ("OVS_KEY_ATTR_UNSPEC", "none"), |
| ("OVS_KEY_ATTR_ENCAP", "none"), |
| ("OVS_KEY_ATTR_PRIORITY", "uint32"), |
| ("OVS_KEY_ATTR_IN_PORT", "uint32"), |
| ("OVS_KEY_ATTR_ETHERNET", "ethaddr"), |
| ("OVS_KEY_ATTR_VLAN", "uint16"), |
| ("OVS_KEY_ATTR_ETHERTYPE", "be16"), |
| ("OVS_KEY_ATTR_IPV4", "ovs_key_ipv4"), |
| ("OVS_KEY_ATTR_IPV6", "ovs_key_ipv6"), |
| ("OVS_KEY_ATTR_TCP", "ovs_key_tcp"), |
| ("OVS_KEY_ATTR_UDP", "ovs_key_udp"), |
| ("OVS_KEY_ATTR_ICMP", "ovs_key_icmp"), |
| ("OVS_KEY_ATTR_ICMPV6", "ovs_key_icmpv6"), |
| ("OVS_KEY_ATTR_ARP", "ovs_key_arp"), |
| ("OVS_KEY_ATTR_ND", "ovs_key_nd"), |
| ("OVS_KEY_ATTR_SKB_MARK", "uint32"), |
| ("OVS_KEY_ATTR_TUNNEL", "none"), |
| ("OVS_KEY_ATTR_SCTP", "ovs_key_sctp"), |
| ("OVS_KEY_ATTR_TCP_FLAGS", "be16"), |
| ("OVS_KEY_ATTR_DP_HASH", "uint32"), |
| ("OVS_KEY_ATTR_RECIRC_ID", "uint32"), |
| ("OVS_KEY_ATTR_MPLS", "array(ovs_key_mpls)"), |
| ("OVS_KEY_ATTR_CT_STATE", "uint32"), |
| ("OVS_KEY_ATTR_CT_ZONE", "uint16"), |
| ("OVS_KEY_ATTR_CT_MARK", "uint32"), |
| ("OVS_KEY_ATTR_CT_LABELS", "none"), |
| ("OVS_KEY_ATTR_CT_ORIG_TUPLE_IPV4", "ovs_key_ct_tuple_ipv4"), |
| ("OVS_KEY_ATTR_CT_ORIG_TUPLE_IPV6", "ovs_key_ct_tuple_ipv6"), |
| ("OVS_KEY_ATTR_NSH", "none"), |
| ("OVS_KEY_ATTR_PACKET_TYPE", "none"), |
| ("OVS_KEY_ATTR_ND_EXTENSIONS", "none"), |
| ("OVS_KEY_ATTR_TUNNEL_INFO", "none"), |
| ("OVS_KEY_ATTR_IPV6_EXTENSIONS", "none"), |
| ) |
| |
| class ovs_key_proto(nla): |
| fields = ( |
| ("src", "!H"), |
| ("dst", "!H"), |
| ) |
| |
| fields_map = ( |
| ("src", "src", "%d", lambda x: int(x) if x else 0, |
| convert_int(16)), |
| ("dst", "dst", "%d", lambda x: int(x) if x else 0, |
| convert_int(16)), |
| ) |
| |
| def __init__( |
| self, |
| protostr, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| self.proto_str = protostr |
| nla.__init__( |
| self, |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| def parse(self, flowstr, typeInst): |
| if not flowstr.startswith(self.proto_str): |
| return None, None |
| |
| k = typeInst() |
| m = typeInst() |
| |
| flowstr = flowstr[len(self.proto_str) :] |
| if flowstr.startswith("("): |
| flowstr = flowstr[1:] |
| |
| keybits = b"" |
| maskbits = b"" |
| for f in self.fields_map: |
| if flowstr.startswith(f[1]): |
| # the following assumes that the field looks |
| # something like 'field.' where '.' is a |
| # character that we don't exactly care about. |
| flowstr = flowstr[len(f[1]) + 1 :] |
| splitchar = 0 |
| for c in flowstr: |
| if c == "," or c == ")": |
| break |
| splitchar += 1 |
| data = flowstr[:splitchar] |
| flowstr = flowstr[splitchar:] |
| else: |
| data = "" |
| |
| if len(f) > 4: |
| k[f[0]], m[f[0]] = f[4](data) |
| else: |
| k[f[0]] = f[3](data) |
| m[f[0]] = f[3](data) |
| |
| flowstr = flowstr[strspn(flowstr, ", ") :] |
| if len(flowstr) == 0: |
| return flowstr, k, m |
| |
| flowstr = flowstr[strspn(flowstr, "), ") :] |
| |
| return flowstr, k, m |
| |
| def dpstr(self, masked=None, more=False): |
| outstr = self.proto_str + "(" |
| first = False |
| for f in self.fields_map: |
| if first: |
| outstr += "," |
| if masked is None: |
| outstr += "%s=" % f[0] |
| if isinstance(f[2], str): |
| outstr += f[2] % self[f[1]] |
| else: |
| outstr += f[2](self[f[1]]) |
| first = True |
| elif more or f[3](masked[f[1]]) != 0: |
| outstr += "%s=" % f[0] |
| if isinstance(f[2], str): |
| outstr += f[2] % self[f[1]] |
| else: |
| outstr += f[2](self[f[1]]) |
| outstr += "/" |
| if isinstance(f[2], str): |
| outstr += f[2] % masked[f[1]] |
| else: |
| outstr += f[2](masked[f[1]]) |
| first = True |
| outstr += ")" |
| return outstr |
| |
| class ethaddr(ovs_key_proto): |
| fields = ( |
| ("src", "!6s"), |
| ("dst", "!6s"), |
| ) |
| |
| fields_map = ( |
| ( |
| "src", |
| "src", |
| macstr, |
| lambda x: int.from_bytes(x, "big"), |
| convert_mac, |
| ), |
| ( |
| "dst", |
| "dst", |
| macstr, |
| lambda x: int.from_bytes(x, "big"), |
| convert_mac, |
| ), |
| ) |
| |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "eth", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_ipv4(ovs_key_proto): |
| fields = ( |
| ("src", "!I"), |
| ("dst", "!I"), |
| ("proto", "B"), |
| ("tos", "B"), |
| ("ttl", "B"), |
| ("frag", "B"), |
| ) |
| |
| fields_map = ( |
| ( |
| "src", |
| "src", |
| lambda x: str(ipaddress.IPv4Address(x)), |
| int, |
| convert_ipv4, |
| ), |
| ( |
| "dst", |
| "dst", |
| lambda x: str(ipaddress.IPv4Address(x)), |
| int, |
| convert_ipv4, |
| ), |
| ("proto", "proto", "%d", lambda x: int(x) if x else 0, |
| convert_int(8)), |
| ("tos", "tos", "%d", lambda x: int(x) if x else 0, |
| convert_int(8)), |
| ("ttl", "ttl", "%d", lambda x: int(x) if x else 0, |
| convert_int(8)), |
| ("frag", "frag", "%d", lambda x: int(x) if x else 0, |
| convert_int(8)), |
| ) |
| |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "ipv4", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_ipv6(ovs_key_proto): |
| fields = ( |
| ("src", "!16s"), |
| ("dst", "!16s"), |
| ("label", "!I"), |
| ("proto", "B"), |
| ("tclass", "B"), |
| ("hlimit", "B"), |
| ("frag", "B"), |
| ) |
| |
| fields_map = ( |
| ( |
| "src", |
| "src", |
| lambda x: str(ipaddress.IPv6Address(x)), |
| lambda x: int.from_bytes(x, "big"), |
| lambda x: ipaddress.IPv6Address(x), |
| ), |
| ( |
| "dst", |
| "dst", |
| lambda x: str(ipaddress.IPv6Address(x)), |
| lambda x: int.from_bytes(x, "big"), |
| lambda x: ipaddress.IPv6Address(x), |
| ), |
| ("label", "label", "%d", int), |
| ("proto", "proto", "%d", int), |
| ("tclass", "tclass", "%d", int), |
| ("hlimit", "hlimit", "%d", int), |
| ("frag", "frag", "%d", int), |
| ) |
| |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "ipv6", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_tcp(ovs_key_proto): |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "tcp", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_udp(ovs_key_proto): |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "udp", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_sctp(ovs_key_proto): |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "sctp", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_icmp(ovs_key_proto): |
| fields = ( |
| ("type", "B"), |
| ("code", "B"), |
| ) |
| |
| fields_map = ( |
| ("type", "type", "%d", lambda x: int(x) if x else 0), |
| ("code", "code", "%d", lambda x: int(x) if x else 0), |
| ) |
| |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "icmp", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_icmpv6(ovs_key_icmp): |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "icmpv6", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_arp(ovs_key_proto): |
| fields = ( |
| ("sip", "!I"), |
| ("tip", "!I"), |
| ("op", "!H"), |
| ("sha", "!6s"), |
| ("tha", "!6s"), |
| ("pad", "xx"), |
| ) |
| |
| fields_map = ( |
| ( |
| "sip", |
| "sip", |
| lambda x: str(ipaddress.IPv4Address(x)), |
| int, |
| convert_ipv4, |
| ), |
| ( |
| "tip", |
| "tip", |
| lambda x: str(ipaddress.IPv4Address(x)), |
| int, |
| convert_ipv4, |
| ), |
| ("op", "op", "%d", lambda x: int(x) if x else 0), |
| ( |
| "sha", |
| "sha", |
| macstr, |
| lambda x: int.from_bytes(x, "big"), |
| convert_mac, |
| ), |
| ( |
| "tha", |
| "tha", |
| macstr, |
| lambda x: int.from_bytes(x, "big"), |
| convert_mac, |
| ), |
| ) |
| |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "arp", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_nd(ovs_key_proto): |
| fields = ( |
| ("target", "!16s"), |
| ("sll", "!6s"), |
| ("tll", "!6s"), |
| ) |
| |
| fields_map = ( |
| ( |
| "target", |
| "target", |
| lambda x: str(ipaddress.IPv6Address(x)), |
| lambda x: int.from_bytes(x, "big"), |
| ), |
| ("sll", "sll", macstr, lambda x: int.from_bytes(x, "big")), |
| ("tll", "tll", macstr, lambda x: int.from_bytes(x, "big")), |
| ) |
| |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "nd", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_ct_tuple_ipv4(ovs_key_proto): |
| fields = ( |
| ("src", "!I"), |
| ("dst", "!I"), |
| ("tp_src", "!H"), |
| ("tp_dst", "!H"), |
| ("proto", "B"), |
| ) |
| |
| fields_map = ( |
| ( |
| "src", |
| "src", |
| lambda x: str(ipaddress.IPv4Address(x)), |
| int, |
| convert_ipv4, |
| ), |
| ( |
| "dst", |
| "dst", |
| lambda x: str(ipaddress.IPv4Address(x)), |
| int, |
| convert_ipv4, |
| ), |
| ("tp_src", "tp_src", "%d", int), |
| ("tp_dst", "tp_dst", "%d", int), |
| ("proto", "proto", "%d", int), |
| ) |
| |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "ct_tuple4", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_ct_tuple_ipv6(nla): |
| fields = ( |
| ("src", "!16s"), |
| ("dst", "!16s"), |
| ("tp_src", "!H"), |
| ("tp_dst", "!H"), |
| ("proto", "B"), |
| ) |
| |
| fields_map = ( |
| ( |
| "src", |
| "src", |
| lambda x: str(ipaddress.IPv6Address(x)), |
| lambda x: int.from_bytes(x, "big", convertmac), |
| ), |
| ( |
| "dst", |
| "dst", |
| lambda x: str(ipaddress.IPv6Address(x)), |
| lambda x: int.from_bytes(x, "big"), |
| ), |
| ("tp_src", "tp_src", "%d", int), |
| ("tp_dst", "tp_dst", "%d", int), |
| ("proto", "proto", "%d", int), |
| ) |
| |
| def __init__( |
| self, |
| data=None, |
| offset=None, |
| parent=None, |
| length=None, |
| init=None, |
| ): |
| ovskey.ovs_key_proto.__init__( |
| self, |
| "ct_tuple6", |
| data=data, |
| offset=offset, |
| parent=parent, |
| length=length, |
| init=init, |
| ) |
| |
| class ovs_key_mpls(nla): |
| fields = (("lse", ">I"),) |
| |
| def parse(self, flowstr, mask=None): |
| for field in ( |
| ("OVS_KEY_ATTR_PRIORITY", "skb_priority", intparse), |
| ("OVS_KEY_ATTR_SKB_MARK", "skb_mark", intparse), |
| ("OVS_KEY_ATTR_RECIRC_ID", "recirc_id", intparse), |
| ("OVS_KEY_ATTR_DP_HASH", "dp_hash", intparse), |
| ("OVS_KEY_ATTR_CT_STATE", "ct_state", parse_ct_state), |
| ("OVS_KEY_ATTR_CT_ZONE", "ct_zone", intparse), |
| ("OVS_KEY_ATTR_CT_MARK", "ct_mark", intparse), |
| ("OVS_KEY_ATTR_IN_PORT", "in_port", intparse), |
| ( |
| "OVS_KEY_ATTR_ETHERNET", |
| "eth", |
| ovskey.ethaddr, |
| ), |
| ( |
| "OVS_KEY_ATTR_ETHERTYPE", |
| "eth_type", |
| lambda x: intparse(x, "0xffff"), |
| ), |
| ( |
| "OVS_KEY_ATTR_IPV4", |
| "ipv4", |
| ovskey.ovs_key_ipv4, |
| ), |
| ( |
| "OVS_KEY_ATTR_IPV6", |
| "ipv6", |
| ovskey.ovs_key_ipv6, |
| ), |
| ( |
| "OVS_KEY_ATTR_ARP", |
| "arp", |
| ovskey.ovs_key_arp, |
| ), |
| ( |
| "OVS_KEY_ATTR_TCP", |
| "tcp", |
| ovskey.ovs_key_tcp, |
| ), |
| ( |
| "OVS_KEY_ATTR_UDP", |
| "udp", |
| ovskey.ovs_key_udp, |
| ), |
| ( |
| "OVS_KEY_ATTR_ICMP", |
| "icmp", |
| ovskey.ovs_key_icmp, |
| ), |
| ( |
| "OVS_KEY_ATTR_TCP_FLAGS", |
| "tcp_flags", |
| lambda x: parse_flags(x, None), |
| ), |
| ): |
| fld = field[1] + "(" |
| if not flowstr.startswith(fld): |
| continue |
| |
| if not isinstance(field[2], types.FunctionType): |
| nk = field[2]() |
| flowstr, k, m = nk.parse(flowstr, field[2]) |
| else: |
| flowstr = flowstr[len(fld) :] |
| flowstr, k, m = field[2](flowstr) |
| |
| if m and mask is not None: |
| mask["attrs"].append([field[0], m]) |
| self["attrs"].append([field[0], k]) |
| |
| flowstr = flowstr[strspn(flowstr, "),") :] |
| |
| return flowstr |
| |
| def dpstr(self, mask=None, more=False): |
| print_str = "" |
| |
| for field in ( |
| ( |
| "OVS_KEY_ATTR_PRIORITY", |
| "skb_priority", |
| "%d", |
| lambda x: False, |
| True, |
| ), |
| ( |
| "OVS_KEY_ATTR_SKB_MARK", |
| "skb_mark", |
| "%d", |
| lambda x: False, |
| True, |
| ), |
| ( |
| "OVS_KEY_ATTR_RECIRC_ID", |
| "recirc_id", |
| "0x%08X", |
| lambda x: False, |
| True, |
| ), |
| ( |
| "OVS_KEY_ATTR_DP_HASH", |
| "dp_hash", |
| "0x%08X", |
| lambda x: False, |
| True, |
| ), |
| ( |
| "OVS_KEY_ATTR_CT_STATE", |
| "ct_state", |
| "0x%04x", |
| lambda x: False, |
| True, |
| ), |
| ( |
| "OVS_KEY_ATTR_CT_ZONE", |
| "ct_zone", |
| "0x%04x", |
| lambda x: False, |
| True, |
| ), |
| ( |
| "OVS_KEY_ATTR_CT_MARK", |
| "ct_mark", |
| "0x%08x", |
| lambda x: False, |
| True, |
| ), |
| ( |
| "OVS_KEY_ATTR_CT_ORIG_TUPLE_IPV4", |
| None, |
| None, |
| False, |
| False, |
| ), |
| ( |
| "OVS_KEY_ATTR_CT_ORIG_TUPLE_IPV6", |
| None, |
| None, |
| False, |
| False, |
| ), |
| ( |
| "OVS_KEY_ATTR_IN_PORT", |
| "in_port", |
| "%d", |
| lambda x: True, |
| True, |
| ), |
| ("OVS_KEY_ATTR_ETHERNET", None, None, False, False), |
| ( |
| "OVS_KEY_ATTR_ETHERTYPE", |
| "eth_type", |
| "0x%04x", |
| lambda x: int(x) == 0xFFFF, |
| True, |
| ), |
| ("OVS_KEY_ATTR_IPV4", None, None, False, False), |
| ("OVS_KEY_ATTR_IPV6", None, None, False, False), |
| ("OVS_KEY_ATTR_ARP", None, None, False, False), |
| ("OVS_KEY_ATTR_TCP", None, None, False, False), |
| ( |
| "OVS_KEY_ATTR_TCP_FLAGS", |
| "tcp_flags", |
| "0x%04x", |
| lambda x: False, |
| True, |
| ), |
| ("OVS_KEY_ATTR_UDP", None, None, False, False), |
| ("OVS_KEY_ATTR_SCTP", None, None, False, False), |
| ("OVS_KEY_ATTR_ICMP", None, None, False, False), |
| ("OVS_KEY_ATTR_ICMPV6", None, None, False, False), |
| ("OVS_KEY_ATTR_ND", None, None, False, False), |
| ): |
| v = self.get_attr(field[0]) |
| if v is not None: |
| m = None if mask is None else mask.get_attr(field[0]) |
| if field[4] is False: |
| print_str += v.dpstr(m, more) |
| print_str += "," |
| else: |
| if m is None or field[3](m): |
| print_str += field[1] + "(" |
| print_str += field[2] % v |
| print_str += ")," |
| elif more or m != 0: |
| print_str += field[1] + "(" |
| print_str += (field[2] % v) + "/" + (field[2] % m) |
| print_str += ")," |
| |
| return print_str |
| |
| |
| class OvsPacket(GenericNetlinkSocket): |
| OVS_PACKET_CMD_MISS = 1 # Flow table miss |
| OVS_PACKET_CMD_ACTION = 2 # USERSPACE action |
| OVS_PACKET_CMD_EXECUTE = 3 # Apply actions to packet |
| |
| class ovs_packet_msg(ovs_dp_msg): |
| nla_map = ( |
| ("OVS_PACKET_ATTR_UNSPEC", "none"), |
| ("OVS_PACKET_ATTR_PACKET", "array(uint8)"), |
| ("OVS_PACKET_ATTR_KEY", "ovskey"), |
| ("OVS_PACKET_ATTR_ACTIONS", "ovsactions"), |
| ("OVS_PACKET_ATTR_USERDATA", "none"), |
| ("OVS_PACKET_ATTR_EGRESS_TUN_KEY", "none"), |
| ("OVS_PACKET_ATTR_UNUSED1", "none"), |
| ("OVS_PACKET_ATTR_UNUSED2", "none"), |
| ("OVS_PACKET_ATTR_PROBE", "none"), |
| ("OVS_PACKET_ATTR_MRU", "uint16"), |
| ("OVS_PACKET_ATTR_LEN", "uint32"), |
| ("OVS_PACKET_ATTR_HASH", "uint64"), |
| ) |
| |
| def __init__(self): |
| GenericNetlinkSocket.__init__(self) |
| self.bind(OVS_PACKET_FAMILY, OvsPacket.ovs_packet_msg) |
| |
| def upcall_handler(self, up=None): |
| print("listening on upcall packet handler:", self.epid) |
| while True: |
| try: |
| msgs = self.get() |
| for msg in msgs: |
| if not up: |
| continue |
| if msg["cmd"] == OvsPacket.OVS_PACKET_CMD_MISS: |
| up.miss(msg) |
| elif msg["cmd"] == OvsPacket.OVS_PACKET_CMD_ACTION: |
| up.action(msg) |
| elif msg["cmd"] == OvsPacket.OVS_PACKET_CMD_EXECUTE: |
| up.execute(msg) |
| else: |
| print("Unkonwn cmd: %d" % msg["cmd"]) |
| except NetlinkError as ne: |
| raise ne |
| |
| |
| class OvsDatapath(GenericNetlinkSocket): |
| OVS_DP_F_VPORT_PIDS = 1 << 1 |
| OVS_DP_F_DISPATCH_UPCALL_PER_CPU = 1 << 3 |
| |
| class dp_cmd_msg(ovs_dp_msg): |
| """ |
| Message class that will be used to communicate with the kernel module. |
| """ |
| |
| nla_map = ( |
| ("OVS_DP_ATTR_UNSPEC", "none"), |
| ("OVS_DP_ATTR_NAME", "asciiz"), |
| ("OVS_DP_ATTR_UPCALL_PID", "array(uint32)"), |
| ("OVS_DP_ATTR_STATS", "dpstats"), |
| ("OVS_DP_ATTR_MEGAFLOW_STATS", "megaflowstats"), |
| ("OVS_DP_ATTR_USER_FEATURES", "uint32"), |
| ("OVS_DP_ATTR_PAD", "none"), |
| ("OVS_DP_ATTR_MASKS_CACHE_SIZE", "uint32"), |
| ("OVS_DP_ATTR_PER_CPU_PIDS", "array(uint32)"), |
| ) |
| |
| class dpstats(nla): |
| fields = ( |
| ("hit", "=Q"), |
| ("missed", "=Q"), |
| ("lost", "=Q"), |
| ("flows", "=Q"), |
| ) |
| |
| class megaflowstats(nla): |
| fields = ( |
| ("mask_hit", "=Q"), |
| ("masks", "=I"), |
| ("padding", "=I"), |
| ("cache_hits", "=Q"), |
| ("pad1", "=Q"), |
| ) |
| |
| def __init__(self): |
| GenericNetlinkSocket.__init__(self) |
| self.bind(OVS_DATAPATH_FAMILY, OvsDatapath.dp_cmd_msg) |
| |
| def info(self, dpname, ifindex=0): |
| msg = OvsDatapath.dp_cmd_msg() |
| msg["cmd"] = OVS_DP_CMD_GET |
| msg["version"] = OVS_DATAPATH_VERSION |
| msg["reserved"] = 0 |
| msg["dpifindex"] = ifindex |
| msg["attrs"].append(["OVS_DP_ATTR_NAME", dpname]) |
| |
| try: |
| reply = self.nlm_request( |
| msg, msg_type=self.prid, msg_flags=NLM_F_REQUEST |
| ) |
| reply = reply[0] |
| except NetlinkError as ne: |
| if ne.code == errno.ENODEV: |
| reply = None |
| else: |
| raise ne |
| |
| return reply |
| |
| def create( |
| self, dpname, shouldUpcall=False, versionStr=None, p=OvsPacket() |
| ): |
| msg = OvsDatapath.dp_cmd_msg() |
| msg["cmd"] = OVS_DP_CMD_NEW |
| if versionStr is None: |
| msg["version"] = OVS_DATAPATH_VERSION |
| else: |
| msg["version"] = int(versionStr.split(":")[0], 0) |
| msg["reserved"] = 0 |
| msg["dpifindex"] = 0 |
| msg["attrs"].append(["OVS_DP_ATTR_NAME", dpname]) |
| |
| dpfeatures = 0 |
| if versionStr is not None and versionStr.find(":") != -1: |
| dpfeatures = int(versionStr.split(":")[1], 0) |
| else: |
| if versionStr is None or versionStr.find(":") == -1: |
| dpfeatures |= OvsDatapath.OVS_DP_F_DISPATCH_UPCALL_PER_CPU |
| dpfeatures &= ~OvsDatapath.OVS_DP_F_VPORT_PIDS |
| |
| nproc = multiprocessing.cpu_count() |
| procarray = [] |
| for i in range(1, nproc): |
| procarray += [int(p.epid)] |
| msg["attrs"].append(["OVS_DP_ATTR_UPCALL_PID", procarray]) |
| msg["attrs"].append(["OVS_DP_ATTR_USER_FEATURES", dpfeatures]) |
| if not shouldUpcall: |
| msg["attrs"].append(["OVS_DP_ATTR_UPCALL_PID", [0]]) |
| |
| try: |
| reply = self.nlm_request( |
| msg, msg_type=self.prid, msg_flags=NLM_F_REQUEST | NLM_F_ACK |
| ) |
| reply = reply[0] |
| except NetlinkError as ne: |
| if ne.code == errno.EEXIST: |
| reply = None |
| else: |
| raise ne |
| |
| return reply |
| |
| def destroy(self, dpname): |
| msg = OvsDatapath.dp_cmd_msg() |
| msg["cmd"] = OVS_DP_CMD_DEL |
| msg["version"] = OVS_DATAPATH_VERSION |
| msg["reserved"] = 0 |
| msg["dpifindex"] = 0 |
| msg["attrs"].append(["OVS_DP_ATTR_NAME", dpname]) |
| |
| try: |
| reply = self.nlm_request( |
| msg, msg_type=self.prid, msg_flags=NLM_F_REQUEST | NLM_F_ACK |
| ) |
| reply = reply[0] |
| except NetlinkError as ne: |
| if ne.code == errno.ENODEV: |
| reply = None |
| else: |
| raise ne |
| |
| return reply |
| |
| |
| class OvsVport(GenericNetlinkSocket): |
| OVS_VPORT_TYPE_NETDEV = 1 |
| OVS_VPORT_TYPE_INTERNAL = 2 |
| OVS_VPORT_TYPE_GRE = 3 |
| OVS_VPORT_TYPE_VXLAN = 4 |
| OVS_VPORT_TYPE_GENEVE = 5 |
| |
| class ovs_vport_msg(ovs_dp_msg): |
| nla_map = ( |
| ("OVS_VPORT_ATTR_UNSPEC", "none"), |
| ("OVS_VPORT_ATTR_PORT_NO", "uint32"), |
| ("OVS_VPORT_ATTR_TYPE", "uint32"), |
| ("OVS_VPORT_ATTR_NAME", "asciiz"), |
| ("OVS_VPORT_ATTR_OPTIONS", "none"), |
| ("OVS_VPORT_ATTR_UPCALL_PID", "array(uint32)"), |
| ("OVS_VPORT_ATTR_STATS", "vportstats"), |
| ("OVS_VPORT_ATTR_PAD", "none"), |
| ("OVS_VPORT_ATTR_IFINDEX", "uint32"), |
| ("OVS_VPORT_ATTR_NETNSID", "uint32"), |
| ) |
| |
| class vportstats(nla): |
| fields = ( |
| ("rx_packets", "=Q"), |
| ("tx_packets", "=Q"), |
| ("rx_bytes", "=Q"), |
| ("tx_bytes", "=Q"), |
| ("rx_errors", "=Q"), |
| ("tx_errors", "=Q"), |
| ("rx_dropped", "=Q"), |
| ("tx_dropped", "=Q"), |
| ) |
| |
| def type_to_str(vport_type): |
| if vport_type == OvsVport.OVS_VPORT_TYPE_NETDEV: |
| return "netdev" |
| elif vport_type == OvsVport.OVS_VPORT_TYPE_INTERNAL: |
| return "internal" |
| elif vport_type == OvsVport.OVS_VPORT_TYPE_GRE: |
| return "gre" |
| elif vport_type == OvsVport.OVS_VPORT_TYPE_VXLAN: |
| return "vxlan" |
| elif vport_type == OvsVport.OVS_VPORT_TYPE_GENEVE: |
| return "geneve" |
| raise ValueError("Unknown vport type:%d" % vport_type) |
| |
| def str_to_type(vport_type): |
| if vport_type == "netdev": |
| return OvsVport.OVS_VPORT_TYPE_NETDEV |
| elif vport_type == "internal": |
| return OvsVport.OVS_VPORT_TYPE_INTERNAL |
| elif vport_type == "gre": |
| return OvsVport.OVS_VPORT_TYPE_INTERNAL |
| elif vport_type == "vxlan": |
| return OvsVport.OVS_VPORT_TYPE_VXLAN |
| elif vport_type == "geneve": |
| return OvsVport.OVS_VPORT_TYPE_GENEVE |
| raise ValueError("Unknown vport type: '%s'" % vport_type) |
| |
| def __init__(self, packet=OvsPacket()): |
| GenericNetlinkSocket.__init__(self) |
| self.bind(OVS_VPORT_FAMILY, OvsVport.ovs_vport_msg) |
| self.upcall_packet = packet |
| |
| def info(self, vport_name, dpifindex=0, portno=None): |
| msg = OvsVport.ovs_vport_msg() |
| |
| msg["cmd"] = OVS_VPORT_CMD_GET |
| msg["version"] = OVS_DATAPATH_VERSION |
| msg["reserved"] = 0 |
| msg["dpifindex"] = dpifindex |
| |
| if portno is None: |
| msg["attrs"].append(["OVS_VPORT_ATTR_NAME", vport_name]) |
| else: |
| msg["attrs"].append(["OVS_VPORT_ATTR_PORT_NO", portno]) |
| |
| try: |
| reply = self.nlm_request( |
| msg, msg_type=self.prid, msg_flags=NLM_F_REQUEST |
| ) |
| reply = reply[0] |
| except NetlinkError as ne: |
| if ne.code == errno.ENODEV: |
| reply = None |
| else: |
| raise ne |
| return reply |
| |
| def attach(self, dpindex, vport_ifname, ptype): |
| msg = OvsVport.ovs_vport_msg() |
| |
| msg["cmd"] = OVS_VPORT_CMD_NEW |
| msg["version"] = OVS_DATAPATH_VERSION |
| msg["reserved"] = 0 |
| msg["dpifindex"] = dpindex |
| port_type = OvsVport.str_to_type(ptype) |
| |
| msg["attrs"].append(["OVS_VPORT_ATTR_TYPE", port_type]) |
| msg["attrs"].append(["OVS_VPORT_ATTR_NAME", vport_ifname]) |
| msg["attrs"].append( |
| ["OVS_VPORT_ATTR_UPCALL_PID", [self.upcall_packet.epid]] |
| ) |
| |
| try: |
| reply = self.nlm_request( |
| msg, msg_type=self.prid, msg_flags=NLM_F_REQUEST | NLM_F_ACK |
| ) |
| reply = reply[0] |
| except NetlinkError as ne: |
| if ne.code == errno.EEXIST: |
| reply = None |
| else: |
| raise ne |
| return reply |
| |
| def reset_upcall(self, dpindex, vport_ifname, p=None): |
| msg = OvsVport.ovs_vport_msg() |
| |
| msg["cmd"] = OVS_VPORT_CMD_SET |
| msg["version"] = OVS_DATAPATH_VERSION |
| msg["reserved"] = 0 |
| msg["dpifindex"] = dpindex |
| msg["attrs"].append(["OVS_VPORT_ATTR_NAME", vport_ifname]) |
| |
| if p == None: |
| p = self.upcall_packet |
| else: |
| self.upcall_packet = p |
| |
| msg["attrs"].append(["OVS_VPORT_ATTR_UPCALL_PID", [p.epid]]) |
| |
| try: |
| reply = self.nlm_request( |
| msg, msg_type=self.prid, msg_flags=NLM_F_REQUEST | NLM_F_ACK |
| ) |
| reply = reply[0] |
| except NetlinkError as ne: |
| raise ne |
| return reply |
| |
| def detach(self, dpindex, vport_ifname): |
| msg = OvsVport.ovs_vport_msg() |
| |
| msg["cmd"] = OVS_VPORT_CMD_DEL |
| msg["version"] = OVS_DATAPATH_VERSION |
| msg["reserved"] = 0 |
| msg["dpifindex"] = dpindex |
| msg["attrs"].append(["OVS_VPORT_ATTR_NAME", vport_ifname]) |
| |
| try: |
| reply = self.nlm_request( |
| msg, msg_type=self.prid, msg_flags=NLM_F_REQUEST | NLM_F_ACK |
| ) |
| reply = reply[0] |
| except NetlinkError as ne: |
| if ne.code == errno.ENODEV: |
| reply = None |
| else: |
| raise ne |
| return reply |
| |
| def upcall_handler(self, handler=None): |
| self.upcall_packet.upcall_handler(handler) |
| |
| |
| class OvsFlow(GenericNetlinkSocket): |
| class ovs_flow_msg(ovs_dp_msg): |
| nla_map = ( |
| ("OVS_FLOW_ATTR_UNSPEC", "none"), |
| ("OVS_FLOW_ATTR_KEY", "ovskey"), |
| ("OVS_FLOW_ATTR_ACTIONS", "ovsactions"), |
| ("OVS_FLOW_ATTR_STATS", "flowstats"), |
| ("OVS_FLOW_ATTR_TCP_FLAGS", "uint8"), |
| ("OVS_FLOW_ATTR_USED", "uint64"), |
| ("OVS_FLOW_ATTR_CLEAR", "none"), |
| ("OVS_FLOW_ATTR_MASK", "ovskey"), |
| ("OVS_FLOW_ATTR_PROBE", "none"), |
| ("OVS_FLOW_ATTR_UFID", "array(uint32)"), |
| ("OVS_FLOW_ATTR_UFID_FLAGS", "uint32"), |
| ) |
| |
| class flowstats(nla): |
| fields = ( |
| ("packets", "=Q"), |
| ("bytes", "=Q"), |
| ) |
| |
| def dpstr(self, more=False): |
| ufid = self.get_attr("OVS_FLOW_ATTR_UFID") |
| ufid_str = "" |
| if ufid is not None: |
| ufid_str = ( |
| "ufid:{:08x}-{:04x}-{:04x}-{:04x}-{:04x}{:08x}".format( |
| ufid[0], |
| ufid[1] >> 16, |
| ufid[1] & 0xFFFF, |
| ufid[2] >> 16, |
| ufid[2] & 0, |
| ufid[3], |
| ) |
| ) |
| |
| key_field = self.get_attr("OVS_FLOW_ATTR_KEY") |
| keymsg = None |
| if key_field is not None: |
| keymsg = key_field |
| |
| mask_field = self.get_attr("OVS_FLOW_ATTR_MASK") |
| maskmsg = None |
| if mask_field is not None: |
| maskmsg = mask_field |
| |
| acts_field = self.get_attr("OVS_FLOW_ATTR_ACTIONS") |
| actsmsg = None |
| if acts_field is not None: |
| actsmsg = acts_field |
| |
| print_str = "" |
| |
| if more: |
| print_str += ufid_str + "," |
| |
| if keymsg is not None: |
| print_str += keymsg.dpstr(maskmsg, more) |
| |
| stats = self.get_attr("OVS_FLOW_ATTR_STATS") |
| if stats is None: |
| print_str += " packets:0, bytes:0," |
| else: |
| print_str += " packets:%d, bytes:%d," % ( |
| stats["packets"], |
| stats["bytes"], |
| ) |
| |
| used = self.get_attr("OVS_FLOW_ATTR_USED") |
| print_str += " used:" |
| if used is None: |
| print_str += "never," |
| else: |
| used_time = int(used) |
| cur_time_sec = time.clock_gettime(time.CLOCK_MONOTONIC) |
| used_time = (cur_time_sec * 1000) - used_time |
| print_str += "{}s,".format(used_time / 1000) |
| |
| print_str += " actions:" |
| if ( |
| actsmsg is None |
| or "attrs" not in actsmsg |
| or len(actsmsg["attrs"]) == 0 |
| ): |
| print_str += "drop" |
| else: |
| print_str += actsmsg.dpstr(more) |
| |
| return print_str |
| |
| def parse(self, flowstr, actstr, dpidx=0): |
| OVS_UFID_F_OMIT_KEY = 1 << 0 |
| OVS_UFID_F_OMIT_MASK = 1 << 1 |
| OVS_UFID_F_OMIT_ACTIONS = 1 << 2 |
| |
| self["cmd"] = 0 |
| self["version"] = 0 |
| self["reserved"] = 0 |
| self["dpifindex"] = 0 |
| |
| if flowstr.startswith("ufid:"): |
| count = 5 |
| while flowstr[count] != ",": |
| count += 1 |
| ufidstr = flowstr[5:count] |
| flowstr = flowstr[count + 1 :] |
| else: |
| ufidstr = str(uuid.uuid4()) |
| uuidRawObj = uuid.UUID(ufidstr).fields |
| |
| self["attrs"].append( |
| [ |
| "OVS_FLOW_ATTR_UFID", |
| [ |
| uuidRawObj[0], |
| uuidRawObj[1] << 16 | uuidRawObj[2], |
| uuidRawObj[3] << 24 |
| | uuidRawObj[4] << 16 |
| | uuidRawObj[5] & (0xFF << 32) >> 32, |
| uuidRawObj[5] & (0xFFFFFFFF), |
| ], |
| ] |
| ) |
| self["attrs"].append( |
| [ |
| "OVS_FLOW_ATTR_UFID_FLAGS", |
| int( |
| OVS_UFID_F_OMIT_KEY |
| | OVS_UFID_F_OMIT_MASK |
| | OVS_UFID_F_OMIT_ACTIONS |
| ), |
| ] |
| ) |
| |
| k = ovskey() |
| m = ovskey() |
| k.parse(flowstr, m) |
| self["attrs"].append(["OVS_FLOW_ATTR_KEY", k]) |
| self["attrs"].append(["OVS_FLOW_ATTR_MASK", m]) |
| |
| a = ovsactions() |
| a.parse(actstr) |
| self["attrs"].append(["OVS_FLOW_ATTR_ACTIONS", a]) |
| |
| def __init__(self): |
| GenericNetlinkSocket.__init__(self) |
| |
| self.bind(OVS_FLOW_FAMILY, OvsFlow.ovs_flow_msg) |
| |
| def add_flow(self, dpifindex, flowmsg): |
| """ |
| Send a new flow message to the kernel. |
| |
| dpifindex should be a valid datapath obtained by calling |
| into the OvsDatapath lookup |
| |
| flowmsg is a flow object obtained by calling a dpparse |
| """ |
| |
| flowmsg["cmd"] = OVS_FLOW_CMD_NEW |
| flowmsg["version"] = OVS_DATAPATH_VERSION |
| flowmsg["reserved"] = 0 |
| flowmsg["dpifindex"] = dpifindex |
| |
| try: |
| reply = self.nlm_request( |
| flowmsg, |
| msg_type=self.prid, |
| msg_flags=NLM_F_REQUEST | NLM_F_ACK, |
| ) |
| reply = reply[0] |
| except NetlinkError as ne: |
| print(flowmsg) |
| raise ne |
| return reply |
| |
| def del_flows(self, dpifindex): |
| """ |
| Send a del message to the kernel that will drop all flows. |
| |
| dpifindex should be a valid datapath obtained by calling |
| into the OvsDatapath lookup |
| """ |
| |
| flowmsg = OvsFlow.ovs_flow_msg() |
| flowmsg["cmd"] = OVS_FLOW_CMD_DEL |
| flowmsg["version"] = OVS_DATAPATH_VERSION |
| flowmsg["reserved"] = 0 |
| flowmsg["dpifindex"] = dpifindex |
| |
| try: |
| reply = self.nlm_request( |
| flowmsg, |
| msg_type=self.prid, |
| msg_flags=NLM_F_REQUEST | NLM_F_ACK, |
| ) |
| reply = reply[0] |
| except NetlinkError as ne: |
| print(flowmsg) |
| raise ne |
| return reply |
| |
| def dump(self, dpifindex, flowspec=None): |
| """ |
| Returns a list of messages containing flows. |
| |
| dpifindex should be a valid datapath obtained by calling |
| into the OvsDatapath lookup |
| |
| flowpsec is a string which represents a flow in the dpctl |
| format. |
| """ |
| msg = OvsFlow.ovs_flow_msg() |
| |
| msg["cmd"] = OVS_FLOW_CMD_GET |
| msg["version"] = OVS_DATAPATH_VERSION |
| msg["reserved"] = 0 |
| msg["dpifindex"] = dpifindex |
| |
| msg_flags = NLM_F_REQUEST | NLM_F_ACK |
| if flowspec is None: |
| msg_flags |= NLM_F_DUMP |
| rep = None |
| |
| try: |
| rep = self.nlm_request( |
| msg, |
| msg_type=self.prid, |
| msg_flags=msg_flags, |
| ) |
| except NetlinkError as ne: |
| raise ne |
| return rep |
| |
| def miss(self, packetmsg): |
| seq = packetmsg["header"]["sequence_number"] |
| keystr = "(none)" |
| key_field = packetmsg.get_attr("OVS_PACKET_ATTR_KEY") |
| if key_field is not None: |
| keystr = key_field.dpstr(None, True) |
| |
| pktdata = packetmsg.get_attr("OVS_PACKET_ATTR_PACKET") |
| pktpres = "yes" if pktdata is not None else "no" |
| |
| print("MISS upcall[%d/%s]: %s" % (seq, pktpres, keystr), flush=True) |
| |
| def execute(self, packetmsg): |
| print("userspace execute command") |
| |
| def action(self, packetmsg): |
| print("userspace action command") |
| |
| |
| def print_ovsdp_full(dp_lookup_rep, ifindex, ndb=NDB(), vpl=OvsVport()): |
| dp_name = dp_lookup_rep.get_attr("OVS_DP_ATTR_NAME") |
| base_stats = dp_lookup_rep.get_attr("OVS_DP_ATTR_STATS") |
| megaflow_stats = dp_lookup_rep.get_attr("OVS_DP_ATTR_MEGAFLOW_STATS") |
| user_features = dp_lookup_rep.get_attr("OVS_DP_ATTR_USER_FEATURES") |
| masks_cache_size = dp_lookup_rep.get_attr("OVS_DP_ATTR_MASKS_CACHE_SIZE") |
| |
| print("%s:" % dp_name) |
| print( |
| " lookups: hit:%d missed:%d lost:%d" |
| % (base_stats["hit"], base_stats["missed"], base_stats["lost"]) |
| ) |
| print(" flows:%d" % base_stats["flows"]) |
| pkts = base_stats["hit"] + base_stats["missed"] |
| avg = (megaflow_stats["mask_hit"] / pkts) if pkts != 0 else 0.0 |
| print( |
| " masks: hit:%d total:%d hit/pkt:%f" |
| % (megaflow_stats["mask_hit"], megaflow_stats["masks"], avg) |
| ) |
| print(" caches:") |
| print(" masks-cache: size:%d" % masks_cache_size) |
| |
| if user_features is not None: |
| print(" features: 0x%X" % user_features) |
| |
| # port print out |
| for iface in ndb.interfaces: |
| rep = vpl.info(iface.ifname, ifindex) |
| if rep is not None: |
| print( |
| " port %d: %s (%s)" |
| % ( |
| rep.get_attr("OVS_VPORT_ATTR_PORT_NO"), |
| rep.get_attr("OVS_VPORT_ATTR_NAME"), |
| OvsVport.type_to_str(rep.get_attr("OVS_VPORT_ATTR_TYPE")), |
| ) |
| ) |
| |
| |
| def main(argv): |
| nlmsg_atoms.ovskey = ovskey |
| nlmsg_atoms.ovsactions = ovsactions |
| |
| # version check for pyroute2 |
| prverscheck = pyroute2.__version__.split(".") |
| if int(prverscheck[0]) == 0 and int(prverscheck[1]) < 6: |
| print("Need to upgrade the python pyroute2 package to >= 0.6.") |
| sys.exit(0) |
| |
| parser = argparse.ArgumentParser() |
| parser.add_argument( |
| "-v", |
| "--verbose", |
| action="count", |
| help="Increment 'verbose' output counter.", |
| default=0, |
| ) |
| subparsers = parser.add_subparsers() |
| |
| showdpcmd = subparsers.add_parser("show") |
| showdpcmd.add_argument( |
| "showdp", metavar="N", type=str, nargs="?", help="Datapath Name" |
| ) |
| |
| adddpcmd = subparsers.add_parser("add-dp") |
| adddpcmd.add_argument("adddp", help="Datapath Name") |
| adddpcmd.add_argument( |
| "-u", |
| "--upcall", |
| action="store_true", |
| help="Leave open a reader for upcalls", |
| ) |
| adddpcmd.add_argument( |
| "-V", |
| "--versioning", |
| required=False, |
| help="Specify a custom version / feature string", |
| ) |
| |
| deldpcmd = subparsers.add_parser("del-dp") |
| deldpcmd.add_argument("deldp", help="Datapath Name") |
| |
| addifcmd = subparsers.add_parser("add-if") |
| addifcmd.add_argument("dpname", help="Datapath Name") |
| addifcmd.add_argument("addif", help="Interface name for adding") |
| addifcmd.add_argument( |
| "-u", |
| "--upcall", |
| action="store_true", |
| help="Leave open a reader for upcalls", |
| ) |
| addifcmd.add_argument( |
| "-t", |
| "--ptype", |
| type=str, |
| default="netdev", |
| choices=["netdev", "internal"], |
| help="Interface type (default netdev)", |
| ) |
| delifcmd = subparsers.add_parser("del-if") |
| delifcmd.add_argument("dpname", help="Datapath Name") |
| delifcmd.add_argument("delif", help="Interface name for adding") |
| |
| dumpflcmd = subparsers.add_parser("dump-flows") |
| dumpflcmd.add_argument("dumpdp", help="Datapath Name") |
| |
| addflcmd = subparsers.add_parser("add-flow") |
| addflcmd.add_argument("flbr", help="Datapath name") |
| addflcmd.add_argument("flow", help="Flow specification") |
| addflcmd.add_argument("acts", help="Flow actions") |
| |
| delfscmd = subparsers.add_parser("del-flows") |
| delfscmd.add_argument("flsbr", help="Datapath name") |
| |
| args = parser.parse_args() |
| |
| if args.verbose > 0: |
| if args.verbose > 1: |
| logging.basicConfig(level=logging.DEBUG) |
| |
| ovspk = OvsPacket() |
| ovsdp = OvsDatapath() |
| ovsvp = OvsVport(ovspk) |
| ovsflow = OvsFlow() |
| ndb = NDB() |
| |
| sys.setrecursionlimit(100000) |
| |
| if hasattr(args, "showdp"): |
| found = False |
| for iface in ndb.interfaces: |
| rep = None |
| if args.showdp is None: |
| rep = ovsdp.info(iface.ifname, 0) |
| elif args.showdp == iface.ifname: |
| rep = ovsdp.info(iface.ifname, 0) |
| |
| if rep is not None: |
| found = True |
| print_ovsdp_full(rep, iface.index, ndb, ovsvp) |
| |
| if not found: |
| msg = "No DP found" |
| if args.showdp is not None: |
| msg += ":'%s'" % args.showdp |
| print(msg) |
| elif hasattr(args, "adddp"): |
| rep = ovsdp.create(args.adddp, args.upcall, args.versioning, ovspk) |
| if rep is None: |
| print("DP '%s' already exists" % args.adddp) |
| else: |
| print("DP '%s' added" % args.adddp) |
| if args.upcall: |
| ovspk.upcall_handler(ovsflow) |
| elif hasattr(args, "deldp"): |
| ovsdp.destroy(args.deldp) |
| elif hasattr(args, "addif"): |
| rep = ovsdp.info(args.dpname, 0) |
| if rep is None: |
| print("DP '%s' not found." % args.dpname) |
| return 1 |
| dpindex = rep["dpifindex"] |
| rep = ovsvp.attach(rep["dpifindex"], args.addif, args.ptype) |
| msg = "vport '%s'" % args.addif |
| if rep and rep["header"]["error"] is None: |
| msg += " added." |
| else: |
| msg += " failed to add." |
| if args.upcall: |
| if rep is None: |
| rep = ovsvp.reset_upcall(dpindex, args.addif, ovspk) |
| ovsvp.upcall_handler(ovsflow) |
| elif hasattr(args, "delif"): |
| rep = ovsdp.info(args.dpname, 0) |
| if rep is None: |
| print("DP '%s' not found." % args.dpname) |
| return 1 |
| rep = ovsvp.detach(rep["dpifindex"], args.delif) |
| msg = "vport '%s'" % args.delif |
| if rep and rep["header"]["error"] is None: |
| msg += " removed." |
| else: |
| msg += " failed to remove." |
| elif hasattr(args, "dumpdp"): |
| rep = ovsdp.info(args.dumpdp, 0) |
| if rep is None: |
| print("DP '%s' not found." % args.dumpdp) |
| return 1 |
| rep = ovsflow.dump(rep["dpifindex"]) |
| for flow in rep: |
| print(flow.dpstr(True if args.verbose > 0 else False)) |
| elif hasattr(args, "flbr"): |
| rep = ovsdp.info(args.flbr, 0) |
| if rep is None: |
| print("DP '%s' not found." % args.flbr) |
| return 1 |
| flow = OvsFlow.ovs_flow_msg() |
| flow.parse(args.flow, args.acts, rep["dpifindex"]) |
| ovsflow.add_flow(rep["dpifindex"], flow) |
| elif hasattr(args, "flsbr"): |
| rep = ovsdp.info(args.flsbr, 0) |
| if rep is None: |
| print("DP '%s' not found." % args.flsbr) |
| ovsflow.del_flows(rep["dpifindex"]) |
| |
| return 0 |
| |
| |
| if __name__ == "__main__": |
| sys.exit(main(sys.argv)) |
