Syscall.cs 75 KB

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