gecko-dev/dom/media/gtest/MockMediaResource.cpp

92 строки
2.2 KiB
C++

/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
#include "MockMediaResource.h"
#include <sys/types.h>
#include <sys/stat.h>
namespace mozilla {
MockMediaResource::MockMediaResource(const char* aFileName)
: mFileHandle(nullptr), mFileName(aFileName) {}
nsresult MockMediaResource::Open() {
mFileHandle = fopen(mFileName, "rb");
if (mFileHandle == nullptr) {
printf_stderr("Can't open %s\n", mFileName);
return NS_ERROR_FAILURE;
}
return NS_OK;
}
MockMediaResource::~MockMediaResource() {
if (mFileHandle != nullptr) {
fclose(mFileHandle);
}
}
nsresult MockMediaResource::ReadAt(int64_t aOffset, char* aBuffer,
uint32_t aCount, uint32_t* aBytes) {
if (mFileHandle == nullptr) {
return NS_ERROR_FAILURE;
}
// Make it fail if we're re-entrant
if (mEntry++) {
MOZ_ASSERT(false);
return NS_ERROR_FAILURE;
}
fseek(mFileHandle, aOffset, SEEK_SET);
*aBytes = fread(aBuffer, 1, aCount, mFileHandle);
mEntry--;
return ferror(mFileHandle) ? NS_ERROR_FAILURE : NS_OK;
}
int64_t MockMediaResource::GetLength() {
if (mFileHandle == nullptr) {
return -1;
}
fseek(mFileHandle, 0, SEEK_END);
return ftell(mFileHandle);
}
void MockMediaResource::MockClearBufferedRanges() { mRanges.Clear(); }
void MockMediaResource::MockAddBufferedRange(int64_t aStart, int64_t aEnd) {
mRanges += MediaByteRange(aStart, aEnd);
}
int64_t MockMediaResource::GetNextCachedData(int64_t aOffset) {
if (!aOffset) {
return mRanges.Length() ? mRanges[0].mStart : -1;
}
for (size_t i = 0; i < mRanges.Length(); i++) {
if (aOffset == mRanges[i].mStart) {
++i;
return i < mRanges.Length() ? mRanges[i].mStart : -1;
}
}
return -1;
}
int64_t MockMediaResource::GetCachedDataEnd(int64_t aOffset) {
for (size_t i = 0; i < mRanges.Length(); i++) {
if (aOffset == mRanges[i].mStart) {
return mRanges[i].mEnd;
}
}
return aOffset;
}
nsresult MockMediaResource::GetCachedRanges(MediaByteRangeSet& aRanges) {
aRanges = mRanges;
return NS_OK;
}
} // namespace mozilla