forked from XTLS/Xray-core
-
Notifications
You must be signed in to change notification settings - Fork 0
/
callbacks.go
41 lines (35 loc) · 1.1 KB
/
callbacks.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
package xray_vless_inbound_callbacks
import (
xray_common_callbacks "github.com/4nd3r5on/Xray-core/common/callbacks"
"github.com/4nd3r5on/Xray-core/common/idsyncmap"
"github.com/4nd3r5on/Xray-core/common/session"
"github.com/4nd3r5on/Xray-core/features/policy"
)
type OnProcessStart func(sessionPolicy *policy.Session) error
type CallbackManager struct {
CbsOnProcess idsyncmap.IDSyncMap[xray_common_callbacks.OnProcess]
CbsOnProcessStart idsyncmap.IDSyncMap[OnProcessStart]
}
func (cm *CallbackManager) ExecOnProcess(inbound *session.Inbound) (id int32, err error) {
for id, callback := range cm.CbsOnProcess.Get() {
err = callback(inbound)
if err != nil {
return id, err
}
}
return id, nil
}
func (cm *CallbackManager) ExecOnProcessStart(sessionPolicy *policy.Session) (id int32, err error) {
for id, callback := range cm.CbsOnProcessStart.Get() {
err = callback(sessionPolicy)
if err != nil {
return id, err
}
}
return id, nil
}
func NewCallbackManager() *CallbackManager {
return &CallbackManager{
CbsOnProcess: idsyncmap.NewIDSyncMap[xray_common_callbacks.OnProcess](),
}
}