DnsProxyListener.cpp

   1 /*
   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
  17 #include <arpa/inet.h>
  18 #include <dirent.h>
  19 #include <errno.h>
  20 #include <linux/if.h>
  21 #include <netdb.h>
  22 #include <netinet/in.h>
  23 #include <stdlib.h>
  24 #include <sys/socket.h>
  25 #include <sys/types.h>
  26 #include <string.h>
  27
  28 #define LOG_TAG "DnsProxyListener"
  29 #define DBG 0
  30
  31 #include <cutils/log.h>
  32 #include <sysutils/SocketClient.h>
  33
  34 #include "DnsProxyListener.h"
  35 #include "ResponseCode.h"
  36
  37 DnsProxyListener::DnsProxyListener() :
  38                  FrameworkListener("dnsproxyd") {
  39     registerCmd(new GetAddrInfoCmd());
  40     registerCmd(new GetHostByAddrCmd());
  41 }
  42
  43 DnsProxyListener::GetAddrInfoHandler::~GetAddrInfoHandler() {
  44     free(mHost);
  45     free(mService);
  46     free(mHints);
  47 }
  48
  49 void DnsProxyListener::GetAddrInfoHandler::start() {
  50     pthread_create(&mThread, NULL,
  51                    DnsProxyListener::GetAddrInfoHandler::threadStart, this);
  52 }
  53
  54 void* DnsProxyListener::GetAddrInfoHandler::threadStart(void* obj) {
  55     GetAddrInfoHandler* handler = reinterpret_cast<GetAddrInfoHandler*>(obj);
  56     handler->run();
  57     delete handler;
  58     pthread_exit(NULL);
  59     return NULL;
  60 }
  61
  62 // Sends 4 bytes of big-endian length, followed by the data.
  63 // Returns true on success.
  64 static bool sendLenAndData(SocketClient *c, const int len, const void* data) {
  65     uint32_t len_be = htonl(len);
  66     return c->sendData(&len_be, 4) == 0 &&
  67         (len == 0 || c->sendData(data, len) == 0);
  68 }
  69
  70 void DnsProxyListener::GetAddrInfoHandler::run() {
  71     if (DBG) {
  72         ALOGD("GetAddrInfoHandler, now for %s / %s", mHost, mService);
  73     }
  74
  75     struct addrinfo* result = NULL;
  76     uint32_t rv = getaddrinfo(mHost, mService, mHints, &result);
  77     if (rv) {
  78         // getaddrinfo failed
  79         mClient->sendBinaryMsg(ResponseCode::DnsProxyOperationFailed, &rv, sizeof(rv));
  80     } else {
  81         bool success = !mClient->sendCode(ResponseCode::DnsProxyQueryResult);
  82         struct addrinfo* ai = result;
  83         while (ai && success) {
  84             success = sendLenAndData(mClient, sizeof(struct addrinfo), ai)
  85                 && sendLenAndData(mClient, ai->ai_addrlen, ai->ai_addr)
  86                 && sendLenAndData(mClient,
  87                                   ai->ai_canonname ? strlen(ai->ai_canonname) + 1 : 0,
  88                                   ai->ai_canonname);
  89             ai = ai->ai_next;
  90         }
  91         success = success && sendLenAndData(mClient, 0, "");
  92         if (!success) {
  93             ALOGW("Error writing DNS result to client");
  94         }
  95     }
  96     if (result) {
  97         freeaddrinfo(result);
  98     }
  99     mClient->decRef();
 100 }
 101
 102 DnsProxyListener::GetAddrInfoCmd::GetAddrInfoCmd() :
 103     NetdCommand("getaddrinfo") {
 104 }
 105
 106 int DnsProxyListener::GetAddrInfoCmd::runCommand(SocketClient *cli,
 107                                             int argc, char **argv) {
 108     if (DBG) {
 109         for (int i = 0; i < argc; i++) {
 110             ALOGD("argv[%i]=%s", i, argv[i]);
 111         }
 112     }
 113     if (argc != 7) {
 114         char* msg = NULL;
 115         asprintf( &msg, "Invalid number of arguments to getaddrinfo: %i", argc);
 116         ALOGW("%s", msg);
 117         cli->sendMsg(ResponseCode::CommandParameterError, msg, false);
 118         free(msg);
 119         return -1;
 120     }
 121
 122     char* name = argv[1];
 123     if (strcmp("^", name) == 0) {
 124         name = NULL;
 125     } else {
 126         name = strdup(name);
 127     }
 128
 129     char* service = argv[2];
 130     if (strcmp("^", service) == 0) {
 131         service = NULL;
 132     } else {
 133         service = strdup(service);
 134     }
 135
 136     struct addrinfo* hints = NULL;
 137     int ai_flags = atoi(argv[3]);
 138     int ai_family = atoi(argv[4]);
 139     int ai_socktype = atoi(argv[5]);
 140     int ai_protocol = atoi(argv[6]);
 141     if (ai_flags != -1 || ai_family != -1 ||
 142         ai_socktype != -1 || ai_protocol != -1) {
 143         hints = (struct addrinfo*) calloc(1, sizeof(struct addrinfo));
 144         hints->ai_flags = ai_flags;
 145         hints->ai_family = ai_family;
 146         hints->ai_socktype = ai_socktype;
 147         hints->ai_protocol = ai_protocol;
 148     }
 149
 150     if (DBG) {
 151         ALOGD("GetAddrInfoHandler for %s / %s",
 152              name ? name : "[nullhost]",
 153              service ? service : "[nullservice]");
 154     }
 155
 156     cli->incRef();
 157     DnsProxyListener::GetAddrInfoHandler* handler =
 158         new DnsProxyListener::GetAddrInfoHandler(cli, name, service, hints);
 159     handler->start();
 160
 161     return 0;
 162 }
 163
 164 /*******************************************************
 165  *                  GetHostByAddr                       *
 166  *******************************************************/
 167 DnsProxyListener::GetHostByAddrCmd::GetHostByAddrCmd() :
 168         NetdCommand("gethostbyaddr") {
 169 }
 170
 171 int DnsProxyListener::GetHostByAddrCmd::runCommand(SocketClient *cli,
 172                                             int argc, char **argv) {
 173     if (DBG) {
 174         for (int i = 0; i < argc; i++) {
 175             ALOGD("argv[%i]=%s", i, argv[i]);
 176         }
 177     }
 178
 179     if (argc != 4) {
 180         char* msg = NULL;
 181         asprintf(&msg, "Invalid number of arguments to gethostbyaddr: %i", argc);
 182         ALOGW("%s", msg);
 183         cli->sendMsg(ResponseCode::CommandParameterError, msg, false);
 184         free(msg);
 185         return -1;
 186     }
 187
 188     char* addrStr = argv[1];
 189     int addrLen = atoi(argv[2]);
 190     int addrFamily = atoi(argv[3]);
 191
 192     void* addr = malloc(sizeof(struct in6_addr));
 193     errno = 0;
 194     int result = inet_pton(addrFamily, addrStr, addr);
 195     if (result <= 0) {
 196         char* msg = NULL;
 197         asprintf(&msg, "inet_pton(\"%s\") failed %s", addrStr, strerror(errno));
 198         ALOGW("%s", msg);
 199         cli->sendMsg(ResponseCode::OperationFailed, msg, false);
 200         free(addr);
 201         free(msg);
 202         return -1;
 203     }
 204
 205     cli->incRef();
 206     DnsProxyListener::GetHostByAddrHandler* handler =
 207             new DnsProxyListener::GetHostByAddrHandler(cli, addr, addrLen, addrFamily);
 208     handler->start();
 209
 210     return 0;
 211 }
 212
 213 DnsProxyListener::GetHostByAddrHandler::~GetHostByAddrHandler() {
 214     free(mAddress);
 215 }
 216
 217 void DnsProxyListener::GetHostByAddrHandler::start() {
 218     pthread_create(&mThread, NULL,
 219                    DnsProxyListener::GetHostByAddrHandler::threadStart, this);
 220 }
 221
 222 void* DnsProxyListener::GetHostByAddrHandler::threadStart(void* obj) {
 223     GetHostByAddrHandler* handler = reinterpret_cast<GetHostByAddrHandler*>(obj);
 224     handler->run();
 225     delete handler;
 226     pthread_exit(NULL);
 227     return NULL;
 228 }
 229
 230 void DnsProxyListener::GetHostByAddrHandler::run() {
 231     if (DBG) {
 232         ALOGD("DnsProxyListener::GetHostByAddrHandler::run\n");
 233     }
 234
 235     struct hostent* hp;
 236
 237     // NOTE gethostbyaddr should take a void* but bionic thinks it should be char*
 238     hp = gethostbyaddr((char*)mAddress, mAddressLen, mAddressFamily);
 239
 240     if (DBG) {
 241         ALOGD("GetHostByAddrHandler::run gethostbyaddr errno: %s hp->h_name = %s, name_len = %d\n",
 242                 hp ? "success" : strerror(errno),
 243                 (hp && hp->h_name) ? hp->h_name: "null",
 244                 (hp && hp->h_name) ? strlen(hp->h_name)+ 1 : 0);
 245     }
 246
 247     bool failed = true;
 248     if (hp) {
 249         failed = mClient->sendBinaryMsg(ResponseCode::DnsProxyQueryResult,
 250                                         hp->h_name ? hp->h_name : "",
 251                                         hp->h_name ? strlen(hp->h_name)+ 1 : 0);
 252     } else {
 253         uint32_t error = h_errno;
 254         failed = mClient->sendBinaryMsg(ResponseCode::DnsProxyOperationFailed,
 255                                         &error, sizeof(error));
 256     }
 257
 258     if (failed) {
 259         ALOGW("GetHostByAddrHandler: Error writing DNS result to client\n");
 260     }
 261     mClient->decRef();
 262 }