1 /* 2 * Copyright (C) 2015 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 17 package com.android.cts.verifier.managedprovisioning; 18 19 import android.app.AlertDialog; 20 import android.app.admin.DevicePolicyManager; 21 import android.content.ComponentName; 22 import android.content.Context; 23 import android.content.Intent; 24 import android.net.VpnService; 25 import android.net.VpnService.Builder; 26 import android.os.Bundle; 27 import android.os.ParcelFileDescriptor; 28 import android.os.UserManager; 29 import android.provider.Settings; 30 import android.util.Log; 31 import android.widget.TextView; 32 import android.net.VpnService; 33 import android.os.ParcelFileDescriptor; 34 35 import com.android.cts.verifier.PassFailButtons; 36 import com.android.cts.verifier.R; 37 38 import java.io.IOException; 39 40 /** 41 * Activity to test Vpn configuration 42 */ 43 public class VpnTestActivity extends PassFailButtons.Activity { 44 45 public static final String ACTION_VPN = "com.android.cts.verifier.managedprovisioning.VPN"; 46 47 public static class MyTestVpnService extends VpnService { 48 /* 49 * MyVpnTestService is just a stub. This class exists because the framework needs a class 50 * inside the app to refer back to, just using VpnService itself won't work. 51 */ 52 } 53 54 private ParcelFileDescriptor descriptor = null; 55 private ComponentName mAdminReceiverComponent; 56 private DevicePolicyManager mDevicePolicyManager; 57 private UserManager mUserManager; 58 private static final String TAG = "DeviceOwnerPositiveTestActivity"; 59 private static final int REQUEST_VPN_CODE = 54321; 60 61 @Override onCreate(Bundle savedInstanceState)62 protected void onCreate(Bundle savedInstanceState) { 63 super.onCreate(savedInstanceState); 64 setContentView(R.layout.vpn_test); 65 setPassFailButtonClickListeners(); 66 mAdminReceiverComponent = new ComponentName(this, DeviceAdminTestReceiver.class.getName()); 67 mDevicePolicyManager = (DevicePolicyManager) getSystemService( 68 Context.DEVICE_POLICY_SERVICE); 69 mDevicePolicyManager.addUserRestriction(mAdminReceiverComponent, 70 UserManager.DISALLOW_CONFIG_VPN); 71 mUserManager = (UserManager) getSystemService(Context.USER_SERVICE); 72 testVpnEstablishFails(); 73 } 74 75 @Override finish()76 public void finish() { 77 mDevicePolicyManager.clearUserRestriction(mAdminReceiverComponent, 78 UserManager.DISALLOW_CONFIG_VPN); 79 super.finish(); 80 } 81 82 @Override onActivityResult(int requestCode, int result, Intent data)83 protected void onActivityResult(int requestCode, int result, Intent data) { 84 if (requestCode == REQUEST_VPN_CODE) { 85 // We don't care about the result - ideally it should automatically cancel, but if 86 // some custom component doesn't do that, try to establish the connection anyway 87 // and see what happens. 88 establishVpn(); 89 } else { 90 Log.w(TAG, "Unexpected request code: " + requestCode); 91 } 92 } 93 testVpnEstablishFails()94 public void testVpnEstablishFails() { 95 Intent newIntent = VpnService.prepare(this); 96 if (newIntent != null) { 97 startActivityForResult(newIntent, REQUEST_VPN_CODE); 98 } else { 99 establishVpn(); 100 } 101 } 102 establishVpn()103 public void establishVpn() { 104 MyTestVpnService service = new MyTestVpnService(); 105 descriptor = service.new Builder().addAddress("8.8.8.8", 30).establish(); 106 if (descriptor == null) { 107 // vpn connection not established, as expected, test case succeeds 108 Log.i(TAG, "Test succeeded: descriptor is null"); 109 populateInfo(R.string.device_owner_no_vpn_connection); 110 return; 111 } 112 // vpn connection established, not expected, test case fails 113 Log.w(TAG, "vpn connection established, not expected, test case fails"); 114 try { 115 descriptor.close(); 116 populateInfo(R.string.device_owner_vpn_connection); 117 } catch (IOException e) { 118 Log.i(TAG, "Closing vpn connection failed. Caught exception: ", e); 119 populateInfo(R.string.device_owner_vpn_connection_close_failed); 120 } 121 } 122 populateInfo(int messageId)123 private void populateInfo(int messageId) { 124 TextView vpnInfoTextView = (TextView) findViewById(R.id.device_owner_vpn_info); 125 vpnInfoTextView.setText(getString(messageId)); 126 } 127 128 } 129