public void PushMessage(IPipe pipe, IMessage message)
{
if (message is ResetMessage)
{
_timeStamper.Reset();
}
else if (message is StatusMessage)
{
StatusMessage statusMsg = message as StatusMessage;
_data.SendStatus(statusMsg.body as StatusASO);
}
else if (message is RtmpMessage)
{
// Make sure chunk size has been sent
if (!_chunkSizeSent)
SendChunkSize();
RtmpMessage rtmpMsg = message as RtmpMessage;
IRtmpEvent msg = rtmpMsg.body;
int eventTime = msg.Timestamp;
#if !SILVERLIGHT
if(log.IsDebugEnabled)
log.Debug(string.Format("Message timestamp: {0}", eventTime));
#endif
if (eventTime < 0)
{
#if !SILVERLIGHT
if (log.IsDebugEnabled)
log.Debug(string.Format("Message has negative timestamp: {0}", eventTime));
#endif
return;
}
byte dataType = msg.DataType;
// Create a new header for the consumer
RtmpHeader header = _timeStamper.GetTimeStamp(dataType, eventTime);
switch (msg.DataType)
{
case Constants.TypeStreamMetadata:
Notify notify = new Notify((msg as Notify).Data);
notify.Header = header;
notify.Timestamp = header.Timer;
_data.Write(notify);
break;
case Constants.TypeFlexStreamEnd:
// TODO: okay to send this also to AMF0 clients?
FlexStreamSend send = new FlexStreamSend((msg as Notify).Data);
send.Header = header;
send.Timestamp = header.Timer;
_data.Write(send);
break;
case Constants.TypeVideoData:
VideoData videoData = new VideoData((msg as VideoData).Data);
videoData.Header = header;
videoData.Timestamp = header.Timer;
_video.Write(videoData);
break;
case Constants.TypeAudioData:
AudioData audioData = new AudioData((msg as AudioData).Data);
audioData.Header = header;
audioData.Timestamp = header.Timer;
_audio.Write(audioData);
break;
case Constants.TypePing:
Ping ping = new Ping((msg as Ping).PingType, (msg as Ping).Value2, (msg as Ping).Value3, (msg as Ping).Value4);
ping.Header = header;
_connection.Ping(ping);
break;
case Constants.TypeBytesRead:
BytesRead bytesRead = new BytesRead((msg as BytesRead).Bytes);
bytesRead.Header = header;
bytesRead.Timestamp = header.Timer;
_connection.GetChannel((byte)2).Write(bytesRead);
break;
default:
_data.Write(msg);
break;
}
}
}
/// <summary>
/// Initializes a new instance of the <see cref="NetStreamAudioEventArgs"/> class.
/// </summary>
/// <param name="audioData">The audio data.</param>
internal NetStreamAudioEventArgs(AudioData audioData)
{
_audioData = audioData;
}
internal void RaiseNetStreamAudio(AudioData audioData)
{
if (_netStreamAudioHandler != null)
{
_netStreamAudioHandler(this, new NetStreamAudioEventArgs(audioData));
}
}
static ByteBuffer EncodeAudioData(RtmpContext context, AudioData audioData)
{
ByteBuffer output = audioData.Data;
return output;
}
public IMessage PullMessage(IPipe pipe)
{
lock (_syncLock)
{
if (_pipe != pipe)
return null;
if (_reader == null)
Init();
if (!_reader.HasMoreTags())
{
// TODO send OOBCM to notify EOF
// Do not unsubscribe as this kills VOD seek while in buffer
// this.pipe.unsubscribe(this);
return null;
}
ITag tag = _reader.ReadTag();
IRtmpEvent msg = null;
int timestamp = tag.Timestamp;
switch (tag.DataType)
{
case Constants.TypeAudioData:
msg = new AudioData(tag.Body);
break;
case Constants.TypeVideoData:
msg = new VideoData(tag.Body);
break;
case Constants.TypeInvoke:
msg = new Invoke(tag.Body);
break;
case Constants.TypeNotify:
msg = new Notify(tag.Body);
break;
case Constants.TypeFlexStreamEnd:
msg = new FlexStreamSend(tag.Body);
break;
default:
log.Warn("Unexpected type " + tag.DataType);
msg = new Unknown(tag.DataType, tag.Body);
break;
}
msg.Timestamp = timestamp;
RtmpMessage rtmpMsg = new RtmpMessage();
rtmpMsg.body = msg;
return rtmpMsg;
}
}
/// <summary>
/// Recieve then send if message is data (not audio or video)
/// </summary>
internal void PullAndPush()
{
lock (this.SyncRoot)
{
if (_playlistSubscriberStream.State == State.PLAYING && _isPullMode && !_waitingForToken)
{
if (_pendingMessage != null)
{
IRtmpEvent body = _pendingMessage.body;
if (!OkayToSendMessage(body))
return;
SendMessage(_pendingMessage);
ReleasePendingMessage();
}
else
{
while (true)
{
IMessage msg = _msgIn.PullMessage();
if (msg == null)
{
// No more packets to send
Stop();
break;
}
else
{
if (msg is RtmpMessage)
{
RtmpMessage rtmpMessage = (RtmpMessage)msg;
IRtmpEvent body = rtmpMessage.body;
if (!_receiveAudio && body is AudioData)
{
// The user doesn't want to get audio packets
//((IStreamData) body).Data.Release();
if (_sendBlankAudio)
{
// Send reset audio packet
_sendBlankAudio = false;
body = new AudioData();
// We need a zero timestamp
if (_lastMessage != null)
{
body.Timestamp = _lastMessage.Timestamp - _timestampOffset;
}
else
{
body.Timestamp = -_timestampOffset;
}
rtmpMessage.body = body;
}
else
{
continue;
}
}
else if (!_receiveVideo && body is VideoData)
{
// The user doesn't want to get video packets
//((IStreamData) body).Data.Release();
continue;
}
// Adjust timestamp when playing lists
body.Timestamp = body.Timestamp + _timestampOffset;
if (OkayToSendMessage(body))
{
if( log.IsDebugEnabled )
log.Debug(string.Format("ts: {0}", rtmpMessage.body.Timestamp));
SendMessage(rtmpMessage);
//((IStreamData) body).Data.Release();
}
else
{
_pendingMessage = rtmpMessage;
}
EnsurePullAndPushRunning();
break;
}
}
}
}
}
}
}
/// <summary>
/// Seek position in file
/// </summary>
/// <param name="position"></param>
public void Seek(int position)
{
lock (this.SyncRoot)
{
if (_playlistSubscriberStream.State != State.PLAYING && _playlistSubscriberStream.State != State.PAUSED && _playlistSubscriberStream.State != State.STOPPED)
{
throw new IllegalStateException();
}
if (!_isPullMode)
{
throw new NotSupportedException();
}
ReleasePendingMessage();
ClearWaitJobs();
_bwController.ResetBuckets(_bwContext);
_waitingForToken = false;
SendClearPing();
SendReset();
SendSeekStatus(_currentItem, position);
SendStartStatus(_currentItem);
int seekPos = SendVODSeekCM(_msgIn, position);
// We seeked to the nearest keyframe so use real timestamp now
if (seekPos == -1)
{
seekPos = position;
}
_playbackStart = System.Environment.TickCount - seekPos;
_playlistSubscriberStream.NotifyItemSeek(_currentItem, seekPos);
bool messageSent = false;
bool startPullPushThread = false;
if ((_playlistSubscriberStream.State == State.PAUSED || _playlistSubscriberStream.State == State.STOPPED) && SendCheckVideoCM(_msgIn))
{
// we send a single snapshot on pause.
// XXX we need to take BWC into account, for
// now send forcefully.
IMessage msg;
try
{
msg = _msgIn.PullMessage();
}
catch (Exception ex)
{
log.Error("Error while pulling message.", ex);
msg = null;
}
while (msg != null)
{
if (msg is RtmpMessage)
{
RtmpMessage rtmpMessage = (RtmpMessage)msg;
IRtmpEvent body = rtmpMessage.body;
if (body is VideoData && ((VideoData)body).FrameType == FrameType.Keyframe)
{
body.Timestamp = seekPos;
DoPushMessage(rtmpMessage);
//rtmpMessage.body.Release();
messageSent = true;
_lastMessage = body;
break;
}
}
try
{
msg = _msgIn.PullMessage();
}
catch (Exception ex)
{
log.Error("Error while pulling message.", ex);
msg = null;
}
}
}
else
{
startPullPushThread = true;
}
if (!messageSent)
{
// Send blank audio packet to notify client about new position
AudioData audio = new AudioData();
audio.Timestamp = seekPos;
audio.Header = new RtmpHeader();
audio.Header.Timer = seekPos;
audio.Header.IsTimerRelative = false;
RtmpMessage audioMessage = new RtmpMessage();
audioMessage.body = audio;
_lastMessage = audio;
DoPushMessage(audioMessage);
}
if (startPullPushThread)
{
EnsurePullAndPushRunning();
}
if (_playlistSubscriberStream.State != State.STOPPED && _currentItem.Length >= 0 && (position - _streamOffset) >= _currentItem.Length)
{
// Seeked after end of stream
Stop();
return;
}
}
}
public void PushMessage(IPipe pipe, IMessage message)
{
lock (this.SyncRoot)
{
if (message is ResetMessage)
{
SendReset();
return;
}
if (message is RtmpMessage)
{
RtmpMessage rtmpMessage = (RtmpMessage)message;
IRtmpEvent body = rtmpMessage.body;
if (!(body is IStreamData))
{
throw new ApplicationException("expected IStreamData but got " + body.GetType().FullName);
}
int size = ((IStreamData)body).Data.Limit;
if (body is VideoData)
{
IVideoStreamCodec videoCodec = null;
if (_msgIn is IBroadcastScope)
{
IClientBroadcastStream stream = ((IBroadcastScope)_msgIn).GetAttribute(Constants.BroadcastScopeStreamAttribute) as IClientBroadcastStream;
if (stream != null && stream.CodecInfo != null)
{
videoCodec = stream.CodecInfo.VideoCodec;
}
}
if (videoCodec == null || videoCodec.CanDropFrames)
{
if (_playlistSubscriberStream.State == State.PAUSED)
{
// The subscriber paused the video
_videoFrameDropper.DropPacket(rtmpMessage);
return;
}
// Only check for frame dropping if the codec supports it
long pendingVideos = GetPendingVideoMessageCount();
if (!_videoFrameDropper.CanSendPacket(rtmpMessage, pendingVideos))
{
// Drop frame as it depends on other frames that were dropped before.
return;
}
bool drop = !_videoBucket.AcquireToken(size, 0);
if (!_receiveVideo || drop)
{
// The client disabled video or the app doesn't have enough bandwidth
// allowed for this stream.
_videoFrameDropper.DropPacket(rtmpMessage);
return;
}
long[] writeDelta = GetWriteDelta();
if (pendingVideos > 1 /*|| writeDelta[0] > writeDelta[1]*/)
{
// We drop because the client has insufficient bandwidth.
long now = System.Environment.TickCount;
if (_bufferCheckInterval > 0 && now >= _nextCheckBufferUnderrun)
{
// Notify client about frame dropping (keyframe)
SendInsufficientBandwidthStatus(_currentItem);
_nextCheckBufferUnderrun = now + _bufferCheckInterval;
}
_videoFrameDropper.DropPacket(rtmpMessage);
return;
}
_videoFrameDropper.SendPacket(rtmpMessage);
}
}
else if (body is AudioData)
{
if (!_receiveAudio && _sendBlankAudio)
{
// Send blank audio packet to reset player
_sendBlankAudio = false;
body = new AudioData();
if (_lastMessage != null)
{
body.Timestamp = _lastMessage.Timestamp;
}
else
{
body.Timestamp = 0;
}
rtmpMessage.body = body;
}
else if (_playlistSubscriberStream.State == State.PAUSED || !_receiveAudio || !_audioBucket.AcquireToken(size, 0))
{
return;
}
}
if (body is IStreamData && ((IStreamData)body).Data != null)
{
_bytesSent += ((IStreamData)body).Data.Limit;
}
_lastMessage = body;
}
_msgOut.PushMessage(message);
}
}
