policy.go

package policy

import (
	"fmt"
	"strconv"
	"sync"
	"time"

	"go.aporeto.io/tg/tglib"
	"go.aporeto.io/trireme-lib/controller/pkg/usertokens"
	"go.uber.org/zap"
)

// PUPolicy captures all policy information related ot the container
type PUPolicy struct {

	// ManagementID is provided for the policy implementations as a means of
	// holding a policy identifier related to the implementation.
	managementID string
	// triremeAction defines what level of policy should be applied to that container.
	triremeAction PUAction
	// dnsACLs is the list of DNS names and the associated ports that the container is
	// allowed to talk to outside the data center
	DNSACLs DNSRuleList
	// applicationACLs is the list of ACLs to be applied when the container talks
	// to IP Addresses outside the data center
	applicationACLs IPRuleList
	// networkACLs is the list of ACLs to be applied from IP Addresses outside
	// the data center
	networkACLs IPRuleList
	// identity is the set of key value pairs that must be send over the wire.
	identity *TagStore
	// annotations are key/value pairs  that should be used for accounting reasons
	annotations *TagStore
	// transmitterRules is the set of rules that implement the label matching at the Transmitter
	transmitterRules TagSelectorList
	// teceiverRules is the set of rules that implement matching at the Receiver
	receiverRules TagSelectorList
	// ips is the set of IP addresses and namespaces that the policy must be applied to
	ips ExtendedMap
	// triremeNetworks is the list of networks that Authorization must be enforced
	triremeNetworks []string
	// triremeUDPNetworks is the list of UDP networks that this policy should apply to
	triremeUDPNetworks []string
	// excludedNetworks a list of networks that must be excluded
	excludedNetworks []string
	// servicesListeningPort is the port that we will use for the proxy.
	servicesListeningPort int
	// exposedServices is the list of services that this PU is exposing.
	exposedServices ApplicationServicesList
	// dependentServices is the list of services that this PU depends on.
	dependentServices ApplicationServicesList
	// servicesCertificate is the services certificate
	servicesCertificate string
	// serviceCertificateExpiration is the time that the certificate expires.
	serviceCertificateExpiration time.Time
	// servicePrivateKey is the service private key
	servicesPrivateKey string
	// servicesCA is the CA to be used for the outgoing services
	servicesCA string
	// scopes are the processing unit granted scopes
	scopes []string

	sync.Mutex
}

// PUAction defines the action types that applies for a specific PU as a whole.
type PUAction int

const (
	// AllowAll allows everything for the specific PU.
	AllowAll = 0x1
	// Police filters on the PU based on the PolicyRules.
	Police = 0x2
)

// NewPUPolicy generates a new ContainerPolicyInfo
// appACLs are the ACLs for packet coming from the Application/PU to the Network.
// netACLs are the ACLs for packet coming from the Network to the Application/PU.
func NewPUPolicy(
	id string,
	action PUAction,
	appACLs IPRuleList,
	netACLs IPRuleList,
	dnsACLs DNSRuleList,
	txtags TagSelectorList,
	rxtags TagSelectorList,
	identity *TagStore,
	annotations *TagStore,
	ips ExtendedMap,
	triremeNetworks []string,
	triremeUDPNetworks []string,
	excludedNetworks []string,
	servicesListeningPort int,
	exposedServices ApplicationServicesList,
	dependentServices ApplicationServicesList,
	scopes []string,
) *PUPolicy {

	if appACLs == nil {
		appACLs = IPRuleList{}
	}
	if netACLs == nil {
		netACLs = IPRuleList{}
	}
	if dnsACLs == nil {
		dnsACLs = DNSRuleList{}
	}
	if txtags == nil {
		txtags = TagSelectorList{}
	}
	if rxtags == nil {
		rxtags = TagSelectorList{}
	}

	if identity == nil {
		identity = NewTagStore()
	}

	if annotations == nil {
		annotations = NewTagStore()
	}

	if ips == nil {
		ips = ExtendedMap{}
	}

	if exposedServices == nil {
		exposedServices = ApplicationServicesList{}
	}

	if dependentServices == nil {
		dependentServices = ApplicationServicesList{}
	}

	return &PUPolicy{
		managementID:          id,
		triremeAction:         action,
		applicationACLs:       appACLs,
		networkACLs:           netACLs,
		DNSACLs:               dnsACLs,
		transmitterRules:      txtags,
		receiverRules:         rxtags,
		identity:              identity,
		annotations:           annotations,
		ips:                   ips,
		triremeNetworks:       triremeNetworks,
		triremeUDPNetworks:    triremeUDPNetworks,
		excludedNetworks:      excludedNetworks,
		servicesListeningPort: servicesListeningPort,
		exposedServices:       exposedServices,
		dependentServices:     dependentServices,
		scopes:                scopes,
	}
}

