FreeDATA/tools/send_file.py

117 lines
3.4 KiB
Python
Executable File

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
@author: DJ2LS
python3 send_file.py --file cleanup.sh --dxcallsign DN2LS-0 --mycallsign DN2LS-2 --attempts 3
"""
import argparse
import socket
import base64
import json
import uuid
import time
import crcengine
# --------------------------------------------GET PARAMETER INPUTS
parser = argparse.ArgumentParser(description='Simons TEST TNC')
parser.add_argument('--port', dest="socket_port", default=3000, help="Set socket listening port.", type=int)
parser.add_argument('--host', dest="socket_host", default='localhost', help="Set the host, the socket is listening on.", type=str)
parser.add_argument('--file', dest="filename", default='', help="Select the file we want to send", type=str)
parser.add_argument('--msg', dest="chatmessage", default='file from cli tool', help="Additional text message appended to file", type=str)
parser.add_argument('--dxcallsign', dest="dxcallsign", default='AA0AA', help="Select the destination callsign", type=str)
parser.add_argument('--mycallsign', dest="mycallsign", default='AA0AA', help="Select the own callsign", type=str)
parser.add_argument('--attempts', dest="attempts", default='5', help="Amount of connection attempts", type=int)
args = parser.parse_args()
HOST, PORT = args.socket_host, args.socket_port
filename = args.filename
dxcallsign = args.dxcallsign
mycallsign = args.mycallsign
attempts = args.attempts
chatmessage = bytes(args.chatmessage, "utf-8")
if filename != "":
# open file by name
f = open(filename, "rb")
file = f.read()
filename = bytes(filename, "utf-8")
else:
file = b""
filename = b""
# convert binary data to base64
#base64_data = base64.b64encode(file).decode("UTF-8")
split_char = b'\0;\1;'
filetype = b"unknown"
timestamp = str(int(time.time()))
# timestamp = timestamp.to_bytes(4, byteorder="big")
timestamp = bytes(timestamp, "utf-8")
msg_with_attachment = timestamp + \
split_char + \
chatmessage + \
split_char + \
filename + \
split_char + \
filetype + \
split_char + \
file
# calculate checksum
crc_algorithm = crcengine.new("crc32") # load crc32 library
crc_data = crc_algorithm(file)
crc_data = crc_data.to_bytes(4, byteorder="big")
datatype = b"m"
command = b"send_message"
checksum = bytes(crc_data.hex(), "utf-8")
uuid_4 = bytes(str(uuid.uuid4()), "utf-8")
data = datatype + \
split_char + \
command + \
split_char + \
checksum + \
split_char + \
uuid_4 + \
split_char + \
msg_with_attachment
data = base64.b64encode(data).decode("UTF-8")
# message
# our command we are going to send
command = {"type": "arq",
"command": "send_raw",
"parameter":
[{"dxcallsign": dxcallsign,
"mycallsign": mycallsign,
"attempts": str(attempts),
"mode": "255",
"n_frames": "1",
"data": data}
]
}
command = json.dumps(command)
print(command)
command = bytes(command + "\n", 'utf-8')
# Create a socket (SOCK_STREAM means a TCP socket)
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as sock:
# Connect to server and send data
sock.connect((HOST, PORT))
sock.sendall(command)
timeout = time.time() + 5
while time.time() < timeout:
pass