org.opendaylight.usc.agent.UscAgentUdp.java Source code

Java tutorial

Introduction

Here is the source code for org.opendaylight.usc.agent.UscAgentUdp.java

Source

/*
 * Copyright (c) 2015 Huawei, Inc and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */

package org.opendaylight.usc.agent;

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.Channel;
import io.netty.channel.ChannelFuture;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.DatagramPacket;
import io.netty.channel.socket.nio.NioDatagramChannel;
import io.netty.handler.logging.LogLevel;
import io.netty.handler.logging.LoggingHandler;

import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.net.UnknownHostException;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import org.opendaylight.usc.manager.UscConfigurationServiceImpl;
import org.opendaylight.usc.manager.api.UscSecureService;
import org.opendaylight.usc.plugin.UscFrameDecoderUdp;
import org.opendaylight.usc.plugin.UscFrameEncoderUdp;
import org.opendaylight.usc.plugin.UscPlugin;
import org.opendaylight.usc.plugin.model.UscSessionImpl;
import org.opendaylight.usc.protocol.UscControl;
import org.opendaylight.usc.util.UscServiceUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.google.common.util.concurrent.SettableFuture;

public class UscAgentUdp implements Runnable, AutoCloseable {
    private static final Logger LOG = LoggerFactory.getLogger(UscAgentUdp.class);
    static final int PORT = Integer.parseInt(System.getProperty("port", "1068"));
    final EventLoopGroup bossGroup = new NioEventLoopGroup(1);
    final Bootstrap b = new Bootstrap();
    final EventLoopGroup callHomeGroup = new NioEventLoopGroup();
    final Bootstrap cb = new Bootstrap();

    private Channel agentServerChannel = null;
    private ConcurrentMap<Integer, SettableFuture<Boolean>> closeFuture = new ConcurrentHashMap<>();
    private UscSecureService secureService = null;

    public UscAgentUdp(boolean callHome) {
        this(callHome, InetAddress.getLoopbackAddress());
    }

    public UscAgentUdp(boolean callHome, InetAddress host) {
        this(callHome, host, "src/test/resources/etc/usc/usc.properties");
    }

    public UscAgentUdp(boolean callHome, InetAddress host, String propertyFile) {
        final UscAgentUdp agent = this;
        UscConfigurationServiceImpl.setDefaultPropertyFilePath(propertyFile);
        secureService = UscServiceUtils.getService(UscSecureService.class);
        b.group(bossGroup);
        b.channel(NioDatagramChannel.class);
        b.handler(new ChannelInitializer<NioDatagramChannel>() {
            @Override
            public void initChannel(NioDatagramChannel ch) throws Exception {
                ChannelPipeline p = ch.pipeline();
                agentServerChannel = ch;
                p.addLast(new ChannelInboundHandlerAdapter() {

                    @Override
                    public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
                        final Channel ch = ctx.channel();
                        final InetSocketAddress remoteAddress = ((DatagramPacket) msg).sender();

                        // this is to deal with UDP channels which don't by
                        // default have remote address
                        if (ch.remoteAddress() == null) {
                            ch.connect(remoteAddress);
                        }
                        ch.pipeline().remove(this);
                        super.channelRead(ctx, msg);
                    }

                });
                if (secureService == null) {
                    LOG.error("UscSecureService is not initialized!");
                    return;
                }
                p.addLast(new LoggingHandler("UscAgnet Handler 5", LogLevel.TRACE));
                p.addLast(secureService.getUdpServerHandler(ch));
                p.addLast(new LoggingHandler("UscAgnet Handler 4", LogLevel.TRACE));
                p.addLast(new UscFrameEncoderUdp());
                p.addLast(new LoggingHandler("UscAgnet Handler 3", LogLevel.TRACE));
                p.addLast(new UscFrameDecoderUdp());
                p.addLast(new LoggingHandler("UscAgnet Handler 2", LogLevel.TRACE));
                p.addLast(new UscAgentUdpHandler(agent, ch));
                p.addLast(new LoggingHandler("UscAgnet Handler 1", LogLevel.TRACE));

            }
        });

