osd-contiki/examples/sky-shell/src/se/sics/contiki/collect/MoteFinder.java

223 lines
7.5 KiB
Java

/*
* Copyright (c) 2008, Swedish Institute of Computer Science.
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. Neither the name of the Institute nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED. IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*
* $Id: MoteFinder.java,v 1.2 2008/09/03 13:35:21 nifi Exp $
*
* -----------------------------------------------------------------
*
* Motelist
*
* Authors : Joakim Eriksson, Niclas Finne
* Created : 4 jul 2008
* Updated : $Date: 2008/09/03 13:35:21 $
* $Revision: 1.2 $
*/
package se.sics.contiki.collect;
import java.awt.Component;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import javax.swing.JOptionPane;
/**
*
*/
public class MoteFinder {
public static final String MOTELIST_WINDOWS = "./tools/motelist-windows.exe";
public static final String MOTELIST_LINUX = "./tools/motelist-linux";
private final Pattern motePattern;
private final boolean isWindows;
private Process moteListProcess;
// private boolean hasVerifiedProcess;
private ArrayList<String> comList = new ArrayList<String>();
private ArrayList<String> moteList = new ArrayList<String>();
public MoteFinder() {
String osName = System.getProperty("os.name", "").toLowerCase();
isWindows = osName.startsWith("win");
motePattern = Pattern.compile("\\s(COM|/dev/[a-zA-Z]+)(\\d+)\\s");
}
public String[] getMotes() throws IOException {
searchForMotes();
return getMoteList();
}
public String[] getComPorts() throws IOException {
searchForMotes();
return getComList();
}
private void searchForMotes() throws IOException {
comList.clear();
moteList.clear();
// hasVerifiedProcess = false;
/* Connect to COM using external serialdump application */
String fullCommand;
if (isWindows) {
fullCommand = MOTELIST_WINDOWS;
} else {
fullCommand = MOTELIST_LINUX;
}
try {
String[] cmd = new String[] { fullCommand };
moteListProcess = Runtime.getRuntime().exec(cmd);
final BufferedReader input = new BufferedReader(new InputStreamReader(moteListProcess.getInputStream()));
final BufferedReader err = new BufferedReader(new InputStreamReader(moteListProcess.getErrorStream()));
/* Start thread listening on stdout */
Thread readInput = new Thread(new Runnable() {
public void run() {
String line;
try {
while ((line = input.readLine()) != null) {
parseIncomingLine(line);
}
input.close();
} catch (IOException e) {
System.err.println("Exception when reading from motelist");
e.printStackTrace();
}
}
}, "read motelist thread");
/* Start thread listening on stderr */
Thread readError = new Thread(new Runnable() {
public void run() {
String line;
try {
while ((line = err.readLine()) != null) {
System.err.println("Motelist error stream> " + line);
}
err.close();
} catch (IOException e) {
System.err.println("Exception when reading from motelist error stream: " + e);
}
}
}, "read motelist error stream thread");
readInput.start();
readError.start();
// Wait for the motelist program to finish executing
readInput.join();
} catch (Exception e) {
throw (IOException) new IOException("Failed to execute '" + fullCommand + "'").initCause(e);
}
}
private String[] getComList() {
return comList.toArray(new String[comList.size()]);
}
private String[] getMoteList() {
return moteList.toArray(new String[moteList.size()]);
}
public void close() {
if (moteListProcess != null) {
moteListProcess.destroy();
moteListProcess = null;
}
}
protected void parseIncomingLine(String line) {
if (line.contains("No devices found") || line.startsWith("Reference")) {
// No Sky connected or title before connected motes
// hasVerifiedProcess = true;
} else if (line.startsWith("-------")) {
// Separator
} else {
Matcher matcher = motePattern.matcher(line);
if (matcher.find()) {
String dev = matcher.group(1);
String no = matcher.group(2);
String comPort = dev + no;
String moteID = comPort;
if (isWindows) {
// Special handling of mote id under Windows
int moteNumber = Integer.parseInt(no);
moteID = Integer.toString(moteNumber - 1);
}
comList.add(comPort);
moteList.add(moteID);
} else {
System.err.println("Motelist> " + line);
}
}
}
public static String selectComPort(Component parent) {
MoteFinder finder = new MoteFinder();
try {
String[] motes = finder.getComPorts();
if (motes == null || motes.length == 0) {
JOptionPane.showMessageDialog(parent, "Could not find any connected motes.", "No mote found", JOptionPane.ERROR_MESSAGE);
return null;
} else if (motes.length == 1) {
// Only one node found
return motes[0];
} else {
// Several motes found
return (String) JOptionPane.showInputDialog(
parent, "Found multiple connected motes. Please select serial port:",
"Select serial port", JOptionPane.QUESTION_MESSAGE, null, motes, motes[0]);
}
} catch (IOException e) {
e.printStackTrace();
JOptionPane.showMessageDialog(parent, "Failed to search for connected motes:\n" + e, "Error", JOptionPane.ERROR_MESSAGE);
return null;
} finally {
finder.close();
}
}
public static void main(String[] args) throws IOException {
MoteFinder finder = new MoteFinder();
String[] comPorts = args.length > 0 && "-v".equals(args[0]) ?
finder.getMotes() : finder.getComPorts();
finder.close();
if (comPorts == null || comPorts.length == 0) {
System.out.println("No motes connected");
} else {
for(String port: comPorts) {
System.out.println("Found Sky at " + port);
}
}
}
}