Offload returns UPF socket

As opposed to load snapshots.
This is because the socket is needed for LoadSnapshot
and needs to be available before it is called

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) (*proto.LoadSnapshotResponse, error) {
+func (s *local) LoadSnapshot(ctx context.Context, req *proto.LoadSnapshotRequest) (*empty.Empty, error) {
 	client, err := s.shimFirecrackerClient(ctx, req.VMID)
 	if err != nil {
 		return nil, err
@@ -672,7 +672,7 @@ func (s *local) LoadSnapshot(ctx context.Context, req *proto.LoadSnapshotRequest
 // Offload Shuts down a VM started through the firecracker go sdk and deletes
 // the corresponding firecracker socket. All of the other resources (shim, other sockets)
 // will persist.
-func (s *local) Offload(ctx context.Context, req *proto.OffloadRequest) (*empty.Empty, error) {
+func (s *local) Offload(ctx context.Context, req *proto.OffloadRequest) (*proto.OffloadResponse, 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) (*proto.LoadSnapshotResponse, error) {
+func (s *service) LoadSnapshot(ctx context.Context, req *proto.LoadSnapshotRequest) (*empty.Empty, error) {
 	log.G(ctx).Debugf("load snapshot request: %+v", req)
 	return s.local.LoadSnapshot(ctx, req)
 }
@@ -137,7 +137,7 @@ func (s *service) CreateSnapshot(ctx context.Context, req *proto.CreateSnapshotR
 	return s.local.CreateSnapshot(ctx, req)
 }
 
-func (s *service) Offload(ctx context.Context, req *proto.OffloadRequest) (*empty.Empty, error) {
+func (s *service) Offload(ctx context.Context, req *proto.OffloadRequest) (*proto.OffloadResponse, error) {
 	log.G(ctx).Debugf("offload request: %+v", req)
 	return s.local.Offload(ctx, req)
 }

proto/firecracker.pb.go

@@ -116,7 +116,7 @@ message LoadSnapshotRequest {
     bool EnableUserPF = 4;
 }
 
-message LoadSnapshotResponse {
+message OffloadResponse {
     string SendSockPath = 1;
 }