// NewPUPolicyWithDefaults sets up a PU policy with defaults
func NewPUPolicyWithDefaults() *PUPolicy {
	return NewPUPolicy("", AllowAll, nil, nil, nil, nil, nil, nil, nil, nil, []string{}, []string{}, []string{}, 0, nil, nil, []string{})
}

// Clone returns a copy of the policy
func (p *PUPolicy) Clone() *PUPolicy {
	p.Lock()
	defer p.Unlock()

	np := NewPUPolicy(
		p.managementID,
		p.triremeAction,
		p.applicationACLs.Copy(),
		p.networkACLs.Copy(),
		p.DNSACLs.Copy(),
		p.transmitterRules.Copy(),
		p.receiverRules.Copy(),
		p.identity.Copy(),
		p.annotations.Copy(),
		p.ips.Copy(),
		p.triremeNetworks,
		p.triremeUDPNetworks,
		p.excludedNetworks,
		p.servicesListeningPort,
		p.exposedServices,
		p.dependentServices,
		p.scopes,
	)

	return np
}

// ManagementID returns the management ID
func (p *PUPolicy) ManagementID() string {
	p.Lock()
	defer p.Unlock()

	return p.managementID
}

// UDPNetworks returns the UDP networks
func (p *PUPolicy) UDPNetworks() []string {
	p.Lock()
	defer p.Unlock()

	return p.triremeUDPNetworks
}

// TriremeAction returns the TriremeAction
func (p *PUPolicy) TriremeAction() PUAction {
	p.Lock()
	defer p.Unlock()

	return p.triremeAction
}

// SetTriremeAction returns the TriremeAction
func (p *PUPolicy) SetTriremeAction(action PUAction) {
	p.Lock()
	defer p.Unlock()

	p.triremeAction = action
}

// ApplicationACLs returns a copy of IPRuleList
func (p *PUPolicy) ApplicationACLs() IPRuleList {
	p.Lock()
	defer p.Unlock()

	return p.applicationACLs.Copy()
}

// NetworkACLs returns a copy of IPRuleList
func (p *PUPolicy) NetworkACLs() IPRuleList {
	p.Lock()
	defer p.Unlock()

	return p.networkACLs.Copy()
}

// DNSNameACLs returns a copy of DNSRuleList
func (p *PUPolicy) DNSNameACLs() DNSRuleList {
	p.Lock()
	defer p.Unlock()

	return p.DNSACLs.Copy()
}

// ReceiverRules returns a copy of TagSelectorList
func (p *PUPolicy) ReceiverRules() TagSelectorList {
	p.Lock()
	defer p.Unlock()

	return p.receiverRules.Copy()
}

// AddReceiverRules adds a receiver rule
func (p *PUPolicy) AddReceiverRules(t TagSelector) {
	p.Lock()
	defer p.Unlock()

	p.receiverRules = append(p.receiverRules, t)
}

// TransmitterRules returns a copy of TagSelectorList
func (p *PUPolicy) TransmitterRules() TagSelectorList {
	p.Lock()
	defer p.Unlock()

	return p.transmitterRules.Copy()
}

// AddTransmitterRules adds a transmitter rule
func (p *PUPolicy) AddTransmitterRules(t TagSelector) {
	p.Lock()
	defer p.Unlock()

	p.transmitterRules = append(p.transmitterRules, t)
}

// Identity returns a copy of the Identity
func (p *PUPolicy) Identity() *TagStore {
	p.Lock()
	defer p.Unlock()

	return p.identity.Copy()
}

// Annotations returns a copy of the annotations
func (p *PUPolicy) Annotations() *TagStore {
	p.Lock()
	defer p.Unlock()

	return p.annotations.Copy()
}

