Syscall.cs 75 KB

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