|  | @@ -14,11 +14,11 @@ import json
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  yes_regex = re.compile(r'([yY][eE][sS]|[yY])+$')
 | 
	
		
			
				|  |  |  if re.search(yes_regex, os.getenv('SKIP_FAIL2BAN', 0)):
 | 
	
		
			
				|  |  | -  print "SKIP_FAIL2BAN=y, Skipping Fail2ban container..."
 | 
	
		
			
				|  |  | +  print 'SKIP_FAIL2BAN=y, Skipping Fail2ban container...'
 | 
	
		
			
				|  |  |    time.sleep(31536000)
 | 
	
		
			
				|  |  |    raise SystemExit
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -r = redis.StrictRedis(host='172.22.1.249', decode_responses=True, port=6379, db=0)
 | 
	
		
			
				|  |  | +r = redis.StrictRedis(host=os.getenv('IPV4_NETWORK', '172.22.1') + '.249', decode_responses=True, port=6379, db=0)
 | 
	
		
			
				|  |  |  pubsub = r.pubsub()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  RULES = {}
 | 
	
	
		
			
				|  | @@ -29,19 +29,23 @@ RULES[4] = '-login: Aborted login \(tried to use disallowed .+\): user=.+, rip=(
 | 
	
		
			
				|  |  |  RULES[5] = 'SOGo.+ Login from \'([0-9a-f\.:]+)\' for user .+ might not have worked'
 | 
	
		
			
				|  |  |  RULES[6] = 'mailcow UI: Invalid password for .+ by ([0-9a-f\.:]+)'
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -r.setnx("F2B_BAN_TIME", "1800")
 | 
	
		
			
				|  |  | -r.setnx("F2B_MAX_ATTEMPTS", "10")
 | 
	
		
			
				|  |  | -r.setnx("F2B_RETRY_WINDOW", "600")
 | 
	
		
			
				|  |  | +r.setnx('F2B_BAN_TIME', '1800')
 | 
	
		
			
				|  |  | +r.setnx('F2B_MAX_ATTEMPTS', '10')
 | 
	
		
			
				|  |  | +r.setnx('F2B_RETRY_WINDOW', '600')
 | 
	
		
			
				|  |  | +r.setnx('F2B_NETBAN_IPV6', '64')
 | 
	
		
			
				|  |  | +r.setnx('F2B_NETBAN_IPV4', '24')
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  bans = {}
 | 
	
		
			
				|  |  |  log = {}
 | 
	
		
			
				|  |  |  quit_now = False
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def ban(address):
 | 
	
		
			
				|  |  | -  BAN_TIME = int(r.get("F2B_BAN_TIME"))
 | 
	
		
			
				|  |  | -  MAX_ATTEMPTS = int(r.get("F2B_MAX_ATTEMPTS"))
 | 
	
		
			
				|  |  | -  RETRY_WINDOW = int(r.get("F2B_RETRY_WINDOW"))
 | 
	
		
			
				|  |  | -  WHITELIST = r.hgetall("F2B_WHITELIST")
 | 
	
		
			
				|  |  | +  BAN_TIME = int(r.get('F2B_BAN_TIME'))
 | 
	
		
			
				|  |  | +  MAX_ATTEMPTS = int(r.get('F2B_MAX_ATTEMPTS'))
 | 
	
		
			
				|  |  | +  RETRY_WINDOW = int(r.get('F2B_RETRY_WINDOW'))
 | 
	
		
			
				|  |  | +  WHITELIST = r.hgetall('F2B_WHITELIST')
 | 
	
		
			
				|  |  | +  NETBAN_IPV6 = '/' + str(r.get('F2B_NETBAN_IPV6'))
 | 
	
		
			
				|  |  | +  NETBAN_IPV4 = '/' + str(r.get('F2B_NETBAN_IPV4'))
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |    ip = ipaddress.ip_address(address.decode('ascii'))
 | 
	
		
			
				|  |  |    if type(ip) is ipaddress.IPv6Address and ip.ipv4_mapped:
 | 
	
	
		
			
				|  | @@ -56,13 +60,13 @@ def ban(address):
 | 
	
		
			
				|  |  |        wl_net = ipaddress.ip_network(wl_key.decode('ascii'), False)
 | 
	
		
			
				|  |  |        if wl_net.overlaps(self_network):
 | 
	
		
			
				|  |  |          log['time'] = int(round(time.time()))
 | 
	
		
			
				|  |  | -        log['priority'] = "info"
 | 
	
		
			
				|  |  | -        log['message'] = "Address %s is whitelisted by rule %s" % (self_network, wl_net)
 | 
	
		
			
				|  |  | -        r.lpush("F2B_LOG", json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | -        print "Address %s is whitelisted by rule %s" % (self_network, wl_net)
 | 
	
		
			
				|  |  | +        log['priority'] = 'info'
 | 
	
		
			
				|  |  | +        log['message'] = 'Address %s is whitelisted by rule %s' % (self_network, wl_net)
 | 
	
		
			
				|  |  | +        r.lpush('F2B_LOG', json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | +        print 'Address %s is whitelisted by rule %s' % (self_network, wl_net)
 | 
	
		
			
				|  |  |          return
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  | -  net = ipaddress.ip_network((address + ('/24' if type(ip) is ipaddress.IPv4Address else '/64')).decode('ascii'), strict=False)
 | 
	
		
			
				|  |  | +  net = ipaddress.ip_network((address + (NETBAN_IPV4 if type(ip) is ipaddress.IPv4Address else NETBAN_IPV6)).decode('ascii'), strict=False)
 | 
	
		
			
				|  |  |    net = str(net)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |    if not net in bans or time.time() - bans[net]['last_attempt'] > RETRY_WINDOW:
 | 
	
	
		
			
				|  | @@ -78,45 +82,45 @@ def ban(address):
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |    if bans[net]['attempts'] >= MAX_ATTEMPTS:
 | 
	
		
			
				|  |  |      log['time'] = int(round(time.time()))
 | 
	
		
			
				|  |  | -    log['priority'] = "crit"
 | 
	
		
			
				|  |  | -    log['message'] = "Banning %s" % net
 | 
	
		
			
				|  |  | -    r.lpush("F2B_LOG", json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | -    print "Banning %s for %d minutes" % (net, BAN_TIME / 60)
 | 
	
		
			
				|  |  | +    log['priority'] = 'crit'
 | 
	
		
			
				|  |  | +    log['message'] = 'Banning %s' % net
 | 
	
		
			
				|  |  | +    r.lpush('F2B_LOG', json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | +    print 'Banning %s for %d minutes' % (net, BAN_TIME / 60)
 | 
	
		
			
				|  |  |      if type(ip) is ipaddress.IPv4Address:
 | 
	
		
			
				|  |  | -      subprocess.call(["iptables", "-I", "INPUT", "-s", net, "-j", "REJECT"])
 | 
	
		
			
				|  |  | -      subprocess.call(["iptables", "-I", "FORWARD", "-s", net, "-j", "REJECT"])
 | 
	
		
			
				|  |  | +      subprocess.call(['iptables', '-I', 'INPUT', '-s', net, '-j', 'REJECT'])
 | 
	
		
			
				|  |  | +      subprocess.call(['iptables', '-I', 'FORWARD', '-s', net, '-j', 'REJECT'])
 | 
	
		
			
				|  |  |      else:
 | 
	
		
			
				|  |  | -      subprocess.call(["ip6tables", "-I", "INPUT", "-s", net, "-j", "REJECT"])
 | 
	
		
			
				|  |  | -      subprocess.call(["ip6tables", "-I", "FORWARD", "-s", net, "-j", "REJECT"])
 | 
	
		
			
				|  |  | -    r.hset("F2B_ACTIVE_BANS", "%s" % net, log['time'] + BAN_TIME)
 | 
	
		
			
				|  |  | +      subprocess.call(['ip6tables', '-I', 'INPUT', '-s', net, '-j', 'REJECT'])
 | 
	
		
			
				|  |  | +      subprocess.call(['ip6tables', '-I', 'FORWARD', '-s', net, '-j', 'REJECT'])
 | 
	
		
			
				|  |  | +    r.hset('F2B_ACTIVE_BANS', '%s' % net, log['time'] + BAN_TIME)
 | 
	
		
			
				|  |  |    else:
 | 
	
		
			
				|  |  |      log['time'] = int(round(time.time()))
 | 
	
		
			
				|  |  | -    log['priority'] = "warn"
 | 
	
		
			
				|  |  | -    log['message'] = "%d more attempts in the next %d seconds until %s is banned" % (MAX_ATTEMPTS - bans[net]['attempts'], RETRY_WINDOW, net)
 | 
	
		
			
				|  |  | -    r.lpush("F2B_LOG", json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | -    print "%d more attempts in the next %d seconds until %s is banned" % (MAX_ATTEMPTS - bans[net]['attempts'], RETRY_WINDOW, net)
 | 
	
		
			
				|  |  | +    log['priority'] = 'warn'
 | 
	
		
			
				|  |  | +    log['message'] = '%d more attempts in the next %d seconds until %s is banned' % (MAX_ATTEMPTS - bans[net]['attempts'], RETRY_WINDOW, net)
 | 
	
		
			
				|  |  | +    r.lpush('F2B_LOG', json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | +    print '%d more attempts in the next %d seconds until %s is banned' % (MAX_ATTEMPTS - bans[net]['attempts'], RETRY_WINDOW, net)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def unban(net):
 | 
	
		
			
				|  |  |    log['time'] = int(round(time.time()))
 | 
	
		
			
				|  |  | -  log['priority'] = "info"
 | 
	
		
			
				|  |  | -  r.lpush("F2B_LOG", json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | +  log['priority'] = 'info'
 | 
	
		
			
				|  |  | +  r.lpush('F2B_LOG', json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  |    if not net in bans:
 | 
	
		
			
				|  |  | -    log['message'] = "%s is not banned, skipping unban and deleting from queue (if any)" % net
 | 
	
		
			
				|  |  | -    r.lpush("F2B_LOG", json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | -    print "%s is not banned, skipping unban and deleting from queue (if any)" % net
 | 
	
		
			
				|  |  | -    r.hdel("F2B_QUEUE_UNBAN", "%s" % net)
 | 
	
		
			
				|  |  | +    log['message'] = '%s is not banned, skipping unban and deleting from queue (if any)' % net
 | 
	
		
			
				|  |  | +    r.lpush('F2B_LOG', json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | +    print '%s is not banned, skipping unban and deleting from queue (if any)' % net
 | 
	
		
			
				|  |  | +    r.hdel('F2B_QUEUE_UNBAN', '%s' % net)
 | 
	
		
			
				|  |  |      return
 | 
	
		
			
				|  |  | -  log['message'] = "Unbanning %s" % net
 | 
	
		
			
				|  |  | -  r.lpush("F2B_LOG", json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | -  print "Unbanning %s" % net
 | 
	
		
			
				|  |  | +  log['message'] = 'Unbanning %s' % net
 | 
	
		
			
				|  |  | +  r.lpush('F2B_LOG', json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | +  print 'Unbanning %s' % net
 | 
	
		
			
				|  |  |    if type(ipaddress.ip_network(net.decode('ascii'))) is ipaddress.IPv4Network:
 | 
	
		
			
				|  |  | -    subprocess.call(["iptables", "-D", "INPUT", "-s", net, "-j", "REJECT"])
 | 
	
		
			
				|  |  | -    subprocess.call(["iptables", "-D", "FORWARD", "-s", net, "-j", "REJECT"])
 | 
	
		
			
				|  |  | +    subprocess.call(['iptables', '-D', 'INPUT', '-s', net, '-j', 'REJECT'])
 | 
	
		
			
				|  |  | +    subprocess.call(['iptables', '-D', 'FORWARD', '-s', net, '-j', 'REJECT'])
 | 
	
		
			
				|  |  |    else:
 | 
	
		
			
				|  |  | -    subprocess.call(["ip6tables", "-D", "INPUT", "-s", net, "-j", "REJECT"])
 | 
	
		
			
				|  |  | -    subprocess.call(["ip6tables", "-D", "FORWARD", "-s", net, "-j", "REJECT"])
 | 
	
		
			
				|  |  | -  r.hdel("F2B_ACTIVE_BANS", "%s" % net)
 | 
	
		
			
				|  |  | -  r.hdel("F2B_QUEUE_UNBAN", "%s" % net)
 | 
	
		
			
				|  |  | +    subprocess.call(['ip6tables', '-D', 'INPUT', '-s', net, '-j', 'REJECT'])
 | 
	
		
			
				|  |  | +    subprocess.call(['ip6tables', '-D', 'FORWARD', '-s', net, '-j', 'REJECT'])
 | 
	
		
			
				|  |  | +  r.hdel('F2B_ACTIVE_BANS', '%s' % net)
 | 
	
		
			
				|  |  | +  r.hdel('F2B_QUEUE_UNBAN', '%s' % net)
 | 
	
		
			
				|  |  |    del bans[net]
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def quit(signum, frame):
 | 
	
	
		
			
				|  | @@ -125,21 +129,21 @@ def quit(signum, frame):
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def clear():
 | 
	
		
			
				|  |  |    log['time'] = int(round(time.time()))
 | 
	
		
			
				|  |  | -  log['priority'] = "info"
 | 
	
		
			
				|  |  | -  log['message'] = "Clearing all bans"
 | 
	
		
			
				|  |  | -  r.lpush("F2B_LOG", json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | -  print "Clearing all bans"
 | 
	
		
			
				|  |  | +  log['priority'] = 'info'
 | 
	
		
			
				|  |  | +  log['message'] = 'Clearing all bans'
 | 
	
		
			
				|  |  | +  r.lpush('F2B_LOG', json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | +  print 'Clearing all bans'
 | 
	
		
			
				|  |  |    for net in bans.copy():
 | 
	
		
			
				|  |  |      unban(net)
 | 
	
		
			
				|  |  |    pubsub.unsubscribe()
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def watch():
 | 
	
		
			
				|  |  |    log['time'] = int(round(time.time()))
 | 
	
		
			
				|  |  | -  log['priority'] = "info"
 | 
	
		
			
				|  |  | -  log['message'] = "Watching Redis channel F2B_CHANNEL"
 | 
	
		
			
				|  |  | -  r.lpush("F2B_LOG", json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | -  pubsub.subscribe("F2B_CHANNEL")
 | 
	
		
			
				|  |  | -  print "Subscribing to Redis channel F2B_CHANNEL"
 | 
	
		
			
				|  |  | +  log['priority'] = 'info'
 | 
	
		
			
				|  |  | +  log['message'] = 'Watching Redis channel F2B_CHANNEL'
 | 
	
		
			
				|  |  | +  r.lpush('F2B_LOG', json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | +  pubsub.subscribe('F2B_CHANNEL')
 | 
	
		
			
				|  |  | +  print 'Subscribing to Redis channel F2B_CHANNEL'
 | 
	
		
			
				|  |  |    while True:
 | 
	
		
			
				|  |  |      for item in pubsub.listen():
 | 
	
		
			
				|  |  |        for rule_id, rule_regex in RULES.iteritems():
 | 
	
	
		
			
				|  | @@ -150,18 +154,18 @@ def watch():
 | 
	
		
			
				|  |  |              ip = ipaddress.ip_address(addr.decode('ascii'))
 | 
	
		
			
				|  |  |              if ip.is_private or ip.is_loopback:
 | 
	
		
			
				|  |  |                continue
 | 
	
		
			
				|  |  | -            print "%s matched rule id %d" % (addr, rule_id)
 | 
	
		
			
				|  |  | +            print '%s matched rule id %d' % (addr, rule_id)
 | 
	
		
			
				|  |  |              log['time'] = int(round(time.time()))
 | 
	
		
			
				|  |  | -            log['priority'] = "warn"
 | 
	
		
			
				|  |  | -            log['message'] = "%s matched rule id %d" % (addr, rule_id)
 | 
	
		
			
				|  |  | -            r.lpush("F2B_LOG", json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  | +            log['priority'] = 'warn'
 | 
	
		
			
				|  |  | +            log['message'] = '%s matched rule id %d' % (addr, rule_id)
 | 
	
		
			
				|  |  | +            r.lpush('F2B_LOG', json.dumps(log, ensure_ascii=False))
 | 
	
		
			
				|  |  |              ban(addr)
 | 
	
		
			
				|  |  |  
 | 
	
		
			
				|  |  |  def autopurge():
 | 
	
		
			
				|  |  |    while not quit_now:
 | 
	
		
			
				|  |  | -    BAN_TIME = int(r.get("F2B_BAN_TIME"))
 | 
	
		
			
				|  |  | -    MAX_ATTEMPTS = int(r.get("F2B_MAX_ATTEMPTS"))
 | 
	
		
			
				|  |  | -    QUEUE_UNBAN = r.hgetall("F2B_QUEUE_UNBAN")
 | 
	
		
			
				|  |  | +    BAN_TIME = int(r.get('F2B_BAN_TIME'))
 | 
	
		
			
				|  |  | +    MAX_ATTEMPTS = int(r.get('F2B_MAX_ATTEMPTS'))
 | 
	
		
			
				|  |  | +    QUEUE_UNBAN = r.hgetall('F2B_QUEUE_UNBAN')
 | 
	
		
			
				|  |  |      if QUEUE_UNBAN:
 | 
	
		
			
				|  |  |        for net in QUEUE_UNBAN:
 | 
	
		
			
				|  |  |          unban(str(net))
 |