wlantvproperties.java

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

package main;

//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.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;
import java.util.Enumeration;
import java.util.Properties;

@SuppressWarnings("serial")
public class WlanTvProperties extends Properties{
	private static String propFile = System.getProperty("user.dir")+File.separator+"wlantv.ini";
	private static String comments =
		"log.dir = updated automatically to last log location\n"+
		"livelog.dir = updated automatically to last live log location\n"+
		"reconstruct.size = size of reconstructed packets\n"+
		"bad.fcs = include | exclude\n"+
		"wireshark.len.compensation = on | off\n";

	public WlanTvProperties() throws Exception {
		if (new File(propFile).exists()) {
			try {
				load(new FileInputStream(propFile));
			} catch (IOException e) {
				System.err.println(e);
			}
		}
		else {
			setProperty("log.dir",System.getProperty("user.dir"));
			setProperty("livelog.dir",System.getProperty("user.dir"));
			setProperty("bad.fcs","include");
			setProperty("reconstruct.size","1470");
			setProperty("wireshark.len.compensation","off");
			store(new FileOutputStream(propFile),comments);
		}
	}

	public void updateProperty(String key, String val) {
//		System.out.println(key+" "+val);
		setProperty(key, val);
		try {
			store(new FileOutputStream(propFile),comments);
		} catch (FileNotFoundException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}

	public static String replaceMacAddrWithAlias(String s,boolean showAddr) {
		String ps;
		for (Enumeration<Object> e = Main.properties.keys(); e.hasMoreElements();) {
			ps = (String)e.nextElement();
			if (s.contains(ps)) {
				if (showAddr)
					s = s.replace(ps, ps+" "+Main.properties.getProperty(ps));
				else
					s = s.replace(ps, Main.properties.getProperty(ps));
			}				
		}
		return s;
	}
}