// Copyright (C) 2017 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.

#include "src/logd/LogEvent.h"

#include <gtest/gtest.h>

#include "flags/FlagProvider.h"
#include "frameworks/proto_logging/stats/atoms.pb.h"
#include "frameworks/proto_logging/stats/enums/stats/launcher/launcher.pb.h"
#include "log/log_event_list.h"
#include "stats_annotations.h"
#include "stats_event.h"
#include "statsd_test_util.h"

#ifdef __ANDROID__

namespace android {
namespace os {
namespace statsd {

using std::string;
using std::vector;
using ::util::ProtoOutputStream;
using ::util::ProtoReader;

namespace {

Field getField(int32_t tag, const vector<int32_t>& pos, int32_t depth,
               const vector<uint8_t>& last) {
    Field f(tag, (int32_t*)pos.data(), depth);

    // only decorate last position for depths with repeated fields (depth 1)
    if (depth > 0 && last[1]) f.decorateLastPos(1);

    return f;
}

bool createFieldWithBoolAnnotationLogEvent(LogEvent* logEvent, uint8_t typeId, uint8_t annotationId,
                                           bool annotationValue, bool doHeaderPrefetch) {
    AStatsEvent* statsEvent = AStatsEvent_obtain();
    createStatsEvent(statsEvent, typeId, /*atomId=*/100);
    AStatsEvent_addBoolAnnotation(statsEvent, annotationId, annotationValue);
    AStatsEvent_build(statsEvent);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    if (doHeaderPrefetch) {
        // Testing LogEvent header prefetch logic
        const LogEvent::BodyBufferInfo bodyInfo = logEvent->parseHeader(buf, size);
        logEvent->parseBody(bodyInfo);
    } else {
        logEvent->parseBuffer(buf, size);
    }
    AStatsEvent_release(statsEvent);

    return logEvent->isValid();
}

bool createFieldWithIntAnnotationLogEvent(LogEvent* logEvent, uint8_t typeId, uint8_t annotationId,
                                          int annotationValue, bool doHeaderPrefetch) {
    AStatsEvent* statsEvent = AStatsEvent_obtain();
    createStatsEvent(statsEvent, typeId, /*atomId=*/100);
    AStatsEvent_addInt32Annotation(statsEvent, annotationId, annotationValue);
    AStatsEvent_build(statsEvent);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    if (doHeaderPrefetch) {
        // Testing LogEvent header prefetch logic
        const LogEvent::BodyBufferInfo bodyInfo = logEvent->parseHeader(buf, size);
        logEvent->parseBody(bodyInfo);
    } else {
        logEvent->parseBuffer(buf, size);
    }
    AStatsEvent_release(statsEvent);

    return logEvent->isValid();
}

bool createAtomLevelIntAnnotationLogEvent(LogEvent* logEvent, uint8_t typeId, uint8_t annotationId,
                                          int annotationValue, bool doHeaderPrefetch) {
    AStatsEvent* statsEvent = AStatsEvent_obtain();
    AStatsEvent_setAtomId(statsEvent, /*atomId=*/100);
    AStatsEvent_addInt32Annotation(statsEvent, annotationId, annotationValue);
    fillStatsEventWithSampleValue(statsEvent, typeId);
    AStatsEvent_build(statsEvent);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    if (doHeaderPrefetch) {
        // Testing LogEvent header prefetch logic
        const LogEvent::BodyBufferInfo bodyInfo = logEvent->parseHeader(buf, size);
        logEvent->parseBody(bodyInfo);
    } else {
        logEvent->parseBuffer(buf, size);
    }
    AStatsEvent_release(statsEvent);

    return logEvent->isValid();
}

bool createAtomLevelBoolAnnotationLogEvent(LogEvent* logEvent, uint8_t typeId, uint8_t annotationId,
                                           bool annotationValue, bool doHeaderPrefetch) {
    AStatsEvent* statsEvent = AStatsEvent_obtain();
    AStatsEvent_setAtomId(statsEvent, /*atomId=*/100);
    AStatsEvent_addBoolAnnotation(statsEvent, annotationId, annotationValue);
    fillStatsEventWithSampleValue(statsEvent, typeId);
    AStatsEvent_build(statsEvent);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    if (doHeaderPrefetch) {
        // Testing LogEvent header prefetch logic
        const LogEvent::BodyBufferInfo bodyInfo = logEvent->parseHeader(buf, size);
        logEvent->parseBody(bodyInfo);
    } else {
        logEvent->parseBuffer(buf, size);
    }
    AStatsEvent_release(statsEvent);

    return logEvent->isValid();
}

}  // anonymous namespace

// Setup for parameterized tests.
class LogEventTestBadAnnotationFieldTypes : public testing::TestWithParam<std::tuple<int, bool>> {
public:
    static std::string ToString(testing::TestParamInfo<std::tuple<int, bool>> info) {
        const std::string boolName = std::get<1>(info.param) ? "_prefetchTrue" : "_prefetchFalse";

        switch (std::get<0>(info.param)) {
            case INT32_TYPE:
                return "Int32" + boolName;
            case INT64_TYPE:
                return "Int64" + boolName;
            case STRING_TYPE:
                return "String" + boolName;
            case LIST_TYPE:
                return "List" + boolName;
            case FLOAT_TYPE:
                return "Float" + boolName;
            case BYTE_ARRAY_TYPE:
                return "ByteArray" + boolName;
            case ATTRIBUTION_CHAIN_TYPE:
                return "AttributionChain" + boolName;
            default:
                return "Unknown" + boolName;
        }
    }
};

// TODO(b/222539899): Add BOOL_TYPE value once parseAnnotations is updated to check specific
// typeIds. BOOL_TYPE should be a bad field type for is_uid, nested, and reset state annotations.
INSTANTIATE_TEST_SUITE_P(BadAnnotationFieldTypes, LogEventTestBadAnnotationFieldTypes,
                         testing::Combine(testing::Values(INT32_TYPE, INT64_TYPE, STRING_TYPE,
                                                          LIST_TYPE, FLOAT_TYPE, BYTE_ARRAY_TYPE,
                                                          ATTRIBUTION_CHAIN_TYPE),
                                          testing::Bool()),
                         LogEventTestBadAnnotationFieldTypes::ToString);

class LogEventTest : public testing::TestWithParam<bool> {
public:
    bool ParseBuffer(LogEvent& logEvent, const uint8_t* buf, size_t size) {
        size_t bufferOffset = 0;
        if (GetParam()) {
            // Testing LogEvent header prefetch logic
            const LogEvent::BodyBufferInfo bodyInfo = logEvent.parseHeader(buf, size);
            EXPECT_TRUE(logEvent.isParsedHeaderOnly());
            const bool parseResult = logEvent.parseBody(bodyInfo);
            EXPECT_EQ(parseResult, logEvent.isValid());
            EXPECT_FALSE(logEvent.isParsedHeaderOnly());
        } else {
            const bool parseResult = logEvent.parseBuffer(buf, size);
            EXPECT_EQ(parseResult, logEvent.isValid());
            EXPECT_FALSE(logEvent.isParsedHeaderOnly());
        }
        return logEvent.isValid();
    }

