cd83fed56a184063215f932ef768f8834328654c,networkml/featurizers/funcs/host.py,HostBase,_df_proto_flags,#HostBase#,239

Before Change

        all_protos = short_row_keys.union(short_frame_protocols) - self.DROP_PROTOS
        all_protos_int = 0
        for proto in all_protos.intersection(WK_IP_PROTOS):
            index = WK_IP_PROTOS.index(proto)
            all_protos_int += 2**index
        return all_protos_int

    def _tshark_all(self, df, srcmacid):
        print("calculating intermediates", end="", flush=True)
        df["_host_key"], df["_srcip"], df["_dstip"], df["_both_private_ip"], df["_ipv4_multicast"], df["_protos_int"] = zip(*df.apply(self._host_key, axis=1))

After Change

    def _df_proto_flags(self, row):
        short_row_keys = frozenset(x.split(".")[0] for x, y in row.items() if not pd.isnull(y) and not x.startswith("_"))
        return self._encode_df_proto_flags(short_row_keys, row["frame.protocols"])

    def _tshark_all(self, df, srcmacid):
        print("calculating intermediates", end="", flush=True)
        df["_host_key"], df["_srcip"], df["_dstip"], df["_both_private_ip"], df["_ipv4_multicast"], df["_protos_int"] = zip(*df.apply(self._host_key, axis=1))

Instances