| // Copyright The Monogon Project Authors. |
| // SPDX-License-Identifier: Apache-2.0 |
| |
| package mgmt |
| |
| import ( |
| "context" |
| |
| "golang.org/x/sys/unix" |
| "google.golang.org/grpc/codes" |
| "google.golang.org/grpc/status" |
| |
| apb "source.monogon.dev/metropolis/proto/api" |
| ) |
| |
| func (s *Service) UpdateNode(ctx context.Context, req *apb.UpdateNodeRequest) (*apb.UpdateNodeResponse, error) { |
| ok := s.updateMutex.TryLock() |
| if ok { |
| defer s.updateMutex.Unlock() |
| } else { |
| return nil, status.Error(codes.Aborted, "another UpdateNode RPC is in progress on this node") |
| } |
| if req.ActivationMode == apb.ActivationMode_ACTIVATION_MODE_INVALID { |
| return nil, status.Errorf(codes.InvalidArgument, "activation_mode needs to be explicitly specified") |
| } |
| if err := s.UpdateService.InstallImage(ctx, req.OsImage, req.ActivationMode == apb.ActivationMode_ACTIVATION_MODE_KEXEC); err != nil { |
| return nil, status.Errorf(codes.Unavailable, "error installing update: %v", err) |
| } |
| if req.ActivationMode != apb.ActivationMode_ACTIVATION_MODE_NONE { |
| |
| methodString, method := "reboot", unix.LINUX_REBOOT_CMD_RESTART |
| if req.ActivationMode == apb.ActivationMode_ACTIVATION_MODE_KEXEC { |
| methodString = "kexec" |
| method = unix.LINUX_REBOOT_CMD_KEXEC |
| } |
| |
| s.LogTree.MustLeveledFor("update").Infof("activating update with method: %s", methodString) |
| s.initiateReboot(method) |
| } |
| |
| return &apb.UpdateNodeResponse{}, nil |
| } |