Add abuseipdb lookup to link, fix differences in version running in prod
This commit is contained in:
		
					parent
					
						
							
								05eeab7c9c
							
						
					
				
			
			
				commit
				
					
						66df6e56e2
					
				
			
		
					 3 changed files with 30 additions and 25 deletions
				
			
		| 
						 | 
				
			
			@ -1,16 +1,29 @@
 | 
			
		|||
import requests
 | 
			
		||||
 | 
			
		||||
def get_abuse_confidence(ip):
 | 
			
		||||
    """ get abuseipdb's confidence level on an ip passed in, and return that value"""
 | 
			
		||||
    base_url = "https://api.abuseipdb.com/api/v2/check"
 | 
			
		||||
    api_key = "<YOUR API KEY>
 | 
			
		||||
    headers = { 'Key': api_key, 'Accept': 'application/json' }
 | 
			
		||||
    data = { 'ipAddress': ip, 'maxAgeInDays': 90 }
 | 
			
		||||
    r = requests.get(base_url, headers=headers, json=data)
 | 
			
		||||
    return r.json()['data']['abuseConfidenceScore']
 | 
			
		||||
 | 
			
		||||
def formatter(data, headers):
 | 
			
		||||
    """format a message sent with crowdsec http endpoints"""
 | 
			
		||||
    """ format a message sent with crowdsec http endpoints"""
 | 
			
		||||
    data_out = ""
 | 
			
		||||
    for row in data["body"]:
 | 
			
		||||
        if "crowdsecurity" in row["scenario"]:
 | 
			
		||||
            source, scenario, *_ = row["scenario"].split("/")
 | 
			
		||||
            row[
 | 
			
		||||
                "scenario"
 | 
			
		||||
            ] = f"[{scenario}](https://hub.crowdsec.net/author/crowdsecurity/configurations/{scenario})"
 | 
			
		||||
        data_out += (
 | 
			
		||||
            f"{row['host']} has been banned {row['duration']} due to {row['scenario']}\n\n"
 | 
			
		||||
            f"[AbuseIPDB](https://www.abuseipdb.com/check/{row['host']})|"
 | 
			
		||||
            f"[Crowdsec](https://app.crowdsec.net/cti/{row['host']})\n\n"
 | 
			
		||||
        )
 | 
			
		||||
       ip = row['host']
 | 
			
		||||
       duration = row['duration']
 | 
			
		||||
       confidence = get_abuse_confidence(ip)
 | 
			
		||||
       if "crowdsecurity" in row['scenario']:
 | 
			
		||||
          source, scenario, *_ = row['scenario'].split('/')
 | 
			
		||||
          row['scenario'] = f"[{scenario}](https://hub.crowdsec.net/author/crowdsecurity/configurations/{scenario})"
 | 
			
		||||
       data_out += (
 | 
			
		||||
                     f"{ip} has been banned {duration} due to {row['scenario']}\n\n"
 | 
			
		||||
                     f"[AbuseIPDB](https://www.abuseipdb.com/check/{row['host']})({confidence}%) | "
 | 
			
		||||
                     f"[Crowdsec](https://app.crowdsec.net/cti/{row['host']})\n\n"
 | 
			
		||||
       )
 | 
			
		||||
    data["body"] = data_out
 | 
			
		||||
    return data
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
	Add table
		Add a link
		
	
		Reference in a new issue