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.nio.ByteBuffer;

    30 

    31 class PushPromiseFrame extends HeaderFrame {

    32 

    33     int padLength;

    34     int promisedStream;

    35 

    36     PushPromiseFrame() {

    37         type = TYPE;

    38     }

    39 

    40     public static final int TYPE = 0x5;

    41 

    42     // Flags

    43     public static final int END_HEADERS = 0x4;

    44     public static final int PADDED = 0x8;

    45 

    46     @Override

    47     public String toString() {

    48         return super.toString() + " promisedStreamid: " + promisedStream

    49                 + " headerLength: " + headerLength;

    50     }

    51 

    52     @Override

    53     String flagAsString(int flag) {

    54         switch (flag) {

    55         case PADDED:

    56             return "PADDED";

    57         case END_HEADERS:

    58             return "END_HEADERS";

    59         }

    60         return super.flagAsString(flag);

    61     }

    62 

    63     public void setPadLength(int padLength) {

    64         this.padLength = padLength;

    65         flags |= PADDED;

    66     }

    67 

    68     public void setPromisedStream(int stream) {

    69         this.promisedStream = stream;

    70     }

    71 

    72     public int getPromisedStream() {

    73         return promisedStream;

    74     }

    75 

    76     /**

    77      */

    78     @Override

    79     void readIncomingImpl(ByteBufferConsumer bc) throws IOException {

    80         if ((flags & PADDED) != 0) {

    81             padLength = bc.getByte();

    82             headerLength = length - (padLength + 5);

    83         } else

    84             headerLength = length - 4;

    85 

    86         promisedStream = bc.getInt() & 0x7fffffff;

    87         headerBlocks = bc.getBuffers(headerLength);

    88     }

    89 

    90     @Override

    91     void computeLength() {

    92         int len = 0;

    93         if ((flags & PADDED) != 0) {

    94             len += (1 + padLength);

    95         }

    96         len += (4 + headerLength);

    97         this.length = len;

    98     }

    99 

   100     @Override

   101     void writeOutgoing(ByteBufferGenerator bg) {

   102         super.writeOutgoing(bg);

   103         ByteBuffer buf = bg.getBuffer(length);

   104         if ((flags & PADDED) != 0) {

   105             buf.put((byte)padLength);

   106         }

   107         buf.putInt(promisedStream);

   108         for (int i=0; i<headerBlocks.length; i++) {

   109             bg.addByteBuffer(headerBlocks[i]);

   110         }

   111         if ((flags & PADDED) != 0) {

   112             bg.addPadding(padLength);

   113         }

   114     }

   115 

   116     @Override

   117     public boolean endHeaders() {

   118         return getFlag(END_HEADERS);

   119     }

   120 }