thegroundzero / openvasreporting_server Goto Github PK

def setup_socket(host, port, report_format):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.bind((host, int(port)))
        print('Server started on {}:{}'.format(host, str(port)))
        print('Will create reports in {} format'.format(report_format))

        s.listen(5)

        while True:
            conn, addr = s.accept()

            file = create_temp_file()
            print('Temp file creates at: {}'.format(file))

            with conn:
                print('Connected by: {}'.format(addr))
                while True:
                    data = conn.recv(1024)
                    if not data:
                        break

                    print('Received data, writing to temp file at: {}'.format(file))
                    write_to_file(file, data)

                    report = '{}/openvas_report_{}'.format(os.getcwd(), file)
                    print('Writing report to: {}.{}'.format(report, report_format))
                    config = openvasreporting.create_config([file], report, filetype=report_format)
                    openvasreporting.convert(config)

            print('Connection closed')
            os.remove(file)

def setup_socket(host, port, report_format):
    with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
        s.bind((host, int(port)))
        print('Server started on {}:{}'.format(host, str(port)))
        print('Will create reports in {} format'.format(report_format))

        s.listen(5)

        while True:
            conn, addr = s.accept()

            file = create_temp_file()
            print('Temp file creates at: {}'.format(file))

            with conn:
                print('Connected by: {}'.format(addr))
                buffer = 8192
                while True:
                    data = conn.recv(buffer)
                    if not data:
                        break
                    print('Received data, writing to temp file at: {}'.format(file))
                    write_to_file(file, data)

                    report = '{}/openvas_report_{}'.format(os.getcwd(), file)
                    print('Writing report to: {}.{}'.format(report, report_format))

            config = openvasreporting.create_config([file], report, filetype=report_format)
            openvasreporting.convert(config)

            print('Connection closed')
            os.remove(file)