blob: 60478c9aa081526470c5c564ac7d9b978f3b9cc4 [file] [log] [blame]
/*
* Copyright (C) 2009 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.
*/
//#define LOG_NDEBUG 0
#define LOG_TAG "MetaDataBase"
#include <inttypes.h>
#include <utils/KeyedVector.h>
#include <utils/Log.h>
#include <stdlib.h>
#include <string.h>
#include <mutex>
#include <media/stagefright/foundation/ADebug.h>
#include <media/stagefright/foundation/AString.h>
#include <media/stagefright/foundation/hexdump.h>
#include <media/stagefright/MetaDataBase.h>
#if defined(__ANDROID__) && !defined(__ANDROID_VNDK__) && !defined(__ANDROID_APEX__)
#include <binder/Parcel.h>
#endif
namespace android {
struct MetaDataBase::typed_data {
typed_data();
~typed_data();
typed_data(const MetaDataBase::typed_data &);
typed_data &operator=(const MetaDataBase::typed_data &);
void clear();
void setData(uint32_t type, const void *data, size_t size);
void getData(uint32_t *type, const void **data, size_t *size) const;
// may include hexdump of binary data if verbose=true
String8 asString(bool verbose) const;
private:
uint32_t mType;
size_t mSize;
union {
void *ext_data;
float reservoir;
} u;
bool usesReservoir() const {
return mSize <= sizeof(u.reservoir);
}
void *allocateStorage(size_t size);
void freeStorage();
void *storage() {
return usesReservoir() ? &u.reservoir : u.ext_data;
}
const void *storage() const {
return usesReservoir() ? &u.reservoir : u.ext_data;
}
};
struct MetaDataBase::Rect {
int32_t mLeft, mTop, mRight, mBottom;
};
struct MetaDataBase::MetaDataInternal {
std::mutex mLock;
KeyedVector<uint32_t, MetaDataBase::typed_data> mItems;
};
MetaDataBase::MetaDataBase()
: mInternalData(new MetaDataInternal()) {
}
MetaDataBase::MetaDataBase(const MetaDataBase &from)
: mInternalData(new MetaDataInternal()) {
mInternalData->mItems = from.mInternalData->mItems;
}
MetaDataBase& MetaDataBase::operator = (const MetaDataBase &rhs) {
this->mInternalData->mItems = rhs.mInternalData->mItems;
return *this;
}
MetaDataBase::~MetaDataBase() {
clear();
delete mInternalData;
}
void MetaDataBase::clear() {
std::lock_guard<std::mutex> guard(mInternalData->mLock);
mInternalData->mItems.clear();
}
bool MetaDataBase::remove(uint32_t key) {
std::lock_guard<std::mutex> guard(mInternalData->mLock);
ssize_t i = mInternalData->mItems.indexOfKey(key);
if (i < 0) {
return false;
}
mInternalData->mItems.removeItemsAt(i);
return true;
}
bool MetaDataBase::setCString(uint32_t key, const char *value) {
return setData(key, TYPE_C_STRING, value, strlen(value) + 1);
}
bool MetaDataBase::setInt32(uint32_t key, int32_t value) {
return setData(key, TYPE_INT32, &value, sizeof(value));
}
bool MetaDataBase::setInt64(uint32_t key, int64_t value) {
return setData(key, TYPE_INT64, &value, sizeof(value));
}
bool MetaDataBase::setFloat(uint32_t key, float value) {
return setData(key, TYPE_FLOAT, &value, sizeof(value));
}
bool MetaDataBase::setPointer(uint32_t key, void *value) {
return setData(key, TYPE_POINTER, &value, sizeof(value));
}
bool MetaDataBase::setRect(
uint32_t key,
int32_t left, int32_t top,
int32_t right, int32_t bottom) {
Rect r;
r.mLeft = left;
r.mTop = top;
r.mRight = right;
r.mBottom = bottom;
return setData(key, TYPE_RECT, &r, sizeof(r));
}
/**
* Note that the returned pointer becomes invalid when additional metadata is set.
*/
bool MetaDataBase::findCString(uint32_t key, const char **value) const {
uint32_t type;
const void *data;
size_t size;
if (!findData(key, &type, &data, &size) || type != TYPE_C_STRING) {
return false;
}
*value = (const char *)data;
return true;
}
bool MetaDataBase::findInt32(uint32_t key, int32_t *value) const {
uint32_t type = 0;
const void *data;
size_t size;
if (!findData(key, &type, &data, &size) || type != TYPE_INT32) {
return false;
}
CHECK_EQ(size, sizeof(*value));
*value = *(int32_t *)data;
return true;
}
bool MetaDataBase::findInt64(uint32_t key, int64_t *value) const {
uint32_t type = 0;
const void *data;
size_t size;
if (!findData(key, &type, &data, &size) || type != TYPE_INT64) {
return false;
}
CHECK_EQ(size, sizeof(*value));
*value = *(int64_t *)data;
return true;
}
bool MetaDataBase::findFloat(uint32_t key, float *value) const {
uint32_t type = 0;
const void *data;
size_t size;
if (!findData(key, &type, &data, &size) || type != TYPE_FLOAT) {
return false;
}
CHECK_EQ(size, sizeof(*value));
*value = *(float *)data;
return true;
}
bool MetaDataBase::findPointer(uint32_t key, void **value) const {
uint32_t type = 0;
const void *data;
size_t size;
if (!findData(key, &type, &data, &size) || type != TYPE_POINTER) {
return false;
}
CHECK_EQ(size, sizeof(*value));
*value = *(void **)data;
return true;
}
bool MetaDataBase::findRect(
uint32_t key,
int32_t *left, int32_t *top,
int32_t *right, int32_t *bottom) const {
uint32_t type = 0;
const void *data;
size_t size;
if (!findData(key, &type, &data, &size) || type != TYPE_RECT) {
return false;
}
CHECK_EQ(size, sizeof(Rect));
const Rect *r = (const Rect *)data;
*left = r->mLeft;
*top = r->mTop;
*right = r->mRight;
*bottom = r->mBottom;
return true;
}
bool MetaDataBase::setData(
uint32_t key, uint32_t type, const void *data, size_t size) {
bool overwrote_existing = true;
std::lock_guard<std::mutex> guard(mInternalData->mLock);
ssize_t i = mInternalData->mItems.indexOfKey(key);
if (i < 0) {
typed_data item;
i = mInternalData->mItems.add(key, item);
overwrote_existing = false;
}
typed_data &item = mInternalData->mItems.editValueAt(i);
item.setData(type, data, size);
return overwrote_existing;
}
bool MetaDataBase::findData(uint32_t key, uint32_t *type,
const void **data, size_t *size) const {
std::lock_guard<std::mutex> guard(mInternalData->mLock);
ssize_t i = mInternalData->mItems.indexOfKey(key);
if (i < 0) {
return false;
}
const typed_data &item = mInternalData->mItems.valueAt(i);
item.getData(type, data, size);
return true;
}
bool MetaDataBase::hasData(uint32_t key) const {
std::lock_guard<std::mutex> guard(mInternalData->mLock);
ssize_t i = mInternalData->mItems.indexOfKey(key);
if (i < 0) {
return false;
}
return true;
}
MetaDataBase::typed_data::typed_data()
: mType(0),
mSize(0) {
}
MetaDataBase::typed_data::~typed_data() {
clear();
}
MetaDataBase::typed_data::typed_data(const typed_data &from)
: mType(from.mType),
mSize(0) {
void *dst = allocateStorage(from.mSize);
if (dst) {
memcpy(dst, from.storage(), mSize);
}
}
MetaDataBase::typed_data &MetaDataBase::typed_data::operator=(
const MetaDataBase::typed_data &from) {
if (this != &from) {
clear();
mType = from.mType;
void *dst = allocateStorage(from.mSize);
if (dst) {
memcpy(dst, from.storage(), mSize);
}
}
return *this;
}
void MetaDataBase::typed_data::clear() {
freeStorage();
mType = 0;
}
void MetaDataBase::typed_data::setData(
uint32_t type, const void *data, size_t size) {
clear();
mType = type;
void *dst = allocateStorage(size);
if (dst) {
memcpy(dst, data, size);
}
}
void MetaDataBase::typed_data::getData(
uint32_t *type, const void **data, size_t *size) const {
*type = mType;
*size = mSize;
*data = storage();
}
void *MetaDataBase::typed_data::allocateStorage(size_t size) {
mSize = size;
if (usesReservoir()) {
return &u.reservoir;
}
u.ext_data = malloc(mSize);
if (u.ext_data == NULL) {
ALOGE("Couldn't allocate %zu bytes for item", size);
mSize = 0;
}
return u.ext_data;
}
void MetaDataBase::typed_data::freeStorage() {
if (!usesReservoir()) {
if (u.ext_data) {
free(u.ext_data);
u.ext_data = NULL;
}
}
mSize = 0;
}
String8 MetaDataBase::typed_data::asString(bool verbose) const {
String8 out;
const void *data = storage();
switch(mType) {
case TYPE_NONE:
out = String8::format("no type, size %zu)", mSize);
break;
case TYPE_C_STRING:
out = String8::format("(char*) %s", (const char *)data);
break;
case TYPE_INT32:
out = String8::format("(int32_t) %d", *(int32_t *)data);
break;
case TYPE_INT64:
out = String8::format("(int64_t) %" PRId64, *(int64_t *)data);
break;
case TYPE_FLOAT:
out = String8::format("(float) %f", *(float *)data);
break;
case TYPE_POINTER:
out = String8::format("(void*) %p", *(void **)data);
break;
case TYPE_RECT:
{
const Rect *r = (const Rect *)data;
out = String8::format("Rect(%d, %d, %d, %d)",
r->mLeft, r->mTop, r->mRight, r->mBottom);
break;
}
default:
out = String8::format("(unknown type %d, size %zu)", mType, mSize);
if (verbose && mSize <= 48) { // if it's less than three lines of hex data, dump it
AString foo;
hexdump(data, mSize, 0, &foo);
out.append("\n");
out.append(foo.c_str());
}
break;
}
return out;
}
static void MakeFourCCString(uint32_t x, char *s) {
s[0] = x >> 24;
s[1] = (x >> 16) & 0xff;
s[2] = (x >> 8) & 0xff;
s[3] = x & 0xff;
s[4] = '\0';
}
String8 MetaDataBase::toString() const {
String8 s;
std::lock_guard<std::mutex> guard(mInternalData->mLock);
for (int i = mInternalData->mItems.size(); --i >= 0;) {
int32_t key = mInternalData->mItems.keyAt(i);
char cc[5];
MakeFourCCString(key, cc);
const typed_data &item = mInternalData->mItems.valueAt(i);
s.appendFormat("%s: %s", cc, item.asString(false).c_str());
if (i != 0) {
s.append(", ");
}
}
return s;
}
void MetaDataBase::dumpToLog() const {
std::lock_guard<std::mutex> guard(mInternalData->mLock);
for (int i = mInternalData->mItems.size(); --i >= 0;) {
int32_t key = mInternalData->mItems.keyAt(i);
char cc[5];
MakeFourCCString(key, cc);
const typed_data &item = mInternalData->mItems.valueAt(i);
ALOGI("%s: %s", cc, item.asString(true /* verbose */).c_str());
}
}
#if defined(__ANDROID__) && !defined(__ANDROID_VNDK__) && !defined(__ANDROID_APEX__)
status_t MetaDataBase::writeToParcel(Parcel &parcel) {
status_t ret;
std::lock_guard<std::mutex> guard(mInternalData->mLock);
size_t numItems = mInternalData->mItems.size();
ret = parcel.writeUint32(uint32_t(numItems));
if (ret) {
return ret;
}
for (size_t i = 0; i < numItems; i++) {
int32_t key = mInternalData->mItems.keyAt(i);
const typed_data &item = mInternalData->mItems.valueAt(i);
uint32_t type;
const void *data;
size_t size;
item.getData(&type, &data, &size);
ret = parcel.writeInt32(key);
if (ret) {
return ret;
}
ret = parcel.writeUint32(type);
if (ret) {
return ret;
}
if (type == TYPE_NONE) {
android::Parcel::WritableBlob blob;
ret = parcel.writeUint32(static_cast<uint32_t>(size));
if (ret) {
return ret;
}
ret = parcel.writeBlob(size, false, &blob);
if (ret) {
return ret;
}
memcpy(blob.data(), data, size);
blob.release();
} else {
ret = parcel.writeByteArray(size, (uint8_t*)data);
if (ret) {
return ret;
}
}
}
return OK;
}
status_t MetaDataBase::updateFromParcel(const Parcel &parcel) {
uint32_t numItems;
if (parcel.readUint32(&numItems) == OK) {
for (size_t i = 0; i < numItems; i++) {
int32_t key;
uint32_t type;
uint32_t size;
status_t ret = parcel.readInt32(&key);
ret |= parcel.readUint32(&type);
ret |= parcel.readUint32(&size);
if (ret != OK) {
break;
}
// copy data from Blob, which may be inline in Parcel storage,
// then advance position
if (type == TYPE_NONE) {
android::Parcel::ReadableBlob blob;
ret = parcel.readBlob(size, &blob);
if (ret != OK) {
break;
}
setData(key, type, blob.data(), size);
blob.release();
} else if (type == TYPE_C_STRING) {
// copy data directly from Parcel storage, then advance position
// NB: readInplace() bumps position, it is NOT idempotent.
const void *src = parcel.readInplace(size);
char *str = (char *) src;
if (src == nullptr || size == 0 || str[size-1] != '\0') {
char ccKey[5];
MakeFourCCString(key, ccKey);
if (src == nullptr) {
ALOGW("ignoring key '%s' string with no data (expected %d)", ccKey, size);
} else {
ALOGW("ignoring key '%s': unterminated string of %d bytes", ccKey, size);
}
} else {
setData(key, type, src, size);
}
} else {
// copy data directly from Parcel storage, then advance position
// verify that the received size is enough
uint32_t needed = 0;
switch (type) {
case TYPE_INT32:
needed = sizeof(int32_t);
break;
case TYPE_INT64:
needed = sizeof(int64_t);
break;
case TYPE_FLOAT:
needed = sizeof(float);
break;
case TYPE_POINTER:
// NB: this rejects passing between 32-bit and 64-bit space.
needed = sizeof(void*);
break;
case TYPE_RECT:
needed = sizeof(Rect);
break;
default:
// non-standard entities can be any size >= 0
needed = 0;
break;
}
const void *src = parcel.readInplace(size);
if (src == nullptr || (needed != 0 && size != needed)) {
char ccKey[5];
MakeFourCCString(key, ccKey);
char ccType[5];
MakeFourCCString(type, ccType);
if (src == nullptr) {
ALOGW("ignoring key '%s' type '%s' missing data (expected %d)",
ccKey, ccType, size);
} else {
ALOGW("ignoring key '%s': type '%s' bytes: expected %d != %d received",
ccKey, ccType, needed, size);
}
} else {
setData(key, type, src, size);
}
}
}
return OK;
}
ALOGW("no metadata in parcel");
return UNKNOWN_ERROR;
}
#endif // defined(__ANDROID__) && !defined(__ANDROID_VNDK__) && !defined(__ANDROID_APEX__)
} // namespace android