Ejemplo n.º 1
0
        private static byte[] ProbeResource(IResourceAccessor accessor, string arguments, int transcoderTimeout)
        {
            ProcessStartInfo startInfo = new ProcessStartInfo()
            {
                FileName               = FFMpegBinary.FFMpegPath,
                Arguments              = arguments,
                UseShellExecute        = false,
                CreateNoWindow         = true,
                RedirectStandardOutput = true,
                RedirectStandardError  = true,
                StandardOutputEncoding = Encoding.UTF8,
                StandardErrorEncoding  = Encoding.UTF8
            };

#if !TRANSCODE_CONSOLE_TEST
            using (ServiceRegistration.Get <IImpersonationService>().CheckImpersonationFor(accessor.CanonicalLocalResourcePath))
            {
                //Only when the server is running as a service it will have elevation rights
                using (ImpersonationProcess ffmpeg = new ImpersonationProcess {
                    StartInfo = startInfo
                })
                {
                    IntPtr userToken = IntPtr.Zero;
                    if (!ImpersonationHelper.GetTokenByProcess(out userToken, true))
                    {
                        return(null);
                    }
#else
            {
                {
                    Process ffmpeg = new Process()
                    {
                        StartInfo = startInfo
                    };
#endif
#if !TRANSCODE_CONSOLE_TEST
                    ffmpeg.StartAsUser(userToken);
#else
                    ffmpeg.Start();
#endif
                    ffmpeg.BeginErrorReadLine();

                    var stream = ffmpeg.StandardOutput.BaseStream;
                    if (!ffmpeg.WaitForExit(transcoderTimeout))
                    {
                        ffmpeg.Kill();
                        stream.Dispose();
                        return(null);
                    }

                    ffmpeg.Close();
                    byte[] data = ReadToEnd(stream);
                    stream.Dispose();
#if !TRANSCODE_CONSOLE_TEST
                    NativeMethods.CloseHandle(userToken);
#endif
                    return(data);
                }
            }
        }
        private async void TranscodeProcessor(object args)
        {
            FFMpegTranscodeThreadData data = (FFMpegTranscodeThreadData)args;
            bool isStream = data.Context.Live && !data.Context.Segmented;
            bool isFile   = true;
            bool isSlimTv = false;

            try
            {
                if (data.Context.Segmented == true)
                {
                    await FFMpegPlaylistManifest.CreatePlaylistFilesAsync(data.TranscodeData).ConfigureAwait(false);
                }

                int  liveChannelId = 0;
                bool runProcess    = true;
                using (IResourceAccessor mediaAccessor = data.TranscodeData.GetFirstResourceAccessor())
                {
                    if (mediaAccessor is ITranscodeLiveAccessor tla)
                    {
                        isSlimTv      = true;
                        liveChannelId = tla.ChannelId;
                    }
                }

                data.Context.Start();
                data.TranscodeData.ClearRuntimeResourcePaths();
                int exitCode = -1;

                string identifier = "Transcode_" + data.TranscodeData.ClientId;
                if (isSlimTv)
                {
                    var result = await _slimtTvHandler.StartTuningAsync(identifier, liveChannelId).ConfigureAwait(false);

                    if (!result.Success)
                    {
                        _logger.Error("FFMpegMediaConverter: Transcoder unable to start timeshifting for channel {0}", liveChannelId);
                        runProcess = false;
                        exitCode   = 5000;
                    }
                    else
                    {
                        using (var mediaAccessor = await _slimtTvHandler.GetDefaultAccessorAsync(liveChannelId).ConfigureAwait(false))
                        {
                            if (mediaAccessor is INetworkResourceAccessor)
                            {
                                int mediaStreamIndex = data.TranscodeData.FirstResourceIndex;
                                data.TranscodeData.AddRuntimeResourcePath(ResourcePath.Deserialize(data.TranscodeData.InputMediaFilePaths[mediaStreamIndex]), mediaAccessor.CanonicalLocalResourcePath.Serialize());
                            }
                            else
                            {
                                _logger.Error("FFMpegMediaConverter: Transcoder unable to start timeshifting for channel {0} because no URL was found", liveChannelId);
                                runProcess = false;
                                exitCode   = 5001;
                            }
                        }
                    }
                }

                ResourcePath firstPath;
                using (var mediaAccessor = data.TranscodeData.GetFirstResourceAccessor())
                {
                    firstPath = mediaAccessor.CanonicalLocalResourcePath;
                    if (mediaAccessor is INetworkResourceAccessor)
                    {
                        isFile = false;
                    }
                }

                if (runProcess)
                {
                    //Prepare resources
                    List <IDisposable> disposables = new List <IDisposable>();
                    foreach (var res in data.TranscodeData.GetResourceAccessors())
                    {
                        if (!(res is INetworkResourceAccessor))
                        {
                            var rah = new LocalFsResourceAccessorHelper(res);
                            data.TranscodeData.AddRuntimeResourcePath(res.CanonicalLocalResourcePath, rah.LocalFsResourceAccessor.LocalFileSystemPath);
                            var accessToken = rah.LocalFsResourceAccessor.EnsureLocalFileSystemAccess();
                            if (accessToken != null)
                            {
                                disposables.Add(accessToken);
                            }
                            disposables.Add(rah);
                        }

                        disposables.Add(res);
                    }

                    ProcessStartInfo startInfo = new ProcessStartInfo()
                    {
                        FileName               = data.TranscodeData.TranscoderBinPath,
                        WorkingDirectory       = data.TranscodeData.WorkPath,
                        Arguments              = data.TranscodeData.TranscoderArguments,
                        UseShellExecute        = false,
                        CreateNoWindow         = true,
                        RedirectStandardOutput = true,
                        RedirectStandardError  = true,
                        RedirectStandardInput  = true,
                        StandardOutputEncoding = Encoding.UTF8,
                        StandardErrorEncoding  = Encoding.UTF8
                    };

                    _logger.Debug("FFMpegMediaConverter: Transcoder '{0}' invoked with command line arguments '{1}'", data.TranscodeData.TranscoderBinPath, data.TranscodeData.TranscoderArguments);

                    IntPtr userToken = IntPtr.Zero;
                    try
                    {
                        //TODO: Fix usages of obsolete and deprecated methods when alternative is available
#if !TRANSCODE_CONSOLE_TEST
                        using (ServiceRegistration.Get <IImpersonationService>().CheckImpersonationFor(firstPath))
                        {
                            //Only when the server is running as a service it will have elevation rights
                            using (ImpersonationProcess ffmpeg = new ImpersonationProcess {
                                StartInfo = startInfo
                            })
                            {
                                if (isFile && !ImpersonationHelper.GetTokenByProcess(out userToken, true))
                                {
                                    return;
                                }
#else
                        {
                            using (Process ffmpeg = new Process()
                            {
                                StartInfo = startInfo
                            })
                            {
#endif
                                ffmpeg.EnableRaisingEvents = true; //Enable raising events because Process does not raise events by default.
                                if (isStream == false)
                                {
                                    ffmpeg.OutputDataReceived += data.Context.OutputDataReceived;
                                }

                                ffmpeg.ErrorDataReceived += data.Context.ErrorDataReceived;
#if !TRANSCODE_CONSOLE_TEST
                                if (isFile)
                                {
                                    ffmpeg.StartAsUser(userToken);
                                }
                                else
                                {
                                    ffmpeg.Start();
                                }
#else
                                ffmpeg.Start();
#endif
                                ffmpeg.BeginErrorReadLine();
                                if (isStream == false)
                                {
                                    ffmpeg.BeginOutputReadLine();
                                }
                                else
                                {
                                    data.TranscodeData.LiveStream = ffmpeg.StandardOutput.BaseStream;
                                }

                                RunTranscodingProcess(ffmpeg, data, isStream);

                                ffmpeg.WaitForExit();
                                exitCode = ffmpeg.ExitCode;
                                ffmpeg.Close();
                            }
                        }
                    }
                    catch (Exception ex)
                    {
                        if (isStream || data.TranscodeData.OutputFilePath == null)
                        {
                            _logger.Error("FFMpegMediaConverter: Transcoder command failed for stream '{0}'", ex, data.TranscodeData.TranscodeId);
                        }
                        else
                        {
                            _logger.Error("FFMpegMediaConverter: Transcoder command failed for file '{0}'", ex, data.TranscodeData.OutputFilePath);
                        }

                        data.Context.Fail();
                    }
                    finally
                    {
#if !TRANSCODE_CONSOLE_TEST
                        if (userToken != IntPtr.Zero)
                        {
                            NativeMethods.CloseHandle(userToken);
                        }
#endif
                        data.TranscodeData.LiveStream?.Dispose();
                        foreach (var disposable in disposables)
                        {
                            disposable?.Dispose();
                        }
                    }
                }

                if (exitCode > 0)
                {
                    data.Context.Fail();
                }
                else
                {
                    data.Context.Stop();
                }
                _ffMpegEncoderHandler.EndEncoding(data.TranscodeData.Encoder, data.TranscodeData.TranscodeId);

                if (isSlimTv)
                {
                    if (await _slimtTvHandler.EndTuningAsync(identifier).ConfigureAwait(false) == false)
                    {
                        _logger.Error("FFMpegMediaConverter: Transcoder unable to stop timeshifting for channel {0}", liveChannelId);
                    }
                }

                string filePath = data.Context.TargetFile;
                bool   isFolder = false;
                if (string.IsNullOrEmpty(data.Context.SegmentDir) == false)
                {
                    filePath = data.Context.SegmentDir;
                    isFolder = true;
                }

                if (exitCode > 0 || data.Context.Aborted)
                {
                    if (exitCode > 0)
                    {
                        if (isStream || data.TranscodeData.OutputFilePath == null)
                        {
                            _logger.Debug("FFMpegMediaConverter: Transcoder command failed with error {1} for stream '{0}'", data.TranscodeData.TranscodeId, exitCode);
                        }
                        else
                        {
                            _logger.Debug("FFMpegMediaConverter: Transcoder command failed with error {1} for file '{0}'", data.TranscodeData.OutputFilePath, exitCode);
                        }
                    }

                    if (data.Context.Aborted)
                    {
                        if (isStream || data.TranscodeData.OutputFilePath == null)
                        {
                            _logger.Debug("FFMpegMediaConverter: Transcoder command aborted for stream '{0}'", data.TranscodeData.TranscodeId);
                        }
                        else
                        {
                            _logger.Debug("FFMpegMediaConverter: Transcoder command aborted for file '{0}'", data.TranscodeData.OutputFilePath);
                        }
                    }
                    else
                    {
                        _logger.Debug("FFMpegMediaConverter: FFMpeg error \n {0}", data.Context.ConsoleErrorOutput);
                    }

                    data.Context.DeleteFiles();
                }
                else
                {
                    //Touch cache files so they will not be cleaned up
                    if (isFolder == false)
                    {
                        TouchFile(filePath);
                    }
                    else
                    {
                        TouchDirectory(filePath);
                    }
                }
            }
            catch (Exception ex)
            {
                if (isStream || data.TranscodeData.OutputFilePath == null)
                {
                    _logger.Error("FFMpegMediaConverter: Transcoder failed processing '{0}'", ex, data.TranscodeData.TranscodeId);
                }
                else
                {
                    _logger.Error("FFMpegMediaConverter: Transcoder failed processing file '{0}'", ex, data.TranscodeData.OutputFilePath);
                }
            }
            finally
            {
                await RemoveTranscodeContextAsync(data.TranscodeData.ClientId, data.TranscodeData.TranscodeId, data.Context).ConfigureAwait(false);
            }
        }