mirror of
https://github.com/v2fly/v2ray-core.git
synced 2024-12-22 18:17:52 -05:00
210 lines
6.1 KiB
Go
Executable File
210 lines
6.1 KiB
Go
Executable File
package core
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"v2ray.com/core/common"
|
|
"v2ray.com/core/common/uuid"
|
|
)
|
|
|
|
// Server is an instance of V2Ray. At any time, there must be at most one Server instance running.
|
|
// Deprecated. Use Instance directly.
|
|
type Server interface {
|
|
common.Runnable
|
|
}
|
|
|
|
// Feature is the interface for V2Ray features. All features must implement this interface.
|
|
// All existing features have an implementation in app directory. These features can be replaced by third-party ones.
|
|
type Feature interface {
|
|
common.Runnable
|
|
}
|
|
|
|
// Instance combines all functionalities in V2Ray.
|
|
type Instance struct {
|
|
dnsClient syncDNSClient
|
|
policyManager syncPolicyManager
|
|
dispatcher syncDispatcher
|
|
router syncRouter
|
|
ihm syncInboundHandlerManager
|
|
ohm syncOutboundHandlerManager
|
|
|
|
access sync.Mutex
|
|
features []Feature
|
|
id uuid.UUID
|
|
running bool
|
|
}
|
|
|
|
// New returns a new V2Ray instance based on given configuration.
|
|
// The instance is not started at this point.
|
|
// To make sure V2Ray instance works properly, the config must contain one Dispatcher, one InboundHandlerManager and one OutboundHandlerManager. Other features are optional.
|
|
func New(config *Config) (*Instance, error) {
|
|
var server = &Instance{
|
|
id: uuid.New(),
|
|
}
|
|
|
|
if err := config.Transport.Apply(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
for _, appSettings := range config.App {
|
|
settings, err := appSettings.GetInstance()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if _, err := server.CreateObject(settings); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
for _, inbound := range config.Inbound {
|
|
rawHandler, err := server.CreateObject(inbound)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
handler, ok := rawHandler.(InboundHandler)
|
|
if !ok {
|
|
return nil, newError("not an InboundHandler")
|
|
}
|
|
if err := server.InboundHandlerManager().AddHandler(context.Background(), handler); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
for _, outbound := range config.Outbound {
|
|
rawHandler, err := server.CreateObject(outbound)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
handler, ok := rawHandler.(OutboundHandler)
|
|
if !ok {
|
|
return nil, newError("not an OutboundHandler")
|
|
}
|
|
if err := server.OutboundHandlerManager().AddHandler(context.Background(), handler); err != nil {
|
|
return nil, err
|
|
}
|
|
}
|
|
|
|
return server, nil
|
|
}
|
|
|
|
func (s *Instance) CreateObject(config interface{}) (interface{}, error) {
|
|
ctx := context.WithValue(context.Background(), v2rayKey, s)
|
|
return common.CreateObject(ctx, config)
|
|
}
|
|
|
|
// ID returns an unique ID for this V2Ray instance.
|
|
func (s *Instance) ID() uuid.UUID {
|
|
return s.id
|
|
}
|
|
|
|
// Close shutdown the V2Ray instance.
|
|
func (s *Instance) Close() error {
|
|
s.access.Lock()
|
|
defer s.access.Unlock()
|
|
|
|
s.running = false
|
|
for _, f := range s.allFeatures() {
|
|
f.Close()
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Start starts the V2Ray instance, including all registered features. When Start returns error, the state of the instance is unknown.
|
|
// A V2Ray instance can be started only once. Upon closing, the instance is not guaranteed to start again.
|
|
func (s *Instance) Start() error {
|
|
s.access.Lock()
|
|
defer s.access.Unlock()
|
|
|
|
s.running = true
|
|
for _, f := range s.allFeatures() {
|
|
if err := f.Start(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
newError("V2Ray ", Version(), " started").AtWarning().WriteToLog()
|
|
|
|
return nil
|
|
}
|
|
|
|
// RegisterFeature registers the given feature into V2Ray.
|
|
// If feature is one of the following types, the corresponding feature in this Instance
|
|
// will be replaced: DNSClient, PolicyManager, Router, Dispatcher, InboundHandlerManager, OutboundHandlerManager.
|
|
func (s *Instance) RegisterFeature(feature interface{}, instance Feature) error {
|
|
running := false
|
|
|
|
switch feature.(type) {
|
|
case DNSClient, *DNSClient:
|
|
s.dnsClient.Set(instance.(DNSClient))
|
|
case PolicyManager, *PolicyManager:
|
|
s.policyManager.Set(instance.(PolicyManager))
|
|
case Router, *Router:
|
|
s.router.Set(instance.(Router))
|
|
case Dispatcher, *Dispatcher:
|
|
s.dispatcher.Set(instance.(Dispatcher))
|
|
case InboundHandlerManager, *InboundHandlerManager:
|
|
s.ihm.Set(instance.(InboundHandlerManager))
|
|
case OutboundHandlerManager, *OutboundHandlerManager:
|
|
s.ohm.Set(instance.(OutboundHandlerManager))
|
|
default:
|
|
s.access.Lock()
|
|
s.features = append(s.features, instance)
|
|
running = s.running
|
|
s.access.Unlock()
|
|
}
|
|
|
|
if running {
|
|
return instance.Start()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *Instance) allFeatures() []Feature {
|
|
return append([]Feature{s.DNSClient(), s.PolicyManager(), s.Dispatcher(), s.Router(), s.InboundHandlerManager(), s.OutboundHandlerManager()}, s.features...)
|
|
}
|
|
|
|
// GetFeature returns a feature that was registered in this Instance. Nil if not found.
|
|
// The returned Feature must implement common.HasType and whose type equals the given feature type.
|
|
func (s *Instance) GetFeature(featureType interface{}) Feature {
|
|
for _, f := range s.features {
|
|
if hasType, ok := f.(common.HasType); ok {
|
|
if hasType.Type() == featureType {
|
|
return f
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// DNSClient returns the DNSClient used by this Instance. The returned DNSClient is always functional.
|
|
func (s *Instance) DNSClient() DNSClient {
|
|
return &(s.dnsClient)
|
|
}
|
|
|
|
// PolicyManager returns the PolicyManager used by this Instance. The returned PolicyManager is always functional.
|
|
func (s *Instance) PolicyManager() PolicyManager {
|
|
return &(s.policyManager)
|
|
}
|
|
|
|
// Router returns the Router used by this Instance. The returned Router is always functional.
|
|
func (s *Instance) Router() Router {
|
|
return &(s.router)
|
|
}
|
|
|
|
// Dispatcher returns the Dispatcher used by this Instance. If Dispatcher was not registered before, the returned value doesn't work, although it is not nil.
|
|
func (s *Instance) Dispatcher() Dispatcher {
|
|
return &(s.dispatcher)
|
|
}
|
|
|
|
// InboundHandlerManager returns the InboundHandlerManager used by this Instance. If InboundHandlerManager was not registered before, the returned value doesn't work.
|
|
func (s *Instance) InboundHandlerManager() InboundHandlerManager {
|
|
return &(s.ihm)
|
|
}
|
|
|
|
// OutboundHandlerManager returns the OutboundHandlerManager used by this Instance. If OutboundHandlerManager was not registered before, the returned value doesn't work.
|
|
func (s *Instance) OutboundHandlerManager() OutboundHandlerManager {
|
|
return &(s.ohm)
|
|
}
|