17882005206

Committed 20 Sep 2025 01:06PM UTC coverage: 70.892% (+1.5%) from 69.377%

Build # 17882005206

Build Type

push

github

Committed by

web-flow

Commit Message

refactor: code clean (#50)

* refactor: code clean

* refactor: code clean,remove hessian1 code

* refactor: fix HessianDebugState

Run Details

1780 of 2691 branches covered (66.15%)

Branch coverage included in aggregate %.

4226 of 5781 relevant lines covered (73.1%)

3.17 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

82.77

hessian2-codec/src/main/java/io/github/wuwen5/hessian/io/HessianEncoder.java

/*
 * Copyright (c) 2001-2008 Caucho Technology, Inc.  All rights reserved.
 *
 * The Apache Software License, Version 1.1
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the
 *    distribution.
 *
 * 3. The end-user documentation included with the redistribution, if
 *    any, must include the following acknowlegement:
 *       "This product includes software developed by the
 *        Caucho Technology (http://www.caucho.com/)."
 *    Alternately, this acknowlegement may appear in the software itself,
 *    if and wherever such third-party acknowlegements normally appear.
 *
 * 4. The names "Burlap", "Resin", and "Caucho" must not be used to
 *    endorse or promote products derived from this software without prior
 *    written permission. For written permission, please contact
 *    info@caucho.com.
 *
 * 5. Products derived from this software may not be called "Resin"
 *    nor may "Resin" appear in their names without prior written
 *    permission of Caucho Technology.
 *
 * THIS SOFTWARE IS PROVIDED ``AS IS'' AND ANY EXPRESSED OR IMPLIED
 * WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
 * DISCLAIMED.  IN NO EVENT SHALL CAUCHO TECHNOLOGY OR ITS CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY,
 * OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT
 * OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR
 * BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
 * WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
 * OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN
 * IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 *
 * @author Scott Ferguson
 */

package io.github.wuwen5.hessian.io;

import io.github.wuwen5.hessian.util.IdentityIntMap;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.HashMap;

/**
 * Output stream for Hessian 2 requests.
 *
 * <p>Since HessianOutput does not depend on any classes other than
 * in the JDK, it can be extracted independently into a smaller package.
 *
 * <p>HessianOutput is unbuffered, so any client needs to provide
 * its own buffering.
 *
 * <pre>
 * OutputStream os = ...; // from http connection
 * Hessian2Output out = new Hessian2Output(os);
 * String value;
 *
 * out.startCall("hello", 1); // start hello call
 * out.writeString("arg1");   // write a string argument
 * out.completeCall();        // complete the call
 * </pre>
 */
public class HessianEncoder extends AbstractHessianEncoder implements Hessian2Constants {

    /**
     * should match Resin buffer size for perf
     */
    public static final int SIZE = 8 * 1024;

    /**
     * the output stream/
     */
    protected OutputStream os;

    /**
     * map of references
     */
    private final IdentityIntMap refs = new IdentityIntMap();

    private int refCount = 0;

    private boolean isCloseStreamOnClose;

    /**
     * map of classes
     */
    private final IdentityIntMap classRefs = new IdentityIntMap();

    /**
     * map of types
     */
    private HashMap<String, Integer> typeRefs;

    protected final byte[] buffer = new byte[SIZE];
    protected int offset;

    private boolean isPacket;

    private boolean isUnshared;

    /**
     * Creates a new Hessian output stream, initialized with an
     * underlying output stream.
     */
    public HessianEncoder() {}

    /**
     * Creates a new Hessian output stream, initialized with an
     * underlying output stream.
     *
     * @param os the underlying output stream.
     */
    public HessianEncoder(OutputStream os) {
        this.os = os;
    }

    @Override
    public void init(OutputStream os) {
        reset();

        this.os = os;
    }

    public void initPacket(OutputStream os) {
        resetReferences();

        this.os = os;
    }

    public void setCloseStreamOnClose(boolean isClose) {
        isCloseStreamOnClose = isClose;
    }

    public boolean isCloseStreamOnClose() {
        return isCloseStreamOnClose;
    }

    /**
     * Sets hessian to be "unshared", meaning it will not detect
     * duplicate or circular references.
     */
    @Override
    public boolean setUnshared(boolean isUnshared) {
        boolean oldIsUnshared = this.isUnshared;

        this.isUnshared = isUnshared;

        return oldIsUnshared;
    }

    public void writeVersion() throws IOException {
        flushIfFull();

        buffer[offset++] = (byte) 'H';
        buffer[offset++] = (byte) 2;
        buffer[offset++] = (byte) 0;
    }

    /**
     * Writes any object to the output stream.
     */
    @Override
    public void writeObject(Object object) throws IOException {
        if (object == null) {
            writeNull();
            return;
        }

        HessianSerializer serializer = findSerializerFactory().getObjectSerializer(object.getClass());

        serializer.writeObject(object, this);
    }

    /**
     * Writes the list header to the stream.  List writers will call
     * <code>writeListBegin</code> followed by the list contents and then
     * call <code>writeListEnd</code>.
     * <p>
     *     <pre>
     * <code>
     * list ::= V type value* Z
     *      ::= v type int value*
     * </code>
     * </pre>
     *
     * @return true for variable lists, false for fixed lists
     */
    @Override
    public boolean writeListBegin(int length, String type) throws IOException {
        flushIfFull();

        if (length < 0) {
            if (type != null) {
                buffer[offset++] = (byte) BC_LIST_VARIABLE;
                writeType(type);
            } else {
                buffer[offset++] = (byte) BC_LIST_VARIABLE_UNTYPED;
            }

            return true;
        } else if (length <= LIST_DIRECT_MAX) {
            if (type != null) {
                buffer[offset++] = (byte) (BC_LIST_DIRECT + length);
                writeType(type);
            } else {
                buffer[offset++] = (byte) (BC_LIST_DIRECT_UNTYPED + length);
            }

            return false;
        } else {
            if (type != null) {
                buffer[offset++] = (byte) BC_LIST_FIXED;
                writeType(type);
            } else {
                buffer[offset++] = (byte) BC_LIST_FIXED_UNTYPED;
            }

            writeInt(length);

            return false;
        }
    }

    /**
     * Writes the tail of the list to the stream for a variable-length list.
     */
    @Override
    public void writeListEnd() throws IOException {
        flushIfFull();

        buffer[offset++] = (byte) BC_END;
    }

    /**
     * Writes the map header to the stream.  Map writers will call
     * <code>writeMapBegin</code> followed by the map contents and then
     * call <code>writeMapEnd</code>.
     *<pre>
     * <code>
     * map ::= M type (value value)* Z
     *     ::= H (value value)* Z
     * </code>
     * </pre>
     */
    @Override
    public void writeMapBegin(String type) throws IOException {
        if (SIZE < offset + 32) {
            flushBuffer();
        }

        if (type != null) {
            buffer[offset++] = BC_MAP;

            writeType(type);
        } else {
            buffer[offset++] = BC_MAP_UNTYPED;
        }
    }

    /**
     * Writes the tail of the map to the stream.
     */
    @Override
    public void writeMapEnd() throws IOException {
        if (SIZE < offset + 32) {
            flushBuffer();
        }

        buffer[offset++] = (byte) BC_END;
    }

    /**
     * Writes the object definition
     *<pre>
     * <code>
     * C &lt;string&gt; &lt;int&gt; &lt;string&gt;*
     * </code>
     * </pre>
     * @return the reference number for the class, or -1 if the class is not
     */
    @Override
    public int writeObjectBegin(String type) throws IOException {
        int newRef = classRefs.size();
        int ref = classRefs.put(type, newRef, false);

        if (SIZE < offset + 32) {
            flushBuffer();
        }
        if (newRef != ref) {

            if (ref <= OBJECT_DIRECT_MAX) {
                buffer[offset++] = (byte) (BC_OBJECT_DIRECT + ref);
            } else {
                buffer[offset++] = (byte) BC_OBJECT;
                writeInt(ref);
            }

            return ref;
        } else {

            buffer[offset++] = (byte) 'C';

            writeString(type);

            return -1;
        }
    }

    /**
     * Writes the tail of the class definition to the stream.
     */
    @Override
    public void writeClassFieldLength(int len) throws IOException {
        writeInt(len);
    }

    /**
     * <pre>
     * <code>
     * type ::= string
     *      ::= int
     * </code>
     * </pre>
     */
    private void writeType(String type) throws IOException {
        flushIfFull();

        int len = type.length();
        if (len == 0) {
            throw new IllegalArgumentException("empty type is not allowed");
        }

        if (typeRefs == null) {
            typeRefs = new HashMap<>(32);
        }

        Integer typeRefV = typeRefs.get(type);

        if (typeRefV != null) {
            int typeRef = typeRefV;

            writeInt(typeRef);
        } else {
            typeRefs.put(type, typeRefs.size());

            writeString(type);
        }
    }

    /**
     * Writes a boolean value to the stream.  The boolean will be written
     * with the following syntax:
     *<pre>
     * <code>
     * T
     * F
     * </code>
     * </pre>
     *
     * @param value the boolean value to write.
     */
    @Override
    public void writeBoolean(boolean value) throws IOException {
        if (SIZE < offset + 16) {
            flushBuffer();
        }

        if (value) {
            buffer[offset++] = (byte) 'T';
        } else {
            buffer[offset++] = (byte) 'F';
        }
    }

    /**
     * Writes an integer value to the stream.  The integer will be written
     * with the following syntax:
     *<pre>
     * <code>
     * I b32 b24 b16 b8
     * </code>
     * </pre>
     *
     * @param value the integer value to write.
     */
    @Override
    public void writeInt(int value) throws IOException {
        int i = this.offset;

        if (SIZE <= i + 16) {
            flushBuffer();
            i = this.offset;
        }

        if (INT_DIRECT_MIN <= value && value <= INT_DIRECT_MAX) {
            buffer[i++] = (byte) (value + BC_INT_ZERO);
        } else if (INT_BYTE_MIN <= value && value <= INT_BYTE_MAX) {
            buffer[i++] = (byte) (BC_INT_BYTE_ZERO + (value >> 8));
            buffer[i++] = (byte) (value);
        } else if (INT_SHORT_MIN <= value && value <= INT_SHORT_MAX) {
            buffer[i++] = (byte) (BC_INT_SHORT_ZERO + (value >> 16));
            buffer[i++] = (byte) (value >> 8);
            buffer[i++] = (byte) (value);
        } else {
            buffer[i++] = (byte) ('I');
            buffer[i++] = (byte) (value >> 24);
            buffer[i++] = (byte) (value >> 16);
            buffer[i++] = (byte) (value >> 8);
            buffer[i++] = (byte) (value);
        }

        this.offset = i;
    }

    /**
     * Writes a long value to the stream.  The long will be written
     * with the following syntax:
     *<pre>
     * <code>
     * L b64 b56 b48 b40 b32 b24 b16 b8
     * </code>
     * </pre>
     *
     * @param value the long value to write.
     */
    @Override
    public void writeLong(long value) throws IOException {
        int i = this.offset;

        if (SIZE <= i + 16) {
            flushBuffer();
            i = this.offset;
        }

        if (LONG_DIRECT_MIN <= value && value <= LONG_DIRECT_MAX) {
            buffer[i++] = (byte) (value + BC_LONG_ZERO);
        } else if (LONG_BYTE_MIN <= value && value <= LONG_BYTE_MAX) {
            buffer[i++] = (byte) (BC_LONG_BYTE_ZERO + (value >> 8));
            buffer[i++] = (byte) (value);
        } else if (LONG_SHORT_MIN <= value && value <= LONG_SHORT_MAX) {
            buffer[i++] = (byte) (BC_LONG_SHORT_ZERO + (value >> 16));
            buffer[i++] = (byte) (value >> 8);
            buffer[i++] = (byte) (value);
        } else if (-0x80000000L <= value && value <= 0x7fffffffL) {
            buffer[i] = (byte) BC_LONG_INT;
            buffer[i + 1] = (byte) (value >> 24);
            buffer[i + 2] = (byte) (value >> 16);
            buffer[i + 3] = (byte) (value >> 8);
            buffer[i + 4] = (byte) (value);

            i += 5;
        } else {
            buffer[i] = (byte) 'L';
            buffer[i + 1] = (byte) (value >> 56);
            buffer[i + 2] = (byte) (value >> 48);
            buffer[i + 3] = (byte) (value >> 40);
            buffer[i + 4] = (byte) (value >> 32);
            buffer[i + 5] = (byte) (value >> 24);
            buffer[i + 6] = (byte) (value >> 16);
            buffer[i + 7] = (byte) (value >> 8);
            buffer[i + 8] = (byte) (value);

            i += 9;
        }

        this.offset = i;
    }

    /**
     * Writes a double value to the stream.  The double will be written
     * with the following syntax:
     * <p>
     *     <pre>
     * <code>
     * D b64 b56 b48 b40 b32 b24 b16 b8
     * </code>
     * </pre>
     *
     * @param value the double value to write.
     */
    @Override
    public void writeDouble(double value) throws IOException {
        int i = this.offset;

        if (SIZE <= i + 16) {
            flushBuffer();
            i = this.offset;
        }

        int intValue = (int) value;

        if (intValue == value) {
            if (intValue == 0) {
                buffer[i++] = (byte) BC_DOUBLE_ZERO;

                this.offset = i;

                return;
            } else if (intValue == 1) {
                buffer[i++] = (byte) BC_DOUBLE_ONE;

                this.offset = i;

                return;
            } else if (-0x80 <= intValue && intValue < 0x80) {
                buffer[i++] = (byte) BC_DOUBLE_BYTE;
                buffer[i++] = (byte) intValue;

                this.offset = i;

                return;
            } else if (-0x8000 <= intValue && intValue < 0x8000) {
                buffer[i] = (byte) BC_DOUBLE_SHORT;
                buffer[i + 1] = (byte) (intValue >> 8);
                buffer[i + 2] = (byte) intValue;

                this.offset = i + 3;

                return;
            }
        }

        int mills = (int) (value * 1000);

        if (0.001 * mills == value) {
            buffer[i] = (byte) (BC_DOUBLE_MILL);
            buffer[i + 1] = (byte) (mills >> 24);
            buffer[i + 2] = (byte) (mills >> 16);
            buffer[i + 3] = (byte) (mills >> 8);
            buffer[i + 4] = (byte) (mills);

            this.offset = i + 5;

            return;
        }

        long bits = Double.doubleToLongBits(value);

        buffer[i] = (byte) 'D';
        buffer[i + 1] = (byte) (bits >> 56);
        buffer[i + 2] = (byte) (bits >> 48);
        buffer[i + 3] = (byte) (bits >> 40);
        buffer[i + 4] = (byte) (bits >> 32);
        buffer[i + 5] = (byte) (bits >> 24);
        buffer[i + 6] = (byte) (bits >> 16);
        buffer[i + 7] = (byte) (bits >> 8);
        buffer[i + 8] = (byte) (bits);

        this.offset = i + 9;
    }

    /**
     * Writes a date to the stream.
     *<pre>
     * <code>
     * date ::= d   b7 b6 b5 b4 b3 b2 b1 b0
     *      ::= x65 b3 b2 b1 b0
     * </code>
     * </pre>
     *
     * @param time the date in milliseconds from the epoch in UTC
     */
    @Override
    public void writeUTCDate(long time) throws IOException {
        if (SIZE < offset + 32) {
            flushBuffer();
        }

        int i = this.offset;

        if (time % 60000L == 0) {
            // compact date ::= x65 b3 b2 b1 b0

            long minutes = time / 60000L;

            if ((minutes >> 31) == 0 || (minutes >> 31) == -1) {
                buffer[i++] = (byte) BC_DATE_MINUTE;
                buffer[i++] = ((byte) (minutes >> 24));
                buffer[i++] = ((byte) (minutes >> 16));
                buffer[i++] = ((byte) (minutes >> 8));
                buffer[i++] = ((byte) (minutes >> 0));

                this.offset = i;
                return;
            }
        }

        buffer[i++] = (byte) BC_DATE;
        buffer[i++] = ((byte) (time >> 56));
        buffer[i++] = ((byte) (time >> 48));
        buffer[i++] = ((byte) (time >> 40));
        buffer[i++] = ((byte) (time >> 32));
        buffer[i++] = ((byte) (time >> 24));
        buffer[i++] = ((byte) (time >> 16));
        buffer[i++] = ((byte) (time >> 8));
        buffer[i++] = ((byte) (time));

        this.offset = i;
    }

    /**
     * Writes a null value to the stream.
     * The null will be written with the following syntax
     * <pre>
     * <code>
     * N
     * </code>
     * </pre>
     */
    @Override
    public void writeNull() throws IOException {
        int i = this.offset;

        if (SIZE <= i + 16) {
            flushBuffer();
            i = this.offset;
        }

        buffer[i++] = BC_NULL;

        this.offset = i;
    }

    /**
     * Writes a string value to the stream using UTF-8 encoding.
     * The string will be written with the following syntax:
     * <pre>
     * <code>
     * S b16 b8 string-value
     * </code>
     * If the value is null, it will be written as
     * <code>
     * N
     * </code>
     * </pre>
     *
     * @param value the string value to write.
     */
    @Override
    public void writeString(String value) throws IOException {
        int i = this.offset;

        if (SIZE <= i + 16) {
            flushBuffer();
            i = this.offset;
        }

        if (value == null) {
            buffer[i++] = (byte) BC_NULL;

            this.offset = i;
        } else {
            int length = value.length();
            int strOffset = 0;

            while (length > 0x8000) {
                int sublen = 0x8000;

                i = this.offset;

                if (SIZE <= i + 16) {
                    flushBuffer();
                    i = this.offset;
                }

                // chunk can't end in high surrogate
                char tail = value.charAt(strOffset + sublen - 1);

                if (0xd800 <= tail && tail <= 0xdbff) {
                    sublen--;
                }

                buffer[i] = (byte) BC_STRING_CHUNK;
                buffer[i + 1] = (byte) (sublen >> 8);
                buffer[i + 2] = (byte) (sublen);

                this.offset = i + 3;

                printString(value, strOffset, sublen);

                length -= sublen;
                strOffset += sublen;
            }

            i = this.offset;

            if (SIZE <= i + 16) {
                flushBuffer();
                i = this.offset;
            }

            if (length <= STRING_DIRECT_MAX) {
                buffer[i++] = (byte) (BC_STRING_DIRECT + length);
            } else if (length <= STRING_SHORT_MAX) {
                buffer[i++] = (byte) (BC_STRING_SHORT + (length >> 8));
                buffer[i++] = (byte) (length);
            } else {
                buffer[i++] = (byte) ('S');
                buffer[i++] = (byte) (length >> 8);
                buffer[i++] = (byte) (length);
            }

            this.offset = i;

            printString(value, strOffset, length);
        }
    }

    /**
     * Writes a string value to the stream using UTF-8 encoding.
     * The string will be written with the following syntax:
     * <pre>
     * <code>
     * S b16 b8 string-value
     * </code>
     * If the value is null, it will be written as
     * <code>
     * N
     * </code>
     * </pre>
     *
     */
    @Override
    public void writeString(char[] buffer, int offset, int length) throws IOException {
        if (buffer == null) {
            if (SIZE < this.offset + 16) {
                flushBuffer();
            }

            this.buffer[this.offset++] = (byte) (BC_NULL);
        } else {
            while (length > 0x8000) {
                int sublen = 0x8000;

                if (SIZE < this.offset + 16) {
                    flushBuffer();
                }

                // chunk can't end in high surrogate
                char tail = buffer[offset + sublen - 1];

                if (0xd800 <= tail && tail <= 0xdbff) {
                    sublen--;
                }

                this.buffer[this.offset++] = (byte) BC_STRING_CHUNK;
                this.buffer[this.offset++] = (byte) (sublen >> 8);
                this.buffer[this.offset++] = (byte) (sublen);

                printString(buffer, offset, sublen);

                length -= sublen;
                offset += sublen;
            }

            if (SIZE < this.offset + 16) {
                flushBuffer();
            }

            if (length <= STRING_DIRECT_MAX) {
                this.buffer[this.offset++] = (byte) (BC_STRING_DIRECT + length);
            } else if (length <= STRING_SHORT_MAX) {
                this.buffer[this.offset++] = (byte) (BC_STRING_SHORT + (length >> 8));
                this.buffer[this.offset++] = (byte) length;
            } else {
                this.buffer[this.offset++] = (byte) ('S');
                this.buffer[this.offset++] = (byte) (length >> 8);
                this.buffer[this.offset++] = (byte) (length);
            }

            printString(buffer, offset, length);
        }
    }

    /**
     * Writes a byte array to the stream.
     * The array will be written with the following syntax:
     * <pre>
     * <code>
     * B b16 b18 bytes
     * </code>
     * If the value is null, it will be written as
     * <code>
     * N
     * </code>
     * </pre>
     *
     */
    @Override
    public void writeBytes(byte[] buffer) throws IOException {
        if (buffer == null) {
            if (SIZE < offset + 16) {
                flushBuffer();
            }

            this.buffer[offset++] = BC_NULL;
        } else {
            writeBytes(buffer, 0, buffer.length);
        }
    }

    /**
     * Writes a byte array to the stream.
     * The array will be written with the following syntax:
     * <pre>
     * <code>
     * B b16 b18 bytes
     * </code>
     * If the value is null, it will be written as
     * <code>
     * N
     * </code>
     * </pre>
     */
    @Override
    public void writeBytes(byte[] buffer, int offset, int length) throws IOException {
        if (buffer == null) {
            if (SIZE < this.offset + 16) {
                flushBuffer();
            }

            this.buffer[this.offset++] = (byte) BC_NULL;
        } else {
            while (SIZE - this.offset - 3 < length) {
                int sublen = SIZE - this.offset - 3;

                if (sublen < 16) {
                    flushBuffer();

                    sublen = SIZE - this.offset - 3;

                    if (length < sublen) {
                        sublen = length;
                    }
                }

                this.buffer[this.offset++] = (byte) BC_BINARY_CHUNK;
                this.buffer[this.offset++] = (byte) (sublen >> 8);
                this.buffer[this.offset++] = (byte) sublen;

                System.arraycopy(buffer, offset, this.buffer, this.offset, sublen);
                this.offset += sublen;

                length -= sublen;
                offset += sublen;

                flushBuffer();
            }

            if (SIZE < this.offset + 16) {
                flushBuffer();
            }

            if (length <= BINARY_DIRECT_MAX) {
                this.buffer[this.offset++] = (byte) (BC_BINARY_DIRECT + length);
            } else if (length <= BINARY_SHORT_MAX) {
                this.buffer[this.offset++] = (byte) (BC_BINARY_SHORT + (length >> 8));
                this.buffer[this.offset++] = (byte) (length);
            } else {
                this.buffer[this.offset++] = (byte) 'B';
                this.buffer[this.offset++] = (byte) (length >> 8);
                this.buffer[this.offset++] = (byte) (length);
            }

            System.arraycopy(buffer, offset, this.buffer, this.offset, length);

            this.offset += length;
        }
    }

    /**
     * Writes a byte buffer to the stream.
     * <pre>
     * <code>
     * b b16 b18 bytes
     * </code>
     * </pre>
     */
    @Override
    public void writeByteBufferPart(byte[] buffer, int offset, int length) throws IOException {
        while (length > 0) {
            flushIfFull();

            int sublen = this.buffer.length - this.offset;

            if (length < sublen) {
                sublen = length;
            }

            this.buffer[this.offset++] = BC_BINARY_CHUNK;
            this.buffer[this.offset++] = (byte) (sublen >> 8);
            this.buffer[this.offset++] = (byte) sublen;

            System.arraycopy(buffer, offset, this.buffer, this.offset, sublen);

            this.offset += sublen;
            length -= sublen;
            offset += sublen;
        }
    }

    /**
     * Writes a byte buffer to the stream.
     * <pre>
     * <code>
     * b b16 b18 bytes
     * </code>
     * </pre>
     */
    @Override
    public void writeByteBufferEnd(byte[] buffer, int offset, int length) throws IOException {
        writeBytes(buffer, offset, length);
    }

    /**
     * Returns an output stream to write binary data.
     */
    public OutputStream getBytesOutputStream() throws IOException {
        return new BytesOutputStream();
    }

    /**
     * Writes a full output stream.
     */
    @Override
    public void writeByteStream(InputStream is) throws IOException {
        while (true) {
            int len = SIZE - offset - 3;

            if (len < 16) {
                flushBuffer();
                len = SIZE - offset - 3;
            }

            len = is.read(buffer, offset + 3, len);

            if (len <= 0) {
                buffer[offset++] = BC_BINARY_DIRECT;
                return;
            }

            buffer[offset] = (byte) BC_BINARY_CHUNK;
            buffer[offset + 1] = (byte) (len >> 8);
            buffer[offset + 2] = (byte) (len);

            offset += len + 3;
        }
    }

    /**
     * Writes a reference.
     * <pre>
     * x51 &lt;int&gt;
     * </pre>
     *
     * @param value the integer value to write.
     */
    @Override
    protected void writeRef(int value) throws IOException {
        if (SIZE < offset + 16) {
            flushBuffer();
        }

        buffer[offset++] = (byte) BC_REF;

        writeInt(value);
    }

    /**
     * If the object has already been written, just write its ref.
     *
     * @return true if we're writing a ref.
     */
    @Override
    public boolean addRef(Object object) throws IOException {
        if (isUnshared) {
            refCount++;
            return false;
        }

        int newRef = refCount;

        int ref = addRef(object, newRef, false);

        if (ref != newRef) {
            writeRef(ref);

            return true;
        } else {
            refCount++;

            return false;
        }
    }

    @Override
    public int getRef(Object obj) {
        if (isUnshared) {
            return -1;
        }

        return refs.get(obj);
    }

    /**
     * Removes a reference.
     */
    public boolean removeRef(Object obj) {
        if (isUnshared) {
            return false;
        } else {
            refs.remove(obj);

            return true;
        }
    }

    /**
     * Replaces a reference from one object to another.
     */
    @Override
    public boolean replaceRef(Object oldRef, Object newRef) {
        if (isUnshared) {
            return false;
        }

        int value = refs.get(oldRef);

        if (value >= 0) {
            addRef(newRef, value, true);

            refs.remove(oldRef);

            return true;
        } else {
            return false;
        }
    }

    private int addRef(Object value, int newRef, boolean isReplace) {

        return refs.put(value, newRef, isReplace);
    }

    protected void registerRef(Object obj, boolean flag) {
        addRef(obj, refCount++, flag);
    }

    /**
     * Starts the streaming message
     *
     * <p>A streaming message starts with 'P'</p>
     *
     * <pre>
     * P x02 x00
     * </pre>
     */
    public void writeStreamingObject(Object obj) throws IOException {
        startPacket();

        writeObject(obj);

        endPacket();
    }

    /**
     * Starts a streaming packet
     *
     * <p>A streaming contains a set of chunks, ending with a zero chunk.
     * Each chunk is a length followed by data where the length is
     * encoded by (b1xxxxxxxx)* b0xxxxxxxx</p>
     */
    public void startPacket() throws IOException {
        refs.clear();
        refCount = 0;

        flushBuffer();

        isPacket = true;
        offset = 4;
        // 0x05 = binary
        buffer[0] = (byte) 0x05;
        buffer[1] = (byte) 0x55;
        buffer[2] = (byte) 0x55;
        buffer[3] = (byte) 0x55;
    }

    public void endPacket() throws IOException {
        int i = this.offset;

        if (os == null) {
            this.offset = 0;
            return;
        }

        int len = i - 4;

        if (len < 0x7e) {
            buffer[2] = buffer[0];
        } else {
            buffer[1] = (byte) (0x7e);
            buffer[2] = (byte) (len >> 8);
        }
        buffer[3] = (byte) (len);

        isPacket = false;
        this.offset = 0;

        if (len < 0x7e) {
            os.write(buffer, 2, i - 2);
        } else {
            os.write(buffer, 0, i);
        }
    }

    /**
     * Prints a string to the stream, encoded as UTF-8 with preceeding length
     *
     * @param v the string to print.
     */
    public void printLenString(String v) throws IOException {
        if (SIZE < offset + 16) {
            flushBuffer();
        }

        if (v == null) {
            buffer[offset++] = (byte) (0);
            buffer[offset++] = (byte) (0);
        } else {
            int len = v.length();
            buffer[offset++] = (byte) (len >> 8);
            buffer[offset++] = (byte) (len);

            printString(v, 0, len);
        }
    }

    /**
     * Prints a string to the stream, encoded as UTF-8
     *
     * @param v the string to print.
     */
    public void printString(String v) throws IOException {
        printString(v, 0, v.length());
    }

    /**
     * Prints a string to the stream, encoded as UTF-8
     *
     * @param v the string to print.
     */
    public void printString(String v, int strOffset, int length) throws IOException {
        int ioffset = this.offset;

        for (int i = 0; i < length; i++) {
            if (SIZE <= ioffset + 16) {
                this.offset = ioffset;
                flushBuffer();
                ioffset = this.offset;
            }

            char ch = v.charAt(i + strOffset);

            if (ch < 0x80) {
                buffer[ioffset++] = (byte) (ch);
            } else if (ch < 0x800) {
                buffer[ioffset++] = (byte) (0xc0 + ((ch >> 6) & 0x1f));
                buffer[ioffset++] = (byte) (0x80 + (ch & 0x3f));
            } else {
                buffer[ioffset++] = (byte) (0xe0 + ((ch >> 12) & 0xf));
                buffer[ioffset++] = (byte) (0x80 + ((ch >> 6) & 0x3f));
                buffer[ioffset++] = (byte) (0x80 + (ch & 0x3f));
            }
        }

        this.offset = ioffset;
    }

    /**
     * Prints a string to the stream, encoded as UTF-8
     *
     * @param v the string to print.
     */
    public void printString(char[] v, int strOffset, int length) throws IOException {
        int ioffset = this.offset;

        for (int i = 0; i < length; i++) {
            if (SIZE <= ioffset + 16) {
                this.offset = ioffset;
                flushBuffer();
                ioffset = this.offset;
            }

            char ch = v[i + strOffset];

            if (ch < 0x80) {
                buffer[ioffset++] = (byte) (ch);
            } else if (ch < 0x800) {
                buffer[ioffset++] = (byte) (0xc0 + ((ch >> 6) & 0x1f));
                buffer[ioffset++] = (byte) (0x80 + (ch & 0x3f));
            } else {
                buffer[ioffset++] = (byte) (0xe0 + ((ch >> 12) & 0xf));
                buffer[ioffset++] = (byte) (0x80 + ((ch >> 6) & 0x3f));
                buffer[ioffset++] = (byte) (0x80 + (ch & 0x3f));
            }
        }

        this.offset = ioffset;
    }

    protected final void flushIfFull() throws IOException {

        if (SIZE < offset + 32) {
            flushBuffer();
        }
    }

    @Override
    public final void flush() throws IOException {
        flushBuffer();

        if (os != null) {
            os.flush();
        }
    }

    public final void flushBuffer() throws IOException {
        int ioffset = this.offset;

        if (!isPacket && ioffset > 0) {
            this.offset = 0;
            if (os != null) {
                os.write(buffer, 0, ioffset);
            }
        } else if (isPacket && ioffset > 4) {
            int len = ioffset - 4;

            buffer[0] |= (byte) 0x80;
            buffer[1] = (byte) (0x7e);
            buffer[2] = (byte) (len >> 8);
            buffer[3] = (byte) (len);
            this.offset = 4;

            if (os != null) {
                os.write(buffer, 0, ioffset);
            }

            buffer[0] = (byte) 0x00;
            buffer[1] = (byte) 0x56;
            buffer[2] = (byte) 0x56;
            buffer[3] = (byte) 0x56;
        }
    }

    @Override
    public void close() throws IOException {
        // hessian/3a8c
        flush();

        OutputStream los = this.os;
        this.os = null;

        if (los != null && isCloseStreamOnClose) {
            los.close();
        }
    }

    public void free() {
        reset();

        os = null;
        isCloseStreamOnClose = false;
    }

    /**
     * Resets the references for streaming.
     */
    @Override
    public void resetReferences() {
        refs.clear();
        refCount = 0;
    }

    /**
     * Resets all counters and references
     */
    public void reset() {
        refs.clear();
        refCount = 0;

        classRefs.clear();
        typeRefs = null;
        offset = 0;
        isPacket = false;
        isUnshared = false;
    }

    class BytesOutputStream extends OutputStream {
        private int startOffset;

        BytesOutputStream() throws IOException {
            if (SIZE < offset + 16) {
                HessianEncoder.this.flushBuffer();
            }

            startOffset = offset;
            offset += 3; // skip 'b' xNN xNN
        }

        @Override
        public void write(int ch) throws IOException {
            if (SIZE <= offset) {
                int length = (offset - startOffset) - 3;

                buffer[startOffset] = (byte) BC_BINARY_CHUNK;
                buffer[startOffset + 1] = (byte) (length >> 8);
                buffer[startOffset + 2] = (byte) (length);

                HessianEncoder.this.flushBuffer();

                startOffset = offset;
                offset += 3;
            }

            buffer[offset++] = (byte) ch;
        }

        @Override
        public void write(byte[] buffer, int offset, int length) throws IOException {
            while (length > 0) {
                int sublen = SIZE - HessianEncoder.this.offset;

                if (length < sublen) {
                    sublen = length;
                }

                if (sublen > 0) {
                    System.arraycopy(buffer, offset, HessianEncoder.this.buffer, HessianEncoder.this.offset, sublen);
                    HessianEncoder.this.offset += sublen;
                }

                length -= sublen;
                offset += sublen;

                if (SIZE <= HessianEncoder.this.offset) {
                    int chunkLength = (HessianEncoder.this.offset - startOffset) - 3;

                    HessianEncoder.this.buffer[startOffset] = (byte) BC_BINARY_CHUNK;
                    HessianEncoder.this.buffer[startOffset + 1] = (byte) (chunkLength >> 8);
                    HessianEncoder.this.buffer[startOffset + 2] = (byte) (chunkLength);

                    HessianEncoder.this.flushBuffer();

                    startOffset = HessianEncoder.this.offset;
                    HessianEncoder.this.offset += 3;
                }
            }
        }

        @Override
        public void close() throws IOException {
            int i = this.startOffset;
            this.startOffset = -1;

            if (i < 0) {
                return;
            }

            int length = (offset - i) - 3;

            buffer[i] = (byte) 'B';
            buffer[i + 1] = (byte) (length >> 8);
            buffer[i + 2] = (byte) (length);

            HessianEncoder.this.flushBuffer();
        }
    }
}

wuwen5 / hessian / 17882005206

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous