You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
2.6 KiB
97 lines
2.6 KiB
"""Описание изолированного стенда; здесь нет сетевых команд и привилегий.""" |
|
|
|
from dataclasses import asdict, dataclass |
|
|
|
|
|
@dataclass(frozen=True) |
|
class Country: |
|
name: str |
|
subnet: str |
|
router: str |
|
gateway: str |
|
|
|
|
|
@dataclass(frozen=True) |
|
class Node: |
|
number: int |
|
country: str |
|
ip: str |
|
gateway: str |
|
nat: str | None |
|
client_type: int |
|
udp_port: int = 32000 |
|
|
|
@property |
|
def name(self): |
|
return f"n{self.number:02d}" |
|
|
|
@property |
|
def mapped_port(self): |
|
return 23000 + self.number |
|
|
|
|
|
@dataclass(frozen=True) |
|
class Nat: |
|
name: str |
|
country: str |
|
ip: str |
|
mode: str |
|
|
|
|
|
@dataclass(frozen=True) |
|
class Wan: |
|
a: str |
|
b: str |
|
ip_a: str |
|
ip_b: str |
|
rtt_ms: int |
|
jitter_ms: float |
|
loss_percent: float |
|
mbps: int |
|
|
|
|
|
COUNTRIES = ( |
|
Country("a", "192.0.2.0/24", "ra", "192.0.2.1"), |
|
Country("b", "198.51.100.0/24", "rb", "198.51.100.1"), |
|
Country("c", "203.0.113.0/24", "rc", "203.0.113.1"), |
|
) |
|
NATS = tuple( |
|
Nat(f"g_{c.name}{i}", c.name, c.gateway.rsplit(".", 1)[0] + f".{100 + i}", mode) |
|
for c in COUNTRIES |
|
for i, mode in ((1, "restricted" if c.name == "c" else "cone"), (2, "strict")) |
|
) |
|
WANS = ( |
|
Wan("a", "b", "198.18.0.1", "198.18.0.2", 50, 2, 0.2, 50), |
|
Wan("a", "c", "198.18.0.5", "198.18.0.6", 100, 5, 0.5, 30), |
|
Wan("b", "c", "198.18.0.9", "198.18.0.10", 150, 10, 1.0, 20), |
|
) |
|
|
|
|
|
def nodes(): |
|
result = [] |
|
number = 1 |
|
for country, private_count in zip(COUNTRIES, (4, 3, 3)): |
|
prefix = country.gateway.rsplit(".", 1)[0] |
|
for public in range(2): |
|
result.append(Node(number, country.name, f"{prefix}.{11 + public}", country.gateway, None, 0 if not public else 1)) |
|
number += 1 |
|
for private in range(private_count): |
|
nat = f"g_{country.name}{1 + private // 2}" |
|
result.append(Node(number, country.name, f"10.64.1.{2 + private % 2}", "10.64.1.1", nat, 2 if number == 16 else 1)) |
|
number += 1 |
|
return tuple(result) |
|
|
|
|
|
NODES = nodes() |
|
|
|
|
|
def wan(a, b): |
|
return next(link for link in WANS if {link.a, link.b} == {a, b}) |
|
|
|
|
|
def manifest(seed): |
|
return {"version": 1, "seed": seed, "countries": [asdict(c) for c in COUNTRIES], |
|
"nodes": [dict(asdict(n), name=n.name, mapped_port=n.mapped_port) for n in NODES], |
|
"nat": [asdict(n) for n in NATS], "wan": [asdict(w) for w in WANS], |
|
"clock": "CLOCK_MONOTONIC, no time namespace", "impairment": "receiver ingress IFB", |
|
"private_subnet": "10.64.1.0/24 (reused independently by every NAT)"}
|
|
|