fwserverthread.java~5~

Firewall Application & source code.

package fwserver;

import java.io.OutputStreamWriter;
import java.io.IOException;
import java.io.FileWriter;
import java.net.ServerSocket;
import java.io.FileInputStream;
import java.io.InputStreamReader;
import java.io.FileNotFoundException;
import java.io.FileReader;
import java.io.BufferedWriter;
import java.io.BufferedReader;
import java.net.Socket;
import fwserver.ShutDownPc;

/**
 * Copyright: Copyright (c) 2005 CH 版权所有
 * 生成日期 2005-6-5
 * 作者姓名 崔灏
 * 类的名称 FwServerThread 父类 Thread
 * 功能描述 建立服务器，监听客户端的连接，交互信息
 * 字段清单 ssock ; 服务器套接字
 *         sock ；客户端连接套接字
 *         suspendFlag；线程执行标志
 * 方法清单  public void SendFirewall()
 *          public void run()
 *          public void mysuspend()
 * @version 1.0
 */
public class FwServerThread extends Thread {
    ServerSocket ssock;
    Socket sock;
    String strpath;
    private boolean suspendFlag = false;

    public FwServerThread(String path) {
        this.strpath = path;
        System.err.println("Thread Start !");
        suspendFlag = true;
        start();

    }

    /**************************************************************
     *函数名称：public void SendFirewall()
     *功能描述：类FwServerThread的方法，完成发送本地防火墙配制文件的功能
     *入口参数：无
     *出口参数：无
     *返回值：无
     *调用者：函数public void run()
     **************************************************************/
    public void SendFirewall() {

        try {
            OutputStreamWriter pstest = new OutputStreamWriter(sock.
                    getOutputStream());
            FileReader freadtest = new FileReader(strpath);
            BufferedReader finputtest = new BufferedReader(freadtest);
            String tempss = new String();
            StringBuffer testbuffer = new StringBuffer();
            tempss = finputtest.readLine();
            while (tempss != null) {
                testbuffer.append(tempss + "\r\n");
                tempss = finputtest.readLine();
            }
            pstest.write(testbuffer.toString());
            pstest.flush();
        } catch (Exception eee) {
            System.err.println("Error In Read FireRule ");
        }

    }

    /**************************************************************
     *函数名称：public void run()
     *功能描述：类FwServerThread的方法，完成建立服务器，监听客户端的连接，交互信息功能
     *入口参数：无
     *出口参数：无
     *返回值：无
     *调用者：函数public void run()
     **************************************************************/

    public void run() {
        FileInputStream fin;
        FileWriter fout = null;
        StringBuffer buffer = new StringBuffer();
        String line;
        String rule;
        String line_new;
        while (suspendFlag) {
            try {
                //建立服务器
                ssock = new ServerSocket(1234);
                System.out.println("Listening");
                sock = ssock.accept();
                ssock.close();
                System.out.println("Get Connection ---");

            } catch (Exception ee) {
                System.err.print(ee);
            }

            try {
             //接收到客户端的消息，分析消息，将规则添加到firewall的规则中
                InputStreamReader pss = new InputStreamReader(sock.
                        getInputStream());
                BufferedReader ins = new BufferedReader(pss);
                String strnumber = ins.readLine();
                int number = (new Integer(strnumber)).intValue();
                System.err.println(number);
                for (int i = 0; i < number; i++) {
                    line = ins.readLine();
                    rule = ins.readLine();
                    System.err.println("line" + line);
                    System.err.println("rule" + rule);
                    if (!line.equals(null)) {
                        line_new = line.substring(1, line.length());
                    } else {
                        line_new = "";
                    }

                    //FileReader fread = new FileReader("/E:/firewall.ini");
                    FileReader fread = new FileReader(strpath);
                    BufferedReader finput = new BufferedReader(fread);
                    byte buf[] = line.getBytes();
                    String temp = new String();
                    temp = finput.readLine();
                    while (temp != null) {
                        if (line_new.equals(" [MANUAL UDP low REJECT]")) { //多一空格
                            while (temp != null) {
                                buffer.append(temp + "\r\n");
                                temp = finput.readLine();
                                if (temp != null) {
                                    while (temp != null) {
                                        if (temp.equals(
                                                "[MANUAL UDP low REJECT]")) {
                                            buffer.append(temp + "\r\n");
                                            buffer.append(rule + "\r\n");
                                            temp = finput.readLine();
                                        } else {
                                            buffer.append(temp + "\r\n");
                                            temp = finput.readLine();
                                        }
                                    }
                                }
                            }
                            break;
                        }
                        if (line.equals("[ShutDownPC]")) {
                            System.err.println("SHUT DOWN");
                            ShutDownPc shutpc = new ShutDownPc();
                            shutpc.shutdown();
                            return;
                        }
                        if (line.equals("[CreatFile]")) {
                            System.err.println("Creat File");
                            SendFirewall();
                            while (temp != null) {
                                buffer.append(temp + "\r\n");
                                temp = finput.readLine();
                                if (temp != null) {
                                    while (temp != null) {
                                        if (temp.equals("[MANUAL IP ACCEPT]")) {
                                            buffer.append(temp + "\r\n");
                                            buffer.append(rule + "\r\n");
                                            temp = finput.readLine();
                                        } else {
                                            buffer.append(temp + "\r\n");
                                            temp = finput.readLine();
                                        }
                                    }
                                }
                            }
                            break;

                        }

                        if (line_new.equals(" [MANUAL IP ACCEPT]")) {
                            while (temp != null) {
                                buffer.append(temp + "\r\n");
                                temp = finput.readLine();
                                if (temp != null) {
                                    while (temp != null) {
                                        if (temp.equals("[MANUAL IP ACCEPT]")) {
                                            buffer.append(temp + "\r\n");
                                            //buffer.append(
                                            //        "REJECT, 1024 - 60000, default, 2005-01-01 00:00:00, PERPETUAL, 1000, unknown" +
                                            //        "\r\n");
                                            buffer.append(rule + "\r\n");
                                            temp = finput.readLine();
                                        } else {
                                            buffer.append(temp + "\r\n");
                                            temp = finput.readLine();
                                        }
                                    }
                                }
                            }
                            break;
                        }
                        temp = finput.readLine();
                    }
                    finput.close();
                    fread.close();
                    //将规则写入文件
                    try {
                        // fout = new FileWriter("/E:/firewall.ini");
                        fout = new FileWriter(strpath);
                        BufferedWriter writer = new BufferedWriter(fout);
                        fout.write(buffer.toString());
                        sock.close();
                        fout.close();

                        buffer.delete(0, buffer.length());

                    } catch (IOException ee) {
                        System.out.println("Write  File  Error ");
                    }

                    //rule = ins.readLine();
                }

            } catch (Exception eee) {
                System.out.println("Read File Error");
                System.out.println(eee);
            }

        }
    }
    /**************************************************************
     *函数名称：public void mysuspend()
     *功能描述：类FwServerThread的方法，完成结束线程功能
     *入口参数：无
     *出口参数：无
     *返回值：无
     *调用者：函数public void run()
     **************************************************************/
    public void mysuspend() {
        suspendFlag = false;
    }


}