feat: add network: endpoint resolver, VM.Config.Options privilege requirement, top talkers dashboard widget, and automatic guest firewall enablement
Add network: prefix support in endpoint_values to resolve network names to IPAM subnet CIDRs for policy matching, extend workload_provider_target to accept vmid: prefix and bare workload names as fallback resolution methods, implement dashboard_top_talkers to aggregate traffic flows by workload with interface_traffic fallback when flows unavailable, add
This commit is contained in:
@@ -202,10 +202,15 @@ def import_discovered_ips(db: Session, cluster_id: str, workload: Workload, addr
|
||||
|
||||
|
||||
def workload_provider_target(db: Session, cluster: Cluster, ref: str) -> tuple[dict, Workload] | None:
|
||||
if not ref.startswith("workload:"):
|
||||
return None
|
||||
workload_id = ref.removeprefix("workload:")
|
||||
workload = db.get(Workload, workload_id)
|
||||
workload: Workload | None = None
|
||||
if ref.startswith("workload:"):
|
||||
workload_ref = ref.removeprefix("workload:")
|
||||
workload = db.get(Workload, workload_ref)
|
||||
elif ref.startswith("vmid:"):
|
||||
workload_ref = ref.removeprefix("vmid:")
|
||||
workload = db.scalar(select(Workload).where(Workload.cluster_id == cluster.id, Workload.external_id == workload_ref))
|
||||
else:
|
||||
workload = db.scalar(select(Workload).where(Workload.cluster_id == cluster.id, Workload.name == ref))
|
||||
if not workload or workload.cluster_id != cluster.id:
|
||||
return None
|
||||
node = db.get(Node, workload.node_id)
|
||||
@@ -237,6 +242,15 @@ def endpoint_values(db: Session, cluster: Cluster, ref: str) -> tuple[list[str |
|
||||
return [], [f"Workload {workload.name} has no assigned IP address for provider-side source/destination matching."]
|
||||
if is_ip_or_cidr(ref):
|
||||
return [ref], []
|
||||
if ref.startswith("network:"):
|
||||
network_name = ref.removeprefix("network:")
|
||||
network = db.scalar(select(Network).where(Network.cluster_id == cluster.id, Network.name == network_name))
|
||||
if not network:
|
||||
return [], [f"Network {network_name} was not found in this cluster."]
|
||||
values = [subnet.cidr for subnet in db.scalars(select(Subnet).where(Subnet.network_id == network.id).order_by(Subnet.cidr)).all()]
|
||||
if values:
|
||||
return values, []
|
||||
return [], [f"Network {network_name} has no IPAM subnets to use as provider-side matcher."]
|
||||
return [], [f"Endpoint {ref} is not yet resolvable to a Proxmox firewall matcher."]
|
||||
|
||||
|
||||
@@ -272,6 +286,48 @@ def flow_endpoint_label(owner: Workload | None, subnets: list[Subnet], value: st
|
||||
return subnet_label_for_ip(subnets, value) or "external"
|
||||
|
||||
|
||||
def dashboard_top_talkers(db: Session) -> list[dict[str, int | str]]:
|
||||
totals: dict[str, int] = {}
|
||||
workloads = db.scalars(select(Workload)).all()
|
||||
workloads_by_node_vmid = {(workload.node_id, workload.external_id): workload for workload in workloads}
|
||||
ip_owners = {
|
||||
address.address: db.get(Workload, address.workload_id)
|
||||
for address in db.scalars(select(IpAddress).where(IpAddress.workload_id.is_not(None))).all()
|
||||
}
|
||||
|
||||
flows = db.scalars(select(TrafficFlow)).all()
|
||||
for flow in flows:
|
||||
raw = flow.raw if isinstance(flow.raw, dict) else {}
|
||||
candidates: list[Workload] = []
|
||||
raw_vmid = str(raw.get("vmid") or "")
|
||||
if raw_vmid:
|
||||
workload = workloads_by_node_vmid.get((flow.node_id, raw_vmid))
|
||||
if workload:
|
||||
candidates.append(workload)
|
||||
for owner in (ip_owners.get(flow.source_ip), ip_owners.get(flow.destination_ip)):
|
||||
if owner and owner not in candidates:
|
||||
candidates.append(owner)
|
||||
for workload in candidates:
|
||||
totals[workload.name] = totals.get(workload.name, 0) + int(flow.bytes or 0)
|
||||
|
||||
if not totals:
|
||||
agents = db.scalars(select(NodeAgent)).all()
|
||||
for agent in agents:
|
||||
payload = agent.last_payload if isinstance(agent.last_payload, dict) else {}
|
||||
for item in payload.get("interface_traffic", []):
|
||||
if not isinstance(item, dict):
|
||||
continue
|
||||
workload = workloads_by_node_vmid.get((agent.node_id, str(item.get("vmid") or "")))
|
||||
if not workload:
|
||||
continue
|
||||
totals[workload.name] = totals.get(workload.name, 0) + int(item.get("bytes") or 0)
|
||||
|
||||
return [
|
||||
{"name": name, "bytes": bytes_value}
|
||||
for name, bytes_value in sorted(totals.items(), key=lambda item: item[1], reverse=True)[:5]
|
||||
]
|
||||
|
||||
|
||||
def proxmox_action(action: str) -> str:
|
||||
return {"allow": "ACCEPT", "deny": "DROP", "reject": "REJECT"}.get(action, "ACCEPT")
|
||||
|
||||
@@ -414,7 +470,7 @@ def dashboard(_: CurrentUser, db: Session = Depends(get_db)) -> dict:
|
||||
{"id": node.id, "name": node.name, "status": node.status, "cluster_id": node.cluster_id}
|
||||
for node in faulty_nodes
|
||||
],
|
||||
"top_talkers": [],
|
||||
"top_talkers": dashboard_top_talkers(db),
|
||||
}
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user