add compatibility with offloading

Signed-off-by: Amory Hoste <[email protected]>
vhive-serverless · Jan 23, 2022 · 032bd10 · 032bd10
1 parent aff3d2b
commit 032bd10
Show file tree

Hide file tree

Showing 5 changed files with 118 additions and 104 deletions.
diff --git a/firecracker-control/local.go b/firecracker-control/local.go
@@ -633,10 +633,12 @@ func (s *local) CreateSnapshot(ctx context.Context, req *proto.CreateSnapshotReq
 func (s *local) LoadSnapshot(ctx context.Context, req *proto.LoadSnapshotRequest) (*proto.LoadResponse, error) {
 	var err error
 
-	// Create shim if not exists yet
-	code, err := s.CreateShim(ctx, req.GetVMID())
-	if err != nil && code != codes.AlreadyExists {
-		return nil, err
+	// Create shim if not exists yet for non-offloaded snapshots
+	if ! req.Offloaded {
+		code, err := s.CreateShim(ctx, req.GetVMID())
+		if err != nil && code != codes.AlreadyExists {
+			return nil, err
+		}
 	}
 
 	client, err := s.shimFirecrackerClient(ctx, req.VMID)

diff --git a/proto/events.pb.go b/proto/events.pb.go
diff --git a/proto/firecracker.pb.go b/proto/firecracker.pb.go
diff --git a/proto/firecracker.proto b/proto/firecracker.proto
@@ -43,6 +43,7 @@ message CreateVMRequest {
     FirecrackerBalloonDevice BalloonDevice = 14;
 
     string NetworkNamespace = 15;
+    bool OffloadEnabled = 16;
 }
 
 message CreateVMResponse {
@@ -121,6 +122,7 @@ message LoadSnapshotRequest {
     bool EnableUserPF = 4;
     string NetworkNamespace = 5;
     string NewSnapshotPath = 6;
+    bool Offloaded = 7;
 }
 
 message LoadResponse {