grpc 1.76.0 → 1.78.0.pre1

data/src/core/ext/transport/chttp2/transport/writable_streams.h

@@ -19,15 +19,22 @@
 #ifndef GRPC_SRC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_WRITABLE_STREAMS_H
 #define GRPC_SRC_CORE_EXT_TRANSPORT_CHTTP2_TRANSPORT_WRITABLE_STREAMS_H
 
+#include <cstdint>
+#include <limits>
+#include <optional>
 #include <queue>
+#include <vector>
 
-#include "absl/log/log.h"
 #include "src/core/ext/transport/chttp2/transport/transport_common.h"
 #include "src/core/lib/promise/if.h"
 #include "src/core/lib/promise/mpsc.h"
 #include "src/core/lib/promise/race.h"
+#include "src/core/lib/promise/status_flag.h"
 #include "src/core/lib/promise/try_seq.h"
 #include "src/core/util/grpc_check.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
+#include "absl/strings/str_cat.h"
 
 namespace grpc_core {
 namespace http2 {
@@ -67,6 +74,16 @@ class WritableStreams {
   WritableStreams(WritableStreams&&) = delete;
   WritableStreams& operator=(WritableStreams&&) = delete;
 
+  absl::Status EnqueueWrapper(const StreamPtr stream,
+                              const WritableStreamPriority priority,
+                              bool transport_tokens_available) {
+    if (transport_tokens_available) {
+      return Enqueue(stream, priority);
+    } else {
+      return BlockedOnTransportFlowControl(stream);
+    }
+  }
+
   // Enqueues a stream id with the given priority.
   // If this returns error, transport MUST be closed.
   absl::Status Enqueue(const StreamPtr stream,
@@ -81,14 +98,13 @@ class WritableStreams {
     StatusFlag status = sender_.UnbufferedImmediateSend(
         StreamIDAndPriority{stream, priority}, /*tokens*/ 1);
     GRPC_WRITABLE_STREAMS_DEBUG
-        << "UnbufferedImmediateEnqueue stream id: " << stream->GetStreamId()
-        << " with priority " << GetWritableStreamPriorityString(priority)
-        << " status " << status;
+        << "UnbufferedImmediateEnqueue stream with priority "
+        << GetWritableStreamPriorityString(priority) << " status " << status;
 
     return (status.ok())
                ? absl::OkStatus()
                : absl::InternalError(absl::StrCat(
-                     "Failed to enqueue stream id: ", stream->GetStreamId()));
+                     "Failed to enqueue stream to list of writable streams "));
   }
 
   // A synchronous function to add a stream id to the transport flow control
@@ -96,8 +112,8 @@ class WritableStreams {
   absl::Status BlockedOnTransportFlowControl(const StreamPtr stream) {
     prioritized_queue_.Push(
         stream, WritableStreamPriority::kWaitForTransportFlowControl);
-    GRPC_WRITABLE_STREAMS_DEBUG << "BlockedOnTransportFlowControl stream id: "
-                                << stream->GetStreamId();
+    GRPC_WRITABLE_STREAMS_DEBUG << "Enqueuing a stream with priority "
+                                   "kWaitForTransportFlowControl ";
     return absl::OkStatus();
   }
 
@@ -161,10 +177,10 @@ class WritableStreams {
                         AddToPrioritizedQueue(batch.value());
                         std::optional<StreamPtr> stream =
                             prioritized_queue_.Pop(transport_tokens_available);
-                        // TODO(akshitpatel) : [PH2][P4] - This DCHECK should
-                        // ideally be fine. But in case if queue_.NextBatch
-                        // spuriously returns an empty batch, move to a Loop
-                        // to avoid this.
+                        // TODO(akshitpatel) : [PH2][P4] - This GRPC_DCHECK
+                        // should ideally be fine. But in case if
+                        // queue_.NextBatch spuriously returns an empty batch,
+                        // move to a Loop to avoid this.
                         GRPC_DCHECK(stream.has_value());
                         GRPC_WRITABLE_STREAMS_DEBUG
                             << "Next stream id: "

data/src/core/ext/transport/chttp2/transport/writing.cc

@@ -30,9 +30,6 @@
 #include <string>
 #include <utility>
 
-#include "absl/container/flat_hash_map.h"
-#include "absl/log/log.h"
-#include "absl/status/status.h"
 #include "src/core/call/metadata_batch.h"
 #include "src/core/channelz/channelz.h"
 #include "src/core/ext/transport/chttp2/transport/call_tracer_wrapper.h"
@@ -75,6 +72,9 @@
 #include "src/core/util/shared_bit_gen.h"
 #include "src/core/util/time.h"
 #include "src/core/util/useful.h"
+#include "absl/container/flat_hash_map.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
 
 // IWYU pragma: no_include "src/core/util/orphanable.h"
 

data/src/core/ext/transport/inproc/inproc_transport.cc

@@ -20,8 +20,6 @@
 #include <atomic>
 #include <memory>
 
-#include "absl/log/log.h"
-#include "absl/status/status.h"
 #include "src/core/call/metadata.h"
 #include "src/core/config/core_configuration.h"
 #include "src/core/ext/transport/inproc/legacy_inproc_transport.h"
@@ -36,6 +34,8 @@
 #include "src/core/util/crash.h"
 #include "src/core/util/debug_location.h"
 #include "src/core/util/grpc_check.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
 
 namespace grpc_core {
 
@@ -94,6 +94,9 @@ class InprocServerTransport final : public ServerTransport {
     ExecCtx::Run(DEBUG_LOCATION, op->on_consumed, absl::OkStatus());
   }
 
+  void StartWatch(RefCountedPtr<StateWatcher>) override {}
+  void StopWatch(RefCountedPtr<StateWatcher>) override {}
+
   void Disconnect(absl::Status error) {
     RefCountedPtr<ConnectedState> connected_state;
     {
@@ -222,6 +225,9 @@ class InprocClientTransport final : public ClientTransport {
   void SetPollsetSet(grpc_stream*, grpc_pollset_set*) override {}
   void PerformOp(grpc_transport_op*) override { Crash("unimplemented"); }
 
+  void StartWatch(RefCountedPtr<StateWatcher>) override {}
+  void StopWatch(RefCountedPtr<StateWatcher>) override {}
+
  private:
   ~InprocClientTransport() override {
     server_transport_->Disconnect(

data/src/core/ext/transport/inproc/legacy_inproc_transport.cc

@@ -35,11 +35,6 @@
 #include <string>
 #include <utility>
 
-#include "absl/log/log.h"
-#include "absl/status/status.h"
-#include "absl/status/statusor.h"
-#include "absl/strings/str_cat.h"
-#include "absl/strings/string_view.h"
 #include "src/core/call/metadata_batch.h"
 #include "src/core/channelz/channelz.h"
 #include "src/core/config/core_configuration.h"
@@ -64,6 +59,11 @@
 #include "src/core/util/ref_counted_ptr.h"
 #include "src/core/util/status_helper.h"
 #include "src/core/util/time.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
+#include "absl/status/statusor.h"
+#include "absl/strings/str_cat.h"
+#include "absl/strings/string_view.h"
 
 namespace {
 struct inproc_stream;
@@ -139,6 +139,9 @@ struct inproc_transport final : public grpc_core::FilterStackTransport {
   void DestroyStream(grpc_stream* gs,
                      grpc_closure* then_schedule_closure) override;
 
+  void StartWatch(grpc_core::RefCountedPtr<StateWatcher>) override {}
+  void StopWatch(grpc_core::RefCountedPtr<StateWatcher>) override {}
+
   void Orphan() override;
 
   void ref() {

data/src/core/filter/auth/auth_filters.h

@@ -24,7 +24,6 @@
 #include <grpc/grpc_security_constants.h>
 #include <grpc/support/port_platform.h>
 
-#include "absl/status/statusor.h"
 #include "src/core/call/status_util.h"
 #include "src/core/credentials/call/call_credentials.h"
 #include "src/core/credentials/transport/security_connector.h"
@@ -36,6 +35,7 @@
 #include "src/core/lib/promise/arena_promise.h"
 #include "src/core/lib/transport/transport.h"
 #include "src/core/util/ref_counted_ptr.h"
+#include "absl/status/statusor.h"
 
 namespace grpc_core {
 
@@ -117,6 +117,9 @@ class ClientAuthFilter final : public ImplementChannelFilter<ClientAuthFilter> {
     static const inline NoInterceptor OnServerToClientMessage;
     static const inline NoInterceptor OnServerTrailingMetadata;
     static const inline NoInterceptor OnFinalize;
+    channelz::PropertyList ChannelzProperties() {
+      return channelz::PropertyList();
+    }
   };
 
  private:
@@ -186,6 +189,9 @@ class ServerAuthFilter final : public ImplementChannelFilter<ServerAuthFilter> {
     static inline const NoInterceptor OnServerToClientMessage;
     static inline const NoInterceptor OnServerTrailingMetadata;
     static inline const NoInterceptor OnFinalize;
+    channelz::PropertyList ChannelzProperties() {
+      return channelz::PropertyList();
+    }
   };
 
  private:

data/src/core/filter/auth/client_auth_filter.cc

@@ -29,8 +29,6 @@
 #include <type_traits>  // IWYU pragma: keep
 #include <utility>
 
-#include "absl/status/status.h"
-#include "absl/status/statusor.h"
 #include "src/core/call/metadata_batch.h"
 #include "src/core/call/security_context.h"
 #include "src/core/call/status_util.h"
@@ -51,6 +49,8 @@
 #include "src/core/transport/auth_context.h"
 #include "src/core/util/debug_location.h"
 #include "src/core/util/ref_counted_ptr.h"
+#include "absl/status/status.h"
+#include "absl/status/statusor.h"
 
 #define MAX_CREDENTIALS_METADATA_COUNT 4
 

data/src/core/filter/auth/server_auth_filter.cc

@@ -30,9 +30,6 @@
 #include <memory>
 #include <utility>
 
-#include "absl/log/log.h"
-#include "absl/status/status.h"
-#include "absl/status/statusor.h"
 #include "src/core/call/metadata_batch.h"
 #include "src/core/call/security_context.h"
 #include "src/core/credentials/transport/transport_credentials.h"
@@ -58,6 +55,9 @@
 #include "src/core/util/grpc_check.h"
 #include "src/core/util/ref_counted_ptr.h"
 #include "src/core/util/status_helper.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
+#include "absl/status/statusor.h"
 
 namespace grpc_core {
 

data/src/core/filter/blackboard.h

@@ -20,14 +20,14 @@
 #include <string>
 #include <utility>
 
-#include "absl/container/flat_hash_map.h"
-#include "absl/strings/string_view.h"
 #include "src/core/resolver/endpoint_addresses.h"
 #include "src/core/util/debug_location.h"
 #include "src/core/util/ref_counted.h"
 #include "src/core/util/ref_counted_ptr.h"
 #include "src/core/util/unique_type_name.h"
 #include "src/core/util/useful.h"
+#include "absl/container/flat_hash_map.h"
+#include "absl/strings/string_view.h"
 
 namespace grpc_core {
 

data/src/core/filter/filter_args.h

@@ -15,12 +15,39 @@
 #ifndef GRPC_SRC_CORE_FILTER_FILTER_ARGS_H
 #define GRPC_SRC_CORE_FILTER_FILTER_ARGS_H
 
+#include <memory>
+
 #include "src/core/filter/blackboard.h"
 #include "src/core/lib/channel/channel_fwd.h"
 #include "src/core/util/match.h"
+#include "src/core/util/ref_counted.h"
+#include "src/core/util/ref_counted_ptr.h"
+#include "src/core/util/unique_type_name.h"
 
 namespace grpc_core {
 
+// A base class for filter configs.
+class FilterConfig : public RefCounted<FilterConfig> {
+ public:
+  virtual UniqueTypeName type() const = 0;
+
+  bool operator==(const FilterConfig& other) const {
+    if (type() != other.type()) return false;
+    return Equals(other);
+  }
+
+  bool operator!=(const FilterConfig& other) const { return !(*this == other); }
+
+  virtual bool Equals(const FilterConfig& other) const = 0;
+
+  virtual std::string ToString() const = 0;
+};
+
+struct FilterAndConfig {
+  const grpc_channel_filter* filter;
+  RefCountedPtr<const FilterConfig> config;
+};
+
 // Filter arguments that are independent of channel args.
 // Here-in should be things that depend on the filters location in the stack, or
 // things that are ephemeral and disjoint from overall channel args.
@@ -31,9 +58,11 @@ class FilterArgs {
              grpc_channel_element* channel_element,
              size_t (*channel_stack_filter_instance_number)(
                  grpc_channel_stack*, grpc_channel_element*),
+             RefCountedPtr<const FilterConfig> config = nullptr,
              const Blackboard* blackboard = nullptr)
       : impl_(ChannelStackBased{channel_stack, channel_element,
                                 channel_stack_filter_instance_number}),
+        config_(std::move(config)),
         blackboard_(blackboard) {}
   // While we're moving to call-v3 we need to have access to
   // grpc_channel_stack & friends here. That means that we can't rely on this
@@ -41,8 +70,12 @@ class FilterArgs {
   // of constructing this object without naming it ===> implicit construction.
   // TODO(ctiller): remove this once we're fully on call-v3
   // NOLINTNEXTLINE(google-explicit-constructor)
-  FilterArgs(size_t instance_id, const Blackboard* blackboard = nullptr)
-      : impl_(V3Based{instance_id}), blackboard_(blackboard) {}
+  FilterArgs(size_t instance_id,
+             RefCountedPtr<const FilterConfig> config = nullptr,
+             const Blackboard* blackboard = nullptr)
+      : impl_(V3Based{instance_id}),
+        config_(std::move(config)),
+        blackboard_(blackboard) {}
 
   ABSL_DEPRECATED("Direct access to channel stack is deprecated")
   grpc_channel_stack* channel_stack() const {
@@ -56,6 +89,8 @@ class FilterArgs {
   // 0 0 0 1 1 0 2.
   // This is useful for filters that need to store per-instance data in a
   // parallel data structure.
+  // TODO(roth): Remove this once server side is migrated to the new
+  // approach for handling xDS filter configs.
   size_t instance_id() const {
     return Match(
         impl_,
@@ -66,6 +101,8 @@ class FilterArgs {
         [](const V3Based& v3) { return v3.instance_id; });
   }
 
+  RefCountedPtr<const FilterConfig> config() const { return config_; }
+
   // Gets the filter state associated with a particular type and key.
   template <typename T>
   RefCountedPtr<T> GetState(const std::string& key) const {
@@ -90,6 +127,7 @@ class FilterArgs {
   using Impl = std::variant<ChannelStackBased, V3Based>;
   Impl impl_;
 
+  const RefCountedPtr<const FilterConfig> config_;
   const Blackboard* blackboard_ = nullptr;
 };
 

data/src/core/handshaker/endpoint_info/endpoint_info_handshaker.cc

@@ -21,8 +21,6 @@
 #include <memory>
 #include <utility>
 
-#include "absl/functional/any_invocable.h"
-#include "absl/status/status.h"
 #include "src/core/handshaker/handshaker.h"
 #include "src/core/handshaker/handshaker_factory.h"
 #include "src/core/handshaker/handshaker_registry.h"
@@ -32,6 +30,8 @@
 #include "src/core/lib/iomgr/exec_ctx.h"
 #include "src/core/util/debug_location.h"
 #include "src/core/util/ref_counted_ptr.h"
+#include "absl/functional/any_invocable.h"
+#include "absl/status/status.h"
 
 namespace grpc_core {
 

data/src/core/handshaker/handshaker.cc

@@ -28,11 +28,6 @@
 #include <string>
 #include <utility>
 
-#include "absl/functional/any_invocable.h"
-#include "absl/log/log.h"
-#include "absl/status/status.h"
-#include "absl/status/statusor.h"
-#include "absl/strings/str_format.h"
 #include "src/core/lib/channel/channel_args.h"
 #include "src/core/lib/debug/trace.h"
 #include "src/core/lib/iomgr/endpoint.h"
@@ -42,6 +37,11 @@
 #include "src/core/util/grpc_check.h"
 #include "src/core/util/status_helper.h"
 #include "src/core/util/time.h"
+#include "absl/functional/any_invocable.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
+#include "absl/status/statusor.h"
+#include "absl/strings/str_format.h"
 
 using ::grpc_event_engine::experimental::EventEngine;
 

data/src/core/handshaker/handshaker.h

@@ -26,8 +26,6 @@
 
 #include <memory>
 
-#include "absl/base/thread_annotations.h"
-#include "absl/container/inlined_vector.h"
 #include "src/core/channelz/channelz.h"
 #include "src/core/lib/channel/channel_args.h"
 #include "src/core/lib/iomgr/closure.h"
@@ -40,6 +38,8 @@
 #include "src/core/util/ref_counted_ptr.h"
 #include "src/core/util/sync.h"
 #include "src/core/util/time.h"
+#include "absl/base/thread_annotations.h"
+#include "absl/container/inlined_vector.h"
 
 namespace grpc_core {
 

data/src/core/handshaker/http_connect/http_connect_handshaker.cc

@@ -30,11 +30,6 @@
 #include <string>
 #include <utility>
 
-#include "absl/base/thread_annotations.h"
-#include "absl/log/log.h"
-#include "absl/status/status.h"
-#include "absl/strings/str_cat.h"
-#include "absl/strings/string_view.h"
 #include "src/core/config/core_configuration.h"
 #include "src/core/handshaker/handshaker.h"
 #include "src/core/handshaker/handshaker_factory.h"
@@ -53,6 +48,11 @@
 #include "src/core/util/ref_counted_ptr.h"
 #include "src/core/util/string.h"
 #include "src/core/util/sync.h"
+#include "absl/base/thread_annotations.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
+#include "absl/strings/str_cat.h"
+#include "absl/strings/string_view.h"
 
 namespace grpc_core {
 

data/src/core/handshaker/http_connect/http_proxy_mapper.cc

@@ -29,17 +29,6 @@
 #include <string>
 #include <utility>
 
-#include "absl/log/log.h"
-#include "absl/status/status.h"
-#include "absl/status/statusor.h"
-#include "absl/strings/ascii.h"
-#include "absl/strings/escaping.h"
-#include "absl/strings/match.h"
-#include "absl/strings/numbers.h"
-#include "absl/strings/str_cat.h"
-#include "absl/strings/str_split.h"
-#include "absl/strings/string_view.h"
-#include "absl/strings/strip.h"
 #include "src/core/handshaker/http_connect/http_connect_handshaker.h"
 #include "src/core/lib/address_utils/parse_address.h"
 #include "src/core/lib/address_utils/sockaddr_utils.h"
@@ -51,6 +40,17 @@
 #include "src/core/util/memory.h"
 #include "src/core/util/string.h"
 #include "src/core/util/uri.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
+#include "absl/status/statusor.h"
+#include "absl/strings/ascii.h"
+#include "absl/strings/escaping.h"
+#include "absl/strings/match.h"
+#include "absl/strings/numbers.h"
+#include "absl/strings/str_cat.h"
+#include "absl/strings/str_split.h"
+#include "absl/strings/string_view.h"
+#include "absl/strings/strip.h"
 
 namespace grpc_core {
 namespace {

data/src/core/handshaker/http_connect/http_proxy_mapper.h

@@ -24,11 +24,11 @@
 #include <optional>
 #include <string>
 
-#include "absl/strings/string_view.h"
 #include "src/core/config/core_configuration.h"
 #include "src/core/handshaker/proxy_mapper.h"
 #include "src/core/lib/channel/channel_args.h"
 #include "src/core/lib/iomgr/resolved_address.h"
+#include "absl/strings/string_view.h"
 
 namespace grpc_core {
 

data/src/core/handshaker/http_connect/xds_http_proxy_mapper.cc

@@ -20,12 +20,12 @@
 #include <optional>
 #include <string>
 
-#include "absl/log/log.h"
 #include "src/core/handshaker/http_connect/http_connect_handshaker.h"
 #include "src/core/lib/address_utils/parse_address.h"
 #include "src/core/lib/address_utils/sockaddr_utils.h"
 #include "src/core/lib/channel/channel_args.h"
 #include "src/core/xds/grpc/xds_endpoint.h"
+#include "absl/log/log.h"
 
 namespace grpc_core {
 

data/src/core/handshaker/http_connect/xds_http_proxy_mapper.h

@@ -20,11 +20,11 @@
 #include <optional>
 #include <string>
 
-#include "absl/strings/string_view.h"
 #include "src/core/config/core_configuration.h"
 #include "src/core/handshaker/proxy_mapper.h"
 #include "src/core/lib/channel/channel_args.h"
 #include "src/core/lib/iomgr/resolved_address.h"
+#include "absl/strings/string_view.h"
 
 namespace grpc_core {
 

data/src/core/handshaker/proxy_mapper.h

@@ -24,9 +24,9 @@
 #include <optional>
 #include <string>
 
-#include "absl/strings/string_view.h"
 #include "src/core/lib/channel/channel_args.h"
 #include "src/core/lib/iomgr/resolved_address.h"
+#include "absl/strings/string_view.h"
 
 namespace grpc_core {
 

data/src/core/handshaker/proxy_mapper_registry.h

@@ -27,10 +27,10 @@
 #include <string>
 #include <vector>
 
-#include "absl/strings/string_view.h"
 #include "src/core/handshaker/proxy_mapper.h"
 #include "src/core/lib/channel/channel_args.h"
 #include "src/core/lib/iomgr/resolved_address.h"
+#include "absl/strings/string_view.h"
 
 namespace grpc_core {
 

data/src/core/handshaker/security/legacy_secure_endpoint.cc

@@ -32,10 +32,6 @@
 #include <optional>
 #include <utility>
 
-#include "absl/base/thread_annotations.h"
-#include "absl/log/log.h"
-#include "absl/status/status.h"
-#include "absl/strings/string_view.h"
 #include "src/core/handshaker/security/secure_endpoint.h"
 #include "src/core/lib/debug/trace.h"
 #include "src/core/lib/experiments/experiments.h"
@@ -57,6 +53,10 @@
 #include "src/core/util/ref_counted_ptr.h"
 #include "src/core/util/string.h"
 #include "src/core/util/sync.h"
+#include "absl/base/thread_annotations.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
+#include "absl/strings/string_view.h"
 
 #define STAGING_BUFFER_SIZE 8192
 

data/src/core/handshaker/security/pipelined_secure_endpoint.cc

@@ -34,13 +34,6 @@
 #include <utility>
 #include <vector>
 
-#include "absl/base/thread_annotations.h"
-#include "absl/functional/any_invocable.h"
-#include "absl/log/log.h"
-#include "absl/status/status.h"
-#include "absl/strings/str_cat.h"
-#include "absl/strings/string_view.h"
-#include "absl/types/span.h"
 #include "src/core/handshaker/security/secure_endpoint.h"
 #include "src/core/lib/channel/channel_args.h"
 #include "src/core/lib/debug/trace.h"
@@ -62,6 +55,13 @@
 #include "src/core/util/ref_counted_ptr.h"
 #include "src/core/util/string.h"
 #include "src/core/util/sync.h"
+#include "absl/base/thread_annotations.h"
+#include "absl/functional/any_invocable.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
+#include "absl/strings/str_cat.h"
+#include "absl/strings/string_view.h"
+#include "absl/types/span.h"
 
 #define STAGING_BUFFER_SIZE 8192
 

data/src/core/handshaker/security/secure_endpoint.cc

@@ -37,11 +37,6 @@
 #include <utility>
 #include <vector>
 
-#include "absl/base/thread_annotations.h"
-#include "absl/log/log.h"
-#include "absl/status/status.h"
-#include "absl/strings/string_view.h"
-#include "absl/types/span.h"
 #include "src/core/lib/debug/trace.h"
 #include "src/core/lib/experiments/experiments.h"
 #include "src/core/lib/iomgr/closure.h"
@@ -63,6 +58,11 @@
 #include "src/core/util/ref_counted_ptr.h"
 #include "src/core/util/string.h"
 #include "src/core/util/sync.h"
+#include "absl/base/thread_annotations.h"
+#include "absl/log/log.h"
+#include "absl/status/status.h"
+#include "absl/strings/string_view.h"
+#include "absl/types/span.h"
 
 #define STAGING_BUFFER_SIZE 8192
 
@@ -71,6 +71,12 @@ static void on_write(void* user_data, grpc_error_handle error);
 
 namespace grpc_core {
 namespace {
+
+grpc_slice AllocSlice(size_t size, void* user_data) {
+  auto* owner = static_cast<MemoryOwner*>(user_data);
+  return owner->MakeSlice(MemoryRequest(size));
+}
+
 class FrameProtector : public RefCounted<FrameProtector> {
  public:
   FrameProtector(tsi_frame_protector* protector,
@@ -94,6 +100,10 @@ class FrameProtector : public RefCounted<FrameProtector> {
       }
     }
     if (zero_copy_protector_ != nullptr) {
+      if (IsTrackZeroCopyAllocationsInResourceQuotaEnabled()) {
+        tsi_zero_copy_grpc_protector_set_allocator(zero_copy_protector_,
+                                                   &AllocSlice, &memory_owner_);
+      }
       read_staging_buffer_ = grpc_empty_slice();
       write_staging_buffer_ = grpc_empty_slice();
     } else {