def handle(self, sock, addr):
logging.debug(u'connection from %s:%s' % addr)
# 调用各个 handler 尝试生成 client
# 检查结果，如果生成了就表示本次客户端请求的是这个协议。
# 剩下的就交由这个协议处理。
_handler = None
_reset_peek_offset = False
mysocket = MySocket(sock)
for Handler in handler.get_handler():
mysocket.reset_peek_offset()
(_handler, _reset_peek_offset) = Handler.create(mysocket, self)
if _handler is not None:
break
if _handler is None:
mysocket.close(safe=False)
logging.warn(u'[Handler]收到未识别的协议，关闭连接。')
return
try:
if _reset_peek_offset:
mysocket.reset_peek_offset()
mysocket.set_peek(False)
# handler 有可能是异步调用
_handler.handler()
except:
logging.exception(u'client.handle()')
mysocket.close(safe=False)
def handle(self, sock, addr):
logging.debug(u'connection from %s:%s' % addr)
# 调用各个 handler 尝试生成 client
# 检查结果，如果生成了就表示本次客户端请求的是这个协议。
# 剩下的就交由这个协议处理。
_handler = None
_reset_peek_offset = False
mysocket = MySocket(sock)
for Handler in handler.get_handler():
mysocket.reset_peek_offset()
(_handler, _reset_peek_offset) = Handler.create(mysocket, self)
if _handler is not None:
break
if _handler is None:
mysocket.close(safe=False)
logging.warn(u'[Handler]收到未识别的协议，关闭连接。')
return
try:
if _reset_peek_offset:
mysocket.reset_peek_offset()
mysocket.set_peek(False)
# handler 有可能是异步调用
_handler.handler()
except:
logging.exception(u'client.handle()')
mysocket.close(safe=False)
class Main(tk.Frame):
def __init__(self, master=None):
super().__init__(master)
self.master = master
self.is_connected = False
self.master.title("Client")
self.master.grid_rowconfigure(0, weight=1)
self.master.grid_columnconfigure(0, weight=1)
self.master.resizable(False, False)
self.grid()
self.socket = None
self.is_running = False
self.subroot = [None] * 6
self.create_widgets()
def create_widgets(self):
self.entry = tk.Entry(self, width=50)
self.entry.grid(row=0, column=0, columnspan=5, padx=10, pady=10)
self.contents = tk.StringVar(self)
self.contents.set("Nhập địa chỉ IP")
self.entry["textvariable"] = self.contents
self.Connect = tk.Button(self, text="Connect", width=10, height=1)
self.Connect.grid(row=0, column=5, sticky=tk.NE, padx=10, pady=10)
self.Connect['command'] = self.connect
self.Screenshot = tk.Button(self,
text="ScreenShot",
height=2,
width=10)
self.Screenshot.grid(row=1, column=0, sticky=tk.NW, padx=10, pady=10)
self.Screenshot["command"] = self.create_screenshot
self.Keystroke = tk.Button(self, text="KeyStroke", width=10, height=2)
self.Keystroke.grid(row=2, column=0, sticky=tk.NW, padx=10, pady=10)
self.Keystroke['command'] = self.create_keystroke
self.Process_Running = tk.Button(self,
text="Process Running",
width=10,
height=10,
wraplength=50)
self.Process_Running.grid(row=1, column=1, rowspan=2, padx=10, pady=10)
self.Process_Running["command"] = self.create_process_running
self.App_Running = tk.Button(self,
text="App Running",
width=10,
height=10,
wraplength=50)
self.App_Running.grid(row=1, column=2, rowspan=2, padx=10, pady=10)
self.App_Running["command"] = self.create_app_running
self.Registry = tk.Button(self, text='Registry', width=10, height=2)
self.Registry.grid(row=1, column=4, padx=10, pady=10, sticky=tk.NE)
self.Registry["command"] = self.create_registry_running
self.master.protocol("WM_DELETE_WINDOW", self.on_exit)
self.Shutdown = tk.Button(self, text="Shut down", width=10, height=2)
self.Shutdown.grid(row=2, column=4, pady=10, padx=10, sticky=tk.NE)
self.Shutdown["command"] = self.create_shutdown
def create_shutdown(self):
if not self.is_connected:
messagebox.showerror("FAIL TO ATTEMPT", "NOT CONNECTED YET")
return
self.socket.client.sendall(bytes("SHUTDOWN", "utf8"))
def create_registry_running(self):
if self.check():
return
if not self.is_connected:
messagebox.showerror("FAIL TO ATTEMPT", "NOT CONNECTED YET")
return
self.socket.client.sendall(bytes("REGISTRY", "utf8"))
registryroot = tk.Toplevel(self.master)
self.subroot[4] = Registry_UI(registryroot, self.socket)
self.subroot[4].mainloop()
def create_app_running(self):
if self.check():
return
if not self.is_connected:
messagebox.showerror("FAIL TO ATTEMPT", "NOT CONNECTED YET")
return
self.socket.client.sendall(bytes("APP", "utf8"))
approot = tk.Toplevel(self.master)
self.subroot[3] = Manager_ui(approot, self.socket, "App")
self.subroot[3].mainloop()
def create_process_running(self):
if self.check():
return
if not self.is_connected:
messagebox.showerror("FAIL TO ATTEMPT", "NOT CONNECTED YET")
return
self.socket.client.sendall(bytes("PROCESS", "utf8"))
processroot = tk.Toplevel(self.master)
self.subroot[2] = Manager_ui(processroot, self.socket, "Process")
self.subroot[2].mainloop()
def create_keystroke(self):
if self.check():
return
if not self.is_connected:
messagebox.showerror("FAIL TO ATTEMPT", "NOT CONNECTED YET")
return
self.socket.client.sendall(bytes("KEYSTROKE", "utf8"))
keystrokeroot = tk.Toplevel(self.master)
self.subroot[0] = keystroke_ui(keystrokeroot, self.socket)
self.subroot[0].mainloop()
def create_screenshot(self):
if self.check():
return
if not self.is_connected:
messagebox.showerror("FAIL TO ATTEMPT", "NOT CONNECTED YET")
return
self.socket.client.sendall(bytes("SCRSHOT", "utf8"))
screenshotroot = tk.Toplevel(self.master)
self.subroot[1] = scrshot_ui(screenshotroot, self.socket)
self.subroot[1].mainloop()
def connect(self):
strings = self.entry.get()
if (self.is_connected):
messagebox.showwarning("FAIL TO ATTEMPT", "Already Connected")
return
#print(strings)
try:
self.socket = MySocket((strings, PORT))
messagebox.showinfo("SUCCESS", "Connected")
self.is_connected = True
except:
messagebox.showerror("ERROR", "Can not connect")
def on_exit(self):
if self.check():
return
if self.socket is None:
self.master.destroy()
return
try:
self.socket.client.sendall(bytes("exit", "utf8"))
except:
dem = 1
self.socket.close()
self.master.destroy()
def check(self):
for i in range(6):
if self.subroot[i] is not None and self.subroot[i].winfo_exists():
return True
return False
