RiverRat  - An MIT iCampus project

http://web.mit.edu/riverrat/

StreamLogger.java

package riverrat.testing;

import java.net.*;
import java.io.*;

public class StreamLogger {
        protected static String file = "out.xml";
        
        protected static String host = "localhost";
        
        protected static int port = 8080;
        
        public static void printUsage() {
                System.out.println("`XMLLogger' is a test data logger for the RiverRat project.");
                System.out.println("usage: java riverrat.testing.XMLLogger [OPTIONS]");
                System.out.println("");
                System.out.println("Options:");
                System.out.println("  -h, --help\t\t\tPrint this message and exit.");
                System.out.println("  -p, --port PORT\t\tUse server port PORT.");
                System.out.println("  -o, --out\t\t\tFile to write to.");
                System.out.println("  -s, --server SERVER\t\tConnect to host SERVER.");
                System.out.println("");
                System.out.println("Author: Scott Torborg 2004");
                System.out.println("RiverRat Project http://mit.edu/riverrat");
                System.exit(1);
        }
        
        public static void main(String args[]) {
                int n = args.length;
                //parse args
                for(int i = 0; i < n; i++) {
                        String s = args[i].toLowerCase();
                        if(s.equals("--help") || s.equals("-h")) {
                                printUsage();
                        } else if(s.equals("--port") || s.equals("-p")) {
                                if(i+1 >= n) printUsage();
                                i++;
                                try {
                                        port = Integer.parseInt(args[i]);
                                } catch(NumberFormatException e) {
                                        System.err.println("Bad TCP port number.");
                                        System.exit(1);
                                }
                        } else if(s.equals("--out") || s.equals("-o")) {
                                if(i+1 >= n) printUsage();
                                i++;
                                file = args[i];
                        } else if(s.equals("--server") || s.equals("-s")) {
                                if(i+1 >= n) printUsage();
                                i++;
                                host = args[i];
                        }
                }
                
                //do the stuff
                
                FileOutputStream output = null;
                
                //open file
                try {
                        output = new FileOutputStream(file);
                } catch(FileNotFoundException e) {
                        System.err.println("File cannot be created.");
                        System.exit(1);
                }
                
                Socket s = null;
                InputStream stream = null;
                int data = 0;
                
                //open socket to host:port
                try {
                        s = new Socket(host, port);
                        try {
                                System.out.println("Connection accepted: "+s);
                                stream = s.getInputStream();
                                //echo data from socket to file
                                while((data = stream.read()) >= 0) {
                                        output.write(data);
                                }
                        } finally {
                                s.close();
                        }
                } catch(SocketException e) {
                        System.err.println("SocketException caught: "+e.getMessage());
                } catch(Exception e) {
                        System.err.println("Error: "+e.getMessage());
                }
        }
}

 

Brought to you by the RiverRat team.