qemu: implement support for OVMF files

Signed-off-by: Morten Linderud <[email protected]>

qemu/qemu.go

@@ -164,6 +164,16 @@ func WithTask(t ...Task) Fn {
 	}
 }
 
+func WithOVMF(vars, code string) Fn {
+	return func(alloc *IDAllocator, opts *Options) error {
+		opts.OVMF = &OVMFFiles{
+			Secboot: code,
+			Vars:    vars,
+		}
+		return nil
+	}
+}
+
 // OptionsFor evaluates the given config functions and returns an Options object.
 func OptionsFor(arch Arch, fns ...Fn) (*Options, error) {
 	var vmTimeout time.Duration
@@ -217,6 +227,11 @@ func StartContext(ctx context.Context, arch Arch, fns ...Fn) (*VM, error) {
 	return o.Start(ctx)
 }
 
+type OVMFFiles struct {
+	Secboot string
+	Vars    string
+}
+
 // Options are VM start-up parameters.
 type Options struct {
 	// arch is the QEMU architecture used.
@@ -264,6 +279,9 @@ type Options struct {
 
 	// ExtraFiles are extra files passed to QEMU on start.
 	ExtraFiles []*os.File
+
+	// OVMF is the OVMF files for UEFI and Secure Boot support.
+	OVMF *OVMFFiles
 }
 
 // AddFile adds the file to the QEMU process and returns the FD it will be in
@@ -447,6 +465,15 @@ func (o *Options) Cmdline() ([]string, error) {
 		args = append(args, "-initrd", o.Initramfs)
 	}
 
+	if o.OVMF != nil {
+		args = append(args, fmt.Sprintf("-drive if=pflash,format=raw,unit=0,file=%s,readonly=on", o.OVMF.Secboot))
+		args = append(args, fmt.Sprintf("-drive if=pflash,format=raw,unit=1,file=%s", o.OVMF.Vars))
+		// Append 2G of ram just in case
+		args = append(args, "-m 2G")
+		// Add specific machine for support
+		args = append(args, "-machine type=q35,smm=on")
+	}
+
 	return args, nil
 }