send.py

#!/usr/bin/env python3

"""
Copyright (c) 2015 Jacob Martin

A command for sending TauNet Messages.
"""

import sys
import os
import socket
from subprocess import call
from taunet import *

args = sys.argv

# Helpful usage hints.
if len(args) != 3:
	print("usage:")
	print("\tpython3 client.py <username> <message>")
	exit()

to_username = args[1]
message     = args[2]

# Fetch the target host from the client table using the username argument.
try:
	target_host = client_table.clients[to_username]
except KeyError:
	print("That client does not exist in the table.")
	exit()

# Construct the full message complete with headers.
full_message  = "version: " + version + "\r\n"
full_message += "from: " + client_table.username + "\r\n"
full_message += "to: " + to_username + "\r\n\r\n"
full_message += message

# Write the message to a file.
with open("message", "w") as messageFile:
	messageFile.write(full_message)

# Encrypt the message file then delete the original file.
call(["./cs2", "encrypt", client_table.key, "message", "encrypted"])
os.remove("message")

# Connect to the target.
target = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
target.connect((target_host, port))

# Send the encrypted file.
with open("encrypted", "rb") as encryptedFile:
	target.send(encryptedFile.read(1024))

# Close the connection and delete the encrypted file.
target.close()
os.remove("encrypted")