command.java

The WLAN Traffic Visualizer is a platform independent Java program providing accurate measurement of

package model;

//Copyright (C) 2008 Harald Unander, Wang Wenjuan
//
//    This file is part of WlanTV.
//
//    WlanTV is free software: you can redistribute it and/or modify
//    it under the terms of the GNU General Public License as published by
//    the Free Software Foundation, either version 3 of the License, or
//    (at your option) any later version.
//
//    WlanTV is distributed in the hope that it will be useful,
//    but WITHOUT ANY WARRANTY; without even the implied warranty of
//    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//    GNU General Public License for more details.
//
//    You should have received a copy of the GNU General Public License
//    along with WlanTV.  If not, see <http://www.gnu.org/licenses/>.

import java.io.BufferedReader;
import java.io.InputStreamReader;

import javax.swing.JOptionPane;

public abstract class Command {
	private Process p;
	private boolean abort;

	public void execute(String cmd) throws Exception {
		System.out.println(cmd);

		p = Runtime.getRuntime().exec(cmd);

		BufferedReader stdInput = new BufferedReader(new 
				InputStreamReader(p.getInputStream()));

		BufferedReader stdError = new BufferedReader(new 
				InputStreamReader(p.getErrorStream()));

		boolean memOk = true;
		abort = false;
		String s;
		while ((s = stdInput.readLine()) != null) {
			if (!heapOk()) {
				p.destroy();
				memOk = false;
				break;
			}
			if (abort) {
				p.destroy();
				break;
			}
			newLine(s);
		}
		
		if (!memOk) {
			throw (new LowMemException());
		}

		while ((s = stdError.readLine()) != null) {
			if (s.startsWith("Capturing on")) {
				throw (new Exception("No write access to log directory"));
			}			
			else if (!s.contains("Ericsson") && !s.contains("This could be dangerous.")) {
				throw (new Exception(s));
			}
		}

		try {
			p.waitFor();
		}
		catch (InterruptedException e) {
//			System.out.println("CMD process was interrupted");
			e.printStackTrace();
			JOptionPane.showMessageDialog(null,e.getMessage());
		}

//		if (p.exitValue() != 0) {
//			System.out.println("CMD exit value was non zero: "+p.exitValue());
//			JOptionPane.showMessageDialog(null,"CMD exit value was non zero: "+p.exitValue());
//		}

		stdError.close();
		stdInput.close();
	}

	public void abortCapture() {
		abort = true;
	}

	public abstract void newLine(String line) throws InterruptedException;
	
	public boolean heapOk() {
		return true;
	}

	@SuppressWarnings("serial")
	public class LowMemException extends Exception {
	}
}