Syscall.cs 79 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501
  1. using Ryujinx.Common;
  2. using Ryujinx.Common.Logging;
  3. using Ryujinx.Cpu;
  4. using Ryujinx.HLE.Exceptions;
  5. using Ryujinx.HLE.HOS.Kernel.Common;
  6. using Ryujinx.HLE.HOS.Kernel.Ipc;
  7. using Ryujinx.HLE.HOS.Kernel.Memory;
  8. using Ryujinx.HLE.HOS.Kernel.Process;
  9. using Ryujinx.HLE.HOS.Kernel.Threading;
  10. using Ryujinx.Memory;
  11. using System;
  12. using System.Threading;
  13. namespace Ryujinx.HLE.HOS.Kernel.SupervisorCall
  14. {
  15. class Syscall
  16. {
  17. private readonly KernelContext _context;
  18. public Syscall(KernelContext context)
  19. {
  20. _context = context;
  21. }
  22. // Process
  23. public KernelResult GetProcessId(int handle, out long pid)
  24. {
  25. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  26. KProcess process = currentProcess.HandleTable.GetKProcess(handle);
  27. if (process == null)
  28. {
  29. KThread thread = currentProcess.HandleTable.GetKThread(handle);
  30. if (thread != null)
  31. {
  32. process = thread.Owner;
  33. }
  34. // TODO: KDebugEvent.
  35. }
  36. pid = process?.Pid ?? 0;
  37. return process != null
  38. ? KernelResult.Success
  39. : KernelResult.InvalidHandle;
  40. }
  41. public KernelResult CreateProcess(
  42. ProcessCreationInfo info,
  43. ReadOnlySpan<int> capabilities,
  44. out int handle,
  45. IProcessContextFactory contextFactory,
  46. ThreadStart customThreadStart = null)
  47. {
  48. handle = 0;
  49. if ((info.Flags & ~ProcessCreationFlags.All) != 0)
  50. {
  51. return KernelResult.InvalidEnumValue;
  52. }
  53. // TODO: Address space check.
  54. if ((info.Flags & ProcessCreationFlags.PoolPartitionMask) > ProcessCreationFlags.PoolPartitionSystemNonSecure)
  55. {
  56. return KernelResult.InvalidEnumValue;
  57. }
  58. if ((info.CodeAddress & 0x1fffff) != 0)
  59. {
  60. return KernelResult.InvalidAddress;
  61. }
  62. if (info.CodePagesCount < 0 || info.SystemResourcePagesCount < 0)
  63. {
  64. return KernelResult.InvalidSize;
  65. }
  66. if (info.Flags.HasFlag(ProcessCreationFlags.OptimizeMemoryAllocation) &&
  67. !info.Flags.HasFlag(ProcessCreationFlags.IsApplication))
  68. {
  69. return KernelResult.InvalidThread;
  70. }
  71. KHandleTable handleTable = KernelStatic.GetCurrentProcess().HandleTable;
  72. KProcess process = new KProcess(_context);
  73. using var _ = new OnScopeExit(process.DecrementReferenceCount);
  74. KResourceLimit resourceLimit;
  75. if (info.ResourceLimitHandle != 0)
  76. {
  77. resourceLimit = handleTable.GetObject<KResourceLimit>(info.ResourceLimitHandle);
  78. if (resourceLimit == null)
  79. {
  80. return KernelResult.InvalidHandle;
  81. }
  82. }
  83. else
  84. {
  85. resourceLimit = _context.ResourceLimit;
  86. }
  87. MemoryRegion memRegion = (info.Flags & ProcessCreationFlags.PoolPartitionMask) switch
  88. {
  89. ProcessCreationFlags.PoolPartitionApplication => MemoryRegion.Application,
  90. ProcessCreationFlags.PoolPartitionApplet => MemoryRegion.Applet,
  91. ProcessCreationFlags.PoolPartitionSystem => MemoryRegion.Service,
  92. ProcessCreationFlags.PoolPartitionSystemNonSecure => MemoryRegion.NvServices,
  93. _ => MemoryRegion.NvServices
  94. };
  95. KernelResult result = process.Initialize(
  96. info,
  97. capabilities,
  98. resourceLimit,
  99. memRegion,
  100. contextFactory,
  101. customThreadStart);
  102. if (result != KernelResult.Success)
  103. {
  104. return result;
  105. }
  106. _context.Processes.TryAdd(process.Pid, process);
  107. return handleTable.GenerateHandle(process, out handle);
  108. }
  109. public KernelResult StartProcess(int handle, int priority, int cpuCore, ulong mainThreadStackSize)
  110. {
  111. KProcess process = KernelStatic.GetCurrentProcess().HandleTable.GetObject<KProcess>(handle);
  112. if (process == null)
  113. {
  114. return KernelResult.InvalidHandle;
  115. }
  116. if ((uint)cpuCore >= KScheduler.CpuCoresCount || !process.IsCpuCoreAllowed(cpuCore))
  117. {
  118. return KernelResult.InvalidCpuCore;
  119. }
  120. if ((uint)priority >= KScheduler.PrioritiesCount || !process.IsPriorityAllowed(priority))
  121. {
  122. return KernelResult.InvalidPriority;
  123. }
  124. process.DefaultCpuCore = cpuCore;
  125. KernelResult result = process.Start(priority, mainThreadStackSize);
  126. if (result != KernelResult.Success)
  127. {
  128. return result;
  129. }
  130. process.IncrementReferenceCount();
  131. return KernelResult.Success;
  132. }
  133. // IPC
  134. public KernelResult ConnectToNamedPort(ulong namePtr, out int handle)
  135. {
  136. handle = 0;
  137. if (!KernelTransfer.UserToKernelString(_context, namePtr, 12, out string name))
  138. {
  139. return KernelResult.UserCopyFailed;
  140. }
  141. return ConnectToNamedPort(name, out handle);
  142. }
  143. public KernelResult ConnectToNamedPort(string name, out int handle)
  144. {
  145. handle = 0;
  146. if (name.Length > 11)
  147. {
  148. return KernelResult.MaximumExceeded;
  149. }
  150. KAutoObject autoObj = KAutoObject.FindNamedObject(_context, name);
  151. if (!(autoObj is KClientPort clientPort))
  152. {
  153. return KernelResult.NotFound;
  154. }
  155. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  156. KernelResult result = currentProcess.HandleTable.ReserveHandle(out handle);
  157. if (result != KernelResult.Success)
  158. {
  159. return result;
  160. }
  161. result = clientPort.Connect(out KClientSession clientSession);
  162. if (result != KernelResult.Success)
  163. {
  164. currentProcess.HandleTable.CancelHandleReservation(handle);
  165. return result;
  166. }
  167. currentProcess.HandleTable.SetReservedHandleObj(handle, clientSession);
  168. clientSession.DecrementReferenceCount();
  169. return result;
  170. }
  171. public KernelResult SendSyncRequest(int handle)
  172. {
  173. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  174. KClientSession session = currentProcess.HandleTable.GetObject<KClientSession>(handle);
  175. if (session == null)
  176. {
  177. return KernelResult.InvalidHandle;
  178. }
  179. return session.SendSyncRequest();
  180. }
  181. public KernelResult SendSyncRequestWithUserBuffer(ulong messagePtr, ulong messageSize, int handle)
  182. {
  183. if (!PageAligned(messagePtr))
  184. {
  185. return KernelResult.InvalidAddress;
  186. }
  187. if (!PageAligned(messageSize) || messageSize == 0)
  188. {
  189. return KernelResult.InvalidSize;
  190. }
  191. if (messagePtr + messageSize <= messagePtr)
  192. {
  193. return KernelResult.InvalidMemState;
  194. }
  195. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  196. KernelResult result = currentProcess.MemoryManager.BorrowIpcBuffer(messagePtr, messageSize);
  197. if (result != KernelResult.Success)
  198. {
  199. return result;
  200. }
  201. KClientSession session = currentProcess.HandleTable.GetObject<KClientSession>(handle);
  202. if (session == null)
  203. {
  204. result = KernelResult.InvalidHandle;
  205. }
  206. else
  207. {
  208. result = session.SendSyncRequest(messagePtr, messageSize);
  209. }
  210. KernelResult result2 = currentProcess.MemoryManager.UnborrowIpcBuffer(messagePtr, messageSize);
  211. if (result == KernelResult.Success)
  212. {
  213. result = result2;
  214. }
  215. return result;
  216. }
  217. public KernelResult SendAsyncRequestWithUserBuffer(ulong messagePtr, ulong messageSize, int handle, out int doneEventHandle)
  218. {
  219. doneEventHandle = 0;
  220. if (!PageAligned(messagePtr))
  221. {
  222. return KernelResult.InvalidAddress;
  223. }
  224. if (!PageAligned(messageSize) || messageSize == 0)
  225. {
  226. return KernelResult.InvalidSize;
  227. }
  228. if (messagePtr + messageSize <= messagePtr)
  229. {
  230. return KernelResult.InvalidMemState;
  231. }
  232. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  233. KernelResult result = currentProcess.MemoryManager.BorrowIpcBuffer(messagePtr, messageSize);
  234. if (result != KernelResult.Success)
  235. {
  236. return result;
  237. }
  238. KResourceLimit resourceLimit = currentProcess.ResourceLimit;
  239. if (resourceLimit != null && !resourceLimit.Reserve(LimitableResource.Event, 1))
  240. {
  241. currentProcess.MemoryManager.UnborrowIpcBuffer(messagePtr, messageSize);
  242. return KernelResult.ResLimitExceeded;
  243. }
  244. KClientSession session = currentProcess.HandleTable.GetObject<KClientSession>(handle);
  245. if (session == null)
  246. {
  247. result = KernelResult.InvalidHandle;
  248. }
  249. else
  250. {
  251. KEvent doneEvent = new KEvent(_context);
  252. result = currentProcess.HandleTable.GenerateHandle(doneEvent.ReadableEvent, out doneEventHandle);
  253. if (result == KernelResult.Success)
  254. {
  255. result = session.SendAsyncRequest(doneEvent.WritableEvent, messagePtr, messageSize);
  256. if (result != KernelResult.Success)
  257. {
  258. currentProcess.HandleTable.CloseHandle(doneEventHandle);
  259. }
  260. }
  261. }
  262. if (result != KernelResult.Success)
  263. {
  264. resourceLimit?.Release(LimitableResource.Event, 1);
  265. currentProcess.MemoryManager.UnborrowIpcBuffer(messagePtr, messageSize);
  266. }
  267. return result;
  268. }
  269. public KernelResult CreateSession(
  270. bool isLight,
  271. ulong namePtr,
  272. out int serverSessionHandle,
  273. out int clientSessionHandle)
  274. {
  275. serverSessionHandle = 0;
  276. clientSessionHandle = 0;
  277. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  278. KResourceLimit resourceLimit = currentProcess.ResourceLimit;
  279. if (resourceLimit != null && !resourceLimit.Reserve(LimitableResource.Session, 1))
  280. {
  281. return KernelResult.ResLimitExceeded;
  282. }
  283. KernelResult result;
  284. if (isLight)
  285. {
  286. KLightSession session = new KLightSession(_context);
  287. result = currentProcess.HandleTable.GenerateHandle(session.ServerSession, out serverSessionHandle);
  288. if (result == KernelResult.Success)
  289. {
  290. result = currentProcess.HandleTable.GenerateHandle(session.ClientSession, out clientSessionHandle);
  291. if (result != KernelResult.Success)
  292. {
  293. currentProcess.HandleTable.CloseHandle(serverSessionHandle);
  294. serverSessionHandle = 0;
  295. }
  296. }
  297. session.ServerSession.DecrementReferenceCount();
  298. session.ClientSession.DecrementReferenceCount();
  299. }
  300. else
  301. {
  302. KSession session = new KSession(_context);
  303. result = currentProcess.HandleTable.GenerateHandle(session.ServerSession, out serverSessionHandle);
  304. if (result == KernelResult.Success)
  305. {
  306. result = currentProcess.HandleTable.GenerateHandle(session.ClientSession, out clientSessionHandle);
  307. if (result != KernelResult.Success)
  308. {
  309. currentProcess.HandleTable.CloseHandle(serverSessionHandle);
  310. serverSessionHandle = 0;
  311. }
  312. }
  313. session.ServerSession.DecrementReferenceCount();
  314. session.ClientSession.DecrementReferenceCount();
  315. }
  316. return result;
  317. }
  318. public KernelResult AcceptSession(int portHandle, out int sessionHandle)
  319. {
  320. sessionHandle = 0;
  321. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  322. KServerPort serverPort = currentProcess.HandleTable.GetObject<KServerPort>(portHandle);
  323. if (serverPort == null)
  324. {
  325. return KernelResult.InvalidHandle;
  326. }
  327. KernelResult result = currentProcess.HandleTable.ReserveHandle(out int handle);
  328. if (result != KernelResult.Success)
  329. {
  330. return result;
  331. }
  332. KAutoObject session;
  333. if (serverPort.IsLight)
  334. {
  335. session = serverPort.AcceptIncomingLightConnection();
  336. }
  337. else
  338. {
  339. session = serverPort.AcceptIncomingConnection();
  340. }
  341. if (session != null)
  342. {
  343. currentProcess.HandleTable.SetReservedHandleObj(handle, session);
  344. session.DecrementReferenceCount();
  345. sessionHandle = handle;
  346. result = KernelResult.Success;
  347. }
  348. else
  349. {
  350. currentProcess.HandleTable.CancelHandleReservation(handle);
  351. result = KernelResult.NotFound;
  352. }
  353. return result;
  354. }
  355. public KernelResult ReplyAndReceive(
  356. ulong handlesPtr,
  357. int handlesCount,
  358. int replyTargetHandle,
  359. long timeout,
  360. out int handleIndex)
  361. {
  362. handleIndex = 0;
  363. if ((uint)handlesCount > 0x40)
  364. {
  365. return KernelResult.MaximumExceeded;
  366. }
  367. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  368. ulong copySize = (ulong)((long)handlesCount * 4);
  369. if (!currentProcess.MemoryManager.InsideAddrSpace(handlesPtr, copySize))
  370. {
  371. return KernelResult.UserCopyFailed;
  372. }
  373. if (handlesPtr + copySize < handlesPtr)
  374. {
  375. return KernelResult.UserCopyFailed;
  376. }
  377. int[] handles = new int[handlesCount];
  378. if (!KernelTransfer.UserToKernelInt32Array(_context, handlesPtr, handles))
  379. {
  380. return KernelResult.UserCopyFailed;
  381. }
  382. return ReplyAndReceive(handles, replyTargetHandle, timeout, out handleIndex);
  383. }
  384. public KernelResult ReplyAndReceive(ReadOnlySpan<int> handles, int replyTargetHandle, long timeout, out int handleIndex)
  385. {
  386. handleIndex = 0;
  387. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  388. KSynchronizationObject[] syncObjs = new KSynchronizationObject[handles.Length];
  389. for (int index = 0; index < handles.Length; index++)
  390. {
  391. KSynchronizationObject obj = currentProcess.HandleTable.GetObject<KSynchronizationObject>(handles[index]);
  392. if (obj == null)
  393. {
  394. return KernelResult.InvalidHandle;
  395. }
  396. syncObjs[index] = obj;
  397. }
  398. KernelResult result = KernelResult.Success;
  399. if (replyTargetHandle != 0)
  400. {
  401. KServerSession replyTarget = currentProcess.HandleTable.GetObject<KServerSession>(replyTargetHandle);
  402. if (replyTarget == null)
  403. {
  404. result = KernelResult.InvalidHandle;
  405. }
  406. else
  407. {
  408. result = replyTarget.Reply();
  409. }
  410. }
  411. if (result == KernelResult.Success)
  412. {
  413. while ((result = _context.Synchronization.WaitFor(syncObjs, timeout, out handleIndex)) == KernelResult.Success)
  414. {
  415. KServerSession session = currentProcess.HandleTable.GetObject<KServerSession>(handles[handleIndex]);
  416. if (session == null)
  417. {
  418. break;
  419. }
  420. if ((result = session.Receive()) != KernelResult.NotFound)
  421. {
  422. break;
  423. }
  424. }
  425. }
  426. return result;
  427. }
  428. public KernelResult ReplyAndReceiveWithUserBuffer(
  429. ulong handlesPtr,
  430. ulong messagePtr,
  431. ulong messageSize,
  432. int handlesCount,
  433. int replyTargetHandle,
  434. long timeout,
  435. out int handleIndex)
  436. {
  437. handleIndex = 0;
  438. if ((uint)handlesCount > 0x40)
  439. {
  440. return KernelResult.MaximumExceeded;
  441. }
  442. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  443. ulong copySize = (ulong)((long)handlesCount * 4);
  444. if (!currentProcess.MemoryManager.InsideAddrSpace(handlesPtr, copySize))
  445. {
  446. return KernelResult.UserCopyFailed;
  447. }
  448. if (handlesPtr + copySize < handlesPtr)
  449. {
  450. return KernelResult.UserCopyFailed;
  451. }
  452. KernelResult result = currentProcess.MemoryManager.BorrowIpcBuffer(messagePtr, messageSize);
  453. if (result != KernelResult.Success)
  454. {
  455. return result;
  456. }
  457. int[] handles = new int[handlesCount];
  458. if (!KernelTransfer.UserToKernelInt32Array(_context, handlesPtr, handles))
  459. {
  460. currentProcess.MemoryManager.UnborrowIpcBuffer(messagePtr, messageSize);
  461. return KernelResult.UserCopyFailed;
  462. }
  463. KSynchronizationObject[] syncObjs = new KSynchronizationObject[handlesCount];
  464. for (int index = 0; index < handlesCount; index++)
  465. {
  466. KSynchronizationObject obj = currentProcess.HandleTable.GetObject<KSynchronizationObject>(handles[index]);
  467. if (obj == null)
  468. {
  469. currentProcess.MemoryManager.UnborrowIpcBuffer(messagePtr, messageSize);
  470. return KernelResult.InvalidHandle;
  471. }
  472. syncObjs[index] = obj;
  473. }
  474. if (replyTargetHandle != 0)
  475. {
  476. KServerSession replyTarget = currentProcess.HandleTable.GetObject<KServerSession>(replyTargetHandle);
  477. if (replyTarget == null)
  478. {
  479. result = KernelResult.InvalidHandle;
  480. }
  481. else
  482. {
  483. result = replyTarget.Reply(messagePtr, messageSize);
  484. }
  485. }
  486. if (result == KernelResult.Success)
  487. {
  488. while ((result = _context.Synchronization.WaitFor(syncObjs, timeout, out handleIndex)) == KernelResult.Success)
  489. {
  490. KServerSession session = currentProcess.HandleTable.GetObject<KServerSession>(handles[handleIndex]);
  491. if (session == null)
  492. {
  493. break;
  494. }
  495. if ((result = session.Receive(messagePtr, messageSize)) != KernelResult.NotFound)
  496. {
  497. break;
  498. }
  499. }
  500. }
  501. currentProcess.MemoryManager.UnborrowIpcBuffer(messagePtr, messageSize);
  502. return result;
  503. }
  504. public KernelResult CreatePort(
  505. int maxSessions,
  506. bool isLight,
  507. ulong namePtr,
  508. out int serverPortHandle,
  509. out int clientPortHandle)
  510. {
  511. serverPortHandle = clientPortHandle = 0;
  512. if (maxSessions < 1)
  513. {
  514. return KernelResult.MaximumExceeded;
  515. }
  516. KPort port = new KPort(_context, maxSessions, isLight, (long)namePtr);
  517. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  518. KernelResult result = currentProcess.HandleTable.GenerateHandle(port.ClientPort, out clientPortHandle);
  519. if (result != KernelResult.Success)
  520. {
  521. return result;
  522. }
  523. result = currentProcess.HandleTable.GenerateHandle(port.ServerPort, out serverPortHandle);
  524. if (result != KernelResult.Success)
  525. {
  526. currentProcess.HandleTable.CloseHandle(clientPortHandle);
  527. }
  528. return result;
  529. }
  530. public KernelResult ManageNamedPort(ulong namePtr, int maxSessions, out int handle)
  531. {
  532. handle = 0;
  533. if (!KernelTransfer.UserToKernelString(_context, namePtr, 12, out string name))
  534. {
  535. return KernelResult.UserCopyFailed;
  536. }
  537. if (name.Length > 11)
  538. {
  539. return KernelResult.MaximumExceeded;
  540. }
  541. return ManageNamedPort(name, maxSessions, out handle);
  542. }
  543. public KernelResult ManageNamedPort(string name, int maxSessions, out int handle)
  544. {
  545. handle = 0;
  546. if (maxSessions < 0)
  547. {
  548. return KernelResult.MaximumExceeded;
  549. }
  550. if (maxSessions == 0)
  551. {
  552. return KAutoObject.RemoveName(_context, name);
  553. }
  554. KPort port = new KPort(_context, maxSessions, false, 0);
  555. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  556. KernelResult result = currentProcess.HandleTable.GenerateHandle(port.ServerPort, out handle);
  557. if (result != KernelResult.Success)
  558. {
  559. return result;
  560. }
  561. result = port.ClientPort.SetName(name);
  562. if (result != KernelResult.Success)
  563. {
  564. currentProcess.HandleTable.CloseHandle(handle);
  565. }
  566. return result;
  567. }
  568. public KernelResult ConnectToPort(int clientPortHandle, out int clientSessionHandle)
  569. {
  570. clientSessionHandle = 0;
  571. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  572. KClientPort clientPort = currentProcess.HandleTable.GetObject<KClientPort>(clientPortHandle);
  573. if (clientPort == null)
  574. {
  575. return KernelResult.InvalidHandle;
  576. }
  577. KernelResult result = currentProcess.HandleTable.ReserveHandle(out int handle);
  578. if (result != KernelResult.Success)
  579. {
  580. return result;
  581. }
  582. KAutoObject session;
  583. if (clientPort.IsLight)
  584. {
  585. result = clientPort.ConnectLight(out KLightClientSession clientSession);
  586. session = clientSession;
  587. }
  588. else
  589. {
  590. result = clientPort.Connect(out KClientSession clientSession);
  591. session = clientSession;
  592. }
  593. if (result != KernelResult.Success)
  594. {
  595. currentProcess.HandleTable.CancelHandleReservation(handle);
  596. return result;
  597. }
  598. currentProcess.HandleTable.SetReservedHandleObj(handle, session);
  599. session.DecrementReferenceCount();
  600. clientSessionHandle = handle;
  601. return result;
  602. }
  603. // Memory
  604. public KernelResult SetHeapSize(ulong size, out ulong position)
  605. {
  606. if ((size & 0xfffffffe001fffff) != 0)
  607. {
  608. position = 0;
  609. return KernelResult.InvalidSize;
  610. }
  611. KProcess process = KernelStatic.GetCurrentProcess();
  612. return process.MemoryManager.SetHeapSize(size, out position);
  613. }
  614. public KernelResult SetMemoryAttribute(
  615. ulong position,
  616. ulong size,
  617. MemoryAttribute attributeMask,
  618. MemoryAttribute attributeValue)
  619. {
  620. if (!PageAligned(position))
  621. {
  622. return KernelResult.InvalidAddress;
  623. }
  624. if (!PageAligned(size) || size == 0)
  625. {
  626. return KernelResult.InvalidSize;
  627. }
  628. MemoryAttribute attributes = attributeMask | attributeValue;
  629. if (attributes != attributeMask ||
  630. (attributes | MemoryAttribute.Uncached) != MemoryAttribute.Uncached)
  631. {
  632. return KernelResult.InvalidCombination;
  633. }
  634. KProcess process = KernelStatic.GetCurrentProcess();
  635. if (!process.MemoryManager.InsideAddrSpace(position, size))
  636. {
  637. return KernelResult.InvalidMemState;
  638. }
  639. KernelResult result = process.MemoryManager.SetMemoryAttribute(
  640. position,
  641. size,
  642. attributeMask,
  643. attributeValue);
  644. return result;
  645. }
  646. public KernelResult MapMemory(ulong dst, ulong src, ulong size)
  647. {
  648. if (!PageAligned(src | dst))
  649. {
  650. return KernelResult.InvalidAddress;
  651. }
  652. if (!PageAligned(size) || size == 0)
  653. {
  654. return KernelResult.InvalidSize;
  655. }
  656. if (src + size <= src || dst + size <= dst)
  657. {
  658. return KernelResult.InvalidMemState;
  659. }
  660. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  661. if (!currentProcess.MemoryManager.InsideAddrSpace(src, size))
  662. {
  663. return KernelResult.InvalidMemState;
  664. }
  665. if (currentProcess.MemoryManager.OutsideStackRegion(dst, size) ||
  666. currentProcess.MemoryManager.InsideHeapRegion(dst, size) ||
  667. currentProcess.MemoryManager.InsideAliasRegion(dst, size))
  668. {
  669. return KernelResult.InvalidMemRange;
  670. }
  671. KProcess process = KernelStatic.GetCurrentProcess();
  672. return process.MemoryManager.Map(dst, src, size);
  673. }
  674. public KernelResult UnmapMemory(ulong dst, ulong src, ulong size)
  675. {
  676. if (!PageAligned(src | dst))
  677. {
  678. return KernelResult.InvalidAddress;
  679. }
  680. if (!PageAligned(size) || size == 0)
  681. {
  682. return KernelResult.InvalidSize;
  683. }
  684. if (src + size <= src || dst + size <= dst)
  685. {
  686. return KernelResult.InvalidMemState;
  687. }
  688. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  689. if (!currentProcess.MemoryManager.InsideAddrSpace(src, size))
  690. {
  691. return KernelResult.InvalidMemState;
  692. }
  693. if (currentProcess.MemoryManager.OutsideStackRegion(dst, size) ||
  694. currentProcess.MemoryManager.InsideHeapRegion(dst, size) ||
  695. currentProcess.MemoryManager.InsideAliasRegion(dst, size))
  696. {
  697. return KernelResult.InvalidMemRange;
  698. }
  699. KProcess process = KernelStatic.GetCurrentProcess();
  700. return process.MemoryManager.Unmap(dst, src, size);
  701. }
  702. public KernelResult QueryMemory(ulong infoPtr, ulong position, out ulong pageInfo)
  703. {
  704. KProcess process = KernelStatic.GetCurrentProcess();
  705. KMemoryInfo blkInfo = process.MemoryManager.QueryMemory(position);
  706. process.CpuMemory.Write(infoPtr + 0x00, blkInfo.Address);
  707. process.CpuMemory.Write(infoPtr + 0x08, blkInfo.Size);
  708. process.CpuMemory.Write(infoPtr + 0x10, (int)blkInfo.State & 0xff);
  709. process.CpuMemory.Write(infoPtr + 0x14, (int)blkInfo.Attribute);
  710. process.CpuMemory.Write(infoPtr + 0x18, (int)blkInfo.Permission);
  711. process.CpuMemory.Write(infoPtr + 0x1c, blkInfo.IpcRefCount);
  712. process.CpuMemory.Write(infoPtr + 0x20, blkInfo.DeviceRefCount);
  713. process.CpuMemory.Write(infoPtr + 0x24, 0);
  714. pageInfo = 0;
  715. return KernelResult.Success;
  716. }
  717. public KernelResult MapSharedMemory(int handle, ulong address, ulong size, KMemoryPermission permission)
  718. {
  719. if (!PageAligned(address))
  720. {
  721. return KernelResult.InvalidAddress;
  722. }
  723. if (!PageAligned(size) || size == 0)
  724. {
  725. return KernelResult.InvalidSize;
  726. }
  727. if (address + size <= address)
  728. {
  729. return KernelResult.InvalidMemState;
  730. }
  731. if ((permission | KMemoryPermission.Write) != KMemoryPermission.ReadAndWrite)
  732. {
  733. return KernelResult.InvalidPermission;
  734. }
  735. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  736. KSharedMemory sharedMemory = currentProcess.HandleTable.GetObject<KSharedMemory>(handle);
  737. if (sharedMemory == null)
  738. {
  739. return KernelResult.InvalidHandle;
  740. }
  741. if (currentProcess.MemoryManager.IsInvalidRegion(address, size) ||
  742. currentProcess.MemoryManager.InsideHeapRegion(address, size) ||
  743. currentProcess.MemoryManager.InsideAliasRegion(address, size))
  744. {
  745. return KernelResult.InvalidMemRange;
  746. }
  747. return sharedMemory.MapIntoProcess(
  748. currentProcess.MemoryManager,
  749. address,
  750. size,
  751. currentProcess,
  752. permission);
  753. }
  754. public KernelResult UnmapSharedMemory(int handle, ulong address, ulong size)
  755. {
  756. if (!PageAligned(address))
  757. {
  758. return KernelResult.InvalidAddress;
  759. }
  760. if (!PageAligned(size) || size == 0)
  761. {
  762. return KernelResult.InvalidSize;
  763. }
  764. if (address + size <= address)
  765. {
  766. return KernelResult.InvalidMemState;
  767. }
  768. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  769. KSharedMemory sharedMemory = currentProcess.HandleTable.GetObject<KSharedMemory>(handle);
  770. if (sharedMemory == null)
  771. {
  772. return KernelResult.InvalidHandle;
  773. }
  774. if (currentProcess.MemoryManager.IsInvalidRegion(address, size) ||
  775. currentProcess.MemoryManager.InsideHeapRegion(address, size) ||
  776. currentProcess.MemoryManager.InsideAliasRegion(address, size))
  777. {
  778. return KernelResult.InvalidMemRange;
  779. }
  780. return sharedMemory.UnmapFromProcess(
  781. currentProcess.MemoryManager,
  782. address,
  783. size,
  784. currentProcess);
  785. }
  786. public KernelResult CreateTransferMemory(ulong address, ulong size, KMemoryPermission permission, out int handle)
  787. {
  788. handle = 0;
  789. if (!PageAligned(address))
  790. {
  791. return KernelResult.InvalidAddress;
  792. }
  793. if (!PageAligned(size) || size == 0)
  794. {
  795. return KernelResult.InvalidSize;
  796. }
  797. if (address + size <= address)
  798. {
  799. return KernelResult.InvalidMemState;
  800. }
  801. if (permission > KMemoryPermission.ReadAndWrite || permission == KMemoryPermission.Write)
  802. {
  803. return KernelResult.InvalidPermission;
  804. }
  805. KProcess process = KernelStatic.GetCurrentProcess();
  806. KResourceLimit resourceLimit = process.ResourceLimit;
  807. if (resourceLimit != null && !resourceLimit.Reserve(LimitableResource.TransferMemory, 1))
  808. {
  809. return KernelResult.ResLimitExceeded;
  810. }
  811. void CleanUpForError()
  812. {
  813. resourceLimit?.Release(LimitableResource.TransferMemory, 1);
  814. }
  815. if (!process.MemoryManager.InsideAddrSpace(address, size))
  816. {
  817. CleanUpForError();
  818. return KernelResult.InvalidMemState;
  819. }
  820. KTransferMemory transferMemory = new KTransferMemory(_context);
  821. KernelResult result = transferMemory.Initialize(address, size, permission);
  822. if (result != KernelResult.Success)
  823. {
  824. CleanUpForError();
  825. return result;
  826. }
  827. result = process.HandleTable.GenerateHandle(transferMemory, out handle);
  828. transferMemory.DecrementReferenceCount();
  829. return result;
  830. }
  831. public KernelResult MapTransferMemory(int handle, ulong address, ulong size, KMemoryPermission permission)
  832. {
  833. if (!PageAligned(address))
  834. {
  835. return KernelResult.InvalidAddress;
  836. }
  837. if (!PageAligned(size) || size == 0)
  838. {
  839. return KernelResult.InvalidSize;
  840. }
  841. if (address + size <= address)
  842. {
  843. return KernelResult.InvalidMemState;
  844. }
  845. if (permission > KMemoryPermission.ReadAndWrite || permission == KMemoryPermission.Write)
  846. {
  847. return KernelResult.InvalidPermission;
  848. }
  849. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  850. KTransferMemory transferMemory = currentProcess.HandleTable.GetObject<KTransferMemory>(handle);
  851. if (transferMemory == null)
  852. {
  853. return KernelResult.InvalidHandle;
  854. }
  855. if (currentProcess.MemoryManager.IsInvalidRegion(address, size) ||
  856. currentProcess.MemoryManager.InsideHeapRegion(address, size) ||
  857. currentProcess.MemoryManager.InsideAliasRegion(address, size))
  858. {
  859. return KernelResult.InvalidMemRange;
  860. }
  861. return transferMemory.MapIntoProcess(
  862. currentProcess.MemoryManager,
  863. address,
  864. size,
  865. currentProcess,
  866. permission);
  867. }
  868. public KernelResult UnmapTransferMemory(int handle, ulong address, ulong size)
  869. {
  870. if (!PageAligned(address))
  871. {
  872. return KernelResult.InvalidAddress;
  873. }
  874. if (!PageAligned(size) || size == 0)
  875. {
  876. return KernelResult.InvalidSize;
  877. }
  878. if (address + size <= address)
  879. {
  880. return KernelResult.InvalidMemState;
  881. }
  882. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  883. KTransferMemory transferMemory = currentProcess.HandleTable.GetObject<KTransferMemory>(handle);
  884. if (transferMemory == null)
  885. {
  886. return KernelResult.InvalidHandle;
  887. }
  888. if (currentProcess.MemoryManager.IsInvalidRegion(address, size) ||
  889. currentProcess.MemoryManager.InsideHeapRegion(address, size) ||
  890. currentProcess.MemoryManager.InsideAliasRegion(address, size))
  891. {
  892. return KernelResult.InvalidMemRange;
  893. }
  894. return transferMemory.UnmapFromProcess(
  895. currentProcess.MemoryManager,
  896. address,
  897. size,
  898. currentProcess);
  899. }
  900. public KernelResult MapPhysicalMemory(ulong address, ulong size)
  901. {
  902. if (!PageAligned(address))
  903. {
  904. return KernelResult.InvalidAddress;
  905. }
  906. if (!PageAligned(size) || size == 0)
  907. {
  908. return KernelResult.InvalidSize;
  909. }
  910. if (address + size <= address)
  911. {
  912. return KernelResult.InvalidMemRange;
  913. }
  914. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  915. if ((currentProcess.PersonalMmHeapPagesCount & 0xfffffffffffff) == 0)
  916. {
  917. return KernelResult.InvalidState;
  918. }
  919. if (!currentProcess.MemoryManager.InsideAddrSpace(address, size) ||
  920. currentProcess.MemoryManager.OutsideAliasRegion(address, size))
  921. {
  922. return KernelResult.InvalidMemRange;
  923. }
  924. KProcess process = KernelStatic.GetCurrentProcess();
  925. return process.MemoryManager.MapPhysicalMemory(address, size);
  926. }
  927. public KernelResult UnmapPhysicalMemory(ulong address, ulong size)
  928. {
  929. if (!PageAligned(address))
  930. {
  931. return KernelResult.InvalidAddress;
  932. }
  933. if (!PageAligned(size) || size == 0)
  934. {
  935. return KernelResult.InvalidSize;
  936. }
  937. if (address + size <= address)
  938. {
  939. return KernelResult.InvalidMemRange;
  940. }
  941. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  942. if ((currentProcess.PersonalMmHeapPagesCount & 0xfffffffffffff) == 0)
  943. {
  944. return KernelResult.InvalidState;
  945. }
  946. if (!currentProcess.MemoryManager.InsideAddrSpace(address, size) ||
  947. currentProcess.MemoryManager.OutsideAliasRegion(address, size))
  948. {
  949. return KernelResult.InvalidMemRange;
  950. }
  951. KProcess process = KernelStatic.GetCurrentProcess();
  952. return process.MemoryManager.UnmapPhysicalMemory(address, size);
  953. }
  954. public KernelResult MapProcessCodeMemory(int handle, ulong dst, ulong src, ulong size)
  955. {
  956. if (!PageAligned(dst) || !PageAligned(src))
  957. {
  958. return KernelResult.InvalidAddress;
  959. }
  960. if (!PageAligned(size) || size == 0)
  961. {
  962. return KernelResult.InvalidSize;
  963. }
  964. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  965. KProcess targetProcess = currentProcess.HandleTable.GetObject<KProcess>(handle);
  966. if (targetProcess == null)
  967. {
  968. return KernelResult.InvalidHandle;
  969. }
  970. if (targetProcess.MemoryManager.OutsideAddrSpace(dst, size) ||
  971. targetProcess.MemoryManager.OutsideAddrSpace(src, size) ||
  972. targetProcess.MemoryManager.InsideAliasRegion(dst, size) ||
  973. targetProcess.MemoryManager.InsideHeapRegion(dst, size))
  974. {
  975. return KernelResult.InvalidMemRange;
  976. }
  977. if (size + dst <= dst || size + src <= src)
  978. {
  979. return KernelResult.InvalidMemState;
  980. }
  981. return targetProcess.MemoryManager.MapProcessCodeMemory(dst, src, size);
  982. }
  983. public KernelResult UnmapProcessCodeMemory(int handle, ulong dst, ulong src, ulong size)
  984. {
  985. if (!PageAligned(dst) || !PageAligned(src))
  986. {
  987. return KernelResult.InvalidAddress;
  988. }
  989. if (!PageAligned(size) || size == 0)
  990. {
  991. return KernelResult.InvalidSize;
  992. }
  993. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  994. KProcess targetProcess = currentProcess.HandleTable.GetObject<KProcess>(handle);
  995. if (targetProcess == null)
  996. {
  997. return KernelResult.InvalidHandle;
  998. }
  999. if (targetProcess.MemoryManager.OutsideAddrSpace(dst, size) ||
  1000. targetProcess.MemoryManager.OutsideAddrSpace(src, size) ||
  1001. targetProcess.MemoryManager.InsideAliasRegion(dst, size) ||
  1002. targetProcess.MemoryManager.InsideHeapRegion(dst, size))
  1003. {
  1004. return KernelResult.InvalidMemRange;
  1005. }
  1006. if (size + dst <= dst || size + src <= src)
  1007. {
  1008. return KernelResult.InvalidMemState;
  1009. }
  1010. return targetProcess.MemoryManager.UnmapProcessCodeMemory(dst, src, size);
  1011. }
  1012. public KernelResult SetProcessMemoryPermission(int handle, ulong src, ulong size, KMemoryPermission permission)
  1013. {
  1014. if (!PageAligned(src))
  1015. {
  1016. return KernelResult.InvalidAddress;
  1017. }
  1018. if (!PageAligned(size) || size == 0)
  1019. {
  1020. return KernelResult.InvalidSize;
  1021. }
  1022. if (permission != KMemoryPermission.None &&
  1023. permission != KMemoryPermission.Read &&
  1024. permission != KMemoryPermission.ReadAndWrite &&
  1025. permission != KMemoryPermission.ReadAndExecute)
  1026. {
  1027. return KernelResult.InvalidPermission;
  1028. }
  1029. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1030. KProcess targetProcess = currentProcess.HandleTable.GetObject<KProcess>(handle);
  1031. if (targetProcess == null)
  1032. {
  1033. return KernelResult.InvalidHandle;
  1034. }
  1035. if (targetProcess.MemoryManager.OutsideAddrSpace(src, size))
  1036. {
  1037. return KernelResult.InvalidMemState;
  1038. }
  1039. return targetProcess.MemoryManager.SetProcessMemoryPermission(src, size, permission);
  1040. }
  1041. private static bool PageAligned(ulong position)
  1042. {
  1043. return (position & (KPageTableBase.PageSize - 1)) == 0;
  1044. }
  1045. // System
  1046. public KernelResult TerminateProcess(int handle)
  1047. {
  1048. KProcess process = KernelStatic.GetCurrentProcess();
  1049. process = process.HandleTable.GetObject<KProcess>(handle);
  1050. KernelResult result;
  1051. if (process != null)
  1052. {
  1053. if (process == KernelStatic.GetCurrentProcess())
  1054. {
  1055. result = KernelResult.Success;
  1056. process.DecrementToZeroWhileTerminatingCurrent();
  1057. }
  1058. else
  1059. {
  1060. result = process.Terminate();
  1061. process.DecrementReferenceCount();
  1062. }
  1063. }
  1064. else
  1065. {
  1066. result = KernelResult.InvalidHandle;
  1067. }
  1068. return result;
  1069. }
  1070. public void ExitProcess()
  1071. {
  1072. KernelStatic.GetCurrentProcess().TerminateCurrentProcess();
  1073. }
  1074. public KernelResult SignalEvent(int handle)
  1075. {
  1076. KProcess process = KernelStatic.GetCurrentProcess();
  1077. KWritableEvent writableEvent = process.HandleTable.GetObject<KWritableEvent>(handle);
  1078. KernelResult result;
  1079. if (writableEvent != null)
  1080. {
  1081. writableEvent.Signal();
  1082. result = KernelResult.Success;
  1083. }
  1084. else
  1085. {
  1086. result = KernelResult.InvalidHandle;
  1087. }
  1088. return result;
  1089. }
  1090. public KernelResult ClearEvent(int handle)
  1091. {
  1092. KernelResult result;
  1093. KProcess process = KernelStatic.GetCurrentProcess();
  1094. KWritableEvent writableEvent = process.HandleTable.GetObject<KWritableEvent>(handle);
  1095. if (writableEvent == null)
  1096. {
  1097. KReadableEvent readableEvent = process.HandleTable.GetObject<KReadableEvent>(handle);
  1098. result = readableEvent?.Clear() ?? KernelResult.InvalidHandle;
  1099. }
  1100. else
  1101. {
  1102. result = writableEvent.Clear();
  1103. }
  1104. return result;
  1105. }
  1106. public KernelResult CloseHandle(int handle)
  1107. {
  1108. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1109. return currentProcess.HandleTable.CloseHandle(handle) ? KernelResult.Success : KernelResult.InvalidHandle;
  1110. }
  1111. public KernelResult ResetSignal(int handle)
  1112. {
  1113. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1114. KReadableEvent readableEvent = currentProcess.HandleTable.GetObject<KReadableEvent>(handle);
  1115. KernelResult result;
  1116. if (readableEvent != null)
  1117. {
  1118. result = readableEvent.ClearIfSignaled();
  1119. }
  1120. else
  1121. {
  1122. KProcess process = currentProcess.HandleTable.GetKProcess(handle);
  1123. if (process != null)
  1124. {
  1125. result = process.ClearIfNotExited();
  1126. }
  1127. else
  1128. {
  1129. result = KernelResult.InvalidHandle;
  1130. }
  1131. }
  1132. return result;
  1133. }
  1134. public ulong GetSystemTick()
  1135. {
  1136. return KernelStatic.GetCurrentThread().Context.CntpctEl0;
  1137. }
  1138. public void Break(ulong reason)
  1139. {
  1140. KThread currentThread = KernelStatic.GetCurrentThread();
  1141. if ((reason & (1UL << 31)) == 0)
  1142. {
  1143. currentThread.PrintGuestStackTrace();
  1144. currentThread.PrintGuestRegisterPrintout();
  1145. // As the process is exiting, this is probably caused by emulation termination.
  1146. if (currentThread.Owner.State == ProcessState.Exiting)
  1147. {
  1148. return;
  1149. }
  1150. // TODO: Debug events.
  1151. currentThread.Owner.TerminateCurrentProcess();
  1152. throw new GuestBrokeExecutionException();
  1153. }
  1154. else
  1155. {
  1156. Logger.Debug?.Print(LogClass.KernelSvc, "Debugger triggered.");
  1157. }
  1158. }
  1159. public void OutputDebugString(ulong strPtr, ulong size)
  1160. {
  1161. KProcess process = KernelStatic.GetCurrentProcess();
  1162. string str = MemoryHelper.ReadAsciiString(process.CpuMemory, strPtr, (long)size);
  1163. Logger.Warning?.Print(LogClass.KernelSvc, str);
  1164. }
  1165. public KernelResult GetInfo(uint id, int handle, long subId, out long value)
  1166. {
  1167. value = 0;
  1168. switch (id)
  1169. {
  1170. case 0:
  1171. case 1:
  1172. case 2:
  1173. case 3:
  1174. case 4:
  1175. case 5:
  1176. case 6:
  1177. case 7:
  1178. case 12:
  1179. case 13:
  1180. case 14:
  1181. case 15:
  1182. case 16:
  1183. case 17:
  1184. case 18:
  1185. case 20:
  1186. case 21:
  1187. case 22:
  1188. {
  1189. if (subId != 0)
  1190. {
  1191. return KernelResult.InvalidCombination;
  1192. }
  1193. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1194. KProcess process = currentProcess.HandleTable.GetKProcess(handle);
  1195. if (process == null)
  1196. {
  1197. return KernelResult.InvalidHandle;
  1198. }
  1199. switch (id)
  1200. {
  1201. case 0: value = process.Capabilities.AllowedCpuCoresMask; break;
  1202. case 1: value = process.Capabilities.AllowedThreadPriosMask; break;
  1203. case 2: value = (long)process.MemoryManager.AliasRegionStart; break;
  1204. case 3:
  1205. value = (long)(process.MemoryManager.AliasRegionEnd -
  1206. process.MemoryManager.AliasRegionStart); break;
  1207. case 4: value = (long)process.MemoryManager.HeapRegionStart; break;
  1208. case 5:
  1209. value = (long)(process.MemoryManager.HeapRegionEnd -
  1210. process.MemoryManager.HeapRegionStart); break;
  1211. case 6: value = (long)process.GetMemoryCapacity(); break;
  1212. case 7: value = (long)process.GetMemoryUsage(); break;
  1213. case 12: value = (long)process.MemoryManager.GetAddrSpaceBaseAddr(); break;
  1214. case 13: value = (long)process.MemoryManager.GetAddrSpaceSize(); break;
  1215. case 14: value = (long)process.MemoryManager.StackRegionStart; break;
  1216. case 15:
  1217. value = (long)(process.MemoryManager.StackRegionEnd -
  1218. process.MemoryManager.StackRegionStart); break;
  1219. case 16: value = (long)process.PersonalMmHeapPagesCount * KPageTableBase.PageSize; break;
  1220. case 17:
  1221. if (process.PersonalMmHeapPagesCount != 0)
  1222. {
  1223. value = process.MemoryManager.GetMmUsedPages() * KPageTableBase.PageSize;
  1224. }
  1225. break;
  1226. case 18: value = (long)process.TitleId; break;
  1227. case 20: value = (long)process.UserExceptionContextAddress; break;
  1228. case 21: value = (long)process.GetMemoryCapacityWithoutPersonalMmHeap(); break;
  1229. case 22: value = (long)process.GetMemoryUsageWithoutPersonalMmHeap(); break;
  1230. }
  1231. break;
  1232. }
  1233. case 8:
  1234. {
  1235. if (handle != 0)
  1236. {
  1237. return KernelResult.InvalidHandle;
  1238. }
  1239. if (subId != 0)
  1240. {
  1241. return KernelResult.InvalidCombination;
  1242. }
  1243. value = KernelStatic.GetCurrentProcess().Debug ? 1 : 0;
  1244. break;
  1245. }
  1246. case 9:
  1247. {
  1248. if (handle != 0)
  1249. {
  1250. return KernelResult.InvalidHandle;
  1251. }
  1252. if (subId != 0)
  1253. {
  1254. return KernelResult.InvalidCombination;
  1255. }
  1256. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1257. if (currentProcess.ResourceLimit != null)
  1258. {
  1259. KHandleTable handleTable = currentProcess.HandleTable;
  1260. KResourceLimit resourceLimit = currentProcess.ResourceLimit;
  1261. KernelResult result = handleTable.GenerateHandle(resourceLimit, out int resLimHandle);
  1262. if (result != KernelResult.Success)
  1263. {
  1264. return result;
  1265. }
  1266. value = (uint)resLimHandle;
  1267. }
  1268. break;
  1269. }
  1270. case 10:
  1271. {
  1272. if (handle != 0)
  1273. {
  1274. return KernelResult.InvalidHandle;
  1275. }
  1276. int currentCore = KernelStatic.GetCurrentThread().CurrentCore;
  1277. if (subId != -1 && subId != currentCore)
  1278. {
  1279. return KernelResult.InvalidCombination;
  1280. }
  1281. value = KTimeManager.ConvertHostTicksToTicks(_context.Schedulers[currentCore].TotalIdleTimeTicks);
  1282. break;
  1283. }
  1284. case 11:
  1285. {
  1286. if (handle != 0)
  1287. {
  1288. return KernelResult.InvalidHandle;
  1289. }
  1290. if ((ulong)subId > 3)
  1291. {
  1292. return KernelResult.InvalidCombination;
  1293. }
  1294. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1295. value = currentProcess.RandomEntropy[subId];
  1296. break;
  1297. }
  1298. case 0xf0000002u:
  1299. {
  1300. if (subId < -1 || subId > 3)
  1301. {
  1302. return KernelResult.InvalidCombination;
  1303. }
  1304. KThread thread = KernelStatic.GetCurrentProcess().HandleTable.GetKThread(handle);
  1305. if (thread == null)
  1306. {
  1307. return KernelResult.InvalidHandle;
  1308. }
  1309. KThread currentThread = KernelStatic.GetCurrentThread();
  1310. int currentCore = currentThread.CurrentCore;
  1311. if (subId != -1 && subId != currentCore)
  1312. {
  1313. return KernelResult.Success;
  1314. }
  1315. KScheduler scheduler = _context.Schedulers[currentCore];
  1316. long timeDelta = PerformanceCounter.ElapsedTicks - scheduler.LastContextSwitchTime;
  1317. if (subId != -1)
  1318. {
  1319. value = KTimeManager.ConvertHostTicksToTicks(timeDelta);
  1320. }
  1321. else
  1322. {
  1323. long totalTimeRunning = thread.TotalTimeRunning;
  1324. if (thread == currentThread)
  1325. {
  1326. totalTimeRunning += timeDelta;
  1327. }
  1328. value = KTimeManager.ConvertHostTicksToTicks(totalTimeRunning);
  1329. }
  1330. break;
  1331. }
  1332. default: return KernelResult.InvalidEnumValue;
  1333. }
  1334. return KernelResult.Success;
  1335. }
  1336. public KernelResult CreateEvent(out int wEventHandle, out int rEventHandle)
  1337. {
  1338. KEvent Event = new KEvent(_context);
  1339. KProcess process = KernelStatic.GetCurrentProcess();
  1340. KernelResult result = process.HandleTable.GenerateHandle(Event.WritableEvent, out wEventHandle);
  1341. if (result == KernelResult.Success)
  1342. {
  1343. result = process.HandleTable.GenerateHandle(Event.ReadableEvent, out rEventHandle);
  1344. if (result != KernelResult.Success)
  1345. {
  1346. process.HandleTable.CloseHandle(wEventHandle);
  1347. }
  1348. }
  1349. else
  1350. {
  1351. rEventHandle = 0;
  1352. }
  1353. return result;
  1354. }
  1355. public KernelResult GetProcessList(ulong address, int maxCount, out int count)
  1356. {
  1357. count = 0;
  1358. if ((maxCount >> 28) != 0)
  1359. {
  1360. return KernelResult.MaximumExceeded;
  1361. }
  1362. if (maxCount != 0)
  1363. {
  1364. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1365. ulong copySize = (ulong)maxCount * 8;
  1366. if (address + copySize <= address)
  1367. {
  1368. return KernelResult.InvalidMemState;
  1369. }
  1370. if (currentProcess.MemoryManager.OutsideAddrSpace(address, copySize))
  1371. {
  1372. return KernelResult.InvalidMemState;
  1373. }
  1374. }
  1375. int copyCount = 0;
  1376. lock (_context.Processes)
  1377. {
  1378. foreach (KProcess process in _context.Processes.Values)
  1379. {
  1380. if (copyCount < maxCount)
  1381. {
  1382. if (!KernelTransfer.KernelToUserInt64(_context, address + (ulong)copyCount * 8, process.Pid))
  1383. {
  1384. return KernelResult.UserCopyFailed;
  1385. }
  1386. }
  1387. copyCount++;
  1388. }
  1389. }
  1390. count = copyCount;
  1391. return KernelResult.Success;
  1392. }
  1393. public KernelResult GetSystemInfo(uint id, int handle, long subId, out long value)
  1394. {
  1395. value = 0;
  1396. if (id > 2)
  1397. {
  1398. return KernelResult.InvalidEnumValue;
  1399. }
  1400. if (handle != 0)
  1401. {
  1402. return KernelResult.InvalidHandle;
  1403. }
  1404. if (id < 2)
  1405. {
  1406. if ((ulong)subId > 3)
  1407. {
  1408. return KernelResult.InvalidCombination;
  1409. }
  1410. KMemoryRegionManager region = _context.MemoryManager.MemoryRegions[subId];
  1411. switch (id)
  1412. {
  1413. // Memory region capacity.
  1414. case 0: value = (long)region.Size; break;
  1415. // Memory region free space.
  1416. case 1:
  1417. {
  1418. ulong freePagesCount = region.GetFreePages();
  1419. value = (long)(freePagesCount * KPageTableBase.PageSize);
  1420. break;
  1421. }
  1422. }
  1423. }
  1424. else /* if (Id == 2) */
  1425. {
  1426. if ((ulong)subId > 1)
  1427. {
  1428. return KernelResult.InvalidCombination;
  1429. }
  1430. switch (subId)
  1431. {
  1432. case 0: value = _context.PrivilegedProcessLowestId; break;
  1433. case 1: value = _context.PrivilegedProcessHighestId; break;
  1434. }
  1435. }
  1436. return KernelResult.Success;
  1437. }
  1438. // Thread
  1439. public KernelResult CreateThread(
  1440. ulong entrypoint,
  1441. ulong argsPtr,
  1442. ulong stackTop,
  1443. int priority,
  1444. int cpuCore,
  1445. out int handle)
  1446. {
  1447. handle = 0;
  1448. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1449. if (cpuCore == -2)
  1450. {
  1451. cpuCore = currentProcess.DefaultCpuCore;
  1452. }
  1453. if ((uint)cpuCore >= KScheduler.CpuCoresCount || !currentProcess.IsCpuCoreAllowed(cpuCore))
  1454. {
  1455. return KernelResult.InvalidCpuCore;
  1456. }
  1457. if ((uint)priority >= KScheduler.PrioritiesCount || !currentProcess.IsPriorityAllowed(priority))
  1458. {
  1459. return KernelResult.InvalidPriority;
  1460. }
  1461. long timeout = KTimeManager.ConvertMillisecondsToNanoseconds(100);
  1462. if (currentProcess.ResourceLimit != null &&
  1463. !currentProcess.ResourceLimit.Reserve(LimitableResource.Thread, 1, timeout))
  1464. {
  1465. return KernelResult.ResLimitExceeded;
  1466. }
  1467. KThread thread = new KThread(_context);
  1468. KernelResult result = currentProcess.InitializeThread(
  1469. thread,
  1470. entrypoint,
  1471. argsPtr,
  1472. stackTop,
  1473. priority,
  1474. cpuCore);
  1475. if (result == KernelResult.Success)
  1476. {
  1477. KProcess process = KernelStatic.GetCurrentProcess();
  1478. result = process.HandleTable.GenerateHandle(thread, out handle);
  1479. }
  1480. else
  1481. {
  1482. currentProcess.ResourceLimit?.Release(LimitableResource.Thread, 1);
  1483. }
  1484. thread.DecrementReferenceCount();
  1485. return result;
  1486. }
  1487. public KernelResult StartThread(int handle)
  1488. {
  1489. KProcess process = KernelStatic.GetCurrentProcess();
  1490. KThread thread = process.HandleTable.GetKThread(handle);
  1491. if (thread != null)
  1492. {
  1493. thread.IncrementReferenceCount();
  1494. KernelResult result = thread.Start();
  1495. if (result == KernelResult.Success)
  1496. {
  1497. thread.IncrementReferenceCount();
  1498. }
  1499. thread.DecrementReferenceCount();
  1500. return result;
  1501. }
  1502. else
  1503. {
  1504. return KernelResult.InvalidHandle;
  1505. }
  1506. }
  1507. public void ExitThread()
  1508. {
  1509. KThread currentThread = KernelStatic.GetCurrentThread();
  1510. currentThread.Exit();
  1511. }
  1512. public void SleepThread(long timeout)
  1513. {
  1514. if (timeout < 1)
  1515. {
  1516. switch (timeout)
  1517. {
  1518. case 0: KScheduler.Yield(_context); break;
  1519. case -1: KScheduler.YieldWithLoadBalancing(_context); break;
  1520. case -2: KScheduler.YieldToAnyThread(_context); break;
  1521. }
  1522. }
  1523. else
  1524. {
  1525. KernelStatic.GetCurrentThread().Sleep(timeout);
  1526. }
  1527. }
  1528. public KernelResult GetThreadPriority(int handle, out int priority)
  1529. {
  1530. KProcess process = KernelStatic.GetCurrentProcess();
  1531. KThread thread = process.HandleTable.GetKThread(handle);
  1532. if (thread != null)
  1533. {
  1534. priority = thread.DynamicPriority;
  1535. return KernelResult.Success;
  1536. }
  1537. else
  1538. {
  1539. priority = 0;
  1540. return KernelResult.InvalidHandle;
  1541. }
  1542. }
  1543. public KernelResult SetThreadPriority(int handle, int priority)
  1544. {
  1545. // TODO: NPDM check.
  1546. KProcess process = KernelStatic.GetCurrentProcess();
  1547. KThread thread = process.HandleTable.GetKThread(handle);
  1548. if (thread == null)
  1549. {
  1550. return KernelResult.InvalidHandle;
  1551. }
  1552. thread.SetPriority(priority);
  1553. return KernelResult.Success;
  1554. }
  1555. public KernelResult GetThreadCoreMask(int handle, out int preferredCore, out long affinityMask)
  1556. {
  1557. KProcess process = KernelStatic.GetCurrentProcess();
  1558. KThread thread = process.HandleTable.GetKThread(handle);
  1559. if (thread != null)
  1560. {
  1561. preferredCore = thread.PreferredCore;
  1562. affinityMask = thread.AffinityMask;
  1563. return KernelResult.Success;
  1564. }
  1565. else
  1566. {
  1567. preferredCore = 0;
  1568. affinityMask = 0;
  1569. return KernelResult.InvalidHandle;
  1570. }
  1571. }
  1572. public KernelResult SetThreadCoreMask(int handle, int preferredCore, long affinityMask)
  1573. {
  1574. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1575. if (preferredCore == -2)
  1576. {
  1577. preferredCore = currentProcess.DefaultCpuCore;
  1578. affinityMask = 1 << preferredCore;
  1579. }
  1580. else
  1581. {
  1582. if ((currentProcess.Capabilities.AllowedCpuCoresMask | affinityMask) !=
  1583. currentProcess.Capabilities.AllowedCpuCoresMask)
  1584. {
  1585. return KernelResult.InvalidCpuCore;
  1586. }
  1587. if (affinityMask == 0)
  1588. {
  1589. return KernelResult.InvalidCombination;
  1590. }
  1591. if ((uint)preferredCore > 3)
  1592. {
  1593. if ((preferredCore | 2) != -1)
  1594. {
  1595. return KernelResult.InvalidCpuCore;
  1596. }
  1597. }
  1598. else if ((affinityMask & (1 << preferredCore)) == 0)
  1599. {
  1600. return KernelResult.InvalidCombination;
  1601. }
  1602. }
  1603. KProcess process = KernelStatic.GetCurrentProcess();
  1604. KThread thread = process.HandleTable.GetKThread(handle);
  1605. if (thread == null)
  1606. {
  1607. return KernelResult.InvalidHandle;
  1608. }
  1609. return thread.SetCoreAndAffinityMask(preferredCore, affinityMask);
  1610. }
  1611. public int GetCurrentProcessorNumber()
  1612. {
  1613. return KernelStatic.GetCurrentThread().CurrentCore;
  1614. }
  1615. public KernelResult GetThreadId(int handle, out long threadUid)
  1616. {
  1617. KProcess process = KernelStatic.GetCurrentProcess();
  1618. KThread thread = process.HandleTable.GetKThread(handle);
  1619. if (thread != null)
  1620. {
  1621. threadUid = thread.ThreadUid;
  1622. return KernelResult.Success;
  1623. }
  1624. else
  1625. {
  1626. threadUid = 0;
  1627. return KernelResult.InvalidHandle;
  1628. }
  1629. }
  1630. public KernelResult SetThreadActivity(int handle, bool pause)
  1631. {
  1632. KProcess process = KernelStatic.GetCurrentProcess();
  1633. KThread thread = process.HandleTable.GetObject<KThread>(handle);
  1634. if (thread == null)
  1635. {
  1636. return KernelResult.InvalidHandle;
  1637. }
  1638. if (thread.Owner != process)
  1639. {
  1640. return KernelResult.InvalidHandle;
  1641. }
  1642. if (thread == KernelStatic.GetCurrentThread())
  1643. {
  1644. return KernelResult.InvalidThread;
  1645. }
  1646. return thread.SetActivity(pause);
  1647. }
  1648. public KernelResult GetThreadContext3(ulong address, int handle)
  1649. {
  1650. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1651. KThread currentThread = KernelStatic.GetCurrentThread();
  1652. KThread thread = currentProcess.HandleTable.GetObject<KThread>(handle);
  1653. if (thread == null)
  1654. {
  1655. return KernelResult.InvalidHandle;
  1656. }
  1657. if (thread.Owner != currentProcess)
  1658. {
  1659. return KernelResult.InvalidHandle;
  1660. }
  1661. if (currentThread == thread)
  1662. {
  1663. return KernelResult.InvalidThread;
  1664. }
  1665. IVirtualMemoryManager memory = currentProcess.CpuMemory;
  1666. memory.Write(address + 0x0, thread.Context.GetX(0));
  1667. memory.Write(address + 0x8, thread.Context.GetX(1));
  1668. memory.Write(address + 0x10, thread.Context.GetX(2));
  1669. memory.Write(address + 0x18, thread.Context.GetX(3));
  1670. memory.Write(address + 0x20, thread.Context.GetX(4));
  1671. memory.Write(address + 0x28, thread.Context.GetX(5));
  1672. memory.Write(address + 0x30, thread.Context.GetX(6));
  1673. memory.Write(address + 0x38, thread.Context.GetX(7));
  1674. memory.Write(address + 0x40, thread.Context.GetX(8));
  1675. memory.Write(address + 0x48, thread.Context.GetX(9));
  1676. memory.Write(address + 0x50, thread.Context.GetX(10));
  1677. memory.Write(address + 0x58, thread.Context.GetX(11));
  1678. memory.Write(address + 0x60, thread.Context.GetX(12));
  1679. memory.Write(address + 0x68, thread.Context.GetX(13));
  1680. memory.Write(address + 0x70, thread.Context.GetX(14));
  1681. memory.Write(address + 0x78, thread.Context.GetX(15));
  1682. memory.Write(address + 0x80, thread.Context.GetX(16));
  1683. memory.Write(address + 0x88, thread.Context.GetX(17));
  1684. memory.Write(address + 0x90, thread.Context.GetX(18));
  1685. memory.Write(address + 0x98, thread.Context.GetX(19));
  1686. memory.Write(address + 0xa0, thread.Context.GetX(20));
  1687. memory.Write(address + 0xa8, thread.Context.GetX(21));
  1688. memory.Write(address + 0xb0, thread.Context.GetX(22));
  1689. memory.Write(address + 0xb8, thread.Context.GetX(23));
  1690. memory.Write(address + 0xc0, thread.Context.GetX(24));
  1691. memory.Write(address + 0xc8, thread.Context.GetX(25));
  1692. memory.Write(address + 0xd0, thread.Context.GetX(26));
  1693. memory.Write(address + 0xd8, thread.Context.GetX(27));
  1694. memory.Write(address + 0xe0, thread.Context.GetX(28));
  1695. memory.Write(address + 0xe8, thread.Context.GetX(29));
  1696. memory.Write(address + 0xf0, thread.Context.GetX(30));
  1697. memory.Write(address + 0xf8, thread.Context.GetX(31));
  1698. memory.Write(address + 0x100, thread.LastPc);
  1699. memory.Write(address + 0x108, (ulong)GetPsr(thread.Context));
  1700. memory.Write(address + 0x110, thread.Context.GetV(0));
  1701. memory.Write(address + 0x120, thread.Context.GetV(1));
  1702. memory.Write(address + 0x130, thread.Context.GetV(2));
  1703. memory.Write(address + 0x140, thread.Context.GetV(3));
  1704. memory.Write(address + 0x150, thread.Context.GetV(4));
  1705. memory.Write(address + 0x160, thread.Context.GetV(5));
  1706. memory.Write(address + 0x170, thread.Context.GetV(6));
  1707. memory.Write(address + 0x180, thread.Context.GetV(7));
  1708. memory.Write(address + 0x190, thread.Context.GetV(8));
  1709. memory.Write(address + 0x1a0, thread.Context.GetV(9));
  1710. memory.Write(address + 0x1b0, thread.Context.GetV(10));
  1711. memory.Write(address + 0x1c0, thread.Context.GetV(11));
  1712. memory.Write(address + 0x1d0, thread.Context.GetV(12));
  1713. memory.Write(address + 0x1e0, thread.Context.GetV(13));
  1714. memory.Write(address + 0x1f0, thread.Context.GetV(14));
  1715. memory.Write(address + 0x200, thread.Context.GetV(15));
  1716. memory.Write(address + 0x210, thread.Context.GetV(16));
  1717. memory.Write(address + 0x220, thread.Context.GetV(17));
  1718. memory.Write(address + 0x230, thread.Context.GetV(18));
  1719. memory.Write(address + 0x240, thread.Context.GetV(19));
  1720. memory.Write(address + 0x250, thread.Context.GetV(20));
  1721. memory.Write(address + 0x260, thread.Context.GetV(21));
  1722. memory.Write(address + 0x270, thread.Context.GetV(22));
  1723. memory.Write(address + 0x280, thread.Context.GetV(23));
  1724. memory.Write(address + 0x290, thread.Context.GetV(24));
  1725. memory.Write(address + 0x2a0, thread.Context.GetV(25));
  1726. memory.Write(address + 0x2b0, thread.Context.GetV(26));
  1727. memory.Write(address + 0x2c0, thread.Context.GetV(27));
  1728. memory.Write(address + 0x2d0, thread.Context.GetV(28));
  1729. memory.Write(address + 0x2e0, thread.Context.GetV(29));
  1730. memory.Write(address + 0x2f0, thread.Context.GetV(30));
  1731. memory.Write(address + 0x300, thread.Context.GetV(31));
  1732. memory.Write(address + 0x310, (int)thread.Context.Fpcr);
  1733. memory.Write(address + 0x314, (int)thread.Context.Fpsr);
  1734. memory.Write(address + 0x318, thread.Context.Tpidr);
  1735. return KernelResult.Success;
  1736. }
  1737. private static int GetPsr(ARMeilleure.State.ExecutionContext context)
  1738. {
  1739. return (context.GetPstateFlag(ARMeilleure.State.PState.NFlag) ? (1 << (int)ARMeilleure.State.PState.NFlag) : 0) |
  1740. (context.GetPstateFlag(ARMeilleure.State.PState.ZFlag) ? (1 << (int)ARMeilleure.State.PState.ZFlag) : 0) |
  1741. (context.GetPstateFlag(ARMeilleure.State.PState.CFlag) ? (1 << (int)ARMeilleure.State.PState.CFlag) : 0) |
  1742. (context.GetPstateFlag(ARMeilleure.State.PState.VFlag) ? (1 << (int)ARMeilleure.State.PState.VFlag) : 0);
  1743. }
  1744. // Thread synchronization
  1745. public KernelResult WaitSynchronization(ulong handlesPtr, int handlesCount, long timeout, out int handleIndex)
  1746. {
  1747. handleIndex = 0;
  1748. if ((uint)handlesCount > KThread.MaxWaitSyncObjects)
  1749. {
  1750. return KernelResult.MaximumExceeded;
  1751. }
  1752. KThread currentThread = KernelStatic.GetCurrentThread();
  1753. var syncObjs = new Span<KSynchronizationObject>(currentThread.WaitSyncObjects).Slice(0, handlesCount);
  1754. if (handlesCount != 0)
  1755. {
  1756. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1757. if (currentProcess.MemoryManager.AddrSpaceStart > handlesPtr)
  1758. {
  1759. return KernelResult.UserCopyFailed;
  1760. }
  1761. long handlesSize = handlesCount * 4;
  1762. if (handlesPtr + (ulong)handlesSize <= handlesPtr)
  1763. {
  1764. return KernelResult.UserCopyFailed;
  1765. }
  1766. if (handlesPtr + (ulong)handlesSize - 1 > currentProcess.MemoryManager.AddrSpaceEnd - 1)
  1767. {
  1768. return KernelResult.UserCopyFailed;
  1769. }
  1770. Span<int> handles = new Span<int>(currentThread.WaitSyncHandles).Slice(0, handlesCount);
  1771. if (!KernelTransfer.UserToKernelInt32Array(_context, handlesPtr, handles))
  1772. {
  1773. return KernelResult.UserCopyFailed;
  1774. }
  1775. int processedHandles = 0;
  1776. for (; processedHandles < handlesCount; processedHandles++)
  1777. {
  1778. KSynchronizationObject syncObj = currentProcess.HandleTable.GetObject<KSynchronizationObject>(handles[processedHandles]);
  1779. if (syncObj == null)
  1780. {
  1781. break;
  1782. }
  1783. syncObjs[processedHandles] = syncObj;
  1784. syncObj.IncrementReferenceCount();
  1785. }
  1786. if (processedHandles != handlesCount)
  1787. {
  1788. // One or more handles are invalid.
  1789. for (int index = 0; index < processedHandles; index++)
  1790. {
  1791. currentThread.WaitSyncObjects[index].DecrementReferenceCount();
  1792. }
  1793. return KernelResult.InvalidHandle;
  1794. }
  1795. }
  1796. KernelResult result = _context.Synchronization.WaitFor(syncObjs, timeout, out handleIndex);
  1797. if (result == KernelResult.PortRemoteClosed)
  1798. {
  1799. result = KernelResult.Success;
  1800. }
  1801. for (int index = 0; index < handlesCount; index++)
  1802. {
  1803. currentThread.WaitSyncObjects[index].DecrementReferenceCount();
  1804. }
  1805. return result;
  1806. }
  1807. public KernelResult CancelSynchronization(int handle)
  1808. {
  1809. KProcess process = KernelStatic.GetCurrentProcess();
  1810. KThread thread = process.HandleTable.GetKThread(handle);
  1811. if (thread == null)
  1812. {
  1813. return KernelResult.InvalidHandle;
  1814. }
  1815. thread.CancelSynchronization();
  1816. return KernelResult.Success;
  1817. }
  1818. public KernelResult ArbitrateLock(int ownerHandle, ulong mutexAddress, int requesterHandle)
  1819. {
  1820. if (IsPointingInsideKernel(mutexAddress))
  1821. {
  1822. return KernelResult.InvalidMemState;
  1823. }
  1824. if (IsAddressNotWordAligned(mutexAddress))
  1825. {
  1826. return KernelResult.InvalidAddress;
  1827. }
  1828. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1829. return currentProcess.AddressArbiter.ArbitrateLock(ownerHandle, mutexAddress, requesterHandle);
  1830. }
  1831. public KernelResult ArbitrateUnlock(ulong mutexAddress)
  1832. {
  1833. if (IsPointingInsideKernel(mutexAddress))
  1834. {
  1835. return KernelResult.InvalidMemState;
  1836. }
  1837. if (IsAddressNotWordAligned(mutexAddress))
  1838. {
  1839. return KernelResult.InvalidAddress;
  1840. }
  1841. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1842. return currentProcess.AddressArbiter.ArbitrateUnlock(mutexAddress);
  1843. }
  1844. public KernelResult WaitProcessWideKeyAtomic(
  1845. ulong mutexAddress,
  1846. ulong condVarAddress,
  1847. int handle,
  1848. long timeout)
  1849. {
  1850. if (IsPointingInsideKernel(mutexAddress))
  1851. {
  1852. return KernelResult.InvalidMemState;
  1853. }
  1854. if (IsAddressNotWordAligned(mutexAddress))
  1855. {
  1856. return KernelResult.InvalidAddress;
  1857. }
  1858. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1859. return currentProcess.AddressArbiter.WaitProcessWideKeyAtomic(
  1860. mutexAddress,
  1861. condVarAddress,
  1862. handle,
  1863. timeout);
  1864. }
  1865. public KernelResult SignalProcessWideKey(ulong address, int count)
  1866. {
  1867. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1868. currentProcess.AddressArbiter.SignalProcessWideKey(address, count);
  1869. return KernelResult.Success;
  1870. }
  1871. public KernelResult WaitForAddress(ulong address, ArbitrationType type, int value, long timeout)
  1872. {
  1873. if (IsPointingInsideKernel(address))
  1874. {
  1875. return KernelResult.InvalidMemState;
  1876. }
  1877. if (IsAddressNotWordAligned(address))
  1878. {
  1879. return KernelResult.InvalidAddress;
  1880. }
  1881. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1882. return type switch
  1883. {
  1884. ArbitrationType.WaitIfLessThan
  1885. => currentProcess.AddressArbiter.WaitForAddressIfLessThan(address, value, false, timeout),
  1886. ArbitrationType.DecrementAndWaitIfLessThan
  1887. => currentProcess.AddressArbiter.WaitForAddressIfLessThan(address, value, true, timeout),
  1888. ArbitrationType.WaitIfEqual
  1889. => currentProcess.AddressArbiter.WaitForAddressIfEqual(address, value, timeout),
  1890. _ => KernelResult.InvalidEnumValue,
  1891. };
  1892. }
  1893. public KernelResult SignalToAddress(ulong address, SignalType type, int value, int count)
  1894. {
  1895. if (IsPointingInsideKernel(address))
  1896. {
  1897. return KernelResult.InvalidMemState;
  1898. }
  1899. if (IsAddressNotWordAligned(address))
  1900. {
  1901. return KernelResult.InvalidAddress;
  1902. }
  1903. KProcess currentProcess = KernelStatic.GetCurrentProcess();
  1904. return type switch
  1905. {
  1906. SignalType.Signal
  1907. => currentProcess.AddressArbiter.Signal(address, count),
  1908. SignalType.SignalAndIncrementIfEqual
  1909. => currentProcess.AddressArbiter.SignalAndIncrementIfEqual(address, value, count),
  1910. SignalType.SignalAndModifyIfEqual
  1911. => currentProcess.AddressArbiter.SignalAndModifyIfEqual(address, value, count),
  1912. _ => KernelResult.InvalidEnumValue
  1913. };
  1914. }
  1915. private bool IsPointingInsideKernel(ulong address)
  1916. {
  1917. return (address + 0x1000000000) < 0xffffff000;
  1918. }
  1919. private bool IsAddressNotWordAligned(ulong address)
  1920. {
  1921. return (address & 3) != 0;
  1922. }
  1923. }
  1924. }