Imported Upstream version 2.88
[platform/upstream/dnsmasq.git] / src / dbus.c
index cbdce9c..fd5d1ca 100644 (file)
@@ -1,4 +1,4 @@
-/* dnsmasq is Copyright (c) 2000-2021 Simon Kelley
+/* dnsmasq is Copyright (c) 2000-2022 Simon Kelley
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
@@ -52,6 +52,9 @@ const char* introspection_xml_template =
 "    <method name=\"SetFilterWin2KOption\">\n"
 "      <arg name=\"filterwin2k\" direction=\"in\" type=\"b\"/>\n"
 "    </method>\n"
+"    <method name=\"SetLocaliseQueriesOption\">\n"
+"      <arg name=\"localise-queries\" direction=\"in\" type=\"b\"/>\n"
+"    </method>\n"
 "    <method name=\"SetBogusPrivOption\">\n"
 "      <arg name=\"boguspriv\" direction=\"in\" type=\"b\"/>\n"
 "    </method>\n"
@@ -88,6 +91,11 @@ const char* introspection_xml_template =
 "    <method name=\"GetMetrics\">\n"
 "      <arg name=\"metrics\" direction=\"out\" type=\"a{su}\"/>\n"
 "    </method>\n"
+"    <method name=\"GetServerMetrics\">\n"
+"      <arg name=\"metrics\" direction=\"out\" type=\"a{ss}\"/>\n"
+"    </method>\n"
+"    <method name=\"ClearMetrics\">\n"
+"    </method>\n"
 "  </interface>\n"
 "</node>\n";
 
@@ -114,7 +122,7 @@ static dbus_bool_t add_watch(DBusWatch *watch, void *data)
   w->next = daemon->watches;
   daemon->watches = w;
 
-  w = data; /* no warning */
+  (void)data; /* no warning */
   return TRUE;
 }
 
@@ -134,16 +142,20 @@ static void remove_watch(DBusWatch *watch, void *data)
        up = &(w->next);
     }
 
-  w = data; /* no warning */
+  (void)data; /* no warning */
 }
 
-static void dbus_read_servers(DBusMessage *message)
+static DBusMessage* dbus_read_servers(DBusMessage *message)
 {
   DBusMessageIter iter;
   union  mysockaddr addr, source_addr;
   char *domain;
   
-  dbus_message_iter_init(message, &iter);
+  if (!dbus_message_iter_init(message, &iter))
+    {
+      return dbus_message_new_error(message, DBUS_ERROR_INVALID_ARGS,
+                                    "Failed to initialize dbus message iter");
+    }
 
   mark_servers(SERV_FROM_DBUS);
   
@@ -222,6 +234,7 @@ static void dbus_read_servers(DBusMessage *message)
    
   /* unlink and free anything still marked. */
   cleanup_servers();
+  return NULL;
 }
 
 #ifdef HAVE_LOOP
@@ -279,6 +292,11 @@ static DBusMessage* dbus_read_servers_ex(DBusMessage *message, int strings)
       u16 flags = 0;
       char interface[IF_NAMESIZE];
       char *str_addr, *str_domain = NULL;
+      struct server_details sdetails = { 0 };
+      sdetails.addr = &addr;
+      sdetails.source_addr = &source_addr;
+      sdetails.interface = interface;
+      sdetails.flags = &flags;
 
       if (strings)
        {
@@ -361,20 +379,6 @@ static DBusMessage* dbus_read_servers_ex(DBusMessage *message, int strings)
          strcpy(str_addr, str);
        }
 
-      /* parse the IP address */
-      if ((addr_err = parse_server(str_addr, &addr, &source_addr, (char *) &interface, &flags)))
-       {
-          error = dbus_message_new_error_printf(message, DBUS_ERROR_INVALID_ARGS,
-                                                "Invalid IP address '%s': %s",
-                                                str, addr_err);
-          break;
-        }
-      
-      /* 0.0.0.0 for server address == NULL, for Dbus */
-      if (addr.in.sin_family == AF_INET &&
-          addr.in.sin_addr.s_addr == 0)
-        flags |= SERV_LITERAL_ADDRESS;
-      
       if (strings)
        {
          char *p;
@@ -388,7 +392,31 @@ static DBusMessage* dbus_read_servers_ex(DBusMessage *message, int strings)
            else 
              p = NULL;
            
-           add_update_server(flags | SERV_FROM_DBUS, &addr, &source_addr, interface, str_domain, NULL);
+            if (strings && strlen(str_addr) == 0)
+              add_update_server(SERV_LITERAL_ADDRESS | SERV_FROM_DBUS, &addr, &source_addr, interface, str_domain, NULL);
+            else
+              {
+                if ((addr_err = parse_server(str_addr, &sdetails)))
+                  {
+                    error = dbus_message_new_error_printf(message, DBUS_ERROR_INVALID_ARGS,
+                                                          "Invalid IP address '%s': %s",
+                                                          str, addr_err);
+                    break;
+                  }
+                
+                while (parse_server_next(&sdetails))
+                  {
+                    if ((addr_err = parse_server_addr(&sdetails)))
+                      {
+                        error = dbus_message_new_error_printf(message, DBUS_ERROR_INVALID_ARGS,
+                                                              "Invalid IP address '%s': %s",
+                                                              str, addr_err);
+                        break;
+                      }
+                    
+                    add_update_server(flags | SERV_FROM_DBUS, &addr, &source_addr, interface, str_domain, NULL);
+                  }
+              }
          } while ((str_domain = p));
        }
       else
