Attachment 1729265 Details for Bug 1638889 – Script for testing the forwarding

Script for testing the forwarding

pmsocat.py (text/plain), 3.42 KB, created by Martin Kletzander on 2020-11-14 01:33:28 UTC

(hide)

Description:

Filename:

MIME Type:

Creator: Martin Kletzander

Created: 2020-11-14 01:33:28 UTC

Size: 3.42 KB

patch

obsolete

>#!/usr/bin/env python3
>
>import argparse
>import asyncio
>import os
>import sys
>
>
>_args = None
>
>async def forward(reader, writer, prefix='='):
>    total = 0
>    while True:
>        data = await reader.read(_args.blocksize)
>        if not data:
>            break
>        new_tot = total + len(data)
>        print(f'{ prefix }  length={ len(data) } from={ total } to={ new_tot - 1}')
>        total = new_tot
>        if _args.output:
>            print(data.decode(encoding='ascii', errors='replace'))
>        writer.write(data)
>        await writer.drain()
>    print(f'{ prefix } close')
>    writer.close()
>
>async def handle_tcp_conn(reader1, writer1):
>    reader2, writer2 = await asyncio.open_unix_connection(_args.socket_path)
>    asyncio.gather(forward(reader1, writer2, '>'),
>                   forward(reader2, writer1, '<'))
>
>async def handle_unix_conn(reader1, writer1):
>    reader2, writer2 = await asyncio.open_connection(_args.host, _args.port)
>    asyncio.gather(forward(reader1, writer2, '>'),
>                   forward(reader2, writer1, '<'))
>
>async def tcp2unix():
>    return await asyncio.start_server(handle_tcp_conn, _args.address, _args.port)
>
>async def unix2tcp():
>    if _args.selinux_context:
>        import selinux
>        if selinux.setsockcreatecon(_args.selinux_context) != 0:
>            return 1
>    try:
>        os.unlink(_args.socket_path)
>    except FileNotFoundError:
>        pass
>    ret = await asyncio.start_unix_server(handle_unix_conn, _args.socket_path)
>    if _args.selinux_context:
>        selinux.setsockcreatecon(None)
>
>    return ret
>
>async def main():
>    parser = argparse.ArgumentParser(description="Poor Man's Socat, but with a SELinux feature")
>    parser.add_argument('-o', '--output',
>                        action='store_true',
>                        help='output forwarded data to stdout (WARNING: slows down everything)')
>    parser.add_argument('-b', '--blocksize',
>                        help='Size of forwarded chunks in Bytes (default 1MB)',
>                        default=2**20,
>                        type=int)
>
>    subparsers = parser.add_subparsers(title='functions',
>                                       dest='function',
>                                       required=True)
>
>    tcp2unix_sp = subparsers.add_parser('tcp2unix',
>                                        help='listen on network and connect to UNIX socket')
>    tcp2unix_sp.add_argument('-a', '--address',
>                             help='listen address (default "0.0.0.0")',
>                             default='0.0.0.0')
>    tcp2unix_sp.add_argument('port', help='port to listen on')
>    tcp2unix_sp.add_argument('socket_path', help='UNIX socket path to connect to')
>    tcp2unix_sp.set_defaults(func=tcp2unix)
>
>    unix2tcp_sp = subparsers.add_parser('unix2tcp',
>                                        help='listen on UNIX socket and connect to network')
>    unix2tcp_sp.add_argument('-c', '--selinux-context',
>                             help='SELinux context used for the listening UNIX socket path to listen on')
>    unix2tcp_sp.add_argument('socket_path', help='UNIX socket path to listen on')
>    unix2tcp_sp.add_argument('host', help='connect to this host/address')
>    unix2tcp_sp.add_argument('port', help='port to listen on')
>    unix2tcp_sp.set_defaults(func=unix2tcp)
>
>    global _args
>    _args = parser.parse_args()
>    server = await _args.func()
>
>    await server.serve_forever()
>
>if __name__ == '__main__':
>    sys.exit(asyncio.run(main(), debug=True))

Actions: View

Attachments on bug 1638889: 1729265 | 1729364 | 1739897