KProcess.cs 34 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117
  1. using ARMeilleure.State;
  2. using Ryujinx.Common;
  3. using Ryujinx.Common.Logging;
  4. using Ryujinx.Cpu;
  5. using Ryujinx.HLE.Exceptions;
  6. using Ryujinx.HLE.HOS.Kernel.Common;
  7. using Ryujinx.HLE.HOS.Kernel.Memory;
  8. using Ryujinx.HLE.HOS.Kernel.Threading;
  9. using System;
  10. using System.Collections.Generic;
  11. using System.Linq;
  12. using System.Threading;
  13. namespace Ryujinx.HLE.HOS.Kernel.Process
  14. {
  15. class KProcess : KSynchronizationObject
  16. {
  17. public const int KernelVersionMajor = 10;
  18. public const int KernelVersionMinor = 4;
  19. public const int KernelVersionRevision = 0;
  20. public const int KernelVersionPacked =
  21. (KernelVersionMajor << 19) |
  22. (KernelVersionMinor << 15) |
  23. (KernelVersionRevision << 0);
  24. public KMemoryManager MemoryManager { get; private set; }
  25. private SortedDictionary<ulong, KTlsPageInfo> _fullTlsPages;
  26. private SortedDictionary<ulong, KTlsPageInfo> _freeTlsPages;
  27. public int DefaultCpuCore { get; set; }
  28. public bool Debug { get; private set; }
  29. public KResourceLimit ResourceLimit { get; private set; }
  30. public ulong PersonalMmHeapPagesCount { get; private set; }
  31. public ProcessState State { get; private set; }
  32. private object _processLock;
  33. private object _threadingLock;
  34. public KAddressArbiter AddressArbiter { get; private set; }
  35. public long[] RandomEntropy { get; private set; }
  36. private bool _signaled;
  37. private bool _useSystemMemBlocks;
  38. public string Name { get; private set; }
  39. private int _threadCount;
  40. public int MmuFlags { get; private set; }
  41. private MemoryRegion _memRegion;
  42. public KProcessCapabilities Capabilities { get; private set; }
  43. public ulong TitleId { get; private set; }
  44. public long Pid { get; private set; }
  45. private long _creationTimestamp;
  46. private ulong _entrypoint;
  47. private ulong _imageSize;
  48. private ulong _mainThreadStackSize;
  49. private ulong _memoryUsageCapacity;
  50. private int _version;
  51. public KHandleTable HandleTable { get; private set; }
  52. public ulong UserExceptionContextAddress { get; private set; }
  53. private LinkedList<KThread> _threads;
  54. public bool IsPaused { get; private set; }
  55. public MemoryManager CpuMemory { get; private set; }
  56. public CpuContext CpuContext { get; private set; }
  57. public HleProcessDebugger Debugger { get; private set; }
  58. public KProcess(KernelContext context) : base(context)
  59. {
  60. _processLock = new object();
  61. _threadingLock = new object();
  62. AddressArbiter = new KAddressArbiter(context);
  63. _fullTlsPages = new SortedDictionary<ulong, KTlsPageInfo>();
  64. _freeTlsPages = new SortedDictionary<ulong, KTlsPageInfo>();
  65. Capabilities = new KProcessCapabilities();
  66. RandomEntropy = new long[KScheduler.CpuCoresCount];
  67. _threads = new LinkedList<KThread>();
  68. Debugger = new HleProcessDebugger(this);
  69. }
  70. public KernelResult InitializeKip(
  71. ProcessCreationInfo creationInfo,
  72. int[] caps,
  73. KPageList pageList,
  74. KResourceLimit resourceLimit,
  75. MemoryRegion memRegion)
  76. {
  77. ResourceLimit = resourceLimit;
  78. _memRegion = memRegion;
  79. AddressSpaceType addrSpaceType = (AddressSpaceType)((creationInfo.MmuFlags >> 1) & 7);
  80. InitializeMemoryManager(addrSpaceType, memRegion);
  81. bool aslrEnabled = ((creationInfo.MmuFlags >> 5) & 1) != 0;
  82. ulong codeAddress = creationInfo.CodeAddress;
  83. ulong codeSize = (ulong)creationInfo.CodePagesCount * KMemoryManager.PageSize;
  84. KMemoryBlockAllocator memoryBlockAllocator = (MmuFlags & 0x40) != 0
  85. ? KernelContext.LargeMemoryBlockAllocator
  86. : KernelContext.SmallMemoryBlockAllocator;
  87. KernelResult result = MemoryManager.InitializeForProcess(
  88. addrSpaceType,
  89. aslrEnabled,
  90. !aslrEnabled,
  91. memRegion,
  92. codeAddress,
  93. codeSize,
  94. memoryBlockAllocator);
  95. if (result != KernelResult.Success)
  96. {
  97. return result;
  98. }
  99. if (!ValidateCodeAddressAndSize(codeAddress, codeSize))
  100. {
  101. return KernelResult.InvalidMemRange;
  102. }
  103. result = MemoryManager.MapPages(
  104. codeAddress,
  105. pageList,
  106. MemoryState.CodeStatic,
  107. MemoryPermission.None);
  108. if (result != KernelResult.Success)
  109. {
  110. return result;
  111. }
  112. result = Capabilities.InitializeForKernel(caps, MemoryManager);
  113. if (result != KernelResult.Success)
  114. {
  115. return result;
  116. }
  117. Pid = KernelContext.NewKipId();
  118. if (Pid == 0 || (ulong)Pid >= KernelConstants.InitialProcessId)
  119. {
  120. throw new InvalidOperationException($"Invalid KIP Id {Pid}.");
  121. }
  122. result = ParseProcessInfo(creationInfo);
  123. return result;
  124. }
  125. public KernelResult Initialize(
  126. ProcessCreationInfo creationInfo,
  127. int[] caps,
  128. KResourceLimit resourceLimit,
  129. MemoryRegion memRegion)
  130. {
  131. ResourceLimit = resourceLimit;
  132. _memRegion = memRegion;
  133. ulong personalMmHeapSize = GetPersonalMmHeapSize((ulong)creationInfo.PersonalMmHeapPagesCount, memRegion);
  134. ulong codePagesCount = (ulong)creationInfo.CodePagesCount;
  135. ulong neededSizeForProcess = personalMmHeapSize + codePagesCount * KMemoryManager.PageSize;
  136. if (neededSizeForProcess != 0 && resourceLimit != null)
  137. {
  138. if (!resourceLimit.Reserve(LimitableResource.Memory, neededSizeForProcess))
  139. {
  140. return KernelResult.ResLimitExceeded;
  141. }
  142. }
  143. void CleanUpForError()
  144. {
  145. if (neededSizeForProcess != 0 && resourceLimit != null)
  146. {
  147. resourceLimit.Release(LimitableResource.Memory, neededSizeForProcess);
  148. }
  149. }
  150. PersonalMmHeapPagesCount = (ulong)creationInfo.PersonalMmHeapPagesCount;
  151. KMemoryBlockAllocator memoryBlockAllocator;
  152. if (PersonalMmHeapPagesCount != 0)
  153. {
  154. memoryBlockAllocator = new KMemoryBlockAllocator(PersonalMmHeapPagesCount * KMemoryManager.PageSize);
  155. }
  156. else
  157. {
  158. memoryBlockAllocator = (MmuFlags & 0x40) != 0
  159. ? KernelContext.LargeMemoryBlockAllocator
  160. : KernelContext.SmallMemoryBlockAllocator;
  161. }
  162. AddressSpaceType addrSpaceType = (AddressSpaceType)((creationInfo.MmuFlags >> 1) & 7);
  163. InitializeMemoryManager(addrSpaceType, memRegion);
  164. bool aslrEnabled = ((creationInfo.MmuFlags >> 5) & 1) != 0;
  165. ulong codeAddress = creationInfo.CodeAddress;
  166. ulong codeSize = codePagesCount * KMemoryManager.PageSize;
  167. KernelResult result = MemoryManager.InitializeForProcess(
  168. addrSpaceType,
  169. aslrEnabled,
  170. !aslrEnabled,
  171. memRegion,
  172. codeAddress,
  173. codeSize,
  174. memoryBlockAllocator);
  175. if (result != KernelResult.Success)
  176. {
  177. CleanUpForError();
  178. return result;
  179. }
  180. if (!ValidateCodeAddressAndSize(codeAddress, codeSize))
  181. {
  182. CleanUpForError();
  183. return KernelResult.InvalidMemRange;
  184. }
  185. result = MemoryManager.MapNewProcessCode(
  186. codeAddress,
  187. codePagesCount,
  188. MemoryState.CodeStatic,
  189. MemoryPermission.None);
  190. if (result != KernelResult.Success)
  191. {
  192. CleanUpForError();
  193. return result;
  194. }
  195. result = Capabilities.InitializeForUser(caps, MemoryManager);
  196. if (result != KernelResult.Success)
  197. {
  198. CleanUpForError();
  199. return result;
  200. }
  201. Pid = KernelContext.NewProcessId();
  202. if (Pid == -1 || (ulong)Pid < KernelConstants.InitialProcessId)
  203. {
  204. throw new InvalidOperationException($"Invalid Process Id {Pid}.");
  205. }
  206. result = ParseProcessInfo(creationInfo);
  207. if (result != KernelResult.Success)
  208. {
  209. CleanUpForError();
  210. }
  211. return result;
  212. }
  213. private bool ValidateCodeAddressAndSize(ulong address, ulong size)
  214. {
  215. ulong codeRegionStart;
  216. ulong codeRegionSize;
  217. switch (MemoryManager.AddrSpaceWidth)
  218. {
  219. case 32:
  220. codeRegionStart = 0x200000;
  221. codeRegionSize = 0x3fe00000;
  222. break;
  223. case 36:
  224. codeRegionStart = 0x8000000;
  225. codeRegionSize = 0x78000000;
  226. break;
  227. case 39:
  228. codeRegionStart = 0x8000000;
  229. codeRegionSize = 0x7ff8000000;
  230. break;
  231. default: throw new InvalidOperationException("Invalid address space width on memory manager.");
  232. }
  233. ulong endAddr = address + size;
  234. ulong codeRegionEnd = codeRegionStart + codeRegionSize;
  235. if (endAddr <= address ||
  236. endAddr - 1 > codeRegionEnd - 1)
  237. {
  238. return false;
  239. }
  240. if (MemoryManager.InsideHeapRegion (address, size) ||
  241. MemoryManager.InsideAliasRegion(address, size))
  242. {
  243. return false;
  244. }
  245. return true;
  246. }
  247. private KernelResult ParseProcessInfo(ProcessCreationInfo creationInfo)
  248. {
  249. // Ensure that the current kernel version is equal or above to the minimum required.
  250. uint requiredKernelVersionMajor = (uint)Capabilities.KernelReleaseVersion >> 19;
  251. uint requiredKernelVersionMinor = ((uint)Capabilities.KernelReleaseVersion >> 15) & 0xf;
  252. if (KernelContext.EnableVersionChecks)
  253. {
  254. if (requiredKernelVersionMajor > KernelVersionMajor)
  255. {
  256. return KernelResult.InvalidCombination;
  257. }
  258. if (requiredKernelVersionMajor != KernelVersionMajor && requiredKernelVersionMajor < 3)
  259. {
  260. return KernelResult.InvalidCombination;
  261. }
  262. if (requiredKernelVersionMinor > KernelVersionMinor)
  263. {
  264. return KernelResult.InvalidCombination;
  265. }
  266. }
  267. KernelResult result = AllocateThreadLocalStorage(out ulong userExceptionContextAddress);
  268. if (result != KernelResult.Success)
  269. {
  270. return result;
  271. }
  272. UserExceptionContextAddress = userExceptionContextAddress;
  273. MemoryHelper.FillWithZeros(CpuMemory, (long)userExceptionContextAddress, KTlsPageInfo.TlsEntrySize);
  274. Name = creationInfo.Name;
  275. State = ProcessState.Created;
  276. _creationTimestamp = PerformanceCounter.ElapsedMilliseconds;
  277. MmuFlags = creationInfo.MmuFlags;
  278. _version = creationInfo.Version;
  279. TitleId = creationInfo.TitleId;
  280. _entrypoint = creationInfo.CodeAddress;
  281. _imageSize = (ulong)creationInfo.CodePagesCount * KMemoryManager.PageSize;
  282. _useSystemMemBlocks = ((MmuFlags >> 6) & 1) != 0;
  283. switch ((AddressSpaceType)((MmuFlags >> 1) & 7))
  284. {
  285. case AddressSpaceType.Addr32Bits:
  286. case AddressSpaceType.Addr36Bits:
  287. case AddressSpaceType.Addr39Bits:
  288. _memoryUsageCapacity = MemoryManager.HeapRegionEnd -
  289. MemoryManager.HeapRegionStart;
  290. break;
  291. case AddressSpaceType.Addr32BitsNoMap:
  292. _memoryUsageCapacity = MemoryManager.HeapRegionEnd -
  293. MemoryManager.HeapRegionStart +
  294. MemoryManager.AliasRegionEnd -
  295. MemoryManager.AliasRegionStart;
  296. break;
  297. default: throw new InvalidOperationException($"Invalid MMU flags value 0x{MmuFlags:x2}.");
  298. }
  299. GenerateRandomEntropy();
  300. return KernelResult.Success;
  301. }
  302. public KernelResult AllocateThreadLocalStorage(out ulong address)
  303. {
  304. KernelContext.CriticalSection.Enter();
  305. KernelResult result;
  306. if (_freeTlsPages.Count > 0)
  307. {
  308. // If we have free TLS pages available, just use the first one.
  309. KTlsPageInfo pageInfo = _freeTlsPages.Values.First();
  310. if (!pageInfo.TryGetFreePage(out address))
  311. {
  312. throw new InvalidOperationException("Unexpected failure getting free TLS page!");
  313. }
  314. if (pageInfo.IsFull())
  315. {
  316. _freeTlsPages.Remove(pageInfo.PageAddr);
  317. _fullTlsPages.Add(pageInfo.PageAddr, pageInfo);
  318. }
  319. result = KernelResult.Success;
  320. }
  321. else
  322. {
  323. // Otherwise, we need to create a new one.
  324. result = AllocateTlsPage(out KTlsPageInfo pageInfo);
  325. if (result == KernelResult.Success)
  326. {
  327. if (!pageInfo.TryGetFreePage(out address))
  328. {
  329. throw new InvalidOperationException("Unexpected failure getting free TLS page!");
  330. }
  331. _freeTlsPages.Add(pageInfo.PageAddr, pageInfo);
  332. }
  333. else
  334. {
  335. address = 0;
  336. }
  337. }
  338. KernelContext.CriticalSection.Leave();
  339. return result;
  340. }
  341. private KernelResult AllocateTlsPage(out KTlsPageInfo pageInfo)
  342. {
  343. pageInfo = default;
  344. if (!KernelContext.UserSlabHeapPages.TryGetItem(out ulong tlsPagePa))
  345. {
  346. return KernelResult.OutOfMemory;
  347. }
  348. ulong regionStart = MemoryManager.TlsIoRegionStart;
  349. ulong regionSize = MemoryManager.TlsIoRegionEnd - regionStart;
  350. ulong regionPagesCount = regionSize / KMemoryManager.PageSize;
  351. KernelResult result = MemoryManager.AllocateOrMapPa(
  352. 1,
  353. KMemoryManager.PageSize,
  354. tlsPagePa,
  355. true,
  356. regionStart,
  357. regionPagesCount,
  358. MemoryState.ThreadLocal,
  359. MemoryPermission.ReadAndWrite,
  360. out ulong tlsPageVa);
  361. if (result != KernelResult.Success)
  362. {
  363. KernelContext.UserSlabHeapPages.Free(tlsPagePa);
  364. }
  365. else
  366. {
  367. pageInfo = new KTlsPageInfo(tlsPageVa);
  368. MemoryHelper.FillWithZeros(CpuMemory, (long)tlsPageVa, KMemoryManager.PageSize);
  369. }
  370. return result;
  371. }
  372. public KernelResult FreeThreadLocalStorage(ulong tlsSlotAddr)
  373. {
  374. ulong tlsPageAddr = BitUtils.AlignDown(tlsSlotAddr, KMemoryManager.PageSize);
  375. KernelContext.CriticalSection.Enter();
  376. KernelResult result = KernelResult.Success;
  377. KTlsPageInfo pageInfo = null;
  378. if (_fullTlsPages.TryGetValue(tlsPageAddr, out pageInfo))
  379. {
  380. // TLS page was full, free slot and move to free pages tree.
  381. _fullTlsPages.Remove(tlsPageAddr);
  382. _freeTlsPages.Add(tlsPageAddr, pageInfo);
  383. }
  384. else if (!_freeTlsPages.TryGetValue(tlsPageAddr, out pageInfo))
  385. {
  386. result = KernelResult.InvalidAddress;
  387. }
  388. if (pageInfo != null)
  389. {
  390. pageInfo.FreeTlsSlot(tlsSlotAddr);
  391. if (pageInfo.IsEmpty())
  392. {
  393. // TLS page is now empty, we should ensure it is removed
  394. // from all trees, and free the memory it was using.
  395. _freeTlsPages.Remove(tlsPageAddr);
  396. KernelContext.CriticalSection.Leave();
  397. FreeTlsPage(pageInfo);
  398. return KernelResult.Success;
  399. }
  400. }
  401. KernelContext.CriticalSection.Leave();
  402. return result;
  403. }
  404. private KernelResult FreeTlsPage(KTlsPageInfo pageInfo)
  405. {
  406. if (!MemoryManager.TryConvertVaToPa(pageInfo.PageAddr, out ulong tlsPagePa))
  407. {
  408. throw new InvalidOperationException("Unexpected failure translating virtual address to physical.");
  409. }
  410. KernelResult result = MemoryManager.UnmapForKernel(pageInfo.PageAddr, 1, MemoryState.ThreadLocal);
  411. if (result == KernelResult.Success)
  412. {
  413. KernelContext.UserSlabHeapPages.Free(tlsPagePa);
  414. }
  415. return result;
  416. }
  417. private void GenerateRandomEntropy()
  418. {
  419. // TODO.
  420. }
  421. public KernelResult Start(int mainThreadPriority, ulong stackSize)
  422. {
  423. lock (_processLock)
  424. {
  425. if (State > ProcessState.CreatedAttached)
  426. {
  427. return KernelResult.InvalidState;
  428. }
  429. if (ResourceLimit != null && !ResourceLimit.Reserve(LimitableResource.Thread, 1))
  430. {
  431. return KernelResult.ResLimitExceeded;
  432. }
  433. KResourceLimit threadResourceLimit = ResourceLimit;
  434. KResourceLimit memoryResourceLimit = null;
  435. if (_mainThreadStackSize != 0)
  436. {
  437. throw new InvalidOperationException("Trying to start a process with a invalid state!");
  438. }
  439. ulong stackSizeRounded = BitUtils.AlignUp(stackSize, KMemoryManager.PageSize);
  440. ulong neededSize = stackSizeRounded + _imageSize;
  441. // Check if the needed size for the code and the stack will fit on the
  442. // memory usage capacity of this Process. Also check for possible overflow
  443. // on the above addition.
  444. if (neededSize > _memoryUsageCapacity ||
  445. neededSize < stackSizeRounded)
  446. {
  447. threadResourceLimit?.Release(LimitableResource.Thread, 1);
  448. return KernelResult.OutOfMemory;
  449. }
  450. if (stackSizeRounded != 0 && ResourceLimit != null)
  451. {
  452. memoryResourceLimit = ResourceLimit;
  453. if (!memoryResourceLimit.Reserve(LimitableResource.Memory, stackSizeRounded))
  454. {
  455. threadResourceLimit?.Release(LimitableResource.Thread, 1);
  456. return KernelResult.ResLimitExceeded;
  457. }
  458. }
  459. KernelResult result;
  460. KThread mainThread = null;
  461. ulong stackTop = 0;
  462. void CleanUpForError()
  463. {
  464. HandleTable.Destroy();
  465. mainThread?.DecrementReferenceCount();
  466. if (_mainThreadStackSize != 0)
  467. {
  468. ulong stackBottom = stackTop - _mainThreadStackSize;
  469. ulong stackPagesCount = _mainThreadStackSize / KMemoryManager.PageSize;
  470. MemoryManager.UnmapForKernel(stackBottom, stackPagesCount, MemoryState.Stack);
  471. _mainThreadStackSize = 0;
  472. }
  473. memoryResourceLimit?.Release(LimitableResource.Memory, stackSizeRounded);
  474. threadResourceLimit?.Release(LimitableResource.Thread, 1);
  475. }
  476. if (stackSizeRounded != 0)
  477. {
  478. ulong stackPagesCount = stackSizeRounded / KMemoryManager.PageSize;
  479. ulong regionStart = MemoryManager.StackRegionStart;
  480. ulong regionSize = MemoryManager.StackRegionEnd - regionStart;
  481. ulong regionPagesCount = regionSize / KMemoryManager.PageSize;
  482. result = MemoryManager.AllocateOrMapPa(
  483. stackPagesCount,
  484. KMemoryManager.PageSize,
  485. 0,
  486. false,
  487. regionStart,
  488. regionPagesCount,
  489. MemoryState.Stack,
  490. MemoryPermission.ReadAndWrite,
  491. out ulong stackBottom);
  492. if (result != KernelResult.Success)
  493. {
  494. CleanUpForError();
  495. return result;
  496. }
  497. _mainThreadStackSize += stackSizeRounded;
  498. stackTop = stackBottom + stackSizeRounded;
  499. }
  500. ulong heapCapacity = _memoryUsageCapacity - _mainThreadStackSize - _imageSize;
  501. result = MemoryManager.SetHeapCapacity(heapCapacity);
  502. if (result != KernelResult.Success)
  503. {
  504. CleanUpForError();
  505. return result;
  506. }
  507. HandleTable = new KHandleTable(KernelContext);
  508. result = HandleTable.Initialize(Capabilities.HandleTableSize);
  509. if (result != KernelResult.Success)
  510. {
  511. CleanUpForError();
  512. return result;
  513. }
  514. mainThread = new KThread(KernelContext);
  515. result = mainThread.Initialize(
  516. _entrypoint,
  517. 0,
  518. stackTop,
  519. mainThreadPriority,
  520. DefaultCpuCore,
  521. this);
  522. if (result != KernelResult.Success)
  523. {
  524. CleanUpForError();
  525. return result;
  526. }
  527. result = HandleTable.GenerateHandle(mainThread, out int mainThreadHandle);
  528. if (result != KernelResult.Success)
  529. {
  530. CleanUpForError();
  531. return result;
  532. }
  533. mainThread.SetEntryArguments(0, mainThreadHandle);
  534. ProcessState oldState = State;
  535. ProcessState newState = State != ProcessState.Created
  536. ? ProcessState.Attached
  537. : ProcessState.Started;
  538. SetState(newState);
  539. // TODO: We can't call KThread.Start from a non-guest thread.
  540. // We will need to make some changes to allow the creation of
  541. // dummy threads that will be used to initialize the current
  542. // thread on KCoreContext so that GetCurrentThread doesn't fail.
  543. /* Result = MainThread.Start();
  544. if (Result != KernelResult.Success)
  545. {
  546. SetState(OldState);
  547. CleanUpForError();
  548. } */
  549. mainThread.Reschedule(ThreadSchedState.Running);
  550. if (result == KernelResult.Success)
  551. {
  552. mainThread.IncrementReferenceCount();
  553. }
  554. mainThread.DecrementReferenceCount();
  555. return result;
  556. }
  557. }
  558. private void SetState(ProcessState newState)
  559. {
  560. if (State != newState)
  561. {
  562. State = newState;
  563. _signaled = true;
  564. Signal();
  565. }
  566. }
  567. public KernelResult InitializeThread(
  568. KThread thread,
  569. ulong entrypoint,
  570. ulong argsPtr,
  571. ulong stackTop,
  572. int priority,
  573. int cpuCore)
  574. {
  575. lock (_processLock)
  576. {
  577. return thread.Initialize(entrypoint, argsPtr, stackTop, priority, cpuCore, this);
  578. }
  579. }
  580. public void SubscribeThreadEventHandlers(ARMeilleure.State.ExecutionContext context)
  581. {
  582. context.Interrupt += InterruptHandler;
  583. context.SupervisorCall += KernelContext.SyscallHandler.SvcCall;
  584. context.Undefined += UndefinedInstructionHandler;
  585. }
  586. private void InterruptHandler(object sender, EventArgs e)
  587. {
  588. KernelContext.Scheduler.ContextSwitch();
  589. }
  590. public void IncrementThreadCount()
  591. {
  592. Interlocked.Increment(ref _threadCount);
  593. KernelContext.ThreadCounter.AddCount();
  594. }
  595. public void DecrementThreadCountAndTerminateIfZero()
  596. {
  597. KernelContext.ThreadCounter.Signal();
  598. if (Interlocked.Decrement(ref _threadCount) == 0)
  599. {
  600. Terminate();
  601. }
  602. }
  603. public void DecrementToZeroWhileTerminatingCurrent()
  604. {
  605. KernelContext.ThreadCounter.Signal();
  606. while (Interlocked.Decrement(ref _threadCount) != 0)
  607. {
  608. Destroy();
  609. TerminateCurrentProcess();
  610. }
  611. // Nintendo panic here because if it reaches this point, the current thread should be already dead.
  612. // As we handle the death of the thread in the post SVC handler and inside the CPU emulator, we don't panic here.
  613. }
  614. public ulong GetMemoryCapacity()
  615. {
  616. ulong totalCapacity = (ulong)ResourceLimit.GetRemainingValue(LimitableResource.Memory);
  617. totalCapacity += MemoryManager.GetTotalHeapSize();
  618. totalCapacity += GetPersonalMmHeapSize();
  619. totalCapacity += _imageSize + _mainThreadStackSize;
  620. if (totalCapacity <= _memoryUsageCapacity)
  621. {
  622. return totalCapacity;
  623. }
  624. return _memoryUsageCapacity;
  625. }
  626. public ulong GetMemoryUsage()
  627. {
  628. return _imageSize + _mainThreadStackSize + MemoryManager.GetTotalHeapSize() + GetPersonalMmHeapSize();
  629. }
  630. public ulong GetMemoryCapacityWithoutPersonalMmHeap()
  631. {
  632. return GetMemoryCapacity() - GetPersonalMmHeapSize();
  633. }
  634. public ulong GetMemoryUsageWithoutPersonalMmHeap()
  635. {
  636. return GetMemoryUsage() - GetPersonalMmHeapSize();
  637. }
  638. private ulong GetPersonalMmHeapSize()
  639. {
  640. return GetPersonalMmHeapSize(PersonalMmHeapPagesCount, _memRegion);
  641. }
  642. private static ulong GetPersonalMmHeapSize(ulong personalMmHeapPagesCount, MemoryRegion memRegion)
  643. {
  644. if (memRegion == MemoryRegion.Applet)
  645. {
  646. return 0;
  647. }
  648. return personalMmHeapPagesCount * KMemoryManager.PageSize;
  649. }
  650. public void AddThread(KThread thread)
  651. {
  652. lock (_threadingLock)
  653. {
  654. thread.ProcessListNode = _threads.AddLast(thread);
  655. }
  656. }
  657. public void RemoveThread(KThread thread)
  658. {
  659. lock (_threadingLock)
  660. {
  661. _threads.Remove(thread.ProcessListNode);
  662. }
  663. }
  664. public bool IsCpuCoreAllowed(int core)
  665. {
  666. return (Capabilities.AllowedCpuCoresMask & (1L << core)) != 0;
  667. }
  668. public bool IsPriorityAllowed(int priority)
  669. {
  670. return (Capabilities.AllowedThreadPriosMask & (1L << priority)) != 0;
  671. }
  672. public override bool IsSignaled()
  673. {
  674. return _signaled;
  675. }
  676. public KernelResult Terminate()
  677. {
  678. KernelResult result;
  679. bool shallTerminate = false;
  680. KernelContext.CriticalSection.Enter();
  681. lock (_processLock)
  682. {
  683. if (State >= ProcessState.Started)
  684. {
  685. if (State == ProcessState.Started ||
  686. State == ProcessState.Crashed ||
  687. State == ProcessState.Attached ||
  688. State == ProcessState.DebugSuspended)
  689. {
  690. SetState(ProcessState.Exiting);
  691. shallTerminate = true;
  692. }
  693. result = KernelResult.Success;
  694. }
  695. else
  696. {
  697. result = KernelResult.InvalidState;
  698. }
  699. }
  700. KernelContext.CriticalSection.Leave();
  701. if (shallTerminate)
  702. {
  703. UnpauseAndTerminateAllThreadsExcept(KernelContext.Scheduler.GetCurrentThread());
  704. HandleTable.Destroy();
  705. SignalExitToDebugTerminated();
  706. SignalExit();
  707. }
  708. return result;
  709. }
  710. public void TerminateCurrentProcess()
  711. {
  712. bool shallTerminate = false;
  713. KernelContext.CriticalSection.Enter();
  714. lock (_processLock)
  715. {
  716. if (State >= ProcessState.Started)
  717. {
  718. if (State == ProcessState.Started ||
  719. State == ProcessState.Attached ||
  720. State == ProcessState.DebugSuspended)
  721. {
  722. SetState(ProcessState.Exiting);
  723. shallTerminate = true;
  724. }
  725. }
  726. }
  727. KernelContext.CriticalSection.Leave();
  728. if (shallTerminate)
  729. {
  730. UnpauseAndTerminateAllThreadsExcept(KernelContext.Scheduler.GetCurrentThread());
  731. HandleTable.Destroy();
  732. // NOTE: this is supposed to be called in receiving of the mailbox.
  733. SignalExitToDebugExited();
  734. SignalExit();
  735. }
  736. }
  737. private void UnpauseAndTerminateAllThreadsExcept(KThread currentThread)
  738. {
  739. lock (_threadingLock)
  740. {
  741. KernelContext.CriticalSection.Enter();
  742. foreach (KThread thread in _threads)
  743. {
  744. if ((thread.SchedFlags & ThreadSchedState.LowMask) != ThreadSchedState.TerminationPending)
  745. {
  746. thread.PrepareForTermination();
  747. }
  748. }
  749. KernelContext.CriticalSection.Leave();
  750. }
  751. KThread blockedThread = null;
  752. lock (_threadingLock)
  753. {
  754. foreach (KThread thread in _threads)
  755. {
  756. if (thread != currentThread && (thread.SchedFlags & ThreadSchedState.LowMask) != ThreadSchedState.TerminationPending)
  757. {
  758. thread.IncrementReferenceCount();
  759. blockedThread = thread;
  760. break;
  761. }
  762. }
  763. }
  764. if (blockedThread != null)
  765. {
  766. blockedThread.Terminate();
  767. blockedThread.DecrementReferenceCount();
  768. }
  769. }
  770. private void SignalExitToDebugTerminated()
  771. {
  772. // TODO: Debug events.
  773. }
  774. private void SignalExitToDebugExited()
  775. {
  776. // TODO: Debug events.
  777. }
  778. private void SignalExit()
  779. {
  780. if (ResourceLimit != null)
  781. {
  782. ResourceLimit.Release(LimitableResource.Memory, GetMemoryUsage());
  783. }
  784. KernelContext.CriticalSection.Enter();
  785. SetState(ProcessState.Exited);
  786. KernelContext.CriticalSection.Leave();
  787. }
  788. public KernelResult ClearIfNotExited()
  789. {
  790. KernelResult result;
  791. KernelContext.CriticalSection.Enter();
  792. lock (_processLock)
  793. {
  794. if (State != ProcessState.Exited && _signaled)
  795. {
  796. _signaled = false;
  797. result = KernelResult.Success;
  798. }
  799. else
  800. {
  801. result = KernelResult.InvalidState;
  802. }
  803. }
  804. KernelContext.CriticalSection.Leave();
  805. return result;
  806. }
  807. private void InitializeMemoryManager(AddressSpaceType addrSpaceType, MemoryRegion memRegion)
  808. {
  809. int addrSpaceBits = addrSpaceType switch
  810. {
  811. AddressSpaceType.Addr32Bits => 32,
  812. AddressSpaceType.Addr36Bits => 36,
  813. AddressSpaceType.Addr32BitsNoMap => 32,
  814. AddressSpaceType.Addr39Bits => 39,
  815. _ => throw new ArgumentException(nameof(addrSpaceType))
  816. };
  817. CpuMemory = new MemoryManager(KernelContext.Memory, 1UL << addrSpaceBits, InvalidAccessHandler);
  818. CpuContext = new CpuContext(CpuMemory);
  819. // TODO: This should eventually be removed.
  820. // The GPU shouldn't depend on the CPU memory manager at all.
  821. KernelContext.Device.Gpu.SetVmm(CpuMemory);
  822. MemoryManager = new KMemoryManager(KernelContext, CpuMemory);
  823. }
  824. private bool InvalidAccessHandler(ulong va)
  825. {
  826. KernelContext.Scheduler.GetCurrentThreadOrNull()?.PrintGuestStackTrace();
  827. Logger.PrintError(LogClass.Cpu, $"Invalid memory access at virtual address 0x{va:X16}.");
  828. return false;
  829. }
  830. private void UndefinedInstructionHandler(object sender, InstUndefinedEventArgs e)
  831. {
  832. KernelContext.Scheduler.GetCurrentThreadOrNull()?.PrintGuestStackTrace();
  833. throw new UndefinedInstructionException(e.Address, e.OpCode);
  834. }
  835. protected override void Destroy()
  836. {
  837. CpuMemory.Dispose();
  838. }
  839. }
  840. }