Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1 | /* |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 2 | * Copyright (C) 2021 The Android Open Source Project |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 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.server; |
| 18 | |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 19 | import static android.net.ConnectivityManager.NETID_UNSET; |
| 20 | import static android.net.nsd.NsdManager.MDNS_SERVICE_EVENT; |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 21 | import static android.provider.DeviceConfig.NAMESPACE_CONNECTIVITY; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 22 | |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 23 | import android.annotation.NonNull; |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 24 | import android.annotation.Nullable; |
paulhu | a262cc1 | 2019-08-12 16:25:11 +0800 | [diff] [blame] | 25 | import android.content.Context; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 26 | import android.content.Intent; |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 27 | import android.net.ConnectivityManager; |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 28 | import android.net.INetd; |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 29 | import android.net.LinkProperties; |
| 30 | import android.net.Network; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 31 | import android.net.mdns.aidl.DiscoveryInfo; |
| 32 | import android.net.mdns.aidl.GetAddressInfo; |
| 33 | import android.net.mdns.aidl.IMDnsEventListener; |
| 34 | import android.net.mdns.aidl.RegistrationInfo; |
| 35 | import android.net.mdns.aidl.ResolutionInfo; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 36 | import android.net.nsd.INsdManager; |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 37 | import android.net.nsd.INsdManagerCallback; |
| 38 | import android.net.nsd.INsdServiceConnector; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 39 | import android.net.nsd.MDnsManager; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 40 | import android.net.nsd.NsdManager; |
paulhu | a262cc1 | 2019-08-12 16:25:11 +0800 | [diff] [blame] | 41 | import android.net.nsd.NsdServiceInfo; |
Hugo Benichi | 803a2f0 | 2017-04-24 11:35:06 +0900 | [diff] [blame] | 42 | import android.os.Handler; |
paulhu | a262cc1 | 2019-08-12 16:25:11 +0800 | [diff] [blame] | 43 | import android.os.HandlerThread; |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 44 | import android.os.IBinder; |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 45 | import android.os.Looper; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 46 | import android.os.Message; |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 47 | import android.os.RemoteException; |
Dianne Hackborn | 692107e | 2012-08-29 18:32:08 -0700 | [diff] [blame] | 48 | import android.os.UserHandle; |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 49 | import android.text.TextUtils; |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 50 | import android.util.Log; |
| 51 | import android.util.Pair; |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 52 | import android.util.SparseArray; |
Hugo Benichi | d2552ae | 2017-04-11 14:42:47 +0900 | [diff] [blame] | 53 | import android.util.SparseIntArray; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 54 | |
paulhu | a262cc1 | 2019-08-12 16:25:11 +0800 | [diff] [blame] | 55 | import com.android.internal.annotations.VisibleForTesting; |
paulhu | a262cc1 | 2019-08-12 16:25:11 +0800 | [diff] [blame] | 56 | import com.android.internal.util.State; |
| 57 | import com.android.internal.util.StateMachine; |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 58 | import com.android.net.module.util.DeviceConfigUtils; |
paulhu | 3ffffe7 | 2021-09-16 10:15:22 +0800 | [diff] [blame] | 59 | import com.android.net.module.util.PermissionUtils; |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 60 | import com.android.server.connectivity.mdns.ExecutorProvider; |
| 61 | import com.android.server.connectivity.mdns.MdnsDiscoveryManager; |
| 62 | import com.android.server.connectivity.mdns.MdnsMultinetworkSocketClient; |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 63 | import com.android.server.connectivity.mdns.MdnsSearchOptions; |
| 64 | import com.android.server.connectivity.mdns.MdnsServiceBrowserListener; |
| 65 | import com.android.server.connectivity.mdns.MdnsServiceInfo; |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 66 | import com.android.server.connectivity.mdns.MdnsSocketClientBase; |
| 67 | import com.android.server.connectivity.mdns.MdnsSocketProvider; |
paulhu | a262cc1 | 2019-08-12 16:25:11 +0800 | [diff] [blame] | 68 | |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 69 | import java.io.FileDescriptor; |
| 70 | import java.io.PrintWriter; |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 71 | import java.net.InetAddress; |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 72 | import java.net.NetworkInterface; |
| 73 | import java.net.SocketException; |
| 74 | import java.net.UnknownHostException; |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 75 | import java.util.HashMap; |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 76 | import java.util.List; |
| 77 | import java.util.regex.Matcher; |
| 78 | import java.util.regex.Pattern; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 79 | |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 80 | /** |
| 81 | * Network Service Discovery Service handles remote service discovery operation requests by |
| 82 | * implementing the INsdManager interface. |
| 83 | * |
| 84 | * @hide |
| 85 | */ |
| 86 | public class NsdService extends INsdManager.Stub { |
| 87 | private static final String TAG = "NsdService"; |
| 88 | private static final String MDNS_TAG = "mDnsConnector"; |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 89 | private static final String MDNS_DISCOVERY_MANAGER_VERSION = "mdns_discovery_manager_version"; |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 90 | private static final String LOCAL_DOMAIN_NAME = "local"; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 91 | |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 92 | private static final boolean DBG = Log.isLoggable(TAG, Log.DEBUG); |
Luke Huang | 92860f9 | 2021-06-23 06:29:30 +0000 | [diff] [blame] | 93 | private static final long CLEANUP_DELAY_MS = 10000; |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 94 | private static final int IFACE_IDX_ANY = 0; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 95 | |
Hugo Benichi | 32be63d | 2017-04-05 14:06:11 +0900 | [diff] [blame] | 96 | private final Context mContext; |
Hugo Benichi | 32be63d | 2017-04-05 14:06:11 +0900 | [diff] [blame] | 97 | private final NsdStateMachine mNsdStateMachine; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 98 | private final MDnsManager mMDnsManager; |
| 99 | private final MDnsEventCallback mMDnsEventCallback; |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 100 | @Nullable |
| 101 | private final MdnsMultinetworkSocketClient mMdnsSocketClient; |
| 102 | @Nullable |
| 103 | private final MdnsDiscoveryManager mMdnsDiscoveryManager; |
| 104 | @Nullable |
| 105 | private final MdnsSocketProvider mMdnsSocketProvider; |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 106 | // WARNING : Accessing these values in any thread is not safe, it must only be changed in the |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 107 | // state machine thread. If change this outside state machine, it will need to introduce |
| 108 | // synchronization. |
| 109 | private boolean mIsDaemonStarted = false; |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 110 | private boolean mIsMonitoringSocketsStarted = false; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 111 | |
| 112 | /** |
| 113 | * Clients receiving asynchronous messages |
| 114 | */ |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 115 | private final HashMap<NsdServiceConnector, ClientInfo> mClients = new HashMap<>(); |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 116 | |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 117 | /* A map from unique id to client info */ |
Hugo Benichi | 32be63d | 2017-04-05 14:06:11 +0900 | [diff] [blame] | 118 | private final SparseArray<ClientInfo> mIdToClientInfoMap= new SparseArray<>(); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 119 | |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 120 | private final long mCleanupDelayMs; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 121 | |
Hugo Benichi | 32be63d | 2017-04-05 14:06:11 +0900 | [diff] [blame] | 122 | private static final int INVALID_ID = 0; |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 123 | private int mUniqueId = 1; |
Luke Huang | f7277ed | 2021-07-12 21:15:10 +0800 | [diff] [blame] | 124 | // The count of the connected legacy clients. |
| 125 | private int mLegacyClientCount = 0; |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 126 | |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 127 | private static class MdnsListener implements MdnsServiceBrowserListener { |
| 128 | protected final int mClientId; |
| 129 | protected final int mTransactionId; |
| 130 | @NonNull |
| 131 | protected final NsdServiceInfo mReqServiceInfo; |
| 132 | @NonNull |
| 133 | protected final String mListenedServiceType; |
| 134 | |
| 135 | MdnsListener(int clientId, int transactionId, @NonNull NsdServiceInfo reqServiceInfo, |
| 136 | @NonNull String listenedServiceType) { |
| 137 | mClientId = clientId; |
| 138 | mTransactionId = transactionId; |
| 139 | mReqServiceInfo = reqServiceInfo; |
| 140 | mListenedServiceType = listenedServiceType; |
| 141 | } |
| 142 | |
| 143 | @NonNull |
| 144 | public String getListenedServiceType() { |
| 145 | return mListenedServiceType; |
| 146 | } |
| 147 | |
| 148 | @Override |
| 149 | public void onServiceFound(@NonNull MdnsServiceInfo serviceInfo) { } |
| 150 | |
| 151 | @Override |
| 152 | public void onServiceUpdated(@NonNull MdnsServiceInfo serviceInfo) { } |
| 153 | |
| 154 | @Override |
| 155 | public void onServiceRemoved(@NonNull MdnsServiceInfo serviceInfo) { } |
| 156 | |
| 157 | @Override |
| 158 | public void onServiceNameDiscovered(@NonNull MdnsServiceInfo serviceInfo) { } |
| 159 | |
| 160 | @Override |
| 161 | public void onServiceNameRemoved(@NonNull MdnsServiceInfo serviceInfo) { } |
| 162 | |
| 163 | @Override |
| 164 | public void onSearchStoppedWithError(int error) { } |
| 165 | |
| 166 | @Override |
| 167 | public void onSearchFailedToStart() { } |
| 168 | |
| 169 | @Override |
| 170 | public void onDiscoveryQuerySent(@NonNull List<String> subtypes, int transactionId) { } |
| 171 | |
| 172 | @Override |
| 173 | public void onFailedToParseMdnsResponse(int receivedPacketNumber, int errorCode) { } |
| 174 | } |
| 175 | |
| 176 | private class DiscoveryListener extends MdnsListener { |
| 177 | |
| 178 | DiscoveryListener(int clientId, int transactionId, @NonNull NsdServiceInfo reqServiceInfo, |
| 179 | @NonNull String listenServiceType) { |
| 180 | super(clientId, transactionId, reqServiceInfo, listenServiceType); |
| 181 | } |
| 182 | |
| 183 | @Override |
| 184 | public void onServiceNameDiscovered(@NonNull MdnsServiceInfo serviceInfo) { |
| 185 | // TODO: implement service name discovered callback. |
| 186 | } |
| 187 | |
| 188 | @Override |
| 189 | public void onServiceNameRemoved(@NonNull MdnsServiceInfo serviceInfo) { |
| 190 | // TODO: implement service name removed callback. |
| 191 | } |
| 192 | } |
| 193 | |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 194 | private class NsdStateMachine extends StateMachine { |
| 195 | |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 196 | private final DefaultState mDefaultState = new DefaultState(); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 197 | private final EnabledState mEnabledState = new EnabledState(); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 198 | |
| 199 | @Override |
Wink Saville | 358f5d4 | 2012-05-29 12:40:46 -0700 | [diff] [blame] | 200 | protected String getWhatToString(int what) { |
Hugo Benichi | 32be63d | 2017-04-05 14:06:11 +0900 | [diff] [blame] | 201 | return NsdManager.nameOf(what); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 202 | } |
| 203 | |
Luke Huang | 92860f9 | 2021-06-23 06:29:30 +0000 | [diff] [blame] | 204 | private void maybeStartDaemon() { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 205 | if (mIsDaemonStarted) { |
| 206 | if (DBG) Log.d(TAG, "Daemon is already started."); |
| 207 | return; |
| 208 | } |
| 209 | mMDnsManager.registerEventListener(mMDnsEventCallback); |
| 210 | mMDnsManager.startDaemon(); |
| 211 | mIsDaemonStarted = true; |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 212 | maybeScheduleStop(); |
| 213 | } |
| 214 | |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 215 | private void maybeStopDaemon() { |
| 216 | if (!mIsDaemonStarted) { |
| 217 | if (DBG) Log.d(TAG, "Daemon has not been started."); |
| 218 | return; |
| 219 | } |
| 220 | mMDnsManager.unregisterEventListener(mMDnsEventCallback); |
| 221 | mMDnsManager.stopDaemon(); |
| 222 | mIsDaemonStarted = false; |
| 223 | } |
| 224 | |
Luke Huang | 92860f9 | 2021-06-23 06:29:30 +0000 | [diff] [blame] | 225 | private boolean isAnyRequestActive() { |
| 226 | return mIdToClientInfoMap.size() != 0; |
| 227 | } |
| 228 | |
| 229 | private void scheduleStop() { |
| 230 | sendMessageDelayed(NsdManager.DAEMON_CLEANUP, mCleanupDelayMs); |
| 231 | } |
| 232 | private void maybeScheduleStop() { |
Luke Huang | f7277ed | 2021-07-12 21:15:10 +0800 | [diff] [blame] | 233 | // The native daemon should stay alive and can't be cleanup |
| 234 | // if any legacy client connected. |
| 235 | if (!isAnyRequestActive() && mLegacyClientCount == 0) { |
Luke Huang | 92860f9 | 2021-06-23 06:29:30 +0000 | [diff] [blame] | 236 | scheduleStop(); |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 237 | } |
| 238 | } |
| 239 | |
Luke Huang | 92860f9 | 2021-06-23 06:29:30 +0000 | [diff] [blame] | 240 | private void cancelStop() { |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 241 | this.removeMessages(NsdManager.DAEMON_CLEANUP); |
| 242 | } |
| 243 | |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 244 | private void maybeStartMonitoringSockets() { |
| 245 | if (mIsMonitoringSocketsStarted) { |
| 246 | if (DBG) Log.d(TAG, "Socket monitoring is already started."); |
| 247 | return; |
| 248 | } |
| 249 | |
| 250 | mMdnsSocketProvider.startMonitoringSockets(); |
| 251 | mIsMonitoringSocketsStarted = true; |
| 252 | } |
| 253 | |
| 254 | private void maybeStopMonitoringSockets() { |
| 255 | if (!mIsMonitoringSocketsStarted) { |
| 256 | if (DBG) Log.d(TAG, "Socket monitoring has not been started."); |
| 257 | return; |
| 258 | } |
| 259 | mMdnsSocketProvider.stopMonitoringSockets(); |
| 260 | mIsMonitoringSocketsStarted = false; |
| 261 | } |
| 262 | |
| 263 | private void maybeStopMonitoringSocketsIfNoActiveRequest() { |
| 264 | if (!isAnyRequestActive()) { |
| 265 | maybeStopMonitoringSockets(); |
| 266 | } |
| 267 | } |
| 268 | |
Hugo Benichi | 803a2f0 | 2017-04-24 11:35:06 +0900 | [diff] [blame] | 269 | NsdStateMachine(String name, Handler handler) { |
| 270 | super(name, handler); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 271 | addState(mDefaultState); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 272 | addState(mEnabledState, mDefaultState); |
paulhu | 5568f45 | 2021-11-30 13:31:29 +0800 | [diff] [blame] | 273 | State initialState = mEnabledState; |
Hugo Benichi | 912db99 | 2017-04-24 16:41:03 +0900 | [diff] [blame] | 274 | setInitialState(initialState); |
Wink Saville | 358f5d4 | 2012-05-29 12:40:46 -0700 | [diff] [blame] | 275 | setLogRecSize(25); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 276 | } |
| 277 | |
| 278 | class DefaultState extends State { |
| 279 | @Override |
| 280 | public boolean processMessage(Message msg) { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 281 | final ClientInfo cInfo; |
| 282 | final int clientId = msg.arg2; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 283 | switch (msg.what) { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 284 | case NsdManager.REGISTER_CLIENT: |
| 285 | final Pair<NsdServiceConnector, INsdManagerCallback> arg = |
| 286 | (Pair<NsdServiceConnector, INsdManagerCallback>) msg.obj; |
| 287 | final INsdManagerCallback cb = arg.second; |
| 288 | try { |
| 289 | cb.asBinder().linkToDeath(arg.first, 0); |
| 290 | cInfo = new ClientInfo(cb); |
| 291 | mClients.put(arg.first, cInfo); |
| 292 | } catch (RemoteException e) { |
| 293 | Log.w(TAG, "Client " + clientId + " has already died"); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 294 | } |
| 295 | break; |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 296 | case NsdManager.UNREGISTER_CLIENT: |
| 297 | final NsdServiceConnector connector = (NsdServiceConnector) msg.obj; |
| 298 | cInfo = mClients.remove(connector); |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 299 | if (cInfo != null) { |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 300 | if (mMdnsDiscoveryManager != null) { |
| 301 | cInfo.unregisterAllListeners(); |
| 302 | } |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 303 | cInfo.expungeAllRequests(); |
Luke Huang | f7277ed | 2021-07-12 21:15:10 +0800 | [diff] [blame] | 304 | if (cInfo.isLegacy()) { |
| 305 | mLegacyClientCount -= 1; |
| 306 | } |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 307 | } |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 308 | if (mMdnsDiscoveryManager != null) { |
| 309 | maybeStopMonitoringSocketsIfNoActiveRequest(); |
| 310 | } |
Luke Huang | f7277ed | 2021-07-12 21:15:10 +0800 | [diff] [blame] | 311 | maybeScheduleStop(); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 312 | break; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 313 | case NsdManager.DISCOVER_SERVICES: |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 314 | cInfo = getClientInfoForReply(msg); |
| 315 | if (cInfo != null) { |
| 316 | cInfo.onDiscoverServicesFailed( |
| 317 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
| 318 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 319 | break; |
| 320 | case NsdManager.STOP_DISCOVERY: |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 321 | cInfo = getClientInfoForReply(msg); |
| 322 | if (cInfo != null) { |
| 323 | cInfo.onStopDiscoveryFailed( |
| 324 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
| 325 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 326 | break; |
| 327 | case NsdManager.REGISTER_SERVICE: |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 328 | cInfo = getClientInfoForReply(msg); |
| 329 | if (cInfo != null) { |
| 330 | cInfo.onRegisterServiceFailed( |
| 331 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
| 332 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 333 | break; |
| 334 | case NsdManager.UNREGISTER_SERVICE: |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 335 | cInfo = getClientInfoForReply(msg); |
| 336 | if (cInfo != null) { |
| 337 | cInfo.onUnregisterServiceFailed( |
| 338 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
| 339 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 340 | break; |
| 341 | case NsdManager.RESOLVE_SERVICE: |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 342 | cInfo = getClientInfoForReply(msg); |
| 343 | if (cInfo != null) { |
| 344 | cInfo.onResolveServiceFailed( |
| 345 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
| 346 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 347 | break; |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 348 | case NsdManager.DAEMON_CLEANUP: |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 349 | maybeStopDaemon(); |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 350 | break; |
Luke Huang | f7277ed | 2021-07-12 21:15:10 +0800 | [diff] [blame] | 351 | // This event should be only sent by the legacy (target SDK < S) clients. |
| 352 | // Mark the sending client as legacy. |
| 353 | case NsdManager.DAEMON_STARTUP: |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 354 | cInfo = getClientInfoForReply(msg); |
Luke Huang | f7277ed | 2021-07-12 21:15:10 +0800 | [diff] [blame] | 355 | if (cInfo != null) { |
| 356 | cancelStop(); |
| 357 | cInfo.setLegacy(); |
| 358 | mLegacyClientCount += 1; |
| 359 | maybeStartDaemon(); |
| 360 | } |
| 361 | break; |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 362 | case NsdManager.MDNS_MONITORING_SOCKETS_CLEANUP: |
| 363 | maybeStopMonitoringSockets(); |
| 364 | break; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 365 | default: |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 366 | Log.e(TAG, "Unhandled " + msg); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 367 | return NOT_HANDLED; |
| 368 | } |
| 369 | return HANDLED; |
| 370 | } |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 371 | |
| 372 | private ClientInfo getClientInfoForReply(Message msg) { |
| 373 | final ListenerArgs args = (ListenerArgs) msg.obj; |
| 374 | return mClients.get(args.connector); |
| 375 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 376 | } |
| 377 | |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 378 | class EnabledState extends State { |
| 379 | @Override |
| 380 | public void enter() { |
| 381 | sendNsdStateChangeBroadcast(true); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 382 | } |
| 383 | |
| 384 | @Override |
| 385 | public void exit() { |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 386 | // TODO: it is incorrect to stop the daemon without expunging all requests |
| 387 | // and sending error callbacks to clients. |
Luke Huang | 92860f9 | 2021-06-23 06:29:30 +0000 | [diff] [blame] | 388 | scheduleStop(); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 389 | } |
| 390 | |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 391 | private boolean requestLimitReached(ClientInfo clientInfo) { |
| 392 | if (clientInfo.mClientIds.size() >= ClientInfo.MAX_LIMIT) { |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 393 | if (DBG) Log.d(TAG, "Exceeded max outstanding requests " + clientInfo); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 394 | return true; |
| 395 | } |
| 396 | return false; |
| 397 | } |
| 398 | |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 399 | private void storeRequestMap(int clientId, int globalId, ClientInfo clientInfo, int what) { |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 400 | clientInfo.mClientIds.put(clientId, globalId); |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 401 | clientInfo.mClientRequests.put(clientId, what); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 402 | mIdToClientInfoMap.put(globalId, clientInfo); |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 403 | // Remove the cleanup event because here comes a new request. |
| 404 | cancelStop(); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 405 | } |
| 406 | |
| 407 | private void removeRequestMap(int clientId, int globalId, ClientInfo clientInfo) { |
Hugo Benichi | d2552ae | 2017-04-11 14:42:47 +0900 | [diff] [blame] | 408 | clientInfo.mClientIds.delete(clientId); |
| 409 | clientInfo.mClientRequests.delete(clientId); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 410 | mIdToClientInfoMap.remove(globalId); |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 411 | maybeScheduleStop(); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 412 | } |
| 413 | |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 414 | private void storeListenerMap(int clientId, int transactionId, MdnsListener listener, |
| 415 | ClientInfo clientInfo) { |
| 416 | clientInfo.mClientIds.put(clientId, transactionId); |
| 417 | clientInfo.mListeners.put(clientId, listener); |
| 418 | mIdToClientInfoMap.put(transactionId, clientInfo); |
| 419 | removeMessages(NsdManager.MDNS_MONITORING_SOCKETS_CLEANUP); |
| 420 | } |
| 421 | |
| 422 | private void removeListenerMap(int clientId, int transactionId, ClientInfo clientInfo) { |
| 423 | clientInfo.mClientIds.delete(clientId); |
| 424 | clientInfo.mListeners.delete(clientId); |
| 425 | mIdToClientInfoMap.remove(transactionId); |
| 426 | maybeStopMonitoringSocketsIfNoActiveRequest(); |
| 427 | } |
| 428 | |
| 429 | /** |
| 430 | * Check the given service type is valid and construct it to a service type |
| 431 | * which can use for discovery / resolution service. |
| 432 | * |
| 433 | * <p> The valid service type should be 2 labels, or 3 labels if the query is for a |
| 434 | * subtype (see RFC6763 7.1). Each label is up to 63 characters and must start with an |
| 435 | * underscore; they are alphanumerical characters or dashes or underscore, except the |
| 436 | * last one that is just alphanumerical. The last label must be _tcp or _udp. |
| 437 | * |
| 438 | * @param serviceType the request service type for discovery / resolution service |
| 439 | * @return constructed service type or null if the given service type is invalid. |
| 440 | */ |
| 441 | @Nullable |
| 442 | private String constructServiceType(String serviceType) { |
| 443 | if (TextUtils.isEmpty(serviceType)) return null; |
| 444 | |
| 445 | final Pattern serviceTypePattern = Pattern.compile( |
| 446 | "^(_[a-zA-Z0-9-_]{1,61}[a-zA-Z0-9]\\.)?" |
| 447 | + "(_[a-zA-Z0-9-_]{1,61}[a-zA-Z0-9]\\._(?:tcp|udp))$"); |
| 448 | final Matcher matcher = serviceTypePattern.matcher(serviceType); |
| 449 | if (!matcher.matches()) return null; |
| 450 | return matcher.group(1) == null |
| 451 | ? serviceType + ".local" |
| 452 | : matcher.group(1) + "._sub" + matcher.group(2) + ".local"; |
| 453 | } |
| 454 | |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 455 | @Override |
| 456 | public boolean processMessage(Message msg) { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 457 | final ClientInfo clientInfo; |
| 458 | final int id; |
| 459 | final int clientId = msg.arg2; |
| 460 | final ListenerArgs args; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 461 | switch (msg.what) { |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 462 | case NsdManager.DISCOVER_SERVICES: |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 463 | if (DBG) Log.d(TAG, "Discover services"); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 464 | args = (ListenerArgs) msg.obj; |
| 465 | clientInfo = mClients.get(args.connector); |
Paul Hu | 116b4c0 | 2022-08-16 07:21:55 +0000 | [diff] [blame] | 466 | // If the binder death notification for a INsdManagerCallback was received |
| 467 | // before any calls are received by NsdService, the clientInfo would be |
| 468 | // cleared and cause NPE. Add a null check here to prevent this corner case. |
| 469 | if (clientInfo == null) { |
| 470 | Log.e(TAG, "Unknown connector in discovery"); |
| 471 | break; |
| 472 | } |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 473 | |
| 474 | if (requestLimitReached(clientInfo)) { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 475 | clientInfo.onDiscoverServicesFailed( |
| 476 | clientId, NsdManager.FAILURE_MAX_LIMIT); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 477 | break; |
| 478 | } |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 479 | |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 480 | final NsdServiceInfo info = args.serviceInfo; |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 481 | id = getUniqueId(); |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 482 | if (mMdnsDiscoveryManager != null) { |
| 483 | final String serviceType = constructServiceType(info.getServiceType()); |
| 484 | if (serviceType == null) { |
| 485 | clientInfo.onDiscoverServicesFailed(clientId, |
| 486 | NsdManager.FAILURE_INTERNAL_ERROR); |
| 487 | break; |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 488 | } |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 489 | |
| 490 | maybeStartMonitoringSockets(); |
| 491 | final MdnsListener listener = |
| 492 | new DiscoveryListener(clientId, id, info, serviceType); |
| 493 | final MdnsSearchOptions options = MdnsSearchOptions.newBuilder() |
| 494 | .setNetwork(info.getNetwork()) |
| 495 | .setIsPassiveMode(true) |
| 496 | .build(); |
| 497 | mMdnsDiscoveryManager.registerListener(serviceType, listener, options); |
| 498 | storeListenerMap(clientId, id, listener, clientInfo); |
| 499 | clientInfo.onDiscoverServicesStarted(clientId, info); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 500 | } else { |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 501 | maybeStartDaemon(); |
| 502 | if (discoverServices(id, info)) { |
| 503 | if (DBG) { |
| 504 | Log.d(TAG, "Discover " + msg.arg2 + " " + id |
| 505 | + info.getServiceType()); |
| 506 | } |
| 507 | storeRequestMap(clientId, id, clientInfo, msg.what); |
| 508 | clientInfo.onDiscoverServicesStarted(clientId, info); |
| 509 | } else { |
| 510 | stopServiceDiscovery(id); |
| 511 | clientInfo.onDiscoverServicesFailed(clientId, |
| 512 | NsdManager.FAILURE_INTERNAL_ERROR); |
| 513 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 514 | } |
| 515 | break; |
| 516 | case NsdManager.STOP_DISCOVERY: |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 517 | if (DBG) Log.d(TAG, "Stop service discovery"); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 518 | args = (ListenerArgs) msg.obj; |
| 519 | clientInfo = mClients.get(args.connector); |
Paul Hu | 116b4c0 | 2022-08-16 07:21:55 +0000 | [diff] [blame] | 520 | // If the binder death notification for a INsdManagerCallback was received |
| 521 | // before any calls are received by NsdService, the clientInfo would be |
| 522 | // cleared and cause NPE. Add a null check here to prevent this corner case. |
| 523 | if (clientInfo == null) { |
| 524 | Log.e(TAG, "Unknown connector in stop discovery"); |
| 525 | break; |
| 526 | } |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 527 | |
| 528 | try { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 529 | id = clientInfo.mClientIds.get(clientId); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 530 | } catch (NullPointerException e) { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 531 | clientInfo.onStopDiscoveryFailed( |
| 532 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 533 | break; |
| 534 | } |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 535 | if (mMdnsDiscoveryManager != null) { |
| 536 | final MdnsListener listener = clientInfo.mListeners.get(clientId); |
| 537 | if (listener == null) { |
| 538 | clientInfo.onStopDiscoveryFailed( |
| 539 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
| 540 | break; |
| 541 | } |
| 542 | mMdnsDiscoveryManager.unregisterListener( |
| 543 | listener.getListenedServiceType(), listener); |
| 544 | removeListenerMap(clientId, id, clientInfo); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 545 | clientInfo.onStopDiscoverySucceeded(clientId); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 546 | } else { |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 547 | removeRequestMap(clientId, id, clientInfo); |
| 548 | if (stopServiceDiscovery(id)) { |
| 549 | clientInfo.onStopDiscoverySucceeded(clientId); |
| 550 | } else { |
| 551 | clientInfo.onStopDiscoveryFailed( |
| 552 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
| 553 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 554 | } |
| 555 | break; |
| 556 | case NsdManager.REGISTER_SERVICE: |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 557 | if (DBG) Log.d(TAG, "Register service"); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 558 | args = (ListenerArgs) msg.obj; |
| 559 | clientInfo = mClients.get(args.connector); |
Paul Hu | 116b4c0 | 2022-08-16 07:21:55 +0000 | [diff] [blame] | 560 | // If the binder death notification for a INsdManagerCallback was received |
| 561 | // before any calls are received by NsdService, the clientInfo would be |
| 562 | // cleared and cause NPE. Add a null check here to prevent this corner case. |
| 563 | if (clientInfo == null) { |
| 564 | Log.e(TAG, "Unknown connector in registration"); |
| 565 | break; |
| 566 | } |
| 567 | |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 568 | if (requestLimitReached(clientInfo)) { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 569 | clientInfo.onRegisterServiceFailed( |
| 570 | clientId, NsdManager.FAILURE_MAX_LIMIT); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 571 | break; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 572 | } |
| 573 | |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 574 | maybeStartDaemon(); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 575 | id = getUniqueId(); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 576 | if (registerService(id, args.serviceInfo)) { |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 577 | if (DBG) Log.d(TAG, "Register " + clientId + " " + id); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 578 | storeRequestMap(clientId, id, clientInfo, msg.what); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 579 | // Return success after mDns reports success |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 580 | } else { |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 581 | unregisterService(id); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 582 | clientInfo.onRegisterServiceFailed( |
| 583 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 584 | } |
| 585 | break; |
| 586 | case NsdManager.UNREGISTER_SERVICE: |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 587 | if (DBG) Log.d(TAG, "unregister service"); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 588 | args = (ListenerArgs) msg.obj; |
| 589 | clientInfo = mClients.get(args.connector); |
Paul Hu | 116b4c0 | 2022-08-16 07:21:55 +0000 | [diff] [blame] | 590 | // If the binder death notification for a INsdManagerCallback was received |
| 591 | // before any calls are received by NsdService, the clientInfo would be |
| 592 | // cleared and cause NPE. Add a null check here to prevent this corner case. |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 593 | if (clientInfo == null) { |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 594 | Log.e(TAG, "Unknown connector in unregistration"); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 595 | break; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 596 | } |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 597 | id = clientInfo.mClientIds.get(clientId); |
| 598 | removeRequestMap(clientId, id, clientInfo); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 599 | if (unregisterService(id)) { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 600 | clientInfo.onUnregisterServiceSucceeded(clientId); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 601 | } else { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 602 | clientInfo.onUnregisterServiceFailed( |
| 603 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 604 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 605 | break; |
| 606 | case NsdManager.RESOLVE_SERVICE: |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 607 | if (DBG) Log.d(TAG, "Resolve service"); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 608 | args = (ListenerArgs) msg.obj; |
| 609 | clientInfo = mClients.get(args.connector); |
Paul Hu | 116b4c0 | 2022-08-16 07:21:55 +0000 | [diff] [blame] | 610 | // If the binder death notification for a INsdManagerCallback was received |
| 611 | // before any calls are received by NsdService, the clientInfo would be |
| 612 | // cleared and cause NPE. Add a null check here to prevent this corner case. |
| 613 | if (clientInfo == null) { |
| 614 | Log.e(TAG, "Unknown connector in resolution"); |
| 615 | break; |
| 616 | } |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 617 | |
| 618 | if (clientInfo.mResolvedService != null) { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 619 | clientInfo.onResolveServiceFailed( |
| 620 | clientId, NsdManager.FAILURE_ALREADY_ACTIVE); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 621 | break; |
| 622 | } |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 623 | |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 624 | maybeStartDaemon(); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 625 | id = getUniqueId(); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 626 | if (resolveService(id, args.serviceInfo)) { |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 627 | clientInfo.mResolvedService = new NsdServiceInfo(); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 628 | storeRequestMap(clientId, id, clientInfo, msg.what); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 629 | } else { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 630 | clientInfo.onResolveServiceFailed( |
| 631 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 632 | } |
| 633 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 634 | case MDNS_SERVICE_EVENT: |
| 635 | if (!handleMDnsServiceEvent(msg.arg1, msg.arg2, msg.obj)) { |
Hugo Benichi | f0c8409 | 2017-04-05 14:43:29 +0900 | [diff] [blame] | 636 | return NOT_HANDLED; |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 637 | } |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 638 | break; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 639 | default: |
Hugo Benichi | f0c8409 | 2017-04-05 14:43:29 +0900 | [diff] [blame] | 640 | return NOT_HANDLED; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 641 | } |
Hugo Benichi | f0c8409 | 2017-04-05 14:43:29 +0900 | [diff] [blame] | 642 | return HANDLED; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 643 | } |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 644 | |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 645 | private boolean handleMDnsServiceEvent(int code, int id, Object obj) { |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 646 | NsdServiceInfo servInfo; |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 647 | ClientInfo clientInfo = mIdToClientInfoMap.get(id); |
| 648 | if (clientInfo == null) { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 649 | Log.e(TAG, String.format("id %d for %d has no client mapping", id, code)); |
Hugo Benichi | f0c8409 | 2017-04-05 14:43:29 +0900 | [diff] [blame] | 650 | return false; |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 651 | } |
| 652 | |
| 653 | /* This goes in response as msg.arg2 */ |
Christopher Lane | 7441122 | 2014-04-25 18:39:07 -0700 | [diff] [blame] | 654 | int clientId = clientInfo.getClientId(id); |
| 655 | if (clientId < 0) { |
Vinit Deshapnde | 930a851 | 2013-06-25 19:45:03 -0700 | [diff] [blame] | 656 | // This can happen because of race conditions. For example, |
| 657 | // SERVICE_FOUND may race with STOP_SERVICE_DISCOVERY, |
| 658 | // and we may get in this situation. |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 659 | Log.d(TAG, String.format("%d for listener id %d that is no longer active", |
| 660 | code, id)); |
Hugo Benichi | f0c8409 | 2017-04-05 14:43:29 +0900 | [diff] [blame] | 661 | return false; |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 662 | } |
Hugo Benichi | 32be63d | 2017-04-05 14:06:11 +0900 | [diff] [blame] | 663 | if (DBG) { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 664 | Log.d(TAG, String.format("MDns service event code:%d id=%d", code, id)); |
Hugo Benichi | 32be63d | 2017-04-05 14:06:11 +0900 | [diff] [blame] | 665 | } |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 666 | switch (code) { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 667 | case IMDnsEventListener.SERVICE_FOUND: { |
| 668 | final DiscoveryInfo info = (DiscoveryInfo) obj; |
| 669 | final String name = info.serviceName; |
| 670 | final String type = info.registrationType; |
| 671 | servInfo = new NsdServiceInfo(name, type); |
| 672 | final int foundNetId = info.netId; |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 673 | if (foundNetId == 0L) { |
| 674 | // Ignore services that do not have a Network: they are not usable |
| 675 | // by apps, as they would need privileged permissions to use |
| 676 | // interfaces that do not have an associated Network. |
| 677 | break; |
| 678 | } |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 679 | setServiceNetworkForCallback(servInfo, info.netId, info.interfaceIdx); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 680 | clientInfo.onServiceFound(clientId, servInfo); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 681 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 682 | } |
| 683 | case IMDnsEventListener.SERVICE_LOST: { |
| 684 | final DiscoveryInfo info = (DiscoveryInfo) obj; |
| 685 | final String name = info.serviceName; |
| 686 | final String type = info.registrationType; |
| 687 | final int lostNetId = info.netId; |
| 688 | servInfo = new NsdServiceInfo(name, type); |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 689 | // The network could be set to null (netId 0) if it was torn down when the |
| 690 | // service is lost |
| 691 | // TODO: avoid returning null in that case, possibly by remembering |
| 692 | // found services on the same interface index and their network at the time |
| 693 | setServiceNetworkForCallback(servInfo, lostNetId, info.interfaceIdx); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 694 | clientInfo.onServiceLost(clientId, servInfo); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 695 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 696 | } |
| 697 | case IMDnsEventListener.SERVICE_DISCOVERY_FAILED: |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 698 | clientInfo.onDiscoverServicesFailed( |
| 699 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 700 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 701 | case IMDnsEventListener.SERVICE_REGISTERED: { |
| 702 | final RegistrationInfo info = (RegistrationInfo) obj; |
| 703 | final String name = info.serviceName; |
| 704 | servInfo = new NsdServiceInfo(name, null /* serviceType */); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 705 | clientInfo.onRegisterServiceSucceeded(clientId, servInfo); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 706 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 707 | } |
| 708 | case IMDnsEventListener.SERVICE_REGISTRATION_FAILED: |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 709 | clientInfo.onRegisterServiceFailed( |
| 710 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 711 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 712 | case IMDnsEventListener.SERVICE_RESOLVED: { |
| 713 | final ResolutionInfo info = (ResolutionInfo) obj; |
Sreeram Ramachandran | a53dd7f | 2014-09-03 15:45:59 -0700 | [diff] [blame] | 714 | int index = 0; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 715 | final String fullName = info.serviceFullName; |
| 716 | while (index < fullName.length() && fullName.charAt(index) != '.') { |
| 717 | if (fullName.charAt(index) == '\\') { |
Sreeram Ramachandran | a53dd7f | 2014-09-03 15:45:59 -0700 | [diff] [blame] | 718 | ++index; |
| 719 | } |
| 720 | ++index; |
| 721 | } |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 722 | if (index >= fullName.length()) { |
| 723 | Log.e(TAG, "Invalid service found " + fullName); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 724 | break; |
| 725 | } |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 726 | |
paulhu | be18660 | 2022-04-12 07:18:23 +0000 | [diff] [blame] | 727 | String name = unescape(fullName.substring(0, index)); |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 728 | String rest = fullName.substring(index); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 729 | String type = rest.replace(".local.", ""); |
| 730 | |
| 731 | clientInfo.mResolvedService.setServiceName(name); |
| 732 | clientInfo.mResolvedService.setServiceType(type); |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 733 | clientInfo.mResolvedService.setPort(info.port); |
| 734 | clientInfo.mResolvedService.setTxtRecords(info.txtRecord); |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 735 | // Network will be added after SERVICE_GET_ADDR_SUCCESS |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 736 | |
| 737 | stopResolveService(id); |
Vinit Deshapnde | 4429e87 | 2013-11-12 15:36:37 -0800 | [diff] [blame] | 738 | removeRequestMap(clientId, id, clientInfo); |
| 739 | |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 740 | final int id2 = getUniqueId(); |
| 741 | if (getAddrInfo(id2, info.hostname, info.interfaceIdx)) { |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 742 | storeRequestMap(clientId, id2, clientInfo, NsdManager.RESOLVE_SERVICE); |
Vinit Deshapnde | 4429e87 | 2013-11-12 15:36:37 -0800 | [diff] [blame] | 743 | } else { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 744 | clientInfo.onResolveServiceFailed( |
| 745 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 746 | clientInfo.mResolvedService = null; |
| 747 | } |
| 748 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 749 | } |
| 750 | case IMDnsEventListener.SERVICE_RESOLUTION_FAILED: |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 751 | /* NNN resolveId errorCode */ |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 752 | stopResolveService(id); |
| 753 | removeRequestMap(clientId, id, clientInfo); |
| 754 | clientInfo.mResolvedService = null; |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 755 | clientInfo.onResolveServiceFailed( |
| 756 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 757 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 758 | case IMDnsEventListener.SERVICE_GET_ADDR_FAILED: |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 759 | /* NNN resolveId errorCode */ |
| 760 | stopGetAddrInfo(id); |
| 761 | removeRequestMap(clientId, id, clientInfo); |
| 762 | clientInfo.mResolvedService = null; |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 763 | clientInfo.onResolveServiceFailed( |
| 764 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 765 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 766 | case IMDnsEventListener.SERVICE_GET_ADDR_SUCCESS: { |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 767 | /* NNN resolveId hostname ttl addr interfaceIdx netId */ |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 768 | final GetAddressInfo info = (GetAddressInfo) obj; |
| 769 | final String address = info.address; |
| 770 | final int netId = info.netId; |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 771 | InetAddress serviceHost = null; |
| 772 | try { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 773 | serviceHost = InetAddress.getByName(address); |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 774 | } catch (UnknownHostException e) { |
| 775 | Log.wtf(TAG, "Invalid host in GET_ADDR_SUCCESS", e); |
| 776 | } |
| 777 | |
| 778 | // If the resolved service is on an interface without a network, consider it |
| 779 | // as a failure: it would not be usable by apps as they would need |
| 780 | // privileged permissions. |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 781 | if (netId != NETID_UNSET && serviceHost != null) { |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 782 | clientInfo.mResolvedService.setHost(serviceHost); |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 783 | setServiceNetworkForCallback(clientInfo.mResolvedService, |
| 784 | netId, info.interfaceIdx); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 785 | clientInfo.onResolveServiceSucceeded( |
| 786 | clientId, clientInfo.mResolvedService); |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 787 | } else { |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 788 | clientInfo.onResolveServiceFailed( |
| 789 | clientId, NsdManager.FAILURE_INTERNAL_ERROR); |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 790 | } |
| 791 | stopGetAddrInfo(id); |
| 792 | removeRequestMap(clientId, id, clientInfo); |
| 793 | clientInfo.mResolvedService = null; |
| 794 | break; |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 795 | } |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 796 | default: |
Hugo Benichi | f0c8409 | 2017-04-05 14:43:29 +0900 | [diff] [blame] | 797 | return false; |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 798 | } |
Hugo Benichi | f0c8409 | 2017-04-05 14:43:29 +0900 | [diff] [blame] | 799 | return true; |
Vairavan Srinivasan | 6ce4818 | 2012-08-05 13:14:12 -0700 | [diff] [blame] | 800 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 801 | } |
| 802 | } |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 803 | |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 804 | private static void setServiceNetworkForCallback(NsdServiceInfo info, int netId, int ifaceIdx) { |
| 805 | switch (netId) { |
| 806 | case NETID_UNSET: |
| 807 | info.setNetwork(null); |
| 808 | break; |
| 809 | case INetd.LOCAL_NET_ID: |
| 810 | // Special case for LOCAL_NET_ID: Networks on netId 99 are not generally |
| 811 | // visible / usable for apps, so do not return it. Store the interface |
| 812 | // index instead, so at least if the client tries to resolve the service |
| 813 | // with that NsdServiceInfo, it will be done on the same interface. |
| 814 | // If they recreate the NsdServiceInfo themselves, resolution would be |
| 815 | // done on all interfaces as before T, which should also work. |
| 816 | info.setNetwork(null); |
| 817 | info.setInterfaceIndex(ifaceIdx); |
| 818 | break; |
| 819 | default: |
| 820 | info.setNetwork(new Network(netId)); |
| 821 | } |
| 822 | } |
| 823 | |
paulhu | be18660 | 2022-04-12 07:18:23 +0000 | [diff] [blame] | 824 | // The full service name is escaped from standard DNS rules on mdnsresponder, making it suitable |
| 825 | // for passing to standard system DNS APIs such as res_query() . Thus, make the service name |
| 826 | // unescape for getting right service address. See "Notes on DNS Name Escaping" on |
| 827 | // external/mdnsresponder/mDNSShared/dns_sd.h for more details. |
| 828 | private String unescape(String s) { |
| 829 | StringBuilder sb = new StringBuilder(s.length()); |
| 830 | for (int i = 0; i < s.length(); ++i) { |
| 831 | char c = s.charAt(i); |
| 832 | if (c == '\\') { |
| 833 | if (++i >= s.length()) { |
| 834 | Log.e(TAG, "Unexpected end of escape sequence in: " + s); |
| 835 | break; |
| 836 | } |
| 837 | c = s.charAt(i); |
| 838 | if (c != '.' && c != '\\') { |
| 839 | if (i + 2 >= s.length()) { |
| 840 | Log.e(TAG, "Unexpected end of escape sequence in: " + s); |
| 841 | break; |
| 842 | } |
| 843 | c = (char) ((c - '0') * 100 + (s.charAt(i + 1) - '0') * 10 |
| 844 | + (s.charAt(i + 2) - '0')); |
| 845 | i += 2; |
| 846 | } |
| 847 | } |
| 848 | sb.append(c); |
| 849 | } |
| 850 | return sb.toString(); |
| 851 | } |
| 852 | |
Hugo Benichi | 803a2f0 | 2017-04-24 11:35:06 +0900 | [diff] [blame] | 853 | @VisibleForTesting |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 854 | NsdService(Context ctx, Handler handler, long cleanupDelayMs) { |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 855 | this(ctx, handler, cleanupDelayMs, new Dependencies()); |
| 856 | } |
| 857 | |
| 858 | @VisibleForTesting |
| 859 | NsdService(Context ctx, Handler handler, long cleanupDelayMs, Dependencies deps) { |
Luke Huang | 0529858 | 2021-06-13 16:52:05 +0000 | [diff] [blame] | 860 | mCleanupDelayMs = cleanupDelayMs; |
Hugo Benichi | 803a2f0 | 2017-04-24 11:35:06 +0900 | [diff] [blame] | 861 | mContext = ctx; |
Hugo Benichi | 803a2f0 | 2017-04-24 11:35:06 +0900 | [diff] [blame] | 862 | mNsdStateMachine = new NsdStateMachine(TAG, handler); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 863 | mNsdStateMachine.start(); |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 864 | mMDnsManager = ctx.getSystemService(MDnsManager.class); |
| 865 | mMDnsEventCallback = new MDnsEventCallback(mNsdStateMachine); |
Paul Hu | 4bd98ef | 2023-01-12 13:42:07 +0800 | [diff] [blame] | 866 | if (deps.isMdnsDiscoveryManagerEnabled(ctx)) { |
| 867 | mMdnsSocketProvider = deps.makeMdnsSocketProvider(ctx, handler.getLooper()); |
| 868 | mMdnsSocketClient = |
| 869 | new MdnsMultinetworkSocketClient(handler.getLooper(), mMdnsSocketProvider); |
| 870 | mMdnsDiscoveryManager = |
| 871 | deps.makeMdnsDiscoveryManager(new ExecutorProvider(), mMdnsSocketClient); |
| 872 | handler.post(() -> mMdnsSocketClient.setCallback(mMdnsDiscoveryManager)); |
| 873 | } else { |
| 874 | mMdnsSocketProvider = null; |
| 875 | mMdnsSocketClient = null; |
| 876 | mMdnsDiscoveryManager = null; |
| 877 | } |
| 878 | } |
| 879 | |
| 880 | /** |
| 881 | * Dependencies of NsdService, for injection in tests. |
| 882 | */ |
| 883 | @VisibleForTesting |
| 884 | public static class Dependencies { |
| 885 | /** |
| 886 | * Check whether or not MdnsDiscoveryManager feature is enabled. |
| 887 | * |
| 888 | * @param context The global context information about an app environment. |
| 889 | * @return true if MdnsDiscoveryManager feature is enabled. |
| 890 | */ |
| 891 | public boolean isMdnsDiscoveryManagerEnabled(Context context) { |
| 892 | return DeviceConfigUtils.isFeatureEnabled(context, NAMESPACE_CONNECTIVITY, |
| 893 | MDNS_DISCOVERY_MANAGER_VERSION, false /* defaultEnabled */); |
| 894 | } |
| 895 | |
| 896 | /** |
| 897 | * @see MdnsDiscoveryManager |
| 898 | */ |
| 899 | public MdnsDiscoveryManager makeMdnsDiscoveryManager( |
| 900 | ExecutorProvider executorProvider, MdnsSocketClientBase socketClient) { |
| 901 | return new MdnsDiscoveryManager(executorProvider, socketClient); |
| 902 | } |
| 903 | |
| 904 | /** |
| 905 | * @see MdnsSocketProvider |
| 906 | */ |
| 907 | public MdnsSocketProvider makeMdnsSocketProvider(Context context, Looper looper) { |
| 908 | return new MdnsSocketProvider(context, looper); |
| 909 | } |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 910 | } |
| 911 | |
paulhu | 1b35e82 | 2022-04-08 14:48:41 +0800 | [diff] [blame] | 912 | public static NsdService create(Context context) { |
Hugo Benichi | 803a2f0 | 2017-04-24 11:35:06 +0900 | [diff] [blame] | 913 | HandlerThread thread = new HandlerThread(TAG); |
| 914 | thread.start(); |
| 915 | Handler handler = new Handler(thread.getLooper()); |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 916 | NsdService service = new NsdService(context, handler, CLEANUP_DELAY_MS); |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 917 | return service; |
| 918 | } |
| 919 | |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 920 | private static class MDnsEventCallback extends IMDnsEventListener.Stub { |
| 921 | private final StateMachine mStateMachine; |
| 922 | |
| 923 | MDnsEventCallback(StateMachine sm) { |
| 924 | mStateMachine = sm; |
| 925 | } |
| 926 | |
| 927 | @Override |
| 928 | public void onServiceRegistrationStatus(final RegistrationInfo status) { |
| 929 | mStateMachine.sendMessage( |
| 930 | MDNS_SERVICE_EVENT, status.result, status.id, status); |
| 931 | } |
| 932 | |
| 933 | @Override |
| 934 | public void onServiceDiscoveryStatus(final DiscoveryInfo status) { |
| 935 | mStateMachine.sendMessage( |
| 936 | MDNS_SERVICE_EVENT, status.result, status.id, status); |
| 937 | } |
| 938 | |
| 939 | @Override |
| 940 | public void onServiceResolutionStatus(final ResolutionInfo status) { |
| 941 | mStateMachine.sendMessage( |
| 942 | MDNS_SERVICE_EVENT, status.result, status.id, status); |
| 943 | } |
| 944 | |
| 945 | @Override |
| 946 | public void onGettingServiceAddressStatus(final GetAddressInfo status) { |
| 947 | mStateMachine.sendMessage( |
| 948 | MDNS_SERVICE_EVENT, status.result, status.id, status); |
| 949 | } |
| 950 | |
| 951 | @Override |
| 952 | public int getInterfaceVersion() throws RemoteException { |
| 953 | return this.VERSION; |
| 954 | } |
| 955 | |
| 956 | @Override |
| 957 | public String getInterfaceHash() throws RemoteException { |
| 958 | return this.HASH; |
| 959 | } |
| 960 | } |
| 961 | |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 962 | @Override |
| 963 | public INsdServiceConnector connect(INsdManagerCallback cb) { |
Hugo Benichi | 803a2f0 | 2017-04-24 11:35:06 +0900 | [diff] [blame] | 964 | mContext.enforceCallingOrSelfPermission(android.Manifest.permission.INTERNET, "NsdService"); |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 965 | final INsdServiceConnector connector = new NsdServiceConnector(); |
| 966 | mNsdStateMachine.sendMessage(mNsdStateMachine.obtainMessage( |
| 967 | NsdManager.REGISTER_CLIENT, new Pair<>(connector, cb))); |
| 968 | return connector; |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 969 | } |
| 970 | |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 971 | private static class ListenerArgs { |
| 972 | public final NsdServiceConnector connector; |
| 973 | public final NsdServiceInfo serviceInfo; |
| 974 | ListenerArgs(NsdServiceConnector connector, NsdServiceInfo serviceInfo) { |
| 975 | this.connector = connector; |
| 976 | this.serviceInfo = serviceInfo; |
| 977 | } |
| 978 | } |
| 979 | |
| 980 | private class NsdServiceConnector extends INsdServiceConnector.Stub |
| 981 | implements IBinder.DeathRecipient { |
| 982 | @Override |
| 983 | public void registerService(int listenerKey, NsdServiceInfo serviceInfo) { |
| 984 | mNsdStateMachine.sendMessage(mNsdStateMachine.obtainMessage( |
| 985 | NsdManager.REGISTER_SERVICE, 0, listenerKey, |
| 986 | new ListenerArgs(this, serviceInfo))); |
| 987 | } |
| 988 | |
| 989 | @Override |
| 990 | public void unregisterService(int listenerKey) { |
| 991 | mNsdStateMachine.sendMessage(mNsdStateMachine.obtainMessage( |
| 992 | NsdManager.UNREGISTER_SERVICE, 0, listenerKey, |
| 993 | new ListenerArgs(this, null))); |
| 994 | } |
| 995 | |
| 996 | @Override |
| 997 | public void discoverServices(int listenerKey, NsdServiceInfo serviceInfo) { |
| 998 | mNsdStateMachine.sendMessage(mNsdStateMachine.obtainMessage( |
| 999 | NsdManager.DISCOVER_SERVICES, 0, listenerKey, |
| 1000 | new ListenerArgs(this, serviceInfo))); |
| 1001 | } |
| 1002 | |
| 1003 | @Override |
| 1004 | public void stopDiscovery(int listenerKey) { |
| 1005 | mNsdStateMachine.sendMessage(mNsdStateMachine.obtainMessage( |
| 1006 | NsdManager.STOP_DISCOVERY, 0, listenerKey, new ListenerArgs(this, null))); |
| 1007 | } |
| 1008 | |
| 1009 | @Override |
| 1010 | public void resolveService(int listenerKey, NsdServiceInfo serviceInfo) { |
| 1011 | mNsdStateMachine.sendMessage(mNsdStateMachine.obtainMessage( |
| 1012 | NsdManager.RESOLVE_SERVICE, 0, listenerKey, |
| 1013 | new ListenerArgs(this, serviceInfo))); |
| 1014 | } |
| 1015 | |
| 1016 | @Override |
| 1017 | public void startDaemon() { |
| 1018 | mNsdStateMachine.sendMessage(mNsdStateMachine.obtainMessage( |
| 1019 | NsdManager.DAEMON_STARTUP, new ListenerArgs(this, null))); |
| 1020 | } |
| 1021 | |
| 1022 | @Override |
| 1023 | public void binderDied() { |
| 1024 | mNsdStateMachine.sendMessage( |
| 1025 | mNsdStateMachine.obtainMessage(NsdManager.UNREGISTER_CLIENT, this)); |
| 1026 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 1027 | } |
| 1028 | |
Hugo Benichi | 912db99 | 2017-04-24 16:41:03 +0900 | [diff] [blame] | 1029 | private void sendNsdStateChangeBroadcast(boolean isEnabled) { |
Irfan Sheriff | 52fc83a | 2012-04-19 10:26:34 -0700 | [diff] [blame] | 1030 | final Intent intent = new Intent(NsdManager.ACTION_NSD_STATE_CHANGED); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 1031 | intent.addFlags(Intent.FLAG_RECEIVER_REGISTERED_ONLY_BEFORE_BOOT); |
Hugo Benichi | 912db99 | 2017-04-24 16:41:03 +0900 | [diff] [blame] | 1032 | int nsdState = isEnabled ? NsdManager.NSD_STATE_ENABLED : NsdManager.NSD_STATE_DISABLED; |
| 1033 | intent.putExtra(NsdManager.EXTRA_NSD_STATE, nsdState); |
Dianne Hackborn | 692107e | 2012-08-29 18:32:08 -0700 | [diff] [blame] | 1034 | mContext.sendStickyBroadcastAsUser(intent, UserHandle.ALL); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 1035 | } |
| 1036 | |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1037 | private int getUniqueId() { |
| 1038 | if (++mUniqueId == INVALID_ID) return ++mUniqueId; |
| 1039 | return mUniqueId; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1040 | } |
| 1041 | |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 1042 | private boolean registerService(int regId, NsdServiceInfo service) { |
Hugo Benichi | 6d70644 | 2017-04-24 16:19:58 +0900 | [diff] [blame] | 1043 | if (DBG) { |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 1044 | Log.d(TAG, "registerService: " + regId + " " + service); |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1045 | } |
Hugo Benichi | 6d70644 | 2017-04-24 16:19:58 +0900 | [diff] [blame] | 1046 | String name = service.getServiceName(); |
| 1047 | String type = service.getServiceType(); |
| 1048 | int port = service.getPort(); |
| 1049 | byte[] textRecord = service.getTxtRecord(); |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 1050 | final int registerInterface = getNetworkInterfaceIndex(service); |
| 1051 | if (service.getNetwork() != null && registerInterface == IFACE_IDX_ANY) { |
Paul Hu | 360a8e9 | 2022-04-26 11:14:14 +0800 | [diff] [blame] | 1052 | Log.e(TAG, "Interface to register service on not found"); |
| 1053 | return false; |
| 1054 | } |
| 1055 | return mMDnsManager.registerService(regId, name, type, port, textRecord, registerInterface); |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1056 | } |
| 1057 | |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1058 | private boolean unregisterService(int regId) { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 1059 | return mMDnsManager.stopOperation(regId); |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1060 | } |
| 1061 | |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 1062 | private boolean discoverServices(int discoveryId, NsdServiceInfo serviceInfo) { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 1063 | final String type = serviceInfo.getServiceType(); |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 1064 | final int discoverInterface = getNetworkInterfaceIndex(serviceInfo); |
| 1065 | if (serviceInfo.getNetwork() != null && discoverInterface == IFACE_IDX_ANY) { |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 1066 | Log.e(TAG, "Interface to discover service on not found"); |
| 1067 | return false; |
| 1068 | } |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 1069 | return mMDnsManager.discover(discoveryId, type, discoverInterface); |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1070 | } |
| 1071 | |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1072 | private boolean stopServiceDiscovery(int discoveryId) { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 1073 | return mMDnsManager.stopOperation(discoveryId); |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1074 | } |
| 1075 | |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 1076 | private boolean resolveService(int resolveId, NsdServiceInfo service) { |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 1077 | final String name = service.getServiceName(); |
| 1078 | final String type = service.getServiceType(); |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 1079 | final int resolveInterface = getNetworkInterfaceIndex(service); |
| 1080 | if (service.getNetwork() != null && resolveInterface == IFACE_IDX_ANY) { |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 1081 | Log.e(TAG, "Interface to resolve service on not found"); |
| 1082 | return false; |
| 1083 | } |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 1084 | return mMDnsManager.resolve(resolveId, name, type, "local.", resolveInterface); |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 1085 | } |
| 1086 | |
| 1087 | /** |
| 1088 | * Guess the interface to use to resolve or discover a service on a specific network. |
| 1089 | * |
| 1090 | * This is an imperfect guess, as for example the network may be gone or not yet fully |
| 1091 | * registered. This is fine as failing is correct if the network is gone, and a client |
| 1092 | * attempting to resolve/discover on a network not yet setup would have a bad time anyway; also |
| 1093 | * this is to support the legacy mdnsresponder implementation, which historically resolved |
| 1094 | * services on an unspecified network. |
| 1095 | */ |
Remi NGUYEN VAN | 1a8ee10 | 2022-05-30 12:42:24 +0900 | [diff] [blame] | 1096 | private int getNetworkInterfaceIndex(NsdServiceInfo serviceInfo) { |
| 1097 | final Network network = serviceInfo.getNetwork(); |
| 1098 | if (network == null) { |
| 1099 | // Fallback to getInterfaceIndex if present (typically if the NsdServiceInfo was |
| 1100 | // provided by NsdService from discovery results, and the service was found on an |
| 1101 | // interface that has no app-usable Network). |
| 1102 | if (serviceInfo.getInterfaceIndex() != 0) { |
| 1103 | return serviceInfo.getInterfaceIndex(); |
| 1104 | } |
| 1105 | return IFACE_IDX_ANY; |
| 1106 | } |
Remi NGUYEN VAN | 2365130 | 2021-12-16 15:31:16 +0900 | [diff] [blame] | 1107 | |
| 1108 | final ConnectivityManager cm = mContext.getSystemService(ConnectivityManager.class); |
| 1109 | if (cm == null) { |
| 1110 | Log.wtf(TAG, "No ConnectivityManager for resolveService"); |
| 1111 | return IFACE_IDX_ANY; |
| 1112 | } |
| 1113 | final LinkProperties lp = cm.getLinkProperties(network); |
| 1114 | if (lp == null) return IFACE_IDX_ANY; |
| 1115 | |
| 1116 | // Only resolve on non-stacked interfaces |
| 1117 | final NetworkInterface iface; |
| 1118 | try { |
| 1119 | iface = NetworkInterface.getByName(lp.getInterfaceName()); |
| 1120 | } catch (SocketException e) { |
| 1121 | Log.e(TAG, "Error querying interface", e); |
| 1122 | return IFACE_IDX_ANY; |
| 1123 | } |
| 1124 | |
| 1125 | if (iface == null) { |
| 1126 | Log.e(TAG, "Interface not found: " + lp.getInterfaceName()); |
| 1127 | return IFACE_IDX_ANY; |
| 1128 | } |
| 1129 | |
| 1130 | return iface.getIndex(); |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1131 | } |
| 1132 | |
| 1133 | private boolean stopResolveService(int resolveId) { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 1134 | return mMDnsManager.stopOperation(resolveId); |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1135 | } |
| 1136 | |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 1137 | private boolean getAddrInfo(int resolveId, String hostname, int interfaceIdx) { |
| 1138 | return mMDnsManager.getServiceAddress(resolveId, hostname, interfaceIdx); |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1139 | } |
| 1140 | |
| 1141 | private boolean stopGetAddrInfo(int resolveId) { |
paulhu | 2b9ed95 | 2022-02-10 21:58:32 +0800 | [diff] [blame] | 1142 | return mMDnsManager.stopOperation(resolveId); |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1143 | } |
| 1144 | |
| 1145 | @Override |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 1146 | public void dump(FileDescriptor fd, PrintWriter pw, String[] args) { |
paulhu | 3ffffe7 | 2021-09-16 10:15:22 +0800 | [diff] [blame] | 1147 | if (!PermissionUtils.checkDumpPermission(mContext, TAG, pw)) return; |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1148 | |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 1149 | for (ClientInfo client : mClients.values()) { |
| 1150 | pw.println("Client Info"); |
| 1151 | pw.println(client); |
| 1152 | } |
| 1153 | |
| 1154 | mNsdStateMachine.dump(fd, pw, args); |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1155 | } |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1156 | |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1157 | /* Information tracked per client */ |
| 1158 | private class ClientInfo { |
| 1159 | |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 1160 | private static final int MAX_LIMIT = 10; |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 1161 | private final INsdManagerCallback mCb; |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1162 | /* Remembers a resolved service until getaddrinfo completes */ |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 1163 | private NsdServiceInfo mResolvedService; |
| 1164 | |
| 1165 | /* A map from client id to unique id sent to mDns */ |
Hugo Benichi | d2552ae | 2017-04-11 14:42:47 +0900 | [diff] [blame] | 1166 | private final SparseIntArray mClientIds = new SparseIntArray(); |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1167 | |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 1168 | /* A map from client id to the type of the request we had received */ |
Hugo Benichi | d2552ae | 2017-04-11 14:42:47 +0900 | [diff] [blame] | 1169 | private final SparseIntArray mClientRequests = new SparseIntArray(); |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 1170 | |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 1171 | /* A map from client id to the MdnsListener */ |
| 1172 | private final SparseArray<MdnsListener> mListeners = new SparseArray<>(); |
| 1173 | |
Luke Huang | f7277ed | 2021-07-12 21:15:10 +0800 | [diff] [blame] | 1174 | // The target SDK of this client < Build.VERSION_CODES.S |
| 1175 | private boolean mIsLegacy = false; |
| 1176 | |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 1177 | private ClientInfo(INsdManagerCallback cb) { |
| 1178 | mCb = cb; |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 1179 | if (DBG) Log.d(TAG, "New client"); |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1180 | } |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 1181 | |
| 1182 | @Override |
| 1183 | public String toString() { |
Jeff Sharkey | 6346538 | 2020-10-17 21:20:13 -0600 | [diff] [blame] | 1184 | StringBuilder sb = new StringBuilder(); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 1185 | sb.append("mResolvedService ").append(mResolvedService).append("\n"); |
Luke Huang | f7277ed | 2021-07-12 21:15:10 +0800 | [diff] [blame] | 1186 | sb.append("mIsLegacy ").append(mIsLegacy).append("\n"); |
Irfan Sheriff | e4c42f4 | 2012-05-03 16:44:27 -0700 | [diff] [blame] | 1187 | for(int i = 0; i< mClientIds.size(); i++) { |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 1188 | int clientID = mClientIds.keyAt(i); |
| 1189 | sb.append("clientId ").append(clientID). |
| 1190 | append(" mDnsId ").append(mClientIds.valueAt(i)). |
| 1191 | append(" type ").append(mClientRequests.get(clientID)).append("\n"); |
Irfan Sheriff | 7500665 | 2012-04-17 23:15:29 -0700 | [diff] [blame] | 1192 | } |
| 1193 | return sb.toString(); |
| 1194 | } |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 1195 | |
Luke Huang | f7277ed | 2021-07-12 21:15:10 +0800 | [diff] [blame] | 1196 | private boolean isLegacy() { |
| 1197 | return mIsLegacy; |
| 1198 | } |
| 1199 | |
| 1200 | private void setLegacy() { |
| 1201 | mIsLegacy = true; |
| 1202 | } |
| 1203 | |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 1204 | // Remove any pending requests from the global map when we get rid of a client, |
| 1205 | // and send cancellations to the daemon. |
| 1206 | private void expungeAllRequests() { |
| 1207 | int globalId, clientId, i; |
Hugo Benichi | d2552ae | 2017-04-11 14:42:47 +0900 | [diff] [blame] | 1208 | // TODO: to keep handler responsive, do not clean all requests for that client at once. |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 1209 | for (i = 0; i < mClientIds.size(); i++) { |
| 1210 | clientId = mClientIds.keyAt(i); |
| 1211 | globalId = mClientIds.valueAt(i); |
| 1212 | mIdToClientInfoMap.remove(globalId); |
paulhu | b222570 | 2021-11-17 09:35:33 +0800 | [diff] [blame] | 1213 | if (DBG) { |
| 1214 | Log.d(TAG, "Terminating client-ID " + clientId |
| 1215 | + " global-ID " + globalId + " type " + mClientRequests.get(clientId)); |
| 1216 | } |
Dave Platt | feff2af | 2014-03-07 14:48:22 -0800 | [diff] [blame] | 1217 | switch (mClientRequests.get(clientId)) { |
| 1218 | case NsdManager.DISCOVER_SERVICES: |
| 1219 | stopServiceDiscovery(globalId); |
| 1220 | break; |
| 1221 | case NsdManager.RESOLVE_SERVICE: |
| 1222 | stopResolveService(globalId); |
| 1223 | break; |
| 1224 | case NsdManager.REGISTER_SERVICE: |
| 1225 | unregisterService(globalId); |
| 1226 | break; |
| 1227 | default: |
| 1228 | break; |
| 1229 | } |
| 1230 | } |
| 1231 | mClientIds.clear(); |
| 1232 | mClientRequests.clear(); |
| 1233 | } |
| 1234 | |
Paul Hu | 23fa202 | 2023-01-13 22:57:24 +0800 | [diff] [blame^] | 1235 | void unregisterAllListeners() { |
| 1236 | for (int i = 0; i < mListeners.size(); i++) { |
| 1237 | final MdnsListener listener = mListeners.valueAt(i); |
| 1238 | mMdnsDiscoveryManager.unregisterListener( |
| 1239 | listener.getListenedServiceType(), listener); |
| 1240 | } |
| 1241 | mListeners.clear(); |
| 1242 | } |
| 1243 | |
Christopher Lane | 7441122 | 2014-04-25 18:39:07 -0700 | [diff] [blame] | 1244 | // mClientIds is a sparse array of listener id -> mDnsClient id. For a given mDnsClient id, |
| 1245 | // return the corresponding listener id. mDnsClient id is also called a global id. |
| 1246 | private int getClientId(final int globalId) { |
Hugo Benichi | d2552ae | 2017-04-11 14:42:47 +0900 | [diff] [blame] | 1247 | int idx = mClientIds.indexOfValue(globalId); |
| 1248 | if (idx < 0) { |
| 1249 | return idx; |
Christopher Lane | 7441122 | 2014-04-25 18:39:07 -0700 | [diff] [blame] | 1250 | } |
Hugo Benichi | d2552ae | 2017-04-11 14:42:47 +0900 | [diff] [blame] | 1251 | return mClientIds.keyAt(idx); |
Christopher Lane | 7441122 | 2014-04-25 18:39:07 -0700 | [diff] [blame] | 1252 | } |
Remi NGUYEN VAN | 62eb76e | 2021-09-09 17:39:05 +0900 | [diff] [blame] | 1253 | |
| 1254 | void onDiscoverServicesStarted(int listenerKey, NsdServiceInfo info) { |
| 1255 | try { |
| 1256 | mCb.onDiscoverServicesStarted(listenerKey, info); |
| 1257 | } catch (RemoteException e) { |
| 1258 | Log.e(TAG, "Error calling onDiscoverServicesStarted", e); |
| 1259 | } |
| 1260 | } |
| 1261 | |
| 1262 | void onDiscoverServicesFailed(int listenerKey, int error) { |
| 1263 | try { |
| 1264 | mCb.onDiscoverServicesFailed(listenerKey, error); |
| 1265 | } catch (RemoteException e) { |
| 1266 | Log.e(TAG, "Error calling onDiscoverServicesFailed", e); |
| 1267 | } |
| 1268 | } |
| 1269 | |
| 1270 | void onServiceFound(int listenerKey, NsdServiceInfo info) { |
| 1271 | try { |
| 1272 | mCb.onServiceFound(listenerKey, info); |
| 1273 | } catch (RemoteException e) { |
| 1274 | Log.e(TAG, "Error calling onServiceFound(", e); |
| 1275 | } |
| 1276 | } |
| 1277 | |
| 1278 | void onServiceLost(int listenerKey, NsdServiceInfo info) { |
| 1279 | try { |
| 1280 | mCb.onServiceLost(listenerKey, info); |
| 1281 | } catch (RemoteException e) { |
| 1282 | Log.e(TAG, "Error calling onServiceLost(", e); |
| 1283 | } |
| 1284 | } |
| 1285 | |
| 1286 | void onStopDiscoveryFailed(int listenerKey, int error) { |
| 1287 | try { |
| 1288 | mCb.onStopDiscoveryFailed(listenerKey, error); |
| 1289 | } catch (RemoteException e) { |
| 1290 | Log.e(TAG, "Error calling onStopDiscoveryFailed", e); |
| 1291 | } |
| 1292 | } |
| 1293 | |
| 1294 | void onStopDiscoverySucceeded(int listenerKey) { |
| 1295 | try { |
| 1296 | mCb.onStopDiscoverySucceeded(listenerKey); |
| 1297 | } catch (RemoteException e) { |
| 1298 | Log.e(TAG, "Error calling onStopDiscoverySucceeded", e); |
| 1299 | } |
| 1300 | } |
| 1301 | |
| 1302 | void onRegisterServiceFailed(int listenerKey, int error) { |
| 1303 | try { |
| 1304 | mCb.onRegisterServiceFailed(listenerKey, error); |
| 1305 | } catch (RemoteException e) { |
| 1306 | Log.e(TAG, "Error calling onRegisterServiceFailed", e); |
| 1307 | } |
| 1308 | } |
| 1309 | |
| 1310 | void onRegisterServiceSucceeded(int listenerKey, NsdServiceInfo info) { |
| 1311 | try { |
| 1312 | mCb.onRegisterServiceSucceeded(listenerKey, info); |
| 1313 | } catch (RemoteException e) { |
| 1314 | Log.e(TAG, "Error calling onRegisterServiceSucceeded", e); |
| 1315 | } |
| 1316 | } |
| 1317 | |
| 1318 | void onUnregisterServiceFailed(int listenerKey, int error) { |
| 1319 | try { |
| 1320 | mCb.onUnregisterServiceFailed(listenerKey, error); |
| 1321 | } catch (RemoteException e) { |
| 1322 | Log.e(TAG, "Error calling onUnregisterServiceFailed", e); |
| 1323 | } |
| 1324 | } |
| 1325 | |
| 1326 | void onUnregisterServiceSucceeded(int listenerKey) { |
| 1327 | try { |
| 1328 | mCb.onUnregisterServiceSucceeded(listenerKey); |
| 1329 | } catch (RemoteException e) { |
| 1330 | Log.e(TAG, "Error calling onUnregisterServiceSucceeded", e); |
| 1331 | } |
| 1332 | } |
| 1333 | |
| 1334 | void onResolveServiceFailed(int listenerKey, int error) { |
| 1335 | try { |
| 1336 | mCb.onResolveServiceFailed(listenerKey, error); |
| 1337 | } catch (RemoteException e) { |
| 1338 | Log.e(TAG, "Error calling onResolveServiceFailed", e); |
| 1339 | } |
| 1340 | } |
| 1341 | |
| 1342 | void onResolveServiceSucceeded(int listenerKey, NsdServiceInfo info) { |
| 1343 | try { |
| 1344 | mCb.onResolveServiceSucceeded(listenerKey, info); |
| 1345 | } catch (RemoteException e) { |
| 1346 | Log.e(TAG, "Error calling onResolveServiceSucceeded", e); |
| 1347 | } |
| 1348 | } |
Irfan Sheriff | e8de246 | 2012-04-11 14:52:19 -0700 | [diff] [blame] | 1349 | } |
Irfan Sheriff | 77ec558 | 2012-03-22 17:01:39 -0700 | [diff] [blame] | 1350 | } |