Syscall.cs 75 KB

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