Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 6c2a9253 authored by Tư Mã Tần Quảng's avatar Tư Mã Tần Quảng Committed by Nolen Johnson
Browse files

Add CLI command to restrict all network usage 

Reference: https://review.lineageos.org/c/324695


Signed-off-by: default avatarTư Mã Tần Quảng <baobaoxich@gmail.com>
Change-Id: I07dd734faca6116356fc05af7dff262bec65b4cb
parent 29908052

services/core/java/com/android/server/net/NetworkPolicyManagerShellCommand.java

+26 −0
Original line number Diff line number Diff line
@@ -18,6 +18,7 @@ package com.android.server.net; 


import static android.net.NetworkPolicyManager.POLICY_ALLOW_METERED_BACKGROUND;

import static android.net.NetworkPolicyManager.POLICY_NONE;

import static android.net.NetworkPolicyManager.POLICY_REJECT_ALL;

import static android.net.NetworkPolicyManager.POLICY_REJECT_METERED_BACKGROUND;

import static android.net.NetworkPolicyManager.POLICY_REJECT_WIFI;

import static android.net.NetworkPolicyManager.POLICY_REJECT_CELLULAR;
@@ -86,6 +87,8 @@ class NetworkPolicyManagerShellCommand extends ShellCommand { 
        pw.println("    Adds a UID to the whitelist for restrict background usage.");

        pw.println("  add restrict-background-blacklist UID");

        pw.println("    Adds a UID to the blacklist for restrict background usage.");

        pw.println("  add restrict-network-usage-blacklist UID");

        pw.println("    Adds a UID to the blacklist for restrict network usage.");

        pw.println("  add restrict-wifi-data-blacklist UID");

        pw.println("    Adds a UID to the blacklist for restrict Wi-Fi data usage.");

        pw.println("  add restrict-mobile-data-blacklist UID");
@@ -104,6 +107,8 @@ class NetworkPolicyManagerShellCommand extends ShellCommand { 
        pw.println("    Lists UIDs that are whitelisted for restrict background usage.");

        pw.println("  list restrict-background-blacklist");

        pw.println("    Lists UIDs that are blacklisted for restrict background usage.");

        pw.println("  list restrict-network-usage-blacklist");

        pw.println("    Lists UIDs that are blacklisted for restrict network usage.");

        pw.println("  list restrict-wifi-data-blacklist");

        pw.println("    Lists UIDs that are blacklisted for restrict Wi-Fi data usage.");

        pw.println("  list restrict-mobile-data-blacklist");
@@ -114,6 +119,8 @@ class NetworkPolicyManagerShellCommand extends ShellCommand { 
        pw.println("    Removes a UID from the whitelist for restrict background usage.");

        pw.println("  remove restrict-background-blacklist UID");

        pw.println("    Removes a UID from the blacklist for restrict background usage.");

        pw.println("  remove restrict-network-usage-blacklist UID");

        pw.println("    Removes a UID from the blacklist for restrict network usage.");

        pw.println("  remove restrict-wifi-data-blacklist UID");

        pw.println("    Removes a UID from the blacklist for restrict Wi-Fi data usage.");

        pw.println("  remove restrict-mobile-data-blacklist UID");
@@ -182,6 +189,8 @@ class NetworkPolicyManagerShellCommand extends ShellCommand { 
                return listRestrictBackgroundWhitelist();

            case "restrict-background-blacklist":

                return listRestrictBackgroundBlacklist();

            case "restrict-network-usage-blacklist":

                return listRestrictNetworkUsageBlacklist();

            case "restrict-wifi-data-blacklist":

                return listRestrictWiFiDataBlacklist();

            case "restrict-mobile-data-blacklist":
@@ -205,6 +214,8 @@ class NetworkPolicyManagerShellCommand extends ShellCommand { 
                return addRestrictBackgroundWhitelist();

            case "restrict-background-blacklist":

                return addRestrictBackgroundBlacklist();

            case "restrict-network-usage-blacklist":

                return addRestrictNetworkUsageBlacklist();

            case "restrict-wifi-data-blacklist":

                return addRestrictWiFiDataBlacklist();

            case "restrict-mobile-data-blacklist":
@@ -230,6 +241,8 @@ class NetworkPolicyManagerShellCommand extends ShellCommand { 
                return removeRestrictBackgroundWhitelist();

            case "restrict-background-blacklist":

                return removeRestrictBackgroundBlacklist();

            case "restrict-network-usage-blacklist":

                return removeRestrictNetworkUsageBlacklist();

            case "restrict-wifi-data-blacklist":

                return removeRestrictWiFiDataBlacklist();

            case "restrict-mobile-data-blacklist":
@@ -290,6 +303,11 @@ class NetworkPolicyManagerShellCommand extends ShellCommand { 
                POLICY_REJECT_METERED_BACKGROUND);

    }



    private int listRestrictNetworkUsageBlacklist() throws RemoteException {

        return listUidPolicies("Restrict network usage blacklisted UIDs",

                POLICY_REJECT_ALL);

    }



    private int listRestrictWiFiDataBlacklist() throws RemoteException {

        return listUidPolicies("Restrict Wi-Fi data blacklisted UIDs",

                POLICY_REJECT_WIFI);
@@ -377,6 +395,10 @@ class NetworkPolicyManagerShellCommand extends ShellCommand { 
        return setUidPolicy(POLICY_REJECT_METERED_BACKGROUND);

    }



    private int addRestrictNetworkUsageBlacklist() throws RemoteException {

        return setUidPolicy(POLICY_REJECT_ALL);

    }



    private int addRestrictWiFiDataBlacklist() throws RemoteException {

        return setUidPolicy(POLICY_REJECT_WIFI);

    }
@@ -393,6 +415,10 @@ class NetworkPolicyManagerShellCommand extends ShellCommand { 
        return resetUidPolicy("not blacklisted", POLICY_REJECT_METERED_BACKGROUND);

    }



    private int removeRestrictNetworkUsageBlacklist() throws RemoteException {

        return resetUidPolicy("not blacklisted", POLICY_REJECT_ALL);

    }



    private int removeRestrictWiFiDataBlacklist() throws RemoteException {

        return resetUidPolicy("not blacklisted", POLICY_REJECT_WIFI);

    }