// AddIdentityTag adds a policy tag
func (p *PUPolicy) AddIdentityTag(k, v string) {
	p.Lock()
	defer p.Unlock()

	p.identity.AppendKeyValue(k, v)
}

// IPAddresses returns all the IP addresses for the processing unit
func (p *PUPolicy) IPAddresses() ExtendedMap {
	p.Lock()
	defer p.Unlock()

	return p.ips.Copy()
}

// SetIPAddresses sets the IP addresses for the processing unit
func (p *PUPolicy) SetIPAddresses(l ExtendedMap) {
	p.Lock()
	defer p.Unlock()

	p.ips = l
}

// TriremeNetworks  returns the list of networks that Trireme must be applied
func (p *PUPolicy) TriremeNetworks() []string {
	p.Lock()
	defer p.Unlock()

	return p.triremeNetworks
}

// ExposedServices returns the exposed services
func (p *PUPolicy) ExposedServices() ApplicationServicesList {
	p.Lock()
	defer p.Unlock()

	return p.exposedServices
}

// DependentServices returns the external services.
func (p *PUPolicy) DependentServices() ApplicationServicesList {
	p.Lock()
	defer p.Unlock()

	return p.dependentServices
}

// ServicesListeningPort returns the port that should be used by the proxies.
func (p *PUPolicy) ServicesListeningPort() string {
	p.Lock()
	defer p.Unlock()

	return strconv.Itoa(p.servicesListeningPort)
}

// UpdateTriremeNetworks updates the set of networks for trireme
func (p *PUPolicy) UpdateTriremeNetworks(networks []string) {
	p.Lock()
	defer p.Unlock()

	p.triremeNetworks = make([]string, len(networks))

	copy(p.triremeNetworks, networks)

}

// UpdateDNSNetworks updates the set of FQDN names allowed by the policy
func (p *PUPolicy) UpdateDNSNetworks(networks DNSRuleList) {
	p.Lock()
	defer p.Unlock()

	p.DNSACLs = make(DNSRuleList, len(networks))

	copy(p.DNSACLs, networks)
}

// ExcludedNetworks returns the list of excluded networks.
func (p *PUPolicy) ExcludedNetworks() []string {
	p.Lock()
	defer p.Unlock()

	return p.excludedNetworks
}

// UpdateExcludedNetworks updates the list of excluded networks.
func (p *PUPolicy) UpdateExcludedNetworks(networks []string) {
	p.Lock()
	defer p.Unlock()

	p.excludedNetworks = make([]string, len(networks))

	copy(p.excludedNetworks, networks)
}

// UpdateServiceCertificates updates the certificate and private key of the policy
func (p *PUPolicy) UpdateServiceCertificates(cert, key string) {
	p.Lock()
	defer p.Unlock()

	if p.servicesCertificate == cert {
		return
	}

	p.servicesCertificate = cert
	p.servicesPrivateKey = key

	certificate, err := tglib.ParseCertificate([]byte(cert))
	if err != nil {
		zap.L().Error("Invalid certificate", zap.Error(err))
		return
	}

	if time.Now().After(certificate.NotAfter) {
		p.serviceCertificateExpiration = time.Now()
	} else {
		p.serviceCertificateExpiration = time.Now().Add(time.Until(certificate.NotAfter) / 2)
	}

}

// ServiceCertificates returns the service certificate.
func (p *PUPolicy) ServiceCertificates() (string, string, string) {
	p.Lock()
	defer p.Unlock()

	return p.servicesCertificate, p.servicesPrivateKey, p.servicesCA
}

// IsServiceCertificateExpired will return true of the service certificate is expired.
func (p *PUPolicy) IsServiceCertificateExpired() bool {
	p.Lock()
	defer p.Unlock()

	return time.Now().After(p.serviceCertificateExpiration)

}

// Scopes returns the scopes of the policy.
func (p *PUPolicy) Scopes() []string {
	p.Lock()
	defer p.Unlock()

	return p.scopes
}

