com.github.sdnwiselab.sdnwise.adapter.AdapterUdp Maven / Gradle / Ivy

Go to download
/* 
 * Copyright (C) 2015 SDN-WISE
 *
 * This program 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.
 *
 * This program 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 this program.  If not, see .
 */
package com.github.sdnwiselab.sdnwise.adapter;

import java.io.IOException;
import java.net.DatagramPacket;
import java.net.DatagramSocket;
import java.net.InetAddress;
import java.net.SocketException;
import java.util.Arrays;
import java.util.Map;
import java.util.Observable;
import java.util.logging.Level;

/**
 * The adapter class for UDP port communication. Configuration data are passed
 * using a Map which contains all the options needed in the
 * constructor of the class.
 *
 * @author Sebastiano Milardo
 * @version 0.1
 */
public class AdapterUdp extends Adapter{

    private final int IN_PORT;
    private final int OUT_PORT;
    private final String OUT_IP;
    final int MAX_PAYLOAD;

    public final boolean FULL_PACKET;

    private UDPServer udpServer;
    private Thread th;
    private DatagramSocket sck;

    /**
     * Creates an AdapterUDP object. The conf map is used to pass the
     * configuration settings for the serial port as strings. Specifically
     * needed parameters are:
     * 
     * OUT_IP
     * OUT_PORT
     * IN_PORT
     * MAX_PAYLOAD
     * 
     *
     * @param conf contains the serial port configuration data.
     */
    public AdapterUdp(Map conf) {
        this.OUT_IP = conf.get("OUT_IP");
        this.OUT_PORT = Integer.parseInt(conf.get("OUT_PORT"));
        this.IN_PORT = Integer.parseInt(conf.get("IN_PORT"));
        this.MAX_PAYLOAD = Integer.parseInt(conf.get("MAX_PAYLOAD"));
        this.FULL_PACKET = Boolean.parseBoolean(conf.get("FULL_PACKET"));
    }

    /**
     * Opens this adapter.
     *
     * @return a boolean indicating the correct completion of the operations
     */
    @Override
    public final boolean open() {
        try {
            sck = new DatagramSocket(IN_PORT);
            if (this.FULL_PACKET) {
                udpServer = new UDPDatagramServer(sck);
            } else {
                udpServer = new UDPServer(sck);
            }
            udpServer.addObserver(this);
            th = new Thread(udpServer);
            th.start();
            return true;
        } catch (SocketException ex) {
            log(Level.SEVERE, ex.toString());
            return false;
        }
    }

    /**
     * Closes this adapter.
     *
     * @return a boolean indicating the correct ending of the operations
     */
    @Override
    public final boolean close() {
        udpServer.isStopped = true;
        sck.close();
        return true;
    }

    /**
     * Sends a byte array using this adapter.
     *
     * @param data the array to be sent
     */
    @Override
    public final void send(byte[] data) {
        try {
            DatagramPacket packet = new DatagramPacket(data, data.length, InetAddress.getByName(OUT_IP), OUT_PORT);
            sck.send(packet);
        } catch (IOException ex) {
            log(Level.SEVERE, ex.toString());
        }
    }

    /**
     * Sends a byte array using this adapter. This method also specifies the
     * destination IP address and UDP port.
     *
     * @param data the array to be sent
     * @param OUT_IP a string containing the IP address of the destination
     * @param OUT_PORT an integer containing the UDP port of the destination
     */
    public final void send(byte[] data, String OUT_IP, int OUT_PORT) {
        try {
            DatagramPacket packet = new DatagramPacket(data, data.length, InetAddress.getByName(OUT_IP), OUT_PORT);
            sck.send(packet);
        } catch (IOException ex) {
            log(Level.SEVERE, ex.toString());
        }
    }

    private class UDPServer extends Observable implements Runnable {

        boolean isStopped;
        DatagramSocket sck;

        UDPServer(DatagramSocket sck) {
            this.sck = sck;
            isStopped = false;
        }

        @Override
        public void run() {
            try {
                byte[] buffer = new byte[MAX_PAYLOAD];
                DatagramPacket packet = new DatagramPacket(buffer, buffer.length);
                while (!isStopped) {
                    sck.receive(packet);
                    setChanged();
                    notifyObservers(Arrays.copyOf(packet.getData(), packet.getLength()));
                }
            } catch (IOException ex) {
                log(Level.SEVERE, ex.toString());
            }
        }
    }

    private class UDPDatagramServer extends UDPServer {

        UDPDatagramServer(DatagramSocket sck) {
            super(sck);
        }

        @Override
        public void run() {
            try {
                byte[] buffer = new byte[MAX_PAYLOAD];
                DatagramPacket packet = new DatagramPacket(buffer, buffer.length);
                while (!isStopped) {
                    sck.receive(packet);
                    setChanged();
                    notifyObservers(packet);
                }
            } catch (IOException ex) {
                log(Level.SEVERE, ex.toString());
            }
        }
    }
}