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
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
|
from ctypes import CDLL, POINTER, c_void_p, cast, pointer
from typing import List, Optional
from eduvpn_common.types import (
cDiscoveryOrganization,
cDiscoveryOrganizations,
cDiscoveryServer,
cDiscoveryServers,
get_ptr_list_strings,
)
class DiscoOrganization:
def __init__(self, display_name: str, org_id: str, secure_internet_home: str, keyword_list: List[str]):
self.display_name = display_name
self.org_id = org_id
self.secure_internet_home = secure_internet_home
self.keyword_list = keyword_list
def __str__(self):
return self.display_name
class DiscoOrganizations:
def __init__(self, version: int, organizations: List[DiscoOrganization]):
self.version = version
self.organizations = organizations
class DiscoServer:
def __init__(
self,
authentication_url_template: str,
base_url: str,
country_code: str,
display_name: str,
keyword_list: List[str],
public_keys: List[str],
server_type: str,
support_contacts: List[str],
):
self.authentication_url_template = authentication_url_template
self.base_url = base_url
self.country_code = country_code
self.display_name = display_name
self.keyword_list = keyword_list
self.public_keys = public_keys
self.server_type = server_type
self.support_contacts = support_contacts
def __str__(self):
return self.display_name
class DiscoServers:
def __init__(self, version, servers):
self.version = version
self.servers = servers
def get_disco_organization(ptr) -> Optional[DiscoOrganization]:
if not ptr:
return None
current_organization = ptr.contents
display_name = current_organization.display_name.decode("utf-8")
org_id = current_organization.org_id.decode("utf-8")
secure_internet_home = current_organization.secure_internet_home.decode("utf-8")
keyword_list = current_organization.keyword_list.decode("utf-8")
return DiscoOrganization(display_name, org_id, secure_internet_home, keyword_list)
def get_disco_server(lib: CDLL, ptr) -> Optional[DiscoServer]:
if not ptr:
return None
current_server = ptr.contents
authentication_url_template = current_server.authentication_url_template.decode(
"utf-8"
)
base_url = current_server.base_url.decode("utf-8")
country_code = current_server.country_code.decode("utf-8")
display_name = current_server.display_name.decode("utf-8")
keyword_list = current_server.keyword_list.decode("utf-8")
public_keys = get_ptr_list_strings(
lib, current_server.public_key_list, current_server.total_public_keys
)
server_type = current_server.server_type.decode("utf-8")
support_contacts = get_ptr_list_strings(
lib, current_server.support_contact, current_server.total_support_contact
)
return DiscoServer(
authentication_url_template,
base_url,
country_code,
display_name,
keyword_list,
public_keys,
server_type,
support_contacts,
)
def get_disco_servers(lib: CDLL, ptr: c_void_p) -> Optional[DiscoServers]:
if ptr:
svrs = cast(ptr, POINTER(cDiscoveryServers)).contents
servers = []
if svrs.servers:
for i in range(svrs.total_servers):
current = get_disco_server(lib, svrs.servers[i])
if current is None:
continue
servers.append(current)
lib.FreeDiscoServers(ptr)
return DiscoServers(svrs.version, servers)
return None
def get_disco_organizations(lib: CDLL, ptr: c_void_p) -> Optional[DiscoOrganizations]:
if ptr:
orgs = cast(ptr, POINTER(cDiscoveryOrganizations)).contents
organizations = []
if orgs.organizations:
for i in range(orgs.total_organizations):
current = get_disco_organization(orgs.organizations[i])
if current is None:
continue
organizations.append(current)
lib.FreeDiscoOrganizations(ptr)
return DiscoOrganizations(orgs.version, organizations)
return None
|
