com.webpieces.http2parser.impl.PaddingImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of http2-parser Show documentation

A re-usable asynchronous http2 parser

There is a newer version: 2.1.1

package com.webpieces.http2parser.impl;

import java.util.List;

import org.webpieces.data.api.DataWrapper;
import org.webpieces.data.api.DataWrapperGenerator;

import com.webpieces.http2parser.api.Padding;
import com.webpieces.http2parser.api.ParseException;
import com.webpieces.http2parser.api.dto.Http2ErrorCode;

public class PaddingImpl implements Padding {
    private byte[] padding = null;
    private boolean padded = false;

    final private DataWrapperGenerator dataGen;

    public PaddingImpl(DataWrapperGenerator dataGen) {
        this.dataGen = dataGen;
    }

    @Override
    public void setPadding(byte[] padding) {
        this.padding = padding;
        this.padded = true;
    }

    @Override
    public boolean isPadded() {
        return padded;
    }

    @Override
    public void setIsPadded(boolean isPadded) {
        this.padded = isPadded;
    }

    @Override
    public byte[] getPadding() {
        return padding;
    }

    @Override
    public DataWrapper extractPayloadAndSetPaddingIfNeeded(DataWrapper data, int streamId) {
        if(isPadded()) {
            short padLength = (short) (data.readByteAt(0) & 0xFF);
            if(padLength > data.getReadableSize()) {
                throw new ParseException(Http2ErrorCode.PROTOCOL_ERROR, streamId, true);
            }
            List split1 = dataGen.split(data, 1);
            List split2 = dataGen.split(split1.get(1), split1.get(1).getReadableSize() - padLength);
            setPadding(split2.get(1).createByteArray());
            return split2.get(0);
        }
        else
            return data;
    }

    @Override
    public DataWrapper padDataIfNeeded(DataWrapper data) {
        if(isPadded()) {
            byte[] length = {(byte) padding.length};
            DataWrapper lengthDW = dataGen.wrapByteArray(length);
            DataWrapper paddingDW = dataGen.wrapByteArray(padding);
            return dataGen.chainDataWrappers(dataGen.chainDataWrappers(lengthDW, data), paddingDW);
        } else
            return data;
    }
}