blob: 3c1f462f9378a3d15ebadd5eaf0eaf4f539ad6c4 [file] [log] [blame]
Irfan Sheriff7d024d32012-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 Sheriff3ef889b2012-04-17 23:15:29 -070020import android.content.ContentResolver;
21import android.content.Intent;
Irfan Sheriff7d024d32012-03-22 17:01:39 -070022import android.content.pm.PackageManager;
Irfan Sheriff919aca52012-06-01 16:44:13 -070023import android.database.ContentObserver;
Irfan Sheriff22af38c2012-05-03 16:44:27 -070024import android.net.nsd.NsdServiceInfo;
Irfan Sheriff7d024d32012-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;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -070034import android.provider.Settings;
Irfan Sheriff7d024d32012-03-22 17:01:39 -070035import android.util.Slog;
Irfan Sheriff22af38c2012-05-03 16:44:27 -070036import android.util.SparseArray;
Irfan Sheriff7d024d32012-03-22 17:01:39 -070037
38import java.io.FileDescriptor;
39import java.io.PrintWriter;
Irfan Sheriff817388e2012-04-11 14:52:19 -070040import java.net.InetAddress;
Irfan Sheriff7d024d32012-03-22 17:01:39 -070041import java.util.ArrayList;
Irfan Sheriff817388e2012-04-11 14:52:19 -070042import java.util.HashMap;
Irfan Sheriff7d024d32012-03-22 17:01:39 -070043import java.util.List;
Irfan Sheriff817388e2012-04-11 14:52:19 -070044import java.util.concurrent.CountDownLatch;
Irfan Sheriff7d024d32012-03-22 17:01:39 -070045
46import com.android.internal.app.IBatteryStats;
47import com.android.internal.telephony.TelephonyIntents;
48import com.android.internal.util.AsyncChannel;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -070049import com.android.internal.util.Protocol;
50import com.android.internal.util.State;
51import com.android.internal.util.StateMachine;
Irfan Sheriff7d024d32012-03-22 17:01:39 -070052import com.android.server.am.BatteryStatsService;
53import com.android.server.NativeDaemonConnector.Command;
54import com.android.internal.R;
55
56/**
57 * Network Service Discovery Service handles remote service discovery operation requests by
58 * implementing the INsdManager interface.
59 *
60 * @hide
61 */
62public class NsdService extends INsdManager.Stub {
63 private static final String TAG = "NsdService";
64 private static final String MDNS_TAG = "mDnsConnector";
65
66 private static final boolean DBG = true;
67
68 private Context mContext;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -070069 private ContentResolver mContentResolver;
70 private NsdStateMachine mNsdStateMachine;
Irfan Sheriff7d024d32012-03-22 17:01:39 -070071
72 /**
73 * Clients receiving asynchronous messages
74 */
Irfan Sheriff817388e2012-04-11 14:52:19 -070075 private HashMap<Messenger, ClientInfo> mClients = new HashMap<Messenger, ClientInfo>();
Irfan Sheriff7d024d32012-03-22 17:01:39 -070076
Irfan Sheriff22af38c2012-05-03 16:44:27 -070077 /* A map from unique id to client info */
78 private SparseArray<ClientInfo> mIdToClientInfoMap= new SparseArray<ClientInfo>();
79
Irfan Sheriff7d024d32012-03-22 17:01:39 -070080 private AsyncChannel mReplyChannel = new AsyncChannel();
81
Irfan Sheriff817388e2012-04-11 14:52:19 -070082 private int INVALID_ID = 0;
83 private int mUniqueId = 1;
84
Irfan Sheriff3ef889b2012-04-17 23:15:29 -070085 private static final int BASE = Protocol.BASE_NSD_MANAGER;
Irfan Sheriff22af38c2012-05-03 16:44:27 -070086 private static final int CMD_TO_STRING_COUNT = NsdManager.RESOLVE_SERVICE - BASE + 1;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -070087 private static String[] sCmdToString = new String[CMD_TO_STRING_COUNT];
Irfan Sheriff7d024d32012-03-22 17:01:39 -070088
Irfan Sheriff3ef889b2012-04-17 23:15:29 -070089 static {
90 sCmdToString[NsdManager.DISCOVER_SERVICES - BASE] = "DISCOVER";
91 sCmdToString[NsdManager.STOP_DISCOVERY - BASE] = "STOP-DISCOVER";
92 sCmdToString[NsdManager.REGISTER_SERVICE - BASE] = "REGISTER";
93 sCmdToString[NsdManager.UNREGISTER_SERVICE - BASE] = "UNREGISTER";
94 sCmdToString[NsdManager.RESOLVE_SERVICE - BASE] = "RESOLVE";
Irfan Sheriff3ef889b2012-04-17 23:15:29 -070095 }
Irfan Sheriff7d024d32012-03-22 17:01:39 -070096
Irfan Sheriff3ef889b2012-04-17 23:15:29 -070097 private static String cmdToString(int cmd) {
98 cmd -= BASE;
99 if ((cmd >= 0) && (cmd < sCmdToString.length)) {
100 return sCmdToString[cmd];
101 } else {
102 return null;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700103 }
104 }
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700105
106 private class NsdStateMachine extends StateMachine {
107
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700108 private final DefaultState mDefaultState = new DefaultState();
109 private final DisabledState mDisabledState = new DisabledState();
110 private final EnabledState mEnabledState = new EnabledState();
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700111
112 @Override
Wink Saville095c58b2012-05-29 12:40:46 -0700113 protected String getWhatToString(int what) {
114 return cmdToString(what);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700115 }
116
Irfan Sheriff919aca52012-06-01 16:44:13 -0700117 /**
118 * Observes the NSD on/off setting, and takes action when changed.
119 */
120 private void registerForNsdSetting() {
121 ContentObserver contentObserver = new ContentObserver(this.getHandler()) {
122 @Override
123 public void onChange(boolean selfChange) {
124 if (isNsdEnabled()) {
125 mNsdStateMachine.sendMessage(NsdManager.ENABLE);
126 } else {
127 mNsdStateMachine.sendMessage(NsdManager.DISABLE);
128 }
129 }
130 };
131
132 mContext.getContentResolver().registerContentObserver(
133 Settings.Secure.getUriFor(Settings.Secure.NSD_ON),
134 false, contentObserver);
135 }
136
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700137 NsdStateMachine(String name) {
138 super(name);
139 addState(mDefaultState);
140 addState(mDisabledState, mDefaultState);
141 addState(mEnabledState, mDefaultState);
142 if (isNsdEnabled()) {
143 setInitialState(mEnabledState);
144 } else {
145 setInitialState(mDisabledState);
146 }
Wink Saville095c58b2012-05-29 12:40:46 -0700147 setLogRecSize(25);
Irfan Sheriff919aca52012-06-01 16:44:13 -0700148 registerForNsdSetting();
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700149 }
150
151 class DefaultState extends State {
152 @Override
153 public boolean processMessage(Message msg) {
154 switch (msg.what) {
155 case AsyncChannel.CMD_CHANNEL_HALF_CONNECTED:
156 if (msg.arg1 == AsyncChannel.STATUS_SUCCESSFUL) {
157 AsyncChannel c = (AsyncChannel) msg.obj;
158 if (DBG) Slog.d(TAG, "New client listening to asynchronous messages");
159 c.sendMessage(AsyncChannel.CMD_CHANNEL_FULLY_CONNECTED);
160 ClientInfo cInfo = new ClientInfo(c, msg.replyTo);
161 mClients.put(msg.replyTo, cInfo);
162 } else {
163 Slog.e(TAG, "Client connection failure, error=" + msg.arg1);
164 }
165 break;
166 case AsyncChannel.CMD_CHANNEL_DISCONNECTED:
167 if (msg.arg1 == AsyncChannel.STATUS_SEND_UNSUCCESSFUL) {
168 Slog.e(TAG, "Send failed, client connection lost");
169 } else {
170 if (DBG) Slog.d(TAG, "Client connection lost with reason: " + msg.arg1);
171 }
172 mClients.remove(msg.replyTo);
173 break;
174 case AsyncChannel.CMD_CHANNEL_FULL_CONNECTION:
175 AsyncChannel ac = new AsyncChannel();
176 ac.connect(mContext, getHandler(), msg.replyTo);
177 break;
178 case NsdManager.DISCOVER_SERVICES:
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700179 replyToMessage(msg, NsdManager.DISCOVER_SERVICES_FAILED,
180 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700181 break;
182 case NsdManager.STOP_DISCOVERY:
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700183 replyToMessage(msg, NsdManager.STOP_DISCOVERY_FAILED,
184 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700185 break;
186 case NsdManager.REGISTER_SERVICE:
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700187 replyToMessage(msg, NsdManager.REGISTER_SERVICE_FAILED,
188 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700189 break;
190 case NsdManager.UNREGISTER_SERVICE:
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700191 replyToMessage(msg, NsdManager.UNREGISTER_SERVICE_FAILED,
192 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700193 break;
194 case NsdManager.RESOLVE_SERVICE:
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700195 replyToMessage(msg, NsdManager.RESOLVE_SERVICE_FAILED,
196 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700197 break;
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700198 case NsdManager.NATIVE_DAEMON_EVENT:
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700199 default:
200 Slog.e(TAG, "Unhandled " + msg);
201 return NOT_HANDLED;
202 }
203 return HANDLED;
204 }
205 }
206
207 class DisabledState extends State {
208 @Override
209 public void enter() {
210 sendNsdStateChangeBroadcast(false);
211 }
212
213 @Override
214 public boolean processMessage(Message msg) {
215 switch (msg.what) {
216 case NsdManager.ENABLE:
217 transitionTo(mEnabledState);
218 break;
219 default:
220 return NOT_HANDLED;
221 }
222 return HANDLED;
223 }
224 }
225
226 class EnabledState extends State {
227 @Override
228 public void enter() {
229 sendNsdStateChangeBroadcast(true);
230 if (mClients.size() > 0) {
231 startMDnsDaemon();
232 }
233 }
234
235 @Override
236 public void exit() {
237 if (mClients.size() > 0) {
238 stopMDnsDaemon();
239 }
240 }
241
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700242 private boolean requestLimitReached(ClientInfo clientInfo) {
243 if (clientInfo.mClientIds.size() >= ClientInfo.MAX_LIMIT) {
244 if (DBG) Slog.d(TAG, "Exceeded max outstanding requests " + clientInfo);
245 return true;
246 }
247 return false;
248 }
249
250 private void storeRequestMap(int clientId, int globalId, ClientInfo clientInfo) {
251 clientInfo.mClientIds.put(clientId, globalId);
252 mIdToClientInfoMap.put(globalId, clientInfo);
253 }
254
255 private void removeRequestMap(int clientId, int globalId, ClientInfo clientInfo) {
256 clientInfo.mClientIds.remove(clientId);
257 mIdToClientInfoMap.remove(globalId);
258 }
259
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700260 @Override
261 public boolean processMessage(Message msg) {
262 ClientInfo clientInfo;
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700263 NsdServiceInfo servInfo;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700264 boolean result = HANDLED;
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700265 int id;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700266 switch (msg.what) {
267 case AsyncChannel.CMD_CHANNEL_HALF_CONNECTED:
268 //First client
269 if (msg.arg1 == AsyncChannel.STATUS_SUCCESSFUL &&
270 mClients.size() == 0) {
271 startMDnsDaemon();
272 }
273 result = NOT_HANDLED;
274 break;
275 case AsyncChannel.CMD_CHANNEL_DISCONNECTED:
276 //Last client
277 if (mClients.size() == 1) {
278 stopMDnsDaemon();
279 }
280 result = NOT_HANDLED;
281 break;
282 case NsdManager.DISABLE:
283 //TODO: cleanup clients
284 transitionTo(mDisabledState);
285 break;
286 case NsdManager.DISCOVER_SERVICES:
287 if (DBG) Slog.d(TAG, "Discover services");
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700288 servInfo = (NsdServiceInfo) msg.obj;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700289 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700290
291 if (requestLimitReached(clientInfo)) {
292 replyToMessage(msg, NsdManager.DISCOVER_SERVICES_FAILED,
293 NsdManager.FAILURE_MAX_LIMIT);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700294 break;
295 }
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700296
297 id = getUniqueId();
298 if (discoverServices(id, servInfo.getServiceType())) {
299 if (DBG) {
300 Slog.d(TAG, "Discover " + msg.arg2 + " " + id +
301 servInfo.getServiceType());
302 }
303 storeRequestMap(msg.arg2, id, clientInfo);
304 replyToMessage(msg, NsdManager.DISCOVER_SERVICES_STARTED, servInfo);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700305 } else {
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700306 stopServiceDiscovery(id);
307 replyToMessage(msg, NsdManager.DISCOVER_SERVICES_FAILED,
308 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700309 }
310 break;
311 case NsdManager.STOP_DISCOVERY:
312 if (DBG) Slog.d(TAG, "Stop service discovery");
313 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700314
315 try {
316 id = clientInfo.mClientIds.get(msg.arg2).intValue();
317 } catch (NullPointerException e) {
318 replyToMessage(msg, NsdManager.STOP_DISCOVERY_FAILED,
319 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700320 break;
321 }
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700322 removeRequestMap(msg.arg2, id, clientInfo);
323 if (stopServiceDiscovery(id)) {
324 replyToMessage(msg, NsdManager.STOP_DISCOVERY_SUCCEEDED);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700325 } else {
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700326 replyToMessage(msg, NsdManager.STOP_DISCOVERY_FAILED,
327 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700328 }
329 break;
330 case NsdManager.REGISTER_SERVICE:
331 if (DBG) Slog.d(TAG, "Register service");
332 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700333 if (requestLimitReached(clientInfo)) {
334 replyToMessage(msg, NsdManager.REGISTER_SERVICE_FAILED,
335 NsdManager.FAILURE_MAX_LIMIT);
336 break;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700337 }
338
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700339 id = getUniqueId();
340 if (registerService(id, (NsdServiceInfo) msg.obj)) {
341 if (DBG) Slog.d(TAG, "Register " + msg.arg2 + " " + id);
342 storeRequestMap(msg.arg2, id, clientInfo);
343 // Return success after mDns reports success
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700344 } else {
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700345 unregisterService(id);
346 replyToMessage(msg, NsdManager.REGISTER_SERVICE_FAILED,
347 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700348 }
349 break;
350 case NsdManager.UNREGISTER_SERVICE:
351 if (DBG) Slog.d(TAG, "unregister service");
352 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700353 try {
354 id = clientInfo.mClientIds.get(msg.arg2).intValue();
355 } catch (NullPointerException e) {
356 replyToMessage(msg, NsdManager.UNREGISTER_SERVICE_FAILED,
357 NsdManager.FAILURE_INTERNAL_ERROR);
358 break;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700359 }
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700360 removeRequestMap(msg.arg2, id, clientInfo);
361 if (unregisterService(id)) {
362 replyToMessage(msg, NsdManager.UNREGISTER_SERVICE_SUCCEEDED);
363 } else {
364 replyToMessage(msg, NsdManager.UNREGISTER_SERVICE_FAILED,
365 NsdManager.FAILURE_INTERNAL_ERROR);
366 }
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700367 break;
368 case NsdManager.RESOLVE_SERVICE:
369 if (DBG) Slog.d(TAG, "Resolve service");
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700370 servInfo = (NsdServiceInfo) msg.obj;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700371 clientInfo = mClients.get(msg.replyTo);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700372
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700373
374 if (clientInfo.mResolvedService != null) {
375 replyToMessage(msg, NsdManager.RESOLVE_SERVICE_FAILED,
376 NsdManager.FAILURE_ALREADY_ACTIVE);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700377 break;
378 }
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700379
380 id = getUniqueId();
381 if (resolveService(id, servInfo)) {
382 clientInfo.mResolvedService = new NsdServiceInfo();
383 storeRequestMap(msg.arg2, id, clientInfo);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700384 } else {
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700385 replyToMessage(msg, NsdManager.RESOLVE_SERVICE_FAILED,
386 NsdManager.FAILURE_INTERNAL_ERROR);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700387 }
388 break;
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700389 case NsdManager.NATIVE_DAEMON_EVENT:
390 NativeEvent event = (NativeEvent) msg.obj;
Vairavan Srinivasan6727e732012-08-05 13:14:12 -0700391 if (!handleNativeEvent(event.code, event.raw,
392 NativeDaemonEvent.unescapeArgs(event.raw))) {
393 result = NOT_HANDLED;
394 }
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700395 break;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700396 default:
397 result = NOT_HANDLED;
398 break;
399 }
400 return result;
401 }
Vairavan Srinivasan6727e732012-08-05 13:14:12 -0700402
403 private boolean handleNativeEvent(int code, String raw, String[] cooked) {
404 boolean handled = true;
405 NsdServiceInfo servInfo;
406 int id = Integer.parseInt(cooked[1]);
407 ClientInfo clientInfo = mIdToClientInfoMap.get(id);
408 if (clientInfo == null) {
409 Slog.e(TAG, "Unique id with no client mapping: " + id);
410 handled = false;
411 return handled;
412 }
413
414 /* This goes in response as msg.arg2 */
415 int clientId = -1;
416 int keyId = clientInfo.mClientIds.indexOfValue(id);
417 if (keyId != -1) {
418 clientId = clientInfo.mClientIds.keyAt(keyId);
419 }
420 switch (code) {
421 case NativeResponseCode.SERVICE_FOUND:
422 /* NNN uniqueId serviceName regType domain */
423 if (DBG) Slog.d(TAG, "SERVICE_FOUND Raw: " + raw);
424 servInfo = new NsdServiceInfo(cooked[2], cooked[3], null);
425 clientInfo.mChannel.sendMessage(NsdManager.SERVICE_FOUND, 0,
426 clientId, servInfo);
427 break;
428 case NativeResponseCode.SERVICE_LOST:
429 /* NNN uniqueId serviceName regType domain */
430 if (DBG) Slog.d(TAG, "SERVICE_LOST Raw: " + raw);
431 servInfo = new NsdServiceInfo(cooked[2], cooked[3], null);
432 clientInfo.mChannel.sendMessage(NsdManager.SERVICE_LOST, 0,
433 clientId, servInfo);
434 break;
435 case NativeResponseCode.SERVICE_DISCOVERY_FAILED:
436 /* NNN uniqueId errorCode */
437 if (DBG) Slog.d(TAG, "SERVICE_DISC_FAILED Raw: " + raw);
438 clientInfo.mChannel.sendMessage(NsdManager.DISCOVER_SERVICES_FAILED,
439 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
440 break;
441 case NativeResponseCode.SERVICE_REGISTERED:
442 /* NNN regId serviceName regType */
443 if (DBG) Slog.d(TAG, "SERVICE_REGISTERED Raw: " + raw);
444 servInfo = new NsdServiceInfo(cooked[2], null, null);
445 clientInfo.mChannel.sendMessage(NsdManager.REGISTER_SERVICE_SUCCEEDED,
446 id, clientId, servInfo);
447 break;
448 case NativeResponseCode.SERVICE_REGISTRATION_FAILED:
449 /* NNN regId errorCode */
450 if (DBG) Slog.d(TAG, "SERVICE_REGISTER_FAILED Raw: " + raw);
451 clientInfo.mChannel.sendMessage(NsdManager.REGISTER_SERVICE_FAILED,
452 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
453 break;
454 case NativeResponseCode.SERVICE_UPDATED:
455 /* NNN regId */
456 break;
457 case NativeResponseCode.SERVICE_UPDATE_FAILED:
458 /* NNN regId errorCode */
459 break;
460 case NativeResponseCode.SERVICE_RESOLVED:
461 /* NNN resolveId fullName hostName port txtlen txtdata */
462 if (DBG) Slog.d(TAG, "SERVICE_RESOLVED Raw: " + raw);
463 int index = cooked[2].indexOf(".");
464 if (index == -1) {
465 Slog.e(TAG, "Invalid service found " + raw);
466 break;
467 }
468 String name = cooked[2].substring(0, index);
469 String rest = cooked[2].substring(index);
470 String type = rest.replace(".local.", "");
471
472 clientInfo.mResolvedService.setServiceName(name);
473 clientInfo.mResolvedService.setServiceType(type);
474 clientInfo.mResolvedService.setPort(Integer.parseInt(cooked[4]));
475
476 stopResolveService(id);
477 if (!getAddrInfo(id, cooked[3])) {
478 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_FAILED,
479 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
480 removeRequestMap(clientId, id, clientInfo);
481 clientInfo.mResolvedService = null;
482 }
483 break;
484 case NativeResponseCode.SERVICE_RESOLUTION_FAILED:
485 /* NNN resolveId errorCode */
486 if (DBG) Slog.d(TAG, "SERVICE_RESOLVE_FAILED Raw: " + raw);
487 stopResolveService(id);
488 removeRequestMap(clientId, id, clientInfo);
489 clientInfo.mResolvedService = null;
490 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_FAILED,
491 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
492 break;
493 case NativeResponseCode.SERVICE_GET_ADDR_FAILED:
494 /* NNN resolveId errorCode */
495 stopGetAddrInfo(id);
496 removeRequestMap(clientId, id, clientInfo);
497 clientInfo.mResolvedService = null;
498 if (DBG) Slog.d(TAG, "SERVICE_RESOLVE_FAILED Raw: " + raw);
499 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_FAILED,
500 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
501 break;
502 case NativeResponseCode.SERVICE_GET_ADDR_SUCCESS:
503 /* NNN resolveId hostname ttl addr */
504 if (DBG) Slog.d(TAG, "SERVICE_GET_ADDR_SUCCESS Raw: " + raw);
505 try {
506 clientInfo.mResolvedService.setHost(InetAddress.getByName(cooked[4]));
507 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_SUCCEEDED,
508 0, clientId, clientInfo.mResolvedService);
509 } catch (java.net.UnknownHostException e) {
510 clientInfo.mChannel.sendMessage(NsdManager.RESOLVE_SERVICE_FAILED,
511 NsdManager.FAILURE_INTERNAL_ERROR, clientId);
512 }
513 stopGetAddrInfo(id);
514 removeRequestMap(clientId, id, clientInfo);
515 clientInfo.mResolvedService = null;
516 break;
517 default:
518 handled = false;
519 break;
520 }
521 return handled;
522 }
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700523 }
524 }
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700525
526 private NativeDaemonConnector mNativeConnector;
527 private final CountDownLatch mNativeDaemonConnected = new CountDownLatch(1);
528
529 private NsdService(Context context) {
530 mContext = context;
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700531 mContentResolver = context.getContentResolver();
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700532
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700533 mNativeConnector = new NativeDaemonConnector(new NativeCallbackReceiver(), "mdns", 10,
534 MDNS_TAG, 25);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700535
536 mNsdStateMachine = new NsdStateMachine(TAG);
537 mNsdStateMachine.start();
538
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700539 Thread th = new Thread(mNativeConnector, MDNS_TAG);
540 th.start();
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700541 }
542
543 public static NsdService create(Context context) throws InterruptedException {
544 NsdService service = new NsdService(context);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700545 service.mNativeDaemonConnected.await();
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700546 return service;
547 }
548
549 public Messenger getMessenger() {
Irfan Sheriff92784672012-04-13 12:15:41 -0700550 mContext.enforceCallingOrSelfPermission(android.Manifest.permission.INTERNET,
551 "NsdService");
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700552 return new Messenger(mNsdStateMachine.getHandler());
553 }
554
555 public void setEnabled(boolean enable) {
556 mContext.enforceCallingOrSelfPermission(android.Manifest.permission.CONNECTIVITY_INTERNAL,
557 "NsdService");
558 Settings.Secure.putInt(mContentResolver, Settings.Secure.NSD_ON, enable ? 1 : 0);
559 if (enable) {
560 mNsdStateMachine.sendMessage(NsdManager.ENABLE);
561 } else {
562 mNsdStateMachine.sendMessage(NsdManager.DISABLE);
563 }
564 }
565
566 private void sendNsdStateChangeBroadcast(boolean enabled) {
Irfan Sheriff54ac7a52012-04-19 10:26:34 -0700567 final Intent intent = new Intent(NsdManager.ACTION_NSD_STATE_CHANGED);
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700568 intent.addFlags(Intent.FLAG_RECEIVER_REGISTERED_ONLY_BEFORE_BOOT);
569 if (enabled) {
570 intent.putExtra(NsdManager.EXTRA_NSD_STATE, NsdManager.NSD_STATE_ENABLED);
571 } else {
572 intent.putExtra(NsdManager.EXTRA_NSD_STATE, NsdManager.NSD_STATE_DISABLED);
573 }
574 mContext.sendStickyBroadcast(intent);
575 }
576
577 private boolean isNsdEnabled() {
578 boolean ret = Settings.Secure.getInt(mContentResolver, Settings.Secure.NSD_ON, 1) == 1;
579 if (DBG) Slog.d(TAG, "Network service discovery enabled " + ret);
580 return ret;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700581 }
582
Irfan Sheriff817388e2012-04-11 14:52:19 -0700583 private int getUniqueId() {
584 if (++mUniqueId == INVALID_ID) return ++mUniqueId;
585 return mUniqueId;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700586 }
587
Irfan Sheriff817388e2012-04-11 14:52:19 -0700588 /* These should be in sync with system/netd/mDnsResponseCode.h */
589 class NativeResponseCode {
590 public static final int SERVICE_DISCOVERY_FAILED = 602;
591 public static final int SERVICE_FOUND = 603;
592 public static final int SERVICE_LOST = 604;
593
594 public static final int SERVICE_REGISTRATION_FAILED = 605;
595 public static final int SERVICE_REGISTERED = 606;
596
597 public static final int SERVICE_RESOLUTION_FAILED = 607;
598 public static final int SERVICE_RESOLVED = 608;
599
600 public static final int SERVICE_UPDATED = 609;
601 public static final int SERVICE_UPDATE_FAILED = 610;
602
603 public static final int SERVICE_GET_ADDR_FAILED = 611;
604 public static final int SERVICE_GET_ADDR_SUCCESS = 612;
605 }
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700606
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700607 private class NativeEvent {
Vairavan Srinivasan6727e732012-08-05 13:14:12 -0700608 final int code;
609 final String raw;
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700610
611 NativeEvent(int code, String raw) {
612 this.code = code;
613 this.raw = raw;
614 }
615 }
616
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700617 class NativeCallbackReceiver implements INativeDaemonConnectorCallbacks {
618 public void onDaemonConnected() {
619 mNativeDaemonConnected.countDown();
620 }
621
622 public boolean onEvent(int code, String raw, String[] cooked) {
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700623 // TODO: NDC translates a message to a callback, we could enhance NDC to
624 // directly interact with a state machine through messages
625 NativeEvent event = new NativeEvent(code, raw);
626 mNsdStateMachine.sendMessage(NsdManager.NATIVE_DAEMON_EVENT, event);
627 return true;
628 }
629 }
Irfan Sheriff817388e2012-04-11 14:52:19 -0700630
Irfan Sheriff817388e2012-04-11 14:52:19 -0700631 private boolean startMDnsDaemon() {
632 if (DBG) Slog.d(TAG, "startMDnsDaemon");
633 try {
634 mNativeConnector.execute("mdnssd", "start-service");
635 } catch(NativeDaemonConnectorException e) {
636 Slog.e(TAG, "Failed to start daemon" + e);
637 return false;
638 }
639 return true;
640 }
641
642 private boolean stopMDnsDaemon() {
643 if (DBG) Slog.d(TAG, "stopMDnsDaemon");
644 try {
645 mNativeConnector.execute("mdnssd", "stop-service");
646 } catch(NativeDaemonConnectorException e) {
647 Slog.e(TAG, "Failed to start daemon" + e);
648 return false;
649 }
650 return true;
651 }
652
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700653 private boolean registerService(int regId, NsdServiceInfo service) {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700654 if (DBG) Slog.d(TAG, "registerService: " + regId + " " + service);
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700655 try {
656 //Add txtlen and txtdata
657 mNativeConnector.execute("mdnssd", "register", regId, service.getServiceName(),
658 service.getServiceType(), service.getPort());
659 } catch(NativeDaemonConnectorException e) {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700660 Slog.e(TAG, "Failed to execute registerService " + e);
661 return false;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700662 }
Irfan Sheriff817388e2012-04-11 14:52:19 -0700663 return true;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700664 }
665
Irfan Sheriff817388e2012-04-11 14:52:19 -0700666 private boolean unregisterService(int regId) {
667 if (DBG) Slog.d(TAG, "unregisterService: " + regId);
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700668 try {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700669 mNativeConnector.execute("mdnssd", "stop-register", regId);
670 } catch(NativeDaemonConnectorException e) {
671 Slog.e(TAG, "Failed to execute unregisterService " + e);
672 return false;
673 }
674 return true;
675 }
676
677 private boolean updateService(int regId, DnsSdTxtRecord t) {
678 if (DBG) Slog.d(TAG, "updateService: " + regId + " " + t);
679 try {
680 if (t == null) return false;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700681 mNativeConnector.execute("mdnssd", "update", regId, t.size(), t.getRawData());
682 } catch(NativeDaemonConnectorException e) {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700683 Slog.e(TAG, "Failed to updateServices " + e);
684 return false;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700685 }
Irfan Sheriff817388e2012-04-11 14:52:19 -0700686 return true;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700687 }
688
Irfan Sheriff817388e2012-04-11 14:52:19 -0700689 private boolean discoverServices(int discoveryId, String serviceType) {
690 if (DBG) Slog.d(TAG, "discoverServices: " + discoveryId + " " + serviceType);
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700691 try {
692 mNativeConnector.execute("mdnssd", "discover", discoveryId, serviceType);
693 } catch(NativeDaemonConnectorException e) {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700694 Slog.e(TAG, "Failed to discoverServices " + e);
695 return false;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700696 }
Irfan Sheriff817388e2012-04-11 14:52:19 -0700697 return true;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700698 }
699
Irfan Sheriff817388e2012-04-11 14:52:19 -0700700 private boolean stopServiceDiscovery(int discoveryId) {
701 if (DBG) Slog.d(TAG, "stopServiceDiscovery: " + discoveryId);
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700702 try {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700703 mNativeConnector.execute("mdnssd", "stop-discover", discoveryId);
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700704 } catch(NativeDaemonConnectorException e) {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700705 Slog.e(TAG, "Failed to stopServiceDiscovery " + e);
706 return false;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700707 }
Irfan Sheriff817388e2012-04-11 14:52:19 -0700708 return true;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700709 }
710
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700711 private boolean resolveService(int resolveId, NsdServiceInfo service) {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700712 if (DBG) Slog.d(TAG, "resolveService: " + resolveId + " " + service);
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700713 try {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700714 mNativeConnector.execute("mdnssd", "resolve", resolveId, service.getServiceName(),
715 service.getServiceType(), "local.");
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700716 } catch(NativeDaemonConnectorException e) {
Irfan Sheriff817388e2012-04-11 14:52:19 -0700717 Slog.e(TAG, "Failed to resolveService " + e);
718 return false;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700719 }
Irfan Sheriff817388e2012-04-11 14:52:19 -0700720 return true;
721 }
722
723 private boolean stopResolveService(int resolveId) {
724 if (DBG) Slog.d(TAG, "stopResolveService: " + resolveId);
725 try {
726 mNativeConnector.execute("mdnssd", "stop-resolve", resolveId);
727 } catch(NativeDaemonConnectorException e) {
728 Slog.e(TAG, "Failed to stop resolve " + e);
729 return false;
730 }
731 return true;
732 }
733
734 private boolean getAddrInfo(int resolveId, String hostname) {
735 if (DBG) Slog.d(TAG, "getAdddrInfo: " + resolveId);
736 try {
737 mNativeConnector.execute("mdnssd", "getaddrinfo", resolveId, hostname);
738 } catch(NativeDaemonConnectorException e) {
739 Slog.e(TAG, "Failed to getAddrInfo " + e);
740 return false;
741 }
742 return true;
743 }
744
745 private boolean stopGetAddrInfo(int resolveId) {
746 if (DBG) Slog.d(TAG, "stopGetAdddrInfo: " + resolveId);
747 try {
748 mNativeConnector.execute("mdnssd", "stop-getaddrinfo", resolveId);
749 } catch(NativeDaemonConnectorException e) {
750 Slog.e(TAG, "Failed to stopGetAddrInfo " + e);
751 return false;
752 }
753 return true;
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700754 }
755
756 @Override
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700757 public void dump(FileDescriptor fd, PrintWriter pw, String[] args) {
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700758 if (mContext.checkCallingOrSelfPermission(android.Manifest.permission.DUMP)
759 != PackageManager.PERMISSION_GRANTED) {
760 pw.println("Permission Denial: can't dump ServiceDiscoverService from from pid="
761 + Binder.getCallingPid()
762 + ", uid=" + Binder.getCallingUid());
763 return;
764 }
765
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700766 for (ClientInfo client : mClients.values()) {
767 pw.println("Client Info");
768 pw.println(client);
769 }
770
771 mNsdStateMachine.dump(fd, pw, args);
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700772 }
Irfan Sheriff817388e2012-04-11 14:52:19 -0700773
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700774 /* arg2 on the source message has an id that needs to be retained in replies
775 * see NsdManager for details */
776 private Message obtainMessage(Message srcMsg) {
777 Message msg = Message.obtain();
778 msg.arg2 = srcMsg.arg2;
779 return msg;
Irfan Sheriff817388e2012-04-11 14:52:19 -0700780 }
781
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700782 private void replyToMessage(Message msg, int what) {
783 if (msg.replyTo == null) return;
784 Message dstMsg = obtainMessage(msg);
785 dstMsg.what = what;
786 mReplyChannel.replyToMessage(msg, dstMsg);
Irfan Sheriff817388e2012-04-11 14:52:19 -0700787 }
788
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700789 private void replyToMessage(Message msg, int what, int arg1) {
790 if (msg.replyTo == null) return;
791 Message dstMsg = obtainMessage(msg);
792 dstMsg.what = what;
793 dstMsg.arg1 = arg1;
794 mReplyChannel.replyToMessage(msg, dstMsg);
795 }
796
797 private void replyToMessage(Message msg, int what, Object obj) {
798 if (msg.replyTo == null) return;
799 Message dstMsg = obtainMessage(msg);
800 dstMsg.what = what;
801 dstMsg.obj = obj;
802 mReplyChannel.replyToMessage(msg, dstMsg);
Irfan Sheriff817388e2012-04-11 14:52:19 -0700803 }
804
805 /* Information tracked per client */
806 private class ClientInfo {
807
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700808 private static final int MAX_LIMIT = 10;
Vairavan Srinivasan6727e732012-08-05 13:14:12 -0700809 private final AsyncChannel mChannel;
810 private final Messenger mMessenger;
Irfan Sheriff817388e2012-04-11 14:52:19 -0700811 /* Remembers a resolved service until getaddrinfo completes */
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700812 private NsdServiceInfo mResolvedService;
813
814 /* A map from client id to unique id sent to mDns */
815 private SparseArray<Integer> mClientIds = new SparseArray<Integer>();
Irfan Sheriff817388e2012-04-11 14:52:19 -0700816
817 private ClientInfo(AsyncChannel c, Messenger m) {
818 mChannel = c;
819 mMessenger = m;
Irfan Sheriff817388e2012-04-11 14:52:19 -0700820 if (DBG) Slog.d(TAG, "New client, channel: " + c + " messenger: " + m);
821 }
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700822
823 @Override
824 public String toString() {
825 StringBuffer sb = new StringBuffer();
826 sb.append("mChannel ").append(mChannel).append("\n");
827 sb.append("mMessenger ").append(mMessenger).append("\n");
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700828 sb.append("mResolvedService ").append(mResolvedService).append("\n");
Irfan Sheriff22af38c2012-05-03 16:44:27 -0700829 for(int i = 0; i< mClientIds.size(); i++) {
830 sb.append("clientId ").append(mClientIds.keyAt(i));
831 sb.append(" mDnsId ").append(mClientIds.valueAt(i)).append("\n");
Irfan Sheriff3ef889b2012-04-17 23:15:29 -0700832 }
833 return sb.toString();
834 }
Irfan Sheriff817388e2012-04-11 14:52:19 -0700835 }
Irfan Sheriff7d024d32012-03-22 17:01:39 -0700836}