    static std::string ToString(testing::TestParamInfo<bool> info) {
        return info.param ? "PrefetchTrue" : "PrefetchFalse";
    }

public:
    void doTestArrayParsing() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestEmptyStringArray() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestArrayTooManyElements() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestEmptyArray() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestEmptyArrayWithAnnotations() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestAnnotationIdIsUid_RepeatedIntAndOtherFields() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestAnnotationIdIsUid_RepeatedIntOneEntry() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestAnnotationIdIsUid_EmptyIntArray() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestAnnotationIdIsUid_BadRepeatedInt64() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestAnnotationIdIsUid_BadRepeatedString() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestUidAnnotationWithInt8MaxValues() __INTRODUCED_IN(__ANDROID_API_T__);
    void doTestInvalidBufferParsing() __INTRODUCED_IN(__ANDROID_API_T__);
};

INSTANTIATE_TEST_SUITE_P(LogEventTestBufferParsing, LogEventTest, testing::Bool(),
                         LogEventTest::ToString);

TEST_P(LogEventTest, TestPrimitiveParsing) {
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeInt32(event, 10);
    AStatsEvent_writeInt64(event, 0x123456789);
    AStatsEvent_writeFloat(event, 2.0);
    AStatsEvent_writeBool(event, true);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());
    EXPECT_FALSE(logEvent.hasAttributionChain());

    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(4, values.size());

    const FieldValue& int32Item = values[0];
    Field expectedField = getField(100, {1, 1, 1}, 0, {false, false, false});
    EXPECT_EQ(expectedField, int32Item.mField);
    EXPECT_EQ(Type::INT, int32Item.mValue.getType());
    EXPECT_EQ(10, int32Item.mValue.int_value);

    const FieldValue& int64Item = values[1];
    expectedField = getField(100, {2, 1, 1}, 0, {false, false, false});
    EXPECT_EQ(expectedField, int64Item.mField);
    EXPECT_EQ(Type::LONG, int64Item.mValue.getType());
    EXPECT_EQ(0x123456789, int64Item.mValue.long_value);

    const FieldValue& floatItem = values[2];
    expectedField = getField(100, {3, 1, 1}, 0, {false, false, false});
    EXPECT_EQ(expectedField, floatItem.mField);
    EXPECT_EQ(Type::FLOAT, floatItem.mValue.getType());
    EXPECT_EQ(2.0, floatItem.mValue.float_value);

    const FieldValue& boolItem = values[3];
    expectedField = getField(100, {4, 1, 1}, 0, {true, false, false});
    EXPECT_EQ(expectedField, boolItem.mField);
    EXPECT_EQ(Type::INT, boolItem.mValue.getType());  // FieldValue does not support boolean type
    EXPECT_EQ(1, boolItem.mValue.int_value);

    AStatsEvent_release(event);
}

TEST_P(LogEventTest, TestEventWithInvalidHeaderParsing) {
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeInt32(event, 10);
    AStatsEvent_writeInt64(event, 0x123456789);
    AStatsEvent_writeFloat(event, 2.0);
    AStatsEvent_writeBool(event, true);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    // Corrupt LogEvent header info
    // OBJECT_TYPE | NUM_FIELDS | TIMESTAMP | ATOM_ID
    // Corrupting first 4 bytes will be sufficient
    uint8_t* bufMod = const_cast<uint8_t*>(buf);
    memset(static_cast<void*>(bufMod), 4, ERROR_TYPE);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_FALSE(ParseBuffer(logEvent, buf, size));
    EXPECT_FALSE(logEvent.isValid());
    EXPECT_FALSE(logEvent.isParsedHeaderOnly());

    AStatsEvent_release(event);
}

TEST(LogEventTestParsing, TestFetchHeaderOnly) {
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeInt32(event, 10);
    AStatsEvent_writeInt64(event, 0x123456789);
    AStatsEvent_writeFloat(event, 2.0);
    AStatsEvent_writeBool(event, true);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    const LogEvent::BodyBufferInfo bodyInfo = logEvent.parseHeader(buf, size);
    EXPECT_TRUE(logEvent.isValid());
    EXPECT_TRUE(logEvent.isParsedHeaderOnly());

    AStatsEvent_release(event);

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());
    EXPECT_FALSE(logEvent.hasAttributionChain());
    ASSERT_EQ(0, logEvent.getValues().size());
}

TEST_P(LogEventTest, TestStringAndByteArrayParsing) {
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    string str = "test";
    AStatsEvent_writeString(event, str.c_str());
    AStatsEvent_writeByteArray(event, (uint8_t*)str.c_str(), str.length());
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());
    EXPECT_FALSE(logEvent.hasAttributionChain());

    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(2, values.size());

    const FieldValue& stringItem = values[0];
    Field expectedField = getField(100, {1, 1, 1}, 0, {false, false, false});
    EXPECT_EQ(expectedField, stringItem.mField);
    EXPECT_EQ(Type::STRING, stringItem.mValue.getType());
    EXPECT_EQ(str, stringItem.mValue.str_value);

