Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in / Register
Toggle navigation
Menu
Open sidebar
adam.huang
go-libp2p
Commits
90eeff42
Unverified
Commit
90eeff42
authored
Sep 12, 2016
by
Jakub Sztandera
Browse files
nat: add locks around nat
parent
b6bfae67
Changes
2
Hide whitespace changes
Inline
Side-by-side
p2p/nat/mapping.go
View file @
90eeff42
...
...
@@ -106,7 +106,9 @@ func (m *mapping) ExternalAddr() (ma.Multiaddr, error) {
return
nil
,
ErrNoMapping
}
m
.
nat
.
natmu
.
Lock
()
ip
,
err
:=
m
.
nat
.
nat
.
GetExternalAddress
()
m
.
nat
.
natmu
.
Unlock
()
if
err
!=
nil
{
return
nil
,
err
}
...
...
p2p/nat/nat.go
View file @
90eeff42
...
...
@@ -52,8 +52,9 @@ func DiscoverNAT() *NAT {
// service that will periodically renew port mappings,
// and keep an up-to-date list of all the external addresses.
type
NAT
struct
{
nat
nat
.
NAT
proc
goprocess
.
Process
// manages nat mappings lifecycle
natmu
sync
.
Mutex
nat
nat
.
NAT
proc
goprocess
.
Process
// manages nat mappings lifecycle
mappingmu
sync
.
RWMutex
// guards mappings
mappings
map
[
*
mapping
]
struct
{}
...
...
@@ -170,11 +171,13 @@ func (nat *NAT) establishMapping(m *mapping) {
comment
=
"libp2p-"
+
m
.
comment
}
nat
.
natmu
.
Lock
()
newport
,
err
:=
nat
.
nat
.
AddPortMapping
(
m
.
Protocol
(),
m
.
InternalPort
(),
comment
,
MappingDuration
)
if
err
!=
nil
{
// Some hardware does not support mappings with timeout, so try that
newport
,
err
=
nat
.
nat
.
AddPortMapping
(
m
.
Protocol
(),
m
.
InternalPort
(),
comment
,
0
)
}
nat
.
natmu
.
Lock
()
failure
:=
func
()
{
m
.
setExternalPort
(
0
)
// clear mapping
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment