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 **/
22 #include "collectd.h"
24 #include "common.h"
25 #include "plugin.h"
26 #include "utils_ignorelist.h"
27 #include "utils_complain.h"
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 #include <libgen.h> /* for basename(3) */
36 /* Plugin name */
37 #define PLUGIN_NAME "virt"
39 static const char *config_keys[] = {
40 "Connection",
42 "RefreshInterval",
44 "Domain",
45 "BlockDevice",
46 "BlockDeviceFormat",
47 "BlockDeviceFormatBasename",
48 "InterfaceDevice",
49 "IgnoreSelected",
51 "HostnameFormat",
52 "InterfaceFormat",
54 "PluginInstanceFormat",
56 NULL
57 };
58 #define NR_CONFIG_KEYS ((sizeof config_keys / sizeof config_keys[0]) - 1)
60 /* Connection. */
61 static virConnectPtr conn = 0;
62 static char *conn_string = NULL;
63 static c_complain_t conn_complain = C_COMPLAIN_INIT_STATIC;
65 /* Seconds between list refreshes, 0 disables completely. */
66 static int interval = 60;
68 /* List of domains, if specified. */
69 static ignorelist_t *il_domains = NULL;
70 /* List of block devices, if specified. */
71 static ignorelist_t *il_block_devices = NULL;
72 /* List of network interface devices, if specified. */
73 static ignorelist_t *il_interface_devices = NULL;
75 static int ignore_device_match (ignorelist_t *,
76 const char *domname, const char *devpath);
78 /* Actual list of domains found on last refresh. */
79 static virDomainPtr *domains = NULL;
80 static int nr_domains = 0;
82 static void free_domains (void);
83 static int add_domain (virDomainPtr dom);
85 /* Actual list of block devices found on last refresh. */
86 struct block_device {
87 virDomainPtr dom; /* domain */
88 char *path; /* name of block device */
89 };
91 static struct block_device *block_devices = NULL;
92 static int nr_block_devices = 0;
94 static void free_block_devices (void);
95 static int add_block_device (virDomainPtr dom, const char *path);
97 /* Actual list of network interfaces found on last refresh. */
98 struct interface_device {
99 virDomainPtr dom; /* domain */
100 char *path; /* name of interface device */
101 char *address; /* mac address of interface device */
102 char *number; /* interface device number */
103 };
105 static struct interface_device *interface_devices = NULL;
106 static int nr_interface_devices = 0;
108 static void free_interface_devices (void);
109 static int add_interface_device (virDomainPtr dom, const char *path, const char *address, unsigned int number);
111 /* HostnameFormat. */
112 #define HF_MAX_FIELDS 3
114 enum hf_field {
115 hf_none = 0,
116 hf_hostname,
117 hf_name,
118 hf_uuid
119 };
121 static enum hf_field hostname_format[HF_MAX_FIELDS] =
122 { hf_name };
124 /* PluginInstanceFormat */
125 #define PLGINST_MAX_FIELDS 2
127 enum plginst_field {
128 plginst_none = 0,
129 plginst_name,
130 plginst_uuid
131 };
133 static enum plginst_field plugin_instance_format[PLGINST_MAX_FIELDS] =
134 { plginst_none };
136 /* BlockDeviceFormat */
137 enum bd_field {
138 target,
139 source
140 };
142 /* InterfaceFormat. */
143 enum if_field {
144 if_address,
145 if_name,
146 if_number
147 };
149 /* BlockDeviceFormatBasename */
150 _Bool blockdevice_format_basename = 0;
151 static enum bd_field blockdevice_format = target;
152 static enum if_field interface_format = if_name;
154 /* Time that we last refreshed. */
155 static time_t last_refresh = (time_t) 0;
157 static int refresh_lists (void);
159 /* ERROR(...) macro for virterrors. */
160 #define VIRT_ERROR(conn,s) do { \
161 virErrorPtr err; \
162 err = (conn) ? virConnGetLastError ((conn)) : virGetLastError (); \
163 if (err) ERROR ("%s: %s", (s), err->message); \
164 } while(0)
166 static void
167 init_value_list (value_list_t *vl, virDomainPtr dom)
168 {
169 int n;
170 const char *name;
171 char uuid[VIR_UUID_STRING_BUFLEN];
173 sstrncpy (vl->plugin, PLUGIN_NAME, sizeof (vl->plugin));
175 vl->host[0] = '\0';
177 /* Construct the hostname field according to HostnameFormat. */
178 for (int i = 0; i < HF_MAX_FIELDS; ++i) {
179 if (hostname_format[i] == hf_none)
180 continue;
182 n = DATA_MAX_NAME_LEN - strlen (vl->host) - 2;
184 if (i > 0 && n >= 1) {
185 strncat (vl->host, ":", 1);
186 n--;
187 }
189 switch (hostname_format[i]) {
190 case hf_none: break;
191 case hf_hostname:
192 strncat (vl->host, hostname_g, n);
193 break;
194 case hf_name:
195 name = virDomainGetName (dom);
196 if (name)
197 strncat (vl->host, name, n);
198 break;
199 case hf_uuid:
200 if (virDomainGetUUIDString (dom, uuid) == 0)
201 strncat (vl->host, uuid, n);
202 break;
203 }
204 }
206 vl->host[sizeof (vl->host) - 1] = '\0';
208 /* Construct the plugin instance field according to PluginInstanceFormat. */
209 for (int i = 0; i < PLGINST_MAX_FIELDS; ++i) {
210 if (plugin_instance_format[i] == plginst_none)
211 continue;
213 n = sizeof(vl->plugin_instance) - strlen (vl->plugin_instance) - 2;
215 if (i > 0 && n >= 1) {
216 strncat (vl->plugin_instance, ":", 1);
217 n--;
218 }
220 switch (plugin_instance_format[i]) {
221 case plginst_none: break;
222 case plginst_name:
223 name = virDomainGetName (dom);
224 if (name)
225 strncat (vl->plugin_instance, name, n);
226 break;
227 case plginst_uuid:
228 if (virDomainGetUUIDString (dom, uuid) == 0)
229 strncat (vl->plugin_instance, uuid, n);
230 break;
231 }
232 }
234 vl->plugin_instance[sizeof (vl->plugin_instance) - 1] = '\0';
236 } /* void init_value_list */
238 static void submit (virDomainPtr dom,
239 char const *type, char const *type_instance,
240 value_t *values, size_t values_len)
241 {
242 value_list_t vl = VALUE_LIST_INIT;
243 init_value_list (&vl, dom);
245 vl.values = values;
246 vl.values_len = values_len;
248 sstrncpy (vl.type, type, sizeof (vl.type));
249 if (type_instance != NULL)
250 sstrncpy (vl.type_instance, type_instance, sizeof (vl.type_instance));
252 plugin_dispatch_values (&vl);
253 }
255 static void
256 memory_submit (gauge_t value, virDomainPtr dom)
257 {
258 submit (dom, "memory", "total", &(value_t) { .gauge = value }, 1);
259 }
261 static void
262 memory_stats_submit (gauge_t value, virDomainPtr dom, int tag_index)
263 {
264 static const char *tags[] = { "swap_in", "swap_out", "major_fault", "minor_fault",
265 "unused", "available", "actual_balloon", "rss"};
267 if ((tag_index < 0) || (tag_index >= STATIC_ARRAY_SIZE (tags))) {
268 ERROR ("virt plugin: Array index out of bounds: tag_index = %d", tag_index);
269 return;
270 }
272 submit (dom, "memory", tags[tag_index], &(value_t) { .gauge = value }, 1);
273 }
275 static void
276 cpu_submit (unsigned long long value,
277 virDomainPtr dom, const char *type)
278 {
279 submit (dom, type, NULL, &(value_t) { .derive = (derive_t) value }, 1);
280 }
282 static void
283 vcpu_submit (derive_t value,
284 virDomainPtr dom, int vcpu_nr, const char *type)
285 {
286 char type_instance[DATA_MAX_NAME_LEN];
288 ssnprintf (type_instance, sizeof (type_instance), "%d", vcpu_nr);
290 submit (dom, type, type_instance, &(value_t) { .derive = value }, 1);
291 }
293 static void
294 submit_derive2 (const char *type, derive_t v0, derive_t v1,
295 virDomainPtr dom, const char *devname)
296 {
297 value_t values[] = {
298 { .derive = v0 },
299 { .derive = v1 },
300 };
302 submit (dom, type, devname, values, STATIC_ARRAY_SIZE (values));
303 } /* void submit_derive2 */
305 static int
306 lv_init (void)
307 {
308 if (virInitialize () != 0)
309 return -1;
310 else
311 return 0;
312 }
314 static int
315 lv_config (const char *key, const char *value)
316 {
317 if (virInitialize () != 0)
318 return 1;
320 if (il_domains == NULL)
321 il_domains = ignorelist_create (1);
322 if (il_block_devices == NULL)
323 il_block_devices = ignorelist_create (1);
324 if (il_interface_devices == NULL)
325 il_interface_devices = ignorelist_create (1);
327 if (strcasecmp (key, "Connection") == 0) {
328 char *tmp = strdup (value);
329 if (tmp == NULL) {
330 ERROR (PLUGIN_NAME " plugin: Connection strdup failed.");
331 return 1;
332 }
333 sfree (conn_string);
334 conn_string = tmp;
335 return 0;
336 }
338 if (strcasecmp (key, "RefreshInterval") == 0) {
339 char *eptr = NULL;
340 interval = strtol (value, &eptr, 10);
341 if (eptr == NULL || *eptr != '\0') return 1;
342 return 0;
343 }
345 if (strcasecmp (key, "Domain") == 0) {
346 if (ignorelist_add (il_domains, value)) return 1;
347 return 0;
348 }
349 if (strcasecmp (key, "BlockDevice") == 0) {
350 if (ignorelist_add (il_block_devices, value)) return 1;
351 return 0;
352 }
354 if (strcasecmp (key, "BlockDeviceFormat") == 0) {
355 if (strcasecmp (value, "target") == 0)
356 blockdevice_format = target;
357 else if (strcasecmp (value, "source") == 0)
358 blockdevice_format = source;
359 else {
360 ERROR (PLUGIN_NAME " plugin: unknown BlockDeviceFormat: %s", value);
361 return -1;
362 }
363 return 0;
364 }
365 if (strcasecmp (key, "BlockDeviceFormatBasename") == 0) {
366 blockdevice_format_basename = IS_TRUE (value);
367 return 0;
368 }
369 if (strcasecmp (key, "InterfaceDevice") == 0) {
370 if (ignorelist_add (il_interface_devices, value)) return 1;
371 return 0;
372 }
374 if (strcasecmp (key, "IgnoreSelected") == 0) {
375 if (IS_TRUE (value))
376 {
377 ignorelist_set_invert (il_domains, 0);
378 ignorelist_set_invert (il_block_devices, 0);
379 ignorelist_set_invert (il_interface_devices, 0);
380 }
381 else
382 {
383 ignorelist_set_invert (il_domains, 1);
384 ignorelist_set_invert (il_block_devices, 1);
385 ignorelist_set_invert (il_interface_devices, 1);
386 }
387 return 0;
388 }
390 if (strcasecmp (key, "HostnameFormat") == 0) {
391 char *value_copy;
392 char *fields[HF_MAX_FIELDS];
393 int n;
395 value_copy = strdup (value);
396 if (value_copy == NULL) {
397 ERROR (PLUGIN_NAME " plugin: strdup failed.");
398 return -1;
399 }
401 n = strsplit (value_copy, fields, HF_MAX_FIELDS);
402 if (n < 1) {
403 sfree (value_copy);
404 ERROR (PLUGIN_NAME " plugin: HostnameFormat: no fields");
405 return -1;
406 }
408 for (int i = 0; i < n; ++i) {
409 if (strcasecmp (fields[i], "hostname") == 0)
410 hostname_format[i] = hf_hostname;
411 else if (strcasecmp (fields[i], "name") == 0)
412 hostname_format[i] = hf_name;
413 else if (strcasecmp (fields[i], "uuid") == 0)
414 hostname_format[i] = hf_uuid;
415 else {
416 ERROR (PLUGIN_NAME " plugin: unknown HostnameFormat field: %s", fields[i]);
417 sfree (value_copy);
418 return -1;
419 }
420 }
421 sfree (value_copy);
423 for (int i = n; i < HF_MAX_FIELDS; ++i)
424 hostname_format[i] = hf_none;
426 return 0;
427 }
429 if (strcasecmp (key, "PluginInstanceFormat") == 0) {
430 char *value_copy;
431 char *fields[PLGINST_MAX_FIELDS];
432 int n;
434 value_copy = strdup (value);
435 if (value_copy == NULL) {
436 ERROR (PLUGIN_NAME " plugin: strdup failed.");
437 return -1;
438 }
440 n = strsplit (value_copy, fields, PLGINST_MAX_FIELDS);
441 if (n < 1) {
442 sfree (value_copy);
443 ERROR (PLUGIN_NAME " plugin: PluginInstanceFormat: no fields");
444 return -1;
445 }
447 for (int i = 0; i < n; ++i) {
448 if (strcasecmp (fields[i], "none") == 0) {
449 plugin_instance_format[i] = plginst_none;
450 break;
451 } else if (strcasecmp (fields[i], "name") == 0)
452 plugin_instance_format[i] = plginst_name;
453 else if (strcasecmp (fields[i], "uuid") == 0)
454 plugin_instance_format[i] = plginst_uuid;
455 else {
456 ERROR (PLUGIN_NAME " plugin: unknown PluginInstanceFormat field: %s", fields[i]);
457 sfree (value_copy);
458 return -1;
459 }
460 }
461 sfree (value_copy);
463 for (int i = n; i < PLGINST_MAX_FIELDS; ++i)
464 plugin_instance_format[i] = plginst_none;
466 return 0;
467 }
469 if (strcasecmp (key, "InterfaceFormat") == 0) {
470 if (strcasecmp (value, "name") == 0)
471 interface_format = if_name;
472 else if (strcasecmp (value, "address") == 0)
473 interface_format = if_address;
474 else if (strcasecmp (value, "number") == 0)
475 interface_format = if_number;
476 else {
477 ERROR (PLUGIN_NAME " plugin: unknown InterfaceFormat: %s", value);
478 return -1;
479 }
480 return 0;
481 }
483 /* Unrecognised option. */
484 return -1;
485 }
487 static int
488 lv_read (void)
489 {
490 time_t t;
492 if (conn == NULL) {
493 /* `conn_string == NULL' is acceptable. */
494 conn = virConnectOpenReadOnly (conn_string);
495 if (conn == NULL) {
496 c_complain (LOG_ERR, &conn_complain,
497 PLUGIN_NAME " plugin: Unable to connect: "
498 "virConnectOpenReadOnly failed.");
499 return -1;
500 }
501 }
502 c_release (LOG_NOTICE, &conn_complain,
503 PLUGIN_NAME " plugin: Connection established.");
505 time (&t);
507 /* Need to refresh domain or device lists? */
508 if ((last_refresh == (time_t) 0) ||
509 ((interval > 0) && ((last_refresh + interval) <= t))) {
510 if (refresh_lists () != 0) {
511 if (conn != NULL)
512 virConnectClose (conn);
513 conn = NULL;
514 return -1;
515 }
516 last_refresh = t;
517 }
519 #if 0
520 for (int i = 0; i < nr_domains; ++i)
521 fprintf (stderr, "domain %s\n", virDomainGetName (domains[i]));
522 for (int i = 0; i < nr_block_devices; ++i)
523 fprintf (stderr, "block device %d %s:%s\n",
524 i, virDomainGetName (block_devices[i].dom),
525 block_devices[i].path);
526 for (int i = 0; i < nr_interface_devices; ++i)
527 fprintf (stderr, "interface device %d %s:%s\n",
528 i, virDomainGetName (interface_devices[i].dom),
529 interface_devices[i].path);
530 #endif
532 /* Get CPU usage, memory, VCPU usage for each domain. */
533 for (int i = 0; i < nr_domains; ++i) {
534 virDomainInfo info;
535 virVcpuInfoPtr vinfo = NULL;
536 virDomainMemoryStatPtr minfo = NULL;
537 int status;
539 status = virDomainGetInfo (domains[i], &info);
540 if (status != 0)
541 {
542 ERROR (PLUGIN_NAME " plugin: virDomainGetInfo failed with status %i.",
543 status);
544 continue;
545 }
547 if (info.state != VIR_DOMAIN_RUNNING)
548 {
549 /* only gather stats for running domains */
550 continue;
551 }
553 cpu_submit (info.cpuTime, domains[i], "virt_cpu_total");
554 memory_submit ((gauge_t) info.memory * 1024, domains[i]);
556 vinfo = malloc (info.nrVirtCpu * sizeof (vinfo[0]));
557 if (vinfo == NULL) {
558 ERROR (PLUGIN_NAME " plugin: malloc failed.");
559 continue;
560 }
562 status = virDomainGetVcpus (domains[i], vinfo, info.nrVirtCpu,
563 /* cpu map = */ NULL, /* cpu map length = */ 0);
564 if (status < 0)
565 {
566 ERROR (PLUGIN_NAME " plugin: virDomainGetVcpus failed with status %i.",
567 status);
568 sfree (vinfo);
569 continue;
570 }
572 for (int j = 0; j < info.nrVirtCpu; ++j)
573 vcpu_submit (vinfo[j].cpuTime,
574 domains[i], vinfo[j].number, "virt_vcpu");
576 sfree (vinfo);
578 minfo = malloc (VIR_DOMAIN_MEMORY_STAT_NR * sizeof (virDomainMemoryStatStruct));
579 if (minfo == NULL) {
580 ERROR ("virt plugin: malloc failed.");
581 continue;
582 }
584 status = virDomainMemoryStats (domains[i], minfo, VIR_DOMAIN_MEMORY_STAT_NR, 0);
586 if (status < 0) {
587 ERROR ("virt plugin: virDomainMemoryStats failed with status %i.",
588 status);
589 sfree (minfo);
590 continue;
591 }
593 for (int j = 0; j < status; j++) {
594 memory_stats_submit ((gauge_t) minfo[j].val * 1024, domains[i], minfo[j].tag);
595 }
597 sfree (minfo);
598 }
601 /* Get block device stats for each domain. */
602 for (int i = 0; i < nr_block_devices; ++i) {
603 struct _virDomainBlockStats stats;
605 if (virDomainBlockStats (block_devices[i].dom, block_devices[i].path,
606 &stats, sizeof stats) != 0)
607 continue;
609 char *type_instance = NULL;
610 if (blockdevice_format_basename && blockdevice_format == source)
611 type_instance = strdup(basename(block_devices[i].path));
612 else
613 type_instance = strdup(block_devices[i].path);
615 if ((stats.rd_req != -1) && (stats.wr_req != -1))
616 submit_derive2 ("disk_ops",
617 (derive_t) stats.rd_req, (derive_t) stats.wr_req,
618 block_devices[i].dom, type_instance);
620 if ((stats.rd_bytes != -1) && (stats.wr_bytes != -1))
621 submit_derive2 ("disk_octets",
622 (derive_t) stats.rd_bytes, (derive_t) stats.wr_bytes,
623 block_devices[i].dom, type_instance);
625 sfree (type_instance);
626 } /* for (nr_block_devices) */
628 /* Get interface stats for each domain. */
629 for (int i = 0; i < nr_interface_devices; ++i) {
630 struct _virDomainInterfaceStats stats;
631 char *display_name = NULL;
633 switch (interface_format) {
634 case if_address:
635 display_name = interface_devices[i].address;
636 break;
637 case if_number:
638 display_name = interface_devices[i].number;
639 break;
640 case if_name:
641 default:
642 display_name = interface_devices[i].path;
643 }
645 if (virDomainInterfaceStats (interface_devices[i].dom,
646 interface_devices[i].path,
647 &stats, sizeof stats) != 0)
648 continue;
650 if ((stats.rx_bytes != -1) && (stats.tx_bytes != -1))
651 submit_derive2 ("if_octets",
652 (derive_t) stats.rx_bytes, (derive_t) stats.tx_bytes,
653 interface_devices[i].dom, display_name);
655 if ((stats.rx_packets != -1) && (stats.tx_packets != -1))
656 submit_derive2 ("if_packets",
657 (derive_t) stats.rx_packets, (derive_t) stats.tx_packets,
658 interface_devices[i].dom, display_name);
660 if ((stats.rx_errs != -1) && (stats.tx_errs != -1))
661 submit_derive2 ("if_errors",
662 (derive_t) stats.rx_errs, (derive_t) stats.tx_errs,
663 interface_devices[i].dom, display_name);
665 if ((stats.rx_drop != -1) && (stats.tx_drop != -1))
666 submit_derive2 ("if_dropped",
667 (derive_t) stats.rx_drop, (derive_t) stats.tx_drop,
668 interface_devices[i].dom, display_name);
669 } /* for (nr_interface_devices) */
671 return 0;
672 }
674 static int
675 refresh_lists (void)
676 {
677 int n;
679 n = virConnectNumOfDomains (conn);
680 if (n < 0) {
681 VIRT_ERROR (conn, "reading number of domains");
682 return -1;
683 }
685 if (n > 0) {
686 int *domids;
688 /* Get list of domains. */
689 domids = malloc (sizeof (*domids) * n);
690 if (domids == NULL) {
691 ERROR (PLUGIN_NAME " plugin: malloc failed.");
692 return -1;
693 }
695 n = virConnectListDomains (conn, domids, n);
696 if (n < 0) {
697 VIRT_ERROR (conn, "reading list of domains");
698 sfree (domids);
699 return -1;
700 }
702 free_block_devices ();
703 free_interface_devices ();
704 free_domains ();
706 /* Fetch each domain and add it to the list, unless ignore. */
707 for (int i = 0; i < n; ++i) {
708 virDomainPtr dom = NULL;
709 const char *name;
710 char *xml = NULL;
711 xmlDocPtr xml_doc = NULL;
712 xmlXPathContextPtr xpath_ctx = NULL;
713 xmlXPathObjectPtr xpath_obj = NULL;
715 dom = virDomainLookupByID (conn, domids[i]);
716 if (dom == NULL) {
717 VIRT_ERROR (conn, "virDomainLookupByID");
718 /* Could be that the domain went away -- ignore it anyway. */
719 continue;
720 }
722 name = virDomainGetName (dom);
723 if (name == NULL) {
724 VIRT_ERROR (conn, "virDomainGetName");
725 goto cont;
726 }
728 if (il_domains && ignorelist_match (il_domains, name) != 0)
729 goto cont;
731 if (add_domain (dom) < 0) {
732 ERROR (PLUGIN_NAME " plugin: malloc failed.");
733 goto cont;
734 }
736 /* Get a list of devices for this domain. */
737 xml = virDomainGetXMLDesc (dom, 0);
738 if (!xml) {
739 VIRT_ERROR (conn, "virDomainGetXMLDesc");
740 goto cont;
741 }
743 /* Yuck, XML. Parse out the devices. */
744 xml_doc = xmlReadDoc ((xmlChar *) xml, NULL, NULL, XML_PARSE_NONET);
745 if (xml_doc == NULL) {
746 VIRT_ERROR (conn, "xmlReadDoc");
747 goto cont;
748 }
750 xpath_ctx = xmlXPathNewContext (xml_doc);
752 /* Block devices. */
753 char *bd_xmlpath = "/domain/devices/disk/target[@dev]";
754 if (blockdevice_format == source)
755 bd_xmlpath = "/domain/devices/disk/source[@dev]";
756 xpath_obj = xmlXPathEval ((xmlChar *) bd_xmlpath, xpath_ctx);
758 if (xpath_obj == NULL || xpath_obj->type != XPATH_NODESET ||
759 xpath_obj->nodesetval == NULL)
760 goto cont;
762 for (int j = 0; j < xpath_obj->nodesetval->nodeNr; ++j) {
763 xmlNodePtr node;
764 char *path = NULL;
766 node = xpath_obj->nodesetval->nodeTab[j];
767 if (!node) continue;
768 path = (char *) xmlGetProp (node, (xmlChar *) "dev");
769 if (!path) continue;
771 if (il_block_devices &&
772 ignore_device_match (il_block_devices, name, path) != 0)
773 goto cont2;
775 add_block_device (dom, path);
776 cont2:
777 if (path) xmlFree (path);
778 }
779 xmlXPathFreeObject (xpath_obj);
781 /* Network interfaces. */
782 xpath_obj = xmlXPathEval
783 ((xmlChar *) "/domain/devices/interface[target[@dev]]",
784 xpath_ctx);
785 if (xpath_obj == NULL || xpath_obj->type != XPATH_NODESET ||
786 xpath_obj->nodesetval == NULL)
787 goto cont;
789 xmlNodeSetPtr xml_interfaces = xpath_obj->nodesetval;
791 for (int j = 0; j < xml_interfaces->nodeNr; ++j) {
792 char *path = NULL;
793 char *address = NULL;
794 xmlNodePtr xml_interface;
796 xml_interface = xml_interfaces->nodeTab[j];
797 if (!xml_interface) continue;
799 for (xmlNodePtr child = xml_interface->children; child; child = child->next) {
800 if (child->type != XML_ELEMENT_NODE) continue;
802 if (xmlStrEqual(child->name, (const xmlChar *) "target")) {
803 path = (char *) xmlGetProp (child, (const xmlChar *) "dev");
804 if (!path) continue;
805 } else if (xmlStrEqual(child->name, (const xmlChar *) "mac")) {
806 address = (char *) xmlGetProp (child, (const xmlChar *) "address");
807 if (!address) continue;
808 }
809 }
811 if (il_interface_devices &&
812 (ignore_device_match (il_interface_devices, name, path) != 0 ||
813 ignore_device_match (il_interface_devices, name, address) != 0))
814 goto cont3;
816 add_interface_device (dom, path, address, j+1);
817 cont3:
818 if (path) xmlFree (path);
819 if (address) xmlFree (address);
820 }
822 cont:
823 if (xpath_obj) xmlXPathFreeObject (xpath_obj);
824 if (xpath_ctx) xmlXPathFreeContext (xpath_ctx);
825 if (xml_doc) xmlFreeDoc (xml_doc);
826 sfree (xml);
827 }
829 sfree (domids);
830 }
832 return 0;
833 }
835 static void
836 free_domains (void)
837 {
838 if (domains) {
839 for (int i = 0; i < nr_domains; ++i)
840 virDomainFree (domains[i]);
841 sfree (domains);
842 }
843 domains = NULL;
844 nr_domains = 0;
845 }
847 static int
848 add_domain (virDomainPtr dom)
849 {
850 virDomainPtr *new_ptr;
851 int new_size = sizeof (domains[0]) * (nr_domains+1);
853 if (domains)
854 new_ptr = realloc (domains, new_size);
855 else
856 new_ptr = malloc (new_size);
858 if (new_ptr == NULL)
859 return -1;
861 domains = new_ptr;
862 domains[nr_domains] = dom;
863 return nr_domains++;
864 }
866 static void
867 free_block_devices (void)
868 {
869 if (block_devices) {
870 for (int i = 0; i < nr_block_devices; ++i)
871 sfree (block_devices[i].path);
872 sfree (block_devices);
873 }
874 block_devices = NULL;
875 nr_block_devices = 0;
876 }
878 static int
879 add_block_device (virDomainPtr dom, const char *path)
880 {
881 struct block_device *new_ptr;
882 int new_size = sizeof (block_devices[0]) * (nr_block_devices+1);
883 char *path_copy;
885 path_copy = strdup (path);
886 if (!path_copy)
887 return -1;
889 if (block_devices)
890 new_ptr = realloc (block_devices, new_size);
891 else
892 new_ptr = malloc (new_size);
894 if (new_ptr == NULL) {
895 sfree (path_copy);
896 return -1;
897 }
898 block_devices = new_ptr;
899 block_devices[nr_block_devices].dom = dom;
900 block_devices[nr_block_devices].path = path_copy;
901 return nr_block_devices++;
902 }
904 static void
905 free_interface_devices (void)
906 {
907 if (interface_devices) {
908 for (int i = 0; i < nr_interface_devices; ++i) {
909 sfree (interface_devices[i].path);
910 sfree (interface_devices[i].address);
911 sfree (interface_devices[i].number);
912 }
913 sfree (interface_devices);
914 }
915 interface_devices = NULL;
916 nr_interface_devices = 0;
917 }
919 static int
920 add_interface_device (virDomainPtr dom, const char *path, const char *address, unsigned int number)
921 {
922 struct interface_device *new_ptr;
923 int new_size = sizeof (interface_devices[0]) * (nr_interface_devices+1);
924 char *path_copy, *address_copy, number_string[15];
926 if ((path == NULL) || (address == NULL))
927 return EINVAL;
929 path_copy = strdup (path);
930 if (!path_copy) return -1;
932 address_copy = strdup (address);
933 if (!address_copy) {
934 sfree(path_copy);
935 return -1;
936 }
938 snprintf(number_string, sizeof (number_string), "interface-%u", number);
940 if (interface_devices)
941 new_ptr = realloc (interface_devices, new_size);
942 else
943 new_ptr = malloc (new_size);
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 }
958 static int
959 ignore_device_match (ignorelist_t *il, const char *domname, const char *devpath)
960 {
961 char *name;
962 int n, r;
964 if ((domname == NULL) || (devpath == NULL))
965 return 0;
967 n = sizeof (char) * (strlen (domname) + strlen (devpath) + 2);
968 name = malloc (n);
969 if (name == NULL) {
970 ERROR (PLUGIN_NAME " plugin: malloc failed.");
971 return 0;
972 }
973 ssnprintf (name, n, "%s:%s", domname, devpath);
974 r = ignorelist_match (il, name);
975 sfree (name);
976 return r;
977 }
979 static int
980 lv_shutdown (void)
981 {
982 free_block_devices ();
983 free_interface_devices ();
984 free_domains ();
986 if (conn != NULL)
987 virConnectClose (conn);
988 conn = NULL;
990 ignorelist_free (il_domains);
991 il_domains = NULL;
992 ignorelist_free (il_block_devices);
993 il_block_devices = NULL;
994 ignorelist_free (il_interface_devices);
995 il_interface_devices = NULL;
997 return 0;
998 }
1000 void
1001 module_register (void)
1002 {
1003 plugin_register_config (PLUGIN_NAME,
1004 lv_config,
1005 config_keys, NR_CONFIG_KEYS);
1006 plugin_register_init (PLUGIN_NAME, lv_init);
1007 plugin_register_read (PLUGIN_NAME, lv_read);
1008 plugin_register_shutdown (PLUGIN_NAME, lv_shutdown);
1009 }
1011 /*
1012 * vim: shiftwidth=4 tabstop=8 softtabstop=4 expandtab fdm=marker
1013 */