blob: 2a7a2eb6873ab1494fec0abaa4166d9e1d14b53b [file] [log] [blame]
Irfan Sheriff77ec5582012-03-22 17:01:39 -07001/*
2 * Copyright (C) 2010 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17package com.android.server;
18
19import android.content.Context;
Irfan Sheriff75006652012-04-17 23:15:29 -070020import android.content.ContentResolver;
21import android.content.Intent;
Irfan Sheriff77ec5582012-03-22 17:01:39 -070022import android.content.pm.PackageManager;
Irfan Sheriff67c80ed2012-06-01 16:44:13 -070023import android.database.ContentObserver;
Irfan Sheriffe4c42f42012-05-03 16:44:27 -070024import android.net.nsd.NsdServiceInfo;
Irfan Sheriff77ec5582012-03-22 17:01:39 -070025import android.net.nsd.DnsSdTxtRecord;
26import android.net.nsd.INsdManager;
27import android.net.nsd.NsdManager;
28import android.os.Binder;
29import android.os.Handler;
30import android.os.HandlerThread;
31import android.os.Message;
32import android.os.Messenger;
33import android.os.IBinder;
Dianne Hackborn692107e2012-08-29 18:32:08 -070034import android.os.UserHandle;
Irfan Sheriff75006652012-04-17 23:15:29 -070035import android.provider.Settings;
Irfan Sheriff77ec5582012-03-22 17:01:39 -070036import android.util.Slog;
Irfan Sheriffe4c42f42012-05-03 16:44:27 -070037import android.util.SparseArray;
Irfan Sheriff77ec5582012-03-22 17:01:39 -070038
39import java.io.FileDescriptor;
40import java.io.PrintWriter;
Irfan Sheriffe8de2462012-04-11 14:52:19 -070041import java.net.InetAddress;
Irfan Sheriff77ec5582012-03-22 17:01:39 -070042import java.util.ArrayList;
Irfan Sheriffe8de2462012-04-11 14:52:19 -070043import java.util.HashMap;
Irfan Sheriff77ec5582012-03-22 17:01:39 -070044import java.util.List;
Irfan Sheriffe8de2462012-04-11 14:52:19 -070045import java.util.concurrent.CountDownLatch;
Irfan Sheriff77ec5582012-03-22 17:01:39 -070046
47import com.android.internal.app.IBatteryStats;
48import com.android.internal.telephony.TelephonyIntents;
49import com.android.internal.util.AsyncChannel;
Irfan Sheriff75006652012-04-17 23:15:29 -070050import com.android.internal.util.Protocol;
51import com.android.internal.util.State;
52import com.android.internal.util.StateMachine;
Irfan Sheriff77ec5582012-03-22 17:01:39 -070053import com.android.server.am.BatteryStatsService;
54import com.android.server.NativeDaemonConnector.Command;
55import com.android.internal.R;
56
57/**
58 * Network Service Discovery Service handles remote service discovery operation requests by
59 * implementing the INsdManager interface.
60 *
61 * @hide
62 */
63public class NsdService extends INsdManager.Stub {
64 private static final String TAG = "NsdService";
65 private static final String MDNS_TAG = "mDnsConnector";
66
67 private static final boolean DBG = true;
68
69 private Context mContext;
Irfan Sheriff75006652012-04-17 23:15:29 -070070 private ContentResolver mContentResolver;
71 private NsdStateMachine mNsdStateMachine;
Irfan Sheriff77ec5582012-03-22 17:01:39 -070072
73 /**
74 * Clients receiving asynchronous messages
75 */
Irfan Sheriffe8de2462012-04-11 14:52:19 -070076 private HashMap<Messenger, ClientInfo> mClients = new HashMap<Messenger, ClientInfo>();
Irfan Sheriff77ec5582012-03-22 17:01:39 -070077
Irfan Sheriffe4c42f42012-05-03 16:44:27 -070078 /* A map from unique id to client info */
79 private SparseArray<ClientInfo> mIdToClientInfoMap= new SparseArray<ClientInfo>();
80
Irfan Sheriff77ec5582012-03-22 17:01:39 -070081 private AsyncChannel mReplyChannel = new AsyncChannel();
82
Irfan Sheriffe8de2462012-04-11 14:52:19 -070083 private int INVALID_ID = 0;
84 private int mUniqueId = 1;
85
Irfan Sheriff75006652012-04-17 23:15:29 -070086 private static final int BASE = Protocol.BASE_NSD_MANAGER;
Irfan Sheriffe4c42f42012-05-03 16:44:27 -070087 private static final int CMD_TO_STRING_COUNT = NsdManager.RESOLVE_SERVICE - BASE + 1;
Irfan Sheriff75006652012-04-17 23:15:29 -070088 private static String[] sCmdToString = new String[CMD_TO_STRING_COUNT];
Irfan Sheriff77ec5582012-03-22 17:01:39 -070089
Irfan Sheriff75006652012-04-17 23:15:29 -070090 static {
91 sCmdToString[NsdManager.DISCOVER_SERVICES - BASE] = "DISCOVER";
92 sCmdToString[NsdManager.STOP_DISCOVERY - BASE] = "STOP-DISCOVER";
93 sCmdToString[NsdManager.REGISTER_SERVICE - BASE] = "REGISTER";
94 sCmdToString[NsdManager.UNREGISTER_SERVICE - BASE] = "UNREGISTER";
95 sCmdToString[NsdManager.RESOLVE_SERVICE - BASE] = "RESOLVE";
Irfan Sheriff75006652012-04-17 23:15:29 -070096 }
Irfan Sheriff77ec5582012-03-22 17:01:39 -070097
Irfan Sheriff75006652012-04-17 23:15:29 -070098 private static String cmdToString(int cmd) {
99 cmd -= BASE;
100 if ((cmd >= 0) && (cmd < sCmdToString.length)) {
101 return sCmdToString[cmd];
102 } else {
103 return null;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700104 }
105 }
Irfan Sheriff75006652012-04-17 23:15:29 -0700106
107 private class NsdStateMachine extends StateMachine {
108
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700109 private final DefaultState mDefaultState = new DefaultState();
110 private final DisabledState mDisabledState = new DisabledState();
111 private final EnabledState mEnabledState = new EnabledState();
Irfan Sheriff75006652012-04-17 23:15:29 -0700112
113 @Override
Wink Saville358f5d42012-05-29 12:40:46 -0700114 protected String getWhatToString(int what) {
115 return cmdToString(what);
Irfan Sheriff75006652012-04-17 23:15:29 -0700116 }
117
Irfan Sheriff67c80ed2012-06-01 16:44:13 -0700118 /**
119 * Observes the NSD on/off setting, and takes action when changed.
120 */
121 private void registerForNsdSetting() {
122 ContentObserver contentObserver = new ContentObserver(this.getHandler()) {
123 @Override
124 public void onChange(boolean selfChange) {
125 if (isNsdEnabled()) {
126 mNsdStateMachine.sendMessage(NsdManager.ENABLE);
127 } else {
128 mNsdStateMachine.sendMessage(NsdManager.DISABLE);
129 }
130 }
131 };
132
133 mContext.getContentResolver().registerContentObserver(
134 Settings.Secure.getUriFor(Settings.Secure.NSD_ON),
135 false, contentObserver);
136 }
137
Irfan Sheriff75006652012-04-17 23:15:29 -0700138 NsdStateMachine(String name) {
139 super(name);
140 addState(mDefaultState);
141 addState(mDisabledState, mDefaultState);
142 addState(mEnabledState, mDefaultState);
143 if (isNsdEnabled()) {
144 setInitialState(mEnabledState);
145 } else {
146 setInitialState(mDisabledState);
147 }
Wink Saville358f5d42012-05-29 12:40:46 -0700148 setLogRecSize(25);
Irfan Sheriff67c80ed2012-06-01 16:44:13 -0700149 registerForNsdSetting();
Irfan Sheriff75006652012-04-17 23:15:29 -0700150 }
151
152 class DefaultState extends State {
153 @Override
154 public boolean processMessage(Message msg) {
155 switch (msg.what) {
156 case AsyncChannel.CMD_CHANNEL_HALF_CONNECTED:
157 if (msg.arg1 == AsyncChannel.STATUS_SUCCESSFUL) {
158 AsyncChannel c = (AsyncChannel) msg.obj;
159 if (DBG) Slog.d(TAG, "New client listening to asynchronous messages");
160 c.sendMessage(AsyncChannel.CMD_CHANNEL_FULLY_CONNECTED);
161 ClientInfo cInfo = new ClientInfo(c, msg.replyTo);
162 mClients.put(msg.replyTo, cInfo);
163 } else {
164 Slog.e(TAG, "Client connection failure, error=" + msg.arg1);
165 }
166 break;
167 case AsyncChannel.CMD_CHANNEL_DISCONNECTED:
168 if (msg.arg1 == AsyncChannel.STATUS_SEND_UNSUCCESSFUL) {
169 Slog.e(TAG, "Send failed, client connection lost");
170 } else {
171 if (DBG) Slog.d(TAG, "Client connection lost with reason: " + msg.arg1);
172 }
173 mClients.remove(msg.replyTo);
174 break;
175 case AsyncChannel.CMD_CHANNEL_FULL_CONNECTION:
176 AsyncChannel ac = new AsyncChannel();
177 ac.connect(mContext, getHandler(), msg.replyTo);
178 break;
179 case NsdManager.DISCOVER_SERVICES:
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700180 replyToMessage(msg, NsdManager.DISCOVER_SERVICES_FAILED,
181 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700182 break;
183 case NsdManager.STOP_DISCOVERY:
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700184 replyToMessage(msg, NsdManager.STOP_DISCOVERY_FAILED,
185 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700186 break;
187 case NsdManager.REGISTER_SERVICE:
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700188 replyToMessage(msg, NsdManager.REGISTER_SERVICE_FAILED,
189 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700190 break;
191 case NsdManager.UNREGISTER_SERVICE:
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700192 replyToMessage(msg, NsdManager.UNREGISTER_SERVICE_FAILED,
193 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700194 break;
195 case NsdManager.RESOLVE_SERVICE:
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700196 replyToMessage(msg, NsdManager.RESOLVE_SERVICE_FAILED,
197 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700198 break;
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700199 case NsdManager.NATIVE_DAEMON_EVENT:
Irfan Sheriff75006652012-04-17 23:15:29 -0700200 default:
201 Slog.e(TAG, "Unhandled " + msg);
202 return NOT_HANDLED;
203 }
204 return HANDLED;
205 }
206 }
207
208 class DisabledState extends State {
209 @Override
210 public void enter() {
211 sendNsdStateChangeBroadcast(false);
212 }
213
214 @Override
215 public boolean processMessage(Message msg) {
216 switch (msg.what) {
217 case NsdManager.ENABLE:
218 transitionTo(mEnabledState);
219 break;
220 default:
221 return NOT_HANDLED;
222 }
223 return HANDLED;
224 }
225 }
226
227 class EnabledState extends State {
228 @Override
229 public void enter() {
230 sendNsdStateChangeBroadcast(true);
231 if (mClients.size() > 0) {
232 startMDnsDaemon();
233 }
234 }
235
236 @Override
237 public void exit() {
238 if (mClients.size() > 0) {
239 stopMDnsDaemon();
240 }
241 }
242
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700243 private boolean requestLimitReached(ClientInfo clientInfo) {
244 if (clientInfo.mClientIds.size() >= ClientInfo.MAX_LIMIT) {
245 if (DBG) Slog.d(TAG, "Exceeded max outstanding requests " + clientInfo);
246 return true;
247 }
248 return false;
249 }
250
251 private void storeRequestMap(int clientId, int globalId, ClientInfo clientInfo) {
252 clientInfo.mClientIds.put(clientId, globalId);
253 mIdToClientInfoMap.put(globalId, clientInfo);
254 }
255
256 private void removeRequestMap(int clientId, int globalId, ClientInfo clientInfo) {
257 clientInfo.mClientIds.remove(clientId);
258 mIdToClientInfoMap.remove(globalId);
259 }
260
Irfan Sheriff75006652012-04-17 23:15:29 -0700261 @Override
262 public boolean processMessage(Message msg) {
263 ClientInfo clientInfo;
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700264 NsdServiceInfo servInfo;
Irfan Sheriff75006652012-04-17 23:15:29 -0700265 boolean result = HANDLED;
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700266 int id;
Irfan Sheriff75006652012-04-17 23:15:29 -0700267 switch (msg.what) {
268 case AsyncChannel.CMD_CHANNEL_HALF_CONNECTED:
269 //First client
270 if (msg.arg1 == AsyncChannel.STATUS_SUCCESSFUL &&
271 mClients.size() == 0) {
272 startMDnsDaemon();
273 }
274 result = NOT_HANDLED;
275 break;
276 case AsyncChannel.CMD_CHANNEL_DISCONNECTED:
277 //Last client
278 if (mClients.size() == 1) {
279 stopMDnsDaemon();
280 }
281 result = NOT_HANDLED;
282 break;
283 case NsdManager.DISABLE:
284 //TODO: cleanup clients
285 transitionTo(mDisabledState);
286 break;
287 case NsdManager.DISCOVER_SERVICES:
288 if (DBG) Slog.d(TAG, "Discover services");
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700289 servInfo = (NsdServiceInfo) msg.obj;
Irfan Sheriff75006652012-04-17 23:15:29 -0700290 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700291
292 if (requestLimitReached(clientInfo)) {
293 replyToMessage(msg, NsdManager.DISCOVER_SERVICES_FAILED,
294 NsdManager.FAILURE_MAX_LIMIT);
Irfan Sheriff75006652012-04-17 23:15:29 -0700295 break;
296 }
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700297
298 id = getUniqueId();
299 if (discoverServices(id, servInfo.getServiceType())) {
300 if (DBG) {
301 Slog.d(TAG, "Discover " + msg.arg2 + " " + id +
302 servInfo.getServiceType());
303 }
304 storeRequestMap(msg.arg2, id, clientInfo);
305 replyToMessage(msg, NsdManager.DISCOVER_SERVICES_STARTED, servInfo);
Irfan Sheriff75006652012-04-17 23:15:29 -0700306 } else {
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700307 stopServiceDiscovery(id);
308 replyToMessage(msg, NsdManager.DISCOVER_SERVICES_FAILED,
309 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700310 }
311 break;
312 case NsdManager.STOP_DISCOVERY:
313 if (DBG) Slog.d(TAG, "Stop service discovery");
314 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700315
316 try {
317 id = clientInfo.mClientIds.get(msg.arg2).intValue();
318 } catch (NullPointerException e) {
319 replyToMessage(msg, NsdManager.STOP_DISCOVERY_FAILED,
320 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700321 break;
322 }
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700323 removeRequestMap(msg.arg2, id, clientInfo);
324 if (stopServiceDiscovery(id)) {
325 replyToMessage(msg, NsdManager.STOP_DISCOVERY_SUCCEEDED);
Irfan Sheriff75006652012-04-17 23:15:29 -0700326 } else {
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700327 replyToMessage(msg, NsdManager.STOP_DISCOVERY_FAILED,
328 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700329 }
330 break;
331 case NsdManager.REGISTER_SERVICE:
332 if (DBG) Slog.d(TAG, "Register service");
333 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700334 if (requestLimitReached(clientInfo)) {
335 replyToMessage(msg, NsdManager.REGISTER_SERVICE_FAILED,
336 NsdManager.FAILURE_MAX_LIMIT);
337 break;
Irfan Sheriff75006652012-04-17 23:15:29 -0700338 }
339
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700340 id = getUniqueId();
341 if (registerService(id, (NsdServiceInfo) msg.obj)) {
342 if (DBG) Slog.d(TAG, "Register " + msg.arg2 + " " + id);
343 storeRequestMap(msg.arg2, id, clientInfo);
344 // Return success after mDns reports success
Irfan Sheriff75006652012-04-17 23:15:29 -0700345 } else {
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700346 unregisterService(id);
347 replyToMessage(msg, NsdManager.REGISTER_SERVICE_FAILED,
348 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700349 }
350 break;
351 case NsdManager.UNREGISTER_SERVICE:
352 if (DBG) Slog.d(TAG, "unregister service");
353 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700354 try {
355 id = clientInfo.mClientIds.get(msg.arg2).intValue();
356 } catch (NullPointerException e) {
357 replyToMessage(msg, NsdManager.UNREGISTER_SERVICE_FAILED,
358 NsdManager.FAILURE_INTERNAL_ERROR);
359 break;
Irfan Sheriff75006652012-04-17 23:15:29 -0700360 }
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700361 removeRequestMap(msg.arg2, id, clientInfo);
362 if (unregisterService(id)) {
363 replyToMessage(msg, NsdManager.UNREGISTER_SERVICE_SUCCEEDED);
364 } else {
365 replyToMessage(msg, NsdManager.UNREGISTER_SERVICE_FAILED,
366 NsdManager.FAILURE_INTERNAL_ERROR);
367 }
Irfan Sheriff75006652012-04-17 23:15:29 -0700368 break;
369 case NsdManager.RESOLVE_SERVICE:
370 if (DBG) Slog.d(TAG, "Resolve service");
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700371 servInfo = (NsdServiceInfo) msg.obj;
Irfan Sheriff75006652012-04-17 23:15:29 -0700372 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriff75006652012-04-17 23:15:29 -0700373
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700374
375 if (clientInfo.mResolvedService != null) {
376 replyToMessage(msg, NsdManager.RESOLVE_SERVICE_FAILED,
377 NsdManager.FAILURE_ALREADY_ACTIVE);
Irfan Sheriff75006652012-04-17 23:15:29 -0700378 break;
379 }
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700380
381 id = getUniqueId();
382 if (resolveService(id, servInfo)) {
383 clientInfo.mResolvedService = new NsdServiceInfo();
384 storeRequestMap(msg.arg2, id, clientInfo);
Irfan Sheriff75006652012-04-17 23:15:29 -0700385 } else {
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700386 replyToMessage(msg, NsdManager.RESOLVE_SERVICE_FAILED,
387 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff75006652012-04-17 23:15:29 -0700388 }
389 break;
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700390 case NsdManager.NATIVE_DAEMON_EVENT:
391 NativeEvent event = (NativeEvent) msg.obj;
392 handleNativeEvent(event.code, event.raw,
393 NativeDaemonEvent.unescapeArgs(event.raw));
394 break;
Irfan Sheriff75006652012-04-17 23:15:29 -0700395 default:
396 result = NOT_HANDLED;
397 break;
398 }
399 return result;
400 }
401 }
402 }
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700403
404 private NativeDaemonConnector mNativeConnector;
405 private final CountDownLatch mNativeDaemonConnected = new CountDownLatch(1);
406
407 private NsdService(Context context) {
408 mContext = context;
Irfan Sheriff75006652012-04-17 23:15:29 -0700409 mContentResolver = context.getContentResolver();
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700410
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700411 mNativeConnector = new NativeDaemonConnector(new NativeCallbackReceiver(), "mdns", 10,
412 MDNS_TAG, 25);
Irfan Sheriff75006652012-04-17 23:15:29 -0700413
414 mNsdStateMachine = new NsdStateMachine(TAG);
415 mNsdStateMachine.start();
416
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700417 Thread th = new Thread(mNativeConnector, MDNS_TAG);
418 th.start();
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700419 }
420
421 public static NsdService create(Context context) throws InterruptedException {
422 NsdService service = new NsdService(context);
Irfan Sheriff75006652012-04-17 23:15:29 -0700423 service.mNativeDaemonConnected.await();
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700424 return service;
425 }
426
427 public Messenger getMessenger() {
Irfan Sheriff18da1322012-04-13 12:15:41 -0700428 mContext.enforceCallingOrSelfPermission(android.Manifest.permission.INTERNET,
429 "NsdService");
Irfan Sheriff75006652012-04-17 23:15:29 -0700430 return new Messenger(mNsdStateMachine.getHandler());
431 }
432
433 public void setEnabled(boolean enable) {
434 mContext.enforceCallingOrSelfPermission(android.Manifest.permission.CONNECTIVITY_INTERNAL,
435 "NsdService");
436 Settings.Secure.putInt(mContentResolver, Settings.Secure.NSD_ON, enable ? 1 : 0);
437 if (enable) {
438 mNsdStateMachine.sendMessage(NsdManager.ENABLE);
439 } else {
440 mNsdStateMachine.sendMessage(NsdManager.DISABLE);
441 }
442 }
443
444 private void sendNsdStateChangeBroadcast(boolean enabled) {
Irfan Sheriff52fc83a2012-04-19 10:26:34 -0700445 final Intent intent = new Intent(NsdManager.ACTION_NSD_STATE_CHANGED);
Irfan Sheriff75006652012-04-17 23:15:29 -0700446 intent.addFlags(Intent.FLAG_RECEIVER_REGISTERED_ONLY_BEFORE_BOOT);
447 if (enabled) {
448 intent.putExtra(NsdManager.EXTRA_NSD_STATE, NsdManager.NSD_STATE_ENABLED);
449 } else {
450 intent.putExtra(NsdManager.EXTRA_NSD_STATE, NsdManager.NSD_STATE_DISABLED);
451 }
Dianne Hackborn692107e2012-08-29 18:32:08 -0700452 mContext.sendStickyBroadcastAsUser(intent, UserHandle.ALL);
Irfan Sheriff75006652012-04-17 23:15:29 -0700453 }
454
455 private boolean isNsdEnabled() {
456 boolean ret = Settings.Secure.getInt(mContentResolver, Settings.Secure.NSD_ON, 1) == 1;
457 if (DBG) Slog.d(TAG, "Network service discovery enabled " + ret);
458 return ret;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700459 }
460
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700461 private int getUniqueId() {
462 if (++mUniqueId == INVALID_ID) return ++mUniqueId;
463 return mUniqueId;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700464 }
465
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700466 /* These should be in sync with system/netd/mDnsResponseCode.h */
467 class NativeResponseCode {
468 public static final int SERVICE_DISCOVERY_FAILED = 602;
469 public static final int SERVICE_FOUND = 603;
470 public static final int SERVICE_LOST = 604;
471
472 public static final int SERVICE_REGISTRATION_FAILED = 605;
473 public static final int SERVICE_REGISTERED = 606;
474
475 public static final int SERVICE_RESOLUTION_FAILED = 607;
476 public static final int SERVICE_RESOLVED = 608;
477
478 public static final int SERVICE_UPDATED = 609;
479 public static final int SERVICE_UPDATE_FAILED = 610;
480
481 public static final int SERVICE_GET_ADDR_FAILED = 611;
482 public static final int SERVICE_GET_ADDR_SUCCESS = 612;
483 }
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700484
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700485 private class NativeEvent {
486 int code;
487 String raw;
488
489 NativeEvent(int code, String raw) {
490 this.code = code;
491 this.raw = raw;
492 }
493 }
494
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700495 class NativeCallbackReceiver implements INativeDaemonConnectorCallbacks {
496 public void onDaemonConnected() {
497 mNativeDaemonConnected.countDown();
498 }
499
500 public boolean onEvent(int code, String raw, String[] cooked) {
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700501 // TODO: NDC translates a message to a callback, we could enhance NDC to
502 // directly interact with a state machine through messages
503 NativeEvent event = new NativeEvent(code, raw);
504 mNsdStateMachine.sendMessage(NsdManager.NATIVE_DAEMON_EVENT, event);
505 return true;
506 }
507 }
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700508
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700509 private void handleNativeEvent(int code, String raw, String[] cooked) {
510 NsdServiceInfo servInfo;
511 int id = Integer.parseInt(cooked[1]);
512 ClientInfo clientInfo = mIdToClientInfoMap.get(id);
513 if (clientInfo == null) {
514 Slog.e(TAG, "Unique id with no client mapping: " + id);
515 return;
516 }
517
518 /* This goes in response as msg.arg2 */
519 int clientId = -1;
520 int keyId = clientInfo.mClientIds.indexOfValue(id);
521 if (keyId != -1) {
522 clientId = clientInfo.mClientIds.keyAt(keyId);
523 }
524 switch (code) {
525 case NativeResponseCode.SERVICE_FOUND:
526 /* NNN uniqueId serviceName regType domain */
527 if (DBG) Slog.d(TAG, "SERVICE_FOUND Raw: " + raw);
528 servInfo = new NsdServiceInfo(cooked[2], cooked[3], null);
529 clientInfo.mChannel.sendMessage(NsdManager.SERVICE_FOUND, 0,
530 clientId, servInfo);
531 break;
532 case NativeResponseCode.SERVICE_LOST:
533 /* NNN uniqueId serviceName regType domain */
534 if (DBG) Slog.d(TAG, "SERVICE_LOST Raw: " + raw);
535 servInfo = new NsdServiceInfo(cooked[2], cooked[3], null);
536 clientInfo.mChannel.sendMessage(NsdManager.SERVICE_LOST, 0,
537 clientId, servInfo);
538 break;
539 case NativeResponseCode.SERVICE_DISCOVERY_FAILED:
540 /* NNN uniqueId errorCode */
541 if (DBG) Slog.d(TAG, "SERVICE_DISC_FAILED Raw: " + raw);
542 clientInfo.mChannel.sendMessage(NsdManager.DISCOVER_SERVICES_FAILED,
543 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
544 break;
545 case NativeResponseCode.SERVICE_REGISTERED:
546 /* NNN regId serviceName regType */
547 if (DBG) Slog.d(TAG, "SERVICE_REGISTERED Raw: " + raw);
548 servInfo = new NsdServiceInfo(cooked[2], null, null);
549 clientInfo.mChannel.sendMessage(NsdManager.REGISTER_SERVICE_SUCCEEDED,
550 id, clientId, servInfo);
551 break;
552 case NativeResponseCode.SERVICE_REGISTRATION_FAILED:
553 /* NNN regId errorCode */
554 if (DBG) Slog.d(TAG, "SERVICE_REGISTER_FAILED Raw: " + raw);
555 clientInfo.mChannel.sendMessage(NsdManager.REGISTER_SERVICE_FAILED,
556 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
557 break;
558 case NativeResponseCode.SERVICE_UPDATED:
559 /* NNN regId */
560 break;
561 case NativeResponseCode.SERVICE_UPDATE_FAILED:
562 /* NNN regId errorCode */
563 break;
564 case NativeResponseCode.SERVICE_RESOLVED:
565 /* NNN resolveId fullName hostName port txtlen txtdata */
566 if (DBG) Slog.d(TAG, "SERVICE_RESOLVED Raw: " + raw);
567 int index = cooked[2].indexOf(".");
568 if (index == -1) {
569 Slog.e(TAG, "Invalid service found " + raw);
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700570 break;
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700571 }
572 String name = cooked[2].substring(0, index);
573 String rest = cooked[2].substring(index);
574 String type = rest.replace(".local.", "");
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700575
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700576 clientInfo.mResolvedService.setServiceName(name);
577 clientInfo.mResolvedService.setServiceType(type);
578 clientInfo.mResolvedService.setPort(Integer.parseInt(cooked[4]));
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700579
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700580 stopResolveService(id);
581 if (!getAddrInfo(id, cooked[3])) {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700582 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_FAILED,
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700583 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
584 mIdToClientInfoMap.remove(id);
585 clientInfo.mResolvedService = null;
586 }
587 break;
588 case NativeResponseCode.SERVICE_RESOLUTION_FAILED:
589 /* NNN resolveId errorCode */
590 if (DBG) Slog.d(TAG, "SERVICE_RESOLVE_FAILED Raw: " + raw);
591 stopResolveService(id);
592 mIdToClientInfoMap.remove(id);
593 clientInfo.mResolvedService = null;
594 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_FAILED,
595 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
596 break;
597 case NativeResponseCode.SERVICE_GET_ADDR_FAILED:
598 /* NNN resolveId errorCode */
599 stopGetAddrInfo(id);
600 mIdToClientInfoMap.remove(id);
601 clientInfo.mResolvedService = null;
602 if (DBG) Slog.d(TAG, "SERVICE_RESOLVE_FAILED Raw: " + raw);
603 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_FAILED,
604 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
605 break;
606 case NativeResponseCode.SERVICE_GET_ADDR_SUCCESS:
607 /* NNN resolveId hostname ttl addr */
608 if (DBG) Slog.d(TAG, "SERVICE_GET_ADDR_SUCCESS Raw: " + raw);
609 try {
610 clientInfo.mResolvedService.setHost(InetAddress.getByName(cooked[4]));
611 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_SUCCEEDED,
612 0, clientId, clientInfo.mResolvedService);
613 } catch (java.net.UnknownHostException e) {
614 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_FAILED,
615 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
616 }
617 stopGetAddrInfo(id);
618 mIdToClientInfoMap.remove(id);
619 clientInfo.mResolvedService = null;
620 break;
621 default:
622 break;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700623 }
624 }
625
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700626 private boolean startMDnsDaemon() {
627 if (DBG) Slog.d(TAG, "startMDnsDaemon");
628 try {
629 mNativeConnector.execute("mdnssd", "start-service");
630 } catch(NativeDaemonConnectorException e) {
631 Slog.e(TAG, "Failed to start daemon" + e);
632 return false;
633 }
634 return true;
635 }
636
637 private boolean stopMDnsDaemon() {
638 if (DBG) Slog.d(TAG, "stopMDnsDaemon");
639 try {
640 mNativeConnector.execute("mdnssd", "stop-service");
641 } catch(NativeDaemonConnectorException e) {
642 Slog.e(TAG, "Failed to start daemon" + e);
643 return false;
644 }
645 return true;
646 }
647
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700648 private boolean registerService(int regId, NsdServiceInfo service) {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700649 if (DBG) Slog.d(TAG, "registerService: " + regId + " " + service);
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700650 try {
651 //Add txtlen and txtdata
652 mNativeConnector.execute("mdnssd", "register", regId, service.getServiceName(),
653 service.getServiceType(), service.getPort());
654 } catch(NativeDaemonConnectorException e) {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700655 Slog.e(TAG, "Failed to execute registerService " + e);
656 return false;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700657 }
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700658 return true;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700659 }
660
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700661 private boolean unregisterService(int regId) {
662 if (DBG) Slog.d(TAG, "unregisterService: " + regId);
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700663 try {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700664 mNativeConnector.execute("mdnssd", "stop-register", regId);
665 } catch(NativeDaemonConnectorException e) {
666 Slog.e(TAG, "Failed to execute unregisterService " + e);
667 return false;
668 }
669 return true;
670 }
671
672 private boolean updateService(int regId, DnsSdTxtRecord t) {
673 if (DBG) Slog.d(TAG, "updateService: " + regId + " " + t);
674 try {
675 if (t == null) return false;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700676 mNativeConnector.execute("mdnssd", "update", regId, t.size(), t.getRawData());
677 } catch(NativeDaemonConnectorException e) {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700678 Slog.e(TAG, "Failed to updateServices " + e);
679 return false;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700680 }
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700681 return true;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700682 }
683
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700684 private boolean discoverServices(int discoveryId, String serviceType) {
685 if (DBG) Slog.d(TAG, "discoverServices: " + discoveryId + " " + serviceType);
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700686 try {
687 mNativeConnector.execute("mdnssd", "discover", discoveryId, serviceType);
688 } catch(NativeDaemonConnectorException e) {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700689 Slog.e(TAG, "Failed to discoverServices " + e);
690 return false;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700691 }
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700692 return true;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700693 }
694
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700695 private boolean stopServiceDiscovery(int discoveryId) {
696 if (DBG) Slog.d(TAG, "stopServiceDiscovery: " + discoveryId);
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700697 try {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700698 mNativeConnector.execute("mdnssd", "stop-discover", discoveryId);
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700699 } catch(NativeDaemonConnectorException e) {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700700 Slog.e(TAG, "Failed to stopServiceDiscovery " + e);
701 return false;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700702 }
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700703 return true;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700704 }
705
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700706 private boolean resolveService(int resolveId, NsdServiceInfo service) {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700707 if (DBG) Slog.d(TAG, "resolveService: " + resolveId + " " + service);
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700708 try {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700709 mNativeConnector.execute("mdnssd", "resolve", resolveId, service.getServiceName(),
710 service.getServiceType(), "local.");
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700711 } catch(NativeDaemonConnectorException e) {
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700712 Slog.e(TAG, "Failed to resolveService " + e);
713 return false;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700714 }
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700715 return true;
716 }
717
718 private boolean stopResolveService(int resolveId) {
719 if (DBG) Slog.d(TAG, "stopResolveService: " + resolveId);
720 try {
721 mNativeConnector.execute("mdnssd", "stop-resolve", resolveId);
722 } catch(NativeDaemonConnectorException e) {
723 Slog.e(TAG, "Failed to stop resolve " + e);
724 return false;
725 }
726 return true;
727 }
728
729 private boolean getAddrInfo(int resolveId, String hostname) {
730 if (DBG) Slog.d(TAG, "getAdddrInfo: " + resolveId);
731 try {
732 mNativeConnector.execute("mdnssd", "getaddrinfo", resolveId, hostname);
733 } catch(NativeDaemonConnectorException e) {
734 Slog.e(TAG, "Failed to getAddrInfo " + e);
735 return false;
736 }
737 return true;
738 }
739
740 private boolean stopGetAddrInfo(int resolveId) {
741 if (DBG) Slog.d(TAG, "stopGetAdddrInfo: " + resolveId);
742 try {
743 mNativeConnector.execute("mdnssd", "stop-getaddrinfo", resolveId);
744 } catch(NativeDaemonConnectorException e) {
745 Slog.e(TAG, "Failed to stopGetAddrInfo " + e);
746 return false;
747 }
748 return true;
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700749 }
750
751 @Override
Irfan Sheriff75006652012-04-17 23:15:29 -0700752 public void dump(FileDescriptor fd, PrintWriter pw, String[] args) {
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700753 if (mContext.checkCallingOrSelfPermission(android.Manifest.permission.DUMP)
754 != PackageManager.PERMISSION_GRANTED) {
755 pw.println("Permission Denial: can't dump ServiceDiscoverService from from pid="
756 + Binder.getCallingPid()
757 + ", uid=" + Binder.getCallingUid());
758 return;
759 }
760
Irfan Sheriff75006652012-04-17 23:15:29 -0700761 for (ClientInfo client : mClients.values()) {
762 pw.println("Client Info");
763 pw.println(client);
764 }
765
766 mNsdStateMachine.dump(fd, pw, args);
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700767 }
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700768
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700769 /* arg2 on the source message has an id that needs to be retained in replies
770 * see NsdManager for details */
771 private Message obtainMessage(Message srcMsg) {
772 Message msg = Message.obtain();
773 msg.arg2 = srcMsg.arg2;
774 return msg;
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700775 }
776
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700777 private void replyToMessage(Message msg, int what) {
778 if (msg.replyTo == null) return;
779 Message dstMsg = obtainMessage(msg);
780 dstMsg.what = what;
781 mReplyChannel.replyToMessage(msg, dstMsg);
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700782 }
783
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700784 private void replyToMessage(Message msg, int what, int arg1) {
785 if (msg.replyTo == null) return;
786 Message dstMsg = obtainMessage(msg);
787 dstMsg.what = what;
788 dstMsg.arg1 = arg1;
789 mReplyChannel.replyToMessage(msg, dstMsg);
790 }
791
792 private void replyToMessage(Message msg, int what, Object obj) {
793 if (msg.replyTo == null) return;
794 Message dstMsg = obtainMessage(msg);
795 dstMsg.what = what;
796 dstMsg.obj = obj;
797 mReplyChannel.replyToMessage(msg, dstMsg);
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700798 }
799
800 /* Information tracked per client */
801 private class ClientInfo {
802
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700803 private static final int MAX_LIMIT = 10;
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700804 private AsyncChannel mChannel;
805 private Messenger mMessenger;
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700806 /* Remembers a resolved service until getaddrinfo completes */
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700807 private NsdServiceInfo mResolvedService;
808
809 /* A map from client id to unique id sent to mDns */
810 private SparseArray<Integer> mClientIds = new SparseArray<Integer>();
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700811
812 private ClientInfo(AsyncChannel c, Messenger m) {
813 mChannel = c;
814 mMessenger = m;
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700815 if (DBG) Slog.d(TAG, "New client, channel: " + c + " messenger: " + m);
816 }
Irfan Sheriff75006652012-04-17 23:15:29 -0700817
818 @Override
819 public String toString() {
820 StringBuffer sb = new StringBuffer();
821 sb.append("mChannel ").append(mChannel).append("\n");
822 sb.append("mMessenger ").append(mMessenger).append("\n");
Irfan Sheriff75006652012-04-17 23:15:29 -0700823 sb.append("mResolvedService ").append(mResolvedService).append("\n");
Irfan Sheriffe4c42f42012-05-03 16:44:27 -0700824 for(int i = 0; i< mClientIds.size(); i++) {
825 sb.append("clientId ").append(mClientIds.keyAt(i));
826 sb.append(" mDnsId ").append(mClientIds.valueAt(i)).append("\n");
Irfan Sheriff75006652012-04-17 23:15:29 -0700827 }
828 return sb.toString();
829 }
Irfan Sheriffe8de2462012-04-11 14:52:19 -0700830 }
Irfan Sheriff77ec5582012-03-22 17:01:39 -0700831}