    const FieldValue& storageItem = values[1];
    expectedField = getField(100, {2, 1, 1}, 0, {true, false, false});
    EXPECT_EQ(expectedField, storageItem.mField);
    EXPECT_EQ(Type::STORAGE, storageItem.mValue.getType());
    vector<uint8_t> expectedValue = {'t', 'e', 's', 't'};
    EXPECT_EQ(expectedValue, storageItem.mValue.storage_value);

    AStatsEvent_release(event);
}

TEST_P(LogEventTest, TestEmptyString) {
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    string empty = "";
    AStatsEvent_writeString(event, empty.c_str());
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());
    EXPECT_FALSE(logEvent.hasAttributionChain());

    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(1, values.size());

    const FieldValue& item = values[0];
    Field expectedField = getField(100, {1, 1, 1}, 0, {true, false, false});
    EXPECT_EQ(expectedField, item.mField);
    EXPECT_EQ(Type::STRING, item.mValue.getType());
    EXPECT_EQ(empty, item.mValue.str_value);

    AStatsEvent_release(event);
}

TEST_P(LogEventTest, TestByteArrayWithNullCharacter) {
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    uint8_t message[] = {'\t', 'e', '\0', 's', 't'};
    AStatsEvent_writeByteArray(event, message, 5);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());

    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(1, values.size());

    const FieldValue& item = values[0];
    Field expectedField = getField(100, {1, 1, 1}, 0, {true, false, false});
    EXPECT_EQ(expectedField, item.mField);
    EXPECT_EQ(Type::STORAGE, item.mValue.getType());
    vector<uint8_t> expectedValue(message, message + 5);
    EXPECT_EQ(expectedValue, item.mValue.storage_value);

    AStatsEvent_release(event);
}

TEST_P(LogEventTest, TestTooManyTopLevelElements) {
    int32_t numElements = 128;
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);

    for (int i = 0; i < numElements; i++) {
        AStatsEvent_writeInt32(event, i);
    }

    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);
    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_FALSE(ParseBuffer(logEvent, buf, size));

    AStatsEvent_release(event);
}

TEST_P(LogEventTest, TestAttributionChain) {
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);

    string tag1 = "tag1";
    string tag2 = "tag2";

    uint32_t uids[] = {1001, 1002};
    const char* tags[] = {tag1.c_str(), tag2.c_str()};

    AStatsEvent_writeAttributionChain(event, uids, tags, 2);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());

    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(4, values.size());  // 2 per attribution node

    std::pair<size_t, size_t> attrIndexRange;
    EXPECT_TRUE(logEvent.hasAttributionChain(&attrIndexRange));
    EXPECT_EQ(0, attrIndexRange.first);
    EXPECT_EQ(3, attrIndexRange.second);

    // Check first attribution node
    const FieldValue& uid1Item = values[0];
    Field expectedField = getField(100, {1, 1, 1}, 2, {true, false, false});
    EXPECT_EQ(expectedField, uid1Item.mField);
    EXPECT_EQ(Type::INT, uid1Item.mValue.getType());
    EXPECT_EQ(1001, uid1Item.mValue.int_value);

    const FieldValue& tag1Item = values[1];
    expectedField = getField(100, {1, 1, 2}, 2, {true, false, true});
    EXPECT_EQ(expectedField, tag1Item.mField);
    EXPECT_EQ(Type::STRING, tag1Item.mValue.getType());
    EXPECT_EQ(tag1, tag1Item.mValue.str_value);

    // Check second attribution nodes
    const FieldValue& uid2Item = values[2];
    expectedField = getField(100, {1, 2, 1}, 2, {true, true, false});
    EXPECT_EQ(expectedField, uid2Item.mField);
    EXPECT_EQ(Type::INT, uid2Item.mValue.getType());
    EXPECT_EQ(1002, uid2Item.mValue.int_value);

    const FieldValue& tag2Item = values[3];
    expectedField = getField(100, {1, 2, 2}, 2, {true, true, true});
    EXPECT_EQ(expectedField, tag2Item.mField);
    EXPECT_EQ(Type::STRING, tag2Item.mValue.getType());
    EXPECT_EQ(tag2, tag2Item.mValue.str_value);

    AStatsEvent_release(event);
}

TEST_P(LogEventTest, TestEmptyAttributionChain) {
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);

    AStatsEvent_writeAttributionChain(event, {}, {}, 0);
    AStatsEvent_writeInt32(event, 10);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_FALSE(ParseBuffer(logEvent, buf, size));

    AStatsEvent_release(event);
}

TEST_P(LogEventTest, TestAttributionChainTooManyElements) {
    int32_t numNodes = 128;
    uint32_t uids[numNodes];
    vector<string> tags(numNodes);  // storage that cTag elements point to
    const char* cTags[numNodes];

    for (int i = 0; i < numNodes; i++) {
        uids[i] = i;
        tags.push_back("test");
        cTags[i] = tags[i].c_str();
    }

    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeAttributionChain(event, uids, cTags, numNodes);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);
    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_FALSE(ParseBuffer(logEvent, buf, size));

    AStatsEvent_release(event);
}