        if (callHome) {
            cb.group(callHomeGroup);
            cb.channel(NioDatagramChannel.class);
            cb.handler(new ChannelInitializer<NioDatagramChannel>() {
                @Override
                public void initChannel(NioDatagramChannel ch) throws Exception {
                    ChannelPipeline p = ch.pipeline();
                    agentServerChannel = ch;
                    p.addLast(new ChannelInboundHandlerAdapter() {

                        @Override
                        public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
                            final Channel ch = ctx.channel();
                            final InetSocketAddress remoteAddress = ((DatagramPacket) msg).sender();

                            // this is to deal with UDP channels which don't by
                            // default have remote address
                            if (ch.remoteAddress() == null) {
                                ch.connect(remoteAddress);
                            }
                            ch.pipeline().remove(this);
                            super.channelRead(ctx, msg);
                        }

                    });
                    if (secureService == null) {
                        LOG.error("UscSecureService is not initialized!");
                        return;
                    }
                    p.addLast(new LoggingHandler("LOG2-5", LogLevel.TRACE));
                    p.addLast(secureService.getUdpClientHandler(ch));
                    p.addLast(new LoggingHandler("LOG2-4", LogLevel.TRACE));
                    p.addLast(new UscFrameEncoderUdp());
                    p.addLast(new LoggingHandler("LOG2-3", LogLevel.TRACE));
                    p.addLast(new UscFrameDecoderUdp());
                    p.addLast(new LoggingHandler("LOG2-2", LogLevel.TRACE));
                    p.addLast(new UscAgentUdpHandler(agent, ch));
                    p.addLast(new LoggingHandler("LOG2-1", LogLevel.TRACE));

                }
            });

            try {
                InetSocketAddress recipient = new InetSocketAddress(host, 1069);
                cb.connect(recipient).sync().channel();
            } catch (Exception e) {
                // TODO Auto-generated catch block
                e.printStackTrace();
            }
        }
    }

    protected ConcurrentMap<Integer, SettableFuture<Boolean>> getCloseFuture() {
        return closeFuture;
    }

    public SettableFuture<Boolean> closeClientInternalConnection(Channel clientChannel) {
        try {
            UscSessionImpl session = clientChannel.attr(UscPlugin.SESSION).get().get();
            closeFuture.remove(session.getSessionId());
            closeFuture.putIfAbsent(session.getSessionId(), SettableFuture.<Boolean>create());

            UscControl data = new UscControl(session.getPort(), session.getSessionId(), 1);
            if (agentServerChannel != null)
                agentServerChannel.writeAndFlush(data);

            LOG.trace("UscAgentTcp closeClientInternalConnection port#: " + session.getPort() + " ,session#: "
                    + session.getSessionId());

            return closeFuture.get(session.getSessionId());

        } catch (Exception e) {
            e.printStackTrace();
            return null;
        }
    }

    @Override
    public void run() {
        // Start the server.
        try {
            ChannelFuture f = b.bind(PORT).sync();
            LOG.trace("UscAgentUdp initialized");
            // Wait until the server socket is closed.
            f.channel().closeFuture().sync();
        } catch (InterruptedException e) {
            // TODO Auto-generated catch block
            e.printStackTrace();
        }
    }

    @Override
    public void close() throws Exception {
        bossGroup.shutdownGracefully();

        callHomeGroup.shutdownGracefully();
    }

    public static void main(String[] args) throws Exception {
        boolean callHome = false;
        InetAddress host = InetAddress.getLoopbackAddress();
        if (args.length > 0) {
            try {
                InetAddress ip = InetAddress.getByName(args[0]);
                callHome = true;
                host = ip;
            } catch (UnknownHostException e) {
                System.err.println("Argument " + args[0] + " must be an iP address (callhome IP).");
                System.exit(1);
            }
        }

        try (UscAgentUdp agent = new UscAgentUdp(callHome)) {
            agent.run();
        }
    }

}