fixes csirtgadgets#106 (csirtgadgets#107)

Author: wesyoung

Diff for: csirtg_indicator/utils/__init__.py

@@ -27,6 +27,7 @@
 RE_FQDN = re.compile('^((?!-))(xn--)?[a-z0-9][a-z0-9-_\.]{0,245}[a-z0-9]{0,1}\.(xn--)?([a-z0-9\-]{1,61}|[a-z0-9-]{1,30}\.[a-z]{2,})$')
 RE_URI_SCHEMES = re.compile('^(https?|ftp)$')
 RE_EMAIL = re.compile('^[_a-z0-9-]+(\.[_a-z0-9-]+)*@[a-z0-9-]+(\.[a-z0-9-]+)*(\.[a-z]{2,4})$')
+RE_ASN = re.compile('^(AS|as)[0-9]{1,5}$')
 
 RE_HASH = {
     'uuid': re.compile('^[0-9a-fA-F]{8}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{4}-[0-9a-fA-F]{12}$'),
@@ -147,6 +148,10 @@ def _email(s):
         if re.match(RE_EMAIL, s):
             return True
 
+    def _asn(s):
+        if re.match(RE_ASN, s):
+            return True
+
     if test_broken and _url_broken(indicator):
         return 'broken_url'
 
@@ -168,6 +173,9 @@ def _email(s):
     elif _fqdn(indicator):
         return 'fqdn'
 
+    elif _asn(indicator):
+        return 'asn'
+
     try:
         error = 'unknown itype for "{}"'.format(indicator)
     except UnicodeEncodeError:

Diff for: test/test_asns.py

@@ -0,0 +1,28 @@
+from csirtg_indicator import Indicator
+from csirtg_indicator.exceptions import InvalidIndicator
+
+
+def _not(data):
+    for d in data:
+        try:
+            d = Indicator(d)
+            assert d.itype is not 'asn'
+        except InvalidIndicator:
+            pass
+
+
+def test_asn_nok():
+    data = [
+        'example.com',
+        'http://example.com:81',
+        '192.168.1.1',
+        '127.0.0./1'
+    ]
+
+    _not(data)
+
+
+def test_asn_random():
+    for d in [1, 10, 500, 65535]:
+        assert Indicator(indicator='AS{}'.format(d)).itype == 'asn'
+        print(Indicator(indicator='AS{}'.format(d)))