Fix DNS routing

This commit is contained in:
世界
2022-09-04 12:39:43 +08:00
parent fa9450da1e
commit 90407bab12
2 changed files with 7 additions and 10 deletions

View File

@@ -77,12 +77,9 @@ func (d *DNS) handleConnection(ctx context.Context, conn net.Conn, metadata adap
if err != nil {
return err
}
if len(message.Questions) > 0 {
question := message.Questions[0]
metadata.Domain = string(question.Name.Data[:question.Name.Length-1])
}
metadataInQuery := metadata
go func() error {
response, err := d.router.Exchange(ctx, &message)
response, err := d.router.Exchange(adapter.WithContext(ctx, &metadataInQuery), &message)
if err != nil {
return err
}
@@ -125,13 +122,10 @@ func (d *DNS) NewPacketConnection(ctx context.Context, conn N.PacketConn, metada
if err != nil {
return err
}
if len(message.Questions) > 0 {
question := message.Questions[0]
metadata.Domain = string(question.Name.Data[:question.Name.Length-1])
}
timeout.Update()
metadataInQuery := metadata
go func() error {
response, err := d.router.Exchange(ctx, &message)
response, err := d.router.Exchange(adapter.WithContext(ctx, &metadataInQuery), &message)
if err != nil {
return err
}