mirror of https://github.com/CIRCL/AIL-framework
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
# -*- coding: utf-8 -*-
|
||
|
|
#
|
||
|
|
# This file is part of AIL framework - Analysis Information Leak framework
|
||
|
|
#
|
||
|
|
# This a simple feeder script feeding data from pystemon to AIL.
|
||
|
|
#
|
||
|
|
# Don't forget to set your pystemonpath and ensure that the
|
||
|
|
# configuration matches this script. Default is Redis DB 10.
|
||
|
|
#
|
||
|
|
# https://github.com/cvandeplas/pystemon/blob/master/pystemon.yaml#L16
|
||
|
|
#
|
||
|
|
# This program is free software: you can redistribute it and/or modify
|
||
|
|
# it under the terms of the GNU Affero General Public License as published by
|
||
|
|
# the Free Software Foundation, either version 3 of the License, or
|
||
|
|
# (at your option) any later version.
|
||
|
|
#
|
||
|
|
# Copyright (c) 2014 Alexandre Dulaunoy - a@foo.be
|
||
|
|
|
||
|
|
|
||
|
|
import zmq
|
||
|
|
import random
|
||
|
|
import sys
|
||
|
|
import time
|
||
|
|
import redis
|
||
|
|
import base64
|
||
|
|
|
||
|
|
port = "5556"
|
||
|
|
pystemonpath = "/home/pystemon/pystemon/"
|
||
|
|
|
||
|
|
context = zmq.Context()
|
||
|
|
socket = context.socket(zmq.PUB)
|
||
|
|
socket.bind("tcp://*:%s" % port)
|
||
|
|
|
||
|
|
# check https://github.com/cvandeplas/pystemon/blob/master/pystemon.yaml#L16
|
||
|
|
r = redis.StrictRedis(host='localhost', db=10)
|
||
|
|
|
||
|
|
# 101 pastes processed feed
|
||
|
|
# 102 raw pastes feed
|
||
|
|
|
||
|
|
while True:
|
||
|
|
time.sleep(1)
|
||
|
|
topic = 101
|
||
|
|
paste = r.lpop("pastes")
|
||
|
|
if paste is None:
|
||
|
|
continue
|
||
|
|
socket.send("%d %s" % (topic, paste))
|
||
|
|
topic = 102
|
||
|
|
messagedata = open(pystemonpath+paste).read()
|
||
|
|
socket.send("%d %s %s" % (topic, paste, base64.b64encode(messagedata)))
|