做自己的Socket网络通信框架【测试篇-remotecmd】

在上一篇里面讲讲了一个简单测试例子，今天讲一个比较靠近实际的应用的例子 remotecmd（在包com.susu.test.nio.remotecmd），远程命令行。测试程序设计思想，在在客户端输入一个命令，发送给服务器执行，然后服务器把执行结果返回给客户端显示。程序流程如下：

1.写一个CmdLine类实现两个功能一个是做为客户端从控制台读取命令，一个功能是作为服务器执行命令返回执行结果。

2.编写几个重要的类服务器输入过滤器 ，客户端输入过滤器，和输出过滤器，客户端和服务器端的启动类。

3.服务启动在9000进行监听

4.客户端连接服务器的9000端口，当客户端连接成功既框架回调ClientInputFilter的onCreateSession方法，在这个方法里面new一个 CmdLine类，从控制台读取数据，都到数据以后通过session发送给服务器执行。

5.服务器在接收到客户端连接以后 既框架回调ServerInputFilter的onCreateSession方法，在这个方法里面建new一个 CmdLine类（这里只是写简单的测试程序，如果有多个客户端连接上来的话，这里显然就有问题） ，服务器在onReceiveMessage里面收到数据时，调 CmdLine类的execCmd方法执行收到的命令，并吧执行结果发送回客户点。

6.客户端收到服务器的执行结果 onReceiveMessage方法里面进行显示执行结果。

下面就是测试类的代码实现：

CmdLine.java

package com.susu.test.nio.remotecmd; import java.io.BufferedReader; import java.io.InputStreamReader; import java.util.Scanner; import com.susu.common.nio.session.Session; public class CmdLine { private boolean isQuit=false; /** * 控制台打印信息 * @param info */ public void print(String info){ System.out.println(info); System.out.print("-->"); } /** * 执行cmd命令返回执行结果 * @param cmd * @return */ public String execCmd(String cmd){ StringBuffer sb= new StringBuffer(); BufferedReader reader =null; try { if(System.getProperty("os.name").indexOf("Windows")!=-1){ cmd="cmd.exe /c "+cmd; } Process p = Runtime.getRuntime().exec(cmd); p.getOutputStream().close(); reader = new BufferedReader(new InputStreamReader(p.getInputStream())); reader.close(); String line = null; while ((line = reader.readLine()) != null) { sb.append(line).append("/n"); } reader = new BufferedReader(new InputStreamReader(p.getErrorStream())); line = null; while ((line = reader.readLine()) != null) { sb.append(line).append("/n"); } } catch (Exception e) { e.printStackTrace(); }finally{ if(reader!=null){ try { reader.close(); } catch (Exception e2) { e2.printStackTrace(); } } } return sb.toString(); } public void cmdLine(Session session){ Scanner sc = new Scanner(System.in); while(!isQuit){ System.out.print("-->"); String cmd=sc.nextLine(); if(session==null){ break; } if(cmd!=null){ session.wirte(cmd); } } } public static void main(String []s){ CmdLine c= new CmdLine(); System.out.println(c.execCmd("wmic process get name,CommandLine")); // System.out.println(c.execCmd("dir")); } }

ClientInputFilter.java

package com.susu.test.nio.remotecmd; import com.susu.common.CommException; import com.susu.common.nio.filter.InputFilter; import com.susu.common.nio.session.Session; /** * 打印收到的报文数据 */ public class ClientInputFilter implements InputFilter { private int order; private CmdLine cmdLine ; public ClientInputFilter(int order){ this.order=order; } public void destroy() throws CommException { System.out.println(this.toString()+"destroy"); } public int getOrder() { return order; } public void init() throws CommException { System.out.println(this.toString()+"init"); } public void onCloseSession(Session session) throws CommException { System.out.println(this.toString()+"onCloseSession"); } public void onCreateSession(Session session) throws CommException { cmdLine = new CmdLine(); cmdLine.cmdLine(session); } public void onReceiveMessage(Session session) throws CommException { String s=new String((byte[])session.getReceiveMessage()); cmdLine.print(s.substring(8)); } public String toString(){ return order+"-inputFilter:"; } }

 

ServerInputFilter.java

package com.susu.test.nio.remotecmd; import com.susu.common.CommException; import com.susu.common.nio.filter.InputFilter; import com.susu.common.nio.session.Session; /** * 打印收到的报文数据 */ public class ServerInputFilter implements InputFilter { private int order; private CmdLine cmdLine ; public ServerInputFilter(int order){ this.order=order; } public void destroy() throws CommException { System.out.println(this.toString()+"destroy"); } public int getOrder() { return order; } public void init() throws CommException { System.out.println(this.toString()+"init"); } public void onCloseSession(Session session) throws CommException { System.out.println(this.toString()+"onCloseSession"); } public void onCreateSession(Session session) throws CommException { cmdLine = new CmdLine(); } public void onReceiveMessage(Session session) throws CommException { String s=new String((byte[])session.getReceiveMessage()); String ret=cmdLine.execCmd(s.substring(8)); if("".equals(ret)){ ret="操作成功"; } session.wirte(ret); } public String toString(){ return order+"-inputFilter:"; } }

TestOutputFilter.java

package com.susu.test.nio.remotecmd; import java.nio.ByteBuffer; import com.susu.common.CommException; import com.susu.common.nio.filter.OutputFilter; /** * 把要发送的数据变为ByteBuffer 类型 */ public class TestOutputFilter implements OutputFilter { private int order; public TestOutputFilter(int order){ this.order=order; } public void destroy() throws CommException { System.out.println(this.toString()+"destroy"); } public int getOrder() { return order; } public String toString(){ return order+"-outputFilter:"; } public void init() throws CommException { System.out.println(this.toString()+"init"); } public Object onSendMsg(Object msg) { int n=msg.toString().getBytes().length; int len=(""+n).length(); StringBuffer sb= new StringBuffer(1000); for(;len<8;len++){ sb.append("0"); } sb.append(n); sb.append(msg.toString()); byte [] src= sb.toString().getBytes(); ByteBuffer bb=ByteBuffer.allocate(src.length); bb.put(src,0,src.length); return bb; } }

ServerTest.java

package com.susu.test.nio.remotecmd; import com.susu.common.nio.acceptor.NioSocketAcceptor; import com.susu.test.nio.simple.TestIoHandler; /** * * @author zhjb */ public class ServerTest { public static void main(String [] s){ try { NioSocketAcceptor acceptor= new NioSocketAcceptor(); TestIoHandler handler = new TestIoHandler(); acceptor.bind("127.0.0.1", 9000,handler); acceptor.addFilter(new ServerInputFilter(1)); acceptor.addFilter(new TestOutputFilter(1)); acceptor.startupAcceptor(); } catch (Exception e) { e.printStackTrace(); } } }

ClientTest.java

package com.susu.test.nio.remotecmd; import com.susu.common.nio.connector.NioSocketConnector; import com.susu.test.nio.simple.TestIoHandler; /** * 客户端测试类，连上服务器后定时给服务器发送报文 * @author zhjb */ public class ClientTest { static boolean isRunning=true; static NioSocketConnector connector ; static TestIoHandler ioHandler; public static void main(String[] args) { try { connector= new NioSocketConnector(); ioHandler=new TestIoHandler(); connector.addFilter(new ClientInputFilter(2)); connector.addFilter(new TestOutputFilter(2)); connector.init(); connector.connect("127.0.0.1",9000,ioHandler); } catch (Exception e) { e.printStackTrace(); } } }