org.bson.io.Bits Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mongo-java-driver
The MongoDB Java Driver uber-artifact, containing mongodb-driver, mongodb-driver-core, and bson
There is a newer version: 3.9.1
/*
 * Copyright (c) 2008-2014 MongoDB, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

// Bits.java


package org.bson.io;

import java.io.EOFException;
import java.io.IOException;
import java.io.InputStream;

public class Bits {

    public static void readFully( InputStream in, byte[] b )
        throws IOException {
        readFully( in , b , b.length );
    }

    public static void readFully( InputStream in, byte[] b, int length )
        throws IOException {
        readFully(in, b, 0, length);
    }

    public static void readFully( InputStream in, byte[] b, int startOffset, int length )
        throws IOException {

        if (b.length < length + startOffset) {
            throw new IllegalArgumentException("Buffer is too small");
        }

        int offset = startOffset;
        int toRead = length;
        while ( toRead > 0 ){
            int bytesRead = in.read( b, offset , toRead );
            if ( bytesRead < 0 )
                throw new EOFException();
            toRead -= bytesRead;
            offset += bytesRead;
        }
    }

    public static int readInt( InputStream in )
        throws IOException {
        return readInt( in , new byte[4] );
    }

    public static int readInt( InputStream in , byte[] data )
        throws IOException {
        readFully(in, data, 4);
        return readInt(data);
    }

    public static int readInt( byte[] data ) {
        return readInt( data , 0 );
    }

    public static int readInt( byte[] data , int offset ) {
        int x = 0;
        x |= ( 0xFF & data[offset+0] ) << 0;
        x |= ( 0xFF & data[offset+1] ) << 8;
        x |= ( 0xFF & data[offset+2] ) << 16;
        x |= ( 0xFF & data[offset+3] ) << 24;
        return x;
    }

    public static int readIntBE( byte[] data , int offset ) {
        int x = 0;
        x |= ( 0xFF & data[offset+0] ) << 24;
        x |= ( 0xFF & data[offset+1] ) << 16;
        x |= ( 0xFF & data[offset+2] ) << 8;
        x |= ( 0xFF & data[offset+3] ) << 0;
        return x;
    }

    public static long readLong( InputStream in )
        throws IOException {
        return readLong( in , new byte[8] );
    }


    public static long readLong( InputStream in , byte[] data )
        throws IOException {
        readFully(in, data, 8);
        return readLong(data);
    }

    public static long readLong( byte[] data ) {
        return readLong( data , 0 );
    }
    
    public static long readLong( byte[] data , int offset ) {
        long x = 0;
        x |= ( 0xFFL & data[offset+0] ) << 0;
        x |= ( 0xFFL & data[offset+1] ) << 8;
        x |= ( 0xFFL & data[offset+2] ) << 16;
        x |= ( 0xFFL & data[offset+3] ) << 24;
        x |= ( 0xFFL & data[offset+4] ) << 32;
        x |= ( 0xFFL & data[offset+5] ) << 40;
        x |= ( 0xFFL & data[offset+6] ) << 48;
        x |= ( 0xFFL & data[offset+7] ) << 56;
        return x;
    }
}