// ToPublicPolicy converts the object to a marshallable object.
func (p *PUPolicy) ToPublicPolicy() *PUPolicyPublic {
	p.Lock()
	defer p.Unlock()

	return &PUPolicyPublic{
		ManagementID:          p.managementID,
		TriremeAction:         p.triremeAction,
		ApplicationACLs:       p.applicationACLs.Copy(),
		NetworkACLs:           p.networkACLs.Copy(),
		DNSACLs:               p.DNSACLs.Copy(),
		TransmitterRules:      p.transmitterRules.Copy(),
		ReceiverRules:         p.receiverRules.Copy(),
		Annotations:           p.annotations.Copy(),
		Identity:              p.identity.Copy(),
		IPs:                   p.ips.Copy(),
		TriremeNetworks:       p.triremeNetworks,
		TriremeUDPNetworks:    p.triremeUDPNetworks,
		ServicesListeningPort: p.servicesListeningPort,
		ExcludedNetworks:      p.excludedNetworks,
		ExposedServices:       p.exposedServices,
		DependentServices:     p.dependentServices,
		Scopes:                p.scopes,
		ServicesCA:            p.servicesCA,
		ServicesCertificate:   p.servicesCertificate,
		ServicesPrivateKey:    p.servicesPrivateKey,
	}
}

// PUPolicyPublic captures all policy information related ot the processing
// unit in an object that can be marshalled and transmitted over the RPC interface.
type PUPolicyPublic struct {
	ManagementID          string                  `json:"managementID,omitempty"`
	TriremeAction         PUAction                `json:"triremeAction,omitempty"`
	ApplicationACLs       IPRuleList              `json:"applicationACLs,omitempty"`
	NetworkACLs           IPRuleList              `json:"networkACLs,omitempty"`
	DNSACLs               DNSRuleList             `json:"dnsACLs,omitempty"`
	Identity              *TagStore               `json:"identity,omitempty"`
	Annotations           *TagStore               `json:"annotations,omitempty"`
	TransmitterRules      TagSelectorList         `json:"transmitterRules,omitempty"`
	ReceiverRules         TagSelectorList         `json:"receiverRules,omitempty"`
	IPs                   ExtendedMap             `json:"IPs,omitempty"`
	TriremeNetworks       []string                `json:"triremeNetworks,omitempty"`
	TriremeUDPNetworks    []string                `json:"triremeUDPNetworks,omitempty"`
	ExcludedNetworks      []string                `json:"excludedNetworks,omitempty"`
	ServicesListeningPort int                     `json:"servicesListeningPort,omitempty"`
	ExposedServices       ApplicationServicesList `json:"exposedServices,omitempty"`
	DependentServices     ApplicationServicesList `json:"dependentServices,omitempty"`
	ServicesCertificate   string                  `json:"servicesCertificate,omitempty"`
	ServicesPrivateKey    string                  `json:"servicesPrivateKey,omitempty"`
	ServicesCA            string                  `json:"servicesCA,omitempty"`
	Scopes                []string                `json:"scopes,omitempty"`
}

// ToPrivatePolicy converts the object to a private object.
func (p *PUPolicyPublic) ToPrivatePolicy(convert bool) (*PUPolicy, error) {
	var err error

	exposedServices := ApplicationServicesList{}
	for _, e := range p.ExposedServices {
		if convert {
			e.UserAuthorizationHandler, err = usertokens.NewVerifier(e.UserAuthorizationHandler)
			if err != nil {
				return nil, fmt.Errorf("unable to initialize user authorization handler for service: %s - error %s", e.ID, err)
			}
		}
		exposedServices = append(exposedServices, e)
	}

	return &PUPolicy{
		managementID:          p.ManagementID,
		triremeAction:         p.TriremeAction,
		applicationACLs:       p.ApplicationACLs,
		networkACLs:           p.NetworkACLs.Copy(),
		DNSACLs:               p.DNSACLs.Copy(),
		transmitterRules:      p.TransmitterRules.Copy(),
		receiverRules:         p.ReceiverRules.Copy(),
		annotations:           p.Annotations.Copy(),
		identity:              p.Identity.Copy(),
		ips:                   p.IPs.Copy(),
		triremeNetworks:       p.TriremeNetworks,
		triremeUDPNetworks:    p.TriremeUDPNetworks,
		servicesListeningPort: p.ServicesListeningPort,
		excludedNetworks:      p.ExcludedNetworks,
		exposedServices:       exposedServices,
		dependentServices:     p.DependentServices,
		scopes:                p.Scopes,
		servicesCA:            p.ServicesCA,
		servicesCertificate:   p.ServicesCertificate,
		servicesPrivateKey:    p.ServicesPrivateKey,
	}, nil
}