1/*
2 * Copyright (C) 2018 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
17package com.android.server.connectivity;
18
19import static android.Manifest.permission.CHANGE_NETWORK_STATE;
20import static android.Manifest.permission.CHANGE_WIFI_STATE;
21import static android.Manifest.permission.CONNECTIVITY_INTERNAL;
22import static android.Manifest.permission.CONNECTIVITY_USE_RESTRICTED_NETWORKS;
23import static android.Manifest.permission.NETWORK_STACK;
24import static android.content.pm.ApplicationInfo.FLAG_SYSTEM;
25import static android.content.pm.PackageManager.GET_PERMISSIONS;
26
27import static org.junit.Assert.assertFalse;
28import static org.junit.Assert.assertTrue;
29import static org.mockito.Mockito.anyInt;
30import static org.mockito.Mockito.eq;
31import static org.mockito.Mockito.when;
32
33import android.content.Context;
34import android.content.pm.ApplicationInfo;
35import android.content.pm.PackageInfo;
36import android.content.pm.PackageManager;
37import android.support.test.filters.SmallTest;
38import android.support.test.runner.AndroidJUnit4;
39
40import org.junit.Before;
41import org.junit.Test;
42import org.junit.runner.RunWith;
43import org.mockito.Mock;
44import org.mockito.MockitoAnnotations;
45
46@RunWith(AndroidJUnit4.class)
47@SmallTest
48public class PermissionMonitorTest {
49    private static final int MOCK_UID = 10001;
50    private static final String[] MOCK_PACKAGE_NAMES = new String[] { "com.foo.bar" };
51
52    @Mock private Context mContext;
53    @Mock private PackageManager mPackageManager;
54
55    private PermissionMonitor mPermissionMonitor;
56
57    @Before
58    public void setUp() throws Exception {
59        MockitoAnnotations.initMocks(this);
60        when(mContext.getPackageManager()).thenReturn(mPackageManager);
61        when(mPackageManager.getPackagesForUid(MOCK_UID)).thenReturn(MOCK_PACKAGE_NAMES);
62        mPermissionMonitor = new PermissionMonitor(mContext, null);
63    }
64
65    private void expectPermission(String[] permissions, boolean preinstalled) throws Exception {
66        final PackageInfo packageInfo = packageInfoWithPermissions(permissions, preinstalled);
67        when(mPackageManager.getPackageInfoAsUser(
68                eq(MOCK_PACKAGE_NAMES[0]), eq(GET_PERMISSIONS), anyInt())).thenReturn(packageInfo);
69    }
70
71    private PackageInfo packageInfoWithPermissions(String[] permissions, boolean preinstalled) {
72        final PackageInfo packageInfo = new PackageInfo();
73        packageInfo.requestedPermissions = permissions;
74        packageInfo.applicationInfo = new ApplicationInfo();
75        packageInfo.applicationInfo.flags = preinstalled ? FLAG_SYSTEM : 0;
76        return packageInfo;
77    }
78
79    @Test
80    public void testHasPermission() {
81        PackageInfo app = packageInfoWithPermissions(new String[] {}, false);
82        assertFalse(mPermissionMonitor.hasPermission(app, CHANGE_NETWORK_STATE));
83        assertFalse(mPermissionMonitor.hasPermission(app, NETWORK_STACK));
84        assertFalse(mPermissionMonitor.hasPermission(app, CONNECTIVITY_USE_RESTRICTED_NETWORKS));
85        assertFalse(mPermissionMonitor.hasPermission(app, CONNECTIVITY_INTERNAL));
86
87        app = packageInfoWithPermissions(new String[] {
88                CHANGE_NETWORK_STATE, NETWORK_STACK
89            }, false);
90        assertTrue(mPermissionMonitor.hasPermission(app, CHANGE_NETWORK_STATE));
91        assertTrue(mPermissionMonitor.hasPermission(app, NETWORK_STACK));
92        assertFalse(mPermissionMonitor.hasPermission(app, CONNECTIVITY_USE_RESTRICTED_NETWORKS));
93        assertFalse(mPermissionMonitor.hasPermission(app, CONNECTIVITY_INTERNAL));
94
95        app = packageInfoWithPermissions(new String[] {
96                CONNECTIVITY_USE_RESTRICTED_NETWORKS, CONNECTIVITY_INTERNAL
97            }, false);
98        assertFalse(mPermissionMonitor.hasPermission(app, CHANGE_NETWORK_STATE));
99        assertFalse(mPermissionMonitor.hasPermission(app, NETWORK_STACK));
100        assertTrue(mPermissionMonitor.hasPermission(app, CONNECTIVITY_USE_RESTRICTED_NETWORKS));
101        assertTrue(mPermissionMonitor.hasPermission(app, CONNECTIVITY_INTERNAL));
102    }
103
104    @Test
105    public void testIsPreinstalledSystemApp() {
106        PackageInfo app = packageInfoWithPermissions(new String[] {}, false);
107        assertFalse(mPermissionMonitor.isPreinstalledSystemApp(app));
108
109        app = packageInfoWithPermissions(new String[] {}, true);
110        assertTrue(mPermissionMonitor.isPreinstalledSystemApp(app));
111    }
112
113    @Test
114    public void testHasUseBackgroundNetworksPermission() throws Exception {
115        expectPermission(new String[] { CHANGE_NETWORK_STATE }, false);
116        assertTrue(mPermissionMonitor.hasUseBackgroundNetworksPermission(MOCK_UID));
117
118        expectPermission(new String[] { NETWORK_STACK, CONNECTIVITY_INTERNAL }, false);
119        assertTrue(mPermissionMonitor.hasUseBackgroundNetworksPermission(MOCK_UID));
120
121        // TODO : make this false when b/31479477 is fixed
122        expectPermission(new String[] {}, true);
123        assertTrue(mPermissionMonitor.hasUseBackgroundNetworksPermission(MOCK_UID));
124        expectPermission(new String[] { CHANGE_WIFI_STATE }, true);
125        assertTrue(mPermissionMonitor.hasUseBackgroundNetworksPermission(MOCK_UID));
126
127        expectPermission(new String[] { NETWORK_STACK, CONNECTIVITY_INTERNAL }, true);
128        assertTrue(mPermissionMonitor.hasUseBackgroundNetworksPermission(MOCK_UID));
129
130        expectPermission(new String[] {}, false);
131        assertFalse(mPermissionMonitor.hasUseBackgroundNetworksPermission(MOCK_UID));
132
133        expectPermission(new String[] { CHANGE_WIFI_STATE }, false);
134        assertFalse(mPermissionMonitor.hasUseBackgroundNetworksPermission(MOCK_UID));
135    }
136}
137