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