libaribb25/aribb25/libaribb25.cpp

// libaribb25.cpp: CB25Decoder クラスのインプリメンテーション
//
//////////////////////////////////////////////////////////////////////
#include "libaribb25.h"

BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
{
	if (fdwReason == DLL_PROCESS_ATTACH) {
		DisableThreadLibraryCalls(hinstDLL);
	} else if (fdwReason == DLL_PROCESS_DETACH) {
		// 未開放の場合はインスタンス開放
		if (CB25Decoder::m_pThis) {
			CB25Decoder::m_pThis->Release();
		}
	}
	return TRUE;
}

//////////////////////////////////////////////////////////////////////
// インスタンス生成メソッド
//////////////////////////////////////////////////////////////////////

extern "C"
{

__declspec(dllexport) IB25Decoder * CreateB25Decoder()
{
	// インスタンス生成
		return dynamic_cast<IB25Decoder *>(new CB25Decoder());
}

__declspec(dllexport) IB25Decoder2 * CreateB25Decoder2()
{
	// インスタンス生成
		return dynamic_cast<IB25Decoder2 *>(new CB25Decoder());
}

}

//////////////////////////////////////////////////////////////////////
// 構築/消滅
//////////////////////////////////////////////////////////////////////

// 静的メンバ初期化
CB25Decoder * CB25Decoder::m_pThis = nullptr;

CB25Decoder::CB25Decoder(void) : _bcas(nullptr), _b25(nullptr), _data(nullptr)
{
	m_pThis = this;
}

CB25Decoder::~CB25Decoder(void)
{
	m_pThis = nullptr;
}

void CB25Decoder::Release()
{
	// インスタンス開放
	if (_data)
		::free(_data);

	std::lock_guard<std::mutex> lock(_mtx);

	if (_b25)
		_b25->release(_b25);

	if (_bcas)
		_bcas->release(_bcas);

	delete this;
}

const BOOL CB25Decoder::Initialize(DWORD dwRound)
{
	std::lock_guard<std::mutex> lock(_mtx);

	if (_b25)
		return Reset();

	_bcas = create_b_cas_card();
	if (!_bcas)
		return FALSE;

	if (_bcas->init(_bcas) < 0)
		goto err;

	_b25 = create_arib_std_b25();
	if (!_b25)
		goto err;

	if (_b25->set_b_cas_card(_b25, _bcas) < 0)
		goto err;

	_b25->set_strip(_b25, 1);
	_b25->set_emm_proc(_b25, 1);
	_b25->set_multi2_round(_b25, dwRound);

	return TRUE;	// success

err:
	if (_b25) {
		_b25->release(_b25);
		_b25 = nullptr;
	}

	if (_bcas) {
		_bcas->release(_bcas);
		_bcas = nullptr;
	}

	_errtime = time(nullptr);
	return FALSE;	// error
}

const BOOL CB25Decoder::Decode(BYTE *pSrcBuf, const DWORD dwSrcSize, BYTE **ppDstBuf, DWORD *pdwDstSize)
{
	if (!pSrcBuf || !dwSrcSize || !ppDstBuf || !pdwDstSize) {
		// 引数が不正
		return FALSE;
	}

	if (!_b25) {
		time_t now = time(nullptr);
		if (difftime(now, _errtime) > RETRY_INTERVAL) {
			if (Initialize() < 0)
				_errtime = now;
		}

		if (!_b25) {
			if (*ppDstBuf != pSrcBuf) {
				*ppDstBuf = pSrcBuf;
				*pdwDstSize = dwSrcSize;
			}
			return FALSE;
		}
	}

	if (_data) {
		::free(_data);
		_data = nullptr;
	}

	ARIB_STD_B25_BUFFER buf;
	buf.data = pSrcBuf;
	buf.size = dwSrcSize;
	const int rc = _b25->put(_b25, &buf);
	if (rc == ARIB_STD_B25_WARN_PAT_NOT_COMPLETE) {
		if (*ppDstBuf != pSrcBuf) {
			*ppDstBuf = pSrcBuf;
			*pdwDstSize = dwSrcSize;
		}
		return TRUE;	// success
	} else if (rc < 0) {
		if (rc >= ARIB_STD_B25_ERROR_NO_ECM_IN_HEAD_32M) {
			// pass through
			_b25->release(_b25);
			_b25 = nullptr;
			_bcas->release(_bcas);
			_bcas = nullptr;
			if (*ppDstBuf != pSrcBuf) {
				*ppDstBuf = pSrcBuf;
				*pdwDstSize = dwSrcSize;
			}
		} else {
			BYTE *p = nullptr;
			_b25->withdraw(_b25, &buf);	// withdraw src buffer
			if (buf.size > 0)
				p = (BYTE *)::malloc(buf.size + dwSrcSize);

			if (p) {
				::memcpy(p, buf.data, buf.size);
				::memcpy(p + buf.size, pSrcBuf, dwSrcSize);
				*ppDstBuf = p;
				*pdwDstSize = buf.size + dwSrcSize;
				_data = p;
			} else {
				if (*ppDstBuf != pSrcBuf) {
					*ppDstBuf = pSrcBuf;
					*pdwDstSize = dwSrcSize;
				}
			}

			if (rc == ARIB_STD_B25_ERROR_ECM_PROC_FAILURE) {
				// pass through
				_b25->release(_b25);
				_b25 = nullptr;
				_bcas->release(_bcas);
				_bcas = nullptr;
			}
		}
		_errtime = time(nullptr);
		return FALSE;	// error
	}
	_b25->get(_b25, &buf);
	*ppDstBuf = buf.data;
	*pdwDstSize = buf.size;
	return TRUE;	// success
}

