distrib/android-emugl/shared/OpenglCodecCommon/ChecksumCalculatorThreadInfo.cpp

/*
* Copyright (C) 2016 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 "ChecksumCalculatorThreadInfo.h"

#include "emugl/common/lazy_instance.h"
#include "emugl/common/thread_store.h"

#include <stdio.h>
#include <atomic>
#include <string>

namespace {

class ChecksumCalculatorThreadStore : public ::emugl::ThreadStore {
public:
    ChecksumCalculatorThreadStore() : ::emugl::ThreadStore(&destructor) {}
#ifdef TRACE_CHECKSUMHELPER
    static std::atomic<size_t> mNumInstances;
#endif  // TRACE_CHECKSUMHELPER

private:
    static void destructor(void* value) {
        LOG_CHECKSUMHELPER("%s: GLprotocol deleted %p (%lu instances)\n",
                           __FUNCTION__, value, (size_t)mNumInstances - 1);
        delete static_cast<ChecksumCalculatorThreadInfo*>(value);
#ifdef TRACE_CHECKSUMHELPER
        mNumInstances--;
#endif  // TRACE_CHECKSUMHELPER
    }
};

static ::emugl::LazyInstance<ChecksumCalculatorThreadStore> s_tls =
        LAZY_INSTANCE_INIT;

ChecksumCalculatorThreadInfo* getChecksumCalculatorThreadInfo() {
    ChecksumCalculatorThreadInfo* pti =
            static_cast<ChecksumCalculatorThreadInfo*>(s_tls->get());
    if (!pti) {
        pti = new ChecksumCalculatorThreadInfo();
#ifdef TRACE_CHECKSUMHELPER
        ChecksumCalculatorThreadStore::mNumInstances++;
#endif  // TRACE_CHECKSUMHELPER
        s_tls->set(pti);
        LOG_CHECKSUMHELPER(
                "%s: GLprotocol created %p (%lu instances)\n", __FUNCTION__,
                pti, (size_t)ChecksumCalculatorThreadStore::mNumInstances);
    }
    return pti;
}

}  // namespace

#ifdef TRACE_CHECKSUMHELPER
std::atomic<size_t> ChecksumCalculatorThreadStore::mNumInstances(0);
#endif  // TRACE_CHECKSUMHELPER

uint32_t ChecksumCalculatorThreadInfo::getVersion() {
    return getChecksumCalculatorThreadInfo()->m_protocol.getVersion();
}

bool ChecksumCalculatorThreadInfo::setVersion(uint32_t version) {
    return getChecksumCalculatorThreadInfo()->m_protocol.setVersion(version);
}

size_t ChecksumCalculatorThreadInfo::checksumByteSize() {
    return getChecksumCalculatorThreadInfo()->m_protocol.checksumByteSize();
}

bool ChecksumCalculatorThreadInfo::writeChecksum(void* buf,
                                                 size_t bufLen,
                                                 void* outputChecksum,
                                                 size_t outputChecksumLen) {
    ChecksumCalculator& protocol =
            getChecksumCalculatorThreadInfo()->m_protocol;
    protocol.addBuffer(buf, bufLen);
    return protocol.writeChecksum(outputChecksum, outputChecksumLen);
}

bool ChecksumCalculatorThreadInfo::validate(void* buf,
                                            size_t bufLen,
                                            void* checksum,
                                            size_t checksumLen) {
    ChecksumCalculator& protocol =
            getChecksumCalculatorThreadInfo()->m_protocol;
    protocol.addBuffer(buf, bufLen);
    return protocol.validate(checksum, checksumLen);
}

void ChecksumCalculatorThreadInfo::validOrDie(void* buf,
                                              size_t bufLen,
                                              void* checksum,
                                              size_t checksumLen,
                                              const char* message) {
    // We should actually call crashhandler_die(message), but I don't think we
    // can link to that library from here
    if (!validate(buf, bufLen, checksum, checksumLen)) {
        abort();
    }
}