fix: correct remote address override in wrapper grpc handling

Signed-off-by: Alexander Trost <[email protected]>
fivenet-app · Oct 12, 2024 · d74481e · d74481e
1 parent 7bc730c
commit d74481e
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/app/composables/grpcws/bridge/index.ts b/app/composables/grpcws/bridge/index.ts
@@ -43,6 +43,7 @@ export class GrpcWSTransport implements RpcTransport {
                 delay: 1150,
             },
             protocols: ['grpc-websocket-channel'],
+
             onConnected(ws) {
                 ws.binaryType = 'arraybuffer';
                 wsInitiated.value = true;

diff --git a/pkg/server/http.go b/pkg/server/http.go
@@ -198,7 +198,14 @@ func NewEngine(p EngineParams) (*gin.Engine, error) {
 		grpcws.WithWebsocketPingInterval(40*time.Second),
 	)
 	ginWrappedGrpc := func(c *gin.Context) {
-		c.Request.RemoteAddr = c.ClientIP()
+		if cip := c.ClientIP(); cip != "" {
+			if strings.Count(cip, ":") > 1 {
+				c.Request.RemoteAddr = "[" + cip + "]:80"
+			} else {
+				c.Request.RemoteAddr = cip + ":80"
+			}
+		}
+
 		wrapperGrpc.ServeHTTP(c.Writer, c.Request)
 	}
 	e.Any("/api/grpc", ginWrappedGrpc)