at.yawk.dbus.protocol.object.ArrayObject Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dbus-protocol Show documentation
The newest version!
/*
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at https://mozilla.org/MPL/2.0/.
 */

package at.yawk.dbus.protocol.object;

import at.yawk.dbus.protocol.type.ArrayTypeDefinition;
import at.yawk.dbus.protocol.type.TypeDefinition;
import io.netty.buffer.ByteBuf;
import java.util.AbstractList;
import java.util.ArrayList;
import java.util.List;
import lombok.EqualsAndHashCode;
import lombok.Getter;
import lombok.ToString;

/**
 * @author yawkat
 */
@Getter
@EqualsAndHashCode
@ToString
public abstract class ArrayObject implements DbusObject {
    static final int ARRAY_MAX_BYTES = 1 << 26; // 64 MiB

    private final ArrayTypeDefinition type;

    ArrayObject(ArrayTypeDefinition type) {
        this.type = type;
    }

    public List getValues() {
        return new AbstractList() {
            @Override
            public DbusObject get(int index) {
                return ArrayObject.this.get(index);
            }

            @Override
            public int size() {
                return ArrayObject.this.size();
            }
        };
    }

    public static ArrayObject create(ArrayTypeDefinition type, List values) {
        for (DbusObject value : values) {
            TypeDefinition valueType = value.getType();
            if (!valueType.equals(type.getMemberType())) {
                throw new IllegalArgumentException(
                        "Mismatched value type " + valueType + ", expected " + type.getMemberType());
            }
        }
        return new SimpleArrayObject(type, values);
    }

    public static ArrayObject deserialize(ArrayTypeDefinition type, AlignableByteBuf buf) {
        buf.alignRead(4);
        int bytes = Math.toIntExact(buf.readUnsignedInt());
        if (bytes > ARRAY_MAX_BYTES) {
            throw new DeserializerException(
                    "Array exceeded length limit (got " + bytes + " bytes, max is " + ARRAY_MAX_BYTES + " bytes)");
        }
        int start = buf.readerIndex();
        List values = new ArrayList<>();
        while ((buf.readerIndex() - start) < bytes) {
            values.add(type.getMemberType().deserialize(buf));
        }
        return new SimpleArrayObject(type, values);
    }

    @Override
    public void serialize(AlignableByteBuf buf) {
        ByteBuf tempBuffer = serializeValues(ArrayObject.allocateBufferForWrite(buf.getBuffer()));

        buf.alignWrite(4);
        buf.getBuffer().writeInt(tempBuffer.writerIndex());
        if (tempBuffer.isReadable()) {
            buf.getBuffer().writeBytes(tempBuffer);
        }
        tempBuffer.release();
    }

    protected abstract ByteBuf serializeValues(ByteBuf writeTarget);

    protected abstract int size();

    public abstract DbusObject get(int i);

    /**
     * Allocate a write buffer that will be written to the given target. Inherits endianness.
     */
    static ByteBuf allocateBufferForWrite(ByteBuf writeTarget) {
        return writeTarget.alloc().buffer().order(writeTarget.order());
    }

    static ByteBuf allocateBufferForWrite(AlignableByteBuf writeTarget) {
        return allocateBufferForWrite(writeTarget.getBuffer());
    }

    @ToString(callSuper = true)
    private static final class SimpleArrayObject extends ArrayObject {
        private final List values;

        SimpleArrayObject(ArrayTypeDefinition type, List values) {
            super(type);
            this.values = values;
        }

        @Override
        protected ByteBuf serializeValues(ByteBuf writeTarget) {
            AlignableByteBuf tempBuffer = AlignableByteBuf.fromAlignedBuffer(allocateBufferForWrite(writeTarget), 8);
            for (DbusObject value : values) {
                // we align to the 8-byte-mark later so we don't need to pass anything but 0
                value.serialize(tempBuffer);
            }
            return tempBuffer.getBuffer();
        }

        @Override
        protected int size() {
            return values.size();
        }

        @Override
        public DbusObject get(int i) {
            return values.get(i);
        }
    }
}