Android/frameworks_base
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge "Fix crash when incrementing data usage request" into tm-dev am: 5b73f4bef4

Browse Source 
Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/18583185

Change-Id: I3ad20d0d4ddea5ed5ed9d3aae0c1b17443c01e92
Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
This commit is contained in:
Junyu Lai
2022-06-01 10:17:37 +00:00
committed by Automerger Merge Worker
parent cc83c18df3 5b73f4bef4
commit e8642adc64
1 changed files with 21 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
services/core/java/com/android/server/connectivity/MultipathPolicyTracker.java
View File

@@ -192,10 +192,13 @@ public class MultipathPolicyTracker {
private final int mSubId;
private long mQuota;
/** Current multipath budget. Nonzero iff we have budget and a UsageCallback is armed. */
private long mMultipathBudget;
/** Current multipath budget. Nonzero iff we have budget. */
// The budget could be accessed by multiple threads, make it volatile to ensure the callers
// on a different thread will not see the stale value.
private volatile long mMultipathBudget;
private final NetworkTemplate mNetworkTemplate;
private final UsageCallback mUsageCallback;
private boolean mUsageCallbackRegistered = false;
private NetworkCapabilities mNetworkCapabilities;
private final NetworkStatsManager mStatsManager;
@@ -234,7 +237,6 @@ public class MultipathPolicyTracker {
@Override
public void onThresholdReached(int networkType, String subscriberId) {
if (DBG) Log.d(TAG, "onThresholdReached for network " + network);
mMultipathBudget = 0;
updateMultipathBudget();
}
};
@@ -376,9 +378,9 @@ public class MultipathPolicyTracker {
if (DBG) {
Log.d(TAG, "Setting callback for " + budget + " bytes on network " + network);
}
registerUsageCallback(budget);
setMultipathBudget(budget);
} else {
maybeUnregisterUsageCallback();
clearMultipathBudget();
}
}
@@ -403,23 +405,30 @@ public class MultipathPolicyTracker {
return mMultipathBudget > 0;
}
private void registerUsageCallback(long budget) {
// Sets the budget and registers a usage callback for it.
private void setMultipathBudget(long budget) {
maybeUnregisterUsageCallback();
if (DBG) Log.d(TAG, "Registering callback, budget is " + mMultipathBudget);
mStatsManager.registerUsageCallback(mNetworkTemplate, budget,
(command) -> mHandler.post(command), mUsageCallback);
mUsageCallbackRegistered = true;
mMultipathBudget = budget;
}
private void maybeUnregisterUsageCallback() {
if (haveMultipathBudget()) {
if (DBG) Log.d(TAG, "Unregistering callback, budget was " + mMultipathBudget);
mStatsManager.unregisterUsageCallback(mUsageCallback);
mMultipathBudget = 0;
}
if (!mUsageCallbackRegistered) return;
if (DBG) Log.d(TAG, "Unregistering callback, budget was " + mMultipathBudget);
mStatsManager.unregisterUsageCallback(mUsageCallback);
mUsageCallbackRegistered = false;
}
private void clearMultipathBudget() {
maybeUnregisterUsageCallback();
mMultipathBudget = 0;
}
void shutdown() {
maybeUnregisterUsageCallback();
clearMultipathBudget();
}
}