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(30000))
  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 fileSystem = new ManagedFileSystem(logManager.GetLogger("FileSystem"), true, true, false, appPaths.TempDirectory);
  285. fileSystem.AddShortcutHandler(new LnkShortcutHandler());
  286. fileSystem.AddShortcutHandler(new MbLinkShortcutHandler(fileSystem));
  287. var imageEncoder = ImageEncoderHelper.GetImageEncoder(_logger, logManager, fileSystem, options, () => _appHost.HttpClient, appPaths);
  288. FileSystem = fileSystem;
  289. _appHost = new WindowsAppHost(appPaths,
  290. logManager,
  291. options,
  292. fileSystem,
  293. new PowerManagement(),
  294. "emby.windows.zip",
  295. new EnvironmentInfo(),
  296. imageEncoder,
  297. new Server.Startup.Common.SystemEvents(logManager.GetLogger("SystemEvents")),
  298. new RecyclableMemoryStreamProvider(),
  299. new Networking.NetworkManager(logManager.GetLogger("NetworkManager")),
  300. GenerateCertificate,
  301. () => Environment.UserDomainName);
  302. var initProgress = new Progress<double>();
  303. if (!runService)
  304. {
  305. if (!options.ContainsOption("-nosplash")) ShowSplashScreen(_appHost.ApplicationVersion, initProgress, logManager.GetLogger("Splash"));
  306. // Not crazy about this but it's the only way to suppress ffmpeg crash dialog boxes
  307. SetErrorMode(ErrorModes.SEM_FAILCRITICALERRORS | ErrorModes.SEM_NOALIGNMENTFAULTEXCEPT |
  308. ErrorModes.SEM_NOGPFAULTERRORBOX | ErrorModes.SEM_NOOPENFILEERRORBOX);
  309. }
  310. var task = _appHost.Init(initProgress);
  311. Task.WaitAll(task);
  312. task = task.ContinueWith(new Action<Task>(a => _appHost.RunStartupTasks()), TaskContinuationOptions.OnlyOnRanToCompletion | TaskContinuationOptions.AttachedToParent);
  313. if (runService)
  314. {
  315. StartService(logManager);
  316. }
  317. else
  318. {
  319. Task.WaitAll(task);
  320. task = InstallVcredist2013IfNeeded(_appHost, _logger);
  321. Task.WaitAll(task);
  322. Microsoft.Win32.SystemEvents.SessionSwitch += SystemEvents_SessionSwitch;
  323. HideSplashScreen();
  324. ShowTrayIcon();
  325. task = ApplicationTaskCompletionSource.Task;
  326. Task.WaitAll(task);
  327. }
  328. }
  329. private static void GenerateCertificate(string certPath, string certHost)
  330. {
  331. CertificateGenerator.CreateSelfSignCertificatePfx(certPath, certHost, _logger);
  332. }
  333. private static ServerNotifyIcon _serverNotifyIcon;
  334. private static TaskScheduler _mainTaskScheduler;
  335. private static void ShowTrayIcon()
  336. {
  337. //Application.EnableVisualStyles();
  338. //Application.SetCompatibleTextRenderingDefault(false);
  339. _serverNotifyIcon = new ServerNotifyIcon(_appHost.LogManager, _appHost, _appHost.ServerConfigurationManager, _appHost.LocalizationManager);
  340. _mainTaskScheduler = TaskScheduler.FromCurrentSynchronizationContext();
  341. Application.Run();
  342. }
  343. internal static SplashForm _splash;
  344. private static Thread _splashThread;
  345. private static void ShowSplashScreen(Version appVersion, Progress<double> progress, ILogger logger)
  346. {
  347. var thread = new Thread(() =>
  348. {
  349. _splash = new SplashForm(appVersion, progress);
  350. _splash.ShowDialog();
  351. });
  352. thread.SetApartmentState(ApartmentState.STA);
  353. thread.IsBackground = true;
  354. thread.Start();
  355. _splashThread = thread;
  356. }
  357. private static void HideSplashScreen()
  358. {
  359. if (_splash != null)
  360. {
  361. Action act = () =>
  362. {
  363. _splash.Close();
  364. _splashThread = null;
  365. };
  366. _splash.Invoke(act);
  367. }
  368. }
  369. static void SystemEvents_SessionSwitch(object sender, SessionSwitchEventArgs e)
  370. {
  371. if (e.Reason == SessionSwitchReason.SessionLogon)
  372. {
  373. BrowserLauncher.OpenDashboard(_appHost);
  374. }
  375. }
  376. public static void Invoke(Action action)
  377. {
  378. if (IsRunningAsService)
  379. {
  380. action();
  381. }
  382. else
  383. {
  384. Task.Factory.StartNew(action, CancellationToken.None, TaskCreationOptions.None, _mainTaskScheduler ?? TaskScheduler.Current);
  385. }
  386. }
  387. /// <summary>
  388. /// Starts the service.
  389. /// </summary>
  390. private static void StartService(ILogManager logManager)
  391. {
  392. var service = new BackgroundService(logManager.GetLogger("Service"));
  393. service.Disposed += service_Disposed;
  394. ServiceBase.Run(service);
  395. }
  396. /// <summary>
  397. /// Handles the Disposed event of the service control.
  398. /// </summary>
  399. /// <param name="sender">The source of the event.</param>
  400. /// <param name="e">The <see cref="EventArgs"/> instance containing the event data.</param>
  401. static void service_Disposed(object sender, EventArgs e)
  402. {
  403. ApplicationTaskCompletionSource.SetResult(true);
  404. OnServiceShutdown();
  405. }
  406. private static void OnServiceShutdown()
  407. {
  408. _logger.Info("Shutting down");
  409. DisposeAppHost();
  410. }
  411. /// <summary>
  412. /// Installs the service.
  413. /// </summary>
  414. private static void InstallService(string applicationPath, ILogger logger)
  415. {
  416. try
  417. {
  418. ManagedInstallerClass.InstallHelper(new[] { applicationPath });
  419. logger.Info("Service installation succeeded");
  420. }
  421. catch (Exception ex)
  422. {
  423. logger.ErrorException("Uninstall failed", ex);
  424. }
  425. }
  426. /// <summary>
  427. /// Uninstalls the service.
  428. /// </summary>
  429. private static void UninstallService(string applicationPath, ILogger logger)
  430. {
  431. try
  432. {
  433. ManagedInstallerClass.InstallHelper(new[] { "/u", applicationPath });
  434. logger.Info("Service uninstallation succeeded");
  435. }
  436. catch (Exception ex)
  437. {
  438. logger.ErrorException("Uninstall failed", ex);
  439. }
  440. }
  441. private static void RunServiceInstallationIfNeeded(string applicationPath)
  442. {
  443. var serviceName = BackgroundService.GetExistingServiceName();
  444. var ctl = ServiceController.GetServices().FirstOrDefault(s => s.ServiceName == serviceName);
  445. if (ctl == null)
  446. {
  447. RunServiceInstallation(applicationPath);
  448. }
  449. }
  450. /// <summary>
  451. /// Runs the service installation.
  452. /// </summary>
  453. private static void RunServiceInstallation(string applicationPath)
  454. {
  455. var startInfo = new ProcessStartInfo
  456. {
  457. FileName = applicationPath,
  458. Arguments = "-installservice",
  459. CreateNoWindow = true,
  460. WindowStyle = ProcessWindowStyle.Hidden,
  461. Verb = "runas",
  462. ErrorDialog = false
  463. };
  464. using (var process = Process.Start(startInfo))
  465. {
  466. process.WaitForExit();
  467. }
  468. }
  469. /// <summary>
  470. /// Runs the service uninstallation.
  471. /// </summary>
  472. private static void RunServiceUninstallation(string applicationPath)
  473. {
  474. var startInfo = new ProcessStartInfo
  475. {
  476. FileName = applicationPath,
  477. Arguments = "-uninstallservice",
  478. CreateNoWindow = true,
  479. WindowStyle = ProcessWindowStyle.Hidden,
  480. Verb = "runas",
  481. ErrorDialog = false
  482. };
  483. using (var process = Process.Start(startInfo))
  484. {
  485. process.WaitForExit();
  486. }
  487. }
  488. /// <summary>
  489. /// Handles the UnhandledException event of the CurrentDomain control.
  490. /// </summary>
  491. /// <param name="sender">The source of the event.</param>
  492. /// <param name="e">The <see cref="UnhandledExceptionEventArgs"/> instance containing the event data.</param>
  493. static void CurrentDomain_UnhandledException(object sender, UnhandledExceptionEventArgs e)
  494. {
  495. var exception = (Exception)e.ExceptionObject;
  496. new UnhandledExceptionWriter(_appHost.ServerConfigurationManager.ApplicationPaths, _logger, _appHost.LogManager, FileSystem, new ConsoleLogger()).Log(exception);
  497. if (!IsRunningAsService)
  498. {
  499. MessageBox.Show("Unhandled exception: " + exception.Message);
  500. }
  501. if (!Debugger.IsAttached)
  502. {
  503. Environment.Exit(Marshal.GetHRForException(exception));
  504. }
  505. }
  506. /// <summary>
  507. /// Performs the update if needed.
  508. /// </summary>
  509. /// <param name="appPaths">The app paths.</param>
  510. /// <param name="logger">The logger.</param>
  511. /// <returns><c>true</c> if XXXX, <c>false</c> otherwise</returns>
  512. private static bool PerformUpdateIfNeeded(ServerApplicationPaths appPaths, ILogger logger)
  513. {
  514. // Not supported
  515. if (IsRunningAsService)
  516. {
  517. return false;
  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. _logger.Info("App host dispose complete");
  577. }
  578. }
  579. private static void ShutdownWindowsApplication()
  580. {
  581. if (_serverNotifyIcon != null)
  582. {
  583. _serverNotifyIcon.Dispose();
  584. _serverNotifyIcon = null;
  585. }
  586. //_logger.Info("Calling Application.Exit");
  587. //Application.Exit();
  588. _logger.Info("Calling Environment.Exit");
  589. Environment.Exit(0);
  590. _logger.Info("Calling ApplicationTaskCompletionSource.SetResult");
  591. ApplicationTaskCompletionSource.SetResult(true);
  592. }
  593. private static void ShutdownWindowsService()
  594. {
  595. _logger.Info("Stopping background service");
  596. var service = new ServiceController(BackgroundService.GetExistingServiceName());
  597. service.Refresh();
  598. if (service.Status == ServiceControllerStatus.Running)
  599. {
  600. service.Stop();
  601. }
  602. }
  603. private static void RestartWindowsService()
  604. {
  605. _logger.Info("Restarting background service");
  606. var startInfo = new ProcessStartInfo
  607. {
  608. FileName = "cmd.exe",
  609. CreateNoWindow = true,
  610. WindowStyle = ProcessWindowStyle.Hidden,
  611. Verb = "runas",
  612. ErrorDialog = false,
  613. Arguments = String.Format("/c sc stop {0} & sc start {0} & sc start {0}", BackgroundService.GetExistingServiceName())
  614. };
  615. Process.Start(startInfo);
  616. }
  617. private static bool CanRestartWindowsService()
  618. {
  619. var startInfo = new ProcessStartInfo
  620. {
  621. FileName = "cmd.exe",
  622. CreateNoWindow = true,
  623. WindowStyle = ProcessWindowStyle.Hidden,
  624. Verb = "runas",
  625. ErrorDialog = false,
  626. Arguments = String.Format("/c sc query {0}", BackgroundService.GetExistingServiceName())
  627. };
  628. using (var process = Process.Start(startInfo))
  629. {
  630. process.WaitForExit();
  631. if (process.ExitCode == 0)
  632. {
  633. return true;
  634. }
  635. else
  636. {
  637. return false;
  638. }
  639. }
  640. }
  641. private static async Task InstallVcredist2013IfNeeded(ApplicationHost appHost, ILogger logger)
  642. {
  643. // Reference
  644. // http://stackoverflow.com/questions/12206314/detect-if-visual-c-redistributable-for-visual-studio-2012-is-installed
  645. try
  646. {
  647. var subkey = Environment.Is64BitProcess
  648. ? "SOFTWARE\\WOW6432Node\\Microsoft\\VisualStudio\\12.0\\VC\\Runtimes\\x64"
  649. : "SOFTWARE\\Microsoft\\VisualStudio\\12.0\\VC\\Runtimes\\x86";
  650. using (RegistryKey ndpKey = RegistryKey.OpenBaseKey(RegistryHive.LocalMachine, RegistryView.Default)
  651. .OpenSubKey(subkey))
  652. {
  653. if (ndpKey != null && ndpKey.GetValue("Version") != null)
  654. {
  655. var installedVersion = ((string)ndpKey.GetValue("Version")).TrimStart('v');
  656. if (installedVersion.StartsWith("12", StringComparison.OrdinalIgnoreCase))
  657. {
  658. return;
  659. }
  660. }
  661. }
  662. }
  663. catch (Exception ex)
  664. {
  665. logger.ErrorException("Error getting .NET Framework version", ex);
  666. return;
  667. }
  668. try
  669. {
  670. await InstallVcredist2013().ConfigureAwait(false);
  671. }
  672. catch (Exception ex)
  673. {
  674. logger.ErrorException("Error installing Visual Studio C++ runtime", ex);
  675. }
  676. }
  677. private async static Task InstallVcredist2013()
  678. {
  679. var httpClient = _appHost.HttpClient;
  680. var tmp = await httpClient.GetTempFile(new HttpRequestOptions
  681. {
  682. Url = GetVcredist2013Url(),
  683. Progress = new Progress<double>()
  684. }).ConfigureAwait(false);
  685. var exePath = Path.ChangeExtension(tmp, ".exe");
  686. File.Copy(tmp, exePath);
  687. var startInfo = new ProcessStartInfo
  688. {
  689. FileName = exePath,
  690. CreateNoWindow = true,
  691. WindowStyle = ProcessWindowStyle.Hidden,
  692. Verb = "runas",
  693. ErrorDialog = false
  694. };
  695. _logger.Info("Running {0}", startInfo.FileName);
  696. using (var process = Process.Start(startInfo))
  697. {
  698. process.WaitForExit();
  699. }
  700. }
  701. private static string GetVcredist2013Url()
  702. {
  703. if (Environment.Is64BitProcess)
  704. {
  705. return "https://github.com/MediaBrowser/Emby.Resources/raw/master/vcredist2013/vcredist_x64.exe";
  706. }
  707. // TODO: ARM url - https://github.com/MediaBrowser/Emby.Resources/raw/master/vcredist2013/vcredist_arm.exe
  708. return "https://github.com/MediaBrowser/Emby.Resources/raw/master/vcredist2013/vcredist_x86.exe";
  709. }
  710. /// <summary>
  711. /// Sets the error mode.
  712. /// </summary>
  713. /// <param name="uMode">The u mode.</param>
  714. /// <returns>ErrorModes.</returns>
  715. [DllImport("kernel32.dll")]
  716. static extern ErrorModes SetErrorMode(ErrorModes uMode);
  717. /// <summary>
  718. /// Enum ErrorModes
  719. /// </summary>
  720. [Flags]
  721. public enum ErrorModes : uint
  722. {
  723. /// <summary>
  724. /// The SYSTE m_ DEFAULT
  725. /// </summary>
  726. SYSTEM_DEFAULT = 0x0,
  727. /// <summary>
  728. /// The SE m_ FAILCRITICALERRORS
  729. /// </summary>
  730. SEM_FAILCRITICALERRORS = 0x0001,
  731. /// <summary>
  732. /// The SE m_ NOALIGNMENTFAULTEXCEPT
  733. /// </summary>
  734. SEM_NOALIGNMENTFAULTEXCEPT = 0x0004,
  735. /// <summary>
  736. /// The SE m_ NOGPFAULTERRORBOX
  737. /// </summary>
  738. SEM_NOGPFAULTERRORBOX = 0x0002,
  739. /// <summary>
  740. /// The SE m_ NOOPENFILEERRORBOX
  741. /// </summary>
  742. SEM_NOOPENFILEERRORBOX = 0x8000
  743. }
  744. }
  745. }