Add Machine::set_fpu_registers() and Machine::run_in_usermode()

Author: fwsGonzo

lib/tinykvm/machine.cpp

@@ -286,6 +286,12 @@ void Machine::run(float timeout)
 	return vcpu.run(timeout * 1000.0);
 }
 
+void Machine::run_in_usermode(float timeout)
+{
+	this->enter_usermode();
+	this->run(timeout);
+}
+
 __attribute__((cold, noreturn))
 void Machine::machine_exception(const char* msg, uint64_t data)
 {

lib/tinykvm/machine.hpp

@@ -28,6 +28,7 @@ struct Machine
 	void setup_linux(const std::vector<std::string>& args,
 					const std::vector<std::string>& env = {});
 	void run(float timeout_secs = 0.f);
+	void run_in_usermode(float timeout_secs = 0.f);
 
 	/* Make a SYSV function call into the VM, with no timeout */
 	template <typename... Args>
@@ -108,6 +109,7 @@ struct Machine
 	const tinykvm_x86regs& registers() const;
 	void set_registers(const tinykvm_x86regs&);
 	tinykvm_fpuregs fpu_registers() const;
+	void set_fpu_registers(const tinykvm_fpuregs&);
 	const kvm_sregs& get_special_registers() const;
 	void set_special_registers(const kvm_sregs&);
 	std::pair<__u64, __u64> get_fsgs() const;

lib/tinykvm/machine_inline.hpp

@@ -36,6 +36,9 @@ inline void Machine::set_registers(const tinykvm_x86regs& regs) {
 inline tinykvm_fpuregs Machine::fpu_registers() const {
 	return vcpu.fpu_registers();
 }
+inline void Machine::set_fpu_registers(const tinykvm_fpuregs& regs) {
+	vcpu.set_fpu_registers(regs);
+}
 inline const struct kvm_sregs& Machine::get_special_registers() const {
 	return vcpu.get_special_registers();
 }

lib/tinykvm/vcpu.cpp

@@ -276,6 +276,12 @@ tinykvm_fpuregs vCPU::fpu_registers() const
 	}
 	return regs;
 }
+void vCPU::set_fpu_registers(const struct tinykvm_fpuregs& regs)
+{
+	if (ioctl(this->fd, KVM_SET_FPU, &regs) < 0) {
+		Machine::machine_exception("KVM_SET_FPU failed");
+	}
+}
 const kvm_sregs& vCPU::get_special_registers() const
 {
 #ifndef TINYKVM_USE_SYNCED_SREGS

lib/tinykvm/vcpu.hpp

@@ -15,6 +15,7 @@ namespace tinykvm
         const tinykvm_x86regs& registers() const;
         void set_registers(const struct tinykvm_x86regs &);
         tinykvm_fpuregs fpu_registers() const;
+		void set_fpu_registers(const struct tinykvm_fpuregs &);
         const struct kvm_sregs& get_special_registers() const;
         struct kvm_sregs& get_special_registers();
         void set_special_registers(const struct kvm_sregs &);