66 lines
2.2 KiB
Python
66 lines
2.2 KiB
Python
|
"""Onionr - Private P2P Communication.
|
||
|
|
|
||
|
|
Discover and publish private-network
|
||
|
|
"""
|
||
|
|
import socket
|
||
|
|
import struct
|
||
|
|
|
||
|
|
from .getip import lan_ips
|
||
|
|
from utils.bettersleep import better_sleep
|
||
|
|
"""
|
||
|
|
This program is free software: you can redistribute it and/or modify
|
||
|
|
it under the terms of the GNU General Public License as published by
|
||
|
|
the Free Software Foundation, either version 3 of the License, or
|
||
|
|
(at your option) any later version.
|
||
|
|
|
||
|
|
This program is distributed in the hope that it will be useful,
|
||
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
GNU General Public License for more details.
|
||
|
|
|
||
|
|
You should have received a copy of the GNU General Public License
|
||
|
|
along with this program. If not, see <https://www.gnu.org/licenses/>.
|
||
|
|
"""
|
||
|
|
MCAST_GRP = '224.13.3.7'
|
||
|
|
MCAST_PORT = 1337
|
||
|
|
IS_ALL_GROUPS = True
|
||
|
|
ANNOUNCE_LOOP_SLEEP = 30
|
||
|
|
|
||
|
|
|
||
|
|
def learn_services():
|
||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
|
||
|
|
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
|
||
|
|
if IS_ALL_GROUPS:
|
||
|
|
# on this port, receives ALL multicast groups
|
||
|
|
sock.bind(('', MCAST_PORT))
|
||
|
|
else:
|
||
|
|
# on this port, listen ONLY to MCAST_GRP
|
||
|
|
sock.bind((MCAST_GRP, MCAST_PORT))
|
||
|
|
mreq = struct.pack("4sl", socket.inet_aton(MCAST_GRP), socket.INADDR_ANY)
|
||
|
|
|
||
|
|
sock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq)
|
||
|
|
|
||
|
|
while True:
|
||
|
|
rec_data = sock.recv(1024)
|
||
|
|
print('reced_data', rec_data)
|
||
|
|
|
||
|
|
sock.shutdown()
|
||
|
|
sock.close()
|
||
|
|
|
||
|
|
|
||
|
|
def advertise_service():
|
||
|
|
# regarding socket.IP_MULTICAST_TTL
|
||
|
|
# ---------------------------------
|
||
|
|
# for all packets sent, after three hops on the network the packet will not
|
||
|
|
# be re-sent/broadcast (see https://www.tldp.org/HOWTO/Multicast-HOWTO-6.html)
|
||
|
|
MULTICAST_TTL = 3
|
||
|
|
ips = '-'.join(lan_ips)
|
||
|
|
|
||
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM, socket.IPPROTO_UDP)
|
||
|
|
sock.setsockopt(socket.IPPROTO_IP, socket.IP_MULTICAST_TTL, MULTICAST_TTL)
|
||
|
|
while True:
|
||
|
|
sock.sendto(f"onionr-{ips}".encode('utf-8'), (MCAST_GRP, MCAST_PORT))
|
||
|
|
better_sleep(ANNOUNCE_LOOP_SLEEP)
|
||
|
|
sock.shutdown()
|
||
|
|
sock.close()
|