From 32fd4ac4189577effb6f546f82e4fe9cf74033c0 Mon Sep 17 00:00:00 2001
From: wujohns <wujohns@163.com>
Date: Tue, 12 Dec 2017 11:32:06 +0800
Subject: [PATCH] Server: add 'user' param to socket

---
 lib/auth/UserPassword.js | 2 +-
 lib/server.js            | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/lib/auth/UserPassword.js b/lib/auth/UserPassword.js
index 0f17eb9..150836f 100644
--- a/lib/auth/UserPassword.js
+++ b/lib/auth/UserPassword.js
@@ -127,7 +127,7 @@ module.exports = function UserPasswordAuthHandlers() {
                       stream.write(BUF_SUCCESS);
                     else
                       stream.write(BUF_FAILURE);
-                    cb(success);
+                    cb(success, user);
                   }
                 });
                 return;
diff --git a/lib/server.js b/lib/server.js
index bcac647..986cdd7 100644
--- a/lib/server.js
+++ b/lib/server.js
@@ -76,8 +76,9 @@ Server.prototype._onConnection = function(socket) {
     for (var a = 0, alen = auths.length; a < alen; ++a) {
       for (var m = 0, mlen = methods.length; m < mlen; ++m) {
         if (methods[m] === auths[a].METHOD) {
-          auths[a].server(socket, function(result) {
+          auths[a].server(socket, function(result, user) {
             if (result === true) {
+              socket.user = user;
               parser.authed = true;
               parser.start();
             } else {