Jędrzej Stuczyński
56 lines
2.2 KiB
Python
56 lines
2.2 KiB
Python
import json
|
|
import os.path
|
|
import sys
|
|
|
|
|
|
def add_mixnode(base_network, base_dir, mix_id, port_delta):
|
|
with open(os.path.join(base_dir, "mix" + str(mix_id) + ".json"), "r") as json_blob:
|
|
mix_data = json.load(json_blob)
|
|
|
|
base_network["mixnodes"][str(mix_id)][0]["identity_key"] = mix_data["identity_key"]
|
|
base_network["mixnodes"][str(mix_id)][0]["sphinx_key"] = mix_data["sphinx_key"]
|
|
base_network["mixnodes"][str(mix_id)][0]["mix_port"] = 10000 + port_delta
|
|
base_network["mixnodes"][str(mix_id)][0]["version"] = mix_data["version"]
|
|
base_network["mixnodes"][str(mix_id)][0]["host"] = "127.0.0.1"
|
|
base_network["mixnodes"][str(mix_id)][0]["layer"] = mix_id % 3 + 1
|
|
base_network["mixnodes"][str(mix_id)][0]["mix_id"] = mix_id
|
|
base_network["mixnodes"][str(mix_id)][0]["owner"] = "n1jw6mp7d5xqc7w6xm79lha27glmd0vdt3l9artf"
|
|
return base_network
|
|
|
|
|
|
def add_gateway(base_network, base_dir, port_delta):
|
|
with open(os.path.join(base_dir, "gateway.json"), "r") as json_blob:
|
|
gateway_data = json.load(json_blob)
|
|
base_network["gateways"][0]["identity_key"] = gateway_data["identity_key"]
|
|
base_network["gateways"][0]["sphinx_key"] = gateway_data["sphinx_key"]
|
|
base_network["gateways"][0]["mix_port"] = 10000 + port_delta
|
|
base_network["gateways"][0]["clients_port"] = 9000
|
|
# base_network["gateways"][0]["version"] = gateway_data["version"]
|
|
base_network["gateways"][0]["host"] = "127.0.0.1"
|
|
base_network["gateways"][0]["owner"] = "n1jw6mp7d5xqc7w6xm79lha27glmd0vdt3l9artf"
|
|
return base_network
|
|
|
|
|
|
def main(args):
|
|
base_network = {
|
|
"mixnodes": {
|
|
"1": [{}],
|
|
"2": [{}],
|
|
"3": [{}],
|
|
},
|
|
"gateways": [{}]
|
|
}
|
|
|
|
base_dir = args[0]
|
|
base_network = add_mixnode(base_network, base_dir, 1, 1)
|
|
base_network = add_mixnode(base_network, base_dir, 2, 2)
|
|
base_network = add_mixnode(base_network, base_dir, 3, 3)
|
|
base_network = add_gateway(base_network, base_dir, 4)
|
|
|
|
with open(os.path.join(base_dir, "network.json"), "w") as out:
|
|
json.dump(base_network, out, indent=2)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main(sys.argv[1:])
|