This Is How We Do It

La ra ra ra ra ra
This is how we do it

.pylintrc

@@ -0,0 +1,6 @@
+[MASTER]
+extension-pkg-whitelist=GeoIP
+
+[FORMAT]
+indent-string='  '
+indent-after-paren=2

Config.py

@@ -3,6 +3,13 @@
 
 import yaml
 
-cnf = {}
-with open('data/config.yaml') as config_file:
-  cnf = yaml.load(config_file)
+class Config(object):
+  def __init__(self):
+    with open('data/config.yaml') as config_file:
+      self.config = yaml.load(config_file)
+
+  def get(self, key, defval=None):
+    return self.config.get(key, defval)
+
+
+cnf = Config()

README.md

@@ -2,130 +2,143 @@
 alpha beta whatever 
 partly works
 
-## roadmap
-refactor README and lib/plugin/plugins/*
-cleanup *
-
-probably Listener shoul be part of Core in order to supervise everything
-
-tasks don't store results currently. need to implement some validation of performed tasks (at least in Executor thread before spreading new tasks)
-http://python-rq.org/docs/results/
 
 ## configuration
 
 `data/config.yaml`
 ```
-# lists top-level services (like listeners, executors, data-manager) and pipelines enabled
+---
+dsl_version: 1
+
 core:
-  services: # should point to correct service
-  - data_manager
+  services:
+  - random_ip
   - rq_executor
+  - tg_feed
   pipelines:
   - ftp
+  - gopher
 
-# describes top-level services and their configurations
 services:
-  data_manager:
-    # REQUIRED package name, just like path to file with dots
-    package: lib.data.Manager
-    # REQUIRED class inherited from Service (lib.Service)
-    service: DataManager
-    # REQUIRED used to select one of storages
-    data:
-      id: pool
-    # there can be more service-specific configuration fields
-    # for now they can be found in code :)
-    sources:
-    - random_ip
-    feeds: 
-    - test_telegram
-  rq_executor:
-    package: lib.exeq.Executor
-    service: RQExecutor
-    data:
-      id: pool
-    redis:
-      host: "127.0.0.1"
-
-# describes datasources for data_manager
-sources:
   random_ip:
     package: lib.plugin.base.lib.IP
     service: RandomIP
-    data:
-      id: random_ip
-
-# describes datafeeds for data_manager
-feeds:
-  test_telegram: # doesn't work yet, eh
+    storage: ip_source
+  rq_executor:
+    package: lib.exec.Executor
+    service: RQExecutor
+    storage: pool
+    redis:
+      host: "127.0.0.1"
+  tg_feed:
     package: lib.plugin.base.lib.Telegram
     service: TelegramFeed
-    data:
-      id: pool
-    token: 
+    storage: pool
+    token: "mocken"
     chats:
-      - id: good_evening
-        pipelines: [ftp, gopher]
-        filter:
-          clause: any-of
-          equal: 
-          - ftp_list_files_status: success
-          - gopher_collect_status: success
+    - id: aiWeipeighah7vufoHa0ieToipooYe
+      if:
+        steps.ftp_apply_tpl: true
+        data.filter: false
+    - id: ohl7AeGah5uo8cho4nae9Eemaeyae3
+      if:
+        steps.gopher_apply_tpl: true
+        data.filter: false
 
-# describes various storages, e.g. data pool for pipelines or queues for datasources 
 storage:
   pool:
-    # REQUIRED
     package: lib.plugin.base.lib.Mongo
     service: MongoStorage
-    size: 40960
-    # service-specific
+    size: 0
     db: "medved"
     coll: 'pool'
-  random_ip:
+  ip_source:
     package: lib.plugin.base.lib.Mongo
     service: MongoStorage
-    size: 500
+    size: 800
     db: "medved"
-    coll: 'randomipsource'
+    coll: 'ip_source'
+
 
-# describes available pipelines
 pipelines:
   ftp:
-    # list of steps with dependencies
+    source: ip_source
     steps:
-    # will pass 10 items to lib.plugin.iscan.tasks.common.scan
-    - name: scan
-      package: lib.plugin.iscan.tasks.common
-      service: scan
-      multiple: 10 # default: False
-      requires: []
-    # will pass 1 item marked with ftp_scan to lib.plugin.iscan.tasks.ftp.connect
-    - name: connect
-      package: lib.plugin.iscan.tasks.ftp
-      service: connect
-      requires:
-      - ftp_scan
-    - name: list_files
-      package: lib.plugin.iscan.tasks.ftp
-      service: list_files
-      requires:
-      - ftp_connect
+    - task: ftp_scan
+      priority: low
+      parallel: 100
+    - task: ftp_connect
+      priority: normal
+      if:
+        steps.ftp_scan: true
+    - task: ftp_list_files
+      priority: high
+      if:
+        steps.ftp_connect: true
+    - task: ftp_apply_tpl
+      priority: high
+      if:
+        steps.ftp_list_files: true
+  gopher:
+    source: ip_source
+    steps:
+    - task: gopher_scan
+      priority: normal
+      parallel: 100
+    - task: gopher_find
+      priority: high
+      if:
+        steps.gopher_scan: true
+    - task: gopher_apply_tpl
+      priority: high
+      if:
+        steps.gopher_find: true
+    
+  http:
+    source: ip_source
+    steps:
+    - task: http_scan
+      priority: low
+      parallel: 25
 
-# various configurations for tasks
 tasks:
+  gopher_scan:
+    package: lib.plugin.iscan.tasks.common
+    service: MasScanTask
+    ports:
+    - 70
+  gopher_find:
+    package: lib.plugin.iscan.tasks.gopher
+    service: GopherFindTask
+  gopher_apply_tpl:
+    package: lib.plugin.base.tasks.text
+    service: Jinja2TemplateTask
+    path: lib/plugin/iscan/templates/gopher.tpl
+
   ftp_scan:
+    package: lib.plugin.iscan.tasks.common
+    service: MasScanTask
     ports:
     - 21
   ftp_connect: 
+    package: lib.plugin.iscan.tasks.ftp
+    service: FTPConnectTask
     logins: data/ftp/logins.txt
     passwords: data/ftp/passwords.txt
     bruteforce: true
     timeout: 15
   ftp_list_files:
+    package: lib.plugin.iscan.tasks.ftp
+    service: FTPListFilesTask
+    filter: true
+  ftp_apply_tpl:
+    package: lib.plugin.base.tasks.text
+    service: Jinja2TemplateTask
+    path: lib/plugin/iscan/templates/ftp.tpl
 
 logging: 
-  Storage: INFO 
+  Storage: DEBUG
+  Loader: DEBUG
 ```
 probably it can be launched with docker, however I didn't test it yet
 
@@ -137,9 +150,10 @@ you'll need working redis and mongodb for default configuration
 
 ## top-level services
 
-### lib.data.Manager.DataManager
-Orchestrates datasources and datafeeds - starts and stops them, also checks pool size. If it is too low - takes data from DS.
-### lib.exeq.Executor.RQExecutor
+### sources ###
+### feeds ###
+
+### lib.exec.Executor.RQExecutor
 Should run pipelines described in configuration. Works via [RedisQueue](http://python-rq.org/), so needs some Redis up and running
 Basically takes data from pool and submits it to workers.
 RQ workers should be launched separately (`rqworker worker` from code root)

data/config.yaml

@@ -3,110 +3,175 @@ dsl_version: 1
 
 core:
   services:
-  - data_manager
-#  - zmq_listener
+  - random_ip
   - rq_executor
+  - GC
+  - tg_feed
   pipelines:
   - ftp
-
+  - gopher
 
 services:
-  data_manager:
-    package: lib.data.Manager
-    service: DataManager
-    data:
-      id: pool
-    sources:
-    - random_ip
-    feeds: 
-    - test_telegram
-  zmq_listener:
-    package: lib.net.Listener
-    service: ZMQListener
-    data:
-      id: pool
-    listen: "0.0.0.0"
-    port: 12321
-  rq_executor:
-    package: lib.exeq.Executor
-    service: RQExecutor
-    data:
-      id: pool
-    redis:
-      host: "127.0.0.1"
-
-
-sources:
   random_ip:
     package: lib.plugin.base.lib.IP
     service: RandomIP
-    data:
-      id: random_ip
-
-
-feeds:
-  test_telegram:
+    storage: ip_source
+  rq_executor:
+    package: lib.exec.Executor
+    service: RQExecutor
+    storage: pool
+    delay: 2
+    redis:
+      host: redis
+  GC:
+    package: lib.plugin.base.lib.GC
+    service: GarbageCollector
+    storage: pool
+    delay: 10
+    if:
+      steps.ftp_scan: false
+      steps.gopher_scan: false
+  tg_feed:
     package: lib.plugin.base.lib.Telegram
     service: TelegramFeed
-    data:
-      id: pool
-    token: 
+    storage: pool
+    token: "358947254:"
     chats:
-      - id: good_evening
-        pipelines: [ftp, gopher]
-        filter:
-          clause: any-of
-          equal: 
-          - ftp_list_files_status: success
-          - gopher_collect_status: success
-
+    - id: aiWeipeighah7vufoHa0ieToipooYe
+      if:
+        steps.ftp_apply_tpl: true
+    - id: ohl7AeGah5uo8cho4nae9Eemaeyae3
+      if:
+        steps.gopher_apply_tpl: true
+        data.filter: false
 
 storage:
   pool:
     package: lib.plugin.base.lib.Mongo
     service: MongoStorage
-    size: 40960
+    url: mongo
+    size: 0
     db: "medved"
     coll: 'pool'
-  random_ip:
+  ip_source:
     package: lib.plugin.base.lib.Mongo
     service: MongoStorage
-    size: 500
+    url: mongo
+    size: 800
     db: "medved"
-    coll: 'randomipsource'
+    coll: 'ip_source'
 
 
 pipelines:
   ftp:
+    source: ip_source
     steps:
-    - name: scan
-      package: lib.plugin.iscan.tasks.common
-      service: scan
-      multiple: 10
-      requires: []
-    - name: connect
-      package: lib.plugin.iscan.tasks.ftp
-      service: connect
-      multiple: False
-      requires:
-      - ftp_scan
-    - name: list_files
-      package: lib.plugin.iscan.tasks.ftp
-      service: list_files
-      multiple: False
-      requires:
-      - ftp_connect
+    - task: ftp_scan
+      priority: low
+      parallel: 100
+    - task: ftp_connect
+      priority: normal
+      if:
+        steps.ftp_scan: true
+    - task: ftp_list_files
+      priority: normal
+      if:
+        steps.ftp_connect: true
+    - task: ftp_filter_files
+      priority: normal
+      parallel: 100
+      if:
+        steps.ftp_list_files: true
+    - task: ftp_apply_tpl
+      priority: high
+      if:
+        steps.ftp_filter_files: true
+        data.filter: false
+  gopher:
+    source: ip_source
+    steps:
+    - task: gopher_scan
+      priority: normal
+      parallel: 100
+    - task: gopher_find
+      priority: high
+      if:
+        steps.gopher_scan: true
+    - task: gopher_apply_tpl
+      priority: high
+      if:
+        steps.gopher_find: true
+    
+  http:
+    source: ip_source
+    steps:
+    - task: http_scan
+      priority: low
+      parallel: 25
 
 tasks:
+  gopher_scan:
+    package: lib.plugin.iscan.tasks.common
+    service: MasScanTask
+    ports:
+    - 70
+  gopher_find:
+    package: lib.plugin.iscan.tasks.gopher
+    service: GopherFindTask
+  gopher_apply_tpl:
+    package: lib.plugin.base.tasks.text
+    service: Jinja2TemplateTask
+    path: lib/plugin/iscan/templates/gopher.tpl
+
+  vnc_scan:
+    package: lib.plugin.iscan.tasks.common
+    service: MasScanTask
+    ports:
+    - 5900
+    - 5901
+  vnc_connect:
+    package: lib.plugin.iscan.tasks.vnc
+    service: VNCConnectTask
+    ports:
+    - 5900
+    - 5901
+
+  http_scan:
+    package: lib.plugin.iscan.tasks.common
+    service: MasScanTask
+    ports: &http_ports
+    - 80
+    - 81
+    - 8080
+    - 8081
+  http_find:
+    package: lib.plugin.iscan.tasks.http
+    service: HTTPFindTask
+    
+
   ftp_scan:
+    package: lib.plugin.iscan.tasks.common
+    service: MasScanTask
     ports:
     - 21
   ftp_connect: 
-    logins: data/ftp/logins.txt
+    package: lib.plugin.iscan.tasks.ftp
+    service: FTPConnectTask
+    usernames: data/ftp/usernames.txt
     passwords: data/ftp/passwords.txt
     bruteforce: true
     timeout: 15
   ftp_list_files:
+    package: lib.plugin.iscan.tasks.ftp
+    service: FTPListFilesTask
+  ftp_filter_files:
+    package: lib.plugin.iscan.tasks.ftp
+    service: FTPFilterFilesTask
+  ftp_apply_tpl:
+    package: lib.plugin.base.tasks.text
+    service: Jinja2TemplateTask
+    path: lib/plugin/iscan/templates/ftp.tpl
 
 logging: 
-  Storage: INFO
+  Storage: DEBUG
+  Loader: INFO

docker-compose.yml

@@ -51,6 +51,8 @@ services:
 
   mongo:
     image: mongo:latest
+    command:
+      - '--quiet'
     volumes:
       - ./docker/lib/mongo:/data/
     env_file:

docker/core/files/run.sh

@@ -7,6 +7,7 @@ export REDIS_IP=$(host ${REDIS_IP} | head -n1 | grep -Po "(\d+\.?){4}")
 
 /tmp/confd -onetime -backend env
 
-sudo -u tor tor
+#sudo -u tor tor
 
-cd /mdvd && proxychains -q python3 medved.py
+#cd /mdvd && proxychains -q python3 medved.py
+cd /mdvd && python3 medved.py

docker/worker/Dockerfile

@@ -1,5 +1,16 @@
 FROM medved_base:latest
 
+RUN pacman -S --noconfirm --needed git libpcap linux-headers clang tor
+
+RUN git clone https://github.com/robertdavidgraham/masscan && \
+    cd masscan && \
+    make -j && \
+    mv bin/masscan /usr/bin/masscan
+
+RUN wget -N http://geolite.maxmind.com/download/geoip/database/GeoLiteCity.dat.gz && gunzip GeoLiteCity.dat.gz
+
+RUN mkdir -p /usr/share/GeoIP/ && mv GeoLiteCity.dat /usr/share/GeoIP/GeoIPCity.dat
+
 ADD files/run.sh /tmp/run.sh
 
 CMD ["/tmp/run.sh"]

docker/worker/files/run.sh

@@ -4,4 +4,4 @@ export CORE_IP=$(host ${CORE_IP} | head -n1 | grep -Po "(\d+\.?){4}")
 
 /tmp/confd -onetime -backend env
 
-cd /mdvd && proxychains -q rq worker common -u "redis://${REDIS_IP}:6379/"
+cd /mdvd && rq worker high normal low -u "redis://${REDIS_IP}:6379/"

lib/Service.py

@@ -1,3 +1,8 @@
+"""
+Provides Service class
+"""
+
+
 from time import sleep
 from threading import Thread
 from lib import Logger, Loader, Loadable
@@ -9,7 +14,7 @@ class Service(Loadable):
   def __init__(self, thread, id, root=cnf):
     super().__init__(id, root)
 
-    self._data = Loader.by_id('storage', self.lcnf.get("data").get("id"))
+    self._data = Loader.by_id('storage', self.lcnf.get("storage"))
 
     self._stop_timeout = 10
     self._running = False
@@ -23,6 +28,7 @@ class Service(Loadable):
     pass
 
   def start(self):
+    """Executes pre_start, starts thread and executes post_start"""
     self._logger.debug('pre_start')
     self._pre_start()
 
@@ -38,6 +44,7 @@ class Service(Loadable):
     self._logger.info('start finished')
 
   def stop(self):
+    """Executes pre_stop, stops thread and executes post_stop"""
     self._logger.debug('pre_stop')
     self._pre_stop()
 

lib/data/Feed.py

@@ -1,20 +1,8 @@
-from queue import LifoQueue
-from time import sleep
-
-import itertools
-
-from lib.net import Remote
 from lib import Service
 
-
 class Feed(Service):
   """Base class for datafeeds"""
   def __init__(self, thread, id, root):
     super().__init__(thread, id, root)
     self._logger.add_field('service', 'Feed')
     self._logger.add_field('vname', self.__class__.__name__)
-
-  def get(self, plugin, count=1, timeout=3):
-    items = self._data.get(count)
-    self._logger.debug("get %s OF %s", len(items), count)
-    return items

lib/data/Item.py

@@ -0,0 +1,17 @@
+class Item(object):
+  """Base class for item"""
+  def __init__(self, source):  
+    self._item = {
+      'source': source,
+      'steps': {},
+      'data': {}
+    }
+
+  def set(self, key, value):
+    elem = self._item['data']
+    upd = {}
+    for x in key.split("."):
+      elem = elem.get(x, {})
+      upd[x] = {}
+    upd[0] = value
+    self._item['data'].update(upd)

lib/data/Manager.py

@@ -1,61 +0,0 @@
-from lib.data import Source, Feed
-from time import sleep
-from lib import Service, Loader
-
-class DataManager(Service):
-  """Actually, we may load feeds, sources and datapools right in core. Not sure that datamanager is required just to pull sources"""
-  def __init__(self, id, root):
-    super().__init__(self.__run, id, root)
-    self._logger.add_field('service', 'DataManager')
-
-    self.sources = {}
-    for s in self.lcnf.get("sources"):
-      self.attach_source(s)
-    self.feeds = {}
-    for f in self.lcnf.get("feeds"):
-      self.attach_feed(f)
-
-  def _pre_start(self):
-    self._logger.debug('starting sources')
-    for _,s  in self.sources.items():
-      s.start()
-    self._logger.debug('starting feeds')
-    for _,f in self.feeds.items():
-      f.start()
-
-  def _pre_stop(self):
-    self._logger.debug('stopping sources')
-    for _,s in self.sources.items():
-      s.stop()
-    self._logger.debug('stopping feeds')
-    for _,f in self.feeds.items():
-      f.stop()
-  
-  def attach_source(self, id):
-    ds = Loader.by_id('sources', id)
-    self.sources[id] = ds
-
-  def attach_feed(self, id):
-    df = Loader.by_id('feeds', id)
-    self.feeds[id] = df
-  
-  def get_source(self, name) -> Source:
-    return self.sources.get(name)
-  
-  def get_feed(self, name) -> Feed:
-    return self.feeds.get(name)
-
-  def __run(self):
-    oneshot = self.lcnf.get("oneshot", 500)
-    while self._running:
-      if self._data.count() < oneshot:
-        while self._running and (self._data.count() + oneshot < self._data.size()):
-          self._logger.debug("fill %s OF %s", self._data.count(), self._data.size())
-          for _,source in self.sources.items():
-            items = source.next(count=oneshot)
-            if items:
-              self._data.put(items)
-          sleep(1)
-      else:
-        self._logger.debug('Pool size is ok: %s', self._data.count())
-      sleep(1)

lib/data/Source.py

@@ -1,21 +1,22 @@
+import copy
+
 from lib import Service
 
 class Source(Service):
   """Base class for datasources"""
   def __init__(self, thread, id, root):
     super().__init__(thread, id, root)
-    self._logger.add_field('service', 'Feed')
+    self._logger.add_field('service', 'Source')
     self._logger.add_field('vname', self.__class__.__name__)
 
-  def item(self, val = None):
-    return {
+    self._item = {
       'source': self._id,
       'steps': {},
-      'data': val
+      'data': {}
     }
 
-  def next(self, count=10, block=False):
-    if self._running or not self._data.count() == 0:
-      return self._data.get(count=count, block=block)
-    elif self._data.count() == 0:
-      raise Exception("Storage is empty, generator is stopped")
+  def _create(self):
+    return copy.deepcopy(self._item)
+
+  def _prepare(self, item):
+    pass

lib/data/Storage.py

@@ -1,4 +1,4 @@
-from queue import LifoQueue, Empty, Full
+import inspect
 
 from lib import Loadable, Logger
 
@@ -7,7 +7,7 @@ class Storage(Loadable):
   def __init__(self, id, root):
     super().__init__(id, root)
 
-    self._size = self.lcnf.get("size")
+    self._size = self.lcnf.get("size", 0)
     self._logger = Logger("Storage")
     self._logger.add_field('vname', self.__class__.__name__)
 
@@ -27,7 +27,9 @@ class Storage(Loadable):
     return items
 
   def get(self, count=1, block=True, filter=None):
-    self._logger.debug("get %s, %s", count, block)
+    """Returns items, removing them from storage"""
+    self._logger.debug("get|%s|%s|%s",
+                       count, block, inspect.stack()[1][0].f_locals["self"].__class__.__name__)
     items = []
     if count == 1:
       items.append(self._get(block, filter))
@@ -44,40 +46,37 @@ class Storage(Loadable):
         self._put(i, block)
 
   def put(self, items, block=True):
+    """Puts provided items"""
+    self._logger.debug("put|%s|%s|%s",
+                       len(items), block, inspect.stack()[1][0].f_locals["self"].__class__.__name__)
     if items:
       items = [i for i in items if i is not None]
-      self._logger.debug("put %s, %s", len(items), block)
       if len(items) == 1:
         self._put(items[0], block)
       elif len(items) > 1:
         self._put_many(items, block)
 
-  def _find(self):
+  def _find(self, filter):
     pass
 
-  def find(self):
-    self._logger.debug("find")
-    return self._find()
+  def find(self, filter):
+    """Returns items without removing them from storage"""
+    return self._find(filter)
 
-
-class LiFoStorage(Storage):
-  def __init__(self, id, root):
-    super().__init__(id, root)
-    self._data = LifoQueue()
-
-  def count(self):
-    return self._data.qsize()
-  
-  def _get(self, block=False, filter=None):
-    try:
-      return self._data.get(block=block)
-    except Empty:
+  def _update(self, items, update):
     pass
 
-  def _put(self, item, block=True):
-    try:
-      self._data.put(item, block=block)
-    except Full:
+  def update(self, items, update=None):
+    """Updates provided items"""
+    self._logger.debug("update|%s|%s",
+                       len(items), inspect.stack()[1][0].f_locals["self"].__class__.__name__)
+    if items:
+      items = [i for i in items if i is not None]
+      self._update(items, update)
+
+  def _remove(self, items):
     pass
 
-
+  def remove(self, items):
+    """Removes provided items"""
+    self._remove(items)

lib/data/Type.py

@@ -1,13 +0,0 @@
-from lib import Loadable, Logger
-
-# dunno
-
-class Type(Loadable):
-  def __init__(self):
-    self.data = {}
-  
-class Host(Type):
-  def __init__(self):
-    self.data = {
-      'ip': ''
-    }

lib/data/__init__.py

@@ -1,7 +1,6 @@
 from .Storage import Storage
 from .Source import Source
 from .Feed import Feed
+from .Item import Item
 
-from .Manager import DataManager
-
-__all__ = ['Storage', 'Source', 'DataManager']
+__all__ = ['Storage', 'Source', 'Feed', 'Item']

lib/exec/Executor.py

@@ -0,0 +1,71 @@
+from lib import Service, Loader, Loadable
+
+from time import sleep
+
+from rq import Queue
+from redis import Redis
+
+
+class Executor(Service):
+  """Base class for executors"""
+  def __init__(self, thread, id, root):
+    super().__init__(thread, id, root)
+    self._logger.add_field('service', 'Executor')
+    self._logger.add_field('vname', self.__class__.__name__)
+
+
+class RQExecutor(Executor):
+  """rq (redis queue) executor"""
+  def __init__(self, id, root):
+    super().__init__(self.__run, id, root)
+
+  def __run(self):
+    redis_conn = Redis(host=self.lcnf.get('redis').get('host'))
+    jobs = []
+    known_sources = {}
+    
+    while self._running:
+      sleep(self.lcnf.get('delay', 0.5))
+      try:
+        for pn, pipeline in self.cnf.get("pipelines").items():
+          if pn not in self.cnf.get('core').get('pipelines'):
+            continue
+          for step in pipeline['steps']:
+            q = Queue(step.get('priority', 'normal'), connection=redis_conn)
+            for job_id in jobs:
+              job = q.fetch_job(job_id)
+              if job:
+                if job.result is not None:
+                  self._logger.debug("%s|%s", job_id, job._status)
+                  self._data.update(job.result)
+                  job.cleanup()
+                  jobs.remove(job_id)
+            if len(jobs) + 1 > self.lcnf.get('qsize', 200):
+              continue
+            filter = {"steps.%s" % step['task']: {'$exists': False}}
+            filter.update({key: value for key, value in step.get("if", {}).items()})
+            count = step.get('parallel', 1)
+            # get as much as possible from own pool
+            items = self._data.get(block=False, count=count, filter=filter)
+            # obtain everything else from source
+            if len(items) < count:
+              source = None
+              source_id = pipeline.get('source')
+              if source_id in known_sources:
+                source = known_sources[source_id]
+              else:
+                source = Loader.by_id('storage', source_id)
+                known_sources[source_id] = source
+              new_items = source.get(block=False, count=(count - len(items)), filter=filter)
+              items.extend(new_items)
+              source.remove(new_items)
+
+            if items:
+              for i in items:
+                i['steps'][step['task']] = None
+              self._data.update(items)
+              job = q.enqueue("lib.exec.Task.run", step['task'], items)
+              self._logger.info("%s|%s|%s|%s", job.id, step.get('priority', 'normal'), step['task'], len(items))
+              jobs.append(job.id)
+      except Exception as e:
+        self._logger.error("Error in executor main thread: %s", e)

lib/exec/Task.py

@@ -0,0 +1,28 @@
+from lib import Loadable, Logger, Loader
+
+from Config import cnf
+
+class Task(Loadable):
+  def __init__(self, id, root):
+    super().__init__(id, root)
+    self._logger = Logger(self.__class__.__name__)
+  
+  def run(self, items):
+    result = self._run(items)
+    return result
+
+  def _run(self, items):
+    for item in items:
+      try:
+        item['steps'][self._id] = self._process(item)
+      except Exception as e:
+        self._logger.debug("Error occured while executing: %s", e)
+        item['steps'][self._id] = False
+    return items
+
+  def _process(self, item):
+    return True
+
+def run(task_name, items):
+  result = Loader.by_id('tasks', task_name).run(items)
+  return result

lib/exec/__init__.py

@@ -1 +1,2 @@
 from .Executor import Executor
+from .Task import Task

lib/exeq/Executor.py

@@ -1,54 +0,0 @@
-from lib import Service, Loader, Loadable
-
-from lib.tasks.worker import worker
-
-from time import sleep
-
-from rq import Queue
-from redis import Redis
-
-
-class Executor(Service):
-  """Base class for executors"""
-  def __init__(self, thread, id, root):
-    super().__init__(thread, id, root)
-    self._logger.add_field('service', 'Executor')
-    self._logger.add_field('vname', self.__class__.__name__)
-
-
-class RQExecutor(Executor):
-  """rq (redis queue) executor - lightweight; workers placed on different nodes"""
-  def __init__(self, id, root):
-    super().__init__(self.__run, id, root)
-
-  def __run(self):
-    while self._running:
-      try:
-        redis_conn = Redis(host=self.lcnf.get('redis').get('host'))
-        q = Queue('worker', connection=redis_conn)
-        if q.count + 1 > self.lcnf.get('size', 100):
-          sleep(self.lcnf.get('delay', 2))
-          continue
-        for pn, pipeline in self.cnf.get("pipelines").items():
-          self._logger.debug("pipeline: %s", pn)
-          for step in pipeline['steps']:
-            self._logger.debug("step: %s", step['name'])
-            filter = {
-              "not_exist": [
-                pn + '_' + step['name']
-              ],
-              "exist": [
-                [tag for tag in step.get("requires")]
-              ]
-            }
-            items = []
-            multiple = step.get('multiple', False)
-            if multiple != False:
-              items = self._data.get(block=False, count=multiple, filter=filter)
-            else:
-              items = self._data.get(block=False, filter=filter)
-            if items:
-              self._logger.debug("enqueueing %s.%s with %s", step['package'], step['service'], items)
-              q.enqueue("%s.%s" % (step['package'], step['service']), items)
-      except Exception as e:
-        self._logger.error(e)

lib/exeq/Pipeline.py

@@ -1,7 +0,0 @@
-from lib import Loadable
-#TODO dunno
-class Pipeline(Loadable):
-  def __init__(self, id, root):
-    super().__init__(id, root)
-  
-  

lib/plugin/Manager.py

@@ -1,9 +0,0 @@
-import importlib
-
-class Manager:
-  def __init__(self):
-    pass
-  
-  @staticmethod
-  def get_plugin(name):
-    return importlib.import_module("lib.plugin.plugins." + name)

lib/plugin/README

@@ -1,6 +0,0 @@
-THIS PART IS LIKE UNDER CONSTRUCTION
-
-get out
-for now
-
-come back later

lib/plugin/Task.py

@@ -1,7 +0,0 @@
-class Task:
-  """Pipelines should consist of tasks??..."""
-  def __init__(self):
-    self._data = None
-  
-  def run(self):
-    pass

lib/plugin/__init__.py

@@ -1 +0,0 @@
-from .Manager import Manager

lib/plugin/base/lib/GC.py

@@ -0,0 +1,25 @@
+"""
+Provides garbage collector
+"""
+
+from time import sleep
+
+from lib import Service
+
+class GarbageCollector(Service):
+  """Simple GarbageCollector, removes items by filter periodically"""
+  def __init__(self, id, root):
+    super().__init__(self.__run, id, root)
+    self._logger.add_field('service', 'GC')
+    self._logger.add_field('vname', self.__class__.__name__)
+
+  def __run(self):
+    while self._running:
+      filter = {key: value for key, value in self.lcnf.get("if", {}).items()}
+      if filter:
+        items = self._data.find(filter=filter)
+        self._logger.info("Removing %s items", items.count())
+        self._data.remove(items)
+      else:
+        self._logger.error("Filter is empty!")
+      sleep(self.lcnf.get('delay', 600))

lib/plugin/base/lib/IP.py

@@ -1,35 +1,51 @@
-from lib.data import Source
-from lib import Loader
-
 from time import sleep
 
-import os
-import netaddr
 import itertools
 import random
 import socket
 import struct
+import netaddr
+
+import GeoIP
+
+from lib.data import Source
 
 class IPSource(Source):
+  """Base source for IPs, appends data.ip and data.geo"""
   def __init__(self, thread, id, root):
     super().__init__(thread, id, root)
-  
-  def item(self, val = None):
-    return {
-      'source': self._id,
+    self._item.update({
       'data': {
-        'ip': val
+        'ip': None,
+        'geo': {
+          'country': None,
+          'city': None
         }
       }
+    })
+    self.geo_ip = GeoIP.open(self.lcnf.get("geoip_dat", "/usr/share/GeoIP/GeoIPCity.dat"),
+                             GeoIP.GEOIP_INDEX_CACHE | GeoIP.GEOIP_CHECK_CACHE)
 
+  def _geoip(self, item):
+    geodata = self.geo_ip.record_by_name(item['data']['ip'])
+    if geodata:
+      if 'country_code3' in geodata and geodata['country_code3']:
+        item['data']['geo']['country'] = geodata['country_code3']
+      if 'city' in geodata and geodata['city']:
+        item['data']['geo']['city'] = geodata['city']
+
+  def _prepare(self, item):
+    self._geoip(item)
 
 class IPRange(IPSource):
+  """Provides IPs from ranges specified in file"""
   def __init__(self, id, root):
     super().__init__(self.__run, id, root)
     self._iprange = []
     self.load_ip_range()
 
   def load_ip_range(self):
+    """Loads IP ranges from specified path"""
     ip_range = []
     with open(self.lcnf.get('path'), "r") as text:
       for line in text:
@@ -40,18 +56,19 @@ class IPRange(IPSource):
           else:
             ip_range.append(netaddr.IPNetwork(diap[0]))
         except Exception as e:
-          raise Exception("Error while adding range {}: {}".format(line, e))
+          raise Exception("Error while adding range %s: %s" % (line, e))
     self._iprange = ip_range
 
   def __run(self):
-    npos = 0
-    apos = 0
+    npos = 0 # network cursor
+    apos = 0 # address cursor
     while self._running:
       try:
-        for _ in itertools.repeat(None, self.lcnf.get('oneshot', 100)):
+        for _ in itertools.repeat(None, self.lcnf.get('oneshot', 200)):
+          item = self._create()
           if self.lcnf.get('ordered', True):
             # put currently selected element
-            self._data.put(str(self._iprange[npos][apos]))
+            item['data']['ip'] = str(self._iprange[npos][apos])
             # rotate next element through networks and addresses
             if apos + 1 < self._iprange[npos].size:
               apos += 1
@@ -65,13 +82,16 @@ class IPRange(IPSource):
                 else:
                   self.stop()
           else:
-            self._data.put(str(random.choice(random.choice(self._iprange))))
+            item['data']['ip'] = str(random.choice(random.choice(self._iprange)))
+          self._prepare(item)
+          self._data.put(item)
         sleep(self.lcnf.get('delay', 0.5))
-      except Exception as e:
-        self._logger.warn(e)
+      except Exception as err:
+        self._logger.warn(err)
 
 
 class RandomIP(IPSource):
+  """Generates completely pseudorandom IPs"""
   def __init__(self, id, root):
     super().__init__(self.__run, id, root)
 
@@ -79,10 +99,13 @@ class RandomIP(IPSource):
     while self._running:
       try:
         items = []
-        for _ in itertools.repeat(None, self.lcnf.get("oneshot", 100)):
+        for _ in itertools.repeat(None, self.lcnf.get("oneshot", 200)):
+          item = self._create()
           randomip = socket.inet_ntoa(struct.pack('>I', random.randint(1, 0xffffffff)))
-          items.append(self.item(str(randomip)))
+          item['data']['ip'] = str(randomip)
+          self._prepare(item)
+          items.append(item)
         self._data.put(items)
-        sleep(self.lcnf.get("delay", 0.5))
-      except Exception as e:
-        self._logger.warn(e)
+        sleep(self.lcnf.get("delay", 0.2))
+      except Exception as err:
+        self._logger.warn(err)

lib/plugin/base/lib/Mongo.py

@@ -1,5 +1,6 @@
-from pymongo import MongoClient
 from time import sleep
+from pymongo import MongoClient
+
 from lib.data import Storage
 
 class MongoStorage(Storage):
@@ -17,41 +18,31 @@ class MongoStorage(Storage):
     return self._coll.count()
 
   def _get(self, block, filter):
-    # TODO cleanup dat BS
     if filter is None:
       filter = {}
-    ne_tags = {}
-    e_tags = {}
-    if filter.get('not_exist'):
-      tags = []
-      for ne in filter.get('not_exist'):
-        tags.append(ne)
-      ne_tags = {'tags': {'$not': {'$all': tags}}}
-      del filter['not_exist']
-    if filter.get('exist'):
-      tags = []
-      for e in filter.get('exist'):
-        tags.append(e)
-      e_tags = {'tags': {'$all': tags}}
-      del filter['exist']
-    filter = {'$and': [ne_tags, e_tags]}
-    item = self._coll.find_one_and_delete(filter=filter)
+
+    item = self._coll.find_one(filter=filter)
     if block:
       while not item:
-        item = self._coll.find_one_and_delete(filter=filter)
+        item = self._coll.find_one(filter=filter)
         sleep(1)
-    
     return item
 
+  def _get_many(self, count, block, filter):
+    if filter is None:
+      filter = {}
+    items = self._coll.find(filter=filter, limit=count)
+    return items
+
   def _put(self, item, block):
-    if block:
+    if block and self.size() is not 0:
       while self.count() + 1 > self.size():
         self._logger.debug('Collection full: %s of %s', self.count(), self.size())
         sleep(1)
     self._coll.insert_one(item)
 
   def _put_many(self, items, block):
-    if block:
+    if block and self.size() is not 0:
       while self.count() + len(items) > self.size():
         self._logger.debug('Collection full: %s of %s', self.count(), self.size())
         sleep(1)
@@ -61,3 +52,14 @@ class MongoStorage(Storage):
     if filter is None:
       filter = {}
     return self._coll.find(filter)
+
+  def _update(self, items, update):
+    if update:
+      filter = {'_id': {'$in': [item['_id'] for item in items]}}
+      self._coll.update_many(filter, update, upsert=True)
+    else:
+      for item in items:
+        self._coll.replace_one({'_id': item['_id']}, item, upsert=True)
+
+  def _remove(self, items):
+    self._coll.delete_many({'_id': {'$in': [item['_id'] for item in items]}})

lib/plugin/base/lib/Telegram.py

@@ -1,9 +1,7 @@
-from lib.data import Feed, Filter
-from lib.plugin import Manager
-
-import telebot
 from time import sleep
+import telebot
 
+from lib.data import Feed
 
 class TelegramFeed(Feed):
   """Send data to Telegram chat"""
@@ -16,24 +14,18 @@ class TelegramFeed(Feed):
     while self._running:
       try:
         for chat in self.lcnf.get("chats"):
-          chat_id = chat['id']
-          sleep(delay)
-          continue
-          # plugins -> pipelines
-          # it is in progress
-          #TODO
-          msg = Manager.get_plugin(plugin).Plugin.TelegramMessage(host)
-          msg.run()
-          if msg.data['txt']:
+          chat_id = chat.get('id')
+          self._logger.debug(chat_id)
+          filter = {"feed.%s" % self._id: {'$exists': False}}
+          filter.update({key: value for key, value in chat.get("if", {}).items()})
+          items = self._data.get(block=False, count=10, filter=filter)
+          self._logger.debug(items)
+          if items:
+            self._data.update(items, {'$set': {'feed.%s' % self._id: True}})
             tbot = telebot.TeleBot(self.lcnf.get('token'), threaded=False)
-            if msg.data['img']:
-              self._logger.debug("Send IP with img %s:%s to %s" % (host['ip'], host['port'], chat_id))
-              tbot.send_photo("@" + chat_id, msg.data['img'], caption=msg.data['txt'])
-            else:
-              self._logger.debug("Send IP %s:%s to %s" % (host['ip'], host['port'], chat_id))
-              tbot.send_message("@" + chat_id, msg.data['txt'])
-          else:
-            self._logger.error('Empty text!')
+            for i in items:
+              self._logger.debug("@%s: %s", chat_id, i['data']['message'])
+              tbot.send_message("@" + chat_id, i['data']['message'])
         sleep(delay)
-      except Exception as e:
-        self._logger.warn(e)
+      except Exception as err:
+        self._logger.warn(err)

lib/plugin/base/tasks/text.py

@@ -0,0 +1,17 @@
+from lib.exec import Task
+from jinja2 import Environment, FileSystemLoader
+
+
+class Jinja2TemplateTask(Task):
+  def __init__(self, id, root):
+    super().__init__(id, root)
+
+  def _process(self, item):
+    template = Environment(loader=FileSystemLoader('.')).get_template(self.lcnf.get('path'))
+    item['data']['message'] = template.render(data = item['data'])
+    item['steps'][self._id] = True
+
+  def _run(self, items):
+    for item in items:
+      self._process(item)
+    return items

lib/plugin/iscan/plugin.yaml

@@ -1,62 +0,0 @@
-plugin:
-  name: iscan
-  version: 0.1
-  pipelines:
-    FTP:
-      actions:
-        - scan
-        - connect
-        - metadata
-        - filetree
-      df:
-        Telegram:
-          action: metadata
-          chats:
-            - xai7poozengee2Aen3poMookohthaZ
-            - aiWeipeighah7vufoHa0ieToipooYe
-    HTTP:
-      actions:
-        - scan
-        - connect
-        - metadata
-        - screenshot
-      df:
-        Telegram:
-          action: screenshot
-          chats:
-            - xai7poozengee2Aen3poMookohthaZ
-            - gohquooFee3duaNaeNuthushoh8di2
-    Gopher:
-      actions:
-        - connect
-        - collect
-      df:
-        Telegram:
-          action: collect
-          chats:
-            - xai7poozengee2Aen3poMookohthaZ
-            - ohl7AeGah5uo8cho4nae9Eemaeyae3
-
-
-df:
-  Telegram:
-    token: TOKEN
-
-ds:
-  IPRange:
-    file: file
-  Remote:
-
-
-docker:
-  services:
-    selenium:
-      image: selenium/standalone-chrome:latest
-      volumes:
-        - /dev/shm:/dev/shm
-      environment:
-        - JAVA_OPTS=-Dselenium.LOGGER.level=WARNING
-      worker_env:
-        - SELENIUM_IP=selenium
-      required_by:
-        - HTTP

lib/plugin/iscan/tasks/common.py

@@ -3,23 +3,19 @@
 import subprocess
 import json
 from jsoncomment import JsonComment
-from lib import Logger
-import GeoIP
-from Config import cnf
 
-logger = Logger("common")
+from lib.exec import Task
 
-
-class MasScan:
-  def __init__(self, bin_path='/usr/bin/masscan', opts="-sS -Pn -n --wait 0 --max-rate 5000"):
-    self.bin_path = bin_path
-    self.opts_list = opts.split(' ')
-
-  def scan(self, ip_list, port_list):
+class MasScanTask(Task):
+  """Provides data.ports for each of items scanned with masscan"""
+  def scan(self, ip_list, port_list, bin_path, opts="-sS -Pn -n --wait 0 --max-rate 5000"):
+    """Executes masscan on given IPs/ports"""
+    bin_path = bin_path
+    opts_list = opts.split(' ')
     port_list = ','.join([str(p) for p in port_list])
     ip_list = ','.join([str(ip) for ip in ip_list])
-    process_list = [self.bin_path]
-    process_list.extend(self.opts_list)
+    process_list = [bin_path]
+    process_list.extend(opts_list)
     process_list.extend(['-oJ', '-', '-p'])
     process_list.append(port_list)
     process_list.append(ip_list)
@@ -30,29 +26,27 @@ class MasScan:
     result = parser.loads(out)
     return result
 
-def scan(items):
-  gi = GeoIP.open(cnf.get("geoip_dat", "/usr/share/GeoIP/GeoIP.dat"), GeoIP.GEOIP_INDEX_CACHE | GeoIP.GEOIP_CHECK_CACHE)
-  logger.debug("Starting scan")
-  ms = MasScan()
-  hosts = ms.scan(ip_list=[i['data']['ip'] for i in items], 
-                  port_list=cnf.get("tasks").get('ftp_scan').get("ports"))
-  logger.debug(hosts)
-  for h in hosts:
-    for port in h['ports']:
-      host = {
-        'ip': h['ip'],
-        'port': port['port'],
-        'data': {
-          'geo': {
-            'country': None,
-            'city': None
-          }
-        }
-      }
-    geodata = gi.record_by_name(host['ip'])
-    if geodata:
-      if 'country_code3' in geodata and geodata['country_code3']:
-        host['data']['geo']['country'] = geodata['country_code3']
-      if 'city' in geodata and geodata['city']:
-        host['data']['geo']['city'] = geodata['city']
-    logger.debug("Found %s:%s", host['ip'], host['port'])
+  def _run(self, items):
+    ip_list = [i['data']['ip'] for i in items]
+    port_list = self.lcnf.get("ports")
+
+    self._logger.debug("Starting scan, port_list=%s", port_list)
+
+    hosts = self.scan(ip_list=ip_list,
+                      port_list=port_list,
+                      bin_path=self.lcnf.get('bin_path', "/usr/bin/masscan"))
+
+    self._logger.debug(hosts)
+    hosts = {h['ip']: h for h in hosts}
+    for item in items:
+      if hosts.get(item['data']['ip']):
+        ports = [p['port'] for p in hosts[item['data']['ip']]['ports']]
+        if 'ports' in item['data']:
+          item['data']['ports'].extend(ports)
+        else:
+          item['data']['ports'] = ports
+        item['steps'][self._id] = True
+        self._logger.debug("Found %s with open ports %s", item['data']['ip'], ports)
+      else:
+        item['steps'][self._id] = False
+    return items

lib/plugin/iscan/tasks/ftp.py

@@ -0,0 +1,91 @@
+"""
+Basic tasks for FTP services
+"""
+
+import ftplib
+
+from lib.exec import Task
+
+class FTPConnectTask(Task): # pylint: disable=too-few-public-methods
+  """Tries to connect FTP service with various credentials"""
+  def _process(self, item):
+    ftp = ftplib.FTP(host=item['data']['ip'], timeout=self.lcnf.get('timeout', 30))
+    try:
+      self._logger.debug('Trying anonymous login')
+      ftp.login()
+    except ftplib.error_perm as err:
+      self._logger.debug('Failed (%s)', err)
+    else:
+      self._logger.info('Succeeded with anonymous')
+      item['data']['username'] = 'anonymous'
+      item['data']['password'] = ''
+      return True
+
+    if self.lcnf.get('bruteforce', False):
+      self._logger.debug('Bruteforce enabled, loading usernames and passwords')
+      usernames = [line.rstrip() for line in open(self.lcnf.get('usernames'), 'r')]
+      passwords = [line.rstrip() for line in open(self.lcnf.get('passwords'), 'r')]
+
+      for username in usernames:
+        for password in passwords:
+          self._logger.debug('Checking %s', username + ':' + password)
+          try:
+            self._logger.debug('Sending NOOP')
+            ftp.voidcmd('NOOP')
+          except IOError as err:
+            self._logger.debug('IOError occured (%s), attempting to open new connection', err)
+            ftp = ftplib.FTP(host=item['data']['ip'], timeout=self.lcnf.get('timeout', 30))
+          try:
+            self._logger.debug('Trying to log in')
+            ftp.login(username, password)
+          except ftplib.error_perm as err:
+            self._logger.debug('Failed (%s)', err)
+            continue
+          else:
+            self._logger.info('Succeeded with %s', username + ':' + password)
+            item['data']['username'] = username
+            item['data']['password'] = password
+            return True
+    self._logger.info('Could not connect')
+    return False
+
+class FTPListFilesTask(Task): # pylint: disable=too-few-public-methods
+  """Executes NLST to list files on FTP"""
+  def _process(self, item):
+    ftp = ftplib.FTP(host=item['data']['ip'], 
+                     user=item['data']['username'],
+                     passwd=item['data']['password'])
+    filelist = ftp.nlst()
+    try:
+      ftp.quit()
+    except ftplib.Error:
+      pass
+
+    item['data']['files'] = []
+    for filename in filelist:
+      item['data']['files'].append(filename)
+    return True
+
+class FTPFilterFilesTask(Task): # pylint: disable=too-few-public-methods
+  """Sets data.filter if FTP contains only junk"""
+  def _process(self, item):
+    junk_list = ['incoming', '..', '.ftpquota', '.', 'pub']
+    files = item['data']['files']
+
+    item['data']['filter'] = False
+
+    try:
+      if not files or files[0] == "total 0":
+        item['data']['filter'] = "Empty"
+    except IndexError:
+      pass
+
+    if 0 < len(files) <= len(junk_list): # pylint: disable=C1801
+      match_count = 0
+      for filename in junk_list:
+        if filename in files:
+          match_count += 1
+        if match_count == len(files):
+          item['data']['filter'] = "EmptyWithBloatDirs"
+
+    return True

lib/plugin/iscan/tasks/gopher.py

@@ -0,0 +1,47 @@
+"""
+Basic tasks for Gopher services
+"""
+
+import socket
+
+from lib.exec import Task
+
+class GopherFindTask(Task): # pylint: disable=too-few-public-methods
+  """Tries to connect Gopher service"""
+  @staticmethod
+  def _recv(sck):
+    total_data = []
+    while True:
+      data = sck.recv(2048)
+      if not data:
+        break
+      total_data.append(data.decode('utf-8'))
+    return ''.join(total_data)
+
+  def _process(self, item):
+    sock = socket.socket()
+    sock.settimeout(self.lcnf.get('timeout', 20))
+    sock.connect((item['data']['ip'], int(70)))
+    sock.sendall(b'\n\n')
+
+    response = self._recv(sock)
+    sock.close()
+
+    self._logger.debug("Parsing result: %s", response)
+    item['data']['files'] = []
+    item['data']['filter'] = False
+    for s in [s for s in response.split("\r\n") if s]:
+      node = {}
+      fields = s.split("\t")
+      self._logger.debug(fields)
+      node['type'] = fields[0][0]
+      if len(fields) == 4:
+        node['name'] = fields[0][1:]
+        node['path'] = fields[1]
+        node['serv'] = f"{fields[2]}:{fields[3]}"
+        item['data']['files'].append(node)
+
+    if not item['data']['files']:
+      raise Exception("Empty server (not Gopher?)")
+
+    return True

lib/plugin/iscan/tasks/http.py

@@ -0,0 +1,60 @@
+from lib.exec import Task
+
+from io import BytesIO
+import json
+import time
+
+import bs4
+import requests
+import urllib3
+from PIL import Image
+from bson.binary import Binary
+
+from selenium import webdriver
+from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
+from selenium.webdriver.common.proxy import Proxy, ProxyType
+import zlib
+import netaddr
+
+
+class HTTPFindTask(Task):
+  def __init__(self, id, root):
+    super().__init__(id, root)
+
+  def _process(self, item):
+      urllib3.disable_warnings()
+      response = requests.get(url='http://%s:%s/' % (self._host['ip'], self._host['port']),
+                              timeout=cnf.stalker.HTTP.timeout,
+                              verify=False)
+
+      if response.status_code in [400, 401, 403, 500]:
+        raise self.PipelineError("Bad response")
+
+      self._host['data']['response'] = {}
+      self._host['data']['response']['code'] = response.status_code
+      self._host['data']['response']['text'] = response.text
+      self._host['data']['response']['content'] = response.content
+      self._host['data']['response']['encoding'] = response.encoding
+      self._host['data']['response']['headers'] = response.headers
+
+      encoding = response.encoding if 'charset' in response.headers.get('content-type', '').lower() else None
+      soup = bs4.BeautifulSoup(response.content, "html.parser", from_encoding=encoding)
+      if soup.original_encoding != 'utf-8':
+        meta = soup.select_one('meta[charset], meta[http-equiv="Content-Type"]')
+        if meta:
+          if 'charset' in meta.attrs:
+            meta['charset'] = 'utf-8'
+          else:
+            meta['content'] = 'text/html; charset=utf-8'
+        self._host['data']['response']['text'] = soup.prettify()  # encodes to UTF-8 by default
+
+      title = soup.select_one('title')
+      if title:
+        if title.string:
+          title = title.string
+        else:
+          title = ""
+      else:
+        title = ""
+
+      self._host['data']['title'] = title

lib/plugin/iscan/tasks/vnc.py

@@ -0,0 +1,10 @@
+import tempfile
+
+from lib.exec import Task
+
+class VNCFindTask(Task): # pylint: disable=too-few-public-methods
+  """Tries to connect FTP service with various credentials"""
+  def _process(self, item):
+    fd, temp_path = tempfile.mkstemp()
+    print(fd, temp_path)
+ 

lib/plugin/iscan/templates/ftp.tpl

@@ -0,0 +1,6 @@
+ftp://{{data['username']}}:{{data['password']}}@{{data['ip']}}
+{% for filename in data['files'] -%}
+ + {{ filename }}
+{% endfor -%}
+Geo: {{data['geo']['country']}}/{{data['geo']['city']}}
+

lib/plugin/iscan/templates/gopher.tpl

@@ -0,0 +1,15 @@
+gopher://{{data['ip']}}/
+Dirs:
+{% for node in data['files'] -%}
+{% if node['type'] == '1' -%}
+ + {{node['path']}}
+{% endif -%}
+{% endfor -%}
+Other nodes:
+{% for node in data['files'] -%}
+{% if node['type'] != '1' and node['type'] != 'i' -%}
+ + {{node['path']}}
+     {{node['name']}}
+{% endif -%}
+{% endfor -%}
+Geo: {{data['geo']['country']}}/{{data['geo']['city']}}

lib/plugin/iscan/templates/http.tpl

@@ -0,0 +1,19 @@
+
+#code{{data['response']['code']}}
+server = self._host['data']['response']['headers'].get('Server', None)
+{% if server != none %}
+
+
+" Server: #%s\n" % server
+
+else:
+
+"\n"
+
+if self._host['data']['title']:
+
+"Title: %s\n" % self._host['data']['title']
+
+"Geo: %s/%s\n" % (self._host['data']['geo']['country'], self._host['data']['geo']['city'])
+"http://%s:%s/\n" % (self._host['ip'], self._host['port'])
+"#http_" + str(int(netaddr.IPAddress(self._host['ip'])))

lib/plugin/plugins/FTP.py

@@ -1,110 +0,0 @@
-import ftplib
-import netaddr
-
-from Config import cnf
-from lib.plugin.plugins import BasePlugin
-
-
-class Plugin(BasePlugin):
-  class TelegramMessage(BasePlugin.TelegramMessage):
-    def _init(self):
-      self._name = "FTP"
-    
-    def _generate(self):
-      self.data['txt'] = "ftp://%s:%s@%s/\n" % \
-                        (self._host['data']['username'], self._host['data']['password'], self._host['ip'])
-      for filename in self._host['data']['files']:
-        self.data['txt'] += " + %s\n" % filename
-      self.data['txt'] += "Geo: %s/%s\n" % (self._host['data']['geo']['country'], self._host['data']['geo']['city'])
-      self.data['txt'] += "#ftp_" + str(int(netaddr.IPAddress(self._host['ip'])))
-
-  class Pipeline(BasePlugin.Pipeline):
-    def _init(self):
-      self._name = "FTP"
-
-    def run(self):
-      try:
-        self._connect()
-        self._find()
-        self._filter()
-        self._push()
-      except Exception as e:
-        self._logger.debug("Error occured: %s (%s)", e, self._host['ip'])
-      else:
-        self._logger.info("Succeeded for %s" % self._host['ip'])
-
-    def _connect(self):
-      self.ftp = ftplib.FTP(host=self._host['ip'], timeout=cnf.stalker.FTP.timeout)
-      try:
-        self._logger.debug('Trying anonymous login')
-        self.ftp.login()
-      except ftplib.error_perm:
-        pass
-      else:
-        self._logger.debug('Succeeded with anonymous')
-        self._host['data']['username'] = 'anonymous'
-        self._host['data']['password'] = ''
-        return
-
-      if cnf.stalker.FTP.bruteforce:
-        usernames = []
-        passwords = []
-
-        with open(cnf.stalker.FTP.logins, 'r') as lfh:
-          for username in lfh:
-            usernames.append(username.rstrip())
-        with open(cnf.stalker.FTP.passwords, 'r') as pfh:
-          for password in pfh:
-            passwords.append(password.rstrip())
-        for username in usernames:
-          for password in passwords:
-            try:
-              self.ftp.voidcmd('NOOP')
-            except IOError:
-              self.ftp = ftplib.FTP(host=self._host['ip'], timeout=cnf.stalker.FTP.timeout)
-            self._logger.debug('Trying %s' % (username + ':' + password))
-            try:
-              self.ftp.login(username, password)
-            except ftplib.error_perm:
-              continue
-            except:
-              raise
-            else:
-              self._logger.debug('Succeeded with %s' %(username + ':' + password))
-              self._host['data']['username'] = username
-              self._host['data']['password'] = password
-              return
-        raise Exception('No matching credentials found')
-
-    def _find(self):
-      filelist = self.ftp.nlst()
-      try:
-        self.ftp.quit()
-      except:
-        # that's weird, but we don't care
-        pass
-
-      try:
-        if len(filelist) == 0 or filelist[0] == "total 0":
-          raise self.PipelineError("Empty server")
-      except IndexError:
-        pass
-
-      self._host['data']['files'] = []
-      for fileName in filelist:
-        self._host['data']['files'].append(fileName)
-
-    def _filter(self):
-      self._host['data']['filter'] = False
-      if len(self._host['data']['files']) == 0:
-        self._host['data']['filter'] = "Empty"
-      elif len(self._host['data']['files']) < 6:
-        match = 0
-        for f in 'incoming', '..', '.ftpquota', '.', 'pub':
-          if f in self._host['data']['files']:
-            match += 1
-          if match == len(self._host['data']['files']):
-            self._host['data']['filter'] = "EmptyWithSystemDirs"
-
-
-

lib/plugin/plugins/Gopher.py

@@ -1,70 +0,0 @@
-import socket
-import netaddr
-
-from Config import cnf
-from lib.plugin.plugins import BasePlugin
-
-
-class Plugin(BasePlugin):
-  class TelegramMessage(BasePlugin.TelegramMessage):
-    def _init(self):
-      self._name = "Gopher"
-
-    def _generate(self):
-      self.data['txt'] = "gopher://%s/\n" % self._host['ip']
-      self.data['txt'] += "Dirs:\n"
-      for dir in [f for f in self._host['data']['files'] if f['type'] == '1']:
-        self.data['txt'] += " + %s\n" % dir['path']
-      self.data['txt'] += "Other nodes:\n"
-      for file in [f for f in self._host['data']['files'] if f['type'] != '1' and f['type'] != 'i']:
-        self.data['txt'] += " + %s\n    %s\n" % (file['path'], file['name'])
-      self.data['txt'] += "Geo: %s/%s\n" % (self._host['data']['geo']['country'], self._host['data']['geo']['city'])
-      self.data['txt'] += "#gopher_" + str(int(netaddr.IPAddress(self._host['ip'])))
-
-  class Pipeline(BasePlugin.Pipeline):
-    def _init(self):
-      self._name = "Gopher"
-
-    def run(self):
-      try:
-        self._find()
-        self._push()
-      except Exception as e:
-        self._logger.debug("Error occured: %s (%s)", e, self._host['ip'])
-      else:
-        self._logger.info("Succeeded for %s" % self._host['ip'])
-
-    def _recv(self, sck):
-      total_data = []
-      while True:
-        data = sck.recv(2048)
-        if not data:
-          break
-        total_data.append(data.decode('utf-8'))
-      return ''.join(total_data)
-
-    def _find(self):
-      sock = socket.socket()
-      sock.settimeout(cnf.stalker.Gopher.timeout)
-      sock.connect((self._host['ip'], int(self._host['port'])))
-      sock.sendall(b'\n\n')
-
-      response = self._recv(sock)
-      sock.close()
-
-      self._logger.debug("Parsing result")
-      self._host['data']['files'] = []
-      self._host['data']['filter'] = False
-      for s in [s for s in response.split("\r\n") if s]:
-        node = {}
-        fields = s.split("\t")
-        self._logger.debug(fields)
-        node['type'] = fields[0][0]
-        if len(fields) == 4:
-          node['name'] = fields[0][1:]
-          node['path'] = fields[1]
-          node['serv'] = f"{fields[2]}:{fields[3]}"
-          self._host['data']['files'].append(node)
-
-      if not self._host['data']['files']:
-        raise self.PipelineError("Empty server (not Gopher?)")

lib/plugin/plugins/README

@@ -1 +0,0 @@
-plugins formed in old plugin format

lib/plugin/test.py

@@ -1,45 +0,0 @@
-from threading import Thread
-from time import sleep
-
-class A: # NOOP
-  def __init__(self, thread = None):
-    if thread:
-      self.__thread = Thread(target=thread)
-    self._running = False
-    self._init()
-  
-  def _init(self):
-    pass
-  
-  def start(self):
-    self._running = True
-    self.__thread.daemon = True
-    self.__thread.start()
-  
-  def stop(self):
-    self._running = False
-    self.__thread.join()
-  
-  def __run(self):
-    while(self._running):
-      print('NOOP')
-      sleep(1)
-
-class B(A): # NOOP
-  def __init__(self):
-    super().__init__(self.__run)
-
-  def __run(self):
-    while(self._running):
-      print('OP')
-      sleep(1)
-
-class C(A): # NOOP
-  def __run(self):
-    while(self._running):
-      print('OP')
-      sleep(1)
-
-  def _init(self):
-    self.__thread = Thread(target=self.__run)
-

lib/tasks/__init__.py

@@ -1,2 +0,0 @@
-#from .scan import scan
-#from .stalk import stalk

lib/tasks/worker.py

@@ -1,8 +0,0 @@
-from lib.plugin import Manager
-# legacy
-# why legacy?
-def worker(host, plugin):
-    p = Manager.get_plugin(plugin)
-    p.Plugin.Pipeline(host, plugin)
-    del p
-    # cool bro

lib/util.py

@@ -1,10 +1,10 @@
 import logging
 
-from Config import cnf as config
-
 import importlib
 import sys, os
 
+from Config import cnf as config
+
 class Logger(logging.Logger):
   """Logger. standard logging logger with some shitcode on the top"""
   def __init__(self, name):
@@ -74,8 +74,8 @@ class Logger(logging.Logger):
 class Loadable:
   """parent for loadable from configuration"""
   def __init__(self, id, root=config):
-    self.cnf = config
-    self.lcnf = root[id]
+    self.cnf = config # global config
+    self.lcnf = root[id] # local config
     self._id = id
 
 
@@ -95,6 +95,9 @@ class Loader:
     return getattr(result, name)
 
   @classmethod
-  def by_id(cls, section, id):
-    l = cls(config[section][id].get('package'))
-    return l.get(config[section][id].get('service'))(id=id, root=config[section])
+  def by_id(cls, section, id) -> Loadable:
+    """Returns instantiated object of class provided in configuration"""
+    # prepares Loader for certain package
+    loader = cls(config.get(section).get(id).get('package'))
+    # loads class from this package and returns instantiated object of this class
+    return loader.get(config.get(section).get(id).get('service'))(id=id, root=config.get(section))

medved.py

@@ -1,5 +1,4 @@
 #!/usr/bin/python3
-# -*- coding: utf-8 -*-
 
 import time
 
@@ -20,22 +19,24 @@ class Core:
       self._services.append(service)
 
   def start(self):
+    """Starts all loaded services"""
     self.logger.info("Starting")
     for service in self._services:
       service.start()
     self.logger.info("Started")
 
   def stop(self):
+    """Stops all loaded services"""
     self.logger.info("Stopping Core")
     for service in self._services:
       service.stop()
     self.logger.info("Stopped")
 
 if __name__ == '__main__':
-  core = Core()
-  core.start()
+  CORE = Core()
+  CORE.start()
   try:
     while True:
       time.sleep(1)
   except KeyboardInterrupt:
-    core.stop()
+    CORE.stop()

requirements.txt

@@ -12,3 +12,4 @@ zmq
 jsoncomment
 rq
 pyyaml
+jinja2