archives/snowflake
2
0
Fork 0
mirror of https://gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake.git synced 2025-10-13 20:11:19 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat(proxy): add sd_notify support

Browse source
This commit is contained in:
KokaKiwi 2025-06-26 02:03:04 +02:00
parent 70974640ab
commit fa5be44b69
No known key found for this signature in database
GPG key ID: FD333F84686EFE78
4 changed files with 43 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
proxy/lib/snowflake.go
View file

@ -822,6 +822,8 @@ func (sf *SnowflakeProxy) Start() error {
}
tokens = newTokens(sf.Capacity)
sdnotifyStatus("Checking NAT type...")
err = sf.checkNATType(config, sf.NATProbeURL)
if err != nil {
// non-fatal error. Log it and continue
@ -830,6 +832,9 @@ func (sf *SnowflakeProxy) Start() error {
}
sf.EventDispatcher.OnNewSnowflakeEvent(event.EventOnCurrentNATTypeDetermined{CurNATType: getCurrentNATType()})
sdnotifyStatus(fmt.Sprintf("NAT type: %s", getCurrentNATType()))
sdnotifyReady()
NatRetestTask := task.Periodic{
Interval: sf.NATTypeMeasurementInterval,
Execute: func() error {
@ -846,6 +851,16 @@ func (sf *SnowflakeProxy) Start() error {
defer NatRetestTask.Close()
}
WatchdogTask := task.Periodic{
Interval: 5 * time.Minute,
Execute: func() error {
sdnotifyWatchdog()
return nil
},
}
WatchdogTask.Start()
defer WatchdogTask.Close()
ticker := time.NewTicker(sf.PollInterval)
defer ticker.Stop()
@ -864,6 +879,7 @@ func (sf *SnowflakeProxy) Start() error {
// Stop closes all existing connections and shuts down the Snowflake.
func (sf *SnowflakeProxy) Stop() {
sdnotifyStopping()
close(sf.shutdown)
}