TEST_P_GUARDED(LogEventTest, TestArrayParsing, __ANDROID_API_T__) {
    size_t numElements = 2;
    int32_t int32Array[2] = {3, 6};
    int64_t int64Array[2] = {1000L, 1002L};
    float floatArray[2] = {0.3f, 0.09f};
    bool boolArray[2] = {0, 1};

    vector<string> stringArray = {"str1", "str2"};
    const char* cStringArray[2];
    for (int i = 0; i < numElements; i++) {
        cStringArray[i] = stringArray[i].c_str();
    }

    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeInt32Array(event, int32Array, numElements);
    AStatsEvent_writeInt64Array(event, int64Array, numElements);
    AStatsEvent_writeFloatArray(event, floatArray, numElements);
    AStatsEvent_writeBoolArray(event, boolArray, numElements);
    AStatsEvent_writeStringArray(event, cStringArray, numElements);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());
    EXPECT_FALSE(logEvent.hasAttributionChain());

    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(10, values.size());  // 2 for each array type

    const FieldValue& int32ArrayItem1 = values[0];
    Field expectedField = getField(100, {1, 1, 1}, 1, {false, false, false});
    EXPECT_EQ(expectedField, int32ArrayItem1.mField);
    EXPECT_EQ(Type::INT, int32ArrayItem1.mValue.getType());
    EXPECT_EQ(3, int32ArrayItem1.mValue.int_value);

    const FieldValue& int32ArrayItem2 = values[1];
    expectedField = getField(100, {1, 2, 1}, 1, {false, true, false});
    EXPECT_EQ(expectedField, int32ArrayItem2.mField);
    EXPECT_EQ(Type::INT, int32ArrayItem2.mValue.getType());
    EXPECT_EQ(6, int32ArrayItem2.mValue.int_value);

    const FieldValue& int64ArrayItem1 = values[2];
    expectedField = getField(100, {2, 1, 1}, 1, {false, false, false});
    EXPECT_EQ(expectedField, int64ArrayItem1.mField);
    EXPECT_EQ(Type::LONG, int64ArrayItem1.mValue.getType());
    EXPECT_EQ(1000L, int64ArrayItem1.mValue.long_value);

    const FieldValue& int64ArrayItem2 = values[3];
    expectedField = getField(100, {2, 2, 1}, 1, {false, true, false});
    EXPECT_EQ(expectedField, int64ArrayItem2.mField);
    EXPECT_EQ(Type::LONG, int64ArrayItem2.mValue.getType());
    EXPECT_EQ(1002L, int64ArrayItem2.mValue.long_value);

    const FieldValue& floatArrayItem1 = values[4];
    expectedField = getField(100, {3, 1, 1}, 1, {false, false, false});
    EXPECT_EQ(expectedField, floatArrayItem1.mField);
    EXPECT_EQ(Type::FLOAT, floatArrayItem1.mValue.getType());
    EXPECT_EQ(0.3f, floatArrayItem1.mValue.float_value);

    const FieldValue& floatArrayItem2 = values[5];
    expectedField = getField(100, {3, 2, 1}, 1, {false, true, false});
    EXPECT_EQ(expectedField, floatArrayItem2.mField);
    EXPECT_EQ(Type::FLOAT, floatArrayItem2.mValue.getType());
    EXPECT_EQ(0.09f, floatArrayItem2.mValue.float_value);

    const FieldValue& boolArrayItem1 = values[6];
    expectedField = getField(100, {4, 1, 1}, 1, {false, false, false});
    EXPECT_EQ(expectedField, boolArrayItem1.mField);
    EXPECT_EQ(Type::INT,
              boolArrayItem1.mValue.getType());  // FieldValue does not support boolean type
    EXPECT_EQ(false, boolArrayItem1.mValue.int_value);

    const FieldValue& boolArrayItem2 = values[7];
    expectedField = getField(100, {4, 2, 1}, 1, {false, true, false});
    EXPECT_EQ(expectedField, boolArrayItem2.mField);
    EXPECT_EQ(Type::INT,
              boolArrayItem2.mValue.getType());  // FieldValue does not support boolean type
    EXPECT_EQ(true, boolArrayItem2.mValue.int_value);

    const FieldValue& stringArrayItem1 = values[8];
    expectedField = getField(100, {5, 1, 1}, 1, {true, false, false});
    EXPECT_EQ(expectedField, stringArrayItem1.mField);
    EXPECT_EQ(Type::STRING, stringArrayItem1.mValue.getType());
    EXPECT_EQ("str1", stringArrayItem1.mValue.str_value);

    const FieldValue& stringArrayItem2 = values[9];
    expectedField = getField(100, {5, 2, 1}, 1, {true, true, false});
    EXPECT_EQ(expectedField, stringArrayItem2.mField);
    EXPECT_EQ(Type::STRING, stringArrayItem2.mValue.getType());
    EXPECT_EQ("str2", stringArrayItem2.mValue.str_value);
}

TEST_P_GUARDED(LogEventTest, TestEmptyStringArray, __ANDROID_API_T__) {
    const char* cStringArray[2];
    string empty = "";
    cStringArray[0] = empty.c_str();
    cStringArray[1] = empty.c_str();

    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeStringArray(event, cStringArray, 2);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());

    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(2, values.size());

    const FieldValue& stringArrayItem1 = values[0];
    Field expectedField = getField(100, {1, 1, 1}, 1, {true, false, false});
    EXPECT_EQ(expectedField, stringArrayItem1.mField);
    EXPECT_EQ(Type::STRING, stringArrayItem1.mValue.getType());
    EXPECT_EQ(empty, stringArrayItem1.mValue.str_value);

    const FieldValue& stringArrayItem2 = values[1];
    expectedField = getField(100, {1, 2, 1}, 1, {true, true, false});
    EXPECT_EQ(expectedField, stringArrayItem2.mField);
    EXPECT_EQ(Type::STRING, stringArrayItem2.mValue.getType());
    EXPECT_EQ(empty, stringArrayItem2.mValue.str_value);

    AStatsEvent_release(event);
}

TEST_P_GUARDED(LogEventTest, TestArrayTooManyElements, __ANDROID_API_T__) {
    int32_t numElements = 128;
    int32_t int32Array[numElements];

    for (int i = 0; i < numElements; i++) {
        int32Array[i] = 1;
    }

    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeInt32Array(event, int32Array, numElements);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_FALSE(ParseBuffer(logEvent, buf, size));

    AStatsEvent_release(event);
}

TEST_P_GUARDED(LogEventTest, TestEmptyArray, __ANDROID_API_T__) {
    int32_t int32Array[0] = {};

    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeInt32Array(event, int32Array, 0);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());

    ASSERT_EQ(logEvent.getValues().size(), 0);

    AStatsEvent_release(event);
}

TEST_P_GUARDED(LogEventTest, TestEmptyArrayWithAnnotations, __ANDROID_API_T__) {
    int32_t int32Array[0] = {};

    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeInt32Array(event, int32Array, 0);
    AStatsEvent_addBoolAnnotation(event, ASTATSLOG_ANNOTATION_ID_IS_UID, true);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    EXPECT_EQ(100, logEvent.GetTagId());
    EXPECT_EQ(1000, logEvent.GetUid());
    EXPECT_EQ(1001, logEvent.GetPid());

    ASSERT_EQ(logEvent.getValues().size(), 0);

    AStatsEvent_release(event);
}

