parent
ef70a0f20b
commit
77287d9e62
@ -1,184 +0,0 @@
|
||||
#!/usr/bin/python
|
||||
# -*- coding: utf-8 -*-
|
||||
|
||||
# TODO
|
||||
# Reconnecting to IRC-Server
|
||||
# Complete switched networks
|
||||
# Reduce debug output / add good output
|
||||
# For more security use packet id + srcnick as key for fragmented packages
|
||||
|
||||
from conf import *
|
||||
import pytap
|
||||
from pytap import TapDevice
|
||||
import select
|
||||
import time
|
||||
import irclib
|
||||
import commands
|
||||
import re
|
||||
import base64
|
||||
import random
|
||||
|
||||
dev = None
|
||||
headerlen = 1 + 5 + 1 # flag + number + whitespace
|
||||
packets = {}
|
||||
|
||||
def getMacFromIf(iface):
|
||||
mac = commands.getoutput("/sbin/ifconfig %s|grep %s|egrep -o '([a-fA-F0-9]{2}(:|.)){6}'" % (iface, iface))
|
||||
print "Debug: %s has mac address %s" % (iface, mac)
|
||||
return mac
|
||||
|
||||
def getDstMacFromPkt(packet):
|
||||
if len(packet) < 10:
|
||||
return None
|
||||
return packet[4:10]
|
||||
|
||||
def binToHexStr(binmac):
|
||||
return "".join(["%02x" % ord(i) for i in binmac])
|
||||
|
||||
def isBroadcast(packet):
|
||||
binmac = getDstMacFromPkt(packet)
|
||||
# normal broadcast
|
||||
if binmac == '\xff\xff\xff\xff\xff\xff':
|
||||
return True
|
||||
# v6 multicast
|
||||
if binmac.startswith('\x33\x33'):
|
||||
return True
|
||||
return False
|
||||
|
||||
def sendToIRC(dev, server):
|
||||
msg = dev.read()
|
||||
oldencmsg = encmsg = base64.encodestring(msg).replace("\n", "")
|
||||
slices = []
|
||||
encmsglen = ircmsglen-headerlen
|
||||
msgid = "%05d" % random.randint(0, 99999)
|
||||
while len(encmsg) > encmsglen:
|
||||
slices.append(encmsg[:encmsglen])
|
||||
encmsg = encmsg[encmsglen:]
|
||||
slices.append(encmsg)
|
||||
|
||||
# HUB or SWITCH?
|
||||
if mode == "SWITCH" and not isBroadcast(package):
|
||||
target = "VPN-%s" % (binToHexStr(getDstMacFromPkt(packet)),)
|
||||
else:
|
||||
target = broadcastchan
|
||||
|
||||
if len(slices) == 0:
|
||||
print "DEBUG: EMPTY PACKAGE FROM DEV?"
|
||||
elif len(slices) == 1:
|
||||
print "Debug: Sending oneliner to dev"
|
||||
server.privmsg(target, "o%s %s" % (msgid, slices[0]))
|
||||
else:
|
||||
print "Debug: Sending fragmented package to dev"
|
||||
print " ---- COMPLETE MESSAGE BEFORE SLICES ---- "
|
||||
print oldencmsg
|
||||
print " ---- SLICES SLICES SLICES ---- "
|
||||
print "\n".join(slices)
|
||||
print " ---- SLICES SLICES SLICES ---- "
|
||||
server.privmsg(target, "b%s %s" % (msgid, slices.pop(0)))
|
||||
while len(slices) > 1:
|
||||
server.privmsg(target, "c%s %s" % (msgid, slices.pop(0)))
|
||||
server.privmsg(target, "e%s %s" % (msgid, slices.pop(0)))
|
||||
|
||||
def sendToDev(dev, server, isBroadcast, c, e):
|
||||
parsed = pkgre.match(e.arguments()[0])
|
||||
if not parsed:
|
||||
print "message could not be parsed", e.arguments()[0]
|
||||
return
|
||||
(flag, msgid, basemsg) = parsed.groups()
|
||||
print "flag: %s msgid: %s msg: %s" % (flag, msgid, basemsg)
|
||||
try:
|
||||
msgid = int(msgid)
|
||||
except ValueError:
|
||||
print "Debug: messageid was not a number"
|
||||
return
|
||||
if not ignoreNonMacUser:
|
||||
# FIXME: ignore the non prefix-mac user
|
||||
if not nickre.match():
|
||||
print "is not allowed in our network"
|
||||
print e.arguments(), irclib.nm_to_n(e.source()), e.target()
|
||||
if flag == "o":
|
||||
# oneliner!
|
||||
print "Debug: Writing onliner to dev"
|
||||
try:
|
||||
msg = base64.decodestring(basemsg)
|
||||
except base64.binascii.Error, e:
|
||||
print "Debug: Error decoding base64 irc message (%s)" % e
|
||||
return
|
||||
#FIXME if packetAllowed(, , chan)
|
||||
dev.write(msg)
|
||||
elif flag == 'b':
|
||||
if packets.has_key(msgid):
|
||||
print "Warning: Overwriting lost package with id %s" % msgid
|
||||
packets[msgid] = basemsg
|
||||
elif flag in ('c', 'e'):
|
||||
if not packets.has_key(msgid):
|
||||
print "Error: Continue package has no matching entry in packets, discarding!"
|
||||
else:
|
||||
packets[msgid] += basemsg
|
||||
if flag == 'e':
|
||||
arrmsg = packets[msgid]
|
||||
del(packets[msgid])
|
||||
try:
|
||||
print "arrmsg is", arrmsg.replace("\n", "\n\n\n")
|
||||
msg = base64.decodestring(arrmsg)
|
||||
except base64.binascii.Error, e:
|
||||
print "Debug: Error decoding base64 irc message (%s)" % e
|
||||
return
|
||||
print "Debug: writing fragmented package to dev"
|
||||
print binToHexStr(msg)
|
||||
dev.write(msg)
|
||||
#print "GOT in %s (broadcast %s) from %s msg %s" (e.target(), isBroadcast, irclib.nm_to_n(e.source()), e.arguments()[0])
|
||||
|
||||
def packetAllowed(src, nick, packet):
|
||||
if not acceptNonMatchingMac:
|
||||
# check if user-mac == packetmac
|
||||
# FIXME if not getDstMacFromPkt(packet) ==
|
||||
pass
|
||||
# FIXME: Maybe move nick check to here
|
||||
if mode != "HUB" and strictSwichedNetwork:
|
||||
if src.startswith("#"): # its a channel
|
||||
return False
|
||||
return True
|
||||
|
||||
def startup():
|
||||
# setup the tap device
|
||||
dev = TapDevice(pytap.IFF_TAP)
|
||||
dev.ifconfig(address=ip, netmask=netmask)
|
||||
if gateway != "":
|
||||
print "Setting default route %s" % gateway
|
||||
os.sys("route add default gw %s")
|
||||
|
||||
# setup IRC foo
|
||||
ircnick = nickPrefix + getMacFromIf(dev.name).replace(":", "")
|
||||
print "Debug: Connectiong to %s as %s" % (ircserver, ircnick)
|
||||
|
||||
irc = irclib.IRC()
|
||||
irc.add_global_handler("privmsg", lambda c, e: sendToDev(dev, server, False, c, e), -20)
|
||||
irc.add_global_handler("pubmsg", lambda c, e: sendToDev(dev, server, True, c, e), -20)
|
||||
server = irc.server()
|
||||
server.connect(ircserver[0], ircserver[1], ircnick)
|
||||
server.join(broadcastchan)
|
||||
|
||||
try:
|
||||
while True:
|
||||
sockets = [dev.__fd__]
|
||||
# get private irc sockets :)
|
||||
sockets.extend(map(lambda x: x._get_socket(), irc.connections))
|
||||
sockets = filter(lambda x: x != None, sockets)
|
||||
print "sockets", sockets
|
||||
(readFDs, _, _) = select.select(sockets, [], [])
|
||||
for readFD in readFDs:
|
||||
print "EVENT BY", readFD
|
||||
if readFD == dev.__fd__:
|
||||
sendToIRC(dev, server)
|
||||
else:
|
||||
print "IRC event"
|
||||
irc.process_once(0.1)
|
||||
except KeyboardInterrupt:
|
||||
pass
|
||||
|
||||
print "Shutting down!"
|
||||
server.quit("RST")
|
||||
|
||||
if __name__ == '__main__':
|
||||
startup()
|
||||
Loading…
Reference in new issue