sender.py

Go to the documentation of this file.

"""
@file sender.py
@brief Sender implementation for QUIC protocol.
@details Sets up a QUIC connection as a sender, creates streams,
         adds file data to the streams, and initiates sending.
"""
 
from constants import Constants
from quic import (QuicConnection)
import os
 
 
def set_file():
    """
    @brief Checks if a file exists at the current directory and creates it if necessary.
 
    @details .
    """
    if not os.path.exists(Constants.FILE_PATH):
        with open(Constants.FILE_PATH, 'wb') as file:
            file.write(b'I' * (Constants.FILE_SIZE * Constants.KILO))
 
 
def main():
    """
    @brief Main function that initializes and runs the sender.
 
    @details Creates a file if needed, sets up a QUIC connection,
             creates streams, adds file data, and starts sending.
    """
    set_file()
    quic_connection = QuicConnection(Constants.CONNECTION_ID_SENDER, Constants.ADDR_SENDER, Constants.ADDR_RECEIVER)
    # Add streams for the files
    streams = []
    for i in range(Constants.MAX_STREAMS):
        streams.append(quic_connection.get_stream(Constants.CONNECTION_ID_SENDER, Constants.UNIDI).get_stream_id())
    # Add files to the streams
    for stream in streams:
        quic_connection.add_file_to_stream(stream, Constants.FILE_PATH)
    # Start sending
    quic_connection.send_packets()
 
 
if __name__ == '__main__':
    main()