KProcess.cs 31 KB

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