Add new iprep sources

Author: michaelschem

suricata/update/commands/addsource.py

@@ -37,6 +37,8 @@ def register(parser):
                         help="Additional HTTP header to add to requests")
     parser.add_argument("--no-checksum", action="store_false",
                         help="Skips downloading the checksum URL")
+    parser.add_argument("--iprep", action="store_true",
+                        help="Identifies source as an IPRep Source")
     parser.set_defaults(func=add_source)
 
 
@@ -67,6 +69,8 @@ def add_source():
 
     header = args.http_header if args.http_header else None
 
+    is_iprep = args.iprep
+
     source_config = sources.SourceConfiguration(
-        name, header=header, url=url, checksum=checksum)
+        name, header=header, url=url, checksum=checksum, is_iprep=is_iprep)
     sources.save_source_config(source_config)

suricata/update/sources.py

@@ -79,12 +79,13 @@ def save_source_config(source_config):
 class SourceConfiguration:
 
     def __init__(self, name, header=None, url=None,
-                 params={}, checksum=True):
+                 params={}, checksum=True, is_iprep=False):
         self.name = name
         self.url = url
         self.params = params
         self.header = header
         self.checksum = checksum
+        self.is_iprep = is_iprep
 
     def dict(self):
         d = {
@@ -98,6 +99,8 @@ def dict(self):
             d["http-header"] = self.header
         if self.checksum:
             d["checksum"] = self.checksum
+        if self.is_iprep:
+            d["is_iprep"] = self.is_iprep
         return d
 
 class Index: