1 /*
   2  * Copyright (c) 2015, 2016, 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.  Oracle designates this
   8  * particular file as subject to the "Classpath" exception as provided
   9  * by Oracle in the LICENSE file that accompanied this code.
  10  *
  11  * This code is distributed in the hope that it will be useful, but WITHOUT
  12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14  * version 2 for more details (a copy is included in the LICENSE file that
  15  * accompanied this code).
  16  *
  17  * You should have received a copy of the GNU General Public License version
  18  * 2 along with this work; if not, write to the Free Software Foundation,
  19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  20  *
  21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  22  * or visit www.oracle.com if you need additional information or have any
  23  * questions.
  24  */
  25 
  26 package java.net.http;
  27 
  28 import java.io.IOException;
  29 import java.net.InetSocketAddress;
  30 import java.net.URI;
  31 import static java.net.http.SettingsFrame.INITIAL_WINDOW_SIZE;
  32 import static java.net.http.SettingsFrame.ENABLE_PUSH;
  33 import static java.net.http.SettingsFrame.HEADER_TABLE_SIZE;
  34 import static java.net.http.SettingsFrame.MAX_CONCURRENT_STREAMS;
  35 import static java.net.http.SettingsFrame.MAX_FRAME_SIZE;
  36 import java.util.Base64;
  37 import java.util.Collections;
  38 import java.util.HashMap;
  39 import java.util.HashSet;
  40 import java.util.Map;
  41 import java.util.Set;
  42 
  43 /**
  44  *  Http2 specific aspects of HttpClientImpl
  45  */
  46 class Http2ClientImpl {
  47 
  48     final private HttpClientImpl client;
  49 
  50     Http2ClientImpl(HttpClientImpl client) {
  51         this.client = client;
  52     }
  53 
  54     /* Map key is "scheme:host:port" */
  55     final private Map<String,Http2Connection> connections =
  56             Collections.synchronizedMap(new HashMap<>());
  57 
  58     final private Set<String> opening = Collections.synchronizedSet(new HashSet<>());
  59 
  60     synchronized boolean haveConnectionFor(URI uri, InetSocketAddress proxy) {
  61         return connections.containsKey(Http2Connection.keyFor(uri,proxy));
  62     }
  63 
  64     /**
  65      * If a https request then blocks and waits until a connection is opened.
  66      * Returns null if the request is 'http' as a different (upgrade)
  67      * mechanism is used.
  68      *
  69      * Only one connection per destination is created. Blocks when opening
  70      * connection, or when waiting for connection to be opened.
  71      * First thread opens the connection and notifies the others when done.
  72      *
  73      * If the request is secure (https) then we open the connection here.
  74      * If not, then the more complicated upgrade from 1.1 to 2 happens (not here)
  75      * In latter case, when the Http2Connection is connected, putConnection() must
  76      * be called to store it.
  77      */
  78     Http2Connection getConnectionFor(HttpRequestImpl req)
  79             throws IOException, InterruptedException {
  80         URI uri = req.uri();
  81         InetSocketAddress proxy = req.proxy();
  82         String key = Http2Connection.keyFor(uri, proxy);
  83         Http2Connection connection;
  84         synchronized (opening) {
  85             while ((connection = connections.get(key)) == null) {
  86                 if (!req.secure()) {
  87                     return null;
  88                 }
  89                 if (!opening.contains(key)) {
  90                     opening.add(key);
  91                     break;
  92                 } else {
  93                     opening.wait();
  94                 }
  95             }
  96         }
  97         if (connection != null) {
  98             return connection;
  99         }
 100         // we are opening the connection here blocking until it is done.
 101         connection = new Http2Connection(req);
 102         synchronized (opening) {
 103             connections.put(key, connection);
 104             opening.remove(key);
 105             opening.notifyAll();
 106         }
 107         return connection;
 108     }
 109 
 110 
 111     /*
 112      * TODO: If there isn't a connection to the same destination, then
 113      * store it. If there is already a connection, then close it
 114      */
 115     synchronized void putConnection(Http2Connection c) {
 116         String key = c.key();
 117         connections.put(key, c);
 118     }
 119 
 120     synchronized void deleteConnection(Http2Connection c) {
 121         String key = c.key();
 122         connections.remove(key);
 123     }
 124 
 125     HttpClientImpl client() {
 126         return client;
 127     }
 128 
 129     /** Returns the client settings as a base64 (url) encoded string */
 130     String getSettingsString() {
 131         SettingsFrame sf = getClientSettings();
 132         ByteBufferGenerator bg = new ByteBufferGenerator(client);
 133         sf.writeOutgoing(bg);
 134         byte[] settings = bg.asByteArray(9); // without the header
 135         Base64.Encoder encoder = Base64.getUrlEncoder()
 136                                        .withoutPadding();
 137         return encoder.encodeToString(settings);
 138     }
 139 
 140     private static final int K = 1024;
 141 
 142     SettingsFrame getClientSettings() {
 143         SettingsFrame frame = new SettingsFrame();
 144         frame.setParameter(HEADER_TABLE_SIZE, Utils.getIntegerNetProperty(
 145             "java.net.httpclient.hpack.maxheadertablesize", 16 * K));
 146         frame.setParameter(ENABLE_PUSH, Utils.getIntegerNetProperty(
 147             "java.net.httpclient.enablepush", 1));
 148         frame.setParameter(MAX_CONCURRENT_STREAMS, Utils.getIntegerNetProperty(
 149             "java.net.httpclient.maxstreams", 16));
 150         frame.setParameter(INITIAL_WINDOW_SIZE, Utils.getIntegerNetProperty(
 151             "java.net.httpclient.windowsize", 32 * K));
 152         frame.setParameter(MAX_FRAME_SIZE, Utils.getIntegerNetProperty(
 153             "java.net.httpclient.maxframesize", 16 * K));
 154         frame.computeLength();
 155         return frame;
 156     }
 157 }