@@ -402,11 +430,40 @@ static DBusMessage* dbus_read_servers_ex(DBusMessage *message, int strings)
            if (dbus_message_iter_get_arg_type(&string_iter) == DBUS_TYPE_STRING)
              dbus_message_iter_get_basic(&string_iter, &str);
            dbus_message_iter_next (&string_iter);
+
+           if ((addr_err = parse_server(str_addr, &sdetails)))
+             {
+               error = dbus_message_new_error_printf(message, DBUS_ERROR_INVALID_ARGS,
+                                                     "Invalid IP address '%s': %s",
+                                                     str, addr_err);
+               break;
+             }
            
-           add_update_server(flags | SERV_FROM_DBUS, &addr, &source_addr, interface, str, NULL);
+           while (parse_server_next(&sdetails))
+             {
+               if ((addr_err = parse_server_addr(&sdetails)))
+                 {
+                   error = dbus_message_new_error_printf(message, DBUS_ERROR_INVALID_ARGS,
+                                                         "Invalid IP address '%s': %s",
+                                                         str, addr_err);
+                   break;
+                 }
+               
+               /* 0.0.0.0 for server address == NULL, for Dbus */
+               if (addr.in.sin_family == AF_INET &&
+                   addr.in.sin_addr.s_addr == 0)
+                 flags |= SERV_LITERAL_ADDRESS;
+               else
+                 flags &= ~SERV_LITERAL_ADDRESS;
+               
+               add_update_server(flags | SERV_FROM_DBUS, &addr, &source_addr, interface, str, NULL);
+             }
          } while (dbus_message_iter_get_arg_type(&string_iter) == DBUS_TYPE_STRING);
        }
-        
+      
+      if (sdetails.orig_hostinfo)
+       freeaddrinfo(sdetails.orig_hostinfo);
+      
       /* jump to next element in outer array */
       dbus_message_iter_next(&array_iter);
     }
@@ -545,6 +602,10 @@ static DBusMessage *dbus_add_lease(DBusMessage* message)
                                         "Invalid IP address '%s'", ipaddr);
    
   hw_len = parse_hex((char*)hwaddr, dhcp_chaddr, DHCP_CHADDR_MAX, NULL, &hw_type);
+  if (hw_len < 0)
+    return dbus_message_new_error_printf(message, DBUS_ERROR_INVALID_ARGS,
+                                        "Invalid HW address '%s'", hwaddr);
+
   if (hw_type == 0 && hw_len != 0)
     hw_type = ARPHRD_ETHER;
   
@@ -632,6 +693,77 @@ static DBusMessage *dbus_get_metrics(DBusMessage* message)
   return reply;
 }
 