TEST_P(LogEventTest, TestAnnotationIdIsUid) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_TRUE(createFieldWithBoolAnnotationLogEvent(&event, INT32_TYPE,
                                                      ASTATSLOG_ANNOTATION_ID_IS_UID, true,
                                                      /*doHeaderPrefetch=*/GetParam()));

    ASSERT_EQ(event.getNumUidFields(), 1);

    const vector<FieldValue>& values = event.getValues();
    ASSERT_EQ(values.size(), 1);
    EXPECT_TRUE(isUidField(values.at(0)));
}

TEST_P_GUARDED(LogEventTest, TestAnnotationIdIsUid_RepeatedIntAndOtherFields, __ANDROID_API_T__) {
    size_t numElements = 2;
    int32_t int32Array[2] = {3, 6};

    vector<string> stringArray = {"str1", "str2"};
    const char* cStringArray[2];
    for (int i = 0; i < numElements; i++) {
        cStringArray[i] = stringArray[i].c_str();
    }

    AStatsEvent* statsEvent = AStatsEvent_obtain();
    AStatsEvent_setAtomId(statsEvent, 100);
    AStatsEvent_writeInt32(statsEvent, 5);
    AStatsEvent_writeInt32Array(statsEvent, int32Array, numElements);
    AStatsEvent_addBoolAnnotation(statsEvent, ASTATSLOG_ANNOTATION_ID_IS_UID, true);
    AStatsEvent_writeStringArray(statsEvent, cStringArray, numElements);
    AStatsEvent_build(statsEvent);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));
    EXPECT_EQ(2, logEvent.getNumUidFields());

    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(values.size(), 5);
    EXPECT_FALSE(isUidField(values.at(0)));
    EXPECT_TRUE(isUidField(values.at(1)));
    EXPECT_TRUE(isUidField(values.at(2)));
    EXPECT_FALSE(isUidField(values.at(3)));
    EXPECT_FALSE(isUidField(values.at(4)));
}

TEST_P_GUARDED(LogEventTest, TestAnnotationIdIsUid_RepeatedIntOneEntry, __ANDROID_API_T__) {
    size_t numElements = 1;
    int32_t int32Array[1] = {3};

    AStatsEvent* statsEvent = AStatsEvent_obtain();
    AStatsEvent_setAtomId(statsEvent, 100);
    AStatsEvent_writeInt32Array(statsEvent, int32Array, numElements);
    AStatsEvent_addBoolAnnotation(statsEvent, ASTATSLOG_ANNOTATION_ID_IS_UID, true);
    AStatsEvent_build(statsEvent);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));
    EXPECT_EQ(1, logEvent.getNumUidFields());

    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(values.size(), 1);
    EXPECT_TRUE(isUidField(values.at(0)));
}

TEST_P_GUARDED(LogEventTest, TestAnnotationIdIsUid_EmptyIntArray, __ANDROID_API_T__) {
    int32_t int32Array[0] = {};

    AStatsEvent* statsEvent = AStatsEvent_obtain();
    AStatsEvent_setAtomId(statsEvent, 100);
    AStatsEvent_writeInt32Array(statsEvent, int32Array, /*numElements*/ 0);
    AStatsEvent_addBoolAnnotation(statsEvent, ASTATSLOG_ANNOTATION_ID_IS_UID, true);
    AStatsEvent_writeInt32(statsEvent, 5);
    AStatsEvent_build(statsEvent);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));
    EXPECT_EQ(0, logEvent.getNumUidFields());

    const vector<FieldValue>& values = logEvent.getValues();
    EXPECT_EQ(values.size(), 1);
}

TEST_P_GUARDED(LogEventTest, TestAnnotationIdIsUid_BadRepeatedInt64, __ANDROID_API_T__) {
    int64_t int64Array[2] = {1000L, 1002L};

    AStatsEvent* statsEvent = AStatsEvent_obtain();
    AStatsEvent_setAtomId(statsEvent, /*atomId=*/100);
    AStatsEvent_writeInt64Array(statsEvent, int64Array, /*numElements*/ 2);
    AStatsEvent_addBoolAnnotation(statsEvent, ASTATSLOG_ANNOTATION_ID_IS_UID, true);
    AStatsEvent_build(statsEvent);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);

    EXPECT_FALSE(ParseBuffer(logEvent, buf, size));
    EXPECT_EQ(0, logEvent.getNumUidFields());

    AStatsEvent_release(statsEvent);
}

TEST_P_GUARDED(LogEventTest, TestAnnotationIdIsUid_BadRepeatedString, __ANDROID_API_T__) {
    size_t numElements = 2;
    vector<string> stringArray = {"str1", "str2"};
    const char* cStringArray[2];
    for (int i = 0; i < numElements; i++) {
        cStringArray[i] = stringArray[i].c_str();
    }

    AStatsEvent* statsEvent = AStatsEvent_obtain();
    AStatsEvent_setAtomId(statsEvent, /*atomId=*/100);
    AStatsEvent_writeStringArray(statsEvent, cStringArray, /*numElements*/ 2);
    AStatsEvent_addBoolAnnotation(statsEvent, ASTATSLOG_ANNOTATION_ID_IS_UID, true);
    AStatsEvent_build(statsEvent);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);

    EXPECT_FALSE(ParseBuffer(logEvent, buf, size));
    EXPECT_EQ(0, logEvent.getNumUidFields());

    AStatsEvent_release(statsEvent);
}

TEST_P(LogEventTestBadAnnotationFieldTypes, TestAnnotationIdIsUid) {
    LogEvent event(/*uid=*/0, /*pid=*/0);

    if (std::get<0>(GetParam()) != INT32_TYPE && std::get<0>(GetParam()) != LIST_TYPE) {
        EXPECT_FALSE(createFieldWithBoolAnnotationLogEvent(
                &event, std::get<0>(GetParam()), ASTATSLOG_ANNOTATION_ID_IS_UID, true,
                /*doHeaderPrefetch=*/std::get<1>(GetParam())));
    }
}

