onmessage expects socket as well

3 years ago · d25f055d13
1 changed files with 2 additions and 2 deletions
--- a/index.js
+++ b/index.js
@ -77,8 +77,8 @@ class DHT extends EventEmitter {
    return this._nat.port
  }

-  onmessage (buf, rinfo) {
-    if (buf.byteLength > 1) this.io.onmessage(null, buf, rinfo)
+  onmessage (socket, buf, rinfo) {
+    if (buf.byteLength > 1) this.io.onmessage(socket, buf, rinfo)
  }

  bind () {