class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "python.ico") self.systray = SysTrayIcon(icon_path, "XX-Net", self.make_menu(), self.on_quit, left_click=self.on_show) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) def make_menu(self): menu_options = (("Config", None, self.on_show), ("Check update", None, self.on_check_update), ("Enable Proxy", None, self.on_enable_proxy), ("Disable Proxy", None, self.on_disable_proxy), ("restart goagent", None, self.on_restart_goagent)) return menu_options def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_goagent(self, widget=None, data=None): module_init.stop_all() module_init.start_all_auto() def on_check_update(self, widget=None, data=None): update.check_update() def set_register(self, reg_path, name, reg_type, value): #_, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, name) winreg.SetValueEx(reg_path, name, 0, reg_type, value) logging.info("set register path:%r name:%s type:%d value:%s", reg_path, name, reg_type, value) def on_enable_proxy(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 1) self.set_register(self.INTERNET_SETTINGS, 'ProxyOverride', 1, '*.local;<local>') # Bypass the proxy for localhost self.set_register(self.INTERNET_SETTINGS, 'ProxyServer', 1, '127.0.0.1:8087') def on_disable_proxy(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 0) def show_control_web(self, widget=None, data=None): webbrowser.open("http://127.0.0.1:8085/") def on_quit(self, widget, data=None): pass def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "python.ico") self.systray = SysTrayIcon(icon_path, "XX-Net", self.make_menu(), self.on_quit, left_click=self.on_show) def make_menu(self): menu_options = (("Show", None, self.on_show), ("Check update", None, self.on_check_update), ("restart goagent", None, self.on_restart_goagent)) return menu_options def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_goagent(self, widget=None, data=None): module_init.stop() module_init.start() def on_check_update(self, widget=None, data=None): update.check_update() def show_control_web(self, widget=None, data=None): webbrowser.open("http://127.0.0.1:8085/") def on_quit(self, widget, data=None): pass def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "web_ui", "favicon.ico") self.systray = SysTrayIcon(icon_path, "XX-Net", self.make_menu(), self.on_quit, left_click=self.on_show, right_click=self.on_right_click) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) proxy_setting = config.get(["modules", "launcher", "proxy"], "pac") if proxy_setting == "pac": self.on_enable_pac() elif proxy_setting == "gae": self.on_enable_gae_proxy() elif proxy_setting == "disable": # Don't disable proxy setting, just do nothing. pass else: xlog.warn("proxy_setting:%r", proxy_setting) def get_proxy_state(self): REG_PATH = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER,REG_PATH,0, winreg.KEY_ALL_ACCESS) try: AutoConfigURL, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, 'AutoConfigURL') if AutoConfigURL: if AutoConfigURL == "http://127.0.0.1:8086/proxy.pac": return "pac" else: return "unknown" except Exception as e: pass try: ProxyEnable, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, 'ProxyEnable') if ProxyEnable: ProxyServer, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, 'ProxyServer') if ProxyServer == "127.0.0.1:8087": return "gae" else: return "unknown" except Exception as e: pass return "disable" def on_right_click(self): self.systray.update(menu=self.make_menu()) self.systray._show_menu() def make_menu(self): import locale lang_code, code_page = locale.getdefaultlocale() proxy_stat = self.get_proxy_state() gae_proxy_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat=="gae" else 0 pac_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat=="pac" else 0 disable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat=="disable" else 0 if lang_code == "zh_CN": menu_options = ((u"设置", None, self.on_show, 0), (u"全局通过GAEProxy代理", None, self.on_enable_gae_proxy, gae_proxy_checked), (u"全局PAC智能代理", None, self.on_enable_pac, pac_checked), (u"取消全局代理", None, self.on_disable_proxy, disable_checked), (u"重启 GAEProxy", None, self.on_restart_gae_proxy, 0)) else: menu_options = ((u"Config", None, self.on_show, 0), (u"Set Global GAEProxy Proxy", None, self.on_enable_gae_proxy, gae_proxy_checked), (u"Set Global PAC Proxy", None, self.on_enable_pac, pac_checked), (u"Disable Global Proxy", None, self.on_disable_proxy, disable_checked), (u"Reset GAEProxy", None, self.on_restart_gae_proxy, 0)) return menu_options def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_gae_proxy(self, widget=None, data=None): module_init.stop_all() module_init.start_all_auto() def on_check_update(self, widget=None, data=None): update.check_update() def on_enable_gae_proxy(self, widget=None, data=None): win32_proxy_manager.set_proxy_server("127.0.0.1", 8087) config.set(["modules", "launcher", "proxy"], "gae") config.save() def on_enable_pac(self, widget=None, data=None): win32_proxy_manager.set_proxy_auto("http://127.0.0.1:8086/proxy.pac") config.set(["modules", "launcher", "proxy"], "pac") config.save() def on_disable_proxy(self, widget=None, data=None): win32_proxy_manager.disable_proxy() config.set(["modules", "launcher", "proxy"], "disable") config.save() def show_control_web(self, widget=None, data=None): host_port = config.get(["modules", "launcher", "control_port"], 8085) webbrowser.open("http://127.0.0.1:%s/" % host_port) ctypes.windll.user32.ShowWindow(ctypes.windll.kernel32.GetConsoleWindow(), 0) def on_quit(self, widget, data=None): proxy_setting = config.get(["modules", "launcher", "proxy"], "disable") if proxy_setting != "disable": win32_proxy_manager.disable_proxy() module_init.stop_all() nid = win32_adapter.NotifyData(self.systray._hwnd, 0) win32_adapter.Shell_NotifyIcon(2, ctypes.byref(nid)) os._exit(0) def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "web_ui", "favicon.ico") self.systray = SysTrayIcon(icon_path, "OSS-FTP", self.make_menu(), self.on_quit, left_click=self.on_show, right_click=self.on_right_click) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) def get_proxy_state(self): REG_PATH = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, REG_PATH, 0, winreg.KEY_ALL_ACCESS) try: AutoConfigURL, reg_type = winreg.QueryValueEx( INTERNET_SETTINGS, 'AutoConfigURL') if AutoConfigURL: return "auto" except Exception as e: pass try: ProxyEnable, reg_type = winreg.QueryValueEx( INTERNET_SETTINGS, 'ProxyEnable') if ProxyEnable: return "enable" except Exception as e: pass return "disable" def on_right_click(self): self.systray.update(menu=self.make_menu()) self.systray._show_menu() def make_menu(self): import locale lang_code, code_page = locale.getdefaultlocale() proxy_stat = self.get_proxy_state() enable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "enable" else 0 auto_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "auto" else 0 if lang_code == "zh_CN": menu_options = ((u"设置", None, self.on_show, 0), (u"重启 OSS Ftp 代理服务器", None, self.on_restart_ossftp_proxy, 0)) else: menu_options = ((u"Config", None, self.on_show, 0), (u"Restart OSS Ftp Proxy", None, self.on_restart_ossftp_proxy, 0)) return menu_options def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_ossftp_proxy(self, widget=None, data=None): module_init.stop_all() module_init.start_all_auto() def on_check_update(self, widget=None, data=None): update.check_update() def show_control_web(self, widget=None, data=None): webbrowser.open("http://127.0.0.1:8192/") ctypes.windll.user32.ShowWindow( ctypes.windll.kernel32.GetConsoleWindow(), 0) def on_quit(self, widget, data=None): win32_proxy_manager.disable_proxy() def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "web_ui", "favicon.ico") self.systray = SysTrayIcon(icon_path, "XX-Net", self.make_menu(), self.on_quit, left_click=self.on_show, right_click=self.on_right_click) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) def get_proxy_state(self): REG_PATH = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, REG_PATH, 0, winreg.KEY_ALL_ACCESS) try: AutoConfigURL, reg_type = winreg.QueryValueEx( INTERNET_SETTINGS, 'AutoConfigURL') if AutoConfigURL: return "auto" ProxyEnable, reg_type = winreg.QueryValueEx( INTERNET_SETTINGS, 'ProxyEnable') if ProxyEnable: return "enable" except Exception as e: pass return "disable" def on_right_click(self): self.systray.update(menu=self.make_menu()) self.systray._show_menu() def make_menu(self): import locale lang_code, code_page = locale.getdefaultlocale() proxy_stat = self.get_proxy_state() enable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "enable" else 0 auto_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "auto" else 0 disable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "disable" else 0 if lang_code == "zh_CN": menu_options = ((u"设置", None, self.on_show, 0), (u"全局通过GoAgent代理", None, self.on_enable_proxy, enable_checked), (u"全局PAC智能代理", None, self.on_enable_pac, auto_checked), (u"取消全局代理", None, self.on_disable_proxy, disable_checked), (u"重启 GoAgent", None, self.on_restart_goagent, 0)) else: menu_options = ((u"Config", None, self.on_show, 0), (u"Set Global GoAgent Proxy", None, self.on_enable_proxy, enable_checked), (u"Set Global PAC Proxy", None, self.on_enable_pac, auto_checked), (u"Disable Global Proxy", None, self.on_disable_proxy, disable_checked), (u"Reset GoAgent", None, self.on_restart_goagent, 0)) return menu_options def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_goagent(self, widget=None, data=None): module_init.stop_all() module_init.start_all_auto() def on_check_update(self, widget=None, data=None): update.check_update() def set_register(self, reg_path, name, reg_type, value): #_, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, name) winreg.SetValueEx(reg_path, name, 0, reg_type, value) logging.info("set register path:%r name:%s type:%d value:%s", reg_path, name, reg_type, value) def on_enable_proxy(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'AutoConfigURL', 1, "") # disable auto proxy self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 1) self.set_register(self.INTERNET_SETTINGS, 'ProxyOverride', 1, '*.local;<local>') # Bypass the proxy for localhost self.set_register(self.INTERNET_SETTINGS, 'ProxyServer', 1, '127.0.0.1:8087') def on_enable_pac(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 0) # disable goagent proxy self.set_register(self.INTERNET_SETTINGS, 'AutoConfigURL', 1, "http://127.0.0.1:8086/proxy.pac") def on_disable_proxy(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 0) # disable goagent proxy self.set_register(self.INTERNET_SETTINGS, 'AutoConfigURL', 1, "") # disable auto proxy def show_control_web(self, widget=None, data=None): webbrowser.open("http://127.0.0.1:8085/") ctypes.windll.user32.ShowWindow( ctypes.windll.kernel32.GetConsoleWindow(), 0) def on_quit(self, widget, data=None): pass def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray: def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "icon", "favicon.ico") self.systray = SysTrayIcon(icon_path, "XX-Net", self.make_menu(), self.on_quit, left_click=self.on_show, right_click=self.on_right_click) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) # proxy_setting = config.get(["modules", "launcher", "proxy"], "smart_router") # if proxy_setting == "pac": # self.on_enable_pac() # elif proxy_setting == "gae": # self.on_enable_gae_proxy() # elif proxy_setting == "x_tunnel": # self.on_enable_x_tunnel() # elif proxy_setting == "smart_router": # self.on_enable_smart_router() # elif proxy_setting == "disable": # # Don't disable proxy setting, just do nothing. # pass # else: # xlog.warn("proxy_setting:%r", proxy_setting) def get_proxy_state(self): try: AutoConfigURL, reg_type = winreg.QueryValueEx( self.INTERNET_SETTINGS, 'AutoConfigURL') if AutoConfigURL: if AutoConfigURL == "https://www.txthinking.com/pac/white.pac": return "pac" else: return "unknown" except: pass try: ProxyEnable, reg_type = winreg.QueryValueEx( self.INTERNET_SETTINGS, 'ProxyEnable') if ProxyEnable: ProxyServer, reg_type = winreg.QueryValueEx( self.INTERNET_SETTINGS, 'ProxyServer') if ProxyServer == "127.0.0.1:8087": return "gae" if ProxyServer == "127.0.0.1:1080": return "Socks5" else: return "unknown" except: pass return "disable" def on_right_click(self): self.systray.update(menu=self.make_menu()) self.systray._show_menu() def make_menu(self): global menu_options proxy_stat = self.get_proxy_state() proxy_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "disable" else 0 pac_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "pac" else 0 # x_tunnel_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "x_tunnel" else 0 # smart_router_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "smart_router" else 0 # pac_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "pac" else 0 # disable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "disable" else 0 if lang_code == "zh_CN": menu_options = [(u"设置(待完善)", None, self.on_show, 0)] # if config.get(["modules", "gae_proxy", "auto_start"], 0) == 1: # menu_options.append((u"全局通过GAEProxy代理", None, self.on_enable_gae_proxy, win32_adapter.fState.MFS_CHECKED)) # if config.get(["modules", "x_tunnel", "auto_start"], 0) == 1: # menu_options.append((u"全局通过X-Tunnel代理", None, self.on_enable_x_tunnel, 0)) # if config.get(["modules", "smart_router", "auto_start"], 0) == 1: # menu_options.append((u"全局通过智能路由代理", None, self.on_enable_smart_router, win32_adapter.fState.MFS_CHECKED)) menu_options += [ (u"全局PAC智能代理", None, self.on_enable_pac, pac_checked), (u"取消全局代理", None, self.on_disable_proxy, proxy_checked), (u'退出', None, SysTrayIcon.QUIT, False) ] return tuple(menu_options) def on_show(self, widget=None, data=None): self.show_control_web() def on_enable_gae_proxy(self, widget=None, data=None): win32_proxy_manager.set_proxy("127.0.0.1:8087") def on_enable_pac(self, widget=None, data=None): curr_path = os.path.dirname(os.path.abspath(__file__)) try: cmd = r'""%s\brook_windows_386.exe" systemproxy -u https://www.txthinking.com/pac/white.pac"' % curr_path os.system(cmd) proxy_stat = self.get_proxy_state() if proxy_stat != "pac": logging.error("failed to enable pac!") except Exception: logging.error('on_enable_pac:%s' % traceback.format_exc()) # win32_proxy_manager.set_proxy("https://www.txthinking.com/pac/white.pac") # config.set(["modules", "launcher", "proxy"], "pac") # config.save() def on_disable_proxy(self, widget=None, data=None): curr_path = os.path.dirname(os.path.abspath(__file__)) try: cmd = r'""%s\brook_windows_386.exe" systemproxy -r"' % curr_path os.system(cmd) proxy_stat = self.get_proxy_state() if proxy_stat != "disable": logging.error("failed to disable pac!") except Exception: logging.error("on_disable_proxy:%s" % traceback.format_exc()) # win32_proxy_manager.set_proxy("") # win32_proxy_manager.disable_proxy() # config.set(["modules", "launcher", "proxy"], "disable") # config.save() def show_control_web(self, widget=None, data=None): # host_port = config.get(["modules", "launcher", "control_port"], 8085) host_port = 8085 webbrowser.open("http://127.0.0.1:%s/" % host_port) ctypes.windll.user32.ShowWindow( ctypes.windll.kernel32.GetConsoleWindow(), 0) def on_quit(self, widget, data=None): # proxy_setting = config.get(["modules", "launcher", "proxy"], "disable") # if proxy_setting != "disable": # win32_proxy_manager.disable_proxy() # module_init.stop_all() cmd = 'taskkill /F /IM brook_windows_386.exe' os.system(cmd) nid = win32_adapter.NotifyData(self.systray._hwnd, 0) win32_adapter.Shell_NotifyIcon(2, ctypes.byref(nid)) os._exit(0) def serve_forever(self): logging.info("\n\n[step3]:enter win_tray") self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "web_ui", "favicon.ico") self.systray = SysTrayIcon(icon_path, "XX-Net", self.make_menu(), self.on_quit, left_click=self.on_show, right_click=self.on_right_click) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) proxy_setting = config.os_proxy_mode if proxy_setting == "pac": self.on_enable_pac() elif proxy_setting == "gae": self.on_enable_gae_proxy() elif proxy_setting == "x_tunnel": self.on_enable_x_tunnel() elif proxy_setting == "smart_router": self.on_enable_smart_router() elif proxy_setting == "disable": # Don't disable proxy setting, just do nothing. pass else: xlog.warn("proxy_setting:%r", proxy_setting) def get_proxy_state(self): try: AutoConfigURL, reg_type = winreg.QueryValueEx(self.INTERNET_SETTINGS, 'AutoConfigURL') if AutoConfigURL: if AutoConfigURL == "http://127.0.0.1:8086/proxy.pac": return "pac" else: return "unknown" except: pass try: ProxyEnable, reg_type = winreg.QueryValueEx(self.INTERNET_SETTINGS, 'ProxyEnable') if ProxyEnable: ProxyServer, reg_type = winreg.QueryValueEx(self.INTERNET_SETTINGS, 'ProxyServer') if ProxyServer == "127.0.0.1:8087": return "gae" if ProxyServer == "127.0.0.1:1080": return "x_tunnel" if ProxyServer == "127.0.0.1:8086": return "smart_router" else: return "unknown" except: pass return "disable" def on_right_click(self): self.systray.update(menu=self.make_menu()) self.systray._show_menu() def make_menu(self): proxy_stat = self.get_proxy_state() gae_proxy_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "gae" else 0 x_tunnel_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "x_tunnel" else 0 smart_router_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "smart_router" else 0 pac_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "pac" else 0 disable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "disable" else 0 if lang_code == "zh_CN": menu_options = [("设置", None, self.on_show, 0)] if config.enable_gae_proxy == 1: menu_options.append(("全局通过GAEProxy代理", None, self.on_enable_gae_proxy, gae_proxy_checked)) if config.enable_x_tunnel == 1: menu_options.append(("全局通过X-Tunnel代理", None, self.on_enable_x_tunnel, x_tunnel_checked)) if config.enable_smart_router == 1: menu_options.append(("全局通过智能路由代理", None, self.on_enable_smart_router, smart_router_checked)) menu_options += [ ("全局PAC智能代理", None, self.on_enable_pac, pac_checked), ("取消全局代理", None, self.on_disable_proxy, disable_checked), ("重启各模块", None, self.on_restart_each_module, 0), ('退出', None, SysTrayIcon.QUIT, False)] else: menu_options = [("Config", None, self.on_show, 0)] if config.enable_gae_proxy == 1: menu_options.append(("Set Global GAEProxy Proxy", None, self.on_enable_gae_proxy, gae_proxy_checked)) if config.enable_x_tunnel == 1: menu_options.append(("Set Global X-Tunnel Proxy", None, self.on_enable_x_tunnel, x_tunnel_checked)) if config.enable_smart_router == 1: menu_options.append(("Set Global Smart-Router Proxy", None, self.on_enable_smart_router, smart_router_checked)) menu_options += [ ("Set Global PAC Proxy", None, self.on_enable_pac, pac_checked), ("Disable Global Proxy", None, self.on_disable_proxy, disable_checked), ("Reset Each module", None, self.on_restart_each_module, 0), ('Quit', None, SysTrayIcon.QUIT, False)] return tuple(menu_options) def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_each_module(self, widget=None, data=None): module_init.stop_all() module_init.start_all_auto() def on_check_update(self, widget=None, data=None): update.check_update() def on_enable_gae_proxy(self, widget=None, data=None): win32_proxy_manager.set_proxy("127.0.0.1:8087") config.os_proxy_mode = "gae" config.save() def on_enable_x_tunnel(self, widget=None, data=None): win32_proxy_manager.set_proxy("127.0.0.1:1080") config.os_proxy_mode = "x_tunnel" config.save() def on_enable_smart_router(self, widget=None, data=None): win32_proxy_manager.set_proxy("127.0.0.1:8086") config.os_proxy_mode = "smart_router" config.save() def on_enable_pac(self, widget=None, data=None): win32_proxy_manager.set_proxy("http://127.0.0.1:8086/proxy.pac") config.os_proxy_mode = "pac" config.save() def on_disable_proxy(self, widget=None, data=None): win32_proxy_manager.disable_proxy() config.os_proxy_mode = "disable" config.save() def show_control_web(self, widget=None, data=None): host_port = config.control_port webbrowser.open("http://127.0.0.1:%s/" % host_port) ctypes.windll.user32.ShowWindow(ctypes.windll.kernel32.GetConsoleWindow(), 0) def on_quit(self, widget, data=None): if self.get_proxy_state() != "unknown": win32_proxy_manager.disable_proxy() module_init.stop_all() nid = win32_adapter.NotifyData(self.systray._hwnd, 0) win32_adapter.Shell_NotifyIcon(2, ctypes.byref(nid)) os._exit(0) def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "web_ui", "favicon.ico") self.systray = SysTrayIcon(icon_path, "OSS-FTP", self.make_menu(), self.on_quit, left_click=self.on_show, right_click=self.on_right_click) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) def get_proxy_state(self): REG_PATH = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER,REG_PATH,0, winreg.KEY_ALL_ACCESS) try: AutoConfigURL, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, 'AutoConfigURL') if AutoConfigURL: return "auto" except Exception as e: pass try: ProxyEnable, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, 'ProxyEnable') if ProxyEnable: return "enable" except Exception as e: pass return "disable" def on_right_click(self): self.systray.update(menu=self.make_menu()) self.systray._show_menu() def make_menu(self): import locale lang_code, code_page = locale.getdefaultlocale() proxy_stat = self.get_proxy_state() enable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat=="enable" else 0 auto_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat=="auto" else 0 if lang_code == "zh_CN": menu_options = ((u"设置", None, self.on_show, 0), (u"重启 OSS Ftp 代理服务器", None, self.on_restart_ossftp_proxy, 0)) else: menu_options = ((u"Config", None, self.on_show, 0), (u"Restart OSS Ftp Proxy", None, self.on_restart_ossftp_proxy, 0)) return menu_options def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_ossftp_proxy(self, widget=None, data=None): module_init.stop_all() module_init.start_all_auto() def on_check_update(self, widget=None, data=None): update.check_update() def show_control_web(self, widget=None, data=None): webbrowser.open("http://127.0.0.1:8192/") ctypes.windll.user32.ShowWindow(ctypes.windll.kernel32.GetConsoleWindow(), 0) def on_quit(self, widget, data=None): win32_proxy_manager.disable_proxy() def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "web_ui", "favicon.ico") self.systray = SysTrayIcon(icon_path, "XX-Net", self.make_menu(), self.on_quit, left_click=self.on_show) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) def make_menu(self): import locale lang_code, code_page = locale.getdefaultlocale() if lang_code == "zh_CN": menu_options = ((u"设置", None, self.on_show), (u"全局通过GoAgent代理上网", None, self.on_enable_proxy), (u"全局PAC智能代理切换", None, self.on_enable_pac), (u"取消全局代理", None, self.on_disable_proxy), (u"重启 GoAgent", None, self.on_restart_goagent)) else: menu_options = ((u"Config", None, self.on_show), (u"Set Global GoAgent Proxy", None, self.on_enable_proxy), (u"Set Global PAC Proxy", None, self.on_enable_pac), (u"Disable Global Proxy", None, self.on_disable_proxy), (u"Reset GoAgent", None, self.on_restart_goagent)) return menu_options def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_goagent(self, widget=None, data=None): module_init.stop_all() module_init.start_all_auto() def on_check_update(self, widget=None, data=None): update.check_update() def set_register(self, reg_path, name, reg_type, value): #_, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, name) winreg.SetValueEx(reg_path, name, 0, reg_type, value) logging.info("set register path:%r name:%s type:%d value:%s", reg_path, name, reg_type, value) def on_enable_proxy(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'AutoConfigURL', 1, "") # disable auto proxy self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 1) self.set_register(self.INTERNET_SETTINGS, 'ProxyOverride', 1, '*.local;<local>') # Bypass the proxy for localhost self.set_register(self.INTERNET_SETTINGS, 'ProxyServer', 1, '127.0.0.1:8087') def on_enable_pac(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 0) # disable goagent proxy self.set_register(self.INTERNET_SETTINGS, 'AutoConfigURL', 1, "http://127.0.0.1:8086/proxy.pac") def on_disable_proxy(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 0) # disable goagent proxy self.set_register(self.INTERNET_SETTINGS, 'AutoConfigURL', 1, "") # disable auto proxy def show_control_web(self, widget=None, data=None): webbrowser.open("http://127.0.0.1:8085/") def on_quit(self, widget, data=None): pass def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "web_ui", "favicon.ico") self.systray = SysTrayIcon(icon_path, "XX-Net", self.make_menu(), self.on_quit, left_click=self.on_show, right_click=self.on_right_click) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) proxy_setting = config.get(["modules", "launcher", "proxy"], "pac") if proxy_setting == "pac": self.on_enable_pac() elif proxy_setting == "gae": self.on_enable_gae_proxy() elif proxy_setting == "disable": # Don't disable proxy setting, just do nothing. pass else: xlog.warn("proxy_setting:%r", proxy_setting) def get_proxy_state(self): REG_PATH = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, REG_PATH, 0, winreg.KEY_ALL_ACCESS) try: AutoConfigURL, reg_type = winreg.QueryValueEx( INTERNET_SETTINGS, 'AutoConfigURL') if AutoConfigURL: if AutoConfigURL == "http://127.0.0.1:8086/proxy.pac": return "pac" else: return "unknown" except Exception as e: pass try: ProxyEnable, reg_type = winreg.QueryValueEx( INTERNET_SETTINGS, 'ProxyEnable') if ProxyEnable: ProxyServer, reg_type = winreg.QueryValueEx( INTERNET_SETTINGS, 'ProxyServer') if ProxyServer == "127.0.0.1:8087": return "gae" else: return "unknown" except Exception as e: pass return "disable" def on_right_click(self): self.systray.update(menu=self.make_menu()) self.systray._show_menu() def make_menu(self): import locale lang_code, code_page = locale.getdefaultlocale() proxy_stat = self.get_proxy_state() gae_proxy_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "gae" else 0 pac_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "pac" else 0 disable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat == "disable" else 0 if lang_code == "zh_CN": menu_options = ((u"设置", None, self.on_show, 0), (u"全局通过GAEProxy代理", None, self.on_enable_gae_proxy, gae_proxy_checked), (u"全局PAC智能代理", None, self.on_enable_pac, pac_checked), (u"取消全局代理", None, self.on_disable_proxy, disable_checked), (u"重启 GAEProxy", None, self.on_restart_gae_proxy, 0)) else: menu_options = ((u"Config", None, self.on_show, 0), (u"Set Global GAEProxy Proxy", None, self.on_enable_gae_proxy, gae_proxy_checked), (u"Set Global PAC Proxy", None, self.on_enable_pac, pac_checked), (u"Disable Global Proxy", None, self.on_disable_proxy, disable_checked), (u"Reset GAEProxy", None, self.on_restart_gae_proxy, 0)) return menu_options def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_gae_proxy(self, widget=None, data=None): module_init.stop_all() module_init.start_all_auto() def on_check_update(self, widget=None, data=None): update.check_update() def on_enable_gae_proxy(self, widget=None, data=None): win32_proxy_manager.set_proxy_server("127.0.0.1", 8087) config.set(["modules", "launcher", "proxy"], "gae") config.save() def on_enable_pac(self, widget=None, data=None): win32_proxy_manager.set_proxy_auto("http://127.0.0.1:8086/proxy.pac") config.set(["modules", "launcher", "proxy"], "pac") config.save() def on_disable_proxy(self, widget=None, data=None): win32_proxy_manager.disable_proxy() config.set(["modules", "launcher", "proxy"], "disable") config.save() def show_control_web(self, widget=None, data=None): host_port = config.get(["modules", "launcher", "control_port"], 8085) webbrowser.open("http://127.0.0.1:%s/" % host_port) ctypes.windll.user32.ShowWindow( ctypes.windll.kernel32.GetConsoleWindow(), 0) def on_quit(self, widget, data=None): proxy_setting = config.get(["modules", "launcher", "proxy"], "disable") if proxy_setting != "disable": win32_proxy_manager.disable_proxy() module_init.stop_all() nid = win32_adapter.NotifyData(self.systray._hwnd, 0) win32_adapter.Shell_NotifyIcon(2, ctypes.byref(nid)) os._exit(0) def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res
class Win_tray(): def __init__(self): icon_path = os.path.join(os.path.dirname(__file__), "web_ui", "favicon.ico") self.systray = SysTrayIcon(icon_path, "XX-Net", self.make_menu(), self.on_quit, left_click=self.on_show, right_click=self.on_right_click) reg_path = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' self.INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER, reg_path, 0, winreg.KEY_ALL_ACCESS) def get_proxy_state(self): REG_PATH = r'Software\Microsoft\Windows\CurrentVersion\Internet Settings' INTERNET_SETTINGS = winreg.OpenKey(winreg.HKEY_CURRENT_USER,REG_PATH,0, winreg.KEY_ALL_ACCESS) try: AutoConfigURL, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, 'AutoConfigURL') if AutoConfigURL: return "auto" ProxyEnable, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, 'ProxyEnable') if ProxyEnable: return "enable" except Exception as e: pass return "disable" def on_right_click(self): self.systray.update(menu=self.make_menu()) self.systray._show_menu() def make_menu(self): import locale lang_code, code_page = locale.getdefaultlocale() proxy_stat = self.get_proxy_state() enable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat=="enable" else 0 auto_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat=="auto" else 0 disable_checked = win32_adapter.fState.MFS_CHECKED if proxy_stat=="disable" else 0 if lang_code == "zh_CN": menu_options = ((u"设置", None, self.on_show, 0), (u"全局通过GAEProxy代理", None, self.on_enable_proxy, enable_checked), (u"全局PAC智能代理", None, self.on_enable_pac, auto_checked), (u"取消全局代理", None, self.on_disable_proxy, disable_checked), (u"重启 GAEProxy", None, self.on_restart_gae_proxy, 0)) else: menu_options = ((u"Config", None, self.on_show, 0), (u"Set Global GAEProxy Proxy", None, self.on_enable_proxy, enable_checked), (u"Set Global PAC Proxy", None, self.on_enable_pac, auto_checked), (u"Disable Global Proxy", None, self.on_disable_proxy, disable_checked), (u"Reset GAEProxy", None, self.on_restart_gae_proxy, 0)) return menu_options def on_show(self, widget=None, data=None): self.show_control_web() def on_restart_gae_proxy(self, widget=None, data=None): module_init.stop_all() module_init.start_all_auto() def on_check_update(self, widget=None, data=None): update.check_update() def set_register(self, reg_path, name, reg_type, value): #_, reg_type = winreg.QueryValueEx(INTERNET_SETTINGS, name) winreg.SetValueEx(reg_path, name, 0, reg_type, value) launcher_log.info("set register path:%r name:%s type:%d value:%s", reg_path, name, reg_type, value) def on_enable_proxy(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'AutoConfigURL', 1, "") # disable auto proxy self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 1) self.set_register(self.INTERNET_SETTINGS, 'ProxyOverride', 1, '*.local;<local>') # Bypass the proxy for localhost self.set_register(self.INTERNET_SETTINGS, 'ProxyServer', 1, '127.0.0.1:8087') def on_enable_pac(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 0) # disable gae_proxy proxy self.set_register(self.INTERNET_SETTINGS, 'AutoConfigURL', 1, "http://127.0.0.1:8086/proxy.pac") def on_disable_proxy(self, widget=None, data=None): self.set_register(self.INTERNET_SETTINGS, 'ProxyEnable', 4, 0) # disable gae_proxy proxy self.set_register(self.INTERNET_SETTINGS, 'AutoConfigURL', 1, "") # disable auto proxy def show_control_web(self, widget=None, data=None): webbrowser.open("http://127.0.0.1:8085/") ctypes.windll.user32.ShowWindow(ctypes.windll.kernel32.GetConsoleWindow(), 0) def on_quit(self, widget, data=None): pass def serve_forever(self): self.systray._message_loop_func() def dialog_yes_no(self, msg="msg", title="Title", data=None, callback=None): res = ctypes.windll.user32.MessageBoxW(None, msg, title, 1) # Yes:1 No:2 if callback: callback(data, res) return res