go.chromium.org/luci@v0.0.0-20240309015107-7cdc2e660f33/server/auth/realms/permissions.go

// Copyright 2020 The LUCI Authors.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//      http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

package realms

import (
	"strings"
	"sync"

	"go.chromium.org/luci/common/errors"
)

var (
	mu            sync.RWMutex
	perms         = map[string]PermissionFlags{}
	forbidChanges bool
)

// PermissionFlags describe how a registered permission is going to be used in
// the current process.
//
// These are purely process-local flags. It is possible for the same single
// permission to be defined with different flags in different processes.
type PermissionFlags uint8

const (
	// UsedInQueryRealms indicates the permission will be used with QueryRealms.
	//
	// It instructs the runtime to build an in-memory index to speed up the query.
	// This flag is necessary for QueryRealms function to work. It implies some
	// performance and memory costs and should be used only with permissions that
	// are going to be passed to QueryRealms.
	UsedInQueryRealms PermissionFlags = 1 << iota
)

// Permission is a symbol that has form "<service>.<subject>.<verb>", which
// describes some elementary action ("<verb>") that can be done to some category
// of resources ("<subject>"), managed by some particular kind of LUCI service
// ("<service>").
//
// Each individual LUCI service should document what permissions it checks and
// when. It becomes a part of service's public API. Usually services should
// check only permissions of resources they own (e.g. "<service>.<subject>.*"),
// but in exceptional cases they may also check permissions intended for other
// services. This is primarily useful for services that somehow "proxy" access
// to resources.
type Permission struct {
	name string
}

// Name is "<service>.<subject>.<verb>" string.
func (p Permission) Name() string {
	return p.name
}

// String allows permissions to be used as "%s" in format strings.
func (p Permission) String() string {
	return p.name
}

// AddFlags appends given flags to a registered permission.
//
// Permissions are usually defined in shared packages used by multiple services.
// Flags that makes sense for one service do not always make sense for another.
// For that reason RegisterPermission and AddFlags are two separate functions.
//
// Must to be called during startup, e.g. in init(). Panics if called when
// the process is already serving requests.
func (p Permission) AddFlags(flags PermissionFlags) {
	mu.Lock()
	defer mu.Unlock()
	if forbidChanges {
		panic("cannot call Permission.AddFlags while already serving requests, do it before starting the serving loop, e.g. in init()")
	}
	perms[p.name] |= flags
}

// clearPermissions removes all registered permissions (for tests).
func clearPermissions() {
	mu.Lock()
	perms = map[string]PermissionFlags{}
	forbidChanges = false
	mu.Unlock()
}

// RegisterPermission adds a new permission with the given name to the process
// registry or returns an existing one.
//
// Panics if the permission name doesn't look like "<service>.<subject>.<verb>".
//
// Must to be called during startup, e.g. in init(). Panics if called when
// the process is already serving requests.
func RegisterPermission(name string) Permission {
	mu.Lock()
	defer mu.Unlock()
	if forbidChanges {
		panic("cannot call RegisterPermission while already serving requests, do it before starting the serving loop, e.g. in init()")
	}
	if err := ValidatePermissionName(name); err != nil {
		panic(err)
	}
	perms[name] |= 0
	return Permission{name: name}
}

// GetPermissions returns the permissions with the matching names. The order of
// the permission is the same as the provided names.
//
// Implicitly calls ForbidPermissionChanges to make sure no new permissions
// are added later.
//
// Returns an error if any of the permission isn't registered.
func GetPermissions(names ...string) ([]Permission, error) {
	mu.RLock()
	var err error
	for _, name := range names {
		if _, ok := perms[name]; !ok {
			err = errors.Reason("permission not registered: %q", name).Err()
			break
		}
	}
	forbiddenAlready := forbidChanges
	mu.RUnlock()

	if err != nil {
		return nil, err
	}

	if !forbiddenAlready {
		mu.Lock()
		forbidChanges = true
		mu.Unlock()
	}

	perms := make([]Permission, 0, len(names))
	for _, name := range names {
		perms = append(perms, Permission{name})
	}
	return perms, nil
}

// RegisteredPermissions returns a snapshot of all registered permissions along
// with their flags.
//
// Implicitly calls ForbidPermissionChanges to make sure no new permissions
// are added later.
func RegisteredPermissions() map[Permission]PermissionFlags {
	mu.RLock()
	all := make(map[Permission]PermissionFlags, len(perms))
	for name, flags := range perms {
		all[Permission{name: name}] = flags
	}
	forbiddenAlready := forbidChanges
	mu.RUnlock()

	if !forbiddenAlready {
		mu.Lock()
		forbidChanges = true
		mu.Unlock()
	}

	return all
}

// ForbidPermissionChanges explicitly forbids registering new permissions or
// changing their flags.
//
// All permissions should be registered before the server starts running its
// loop. The runtime relies on this when building various caches. After this
// function is called, RegisterPermissions and AddFlags would start panicking.
//
// Intended for internal server code.
func ForbidPermissionChanges() {
	mu.Lock()
	forbidChanges = true
	mu.Unlock()
}

// ValidatePermissionName returns an error if the permission name is invalid.
//
// It checks the name looks like "<service>.<subject>.<verb>".
func ValidatePermissionName(name string) error {
	if parts := strings.Split(name, "."); len(parts) == 3 {
		good := true
		for _, p := range parts {
			if p == "" {
				good = false
				break
			}
		}
		if good {
			return nil
		}
	}
	return errors.Reason("bad permission %q - must have form <service>.<subject>.<verb>", name).Err()
}