Syscall.cs 76 KB

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