package workers import ( "context" "fmt" "strings" "github.com/hashicorp/boundary/globals" "github.com/hashicorp/boundary/internal/daemon/controller/auth" "github.com/hashicorp/boundary/internal/daemon/controller/common" "github.com/hashicorp/boundary/internal/daemon/controller/common/scopeids" "github.com/hashicorp/boundary/internal/daemon/controller/handlers" "github.com/hashicorp/boundary/internal/errors" pbs "github.com/hashicorp/boundary/internal/gen/controller/api/services" "github.com/hashicorp/boundary/internal/perms" "github.com/hashicorp/boundary/internal/requests" "github.com/hashicorp/boundary/internal/server" "github.com/hashicorp/boundary/internal/server/store" "github.com/hashicorp/boundary/internal/types/action" "github.com/hashicorp/boundary/internal/types/resource" "github.com/hashicorp/boundary/internal/types/scope" pb "github.com/hashicorp/boundary/sdk/pbs/controller/api/resources/workers" "github.com/hashicorp/go-secure-stdlib/strutil" "github.com/hashicorp/nodeenrollment/types" "github.com/mr-tron/base58" "google.golang.org/grpc/codes" "google.golang.org/protobuf/proto" "google.golang.org/protobuf/types/known/structpb" "google.golang.org/protobuf/types/known/wrapperspb" ) const ( PkiWorkerType = "pki" KmsWorkerType = "kms" ) var ( maskManager handlers.MaskManager // IdActions contains the set of actions that can be performed on // individual resources IdActions = action.ActionSet{ action.NoOp, action.Read, action.Update, action.Delete, action.AddWorkerTags, action.SetWorkerTags, action.RemoveWorkerTags, } // CollectionActions contains the set of actions that can be performed on // this collection CollectionActions = action.ActionSet{ action.CreateWorkerLed, action.List, } ) func init() { var err error if maskManager, err = handlers.NewMaskManager(handlers.MaskDestination{&store.Worker{}}, handlers.MaskSource{&pb.Worker{}}); err != nil { panic(err) } } // Service handles request as described by the pbs.WorkerServiceServer interface. type Service struct { pbs.UnimplementedWorkerServiceServer repoFn common.ServersRepoFactory iamRepoFn common.IamRepoFactory } // NewService returns a worker service which handles worker related requests to boundary. func NewService(ctx context.Context, repo common.ServersRepoFactory, iamRepoFn common.IamRepoFactory) (Service, error) { const op = "workers.NewService" if repo == nil { return Service{}, errors.New(ctx, errors.InvalidParameter, op, "missing server repository") } if iamRepoFn == nil { return Service{}, errors.New(ctx, errors.InvalidParameter, op, "missing iam repository") } return Service{repoFn: repo, iamRepoFn: iamRepoFn}, nil } var _ pbs.WorkerServiceServer = Service{} // ListWorkers implements the interface pbs.WorkerServiceServer. func (s Service) ListWorkers(ctx context.Context, req *pbs.ListWorkersRequest) (*pbs.ListWorkersResponse, error) { if err := validateListRequest(req); err != nil { return nil, err } authResults := s.authResult(ctx, req.GetScopeId(), action.List) if authResults.Error != nil { // If it's forbidden, and it's a recursive request, and they're // successfully authenticated but just not authorized, keep going as we // may have authorization on downstream scopes. Or, if they've not // authenticated, still process in case u_anon has permissions. if (authResults.Error == handlers.ForbiddenError() || authResults.Error == handlers.UnauthenticatedError()) && req.GetRecursive() && authResults.AuthenticationFinished { } else { return nil, authResults.Error } } scopeIds, scopeInfoMap, err := scopeids.GetListingScopeIds( ctx, s.iamRepoFn, authResults, req.GetScopeId(), resource.Worker, req.GetRecursive()) if err != nil { return nil, err } // If no scopes match, return an empty response if len(scopeIds) == 0 { return &pbs.ListWorkersResponse{}, nil } ul, err := s.listFromRepo(ctx, scopeIds) if err != nil { return nil, err } if len(ul) == 0 { return &pbs.ListWorkersResponse{}, nil } filter, err := handlers.NewFilter(req.GetFilter()) if err != nil { return nil, err } finalItems := make([]*pb.Worker, 0, len(ul)) res := perms.Resource{ Type: resource.Worker, } for _, item := range ul { res.Id = item.GetPublicId() res.ScopeId = item.GetScopeId() authorizedActions := authResults.FetchActionSetForId(ctx, item.GetPublicId(), IdActions, auth.WithResource(&res)).Strings() if len(authorizedActions) == 0 { continue } outputFields := authResults.FetchOutputFields(res, action.List).SelfOrDefaults(authResults.UserId) outputOpts := make([]handlers.Option, 0, 3) outputOpts = append(outputOpts, handlers.WithOutputFields(&outputFields)) if outputFields.Has(globals.ScopeField) { outputOpts = append(outputOpts, handlers.WithScope(scopeInfoMap[item.GetScopeId()])) } if outputFields.Has(globals.AuthorizedActionsField) { outputOpts = append(outputOpts, handlers.WithAuthorizedActions(authorizedActions)) } item, err := toProto(ctx, item, outputOpts...) if err != nil { return nil, err } if filter.Match(item) { finalItems = append(finalItems, item) } } return &pbs.ListWorkersResponse{Items: finalItems}, nil } // GetWorker implements the interface pbs.WorkerServiceServer. func (s Service) GetWorker(ctx context.Context, req *pbs.GetWorkerRequest) (*pbs.GetWorkerResponse, error) { const op = "workers.(Service).GetWorker" if err := validateGetRequest(req); err != nil { return nil, err } authResults := s.authResult(ctx, req.GetId(), action.Read) if authResults.Error != nil { return nil, authResults.Error } w, err := s.getFromRepo(ctx, req.GetId()) if err != nil { return nil, err } outputFields, ok := requests.OutputFields(ctx) if !ok { return nil, errors.New(ctx, errors.Internal, op, "no request context found") } outputOpts := make([]handlers.Option, 0, 3) outputOpts = append(outputOpts, handlers.WithOutputFields(&outputFields)) if outputFields.Has(globals.ScopeField) { outputOpts = append(outputOpts, handlers.WithScope(authResults.Scope)) } if outputFields.Has(globals.AuthorizedActionsField) { outputOpts = append(outputOpts, handlers.WithAuthorizedActions(authResults.FetchActionSetForId(ctx, w.GetPublicId(), IdActions).Strings())) } item, err := toProto(ctx, w, outputOpts...) if err != nil { return nil, err } return &pbs.GetWorkerResponse{Item: item}, nil } // CreateWorker implements the interface pbs.WorkerServiceServer. func (s Service) CreateWorkerLed(ctx context.Context, req *pbs.CreateWorkerLedRequest) (*pbs.CreateWorkerLedResponse, error) { const op = "workers.(Service).CreateWorkerLed" if err := validateCreateRequest(req); err != nil { return nil, err } authResults := s.authResult(ctx, req.GetItem().GetScopeId(), action.CreateWorkerLed) if authResults.Error != nil { return nil, authResults.Error } reqBytes, err := base58.FastBase58Decoding(req.GetItem().WorkerGeneratedAuthToken.GetValue()) if err != nil { return nil, fmt.Errorf("%s: error decoding node_credentials_token: %w", op, err) } // Decode the proto into the request creds := new(types.FetchNodeCredentialsRequest) if err := proto.Unmarshal(reqBytes, creds); err != nil { return nil, fmt.Errorf("%s: error unmarshaling node_credentials_token: %w", op, err) } created, err := s.createInRepo(ctx, req.GetItem(), creds) if err != nil { return nil, fmt.Errorf("%s: error creating worker: %w", op, err) } outputFields, ok := requests.OutputFields(ctx) if !ok { return nil, errors.New(ctx, errors.Internal, op, "no request context found") } outputOpts := make([]handlers.Option, 0, 3) outputOpts = append(outputOpts, handlers.WithOutputFields(&outputFields)) if outputFields.Has(globals.ScopeField) { outputOpts = append(outputOpts, handlers.WithScope(authResults.Scope)) } if outputFields.Has(globals.AuthorizedActionsField) { outputOpts = append(outputOpts, handlers.WithAuthorizedActions(authResults.FetchActionSetForId(ctx, created.GetPublicId(), IdActions).Strings())) } item, err := toProto(ctx, created, outputOpts...) if err != nil { return nil, err } return &pbs.CreateWorkerLedResponse{Item: item}, nil } // DeleteWorker implements the interface pbs.WorkerServiceServer. func (s Service) DeleteWorker(ctx context.Context, req *pbs.DeleteWorkerRequest) (*pbs.DeleteWorkerResponse, error) { if err := validateDeleteRequest(req); err != nil { return nil, err } authResults := s.authResult(ctx, req.GetId(), action.Delete) if authResults.Error != nil { return nil, authResults.Error } _, err := s.deleteFromRepo(ctx, req.GetId()) if err != nil { return nil, err } return nil, nil } // UpdateWorker implements the interface pbs.WorkerServiceServer. func (s Service) UpdateWorker(ctx context.Context, req *pbs.UpdateWorkerRequest) (*pbs.UpdateWorkerResponse, error) { const op = "workers.(Service).UpdateWorker" if err := validateUpdateRequest(req); err != nil { return nil, err } authResults := s.authResult(ctx, req.GetId(), action.Update) if authResults.Error != nil { return nil, authResults.Error } w, err := s.updateInRepo(ctx, authResults.Scope.GetId(), req.GetId(), req.GetUpdateMask().GetPaths(), req.GetItem()) if err != nil { return nil, err } outputFields, ok := requests.OutputFields(ctx) if !ok { return nil, errors.New(ctx, errors.Internal, op, "no request context found") } outputOpts := make([]handlers.Option, 0, 3) outputOpts = append(outputOpts, handlers.WithOutputFields(&outputFields)) if outputFields.Has(globals.ScopeField) { outputOpts = append(outputOpts, handlers.WithScope(authResults.Scope)) } if outputFields.Has(globals.AuthorizedActionsField) { outputOpts = append(outputOpts, handlers.WithAuthorizedActions(authResults.FetchActionSetForId(ctx, w.GetPublicId(), IdActions).Strings())) } item, err := toProto(ctx, w, outputOpts...) if err != nil { return nil, err } return &pbs.UpdateWorkerResponse{Item: item}, nil } // AddWorkerTags implements the interface pbs.WorkerServiceServer. func (s Service) AddWorkerTags(ctx context.Context, req *pbs.AddWorkerTagsRequest) (*pbs.AddWorkerTagsResponse, error) { const op = "workers.(Service).AddWorkerTags" if err := validateAddTagsRequest(req); err != nil { return nil, err } authResults := s.authResult(ctx, req.GetId(), action.AddWorkerTags) if authResults.Error != nil { return nil, authResults.Error } w, err := s.addTagsInRepo(ctx, req.GetId(), req.GetVersion(), req.GetApiTags()) if err != nil { return nil, err } outputFields, ok := requests.OutputFields(ctx) if !ok { return nil, errors.New(ctx, errors.Internal, op, "no request context found") } outputOpts := make([]handlers.Option, 0, 3) outputOpts = append(outputOpts, handlers.WithOutputFields(&outputFields)) if outputFields.Has(globals.ScopeField) { outputOpts = append(outputOpts, handlers.WithScope(authResults.Scope)) } if outputFields.Has(globals.AuthorizedActionsField) { outputOpts = append(outputOpts, handlers.WithAuthorizedActions(authResults.FetchActionSetForId(ctx, w.GetPublicId(), IdActions).Strings())) } item, err := toProto(ctx, w, outputOpts...) if err != nil { return nil, err } return &pbs.AddWorkerTagsResponse{Item: item}, nil } // SetWorkerTags implements the interface pbs.WorkerServiceServer. func (s Service) SetWorkerTags(ctx context.Context, req *pbs.SetWorkerTagsRequest) (*pbs.SetWorkerTagsResponse, error) { const op = "workers.(Service).SetWorkerTags" if err := validateSetTagsRequest(req); err != nil { return nil, err } authResults := s.authResult(ctx, req.GetId(), action.SetWorkerTags) if authResults.Error != nil { return nil, authResults.Error } w, err := s.setTagsInRepo(ctx, req.GetId(), req.GetVersion(), req.GetApiTags()) if err != nil { return nil, err } outputFields, ok := requests.OutputFields(ctx) if !ok { return nil, errors.New(ctx, errors.Internal, op, "no request context found") } outputOpts := make([]handlers.Option, 0, 3) outputOpts = append(outputOpts, handlers.WithOutputFields(&outputFields)) if outputFields.Has(globals.ScopeField) { outputOpts = append(outputOpts, handlers.WithScope(authResults.Scope)) } if outputFields.Has(globals.AuthorizedActionsField) { outputOpts = append(outputOpts, handlers.WithAuthorizedActions(authResults.FetchActionSetForId(ctx, w.GetPublicId(), IdActions).Strings())) } item, err := toProto(ctx, w, outputOpts...) if err != nil { return nil, err } return &pbs.SetWorkerTagsResponse{Item: item}, nil } // RemoveWorkerTags implements the interface pbs.WorkerServiceServer. func (s Service) RemoveWorkerTags(ctx context.Context, req *pbs.RemoveWorkerTagsRequest) (*pbs.RemoveWorkerTagsResponse, error) { const op = "workers.(Service).RemoveWorkerTags" if err := validateRemoveTagsRequest(req); err != nil { return nil, err } authResults := s.authResult(ctx, req.GetId(), action.RemoveWorkerTags) if authResults.Error != nil { return nil, authResults.Error } w, err := s.removeTagsInRepo(ctx, req.GetId(), req.GetVersion(), req.GetApiTags()) if err != nil { return nil, err } outputFields, ok := requests.OutputFields(ctx) if !ok { return nil, errors.New(ctx, errors.Internal, op, "no request context found") } outputOpts := make([]handlers.Option, 0, 3) outputOpts = append(outputOpts, handlers.WithOutputFields(&outputFields)) if outputFields.Has(globals.ScopeField) { outputOpts = append(outputOpts, handlers.WithScope(authResults.Scope)) } if outputFields.Has(globals.AuthorizedActionsField) { outputOpts = append(outputOpts, handlers.WithAuthorizedActions(authResults.FetchActionSetForId(ctx, w.GetPublicId(), IdActions).Strings())) } item, err := toProto(ctx, w, outputOpts...) if err != nil { return nil, err } return &pbs.RemoveWorkerTagsResponse{Item: item}, nil } func (s Service) listFromRepo(ctx context.Context, scopeIds []string) ([]*server.Worker, error) { repo, err := s.repoFn() if err != nil { return nil, err } wl, err := repo.ListWorkers(ctx, scopeIds, server.WithLiveness(-1)) if err != nil { return nil, err } return wl, nil } func (s Service) getFromRepo(ctx context.Context, id string) (*server.Worker, error) { repo, err := s.repoFn() if err != nil { return nil, err } w, err := repo.LookupWorker(ctx, id) if err != nil { if errors.IsNotFoundError(err) { return nil, handlers.NotFoundErrorf("Worker %q doesn't exist.", id) } return nil, err } if w == nil { return nil, handlers.NotFoundErrorf("Worker %q doesn't exist.", id) } return w, nil } func (s Service) createInRepo(ctx context.Context, worker *pb.Worker, creds *types.FetchNodeCredentialsRequest) (*server.Worker, error) { const op = "workers.(Service).createInRepo" repo, err := s.repoFn() if err != nil { return nil, err } newWorker := server.NewWorker( worker.GetScopeId(), server.WithName(worker.GetName().GetValue()), server.WithDescription(worker.GetDescription().GetValue()), ) retWorker, err := repo.CreateWorker(ctx, newWorker, server.WithFetchNodeCredentialsRequest(creds)) if err != nil { return nil, errors.Wrap(ctx, err, op, errors.WithMsg("unable to create worker")) } return retWorker, nil } func (s Service) deleteFromRepo(ctx context.Context, id string) (bool, error) { const op = "workers.(Service).deleteFromRepo" repo, err := s.repoFn() if err != nil { return false, err } rows, err := repo.DeleteWorker(ctx, id) if err != nil { if errors.IsNotFoundError(err) { return false, nil } return false, errors.Wrap(ctx, err, op, errors.WithMsg("unable to delete worker")) } return rows > 0, nil } func (s Service) updateInRepo(ctx context.Context, scopeId, id string, mask []string, item *pb.Worker) (*server.Worker, error) { const op = "workers.(Service).updateInRepo" var opts []server.Option if desc := item.GetDescription(); desc != nil { opts = append(opts, server.WithDescription(desc.GetValue())) } if name := item.GetName(); name != nil { opts = append(opts, server.WithName(name.GetValue())) } w := server.NewWorker(scopeId, opts...) w.PublicId = id dbMask := maskManager.Translate(mask) if len(dbMask) == 0 { return nil, handlers.InvalidArgumentErrorf("No valid fields included in the update mask.", map[string]string{"update_mask": "No valid fields provided in the update mask."}) } repo, err := s.repoFn() if err != nil { return nil, err } out, rowsUpdated, err := repo.UpdateWorker(ctx, w, item.GetVersion(), dbMask) if err != nil { return nil, errors.Wrap(ctx, err, op, errors.WithMsg("unable to update worker")) } if rowsUpdated == 0 { return nil, handlers.NotFoundErrorf("Worker %q doesn't exist or incorrect version provided.", id) } return out, nil } func (s Service) addTagsInRepo(ctx context.Context, workerId string, workerVersion uint32, addTags map[string]*structpb.ListValue) (*server.Worker, error) { const op = "workers.(Service).addTagsInRepo" repo, err := s.repoFn() if err != nil { return nil, err } tags := make([]*server.Tag, 0, len(addTags)) for k, lv := range addTags { for _, v := range lv.GetValues() { tags = append(tags, &server.Tag{Key: k, Value: v.GetStringValue()}) } } _, err = repo.AddWorkerTags(ctx, workerId, workerVersion, tags) if err != nil { return nil, handlers.ApiErrorWithCodeAndMessage(codes.Internal, "Unable to add worker tags in repo: %v.", err) } w, err := repo.LookupWorker(ctx, workerId) if err != nil { return nil, errors.Wrap(ctx, err, op, errors.WithMsg("unable to look up worker after adding tags")) } if w == nil { return nil, handlers.ApiErrorWithCodeAndMessage(codes.Internal, "Unable to look up worker after adding tags.") } return w, nil } func (s Service) setTagsInRepo(ctx context.Context, workerId string, workerVersion uint32, setTags map[string]*structpb.ListValue) (*server.Worker, error) { const op = "workers.(Service).setTagsInRepo" repo, err := s.repoFn() if err != nil { return nil, err } tags := make([]*server.Tag, 0, len(setTags)) for k, lv := range setTags { for _, v := range lv.GetValues() { tags = append(tags, &server.Tag{Key: k, Value: v.GetStringValue()}) } } _, err = repo.SetWorkerTags(ctx, workerId, workerVersion, tags) if err != nil { return nil, handlers.ApiErrorWithCodeAndMessage(codes.Internal, "Unable to set worker tags in repo: %v.", err) } w, err := repo.LookupWorker(ctx, workerId) if err != nil { return nil, errors.Wrap(ctx, err, op, errors.WithMsg("unable to look up worker after setting tags")) } if w == nil { return nil, handlers.ApiErrorWithCodeAndMessage(codes.Internal, "Unable to look up worker after setting tags.") } return w, nil } func (s Service) removeTagsInRepo(ctx context.Context, workerId string, workerVersion uint32, removeTags map[string]*structpb.ListValue) (*server.Worker, error) { const op = "workers.(Service).removeTagsInRepo" repo, err := s.repoFn() if err != nil { return nil, err } tags := make([]*server.Tag, 0, len(removeTags)) for k, lv := range removeTags { for _, v := range lv.GetValues() { tags = append(tags, &server.Tag{Key: k, Value: v.GetStringValue()}) } } _, err = repo.DeleteWorkerTags(ctx, workerId, workerVersion, tags) if err != nil { return nil, handlers.ApiErrorWithCodeAndMessage(codes.Internal, "Unable to remove worker tags in repo: %v.", err) } w, err := repo.LookupWorker(ctx, workerId) if err != nil { return nil, errors.Wrap(ctx, err, op, errors.WithMsg("unable to look up worker after removing tags")) } if w == nil { return nil, handlers.ApiErrorWithCodeAndMessage(codes.Internal, "Unable to look up worker after removing tags.") } return w, nil } func (s Service) authResult(ctx context.Context, id string, a action.Type) auth.VerifyResults { res := auth.VerifyResults{} repo, err := s.repoFn() if err != nil { res.Error = err return res } var parentId string opts := []auth.Option{auth.WithType(resource.Worker), auth.WithAction(a)} switch a { case action.List, action.CreateWorkerLed: parentId = id default: w, err := repo.LookupWorker(ctx, id) if err != nil { res.Error = err return res } if w == nil { res.Error = handlers.NotFoundError() return res } parentId = w.GetScopeId() opts = append(opts, auth.WithId(id)) } opts = append(opts, auth.WithScopeId(parentId)) return auth.Verify(ctx, opts...) } func toProto(ctx context.Context, in *server.Worker, opt ...handlers.Option) (*pb.Worker, error) { const op = "workers.toProto" opts := handlers.GetOpts(opt...) if opts.WithOutputFields == nil { return nil, handlers.ApiErrorWithCodeAndMessage(codes.Internal, "output fields not found when building worker proto") } outputFields := *opts.WithOutputFields out := pb.Worker{} if outputFields.Has(globals.IdField) { out.Id = in.GetPublicId() } if outputFields.Has(globals.ScopeIdField) { out.ScopeId = in.GetScopeId() } if outputFields.Has(globals.DescriptionField) && in.GetDescription() != "" { out.Description = wrapperspb.String(in.GetDescription()) } if outputFields.Has(globals.NameField) && in.GetName() != "" { out.Name = wrapperspb.String(in.GetName()) } if outputFields.Has(globals.CreatedTimeField) { out.CreatedTime = in.GetCreateTime().GetTimestamp() } if outputFields.Has(globals.UpdatedTimeField) { out.UpdatedTime = in.GetUpdateTime().GetTimestamp() } if outputFields.Has(globals.VersionField) { out.Version = in.GetVersion() } if outputFields.Has(globals.ScopeField) { out.Scope = opts.WithScope } if outputFields.Has(globals.AuthorizedActionsField) { out.AuthorizedActions = opts.WithAuthorizedActions if in.Type == KmsWorkerType && out.AuthorizedActions != nil { // KMS workers cannot be updated through the API allActions := out.AuthorizedActions out.AuthorizedActions = make([]string, 0, len(allActions)) for _, act := range allActions { if act != action.Update.String() { out.AuthorizedActions = append(out.AuthorizedActions, act) } } } } if outputFields.Has(globals.AddressField) && in.GetAddress() != "" { out.Address = in.GetAddress() } if outputFields.Has(globals.TypeField) && in.GetType() != "" { out.Type = in.GetType() } if outputFields.Has(globals.LastStatusTimeField) { out.LastStatusTime = in.GetLastStatusTime().GetTimestamp() } if outputFields.Has(globals.ActiveConnectionCountField) { out.ActiveConnectionCount = &wrapperspb.UInt32Value{Value: in.ActiveConnectionCount()} } if outputFields.Has(globals.ConfigTagsField) && len(in.GetConfigTags()) > 0 { var err error out.ConfigTags, err = tagsToMapProto(in.GetConfigTags()) if err != nil { return nil, errors.Wrap(ctx, err, op, errors.WithMsg("error preparing config tags proto")) } } if outputFields.Has(globals.ApiTagsField) && len(in.GetApiTags()) > 0 { var err error out.ApiTags, err = tagsToMapProto(in.GetApiTags()) if err != nil { return nil, errors.Wrap(ctx, err, op, errors.WithMsg("error preparing api tags proto")) } } if outputFields.Has(globals.CanonicalTagsField) && len(in.CanonicalTags()) > 0 { var err error out.CanonicalTags, err = tagsToMapProto(in.CanonicalTags()) if err != nil { return nil, errors.Wrap(ctx, err, op, errors.WithMsg("error preparing canonical tags proto")) } } return &out, nil } func tagsToMapProto(in map[string][]string) (map[string]*structpb.ListValue, error) { b := make(map[string][]interface{}) for k, v := range in { result := make([]interface{}, 0, len(v)) for _, t := range v { result = append(result, t) } b[k] = result } ret := make(map[string]*structpb.ListValue) var err error for k, v := range b { ret[k], err = structpb.NewList(v) if err != nil { return nil, err } } return ret, nil } // A validateX method should exist for each method above. These methods do not make calls to any backing service but enforce // requirements on the structure of the request. They verify that: // * The path passed in is correctly formatted // * All required parameters are set // * There are no conflicting parameters provided func validateGetRequest(req *pbs.GetWorkerRequest) error { return handlers.ValidateGetRequest(handlers.NoopValidatorFn, req, server.WorkerPrefix) } func validateListRequest(req *pbs.ListWorkersRequest) error { badFields := map[string]string{} if req.GetScopeId() != scope.Global.String() { badFields["scope_id"] = "Must be 'global' when listing." } if _, err := handlers.NewFilter(req.GetFilter()); err != nil { badFields["filter"] = fmt.Sprintf("This field could not be parsed. %v", err) } if len(badFields) > 0 { return handlers.InvalidArgumentErrorf("Error in provided request.", badFields) } return nil } func validateDeleteRequest(req *pbs.DeleteWorkerRequest) error { return handlers.ValidateDeleteRequest(handlers.NoopValidatorFn, req, server.WorkerPrefix) } func validateUpdateRequest(req *pbs.UpdateWorkerRequest) error { return handlers.ValidateUpdateRequest(req, req.GetItem(), func() map[string]string { badFields := map[string]string{} if req.GetItem().GetAddress() != "" { badFields[globals.AddressField] = "This is a read only field." } if req.GetItem().GetCanonicalTags() != nil { badFields[globals.CanonicalAddressField] = "This is a read only field." } if req.GetItem().GetConfigTags() != nil { badFields[globals.TagsField] = "This is a read only field." } nameString := req.GetItem().GetName().String() if !strutil.Printable(nameString) { badFields[globals.NameField] = "Contains non-printable characters" } if strings.ToLower(nameString) != nameString { badFields[globals.NameField] = "Must be all lowercase." } descriptionString := req.GetItem().GetDescription().String() if !strutil.Printable(descriptionString) { badFields[globals.DescriptionField] = "Contains non-printable characters." } return badFields }, server.WorkerPrefix) } func validateCreateRequest(req *pbs.CreateWorkerLedRequest) error { return handlers.ValidateCreateRequest(req.GetItem(), func() map[string]string { const ( mustBeGlobalMsg = "Must be 'global'" cannotBeEmptyMsg = "Cannot be empty." readOnlyFieldMsg = "This is a read only field." ) badFields := map[string]string{} if scope.Global.String() != req.GetItem().GetScopeId() { badFields[globals.ScopeIdField] = mustBeGlobalMsg } // FIXME: in the future, we won't require this token since we'll support // the server led flow where a token is returned when a worker is created. if req.GetItem().WorkerGeneratedAuthToken == nil { badFields[globals.WorkerGeneratedAuthTokenField] = cannotBeEmptyMsg } if req.GetItem().Address != "" { badFields[globals.CanonicalAddressField] = readOnlyFieldMsg } if req.GetItem().CanonicalTags != nil { badFields[globals.CanonicalTagsField] = readOnlyFieldMsg } if req.GetItem().ConfigTags != nil { badFields[globals.ConfigTagsField] = readOnlyFieldMsg } if req.GetItem().LastStatusTime != nil { badFields[globals.LastStatusTimeField] = readOnlyFieldMsg } if req.GetItem().AuthorizedActions != nil { badFields[globals.AuthorizedActionsField] = readOnlyFieldMsg } nameString := req.GetItem().GetName().String() if !strutil.Printable(nameString) { badFields[globals.NameField] = "Name contains non-printable characters." } if strings.ToLower(nameString) != nameString { badFields[globals.NameField] = "Name must be all lowercase." } descriptionString := req.GetItem().GetDescription().String() if !strutil.Printable(descriptionString) { badFields[globals.DescriptionField] = "Description contains non-printable characters." } return badFields }) } // validateStringForDb checks a string is valid for db storage and returns a string for an error message if needed. // returns an empty string otherwise. func validateStringForDb(str string) string { switch { case len(str) <= 0: return "must be non-empty." case len(str) > 512: return "must be within 512 characters." default: return "" } } func validateAddTagsRequest(req *pbs.AddWorkerTagsRequest) error { badFields := map[string]string{} if !handlers.ValidId(handlers.Id(req.GetId()), server.WorkerPrefix) { badFields[globals.IdField] = "Incorrectly formatted identifier." } if req.GetVersion() == 0 { badFields[globals.VersionField] = "Required field." } if req.GetApiTags() == nil { badFields[globals.ApiTagsField] = "Must be non-empty." } for k, lv := range req.GetApiTags() { if err := validateStringForDb(k); err != "" { badFields[globals.ApiTagsField] = "Tag keys " + err break } if lv.GetValues() == nil { badFields[globals.ApiTagsField] = "Tag values must be non-empty." break } for _, v := range lv.GetValues() { if _, ok := v.GetKind().(*structpb.Value_StringValue); !ok { badFields[globals.ApiTagsField] = "Tag values must be strings." break } if err := validateStringForDb(v.GetStringValue()); err != "" { badFields[globals.ApiTagsField] = "Tag values " + err break } } } if len(badFields) > 0 { return handlers.InvalidArgumentErrorf("Errors in provided fields.", badFields) } return nil } func validateSetTagsRequest(req *pbs.SetWorkerTagsRequest) error { badFields := map[string]string{} if !handlers.ValidId(handlers.Id(req.GetId()), server.WorkerPrefix) { badFields[globals.IdField] = "Incorrectly formatted identifier." } if req.GetVersion() == 0 { badFields[globals.VersionField] = "Required field." } for k, lv := range req.GetApiTags() { if err := validateStringForDb(k); err != "" { badFields[globals.ApiTagsField] = "Tag keys " + err break } if lv.GetValues() == nil { badFields[globals.ApiTagsField] = "Tag values must be non-empty." break } for _, v := range lv.GetValues() { if _, ok := v.GetKind().(*structpb.Value_StringValue); !ok { badFields[globals.ApiTagsField] = "Tag values must be strings." break } if err := validateStringForDb(v.GetStringValue()); err != "" { badFields[globals.ApiTagsField] = "Tag values " + err break } } } if len(badFields) > 0 { return handlers.InvalidArgumentErrorf("Errors in provided fields.", badFields) } return nil } func validateRemoveTagsRequest(req *pbs.RemoveWorkerTagsRequest) error { badFields := map[string]string{} if !handlers.ValidId(handlers.Id(req.GetId()), server.WorkerPrefix) { badFields[globals.IdField] = "Incorrectly formatted identifier." } if req.GetVersion() == 0 { badFields[globals.VersionField] = "Required field." } if req.GetApiTags() == nil { badFields[globals.ApiTagsField] = "Must be non-empty." } for k, lv := range req.GetApiTags() { if err := validateStringForDb(k); err != "" { badFields[globals.ApiTagsField] = "Tag keys " + err break } if lv.GetValues() == nil { badFields[globals.ApiTagsField] = "Tag values must be non-empty." break } for _, v := range lv.GetValues() { if _, ok := v.GetKind().(*structpb.Value_StringValue); !ok { badFields[globals.ApiTagsField] = "Tag values must be strings." break } if err := validateStringForDb(v.GetStringValue()); err != "" { badFields[globals.ApiTagsField] = "Tag values " + err break } } } if len(badFields) > 0 { return handlers.InvalidArgumentErrorf("Errors in provided fields.", badFields) } return nil }