65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
import base64
|
|
import hashlib
|
|
|
|
from app.brm.brn import BRN
|
|
from app.extensions import db
|
|
from app.models import AbstractConfiguration, AbstractResource
|
|
|
|
|
|
class Onion(AbstractConfiguration):
|
|
@property
|
|
def brn(self) -> BRN:
|
|
return BRN(
|
|
group_id=self.group_id,
|
|
product="eotk",
|
|
provider="*",
|
|
resource_type="onion",
|
|
resource_id=self.onion_name
|
|
)
|
|
|
|
group_id = db.Column(db.Integer(), db.ForeignKey("group.id"), nullable=False)
|
|
domain_name = db.Column(db.String(255), nullable=False)
|
|
|
|
onion_public_key = db.Column(db.LargeBinary, nullable=False)
|
|
onion_private_key = db.Column(db.LargeBinary, nullable=False)
|
|
|
|
tls_public_key = db.Column(db.LargeBinary, nullable=False)
|
|
tls_private_key = db.Column(db.LargeBinary, nullable=False)
|
|
|
|
group = db.relationship("Group", back_populates="onions")
|
|
|
|
@property
|
|
def onion_name(self) -> str:
|
|
p = self.onion_public_key[32:]
|
|
|
|
h = hashlib.sha3_256()
|
|
h.update(b".onion checksum")
|
|
h.update(p)
|
|
h.update(b"\x03")
|
|
checksum = h.digest()
|
|
|
|
result = bytearray(p)
|
|
result.extend(checksum[0:2])
|
|
result.append(0x03)
|
|
|
|
onion = base64.b32encode(result).decode("utf-8").strip("=")
|
|
return onion.lower()
|
|
|
|
|
|
class Eotk(AbstractResource):
|
|
group_id = db.Column(db.Integer(), db.ForeignKey("group.id"), nullable=False)
|
|
instance_id = db.Column(db.String(100), nullable=True)
|
|
provider = db.Column(db.String(20), nullable=False)
|
|
region = db.Column(db.String(20), nullable=False)
|
|
|
|
group = db.relationship("Group", back_populates="eotks")
|
|
|
|
@property
|
|
def brn(self) -> BRN:
|
|
return BRN(
|
|
group_id=self.group_id,
|
|
provider=self.provider,
|
|
product="eotk",
|
|
resource_type="instance",
|
|
resource_id=self.region
|
|
)
|