Syscall.cs 74 KB

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