Merge branch 'master' of github.com:jyu213/ssh2-sftp-client

theophilusx · Jun 7, 2016 · 753bf8e · 753bf8e
2 parents 4deddea + 78d1177
commit 753bf8e
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/index.js b/src/index.js
@@ -67,7 +67,11 @@ SftpClient.prototype.get = function(path, useCompression) {
 
         if (sftp) {
             try {
-                resolve(sftp.createReadStream(path, useCompression));
+                let stream = sftp.createReadStream(path, useCompression);
+
+                stream.on('error', reject);
+
+                resolve(stream);
             } catch(err) {
                 reject(err);
             }
@@ -105,9 +109,8 @@ SftpClient.prototype.put = function(input, remotePath, useCompression) {
             let stream = sftp.createWriteStream(remotePath, useCompression);
             let data;
 
-            stream.on('close', () => {
-                resolve();
-            });
+            stream.on('error', reject);
+            stream.on('close', resolve);
 
             if (input instanceof Buffer) {
                 data = stream.end(input);