51Testing软件测试论坛

标题: loadrunner远程调用sh脚本 [打印本页]

作者: junior777 时间: 2013-3-11 14:09
标题: loadrunner远程调用sh脚本
本帖最后由 junior777 于 2013-3-19 14:09 编辑

package com.sun.junior;
import java.io.InputStream;
import java.io.PrintStream;
import org.apache.commons.net.telnet.TelnetClient;

public class Shell
{
private TelnetClient telnet = new TelnetClient();

private InputStream in;

private PrintStream out;

private char prompt = '$';// 普通用户结束

public Shell(String ip, int port, String user, String password)
{
      try
      {
         telnet.connect(ip, port);
         in = telnet.getInputStream();
         out = new PrintStream(telnet.getOutputStream());
         // 根据root用户设置结束符
         this.prompt = user.equals("root") ? '#' : '$';
         login(user, password);
      }
      catch (Exception e)
      {
         e.printStackTrace();
      }
}

/**
   * 登录
   *
   * @param user
   * @param password
   */
public void login(String user, String password)
{
//       readUntil("login:");
      readUntil("login:");
      write(user);
      readUntil("Password:");
      write(password);
      readUntil(prompt + "");
}

/**
   * 读取分析结果
   *
   * @param pattern
   * @return
   */
public String readUntil(String pattern)
{
      try
      {
         char lastChar = pattern.charAt(pattern.length() - 1);
         StringBuffer sb = new StringBuffer();
         char ch = (char)in.read();
         while (true)
         {
            sb.append(ch);
            if (ch == lastChar)
            {
                  if (sb.toString().endsWith(pattern))
                  {
                     return sb.toString();
                  }
            }
            ch = (char)in.read();
            System.out.print(ch);
         }
      }
      catch (Exception e)
      {
         e.printStackTrace();
      }
      return null;
}

/**
   * 写操作
   *
   * @param value
   */
public void write(String value)
{
      try
      {
         out.println(value);
         out.flush();
      }
      catch (Exception e)
      {
         e.printStackTrace();
      }
}

/**
   * 向目标发送命令字符串
   *
   * @param command
   * @return
   */
public String sendCommand(String command)
{
      try
      {
         write(command);
         return readUntil(prompt + "");
      }
      catch (Exception e)
      {
         e.printStackTrace();
      }
      return null;
}

/**
   * 关闭连接
   */
public void disconnect()
{
      try
      {
         telnet.disconnect();
      }
      catch (Exception e)
      {
         e.printStackTrace();
      }
}
public static void main(String[] args) {
      TelnetClient telnet = new TelnetClient();
      try {

         String cmd="sh /home/test.sh";// sh脚本
         String ip="*.16.16.233";//远程服务器IP地址
         int port=23;//端口号
         String user="root";//远程服务器登录用户名
         String pass="root";//远程服务器登录密码

         Shell she =new Shell(ip, port, user,pass); //远程登录

         she.sendCommand("export LANG=en");//语言
         she.sendCommand(cmd);  //执行sh脚本

         System.out.println("********"+she.sendCommand(cmd));
         she.disconnect();  //关闭

      }catch (Exception e) {
         // TODO: handle exception
      }

}
}

作者: shotting 时间: 2013-3-11 15:19
太水了，这种东西也敢卖分？

欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/)