TEST_P(LogEventTest, TestAnnotationIdIsUid_NotIntAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createFieldWithIntAnnotationLogEvent(&event, INT32_TYPE,
                                                      ASTATSLOG_ANNOTATION_ID_IS_UID, 10,
                                                      /*doHeaderPrefetch=*/GetParam()));
}

TEST_P(LogEventTest, TestAnnotationIdStateNested) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_TRUE(createFieldWithBoolAnnotationLogEvent(&event, INT32_TYPE,
                                                      ASTATSLOG_ANNOTATION_ID_STATE_NESTED, true,
                                                      /*doHeaderPrefetch=*/GetParam()));

    const vector<FieldValue>& values = event.getValues();
    ASSERT_EQ(values.size(), 1);
    EXPECT_TRUE(values[0].mAnnotations.isNested());
}

TEST_P(LogEventTestBadAnnotationFieldTypes, TestAnnotationIdStateNested) {
    LogEvent event(/*uid=*/0, /*pid=*/0);

    if (std::get<0>(GetParam()) != INT32_TYPE &&
        (std::get<0>(GetParam()) != LIST_TYPE || isAtLeastT())) {
        EXPECT_FALSE(createFieldWithBoolAnnotationLogEvent(
                &event, std::get<0>(GetParam()), ASTATSLOG_ANNOTATION_ID_STATE_NESTED, true,
                /*doHeaderPrefetch=*/std::get<1>(GetParam())));
    }
}

TEST_P(LogEventTest, TestAnnotationIdStateNested_NotIntAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createFieldWithIntAnnotationLogEvent(&event, INT32_TYPE,
                                                      ASTATSLOG_ANNOTATION_ID_STATE_NESTED, 10,
                                                      /*doHeaderPrefetch=*/GetParam()));
}

TEST_P(LogEventTest, TestPrimaryFieldAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_TRUE(createFieldWithBoolAnnotationLogEvent(&event, INT32_TYPE,
                                                      ASTATSLOG_ANNOTATION_ID_PRIMARY_FIELD, true,
                                                      /*doHeaderPrefetch=*/GetParam()));

    const vector<FieldValue>& values = event.getValues();
    ASSERT_EQ(values.size(), 1);
    EXPECT_TRUE(values[0].mAnnotations.isPrimaryField());
}

TEST_P(LogEventTestBadAnnotationFieldTypes, TestPrimaryFieldAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);

    if ((std::get<0>(GetParam()) == LIST_TYPE && isAtLeastT()) ||
        std::get<0>(GetParam()) == ATTRIBUTION_CHAIN_TYPE) {
        EXPECT_FALSE(createFieldWithBoolAnnotationLogEvent(
                &event, std::get<0>(GetParam()), ASTATSLOG_ANNOTATION_ID_PRIMARY_FIELD, true,
                /*doHeaderPrefetch=*/std::get<1>(GetParam())));
    }
}

TEST_P(LogEventTest, TestPrimaryFieldAnnotation_NotIntAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createFieldWithIntAnnotationLogEvent(&event, INT32_TYPE,
                                                      ASTATSLOG_ANNOTATION_ID_PRIMARY_FIELD, 10,
                                                      /*doHeaderPrefetch=*/GetParam()));
}

TEST_P(LogEventTest, TestExclusiveStateAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_TRUE(createFieldWithBoolAnnotationLogEvent(&event, INT32_TYPE,
                                                      ASTATSLOG_ANNOTATION_ID_EXCLUSIVE_STATE, true,
                                                      /*doHeaderPrefetch=*/GetParam()));

    const vector<FieldValue>& values = event.getValues();
    ASSERT_EQ(values.size(), 1);
    EXPECT_TRUE(values[0].mAnnotations.isExclusiveState());
}

TEST_P(LogEventTestBadAnnotationFieldTypes, TestExclusiveStateAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);

    if (std::get<0>(GetParam()) != INT32_TYPE &&
        (std::get<0>(GetParam()) != LIST_TYPE || isAtLeastT())) {
        EXPECT_FALSE(createFieldWithBoolAnnotationLogEvent(
                &event, std::get<0>(GetParam()), ASTATSLOG_ANNOTATION_ID_EXCLUSIVE_STATE, true,
                /*doHeaderPrefetch=*/std::get<1>(GetParam())));
    }
}

TEST_P(LogEventTest, TestExclusiveStateAnnotation_NotIntAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createFieldWithIntAnnotationLogEvent(&event, INT32_TYPE,
                                                      ASTATSLOG_ANNOTATION_ID_EXCLUSIVE_STATE, 10,
                                                      /*doHeaderPrefetch=*/GetParam()));
}

TEST_P(LogEventTest, TestPrimaryFieldFirstUidAnnotation) {
    // Event has 10 ints and then an attribution chain
    int numInts = 10;
    int firstUidInChainIndex = numInts;
    string tag1 = "tag1";
    string tag2 = "tag2";
    uint32_t uids[] = {1001, 1002};
    const char* tags[] = {tag1.c_str(), tag2.c_str()};

    // Construct AStatsEvent
    AStatsEvent* statsEvent = AStatsEvent_obtain();
    AStatsEvent_setAtomId(statsEvent, 100);
    for (int i = 0; i < numInts; i++) {
        AStatsEvent_writeInt32(statsEvent, 10);
    }
    AStatsEvent_writeAttributionChain(statsEvent, uids, tags, 2);
    AStatsEvent_addBoolAnnotation(statsEvent, ASTATSLOG_ANNOTATION_ID_PRIMARY_FIELD_FIRST_UID,
                                  true);
    AStatsEvent_build(statsEvent);

    // Construct LogEvent
    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(statsEvent, &size);
    LogEvent logEvent(/*uid=*/0, /*pid=*/0);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));
    AStatsEvent_release(statsEvent);

    // Check annotation
    const vector<FieldValue>& values = logEvent.getValues();
    ASSERT_EQ(values.size(), numInts + 4);
    EXPECT_TRUE(values[firstUidInChainIndex].mAnnotations.isPrimaryField());
}

