Java 通过telnet协议 操作远程交换机
/**
*@date 2015年2月4日 下午3:15:13
*@author XUQIANG
*@filename TelnetConnection.java
*@package com.merit.monitor.device.parameter.sdk
*/
package com.merit.monitor.device.parameter.sdk;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.PrintStream;
import java.net.SocketException;
import org.apache.commons.net.telnet.TelnetClient;
import com.merit.monitor.device.parameter.logs.ILog;
import com.merit.monitor.device.parameter.logs.LogImpl;
/**
- @author XUQIANG
*/
public class TelnetConnection {
private static ILog log = new LogImpl();
private TelnetClient telnet = null;
// private String username = "";
private String password = "";
private String prompt = "#";
private InputStream in;
private PrintStream out;
public TelnetConnection(String host, int port) {
if(telnet == null) {
telnet = new TelnetClient();
try {
telnet.connect(host, port);
in = telnet.getInputStream();
out = new PrintStream(telnet.getOutputStream());
} catch (SocketException e) {
log.logForException(TelnetConnection.class, e.getMessage());
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
}
/**
* @return the prompt
*/
public String getPrompt() {
return prompt;
}
/**
* @param prompt the prompt to set
*/
public void setPrompt(String prompt) {
this.prompt = prompt;
}
/**
* 关闭打开的连接
* @param telnet
*/
public void close(TelnetClient telnet) {
if(telnet != null) {
try {
telnet.disconnect();
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
if(this.telnet != null) {
try {
this.telnet.disconnect();
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
}
/**
* 登录到远程机器中
*/
public void login(String username, String password, String prompt) {
//处理命令行的提示字符
if(prompt == null || "".equals(prompt)) {
if(this.prompt.equals("#"))
prompt = ("root".equals(username)) ? "#" : "$";
} else {
this.prompt = prompt;
}
this.password = password;
readUntil("Username:");
write(username);
readUntil("Password:");
write(this.password);
readUntil(this.prompt + "");
// 其它交换机登录后如果有提示信息也需要添加
if(this.prompt.indexOf("Quidway") != -1)
readUntil("login");
}
/** * 读取分析结果 * * @param pattern * @return */
public String readUntil(String pattern) {
StringBuffer sb = new StringBuffer();
try {
int len = 0;
while((len = in.read()) != -1) {
sb.append((char)len);
if(sb.toString().endsWith(pattern)) {
return sb.toString();
}
}
} catch (IOException e1) {
// TODO Auto-generated catch block
e1.printStackTrace();
}
return "";
}
/** * 写操作 * * @param value */
public void write(String value) {
try {
out.println(value);
out.flush();
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
/** * 向目标发送命令字符串 * * @param command * @return */
public String sendCommand(String command) {
try {
write(command);
return readUntil(prompt + "");
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
return "";
}
/** * 关闭连接 */
public void disconnect() {
try {
telnet.disconnect();
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
}
Junit
@Test
public void testTelnet() {
try {
System.out.println("启动Telnet...");
String ip = "191.168.*.*";
int port = 23;
String user = "admin";
String password = "****";
TelnetConnection telnet = new TelnetConnection(ip, port);
telnet.setPrompt("");
telnet.login(user, password, "");
telnet.setPrompt("[Quidway]");
String r1 = telnet.sendCommand("system-view");//display snmp-agent local-engineid
telnet.setPrompt("[Quidway-Ethernet1/0/5]");
String r2 = telnet.sendCommand("interface Ethernet1/0/5");
String r3 = telnet.sendCommand("undo shutdown");
System.out.println("显示结果");
System.out.println(r1);
System.out.println(r2);
System.out.println(r3);
System.out.println((r3.indexOf(telnet.getPrompt()) != -1) && r3.indexOf("^") == -1);
telnet.disconnect();
} catch (Exception e) {
e.printStackTrace();
}
}
改进后的实现：
登录到远程机器中
说明：在登录前，先确认输入用户名的提示符，如果不是Username：，需要设置该值，使用setUsernamePrompt(prompt)；第二，确认输入密码时的提示符，如果不是Password：，需要设置该值,使用setPasswordPrompt(prompt)； 第三，确认登录后查看是否有登录后的提示信息：如：%Apr 17 04:26:32:256 2000 Quidway SHELL/5/LOGIN:- 1 - admin(191.168.2.227) in unit1 login 如果末尾不是login,需要指定最后一个单词，使用setLoginPrompt(prompt)。 如果没有登录提示，设置setLoginPrompt(null); 第四，执行命令时，如果提示符不是 #、$、>、]中的一个，也需要指定最后一个符号，使用setPrompt(prompt).
/**
*@date 2015年2月4日 下午3:15:13
*@author XUQIANG
*@filename TelnetConnection.java
*@package com.merit.monitor.device.parameter.sdk
*/
package com.meritit.product.eism.telnet.sdk;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintStream;
import java.net.SocketException;
import org.apache.commons.net.telnet.TelnetClient;
import com.meritit.product.eism.telnet.log.ILog;
import com.meritit.product.eism.telnet.log.LogImpl;
/**
* @author XUQIANG
*
*/
public class TelnetConnection {
private static ILog log = new LogImpl();
private TelnetClient telnet = null;
private String prompt = "#$>]";
private String loginPrompt = "login";
private String usernamePrompt = "Username:";
private String passwordPrompt = "Password:";
private InputStream in;
private PrintStream out;
public TelnetConnection(String host, int port) {
if(telnet == null) {
telnet = new TelnetClient();
try {
telnet.connect(host, port);
in = telnet.getInputStream();
out = new PrintStream(telnet.getOutputStream());
} catch (SocketException e) {
log.logForException(TelnetConnection.class, e.getMessage());
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
}
/**
* 登录到远程机器中
* 说明：在登录前，先确认输入用户名的提示符，如果不是Username：，需要设置该值，使用setUsernamePrompt(prompt)；
* 第二，确认输入密码时的提示符，如果不是Password：，需要设置该值,使用setPasswordPrompt(prompt)；
* 第三，确认登录后查看是否有登录后的提示信息：如：%Apr 17 04:26:32:256 2000 Quidway SHELL/5/LOGIN:- 1 - admin(191.168.2.227) in unit1 login
* 如果末尾不是login,需要指定最后一个单词，使用setLoginPrompt(prompt)。
如果没有登录提示，设置setLoginPrompt(null);
* 第四，执行命令时，如果提示符不是 #、$、>、]中的一个，也需要指定最后一个符号，使用setPrompt(prompt).
*/
public void login(String username, String password, String prompt) {
//处理命令行的提示字符
if(prompt != null && !"".equals(prompt)) {
this.prompt = prompt;
}
readUntil(this.usernamePrompt);
write(username);
readUntil(this.passwordPrompt);
write(password);
readUntil(this.prompt);
if(this.loginPrompt ！= null)
readUntil(this.loginPrompt);
}
/** * 读取分析结果 * * @param pattern * @return */
public String readUntil(String pattern) {
StringBuffer sb = new StringBuffer();
try {
int len = 0;
while((len = in.read()) != -1) {
sb.append((char)len);
if(pattern.indexOf((char)len) != -1 || sb.toString().endsWith(pattern)) {
return sb.toString();
}
}
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
return "";
}
/** * 写操作 * * @param value */
public void write(String value) {
try {
out.println(value);
out.flush();
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
/** * 向目标发送命令字符串 * * @param command * @return */
public String sendCommand(String command) {
try {
write(command);
return readUntil(prompt + "");
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
return "";
}
/** * 关闭连接 */
public void disconnect() {
try {
telnet.disconnect();
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
/**
* @return the prompt
*/
public String getPrompt() {
return prompt;
}
/**
* @param prompt the prompt to set
*/
public void setPrompt(String prompt) {
this.prompt = prompt;
}
/**
* @return the usernamePrompt
*/
public String getUsernamePrompt() {
return usernamePrompt;
}
/**
* @param usernamePrompt the usernamePrompt to set
*/
public void setUsernamePrompt(String usernamePrompt) {
this.usernamePrompt = usernamePrompt;
}
/**
* @return the passwordPrompt
*/
public String getPasswordPrompt() {
return passwordPrompt;
}
/**
* @param passwordPrompt the passwordPrompt to set
*/
public void setPasswordPrompt(String passwordPrompt) {
this.passwordPrompt = passwordPrompt;
}
/**
* @return the loginPrompt
*/
public String getLoginPrompt() {
return loginPrompt;
}
/**
* @param loginPrompt the loginPrompt to set
*/
public void setLoginPrompt(String loginPrompt) {
this.loginPrompt = loginPrompt;
}
/**
* 关闭打开的连接
* @param telnet
*/
public void close(TelnetClient telnet) {
if(telnet != null) {
try {
telnet.disconnect();
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
if(this.telnet != null) {
try {
this.telnet.disconnect();
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
}
}
=====================================================================
添加了telnet tab键命令补全功能。
简单记录一下：
单击tab键时，将命令和\t符号一同发送到交换机中，但是不执行命令，所以就不能使用out.println().只能使用out.print().
连续tab键时，需要向服务器发送的命令是，与上次命令不同的地方。就是把上次相同的命令字符要去掉。
例如：
第一次tab时执行: int
发送命令为: int\t
执行程序为：out.print(“int\t”);
返回值为：interface
第二次tab时执行： interface e
发送命令为：e\t
执行程序为：out.print(“e\t”);
返回值为：interface Ethernet
第三次按回车是执行：interface Ethernet 1/0/5
发送命令： 1/0/5 说明：1/0/5前面有个空格
执行程序为：out.println(" 1/0/5");
返回结果为：[Quidway-Ethernet1/0/5] 说明：只有提示符。
源程序如下：
/**
*@date 2015年3月19日 下午3:15:13
*@author XUQIANG
*@filename TelnetConnection.java
*@package com.merit.monitor.device.parameter.sdk
*/
package com.meritit.product.eism.telnet.sdk;
import java.io.IOException;
import java.io.InputStream;
import java.io.PrintStream;
import java.net.SocketException;
import org.apache.commons.net.telnet.TelnetClient;
import com.meritit.product.eism.telnet.log.ILog;
import com.meritit.product.eism.telnet.log.LogImpl;
/**
* @author XUQIANG
*
*/
public class TelnetConnection {
private static ILog log = new LogImpl();
private TelnetClient telnet = null;
private String prompt = "#$>]";
private String loginPrompt = "login";
private String usernamePrompt = "Username:";
private String passwordPrompt = "Password:";
private InputStream in;
private PrintStream out;
public TelnetConnection(String host, int port) {
if(telnet == null) {
telnet = new TelnetClient();
try {
telnet.connect(host, port);
in = telnet.getInputStream();
out = new PrintStream(telnet.getOutputStream());
} catch (SocketException e) {
log.logForException(TelnetConnection.class, e.getMessage());
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
}
/**
* 登录到远程机器中
* 说明：在登录前，先确认输入用户名的提示符，如果不是Username：，需要设置该值，使用setUsernamePrompt(prompt)；
* 第二，确认输入密码时的提示符，如果不是Password：，需要设置该值,使用setPasswordPrompt(prompt)；
* 第三，确认登录后查看是否有登录后的提示信息：如：%Apr 17 04:26:32:256 2000 Quidway SHELL/5/LOGIN:- 1 - admin(191.168.2.227) in unit1 login
* 如果末尾不是login,需要指定最后一个单词，使用setLoginPrompt(prompt).
* 如果没有登录提示，设置setLoginPrompt(null);
* 第四，执行命令时，如果提示符不是 #、$、>、]中的一个，也需要指定最后一个符号，使用setPrompt(prompt).
*/
public void login(String username, String password, String prompt) {
loginForWin(username, password, prompt);
}
public String loginForWin(String username, String password, String prompt) {
//处理命令行的提示字符
if(prompt != null && !"".equals(prompt)) {
this.prompt = prompt;
}
String result = "";
readUntil(this.usernamePrompt);
write(username);
readUntil(this.passwordPrompt);
write(password);
result = readUntil(this.prompt);
if(this.loginPrompt != null)
result += readUntil(this.loginPrompt);
return result;
}
/** * 读取分析结果 * * @param pattern * @return */
private String readUntil(String pattern) {
StringBuffer sb = new StringBuffer();
try {
int len = 0;
while((len = in.read()) != -1) {
char ch = (char)len;
sb.append(ch);
if(pattern.indexOf(ch) != -1 || sb.toString().endsWith(pattern)) {
return sb.toString();
}
}
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
return "";
}
/**
* tab提示 命令补全
* @return
*/
private String readUntil() {
StringBuffer sb = new StringBuffer();
boolean flag = false;
try {
int len = 0;
int i = 0;
Thread.sleep(500);
int count = in.available();
// System.out.println(count);
while((len = in.read()) != -1) {
i++;
char ch = (char)len;
sb.append(ch);
// System.out.print(ch);
if(prompt.indexOf(ch) != -1) {
flag = true;
}
if(flag && i == count) {
return sb.toString();
}
}
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
} catch (InterruptedException e) {
e.printStackTrace();
}
System.out.println(sb.toString());
return sb.toString();
}
/** * 写操作 * * @param value */
private void write(String value) {
try {
out.println(value);
out.flush();
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
/** * 写操作 * * @param value */
private void writeTab(String value) {
try {
// out.println();
out.print(value);
out.flush();
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
/** * 向目标发送命令字符串 * * @param command * @return */
public String sendCommand(String command) {
try {
write(command);
return readUntil(prompt + "");
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
return "";
}
/** * 向目标发送命令字符串 * * @param command * @return */
public String sendCommand(String command, boolean prom) {
if(prom && command.indexOf("\t") == -1)
command += "\t";
try {
writeTab(command);
return readUntil();
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
return "";
}
/** * 关闭连接 */
public void disconnect() {
try {
telnet.disconnect();
} catch (Exception e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
/**
* @return the prompt
*/
public String getPrompt() {
return prompt;
}
/**
* @param prompt the prompt to set
*/
public void setPrompt(String prompt) {
this.prompt = prompt;
}
/**
* @return the usernamePrompt
*/
public String getUsernamePrompt() {
return usernamePrompt;
}
/**
* @param usernamePrompt the usernamePrompt to set
*/
public void setUsernamePrompt(String usernamePrompt) {
this.usernamePrompt = usernamePrompt;
}
/**
* @return the passwordPrompt
*/
public String getPasswordPrompt() {
return passwordPrompt;
}
/**
* @param passwordPrompt the passwordPrompt to set
*/
public void setPasswordPrompt(String passwordPrompt) {
this.passwordPrompt = passwordPrompt;
}
/**
* @return the loginPrompt
*/
public String getLoginPrompt() {
return loginPrompt;
}
/**
* @param loginPrompt the loginPrompt to set
*/
public void setLoginPrompt(String loginPrompt) {
this.loginPrompt = loginPrompt;
}
/**
* 关闭打开的连接
* @param telnet
*/
public void close(TelnetClient telnet) {
if(telnet != null) {
try {
telnet.disconnect();
telnet = null;
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
if(this.telnet != null) {
try {
this.telnet.disconnect();
this.telnet = null;
} catch (IOException e) {
log.logForException(TelnetConnection.class, e.getMessage());
}
}
}
}