1 #include <avahi-client/publish.h>
2 #include <avahi-client/lookup.h>
4 #include <avahi-common/alternative.h>
5 #include <avahi-common/simple-watch.h>
6 #include <avahi-common/malloc.h>
7 #include <avahi-common/error.h>
8 #include <avahi-common/timeval.h>
9 #include <avahi-common/strlst.h>
11 #include <libvirt/libvirt.h>
16 unsigned int domainid
;
17 AvahiEntryGroup
*group
;
21 typedef struct list_el domu
;
23 domu
* domus
= NULL
; /* list of domains */
24 unsigned int domu_count
= 0;
27 static AvahiStringList
* domainProperties(virDomainPtr thisDomain
) {
28 AvahiStringList
*new = NULL
;
29 // avahi_string_list_new_from_array (NULL, 0);
31 virDomainInterfaceStatsStruct stats
;
32 virDomainGetInfo(thisDomain
, &info
);
33 new = avahi_string_list_add_printf(new, "maxMem=%lu", info
.maxMem
);
34 new = avahi_string_list_add_printf(new, "memory=%lu", info
.memory
);
35 new = avahi_string_list_add_printf(new, "cpuTime=%llu", info
.cpuTime
);
37 /* TODO: multiple network interfaces :( */
39 snprintf(vifname
, 14, "vif%d.0", virDomainGetID(thisDomain
));
43 virDomainInterfaceStats(thisDomain
, vifname
, &stats
, sizeof(stats
));
45 new = avahi_string_list_add_printf(new, "interfaces=eth%d", no
);
46 new = avahi_string_list_add_printf(new, "interface_eth%d_rxbytes=%lld", no
, stats
.rx_bytes
);
47 new = avahi_string_list_add_printf(new, "interface_eth%d_rxpackets=%lld", no
, stats
.rx_packets
);
48 // new = avahi_string_list_add_printf(new, "interface_eth%d_rxerrs=%lld", no, stats.rx_errs);
49 // new = avahi_string_list_add_printf(new, "interface_eth%d_rxdrop=%lld", no, stats.rx_drop);
50 new = avahi_string_list_add_printf(new, "interface_eth%d_txbytes=%lld", no
, stats
.tx_bytes
);
51 new = avahi_string_list_add_printf(new, "interface_eth%d_txpackets=%lld", no
, stats
.tx_packets
);
52 // new = avahi_string_list_add_printf(new, "interface_eth%d_txerrs=%lld", no, stats.tx_errs);
53 // new = avahi_string_list_add_printf(new, "interface_eth%d_txdrop=%lld", no, stats.tx_drop);
58 void remove_everything() {
61 for (i
= 0; i
< domu_count
; i
++) {
62 avahi_entry_group_free(domus
[i
].group
);
63 domus
[i
].group
= NULL
;
72 void create_services(AvahiClient
*c
) {
73 virConnectPtr conn
= virConnectOpenReadOnly(NULL
);
76 int maxid
= virConnectNumOfDomains(conn
);
77 if (maxid
> 1) { /* ignore dom0 */
78 int *ids
= (int *) malloc(sizeof(int) * maxid
);
79 if ((maxid
= virConnectListDomains(conn
, &ids
[0], maxid
)) < 0) {
83 unsigned int domu_count_new
= (maxid
- 1);
84 domu
* domus_old
= domus
;
85 domus
= (domu
*) malloc(sizeof(domu
) * domu_count_new
);
86 const char *type
= "_domu._tcp";
88 for (i
= 0; i
< domu_count_new
; i
++) {
90 domus
[i
].group
= NULL
;
91 domus
[i
].domainid
= ids
[i
+1];
94 // fprintf(stderr, "Looking for %d\n", domus[i].domainid);
96 for (j
= 0; j
< domu_count
; j
++) {
97 // fprintf(stderr, "--> %d\n", domus_old[j].domainid);
98 if (ids
[i
+1] == domus_old
[j
].domainid
) {
99 // fprintf(stderr, "got it\n");
100 virDomainPtr thisDomain
= virDomainLookupByID(conn
, domus
[i
].domainid
);
103 name
= virDomainGetName(thisDomain
);
105 domus
[i
].group
= domus_old
[j
].group
;
106 AvahiStringList
*domainProps
;
107 domainProps
= domainProperties(thisDomain
);
108 domus_old
[j
].keep
= 1;
109 avahi_entry_group_update_service_txt_strlst(domus
[i
].group
, AVAHI_IF_UNSPEC
, AVAHI_PROTO_UNSPEC
, 0, name
, type
, NULL
, domainProps
);
110 avahi_string_list_free(domainProps
);
112 virDomainFree(thisDomain
);
116 if (i
> domu_count
|| domus
[i
].group
== NULL
) {
118 AvahiStringList
*domainProps
;
119 virDomainPtr thisDomain
= virDomainLookupByID(conn
, ids
[i
+1]);
121 domus
[i
].group
= avahi_entry_group_new(c
, entry_group_callback
, NULL
);
122 domainProps
= domainProperties(thisDomain
);
123 name
= virDomainGetName(thisDomain
);
126 avahi_entry_group_add_service_strlst(domus
[i
].group
, AVAHI_IF_UNSPEC
, AVAHI_PROTO_UNSPEC
, 0, name
, type
, NULL
, NULL
, 651, domainProps
);
127 avahi_entry_group_commit(domus
[i
].group
);
132 avahi_string_list_free(domainProps
);
133 virDomainFree(thisDomain
);
138 for (i
= 0; i
< domu_count
; i
++) {
139 // fprintf(stderr, "id: %d, keep: %d\n", domus_old[i].domainid, domus_old[i].keep);
140 if (domus_old
[i
].keep
== 0 && domus_old
[i
].group
!= NULL
) {
141 avahi_entry_group_free(domus_old
[i
].group
);
142 domus_old
[i
].group
= NULL
;
147 domu_count
= domu_count_new
;
153 virConnectClose(conn
);
154 //fprintf(stderr, "exit: %s\n", __func__);