1 /*
   2  * Copyright (c) 2007, 2011, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /* @test
  25  * @bug 4640544 8044773
  26  * @summary Unit test to check SocketChannel setOption/getOption/options
  27  *          methods.
  28  * @run main SocketOptionTests
  29  * @run main/othervm --limit-modules=java.base SocketOptionTests
  30  */
  31 
  32 import java.nio.*;
  33 import java.nio.channels.*;
  34 import java.net.*;
  35 import java.io.IOException;
  36 import java.util.*;
  37 import static java.net.StandardSocketOptions.*;
  38 
  39 public class SocketOptionTests {
  40 
  41     static void checkOption(SocketChannel sc, SocketOption name, Object expectedValue)
  42         throws IOException
  43     {
  44         Object value = sc.getOption(name);
  45         if (!value.equals(expectedValue))
  46             throw new RuntimeException("value not as expected");
  47     }
  48 
  49     public static void main(String[] args) throws IOException {
  50         SocketChannel sc = SocketChannel.open();
  51 
  52         // check supported options
  53         Set<SocketOption<?>> options = sc.supportedOptions();
  54         List<? extends SocketOption> expected = Arrays.asList(SO_SNDBUF, SO_RCVBUF,
  55             SO_KEEPALIVE, SO_REUSEADDR, SO_LINGER, TCP_NODELAY);
  56         for (SocketOption opt: expected) {
  57             if (!options.contains(opt))
  58                 throw new RuntimeException(opt.name() + " should be supported");
  59         }
  60 
  61         // check specified defaults
  62         int linger = sc.<Integer>getOption(SO_LINGER);
  63         if (linger >= 0)
  64             throw new RuntimeException("initial value of SO_LINGER should be < 0");
  65         checkOption(sc, SO_KEEPALIVE, false);
  66         checkOption(sc, TCP_NODELAY, false);
  67 
  68         // allowed to change when not bound
  69         sc.setOption(SO_KEEPALIVE, true);
  70         checkOption(sc, SO_KEEPALIVE, true);
  71         sc.setOption(SO_KEEPALIVE, false);
  72         checkOption(sc, SO_KEEPALIVE, false);
  73         sc.setOption(SO_SNDBUF, 128*1024);      // can't check
  74         sc.setOption(SO_RCVBUF, 256*1024);      // can't check
  75         int before, after;
  76         before = sc.getOption(SO_SNDBUF);
  77         after = sc.setOption(SO_SNDBUF, Integer.MAX_VALUE).getOption(SO_SNDBUF);
  78         if (after < before)
  79             throw new RuntimeException("setOption caused SO_SNDBUF to decrease");
  80         before = sc.getOption(SO_RCVBUF);
  81         after = sc.setOption(SO_RCVBUF, Integer.MAX_VALUE).getOption(SO_RCVBUF);
  82         if (after < before)
  83             throw new RuntimeException("setOption caused SO_RCVBUF to decrease");
  84         sc.setOption(SO_REUSEADDR, true);
  85         checkOption(sc, SO_REUSEADDR, true);
  86         sc.setOption(SO_REUSEADDR, false);
  87         checkOption(sc, SO_REUSEADDR, false);
  88         sc.setOption(SO_LINGER, 10);
  89         linger = sc.<Integer>getOption(SO_LINGER);
  90         if (linger < 1)
  91             throw new RuntimeException("expected linger to be enabled");
  92         sc.setOption(SO_LINGER, -1);
  93         linger = sc.<Integer>getOption(SO_LINGER);
  94         if (linger >= 0)
  95             throw new RuntimeException("expected linger to be disabled");
  96         sc.setOption(TCP_NODELAY, true);
  97         checkOption(sc, TCP_NODELAY, true);
  98         sc.setOption(TCP_NODELAY, false);       // can't check
  99 
 100         // bind socket
 101         sc.bind(new InetSocketAddress(0));
 102 
 103         // allow to change when bound
 104         sc.setOption(SO_KEEPALIVE, true);
 105         checkOption(sc, SO_KEEPALIVE, true);
 106         sc.setOption(SO_KEEPALIVE, false);
 107         checkOption(sc, SO_KEEPALIVE, false);
 108 
 109         sc.setOption(SO_LINGER, 10);
 110         linger = sc.<Integer>getOption(SO_LINGER);
 111         if (linger < 1)
 112             throw new RuntimeException("expected linger to be enabled");
 113         sc.setOption(SO_LINGER, -1);
 114         linger = sc.<Integer>getOption(SO_LINGER);
 115         if (linger >= 0)
 116             throw new RuntimeException("expected linger to be disabled");
 117         sc.setOption(TCP_NODELAY, true);        // can't check
 118         sc.setOption(TCP_NODELAY, false);       // can't check
 119 
 120         // NullPointerException
 121         try {
 122             sc.setOption(null, "value");
 123             throw new RuntimeException("NullPointerException not thrown");
 124         } catch (NullPointerException x) {
 125         }
 126         try {
 127             sc.getOption(null);
 128             throw new RuntimeException("NullPointerException not thrown");
 129         } catch (NullPointerException x) {
 130         }
 131 
 132         // ClosedChannelException
 133         sc.close();
 134         try {
 135             sc.setOption(TCP_NODELAY, true);
 136             throw new RuntimeException("ClosedChannelException not thrown");
 137         } catch (ClosedChannelException x) {
 138         }
 139     }
 140 }