make it nicer to websockets

Author: riccardobl

src/Gateway.ts

@@ -4,6 +4,7 @@ import Net from "net";
 // @ts-ignore
 import b4a from "b4a";
 import UDPNet from "./UDPNet.js";
+import TCPNet from "./TCPNet.js";
 import { RoutingEntry, RoutingTable } from "./Router.js";
 import { Socket as NetSocket } from "net";
 import Utils from "./Utils.js";
@@ -288,7 +289,7 @@ export default class Gateway extends Peer {
                                         gatePort: gatePort,
                                     }),
                                 );
-                                const timeout = setTimeout(() => rej("timeout"), 5000); // timeout open request
+                                const timeout = setTimeout(() => rej("route timeout"), 5000); // timeout open request
                                 this.addMessageHandler((peer, msg) => {
                                     if (msg.actionId == MessageActions.open && msg.channelPort == channelPort) {
                                         if (msg.error) {
@@ -368,7 +369,7 @@ export default class Gateway extends Peer {
             refreshId: this.refreshId,
             channels: [],
         };
-        const conn = (protocol == "udp" ? UDPNet : Net).createServer((socket) => {
+        const conn = (protocol == "udp" ? UDPNet : TCPNet).createServer((socket) => {
             onConnection(gate, socket);
         });
 

src/ServiceProvider.ts

@@ -1,11 +1,11 @@
 import Peer from "./Peer.js";
 import Message, { MessageContent } from "./Message.js";
-import Net from "net";
 import UDPNet from "./UDPNet.js";
 import { MessageActions } from "./Message.js";
 import { AuthorizedPeer } from "./Peer.js";
 import { RoutingTable, Service } from "./Router.js";
 import Utils from "./Utils.js";
+import TCPNet from "./TCPNet.js";
 
 export default class ServiceProvider extends Peer {
     private services: Array<Service> = [];
@@ -125,7 +125,7 @@ export default class ServiceProvider extends Peer {
                 // connect to service
                 console.log("Connect to", service.serviceHost, service.servicePort, isUDP ? "UDP" : "TCP", "on channel", msg.channelPort);
 
-                const serviceConn = (isUDP ? UDPNet : Net).connect({
+                const serviceConn = (isUDP ? UDPNet : TCPNet).connect({
                     host: service.serviceHost,
                     port: service.servicePort,
                     allowHalfOpen: true,

src/TCPNet.ts

@@ -0,0 +1,21 @@
+import Net from 'net';
+
+const timeout = 60 * 60 * 1000;
+
+export default class TCPNet {
+    public static connect(options: Net.NetConnectOpts, connectionListener?: () => void): Net.Socket{
+        options.timeout = options.timeout ?? timeout;
+        const socket:Net.Socket = Net.connect(options, connectionListener);        
+        socket.setKeepAlive(true);
+        return socket;
+    }
+
+    public static createServer(connectionListener?: (socket: Net.Socket) => void): Net.Server{
+        const server:Net.Server = Net.createServer(connectionListener);
+        server.on('connection', (socket) => {
+            socket.setKeepAlive(true);
+            socket.setTimeout(timeout);
+        });
+        return server;
+    }
+}