浏览代码

Fix UDP exchange for local/dhcp DNS servers

世界 1 月之前
父节点
当前提交
510bf05e36
共有 2 个文件被更改,包括 2 次插入2 次删除
  1. 1 1
      dns/transport/dhcp/dhcp_shared.go
  2. 1 1
      dns/transport/local/local.go

+ 1 - 1
dns/transport/dhcp/dhcp_shared.go

@@ -121,7 +121,7 @@ func (t *Transport) exchangeUDP(ctx context.Context, server M.Socksaddr, request
 	if deadline, loaded := ctx.Deadline(); loaded && !deadline.IsZero() {
 	if deadline, loaded := ctx.Deadline(); loaded && !deadline.IsZero() {
 		conn.SetDeadline(deadline)
 		conn.SetDeadline(deadline)
 	}
 	}
-	buffer := buf.Get(1 + request.Len())
+	buffer := buf.Get(buf.UDPBufferSize)
 	defer buf.Put(buffer)
 	defer buf.Put(buffer)
 	rawMessage, err := request.PackBuffer(buffer)
 	rawMessage, err := request.PackBuffer(buffer)
 	if err != nil {
 	if err != nil {

+ 1 - 1
dns/transport/local/local.go

@@ -182,7 +182,7 @@ func (t *Transport) exchangeUDP(ctx context.Context, server M.Socksaddr, request
 		}
 		}
 		conn.SetDeadline(deadline)
 		conn.SetDeadline(deadline)
 	}
 	}
-	buffer := buf.Get(1 + request.Len())
+	buffer := buf.Get(buf.UDPBufferSize)
 	defer buf.Put(buffer)
 	defer buf.Put(buffer)
 	rawMessage, err := request.PackBuffer(buffer)
 	rawMessage, err := request.PackBuffer(buffer)
 	if err != nil {
 	if err != nil {