/*
    * Copyright 2012 The Netty Project
    *
    * The Netty Project licenses this file to you under the Apache License,
    * version 2.0 (the "License"); you may not use this file except in compliance
    * with the License. You may obtain a copy of the License at:
    *
    *   http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
   * WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
   * License for the specific language governing permissions and limitations
   * under the License.
   */
  package io.netty.channel.socket.oio;
  
  import io.netty.buffer.ByteBuf;
  import io.netty.channel.Channel;
  import io.netty.channel.ChannelException;
  import io.netty.channel.ChannelFuture;
  import io.netty.channel.ChannelPromise;
  import io.netty.channel.ConnectTimeoutException;
  import io.netty.channel.EventLoop;
  import io.netty.channel.oio.OioByteStreamChannel;
  import io.netty.channel.socket.ServerSocketChannel;
  import io.netty.channel.socket.SocketChannel;
  import io.netty.util.internal.SocketUtils;
  import io.netty.util.internal.UnstableApi;
  import io.netty.util.internal.logging.InternalLogger;
  import io.netty.util.internal.logging.InternalLoggerFactory;
  
  import java.io.IOException;
  import java.net.InetSocketAddress;
  import java.net.Socket;
  import java.net.SocketAddress;
  import java.net.SocketTimeoutException;
  
  /**
   * A {@link SocketChannel} which is using Old-Blocking-IO
   */
  public class OioSocketChannel extends OioByteStreamChannel
                                implements SocketChannel {
  
      private static final InternalLogger logger =
              InternalLoggerFactory.getInstance(OioSocketChannel.class);
  
      private final Socket socket;
      private final OioSocketChannelConfig config;
  
      /**
       * Create a new instance with an new {@link Socket}
       */
      public OioSocketChannel() {
          this(new Socket());
      }
  
      /**
       * Create a new instance from the given {@link Socket}
       *
       * @param socket    the {@link Socket} which is used by this instance
       */
      public OioSocketChannel(Socket socket) {
          this(null, socket);
      }
  
      /**
       * Create a new instance from the given {@link Socket}
       *
       * @param parent    the parent {@link Channel} which was used to create this instance. This can be null if the
       *                  {@link} has no parent as it was created by your self.
       * @param socket    the {@link Socket} which is used by this instance
       */
      public OioSocketChannel(Channel parent, Socket socket) {
          super(parent);
          this.socket = socket;
          config = new DefaultOioSocketChannelConfig(this, socket);
  
          boolean success = false;
          try {
              if (socket.isConnected()) {
                  activate(socket.getInputStream(), socket.getOutputStream());
              }
              socket.setSoTimeout(SO_TIMEOUT);
              success = true;
          } catch (Exception e) {
              throw new ChannelException("failed to initialize a socket", e);
          } finally {
              if (!success) {
                  try {
                      socket.close();
                  } catch (IOException e) {
                      logger.warn("Failed to close a socket.", e);
                  }
              }
          }
      }
  
      @Override
     public ServerSocketChannel parent() {
         return (ServerSocketChannel) super.parent();
     }
 
     @Override
     public OioSocketChannelConfig config() {
         return config;
     }
 
     @Override
     public boolean isOpen() {
         return !socket.isClosed();
     }
 
     @Override
     public boolean isActive() {
         return !socket.isClosed() && socket.isConnected();
     }
 
     @Override
     public boolean isInputShutdown() {
         return super.isInputShutdown();
     }
 
     @Override
     public boolean isOutputShutdown() {
         return socket.isOutputShutdown() || !isActive();
     }
 
     @UnstableApi
     @Override
     protected final void doShutdownOutput() throws Exception {
         shutdownOutput0();
     }
 
     @Override
     public ChannelFuture shutdownOutput() {
         return shutdownOutput(newPromise());
     }
 
     @Override
     protected int doReadBytes(ByteBuf buf) throws Exception {
         if (socket.isClosed()) {
             return -1;
         }
         try {
             return super.doReadBytes(buf);
         } catch (SocketTimeoutException ignored) {
             return 0;
         }
     }
 
     @Override
     public ChannelFuture shutdownOutput(final ChannelPromise promise) {
         EventLoop loop = eventLoop();
         if (loop.inEventLoop()) {
             shutdownOutput0(promise);
         } else {
             loop.execute(new Runnable() {
                 @Override
                 public void run() {
                     shutdownOutput0(promise);
                 }
             });
         }
         return promise;
     }
 
     private void shutdownOutput0(ChannelPromise promise) {
         try {
             shutdownOutput0();
             promise.setSuccess();
         } catch (Throwable t) {
             promise.setFailure(t);
         }
     }
 
     private void shutdownOutput0() throws IOException {
         socket.shutdownOutput();
     }
 
     @Override
     public InetSocketAddress localAddress() {
         return (InetSocketAddress) super.localAddress();
     }
 
     @Override
     public InetSocketAddress remoteAddress() {
         return (InetSocketAddress) super.remoteAddress();
     }
 
     @Override
     protected SocketAddress localAddress0() {
         return socket.getLocalSocketAddress();
     }
 
     @Override
     protected SocketAddress remoteAddress0() {
         return socket.getRemoteSocketAddress();
     }
 
     @Override
     protected void doBind(SocketAddress localAddress) throws Exception {
         SocketUtils.bind(socket, localAddress);
     }
 
     @Override
     protected void doConnect(SocketAddress remoteAddress,
             SocketAddress localAddress) throws Exception {
         if (localAddress != null) {
             SocketUtils.bind(socket, localAddress);
         }
 
         boolean success = false;
         try {
             SocketUtils.connect(socket, remoteAddress, config().getConnectTimeoutMillis());
             activate(socket.getInputStream(), socket.getOutputStream());
             success = true;
         } catch (SocketTimeoutException e) {
             ConnectTimeoutException cause = new ConnectTimeoutException("connection timed out: " + remoteAddress);
             cause.setStackTrace(e.getStackTrace());
             throw cause;
         } finally {
             if (!success) {
                 doClose();
             }
         }
     }
 
     @Override
     protected void doDisconnect() throws Exception {
         doClose();
     }
 
     @Override
     protected void doClose() throws Exception {
         socket.close();
     }
 
     @Override
     protected boolean checkInputShutdown() {
         if (isInputShutdown()) {
             try {
                 Thread.sleep(config().getSoTimeout());
             } catch (Throwable e) {
                 // ignore
             }
             return true;
         }
         return false;
     }
 
     @Override
     protected void setReadPending(boolean readPending) {
         super.setReadPending(readPending);
     }
 }