Syscall.cs 80 KB

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