pw.simplyintricate.bitcoin.models.datastructures.VariableString.java Source code

Introduction

Here is the source code for pw.simplyintricate.bitcoin.models.datastructures.VariableString.java

Source

/*
 * lib-bitcoinj
 * Copyright (c) 2014, Stephen Liang, All rights reserved.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Affero General Public
 * License as published by the Free Software Foundation; either
 * version 3.0 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public
 * License along with this library.
 */

package pw.simplyintricate.bitcoin.models.datastructures;

import com.google.common.io.LittleEndianDataInputStream;
import com.google.common.primitives.UnsignedInteger;
import org.apache.commons.lang3.builder.ToStringBuilder;
import pw.simplyintricate.bitcoin.io.HybridByteArrayDataOutput;

import java.io.IOException;
import java.io.UnsupportedEncodingException;

public class VariableString {
    private final VariableInteger variableInteger;
    private final byte[] string;

    public VariableString(String string) {
        try {
            byte[] stringBytes = string.getBytes("iso-8859-1");
            byte[] cStyleString = new byte[stringBytes.length + 1];

            System.arraycopy(stringBytes, 0, cStyleString, 0, stringBytes.length);
            this.string = cStyleString;
            this.variableInteger = new VariableInteger(UnsignedInteger.fromIntBits(stringBytes.length));
        } catch (UnsupportedEncodingException e) {
            throw new IllegalArgumentException("The string provided is invalid", e);
        }
    }

    public byte[] toByteArray() {
        HybridByteArrayDataOutput writer = new HybridByteArrayDataOutput();

        writer.write(variableInteger.toByteArray());
        writer.write(string, 0, string.length - 1);

        return writer.toByteArray();
    }

    public static VariableString fromInputStream(LittleEndianDataInputStream reader) throws IOException {
        VariableInteger variableInteger = VariableInteger.fromInputStream(reader);
        int stringSize = variableInteger.getValue().intValue();
        byte[] stringBuffer = new byte[stringSize];
        reader.read(stringBuffer, 0, stringSize);

        VariableString variableString = new VariableString(new String(stringBuffer));

        return variableString;
    }

    @Override
    public String toString() {
        return ToStringBuilder.reflectionToString(this);
    }
}