UPF sock pass back to orch

Signed-off-by: Plamen Petrov <[email protected]>

Author: plamenmpetrov

firecracker-control/local.go

@@ -652,7 +652,7 @@ func (s *local) CreateSnapshot(ctx context.Context, req *proto.CreateSnapshotReq
 }
 
 // LoadSnapshot Loads a snapshot of a VM
-func (s *local) LoadSnapshot(ctx context.Context, req *proto.LoadSnapshotRequest) (*empty.Empty, error) {
+func (s *local) LoadSnapshot(ctx context.Context, req *proto.LoadSnapshotRequest) (*proto.LoadSnapshotResponse, error) {
 	client, err := s.shimFirecrackerClient(ctx, req.VMID)
 	if err != nil {
 		return nil, err

firecracker-control/service.go

@@ -127,7 +127,7 @@ func (s *service) UpdateBalloonStats(ctx context.Context, req *proto.UpdateBallo
 	return s.local.UpdateBalloonStats(ctx, req)
 }
 
-func (s *service) LoadSnapshot(ctx context.Context, req *proto.LoadSnapshotRequest) (*empty.Empty, error) {
+func (s *service) LoadSnapshot(ctx context.Context, req *proto.LoadSnapshotRequest) (*proto.LoadSnapshotResponse, error) {
 	log.G(ctx).Debugf("load snapshot request: %+v", req)
 	return s.local.LoadSnapshot(ctx, req)
 }

proto/firecracker.pb.go

@@ -116,6 +116,10 @@ message LoadSnapshotRequest {
     bool EnableUserPF = 4;
 }
 
+message LoadSnapshotResponse {
+    string SendSockPath = 1;
+}
+
 message OffloadRequest {
     string VMID = 1;
 }