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
17package com.android.settings.wifi;
18
19import android.content.Context;
20import android.content.DialogInterface;
21import android.content.Intent;
22import android.net.ConnectivityManager;
23import android.net.ConnectivityManager.NetworkCallback;
24import android.net.Network;
25import android.net.NetworkCapabilities;
26import android.net.NetworkInfo;
27import android.net.NetworkRequest;
28import android.os.Bundle;
29import android.provider.Settings;
30import android.util.Log;
31import android.view.LayoutInflater;
32import android.view.View;
33import android.widget.CheckBox;
34
35import com.android.internal.app.AlertActivity;
36import com.android.internal.app.AlertController;
37import com.android.settings.R;
38
39import static android.net.ConnectivityManager.ACTION_PROMPT_LOST_VALIDATION;
40import static android.net.ConnectivityManager.ACTION_PROMPT_UNVALIDATED;
41import static android.net.NetworkCapabilities.NET_CAPABILITY_VALIDATED;
42
43public final class WifiNoInternetDialog extends AlertActivity implements
44        DialogInterface.OnClickListener {
45    private static final String TAG = "WifiNoInternetDialog";
46
47    private ConnectivityManager mCM;
48    private Network mNetwork;
49    private String mNetworkName;
50    private ConnectivityManager.NetworkCallback mNetworkCallback;
51    private CheckBox mAlwaysAllow;
52    private String mAction;
53
54    private boolean isKnownAction(Intent intent) {
55        return intent.getAction().equals(ACTION_PROMPT_UNVALIDATED) ||
56                intent.getAction().equals(ACTION_PROMPT_LOST_VALIDATION);
57    }
58
59    @Override
60    public void onCreate(Bundle savedInstanceState) {
61        super.onCreate(savedInstanceState);
62
63        final Intent intent = getIntent();
64        if (intent == null || !isKnownAction(intent) || !"netId".equals(intent.getScheme())) {
65            Log.e(TAG, "Unexpected intent " + intent + ", exiting");
66            finish();
67            return;
68        }
69
70        mAction = intent.getAction();
71
72        try {
73            mNetwork = new Network(Integer.parseInt(intent.getData().getSchemeSpecificPart()));
74        } catch (NullPointerException|NumberFormatException e) {
75            mNetwork = null;
76        }
77
78        if (mNetwork == null) {
79            Log.e(TAG, "Can't determine network from '" + intent.getData() + "' , exiting");
80            finish();
81            return;
82        }
83
84        // TODO: add a registerNetworkCallback(Network network, NetworkCallback networkCallback) and
85        // simplify this.
86        final NetworkRequest request = new NetworkRequest.Builder().clearCapabilities().build();
87        mNetworkCallback = new NetworkCallback() {
88            @Override
89            public void onLost(Network network) {
90                // Close the dialog if the network disconnects.
91                if (mNetwork.equals(network)) {
92                    Log.d(TAG, "Network " + mNetwork + " disconnected");
93                    finish();
94                }
95            }
96            @Override
97            public void onCapabilitiesChanged(Network network, NetworkCapabilities nc) {
98                // Close the dialog if the network validates.
99                if (mNetwork.equals(network) && nc.hasCapability(NET_CAPABILITY_VALIDATED)) {
100                    Log.d(TAG, "Network " + mNetwork + " validated");
101                    finish();
102                }
103            }
104        };
105
106        mCM = (ConnectivityManager) getSystemService(Context.CONNECTIVITY_SERVICE);
107        mCM.registerNetworkCallback(request, mNetworkCallback);
108
109        final NetworkInfo ni = mCM.getNetworkInfo(mNetwork);
110        if (ni == null || !ni.isConnectedOrConnecting()) {
111            Log.d(TAG, "Network " + mNetwork + " is not connected: " + ni);
112            finish();
113            return;
114        }
115        mNetworkName = ni.getExtraInfo();
116        if (mNetworkName != null) {
117            mNetworkName = mNetworkName.replaceAll("^\"|\"$", "");  // Remove double quotes
118        }
119
120        createDialog();
121    }
122
123    private void createDialog() {
124        mAlert.setIcon(R.drawable.ic_settings_wireless);
125
126        final AlertController.AlertParams ap = mAlertParams;
127        if (ACTION_PROMPT_UNVALIDATED.equals(mAction)) {
128            ap.mTitle = mNetworkName;
129            ap.mMessage = getString(R.string.no_internet_access_text);
130            ap.mPositiveButtonText = getString(R.string.yes);
131            ap.mNegativeButtonText = getString(R.string.no);
132        } else {
133            ap.mTitle = getString(R.string.lost_internet_access_title);
134            ap.mMessage = getString(R.string.lost_internet_access_text);
135            ap.mPositiveButtonText = getString(R.string.lost_internet_access_switch);
136            ap.mNegativeButtonText = getString(R.string.lost_internet_access_cancel);
137        }
138        ap.mPositiveButtonListener = this;
139        ap.mNegativeButtonListener = this;
140
141        final LayoutInflater inflater = LayoutInflater.from(ap.mContext);
142        final View checkbox = inflater.inflate(
143                com.android.internal.R.layout.always_use_checkbox, null);
144        ap.mView = checkbox;
145        mAlwaysAllow = (CheckBox) checkbox.findViewById(com.android.internal.R.id.alwaysUse);
146
147        if (ACTION_PROMPT_UNVALIDATED.equals(mAction)) {
148            mAlwaysAllow.setText(getString(R.string.no_internet_access_remember));
149        } else {
150            mAlwaysAllow.setText(getString(R.string.lost_internet_access_persist));
151        }
152
153        setupAlert();
154    }
155
156    @Override
157    protected void onDestroy() {
158        if (mNetworkCallback != null) {
159            mCM.unregisterNetworkCallback(mNetworkCallback);
160            mNetworkCallback = null;
161        }
162        super.onDestroy();
163    }
164
165    public void onClick(DialogInterface dialog, int which) {
166        if (which != BUTTON_NEGATIVE && which != BUTTON_POSITIVE) return;
167        final boolean always = mAlwaysAllow.isChecked();
168        final String what, action;
169
170        if (ACTION_PROMPT_UNVALIDATED.equals(mAction)) {
171            what = "NO_INTERNET";
172            final boolean accept = (which == BUTTON_POSITIVE);
173            action = (accept ? "Connect" : "Ignore");
174            mCM.setAcceptUnvalidated(mNetwork, accept, always);
175        } else {
176            what = "LOST_INTERNET";
177            final boolean avoid = (which == BUTTON_POSITIVE);
178            action = (avoid ? "Switch away" : "Get stuck");
179            if (always) {
180                Settings.Global.putString(mAlertParams.mContext.getContentResolver(),
181                        Settings.Global.NETWORK_AVOID_BAD_WIFI, avoid ? "1" : "0");
182            } else if (avoid) {
183                mCM.setAvoidUnvalidated(mNetwork);
184            }
185        }
186        Log.d(TAG, what + ": " + action +  " network=" + mNetwork +
187                (always ? " and remember" : ""));
188    }
189}
190