/*
* Copyright (C) 2018 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 com.googlecode.android_scripting.facade;
import static android.net.ConnectivityManager.TYPE_MOBILE;
import static android.net.NetworkStats.SET_ALL;
import static android.net.NetworkStatsHistory.FIELD_RX_BYTES;
import static android.net.NetworkStatsHistory.FIELD_TX_BYTES;
import static android.telephony.TelephonyManager.SIM_STATE_READY;
import static android.text.format.DateUtils.FORMAT_ABBREV_MONTH;
import static android.text.format.DateUtils.FORMAT_SHOW_DATE;
import static android.text.format.DateUtils.FORMAT_SHOW_TIME;
import static android.text.format.DateUtils.FORMAT_SHOW_YEAR;
import android.content.Context;
import android.net.ConnectivityManager;
import android.net.INetworkStatsService;
import android.net.INetworkStatsSession;
import android.net.NetworkPolicy;
import android.net.NetworkPolicyManager;
import android.net.NetworkStatsHistory;
import android.net.NetworkTemplate;
import android.os.RemoteException;
import android.os.ServiceManager;
import android.telephony.TelephonyManager;
import android.text.format.DateUtils;
import android.util.Log;
import android.util.Pair;
import java.time.ZonedDateTime;
import java.util.Locale;
/**
* DataUsageController.
*/
public class DataUsageController {
private static final String TAG = "DataUsageController";
private static final boolean DEBUG = Log.isLoggable(TAG, Log.DEBUG);
private static final int FIELDS = FIELD_RX_BYTES | FIELD_TX_BYTES;
private static final StringBuilder PERIOD_BUILDER = new StringBuilder(50);
private static final java.util.Formatter PERIOD_FORMATTER = new java.util.Formatter(
PERIOD_BUILDER, Locale.getDefault());
private final Context mContext;
private final TelephonyManager mTelephonyManager;
private final ConnectivityManager mConnectivityManager;
private final INetworkStatsService mStatsService;
private final NetworkPolicyManager mPolicyManager;
private INetworkStatsSession mSession;
private Callback mCallback;
private NetworkNameProvider mNetworkController;
public DataUsageController(Context context) {
mContext = context;
mTelephonyManager = TelephonyManager.from(context);
mConnectivityManager = ConnectivityManager.from(context);
mStatsService = INetworkStatsService.Stub.asInterface(
ServiceManager.getService(Context.NETWORK_STATS_SERVICE));
mPolicyManager = NetworkPolicyManager.from(mContext);
}
public void setNetworkController(NetworkNameProvider networkController) {
mNetworkController = networkController;
}
private INetworkStatsSession getSession() {
if (mSession == null) {
try {
mSession = mStatsService.openSession();
} catch (RemoteException e) {
Log.w(TAG, "Failed to open stats session", e);
} catch (RuntimeException e) {
Log.w(TAG, "Failed to open stats session", e);
}
}
return mSession;
}
public void setCallback(Callback callback) {
mCallback = callback;
}
private DataUsageInfo warn(String msg) {
Log.w(TAG, "Failed to get data usage, " + msg);
return null;
}
/**
* Get mobile data usage info.
* @return DataUsageInfo: The Mobile data usage information.
*/
public DataUsageInfo getMobileDataUsageInfoForSubscriber(String subscriberId) {
if (subscriberId == null) {
return warn("no subscriber id");
}
NetworkTemplate template = NetworkTemplate.buildTemplateMobileAll(subscriberId);
template = NetworkTemplate.normalize(template, mTelephonyManager.getMergedSubscriberIds());
return getDataUsageInfo(template);
}
/**
* Get mobile data usage info.
* @return DataUsageInfo: The Mobile data usage information.
*/
public DataUsageInfo getMobileDataUsageInfoForUid(Integer uId, String subscriberId) {
if (subscriberId == null) {
return warn("no subscriber id");
}
NetworkTemplate template = NetworkTemplate.buildTemplateMobileAll(subscriberId);
template = NetworkTemplate.normalize(template, mTelephonyManager.getMergedSubscriberIds());
return getDataUsageInfo(template, uId);
}
/**
* Get wifi data usage info.
* @return DataUsageInfo: The Wifi data usage information.
*/
public DataUsageInfo getWifiDataUsageInfo() {
NetworkTemplate template = NetworkTemplate.buildTemplateWifiWildcard();
return getDataUsageInfo(template);
}
/**
* Get data usage info for a given template.
* @param template A given template.
* @return DataUsageInfo: The data usage information.
*/
public DataUsageInfo getDataUsageInfo(NetworkTemplate template) {
return getDataUsageInfo(template, -1);
}
/**
* Get data usage info for a given template.
* @param template A given template.
* @return DataUsageInfo: The data usage information.
*/
public DataUsageInfo getDataUsageInfo(NetworkTemplate template, int uId) {
final INetworkStatsSession session = getSession();
if (session == null) {
return warn("no stats session");
}
final NetworkPolicy policy = findNetworkPolicy(template);
try {
final NetworkStatsHistory mHistory;
if (uId == -1) {
mHistory = session.getHistoryForNetwork(template, FIELDS);
} else {
mHistory = session.getHistoryForUid(template, uId, SET_ALL, 0, FIELDS);
}
final long now = System.currentTimeMillis();
final long start, end;
if (policy != null) {
final Pair<ZonedDateTime, ZonedDateTime> cycle = NetworkPolicyManager
.cycleIterator(policy).next();
start = cycle.first.toInstant().toEpochMilli();
end = cycle.second.toInstant().toEpochMilli();
} else {
// period = last 4 wks
end = now;
start = now - DateUtils.WEEK_IN_MILLIS * 4;
}
//final long callStart = System.currentTimeMillis();
final NetworkStatsHistory.Entry entry = mHistory.getValues(start, end, now, null);
if (entry == null) {
return warn("no entry data");
}
final DataUsageInfo usage = new DataUsageInfo();
usage.subscriberId = template.getSubscriberId();
usage.startEpochMilli = start;
usage.endEpochMilli = end;
usage.usageLevel = mHistory.getTotalBytes();
usage.period = formatDateRange(start, end);
usage.cycleStart = DateUtils.formatDateTime(mContext, start,
FORMAT_SHOW_DATE + FORMAT_SHOW_YEAR + FORMAT_SHOW_TIME);
usage.cycleEnd = DateUtils.formatDateTime(mContext, end,
FORMAT_SHOW_DATE + FORMAT_SHOW_YEAR + FORMAT_SHOW_TIME);
if (policy != null) {
usage.limitLevel = policy.limitBytes > 0 ? policy.limitBytes : -1;
usage.warningLevel = policy.warningBytes > 0 ? policy.warningBytes : -1;
}
if (uId != -1) {
usage.uId = uId;
}
return usage;
} catch (RemoteException e) {
return warn("remote call failed");
}
}
private NetworkPolicy findNetworkPolicy(NetworkTemplate template) {
if (mPolicyManager == null || template == null) return null;
final NetworkPolicy[] policies = mPolicyManager.getNetworkPolicies();
if (policies == null) return null;
final int mLength = policies.length;
for (int i = 0; i < mLength; i++) {
final NetworkPolicy policy = policies[i];
if (policy != null && template.equals(policy.template)) {
return policy;
}
}
return null;
}
private static String historyEntryToString(NetworkStatsHistory.Entry entry) {
return entry == null ? null : new StringBuilder("Entry[")
.append("bucketDuration=").append(entry.bucketDuration)
.append(",bucketStart=").append(entry.bucketStart)
.append(",activeTime=").append(entry.activeTime)
.append(",rxBytes=").append(entry.rxBytes)
.append(",rxPackets=").append(entry.rxPackets)
.append(",txBytes=").append(entry.txBytes)
.append(",txPackets=").append(entry.txPackets)
.append(",operations=").append(entry.operations)
.append(']').toString();
}
/**
* Enable or disable mobile data.
* @param enabled Enable data: True: Disable data: False.
*/
public void setMobileDataEnabled(boolean enabled) {
Log.d(TAG, "setMobileDataEnabled: enabled=" + enabled);
mTelephonyManager.setDataEnabled(enabled);
if (mCallback != null) {
mCallback.onMobileDataEnabled(enabled);
}
}
/**
* Check if mobile data is supported.
* @return True supported, False: Not supported.
*/
public boolean isMobileDataSupported() {
// require both supported network and ready SIM
return mConnectivityManager.isNetworkSupported(TYPE_MOBILE)
&& mTelephonyManager.getSimState() == SIM_STATE_READY;
}
/**
* Check if mobile data is enabled.
* @return True: enabled, False: Not enabled.
*/
public boolean isMobileDataEnabled() {
return mTelephonyManager.getDataEnabled();
}
private String formatDateRange(long start, long end) {
final int flags = FORMAT_SHOW_DATE | FORMAT_ABBREV_MONTH;
synchronized (PERIOD_BUILDER) {
PERIOD_BUILDER.setLength(0);
return DateUtils.formatDateRange(mContext, PERIOD_FORMATTER, start, end, flags, null)
.toString();
}
}
public interface NetworkNameProvider {
String getMobileDataNetworkName();
}
public static class DataUsageInfo {
public String subscriberId;
public String period;
public Integer uId;
public long startEpochMilli;
public long endEpochMilli;
public long limitLevel;
public long warningLevel;
public long usageLevel;
public String cycleStart;
public String cycleEnd;
}
public interface Callback {
void onMobileDataEnabled(boolean enabled);
}
}