/*
* Copyright (C) 2018 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.android.server.notification;
import android.app.ActivityManager;
import android.app.INotificationManager;
import android.app.Notification;
import android.app.NotificationChannel;
import android.app.NotificationManager;
import android.app.PendingIntent;
import android.app.Person;
import android.content.ComponentName;
import android.content.Context;
import android.content.Intent;
import android.content.pm.ParceledListSlice;
import android.content.res.Resources;
import android.graphics.drawable.BitmapDrawable;
import android.graphics.drawable.Drawable;
import android.graphics.drawable.Icon;
import android.net.Uri;
import android.os.Binder;
import android.os.RemoteException;
import android.os.ShellCommand;
import android.os.UserHandle;
import android.text.TextUtils;
import android.util.Slog;
import java.io.PrintWriter;
import java.net.URISyntaxException;
import java.util.Collections;
/**
* Implementation of `cmd notification` in NotificationManagerService.
*/
public class NotificationShellCmd extends ShellCommand {
private static final String USAGE =
"usage: cmd notification SUBCMD [args]\n\n"
+ "SUBCMDs:\n"
+ " allow_listener COMPONENT [user_id (current user if not specified)]\n"
+ " disallow_listener COMPONENT [user_id (current user if not specified)]\n"
+ " allow_assistant COMPONENT [user_id (current user if not specified)]\n"
+ " remove_assistant COMPONENT [user_id (current user if not specified)]\n"
+ " allow_dnd PACKAGE [user_id (current user if not specified)]\n"
+ " disallow_dnd PACKAGE [user_id (current user if not specified)]\n"
+ " suspend_package PACKAGE\n"
+ " unsuspend_package PACKAGE\n"
+ " reset_assistant_user_set [user_id (current user if not specified)]\n"
+ " get_approved_assistant [user_id (current user if not specified)]\n"
+ " post [--help | flags] TAG TEXT";
private static final String NOTIFY_USAGE =
"usage: cmd notification post [flags] <tag> <text>\n\n"
+ "flags:\n"
+ " -h|--help\n"
+ " -v|--verbose\n"
+ " -t|--title <text>\n"
+ " -i|--icon <iconspec>\n"
+ " -I|--large-icon <iconspec>\n"
+ " -S|--style <style> [styleargs]\n"
+ " -c|--content-intent <intentspec>\n"
+ "\n"
+ "styles: (default none)\n"
+ " bigtext\n"
+ " bigpicture --picture <iconspec>\n"
+ " inbox --line <text> --line <text> ...\n"
+ " messaging --conversation <title> --message <who>:<text> ...\n"
+ " media\n"
+ "\n"
+ "an <iconspec> is one of\n"
+ " file:///data/local/tmp/<img.png>\n"
+ " content://<provider>/<path>\n"
+ " @[<package>:]drawable/<img>\n"
+ " data:base64,<B64DATA==>\n"
+ "\n"
+ "an <intentspec> is (broadcast|service|activity) <args>\n"
+ " <args> are as described in `am start`";
public static final int NOTIFICATION_ID = 1138;
public static final String NOTIFICATION_PACKAGE = "com.android.shell";
public static final String CHANNEL_ID = "shellcmd";
public static final String CHANNEL_NAME = "Shell command";
public static final int CHANNEL_IMP = NotificationManager.IMPORTANCE_DEFAULT;
private final NotificationManagerService mDirectService;
private final INotificationManager mBinderService;
public NotificationShellCmd(NotificationManagerService service) {
mDirectService = service;
mBinderService = service.getBinderService();
}
@Override
public int onCommand(String cmd) {
if (cmd == null) {
return handleDefaultCommands(cmd);
}
final PrintWriter pw = getOutPrintWriter();
try {
switch (cmd.replace('-', '_')) {
case "allow_dnd": {
String packageName = getNextArgRequired();
int userId = ActivityManager.getCurrentUser();
if (peekNextArg() != null) {
userId = Integer.parseInt(getNextArgRequired());
}
mBinderService.setNotificationPolicyAccessGrantedForUser(
packageName, userId, true);
}
break;
case "disallow_dnd": {
String packageName = getNextArgRequired();
int userId = ActivityManager.getCurrentUser();
if (peekNextArg() != null) {
userId = Integer.parseInt(getNextArgRequired());
}
mBinderService.setNotificationPolicyAccessGrantedForUser(
packageName, userId, false);
}
break;
case "allow_listener": {
ComponentName cn = ComponentName.unflattenFromString(getNextArgRequired());
if (cn == null) {
pw.println("Invalid listener - must be a ComponentName");
return -1;
}
int userId = ActivityManager.getCurrentUser();
if (peekNextArg() != null) {
userId = Integer.parseInt(getNextArgRequired());
}
mBinderService.setNotificationListenerAccessGrantedForUser(cn, userId, true);
}
break;
case "disallow_listener": {
ComponentName cn = ComponentName.unflattenFromString(getNextArgRequired());
if (cn == null) {
pw.println("Invalid listener - must be a ComponentName");
return -1;
}
int userId = ActivityManager.getCurrentUser();
if (peekNextArg() != null) {
userId = Integer.parseInt(getNextArgRequired());
}
mBinderService.setNotificationListenerAccessGrantedForUser(cn, userId, false);
}
break;
case "allow_assistant": {
ComponentName cn = ComponentName.unflattenFromString(getNextArgRequired());
if (cn == null) {
pw.println("Invalid assistant - must be a ComponentName");
return -1;
}
int userId = ActivityManager.getCurrentUser();
if (peekNextArg() != null) {
userId = Integer.parseInt(getNextArgRequired());
}
mBinderService.setNotificationAssistantAccessGrantedForUser(cn, userId, true);
}
break;
case "disallow_assistant": {
ComponentName cn = ComponentName.unflattenFromString(getNextArgRequired());
if (cn == null) {
pw.println("Invalid assistant - must be a ComponentName");
return -1;
}
int userId = ActivityManager.getCurrentUser();
if (peekNextArg() != null) {
userId = Integer.parseInt(getNextArgRequired());
}
mBinderService.setNotificationAssistantAccessGrantedForUser(cn, userId, false);
}
break;
case "suspend_package": {
// only use for testing
mDirectService.simulatePackageSuspendBroadcast(true, getNextArgRequired());
}
break;
case "unsuspend_package": {
// only use for testing
mDirectService.simulatePackageSuspendBroadcast(false, getNextArgRequired());
}
break;
case "distract_package": {
// only use for testing
// Flag values are in
// {@link android.content.pm.PackageManager.DistractionRestriction}.
mDirectService.simulatePackageDistractionBroadcast(
Integer.parseInt(getNextArgRequired()),
getNextArgRequired().split(","));
break;
}
case "reset_assistant_user_set": {
int userId = ActivityManager.getCurrentUser();
if (peekNextArg() != null) {
userId = Integer.parseInt(getNextArgRequired());
}
mDirectService.resetAssistantUserSet(userId);
break;
}
case "get_approved_assistant": {
int userId = ActivityManager.getCurrentUser();
if (peekNextArg() != null) {
userId = Integer.parseInt(getNextArgRequired());
}
ComponentName approvedAssistant = mDirectService.getApprovedAssistant(userId);
if (approvedAssistant == null) {
pw.println("null");
} else {
pw.println(approvedAssistant.flattenToString());
}
break;
}
case "post":
case "notify":
doNotify(pw);
break;
default:
return handleDefaultCommands(cmd);
}
} catch (Exception e) {
pw.println("Error occurred. Check logcat for details. " + e.getMessage());
Slog.e(NotificationManagerService.TAG, "Error running shell command", e);
}
return 0;
}
void ensureChannel() throws RemoteException {
final int uid = Binder.getCallingUid();
final int userid = UserHandle.getCallingUserId();
final long token = Binder.clearCallingIdentity();
try {
if (mBinderService.getNotificationChannelForPackage(NOTIFICATION_PACKAGE,
uid, CHANNEL_ID, false) == null) {
final NotificationChannel chan = new NotificationChannel(CHANNEL_ID, CHANNEL_NAME,
CHANNEL_IMP);
Slog.v(NotificationManagerService.TAG,
"creating shell channel for user " + userid + " uid " + uid + ": " + chan);
mBinderService.createNotificationChannelsForPackage(NOTIFICATION_PACKAGE, uid,
new ParceledListSlice<NotificationChannel>(
Collections.singletonList(chan)));
Slog.v(NotificationManagerService.TAG, "created channel: "
+ mBinderService.getNotificationChannelForPackage(NOTIFICATION_PACKAGE,
uid, CHANNEL_ID, false));
}
} finally {
Binder.restoreCallingIdentity(token);
}
}
Icon parseIcon(Resources res, String encoded) throws IllegalArgumentException {
if (TextUtils.isEmpty(encoded)) return null;
if (encoded.startsWith("/")) {
encoded = "file://" + encoded;
}
if (encoded.startsWith("http:")
|| encoded.startsWith("https:")
|| encoded.startsWith("content:")
|| encoded.startsWith("file:")
|| encoded.startsWith("android.resource:")) {
Uri asUri = Uri.parse(encoded);
return Icon.createWithContentUri(asUri);
} else if (encoded.startsWith("@")) {
final int resid = res.getIdentifier(encoded.substring(1),
"drawable", "android");
if (resid != 0) {
return Icon.createWithResource(res, resid);
}
} else if (encoded.startsWith("data:")) {
encoded = encoded.substring(encoded.indexOf(',') + 1);
byte[] bits = android.util.Base64.decode(encoded, android.util.Base64.DEFAULT);
return Icon.createWithData(bits, 0, bits.length);
}
return null;
}
private int doNotify(PrintWriter pw) throws RemoteException, URISyntaxException {
final Context context = mDirectService.getContext();
final Resources res = context.getResources();
final Notification.Builder builder = new Notification.Builder(context, CHANNEL_ID);
String opt;
boolean verbose = false;
Notification.BigPictureStyle bigPictureStyle = null;
Notification.BigTextStyle bigTextStyle = null;
Notification.InboxStyle inboxStyle = null;
Notification.MediaStyle mediaStyle = null;
Notification.MessagingStyle messagingStyle = null;
Icon smallIcon = null;
while ((opt = getNextOption()) != null) {
boolean large = false;
switch (opt) {
case "-v":
case "--verbose":
verbose = true;
break;
case "-t":
case "--title":
case "title":
builder.setContentTitle(getNextArgRequired());
break;
case "-I":
case "--large-icon":
case "--largeicon":
case "largeicon":
case "large-icon":
large = true;
// fall through
case "-i":
case "--icon":
case "icon":
final String iconSpec = getNextArgRequired();
final Icon icon = parseIcon(res, iconSpec);
if (icon == null) {
pw.println("error: invalid icon: " + iconSpec);
return -1;
}
if (large) {
builder.setLargeIcon(icon);
large = false;
} else {
smallIcon = icon;
}
break;
case "-c":
case "--content-intent":
case "content-intent":
case "--intent":
case "intent":
String intentKind = null;
switch (peekNextArg()) {
case "broadcast":
case "service":
case "activity":
intentKind = getNextArg();
}
final Intent intent = Intent.parseCommandArgs(this, null);
if (intent.getData() == null) {
// force unique intents unless you know what you're doing
intent.setData(Uri.parse("xyz:" + System.currentTimeMillis()));
}
final PendingIntent pi;
if ("broadcast".equals(intentKind)) {
pi = PendingIntent.getBroadcastAsUser(
context, 0, intent, PendingIntent.FLAG_UPDATE_CURRENT,
UserHandle.CURRENT);
} else if ("service".equals(intentKind)) {
pi = PendingIntent.getService(
context, 0, intent, PendingIntent.FLAG_UPDATE_CURRENT);
} else {
pi = PendingIntent.getActivityAsUser(
context, 0, intent, PendingIntent.FLAG_UPDATE_CURRENT, null,
UserHandle.CURRENT);
}
builder.setContentIntent(pi);
break;
case "-S":
case "--style":
final String styleSpec = getNextArgRequired().toLowerCase();
switch (styleSpec) {
case "bigtext":
bigTextStyle = new Notification.BigTextStyle();
builder.setStyle(bigTextStyle);
break;
case "bigpicture":
bigPictureStyle = new Notification.BigPictureStyle();
builder.setStyle(bigPictureStyle);
break;
case "inbox":
inboxStyle = new Notification.InboxStyle();
builder.setStyle(inboxStyle);
break;
case "messaging":
String name = "You";
if ("--user".equals(peekNextArg())) {
getNextArg();
name = getNextArgRequired();
}
messagingStyle = new Notification.MessagingStyle(
new Person.Builder().setName(name).build());
builder.setStyle(messagingStyle);
break;
case "media":
mediaStyle = new Notification.MediaStyle();
builder.setStyle(mediaStyle);
break;
default:
throw new IllegalArgumentException(
"unrecognized notification style: " + styleSpec);
}
break;
case "--bigText": case "--bigtext": case "--big-text":
if (bigTextStyle == null) {
throw new IllegalArgumentException("--bigtext requires --style bigtext");
}
bigTextStyle.bigText(getNextArgRequired());
break;
case "--picture":
if (bigPictureStyle == null) {
throw new IllegalArgumentException("--picture requires --style bigpicture");
}
final String pictureSpec = getNextArgRequired();
final Icon pictureAsIcon = parseIcon(res, pictureSpec);
if (pictureAsIcon == null) {
throw new IllegalArgumentException("bad picture spec: " + pictureSpec);
}
final Drawable d = pictureAsIcon.loadDrawable(context);
if (d instanceof BitmapDrawable) {
bigPictureStyle.bigPicture(((BitmapDrawable) d).getBitmap());
} else {
throw new IllegalArgumentException("not a bitmap: " + pictureSpec);
}
break;
case "--line":
if (inboxStyle == null) {
throw new IllegalArgumentException("--line requires --style inbox");
}
inboxStyle.addLine(getNextArgRequired());
break;
case "--message":
if (messagingStyle == null) {
throw new IllegalArgumentException(
"--message requires --style messaging");
}
String arg = getNextArgRequired();
String[] parts = arg.split(":", 2);
if (parts.length > 1) {
messagingStyle.addMessage(parts[1], System.currentTimeMillis(),
parts[0]);
} else {
messagingStyle.addMessage(parts[0], System.currentTimeMillis(),
new String[]{
messagingStyle.getUserDisplayName().toString(),
"Them"
}[messagingStyle.getMessages().size() % 2]);
}
break;
case "--conversation":
if (messagingStyle == null) {
throw new IllegalArgumentException(
"--conversation requires --style messaging");
}
messagingStyle.setConversationTitle(getNextArgRequired());
break;
case "-h":
case "--help":
case "--wtf":
default:
pw.println(NOTIFY_USAGE);
return 0;
}
}
final String tag = getNextArg();
final String text = getNextArg();
if (tag == null || text == null) {
pw.println(NOTIFY_USAGE);
return -1;
}
builder.setContentText(text);
if (smallIcon == null) {
// uh oh, let's substitute something
builder.setSmallIcon(com.android.internal.R.drawable.stat_notify_chat);
} else {
builder.setSmallIcon(smallIcon);
}
ensureChannel();
final Notification n = builder.build();
pw.println("posting:\n " + n);
Slog.v("NotificationManager", "posting: " + n);
final int userId = UserHandle.getCallingUserId();
final long token = Binder.clearCallingIdentity();
try {
mBinderService.enqueueNotificationWithTag(
NOTIFICATION_PACKAGE, "android",
tag, NOTIFICATION_ID,
n, userId);
} finally {
Binder.restoreCallingIdentity(token);
}
if (verbose) {
NotificationRecord nr = mDirectService.findNotificationLocked(
NOTIFICATION_PACKAGE, tag, NOTIFICATION_ID, userId);
for (int tries = 3; tries-- > 0; ) {
if (nr != null) break;
try {
pw.println("waiting for notification to post...");
Thread.sleep(500);
} catch (InterruptedException e) {
}
nr = mDirectService.findNotificationLocked(
NOTIFICATION_PACKAGE, tag, NOTIFICATION_ID, userId);
}
if (nr == null) {
pw.println("warning: couldn't find notification after enqueueing");
} else {
pw.println("posted: ");
nr.dump(pw, " ", context, false);
}
}
return 0;
}
@Override
public void onHelp() {
getOutPrintWriter().println(USAGE);
}
}