最近，很多客户向我们反馈终端启动后异常的问题，因此，我自己做了一个远程开关的小工具，该工具的目的在于通过批量的方式来控制终端启动。其设计逻辑是通过服务端发送cmd指令 ,客户端接受并执行指令,把结果返回给服务端。
1.客户端
package testSocket; import java.awt.Dimension;
import java.awt.Rectangle;
import java.awt.Robot;
import java.awt.Toolkit;
import java.awt.image.BufferedImage;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.net.Socket;
import java.net.UnknownHostException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Scanner; import javax.imageio.ImageIO;
import javax.swing.JFrame; //客户端
public class Kehu {
private static String ServiceIp;
private static Integer ServicePort;
private static String imageFormat = "jpg"; //图像文件的格式
private static Dimension d = Toolkit.getDefaultToolkit().getScreenSize();
private static SimpleDateFormat sdfymdhm = new SimpleDateFormat("yyyyMMddHHmm");
private static Socket socket;
private static DataInputStream in;
private static DataOutputStream out; @SuppressWarnings("static-access")
public static void main(String[] args) throws UnknownHostException, IOException, InterruptedException {
JFrame a = new JFrame();
a.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);//设置默认关闭,当关闭窗口时，程序即结束
a.setBounds(100, 100, 300, 200);//设置窗口坐标和大小
a.setLocationRelativeTo(null);//窗口居中
a.setVisible(true);
a.setTitle("客户端");
a.setResizable(false); String curAddr = System.getProperty("user.dir");
//String configAddr = curAddr+"\\mainConfig.txt"; //exe执行时路径
String configAddr = curAddr+"\\src\\config\\mainConfig.txt"; //ide工具执行路径
String[] text=getFile(configAddr);
for (String con : text) {
if("ServiceIp".equals(con.substring(0, con.indexOf("=")))){
ServiceIp = con.substring(con.indexOf("=")+1, con.length());
}else if("ServicePort".equals(con.substring(0, con.indexOf("=")))){
ServicePort = Integer.parseInt(con.substring(con.indexOf("=")+1, con.length()));
}
}
while(true){
socket = new Socket(ServiceIp,ServicePort);
in = new DataInputStream(socket.getInputStream());
out = new DataOutputStream(socket.getOutputStream());
String s = "";
try {
s = in.readUTF();
} catch (Exception e) {
while(true){
try {
socket = new Socket(ServiceIp,ServicePort);
in = new DataInputStream(socket.getInputStream());
out = new DataOutputStream(socket.getOutputStream());
} catch (Exception e2) {
e.printStackTrace();
Thread.currentThread().sleep(1000*60*5);
} finally{
if(socket!=null){
break;
}
}
}
}
snapShot(curAddr,out);
out.close();
in.close();
socket.close();
callCmd(s);
}
} public static String[] getFile(String realpath) {
Scanner scanner = null;
String[] text = new String[4];
int i = 0;
try {
File file= new File(realpath);
scanner = new Scanner(file);
} catch (FileNotFoundException e) {
e.printStackTrace();
}
if(scanner!=null){
while(scanner.hasNextLine()){
text[i]= scanner.nextLine();
i++;
}
scanner.close();
}
return text;
} public static void snapShot(String addr,DataOutputStream outs) {
try {
//拷贝屏幕到一个BufferedImage对象screenshot
BufferedImage screenshot = (new Robot()).createScreenCapture(new Rectangle(0, 0,(int) d.getWidth(), (int) d.getHeight()));
//根据文件前缀变量和文件格式变量，自动生成文件名
String name = addr+"\\"+sdfymdhm.format(new Date())+".jpg";
File f = new File(name);
ImageIO.write(screenshot, imageFormat, f);
FileInputStream fin = new FileInputStream(f);
byte[] sendByte = new byte[1024];
outs.writeUTF(f.getName());
int length = 0;
while((length = fin.read(sendByte, 0, sendByte.length))>0){
outs.write(sendByte,0,length);
outs.flush();
}
fin.close();
f.delete();
} catch (Exception ex) {
System.out.println(ex);
}
} //执行CMD命令
public static void callCmd(String cmd) throws IOException{
Runtime.getRuntime().exec(cmd);
} }
2.服务端
package testFrame; import java.awt.Color;
import java.awt.Font;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.Scanner;
import java.util.Timer; import javax.swing.BorderFactory;
import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JOptionPane;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea; //服务器
public class testMain extends JFrame{
private static final long serialVersionUID = 1L;
private static String logAddr = "";
private static String logImageAddr = "";
private static SimpleDateFormat sdfymd = new SimpleDateFormat("yyyy-MM-dd");
private static SimpleDateFormat sdfymdhm = new SimpleDateFormat("yyyy-MM-dd HH:mm");
private static SimpleDateFormat sdfymdhms = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
private static String curLogAddr = "";
final String LOOPBACK_ADDRESS = "127.0.0.1";
private static DataOutputStream out;
private static DataInputStream in;
private static Socket CurSocket;
private static String ServiceIp;
private static Integer ServicePort;
private static Map<String,Socket> socketList = new HashMap<String,Socket>(); //客户端集合信息
private static FileOutputStream fout = null;
private static byte[] inputByte = null;
private static Timer timer;
public static void main(String[] args) throws IOException {
Date d = new Date();
String logDateFile = sdfymd.format(d); String curAddr = System.getProperty("user.dir");
//String configAddr = curAddr+"\\mainConfig.txt"; //exe执行时路径
String configAddr = curAddr+"\\src\\config\\mainConfig.txt"; //ide工具执行路径
String[] text=getFile(configAddr);
for (String con : text) {
if("logAddr".equals(con.substring(0, con.indexOf("=")))){
logAddr = con.substring(con.indexOf("=")+1, con.length());
judeDirExists(new File(con.substring(con.indexOf("=")+1, con.length())));
}else if("logImageAddr".equals(con.substring(0, con.indexOf("=")))){
logImageAddr = con.substring(con.indexOf("=")+1, con.length());
judeDirExists(new File(con.substring(con.indexOf("=")+1, con.length())));
}else if("ServiceIp".equals(con.substring(0, con.indexOf("=")))){
ServiceIp = con.substring(con.indexOf("=")+1, con.length());
}else if("ServicePort".equals(con.substring(0, con.indexOf("=")))){
ServicePort = Integer.parseInt(con.substring(con.indexOf("=")+1, con.length()));
}
}
curLogAddr = logAddr+"\\unitTest_"+logDateFile+".log";
judeFileExists(new File(curLogAddr));
FileWriter fw = new FileWriter(curLogAddr,true);
PrintWriter pw = new PrintWriter(fw);
if(!"".equals(logAddr)||!"".equals(logImageAddr)||!"".equals(ServiceIp)||!"".equals(ServicePort)){
pw.write(sdfymdhms.format(new Date())+": 启动成功!\r\n");
pw.flush();
fw.flush();
pw.close();
fw.close();
testMain myFrame = new testMain();
myFrame.setVisible(true);
myFrame.setTitle("自动开关机工具");
myFrame.setResizable(false);
newService();
}else{
pw.write(sdfymdhms.format(new Date())+": 启动失败,mainConfig.txt配置文件异常!\r\n");
pw.flush();
fw.flush();
pw.close();
fw.close();
}
} public testMain() throws IOException{
setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);//设置默认关闭,当关闭窗口时，程序即结束
setBounds(100, 100, 600, 360);//设置窗口坐标和大小
setLocationRelativeTo(null);//窗口居中
Font fontStyle = new Font(Font.DIALOG,Font.BOLD, 18);
Font fontConton = new Font(Font.DIALOG,Font.LAYOUT_NO_LIMIT_CONTEXT, 18); JPanel panel = new JPanel(); JLabel label = new JLabel("IP地址:");
label.setFont(fontStyle);
panel.add(label); final JTextArea IpArea = new JTextArea(10, 35);
IpArea.setLineWrap(true);
IpArea.setFont(fontConton);
JScrollPane IpPane =new JScrollPane(IpArea);
IpPane.setVerticalScrollBarPolicy(JScrollPane.VERTICAL_SCROLLBAR_AS_NEEDED);
panel.add(IpPane); JLabel labela = new JLabel("重启效率:");
labela.setFont(fontStyle);
panel.add(labela); final JTextArea DateArea = new JTextArea(1, 3);
DateArea.setLineWrap(false);
DateArea.setBorder(BorderFactory.createLineBorder(Color.black,1));
DateArea.setFont(fontConton);
panel.add(DateArea); JLabel labeb = new JLabel("(分钟/每次) ");
labeb.setFont(fontStyle);
panel.add(labeb); final JLabel labec = new JLabel(""); JButton btn = new JButton("发送");
btn.setFont(fontStyle);
btn.addActionListener(new ActionListener() {
@Override
public void actionPerformed(ActionEvent e) {
if(DateArea.getText().isEmpty()||IpArea.getText().isEmpty()){
JOptionPane.showMessageDialog(null, "请输入IP地址和重启效率");
}else{
try {
timerFunction(DateArea.getText(),IpArea.getText(),labec);
} catch (IOException e1) {
e1.printStackTrace();
}
}
}
});
panel.add(btn); labec.setFont(fontStyle);
labec.setForeground(Color.red);
panel.add(labec); this.setContentPane(panel); this.addWindowListener(new WindowAdapter() {
@Override
public void windowClosing(WindowEvent e){
FileWriter fw = null;
try {
fw = new FileWriter(curLogAddr,true);
} catch (IOException e1) {
e1.printStackTrace();
}
PrintWriter pw = new PrintWriter(fw);
pw.write(sdfymdhms.format(new Date())+": 关闭成功!\r\n");
try {
pw.flush();
fw.flush();
pw.close();
fw.close();
} catch (IOException e1) {
e1.printStackTrace();
}
System.exit(0);
}
}); } public static void timerFunction(final String timeSize,final String iptext,final JLabel labec) throws IOException{
final int timecount = Integer.parseInt(timeSize);
if(timer!=null){
timer.cancel();
}
timer = new Timer();
timer.schedule(
new java.util.TimerTask() {
public void run() {
timeFunctionContion(iptext,labec);
}
}, 0, timecount*60*1000);
} public static String[] getFile(String realpath) {
Scanner scanner = null;
String[] text = new String[4];
int i = 0;
try {
File file= new File(realpath);
scanner = new Scanner(file);
} catch (FileNotFoundException e) {
e.printStackTrace();
}
if(scanner!=null){
while(scanner.hasNextLine()){
text[i]= scanner.nextLine();
i++;
}
scanner.close();
}
return text;
} //校验IP
public static boolean ipCheck(String text) {
if (text != null && !text.isEmpty()) {
// 定义正则表达式
String regex = "^(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|[1-9])\\."
+"(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|\\d)\\."
+"(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|\\d)\\."
+"(1\\d{2}|2[0-4]\\d|25[0-5]|[1-9]\\d|\\d)$";
// 判断ip地址是否与正则表达式匹配
if (text.matches(regex)) {
return true;
} else {
return false;
}
}
return false;
} //判断文件夹是否存在
public static void judeDirExists(File file) {
if (!file.exists()) {
file.mkdir();
}
} //判断文件是否存在
public static void judeFileExists(File file) {
if (!file.exists()) {
try {
file.createNewFile();
} catch (Exception e) {
e.printStackTrace();
}
}
} //通过IP地址获取MAC地址
public static String getMACAddress(String ip) throws Exception {
String macAddress = "";
try {
String cmd = "ping " + ip;
String ret = callCmd(cmd);
if(ret!=null &&!"".equals(ret)){//判断IP地址是否存在
cmd = "nbtstat -a " + ip;
macAddress = callCmdMAC(cmd);
}
} catch (IOException e) {
e.printStackTrace();
}
return macAddress;
} //获取MAC地址
public static String callCmdMAC(String cmd) throws IOException{
Process p = Runtime.getRuntime().exec(cmd);
InputStreamReader isr = new InputStreamReader(p.getInputStream(),"GB2312");
BufferedReader br = new BufferedReader(isr);
String line="";
int startIndex = 0;
while ((line = br.readLine()) != null) {
startIndex = line.indexOf("MAC 地址");
if(startIndex>0){
line = line.substring(startIndex+8, line.length()).trim().toUpperCase();
break;
}
}
br.close();
return line;
} //创建服务器端
public static String callCmd(String cmd) throws IOException{
Process p = Runtime.getRuntime().exec(cmd);
InputStreamReader isr = new InputStreamReader(p.getInputStream());
BufferedReader br = new BufferedReader(isr);
String line="";
String ret="";
while ((line = br.readLine()) != null) {
if (line != null) {
ret =line;
}
}
br.close();
return ret;
} //创建服务器端
public static void newService() throws IOException{
ServerSocket serverSocket=new ServerSocket(ServicePort);
while(true){
Socket sk = serverSocket.accept(); //当有客户端连接时，产生阻塞
socketList.put(sk.getInetAddress().toString(), sk);
FileWriter fw = new FileWriter(curLogAddr,true);
PrintWriter pw = new PrintWriter(fw);
pw.write(sdfymdhms.format(new Date())+": 新的客户端连接!"+sk.getInetAddress()+":"+sk.getPort()+"\r\n");
System.out.println("新的客户端连接"+sk.getInetAddress()+":"+sk.getPort());
pw.flush();
fw.flush();
pw.close();
fw.close();
}
} public static void timeFunctionContion(String iptext,JLabel labec){
FileWriter fw = null;
try {
fw = new FileWriter(curLogAddr,true);
} catch (IOException e) {
e.printStackTrace();
}
PrintWriter pw = new PrintWriter(fw);
Date d = new Date();
String dateNowStr = sdfymdhm.format(d);
labec.setText("发送成功,上次发送时间"+dateNowStr);
String[] iplist = iptext.split(",");
for (String ip : iplist) {
if(ipCheck(ip)){
try {
CurSocket = socketList.get("/"+ip);
out=new DataOutputStream(CurSocket.getOutputStream()); //信息接发流
out.writeUTF("shutdown -r -t 20");
in = new DataInputStream(CurSocket.getInputStream());
String curIPImgAddFol = logImageAddr+"\\"+ip;
judeDirExists(new File(curIPImgAddFol));
fout = new FileOutputStream(new File(curIPImgAddFol+"\\"+in.readUTF()));
inputByte = new byte[1024];
int length = 0;
while (true) {
if (in != null) {
length = in.read(inputByte, 0, inputByte.length);
}
if (length == -1) {
break;
}
fout.write(inputByte, 0, length);
fout.flush();
}
pw.write(sdfymdhms.format(new Date())+": "+ip+"重启成功\r\n");
fout.close();
} catch (Exception e) {
e.printStackTrace();
}
}else{
pw.write(sdfymdhms.format(new Date())+": "+ip+" 地址不合法!\r\n");
}
}
try {
fw.flush();
pw.close();
fw.close();
pw.flush();
} catch (IOException e) {
e.printStackTrace();
}
}
}
如有问题联系我QQ：316349645