diff options
Diffstat (limited to 'dwt.py')
| -rw-r--r-- | dwt.py | 426 |
1 files changed, 426 insertions, 0 deletions
@@ -0,0 +1,426 @@ +# Copyright (C) 10se1ucgo 2015-2016 +# +# This file is part of DisableWinTracking. +# +# DisableWinTracking 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. +# +# DisableWinTracking 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 DisableWinTracking. If not, see <http://www.gnu.org/licenses/>. +import os +import logging +import sys +import io +import platform +import traceback +from ctypes import windll + +import wx +from wx.lib.itemspicker import ItemsPicker, IP_SORT_SELECTED, IP_SORT_CHOICES, IP_REMOVE_FROM_CHOICES + +import dwt_about +import dwt_util + + +class RedirectText(io.StringIO): + def __init__(self, console, old_stdout): + super(RedirectText, self).__init__() + + self.out = console + self.old_out = old_stdout + + def write(self, string): + # Oh my god this is the DUMBEST THING I've ever done. (Keeping a reference to the old stdout) + self.old_out.write(string) + self.out.WriteText(string) + + +class ConsoleDialog(wx.Dialog): + def __init__(self, old_stdout): + wx.Dialog.__init__(self, parent=wx.GetApp().TopWindow, title="Console Output", size=(500, 200), + style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER) + + console_box = wx.TextCtrl(self, style=wx.TE_MULTILINE | wx.TE_READONLY) + sys.stdout = RedirectText(console_box, old_stdout) + + top_sizer = wx.BoxSizer(wx.VERTICAL) + console_sizer = wx.BoxSizer(wx.VERTICAL) + button_sizer = self.CreateButtonSizer(wx.OK | wx.CANCEL) + + report_button = wx.FindWindowById(wx.ID_CANCEL, self) + report_button.SetLabel("Report an issue") + + console_sizer.Add(console_box, 1, wx.LEFT | wx.RIGHT | wx.EXPAND | wx.ALIGN_TOP, 5) + + top_sizer.Add(console_sizer, 1, wx.ALL | wx.EXPAND, 5) + top_sizer.Add(button_sizer, 0, wx.ALL | wx.ALIGN_LEFT, 5) + + self.Bind(wx.EVT_CLOSE, handler=lambda x: sys.exit(0)) + self.Bind(wx.EVT_BUTTON, handler=lambda x: sys.exit(0), id=wx.ID_OK) + self.Bind(wx.EVT_BUTTON, source=report_button, handler=self.submit_issue) + self.SetSizer(top_sizer) + + def submit_issue(self, event): + os.startfile("https://github.com/10se1ucgo/DisableWinTracking/issues/new") + + +class MainFrame(wx.Frame): + def __init__(self): + super(MainFrame, self).__init__(parent=wx.GetApp().GetTopWindow(), title="Disable Windows 10 Tracking", + size=(415, 450)) + self.SetMinSize(self.GetSize()) + panel = MainPanel(self) + + file_menu = wx.Menu() + settings = file_menu.Append(wx.ID_SETUP, "&Settings", "DWT settings") + + help_menu = wx.Menu() + about = help_menu.Append(wx.ID_ABOUT, "&About", "About DWT") + licenses = help_menu.Append(wx.ID_ANY, "&Licenses", "Open-source licenses") + + menu_bar = wx.MenuBar() + menu_bar.Append(file_menu, "&File") + menu_bar.Append(help_menu, "&Help") + self.SetMenuBar(menu_bar) + + if bool(windll.advpack.IsNTAdmin(0, None)): + warn = wx.MessageDialog(parent=self, + message="Program requires elevation, please run it as an administrator.", + caption="ERROR!", style=wx.OK | wx.ICON_WARNING) + warn.ShowModal() + warn.Destroy() + sys.exit(1) + + self.SetIcon(wx.Icon(sys.executable, wx.BITMAP_TYPE_ICO)) + self.Bind(wx.EVT_MENU, lambda x: dwt_about.about_dialog(self), about) + self.Bind(wx.EVT_MENU, panel.settings, settings) + self.Bind(wx.EVT_MENU, lambda x: dwt_about.Licenses(self), licenses) + + +class MainPanel(wx.Panel): + def __init__(self, parent): + super(MainPanel, self).__init__(parent) + + self.parent = parent + self.picked_normal = [] + self.picked_extra = [] + self.picked_ips = [] + + self.service_check = wx.CheckBox(self, label="Services") + self.service_check.SetToolTip("Disables or deletes tracking services. Choose option in 'Services Method'") + + self.diagtrack_check = wx.CheckBox(self, label="Clear DiagTrack log") + self.diagtrack_check.SetToolTip("Clears Dianostic Tracking log and prevents modification to it. " + "Cannot be undone automatically.") + + # Telemetry checkbox + self.telemetry_check = wx.CheckBox(self, label="Telemetry") + self.telemetry_check.SetToolTip("Sets 'AllowTelemetry' to 0. " + "On non-Enterprise OS editions, requires HOSTS file modification.") + + # HOSTS file checkbox + self.host_check = wx.CheckBox(self, label="Block tracking domains") + self.host_check.SetToolTip("Adds known tracking domains to HOSTS file. Required to disable Telemetry") + + # Extra HOSTS checkbox + self.extra_host_check = wx.CheckBox(self, label="Block even more tracking domains") + self.extra_host_check.SetToolTip("For the paranoid. Adds extra domains to the HOSTS file.\n" + "May cause issues with Skype, Dr. Watson, Hotmail and/or Error Reporting.") + + # IP block checkbox + self.ip_check = wx.CheckBox(self, label="Block tracking IP addresses") + self.ip_check.SetToolTip("Blocks known tracking IP addresses with Windows Firewall.") + + # Windows Privacy Regs (Policy Manager) + self.defender_check = wx.CheckBox(self, label="Windows Defender collection") + self.defender_check.SetToolTip("Modifies registry to prevent Defender collection") + + # WifiSense checkbox + self.wifisense_check = wx.CheckBox(self, label="WifiSense") + + # OneDrive uninstall checkbox + self.onedrive_check = wx.CheckBox(self, label="Uninstall OneDrive") + self.onedrive_check.SetToolTip("Uninstalls OneDrive from your computer and removes it from Explorer.") + + self.service_rad = wx.RadioBox(self, label="Service Method", choices=("Disable", "Delete")) + self.service_rad.SetItemToolTip(item=0, text="Simply disables the services. This can be undone.") + self.service_rad.SetItemToolTip(item=1, text="Deletes the services completely. This can't be undone.") + + self.mode_rad = wx.RadioBox(self, label="Mode", choices=("Privacy", "Revert")) + self.mode_rad.SetItemToolTip(item=0, text="Applies the selected settings.") + self.mode_rad.SetItemToolTip(item=1, text="Reverts the selected settings.") + + go_button = wx.Button(self, label="Go!") + + self.app_box = wx.StaticBoxSizer(wx.VERTICAL, self, "Built-in Apps") + stat_box = self.app_box.GetStaticBox + top_app_sizer = wx.BoxSizer(wx.HORIZONTAL) + button_app_sizer = wx.BoxSizer(wx.HORIZONTAL) + left_app_sizer = wx.BoxSizer(wx.VERTICAL) + middle_app_sizer = wx.BoxSizer(wx.VERTICAL) + right_app_sizer = wx.BoxSizer(wx.VERTICAL) + + # wx.CheckBox(app_box.GetStaticBox(), label="Name", name="search_name") + wx.CheckBox(stat_box(), label="3D Builder", name="3dbuilder") + wx.CheckBox(stat_box(), label="Calender and Mail", name="windowscommunicationsapps") + wx.CheckBox(stat_box(), label="Camera", name="windowscamera") + wx.CheckBox(stat_box(), label="Get Office App", name="officehub") + wx.CheckBox(stat_box(), label="Get Skype App", name="skypeapp") + wx.CheckBox(stat_box(), label="Get Started App", name="getstarted") + wx.CheckBox(stat_box(), label="Groove Music", name="zunemusic") + wx.CheckBox(stat_box(), label="Maps", name="windowsmaps") + wx.CheckBox(stat_box(), label="Solitaire Collection", name="solitairecollection") + wx.CheckBox(stat_box(), label="Money", name="bingfinance") + wx.CheckBox(stat_box(), label="Movies && TV", name="zunevideo") + wx.CheckBox(stat_box(), label="News", name="bingnews") + wx.CheckBox(stat_box(), label="OneNote App", name="onenote") + wx.CheckBox(stat_box(), label="People", name="people") + wx.CheckBox(stat_box(), label="Phone Companion", name="windowsphone") + wx.CheckBox(stat_box(), label="Photos", name="photos") + wx.CheckBox(stat_box(), label="Sports", name="bingsports") + wx.CheckBox(stat_box(), label="Voice Recorder", name="soundrecorder") + wx.CheckBox(stat_box(), label="Weather", name="bingweather") + wx.CheckBox(stat_box(), label="Xbox", name="xboxapp") + remove_app_button = wx.Button(stat_box(), label="Remove selected apps") + select_all_check = wx.CheckBox(stat_box(), label="Select all") + + sorted_list = sorted(stat_box().GetChildren(), key=lambda x: x.GetLabel()) + for index, item in enumerate([x for x in sorted_list if isinstance(x, wx.CheckBox) and x != select_all_check]): + n = len(sorted_list) // 3 + if index <= n: + left_app_sizer.Add(item, 1, wx.ALL, 1) + elif index <= n * 2: + middle_app_sizer.Add(item, 1, wx.ALL, 1) + else: + right_app_sizer.Add(item, 1, wx.ALL, 1) + + top_app_sizer.Add(left_app_sizer, 1, wx.ALL, 1) + top_app_sizer.Add(middle_app_sizer, 1, wx.ALL, 1) + top_app_sizer.Add(right_app_sizer, 1, wx.ALL, 1) + button_app_sizer.Add(remove_app_button, 0, wx.ALL, 1) + button_app_sizer.Add(select_all_check, 1, wx.ALL, 5) + self.app_box.Add(top_app_sizer) + self.app_box.Add(button_app_sizer) + + top_sizer = wx.BoxSizer(wx.VERTICAL) + top_row_sizer = wx.BoxSizer(wx.HORIZONTAL) + check_sizer = wx.BoxSizer(wx.VERTICAL) + rad_sizer = wx.BoxSizer(wx.VERTICAL) + + top_sizer.Add(top_row_sizer, 0, wx.ALL, 5) + top_sizer.Add(self.app_box, 0, wx.ALL, 5) + top_row_sizer.Add(check_sizer, 0, wx.ALL) + top_row_sizer.Add(rad_sizer, 0, wx.ALL) + rad_sizer.Add(self.service_rad, 0, wx.ALL, 10) + rad_sizer.Add(go_button, 0, wx.ALL ^ wx.BOTTOM | wx.ALIGN_CENTER, 10) + rad_sizer.Add(self.mode_rad, 0, wx.ALL, 10) + check_sizer.Add(self.service_check, 0, wx.ALL, 1) + check_sizer.Add(self.diagtrack_check, 0, wx.ALL, 1) + check_sizer.Add(self.telemetry_check, 0, wx.ALL, 1) + check_sizer.Add(self.host_check, 0, wx.ALL, 1) + check_sizer.Add(self.extra_host_check, 0, wx.ALL, 1) + check_sizer.Add(self.ip_check, 0, wx.ALL, 1) + check_sizer.Add(self.defender_check, 0, wx.ALL, 1) + check_sizer.Add(self.wifisense_check, 0, wx.ALL, 1) + check_sizer.Add(self.onedrive_check, 0, wx.ALL, 1) + + self.Bind(wx.EVT_CHECKBOX, handler=self.select_all_apps, source=select_all_check) + self.Bind(wx.EVT_BUTTON, handler=self.remove_apps, source=remove_app_button) + self.Bind(wx.EVT_BUTTON, handler=self.go, source=go_button) + + self.SetSizer(top_sizer) + print(self.GetSize()) + + def select_all_apps(self, event): + # Iters through all children of the wxStaticBox of the wxStaticBoxSizer and checks/un checks all wxCheckBoxes. + for child in self.app_box.GetStaticBox().GetChildren(): + if isinstance(child, wx.CheckBox): + child.SetValue(event.IsChecked()) + + def go(self, event): + if not all((self.picked_ips, self.picked_extra, self.picked_normal)): + self.settings(event=None) + + undo = bool(self.mode_rad.GetSelection()) + + if self.ip_check.IsChecked(): + dwt_util.ip_block(self.picked_ips, undo=undo) + if self.diagtrack_check.IsChecked(): + dwt_util.clear_diagtrack() + if self.service_check.IsChecked(): + if self.service_rad.GetSize(): + dwt_util.delete_service("dmwappushsvc") + dwt_util.delete_service("DiagTrack") + else: + dwt_util.disable_service("dmwappushsvc") + dwt_util.disable_service("DiagTrack") + dwt_util.services(undo=undo) + if self.telemetry_check.IsChecked(): + dwt_util.telemetry(undo=undo) + if self.host_check.IsChecked(): + dwt_util.host_file(self.picked_normal, undo=undo) + if self.extra_host_check.IsChecked(): + dwt_util.host_file(self.picked_extra, undo=undo) + if self.defender_check.IsChecked(): + dwt_util.defender(undo=undo) + if self.wifisense_check.IsChecked(): + dwt_util.wifisense(undo=undo) + if self.onedrive_check.IsChecked(): + dwt_util.onedrive(undo=undo) + logger.info("Done. It's recommended that you reboot as soon as possible for the full effect.") + logger.info(("If you feel something didn't work properly, please press the 'Report an issue" + "button and follow the directions")) + console.Center() + console.Show() + + def remove_apps(self, event): + children = [child for child in self.app_box.GetStaticBox().GetChildren() if child.GetName() != "check"] + app_list = [child.GetName() for child in children if isinstance(child, wx.CheckBox) and child.IsChecked()] + dwt_util.app_manager(app_list, undo=False) + + def settings(self, event): + dialog = wx.Dialog(parent=self, title="Settings", style=wx.DEFAULT_DIALOG_STYLE | wx.RESIZE_BORDER) + sizer = wx.BoxSizer(wx.VERTICAL) + + normal_domains = ( + 'a-0001.a-msedge.net', 'a-0002.a-msedge.net', 'a-0003.a-msedge.net', 'a-0004.a-msedge.net', + 'a-0005.a-msedge.net', 'a-0006.a-msedge.net', 'a-0007.a-msedge.net', 'a-0008.a-msedge.net', + 'a-0009.a-msedge.net', 'a-msedge.net', 'a.ads1.msn.com', 'a.ads2.msads.net', 'a.ads2.msn.com', + 'a.rad.msn.com', 'ac3.msn.com', 'ad.doubleclick.net', 'adnexus.net', 'adnxs.com', 'ads.msn.com', + 'ads1.msads.net', 'ads1.msn.com', 'aidps.atdmt.com', 'aka-cdn-ns.adtech.de', + 'az361816.vo.msecnd.net', 'az512334.vo.msecnd.net', 'b.ads1.msn.com', 'b.ads2.msads.net', + 'b.rad.msn.com', 'bs.serving-sys.com', 'c.atdmt.com', 'c.msn.com', 'cdn.atdmt.com', + 'cds26.ams9.msecn.net', 'choice.microsoft.com', 'choice.microsoft.com.nsatc.net', + 'compatexchange.cloudapp.net', 'corp.sts.microsoft.com', 'corpext.msitadfs.glbdns2.microsoft.com', + 'cs1.wpc.v0cdn.net', 'db3aqu.atdmt.com', 'df.telemetry.microsoft.com', + 'diagnostics.support.microsoft.com', 'ec.atdmt.com', 'feedback.microsoft-hohm.com', + 'feedback.search.microsoft.com', 'feedback.windows.com', 'flex.msn.com', 'g.msn.com', 'h1.msn.com', + 'i1.services.social.microsoft.com', 'i1.services.social.microsoft.com.nsatc.net', + 'lb1.www.ms.akadns.net', 'live.rads.msn.com', 'm.adnxs.com', 'msedge.net', 'msftncsi.com', + 'msnbot-65-55-108-23.search.msn.com', 'msntest.serving-sys.com', 'oca.telemetry.microsoft.com', + 'oca.telemetry.microsoft.com.nsatc.net', 'pre.footprintpredict.com', 'preview.msn.com', + 'rad.live.com', 'rad.msn.com', 'redir.metaservices.microsoft.com', + 'reports.wes.df.telemetry.microsoft.com', 'schemas.microsoft.akadns.net', + 'secure.adnxs.com', 'secure.flashtalking.com', 'services.wes.df.telemetry.microsoft.com', + 'settings-sandbox.data.microsoft.com', 'settings-win.data.microsoft.com', + 'sls.update.microsoft.com.akadns.net', 'sqm.df.telemetry.microsoft.com', + 'sqm.telemetry.microsoft.com', 'sqm.telemetry.microsoft.com.nsatc.net', 'ssw.live.com', + 'static.2mdn.net', 'statsfe1.ws.microsoft.com', 'statsfe2.ws.microsoft.com', + 'telecommand.telemetry.microsoft.com', 'telecommand.telemetry.microsoft.com.nsatc.net', + 'telemetry.appex.bing.net', 'telemetry.microsoft.com', 'telemetry.urs.microsoft.com', + 'v10.vortex-win.data.microsoft.com', 'vortex-bn2.metron.live.com.nsatc.net', + 'vortex-cy2.metron.live.com.nsatc.net', 'vortex-sandbox.data.microsoft.com', + 'vortex-win.data.metron.live.com.nsatc.net', 'vortex-win.data.microsoft.com', + 'vortex.data.glbdns2.microsoft.com', 'vortex.data.microsoft.com', 'watson.live.com', + 'web.vortex.data.microsoft.com', 'www.msftncsi.com' + ) + + extra_domains = ( + 'fe2.update.microsoft.com.akadns.net', 's0.2mdn.net', 'statsfe2.update.microsoft.com.akadns.net', + 'survey.watson.microsoft.com', 'view.atdmt.com', 'watson.microsoft.com', + 'watson.ppe.telemetry.microsoft.com', 'watson.telemetry.microsoft.com', + 'watson.telemetry.microsoft.com.nsatc.net', 'wes.df.telemetry.microsoft.com', 'ui.skype.com', + 'pricelist.skype.com', 'apps.skype.com', 'm.hotmail.com', 's.gateway.messenger.live.com' + ) + + ip_addresses = ( + '2.22.61.43', '2.22.61.66', '65.39.117.230', '65.55.108.23', '23.218.212.69', '134.170.30.202', + '137.116.81.24', '157.56.106.189', '204.79.197.200', '65.52.108.33', '64.4.54.254' + ) + + normal_domain_picker = ItemsPicker(dialog, choices=[], selectedLabel="Domains to be blocked", + ipStyle=IP_SORT_SELECTED | IP_SORT_CHOICES | IP_REMOVE_FROM_CHOICES) + if self.picked_normal: + normal_domain_picker.SetSelections(self.picked_normal) + normal_domain_picker.SetItems([domain for domain in normal_domains if domain not in self.picked_normal]) + else: + normal_domain_picker.SetSelections(normal_domains) + + extra_domain_picker = ItemsPicker(dialog, choices=[], selectedLabel="Extra domains to be blocked", + ipStyle=IP_SORT_SELECTED | IP_SORT_CHOICES | IP_REMOVE_FROM_CHOICES) + if self.picked_normal: + extra_domain_picker.SetSelections(self.picked_extra) + extra_domain_picker.SetItems([domain for domain in extra_domains if domain not in self.picked_extra]) + else: + extra_domain_picker.SetSelections(extra_domains) + + ip_picker = ItemsPicker(dialog, choices=[], selectedLabel="IP addresses to be blocked", + ipStyle=IP_SORT_SELECTED | IP_SORT_CHOICES | IP_REMOVE_FROM_CHOICES) + if self.picked_ips: + ip_picker.SetSelections(self.picked_ips) + ip_picker.SetItems([ip for ip in ip_addresses if ip not in self.picked_ips]) + else: + ip_picker.SetSelections(ip_addresses) + + sizer.Add(normal_domain_picker, 0, wx.EXPAND) + sizer.Add(extra_domain_picker, 0, wx.EXPAND) + sizer.Add(ip_picker, 0, wx.EXPAND) + if event is not None: + dialog.SetSizerAndFit(sizer) + dialog.Center() + dialog.ShowModal() + dialog.Destroy() + self.picked_normal = normal_domain_picker.GetSelections() + self.picked_extra = extra_domain_picker.GetSelections() + self.picked_ips = ip_picker.GetSelections() + + +def setup_logging(): + global logger + logger = logging.getLogger('dwt') + logger.setLevel(logging.DEBUG) + + formatter = logging.Formatter(fmt='%(asctime)s %(levelname)s: %(message)s', datefmt='%H:%M:%S') + + stdout_log = logging.StreamHandler(sys.stdout) + stdout_log.setLevel(logging.DEBUG) + stdout_log.setFormatter(formatter) + logger.addHandler(stdout_log) + + try: + file_log = logging.FileHandler(filename='dwt.log') + file_log.setLevel(logging.DEBUG) + file_log.setFormatter(formatter) + logger.addHandler(file_log) + except (OSError, IOError): + error_dialog = wx.MessageDialog(parent=wx.GetApp().GetTopWindow(), + message="Could not create log file, errors will not be recorded!", + caption="ERROR!", style=wx.OK | wx.ICON_ERROR) + error_dialog.ShowModal() + error_dialog.Destroy() + logger.exception("Could not create log file.") + + logger.info("Python {version} on {platform}".format(version=sys.version, platform=sys.platform)) + logger.info(platform.uname()) + logger.info("DisableWinTracking version {v}".format(v=dwt_about.__version__)) + + +def exception_hook(error, value, trace): + error_message = ''.join(traceback.format_exception(error, value, trace)) + logger.critical(error_message) + error_dialog = wx.MessageDialog(parent=wx.GetApp().GetTopWindow(), + message="An error has occured!\n\n" + error_message, + caption="ERROR!", style=wx.OK | wx.CANCEL | wx.ICON_ERROR) + error_dialog.SetOKCancelLabels("Ignore", "Quit") + if error_dialog.ShowModal() == wx.ID_OK: + error_dialog.Destroy() + else: + error_dialog.Destroy() + sys.exit(1) + + +if __name__ == '__main__': + wx_app = wx.App() + frame = MainFrame() + console = ConsoleDialog(sys.stdout) + setup_logging() + sys.excepthook = exception_hook + frame.Show() + wx_app.MainLoop() |