271 lines
8.1 KiB
Java
271 lines
8.1 KiB
Java
/*
|
|
* Copyright (c) 2010, 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.
|
|
*
|
|
*/
|
|
|
|
import java.awt.BorderLayout;
|
|
import java.awt.Dimension;
|
|
import java.io.DataInputStream;
|
|
import java.io.DataOutputStream;
|
|
import java.io.IOException;
|
|
import java.net.ServerSocket;
|
|
import java.net.Socket;
|
|
import java.util.Collection;
|
|
import java.util.Observable;
|
|
import java.util.Observer;
|
|
|
|
import javax.swing.BorderFactory;
|
|
import javax.swing.Box;
|
|
import javax.swing.JComponent;
|
|
import javax.swing.JLabel;
|
|
import javax.swing.JPanel;
|
|
import javax.swing.SwingUtilities;
|
|
|
|
import org.apache.log4j.Logger;
|
|
import org.jdom.Element;
|
|
|
|
import se.sics.cooja.ClassDescription;
|
|
import se.sics.cooja.GUI;
|
|
import se.sics.cooja.Mote;
|
|
import se.sics.cooja.MotePlugin;
|
|
import se.sics.cooja.PluginType;
|
|
import se.sics.cooja.Simulation;
|
|
import se.sics.cooja.VisPlugin;
|
|
import se.sics.cooja.interfaces.SerialPort;
|
|
|
|
/**
|
|
* Socket to simulated serial port forwarder. Server version.
|
|
*
|
|
* @author Fredrik Osterlind
|
|
*/
|
|
@ClassDescription("Serial Socket (SERVER)")
|
|
@PluginType(PluginType.MOTE_PLUGIN)
|
|
public class SerialSocketServer extends VisPlugin implements MotePlugin {
|
|
private static final long serialVersionUID = 1L;
|
|
private static Logger logger = Logger.getLogger(SerialSocketServer.class);
|
|
|
|
private final static int LABEL_WIDTH = 100;
|
|
private final static int LABEL_HEIGHT = 15;
|
|
|
|
public final int LISTEN_PORT;
|
|
|
|
private SerialPort serialPort;
|
|
private Observer serialDataObserver;
|
|
|
|
private JLabel statusLabel, inLabel, outLabel;
|
|
private int inBytes = 0, outBytes = 0;
|
|
|
|
private ServerSocket server;
|
|
private Socket client;
|
|
private DataInputStream in;
|
|
private DataOutputStream out;
|
|
|
|
private Mote mote;
|
|
|
|
public SerialSocketServer(Mote mote, Simulation simulation, final GUI gui) {
|
|
super("Serial Socket (SERVER) (" + mote + ")", gui, false);
|
|
this.mote = mote;
|
|
|
|
LISTEN_PORT = 60000 + mote.getID();
|
|
|
|
/* GUI components */
|
|
if (GUI.isVisualized()) {
|
|
Box northBox = Box.createHorizontalBox();
|
|
northBox.setBorder(BorderFactory.createEmptyBorder(5, 5, 5, 5));
|
|
statusLabel = configureLabel(northBox, "", "");
|
|
|
|
Box mainBox = Box.createHorizontalBox();
|
|
mainBox.setBorder(BorderFactory.createEmptyBorder(0, 5, 5, 5));
|
|
inLabel = configureLabel(mainBox, "socket -> mote:", "0 bytes");
|
|
outLabel = configureLabel(mainBox, "mote -> socket", "0 bytes");
|
|
|
|
getContentPane().add(BorderLayout.NORTH, northBox);
|
|
getContentPane().add(BorderLayout.CENTER, mainBox);
|
|
pack();
|
|
}
|
|
|
|
/* Mote serial port */
|
|
serialPort = (SerialPort) mote.getInterfaces().getLog();
|
|
if (serialPort == null) {
|
|
throw new RuntimeException("No mote serial port");
|
|
}
|
|
|
|
try {
|
|
logger.info("Listening on port: " + LISTEN_PORT);
|
|
if (GUI.isVisualized()) {
|
|
statusLabel.setText("Listening on port: " + LISTEN_PORT);
|
|
}
|
|
server = new ServerSocket(LISTEN_PORT);
|
|
new Thread() {
|
|
public void run() {
|
|
while (server != null) {
|
|
try {
|
|
client = server.accept();
|
|
in = new DataInputStream(client.getInputStream());
|
|
out = new DataOutputStream(client.getOutputStream());
|
|
out.flush();
|
|
|
|
startSocketReadThread(in);
|
|
if (GUI.isVisualized()) {
|
|
statusLabel.setText("Client connected: " + client.getInetAddress());
|
|
}
|
|
} catch (IOException e) {
|
|
logger.fatal("Listening thread shut down: " + e.getMessage());
|
|
server = null;
|
|
cleanupClient();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}.start();
|
|
} catch (Exception e) {
|
|
throw (RuntimeException) new RuntimeException(
|
|
"Connection error: " + e.getMessage()).initCause(e);
|
|
}
|
|
|
|
/* Observe serial port for outgoing data */
|
|
serialPort.addSerialDataObserver(serialDataObserver = new Observer() {
|
|
public void update(Observable obs, Object obj) {
|
|
try {
|
|
if (out == null) {
|
|
/*logger.debug("out is null");*/
|
|
return;
|
|
}
|
|
out.write(serialPort.getLastSerialData());
|
|
out.flush();
|
|
outBytes++;
|
|
if (GUI.isVisualized()) {
|
|
outLabel.setText(outBytes + " bytes");
|
|
}
|
|
} catch (IOException e) {
|
|
cleanupClient();
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
private void startSocketReadThread(final DataInputStream in) {
|
|
/* Forward data: virtual port -> mote */
|
|
Thread incomingDataThread = new Thread(new Runnable() {
|
|
public void run() {
|
|
int numRead = 0;
|
|
byte[] data = new byte[1024];
|
|
logger.info("Forwarder: socket -> serial port");
|
|
while (true) {
|
|
numRead = -1;
|
|
try {
|
|
numRead = in.read(data);
|
|
} catch (IOException e) {
|
|
numRead = -1;
|
|
}
|
|
|
|
if (numRead >= 0) {
|
|
for (int i=0; i < numRead; i++) {
|
|
serialPort.writeByte(data[i]);
|
|
}
|
|
inBytes += numRead;
|
|
if (GUI.isVisualized()) {
|
|
inLabel.setText(inBytes + " bytes");
|
|
}
|
|
} else {
|
|
cleanupClient();
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
});
|
|
incomingDataThread.start();
|
|
}
|
|
|
|
private JLabel configureLabel(JComponent pane, String desc, String value) {
|
|
JPanel smallPane = new JPanel(new BorderLayout());
|
|
JLabel label = new JLabel(desc);
|
|
label.setPreferredSize(new Dimension(LABEL_WIDTH,LABEL_HEIGHT));
|
|
smallPane.add(BorderLayout.WEST, label);
|
|
label = new JLabel(value);
|
|
label.setPreferredSize(new Dimension(LABEL_WIDTH,LABEL_HEIGHT));
|
|
smallPane.add(BorderLayout.CENTER, label);
|
|
pane.add(smallPane);
|
|
return label;
|
|
}
|
|
|
|
public boolean setConfigXML(Collection<Element> configXML, boolean visAvailable) {
|
|
return true;
|
|
}
|
|
|
|
public Collection<Element> getConfigXML() {
|
|
return null;
|
|
}
|
|
|
|
private void cleanupClient() {
|
|
try {
|
|
if (client != null) {
|
|
client.close();
|
|
client = null;
|
|
}
|
|
} catch (IOException e1) {
|
|
}
|
|
try {
|
|
if (in != null) {
|
|
in.close();
|
|
in = null;
|
|
}
|
|
} catch (IOException e) {
|
|
}
|
|
try {
|
|
if (out != null) {
|
|
out.close();
|
|
out = null;
|
|
}
|
|
} catch (IOException e) {
|
|
}
|
|
|
|
if (GUI.isVisualized()) {
|
|
SwingUtilities.invokeLater(new Runnable() {
|
|
public void run() {
|
|
statusLabel.setText("Listening on port: " + LISTEN_PORT);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
public void closePlugin() {
|
|
cleanupClient();
|
|
serialPort.deleteSerialDataObserver(serialDataObserver);
|
|
try {
|
|
server.close();
|
|
} catch (IOException e) {
|
|
}
|
|
}
|
|
|
|
public Mote getMote() {
|
|
return mote;
|
|
}
|
|
|
|
}
|
|
|