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 #include "common.h"
  24 #include "plugin.h"
  25 #include "configfile.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_name };
 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 i, 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 (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 (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
 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 i, 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 (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                 sfree (value_copy);
 421                 ERROR (PLUGIN_NAME " plugin: unknown HostnameFormat field: %s", fields[i]);
 422                 return -1;
 423             }
 424         }
 425         sfree (value_copy);
 426
 427         for (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 i, 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 (i = 0; i < n; ++i) {
 452             if (strcasecmp (fields[i], "name") == 0)
 453                 plugin_instance_format[i] = plginst_name;
 454             else if (strcasecmp (fields[i], "uuid") == 0)
 455                 plugin_instance_format[i] = plginst_uuid;
 456             else {
 457                 sfree (value_copy);
 458                 ERROR (PLUGIN_NAME " plugin: unknown HostnameFormat field: %s", fields[i]);
 459                 return -1;
 460             }
 461         }
 462         sfree (value_copy);
 463
 464         for (i = n; i < PLGINST_MAX_FIELDS; ++i)
 465             plugin_instance_format[i] = plginst_none;
 466
 467         return 0;
 468     }
 469
 470     if (strcasecmp (key, "InterfaceFormat") == 0) {
 471         if (strcasecmp (value, "name") == 0)
 472             interface_format = if_name;
 473         else if (strcasecmp (value, "address") == 0)
 474             interface_format = if_address;
 475         else if (strcasecmp (value, "number") == 0)
 476             interface_format = if_number;
 477         else {
 478             ERROR (PLUGIN_NAME " plugin: unknown InterfaceFormat: %s", value);
 479             return -1;
 480         }
 481         return 0;
 482     }
 483
 484     /* Unrecognised option. */
 485     return -1;
 486 }
 487
 488 static int
 489 lv_read (void)
 490 {
 491     time_t t;
 492     int i;
 493
 494     if (conn == NULL) {
 495         /* `conn_string == NULL' is acceptable. */
 496         conn = virConnectOpenReadOnly (conn_string);
 497         if (conn == NULL) {
 498             c_complain (LOG_ERR, &conn_complain,
 499                     PLUGIN_NAME " plugin: Unable to connect: "
 500                     "virConnectOpenReadOnly failed.");
 501             return -1;
 502         }
 503     }
 504     c_release (LOG_NOTICE, &conn_complain,
 505             PLUGIN_NAME " plugin: Connection established.");
 506
 507     time (&t);
 508
 509     /* Need to refresh domain or device lists? */
 510     if ((last_refresh == (time_t) 0) ||
 511             ((interval > 0) && ((last_refresh + interval) <= t))) {
 512         if (refresh_lists () != 0) {
 513             if (conn != NULL)
 514                 virConnectClose (conn);
 515             conn = NULL;
 516             return -1;
 517         }
 518         last_refresh = t;
 519     }
 520
 521 #if 0
 522     for (i = 0; i < nr_domains; ++i)
 523         fprintf (stderr, "domain %s\n", virDomainGetName (domains[i]));
 524     for (i = 0; i < nr_block_devices; ++i)
 525         fprintf  (stderr, "block device %d %s:%s\n",
 526                   i, virDomainGetName (block_devices[i].dom),
 527                   block_devices[i].path);
 528     for (i = 0; i < nr_interface_devices; ++i)
 529         fprintf (stderr, "interface device %d %s:%s\n",
 530                  i, virDomainGetName (interface_devices[i].dom),
 531                  interface_devices[i].path);
 532 #endif
 533
 534     /* Get CPU usage, memory, VCPU usage for each domain. */
 535     for (i = 0; i < nr_domains; ++i) {
 536         virDomainInfo info;
 537         virVcpuInfoPtr vinfo = NULL;
 538         virDomainMemoryStatPtr minfo = NULL;
 539         int status;
 540         int j;
 541
 542         status = virDomainGetInfo (domains[i], &info);
 543         if (status != 0)
 544         {
 545             ERROR (PLUGIN_NAME " plugin: virDomainGetInfo failed with status %i.",
 546                     status);
 547             continue;
 548         }
 549
 550         if (info.state != VIR_DOMAIN_RUNNING)
 551         {
 552             /* only gather stats for running domains */
 553             continue;
 554         }
 555
 556         cpu_submit (info.cpuTime, domains[i], "virt_cpu_total");
 557         memory_submit ((gauge_t) info.memory * 1024, domains[i]);
 558
 559         vinfo = malloc (info.nrVirtCpu * sizeof (vinfo[0]));
 560         if (vinfo == NULL) {
 561             ERROR (PLUGIN_NAME " plugin: malloc failed.");
 562             continue;
 563         }
 564
 565         status = virDomainGetVcpus (domains[i], vinfo, info.nrVirtCpu,
 566                 /* cpu map = */ NULL, /* cpu map length = */ 0);
 567         if (status < 0)
 568         {
 569             ERROR (PLUGIN_NAME " plugin: virDomainGetVcpus failed with status %i.",
 570                     status);
 571             sfree (vinfo);
 572             continue;
 573         }
 574
 575         for (j = 0; j < info.nrVirtCpu; ++j)
 576             vcpu_submit (vinfo[j].cpuTime,
 577                     domains[i], vinfo[j].number, "virt_vcpu");
 578
 579         sfree (vinfo);
 580
 581         minfo = malloc (VIR_DOMAIN_MEMORY_STAT_NR * sizeof (virDomainMemoryStatStruct));
 582         if (minfo == NULL) {
 583             ERROR ("virt plugin: malloc failed.");
 584             continue;
 585         }
 586
 587         status =  virDomainMemoryStats (domains[i], minfo, VIR_DOMAIN_MEMORY_STAT_NR, 0);
 588
 589         if (status < 0) {
 590             ERROR ("virt plugin: virDomainMemoryStats failed with status %i.",
 591                     status);
 592             sfree (minfo);
 593             continue;
 594         }
 595
 596         for (j = 0; j < status; j++) {
 597             memory_stats_submit ((gauge_t) minfo[j].val * 1024, domains[i], minfo[j].tag);
 598         }
 599
 600         sfree (minfo);
 601     }
 602
 603
 604     /* Get block device stats for each domain. */
 605     for (i = 0; i < nr_block_devices; ++i) {
 606         struct _virDomainBlockStats stats;
 607
 608         if (virDomainBlockStats (block_devices[i].dom, block_devices[i].path,
 609                     &stats, sizeof stats) != 0)
 610             continue;
 611
 612         if ((stats.rd_req != -1) && (stats.wr_req != -1))
 613             submit_derive2 ("disk_ops",
 614                     (derive_t) stats.rd_req, (derive_t) stats.wr_req,
 615                     block_devices[i].dom, block_devices[i].path);
 616
 617         if ((stats.rd_bytes != -1) && (stats.wr_bytes != -1))
 618             submit_derive2 ("disk_octets",
 619                     (derive_t) stats.rd_bytes, (derive_t) stats.wr_bytes,
 620                     block_devices[i].dom, block_devices[i].path);
 621     } /* for (nr_block_devices) */
 622
 623     /* Get interface stats for each domain. */
 624     for (i = 0; i < nr_interface_devices; ++i) {
 625         struct _virDomainInterfaceStats stats;
 626         char *display_name = NULL;
 627
 628
 629         switch (interface_format) {
 630             case if_address:
 631                 display_name = interface_devices[i].address;
 632                 break;
 633             case if_number:
 634                 display_name = interface_devices[i].number;
 635                 break;
 636             case if_name:
 637             default:
 638                 display_name = interface_devices[i].path;
 639         }
 640
 641         if (virDomainInterfaceStats (interface_devices[i].dom,
 642                     interface_devices[i].path,
 643                     &stats, sizeof stats) != 0)
 644             continue;
 645
 646         if ((stats.rx_bytes != -1) && (stats.tx_bytes != -1))
 647             submit_derive2 ("if_octets",
 648                     (derive_t) stats.rx_bytes, (derive_t) stats.tx_bytes,
 649                     interface_devices[i].dom, display_name);
 650
 651         if ((stats.rx_packets != -1) && (stats.tx_packets != -1))
 652             submit_derive2 ("if_packets",
 653                     (derive_t) stats.rx_packets, (derive_t) stats.tx_packets,
 654                     interface_devices[i].dom, display_name);
 655
 656         if ((stats.rx_errs != -1) && (stats.tx_errs != -1))
 657             submit_derive2 ("if_errors",
 658                     (derive_t) stats.rx_errs, (derive_t) stats.tx_errs,
 659                     interface_devices[i].dom, display_name);
 660
 661         if ((stats.rx_drop != -1) && (stats.tx_drop != -1))
 662             submit_derive2 ("if_dropped",
 663                     (derive_t) stats.rx_drop, (derive_t) stats.tx_drop,
 664                     interface_devices[i].dom, display_name);
 665     } /* for (nr_interface_devices) */
 666
 667     return 0;
 668 }
 669
 670 static int
 671 refresh_lists (void)
 672 {
 673     int n;
 674
 675     n = virConnectNumOfDomains (conn);
 676     if (n < 0) {
 677         VIRT_ERROR (conn, "reading number of domains");
 678         return -1;
 679     }
 680
 681     if (n > 0) {
 682         int i;
 683         int *domids;
 684
 685         /* Get list of domains. */
 686         domids = malloc (sizeof (int) * n);
 687         if (domids == 0) {
 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 (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             int j;
 712
 713             dom = virDomainLookupByID (conn, domids[i]);
 714             if (dom == NULL) {
 715                 VIRT_ERROR (conn, "virDomainLookupByID");
 716                 /* Could be that the domain went away -- ignore it anyway. */
 717                 continue;
 718             }
 719
 720             name = virDomainGetName (dom);
 721             if (name == NULL) {
 722                 VIRT_ERROR (conn, "virDomainGetName");
 723                 goto cont;
 724             }
 725
 726             if (il_domains && ignorelist_match (il_domains, name) != 0)
 727                 goto cont;
 728
 729             if (add_domain (dom) < 0) {
 730                 ERROR (PLUGIN_NAME " plugin: malloc failed.");
 731                 goto cont;
 732             }
 733
 734             /* Get a list of devices for this domain. */
 735             xml = virDomainGetXMLDesc (dom, 0);
 736             if (!xml) {
 737                 VIRT_ERROR (conn, "virDomainGetXMLDesc");
 738                 goto cont;
 739             }
 740
 741             /* Yuck, XML.  Parse out the devices. */
 742             xml_doc = xmlReadDoc ((xmlChar *) xml, NULL, NULL, XML_PARSE_NONET);
 743             if (xml_doc == NULL) {
 744                 VIRT_ERROR (conn, "xmlReadDoc");
 745                 goto cont;
 746             }
 747
 748             xpath_ctx = xmlXPathNewContext (xml_doc);
 749
 750             /* Block devices. */
 751             xpath_obj = xmlXPathEval
 752                 ((xmlChar *) "/domain/devices/disk/target[@dev]",
 753                  xpath_ctx);
 754             if (xpath_obj == NULL || xpath_obj->type != XPATH_NODESET ||
 755                 xpath_obj->nodesetval == NULL)
 756                 goto cont;
 757
 758             for (j = 0; j < xpath_obj->nodesetval->nodeNr; ++j) {
 759                 xmlNodePtr node;
 760                 char *path = NULL;
 761
 762                 node = xpath_obj->nodesetval->nodeTab[j];
 763                 if (!node) continue;
 764                 path = (char *) xmlGetProp (node, (xmlChar *) "dev");
 765                 if (!path) continue;
 766
 767                 if (il_block_devices &&
 768                     ignore_device_match (il_block_devices, name, path) != 0)
 769                     goto cont2;
 770
 771                 add_block_device (dom, path);
 772             cont2:
 773                 if (path) xmlFree (path);
 774             }
 775             xmlXPathFreeObject (xpath_obj);
 776
 777             /* Network interfaces. */
 778             xpath_obj = xmlXPathEval
 779                 ((xmlChar *) "/domain/devices/interface[target[@dev]]",
 780                  xpath_ctx);
 781             if (xpath_obj == NULL || xpath_obj->type != XPATH_NODESET ||
 782                 xpath_obj->nodesetval == NULL)
 783                 goto cont;
 784
 785             xmlNodeSetPtr xml_interfaces = xpath_obj->nodesetval;
 786
 787             for (j = 0; j < xml_interfaces->nodeNr; ++j) {
 788                 char *path = NULL;
 789                 char *address = NULL;
 790                 xmlNodePtr xml_interface;
 791
 792                 xml_interface = xml_interfaces->nodeTab[j];
 793                 if (!xml_interface) continue;
 794                 xmlNodePtr child = NULL;
 795
 796                 for (child = xml_interface->children; child; child = child->next) {
 797                     if (child->type != XML_ELEMENT_NODE) continue;
 798
 799                     if (xmlStrEqual(child->name, (const xmlChar *) "target")) {
 800                         path = (char *) xmlGetProp (child, (const xmlChar *) "dev");
 801                         if (!path) continue;
 802                     } else if (xmlStrEqual(child->name, (const xmlChar *) "mac")) {
 803                         address = (char *) xmlGetProp (child, (const xmlChar *) "address");
 804                         if (!address) continue;
 805                     }
 806                 }
 807
 808                 if (il_interface_devices &&
 809                     (ignore_device_match (il_interface_devices, name, path) != 0 ||
 810                      ignore_device_match (il_interface_devices, name, address) != 0))
 811                     goto cont3;
 812
 813                 add_interface_device (dom, path, address, j+1);
 814                 cont3:
 815                     if (path) xmlFree (path);
 816                     if (address) xmlFree (address);
 817             }
 818
 819         cont:
 820             if (xpath_obj) xmlXPathFreeObject (xpath_obj);
 821             if (xpath_ctx) xmlXPathFreeContext (xpath_ctx);
 822             if (xml_doc) xmlFreeDoc (xml_doc);
 823             sfree (xml);
 824         }
 825
 826         sfree (domids);
 827     }
 828
 829     return 0;
 830 }
 831
 832 static void
 833 free_domains ()
 834 {
 835     int i;
 836
 837     if (domains) {
 838         for (i = 0; i < nr_domains; ++i)
 839             virDomainFree (domains[i]);
 840         sfree (domains);
 841     }
 842     domains = NULL;
 843     nr_domains = 0;
 844 }
 845
 846 static int
 847 add_domain (virDomainPtr dom)
 848 {
 849     virDomainPtr *new_ptr;
 850     int new_size = sizeof (domains[0]) * (nr_domains+1);
 851
 852     if (domains)
 853         new_ptr = realloc (domains, new_size);
 854     else
 855         new_ptr = malloc (new_size);
 856
 857     if (new_ptr == NULL)
 858         return -1;
 859
 860     domains = new_ptr;
 861     domains[nr_domains] = dom;
 862     return nr_domains++;
 863 }
 864
 865 static void
 866 free_block_devices ()
 867 {
 868     int i;
 869
 870     if (block_devices) {
 871         for (i = 0; i < nr_block_devices; ++i)
 872             sfree (block_devices[i].path);
 873         sfree (block_devices);
 874     }
 875     block_devices = NULL;
 876     nr_block_devices = 0;
 877 }
 878
 879 static int
 880 add_block_device (virDomainPtr dom, const char *path)
 881 {
 882     struct block_device *new_ptr;
 883     int new_size = sizeof (block_devices[0]) * (nr_block_devices+1);
 884     char *path_copy;
 885
 886     path_copy = strdup (path);
 887     if (!path_copy)
 888         return -1;
 889
 890     if (block_devices)
 891         new_ptr = realloc (block_devices, new_size);
 892     else
 893         new_ptr = malloc (new_size);
 894
 895     if (new_ptr == NULL) {
 896         sfree (path_copy);
 897         return -1;
 898     }
 899     block_devices = new_ptr;
 900     block_devices[nr_block_devices].dom = dom;
 901     block_devices[nr_block_devices].path = path_copy;
 902     return nr_block_devices++;
 903 }
 904
 905 static void
 906 free_interface_devices ()
 907 {
 908     int i;
 909
 910     if (interface_devices) {
 911         for (i = 0; i < nr_interface_devices; ++i) {
 912             sfree (interface_devices[i].path);
 913             sfree (interface_devices[i].address);
 914             sfree (interface_devices[i].number);
 915         }
 916         sfree (interface_devices);
 917     }
 918     interface_devices = NULL;
 919     nr_interface_devices = 0;
 920 }
 921
 922 static int
 923 add_interface_device (virDomainPtr dom, const char *path, const char *address, unsigned int number)
 924 {
 925     struct interface_device *new_ptr;
 926     int new_size = sizeof (interface_devices[0]) * (nr_interface_devices+1);
 927     char *path_copy, *address_copy, number_string[15];
 928
 929     path_copy = strdup (path);
 930     if (!path_copy) return -1;
 931
 932     address_copy = strdup (address);
 933     if (!address_copy) {
 934         sfree(path_copy);
 935         return -1;
 936     }
 937
 938     snprintf(number_string, sizeof (number_string), "interface-%u", number);
 939
 940     if (interface_devices)
 941         new_ptr = realloc (interface_devices, new_size);
 942     else
 943         new_ptr = malloc (new_size);
 944
 945     if (new_ptr == NULL) {
 946         sfree (path_copy);
 947         sfree (address_copy);
 948         return -1;
 949     }
 950     interface_devices = new_ptr;
 951     interface_devices[nr_interface_devices].dom = dom;
 952     interface_devices[nr_interface_devices].path = path_copy;
 953     interface_devices[nr_interface_devices].address = address_copy;
 954     interface_devices[nr_interface_devices].number = strdup(number_string);
 955     return nr_interface_devices++;
 956 }
 957
 958 static int
 959 ignore_device_match (ignorelist_t *il, const char *domname, const char *devpath)
 960 {
 961     char *name;
 962     int n, r;
 963
 964     n = sizeof (char) * (strlen (domname) + strlen (devpath) + 2);
 965     name = malloc (n);
 966     if (name == NULL) {
 967         ERROR (PLUGIN_NAME " plugin: malloc failed.");
 968         return 0;
 969     }
 970     ssnprintf (name, n, "%s:%s", domname, devpath);
 971     r = ignorelist_match (il, name);
 972     sfree (name);
 973     return r;
 974 }
 975
 976 static int
 977 lv_shutdown (void)
 978 {
 979     free_block_devices ();
 980     free_interface_devices ();
 981     free_domains ();
 982
 983     if (conn != NULL)
 984         virConnectClose (conn);
 985     conn = NULL;
 986
 987     ignorelist_free (il_domains);
 988     il_domains = NULL;
 989     ignorelist_free (il_block_devices);
 990     il_block_devices = NULL;
 991     ignorelist_free (il_interface_devices);
 992     il_interface_devices = NULL;
 993
 994     return 0;
 995 }
 996
 997 void
 998 module_register (void)
 999 {
1000     plugin_register_config (PLUGIN_NAME,
1001     lv_config,
1002     config_keys, NR_CONFIG_KEYS);
1003     plugin_register_init (PLUGIN_NAME, lv_init);
1004     plugin_register_read (PLUGIN_NAME, lv_read);
1005     plugin_register_shutdown (PLUGIN_NAME, lv_shutdown);
1006 }
1007
1008 /*
1009  * vim: shiftwidth=4 tabstop=8 softtabstop=4 expandtab fdm=marker
1010  */