1 files changed, 6 insertions, 2 deletions
diff --git a/src/core/iomgr/udp_server.h b/src/core/iomgr/udp_server.h
index 76082d7761..b6cf3c8ed9 100644
--- a/src/core/iomgr/udp_server.h
+++ b/src/core/iomgr/udp_server.h
@@ -36,18 +36,22 @@
 
 #include "src/core/iomgr/endpoint.h"
 
+/* Forward decl of grpc_server */
+typedef struct grpc_server grpc_server;
+
 /* Forward decl of grpc_udp_server */
 typedef struct grpc_udp_server grpc_udp_server;
 
 /* Called when data is available to read from the socket. */
-typedef void (*grpc_udp_server_read_cb)(int fd);
+typedef void (*grpc_udp_server_read_cb)(int fd, grpc_server* server);
 
 /* Create a server, initially not bound to any ports */
 grpc_udp_server *grpc_udp_server_create(void);
 
 /* Start listening to bound ports */
 void grpc_udp_server_start(grpc_exec_ctx *exec_ctx, grpc_udp_server *udp_server,
-                           grpc_pollset **pollsets, size_t pollset_count);
+                           grpc_pollset **pollsets, size_t pollset_count,
+                           grpc_server *server);
 
 int grpc_udp_server_get_fd(grpc_udp_server *s, unsigned index);