void WS::Unmask(BinaryReader& reader) {
UInt32 i;
UInt8 mask[4];
reader.read(sizeof(mask),mask);
UInt8* bytes = (UInt8*)reader.current();
for(i=0;i<reader.available();++i)
bytes[i] ^= mask[i%4];
}
void RTMFPSession::receive(const SocketAddress& address, BinaryReader& packet) {
if (!Session::receive(address, packet))
return;
// Read packet
UInt8 marker = packet.read8()|0xF0;
_timeSent = packet.read16();
// with time echo
if(marker == 0xFD) {
UInt16 time = RTMFP::TimeNow();
UInt16 timeEcho = packet.read16();
if(timeEcho>time) {
if(timeEcho-time<30)
time=0;
else
time += 0xFFFF-timeEcho;
timeEcho = 0;
}
peer.setPing((time-timeEcho)*RTMFP_TIMESTAMP_SCALE);
}
else if(marker != 0xF9)
WARN("RTMFPPacket marker unknown : ", Format<UInt8>("%02x",marker));
// Variables for request (0x10 and 0x11)
UInt8 flags;
RTMFPFlow* pFlow=NULL;
UInt64 stage=0;
UInt64 deltaNAck=0;
UInt8 type = packet.available()>0 ? packet.read8() : 0xFF;
bool answer = false;
// Can have nested queries
while(type!=0xFF) {
UInt16 size = packet.read16();
PacketReader message(packet.current(),size);
switch(type) {
case 0x0c :
fail("failed on client side");
break;
case 0x4c :
/// Session death!
_failed = true; // to avoid the fail signal!!
kill();
return;
/// KeepAlive
case 0x01 :
if(!peer.connected)
fail("Timeout connection client");
else
writeMessage(0x41,0);
case 0x41 :
_timesKeepalive=0;
break;
case 0x5e : {
// RTMFPFlow exception!
UInt64 id = message.read7BitLongValue();
RTMFPWriter* pRTMFPWriter = writer(id);
if(pRTMFPWriter)
pRTMFPWriter->fail("Writer rejected on session ",name());
else
WARN("RTMFPWriter ", id, " unfound for failed signal on session ", name());
break;
}
case 0x18 :
/// This response is sent when we answer with a Acknowledgment negative
// It contains the id flow
// I don't unsertand the usefulness...
//pFlow = &flow(message.read8());
//stage = pFlow->stageSnd();
// For the moment, we considerate it like a exception
fail("ack negative from server"); // send fail message immediatly
break;
case 0x51 : {
/// Acknowledgment
UInt64 id = message.read7BitLongValue();
RTMFPWriter* pRTMFPWriter = writer(id);
if(pRTMFPWriter)
pRTMFPWriter->acknowledgment(message);
else
WARN("RTMFPWriter ",id," unfound for acknowledgment on session ",name());
break;
}
/// Request
// 0x10 normal request
// 0x11 special request, in repeat case (following stage request)
case 0x10 : {
flags = message.read8();
UInt64 idFlow = message.read7BitLongValue();
stage = message.read7BitLongValue()-1;
deltaNAck = message.read7BitLongValue()-1;
if (_failed)
break;
map<UInt64,RTMFPFlow*>::const_iterator it = _flows.find(idFlow);
pFlow = it==_flows.end() ? NULL : it->second;
// Header part if present
if(flags & MESSAGE_HEADER) {
string signature;
message.read(message.read8(),signature);
if(!pFlow)
pFlow = createFlow(idFlow,signature);
if(message.read8()>0) {
// Fullduplex header part
if(message.read8()!=0x0A)
WARN("Unknown fullduplex header part for the flow ",idFlow)
else
message.read7BitLongValue(); // Fullduplex useless here! Because we are creating a new RTMFPFlow!
// Useless header part
UInt8 length=message.read8();
while(length>0 && message.available()) {
WARN("Unknown message part on flow ",idFlow);
message.next(length);
length=message.read8();
}
if(length>0)
ERROR("Bad header message part, finished before scheduled");
}
}
if(!pFlow) {
WARN("RTMFPFlow ",idFlow," unfound");
if (_pFlowNull)
((UInt64&)_pFlowNull->id) = idFlow;
pFlow = _pFlowNull;
}
}
case 0x11 : {
++stage;
++deltaNAck;
// has Header?
if(type==0x11)
flags = message.read8();
// Process request
if (pFlow && !_failed) {
bool wasConnected(peer.connected);
pFlow->receive(stage, deltaNAck, message, flags);
if (!wasConnected && peer.connected) {
for (auto& it : _flowWriters)
it.second->open();
}
}
break;
}
default :
ERROR("RTMFPMessage type '", Format<UInt8>("%02x", type), "' unknown");
}
// Next
packet.next(size);
type = packet.available()>0 ? packet.read8() : 0xFF;
// Commit RTMFPFlow (pFlow means 0x11 or 0x10 message)
if(pFlow && type!= 0x11) {
pFlow->commit();
if(pFlow->consumed()) {
if (pFlow->critical()) {
if (!peer.connected) {
// without connection, nothing must be sent!
for (auto& it : _flowWriters)
it.second->clear();
}
fail(); // If connection fails, log is already displayed, and so fail the whole session!
} else {
_flows.erase(pFlow->id);
delete pFlow;
}
}
pFlow=NULL;
}
}
