From 0774c7d9df5515482b6c7d5e813793642575d23d Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=D0=92=D0=B0=D1=81=D1=8F=20=D0=A7=D0=B0=D0=B9=D0=BA=D0=BE?=
 <vasya@chajko.ru>
Date: Thu, 2 Jan 2025 01:03:15 +0700
Subject: [PATCH] Fix error: empty respose in tcp client when server response
 is async

---
 client/tcp.js |  3 ++-
 test/index.js | 34 ++++++++++++++++++++++++++++++++--
 2 files changed, 34 insertions(+), 3 deletions(-)

diff --git a/client/tcp.js b/client/tcp.js
index 4a34d5c..7c9023f 100644
--- a/client/tcp.js
+++ b/client/tcp.js
@@ -19,7 +19,7 @@ const makeQuery = ({ name, type = 'A', cls = Packet.CLASS.IN, clientIp, recursiv
 const sendQuery = (client, message) => {
   const len = Buffer.alloc(2);
   len.writeUInt16BE(message.length);
-  client.end(Buffer.concat([ len, message ]));
+  client.write(Buffer.concat([ len, message ]));
 };
 
 const protocols = {
@@ -39,6 +39,7 @@ const TCPClient = ({ dns, protocol = 'tcp:', port = protocol === 'tls:' ? 853 :
 
     sendQuery(client, message);
     const data = await Packet.readStream(client);
+    client.end();
 
     if (!data.length) throw new Error('Empty response');
     return Packet.parse(data);
diff --git a/test/index.js b/test/index.js
index 492b3a1..9909ea9 100644
--- a/test/index.js
+++ b/test/index.js
@@ -1,7 +1,7 @@
 const assert = require('assert');
 const test = require('./test');
-const { Packet, createDOHServer, createServer } = require('..');
-// const { Packet, createDOHServer, createServer, TCPClient, DOHClient, UDPClient } = require('..');
+// const { Packet, createDOHServer, createServer } = require('..');
+const { Packet, createDOHServer, createServer, TCPClient, DOHClient, UDPClient } = require('..');
 const http = require('http');
 const tcp = require('net');
 const udp = require('dgram');
@@ -371,6 +371,36 @@ test('server/doh#cors - cors function', async function() {
 //   await server.close();
 // });
 
+test('server/udp-tcp#simple-request-async-response', async() => {
+  const server = createServer({
+    tcp : true,
+    udp : true,
+    handle(request, send, _info) {
+      const [ question ] = request.questions;
+      assert.deepEqual(request.questions, [ { name: 'test.com', type: 1, class: 1 } ]);
+      const response = Packet.createResponseFromRequest(request);
+      response.answers.push({
+        name  : question.name,
+        type  : Packet.TYPE.TXT,
+        class : Packet.CLASS.IN,
+        ttl   : 300,
+        data  : [ 'Hello World' ],
+      });
+
+      (new Promise((resolve) => setTimeout(() => resolve(), 1))).then(() => send(response));
+    },
+  });
+  const servers = await server.listen();
+  assert.ok(servers.udp.port > 1000);
+  assert.ok(servers.tcp.port > 1000);
+  const tcp = TCPClient({ dns: '127.0.0.1', port: servers.tcp.port });
+  const udp = UDPClient({ dns: '127.0.0.1', port: servers.udp.port });
+  const expected = [ { name: 'test.com', ttl: 300, type: 16, class: 1, data: 'Hello World' } ];
+  assert.deepEqual((await tcp('test.com')).answers, expected);
+  assert.deepEqual((await udp('test.com')).answers, expected);
+  await server.close();
+});
+
 test('server/all#invalid-request', async() => {
   const server = createServer({
     doh    : true,