src/virt.c

   1 /**
   2  * collectd - src/virt.c
   3  * Copyright (C) 2006-2008  Red Hat Inc.
   4  *
   5  * This program is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License as published by the
   7  * Free Software Foundation; only version 2 of the license is applicable.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License along
  15  * with this program; if not, write to the Free Software Foundation, Inc.,
  16  * 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
  17  *
  18  * Authors:
  19  *   Richard W.M. Jones <rjones@redhat.com>
  20  **/
  21
  22 #include "collectd.h"
  23
  24 #include "common.h"
  25 #include "plugin.h"
  26 #include "utils_ignorelist.h"
  27 #include "utils_complain.h"
  28
  29 #include <libvirt/libvirt.h>
  30 #include <libvirt/virterror.h>
  31 #include <libxml/parser.h>
  32 #include <libxml/tree.h>
  33 #include <libxml/xpath.h>
  34
  35 /* Plugin name */
  36 #define PLUGIN_NAME "virt"
  37
  38 static const char *config_keys[] = {
  39     "Connection",
  40
  41     "RefreshInterval",
  42
  43     "Domain",
  44     "BlockDevice",
  45     "InterfaceDevice",
  46     "IgnoreSelected",
  47
  48     "HostnameFormat",
  49     "InterfaceFormat",
  50
  51     "PluginInstanceFormat",
  52
  53     NULL
  54 };
  55 #define NR_CONFIG_KEYS ((sizeof config_keys / sizeof config_keys[0]) - 1)
  56
  57 /* Connection. */
  58 static virConnectPtr conn = 0;
  59 static char *conn_string = NULL;
  60 static c_complain_t conn_complain = C_COMPLAIN_INIT_STATIC;
  61
  62 /* Seconds between list refreshes, 0 disables completely. */
  63 static int interval = 60;
  64
  65 /* List of domains, if specified. */
  66 static ignorelist_t *il_domains = NULL;
  67 /* List of block devices, if specified. */
  68 static ignorelist_t *il_block_devices = NULL;
  69 /* List of network interface devices, if specified. */
  70 static ignorelist_t *il_interface_devices = NULL;
  71
  72 static int ignore_device_match (ignorelist_t *,
  73                                 const char *domname, const char *devpath);
  74
  75 /* Actual list of domains found on last refresh. */
  76 static virDomainPtr *domains = NULL;
  77 static int nr_domains = 0;
  78
  79 static void free_domains (void);
  80 static int add_domain (virDomainPtr dom);
  81
  82 /* Actual list of block devices found on last refresh. */
  83 struct block_device {
  84     virDomainPtr dom;           /* domain */
  85     char *path;                 /* name of block device */
  86 };
  87
  88 static struct block_device *block_devices = NULL;
  89 static int nr_block_devices = 0;
  90
  91 static void free_block_devices (void);
  92 static int add_block_device (virDomainPtr dom, const char *path);
  93
  94 /* Actual list of network interfaces found on last refresh. */
  95 struct interface_device {
  96     virDomainPtr dom;           /* domain */
  97     char *path;                 /* name of interface device */
  98     char *address;              /* mac address of interface device */
  99     char *number;               /* interface device number */
 100 };
 101
 102 static struct interface_device *interface_devices = NULL;
 103 static int nr_interface_devices = 0;
 104
 105 static void free_interface_devices (void);
 106 static int add_interface_device (virDomainPtr dom, const char *path, const char *address, unsigned int number);
 107
 108 /* HostnameFormat. */
 109 #define HF_MAX_FIELDS 3
 110
 111 enum hf_field {
 112     hf_none = 0,
 113     hf_hostname,
 114     hf_name,
 115     hf_uuid
 116 };
 117
 118 static enum hf_field hostname_format[HF_MAX_FIELDS] =
 119     { hf_name };
 120
 121 /* PluginInstanceFormat */
 122 #define PLGINST_MAX_FIELDS 2
 123
 124 enum plginst_field {
 125     plginst_none = 0,
 126     plginst_name,
 127     plginst_uuid
 128 };
 129
 130 static enum plginst_field plugin_instance_format[PLGINST_MAX_FIELDS] =
 131     { plginst_none };
 132
 133 /* InterfaceFormat. */
 134 enum if_field {
 135     if_address,
 136     if_name,
 137     if_number
 138 };
 139
 140 static enum if_field interface_format = if_name;
 141
 142 /* Time that we last refreshed. */
 143 static time_t last_refresh = (time_t) 0;
 144
 145 static int refresh_lists (void);
 146
 147 /* ERROR(...) macro for virterrors. */
 148 #define VIRT_ERROR(conn,s) do {                 \
 149         virErrorPtr err;                        \
 150         err = (conn) ? virConnGetLastError ((conn)) : virGetLastError (); \
 151         if (err) ERROR ("%s: %s", (s), err->message);                   \
 152     } while(0)
 153
 154 static void
 155 init_value_list (value_list_t *vl, virDomainPtr dom)
 156 {
 157     int n;
 158     const char *name;
 159     char uuid[VIR_UUID_STRING_BUFLEN];
 160
 161     sstrncpy (vl->plugin, PLUGIN_NAME, sizeof (vl->plugin));
 162
 163     vl->host[0] = '\0';
 164
 165     /* Construct the hostname field according to HostnameFormat. */
 166     for (int i = 0; i < HF_MAX_FIELDS; ++i) {
 167         if (hostname_format[i] == hf_none)
 168             continue;
 169
 170         n = DATA_MAX_NAME_LEN - strlen (vl->host) - 2;
 171
 172         if (i > 0 && n >= 1) {
 173             strncat (vl->host, ":", 1);
 174             n--;
 175         }
 176
 177         switch (hostname_format[i]) {
 178         case hf_none: break;
 179         case hf_hostname:
 180             strncat (vl->host, hostname_g, n);
 181             break;
 182         case hf_name:
 183             name = virDomainGetName (dom);
 184             if (name)
 185                 strncat (vl->host, name, n);
 186             break;
 187         case hf_uuid:
 188             if (virDomainGetUUIDString (dom, uuid) == 0)
 189                 strncat (vl->host, uuid, n);
 190             break;
 191         }
 192     }
 193
 194     vl->host[sizeof (vl->host) - 1] = '\0';
 195
 196     /* Construct the plugin instance field according to PluginInstanceFormat. */
 197     for (int i = 0; i < PLGINST_MAX_FIELDS; ++i) {
 198         if (plugin_instance_format[i] == plginst_none)
 199             continue;
 200
 201         n = sizeof(vl->plugin_instance) - strlen (vl->plugin_instance) - 2;
 202
 203         if (i > 0 && n >= 1) {
 204             strncat (vl->plugin_instance, ":", 1);
 205             n--;
 206         }
 207
 208         switch (plugin_instance_format[i]) {
 209         case plginst_none: break;
 210         case plginst_name:
 211             name = virDomainGetName (dom);
 212             if (name)
 213                 strncat (vl->plugin_instance, name, n);
 214             break;
 215         case plginst_uuid:
 216             if (virDomainGetUUIDString (dom, uuid) == 0)
 217                 strncat (vl->plugin_instance, uuid, n);
 218             break;
 219         }
 220     }
 221
 222     vl->plugin_instance[sizeof (vl->plugin_instance) - 1] = '\0';
 223
 224 } /* void init_value_list */
 225
 226 static void
 227 memory_submit (gauge_t memory, virDomainPtr dom)
 228 {
 229     value_t values[1];
 230     value_list_t vl = VALUE_LIST_INIT;
 231
 232     init_value_list (&vl, dom);
 233
 234     values[0].gauge = memory;
 235
 236     vl.values = values;
 237     vl.values_len = 1;
 238
 239     sstrncpy (vl.type, "memory", sizeof (vl.type));
 240     sstrncpy (vl.type_instance, "total", sizeof (vl.type_instance));
 241
 242     plugin_dispatch_values (&vl);
 243 }
 244
 245 static void
 246 memory_stats_submit (gauge_t memory, virDomainPtr dom, int tag_index)
 247 {
 248     static const char *tags[] = { "swap_in", "swap_out", "major_fault", "minor_fault",
 249                                     "unused", "available", "actual_balloon", "rss"};
 250
 251     value_t values[1];
 252     value_list_t vl = VALUE_LIST_INIT;
 253
 254     init_value_list (&vl, dom);
 255
 256     values[0].gauge = memory;
 257
 258     vl.values = values;
 259     vl.values_len = 1;
 260
 261     sstrncpy (vl.type, "memory", sizeof (vl.type));
 262     sstrncpy (vl.type_instance, tags[tag_index], sizeof (vl.type_instance));
 263
 264     plugin_dispatch_values (&vl);
 265 }
 266
 267 static void
 268 cpu_submit (unsigned long long cpu_time,
 269             virDomainPtr dom, const char *type)
 270 {
 271     value_t values[1];
 272     value_list_t vl = VALUE_LIST_INIT;
 273
 274     init_value_list (&vl, dom);
 275
 276     values[0].derive = cpu_time;
 277
 278     vl.values = values;
 279     vl.values_len = 1;
 280
 281     sstrncpy (vl.type, type, sizeof (vl.type));
 282
 283     plugin_dispatch_values (&vl);
 284 }
 285
 286 static void
 287 vcpu_submit (derive_t cpu_time,
 288              virDomainPtr dom, int vcpu_nr, const char *type)
 289 {
 290     value_t values[1];
 291     value_list_t vl = VALUE_LIST_INIT;
 292
 293     init_value_list (&vl, dom);
 294
 295     values[0].derive = cpu_time;
 296     vl.values = values;
 297     vl.values_len = 1;
 298
 299     sstrncpy (vl.type, type, sizeof (vl.type));
 300     ssnprintf (vl.type_instance, sizeof (vl.type_instance), "%d", vcpu_nr);
 301
 302     plugin_dispatch_values (&vl);
 303 }
 304
 305 static void
 306 submit_derive2 (const char *type, derive_t v0, derive_t v1,
 307              virDomainPtr dom, const char *devname)
 308 {
 309     value_t values[2];
 310     value_list_t vl = VALUE_LIST_INIT;
 311
 312     init_value_list (&vl, dom);
 313
 314     values[0].derive = v0;
 315     values[1].derive = v1;
 316     vl.values = values;
 317     vl.values_len = 2;
 318
 319     sstrncpy (vl.type, type, sizeof (vl.type));
 320     sstrncpy (vl.type_instance, devname, sizeof (vl.type_instance));
 321
 322     plugin_dispatch_values (&vl);
 323 } /* void submit_derive2 */
 324
 325 static int
 326 lv_init (void)
 327 {
 328     if (virInitialize () != 0)
 329         return -1;
 330     else
 331         return 0;
 332 }
 333
 334 static int
 335 lv_config (const char *key, const char *value)
 336 {
 337     if (virInitialize () != 0)
 338         return 1;
 339
 340     if (il_domains == NULL)
 341         il_domains = ignorelist_create (1);
 342     if (il_block_devices == NULL)
 343         il_block_devices = ignorelist_create (1);
 344     if (il_interface_devices == NULL)
 345         il_interface_devices = ignorelist_create (1);
 346
 347     if (strcasecmp (key, "Connection") == 0) {
 348         char *tmp = strdup (value);
 349         if (tmp == NULL) {
 350             ERROR (PLUGIN_NAME " plugin: Connection strdup failed.");
 351             return 1;
 352         }
 353         sfree (conn_string);
 354         conn_string = tmp;
 355         return 0;
 356     }
 357
 358     if (strcasecmp (key, "RefreshInterval") == 0) {
 359         char *eptr = NULL;
 360         interval = strtol (value, &eptr, 10);
 361         if (eptr == NULL || *eptr != '\0') return 1;
 362         return 0;
 363     }
 364
 365     if (strcasecmp (key, "Domain") == 0) {
 366         if (ignorelist_add (il_domains, value)) return 1;
 367         return 0;
 368     }
 369     if (strcasecmp (key, "BlockDevice") == 0) {
 370         if (ignorelist_add (il_block_devices, value)) return 1;
 371         return 0;
 372     }
 373     if (strcasecmp (key, "InterfaceDevice") == 0) {
 374         if (ignorelist_add (il_interface_devices, value)) return 1;
 375         return 0;
 376     }
 377
 378     if (strcasecmp (key, "IgnoreSelected") == 0) {
 379         if (IS_TRUE (value))
 380         {
 381             ignorelist_set_invert (il_domains, 0);
 382             ignorelist_set_invert (il_block_devices, 0);
 383             ignorelist_set_invert (il_interface_devices, 0);
 384         }
 385         else
 386         {
 387             ignorelist_set_invert (il_domains, 1);
 388             ignorelist_set_invert (il_block_devices, 1);
 389             ignorelist_set_invert (il_interface_devices, 1);
 390         }
 391         return 0;
 392     }
 393
 394     if (strcasecmp (key, "HostnameFormat") == 0) {
 395         char *value_copy;
 396         char *fields[HF_MAX_FIELDS];
 397         int n;
 398
 399         value_copy = strdup (value);
 400         if (value_copy == NULL) {
 401             ERROR (PLUGIN_NAME " plugin: strdup failed.");
 402             return -1;
 403         }
 404
 405         n = strsplit (value_copy, fields, HF_MAX_FIELDS);
 406         if (n < 1) {
 407             sfree (value_copy);
 408             ERROR (PLUGIN_NAME " plugin: HostnameFormat: no fields");
 409             return -1;
 410         }
 411
 412         for (int i = 0; i < n; ++i) {
 413             if (strcasecmp (fields[i], "hostname") == 0)
 414                 hostname_format[i] = hf_hostname;
 415             else if (strcasecmp (fields[i], "name") == 0)
 416                 hostname_format[i] = hf_name;
 417             else if (strcasecmp (fields[i], "uuid") == 0)
 418                 hostname_format[i] = hf_uuid;
 419             else {
 420                 ERROR (PLUGIN_NAME " plugin: unknown HostnameFormat field: %s", fields[i]);
 421                 sfree (value_copy);
 422                 return -1;
 423             }
 424         }
 425         sfree (value_copy);
 426
 427         for (int i = n; i < HF_MAX_FIELDS; ++i)
 428             hostname_format[i] = hf_none;
 429
 430         return 0;
 431     }
 432
 433     if (strcasecmp (key, "PluginInstanceFormat") == 0) {
 434         char *value_copy;
 435         char *fields[PLGINST_MAX_FIELDS];
 436         int n;
 437
 438         value_copy = strdup (value);
 439         if (value_copy == NULL) {
 440             ERROR (PLUGIN_NAME " plugin: strdup failed.");
 441             return -1;
 442         }
 443
 444         n = strsplit (value_copy, fields, PLGINST_MAX_FIELDS);
 445         if (n < 1) {
 446             sfree (value_copy);
 447             ERROR (PLUGIN_NAME " plugin: PluginInstanceFormat: no fields");
 448             return -1;
 449         }
 450
 451         for (int i = 0; i < n; ++i) {
 452             if (strcasecmp (fields[i], "none") == 0) {
 453                 plugin_instance_format[i] = plginst_none;
 454                 break;
 455             } else if (strcasecmp (fields[i], "name") == 0)
 456                 plugin_instance_format[i] = plginst_name;
 457             else if (strcasecmp (fields[i], "uuid") == 0)
 458                 plugin_instance_format[i] = plginst_uuid;
 459             else {
 460                 ERROR (PLUGIN_NAME " plugin: unknown PluginInstanceFormat field: %s", fields[i]);
 461                 sfree (value_copy);
 462                 return -1;
 463             }
 464         }
 465         sfree (value_copy);
 466
 467         for (int i = n; i < PLGINST_MAX_FIELDS; ++i)
 468             plugin_instance_format[i] = plginst_none;
 469
 470         return 0;
 471     }
 472
 473     if (strcasecmp (key, "InterfaceFormat") == 0) {
 474         if (strcasecmp (value, "name") == 0)
 475             interface_format = if_name;
 476         else if (strcasecmp (value, "address") == 0)
 477             interface_format = if_address;
 478         else if (strcasecmp (value, "number") == 0)
 479             interface_format = if_number;
 480         else {
 481             ERROR (PLUGIN_NAME " plugin: unknown InterfaceFormat: %s", value);
 482             return -1;
 483         }
 484         return 0;
 485     }
 486
 487     /* Unrecognised option. */
 488     return -1;
 489 }
 490
 491 static int
 492 lv_read (void)
 493 {
 494     time_t t;
 495
 496     if (conn == NULL) {
 497         /* `conn_string == NULL' is acceptable. */
 498         conn = virConnectOpenReadOnly (conn_string);
 499         if (conn == NULL) {
 500             c_complain (LOG_ERR, &conn_complain,
 501                     PLUGIN_NAME " plugin: Unable to connect: "
 502                     "virConnectOpenReadOnly failed.");
 503             return -1;
 504         }
 505     }
 506     c_release (LOG_NOTICE, &conn_complain,
 507             PLUGIN_NAME " plugin: Connection established.");
 508
 509     time (&t);
 510
 511     /* Need to refresh domain or device lists? */
 512     if ((last_refresh == (time_t) 0) ||
 513             ((interval > 0) && ((last_refresh + interval) <= t))) {
 514         if (refresh_lists () != 0) {
 515             if (conn != NULL)
 516                 virConnectClose (conn);
 517             conn = NULL;
 518             return -1;
 519         }
 520         last_refresh = t;
 521     }
 522
 523 #if 0
 524     for (int i = 0; i < nr_domains; ++i)
 525         fprintf (stderr, "domain %s\n", virDomainGetName (domains[i]));
 526     for (int i = 0; i < nr_block_devices; ++i)
 527         fprintf  (stderr, "block device %d %s:%s\n",
 528                   i, virDomainGetName (block_devices[i].dom),
 529                   block_devices[i].path);
 530     for (int i = 0; i < nr_interface_devices; ++i)
 531         fprintf (stderr, "interface device %d %s:%s\n",
 532                  i, virDomainGetName (interface_devices[i].dom),
 533                  interface_devices[i].path);
 534 #endif
 535
 536     /* Get CPU usage, memory, VCPU usage for each domain. */
 537     for (int i = 0; i < nr_domains; ++i) {
 538         virDomainInfo info;
 539         virVcpuInfoPtr vinfo = NULL;
 540         virDomainMemoryStatPtr minfo = NULL;
 541         int status;
 542
 543         status = virDomainGetInfo (domains[i], &info);
 544         if (status != 0)
 545         {
 546             ERROR (PLUGIN_NAME " plugin: virDomainGetInfo failed with status %i.",
 547                     status);
 548             continue;
 549         }
 550
 551         if (info.state != VIR_DOMAIN_RUNNING)
 552         {
 553             /* only gather stats for running domains */
 554             continue;
 555         }
 556
 557         cpu_submit (info.cpuTime, domains[i], "virt_cpu_total");
 558         memory_submit ((gauge_t) info.memory * 1024, domains[i]);
 559
 560         vinfo = malloc (info.nrVirtCpu * sizeof (vinfo[0]));
 561         if (vinfo == NULL) {
 562             ERROR (PLUGIN_NAME " plugin: malloc failed.");
 563             continue;
 564         }
 565
 566         status = virDomainGetVcpus (domains[i], vinfo, info.nrVirtCpu,
 567                 /* cpu map = */ NULL, /* cpu map length = */ 0);
 568         if (status < 0)
 569         {
 570             ERROR (PLUGIN_NAME " plugin: virDomainGetVcpus failed with status %i.",
 571                     status);
 572             sfree (vinfo);
 573             continue;
 574         }
 575
 576         for (int j = 0; j < info.nrVirtCpu; ++j)
 577             vcpu_submit (vinfo[j].cpuTime,
 578                     domains[i], vinfo[j].number, "virt_vcpu");
 579
 580         sfree (vinfo);
 581
 582         minfo = malloc (VIR_DOMAIN_MEMORY_STAT_NR * sizeof (virDomainMemoryStatStruct));
 583         if (minfo == NULL) {
 584             ERROR ("virt plugin: malloc failed.");
 585             continue;
 586         }
 587
 588         status = virDomainMemoryStats (domains[i], minfo, VIR_DOMAIN_MEMORY_STAT_NR, 0);
 589
 590         if (status < 0) {
 591             ERROR ("virt plugin: virDomainMemoryStats failed with status %i.",
 592                     status);
 593             sfree (minfo);
 594             continue;
 595         }
 596
 597         for (int j = 0; j < status; j++) {
 598             memory_stats_submit ((gauge_t) minfo[j].val * 1024, domains[i], minfo[j].tag);
 599         }
 600
 601         sfree (minfo);
 602     }
 603
 604
 605     /* Get block device stats for each domain. */
 606     for (int i = 0; i < nr_block_devices; ++i) {
 607         struct _virDomainBlockStats stats;
 608
 609         if (virDomainBlockStats (block_devices[i].dom, block_devices[i].path,
 610                     &stats, sizeof stats) != 0)
 611             continue;
 612
 613         if ((stats.rd_req != -1) && (stats.wr_req != -1))
 614             submit_derive2 ("disk_ops",
 615                     (derive_t) stats.rd_req, (derive_t) stats.wr_req,
 616                     block_devices[i].dom, block_devices[i].path);
 617
 618         if ((stats.rd_bytes != -1) && (stats.wr_bytes != -1))
 619             submit_derive2 ("disk_octets",
 620                     (derive_t) stats.rd_bytes, (derive_t) stats.wr_bytes,
 621                     block_devices[i].dom, block_devices[i].path);
 622     } /* for (nr_block_devices) */
 623
 624     /* Get interface stats for each domain. */
 625     for (int i = 0; i < nr_interface_devices; ++i) {
 626         struct _virDomainInterfaceStats stats;
 627         char *display_name = NULL;
 628
 629
 630         switch (interface_format) {
 631             case if_address:
 632                 display_name = interface_devices[i].address;
 633                 break;
 634             case if_number:
 635                 display_name = interface_devices[i].number;
 636                 break;
 637             case if_name:
 638             default:
 639                 display_name = interface_devices[i].path;
 640         }
 641
 642         if (virDomainInterfaceStats (interface_devices[i].dom,
 643                     interface_devices[i].path,
 644                     &stats, sizeof stats) != 0)
 645             continue;
 646
 647         if ((stats.rx_bytes != -1) && (stats.tx_bytes != -1))
 648             submit_derive2 ("if_octets",
 649                     (derive_t) stats.rx_bytes, (derive_t) stats.tx_bytes,
 650                     interface_devices[i].dom, display_name);
 651
 652         if ((stats.rx_packets != -1) && (stats.tx_packets != -1))
 653             submit_derive2 ("if_packets",
 654                     (derive_t) stats.rx_packets, (derive_t) stats.tx_packets,
 655                     interface_devices[i].dom, display_name);
 656
 657         if ((stats.rx_errs != -1) && (stats.tx_errs != -1))
 658             submit_derive2 ("if_errors",
 659                     (derive_t) stats.rx_errs, (derive_t) stats.tx_errs,
 660                     interface_devices[i].dom, display_name);
 661
 662         if ((stats.rx_drop != -1) && (stats.tx_drop != -1))
 663             submit_derive2 ("if_dropped",
 664                     (derive_t) stats.rx_drop, (derive_t) stats.tx_drop,
 665                     interface_devices[i].dom, display_name);
 666     } /* for (nr_interface_devices) */
 667
 668     return 0;
 669 }
 670
 671 static int
 672 refresh_lists (void)
 673 {
 674     int n;
 675
 676     n = virConnectNumOfDomains (conn);
 677     if (n < 0) {
 678         VIRT_ERROR (conn, "reading number of domains");
 679         return -1;
 680     }
 681
 682     if (n > 0) {
 683         int *domids;
 684
 685         /* Get list of domains. */
 686         domids = malloc (sizeof (*domids) * n);
 687         if (domids == NULL) {
 688             ERROR (PLUGIN_NAME " plugin: malloc failed.");
 689             return -1;
 690         }
 691
 692         n = virConnectListDomains (conn, domids, n);
 693         if (n < 0) {
 694             VIRT_ERROR (conn, "reading list of domains");
 695             sfree (domids);
 696             return -1;
 697         }
 698
 699         free_block_devices ();
 700         free_interface_devices ();
 701         free_domains ();
 702
 703         /* Fetch each domain and add it to the list, unless ignore. */
 704         for (int i = 0; i < n; ++i) {
 705             virDomainPtr dom = NULL;
 706             const char *name;
 707             char *xml = NULL;
 708             xmlDocPtr xml_doc = NULL;
 709             xmlXPathContextPtr xpath_ctx = NULL;
 710             xmlXPathObjectPtr xpath_obj = NULL;
 711
 712             dom = virDomainLookupByID (conn, domids[i]);
 713             if (dom == NULL) {
 714                 VIRT_ERROR (conn, "virDomainLookupByID");
 715                 /* Could be that the domain went away -- ignore it anyway. */
 716                 continue;
 717             }
 718
 719             name = virDomainGetName (dom);
 720             if (name == NULL) {
 721                 VIRT_ERROR (conn, "virDomainGetName");
 722                 goto cont;
 723             }
 724
 725             if (il_domains && ignorelist_match (il_domains, name) != 0)
 726                 goto cont;
 727
 728             if (add_domain (dom) < 0) {
 729                 ERROR (PLUGIN_NAME " plugin: malloc failed.");
 730                 goto cont;
 731             }
 732
 733             /* Get a list of devices for this domain. */
 734             xml = virDomainGetXMLDesc (dom, 0);
 735             if (!xml) {
 736                 VIRT_ERROR (conn, "virDomainGetXMLDesc");
 737                 goto cont;
 738             }
 739
 740             /* Yuck, XML.  Parse out the devices. */
 741             xml_doc = xmlReadDoc ((xmlChar *) xml, NULL, NULL, XML_PARSE_NONET);
 742             if (xml_doc == NULL) {
 743                 VIRT_ERROR (conn, "xmlReadDoc");
 744                 goto cont;
 745             }
 746
 747             xpath_ctx = xmlXPathNewContext (xml_doc);
 748
 749             /* Block devices. */
 750             xpath_obj = xmlXPathEval
 751                 ((xmlChar *) "/domain/devices/disk/target[@dev]",
 752                  xpath_ctx);
 753             if (xpath_obj == NULL || xpath_obj->type != XPATH_NODESET ||
 754                 xpath_obj->nodesetval == NULL)
 755                 goto cont;
 756
 757             for (int j = 0; j < xpath_obj->nodesetval->nodeNr; ++j) {
 758                 xmlNodePtr node;
 759                 char *path = NULL;
 760
 761                 node = xpath_obj->nodesetval->nodeTab[j];
 762                 if (!node) continue;
 763                 path = (char *) xmlGetProp (node, (xmlChar *) "dev");
 764                 if (!path) continue;
 765
 766                 if (il_block_devices &&
 767                     ignore_device_match (il_block_devices, name, path) != 0)
 768                     goto cont2;
 769
 770                 add_block_device (dom, path);
 771             cont2:
 772                 if (path) xmlFree (path);
 773             }
 774             xmlXPathFreeObject (xpath_obj);
 775
 776             /* Network interfaces. */
 777             xpath_obj = xmlXPathEval
 778                 ((xmlChar *) "/domain/devices/interface[target[@dev]]",
 779                  xpath_ctx);
 780             if (xpath_obj == NULL || xpath_obj->type != XPATH_NODESET ||
 781                 xpath_obj->nodesetval == NULL)
 782                 goto cont;
 783
 784             xmlNodeSetPtr xml_interfaces = xpath_obj->nodesetval;
 785
 786             for (int j = 0; j < xml_interfaces->nodeNr; ++j) {
 787                 char *path = NULL;
 788                 char *address = NULL;
 789                 xmlNodePtr xml_interface;
 790
 791                 xml_interface = xml_interfaces->nodeTab[j];
 792                 if (!xml_interface) continue;
 793
 794                 for (xmlNodePtr child = xml_interface->children; child; child = child->next) {
 795                     if (child->type != XML_ELEMENT_NODE) continue;
 796
 797                     if (xmlStrEqual(child->name, (const xmlChar *) "target")) {
 798                         path = (char *) xmlGetProp (child, (const xmlChar *) "dev");
 799                         if (!path) continue;
 800                     } else if (xmlStrEqual(child->name, (const xmlChar *) "mac")) {
 801                         address = (char *) xmlGetProp (child, (const xmlChar *) "address");
 802                         if (!address) continue;
 803                     }
 804                 }
 805
 806                 if (il_interface_devices &&
 807                     (ignore_device_match (il_interface_devices, name, path) != 0 ||
 808                      ignore_device_match (il_interface_devices, name, address) != 0))
 809                     goto cont3;
 810
 811                 add_interface_device (dom, path, address, j+1);
 812                 cont3:
 813                     if (path) xmlFree (path);
 814                     if (address) xmlFree (address);
 815             }
 816
 817         cont:
 818             if (xpath_obj) xmlXPathFreeObject (xpath_obj);
 819             if (xpath_ctx) xmlXPathFreeContext (xpath_ctx);
 820             if (xml_doc) xmlFreeDoc (xml_doc);
 821             sfree (xml);
 822         }
 823
 824         sfree (domids);
 825     }
 826
 827     return 0;
 828 }
 829
 830 static void
 831 free_domains (void)
 832 {
 833     if (domains) {
 834         for (int i = 0; i < nr_domains; ++i)
 835             virDomainFree (domains[i]);
 836         sfree (domains);
 837     }
 838     domains = NULL;
 839     nr_domains = 0;
 840 }
 841
 842 static int
 843 add_domain (virDomainPtr dom)
 844 {
 845     virDomainPtr *new_ptr;
 846     int new_size = sizeof (domains[0]) * (nr_domains+1);
 847
 848     if (domains)
 849         new_ptr = realloc (domains, new_size);
 850     else
 851         new_ptr = malloc (new_size);
 852
 853     if (new_ptr == NULL)
 854         return -1;
 855
 856     domains = new_ptr;
 857     domains[nr_domains] = dom;
 858     return nr_domains++;
 859 }
 860
 861 static void
 862 free_block_devices (void)
 863 {
 864     if (block_devices) {
 865         for (int i = 0; i < nr_block_devices; ++i)
 866             sfree (block_devices[i].path);
 867         sfree (block_devices);
 868     }
 869     block_devices = NULL;
 870     nr_block_devices = 0;
 871 }
 872
 873 static int
 874 add_block_device (virDomainPtr dom, const char *path)
 875 {
 876     struct block_device *new_ptr;
 877     int new_size = sizeof (block_devices[0]) * (nr_block_devices+1);
 878     char *path_copy;
 879
 880     path_copy = strdup (path);
 881     if (!path_copy)
 882         return -1;
 883
 884     if (block_devices)
 885         new_ptr = realloc (block_devices, new_size);
 886     else
 887         new_ptr = malloc (new_size);
 888
 889     if (new_ptr == NULL) {
 890         sfree (path_copy);
 891         return -1;
 892     }
 893     block_devices = new_ptr;
 894     block_devices[nr_block_devices].dom = dom;
 895     block_devices[nr_block_devices].path = path_copy;
 896     return nr_block_devices++;
 897 }
 898
 899 static void
 900 free_interface_devices (void)
 901 {
 902     if (interface_devices) {
 903         for (int i = 0; i < nr_interface_devices; ++i) {
 904             sfree (interface_devices[i].path);
 905             sfree (interface_devices[i].address);
 906             sfree (interface_devices[i].number);
 907         }
 908         sfree (interface_devices);
 909     }
 910     interface_devices = NULL;
 911     nr_interface_devices = 0;
 912 }
 913
 914 static int
 915 add_interface_device (virDomainPtr dom, const char *path, const char *address, unsigned int number)
 916 {
 917     struct interface_device *new_ptr;
 918     int new_size = sizeof (interface_devices[0]) * (nr_interface_devices+1);
 919     char *path_copy, *address_copy, number_string[15];
 920
 921     if ((path == NULL) || (address == NULL))
 922         return EINVAL;
 923
 924     path_copy = strdup (path);
 925     if (!path_copy) return -1;
 926
 927     address_copy = strdup (address);
 928     if (!address_copy) {
 929         sfree(path_copy);
 930         return -1;
 931     }
 932
 933     snprintf(number_string, sizeof (number_string), "interface-%u", number);
 934
 935     if (interface_devices)
 936         new_ptr = realloc (interface_devices, new_size);
 937     else
 938         new_ptr = malloc (new_size);
 939
 940     if (new_ptr == NULL) {
 941         sfree (path_copy);
 942         sfree (address_copy);
 943         return -1;
 944     }
 945     interface_devices = new_ptr;
 946     interface_devices[nr_interface_devices].dom = dom;
 947     interface_devices[nr_interface_devices].path = path_copy;
 948     interface_devices[nr_interface_devices].address = address_copy;
 949     interface_devices[nr_interface_devices].number = strdup(number_string);
 950     return nr_interface_devices++;
 951 }
 952
 953 static int
 954 ignore_device_match (ignorelist_t *il, const char *domname, const char *devpath)
 955 {
 956     char *name;
 957     int n, r;
 958
 959     if ((domname == NULL) || (devpath == NULL))
 960         return 0;
 961
 962     n = sizeof (char) * (strlen (domname) + strlen (devpath) + 2);
 963     name = malloc (n);
 964     if (name == NULL) {
 965         ERROR (PLUGIN_NAME " plugin: malloc failed.");
 966         return 0;
 967     }
 968     ssnprintf (name, n, "%s:%s", domname, devpath);
 969     r = ignorelist_match (il, name);
 970     sfree (name);
 971     return r;
 972 }
 973
 974 static int
 975 lv_shutdown (void)
 976 {
 977     free_block_devices ();
 978     free_interface_devices ();
 979     free_domains ();
 980
 981     if (conn != NULL)
 982         virConnectClose (conn);
 983     conn = NULL;
 984
 985     ignorelist_free (il_domains);
 986     il_domains = NULL;
 987     ignorelist_free (il_block_devices);
 988     il_block_devices = NULL;
 989     ignorelist_free (il_interface_devices);
 990     il_interface_devices = NULL;
 991
 992     return 0;
 993 }
 994
 995 void
 996 module_register (void)
 997 {
 998     plugin_register_config (PLUGIN_NAME,
 999     lv_config,
1000     config_keys, NR_CONFIG_KEYS);
1001     plugin_register_init (PLUGIN_NAME, lv_init);
1002     plugin_register_read (PLUGIN_NAME, lv_read);
1003     plugin_register_shutdown (PLUGIN_NAME, lv_shutdown);
1004 }
1005
1006 /*
1007  * vim: shiftwidth=4 tabstop=8 softtabstop=4 expandtab fdm=marker
1008  */