TEST_P(LogEventTestBadAnnotationFieldTypes, TestPrimaryFieldFirstUidAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);

    if (std::get<0>(GetParam()) != ATTRIBUTION_CHAIN_TYPE &&
        (std::get<0>(GetParam()) != LIST_TYPE || isAtLeastT())) {
        EXPECT_FALSE(createFieldWithBoolAnnotationLogEvent(
                &event, std::get<0>(GetParam()), ASTATSLOG_ANNOTATION_ID_PRIMARY_FIELD_FIRST_UID,
                true,
                /*doHeaderPrefetch=*/std::get<1>(GetParam())));
    }
}

TEST_P(LogEventTest, TestPrimaryFieldFirstUidAnnotation_NotIntAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createFieldWithIntAnnotationLogEvent(
            &event, ATTRIBUTION_CHAIN_TYPE, ASTATSLOG_ANNOTATION_ID_PRIMARY_FIELD_FIRST_UID, 10,
            /*doHeaderPrefetch=*/GetParam()));
}

TEST_P(LogEventTest, TestResetStateAnnotation) {
    int32_t resetState = 10;
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_TRUE(createFieldWithIntAnnotationLogEvent(
            &event, INT32_TYPE, ASTATSLOG_ANNOTATION_ID_TRIGGER_STATE_RESET, resetState,
            /*doHeaderPrefetch=*/GetParam()));

    const vector<FieldValue>& values = event.getValues();
    ASSERT_EQ(values.size(), 1);
    EXPECT_EQ(event.getResetState(), resetState);
}

TEST_P(LogEventTest, TestRestrictionCategoryAnnotation) {
    if (!isAtLeastU()) {
        GTEST_SKIP();
    }
    int32_t restrictionCategory = ASTATSLOG_RESTRICTION_CATEGORY_DIAGNOSTIC;
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_TRUE(createAtomLevelIntAnnotationLogEvent(
            &event, INT32_TYPE, ASTATSLOG_ANNOTATION_ID_RESTRICTION_CATEGORY, restrictionCategory,
            /*doHeaderPrefetch=*/GetParam()));

    ASSERT_EQ(event.getRestrictionCategory(), restrictionCategory);
}

TEST_P(LogEventTest, TestInvalidRestrictionCategoryAnnotation) {
    if (!isAtLeastU()) {
        GTEST_SKIP();
    }
    int32_t restrictionCategory = 619;  // unknown category
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createAtomLevelIntAnnotationLogEvent(
            &event, INT32_TYPE, ASTATSLOG_ANNOTATION_ID_RESTRICTION_CATEGORY, restrictionCategory,
            /*doHeaderPrefetch=*/GetParam()));
}

TEST_P(LogEventTest, TestRestrictionCategoryAnnotationBelowUDevice) {
    if (isAtLeastU()) {
        GTEST_SKIP();
    }
    int32_t restrictionCategory = ASTATSLOG_RESTRICTION_CATEGORY_DIAGNOSTIC;
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createAtomLevelIntAnnotationLogEvent(
            &event, INT32_TYPE, ASTATSLOG_ANNOTATION_ID_RESTRICTION_CATEGORY, restrictionCategory,
            /*doHeaderPrefetch=*/GetParam()));
}

TEST_P(LogEventTestBadAnnotationFieldTypes, TestResetStateAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    int32_t resetState = 10;

    if (std::get<0>(GetParam()) != INT32_TYPE &&
        (std::get<0>(GetParam()) != LIST_TYPE || isAtLeastT())) {
        EXPECT_FALSE(createFieldWithIntAnnotationLogEvent(
                &event, std::get<0>(GetParam()), ASTATSLOG_ANNOTATION_ID_TRIGGER_STATE_RESET,
                resetState,
                /*doHeaderPrefetch=*/std::get<1>(GetParam())));
    }
}

TEST_P(LogEventTest, TestResetStateAnnotation_NotBoolAnnotation) {
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createFieldWithBoolAnnotationLogEvent(
            &event, INT32_TYPE, ASTATSLOG_ANNOTATION_ID_TRIGGER_STATE_RESET, true,
            /*doHeaderPrefetch=*/GetParam()));
}

TEST_P_GUARDED(LogEventTest, TestUidAnnotationWithInt8MaxValues, __ANDROID_API_T__) {
    int32_t numElements = INT8_MAX;
    int32_t int32Array[numElements];

    for (int i = 0; i < numElements; i++) {
        int32Array[i] = i;
    }

    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);
    AStatsEvent_writeInt32Array(event, int32Array, numElements);
    AStatsEvent_writeInt32(event, 10);
    AStatsEvent_writeInt32(event, 11);
    AStatsEvent_addBoolAnnotation(event, ASTATSLOG_ANNOTATION_ID_IS_UID, true);
    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);
    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_TRUE(ParseBuffer(logEvent, buf, size));

    AStatsEvent_release(event);
}

TEST_P(LogEventTest, TestEmptyAttributionChainWithPrimaryFieldFirstUidAnnotation) {
    AStatsEvent* event = AStatsEvent_obtain();
    AStatsEvent_setAtomId(event, 100);

    uint32_t uids[] = {};
    const char* tags[] = {};

    AStatsEvent_writeInt32(event, 10);
    AStatsEvent_writeAttributionChain(event, uids, tags, 0);
    AStatsEvent_addBoolAnnotation(event, ASTATSLOG_ANNOTATION_ID_PRIMARY_FIELD_FIRST_UID, true);

    AStatsEvent_build(event);

    size_t size;
    const uint8_t* buf = AStatsEvent_getBuffer(event, &size);

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_FALSE(ParseBuffer(logEvent, buf, size));

    AStatsEvent_release(event);
}

