add compatibility with offloading

Signed-off-by: Amory Hoste <[email protected]>

Author: amohoste

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)

proto/events.pb.go

@@ -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 {