shaba/openuds
1
0
Fork 1
mirror of https://github.com/dkmstr/openuds.git synced 2024-12-22 13:34:04 +03:00
Code Issues Releases Wiki Activity

upgrading samples

Browse Source
This commit is contained in:
Adolfo Gómez García 2023-03-28 22:57:55 +02:00
parent 01fb43d51b
commit e48237f6d4
No known key found for this signature in database
GPG Key ID: DD1ABF20724CDA23
5 changed files with 6 additions and 170 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tunnel-server/samples/echo_client_ssl.py
View File

@ -1,16 +1,14 @@
#!/usr/bin/env python3
# -*- coding=utf-8 -*-
from functools import WRAPPER_ASSIGNMENTS
import ssl
import asyncio
import logging
import typing
import certifi # In order to get valid ca certificates
if typing.TYPE_CHECKING:
from asyncio.streams import StreamReader, StreamWriter
pass
async def tcp_echo_client():

50
tunnel-server/samples/echo_server.py
View File

@ -1,50 +0,0 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Copyright (c) 2020 Virtual Cable S.L.U.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without modification,
# are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions and the following disclaimer in the documentation
# and/or other materials provided with the distribution.
# * Neither the name of Virtual Cable S.L. nor the names of its contributors
# may be used to endorse or promote products derived from this software
# without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
'''
@author: Adolfo Gómez, dkmaster at dkmon dot com
'''
import curio
BUFFER_SIZE = 1024
async def echo_server(client, address) -> None:
print(f'Connection fro {address!r}')
while True:
data = await client.recv(BUFFER_SIZE)
if not data:
break
print(f'received {data}')
await client.sendall(data)
print('Closed')
if __name__ == '__main__':
curio.run(curio.tcp_server, 'localhost', 7777, echo_server)

117
tunnel-server/samples/echo_server_ssl.py
View File

@ -1,117 +0,0 @@
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Copyright (c) 2020 Virtual Cable S.L.U.
# All rights reserved.
#
# Redistribution and use in source and binary forms, with or without modification,
# are permitted provided that the following conditions are met:
#
# * Redistributions of source code must retain the above copyright notice,
# this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
# this list of conditions and the following disclaimer in the documentation
# and/or other materials provided with the distribution.
# * Neither the name of Virtual Cable S.L. nor the names of its contributors
# may be used to endorse or promote products derived from this software
# without specific prior written permission.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE
# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR
# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,
# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
'''
@author: Adolfo Gómez, dkmaster at dkmon dot com
'''
import multiprocessing
import typing
import socket
import curio
import curio.ssl
import curio.io
BUFFER_SIZE = 1024
if typing.TYPE_CHECKING:
from multiprocessing.connection import Connection
def get_socket(pipe: 'Connection') -> typing.Any:
sock, address = pipe.recv()
print(f'Sock: {sock}, f{address}')
return (sock, address)
async def echo_server_async(pipe: 'Connection'):
async def run_server(pipe: 'Connection', group: curio.TaskGroup) -> None:
while True:
sock, address = await curio.run_in_thread(get_socket, pipe)
await group.spawn(echo_server, sock, address)
del sock
async with curio.TaskGroup() as tg:
await tg.spawn(run_server, pipe, tg)
# Reap all of the children tasks as they complete
async for task in tg:
print(f'Deleting {task!r}')
task.joined = True
del task
async def echo_server(iclient, address) -> None:
print(f'Connection from {address!r}')
context = curio.ssl.SSLContext(curio.ssl.PROTOCOL_TLS_SERVER)
context.load_cert_chain('testing.pem', 'testing.key')
context.set_ciphers('ECDHE-RSA-AES256-GCM-SHA512:DHE-RSA-AES256-GCM-SHA512:ECDHE-RSA-AES256-GCM-SHA384:DHE-RSA-AES256-GCM-SHA384:ECDHE-RSA-AES256-SHA384')
client = await context.wrap_socket(curio.io.Socket(iclient), server_side=True)
while True:
data = await client.recv(BUFFER_SIZE)
if not data:
break
print(f'received {data}')
await client.sendall(data)
print('Closed')
def main():
own_conn, child_conn = multiprocessing.Pipe()
task = multiprocessing.Process(target=curio.run, args=(echo_server_async, child_conn,))
task.start()
host, port = 'fake.udsenterprise.com', 7777
backlog = 100
sock = None
try:
# Wait for socket incoming connections and spread them
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, True)
try:
sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, True)
except (AttributeError, OSError) as e:
# log.warning('reuse_port=True option failed', exc_info=True)
pass
sock.bind((host, port))
sock.listen(backlog)
while True:
print('Waiting...')
client, addr = sock.accept()
print('Sending...')
own_conn.send((client, addr))
except Exception:
pass
if sock:
sock.close()
if __name__ == "__main__":
main()

4
tunnel-server/src/uds_tunnel/stats.py
View File

@ -105,6 +105,10 @@ class StatsManager:
self.sent += size
self.update()
def decrement_connections(self):
self.ns.current -= 1
self.ns.total -= 1
@property
def as_sent_counter(self) -> 'StatsSingleCounter':
return StatsSingleCounter(self, False)

1
tunnel-server/src/uds_tunnel/tunnel.py
View File

@ -164,6 +164,7 @@ class TunnelProtocol(asyncio.Protocol):
self.runner = self.do_proxy
def process_stats(self, full: bool) -> None:
self.stats_manager.decrement_connections() # This connection does not count, it's just "stats"
# if pasword is not already received, wait for it
if len(self.cmd) < consts.PASSWORD_LENGTH + consts.COMMAND_LENGTH:
return