/*

  *   SONEWS News Server

  *   see AUTHORS for the list of contributors

  *

  *   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 <http://www.gnu.org/licenses/>.

  */

 package org.sonews.daemon;

 import org.sonews.config.Config;

 import org.sonews.Main;

 import org.sonews.util.Log;

 import java.io.IOException;

 import java.net.BindException;

 import java.net.InetSocketAddress;

 import java.net.ServerSocket;

 import java.nio.channels.CancelledKeyException;

 import java.nio.channels.ClosedChannelException;

 import java.nio.channels.SelectionKey;

 import java.nio.channels.Selector;

 import java.nio.channels.ServerSocketChannel;

 import java.nio.channels.SocketChannel;

 /**

  * NNTP daemon using SelectableChannels.

  * @author Christian Lins

  * @since sonews/0.5.0

  */

 public final class NNTPDaemon extends AbstractDaemon

 {

   public static final Object RegisterGate = new Object();

   private static NNTPDaemon instance = null;

   public static synchronized NNTPDaemon createInstance(int port)

   {

     if(instance == null)

     {

       instance = new NNTPDaemon(port);

       return instance;

     }

     else

     {

       throw new RuntimeException("NNTPDaemon.createInstance() called twice");

     }

   }

   private int port;

   private NNTPDaemon(final int port)

   {

     Log.msg("Server listening on port " + port, false);

     this.port = port;

   }

   @Override

   public void run()

   {

     try

     {

       // Create a Selector that handles the SocketChannel multiplexing

       final Selector readSelector  = Selector.open();

       final Selector writeSelector = Selector.open();

       // Start working threads

       final int workerThreads = Runtime.getRuntime().availableProcessors() * 4;

       ConnectionWorker[] cworkers = new ConnectionWorker[workerThreads];

       for(int n = 0; n < workerThreads; n++)

       {

         cworkers[n] = new ConnectionWorker();

         cworkers[n].start();

       }

       ChannelWriter.getInstance().setSelector(writeSelector);

       ChannelReader.getInstance().setSelector(readSelector);

       ChannelWriter.getInstance().start();

       ChannelReader.getInstance().start();

       final ServerSocketChannel serverSocketChannel = ServerSocketChannel.open();

       serverSocketChannel.configureBlocking(true);  // Set to blocking mode

       // Configure ServerSocket; bind to socket...

       final ServerSocket serverSocket = serverSocketChannel.socket();

       serverSocket.bind(new InetSocketAddress(this.port));

       while(isRunning())

       {

         SocketChannel socketChannel;

         try

         {

           // As we set the server socket channel to blocking mode the accept()

           // method will block.

           socketChannel = serverSocketChannel.accept();

           socketChannel.configureBlocking(false);

           assert socketChannel.isConnected();

           assert socketChannel.finishConnect();

         }

         catch(IOException ex)

         {

           // Under heavy load an IOException "Too many open files may

           // be thrown. It most cases we should slow down the connection

           // accepting, to give the worker threads some time to process work.

           Log.msg("IOException while accepting connection: " + ex.getMessage(), false);

           Log.msg("Connection accepting sleeping for seconds...", true);

           Thread.sleep(5000); // 5 seconds

           continue;

         }

         final NNTPConnection conn;

         try

         {

           conn = new NNTPConnection(socketChannel);

           Connections.getInstance().add(conn);

         }

         catch(IOException ex)

         {

           Log.msg(ex.getLocalizedMessage(), false);

           socketChannel.close();

           continue;

         }

         try

         {

           SelectionKey selKeyWrite =

             registerSelector(writeSelector, socketChannel, SelectionKey.OP_WRITE);

           registerSelector(readSelector, socketChannel, SelectionKey.OP_READ);

           Log.msg("Connected: " + socketChannel.socket().getRemoteSocketAddress(), true);

           // Set write selection key and send hello to client

           conn.setWriteSelectionKey(selKeyWrite);

           conn.println("200 " + Config.inst().get(Config.HOSTNAME, "localhost")

               + " " + Main.VERSION + " news server ready - (posting ok).");

         }

         catch(CancelledKeyException cke)

         {

           Log.msg("CancelledKeyException " + cke.getMessage() + " was thrown: " 

             + socketChannel.socket(), false);

         }

         catch(ClosedChannelException cce)

         {

           Log.msg("ClosedChannelException " + cce.getMessage() + " was thrown: " 

             + socketChannel.socket(), false);

         }

       }

     }

     catch(BindException ex)

     {

       // Could not bind to socket; this is a fatal problem; so perform shutdown

       ex.printStackTrace();

       System.exit(1);

     }

     catch(IOException ex)

     {

       ex.printStackTrace();

     }

     catch(Exception ex)

     {

       ex.printStackTrace();

     }

   }

   public static SelectionKey registerSelector(final Selector selector,

     final SocketChannel channel, final int op)

     throws CancelledKeyException, ClosedChannelException

   {

     // Register the selector at the channel, so that it will be notified

     // on the socket's events

     synchronized(RegisterGate)

     {

       // Wakeup the currently blocking reader/writer thread; we have locked

       // the RegisterGate to prevent the awakened thread to block again

       selector.wakeup();

       // Lock the selector to prevent the waiting worker threads going into

       // selector.select() which would block the selector.

       synchronized (selector)

       {

         return channel.register(selector, op, null);

       }

     }

   }

 }