testthread.java~22~

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;

/**
 * <p>Title: </p>
 *
 * <p>Description: </p>
 *
 * <p>Copyright: Copyright (c) 2005</p>
 *
 * <p>Company: </p>
 *
 * @author not attributable
 * @version 1.0
 */
class testThread extends Thread {
    ServerSocket ssock;
    Socket sock;
    String strpath;
    private boolean suspendFlag = false;
    public testThread(String path) {
        this.strpath = path;
        System.err.println("Thread Start !");
        suspendFlag = true;
        start();
        try {
            jbInit();
        } catch (Exception ex) {
            ex.printStackTrace();
        }
    }

    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() {
        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 {
                //FileReader fread = new FileReader("1234.txt");
                // BufferedReader finput = new BufferedReader(fread);
                //BufferedReader finput = new BufferedReader(fread);
                // Receive the Client's Information

                // ObjectInputStream input = new ObjectInputStream(sock.
                //         getInputStream());
                // line = (String) input.readObject();
                //SendFirewall();
                InputStreamReader pss = new InputStreamReader(sock.
                        getInputStream());
                BufferedReader ins = new BufferedReader(pss);
                String strnumber = ins.readLine();
                int number = (new Integer(strnumber)).intValue();
                //line = ins.readLine();
                //line_new = line.substring(1, line.length());
                //System.err.println(line_new);
                //rule = ins.readLine();
                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(
                                            //        "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;
                        }
                        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);
                        /*String str ="[auto IP xxxx]";
                                             int indextemp=buffer.indexOf(str);
                         buffer.insert(indextemp+str.length()+1,"\r\n"+"insert a line rule"+  "\r\n");*/
                        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() {
        suspendFlag = false;
    }

    private void jbInit() throws Exception {
    }
}