301 lines
9.7 KiB
C++
301 lines
9.7 KiB
C++
/*
|
|
* Copyright (C) 2014-2025 Amlogic, Inc. All rights reserved.
|
|
*
|
|
* All information contained herein is Amlogic confidential.
|
|
*
|
|
* This software is provided to you pursuant to Software License Agreement
|
|
* (SLA) with Amlogic Inc ("Amlogic"). This software may be used
|
|
* only in accordance with the terms of this agreement.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification is strictly prohibited without prior written permission from
|
|
* Amlogic.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
* "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
* LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
* A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
* OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
* THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#define LOG_TAG "DeviceSource"
|
|
|
|
#include "DeviceSource.h"
|
|
|
|
#include <fcntl.h>
|
|
#include <pthread.h>
|
|
#include <stdio.h>
|
|
#include <sys/ioctl.h>
|
|
#include <sys/poll.h>
|
|
#include <sys/stat.h>
|
|
#include <sys/types.h>
|
|
#include <unistd.h>
|
|
#include <utils/CallStack.h>
|
|
|
|
#include <string>
|
|
|
|
#include "SubtitleLog.h"
|
|
|
|
static const std::string SYSFS_VIDEO_PTS = "/sys/class/tsync/pts_video_u64";
|
|
static const std::string SYSFS_SUBTITLE_TYPE = "/sys/class/subtitle/subtype";
|
|
static const std::string SYSFS_SUBTITLE_SIZE = "/sys/class/subtitle/size";
|
|
static const std::string SYSFS_SUBTITLE_TOTAL = "/sys/class/subtitle/total";
|
|
static const std::string SUBTITLE_READ_DEVICE = "/dev/amstream_sub_read";
|
|
|
|
#define AMSTREAM_IOC_MAGIC 'S'
|
|
#define AMSTREAM_IOC_SUB_LENGTH _IOR(AMSTREAM_IOC_MAGIC, 0x1b, int)
|
|
#define AMSTREAM_IOC_SUB_RESET _IOW(AMSTREAM_IOC_MAGIC, 0x1a, int)
|
|
|
|
static inline bool pollFd(int fd, int timeout) {
|
|
struct pollfd pollFd[1];
|
|
if (fd <= 0) {
|
|
return false;
|
|
}
|
|
pollFd[0].fd = fd;
|
|
pollFd[0].events = POLLOUT;
|
|
return (::poll(pollFd, 1, timeout) > 0);
|
|
}
|
|
|
|
static inline int64_t sysfsReadInt(const char *path, int base) {
|
|
int fd;
|
|
int64_t val = 0;
|
|
char bcmd[64] = {0};
|
|
fd = ::open(path, O_RDONLY);
|
|
if (fd >= 0) {
|
|
int c = read(fd, bcmd, sizeof(bcmd));
|
|
if (c > 0) val = strtoll(bcmd, NULL, base);
|
|
::close(fd);
|
|
} else {
|
|
SUBTITLE_LOGE("unable to open file %s,err: %s", path, strerror(errno));
|
|
}
|
|
return val;
|
|
}
|
|
|
|
DeviceSource::DeviceSource() {
|
|
mState = E_SOURCE_INV;
|
|
mExitRequested = false;
|
|
mDumpFd = -1;
|
|
mRdFd = ::open(SUBTITLE_READ_DEVICE.c_str(), O_RDONLY);
|
|
SUBTITLE_LOGI("DeviceSource: %s=%d", SUBTITLE_READ_DEVICE.c_str(), mRdFd);
|
|
}
|
|
|
|
DeviceSource::~DeviceSource() {
|
|
SUBTITLE_LOGI("~DeviceSource: %s=%d", SUBTITLE_READ_DEVICE.c_str(), mRdFd);
|
|
|
|
mExitRequested = true;
|
|
if (mRenderTimeThread != nullptr) {
|
|
mRenderTimeThread->join();
|
|
mRenderTimeThread = nullptr;
|
|
}
|
|
|
|
::close(mRdFd);
|
|
if (mReadThread != nullptr) {
|
|
mReadThread->join();
|
|
mReadThread = nullptr;
|
|
}
|
|
|
|
if (mDumpFd > 0) ::close(mDumpFd);
|
|
}
|
|
|
|
bool DeviceSource::notifyInfoChange() {
|
|
std::unique_lock<std::mutex> autolock(mLock);
|
|
for (auto it = mInfoListeners.begin(); it != mInfoListeners.end(); it++) {
|
|
auto wk_listener = (*it);
|
|
|
|
// coverity[INTEGER_OVERFLOW]
|
|
if (auto lstn = wk_listener.lock()) {
|
|
auto value = sysfsReadInt(SYSFS_SUBTITLE_TYPE.c_str(), 10);
|
|
if (value > 0) { // 0:no sub
|
|
lstn->onTypeChanged(static_cast<int>(value));
|
|
}
|
|
|
|
value = sysfsReadInt(SYSFS_SUBTITLE_TOTAL.c_str(), 10);
|
|
if (value > 0) {
|
|
lstn->onSubtitleChanged(static_cast<int>(value));
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
void DeviceSource::loopRenderTime() {
|
|
while (!mExitRequested) {
|
|
mLock.lock();
|
|
for (auto it = mInfoListeners.begin(); it != mInfoListeners.end(); it++) {
|
|
auto wk_listener = (*it);
|
|
|
|
if (wk_listener.expired()) {
|
|
SUBTITLE_LOGI("[threadLoop] lstn null.\n");
|
|
continue;
|
|
}
|
|
|
|
int64_t value = sysfsReadInt(SYSFS_VIDEO_PTS.c_str(), 16);
|
|
|
|
static int i = 0;
|
|
if (i++ % 300 == 0) {
|
|
SUBTITLE_LOGI("read pts: %" PRId64, value);
|
|
}
|
|
if (!mExitRequested) {
|
|
if (auto lstn = wk_listener.lock()) {
|
|
lstn->onRenderTimeChanged(value);
|
|
}
|
|
}
|
|
}
|
|
mLock.unlock();
|
|
std::this_thread::sleep_for(std::chrono::milliseconds(16));
|
|
}
|
|
}
|
|
|
|
void DeviceSource::loopDriverData() {
|
|
while (!mExitRequested && mRdFd > 0) {
|
|
if (pollFd(mRdFd, 10)) {
|
|
unsigned long size = 0;
|
|
int r = ::ioctl(mRdFd, AMSTREAM_IOC_SUB_LENGTH, &size);
|
|
if (r < 0) continue;
|
|
std::shared_ptr<char> spBuf(new char[size], [](char *buf) { delete[] buf; });
|
|
int read = readDriverData(spBuf.get(), size);
|
|
mSegment->push(std::move(spBuf), read);
|
|
// coverity[RESOURCE_LEAK]
|
|
} else {
|
|
if (mExitRequested || mRdFd < 0) return;
|
|
usleep(1); // in case of little page subtitle loss. improve read freq
|
|
}
|
|
}
|
|
}
|
|
|
|
bool DeviceSource::start() {
|
|
if (E_SOURCE_STARTED == mState) {
|
|
SUBTITLE_LOGE("already stated");
|
|
return false;
|
|
}
|
|
mState = E_SOURCE_STARTED;
|
|
auto total = sysfsReadInt(SYSFS_SUBTITLE_TOTAL.c_str(), 10);
|
|
auto subtype =
|
|
sysfsReadInt(SYSFS_SUBTITLE_TYPE.c_str(), 10); // scte27 subtype:0,not start thread
|
|
|
|
mSegment = std::shared_ptr<BufferSegment>(new BufferSegment());
|
|
|
|
if (total > 0 && subtype > 0) {
|
|
mRenderTimeThread =
|
|
std::shared_ptr<std::thread>(new std::thread(&DeviceSource::loopRenderTime, this));
|
|
}
|
|
|
|
mReadThread =
|
|
std::shared_ptr<std::thread>(new std::thread(&DeviceSource::loopDriverData, this));
|
|
|
|
// first check types
|
|
notifyInfoChange();
|
|
|
|
return false;
|
|
}
|
|
|
|
bool DeviceSource::stop() {
|
|
mState = E_SOURCE_STOPPED;
|
|
auto ret = ::ioctl(mRdFd, AMSTREAM_IOC_SUB_RESET);
|
|
if (ret >= 0) {
|
|
// If parser has problem, it read more data than provided
|
|
// then stop cannot stopped. need notify exit here.
|
|
mSegment->notifyExit();
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
SubtitleIOType DeviceSource::type() {
|
|
return E_SUBTITLE_DEV;
|
|
}
|
|
|
|
bool DeviceSource::isFileAvailable() {
|
|
return false;
|
|
}
|
|
|
|
size_t DeviceSource::availableDataSize() {
|
|
if (mSegment == nullptr) return 0;
|
|
|
|
return mSegment->availableDataSize();
|
|
}
|
|
|
|
size_t DeviceSource::readDriverData(void *buffer, size_t size) {
|
|
size_t data_size = size, read_done = 0;
|
|
char *buf = static_cast<char *>(buffer);
|
|
|
|
while (data_size && mState == E_SOURCE_STARTED) {
|
|
auto r = ::read(mRdFd, buf + read_done, data_size);
|
|
if (r < 0) {
|
|
return 0;
|
|
} else {
|
|
data_size -= r;
|
|
read_done += r;
|
|
}
|
|
}
|
|
|
|
if (isSubtitleDumpAvailable()) {
|
|
if (mDumpFd == -1) {
|
|
mDumpFd = ::open(DUMP_DATA_FILE_PATH, O_RDWR | O_CREAT, 0666);
|
|
SUBTITLE_LOGI("need dump Source: mDumpFd=%d %d", mDumpFd, errno);
|
|
}
|
|
|
|
if (mDumpFd > 0) {
|
|
size_t ret = write(mDumpFd, buffer, size);
|
|
if (ret != size) {
|
|
SUBTITLE_LOGE("wrote %zu/%zu", ret, size);
|
|
}
|
|
}
|
|
}
|
|
|
|
// coverity[INTEGER_OVERFLOW]
|
|
return read_done;
|
|
}
|
|
|
|
size_t DeviceSource::read(void *buffer, size_t size) {
|
|
size_t read = 0;
|
|
|
|
// Current design of Parser Read, do not need add lock protection.
|
|
// because all the read, is in Parser's parser thread.
|
|
// We only need add lock here, is for protect access the mCurrentItem's
|
|
// member function multi-thread.
|
|
// Current Impl do not need lock, if required, then redesign the SegmentBuffer.
|
|
|
|
// in case of applied size more than 1024*2, such as dvb subtitle,
|
|
// and data process error for two reads.
|
|
// so add until read applied data then exit.
|
|
while (read != size && mState == E_SOURCE_STARTED) {
|
|
if (mCurrentItem != nullptr && !mCurrentItem->isEmpty()) {
|
|
read += mCurrentItem->read_l(((char *)buffer + read), size - read);
|
|
// SUBTITLE_LOGI("read:%d,size:%d", read, size);
|
|
if (read == size) return read;
|
|
} else {
|
|
// SUBTITLE_LOGI("mCurrentItem null, pop next buffer item");
|
|
mCurrentItem = mSegment->pop();
|
|
}
|
|
}
|
|
// read += mCurrentItem->read(((char *)buffer+read), size-read);
|
|
return read;
|
|
}
|
|
|
|
void DeviceSource::dump(int fd, const char *prefix) {
|
|
dprintf(fd, "%s nDeviceSource:\n", prefix);
|
|
{
|
|
std::unique_lock<std::mutex> autolock(mLock);
|
|
for (auto it = mInfoListeners.begin(); it != mInfoListeners.end(); it++) {
|
|
auto wk_listener = (*it);
|
|
if (auto lstn = wk_listener.lock())
|
|
dprintf(fd, "%s InfoListener: %p\n", prefix, lstn.get());
|
|
}
|
|
}
|
|
dprintf(fd, "%s state:%d\n\n", prefix, mState);
|
|
dprintf(fd, "%s Total Subtitle Tracks: (%s)%" PRId64 "\n", prefix,
|
|
SYSFS_SUBTITLE_TOTAL.c_str(), sysfsReadInt(SYSFS_SUBTITLE_TOTAL.c_str(), 10));
|
|
dprintf(fd, "%s Current Subtitle type: (%s)%" PRId64 "\n", prefix,
|
|
SYSFS_SUBTITLE_TYPE.c_str(), sysfsReadInt(SYSFS_SUBTITLE_TYPE.c_str(), 10));
|
|
dprintf(fd, "\n%s Current Unconsumed Data Size: %zu\n", prefix, availableDataSize());
|
|
}
|