KProcess.cs 32 KB

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