+static void add_dict_entry(DBusMessageIter *container, const char *key, const char *val)
+{
+  DBusMessageIter dict;
+
+  dbus_message_iter_open_container(container, DBUS_TYPE_DICT_ENTRY, NULL, &dict);
+  dbus_message_iter_append_basic(&dict, DBUS_TYPE_STRING, &key);
+  dbus_message_iter_append_basic(&dict, DBUS_TYPE_STRING, &val);
+  dbus_message_iter_close_container(container, &dict);
+}
+
+static void add_dict_int(DBusMessageIter *container, const char *key, const unsigned int val)
+{
+  snprintf(daemon->namebuff, MAXDNAME, "%u", val);
+  
+  add_dict_entry(container, key, daemon->namebuff);
+}
+
+static DBusMessage *dbus_get_server_metrics(DBusMessage* message)
+{
+  DBusMessage *reply = dbus_message_new_method_return(message);
+  DBusMessageIter server_array, dict_array, server_iter;
+  struct server *serv;
+  
+  dbus_message_iter_init_append(reply, &server_iter);
+  dbus_message_iter_open_container(&server_iter, DBUS_TYPE_ARRAY, "a{ss}", &server_array);
+
+  /* sum counts from different records for same server */
+  for (serv = daemon->servers; serv; serv = serv->next)
+    serv->flags &= ~SERV_MARK;
+  
+  for (serv = daemon->servers; serv; serv = serv->next)
+    if (!(serv->flags & SERV_MARK))
+      {
+       unsigned int port;
+       unsigned int queries = 0, failed_queries = 0, nxdomain_replies = 0, retrys = 0;
+       unsigned int sigma_latency = 0, count_latency = 0;
+       
+       struct server *serv1;
+
+       for (serv1 = serv; serv1; serv1 = serv1->next)
+         if (!(serv1->flags & SERV_MARK) && sockaddr_isequal(&serv->addr, &serv1->addr))
+           {
+             serv1->flags |= SERV_MARK;
+             queries += serv1->queries;
+             failed_queries += serv1->failed_queries;
+             nxdomain_replies += serv1->nxdomain_replies;
+             retrys += serv1->retrys;
+             sigma_latency += serv1->query_latency;
+             count_latency++;
+           }
+       
+       dbus_message_iter_open_container(&server_array, DBUS_TYPE_ARRAY, "{ss}", &dict_array);
+       
+       port = prettyprint_addr(&serv->addr, daemon->namebuff);
+       add_dict_entry(&dict_array, "address", daemon->namebuff);
+       
+       add_dict_int(&dict_array, "port", port);
+       add_dict_int(&dict_array, "queries", serv->queries);
+       add_dict_int(&dict_array, "failed_queries", serv->failed_queries);
+       add_dict_int(&dict_array, "nxdomain", serv->nxdomain_replies);
+       add_dict_int(&dict_array, "retries", serv->retrys);
+       add_dict_int(&dict_array, "latency", sigma_latency/count_latency);
+       
+       dbus_message_iter_close_container(&server_array, &dict_array);
+      }
+  
+  dbus_message_iter_close_container(&server_iter, &server_array);
+  
+  return reply;
+}
+
 DBusHandlerResult message_handler(DBusConnection *connection, 
                                  DBusMessage *message, 
                                  void *user_data)
@@ -668,7 +800,7 @@ DBusHandlerResult message_handler(DBusConnection *connection,
 #endif
   else if (strcmp(method, "SetServers") == 0)
     {
-      dbus_read_servers(message);
+      reply = dbus_read_servers(message);
       new_servers = 1;
     }
   else if (strcmp(method, "SetServersEx") == 0)
@@ -685,6 +817,10 @@ DBusHandlerResult message_handler(DBusConnection *connection,
     {
       reply = dbus_set_bool(message, OPT_FILTER, "filterwin2k");
     }
+  else if (strcmp(method, "SetLocaliseQueriesOption") == 0)
+    {
+      reply = dbus_set_bool(message, OPT_LOCALISE, "localise-queries");
+    }
   else if (strcmp(method, "SetBogusPrivOption") == 0)
     {
       reply = dbus_set_bool(message, OPT_BOGUSPRIV, "bogus-priv");
@@ -703,6 +839,14 @@ DBusHandlerResult message_handler(DBusConnection *connection,
     {
       reply = dbus_get_metrics(message);
     }
+  else if (strcmp(method, "GetServerMetrics") == 0)
+    {
+      reply = dbus_get_server_metrics(message);
+    }
+  else if (strcmp(method, "ClearMetrics") == 0)
+    {
+      clear_metrics();
+    }
   else if (strcmp(method, "ClearCache") == 0)
     clear_cache = 1;
   else
@@ -719,7 +863,7 @@ DBusHandlerResult message_handler(DBusConnection *connection,
   if (clear_cache)
     clear_cache_and_reload(dnsmasq_time());
   
-  method = user_data; /* no warning */
+  (void)user_data; /* no warning */
 
   /* If no reply or no error, return nothing */
   if (!reply)
@@ -745,8 +889,11 @@ char *dbus_init(void)
 
   dbus_error_init (&dbus_error);
   if (!(connection = dbus_bus_get (DBUS_BUS_SYSTEM, &dbus_error)))
-    return NULL;
-    
+    {
+      dbus_error_free(&dbus_error);
+      return NULL;
+    }
+  
   dbus_connection_set_exit_on_disconnect(connection, FALSE);
   dbus_connection_set_watch_functions(connection, add_watch, remove_watch, 
                                      NULL, NULL, NULL);