MainStartup.cs 30 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879
  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.Browser;
  27. using Emby.Server.Implementations.IO;
  28. using ImageMagickSharp;
  29. using MediaBrowser.Common.Net;
  30. using MediaBrowser.Server.Startup.Common.IO;
  31. namespace MediaBrowser.ServerApplication
  32. {
  33. public class MainStartup
  34. {
  35. private static ApplicationHost _appHost;
  36. private static ILogger _logger;
  37. public static bool IsRunningAsService = false;
  38. private static bool _canRestartService = false;
  39. private static bool _appHostDisposed;
  40. [DllImport("kernel32.dll", SetLastError = true)]
  41. static extern bool SetDllDirectory(string lpPathName);
  42. public static string ApplicationPath;
  43. public static bool TryGetLocalFromUncDirectory(string local, out string unc)
  44. {
  45. if ((local == null) || (local == ""))
  46. {
  47. unc = "";
  48. throw new ArgumentNullException("local");
  49. }
  50. ManagementObjectSearcher searcher = new ManagementObjectSearcher("SELECT Name FROM Win32_share WHERE path ='" + local.Replace("\\", "\\\\") + "'");
  51. ManagementObjectCollection coll = searcher.Get();
  52. if (coll.Count == 1)
  53. {
  54. foreach (ManagementObject share in searcher.Get())
  55. {
  56. unc = share["Name"] as String;
  57. unc = "\\\\" + SystemInformation.ComputerName + "\\" + unc;
  58. return true;
  59. }
  60. }
  61. unc = "";
  62. return false;
  63. }
  64. /// <summary>
  65. /// Defines the entry point of the application.
  66. /// </summary>
  67. public static void Main()
  68. {
  69. var options = new StartupOptions();
  70. IsRunningAsService = options.ContainsOption("-service");
  71. if (IsRunningAsService)
  72. {
  73. //_canRestartService = CanRestartWindowsService();
  74. }
  75. var currentProcess = Process.GetCurrentProcess();
  76. ApplicationPath = currentProcess.MainModule.FileName;
  77. var architecturePath = Path.Combine(Path.GetDirectoryName(ApplicationPath), Environment.Is64BitProcess ? "x64" : "x86");
  78. Wand.SetMagickCoderModulePath(architecturePath);
  79. var success = SetDllDirectory(architecturePath);
  80. var appPaths = CreateApplicationPaths(ApplicationPath, IsRunningAsService);
  81. var logManager = new NlogManager(appPaths.LogDirectoryPath, "server");
  82. logManager.ReloadLogger(LogSeverity.Debug);
  83. logManager.AddConsoleOutput();
  84. var logger = _logger = logManager.GetLogger("Main");
  85. ApplicationHost.LogEnvironmentInfo(logger, appPaths, true);
  86. // Install directly
  87. if (options.ContainsOption("-installservice"))
  88. {
  89. logger.Info("Performing service installation");
  90. InstallService(ApplicationPath, logger);
  91. return;
  92. }
  93. // Restart with admin rights, then install
  94. if (options.ContainsOption("-installserviceasadmin"))
  95. {
  96. logger.Info("Performing service installation");
  97. RunServiceInstallation(ApplicationPath);
  98. return;
  99. }
  100. // Uninstall directly
  101. if (options.ContainsOption("-uninstallservice"))
  102. {
  103. logger.Info("Performing service uninstallation");
  104. UninstallService(ApplicationPath, logger);
  105. return;
  106. }
  107. // Restart with admin rights, then uninstall
  108. if (options.ContainsOption("-uninstallserviceasadmin"))
  109. {
  110. logger.Info("Performing service uninstallation");
  111. RunServiceUninstallation(ApplicationPath);
  112. return;
  113. }
  114. AppDomain.CurrentDomain.UnhandledException += CurrentDomain_UnhandledException;
  115. RunServiceInstallationIfNeeded(ApplicationPath);
  116. if (IsAlreadyRunning(ApplicationPath, currentProcess))
  117. {
  118. logger.Info("Shutting down because another instance of Emby Server is already running.");
  119. return;
  120. }
  121. if (PerformUpdateIfNeeded(appPaths, logger))
  122. {
  123. logger.Info("Exiting to perform application update.");
  124. return;
  125. }
  126. try
  127. {
  128. RunApplication(appPaths, logManager, IsRunningAsService, options);
  129. }
  130. finally
  131. {
  132. OnServiceShutdown();
  133. }
  134. }
  135. /// <summary>
  136. /// Determines whether [is already running] [the specified current process].
  137. /// </summary>
  138. /// <param name="applicationPath">The application path.</param>
  139. /// <param name="currentProcess">The current process.</param>
  140. /// <returns><c>true</c> if [is already running] [the specified current process]; otherwise, <c>false</c>.</returns>
  141. private static bool IsAlreadyRunning(string applicationPath, Process currentProcess)
  142. {
  143. var duplicate = Process.GetProcesses().FirstOrDefault(i =>
  144. {
  145. try
  146. {
  147. if (currentProcess.Id == i.Id)
  148. {
  149. return false;
  150. }
  151. }
  152. catch (Exception)
  153. {
  154. return false;
  155. }
  156. try
  157. {
  158. //_logger.Info("Module: {0}", i.MainModule.FileName);
  159. if (string.Equals(applicationPath, i.MainModule.FileName, StringComparison.OrdinalIgnoreCase))
  160. {
  161. return true;
  162. }
  163. return false;
  164. }
  165. catch (Exception)
  166. {
  167. return false;
  168. }
  169. });
  170. if (duplicate != null)
  171. {
  172. _logger.Info("Found a duplicate process. Giving it time to exit.");
  173. if (!duplicate.WaitForExit(30000))
  174. {
  175. _logger.Info("The duplicate process did not exit.");
  176. return true;
  177. }
  178. }
  179. if (!IsRunningAsService)
  180. {
  181. return IsAlreadyRunningAsService(applicationPath);
  182. }
  183. return false;
  184. }
  185. private static bool IsAlreadyRunningAsService(string applicationPath)
  186. {
  187. var serviceName = BackgroundService.GetExistingServiceName();
  188. WqlObjectQuery wqlObjectQuery = new WqlObjectQuery(string.Format("SELECT * FROM Win32_Service WHERE State = 'Running' AND Name = '{0}'", serviceName));
  189. ManagementObjectSearcher managementObjectSearcher = new ManagementObjectSearcher(wqlObjectQuery);
  190. ManagementObjectCollection managementObjectCollection = managementObjectSearcher.Get();
  191. foreach (ManagementObject managementObject in managementObjectCollection)
  192. {
  193. var obj = managementObject.GetPropertyValue("PathName");
  194. if (obj == null)
  195. {
  196. continue;
  197. }
  198. var path = obj.ToString();
  199. _logger.Info("Service path: {0}", path);
  200. // Need to use indexOf instead of equality because the path will have the full service command line
  201. if (path.IndexOf(applicationPath, StringComparison.OrdinalIgnoreCase) != -1)
  202. {
  203. _logger.Info("The windows service is already running");
  204. 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.");
  205. return true;
  206. }
  207. }
  208. return false;
  209. }
  210. /// <summary>
  211. /// Creates the application paths.
  212. /// </summary>
  213. /// <param name="applicationPath">The application path.</param>
  214. /// <param name="runAsService">if set to <c>true</c> [run as service].</param>
  215. /// <returns>ServerApplicationPaths.</returns>
  216. private static ServerApplicationPaths CreateApplicationPaths(string applicationPath, bool runAsService)
  217. {
  218. var appFolderPath = Path.GetDirectoryName(applicationPath);
  219. var resourcesPath = Path.GetDirectoryName(applicationPath);
  220. if (runAsService)
  221. {
  222. var systemPath = Path.GetDirectoryName(applicationPath);
  223. var programDataPath = Path.GetDirectoryName(systemPath);
  224. return new ServerApplicationPaths(programDataPath, appFolderPath, resourcesPath);
  225. }
  226. return new ServerApplicationPaths(ApplicationPathHelper.GetProgramDataPath(applicationPath), appFolderPath, resourcesPath);
  227. }
  228. /// <summary>
  229. /// Gets a value indicating whether this instance can self restart.
  230. /// </summary>
  231. /// <value><c>true</c> if this instance can self restart; otherwise, <c>false</c>.</value>
  232. public static bool CanSelfRestart
  233. {
  234. get
  235. {
  236. if (IsRunningAsService)
  237. {
  238. return _canRestartService;
  239. }
  240. else
  241. {
  242. return true;
  243. }
  244. }
  245. }
  246. /// <summary>
  247. /// Gets a value indicating whether this instance can self update.
  248. /// </summary>
  249. /// <value><c>true</c> if this instance can self update; otherwise, <c>false</c>.</value>
  250. public static bool CanSelfUpdate
  251. {
  252. get
  253. {
  254. #if DEBUG
  255. return false;
  256. #endif
  257. if (IsRunningAsService)
  258. {
  259. return _canRestartService;
  260. }
  261. else
  262. {
  263. return true;
  264. }
  265. }
  266. }
  267. private static readonly TaskCompletionSource<bool> ApplicationTaskCompletionSource = new TaskCompletionSource<bool>();
  268. /// <summary>
  269. /// Runs the application.
  270. /// </summary>
  271. /// <param name="appPaths">The app paths.</param>
  272. /// <param name="logManager">The log manager.</param>
  273. /// <param name="runService">if set to <c>true</c> [run service].</param>
  274. /// <param name="options">The options.</param>
  275. private static void RunApplication(ServerApplicationPaths appPaths, ILogManager logManager, bool runService, StartupOptions options)
  276. {
  277. var fileSystem = new ManagedFileSystem(logManager.GetLogger("FileSystem"), true, true, true);
  278. fileSystem.AddShortcutHandler(new LnkShortcutHandler());
  279. fileSystem.AddShortcutHandler(new MbLinkShortcutHandler(fileSystem));
  280. var imageEncoder = ImageEncoderHelper.GetImageEncoder(_logger, logManager, fileSystem, options, () => _appHost.HttpClient, appPaths);
  281. _appHost = new WindowsAppHost(appPaths,
  282. logManager,
  283. options,
  284. fileSystem,
  285. new PowerManagement(),
  286. "emby.windows.zip",
  287. new EnvironmentInfo(),
  288. imageEncoder,
  289. new Server.Startup.Common.SystemEvents(logManager.GetLogger("SystemEvents")),
  290. new RecyclableMemoryStreamProvider(),
  291. new NetworkManager(logManager.GetLogger("NetworkManager")),
  292. GenerateCertificate,
  293. () => Environment.UserDomainName);
  294. var initProgress = new Progress<double>();
  295. if (!runService)
  296. {
  297. if (!options.ContainsOption("-nosplash")) ShowSplashScreen(_appHost.ApplicationVersion, initProgress, logManager.GetLogger("Splash"));
  298. // Not crazy about this but it's the only way to suppress ffmpeg crash dialog boxes
  299. SetErrorMode(ErrorModes.SEM_FAILCRITICALERRORS | ErrorModes.SEM_NOALIGNMENTFAULTEXCEPT |
  300. ErrorModes.SEM_NOGPFAULTERRORBOX | ErrorModes.SEM_NOOPENFILEERRORBOX);
  301. }
  302. var task = _appHost.Init(initProgress);
  303. Task.WaitAll(task);
  304. task = task.ContinueWith(new Action<Task>(a => _appHost.RunStartupTasks()), TaskContinuationOptions.OnlyOnRanToCompletion | TaskContinuationOptions.AttachedToParent);
  305. if (runService)
  306. {
  307. StartService(logManager);
  308. }
  309. else
  310. {
  311. Task.WaitAll(task);
  312. task = InstallVcredist2013IfNeeded(_appHost, _logger);
  313. Task.WaitAll(task);
  314. Microsoft.Win32.SystemEvents.SessionEnding += SystemEvents_SessionEnding;
  315. Microsoft.Win32.SystemEvents.SessionSwitch += SystemEvents_SessionSwitch;
  316. HideSplashScreen();
  317. ShowTrayIcon();
  318. task = ApplicationTaskCompletionSource.Task;
  319. Task.WaitAll(task);
  320. }
  321. }
  322. private static void GenerateCertificate(string certPath, string certHost)
  323. {
  324. CertificateGenerator.CreateSelfSignCertificatePfx(certPath, certHost, _logger);
  325. }
  326. private static ServerNotifyIcon _serverNotifyIcon;
  327. private static TaskScheduler _mainTaskScheduler;
  328. private static void ShowTrayIcon()
  329. {
  330. //Application.EnableVisualStyles();
  331. //Application.SetCompatibleTextRenderingDefault(false);
  332. _serverNotifyIcon = new ServerNotifyIcon(_appHost.LogManager, _appHost, _appHost.ServerConfigurationManager, _appHost.LocalizationManager);
  333. _mainTaskScheduler = TaskScheduler.FromCurrentSynchronizationContext();
  334. Application.Run();
  335. }
  336. internal static SplashForm _splash;
  337. private static Thread _splashThread;
  338. private static void ShowSplashScreen(Version appVersion, Progress<double> progress, ILogger logger)
  339. {
  340. var thread = new Thread(() =>
  341. {
  342. _splash = new SplashForm(appVersion, progress);
  343. _splash.ShowDialog();
  344. });
  345. thread.SetApartmentState(ApartmentState.STA);
  346. thread.IsBackground = true;
  347. thread.Start();
  348. _splashThread = thread;
  349. }
  350. private static void HideSplashScreen()
  351. {
  352. if (_splash != null)
  353. {
  354. Action act = () =>
  355. {
  356. _splash.Close();
  357. _splashThread = null;
  358. };
  359. _splash.Invoke(act);
  360. }
  361. }
  362. static void SystemEvents_SessionSwitch(object sender, SessionSwitchEventArgs e)
  363. {
  364. if (e.Reason == SessionSwitchReason.SessionLogon)
  365. {
  366. BrowserLauncher.OpenDashboard(_appHost);
  367. }
  368. }
  369. public static void Invoke(Action action)
  370. {
  371. if (IsRunningAsService)
  372. {
  373. action();
  374. }
  375. else
  376. {
  377. Task.Factory.StartNew(action, CancellationToken.None, TaskCreationOptions.None, _mainTaskScheduler ?? TaskScheduler.Current);
  378. }
  379. }
  380. /// <summary>
  381. /// Starts the service.
  382. /// </summary>
  383. private static void StartService(ILogManager logManager)
  384. {
  385. var service = new BackgroundService(logManager.GetLogger("Service"));
  386. service.Disposed += service_Disposed;
  387. ServiceBase.Run(service);
  388. }
  389. /// <summary>
  390. /// Handles the Disposed event of the service control.
  391. /// </summary>
  392. /// <param name="sender">The source of the event.</param>
  393. /// <param name="e">The <see cref="EventArgs"/> instance containing the event data.</param>
  394. static void service_Disposed(object sender, EventArgs e)
  395. {
  396. ApplicationTaskCompletionSource.SetResult(true);
  397. OnServiceShutdown();
  398. }
  399. private static void OnServiceShutdown()
  400. {
  401. _logger.Info("Shutting down");
  402. DisposeAppHost();
  403. }
  404. /// <summary>
  405. /// Installs the service.
  406. /// </summary>
  407. private static void InstallService(string applicationPath, ILogger logger)
  408. {
  409. try
  410. {
  411. ManagedInstallerClass.InstallHelper(new[] { applicationPath });
  412. logger.Info("Service installation succeeded");
  413. }
  414. catch (Exception ex)
  415. {
  416. logger.ErrorException("Uninstall failed", ex);
  417. }
  418. }
  419. /// <summary>
  420. /// Uninstalls the service.
  421. /// </summary>
  422. private static void UninstallService(string applicationPath, ILogger logger)
  423. {
  424. try
  425. {
  426. ManagedInstallerClass.InstallHelper(new[] { "/u", applicationPath });
  427. logger.Info("Service uninstallation succeeded");
  428. }
  429. catch (Exception ex)
  430. {
  431. logger.ErrorException("Uninstall failed", ex);
  432. }
  433. }
  434. private static void RunServiceInstallationIfNeeded(string applicationPath)
  435. {
  436. var serviceName = BackgroundService.GetExistingServiceName();
  437. var ctl = ServiceController.GetServices().FirstOrDefault(s => s.ServiceName == serviceName);
  438. if (ctl == null)
  439. {
  440. RunServiceInstallation(applicationPath);
  441. }
  442. }
  443. /// <summary>
  444. /// Runs the service installation.
  445. /// </summary>
  446. private static void RunServiceInstallation(string applicationPath)
  447. {
  448. var startInfo = new ProcessStartInfo
  449. {
  450. FileName = applicationPath,
  451. Arguments = "-installservice",
  452. CreateNoWindow = true,
  453. WindowStyle = ProcessWindowStyle.Hidden,
  454. Verb = "runas",
  455. ErrorDialog = false
  456. };
  457. using (var process = Process.Start(startInfo))
  458. {
  459. process.WaitForExit();
  460. }
  461. }
  462. /// <summary>
  463. /// Runs the service uninstallation.
  464. /// </summary>
  465. private static void RunServiceUninstallation(string applicationPath)
  466. {
  467. var startInfo = new ProcessStartInfo
  468. {
  469. FileName = applicationPath,
  470. Arguments = "-uninstallservice",
  471. CreateNoWindow = true,
  472. WindowStyle = ProcessWindowStyle.Hidden,
  473. Verb = "runas",
  474. ErrorDialog = false
  475. };
  476. using (var process = Process.Start(startInfo))
  477. {
  478. process.WaitForExit();
  479. }
  480. }
  481. /// <summary>
  482. /// Handles the SessionEnding event of the SystemEvents control.
  483. /// </summary>
  484. /// <param name="sender">The source of the event.</param>
  485. /// <param name="e">The <see cref="SessionEndingEventArgs"/> instance containing the event data.</param>
  486. static void SystemEvents_SessionEnding(object sender, SessionEndingEventArgs e)
  487. {
  488. if (e.Reason == SessionEndReasons.SystemShutdown || !IsRunningAsService)
  489. {
  490. Shutdown();
  491. }
  492. }
  493. /// <summary>
  494. /// Handles the UnhandledException event of the CurrentDomain control.
  495. /// </summary>
  496. /// <param name="sender">The source of the event.</param>
  497. /// <param name="e">The <see cref="UnhandledExceptionEventArgs"/> instance containing the event data.</param>
  498. static void CurrentDomain_UnhandledException(object sender, UnhandledExceptionEventArgs e)
  499. {
  500. var exception = (Exception)e.ExceptionObject;
  501. new UnhandledExceptionWriter(_appHost.ServerConfigurationManager.ApplicationPaths, _logger, _appHost.LogManager).Log(exception);
  502. if (!IsRunningAsService)
  503. {
  504. MessageBox.Show("Unhandled exception: " + exception.Message);
  505. }
  506. if (!Debugger.IsAttached)
  507. {
  508. Environment.Exit(Marshal.GetHRForException(exception));
  509. }
  510. }
  511. /// <summary>
  512. /// Performs the update if needed.
  513. /// </summary>
  514. /// <param name="appPaths">The app paths.</param>
  515. /// <param name="logger">The logger.</param>
  516. /// <returns><c>true</c> if XXXX, <c>false</c> otherwise</returns>
  517. private static bool PerformUpdateIfNeeded(ServerApplicationPaths appPaths, ILogger logger)
  518. {
  519. // Look for the existence of an update archive
  520. var updateArchive = Path.Combine(appPaths.TempUpdatePath, "MBServer" + ".zip");
  521. if (File.Exists(updateArchive))
  522. {
  523. logger.Info("An update is available from {0}", updateArchive);
  524. // Update is there - execute update
  525. try
  526. {
  527. var serviceName = IsRunningAsService ? BackgroundService.GetExistingServiceName() : string.Empty;
  528. new ApplicationUpdater().UpdateApplication(appPaths, updateArchive, logger, serviceName);
  529. // And just let the app exit so it can update
  530. return true;
  531. }
  532. catch (Exception e)
  533. {
  534. logger.ErrorException("Error starting updater.", e);
  535. MessageBox.Show(string.Format("Error attempting to update application.\n\n{0}\n\n{1}", e.GetType().Name, e.Message));
  536. }
  537. }
  538. return false;
  539. }
  540. public static void Shutdown()
  541. {
  542. if (IsRunningAsService)
  543. {
  544. ShutdownWindowsService();
  545. }
  546. else
  547. {
  548. DisposeAppHost();
  549. ShutdownWindowsApplication();
  550. }
  551. }
  552. public static void Restart()
  553. {
  554. DisposeAppHost();
  555. if (IsRunningAsService)
  556. {
  557. RestartWindowsService();
  558. }
  559. else
  560. {
  561. //_logger.Info("Hiding server notify icon");
  562. //_serverNotifyIcon.Visible = false;
  563. _logger.Info("Starting new instance");
  564. //Application.Restart();
  565. Process.Start(ApplicationPath);
  566. ShutdownWindowsApplication();
  567. }
  568. }
  569. private static void DisposeAppHost()
  570. {
  571. if (!_appHostDisposed)
  572. {
  573. _logger.Info("Disposing app host");
  574. _appHostDisposed = true;
  575. _appHost.Dispose();
  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. }