| /* |
| * 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. |
| */ |
| |
| #define LOG_TAG "AHardwareBuffer" |
| |
| #include <vndk/hardware_buffer.h> |
| |
| #include <errno.h> |
| #include <sys/socket.h> |
| #include <memory> |
| |
| #include <cutils/native_handle.h> |
| #include <log/log.h> |
| #include <utils/StrongPointer.h> |
| #include <ui/GraphicBuffer.h> |
| #include <system/graphics.h> |
| |
| #include <private/android/AHardwareBufferHelpers.h> |
| #include <android/hardware/graphics/common/1.0/types.h> |
| |
| |
| static constexpr int kFdBufferSize = 128 * sizeof(int); // 128 ints |
| |
| using namespace android; |
| |
| // ---------------------------------------------------------------------------- |
| // Public functions |
| // ---------------------------------------------------------------------------- |
| |
| int AHardwareBuffer_allocate(const AHardwareBuffer_Desc* desc, AHardwareBuffer** outBuffer) { |
| if (!outBuffer || !desc) |
| return BAD_VALUE; |
| |
| int format = AHardwareBuffer_convertToPixelFormat(desc->format); |
| if (format == 0) { |
| ALOGE("Invalid pixel format %u", desc->format); |
| return BAD_VALUE; |
| } |
| |
| if (desc->reserved != 0) { |
| ALOGE("AHardwareBuffer_Desc::reserved field must be 0"); |
| return BAD_VALUE; |
| } |
| |
| if (desc->format == AHARDWAREBUFFER_FORMAT_BLOB && desc->height != 1) { |
| ALOGE("Height must be 1 when using the AHARDWAREBUFFER_FORMAT_BLOB format"); |
| return BAD_VALUE; |
| } |
| |
| uint64_t usage = AHardwareBuffer_convertToGrallocUsageBits(desc->usage); |
| sp<GraphicBuffer> gbuffer(new GraphicBuffer( |
| desc->width, desc->height, format, desc->layers, usage, |
| std::string("AHardwareBuffer pid [") + std::to_string(getpid()) + "]")); |
| |
| status_t err = gbuffer->initCheck(); |
| if (err != 0 || gbuffer->handle == 0) { |
| if (err == NO_MEMORY) { |
| GraphicBuffer::dumpAllocationsToSystemLog(); |
| } |
| ALOGE("GraphicBuffer(w=%u, h=%u, lc=%u) failed (%s), handle=%p", |
| desc->width, desc->height, desc->layers, strerror(-err), gbuffer->handle); |
| return err; |
| } |
| |
| *outBuffer = AHardwareBuffer_from_GraphicBuffer(gbuffer.get()); |
| |
| // Ensure the buffer doesn't get destroyed when the sp<> goes away. |
| AHardwareBuffer_acquire(*outBuffer); |
| return NO_ERROR; |
| } |
| |
| void AHardwareBuffer_acquire(AHardwareBuffer* buffer) { |
| // incStrong/decStrong token must be the same, doesn't matter what it is |
| AHardwareBuffer_to_GraphicBuffer(buffer)->incStrong((void*)AHardwareBuffer_acquire); |
| } |
| |
| void AHardwareBuffer_release(AHardwareBuffer* buffer) { |
| // incStrong/decStrong token must be the same, doesn't matter what it is |
| AHardwareBuffer_to_GraphicBuffer(buffer)->decStrong((void*)AHardwareBuffer_acquire); |
| } |
| |
| void AHardwareBuffer_describe(const AHardwareBuffer* buffer, |
| AHardwareBuffer_Desc* outDesc) { |
| if (!buffer || !outDesc) return; |
| |
| const GraphicBuffer* gbuffer = AHardwareBuffer_to_GraphicBuffer(buffer); |
| |
| outDesc->width = gbuffer->getWidth(); |
| outDesc->height = gbuffer->getHeight(); |
| outDesc->layers = gbuffer->getLayerCount(); |
| outDesc->format = AHardwareBuffer_convertFromPixelFormat( |
| static_cast<uint32_t>(gbuffer->getPixelFormat())); |
| outDesc->usage = AHardwareBuffer_convertFromGrallocUsageBits(gbuffer->getUsage()); |
| outDesc->reserved = 0; |
| } |
| |
| int AHardwareBuffer_lock(AHardwareBuffer* buffer, uint64_t usage, |
| int32_t fence, const ARect* rect, void** outVirtualAddress) { |
| if (!buffer) return BAD_VALUE; |
| |
| if (usage & ~(AHARDWAREBUFFER_USAGE_CPU_READ_MASK | |
| AHARDWAREBUFFER_USAGE_CPU_WRITE_MASK)) { |
| ALOGE("Invalid usage flags passed to AHardwareBuffer_lock; only " |
| " AHARDWAREBUFFER_USAGE_CPU_* flags are allowed"); |
| return BAD_VALUE; |
| } |
| |
| usage = AHardwareBuffer_convertToGrallocUsageBits(usage); |
| GraphicBuffer* gBuffer = AHardwareBuffer_to_GraphicBuffer(buffer); |
| Rect bounds; |
| if (!rect) { |
| bounds.set(Rect(gBuffer->getWidth(), gBuffer->getHeight())); |
| } else { |
| bounds.set(Rect(rect->left, rect->top, rect->right, rect->bottom)); |
| } |
| return gBuffer->lockAsync(usage, usage, bounds, outVirtualAddress, fence); |
| } |
| |
| int AHardwareBuffer_unlock(AHardwareBuffer* buffer, int32_t* fence) { |
| if (!buffer) return BAD_VALUE; |
| |
| GraphicBuffer* gBuffer = AHardwareBuffer_to_GraphicBuffer(buffer); |
| return gBuffer->unlockAsync(fence); |
| } |
| |
| int AHardwareBuffer_sendHandleToUnixSocket(const AHardwareBuffer* buffer, int socketFd) { |
| if (!buffer) return BAD_VALUE; |
| const GraphicBuffer* gBuffer = AHardwareBuffer_to_GraphicBuffer(buffer); |
| |
| size_t flattenedSize = gBuffer->getFlattenedSize(); |
| size_t fdCount = gBuffer->getFdCount(); |
| |
| std::unique_ptr<uint8_t[]> data(new uint8_t[flattenedSize]); |
| std::unique_ptr<int[]> fds(new int[fdCount]); |
| |
| // Make copies of needed items since flatten modifies them, and we don't |
| // want to send anything if there's an error during flatten. |
| size_t flattenedSizeCopy = flattenedSize; |
| size_t fdCountCopy = fdCount; |
| void* dataStart = data.get(); |
| int* fdsStart = fds.get(); |
| status_t err = gBuffer->flatten(dataStart, flattenedSizeCopy, fdsStart, |
| fdCountCopy); |
| if (err != NO_ERROR) { |
| return err; |
| } |
| |
| struct iovec iov[1]; |
| iov[0].iov_base = data.get(); |
| iov[0].iov_len = flattenedSize; |
| |
| char buf[CMSG_SPACE(kFdBufferSize)]; |
| struct msghdr msg = { |
| .msg_control = buf, |
| .msg_controllen = sizeof(buf), |
| .msg_iov = &iov[0], |
| .msg_iovlen = 1, |
| }; |
| |
| struct cmsghdr* cmsg = CMSG_FIRSTHDR(&msg); |
| cmsg->cmsg_level = SOL_SOCKET; |
| cmsg->cmsg_type = SCM_RIGHTS; |
| cmsg->cmsg_len = CMSG_LEN(sizeof(int) * fdCount); |
| int* fdData = reinterpret_cast<int*>(CMSG_DATA(cmsg)); |
| memcpy(fdData, fds.get(), sizeof(int) * fdCount); |
| msg.msg_controllen = cmsg->cmsg_len; |
| |
| int result = sendmsg(socketFd, &msg, 0); |
| if (result <= 0) { |
| ALOGE("Error writing AHardwareBuffer to socket: error %#x (%s)", |
| result, strerror(errno)); |
| return result; |
| } |
| return NO_ERROR; |
| } |
| |
| int AHardwareBuffer_recvHandleFromUnixSocket(int socketFd, AHardwareBuffer** outBuffer) { |
| if (!outBuffer) return BAD_VALUE; |
| |
| static constexpr int kMessageBufferSize = 4096 * sizeof(int); |
| |
| std::unique_ptr<char[]> dataBuf(new char[kMessageBufferSize]); |
| char fdBuf[CMSG_SPACE(kFdBufferSize)]; |
| struct iovec iov[1]; |
| iov[0].iov_base = dataBuf.get(); |
| iov[0].iov_len = kMessageBufferSize; |
| |
| struct msghdr msg = { |
| .msg_control = fdBuf, |
| .msg_controllen = sizeof(fdBuf), |
| .msg_iov = &iov[0], |
| .msg_iovlen = 1, |
| }; |
| |
| int result = recvmsg(socketFd, &msg, 0); |
| if (result <= 0) { |
| ALOGE("Error reading AHardwareBuffer from socket: error %#x (%s)", |
| result, strerror(errno)); |
| return result; |
| } |
| |
| if (msg.msg_iovlen != 1) { |
| ALOGE("Error reading AHardwareBuffer from socket: bad data length"); |
| return INVALID_OPERATION; |
| } |
| |
| if (msg.msg_controllen % sizeof(int) != 0) { |
| ALOGE("Error reading AHardwareBuffer from socket: bad fd length"); |
| return INVALID_OPERATION; |
| } |
| |
| size_t dataLen = msg.msg_iov[0].iov_len; |
| const void* data = static_cast<const void*>(msg.msg_iov[0].iov_base); |
| if (!data) { |
| ALOGE("Error reading AHardwareBuffer from socket: no buffer data"); |
| return INVALID_OPERATION; |
| } |
| |
| struct cmsghdr* cmsg = CMSG_FIRSTHDR(&msg); |
| if (!cmsg) { |
| ALOGE("Error reading AHardwareBuffer from socket: no fd header"); |
| return INVALID_OPERATION; |
| } |
| |
| size_t fdCount = msg.msg_controllen >> 2; |
| const int* fdData = reinterpret_cast<const int*>(CMSG_DATA(cmsg)); |
| if (!fdData) { |
| ALOGE("Error reading AHardwareBuffer from socket: no fd data"); |
| return INVALID_OPERATION; |
| } |
| |
| GraphicBuffer* gBuffer = new GraphicBuffer(); |
| status_t err = gBuffer->unflatten(data, dataLen, fdData, fdCount); |
| if (err != NO_ERROR) { |
| return err; |
| } |
| *outBuffer = AHardwareBuffer_from_GraphicBuffer(gBuffer); |
| // Ensure the buffer has a positive ref-count. |
| AHardwareBuffer_acquire(*outBuffer); |
| |
| return NO_ERROR; |
| } |
| |
| |
| // ---------------------------------------------------------------------------- |
| // VNDK functions |
| // ---------------------------------------------------------------------------- |
| |
| const native_handle_t* AHardwareBuffer_getNativeHandle( |
| const AHardwareBuffer* buffer) { |
| if (!buffer) return nullptr; |
| const GraphicBuffer* gbuffer = AHardwareBuffer_to_GraphicBuffer(buffer); |
| return gbuffer->handle; |
| } |
| |
| |
| // ---------------------------------------------------------------------------- |
| // Helpers implementation |
| // ---------------------------------------------------------------------------- |
| |
| namespace android { |
| |
| // A 1:1 mapping of AHardwaqreBuffer bitmasks to gralloc1 bitmasks. |
| struct UsageMaskMapping { |
| uint64_t hardwareBufferMask; |
| uint64_t grallocMask; |
| }; |
| |
| static inline bool containsBits(uint64_t mask, uint64_t bitsToCheck) { |
| return (mask & bitsToCheck) == bitsToCheck && bitsToCheck; |
| } |
| |
| uint32_t AHardwareBuffer_convertFromPixelFormat(uint32_t format) { |
| switch (format) { |
| case HAL_PIXEL_FORMAT_RGBA_8888: return AHARDWAREBUFFER_FORMAT_R8G8B8A8_UNORM; |
| case HAL_PIXEL_FORMAT_RGBX_8888: return AHARDWAREBUFFER_FORMAT_R8G8B8X8_UNORM; |
| case HAL_PIXEL_FORMAT_RGB_565: return AHARDWAREBUFFER_FORMAT_R5G6B5_UNORM; |
| case HAL_PIXEL_FORMAT_RGB_888: return AHARDWAREBUFFER_FORMAT_R8G8B8_UNORM; |
| case HAL_PIXEL_FORMAT_RGBA_FP16: return AHARDWAREBUFFER_FORMAT_R16G16B16A16_FLOAT; |
| case HAL_PIXEL_FORMAT_RGBA_1010102: return AHARDWAREBUFFER_FORMAT_R10G10B10A2_UNORM; |
| case HAL_PIXEL_FORMAT_BLOB: return AHARDWAREBUFFER_FORMAT_BLOB; |
| default:ALOGE("Unknown pixel format %u", format); |
| return 0; |
| } |
| } |
| |
| uint32_t AHardwareBuffer_convertToPixelFormat(uint32_t format) { |
| switch (format) { |
| case AHARDWAREBUFFER_FORMAT_R8G8B8A8_UNORM: return HAL_PIXEL_FORMAT_RGBA_8888; |
| case AHARDWAREBUFFER_FORMAT_R8G8B8X8_UNORM: return HAL_PIXEL_FORMAT_RGBX_8888; |
| case AHARDWAREBUFFER_FORMAT_R5G6B5_UNORM: return HAL_PIXEL_FORMAT_RGB_565; |
| case AHARDWAREBUFFER_FORMAT_R8G8B8_UNORM: return HAL_PIXEL_FORMAT_RGB_888; |
| case AHARDWAREBUFFER_FORMAT_R16G16B16A16_FLOAT: return HAL_PIXEL_FORMAT_RGBA_FP16; |
| case AHARDWAREBUFFER_FORMAT_R10G10B10A2_UNORM: return HAL_PIXEL_FORMAT_RGBA_1010102; |
| case AHARDWAREBUFFER_FORMAT_BLOB: return HAL_PIXEL_FORMAT_BLOB; |
| default:ALOGE("Unknown AHardwareBuffer format %u", format); |
| return 0; |
| } |
| } |
| |
| uint64_t AHardwareBuffer_convertToGrallocUsageBits(uint64_t usage) { |
| using android::hardware::graphics::common::V1_0::BufferUsage; |
| static_assert(AHARDWAREBUFFER_USAGE_CPU_READ_NEVER == (uint64_t)BufferUsage::CPU_READ_NEVER, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_CPU_READ_RARELY == (uint64_t)BufferUsage::CPU_READ_RARELY, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_CPU_READ_OFTEN == (uint64_t)BufferUsage::CPU_READ_OFTEN, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_CPU_WRITE_NEVER == (uint64_t)BufferUsage::CPU_WRITE_NEVER, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_CPU_WRITE_RARELY == (uint64_t)BufferUsage::CPU_WRITE_RARELY, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_CPU_WRITE_OFTEN == (uint64_t)BufferUsage::CPU_WRITE_OFTEN, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_GPU_SAMPLED_IMAGE == (uint64_t)BufferUsage::GPU_TEXTURE, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_GPU_COLOR_OUTPUT == (uint64_t)BufferUsage::GPU_RENDER_TARGET, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_PROTECTED_CONTENT == (uint64_t)BufferUsage::PROTECTED, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_VIDEO_ENCODE == (uint64_t)BufferUsage::VIDEO_ENCODER, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_GPU_DATA_BUFFER == (uint64_t)BufferUsage::GPU_DATA_BUFFER, |
| "gralloc and AHardwareBuffer flags don't match"); |
| static_assert(AHARDWAREBUFFER_USAGE_SENSOR_DIRECT_DATA == (uint64_t)BufferUsage::SENSOR_DIRECT_DATA, |
| "gralloc and AHardwareBuffer flags don't match"); |
| return usage; |
| } |
| |
| uint64_t AHardwareBuffer_convertFromGrallocUsageBits(uint64_t usage) { |
| return usage; |
| } |
| |
| const GraphicBuffer* AHardwareBuffer_to_GraphicBuffer(const AHardwareBuffer* buffer) { |
| return reinterpret_cast<const GraphicBuffer*>(buffer); |
| } |
| |
| GraphicBuffer* AHardwareBuffer_to_GraphicBuffer(AHardwareBuffer* buffer) { |
| return reinterpret_cast<GraphicBuffer*>(buffer); |
| } |
| |
| const ANativeWindowBuffer* AHardwareBuffer_to_ANativeWindowBuffer(const AHardwareBuffer* buffer) { |
| return AHardwareBuffer_to_GraphicBuffer(buffer)->getNativeBuffer(); |
| } |
| |
| ANativeWindowBuffer* AHardwareBuffer_to_ANativeWindowBuffer(AHardwareBuffer* buffer) { |
| return AHardwareBuffer_to_GraphicBuffer(buffer)->getNativeBuffer(); |
| } |
| |
| AHardwareBuffer* AHardwareBuffer_from_GraphicBuffer(GraphicBuffer* buffer) { |
| return reinterpret_cast<AHardwareBuffer*>(buffer); |
| } |
| |
| } // namespace android |