vm: add context to Pool.Create()

Enable external abortion of the instance creation process. This is
especially useful for the qemu case where we retry the creation/boot up
to 1000 times, which can take significant time (e.g. it timeouts
syz-cluster pods on unstable kernels).

The context can be further propagated to WaitForSSH, but that requires
another quite significant vm/ refactoring.

1 files changed, 2 insertions, 2 deletions

diff --git a/vm/cuttlefish/cuttlefish.go b/vm/cuttlefish/cuttlefish.go
index 55e56d2e5..dbb95d51d 100644
--- a/vm/cuttlefish/cuttlefish.go
+++ b/vm/cuttlefish/cuttlefish.go
@@ -66,8 +66,8 @@ func (pool *Pool) Count() int {
 	return pool.gcePool.Count()
 }
 
-func (pool *Pool) Create(workdir string, index int) (vmimpl.Instance, error) {
-	gceInst, err := pool.gcePool.Create(workdir, index)
+func (pool *Pool) Create(ctx context.Context, workdir string, index int) (vmimpl.Instance, error) {
+	gceInst, err := pool.gcePool.Create(ctx, workdir, index)
 	if err != nil {
 		return nil, fmt.Errorf("failed to create underlying gce instance: %w", err)
 	}