monkey/monkey_island/cc/resources/pthmap.py

import hashlib
import binascii
import copy
import flask_restful
from pthreport import PassTheHashReport, Machine, get_report_html

from cc.auth import jwt_required
from cc.services.edge import EdgeService
from cc.services.node import NodeService
from cc.database import mongo

class PthMap(flask_restful.Resource):
    @jwt_required()
    def get(self, **kw):
        pth = PassTheHashReport()
        
        v = copy.deepcopy(pth.vertices)
        e = copy.deepcopy(pth.edges)
        
        return \
            {
                "graph": {
                    "nodes": [{"id": x, "label": Machine(x).GetIp()} for x in v],
                    "edges": [{"id": str(s) + str(t), "from": s, "to": t, "label": label} for s, t, label in e]
                },
                
                "report_html": get_report_html()
            }