TEST_P_GUARDED(LogEventTest, TestInvalidBufferParsing, __ANDROID_API_T__) {
    size_t emptyAtomBufferSize = 0;
    {
        // creating valid event to get valid buffer header size when no data fields
        AStatsEvent* event = AStatsEvent_obtain();
        AStatsEvent_setAtomId(event, 100);
        AStatsEvent_build(event);
        AStatsEvent_getBuffer(event, &emptyAtomBufferSize);
        AStatsEvent_release(event);
    }

    uint8_t buffer[4096];
    memset(buffer, 0, 4096);
    {
        // creating valid event to get valid buffer header with 1 array field and 1 annotation
        AStatsEvent* event = AStatsEvent_obtain();
        AStatsEvent_setAtomId(event, 100);
        AStatsEvent_writeInt32Array(event, nullptr, 0);
        AStatsEvent_addBoolAnnotation(event, ASTATSLOG_ANNOTATION_ID_IS_UID, true);
        EXPECT_EQ(AStatsEvent_getErrors(event), 0);
        AStatsEvent_build(event);

        const uint8_t* buf = AStatsEvent_getBuffer(event, nullptr);
        memcpy(buffer, buf, emptyAtomBufferSize);
        AStatsEvent_release(event);
    }

    size_t bufferPosWithAlteredData = emptyAtomBufferSize;

    // adding extra data to test the logEvent parser logic

    buffer[bufferPosWithAlteredData++] = 0x13;  // array with 1 annotation
    buffer[bufferPosWithAlteredData++] = 0x00;  // size of array is 0
    buffer[bufferPosWithAlteredData++] = 0x00;  // type of array is int32
    buffer[bufferPosWithAlteredData++] = 0x01;  // annotation type is isUid
    buffer[bufferPosWithAlteredData++] = 0x01;  // annotation value type is not bool

    LogEvent logEvent(/*uid=*/1000, /*pid=*/1001);
    EXPECT_FALSE(ParseBuffer(logEvent, buffer, bufferPosWithAlteredData));
}

// Setup for parameterized tests.
class LogEvent_FieldRestrictionTest : public testing::TestWithParam<std::tuple<int, bool>> {
public:
    static std::string ToString(testing::TestParamInfo<std::tuple<int, bool>> info) {
        const std::string boolName = std::get<1>(info.param) ? "_prefetchTrue" : "_prefetchFalse";

        switch (std::get<0>(info.param)) {
            case ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_PERIPHERAL_DEVICE_INFO:
                return "PeripheralDeviceInfo" + boolName;
            case ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_APP_USAGE:
                return "AppUsage" + boolName;
            case ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_APP_ACTIVITY:
                return "AppActivity" + boolName;
            case ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_HEALTH_CONNECT:
                return "HealthConnect" + boolName;
            case ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_ACCESSIBILITY:
                return "Accessibility" + boolName;
            case ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_SYSTEM_SEARCH:
                return "SystemSearch" + boolName;
            case ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_USER_ENGAGEMENT:
                return "UserEngagement" + boolName;
            case ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_AMBIENT_SENSING:
                return "AmbientSensing" + boolName;
            case ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_DEMOGRAPHIC_CLASSIFICATION:
                return "DemographicClassification" + boolName;
            default:
                return "Unknown" + boolName;
        }
    }
    void TearDown() override {
        FlagProvider::getInstance().resetOverrides();
    }
};

// TODO(b/222539899): Add BOOL_TYPE value once parseAnnotations is updated to check specific
// typeIds. BOOL_TYPE should be a bad field type for is_uid, nested, and reset state annotations.
INSTANTIATE_TEST_SUITE_P(
        LogEvent_FieldRestrictionTest, LogEvent_FieldRestrictionTest,
        testing::Combine(
                testing::Values(
                        ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_PERIPHERAL_DEVICE_INFO,
                        ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_APP_USAGE,
                        ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_APP_ACTIVITY,
                        ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_HEALTH_CONNECT,
                        ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_ACCESSIBILITY,
                        ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_SYSTEM_SEARCH,
                        ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_USER_ENGAGEMENT,
                        ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_AMBIENT_SENSING,
                        ASTATSLOG_ANNOTATION_ID_FIELD_RESTRICTION_DEMOGRAPHIC_CLASSIFICATION),
                testing::Bool()),
        LogEvent_FieldRestrictionTest::ToString);

TEST_P(LogEvent_FieldRestrictionTest, TestFieldRestrictionAnnotation) {
    if (!isAtLeastU()) {
        GTEST_SKIP();
    }
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_TRUE(
            createFieldWithBoolAnnotationLogEvent(&event, INT32_TYPE, std::get<0>(GetParam()), true,
                                                  /*doHeaderPrefetch=*/std::get<1>(GetParam())));
    // Some basic checks to make sure the event is parsed correctly.
    EXPECT_EQ(event.GetTagId(), 100);
    ASSERT_EQ(event.getValues().size(), 1);
    EXPECT_EQ(event.getValues()[0].mValue.getType(), Type::INT);
}

TEST_P(LogEvent_FieldRestrictionTest, TestInvalidAnnotationIntType) {
    if (!isAtLeastU()) {
        GTEST_SKIP();
    }
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createFieldWithIntAnnotationLogEvent(
            &event, STRING_TYPE, std::get<0>(GetParam()),
            /*random int*/ 15, /*doHeaderPrefetch=*/std::get<1>(GetParam())));
}

TEST_P(LogEvent_FieldRestrictionTest, TestInvalidAnnotationAtomLevel) {
    if (!isAtLeastU()) {
        GTEST_SKIP();
    }
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(createAtomLevelBoolAnnotationLogEvent(
            &event, STRING_TYPE, std::get<0>(GetParam()), true,
            /*doHeaderPrefetch=*/std::get<1>(GetParam())));
}

TEST_P(LogEvent_FieldRestrictionTest, TestRestrictionCategoryAnnotationBelowUDevice) {
    if (isAtLeastU()) {
        GTEST_SKIP();
    }
    int32_t restrictionCategory = ASTATSLOG_RESTRICTION_CATEGORY_DIAGNOSTIC;
    LogEvent event(/*uid=*/0, /*pid=*/0);
    EXPECT_FALSE(
            createFieldWithBoolAnnotationLogEvent(&event, INT32_TYPE, std::get<0>(GetParam()), true,
                                                  /*doHeaderPrefetch=*/std::get<1>(GetParam())));
}

}  // namespace statsd
}  // namespace os
}  // namespace android
#else
GTEST_LOG_(INFO) << "This test does nothing.\n";
#endif