zishuo/go-upnp
1
0
Fork 0
mirror of https://gitlab.com/NebulousLabs/go-upnp.git synced 2025-10-05 23:16:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Rate-limit Add/Delete port mappings

Browse Source 
Fixes #10
This commit is contained in:
Serhat Şevki Dinçer
2017-12-21 00:23:09 +03:00
parent 1be4557baa
commit a625a89506
2 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
upnp.go
View File

@@ -63,17 +63,23 @@ func (d *IGD) Forward(port uint16, desc string) error {
return err return err
} }
time.Sleep(time.Millisecond)
err = d.client.AddPortMapping("", port, "TCP", port, ip, true, desc, 0) err = d.client.AddPortMapping("", port, "TCP", port, ip, true, desc, 0)
if err != nil { if err != nil {
return err return err
} }
time.Sleep(time.Millisecond)
return d.client.AddPortMapping("", port, "UDP", port, ip, true, desc, 0) return d.client.AddPortMapping("", port, "UDP", port, ip, true, desc, 0)
} }
// Clear un-forwards a port, removing it from the router's port mapping table. // Clear un-forwards a port, removing it from the router's port mapping table.
func (d *IGD) Clear(port uint16) error { func (d *IGD) Clear(port uint16) error {
time.Sleep(time.Millisecond)
tcpErr := d.client.DeletePortMapping("", port, "TCP") tcpErr := d.client.DeletePortMapping("", port, "TCP")
time.Sleep(time.Millisecond)
udpErr := d.client.DeletePortMapping("", port, "UDP") udpErr := d.client.DeletePortMapping("", port, "UDP")
// only return an error if both deletions failed // only return an error if both deletions failed
if tcpErr != nil && udpErr != nil { if tcpErr != nil && udpErr != nil {
return tcpErr return tcpErr

1
upnp_test.go
View File

@@ -71,6 +71,7 @@ func TestIGD(t *testing.T) {
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
t.Log("Loc:", loc)
// connect to router directly // connect to router directly
d, err = Load(loc) d, err = Load(loc)