From 7806db1d3f0fc0574235c92f3936507f1530a4ba Mon Sep 17 00:00:00 2001
From: guanghongwei <hongwei.guang@yolu-inc.com>
Date: Mon, 22 Dec 2014 16:22:04 +0800
Subject: [PATCH] catch excepition

---
 connect.py | 12 +++++++++++-
 1 file changed, 11 insertions(+), 1 deletion(-)

diff --git a/connect.py b/connect.py
index cb692c265..c1fb3ace4 100755
--- a/connect.py
+++ b/connect.py
@@ -10,6 +10,7 @@ import sys
 import struct
 import fcntl
 import signal
+import socket
 
 try:
     import termios
@@ -112,7 +113,16 @@ def connect(username, password, host, port):
     ssh = paramiko.SSHClient()
     ssh.load_system_host_keys()
     ssh.set_missing_host_key_policy(paramiko.AutoAddPolicy())
-    ssh.connect(host, port=port, username=username, password=password, compress=True)
+    try:
+        ssh.connect(host, port=port, username=username, password=password, compress=True)
+    except paramiko.ssh_exception.AuthenticationException:
+        print 'Password Error, Please Correct it.'
+        time.sleep(2)
+        sys.exit()
+    except socket.error:
+        print 'Connect SSH Socket Port Error, Please Correct it.'
+        time.sleep(2)
+        sys.exit()
 
     # Make a channel and set windows size
     global channel