MainStartup.cs 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884
  1. using MediaBrowser.Model.Logging;
  2. using MediaBrowser.Server.Implementations;
  3. using MediaBrowser.Server.Startup.Common;
  4. using MediaBrowser.ServerApplication.Native;
  5. using MediaBrowser.ServerApplication.Splash;
  6. using MediaBrowser.ServerApplication.Updates;
  7. using Microsoft.Win32;
  8. using System;
  9. using System.Configuration.Install;
  10. using System.Diagnostics;
  11. using System.IO;
  12. using System.Linq;
  13. using System.Management;
  14. using System.Runtime.InteropServices;
  15. using System.ServiceProcess;
  16. using System.Text;
  17. using System.Threading;
  18. using System.Threading.Tasks;
  19. using System.Windows.Forms;
  20. using Emby.Common.Implementations.EnvironmentInfo;
  21. using Emby.Common.Implementations.IO;
  22. using Emby.Common.Implementations.Logging;
  23. using Emby.Common.Implementations.Networking;
  24. using Emby.Common.Implementations.Security;
  25. using Emby.Server.Core;
  26. using Emby.Server.Core.Logging;
  27. using Emby.Server.Implementations;
  28. using Emby.Server.Implementations.Browser;
  29. using Emby.Server.Implementations.IO;
  30. using Emby.Server.Implementations.Logging;
  31. using ImageMagickSharp;
  32. using MediaBrowser.Common.Net;
  33. using MediaBrowser.Model.IO;
  34. using MediaBrowser.Server.Startup.Common.IO;
  35. namespace MediaBrowser.ServerApplication
  36. {
  37. public class MainStartup
  38. {
  39. private static ApplicationHost _appHost;
  40. private static ILogger _logger;
  41. public static bool IsRunningAsService = false;
  42. private static bool _canRestartService = false;
  43. private static bool _appHostDisposed;
  44. [DllImport("kernel32.dll", SetLastError = true)]
  45. static extern bool SetDllDirectory(string lpPathName);
  46. public static string ApplicationPath;
  47. private static IFileSystem FileSystem;
  48. public static bool TryGetLocalFromUncDirectory(string local, out string unc)
  49. {
  50. if ((local == null) || (local == ""))
  51. {
  52. unc = "";
  53. throw new ArgumentNullException("local");
  54. }
  55. ManagementObjectSearcher searcher = new ManagementObjectSearcher("SELECT Name FROM Win32_share WHERE path ='" + local.Replace("\\", "\\\\") + "'");
  56. ManagementObjectCollection coll = searcher.Get();
  57. if (coll.Count == 1)
  58. {
  59. foreach (ManagementObject share in searcher.Get())
  60. {
  61. unc = share["Name"] as String;
  62. unc = "\\\\" + SystemInformation.ComputerName + "\\" + unc;
  63. return true;
  64. }
  65. }
  66. unc = "";
  67. return false;
  68. }
  69. /// <summary>
  70. /// Defines the entry point of the application.
  71. /// </summary>
  72. public static void Main()
  73. {
  74. var options = new StartupOptions(Environment.GetCommandLineArgs());
  75. IsRunningAsService = options.ContainsOption("-service");
  76. if (IsRunningAsService)
  77. {
  78. //_canRestartService = CanRestartWindowsService();
  79. }
  80. var currentProcess = Process.GetCurrentProcess();
  81. ApplicationPath = currentProcess.MainModule.FileName;
  82. var architecturePath = Path.Combine(Path.GetDirectoryName(ApplicationPath), Environment.Is64BitProcess ? "x64" : "x86");
  83. Wand.SetMagickCoderModulePath(architecturePath);
  84. var success = SetDllDirectory(architecturePath);
  85. SQLitePCL.raw.SetProvider(new SQLitePCL.SQLite3Provider_sqlite3());
  86. var appPaths = CreateApplicationPaths(ApplicationPath, IsRunningAsService);
  87. var logManager = new NlogManager(appPaths.LogDirectoryPath, "server");
  88. logManager.ReloadLogger(LogSeverity.Debug);
  89. logManager.AddConsoleOutput();
  90. var logger = _logger = logManager.GetLogger("Main");
  91. ApplicationHost.LogEnvironmentInfo(logger, appPaths, true);
  92. // Install directly
  93. if (options.ContainsOption("-installservice"))
  94. {
  95. logger.Info("Performing service installation");
  96. InstallService(ApplicationPath, logger);
  97. return;
  98. }
  99. // Restart with admin rights, then install
  100. if (options.ContainsOption("-installserviceasadmin"))
  101. {
  102. logger.Info("Performing service installation");
  103. RunServiceInstallation(ApplicationPath);
  104. return;
  105. }
  106. // Uninstall directly
  107. if (options.ContainsOption("-uninstallservice"))
  108. {
  109. logger.Info("Performing service uninstallation");
  110. UninstallService(ApplicationPath, logger);
  111. return;
  112. }
  113. // Restart with admin rights, then uninstall
  114. if (options.ContainsOption("-uninstallserviceasadmin"))
  115. {
  116. logger.Info("Performing service uninstallation");
  117. RunServiceUninstallation(ApplicationPath);
  118. return;
  119. }
  120. AppDomain.CurrentDomain.UnhandledException += CurrentDomain_UnhandledException;
  121. RunServiceInstallationIfNeeded(ApplicationPath);
  122. if (IsAlreadyRunning(ApplicationPath, currentProcess))
  123. {
  124. logger.Info("Shutting down because another instance of Emby Server is already running.");
  125. return;
  126. }
  127. if (PerformUpdateIfNeeded(appPaths, logger))
  128. {
  129. logger.Info("Exiting to perform application update.");
  130. return;
  131. }
  132. try
  133. {
  134. RunApplication(appPaths, logManager, IsRunningAsService, options);
  135. }
  136. finally
  137. {
  138. OnServiceShutdown();
  139. }
  140. }
  141. /// <summary>
  142. /// Determines whether [is already running] [the specified current process].
  143. /// </summary>
  144. /// <param name="applicationPath">The application path.</param>
  145. /// <param name="currentProcess">The current process.</param>
  146. /// <returns><c>true</c> if [is already running] [the specified current process]; otherwise, <c>false</c>.</returns>
  147. private static bool IsAlreadyRunning(string applicationPath, Process currentProcess)
  148. {
  149. var duplicate = Process.GetProcesses().FirstOrDefault(i =>
  150. {
  151. try
  152. {
  153. if (currentProcess.Id == i.Id)
  154. {
  155. return false;
  156. }
  157. }
  158. catch (Exception)
  159. {
  160. return false;
  161. }
  162. try
  163. {
  164. //_logger.Info("Module: {0}", i.MainModule.FileName);
  165. if (string.Equals(applicationPath, i.MainModule.FileName, StringComparison.OrdinalIgnoreCase))
  166. {
  167. return true;
  168. }
  169. return false;
  170. }
  171. catch (Exception)
  172. {
  173. return false;
  174. }
  175. });
  176. if (duplicate != null)
  177. {
  178. _logger.Info("Found a duplicate process. Giving it time to exit.");
  179. if (!duplicate.WaitForExit(40000))
  180. {
  181. _logger.Info("The duplicate process did not exit.");
  182. return true;
  183. }
  184. }
  185. if (!IsRunningAsService)
  186. {
  187. return IsAlreadyRunningAsService(applicationPath);
  188. }
  189. return false;
  190. }
  191. private static bool IsAlreadyRunningAsService(string applicationPath)
  192. {
  193. var serviceName = BackgroundService.GetExistingServiceName();
  194. WqlObjectQuery wqlObjectQuery = new WqlObjectQuery(string.Format("SELECT * FROM Win32_Service WHERE State = 'Running' AND Name = '{0}'", serviceName));
  195. ManagementObjectSearcher managementObjectSearcher = new ManagementObjectSearcher(wqlObjectQuery);
  196. ManagementObjectCollection managementObjectCollection = managementObjectSearcher.Get();
  197. foreach (ManagementObject managementObject in managementObjectCollection)
  198. {
  199. var obj = managementObject.GetPropertyValue("PathName");
  200. if (obj == null)
  201. {
  202. continue;
  203. }
  204. var path = obj.ToString();
  205. _logger.Info("Service path: {0}", path);
  206. // Need to use indexOf instead of equality because the path will have the full service command line
  207. if (path.IndexOf(applicationPath, StringComparison.OrdinalIgnoreCase) != -1)
  208. {
  209. _logger.Info("The windows service is already running");
  210. MessageBox.Show("Emby Server is already running as a Windows Service. Only one instance is allowed at a time. To run as a tray icon, shut down the Windows Service.");
  211. return true;
  212. }
  213. }
  214. return false;
  215. }
  216. /// <summary>
  217. /// Creates the application paths.
  218. /// </summary>
  219. /// <param name="applicationPath">The application path.</param>
  220. /// <param name="runAsService">if set to <c>true</c> [run as service].</param>
  221. /// <returns>ServerApplicationPaths.</returns>
  222. private static ServerApplicationPaths CreateApplicationPaths(string applicationPath, bool runAsService)
  223. {
  224. var appFolderPath = Path.GetDirectoryName(applicationPath);
  225. var resourcesPath = Path.GetDirectoryName(applicationPath);
  226. Action<string> createDirectoryFn = s => Directory.CreateDirectory(s);
  227. if (runAsService)
  228. {
  229. var systemPath = Path.GetDirectoryName(applicationPath);
  230. var programDataPath = Path.GetDirectoryName(systemPath);
  231. return new ServerApplicationPaths(programDataPath, appFolderPath, resourcesPath, createDirectoryFn);
  232. }
  233. return new ServerApplicationPaths(ApplicationPathHelper.GetProgramDataPath(applicationPath), appFolderPath, resourcesPath, createDirectoryFn);
  234. }
  235. /// <summary>
  236. /// Gets a value indicating whether this instance can self restart.
  237. /// </summary>
  238. /// <value><c>true</c> if this instance can self restart; otherwise, <c>false</c>.</value>
  239. public static bool CanSelfRestart
  240. {
  241. get
  242. {
  243. if (IsRunningAsService)
  244. {
  245. return _canRestartService;
  246. }
  247. else
  248. {
  249. return true;
  250. }
  251. }
  252. }
  253. /// <summary>
  254. /// Gets a value indicating whether this instance can self update.
  255. /// </summary>
  256. /// <value><c>true</c> if this instance can self update; otherwise, <c>false</c>.</value>
  257. public static bool CanSelfUpdate
  258. {
  259. get
  260. {
  261. #if DEBUG
  262. return false;
  263. #endif
  264. if (IsRunningAsService)
  265. {
  266. return _canRestartService;
  267. }
  268. else
  269. {
  270. return true;
  271. }
  272. }
  273. }
  274. private static readonly TaskCompletionSource<bool> ApplicationTaskCompletionSource = new TaskCompletionSource<bool>();
  275. /// <summary>
  276. /// Runs the application.
  277. /// </summary>
  278. /// <param name="appPaths">The app paths.</param>
  279. /// <param name="logManager">The log manager.</param>
  280. /// <param name="runService">if set to <c>true</c> [run service].</param>
  281. /// <param name="options">The options.</param>
  282. private static void RunApplication(ServerApplicationPaths appPaths, ILogManager logManager, bool runService, StartupOptions options)
  283. {
  284. var environmentInfo = new EnvironmentInfo();
  285. var fileSystem = new ManagedFileSystem(logManager.GetLogger("FileSystem"), environmentInfo, appPaths.TempDirectory);
  286. var imageEncoder = ImageEncoderHelper.GetImageEncoder(_logger, logManager, fileSystem, options, () => _appHost.HttpClient, appPaths);
  287. FileSystem = fileSystem;
  288. _appHost = new WindowsAppHost(appPaths,
  289. logManager,
  290. options,
  291. fileSystem,
  292. new PowerManagement(),
  293. "emby.windows.zip",
  294. environmentInfo,
  295. imageEncoder,
  296. new Server.Startup.Common.SystemEvents(logManager.GetLogger("SystemEvents")),
  297. new RecyclableMemoryStreamProvider(),
  298. new Networking.NetworkManager(logManager.GetLogger("NetworkManager")),
  299. GenerateCertificate,
  300. () => Environment.UserDomainName);
  301. var initProgress = new Progress<double>();
  302. if (!runService)
  303. {
  304. if (!options.ContainsOption("-nosplash")) ShowSplashScreen(_appHost.ApplicationVersion, initProgress, logManager.GetLogger("Splash"));
  305. // Not crazy about this but it's the only way to suppress ffmpeg crash dialog boxes
  306. SetErrorMode(ErrorModes.SEM_FAILCRITICALERRORS | ErrorModes.SEM_NOALIGNMENTFAULTEXCEPT |
  307. ErrorModes.SEM_NOGPFAULTERRORBOX | ErrorModes.SEM_NOOPENFILEERRORBOX);
  308. }
  309. var task = _appHost.Init(initProgress);
  310. Task.WaitAll(task);
  311. task = task.ContinueWith(new Action<Task>(a => _appHost.RunStartupTasks()), TaskContinuationOptions.OnlyOnRanToCompletion | TaskContinuationOptions.AttachedToParent);
  312. if (runService)
  313. {
  314. StartService(logManager);
  315. }
  316. else
  317. {
  318. Task.WaitAll(task);
  319. task = InstallVcredist2013IfNeeded(_appHost, _logger);
  320. Task.WaitAll(task);
  321. Microsoft.Win32.SystemEvents.SessionSwitch += SystemEvents_SessionSwitch;
  322. HideSplashScreen();
  323. ShowTrayIcon();
  324. task = ApplicationTaskCompletionSource.Task;
  325. Task.WaitAll(task);
  326. }
  327. }
  328. private static void GenerateCertificate(string certPath, string certHost, string certPassword)
  329. {
  330. CertificateGenerator.CreateSelfSignCertificatePfx(certPath, certHost, certPassword, _logger);
  331. }
  332. private static ServerNotifyIcon _serverNotifyIcon;
  333. private static TaskScheduler _mainTaskScheduler;
  334. private static void ShowTrayIcon()
  335. {
  336. //Application.EnableVisualStyles();
  337. //Application.SetCompatibleTextRenderingDefault(false);
  338. _serverNotifyIcon = new ServerNotifyIcon(_appHost.LogManager, _appHost, _appHost.ServerConfigurationManager, _appHost.LocalizationManager);
  339. _mainTaskScheduler = TaskScheduler.FromCurrentSynchronizationContext();
  340. Application.Run();
  341. }
  342. internal static SplashForm _splash;
  343. private static Thread _splashThread;
  344. private static void ShowSplashScreen(Version appVersion, Progress<double> progress, ILogger logger)
  345. {
  346. var thread = new Thread(() =>
  347. {
  348. _splash = new SplashForm(appVersion, progress);
  349. _splash.ShowDialog();
  350. });
  351. thread.SetApartmentState(ApartmentState.STA);
  352. thread.IsBackground = true;
  353. thread.Start();
  354. _splashThread = thread;
  355. }
  356. private static void HideSplashScreen()
  357. {
  358. if (_splash != null)
  359. {
  360. Action act = () =>
  361. {
  362. _splash.Close();
  363. _splashThread = null;
  364. };
  365. _splash.Invoke(act);
  366. }
  367. }
  368. static void SystemEvents_SessionSwitch(object sender, SessionSwitchEventArgs e)
  369. {
  370. if (e.Reason == SessionSwitchReason.SessionLogon)
  371. {
  372. BrowserLauncher.OpenDashboard(_appHost);
  373. }
  374. }
  375. public static void Invoke(Action action)
  376. {
  377. if (IsRunningAsService)
  378. {
  379. action();
  380. }
  381. else
  382. {
  383. Task.Factory.StartNew(action, CancellationToken.None, TaskCreationOptions.None, _mainTaskScheduler ?? TaskScheduler.Current);
  384. }
  385. }
  386. /// <summary>
  387. /// Starts the service.
  388. /// </summary>
  389. private static void StartService(ILogManager logManager)
  390. {
  391. var service = new BackgroundService(logManager.GetLogger("Service"));
  392. service.Disposed += service_Disposed;
  393. ServiceBase.Run(service);
  394. }
  395. /// <summary>
  396. /// Handles the Disposed event of the service control.
  397. /// </summary>
  398. /// <param name="sender">The source of the event.</param>
  399. /// <param name="e">The <see cref="EventArgs"/> instance containing the event data.</param>
  400. static void service_Disposed(object sender, EventArgs e)
  401. {
  402. ApplicationTaskCompletionSource.SetResult(true);
  403. OnServiceShutdown();
  404. }
  405. private static void OnServiceShutdown()
  406. {
  407. _logger.Info("Shutting down");
  408. DisposeAppHost();
  409. }
  410. /// <summary>
  411. /// Installs the service.
  412. /// </summary>
  413. private static void InstallService(string applicationPath, ILogger logger)
  414. {
  415. try
  416. {
  417. ManagedInstallerClass.InstallHelper(new[] { applicationPath });
  418. logger.Info("Service installation succeeded");
  419. }
  420. catch (Exception ex)
  421. {
  422. logger.ErrorException("Uninstall failed", ex);
  423. }
  424. }
  425. /// <summary>
  426. /// Uninstalls the service.
  427. /// </summary>
  428. private static void UninstallService(string applicationPath, ILogger logger)
  429. {
  430. try
  431. {
  432. ManagedInstallerClass.InstallHelper(new[] { "/u", applicationPath });
  433. logger.Info("Service uninstallation succeeded");
  434. }
  435. catch (Exception ex)
  436. {
  437. logger.ErrorException("Uninstall failed", ex);
  438. }
  439. }
  440. private static void RunServiceInstallationIfNeeded(string applicationPath)
  441. {
  442. var serviceName = BackgroundService.GetExistingServiceName();
  443. var ctl = ServiceController.GetServices().FirstOrDefault(s => s.ServiceName == serviceName);
  444. if (ctl == null)
  445. {
  446. RunServiceInstallation(applicationPath);
  447. }
  448. }
  449. /// <summary>
  450. /// Runs the service installation.
  451. /// </summary>
  452. private static void RunServiceInstallation(string applicationPath)
  453. {
  454. var startInfo = new ProcessStartInfo
  455. {
  456. FileName = applicationPath,
  457. Arguments = "-installservice",
  458. CreateNoWindow = true,
  459. WindowStyle = ProcessWindowStyle.Hidden,
  460. Verb = "runas",
  461. ErrorDialog = false
  462. };
  463. using (var process = Process.Start(startInfo))
  464. {
  465. process.WaitForExit();
  466. }
  467. }
  468. /// <summary>
  469. /// Runs the service uninstallation.
  470. /// </summary>
  471. private static void RunServiceUninstallation(string applicationPath)
  472. {
  473. var startInfo = new ProcessStartInfo
  474. {
  475. FileName = applicationPath,
  476. Arguments = "-uninstallservice",
  477. CreateNoWindow = true,
  478. WindowStyle = ProcessWindowStyle.Hidden,
  479. Verb = "runas",
  480. ErrorDialog = false
  481. };
  482. using (var process = Process.Start(startInfo))
  483. {
  484. process.WaitForExit();
  485. }
  486. }
  487. /// <summary>
  488. /// Handles the UnhandledException event of the CurrentDomain control.
  489. /// </summary>
  490. /// <param name="sender">The source of the event.</param>
  491. /// <param name="e">The <see cref="UnhandledExceptionEventArgs"/> instance containing the event data.</param>
  492. static void CurrentDomain_UnhandledException(object sender, UnhandledExceptionEventArgs e)
  493. {
  494. var exception = (Exception)e.ExceptionObject;
  495. new UnhandledExceptionWriter(_appHost.ServerConfigurationManager.ApplicationPaths, _logger, _appHost.LogManager, FileSystem, new ConsoleLogger()).Log(exception);
  496. if (!IsRunningAsService)
  497. {
  498. MessageBox.Show("Unhandled exception: " + exception.Message);
  499. }
  500. if (!Debugger.IsAttached)
  501. {
  502. Environment.Exit(Marshal.GetHRForException(exception));
  503. }
  504. }
  505. /// <summary>
  506. /// Performs the update if needed.
  507. /// </summary>
  508. /// <param name="appPaths">The app paths.</param>
  509. /// <param name="logger">The logger.</param>
  510. /// <returns><c>true</c> if XXXX, <c>false</c> otherwise</returns>
  511. private static bool PerformUpdateIfNeeded(ServerApplicationPaths appPaths, ILogger logger)
  512. {
  513. // Not supported
  514. if (IsRunningAsService)
  515. {
  516. return false;
  517. }
  518. // Look for the existence of an update archive
  519. var updateArchive = Path.Combine(appPaths.TempUpdatePath, "MBServer" + ".zip");
  520. if (File.Exists(updateArchive))
  521. {
  522. logger.Info("An update is available from {0}", updateArchive);
  523. // Update is there - execute update
  524. try
  525. {
  526. var serviceName = IsRunningAsService ? BackgroundService.GetExistingServiceName() : string.Empty;
  527. new ApplicationUpdater().UpdateApplication(appPaths, updateArchive, logger, serviceName);
  528. // And just let the app exit so it can update
  529. return true;
  530. }
  531. catch (Exception e)
  532. {
  533. logger.ErrorException("Error starting updater.", e);
  534. MessageBox.Show(string.Format("Error attempting to update application.\n\n{0}\n\n{1}", e.GetType().Name, e.Message));
  535. }
  536. }
  537. return false;
  538. }
  539. public static void Shutdown()
  540. {
  541. if (IsRunningAsService)
  542. {
  543. ShutdownWindowsService();
  544. }
  545. else
  546. {
  547. DisposeAppHost();
  548. ShutdownWindowsApplication();
  549. }
  550. }
  551. public static void Restart()
  552. {
  553. DisposeAppHost();
  554. if (IsRunningAsService)
  555. {
  556. RestartWindowsService();
  557. }
  558. else
  559. {
  560. //_logger.Info("Hiding server notify icon");
  561. //_serverNotifyIcon.Visible = false;
  562. _logger.Info("Starting new instance");
  563. //Application.Restart();
  564. Process.Start(ApplicationPath);
  565. ShutdownWindowsApplication();
  566. }
  567. }
  568. private static void DisposeAppHost()
  569. {
  570. if (!_appHostDisposed)
  571. {
  572. _logger.Info("Disposing app host");
  573. _appHostDisposed = true;
  574. _appHost.Dispose();
  575. _logger.Info("App host dispose complete");
  576. }
  577. }
  578. private static void ShutdownWindowsApplication()
  579. {
  580. if (_serverNotifyIcon != null)
  581. {
  582. _serverNotifyIcon.Dispose();
  583. _serverNotifyIcon = null;
  584. }
  585. //_logger.Info("Calling Application.Exit");
  586. //Application.Exit();
  587. _logger.Info("Calling Environment.Exit");
  588. Environment.Exit(0);
  589. _logger.Info("Calling ApplicationTaskCompletionSource.SetResult");
  590. ApplicationTaskCompletionSource.SetResult(true);
  591. }
  592. private static void ShutdownWindowsService()
  593. {
  594. _logger.Info("Stopping background service");
  595. var service = new ServiceController(BackgroundService.GetExistingServiceName());
  596. service.Refresh();
  597. if (service.Status == ServiceControllerStatus.Running)
  598. {
  599. service.Stop();
  600. }
  601. }
  602. private static void RestartWindowsService()
  603. {
  604. _logger.Info("Restarting background service");
  605. var startInfo = new ProcessStartInfo
  606. {
  607. FileName = "cmd.exe",
  608. CreateNoWindow = true,
  609. WindowStyle = ProcessWindowStyle.Hidden,
  610. Verb = "runas",
  611. ErrorDialog = false,
  612. Arguments = String.Format("/c sc stop {0} & sc start {0} & sc start {0}", BackgroundService.GetExistingServiceName())
  613. };
  614. Process.Start(startInfo);
  615. }
  616. private static bool CanRestartWindowsService()
  617. {
  618. var startInfo = new ProcessStartInfo
  619. {
  620. FileName = "cmd.exe",
  621. CreateNoWindow = true,
  622. WindowStyle = ProcessWindowStyle.Hidden,
  623. Verb = "runas",
  624. ErrorDialog = false,
  625. Arguments = String.Format("/c sc query {0}", BackgroundService.GetExistingServiceName())
  626. };
  627. using (var process = Process.Start(startInfo))
  628. {
  629. process.WaitForExit();
  630. if (process.ExitCode == 0)
  631. {
  632. return true;
  633. }
  634. else
  635. {
  636. return false;
  637. }
  638. }
  639. }
  640. private static async Task InstallVcredist2013IfNeeded(ApplicationHost appHost, ILogger logger)
  641. {
  642. // Reference
  643. // http://stackoverflow.com/questions/12206314/detect-if-visual-c-redistributable-for-visual-studio-2012-is-installed
  644. try
  645. {
  646. var subkey = Environment.Is64BitProcess
  647. ? "SOFTWARE\\WOW6432Node\\Microsoft\\VisualStudio\\12.0\\VC\\Runtimes\\x64"
  648. : "SOFTWARE\\Microsoft\\VisualStudio\\12.0\\VC\\Runtimes\\x86";
  649. using (RegistryKey ndpKey = RegistryKey.OpenBaseKey(RegistryHive.LocalMachine, RegistryView.Default)
  650. .OpenSubKey(subkey))
  651. {
  652. if (ndpKey != null && ndpKey.GetValue("Version") != null)
  653. {
  654. var installedVersion = ((string)ndpKey.GetValue("Version")).TrimStart('v');
  655. if (installedVersion.StartsWith("12", StringComparison.OrdinalIgnoreCase))
  656. {
  657. return;
  658. }
  659. }
  660. }
  661. }
  662. catch (Exception ex)
  663. {
  664. logger.ErrorException("Error getting .NET Framework version", ex);
  665. return;
  666. }
  667. try
  668. {
  669. await InstallVcredist2013().ConfigureAwait(false);
  670. }
  671. catch (Exception ex)
  672. {
  673. logger.ErrorException("Error installing Visual Studio C++ runtime", ex);
  674. }
  675. }
  676. private async static Task InstallVcredist2013()
  677. {
  678. var httpClient = _appHost.HttpClient;
  679. var tmp = await httpClient.GetTempFile(new HttpRequestOptions
  680. {
  681. Url = GetVcredist2013Url(),
  682. Progress = new Progress<double>()
  683. }).ConfigureAwait(false);
  684. var exePath = Path.ChangeExtension(tmp, ".exe");
  685. File.Copy(tmp, exePath);
  686. var startInfo = new ProcessStartInfo
  687. {
  688. FileName = exePath,
  689. CreateNoWindow = true,
  690. WindowStyle = ProcessWindowStyle.Hidden,
  691. Verb = "runas",
  692. ErrorDialog = false
  693. };
  694. _logger.Info("Running {0}", startInfo.FileName);
  695. using (var process = Process.Start(startInfo))
  696. {
  697. process.WaitForExit();
  698. }
  699. }
  700. private static string GetVcredist2013Url()
  701. {
  702. if (Environment.Is64BitProcess)
  703. {
  704. return "https://github.com/MediaBrowser/Emby.Resources/raw/master/vcredist2013/vcredist_x64.exe";
  705. }
  706. // TODO: ARM url - https://github.com/MediaBrowser/Emby.Resources/raw/master/vcredist2013/vcredist_arm.exe
  707. return "https://github.com/MediaBrowser/Emby.Resources/raw/master/vcredist2013/vcredist_x86.exe";
  708. }
  709. /// <summary>
  710. /// Sets the error mode.
  711. /// </summary>
  712. /// <param name="uMode">The u mode.</param>
  713. /// <returns>ErrorModes.</returns>
  714. [DllImport("kernel32.dll")]
  715. static extern ErrorModes SetErrorMode(ErrorModes uMode);
  716. /// <summary>
  717. /// Enum ErrorModes
  718. /// </summary>
  719. [Flags]
  720. public enum ErrorModes : uint
  721. {
  722. /// <summary>
  723. /// The SYSTE m_ DEFAULT
  724. /// </summary>
  725. SYSTEM_DEFAULT = 0x0,
  726. /// <summary>
  727. /// The SE m_ FAILCRITICALERRORS
  728. /// </summary>
  729. SEM_FAILCRITICALERRORS = 0x0001,
  730. /// <summary>
  731. /// The SE m_ NOALIGNMENTFAULTEXCEPT
  732. /// </summary>
  733. SEM_NOALIGNMENTFAULTEXCEPT = 0x0004,
  734. /// <summary>
  735. /// The SE m_ NOGPFAULTERRORBOX
  736. /// </summary>
  737. SEM_NOGPFAULTERRORBOX = 0x0002,
  738. /// <summary>
  739. /// The SE m_ NOOPENFILEERRORBOX
  740. /// </summary>
  741. SEM_NOOPENFILEERRORBOX = 0x8000
  742. }
  743. }
  744. }