void ProtocolManager::GetNetworkedProtocols(map<uint32_t, BaseProtocol *> &result) {
result.clear();
FOR_MAP(_activeProtocols, uint32_t, BaseProtocol *, i) {
BaseProtocol *pProtocol = MAP_VAL(i)->GetNearEndpoint();
if (MAP_HAS1(result, pProtocol->GetId()))
continue;
IOHandler *pIOHandler = pProtocol->GetIOHandler();
if ((pIOHandler == NULL)
|| ((pIOHandler->GetType() != IOHT_TCP_CARRIER)
&& (pIOHandler->GetType() != IOHT_UDP_CARRIER)))
continue;
result[pProtocol->GetId()] = pProtocol;
}
File* BaseInFileStream::GetFile(string filePath, uint32_t windowSize) {
File *pResult = NULL;
if (!MAP_HAS1(_fileCache, filePath)) {
pResult = new File();
if (!pResult->Initialize(filePath)) {
delete pResult;
return NULL;
}
_fileCache[filePath] = pair<uint32_t, File *>(1, pResult);
} else {
pResult = _fileCache[filePath].second;
_fileCache[filePath].first++;
}
return pResult;
}
bool OutboundConnectivity::RegisterUDPAudioClient1(uint32_t rtspProtocolId,
sockaddr_in &data, sockaddr_in &rtcp) {
if (!MAP_HAS1(_clients, rtspProtocolId))
_clients[rtspProtocolId] = RTPClient();
RTPClient &client = _clients[rtspProtocolId];
if (client.hasAudio) {
FATAL("Client already registered for audio feed");
return false;
}
client.hasAudio = true;
client.isUdp = true;
client.audioDataAddress = data;
client.audioRtcpAddress = rtcp;
return true;
}
bool AtomMVEX::AtomCreated(BaseAtom *pAtom) {
switch (pAtom->GetTypeNumeric()) {
case A_TREX:
{
AtomTREX *pTemp = (AtomTREX *) pAtom;
if (MAP_HAS1(_trex, pTemp->GetTrackID())) {
FATAL("Track already registered");
return false;
}
_trex[pTemp->GetTrackID()] = pTemp;
return true;
}
default:
{
FATAL("Invalid atom type: %s", STR(pAtom->GetTypeString()));
return false;
}
}
}
bool StreamsManager::RegisterStream(BaseStream *pStream) {
//1. Test to see if we had registered this stream before
if (MAP_HAS1(_streamsByUniqueId, pStream->GetUniqueId())) {
FATAL("Stream with unique ID %u already registered", pStream->GetUniqueId());
return false;
}
//2. Register the stream
_streamsByUniqueId[pStream->GetUniqueId()] = pStream;
if (pStream->GetProtocol() != NULL)
_streamsByProtocolId[pStream->GetProtocol()->GetId()][pStream->GetUniqueId()] = pStream;
_streamsByType[pStream->GetType()][pStream->GetUniqueId()] = pStream;
_streamsByName[pStream->GetName()][pStream->GetUniqueId()] = pStream;
_pApplication->SignalStreamRegistered(pStream);
//3. Done
return true;
}
bool parseURI(string stringUri, URI &uri) {
/*
* schema://[username[:password]@]host[:port][/[path[?parameters]]]
*/
LOG_URI("------------------------");
LOG_URI("stringUri: `%s`", STR(stringUri));
string fullUri;
string fullUriWithAuth = stringUri;
string scheme;
string authentication;
string username;
string password;
string hostPort;
string host;
string portString;
uint16_t port;
bool portSpecified;
string fullDocumentPathWithParameters;
string fullDocumentPath;
string fullParameters;
string documentPath;
string document;
string documentWithFullParameters;
Variant parameters;
string::size_type cursor = 0;
string::size_type pos = 0;
//1. Reset
uri.Reset();
//2. trim
trim(stringUri);
if (stringUri == "") {
FATAL("Empty uri");
return false;
}
//2. Get the scheme and the default port
pos = stringUri.find("://", cursor);
if (pos == string::npos) {
FATAL("Unable to determine scheme");
return false;
}
scheme = lowerCase(stringUri.substr(cursor, pos - cursor));
cursor = pos + 3;
if (_schemeToPort.size() == 0) {
_schemeToPort["http"] = 80;
_schemeToPort["rtmpt"] = 80;
_schemeToPort["rtmpte"] = 80;
_schemeToPort["https"] = 443;
_schemeToPort["rtmps"] = 443;
_schemeToPort["rtsp"] = 554;
_schemeToPort["rtmp"] = 1935;
_schemeToPort["rtmpe"] = 1935;
_schemeToPort["mms"] = 1755;
}
if (MAP_HAS1(_schemeToPort, scheme)) {
port = _schemeToPort[scheme];
}
// else {
// FATAL("Scheme `%s` not supported", STR(scheme));
// return false;
// }
LOG_URI("scheme: %s; default port: %"PRIu16, STR(scheme), port);
//3. get the authentication portion. the search starts from
//where the scheme detection left and up to the first / character
string::size_type limit = stringUri.find("/", cursor);
bool hasAuthentication = false;
pos = stringUri.find("@", cursor);
if (pos != string::npos) {
if (limit != string::npos) {
hasAuthentication = pos<limit;
}
hasAuthentication = true;
}
if (hasAuthentication) {
authentication = stringUri.substr(cursor, pos - cursor);
fullUri = stringUri.substr(0, cursor);
fullUri += stringUri.substr(pos + 1);
cursor = pos + 1;
} else {
fullUri = fullUriWithAuth;
}
if (authentication != "") {
pos = authentication.find(":");
if (pos != string::npos) {
username = authentication.substr(0, pos);
password = authentication.substr(pos + 1);
} else {
username = authentication;
password = "";
}
}
LOG_URI("fullUri: `%s`; fullUriWithAuth: `%s`", STR(fullUri), STR(fullUriWithAuth));
LOG_URI("username: `%s`; password: `%s`", STR(username), STR(password));
//4. Get the host:port
pos = stringUri.find("/", cursor);
if (pos == string::npos) {
hostPort = stringUri.substr(cursor);
cursor = stringUri.size() - 1;
fullDocumentPathWithParameters = "/";
} else {
hostPort = stringUri.substr(cursor, pos - cursor);
cursor = pos + 1;
fullDocumentPathWithParameters = "/" + stringUri.substr(cursor);
}
trim(hostPort);
if (hostPort == "") {
FATAL("Invalid host:port specified");
return false;
}
pos = hostPort.find(":");
if (pos == string::npos) {
host = hostPort;
portSpecified = false;
} else {
host = hostPort.substr(0, pos);
trim(host);
portString = hostPort.substr(pos + 1);
portSpecified = true;
port = (uint16_t) atoi(STR(portString));
if (format("%"PRIu16, port) != portString) {
FATAL("Invalid port number specified: `%s`", STR(portString));
return false;
}
}
LOG_URI("host: %s; port: %"PRIu16"; portSpecified: %d", STR(host), port, portSpecified);
//5. fullDocumentPathWithParameters
fullDocumentPath = "/";
fullParameters = "";
documentPath = "/";
document = "";
documentWithFullParameters = "";
parameters.Reset();
parameters.IsArray(false);
if (fullDocumentPathWithParameters != "/") {
pos = fullDocumentPathWithParameters.find("?");
if (pos == string::npos) {
fullDocumentPath = fullDocumentPathWithParameters;
fullParameters = "";
} else {
fullDocumentPath = fullDocumentPathWithParameters.substr(0, pos);
fullParameters = fullDocumentPathWithParameters.substr(pos + 1);
}
trim(fullParameters);
if (fullParameters != "") {
vector<string> elements;
split(fullParameters, "&", elements);
for (uint32_t i = 0; i < elements.size(); i++) {
string kvp = elements[i];
if (kvp == "")
continue;
string k = "";
string v = "";
pos = kvp.find("=");
if (pos == string::npos) {
k = kvp;
v = "";
} else {
k = kvp.substr(0, pos);
v = kvp.substr(pos + 1);
}
if (k == "")
continue;
parameters[k] = v;
}
}
for (string::size_type i = fullDocumentPath.size() - 1; i >= 0; i--) {
if (fullDocumentPath[i] == '/')
break;
document = fullDocumentPath[i] + document;
}
documentPath = fullDocumentPath.substr(0, fullDocumentPath.size() - document.size());
documentWithFullParameters = document;
if (fullParameters != "")
documentWithFullParameters += "?" + fullParameters;
}
LOG_URI("fullDocumentPathWithParameters: `%s`", STR(fullDocumentPathWithParameters));
LOG_URI("fullDocumentPath: `%s`", STR(fullDocumentPath));
LOG_URI("fullParameters: `%s`", STR(fullParameters));
LOG_URI("documentPath: `%s`", STR(documentPath));
LOG_URI("document: `%s`", STR(document));
LOG_URI("documentWithFullParameters: `%s`", STR(documentWithFullParameters));
LOG_URI("parameters:");
#ifdef DEBUG_URI
FOR_MAP(parameters, string, Variant, i) {
LOG_URI("\t`%s`: `%s`", STR(MAP_KEY(i)), STR(MAP_VAL(i)));
}
void ProtocolManager::UnRegisterProtocol(BaseProtocol *pProtocol) {
if (MAP_HAS1(_activeProtocols, pProtocol->GetId()))
_activeProtocols.erase(pProtocol->GetId());
if (MAP_HAS1(_deadProtocols, pProtocol->GetId()))
_deadProtocols.erase(pProtocol->GetId());
}
bool parseURI(string stringUri, URI &uri) {
LOG_URI_SPLIT("stringUri: %s", STR(stringUri));
uri.Reset();
trim(stringUri);
if (stringUri == "")
return false;
if (stringUri.size() > 1024)
return false;
uri.fullUri = stringUri;
LOG_URI_SPLIT("uri.fullUri: %s", STR(uri.fullUri));
// scheme://user:pwd@host:port/document/parts/here
vector<string> components;
split(stringUri, "/", components);
for (uint32_t i = 0; i < components.size(); i++) {
LOG_URI_SPLIT("%u: %s", i, STR(components[i]));
}
//0 - scheme:
if (components[0] == "")
return false;
if (components[0][components[0].size() - 1] != ':')
return false;
uri.scheme = lowerCase(components[0].substr(0, components[0].size() - 1));
LOG_URI_SPLIT("uri.scheme: %s", STR(uri.scheme));
//1 - nothing
if (components[1] != "")
return false;
//2 - user:pwd@host:port
vector<string> hostComponents;
if (components[2].find("@") != string::npos) {
split(components[2], "@", hostComponents);
if (hostComponents.size() != 2)
return false;
if ((hostComponents[0] == "")
|| (hostComponents[1] == "")) {
return false;
}
components[2] = hostComponents[1];
vector<string> userNamePasswordComponents;
split(hostComponents[0], ":", userNamePasswordComponents);
if (userNamePasswordComponents.size() != 2)
return false;
if ((userNamePasswordComponents[0] == "")
|| (userNamePasswordComponents[1] == "")) {
return false;
}
uri.userName = userNamePasswordComponents[0];
LOG_URI_SPLIT("uri.userName: %s", STR(uri.userName));
uri.password = userNamePasswordComponents[1];
LOG_URI_SPLIT("uri.password: %s", STR(uri.password));
}
split(components[2], ":", hostComponents);
if (hostComponents.size() == 1) {
if (hostComponents[0] == "")
return false;
uri.host = hostComponents[0];
LOG_URI_SPLIT("uri.host: %s", STR(uri.host));
} else if (hostComponents.size() == 2) {
if ((hostComponents[0] == "")
|| (hostComponents[0] == ""))
return false;
uri.host = hostComponents[0];
LOG_URI_SPLIT("uri.host: %s", STR(uri.host));
int32_t port = atoi(STR(hostComponents[1]));
if ((port <= 0) || (port > 65535))
return false;
uri.port = (uint16_t) port;
} else {
return false;
}
if (uri.port == 0) {
if (___schemeToPort.size() == 0) {
___schemeToPort["http"] = 80;
___schemeToPort["rtmpt"] = 80;
___schemeToPort["rtmpte"] = 80;
___schemeToPort["https"] = 443;
___schemeToPort["rtmps"] = 443;
___schemeToPort["rtsp"] = 554;
___schemeToPort["rtmp"] = 1935;
___schemeToPort["rtmpe"] = 1935;
___schemeToPort["mms"] = 1755;
}
if (MAP_HAS1(___schemeToPort, uri.scheme))
uri.port = ___schemeToPort[uri.scheme];
else
return false;
}
LOG_URI_SPLIT("uri.port: %u", uri.port);
for (uint32_t i = 3; i < components.size(); i++) {
uri.fullDocumentPath += "/" + components[i];
}
LOG_URI_SPLIT("uri.fullDocumentPath: %s", STR(uri.fullDocumentPath));
uri.documentPath = "/";
for (uint32_t i = 3; i < components.size() - 1; i++) {
uri.documentPath += components[i];
if (i != components.size() - 2)
uri.documentPath += "/";
}
LOG_URI_SPLIT("uri.documentPath: %s", STR(uri.documentPath));
if ((components.size() - 1) >= 3) {
uri.document = components[components.size() - 1];
vector<string> documentComponents;
split(uri.document, "?", documentComponents);
if (documentComponents.size() == 2) {
uri.document = documentComponents[0];
map<string, string> params;
params = mapping(documentComponents[1], "&", "=", true);
FOR_MAP(params, string, string, i) {
uri.parameters[MAP_KEY(i)] = MAP_VAL(i);
LOG_URI_SPLIT("uri.parameters[\"%s\"]: %s",
STR(MAP_KEY(i)),
STR(MAP_VAL(i)));
}
void OutboundConnectivity::UnRegisterClient(uint32_t protocolId) {
if (MAP_HAS1(_clients, protocolId))
_clients.erase(protocolId);
}