const BOOL CB25Decoder::Flush(BYTE **ppDstBuf, DWORD *pdwDstSize)
{
	BOOL ret = TRUE;
	
	if (_b25) {
		int rc = _b25->flush(_b25);
		ret = (rc < 0) ? FALSE : TRUE;
	}

	*ppDstBuf = nullptr;
	*pdwDstSize = 0;

	return ret;
}

const BOOL CB25Decoder::Reset(void)
{
	BOOL ret = TRUE;
	
	if (_b25) {
		int rc = _b25->reset(_b25);
		ret = (rc < 0) ? FALSE : TRUE;
	}

	return ret;
}

void CB25Decoder::DiscardNullPacket(const bool bEnable)
{
	// NULLパケット破棄の有無を設定
	_b25->set_strip(_b25, bEnable);
}

void CB25Decoder::DiscardScramblePacket(const bool bEnable)
{
	// 復号漏れパケット破棄の有無を設定
}

void CB25Decoder::EnableEmmProcess(const bool bEnable)
{
	// EMM処理の有効/無効を設定
	_b25->set_emm_proc(_b25, bEnable);
}

void CB25Decoder::SetMulti2Round(const int32_t round)
{
	// ラウンド段数を設定
	_b25->set_multi2_round(_b25, round);
}

const DWORD CB25Decoder::GetDescramblingState(const WORD wProgramID)
{
	// 指定したプログラムIDの復号状態を返す
	return 0;
}

void CB25Decoder::ResetStatistics(void)
{
}

const DWORD CB25Decoder::GetPacketStride(void)
{
	// パケット周期を返す
	return 0;
}

const DWORD CB25Decoder::GetInputPacketNum(const WORD wPID)
{
	// 入力パケット数を返す　※TS_INVALID_PID指定時は全PIDの合計を返す
	return 0;
}

const DWORD CB25Decoder::GetOutputPacketNum(const WORD wPID)
{
	// 出力パケット数を返す　※TS_INVALID_PID指定時は全PIDの合計を返す
	return 0;
}

const DWORD CB25Decoder::GetSyncErrNum(void)
{
	// 同期エラー数を返す
	return 0;
}

const DWORD CB25Decoder::GetFormatErrNum(void)
{
	// フォーマットエラー数を返す
	return 0;
}

const DWORD CB25Decoder::GetTransportErrNum(void)
{
	// ビットエラー数を返す
	return 0;
}

const DWORD CB25Decoder::GetContinuityErrNum(const WORD wPID)
{
	// ドロップパケット数を返す
	return 0;
}

const DWORD CB25Decoder::GetScramblePacketNum(const WORD wPID)
{
	// 復号漏れパケット数を返す
	return 0;
}

const DWORD CB25Decoder::GetEcmProcessNum(void)
{
	// ECM処理数を返す(B-CASカードアクセス回数)
	return 0;
}

const DWORD CB25Decoder::GetEmmProcessNum(void)
{
	// EMM処理数を返す(B-CASカードアクセス回数)
	return 0;
}