blob: b3e9ab007ce0a3ca8049c84200431ea629f98fc4 [file] [log] [blame]
/*
* Copyright (c) 2011 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#include "ssrc_database.h"
#include "critical_section_wrapper.h"
#include "trace.h"
#include <stdlib.h>
#include <cassert>
#ifdef _WIN32
#include <windows.h>
#include <MMSystem.h> //timeGetTime
// TODO(hellner): investigate if it is necessary to disable these warnings.
#pragma warning(disable:4311)
#pragma warning(disable:4312)
#else
#include <stdio.h>
#include <string.h>
#include <time.h>
#include <sys/time.h>
#endif
namespace webrtc {
SSRCDatabase*
SSRCDatabase::StaticInstance(CountOperation count_operation)
{
SSRCDatabase* impl =
GetStaticInstance<SSRCDatabase>(count_operation);
return impl;
}
SSRCDatabase*
SSRCDatabase::GetSSRCDatabase()
{
return StaticInstance(kAddRef);
}
void
SSRCDatabase::ReturnSSRCDatabase()
{
StaticInstance(kRelease);
}
WebRtc_UWord32
SSRCDatabase::CreateSSRC()
{
CriticalSectionScoped lock(_critSect);
WebRtc_UWord32 ssrc = GenerateRandom();
#ifndef WEBRTC_NO_STL
while(_ssrcMap.find(ssrc) != _ssrcMap.end())
{
ssrc = GenerateRandom();
}
_ssrcMap[ssrc] = 0;
#else
if(_sizeOfSSRC <= _numberOfSSRC)
{
// allocate more space
const int newSize = _sizeOfSSRC + 10;
WebRtc_UWord32* tempSSRCVector = new WebRtc_UWord32[newSize];
memcpy(tempSSRCVector, _ssrcVector, _sizeOfSSRC*sizeof(WebRtc_UWord32));
delete [] _ssrcVector;
_ssrcVector = tempSSRCVector;
_sizeOfSSRC = newSize;
}
// check if in DB
if(_ssrcVector)
{
for (int i=0; i<_numberOfSSRC; i++)
{
if (_ssrcVector[i] == ssrc)
{
// we have a match
i = 0; // start over with a new ssrc
ssrc = GenerateRandom();
}
}
// add to database
_ssrcVector[_numberOfSSRC] = ssrc;
_numberOfSSRC++;
}
#endif
return ssrc;
}
WebRtc_Word32
SSRCDatabase::RegisterSSRC(const WebRtc_UWord32 ssrc)
{
CriticalSectionScoped lock(_critSect);
#ifndef WEBRTC_NO_STL
_ssrcMap[ssrc] = 0;
#else
if(_sizeOfSSRC <= _numberOfSSRC)
{
// allocate more space
const int newSize = _sizeOfSSRC + 10;
WebRtc_UWord32* tempSSRCVector = new WebRtc_UWord32[newSize];
memcpy(tempSSRCVector, _ssrcVector, _sizeOfSSRC*sizeof(WebRtc_UWord32));
delete [] _ssrcVector;
_ssrcVector = tempSSRCVector;
_sizeOfSSRC = newSize;
}
// check if in DB
if(_ssrcVector)
{
for (int i=0; i<_numberOfSSRC; i++)
{
if (_ssrcVector[i] == ssrc)
{
// we have a match
return -1;
}
}
// add to database
_ssrcVector[_numberOfSSRC] = ssrc;
_numberOfSSRC++;
}
#endif
return 0;
}
WebRtc_Word32
SSRCDatabase::ReturnSSRC(const WebRtc_UWord32 ssrc)
{
CriticalSectionScoped lock(_critSect);
#ifndef WEBRTC_NO_STL
_ssrcMap.erase(ssrc);
#else
if(_ssrcVector)
{
for (int i=0; i<_numberOfSSRC; i++)
{
if (_ssrcVector[i] == ssrc)
{
// we have a match
// remove from database
_ssrcVector[i] = _ssrcVector[_numberOfSSRC-1];
_numberOfSSRC--;
break;
}
}
}
#endif
return 0;
}
SSRCDatabase::SSRCDatabase()
{
// we need to seed the random generator, otherwise we get 26500 each time, hardly a random value :)
#ifdef _WIN32
srand(timeGetTime());
#else
struct timeval tv;
struct timezone tz;
gettimeofday(&tv, &tz);
srand(tv.tv_usec);
#endif
#ifdef WEBRTC_NO_STL
_sizeOfSSRC = 10;
_numberOfSSRC = 0;
_ssrcVector = new WebRtc_UWord32[10];
#endif
_critSect = CriticalSectionWrapper::CreateCriticalSection();
WEBRTC_TRACE(kTraceMemory, kTraceRtpRtcp, -1, "%s created", __FUNCTION__);
}
SSRCDatabase::~SSRCDatabase()
{
#ifdef WEBRTC_NO_STL
delete [] _ssrcVector;
#else
_ssrcMap.clear();
#endif
delete _critSect;
WEBRTC_TRACE(kTraceMemory, kTraceRtpRtcp, -1, "%s deleted", __FUNCTION__);
}
WebRtc_UWord32 SSRCDatabase::GenerateRandom()
{
WebRtc_UWord32 ssrc = 0;
do
{
ssrc = rand();
ssrc = ssrc <<16;
ssrc += rand();
} while (ssrc == 0 || ssrc == 0xffffffff);
return ssrc;
}
} // namespace webrtc