From 0fdbefbb86bcf959875358b732682af65eddacc2 Mon Sep 17 00:00:00 2001
From: Jannis R <mail@jannisr.de>
Date: Fri, 16 Oct 2020 14:49:33 +0200
Subject: client & server: improved error handling & timeouts 🐛
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 client.js | 27 +++++++++++++++++----------
 1 file changed, 17 insertions(+), 10 deletions(-)

(limited to 'client.js')

diff --git a/client.js b/client.js
index 4614921..2748da1 100644
--- a/client.js
+++ b/client.js
@@ -3,6 +3,7 @@
 const debug = require('debug')('gemini:client')
 const {parse: parseUrl} = require('url')
 const pem = require('pem')
+const {pipeline: pipe} = require('stream')
 const connect = require('./connect')
 const createParser = require('./lib/response-parser')
 const {
@@ -33,20 +34,26 @@ const _request = (pathOrUrl, opt, cb) => {
 		}
 
 		const res = createParser()
-		socket.pipe(res)
-		socket.once('error', (err) => {
-			socket.unpipe(res)
-			res.destroy(err)
-		})
-
-		const close = () => {
-			socket.destroy()
-			res.destroy()
+		pipe(
+			socket,
+			res,
+			(err) => {
+				if (err) debug('error receiving response', err)
+				// Control over the socket has been given to the caller
+				// already, so we swallow the error here.
+				if (!timeout) return;
+				if (!err) cb(new Error('socket closed while waiting for header'))
+			},
+		)
+
+		const reportTimeout = () => {
+			socket.destroy(new Error('timeout waiting for header'))
 		}
-		let timeout = setTimeout(close, 20 * 1000)
+		let timeout = setTimeout(reportTimeout, 20 * 1000)
 
 		res.once('header', (header) => {
 			clearTimeout(timeout)
+			timeout = null
 			debug('received header', header)
 
 			// prepare res
-- 
cgit v1.3