forked from alfredkrohmer/profinet
/
server.py
110 lines (83 loc) · 3.31 KB
/
server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
from struct import unpack
import argparse
import binascii
from flask import Flask, request, render_template, redirect, url_for
from protocol import *
import dcp, rpc
app = Flask(__name__)
conns = {}
parser = argparse.ArgumentParser()
parser.add_argument("-i", required=True,
help="use INTERFACE", metavar="INTERFACE")
args = parser.parse_args()
s = ethernet_socket(args.i, 3)
src = get_mac(args.i)
def get_connection(name):
if name not in conns.keys():
info = rpc.get_station_info(s, src, name)
con = rpc.RPCCon(info)
con.connect(src)
conns[name] = (info, con)
return conns[name]
@app.route("/")
def index():
dcp.send_discover(s, src)
resp = [dcp.DCPDeviceDescription(mac, blocks) for mac, blocks in dcp.read_response(s, src).items()]
return render_template("discover.html", resp=resp)
@app.route("/device")
def device():
name = request.values.get('name')
info, con = get_connection(name)
data = con.read_inm0filter()
return render_template("device.html", info=info, data=data)
@app.route("/inm0", methods=["GET", "POST"])
def inm0():
name = request.values.get('name')
api = int(request.values.get('api'))
slot = int(request.values.get('slot'))
subslot = int(request.values.get('subslot'))
info, con = get_connection(name)
payload = con.read(api, slot, subslot, idx=PNInM0.IDX).payload
if len(payload) != 0:
data = PNInM0(payload)
else:
data = None
idx = request.values.get('idx')
paramdata = None
if idx is not None:
_idx = int(idx, 16)
if request.values.get('action') == "write":
paramdata = request.values.get('data')
con.write(api, slot, subslot, _idx, binascii.unhexlify(paramdata.replace(":", "")))
else:
paramdata = to_hex(con.read(api, slot, subslot, idx=_idx).payload)
else:
idx = ""
return render_template("inm0.html", name=name, api=api, slot=slot, subslot=subslot, idx=idx, data=data, paramdata=paramdata, inm1_supported=(data.im_supported&1<<1 if data is not None else False))
@app.route("/inm1", methods=["GET", "POST"])
def inm1():
name = request.values.get('name')
api = int(request.values.get('api'))
slot = int(request.values.get('slot'))
subslot = int(request.values.get('subslot'))
info, con = get_connection(name)
data = PNInM1(con.read(api, slot, subslot, idx=PNInM1.IDX).payload)
if request.method == 'POST':
function = request.values.get('function')
location = request.values.get('location')
inm1 = PNInM1(data.block_header, bytes(function, "utf-8"), bytes(location, "utf-8"))
con.write(api, slot, subslot, PNInM1.IDX, inm1)
return redirect(url_for("inm1", name=name, api=api, slot=slot, subslot=subslot))
else:
return render_template("inm1.html", name=name, api=api, slot=slot, subslot=subslot, data=data)
@app.route("/rename")
def rename():
mac = request.values.get('mac')
name = request.values.get('name')
old_name = dcp.get_param(s, src, mac, "name").decode()
# delete connection
if old_name in conns.keys():
del conns[old_name]
dcp.set_param(s, src, mac, "name", name)
return redirect(url_for("device", name=name))
app.run(debug=True)