From 7942b23e79567d74b6b9d8796ab15f985f6926e2 Mon Sep 17 00:00:00 2001
From: Michael Lumish <mlumish@google.com>
Date: Mon, 10 Oct 2022 14:11:16 -0700
Subject: [PATCH] grpc-js-xds: Validate that endpoint weights sum to no more
 than 32 bit uint max per priority

---
 packages/grpc-js-xds/src/xds-stream-state/eds-state.ts | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/packages/grpc-js-xds/src/xds-stream-state/eds-state.ts b/packages/grpc-js-xds/src/xds-stream-state/eds-state.ts
index 91cb6f30..dd1adf18 100644
--- a/packages/grpc-js-xds/src/xds-stream-state/eds-state.ts
+++ b/packages/grpc-js-xds/src/xds-stream-state/eds-state.ts
@@ -50,6 +50,7 @@ export class EdsState extends BaseXdsStreamState<ClusterLoadAssignment__Output>
    */
   public validateResponse(message: ClusterLoadAssignment__Output) {
     const seenLocalities: {locality: Locality__Output, priority: number}[] = [];
+    const priorityTotalWeights: Map<number,  number> = new Map();
     for (const endpoint of message.endpoints) {
       if (!endpoint.locality) {
         return false;
@@ -72,6 +73,12 @@ export class EdsState extends BaseXdsStreamState<ClusterLoadAssignment__Output>
           return false;
         }
       }
+      priorityTotalWeights.set(endpoint.priority, (priorityTotalWeights.get(endpoint.priority) ?? 0) + (endpoint.load_balancing_weight?.value ?? 0));
+    }
+    for (const totalWeight of priorityTotalWeights.values()) {
+      if (totalWeight >= 1<<32) {
+        return false;
+      }
     }
     return true;
   }