16e40ab0ab40d27950ef70e79b35f60c888c4ba6,networkml/featurizers/funcs/host.py,HostBase,_tshark_ratio_ports,#HostBase#Any#,138

Before Change

                continue
            src = proto_df["%s.srcport" % ip_proto]
            dst = proto_df["%s.dstport" % ip_proto]
            for field_name, wk_ports in (
                    ("priv", self.WK_PRIV_TCPUDP_PORTS),
                    ("nonpriv", self.WK_NONPRIV_TCPUDP_PORTS),
                ):
                src_counts = src[src.isin(wk_ports)].value_counts()
                dst_counts = dst[dst.isin(wk_ports)].value_counts()
                for port in wk_ports:
                    src_count = src_counts.get(port, None)
                    dst_count = dst_counts.get(port, None)

After Change

                    dst_count = dst_counts.get(port, None)
                    mac_row_ports.update({
                        "tshark_%s_%s_packet_ratio_io_port_%s" % (ip_proto, field_name, port): calc_ratio(src_count, dst_count)})
                src_count = port_src[~port_src.isin(wk_ports)].value_counts().sum()
                dst_count = port_dst[~port_dst.isin(wk_ports)].value_counts().sum()
                mac_row_ports.update({
                    "tshark_%s_%s_packet_ratio_io_port_%s" % (ip_proto, field_name, "other"): calc_ratio(src_count, dst_count)})

Instances