Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
R
re6stnet
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Labels
Merge Requests
0
Merge Requests
0
Analytics
Analytics
Repository
Value Stream
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Commits
Open sidebar
Kirill Smelkov
re6stnet
Commits
abacdb1a
Commit
abacdb1a
authored
Jul 06, 2012
by
Ulysse Beaugnon
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Bug solved :
The connections who had failed are now removed correctly from connection_dict
parent
004aa1f0
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
47 additions
and
27 deletions
+47
-27
main.py
main.py
+45
-25
openvpn.py
openvpn.py
+2
-2
No files found.
main.py
View file @
abacdb1a
...
@@ -16,6 +16,35 @@ def log_message(message, verbose_level):
...
@@ -16,6 +16,35 @@ def log_message(message, verbose_level):
# TODO : How do we get our vifib ip ?
# TODO : How do we get our vifib ip ?
class
PeersDB
:
def
__init__
(
self
,
dbPath
):
log_message
(
'Connectiong to peers database'
,
4
)
self
.
db
=
sqlite3
.
connect
(
dbPath
,
isolation_level
=
None
)
log_message
(
'Initializing peers database'
,
4
)
self
.
db
.
execute
(
"""CREATE TABLE IF NOT EXISTS peers
( id INTEGER PRIMARY KEY AUTOINCREMENT,
ip TEXT NOT NULL,
port INTEGER NOT NULL,
proto TEXT NOT NULL,
used INTEGER NOT NULL)"""
)
self
.
db
.
execute
(
"CREATE INDEX IF NOT EXISTS _peers_used ON peers(used)"
)
self
.
db
.
execute
(
"CREATE INDEX IF NOT EXISTS _peers_ip ON peers(ip)"
)
self
.
db
.
execute
(
"UPDATE peers SET used = 0"
)
def
getUnusedPeers
(
self
,
nPeers
):
return
self
.
db
.
execute
(
"SELECT id, ip, port, proto FROM peers WHERE used = 0 "
"ORDER BY RANDOM() LIMIT ?"
,
(
nPeers
,))
def
usePeer
(
self
,
id
):
log_message
(
'Updating peers database : using peer '
+
str
(
id
),
5
)
self
.
db
.
execute
(
"UPDATE peers SET used = 1 WHERE id = ?"
,
(
id
,))
def
unusePeer
(
self
,
id
):
log_message
(
'Updating peers database : unusing peer '
+
str
(
id
),
5
)
self
.
db
.
execute
(
"UPDATE peers SET used = 0 WHERE id = ?"
,
(
id
,))
def
babel
(
network_ip
,
network_mask
,
verbose_level
):
def
babel
(
network_ip
,
network_mask
,
verbose_level
):
args
=
[
'babeld'
,
args
=
[
'babeld'
,
'-C'
,
'redistribute local ip %s/%s'
%
(
network_ip
,
network_mask
),
'-C'
,
'redistribute local ip %s/%s'
%
(
network_ip
,
network_mask
),
...
@@ -74,14 +103,12 @@ def getConfig():
...
@@ -74,14 +103,12 @@ def getConfig():
def
startNewConnection
(
n
):
def
startNewConnection
(
n
):
try
:
try
:
for
id
,
ip
,
port
,
proto
in
peer_db
.
execute
(
for
id
,
ip
,
port
,
proto
in
peers_db
.
getUnusedPeers
(
n
):
"SELECT id, ip, port, proto FROM peers WHERE used = 0 ORDER BY RANDOM() LIMIT ?"
,
(
n
,)):
log_message
(
'Establishing a connection with id %s (%s:%s)'
%
(
id
,
ip
,
port
),
2
)
log_message
(
'Establishing a connection with id %s (%s:%s)'
%
(
id
,
ip
,
port
),
2
)
iface
=
free_interface_set
.
pop
()
iface
=
free_interface_set
.
pop
()
connection_dict
[
id
]
=
(
openvpn
.
client
(
ip
,
'--dev'
,
iface
,
'--proto'
,
proto
,
'--rport'
,
str
(
port
),
connection_dict
[
id
]
=
(
openvpn
.
client
(
ip
,
'--dev'
,
iface
,
'--proto'
,
proto
,
'--rport'
,
str
(
port
),
stdout
=
os
.
open
(
config
.
client_log
+
'vifibnet.client.'
+
str
(
id
)
+
'.log'
,
os
.
O_WRONLY
|
os
.
O_CREAT
|
os
.
O_TRUNC
)
)
,
iface
)
stdout
=
os
.
open
(
config
.
client_log
+
'vifibnet.client.'
+
str
(
id
)
+
'.log'
,
os
.
O_WRONLY
|
os
.
O_CREAT
|
os
.
O_TRUNC
)
)
,
iface
)
log_message
(
'Updating peers database'
,
5
)
peers_db
.
usePeer
(
id
)
peer_db
.
execute
(
"UPDATE peers SET used = 1 WHERE id = ?"
,
(
id
,))
except
KeyError
:
except
KeyError
:
log_message
(
"Can't establish connection with %s : no available interface"
%
ip
,
2
)
log_message
(
"Can't establish connection with %s : no available interface"
%
ip
,
2
)
pass
pass
...
@@ -94,8 +121,7 @@ def killConnection(id):
...
@@ -94,8 +121,7 @@ def killConnection(id):
p
,
iface
=
connection_dict
.
pop
(
id
)
p
,
iface
=
connection_dict
.
pop
(
id
)
p
.
kill
()
p
.
kill
()
free_interface_set
.
add
(
iface
)
free_interface_set
.
add
(
iface
)
log_message
(
'Updating peers database'
,
5
)
peers_db
.
unusedPeer
(
id
)
peer_db
.
execute
(
"UPDATE peers SET used = 0 WHERE id = ?"
,
(
id
,))
except
KeyError
:
except
KeyError
:
log_message
(
"Can't kill connection to "
+
peer
+
": no existing connection"
,
1
)
log_message
(
"Can't kill connection to "
+
peer
+
": no existing connection"
,
1
)
pass
pass
...
@@ -104,13 +130,14 @@ def killConnection(id):
...
@@ -104,13 +130,14 @@ def killConnection(id):
pass
pass
def
checkConnections
():
def
checkConnections
():
for
id
in
connection_dict
:
toDel
=
set
([])
for
id
in
connection_dict
.
keys
():
p
,
iface
=
connection_dict
[
id
]
p
,
iface
=
connection_dict
[
id
]
if
p
.
poll
()
!=
None
:
if
p
.
poll
()
!=
None
:
log_message
(
'Connection with '
+
str
(
id
)
+
' has failed
'
,
3
)
log_message
(
'Connection with '
+
str
(
id
)
+
' has failed
with return code '
+
str
(
p
.
returncode
)
,
3
)
free_interface_set
.
add
(
iface
)
free_interface_set
.
add
(
iface
)
log_message
(
'Updating peers database'
,
5
)
peers_db
.
unusePeer
(
id
)
peer_db
.
execute
(
"UPDATE peers SET used = 0 WHERE id = ?"
,
(
id
,))
del
connection_dict
[
id
]
def
refreshConnections
():
def
refreshConnections
():
checkConnections
()
checkConnections
()
...
@@ -128,10 +155,11 @@ def handle_message(msg):
...
@@ -128,10 +155,11 @@ def handle_message(msg):
words
=
msg
.
split
()
words
=
msg
.
split
()
if
words
[
0
]
==
'CLIENT_CONNECTED'
:
if
words
[
0
]
==
'CLIENT_CONNECTED'
:
log_message
(
'Incomming connection from '
+
words
[
1
],
3
)
log_message
(
'Incomming connection from '
+
words
[
1
],
3
)
# TODO : check if we are not already connected to it
elif
words
[
0
]
==
'CLIENT_DISCONNECTED'
:
elif
words
[
0
]
==
'CLIENT_DISCONNECTED'
:
log_message
(
words
[
1
]
+
' has disconnected'
,
3
)
log_message
(
words
[
1
]
+
' has disconnected'
,
3
)
else
:
else
:
log_message
(
'Unknow message recieved : '
+
msg
,
1
)
log_message
(
'Unknow message recieved
from the openvpn pipe
: '
+
msg
,
1
)
def
main
():
def
main
():
# Get arguments
# Get arguments
...
@@ -139,18 +167,8 @@ def main():
...
@@ -139,18 +167,8 @@ def main():
(
externalIp
,
externalPort
)
=
upnpigd
.
GetExternalInfo
(
1194
)
(
externalIp
,
externalPort
)
=
upnpigd
.
GetExternalInfo
(
1194
)
# Setup database
# Setup database
global
peer_db
# stop using global variables for everything ?
global
peers_db
# stop using global variables for everything ?
log_message
(
'Connectiong to peers database'
,
4
)
peers_db
=
PeersDB
(
config
.
db
)
peer_db
=
sqlite3
.
connect
(
config
.
db
,
isolation_level
=
None
)
log_message
(
'Initializing peers database'
,
4
)
peer_db
.
execute
(
"""CREATE TABLE IF NOT EXISTS peers
( id INTEGER PRIMARY KEY AUTOINCREMENT,
ip TEXT NOT NULL,
port INTEGER NOT NULL,
proto TEXT NOT NULL,
used INTEGER NOT NULL)"""
)
peer_db
.
execute
(
"CREATE INDEX IF NOT EXISTS _peers_used ON peers(used)"
)
peer_db
.
execute
(
"UPDATE peers SET used = 0"
)
# Create and open read_only pipe to get connect/disconnect events from openvpn
# Create and open read_only pipe to get connect/disconnect events from openvpn
log_message
(
'Creating pipe for openvpn events'
,
3
)
log_message
(
'Creating pipe for openvpn events'
,
3
)
...
@@ -160,14 +178,16 @@ def main():
...
@@ -160,14 +178,16 @@ def main():
# Establish connections
# Establish connections
log_message
(
'Starting openvpn server'
,
3
)
log_message
(
'Starting openvpn server'
,
3
)
serverProcess
=
openvpn
.
server
(
config
.
ip
,
write_pipe
,
serverProcess
=
openvpn
.
server
(
config
.
ip
,
write_pipe
,
'--dev'
,
'vifibnet'
,
stdout
=
os
.
open
(
config
.
server_log
,
os
.
O_WRONLY
|
os
.
O_CREAT
|
os
.
O_TRUNC
))
'--dev'
,
'vifibnet'
,
stdout
=
os
.
open
(
config
.
server_log
,
os
.
O_WRONLY
|
os
.
O_CREAT
|
os
.
O_TRUNC
))
startNewConnection
(
config
.
client_count
)
startNewConnection
(
config
.
client_count
)
next_refresh
=
time
.
time
()
+
config
.
refresh_time
next_refresh
=
time
.
time
()
+
config
.
refresh_time
# main loop
# main loop
try
:
try
:
while
True
:
while
True
:
ready
,
tmp1
,
tmp2
=
select
.
select
([
read_pipe
],
[],
[],
max
(
0
,
next_refresh
-
time
.
time
()))
ready
,
tmp1
,
tmp2
=
select
.
select
([
read_pipe
],
[],
[],
max
(
0
,
next_refresh
-
time
.
time
()))
if
ready
:
if
ready
:
handle_message
(
read_pipe
.
readline
())
handle_message
(
read_pipe
.
readline
())
if
time
.
time
()
>=
next_refresh
:
if
time
.
time
()
>=
next_refresh
:
...
...
openvpn.py
View file @
abacdb1a
...
@@ -8,12 +8,12 @@ def openvpn(*args, **kw):
...
@@ -8,12 +8,12 @@ def openvpn(*args, **kw):
'--persist-key'
,
'--persist-key'
,
'--script-security'
,
'2'
,
'--script-security'
,
'2'
,
'--user'
,
'nobody'
,
'--user'
,
'nobody'
,
'--ping'
,
'
0.5
'
,
'--ping'
,
'
1
'
,
# I don't kown how Babel works, but if it test the
# I don't kown how Babel works, but if it test the
# connection often, the ping directive might not be needed
# connection often, the ping directive might not be needed
# if it test the connection very often, we could also decrease
# if it test the connection very often, we could also decrease
# ping-exit to 1 sec
# ping-exit to 1 sec
'--ping-exit'
,
'
2
'
,
'--ping-exit'
,
'
3
'
,
'--group'
,
'nogroup'
,
'--group'
,
'nogroup'
,
'--verb'
,
str
(
config
.
verbose
),
'--verb'
,
str
(
config
.
verbose
),
]
+
list
(
args
)
+
config
.
openvpn_args
]
+
list
(
args
)
+
config
.
openvpn_args
...
...
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