services.go

/*
   Velociraptor - Hunting Evil
   Copyright (C) 2019 Velocidex Innovations.

   This program is free software: you can redistribute it and/or modify
   it under the terms of the GNU Affero General Public License as published
   by the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   This program is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU Affero General Public License for more details.

   You should have received a copy of the GNU Affero General Public License
   along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/
package services

import (
	config_proto "www.velocidex.com/golang/velociraptor/config/proto"
	"www.velocidex.com/golang/velociraptor/notifications"
	"www.velocidex.com/golang/velociraptor/users"
)

// A manager responsible for starting and shutting down all the
// services in an orderly fashion.
type ServicesManager struct {
	hunt_manager      *HuntManager
	hunt_dispatcher   *HuntDispatcher
	user_manager      *users.UserNotificationManager
	stats_collector   *StatsCollector
	server_monitoring *EventTable
	server_artifacts  *ServerArtifactsRunner
	dyn_dns           *DynDNSService
	interrogation     *InterrogationService
	sanity_checker    *SanityChecks
	vfs_service       *VFSService
}

func (self *ServicesManager) Close() {
	self.hunt_manager.Close()
	self.hunt_dispatcher.Close()
	self.user_manager.Close()
	self.stats_collector.Close()
	self.server_monitoring.Close()
	self.server_artifacts.Close()
	self.dyn_dns.Close()
	self.interrogation.Close()
	self.sanity_checker.Close()
	self.vfs_service.Close()
}

// Start all the server services.
func StartServices(
	config_obj *config_proto.Config,
	notifier *notifications.NotificationPool) (*ServicesManager, error) {
	result := &ServicesManager{}

	hunt_manager, err := startHuntManager(config_obj)
	if err != nil {
		return nil, err
	}
	result.hunt_manager = hunt_manager

	hunt_dispatcher, err := StartHuntDispatcher(config_obj)
	if err != nil {
		return nil, err
	}
	result.hunt_dispatcher = hunt_dispatcher

	user_manager, err := users.StartUserNotificationManager(config_obj)
	if err != nil {
		return nil, err
	}
	result.user_manager = user_manager

	stats_collector, err := startStatsCollector(config_obj)
	if err != nil {
		return nil, err
	}
	result.stats_collector = stats_collector

	server_monitoring, err := startServerMonitoringService(config_obj)
	if err != nil {
		return nil, err
	}
	result.server_monitoring = server_monitoring

	server_artifacts, err := startServerArtifactService(config_obj, notifier)
	if err != nil {
		return nil, err
	}
	result.server_artifacts = server_artifacts

	err = StartClientMonitoringService(config_obj)
	if err != nil {
		return nil, err
	}

	dyndns, err := startDynDNSService(config_obj)
	if err != nil {
		return nil, err
	}
	result.dyn_dns = dyndns

	interrogation := startInterrogationService(config_obj)
	result.interrogation = interrogation

	sanity_checker, err := startSanityCheckService(config_obj)
	if err != nil {
		return nil, err
	}
	result.sanity_checker = sanity_checker

	vfs_service, err := startVFSService(config_obj)
	if err != nil {
		return nil, err
	}
	result.vfs_service = vfs_service

	return result, nil
}