forked from p15670423/monkey
25 lines
703 B
Python
25 lines
703 B
Python
import hashlib
|
|
import binascii
|
|
import copy
|
|
import flask_restful
|
|
from pthreport import PassTheHashReport, Machine
|
|
|
|
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 \
|
|
{
|
|
"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]
|
|
}
|