content: /*
   * Copyright (C) 2011 The Android Open Source Project
   *
   * Licensed under the Apache License, Version 2.0 (the "License");
   * you may not use this file except in compliance with the License.
   * You may obtain a copy of the License at
   *
   *      http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package android.net;
  
  import android.content.Context;
  import android.content.Intent;
  import android.content.pm.PackageManager;
  import android.content.pm.PackageManager.NameNotFoundException;
  import android.content.pm.Signature;
  import android.os.RemoteException;
  import android.text.format.Time;
  import com.google.android.collect.Sets;
  
  import java.io.PrintWriter;
  import java.util.HashSet;
  
  import static android.content.pm.PackageManager.GET_SIGNATURES;
  import static android.net.NetworkPolicy.CYCLE_NONE;
  import static android.text.format.Time.MONTH_DAY;
  
  /**
   * Manager for creating and modifying network policy rules.
   * <p/>
   * {@hide}
   */
  public class NetworkPolicyManager {
  
      /**
       * No specific network policy, use system default.
       */
      public static final int POLICY_NONE = 0x0;
      /**
       * Reject network usage on metered networks when application in background.
       */
      public static final int POLICY_REJECT_METERED_BACKGROUND = 0x1;
  
      /**
       * All network traffic should be allowed.
       */
      public static final int RULE_ALLOW_ALL = 0x0;
      /**
       * Reject traffic on metered networks.
       */
      public static final int RULE_REJECT_METERED = 0x1;
  
      private static final boolean ALLOW_PLATFORM_APP_POLICY = true;
  
      /**
       * {@link Intent} extra that indicates which {@link NetworkTemplate} rule it
       * applies to.
       */
      public static final String EXTRA_NETWORK_TEMPLATE = "android.net.NETWORK_TEMPLATE";
  
      private INetworkPolicyManager mService;
  
      public NetworkPolicyManager(INetworkPolicyManager service) {
          if (service == null) {
              throw new IllegalArgumentException("missing INetworkPolicyManager");
          }
          mService = service;
      }
  
      /**
       * Set policy flags for specific UID.
       *
       * @param policy {@link #POLICY_NONE} or combination of flags like
       *               {@link #POLICY_REJECT_METERED_BACKGROUND}.
       */
      public void setUidPolicy(int uid, int policy) {
          try {
              mService.setUidPolicy(uid, policy);
          } catch (RemoteException e) {
          }
      }
  
      public int getUidPolicy(int uid) {
          try {
              return mService.getUidPolicy(uid);
          } catch (RemoteException e) {
              return POLICY_NONE;
          }
      }
  
      public int[] getUidsWithPolicy(int policy) {
          try {
              return mService.getUidsWithPolicy(policy);
          } catch (RemoteException e) {
              return new int[0];
          }
      }
  
      public void registerListener(INetworkPolicyListener listener) {
          try {
              mService.registerListener(listener);
          } catch (RemoteException e) {
          }
      }
  
      public void unregisterListener(INetworkPolicyListener listener) {
          try {
              mService.unregisterListener(listener);
          } catch (RemoteException e) {
          }
      }
  
      public void setNetworkPolicies(NetworkPolicy[] policies) {
          try {
              mService.setNetworkPolicies(policies);
          } catch (RemoteException e) {
          }
      }
  
      public NetworkPolicy[] getNetworkPolicies() {
          try {
              return mService.getNetworkPolicies();
          } catch (RemoteException e) {
              return null;
          }
      }
  
      public void setRestrictBackground(boolean restrictBackground) {
          try {
              mService.setRestrictBackground(restrictBackground);
          } catch (RemoteException e) {
          }
      }
  
      public boolean getRestrictBackground() {
          try {
              return mService.getRestrictBackground();
          } catch (RemoteException e) {
              return false;
          }
      }
  
      /**
       * Compute the last cycle boundary for the given {@link NetworkPolicy}. For
       * example, if cycle day is 20th, and today is June 15th, it will return May
       * 20th. When cycle day doesn't exist in current month, it snaps to the 1st
       * of following month.
       *
       * @hide
       */
      public static long computeLastCycleBoundary(long currentTime, NetworkPolicy policy) {
          if (policy.cycleDay == CYCLE_NONE) {
              throw new IllegalArgumentException("Unable to compute boundary without cycleDay");
          }
  
          final Time now = new Time(policy.cycleTimezone);
          now.set(currentTime);
  
          // first, find cycle boundary for current month
          final Time cycle = new Time(now);
          cycle.hour = cycle.minute = cycle.second = 0;
          snapToCycleDay(cycle, policy.cycleDay);
  
          if (Time.compare(cycle, now) >= 0) {
              // cycle boundary is beyond now, use last cycle boundary; start by
              // pushing ourselves squarely into last month.
              final Time lastMonth = new Time(now);
              lastMonth.hour = lastMonth.minute = lastMonth.second = 0;
              lastMonth.monthDay = 1;
              lastMonth.month -= 1;
              lastMonth.normalize(true);
  
              cycle.set(lastMonth);
              snapToCycleDay(cycle, policy.cycleDay);
          }
  
          return cycle.toMillis(true);
      }
  
      /**
       * {@hide}
       */
      public static long computeNextCycleBoundary(long currentTime, NetworkPolicy policy) {
          if (policy.cycleDay == CYCLE_NONE) {
              throw new IllegalArgumentException("Unable to compute boundary without cycleDay");
          }
  
          final Time now = new Time(policy.cycleTimezone);
          now.set(currentTime);
  
          // first, find cycle boundary for current month
          final Time cycle = new Time(now);
          cycle.hour = cycle.minute = cycle.second = 0;
          snapToCycleDay(cycle, policy.cycleDay);
  
          if (Time.compare(cycle, now) <= 0) {
              // cycle boundary is before now, use next cycle boundary; start by
              // pushing ourselves squarely into next month.
              final Time nextMonth = new Time(now);
              nextMonth.hour = nextMonth.minute = nextMonth.second = 0;
              nextMonth.monthDay = 1;
              nextMonth.month += 1;
              nextMonth.normalize(true);
  
              cycle.set(nextMonth);
              snapToCycleDay(cycle, policy.cycleDay);
          }
  
          return cycle.toMillis(true);
      }
  
      /**
       * Snap to the cycle day for the current month given; when cycle day doesn't
       * exist, it snaps to last second of current month.
       *
       * @hide
       */
      public static void snapToCycleDay(Time time, int cycleDay) {
          if (cycleDay > time.getActualMaximum(MONTH_DAY)) {
              // cycle day isn't valid this month; snap to last second of month
              time.month += 1;
              time.monthDay = 1;
              time.second = -1;
          } else {
              time.monthDay = cycleDay;
          }
          time.normalize(true);
      }
  
      /**
       * Check if given UID can have a {@link #setUidPolicy(int, int)} defined,
       * usually to protect critical system services.
       */
      @Deprecated
      public static boolean isUidValidForPolicy(Context context, int uid) {
  
          if (!ALLOW_PLATFORM_APP_POLICY) {
              final PackageManager pm = context.getPackageManager();
              final HashSet<Signature> systemSignature;
              try {
                  systemSignature = Sets.newHashSet(
                          pm.getPackageInfo("android", GET_SIGNATURES).signatures);
              } catch (NameNotFoundException e) {
                  throw new RuntimeException("problem finding system signature", e);
              }
  
              try {
                  // reject apps signed with platform cert
                  for (String packageName : pm.getPackagesForUid(uid)) {
                      final HashSet<Signature> packageSignature = Sets.newHashSet(
                              pm.getPackageInfo(packageName, GET_SIGNATURES).signatures);
                      if (packageSignature.containsAll(systemSignature)) {
                          return false;
                      }
                  }
              } catch (NameNotFoundException e) {
              }
          }
  
          // nothing found above; we can apply policy to UID
          return true;
      }
  
      /**
       * {@hide}
       */
      public static void dumpPolicy(PrintWriter fout, int policy) {
          fout.write("[");
          if ((policy & POLICY_REJECT_METERED_BACKGROUND) != 0) {
              fout.write("REJECT_METERED_BACKGROUND");
          }
          fout.write("]");
      }
  
      /**
       * {@hide}
       */
      public static void dumpRules(PrintWriter fout, int rules) {
          fout.write("[");
          if ((rules & RULE_REJECT_METERED) != 0) {
              fout.write("REJECT_METERED");
          }
          fout.write("]");
      }
  
  }
  