1 /* 2 * Copyright (C) 2016 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 package com.android.server.pm; 17 18 import android.annotation.NonNull; 19 import android.annotation.Nullable; 20 import android.annotation.UserIdInt; 21 import android.content.ComponentName; 22 import android.content.Intent; 23 import android.content.pm.PackageInfo; 24 import android.content.pm.ShortcutInfo; 25 import android.content.res.Resources; 26 import android.os.Bundle; 27 import android.os.PersistableBundle; 28 import android.text.format.Formatter; 29 import android.util.ArrayMap; 30 import android.util.ArraySet; 31 import android.util.Log; 32 import android.util.Slog; 33 34 import com.android.internal.annotations.VisibleForTesting; 35 import com.android.internal.util.Preconditions; 36 import com.android.internal.util.XmlUtils; 37 import com.android.server.pm.ShortcutService.ShortcutOperation; 38 import com.android.server.pm.ShortcutService.Stats; 39 40 import org.json.JSONException; 41 import org.json.JSONObject; 42 import org.xmlpull.v1.XmlPullParser; 43 import org.xmlpull.v1.XmlPullParserException; 44 import org.xmlpull.v1.XmlSerializer; 45 46 import java.io.File; 47 import java.io.IOException; 48 import java.io.PrintWriter; 49 import java.util.ArrayList; 50 import java.util.Collections; 51 import java.util.Comparator; 52 import java.util.List; 53 import java.util.Set; 54 import java.util.function.Predicate; 55 56 /** 57 * Package information used by {@link ShortcutService}. 58 * User information used by {@link ShortcutService}. 59 * 60 * All methods should be guarded by {@code #mShortcutUser.mService.mLock}. 61 */ 62 class ShortcutPackage extends ShortcutPackageItem { 63 private static final String TAG = ShortcutService.TAG; 64 private static final String TAG_VERIFY = ShortcutService.TAG + ".verify"; 65 66 static final String TAG_ROOT = "package"; 67 private static final String TAG_INTENT_EXTRAS_LEGACY = "intent-extras"; 68 private static final String TAG_INTENT = "intent"; 69 private static final String TAG_EXTRAS = "extras"; 70 private static final String TAG_SHORTCUT = "shortcut"; 71 private static final String TAG_CATEGORIES = "categories"; 72 73 private static final String ATTR_NAME = "name"; 74 private static final String ATTR_CALL_COUNT = "call-count"; 75 private static final String ATTR_LAST_RESET = "last-reset"; 76 private static final String ATTR_ID = "id"; 77 private static final String ATTR_ACTIVITY = "activity"; 78 private static final String ATTR_TITLE = "title"; 79 private static final String ATTR_TITLE_RES_ID = "titleid"; 80 private static final String ATTR_TITLE_RES_NAME = "titlename"; 81 private static final String ATTR_TEXT = "text"; 82 private static final String ATTR_TEXT_RES_ID = "textid"; 83 private static final String ATTR_TEXT_RES_NAME = "textname"; 84 private static final String ATTR_DISABLED_MESSAGE = "dmessage"; 85 private static final String ATTR_DISABLED_MESSAGE_RES_ID = "dmessageid"; 86 private static final String ATTR_DISABLED_MESSAGE_RES_NAME = "dmessagename"; 87 private static final String ATTR_INTENT_LEGACY = "intent"; 88 private static final String ATTR_INTENT_NO_EXTRA = "intent-base"; 89 private static final String ATTR_RANK = "rank"; 90 private static final String ATTR_TIMESTAMP = "timestamp"; 91 private static final String ATTR_FLAGS = "flags"; 92 private static final String ATTR_ICON_RES_ID = "icon-res"; 93 private static final String ATTR_ICON_RES_NAME = "icon-resname"; 94 private static final String ATTR_BITMAP_PATH = "bitmap-path"; 95 96 private static final String NAME_CATEGORIES = "categories"; 97 98 private static final String TAG_STRING_ARRAY_XMLUTILS = "string-array"; 99 private static final String ATTR_NAME_XMLUTILS = "name"; 100 101 private static final String KEY_DYNAMIC = "dynamic"; 102 private static final String KEY_MANIFEST = "manifest"; 103 private static final String KEY_PINNED = "pinned"; 104 private static final String KEY_BITMAPS = "bitmaps"; 105 private static final String KEY_BITMAP_BYTES = "bitmapBytes"; 106 107 /** 108 * All the shortcuts from the package, keyed on IDs. 109 */ 110 final private ArrayMap<String, ShortcutInfo> mShortcuts = new ArrayMap<>(); 111 112 /** 113 * # of times the package has called rate-limited APIs. 114 */ 115 private int mApiCallCount; 116 117 /** 118 * When {@link #mApiCallCount} was reset last time. 119 */ 120 private long mLastResetTime; 121 122 private final int mPackageUid; 123 124 private long mLastKnownForegroundElapsedTime; 125 126 private ShortcutPackage(ShortcutUser shortcutUser, 127 int packageUserId, String packageName, ShortcutPackageInfo spi) { 128 super(shortcutUser, packageUserId, packageName, 129 spi != null ? spi : ShortcutPackageInfo.newEmpty()); 130 131 mPackageUid = shortcutUser.mService.injectGetPackageUid(packageName, packageUserId); 132 } 133 134 public ShortcutPackage(ShortcutUser shortcutUser, int packageUserId, String packageName) { 135 this(shortcutUser, packageUserId, packageName, null); 136 } 137 138 @Override 139 public int getOwnerUserId() { 140 // For packages, always owner user == package user. 141 return getPackageUserId(); 142 } 143 144 public int getPackageUid() { 145 return mPackageUid; 146 } 147 148 /** 149 * Called when a shortcut is about to be published. At this point we know the publisher 150 * package 151 * exists (as opposed to Launcher trying to fetch shortcuts from a non-existent package), so 152 * we do some initialization for the package. 153 */ 154 private void ensurePackageVersionInfo() { 155 // Make sure we have the version code for the app. We need the version code in 156 // handlePackageUpdated(). 157 if (getPackageInfo().getVersionCode() < 0) { 158 final ShortcutService s = mShortcutUser.mService; 159 160 final PackageInfo pi = s.getPackageInfo(getPackageName(), getOwnerUserId()); 161 if (pi != null) { 162 if (ShortcutService.DEBUG) { 163 Slog.d(TAG, String.format("Package %s version = %d", getPackageName(), 164 pi.versionCode)); 165 } 166 getPackageInfo().updateVersionInfo(pi); 167 s.scheduleSaveUser(getOwnerUserId()); 168 } 169 } 170 } 171 172 @Nullable 173 public Resources getPackageResources() { 174 return mShortcutUser.mService.injectGetResourcesForApplicationAsUser( 175 getPackageName(), getPackageUserId()); 176 } 177 178 @Override 179 protected void onRestoreBlocked() { 180 // Can't restore due to version/signature mismatch. Remove all shortcuts. 181 mShortcuts.clear(); 182 } 183 184 @Override 185 protected void onRestored() { 186 // Because some launchers may not have been restored (e.g. allowBackup=false), 187 // we need to re-calculate the pinned shortcuts. 188 refreshPinnedFlags(); 189 } 190 191 /** 192 * Note this does *not* provide a correct view to the calling launcher. 193 */ 194 @Nullable 195 public ShortcutInfo findShortcutById(String id) { 196 return mShortcuts.get(id); 197 } 198 199 private void ensureNotImmutable(@Nullable ShortcutInfo shortcut) { 200 if (shortcut != null && shortcut.isImmutable()) { 201 throw new IllegalArgumentException( 202 "Manifest shortcut ID=" + shortcut.getId() 203 + " may not be manipulated via APIs"); 204 } 205 } 206 207 private void ensureNotImmutable(@NonNull String id) { 208 ensureNotImmutable(mShortcuts.get(id)); 209 } 210 211 public void ensureImmutableShortcutsNotIncludedWithIds(@NonNull List<String> shortcutIds) { 212 for (int i = shortcutIds.size() - 1; i >= 0; i--) { 213 ensureNotImmutable(shortcutIds.get(i)); 214 } 215 } 216 217 public void ensureImmutableShortcutsNotIncluded(@NonNull List<ShortcutInfo> shortcuts) { 218 for (int i = shortcuts.size() - 1; i >= 0; i--) { 219 ensureNotImmutable(shortcuts.get(i).getId()); 220 } 221 } 222 223 private ShortcutInfo deleteShortcutInner(@NonNull String id) { 224 final ShortcutInfo shortcut = mShortcuts.remove(id); 225 if (shortcut != null) { 226 mShortcutUser.mService.removeIcon(getPackageUserId(), shortcut); 227 shortcut.clearFlags(ShortcutInfo.FLAG_DYNAMIC | ShortcutInfo.FLAG_PINNED 228 | ShortcutInfo.FLAG_MANIFEST); 229 } 230 return shortcut; 231 } 232 233 private void addShortcutInner(@NonNull ShortcutInfo newShortcut) { 234 final ShortcutService s = mShortcutUser.mService; 235 236 deleteShortcutInner(newShortcut.getId()); 237 238 // Extract Icon and update the icon res ID and the bitmap path. 239 s.saveIconAndFixUpShortcut(getPackageUserId(), newShortcut); 240 s.fixUpShortcutResourceNamesAndValues(newShortcut); 241 mShortcuts.put(newShortcut.getId(), newShortcut); 242 } 243 244 /** 245 * Add a shortcut, or update one with the same ID, with taking over existing flags. 246 * 247 * It checks the max number of dynamic shortcuts. 248 */ 249 public void addOrUpdateDynamicShortcut(@NonNull ShortcutInfo newShortcut) { 250 251 Preconditions.checkArgument(newShortcut.isEnabled(), 252 "add/setDynamicShortcuts() cannot publish disabled shortcuts"); 253 254 ensurePackageVersionInfo(); 255 256 newShortcut.addFlags(ShortcutInfo.FLAG_DYNAMIC); 257 258 final ShortcutInfo oldShortcut = mShortcuts.get(newShortcut.getId()); 259 260 final boolean wasPinned; 261 262 if (oldShortcut == null) { 263 wasPinned = false; 264 } else { 265 // It's an update case. 266 // Make sure the target is updatable. (i.e. should be mutable.) 267 oldShortcut.ensureUpdatableWith(newShortcut); 268 269 wasPinned = oldShortcut.isPinned(); 270 } 271 272 // If it was originally pinned, the new one should be pinned too. 273 if (wasPinned) { 274 newShortcut.addFlags(ShortcutInfo.FLAG_PINNED); 275 } 276 277 addShortcutInner(newShortcut); 278 } 279 280 /** 281 * Remove all shortcuts that aren't pinned nor dynamic. 282 */ 283 private void removeOrphans() { 284 ArrayList<String> removeList = null; // Lazily initialize. 285 286 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 287 final ShortcutInfo si = mShortcuts.valueAt(i); 288 289 if (si.isAlive()) continue; 290 291 if (removeList == null) { 292 removeList = new ArrayList<>(); 293 } 294 removeList.add(si.getId()); 295 } 296 if (removeList != null) { 297 for (int i = removeList.size() - 1; i >= 0; i--) { 298 deleteShortcutInner(removeList.get(i)); 299 } 300 } 301 } 302 303 /** 304 * Remove all dynamic shortcuts. 305 */ 306 public void deleteAllDynamicShortcuts() { 307 final long now = mShortcutUser.mService.injectCurrentTimeMillis(); 308 309 boolean changed = false; 310 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 311 final ShortcutInfo si = mShortcuts.valueAt(i); 312 if (si.isDynamic()) { 313 changed = true; 314 315 si.setTimestamp(now); 316 si.clearFlags(ShortcutInfo.FLAG_DYNAMIC); 317 si.setRank(0); // It may still be pinned, so clear the rank. 318 } 319 } 320 if (changed) { 321 removeOrphans(); 322 } 323 } 324 325 /** 326 * Remove a dynamic shortcut by ID. It'll be removed from the dynamic set, but if the shortcut 327 * is pinned, it'll remain as a pinned shortcut, and is still enabled. 328 * 329 * @return true if it's actually removed because it wasn't pinned, or false if it's still 330 * pinned. 331 */ 332 public boolean deleteDynamicWithId(@NonNull String shortcutId) { 333 final ShortcutInfo removed = deleteOrDisableWithId( 334 shortcutId, /* disable =*/ false, /* overrideImmutable=*/ false); 335 return removed == null; 336 } 337 338 /** 339 * Disable a dynamic shortcut by ID. It'll be removed from the dynamic set, but if the shortcut 340 * is pinned, it'll remain as a pinned shortcut, but will be disabled. 341 * 342 * @return true if it's actually removed because it wasn't pinned, or false if it's still 343 * pinned. 344 */ 345 private boolean disableDynamicWithId(@NonNull String shortcutId) { 346 final ShortcutInfo disabled = deleteOrDisableWithId( 347 shortcutId, /* disable =*/ true, /* overrideImmutable=*/ false); 348 return disabled == null; 349 } 350 351 /** 352 * Disable a dynamic shortcut by ID. It'll be removed from the dynamic set, but if the shortcut 353 * is pinned, it'll remain as a pinned shortcut but will be disabled. 354 */ 355 public void disableWithId(@NonNull String shortcutId, String disabledMessage, 356 int disabledMessageResId, boolean overrideImmutable) { 357 final ShortcutInfo disabled = deleteOrDisableWithId(shortcutId, /* disable =*/ true, 358 overrideImmutable); 359 360 if (disabled != null) { 361 if (disabledMessage != null) { 362 disabled.setDisabledMessage(disabledMessage); 363 } else if (disabledMessageResId != 0) { 364 disabled.setDisabledMessageResId(disabledMessageResId); 365 366 mShortcutUser.mService.fixUpShortcutResourceNamesAndValues(disabled); 367 } 368 } 369 } 370 371 @Nullable 372 private ShortcutInfo deleteOrDisableWithId(@NonNull String shortcutId, boolean disable, 373 boolean overrideImmutable) { 374 final ShortcutInfo oldShortcut = mShortcuts.get(shortcutId); 375 376 if (oldShortcut == null || !oldShortcut.isEnabled()) { 377 return null; // Doesn't exist or already disabled. 378 } 379 if (!overrideImmutable) { 380 ensureNotImmutable(oldShortcut); 381 } 382 if (oldShortcut.isPinned()) { 383 384 oldShortcut.setRank(0); 385 oldShortcut.clearFlags(ShortcutInfo.FLAG_DYNAMIC | ShortcutInfo.FLAG_MANIFEST); 386 if (disable) { 387 oldShortcut.addFlags(ShortcutInfo.FLAG_DISABLED); 388 } 389 oldShortcut.setTimestamp(mShortcutUser.mService.injectCurrentTimeMillis()); 390 391 return oldShortcut; 392 } else { 393 deleteShortcutInner(shortcutId); 394 return null; 395 } 396 } 397 398 public void enableWithId(@NonNull String shortcutId) { 399 final ShortcutInfo shortcut = mShortcuts.get(shortcutId); 400 if (shortcut != null) { 401 ensureNotImmutable(shortcut); 402 shortcut.clearFlags(ShortcutInfo.FLAG_DISABLED); 403 } 404 } 405 406 /** 407 * Called after a launcher updates the pinned set. For each shortcut in this package, 408 * set FLAG_PINNED if any launcher has pinned it. Otherwise, clear it. 409 * 410 * <p>Then remove all shortcuts that are not dynamic and no longer pinned either. 411 */ 412 public void refreshPinnedFlags() { 413 // First, un-pin all shortcuts 414 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 415 mShortcuts.valueAt(i).clearFlags(ShortcutInfo.FLAG_PINNED); 416 } 417 418 // Then, for the pinned set for each launcher, set the pin flag one by one. 419 mShortcutUser.mService.getUserShortcutsLocked(getPackageUserId()) 420 .forAllLaunchers(launcherShortcuts -> { 421 final ArraySet<String> pinned = launcherShortcuts.getPinnedShortcutIds( 422 getPackageName(), getPackageUserId()); 423 424 if (pinned == null || pinned.size() == 0) { 425 return; 426 } 427 for (int i = pinned.size() - 1; i >= 0; i--) { 428 final String id = pinned.valueAt(i); 429 final ShortcutInfo si = mShortcuts.get(id); 430 if (si == null) { 431 // This happens if a launcher pinned shortcuts from this package, then backup& 432 // restored, but this package doesn't allow backing up. 433 // In that case the launcher ends up having a dangling pinned shortcuts. 434 // That's fine, when the launcher is restored, we'll fix it. 435 continue; 436 } 437 si.addFlags(ShortcutInfo.FLAG_PINNED); 438 } 439 }); 440 441 // Lastly, remove the ones that are no longer pinned nor dynamic. 442 removeOrphans(); 443 } 444 445 /** 446 * Number of calls that the caller has made, since the last reset. 447 * 448 * <p>This takes care of the resetting the counter for foreground apps as well as after 449 * locale changes. 450 */ 451 public int getApiCallCount() { 452 final ShortcutService s = mShortcutUser.mService; 453 454 // Reset the counter if: 455 // - the package is in foreground now. 456 // - the package is *not* in foreground now, but was in foreground at some point 457 // since the previous time it had been. 458 if (s.isUidForegroundLocked(mPackageUid) 459 || mLastKnownForegroundElapsedTime 460 < s.getUidLastForegroundElapsedTimeLocked(mPackageUid)) { 461 mLastKnownForegroundElapsedTime = s.injectElapsedRealtime(); 462 resetRateLimiting(); 463 } 464 465 // Note resetThrottlingIfNeeded() and resetRateLimiting() will set 0 to mApiCallCount, 466 // but we just can't return 0 at this point, because we may have to update 467 // mLastResetTime. 468 469 final long last = s.getLastResetTimeLocked(); 470 471 final long now = s.injectCurrentTimeMillis(); 472 if (ShortcutService.isClockValid(now) && mLastResetTime > now) { 473 Slog.w(TAG, "Clock rewound"); 474 // Clock rewound. 475 mLastResetTime = now; 476 mApiCallCount = 0; 477 return mApiCallCount; 478 } 479 480 // If not reset yet, then reset. 481 if (mLastResetTime < last) { 482 if (ShortcutService.DEBUG) { 483 Slog.d(TAG, String.format("%s: last reset=%d, now=%d, last=%d: resetting", 484 getPackageName(), mLastResetTime, now, last)); 485 } 486 mApiCallCount = 0; 487 mLastResetTime = last; 488 } 489 return mApiCallCount; 490 } 491 492 /** 493 * If the caller app hasn't been throttled yet, increment {@link #mApiCallCount} 494 * and return true. Otherwise just return false. 495 * 496 * <p>This takes care of the resetting the counter for foreground apps as well as after 497 * locale changes, which is done internally by {@link #getApiCallCount}. 498 */ 499 public boolean tryApiCall() { 500 final ShortcutService s = mShortcutUser.mService; 501 502 if (getApiCallCount() >= s.mMaxUpdatesPerInterval) { 503 return false; 504 } 505 mApiCallCount++; 506 s.scheduleSaveUser(getOwnerUserId()); 507 return true; 508 } 509 510 public void resetRateLimiting() { 511 if (ShortcutService.DEBUG) { 512 Slog.d(TAG, "resetRateLimiting: " + getPackageName()); 513 } 514 if (mApiCallCount > 0) { 515 mApiCallCount = 0; 516 mShortcutUser.mService.scheduleSaveUser(getOwnerUserId()); 517 } 518 } 519 520 public void resetRateLimitingForCommandLineNoSaving() { 521 mApiCallCount = 0; 522 mLastResetTime = 0; 523 } 524 525 /** 526 * Find all shortcuts that match {@code query}. 527 */ 528 public void findAll(@NonNull List<ShortcutInfo> result, 529 @Nullable Predicate<ShortcutInfo> query, int cloneFlag) { 530 findAll(result, query, cloneFlag, null, 0); 531 } 532 533 /** 534 * Find all shortcuts that match {@code query}. 535 * 536 * This will also provide a "view" for each launcher -- a non-dynamic shortcut that's not pinned 537 * by the calling launcher will not be included in the result, and also "isPinned" will be 538 * adjusted for the caller too. 539 */ 540 public void findAll(@NonNull List<ShortcutInfo> result, 541 @Nullable Predicate<ShortcutInfo> query, int cloneFlag, 542 @Nullable String callingLauncher, int launcherUserId) { 543 if (getPackageInfo().isShadow()) { 544 // Restored and the app not installed yet, so don't return any. 545 return; 546 } 547 548 final ShortcutService s = mShortcutUser.mService; 549 550 // Set of pinned shortcuts by the calling launcher. 551 final ArraySet<String> pinnedByCallerSet = (callingLauncher == null) ? null 552 : s.getLauncherShortcutsLocked(callingLauncher, getPackageUserId(), launcherUserId) 553 .getPinnedShortcutIds(getPackageName(), getPackageUserId()); 554 555 for (int i = 0; i < mShortcuts.size(); i++) { 556 final ShortcutInfo si = mShortcuts.valueAt(i); 557 558 // Need to adjust PINNED flag depending on the caller. 559 // Basically if the caller is a launcher (callingLauncher != null) and the launcher 560 // isn't pinning it, then we need to clear PINNED for this caller. 561 final boolean isPinnedByCaller = (callingLauncher == null) 562 || ((pinnedByCallerSet != null) && pinnedByCallerSet.contains(si.getId())); 563 564 if (si.isFloating()) { 565 if (!isPinnedByCaller) { 566 continue; 567 } 568 } 569 final ShortcutInfo clone = si.clone(cloneFlag); 570 571 // Fix up isPinned for the caller. Note we need to do it before the "test" callback, 572 // since it may check isPinned. 573 if (!isPinnedByCaller) { 574 clone.clearFlags(ShortcutInfo.FLAG_PINNED); 575 } 576 if (query == null || query.test(clone)) { 577 result.add(clone); 578 } 579 } 580 } 581 582 public void resetThrottling() { 583 mApiCallCount = 0; 584 } 585 586 /** 587 * Return the filenames (excluding path names) of icon bitmap files from this package. 588 */ 589 public ArraySet<String> getUsedBitmapFiles() { 590 final ArraySet<String> usedFiles = new ArraySet<>(mShortcuts.size()); 591 592 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 593 final ShortcutInfo si = mShortcuts.valueAt(i); 594 if (si.getBitmapPath() != null) { 595 usedFiles.add(getFileName(si.getBitmapPath())); 596 } 597 } 598 return usedFiles; 599 } 600 601 private static String getFileName(@NonNull String path) { 602 final int sep = path.lastIndexOf(File.separatorChar); 603 if (sep == -1) { 604 return path; 605 } else { 606 return path.substring(sep + 1); 607 } 608 } 609 610 /** 611 * @return false if any of the target activities are no longer enabled. 612 */ 613 private boolean areAllActivitiesStillEnabled() { 614 if (mShortcuts.size() == 0) { 615 return true; 616 } 617 final ShortcutService s = mShortcutUser.mService; 618 619 // Normally the number of target activities is 1 or so, so no need to use a complex 620 // structure like a set. 621 final ArrayList<ComponentName> checked = new ArrayList<>(4); 622 623 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 624 final ShortcutInfo si = mShortcuts.valueAt(i); 625 final ComponentName activity = si.getActivity(); 626 627 if (checked.contains(activity)) { 628 continue; // Already checked. 629 } 630 checked.add(activity); 631 632 if (!s.injectIsActivityEnabledAndExported(activity, getOwnerUserId())) { 633 return false; 634 } 635 } 636 return true; 637 } 638 639 /** 640 * Called when the package may be added or updated, or its activities may be disabled, and 641 * if so, rescan the package and do the necessary stuff. 642 * 643 * Add case: 644 * - Publish manifest shortcuts. 645 * 646 * Update case: 647 * - Re-publish manifest shortcuts. 648 * - If there are shortcuts with resources (icons or strings), update their timestamps. 649 * - Disable shortcuts whose target activities are disabled. 650 * 651 * @return TRUE if any shortcuts have been changed. 652 */ 653 public boolean rescanPackageIfNeeded(boolean isNewApp, boolean forceRescan) { 654 final ShortcutService s = mShortcutUser.mService; 655 final long start = s.injectElapsedRealtime(); 656 657 final PackageInfo pi; 658 try { 659 pi = mShortcutUser.mService.getPackageInfo( 660 getPackageName(), getPackageUserId()); 661 if (pi == null) { 662 return false; // Shouldn't happen. 663 } 664 665 if (!isNewApp && !forceRescan) { 666 // Return if the package hasn't changed, ie: 667 // - version code hasn't change 668 // - lastUpdateTime hasn't change 669 // - all target activities are still enabled. 670 if ((getPackageInfo().getVersionCode() == pi.versionCode) 671 && (getPackageInfo().getLastUpdateTime() == pi.lastUpdateTime) 672 && areAllActivitiesStillEnabled()) { 673 return false; 674 } 675 } 676 } finally { 677 s.logDurationStat(Stats.PACKAGE_UPDATE_CHECK, start); 678 } 679 680 // Now prepare to publish manifest shortcuts. 681 List<ShortcutInfo> newManifestShortcutList = null; 682 try { 683 newManifestShortcutList = ShortcutParser.parseShortcuts(mShortcutUser.mService, 684 getPackageName(), getPackageUserId()); 685 } catch (IOException|XmlPullParserException e) { 686 Slog.e(TAG, "Failed to load shortcuts from AndroidManifest.xml.", e); 687 } 688 final int manifestShortcutSize = newManifestShortcutList == null ? 0 689 : newManifestShortcutList.size(); 690 if (ShortcutService.DEBUG) { 691 Slog.d(TAG, String.format("Package %s has %d manifest shortcut(s)", 692 getPackageName(), manifestShortcutSize)); 693 } 694 if (isNewApp && (manifestShortcutSize == 0)) { 695 // If it's a new app, and it doesn't have manifest shortcuts, then nothing to do. 696 697 // If it's an update, then it may already have manifest shortcuts, which need to be 698 // disabled. 699 return false; 700 } 701 if (ShortcutService.DEBUG) { 702 Slog.d(TAG, String.format("Package %s %s, version %d -> %d", getPackageName(), 703 (isNewApp ? "added" : "updated"), 704 getPackageInfo().getVersionCode(), pi.versionCode)); 705 } 706 707 getPackageInfo().updateVersionInfo(pi); 708 709 boolean changed = false; 710 711 // For existing shortcuts, update timestamps if they have any resources. 712 // Also check if shortcuts' activities are still main activities. Otherwise, disable them. 713 if (!isNewApp) { 714 Resources publisherRes = null; 715 716 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 717 final ShortcutInfo si = mShortcuts.valueAt(i); 718 719 if (si.isDynamic()) { 720 if (!s.injectIsMainActivity(si.getActivity(), getPackageUserId())) { 721 Slog.w(TAG, String.format( 722 "%s is no longer main activity. Disabling shorcut %s.", 723 getPackageName(), si.getId())); 724 if (disableDynamicWithId(si.getId())) { 725 continue; // Actually removed. 726 } 727 // Still pinned, so fall-through and possibly update the resources. 728 } 729 changed = true; 730 } 731 732 if (si.hasAnyResources()) { 733 if (!si.isOriginallyFromManifest()) { 734 if (publisherRes == null) { 735 publisherRes = getPackageResources(); 736 if (publisherRes == null) { 737 break; // Resources couldn't be loaded. 738 } 739 } 740 741 // If this shortcut is not from a manifest, then update all resource IDs 742 // from resource names. (We don't allow resource strings for 743 // non-manifest at the moment, but icons can still be resources.) 744 si.lookupAndFillInResourceIds(publisherRes); 745 } 746 changed = true; 747 si.setTimestamp(s.injectCurrentTimeMillis()); 748 } 749 } 750 } 751 752 // (Re-)publish manifest shortcut. 753 changed |= publishManifestShortcuts(newManifestShortcutList); 754 755 if (newManifestShortcutList != null) { 756 changed |= pushOutExcessShortcuts(); 757 } 758 759 s.verifyStates(); 760 761 if (changed) { 762 // This will send a notification to the launcher, and also save . 763 s.packageShortcutsChanged(getPackageName(), getPackageUserId()); 764 } else { 765 // Still save the version code. 766 s.scheduleSaveUser(getPackageUserId()); 767 } 768 return changed; 769 } 770 771 private boolean publishManifestShortcuts(List<ShortcutInfo> newManifestShortcutList) { 772 if (ShortcutService.DEBUG) { 773 Slog.d(TAG, String.format( 774 "Package %s: publishing manifest shortcuts", getPackageName())); 775 } 776 boolean changed = false; 777 778 // Keep the previous IDs. 779 ArraySet<String> toDisableList = null; 780 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 781 final ShortcutInfo si = mShortcuts.valueAt(i); 782 783 if (si.isManifestShortcut()) { 784 if (toDisableList == null) { 785 toDisableList = new ArraySet<>(); 786 } 787 toDisableList.add(si.getId()); 788 } 789 } 790 791 // Publish new ones. 792 if (newManifestShortcutList != null) { 793 final int newListSize = newManifestShortcutList.size(); 794 795 for (int i = 0; i < newListSize; i++) { 796 changed = true; 797 798 final ShortcutInfo newShortcut = newManifestShortcutList.get(i); 799 final boolean newDisabled = !newShortcut.isEnabled(); 800 801 final String id = newShortcut.getId(); 802 final ShortcutInfo oldShortcut = mShortcuts.get(id); 803 804 boolean wasPinned = false; 805 806 if (oldShortcut != null) { 807 if (!oldShortcut.isOriginallyFromManifest()) { 808 Slog.e(TAG, "Shortcut with ID=" + newShortcut.getId() 809 + " exists but is not from AndroidManifest.xml, not updating."); 810 continue; 811 } 812 // Take over the pinned flag. 813 if (oldShortcut.isPinned()) { 814 wasPinned = true; 815 newShortcut.addFlags(ShortcutInfo.FLAG_PINNED); 816 } 817 } 818 if (newDisabled && !wasPinned) { 819 // If the shortcut is disabled, and it was *not* pinned, then this 820 // just doesn't have to be published. 821 // Just keep it in toDisableList, so the previous one would be removed. 822 continue; 823 } 824 825 // Note even if enabled=false, we still need to update all fields, so do it 826 // regardless. 827 addShortcutInner(newShortcut); // This will clean up the old one too. 828 829 if (!newDisabled && toDisableList != null) { 830 // Still alive, don't remove. 831 toDisableList.remove(id); 832 } 833 } 834 } 835 836 // Disable the previous manifest shortcuts that are no longer in the manifest. 837 if (toDisableList != null) { 838 if (ShortcutService.DEBUG) { 839 Slog.d(TAG, String.format( 840 "Package %s: disabling %d stale shortcuts", getPackageName(), 841 toDisableList.size())); 842 } 843 for (int i = toDisableList.size() - 1; i >= 0; i--) { 844 changed = true; 845 846 final String id = toDisableList.valueAt(i); 847 848 disableWithId(id, /* disable message =*/ null, /* disable message resid */ 0, 849 /* overrideImmutable=*/ true); 850 } 851 removeOrphans(); 852 } 853 adjustRanks(); 854 return changed; 855 } 856 857 /** 858 * For each target activity, make sure # of dynamic + manifest shortcuts <= max. 859 * If too many, we'll remove the dynamic with the lowest ranks. 860 */ 861 private boolean pushOutExcessShortcuts() { 862 final ShortcutService service = mShortcutUser.mService; 863 final int maxShortcuts = service.getMaxActivityShortcuts(); 864 865 boolean changed = false; 866 867 final ArrayMap<ComponentName, ArrayList<ShortcutInfo>> all = 868 sortShortcutsToActivities(); 869 for (int outer = all.size() - 1; outer >= 0; outer--) { 870 final ArrayList<ShortcutInfo> list = all.valueAt(outer); 871 if (list.size() <= maxShortcuts) { 872 continue; 873 } 874 // Sort by isManifestShortcut() and getRank(). 875 Collections.sort(list, mShortcutTypeAndRankComparator); 876 877 // Keep [0 .. max), and remove (as dynamic) [max .. size) 878 for (int inner = list.size() - 1; inner >= maxShortcuts; inner--) { 879 final ShortcutInfo shortcut = list.get(inner); 880 881 if (shortcut.isManifestShortcut()) { 882 // This shouldn't happen -- excess shortcuts should all be non-manifest. 883 // But just in case. 884 service.wtf("Found manifest shortcuts in excess list."); 885 continue; 886 } 887 deleteDynamicWithId(shortcut.getId()); 888 } 889 } 890 891 return changed; 892 } 893 894 /** 895 * To sort by isManifestShortcut() and getRank(). i.e. manifest shortcuts come before 896 * non-manifest shortcuts, then sort by rank. 897 * 898 * This is used to decide which dynamic shortcuts to remove when an upgraded version has more 899 * manifest shortcuts than before and as a result we need to remove some of the dynamic 900 * shortcuts. We sort manifest + dynamic shortcuts by this order, and remove the ones with 901 * the last ones. 902 * 903 * (Note the number of manifest shortcuts is always <= the max number, because if there are 904 * more, ShortcutParser would ignore the rest.) 905 */ 906 final Comparator<ShortcutInfo> mShortcutTypeAndRankComparator = (ShortcutInfo a, 907 ShortcutInfo b) -> { 908 if (a.isManifestShortcut() && !b.isManifestShortcut()) { 909 return -1; 910 } 911 if (!a.isManifestShortcut() && b.isManifestShortcut()) { 912 return 1; 913 } 914 return Integer.compare(a.getRank(), b.getRank()); 915 }; 916 917 /** 918 * Build a list of shortcuts for each target activity and return as a map. The result won't 919 * contain "floating" shortcuts because they don't belong on any activities. 920 */ 921 private ArrayMap<ComponentName, ArrayList<ShortcutInfo>> sortShortcutsToActivities() { 922 final ArrayMap<ComponentName, ArrayList<ShortcutInfo>> activitiesToShortcuts 923 = new ArrayMap<>(); 924 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 925 final ShortcutInfo si = mShortcuts.valueAt(i); 926 if (si.isFloating()) { 927 continue; // Ignore floating shortcuts, which are not tied to any activities. 928 } 929 930 final ComponentName activity = si.getActivity(); 931 932 ArrayList<ShortcutInfo> list = activitiesToShortcuts.get(activity); 933 if (list == null) { 934 list = new ArrayList<>(); 935 activitiesToShortcuts.put(activity, list); 936 } 937 list.add(si); 938 } 939 return activitiesToShortcuts; 940 } 941 942 /** Used by {@link #enforceShortcutCountsBeforeOperation} */ 943 private void incrementCountForActivity(ArrayMap<ComponentName, Integer> counts, 944 ComponentName cn, int increment) { 945 Integer oldValue = counts.get(cn); 946 if (oldValue == null) { 947 oldValue = 0; 948 } 949 950 counts.put(cn, oldValue + increment); 951 } 952 953 /** 954 * Called by 955 * {@link android.content.pm.ShortcutManager#setDynamicShortcuts}, 956 * {@link android.content.pm.ShortcutManager#addDynamicShortcuts}, and 957 * {@link android.content.pm.ShortcutManager#updateShortcuts} before actually performing 958 * the operation to make sure the operation wouldn't result in the target activities having 959 * more than the allowed number of dynamic/manifest shortcuts. 960 * 961 * @param newList shortcut list passed to set, add or updateShortcuts(). 962 * @param operation add, set or update. 963 * @throws IllegalArgumentException if the operation would result in going over the max 964 * shortcut count for any activity. 965 */ 966 public void enforceShortcutCountsBeforeOperation(List<ShortcutInfo> newList, 967 @ShortcutOperation int operation) { 968 final ShortcutService service = mShortcutUser.mService; 969 970 // Current # of dynamic / manifest shortcuts for each activity. 971 // (If it's for update, then don't count dynamic shortcuts, since they'll be replaced 972 // anyway.) 973 final ArrayMap<ComponentName, Integer> counts = new ArrayMap<>(4); 974 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 975 final ShortcutInfo shortcut = mShortcuts.valueAt(i); 976 977 if (shortcut.isManifestShortcut()) { 978 incrementCountForActivity(counts, shortcut.getActivity(), 1); 979 } else if (shortcut.isDynamic() && (operation != ShortcutService.OPERATION_SET)) { 980 incrementCountForActivity(counts, shortcut.getActivity(), 1); 981 } 982 } 983 984 for (int i = newList.size() - 1; i >= 0; i--) { 985 final ShortcutInfo newShortcut = newList.get(i); 986 final ComponentName newActivity = newShortcut.getActivity(); 987 if (newActivity == null) { 988 if (operation != ShortcutService.OPERATION_UPDATE) { 989 service.wtf("Activity must not be null at this point"); 990 continue; // Just ignore this invalid case. 991 } 992 continue; // Activity can be null for update. 993 } 994 995 final ShortcutInfo original = mShortcuts.get(newShortcut.getId()); 996 if (original == null) { 997 if (operation == ShortcutService.OPERATION_UPDATE) { 998 continue; // When updating, ignore if there's no target. 999 } 1000 // Add() or set(), and there's no existing shortcut with the same ID. We're 1001 // simply publishing (as opposed to updating) this shortcut, so just +1. 1002 incrementCountForActivity(counts, newActivity, 1); 1003 continue; 1004 } 1005 if (original.isFloating() && (operation == ShortcutService.OPERATION_UPDATE)) { 1006 // Updating floating shortcuts doesn't affect the count, so ignore. 1007 continue; 1008 } 1009 1010 // If it's add() or update(), then need to decrement for the previous activity. 1011 // Skip it for set() since it's already been taken care of by not counting the original 1012 // dynamic shortcuts in the first loop. 1013 if (operation != ShortcutService.OPERATION_SET) { 1014 final ComponentName oldActivity = original.getActivity(); 1015 if (!original.isFloating()) { 1016 incrementCountForActivity(counts, oldActivity, -1); 1017 } 1018 } 1019 incrementCountForActivity(counts, newActivity, 1); 1020 } 1021 1022 // Then make sure none of the activities have more than the max number of shortcuts. 1023 for (int i = counts.size() - 1; i >= 0; i--) { 1024 service.enforceMaxActivityShortcuts(counts.valueAt(i)); 1025 } 1026 } 1027 1028 /** 1029 * For all the text fields, refresh the string values if they're from resources. 1030 */ 1031 public void resolveResourceStrings() { 1032 final ShortcutService s = mShortcutUser.mService; 1033 boolean changed = false; 1034 1035 Resources publisherRes = null; 1036 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 1037 final ShortcutInfo si = mShortcuts.valueAt(i); 1038 1039 if (si.hasStringResources()) { 1040 changed = true; 1041 1042 if (publisherRes == null) { 1043 publisherRes = getPackageResources(); 1044 if (publisherRes == null) { 1045 break; // Resources couldn't be loaded. 1046 } 1047 } 1048 1049 si.resolveResourceStrings(publisherRes); 1050 si.setTimestamp(s.injectCurrentTimeMillis()); 1051 } 1052 } 1053 if (changed) { 1054 s.packageShortcutsChanged(getPackageName(), getPackageUserId()); 1055 } 1056 } 1057 1058 /** Clears the implicit ranks for all shortcuts. */ 1059 public void clearAllImplicitRanks() { 1060 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 1061 final ShortcutInfo si = mShortcuts.valueAt(i); 1062 si.clearImplicitRankAndRankChangedFlag(); 1063 } 1064 } 1065 1066 /** 1067 * Used to sort shortcuts for rank auto-adjusting. 1068 */ 1069 final Comparator<ShortcutInfo> mShortcutRankComparator = (ShortcutInfo a, ShortcutInfo b) -> { 1070 // First, sort by rank. 1071 int ret = Integer.compare(a.getRank(), b.getRank()); 1072 if (ret != 0) { 1073 return ret; 1074 } 1075 // When ranks are tie, then prioritize the ones that have just been assigned new ranks. 1076 // e.g. when there are 3 shortcuts, "s1" "s2" and "s3" with rank 0, 1, 2 respectively, 1077 // adding a shortcut "s4" with rank 1 will "insert" it between "s1" and "s2", because 1078 // "s2" and "s4" have the same rank 1 but s4 has isRankChanged() set. 1079 // Similarly, updating s3's rank to 1 will insert it between s1 and s2. 1080 if (a.isRankChanged() != b.isRankChanged()) { 1081 return a.isRankChanged() ? -1 : 1; 1082 } 1083 // If they're still tie, sort by implicit rank -- i.e. preserve the order in which 1084 // they're passed to the API. 1085 ret = Integer.compare(a.getImplicitRank(), b.getImplicitRank()); 1086 if (ret != 0) { 1087 return ret; 1088 } 1089 // If they're stil tie, just sort by their IDs. 1090 // This may happen with updateShortcuts() -- see 1091 // the testUpdateShortcuts_noManifestShortcuts() test. 1092 return a.getId().compareTo(b.getId()); 1093 }; 1094 1095 /** 1096 * Re-calculate the ranks for all shortcuts. 1097 */ 1098 public void adjustRanks() { 1099 final ShortcutService s = mShortcutUser.mService; 1100 final long now = s.injectCurrentTimeMillis(); 1101 1102 // First, clear ranks for floating shortcuts. 1103 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 1104 final ShortcutInfo si = mShortcuts.valueAt(i); 1105 if (si.isFloating()) { 1106 if (si.getRank() != 0) { 1107 si.setTimestamp(now); 1108 si.setRank(0); 1109 } 1110 } 1111 } 1112 1113 // Then adjust ranks. Ranks are unique for each activity, so we first need to sort 1114 // shortcuts to each activity. 1115 // Then sort the shortcuts within each activity with mShortcutRankComparator, and 1116 // assign ranks from 0. 1117 final ArrayMap<ComponentName, ArrayList<ShortcutInfo>> all = 1118 sortShortcutsToActivities(); 1119 for (int outer = all.size() - 1; outer >= 0; outer--) { // For each activity. 1120 final ArrayList<ShortcutInfo> list = all.valueAt(outer); 1121 1122 // Sort by ranks and other signals. 1123 Collections.sort(list, mShortcutRankComparator); 1124 1125 int rank = 0; 1126 1127 final int size = list.size(); 1128 for (int i = 0; i < size; i++) { 1129 final ShortcutInfo si = list.get(i); 1130 if (si.isManifestShortcut()) { 1131 // Don't adjust ranks for manifest shortcuts. 1132 continue; 1133 } 1134 // At this point, it must be dynamic. 1135 if (!si.isDynamic()) { 1136 s.wtf("Non-dynamic shortcut found."); 1137 continue; 1138 } 1139 final int thisRank = rank++; 1140 if (si.getRank() != thisRank) { 1141 si.setTimestamp(now); 1142 si.setRank(thisRank); 1143 } 1144 } 1145 } 1146 } 1147 1148 public void dump(@NonNull PrintWriter pw, @NonNull String prefix) { 1149 pw.println(); 1150 1151 pw.print(prefix); 1152 pw.print("Package: "); 1153 pw.print(getPackageName()); 1154 pw.print(" UID: "); 1155 pw.print(mPackageUid); 1156 pw.println(); 1157 1158 pw.print(prefix); 1159 pw.print(" "); 1160 pw.print("Calls: "); 1161 pw.print(getApiCallCount()); 1162 pw.println(); 1163 1164 // getApiCallCount() may have updated mLastKnownForegroundElapsedTime. 1165 pw.print(prefix); 1166 pw.print(" "); 1167 pw.print("Last known FG: "); 1168 pw.print(mLastKnownForegroundElapsedTime); 1169 pw.println(); 1170 1171 // This should be after getApiCallCount(), which may update it. 1172 pw.print(prefix); 1173 pw.print(" "); 1174 pw.print("Last reset: ["); 1175 pw.print(mLastResetTime); 1176 pw.print("] "); 1177 pw.print(ShortcutService.formatTime(mLastResetTime)); 1178 pw.println(); 1179 1180 getPackageInfo().dump(pw, prefix + " "); 1181 pw.println(); 1182 1183 pw.print(prefix); 1184 pw.println(" Shortcuts:"); 1185 long totalBitmapSize = 0; 1186 final ArrayMap<String, ShortcutInfo> shortcuts = mShortcuts; 1187 final int size = shortcuts.size(); 1188 for (int i = 0; i < size; i++) { 1189 final ShortcutInfo si = shortcuts.valueAt(i); 1190 pw.print(prefix); 1191 pw.print(" "); 1192 pw.println(si.toInsecureString()); 1193 if (si.getBitmapPath() != null) { 1194 final long len = new File(si.getBitmapPath()).length(); 1195 pw.print(prefix); 1196 pw.print(" "); 1197 pw.print("bitmap size="); 1198 pw.println(len); 1199 1200 totalBitmapSize += len; 1201 } 1202 } 1203 pw.print(prefix); 1204 pw.print(" "); 1205 pw.print("Total bitmap size: "); 1206 pw.print(totalBitmapSize); 1207 pw.print(" ("); 1208 pw.print(Formatter.formatFileSize(mShortcutUser.mService.mContext, totalBitmapSize)); 1209 pw.println(")"); 1210 } 1211 1212 @Override 1213 public JSONObject dumpCheckin(boolean clear) throws JSONException { 1214 final JSONObject result = super.dumpCheckin(clear); 1215 1216 int numDynamic = 0; 1217 int numPinned = 0; 1218 int numManifest = 0; 1219 int numBitmaps = 0; 1220 long totalBitmapSize = 0; 1221 1222 final ArrayMap<String, ShortcutInfo> shortcuts = mShortcuts; 1223 final int size = shortcuts.size(); 1224 for (int i = 0; i < size; i++) { 1225 final ShortcutInfo si = shortcuts.valueAt(i); 1226 1227 if (si.isDynamic()) numDynamic++; 1228 if (si.isDeclaredInManifest()) numManifest++; 1229 if (si.isPinned()) numPinned++; 1230 1231 if (si.getBitmapPath() != null) { 1232 numBitmaps++; 1233 totalBitmapSize += new File(si.getBitmapPath()).length(); 1234 } 1235 } 1236 1237 result.put(KEY_DYNAMIC, numDynamic); 1238 result.put(KEY_MANIFEST, numManifest); 1239 result.put(KEY_PINNED, numPinned); 1240 result.put(KEY_BITMAPS, numBitmaps); 1241 result.put(KEY_BITMAP_BYTES, totalBitmapSize); 1242 1243 // TODO Log update frequency too. 1244 1245 return result; 1246 } 1247 1248 @Override 1249 public void saveToXml(@NonNull XmlSerializer out, boolean forBackup) 1250 throws IOException, XmlPullParserException { 1251 final int size = mShortcuts.size(); 1252 1253 if (size == 0 && mApiCallCount == 0) { 1254 return; // nothing to write. 1255 } 1256 1257 out.startTag(null, TAG_ROOT); 1258 1259 ShortcutService.writeAttr(out, ATTR_NAME, getPackageName()); 1260 ShortcutService.writeAttr(out, ATTR_CALL_COUNT, mApiCallCount); 1261 ShortcutService.writeAttr(out, ATTR_LAST_RESET, mLastResetTime); 1262 getPackageInfo().saveToXml(out); 1263 1264 for (int j = 0; j < size; j++) { 1265 saveShortcut(out, mShortcuts.valueAt(j), forBackup); 1266 } 1267 1268 out.endTag(null, TAG_ROOT); 1269 } 1270 1271 private static void saveShortcut(XmlSerializer out, ShortcutInfo si, boolean forBackup) 1272 throws IOException, XmlPullParserException { 1273 if (forBackup) { 1274 if (!(si.isPinned() && si.isEnabled())) { 1275 return; // We only backup pinned shortcuts that are enabled. 1276 } 1277 } 1278 out.startTag(null, TAG_SHORTCUT); 1279 ShortcutService.writeAttr(out, ATTR_ID, si.getId()); 1280 // writeAttr(out, "package", si.getPackageName()); // not needed 1281 ShortcutService.writeAttr(out, ATTR_ACTIVITY, si.getActivity()); 1282 // writeAttr(out, "icon", si.getIcon()); // We don't save it. 1283 ShortcutService.writeAttr(out, ATTR_TITLE, si.getTitle()); 1284 ShortcutService.writeAttr(out, ATTR_TITLE_RES_ID, si.getTitleResId()); 1285 ShortcutService.writeAttr(out, ATTR_TITLE_RES_NAME, si.getTitleResName()); 1286 ShortcutService.writeAttr(out, ATTR_TEXT, si.getText()); 1287 ShortcutService.writeAttr(out, ATTR_TEXT_RES_ID, si.getTextResId()); 1288 ShortcutService.writeAttr(out, ATTR_TEXT_RES_NAME, si.getTextResName()); 1289 ShortcutService.writeAttr(out, ATTR_DISABLED_MESSAGE, si.getDisabledMessage()); 1290 ShortcutService.writeAttr(out, ATTR_DISABLED_MESSAGE_RES_ID, 1291 si.getDisabledMessageResourceId()); 1292 ShortcutService.writeAttr(out, ATTR_DISABLED_MESSAGE_RES_NAME, 1293 si.getDisabledMessageResName()); 1294 ShortcutService.writeAttr(out, ATTR_TIMESTAMP, 1295 si.getLastChangedTimestamp()); 1296 if (forBackup) { 1297 // Don't write icon information. Also drop the dynamic flag. 1298 ShortcutService.writeAttr(out, ATTR_FLAGS, 1299 si.getFlags() & 1300 ~(ShortcutInfo.FLAG_HAS_ICON_FILE | ShortcutInfo.FLAG_HAS_ICON_RES 1301 | ShortcutInfo.FLAG_DYNAMIC)); 1302 } else { 1303 // When writing for backup, ranks shouldn't be saved, since shortcuts won't be restored 1304 // as dynamic. 1305 ShortcutService.writeAttr(out, ATTR_RANK, si.getRank()); 1306 1307 ShortcutService.writeAttr(out, ATTR_FLAGS, si.getFlags()); 1308 ShortcutService.writeAttr(out, ATTR_ICON_RES_ID, si.getIconResourceId()); 1309 ShortcutService.writeAttr(out, ATTR_ICON_RES_NAME, si.getIconResName()); 1310 ShortcutService.writeAttr(out, ATTR_BITMAP_PATH, si.getBitmapPath()); 1311 } 1312 1313 { 1314 final Set<String> cat = si.getCategories(); 1315 if (cat != null && cat.size() > 0) { 1316 out.startTag(null, TAG_CATEGORIES); 1317 XmlUtils.writeStringArrayXml(cat.toArray(new String[cat.size()]), 1318 NAME_CATEGORIES, out); 1319 out.endTag(null, TAG_CATEGORIES); 1320 } 1321 } 1322 final Intent[] intentsNoExtras = si.getIntentsNoExtras(); 1323 final PersistableBundle[] intentsExtras = si.getIntentPersistableExtrases(); 1324 final int numIntents = intentsNoExtras.length; 1325 for (int i = 0; i < numIntents; i++) { 1326 out.startTag(null, TAG_INTENT); 1327 ShortcutService.writeAttr(out, ATTR_INTENT_NO_EXTRA, intentsNoExtras[i]); 1328 ShortcutService.writeTagExtra(out, TAG_EXTRAS, intentsExtras[i]); 1329 out.endTag(null, TAG_INTENT); 1330 } 1331 1332 ShortcutService.writeTagExtra(out, TAG_EXTRAS, si.getExtras()); 1333 1334 out.endTag(null, TAG_SHORTCUT); 1335 } 1336 1337 public static ShortcutPackage loadFromXml(ShortcutService s, ShortcutUser shortcutUser, 1338 XmlPullParser parser, boolean fromBackup) 1339 throws IOException, XmlPullParserException { 1340 1341 final String packageName = ShortcutService.parseStringAttribute(parser, 1342 ATTR_NAME); 1343 1344 final ShortcutPackage ret = new ShortcutPackage(shortcutUser, 1345 shortcutUser.getUserId(), packageName); 1346 1347 ret.mApiCallCount = 1348 ShortcutService.parseIntAttribute(parser, ATTR_CALL_COUNT); 1349 ret.mLastResetTime = 1350 ShortcutService.parseLongAttribute(parser, ATTR_LAST_RESET); 1351 1352 final int outerDepth = parser.getDepth(); 1353 int type; 1354 while ((type = parser.next()) != XmlPullParser.END_DOCUMENT 1355 && (type != XmlPullParser.END_TAG || parser.getDepth() > outerDepth)) { 1356 if (type != XmlPullParser.START_TAG) { 1357 continue; 1358 } 1359 final int depth = parser.getDepth(); 1360 final String tag = parser.getName(); 1361 if (depth == outerDepth + 1) { 1362 switch (tag) { 1363 case ShortcutPackageInfo.TAG_ROOT: 1364 ret.getPackageInfo().loadFromXml(parser, fromBackup); 1365 continue; 1366 case TAG_SHORTCUT: 1367 final ShortcutInfo si = parseShortcut(parser, packageName, 1368 shortcutUser.getUserId()); 1369 1370 // Don't use addShortcut(), we don't need to save the icon. 1371 ret.mShortcuts.put(si.getId(), si); 1372 continue; 1373 } 1374 } 1375 ShortcutService.warnForInvalidTag(depth, tag); 1376 } 1377 return ret; 1378 } 1379 1380 private static ShortcutInfo parseShortcut(XmlPullParser parser, String packageName, 1381 @UserIdInt int userId) throws IOException, XmlPullParserException { 1382 String id; 1383 ComponentName activityComponent; 1384 // Icon icon; 1385 String title; 1386 int titleResId; 1387 String titleResName; 1388 String text; 1389 int textResId; 1390 String textResName; 1391 String disabledMessage; 1392 int disabledMessageResId; 1393 String disabledMessageResName; 1394 Intent intentLegacy; 1395 PersistableBundle intentPersistableExtrasLegacy = null; 1396 ArrayList<Intent> intents = new ArrayList<>(); 1397 int rank; 1398 PersistableBundle extras = null; 1399 long lastChangedTimestamp; 1400 int flags; 1401 int iconResId; 1402 String iconResName; 1403 String bitmapPath; 1404 ArraySet<String> categories = null; 1405 1406 id = ShortcutService.parseStringAttribute(parser, ATTR_ID); 1407 activityComponent = ShortcutService.parseComponentNameAttribute(parser, 1408 ATTR_ACTIVITY); 1409 title = ShortcutService.parseStringAttribute(parser, ATTR_TITLE); 1410 titleResId = ShortcutService.parseIntAttribute(parser, ATTR_TITLE_RES_ID); 1411 titleResName = ShortcutService.parseStringAttribute(parser, ATTR_TITLE_RES_NAME); 1412 text = ShortcutService.parseStringAttribute(parser, ATTR_TEXT); 1413 textResId = ShortcutService.parseIntAttribute(parser, ATTR_TEXT_RES_ID); 1414 textResName = ShortcutService.parseStringAttribute(parser, ATTR_TEXT_RES_NAME); 1415 disabledMessage = ShortcutService.parseStringAttribute(parser, ATTR_DISABLED_MESSAGE); 1416 disabledMessageResId = ShortcutService.parseIntAttribute(parser, 1417 ATTR_DISABLED_MESSAGE_RES_ID); 1418 disabledMessageResName = ShortcutService.parseStringAttribute(parser, 1419 ATTR_DISABLED_MESSAGE_RES_NAME); 1420 intentLegacy = ShortcutService.parseIntentAttributeNoDefault(parser, ATTR_INTENT_LEGACY); 1421 rank = (int) ShortcutService.parseLongAttribute(parser, ATTR_RANK); 1422 lastChangedTimestamp = ShortcutService.parseLongAttribute(parser, ATTR_TIMESTAMP); 1423 flags = (int) ShortcutService.parseLongAttribute(parser, ATTR_FLAGS); 1424 iconResId = (int) ShortcutService.parseLongAttribute(parser, ATTR_ICON_RES_ID); 1425 iconResName = ShortcutService.parseStringAttribute(parser, ATTR_ICON_RES_NAME); 1426 bitmapPath = ShortcutService.parseStringAttribute(parser, ATTR_BITMAP_PATH); 1427 1428 final int outerDepth = parser.getDepth(); 1429 int type; 1430 while ((type = parser.next()) != XmlPullParser.END_DOCUMENT 1431 && (type != XmlPullParser.END_TAG || parser.getDepth() > outerDepth)) { 1432 if (type != XmlPullParser.START_TAG) { 1433 continue; 1434 } 1435 final int depth = parser.getDepth(); 1436 final String tag = parser.getName(); 1437 if (ShortcutService.DEBUG_LOAD) { 1438 Slog.d(TAG, String.format(" depth=%d type=%d name=%s", 1439 depth, type, tag)); 1440 } 1441 switch (tag) { 1442 case TAG_INTENT_EXTRAS_LEGACY: 1443 intentPersistableExtrasLegacy = PersistableBundle.restoreFromXml(parser); 1444 continue; 1445 case TAG_INTENT: 1446 intents.add(parseIntent(parser)); 1447 continue; 1448 case TAG_EXTRAS: 1449 extras = PersistableBundle.restoreFromXml(parser); 1450 continue; 1451 case TAG_CATEGORIES: 1452 // This just contains string-array. 1453 continue; 1454 case TAG_STRING_ARRAY_XMLUTILS: 1455 if (NAME_CATEGORIES.equals(ShortcutService.parseStringAttribute(parser, 1456 ATTR_NAME_XMLUTILS))) { 1457 final String[] ar = XmlUtils.readThisStringArrayXml( 1458 parser, TAG_STRING_ARRAY_XMLUTILS, null); 1459 categories = new ArraySet<>(ar.length); 1460 for (int i = 0; i < ar.length; i++) { 1461 categories.add(ar[i]); 1462 } 1463 } 1464 continue; 1465 } 1466 throw ShortcutService.throwForInvalidTag(depth, tag); 1467 } 1468 1469 if (intentLegacy != null) { 1470 // For the legacy file format which supported only one intent per shortcut. 1471 ShortcutInfo.setIntentExtras(intentLegacy, intentPersistableExtrasLegacy); 1472 intents.clear(); 1473 intents.add(intentLegacy); 1474 } 1475 1476 return new ShortcutInfo( 1477 userId, id, packageName, activityComponent, /* icon =*/ null, 1478 title, titleResId, titleResName, text, textResId, textResName, 1479 disabledMessage, disabledMessageResId, disabledMessageResName, 1480 categories, 1481 intents.toArray(new Intent[intents.size()]), 1482 rank, extras, lastChangedTimestamp, flags, 1483 iconResId, iconResName, bitmapPath); 1484 } 1485 1486 private static Intent parseIntent(XmlPullParser parser) 1487 throws IOException, XmlPullParserException { 1488 1489 Intent intent = ShortcutService.parseIntentAttribute(parser, 1490 ATTR_INTENT_NO_EXTRA); 1491 1492 final int outerDepth = parser.getDepth(); 1493 int type; 1494 while ((type = parser.next()) != XmlPullParser.END_DOCUMENT 1495 && (type != XmlPullParser.END_TAG || parser.getDepth() > outerDepth)) { 1496 if (type != XmlPullParser.START_TAG) { 1497 continue; 1498 } 1499 final int depth = parser.getDepth(); 1500 final String tag = parser.getName(); 1501 if (ShortcutService.DEBUG_LOAD) { 1502 Slog.d(TAG, String.format(" depth=%d type=%d name=%s", 1503 depth, type, tag)); 1504 } 1505 switch (tag) { 1506 case TAG_EXTRAS: 1507 ShortcutInfo.setIntentExtras(intent, 1508 PersistableBundle.restoreFromXml(parser)); 1509 continue; 1510 } 1511 throw ShortcutService.throwForInvalidTag(depth, tag); 1512 } 1513 return intent; 1514 } 1515 1516 @VisibleForTesting 1517 List<ShortcutInfo> getAllShortcutsForTest() { 1518 return new ArrayList<>(mShortcuts.values()); 1519 } 1520 1521 @Override 1522 public void verifyStates() { 1523 super.verifyStates(); 1524 1525 boolean failed = false; 1526 1527 final ArrayMap<ComponentName, ArrayList<ShortcutInfo>> all = 1528 sortShortcutsToActivities(); 1529 1530 // Make sure each activity won't have more than max shortcuts. 1531 for (int outer = all.size() - 1; outer >= 0; outer--) { 1532 final ArrayList<ShortcutInfo> list = all.valueAt(outer); 1533 if (list.size() > mShortcutUser.mService.getMaxActivityShortcuts()) { 1534 failed = true; 1535 Log.e(TAG_VERIFY, "Package " + getPackageName() + ": activity " + all.keyAt(outer) 1536 + " has " + all.valueAt(outer).size() + " shortcuts."); 1537 } 1538 1539 // Sort by rank. 1540 Collections.sort(list, (a, b) -> Integer.compare(a.getRank(), b.getRank())); 1541 1542 // Split into two arrays for each kind. 1543 final ArrayList<ShortcutInfo> dynamicList = new ArrayList<>(list); 1544 dynamicList.removeIf((si) -> !si.isDynamic()); 1545 1546 final ArrayList<ShortcutInfo> manifestList = new ArrayList<>(list); 1547 dynamicList.removeIf((si) -> !si.isManifestShortcut()); 1548 1549 verifyRanksSequential(dynamicList); 1550 verifyRanksSequential(manifestList); 1551 } 1552 1553 // Verify each shortcut's status. 1554 for (int i = mShortcuts.size() - 1; i >= 0; i--) { 1555 final ShortcutInfo si = mShortcuts.valueAt(i); 1556 if (!(si.isDeclaredInManifest() || si.isDynamic() || si.isPinned())) { 1557 failed = true; 1558 Log.e(TAG_VERIFY, "Package " + getPackageName() + ": shortcut " + si.getId() 1559 + " is not manifest, dynamic or pinned."); 1560 } 1561 if (si.isDeclaredInManifest() && si.isDynamic()) { 1562 failed = true; 1563 Log.e(TAG_VERIFY, "Package " + getPackageName() + ": shortcut " + si.getId() 1564 + " is both dynamic and manifest at the same time."); 1565 } 1566 if (si.getActivity() == null) { 1567 failed = true; 1568 Log.e(TAG_VERIFY, "Package " + getPackageName() + ": shortcut " + si.getId() 1569 + " has null activity."); 1570 } 1571 if ((si.isDynamic() || si.isManifestShortcut()) && !si.isEnabled()) { 1572 failed = true; 1573 Log.e(TAG_VERIFY, "Package " + getPackageName() + ": shortcut " + si.getId() 1574 + " is not floating, but is disabled."); 1575 } 1576 if (si.isFloating() && si.getRank() != 0) { 1577 failed = true; 1578 Log.e(TAG_VERIFY, "Package " + getPackageName() + ": shortcut " + si.getId() 1579 + " is floating, but has rank=" + si.getRank()); 1580 } 1581 if (si.getIcon() != null) { 1582 failed = true; 1583 Log.e(TAG_VERIFY, "Package " + getPackageName() + ": shortcut " + si.getId() 1584 + " still has an icon"); 1585 } 1586 if (si.hasIconFile() && si.hasIconResource()) { 1587 failed = true; 1588 Log.e(TAG_VERIFY, "Package " + getPackageName() + ": shortcut " + si.getId() 1589 + " has both resource and bitmap icons"); 1590 } 1591 } 1592 1593 if (failed) { 1594 throw new IllegalStateException("See logcat for errors"); 1595 } 1596 } 1597 1598 private boolean verifyRanksSequential(List<ShortcutInfo> list) { 1599 boolean failed = false; 1600 1601 for (int i = 0; i < list.size(); i++) { 1602 final ShortcutInfo si = list.get(i); 1603 if (si.getRank() != i) { 1604 failed = true; 1605 Log.e(TAG_VERIFY, "Package " + getPackageName() + ": shortcut " + si.getId() 1606 + " rank=" + si.getRank() + " but expected to be "+ i); 1607 } 1608 } 1609 return failed; 1610 } 1611 } 1612