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         cpu_submit (info.cpuTime, domains[i], "virt_cpu_total");
 551         memory_submit ((gauge_t) info.memory * 1024, domains[i]);
 552
 553         vinfo = malloc (info.nrVirtCpu * sizeof (vinfo[0]));
 554         if (vinfo == NULL) {
 555             ERROR (PLUGIN_NAME " plugin: malloc failed.");
 556             continue;
 557         }
 558
 559         status = virDomainGetVcpus (domains[i], vinfo, info.nrVirtCpu,
 560                 /* cpu map = */ NULL, /* cpu map length = */ 0);
 561         if (status < 0)
 562         {
 563             ERROR (PLUGIN_NAME " plugin: virDomainGetVcpus failed with status %i.",
 564                     status);
 565             sfree (vinfo);
 566             continue;
 567         }
 568
 569         for (j = 0; j < info.nrVirtCpu; ++j)
 570             vcpu_submit (vinfo[j].cpuTime,
 571                     domains[i], vinfo[j].number, "virt_vcpu");
 572
 573         sfree (vinfo);
 574
 575         minfo = malloc (VIR_DOMAIN_MEMORY_STAT_NR * sizeof (virDomainMemoryStatStruct));
 576         if (minfo == NULL) {
 577             ERROR ("virt plugin: malloc failed.");
 578             continue;
 579         }
 580
 581         status =  virDomainMemoryStats (domains[i], minfo, VIR_DOMAIN_MEMORY_STAT_NR, 0);
 582
 583         if (status < 0) {
 584             ERROR ("virt plugin: virDomainMemoryStats failed with status %i.",
 585                     status);
 586             sfree (minfo);
 587             continue;
 588         }
 589
 590         for (j = 0; j < status; j++) {
 591             memory_stats_submit ((gauge_t) minfo[j].val * 1024, domains[i], minfo[j].tag);
 592         }
 593
 594         sfree (minfo);
 595     }
 596
 597
 598     /* Get block device stats for each domain. */
 599     for (i = 0; i < nr_block_devices; ++i) {
 600         struct _virDomainBlockStats stats;
 601
 602         if (virDomainBlockStats (block_devices[i].dom, block_devices[i].path,
 603                     &stats, sizeof stats) != 0)
 604             continue;
 605
 606         if ((stats.rd_req != -1) && (stats.wr_req != -1))
 607             submit_derive2 ("disk_ops",
 608                     (derive_t) stats.rd_req, (derive_t) stats.wr_req,
 609                     block_devices[i].dom, block_devices[i].path);
 610
 611         if ((stats.rd_bytes != -1) && (stats.wr_bytes != -1))
 612             submit_derive2 ("disk_octets",
 613                     (derive_t) stats.rd_bytes, (derive_t) stats.wr_bytes,
 614                     block_devices[i].dom, block_devices[i].path);
 615     } /* for (nr_block_devices) */
 616
 617     /* Get interface stats for each domain. */
 618     for (i = 0; i < nr_interface_devices; ++i) {
 619         struct _virDomainInterfaceStats stats;
 620         char *display_name = NULL;
 621
 622
 623         switch (interface_format) {
 624             case if_address:
 625                 display_name = interface_devices[i].address;
 626                 break;
 627             case if_number:
 628                 display_name = interface_devices[i].number;
 629                 break;
 630             case if_name:
 631             default:
 632                 display_name = interface_devices[i].path;
 633         }
 634
 635         if (virDomainInterfaceStats (interface_devices[i].dom,
 636                     interface_devices[i].path,
 637                     &stats, sizeof stats) != 0)
 638             continue;
 639
 640         if ((stats.rx_bytes != -1) && (stats.tx_bytes != -1))
 641             submit_derive2 ("if_octets",
 642                     (derive_t) stats.rx_bytes, (derive_t) stats.tx_bytes,
 643                     interface_devices[i].dom, display_name);
 644
 645         if ((stats.rx_packets != -1) && (stats.tx_packets != -1))
 646             submit_derive2 ("if_packets",
 647                     (derive_t) stats.rx_packets, (derive_t) stats.tx_packets,
 648                     interface_devices[i].dom, display_name);
 649
 650         if ((stats.rx_errs != -1) && (stats.tx_errs != -1))
 651             submit_derive2 ("if_errors",
 652                     (derive_t) stats.rx_errs, (derive_t) stats.tx_errs,
 653                     interface_devices[i].dom, display_name);
 654
 655         if ((stats.rx_drop != -1) && (stats.tx_drop != -1))
 656             submit_derive2 ("if_dropped",
 657                     (derive_t) stats.rx_drop, (derive_t) stats.tx_drop,
 658                     interface_devices[i].dom, display_name);
 659     } /* for (nr_interface_devices) */
 660
 661     return 0;
 662 }
 663
 664 static int
 665 refresh_lists (void)
 666 {
 667     int n;
 668
 669     n = virConnectNumOfDomains (conn);
 670     if (n < 0) {
 671         VIRT_ERROR (conn, "reading number of domains");
 672         return -1;
 673     }
 674
 675     if (n > 0) {
 676         int i;
 677         int *domids;
 678
 679         /* Get list of domains. */
 680         domids = malloc (sizeof (int) * n);
 681         if (domids == 0) {
 682             ERROR (PLUGIN_NAME " plugin: malloc failed.");
 683             return -1;
 684         }
 685
 686         n = virConnectListDomains (conn, domids, n);
 687         if (n < 0) {
 688             VIRT_ERROR (conn, "reading list of domains");
 689             sfree (domids);
 690             return -1;
 691         }
 692
 693         free_block_devices ();
 694         free_interface_devices ();
 695         free_domains ();
 696
 697         /* Fetch each domain and add it to the list, unless ignore. */
 698         for (i = 0; i < n; ++i) {
 699             virDomainPtr dom = NULL;
 700             const char *name;
 701             char *xml = NULL;
 702             xmlDocPtr xml_doc = NULL;
 703             xmlXPathContextPtr xpath_ctx = NULL;
 704             xmlXPathObjectPtr xpath_obj = NULL;
 705             int j;
 706
 707             dom = virDomainLookupByID (conn, domids[i]);
 708             if (dom == NULL) {
 709                 VIRT_ERROR (conn, "virDomainLookupByID");
 710                 /* Could be that the domain went away -- ignore it anyway. */
 711                 continue;
 712             }
 713
 714             name = virDomainGetName (dom);
 715             if (name == NULL) {
 716                 VIRT_ERROR (conn, "virDomainGetName");
 717                 goto cont;
 718             }
 719
 720             if (il_domains && ignorelist_match (il_domains, name) != 0)
 721                 goto cont;
 722
 723             if (add_domain (dom) < 0) {
 724                 ERROR (PLUGIN_NAME " plugin: malloc failed.");
 725                 goto cont;
 726             }
 727
 728             /* Get a list of devices for this domain. */
 729             xml = virDomainGetXMLDesc (dom, 0);
 730             if (!xml) {
 731                 VIRT_ERROR (conn, "virDomainGetXMLDesc");
 732                 goto cont;
 733             }
 734
 735             /* Yuck, XML.  Parse out the devices. */
 736             xml_doc = xmlReadDoc ((xmlChar *) xml, NULL, NULL, XML_PARSE_NONET);
 737             if (xml_doc == NULL) {
 738                 VIRT_ERROR (conn, "xmlReadDoc");
 739                 goto cont;
 740             }
 741
 742             xpath_ctx = xmlXPathNewContext (xml_doc);
 743
 744             /* Block devices. */
 745             xpath_obj = xmlXPathEval
 746                 ((xmlChar *) "/domain/devices/disk/target[@dev]",
 747                  xpath_ctx);
 748             if (xpath_obj == NULL || xpath_obj->type != XPATH_NODESET ||
 749                 xpath_obj->nodesetval == NULL)
 750                 goto cont;
 751
 752             for (j = 0; j < xpath_obj->nodesetval->nodeNr; ++j) {
 753                 xmlNodePtr node;
 754                 char *path = NULL;
 755
 756                 node = xpath_obj->nodesetval->nodeTab[j];
 757                 if (!node) continue;
 758                 path = (char *) xmlGetProp (node, (xmlChar *) "dev");
 759                 if (!path) continue;
 760
 761                 if (il_block_devices &&
 762                     ignore_device_match (il_block_devices, name, path) != 0)
 763                     goto cont2;
 764
 765                 add_block_device (dom, path);
 766             cont2:
 767                 if (path) xmlFree (path);
 768             }
 769             xmlXPathFreeObject (xpath_obj);
 770
 771             /* Network interfaces. */
 772             xpath_obj = xmlXPathEval
 773                 ((xmlChar *) "/domain/devices/interface[target[@dev]]",
 774                  xpath_ctx);
 775             if (xpath_obj == NULL || xpath_obj->type != XPATH_NODESET ||
 776                 xpath_obj->nodesetval == NULL)
 777                 goto cont;
 778
 779             xmlNodeSetPtr xml_interfaces = xpath_obj->nodesetval;
 780
 781             for (j = 0; j < xml_interfaces->nodeNr; ++j) {
 782                 char *path = NULL;
 783                 char *address = NULL;
 784                 xmlNodePtr xml_interface;
 785
 786                 xml_interface = xml_interfaces->nodeTab[j];
 787                 if (!xml_interface) continue;
 788                 xmlNodePtr child = NULL;
 789
 790                 for (child = xml_interface->children; child; child = child->next) {
 791                     if (child->type != XML_ELEMENT_NODE) continue;
 792
 793                     if (xmlStrEqual(child->name, (const xmlChar *) "target")) {
 794                         path = (char *) xmlGetProp (child, (const xmlChar *) "dev");
 795                         if (!path) continue;
 796                     } else if (xmlStrEqual(child->name, (const xmlChar *) "mac")) {
 797                         address = (char *) xmlGetProp (child, (const xmlChar *) "address");
 798                         if (!address) continue;
 799                     }
 800                 }
 801
 802                 if (il_interface_devices &&
 803                     (ignore_device_match (il_interface_devices, name, path) != 0 ||
 804                      ignore_device_match (il_interface_devices, name, address) != 0))
 805                     goto cont3;
 806
 807                 add_interface_device (dom, path, address, j+1);
 808                 cont3:
 809                     if (path) xmlFree (path);
 810                     if (address) xmlFree (address);
 811             }
 812
 813         cont:
 814             if (xpath_obj) xmlXPathFreeObject (xpath_obj);
 815             if (xpath_ctx) xmlXPathFreeContext (xpath_ctx);
 816             if (xml_doc) xmlFreeDoc (xml_doc);
 817             sfree (xml);
 818         }
 819
 820         sfree (domids);
 821     }
 822
 823     return 0;
 824 }
 825
 826 static void
 827 free_domains ()
 828 {
 829     int i;
 830
 831     if (domains) {
 832         for (i = 0; i < nr_domains; ++i)
 833             virDomainFree (domains[i]);
 834         sfree (domains);
 835     }
 836     domains = NULL;
 837     nr_domains = 0;
 838 }
 839
 840 static int
 841 add_domain (virDomainPtr dom)
 842 {
 843     virDomainPtr *new_ptr;
 844     int new_size = sizeof (domains[0]) * (nr_domains+1);
 845
 846     if (domains)
 847         new_ptr = realloc (domains, new_size);
 848     else
 849         new_ptr = malloc (new_size);
 850
 851     if (new_ptr == NULL)
 852         return -1;
 853
 854     domains = new_ptr;
 855     domains[nr_domains] = dom;
 856     return nr_domains++;
 857 }
 858
 859 static void
 860 free_block_devices ()
 861 {
 862     int i;
 863
 864     if (block_devices) {
 865         for (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 ()
 901 {
 902     int i;
 903
 904     if (interface_devices) {
 905         for (i = 0; i < nr_interface_devices; ++i) {
 906             sfree (interface_devices[i].path);
 907             sfree (interface_devices[i].address);
 908             sfree (interface_devices[i].number);
 909         }
 910         sfree (interface_devices);
 911     }
 912     interface_devices = NULL;
 913     nr_interface_devices = 0;
 914 }
 915
 916 static int
 917 add_interface_device (virDomainPtr dom, const char *path, const char *address, unsigned int number)
 918 {
 919     struct interface_device *new_ptr;
 920     int new_size = sizeof (interface_devices[0]) * (nr_interface_devices+1);
 921     char *path_copy, *address_copy, number_string[15];
 922
 923     path_copy = strdup (path);
 924     if (!path_copy) return -1;
 925
 926     address_copy = strdup (address);
 927     if (!address_copy) {
 928         sfree(path_copy);
 929         return -1;
 930     }
 931
 932     snprintf(number_string, sizeof (number_string), "interface-%u", number);
 933
 934     if (interface_devices)
 935         new_ptr = realloc (interface_devices, new_size);
 936     else
 937         new_ptr = malloc (new_size);
 938
 939     if (new_ptr == NULL) {
 940         sfree (path_copy);
 941         sfree (address_copy);
 942         return -1;
 943     }
 944     interface_devices = new_ptr;
 945     interface_devices[nr_interface_devices].dom = dom;
 946     interface_devices[nr_interface_devices].path = path_copy;
 947     interface_devices[nr_interface_devices].address = address_copy;
 948     interface_devices[nr_interface_devices].number = strdup(number_string);
 949     return nr_interface_devices++;
 950 }
 951
 952 static int
 953 ignore_device_match (ignorelist_t *il, const char *domname, const char *devpath)
 954 {
 955     char *name;
 956     int n, r;
 957
 958     n = sizeof (char) * (strlen (domname) + strlen (devpath) + 2);
 959     name = malloc (n);
 960     if (name == NULL) {
 961         ERROR (PLUGIN_NAME " plugin: malloc failed.");
 962         return 0;
 963     }
 964     ssnprintf (name, n, "%s:%s", domname, devpath);
 965     r = ignorelist_match (il, name);
 966     sfree (name);
 967     return r;
 968 }
 969
 970 static int
 971 lv_shutdown (void)
 972 {
 973     free_block_devices ();
 974     free_interface_devices ();
 975     free_domains ();
 976
 977     if (conn != NULL)
 978         virConnectClose (conn);
 979     conn = NULL;
 980
 981     ignorelist_free (il_domains);
 982     il_domains = NULL;
 983     ignorelist_free (il_block_devices);
 984     il_block_devices = NULL;
 985     ignorelist_free (il_interface_devices);
 986     il_interface_devices = NULL;
 987
 988     return 0;
 989 }
 990
 991 void
 992 module_register (void)
 993 {
 994     plugin_register_config (PLUGIN_NAME,
 995     lv_config,
 996     config_keys, NR_CONFIG_KEYS);
 997     plugin_register_init (PLUGIN_NAME, lv_init);
 998     plugin_register_read (PLUGIN_NAME, lv_read);
 999     plugin_register_shutdown (PLUGIN_NAME, lv_shutdown);
1000 }
1001
1002 /*
1003  * vim: shiftwidth=4 tabstop=8 softtabstop=4 expandtab fdm=marker
1004  */