networksource.java

来自「用于传感器网络的节点操作系统 TinyOS 结构设计非常有意思」· Java 代码 · 共 138 行

/*
 * "Copyright (c) 2001 and The Regents of the University
 * of California.  All rights reserved.
 *
 * Permission to use, copy, modify, and distribute this software and its
 * documentation for any purpose, without fee, and without written agreement is
 * hereby granted, provided that the above copyright notice and the following
 * two paragraphs appear in all copies of this software.
 *
 * IN NO EVENT SHALL THE UNIVERSITY OF CALIFORNIA BE LIABLE TO ANY PARTY FOR
 * DIRECT, INDIRECT, SPECIAL, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING OUT
 * OF THE USE OF THIS SOFTWARE AND ITS DOCUMENTATION, EVEN IF THE UNIVERSITY OF
 * CALIFORNIA HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * THE UNIVERSITY OF CALIFORNIA SPECIFICALLY DISCLAIMS ANY WARRANTIES,
 * INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY
 * AND FITNESS FOR A PARTICULAR PURPOSE.  THE SOFTWARE PROVIDED HEREUNDER IS
 * ON AN "AS IS" BASIS, AND THE UNIVERSITY OF CALIFORNIA HAS NO OBLIGATION TO
 * PROVIDE MAINTENANCE, SUPPORT, UPDATES, ENHANCEMENTS, OR MODIFICATIONS."
 *
} * $\Id$
 */

package net.tinyos.sf;

import java.util.*;
import java.io.*;
import javax.comm.*;
import java.net.*;

public class NetworkSource implements DataSource {
    private InputStream      m_is = null;
    private OutputStream     m_os = null;
    private Socket           m_socket = null;
    private String           CLASS_NAME = "SerialSource";
    private boolean          m_bInitialized = false;
    private boolean          m_bShutdown = false;
    private SerialPort       serialPort      = null;

    public NetworkSource() {}

    public boolean OpenSource() {
        m_bShutdown                  = false;
        m_bInitialized               = false;

        try {
            OpenSocket();
            SerialForward.VERBOSE("Successfully opened " + SerialForward.commHost);
            m_is = m_socket.getInputStream();
            m_os = m_socket.getOutputStream();
            m_bInitialized = true;
        }
        catch (Exception e) {
            SerialForward.VERBOSE ("Unable to open socket to host [" +
                    SerialForward.commHost + ", port " +
                    SerialForward.commTCPPort + "] as serial port");
            m_is = null;
            m_os = null;
            return false;
        }

        return true;
    }

    public byte[] ReadPacket() {
        int     serialByte;
        int     nPacketSize = SerialForward.PACKET_SIZE;
        int     count = 0;
        byte[]  packet = new byte[SerialForward.PACKET_SIZE];

        if (m_is == null) {
            // serial port must not have opened correctly
            m_bShutdown = true;
        }

        try {
            SerialForward.VERBOSE("SerialPortIO: Reading port");
            while (!m_bShutdown && (serialByte = m_is.read()) != -1) {
            //while ((serialByte = m_is.read()) != -1) {
                packet[count] = (byte) serialByte;
                count++;
                SerialForward.nBytesRead++;

                if (count == nPacketSize) {
                  return packet;
                }
                else if(count == 1 && serialByte != 0x7e) {
                    count = 0;
                }
            }
        }
        catch(IOException e) {
            m_bShutdown = true;
        }

        return null;
    }

    public boolean CloseSource() {
        if (m_os != null) {
            try {m_os.close();}
            catch (IOException e) {}
        }
        if (m_is != null) {
            try {m_is.close();}
            catch (IOException e) {}
        }

        m_bInitialized = false;
        m_bShutdown    = true;
        m_is           = null;
        m_os           = null;

	return true;
    }

    public boolean WritePacket (byte[] packet) {
        try {
            if (m_os != null) {
                m_os.write(packet);
                return true;
            }
        }
        catch (IOException e) {
            SerialForward.VERBOSE("Unable to write data to mote");
        }

	return false;
    }

    private void OpenSocket() throws UnknownHostException, IOException, NumberFormatException {
	// use a socket as the serial port
	m_socket = new Socket(SerialForward.commHost,
			      Integer.parseInt(SerialForward.commTCPPort));
    }

}