1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
|
#define _GNU_SOURCE
#include <stdlib.h>
#include <stdint.h>
#include <string.h>
#include <stdio.h>
#include <xenctrl.h>
#include <xenstore.h>
#include <libxl.h>
#include "init-dom-json.h"
#define DOMNAME_PATH "/local/domain/0/name"
#define DOMID_PATH "/local/domain/0/domid"
static int clear_domid_history(void)
{
int rc = 1;
xentoollog_logger_stdiostream *logger;
libxl_ctx *ctx;
logger = xtl_createlogger_stdiostream(stderr, XTL_ERROR, 0);
if (!logger)
return 1;
if (libxl_ctx_alloc(&ctx, LIBXL_VERSION, 0,
(xentoollog_logger *)logger)) {
fprintf(stderr, "cannot init libxl context\n");
goto outlog;
}
if (!libxl_clear_domid_history(ctx))
rc = 0;
libxl_ctx_free(ctx);
outlog:
xtl_logger_destroy((xentoollog_logger *)logger);
return rc;
}
int main(int argc, char **argv)
{
int rc;
struct xs_handle *xsh = NULL;
xc_interface *xch = NULL;
char *domname_string = NULL, *domid_string = NULL,
*pool_path = NULL, *pool_name = NULL;
xc_cpupoolinfo_t *xcinfo;
unsigned int pool_id = 0;
libxl_uuid uuid;
/* Accept 0 or 1 argument */
if (argc > 2) {
fprintf(stderr, "too many arguments\n");
rc = 1;
goto out;
}
xsh = xs_open(0);
if (!xsh) {
perror("cannot open xenstore connection");
rc = 1;
goto out;
}
xch = xc_interface_open(NULL, NULL, 0);
if (!xch) {
perror("xc_interface_open() failed");
rc = 1;
goto out;
}
/* Sanity check: this program can only be run once. */
domid_string = xs_read(xsh, XBT_NULL, DOMID_PATH, NULL);
domname_string = xs_read(xsh, XBT_NULL, DOMNAME_PATH, NULL);
if (domid_string && domname_string) {
fprintf(stderr, "Dom0 is already set up\n");
rc = 0;
goto out;
}
libxl_uuid_clear(&uuid);
/* If UUID is supplied, parse it. */
if (argc == 2 && libxl_uuid_from_string(&uuid, argv[1])) {
fprintf(stderr, "failed to parse UUID %s\n", argv[1]);
rc = 1;
goto out;
}
if (!libxl_uuid_is_nil(&uuid) &&
xc_domain_sethandle(xch, 0, libxl_uuid_bytearray(&uuid))) {
perror("failed to set Dom0 UUID");
rc = 1;
goto out;
}
rc = gen_stub_json_config(0, &uuid);
if (rc)
goto out;
rc = clear_domid_history();
if (rc)
goto out;
/* Write xenstore entries. */
if (!xs_write(xsh, XBT_NULL, DOMID_PATH, "0", strlen("0"))) {
fprintf(stderr, "cannot set domid for Dom0\n");
rc = 1;
goto out;
}
if (!xs_write(xsh, XBT_NULL, DOMNAME_PATH, "Domain-0",
strlen("Domain-0"))) {
fprintf(stderr, "cannot set domain name for Dom0\n");
rc = 1;
goto out;
}
/* Create an entry in xenstore for each cpupool on the system */
do {
xcinfo = xc_cpupool_getinfo(xch, pool_id);
if (xcinfo != NULL) {
if (xcinfo->cpupool_id != pool_id)
pool_id = xcinfo->cpupool_id;
xc_cpupool_infofree(xch, xcinfo);
if (asprintf(&pool_path, "/local/pool/%d/name", pool_id) <= 0) {
fprintf(stderr, "cannot allocate memory for pool path\n");
rc = 1;
goto out;
}
if (asprintf(&pool_name, "Pool-%d", pool_id) <= 0) {
fprintf(stderr, "cannot allocate memory for pool name\n");
rc = 1;
goto out;
}
pool_id++;
if (!xs_write(xsh, XBT_NULL, pool_path, pool_name,
strlen(pool_name))) {
fprintf(stderr, "cannot set pool name\n");
rc = 1;
goto out;
}
free(pool_name);
free(pool_path);
pool_path = pool_name = NULL;
}
} while(xcinfo != NULL);
printf("Done setting up Dom0\n");
out:
free(pool_path);
free(pool_name);
free(domid_string);
free(domname_string);
xs_close(xsh);
xc_interface_close(xch);
return rc;
}
/*
* Local variables:
* mode: C
* c-basic-offset: 4
* indent-tabs-mode: nil
* End:
*/
|
