/*
 * Copyright (C) 2020 The Android Open Source Project
 *
 * 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.
 */

package com.android.test.protoinputstream;

import android.util.proto.ProtoInputStream;
import android.util.proto.ProtoParseException;
import android.util.proto.ProtoStream;

import junit.framework.TestCase;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;

public class ProtoInputStreamIncompleteValueTest extends TestCase {

    /**
     * Test that an incomplete varint at the end of a stream throws an exception
     */
    public void testIncompleteVarint() throws IOException {
        final long fieldFlags = ProtoStream.FIELD_COUNT_SINGLE | ProtoStream.FIELD_TYPE_INT32;

        final long fieldId1 = fieldFlags | ((long) 1 & 0x0ffffffffL);

        final byte[] protobuf = new byte[]{
                // 1 : varint -> invalid varint value
                (byte) 0x08,
                (byte) 0xff,
        };

        InputStream stream = new ByteArrayInputStream(protobuf);
        final ProtoInputStream pi = new ProtoInputStream(stream);

        while (pi.nextField() != ProtoInputStream.NO_MORE_FIELDS) {
            try {
                switch (pi.getFieldNumber()) {
                    case (int) fieldId1:
                        pi.readInt(fieldId1);
                        fail("Should have thrown a ProtoParseException");
                        break;
                    default:
                        fail("Unexpected field id " + pi.getFieldNumber());
                }
            } catch (ProtoParseException ppe) {
                // good
                stream.close();
                return;
            }
        }
        stream.close();
        fail("Test should not have reached this point...");
    }

    /**
     * Test that an incomplete fixed64 at the end of a stream throws an exception
     */
    public void testIncompleteFixed64() throws IOException {
        final long fieldFlags = ProtoStream.FIELD_COUNT_SINGLE | ProtoStream.FIELD_TYPE_FIXED64;

        final long fieldId2 = fieldFlags | ((long) 2 & 0x0ffffffffL);

        final byte[] protobuf = new byte[]{
                // 2 -> invalid fixed64
                (byte) 0x11,
                (byte) 0x01, (byte) 0x00, (byte) 0x00, (byte) 0x00,
                (byte) 0x00, (byte) 0x00, (byte) 0x00,
        };

        InputStream stream = new ByteArrayInputStream(protobuf);
        final ProtoInputStream pi = new ProtoInputStream(stream);

        while (pi.nextField() != ProtoInputStream.NO_MORE_FIELDS) {
            try {
                switch (pi.getFieldNumber()) {
                    case (int) fieldId2:
                        pi.readLong(fieldId2);
                        fail("Should have thrown a ProtoParseException");
                        break;
                    default:
                        fail("Unexpected field id " + pi.getFieldNumber());
                }
            } catch (ProtoParseException ppe) {
                // good
                stream.close();
                return;
            }
        }
        stream.close();
        fail("Test should not have reached this point...");
    }

    /**
     * Test that an incomplete length delimited value at the end of a stream throws an exception
     */
    public void testIncompleteLengthDelimited() throws IOException {
        final long fieldFlags = ProtoStream.FIELD_COUNT_SINGLE | ProtoStream.FIELD_TYPE_BYTES;

        final long fieldId5 = fieldFlags | ((long) 5 & 0x0ffffffffL);

        final byte[] protobuf = new byte[]{
                // 5 -> invalid byte array (has size 5 but only 4 values)
                (byte) 0x2a,
                (byte) 0x05,
                (byte) 0x00, (byte) 0x01, (byte) 0x02, (byte) 0x03,
        };

        InputStream stream = new ByteArrayInputStream(protobuf);
        final ProtoInputStream pi = new ProtoInputStream(stream);

        while (pi.nextField() != ProtoInputStream.NO_MORE_FIELDS) {
            try {
                switch (pi.getFieldNumber()) {
                    case (int) fieldId5:
                        pi.readBytes(fieldId5);
                        fail("Should have thrown a ProtoParseException");
                        break;
                    default:
                        fail("Unexpected field id " + pi.getFieldNumber());
                }
            } catch (ProtoParseException ppe) {
                // good
                stream.close();
                return;
            }
        }
        stream.close();
        fail("Test should not have reached this point...");
    }

    /**
     * Test that an incomplete fixed32 at the end of a stream throws an exception
     */
    public void testIncompleteFixed32() throws IOException {
        final long fieldFlags = ProtoStream.FIELD_COUNT_SINGLE | ProtoStream.FIELD_TYPE_FIXED32;

        final long fieldId2 = fieldFlags | ((long) 2 & 0x0ffffffffL);

        final byte[] protobuf = new byte[]{
                // 2 ->  invalid fixed32
                (byte) 0x15,
                (byte) 0x01, (byte) 0x00, (byte) 0x00,
        };

        InputStream stream = new ByteArrayInputStream(protobuf);
        final ProtoInputStream pi = new ProtoInputStream(stream);

        while (pi.nextField() != ProtoInputStream.NO_MORE_FIELDS) {
            try {
                switch (pi.getFieldNumber()) {
                    case (int) fieldId2:
                        pi.readInt(fieldId2);
                        fail("Should have thrown a ProtoParseException");
                        break;
                    default:
                        fail("Unexpected field id " + pi.getFieldNumber());
                }
            } catch (ProtoParseException ppe) {
                // good
                stream.close();
                return;
            }
        }
        stream.close();
        fail("Test should not have reached this point...");
    }
}