thor: Add a GetNumCpu request to kerncfg

Dennisbonke · Sep 23, 2024 · c2e1332 · c2e1332
1 parent 6107613
commit c2e1332
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/kernel/thor/generic/kerncfg.cpp b/kernel/thor/generic/kerncfg.cpp
@@ -90,6 +90,23 @@ struct KerncfgBusObject : private KernelBusObject {
 			auto respError = co_await SendBufferSender{lane, std::move(respBuffer)};
 			if(respError != Error::success)
 				co_return respError;
+		}else if(preamble.id() == bragi::message_id<managarm::kerncfg::GetNumCpuRequest>) {
+			auto req = bragi::parse_head_only<managarm::kerncfg::GetNumCpuRequest>(reqBuffer, *kernelAlloc);
+
+			if (!req) {
+				co_return Error::protocolViolation;
+			}
+
+			managarm::kerncfg::GetNumCpuResponse<KernelAlloc> resp(*kernelAlloc);
+			resp.set_error(managarm::kerncfg::Error::SUCCESS);
+			resp.set_num_cpu(getCpuCount());
+
+			frg::unique_memory<KernelAlloc> respBuffer{*kernelAlloc, resp.size_of_head()};
+			bragi::write_head_only(resp, respBuffer);
+			auto respError = co_await SendBufferSender{lane, std::move(respBuffer)};
+			if(respError != Error::success) {
+				co_return respError;
+			}
 		}else{
 			managarm::kerncfg::SvrResponse<KernelAlloc> resp(*kernelAlloc);
 			resp.set_error(managarm::kerncfg::Error::ILLEGAL_REQUEST);

diff --git a/protocols/kerncfg/kerncfg.bragi b/protocols/kerncfg/kerncfg.bragi
@@ -47,3 +47,13 @@ head(128):
 	uint64 available_memory;
 	uint64 memory_unit;
 }
+
+message GetNumCpuRequest 6 {
+head(128):
+}
+
+message GetNumCpuResponse 7 {
+head(128):
+	Error error;
+	uint64 num_cpu;
+}