xen-vtx-unstable
annotate tools/xenstore/xs.c @ 6043:a9ee400a5da9
1) More testing: include tests which I forgot in previous patch, remove
xs_watch_stress, reduce cycles in "make check" random test.
2) xs_crashme: corrupt random packets going to xenstored, watch it
crash.
3) Handle second input from before we finished output on first one.
4) Fix bug where one-arg operations are given zero args.
5) Fix bug where SET_PERMS fails after blocking on transaction.
6) Fix memory leak when DIRECTORY op given no argument.
7) Fail on first memory leak, for better testing.
8) Fix missing waiting_for_ack initialization for new connections.
9) Ensure all input and output is handled for domains so we don't stall.
10) Fix overrun bug in xs_count_strings on non-nul-terminated strings.
11) New test for clients which write without waiting for response.
Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
xs_watch_stress, reduce cycles in "make check" random test.
2) xs_crashme: corrupt random packets going to xenstored, watch it
crash.
3) Handle second input from before we finished output on first one.
4) Fix bug where one-arg operations are given zero args.
5) Fix bug where SET_PERMS fails after blocking on transaction.
6) Fix memory leak when DIRECTORY op given no argument.
7) Fail on first memory leak, for better testing.
8) Fix missing waiting_for_ack initialization for new connections.
9) Ensure all input and output is handled for domains so we don't stall.
10) Fix overrun bug in xs_count_strings on non-nul-terminated strings.
11) New test for clients which write without waiting for response.
Signed-off-by: Rusty Russell <rusty@rustcorp.com.au>
author | kaf24@firebug.cl.cam.ac.uk |
---|---|
date | Mon Aug 08 09:13:19 2005 +0000 (2005-08-08) |
parents | f5b8d2720d6c |
children | 8799d14bef77 9312a3e8a6f8 23979fb12c49 84ee014ebd41 99914b54f7bf 81576d3d1ca8 3a8f27c6d56c cc5f88b719d0 fa0754a9f64f 112d44270733 |
rev | line source |
---|---|
cl349@5357 | 1 /* |
cl349@5357 | 2 Xen Store Daemon interface providing simple tree-like database. |
cl349@5357 | 3 Copyright (C) 2005 Rusty Russell IBM Corporation |
cl349@5357 | 4 |
kaf24@6008 | 5 This library is free software; you can redistribute it and/or |
kaf24@6008 | 6 modify it under the terms of the GNU Lesser General Public |
kaf24@6008 | 7 License as published by the Free Software Foundation; either |
kaf24@6008 | 8 version 2.1 of the License, or (at your option) any later version. |
cl349@5357 | 9 |
kaf24@6008 | 10 This library is distributed in the hope that it will be useful, |
cl349@5357 | 11 but WITHOUT ANY WARRANTY; without even the implied warranty of |
kaf24@6008 | 12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
kaf24@6008 | 13 Lesser General Public License for more details. |
cl349@5357 | 14 |
kaf24@6008 | 15 You should have received a copy of the GNU Lesser General Public |
kaf24@6008 | 16 License along with this library; if not, write to the Free Software |
kaf24@6008 | 17 Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA |
cl349@5357 | 18 */ |
cl349@5357 | 19 |
cl349@5357 | 20 #include <sys/types.h> |
cl349@5357 | 21 #include <sys/stat.h> |
cl349@5357 | 22 #include <fcntl.h> |
cl349@5357 | 23 #include <sys/socket.h> |
cl349@5357 | 24 #include <sys/un.h> |
cl349@5357 | 25 #include <string.h> |
cl349@5357 | 26 #include <unistd.h> |
cl349@5357 | 27 #include <stdbool.h> |
cl349@5357 | 28 #include <stdlib.h> |
cl349@5357 | 29 #include <assert.h> |
cl349@5357 | 30 #include <stdio.h> |
cl349@5357 | 31 #include <signal.h> |
cl349@5357 | 32 #include <stdint.h> |
cl349@5357 | 33 #include <errno.h> |
cl349@5357 | 34 #include "xs.h" |
cl349@5357 | 35 #include "xenstored.h" |
cl349@5357 | 36 #include "xs_lib.h" |
cl349@5357 | 37 #include "utils.h" |
cl349@5357 | 38 |
cl349@5357 | 39 struct xs_handle |
cl349@5357 | 40 { |
cl349@5357 | 41 int fd; |
cl349@5357 | 42 }; |
cl349@5357 | 43 |
cl349@5357 | 44 /* Get the socket from the store daemon handle. |
cl349@5357 | 45 */ |
cl349@5357 | 46 int xs_fileno(struct xs_handle *h) |
cl349@5357 | 47 { |
cl349@5357 | 48 return h->fd; |
cl349@5357 | 49 } |
cl349@5357 | 50 |
cl349@5357 | 51 static struct xs_handle *get_socket(const char *connect_to) |
cl349@5357 | 52 { |
cl349@5357 | 53 struct sockaddr_un addr; |
cl349@5357 | 54 int sock, saved_errno; |
cl349@5357 | 55 struct xs_handle *h = NULL; |
cl349@5357 | 56 |
cl349@5357 | 57 sock = socket(PF_UNIX, SOCK_STREAM, 0); |
cl349@5357 | 58 if (sock < 0) |
cl349@5357 | 59 return NULL; |
cl349@5357 | 60 |
cl349@5357 | 61 addr.sun_family = AF_UNIX; |
cl349@5357 | 62 strcpy(addr.sun_path, connect_to); |
cl349@5357 | 63 |
cl349@5357 | 64 if (connect(sock, (struct sockaddr *)&addr, sizeof(addr)) == 0) { |
cl349@5357 | 65 h = malloc(sizeof(*h)); |
cl349@5357 | 66 if (h) { |
cl349@5357 | 67 h->fd = sock; |
cl349@5357 | 68 return h; |
cl349@5357 | 69 } |
cl349@5357 | 70 } |
cl349@5357 | 71 |
cl349@5357 | 72 saved_errno = errno; |
cl349@5357 | 73 close(sock); |
cl349@5357 | 74 free(h); |
cl349@5357 | 75 errno = saved_errno; |
cl349@5357 | 76 return NULL; |
cl349@5357 | 77 } |
cl349@5357 | 78 |
cl349@5357 | 79 struct xs_handle *xs_daemon_open(void) |
cl349@5357 | 80 { |
cl349@5357 | 81 return get_socket(xs_daemon_socket()); |
cl349@5357 | 82 } |
cl349@5357 | 83 |
cl349@5357 | 84 struct xs_handle *xs_daemon_open_readonly(void) |
cl349@5357 | 85 { |
cl349@5357 | 86 return get_socket(xs_daemon_socket_ro()); |
cl349@5357 | 87 } |
cl349@5357 | 88 |
cl349@5357 | 89 void xs_daemon_close(struct xs_handle *h) |
cl349@5357 | 90 { |
cl349@5357 | 91 if (h->fd >= 0) |
cl349@5357 | 92 close(h->fd); |
cl349@5357 | 93 free(h); |
cl349@5357 | 94 } |
cl349@5357 | 95 |
cl349@5357 | 96 static bool read_all(int fd, void *data, unsigned int len) |
cl349@5357 | 97 { |
cl349@5357 | 98 while (len) { |
cl349@5357 | 99 int done; |
cl349@5357 | 100 |
cl349@5357 | 101 done = read(fd, data, len); |
cl349@5357 | 102 if (done < 0) { |
cl349@5357 | 103 if (errno == EINTR) |
cl349@5357 | 104 continue; |
cl349@5357 | 105 return false; |
cl349@5357 | 106 } |
cl349@5357 | 107 if (done == 0) { |
cl349@5357 | 108 /* It closed fd on us? EBADF is appropriate. */ |
cl349@5357 | 109 errno = EBADF; |
cl349@5357 | 110 return false; |
cl349@5357 | 111 } |
cl349@5357 | 112 data += done; |
cl349@5357 | 113 len -= done; |
cl349@5357 | 114 } |
cl349@5357 | 115 |
cl349@5357 | 116 return true; |
cl349@5357 | 117 } |
cl349@5357 | 118 |
cl349@5357 | 119 #ifdef XSTEST |
cl349@5357 | 120 #define read_all read_all_choice |
kaf24@5423 | 121 #define xs_write_all write_all_choice |
cl349@5357 | 122 #endif |
cl349@5357 | 123 |
cl349@5357 | 124 static int get_error(const char *errorstring) |
cl349@5357 | 125 { |
cl349@5357 | 126 unsigned int i; |
cl349@5357 | 127 |
cl349@5357 | 128 for (i = 0; !streq(errorstring, xsd_errors[i].errstring); i++) |
cl349@5357 | 129 if (i == ARRAY_SIZE(xsd_errors) - 1) |
cl349@5357 | 130 return EINVAL; |
cl349@5357 | 131 return xsd_errors[i].errnum; |
cl349@5357 | 132 } |
cl349@5357 | 133 |
cl349@5522 | 134 /* Adds extra nul terminator, because we generally (always?) hold strings. */ |
cl349@5357 | 135 static void *read_reply(int fd, enum xsd_sockmsg_type *type, unsigned int *len) |
cl349@5357 | 136 { |
cl349@5357 | 137 struct xsd_sockmsg msg; |
cl349@5357 | 138 void *ret; |
cl349@5357 | 139 int saved_errno; |
cl349@5357 | 140 |
cl349@5357 | 141 if (!read_all(fd, &msg, sizeof(msg))) |
cl349@5357 | 142 return NULL; |
cl349@5357 | 143 |
cl349@5522 | 144 ret = malloc(msg.len + 1); |
cl349@5357 | 145 if (!ret) |
cl349@5357 | 146 return NULL; |
cl349@5357 | 147 |
cl349@5357 | 148 if (!read_all(fd, ret, msg.len)) { |
cl349@5357 | 149 saved_errno = errno; |
cl349@5357 | 150 free(ret); |
cl349@5357 | 151 errno = saved_errno; |
cl349@5357 | 152 return NULL; |
cl349@5357 | 153 } |
cl349@5357 | 154 |
cl349@5357 | 155 *type = msg.type; |
cl349@5357 | 156 if (len) |
cl349@5357 | 157 *len = msg.len; |
cl349@5522 | 158 ((char *)ret)[msg.len] = '\0'; |
cl349@5357 | 159 return ret; |
cl349@5357 | 160 } |
cl349@5357 | 161 |
cl349@5357 | 162 /* Send message to xs, get malloc'ed reply. NULL and set errno on error. */ |
cl349@5357 | 163 static void *xs_talkv(struct xs_handle *h, enum xsd_sockmsg_type type, |
cl349@5478 | 164 const struct iovec *iovec, unsigned int num_vecs, |
cl349@5357 | 165 unsigned int *len) |
cl349@5357 | 166 { |
cl349@5357 | 167 struct xsd_sockmsg msg; |
cl349@5357 | 168 void *ret = NULL; |
cl349@5357 | 169 int saved_errno; |
cl349@5357 | 170 unsigned int i; |
cl349@5357 | 171 struct sigaction ignorepipe, oldact; |
cl349@5357 | 172 |
cl349@5357 | 173 msg.type = type; |
cl349@5357 | 174 msg.len = 0; |
cl349@5357 | 175 for (i = 0; i < num_vecs; i++) |
cl349@5357 | 176 msg.len += iovec[i].iov_len; |
cl349@5357 | 177 |
cl349@5357 | 178 ignorepipe.sa_handler = SIG_IGN; |
cl349@5357 | 179 sigemptyset(&ignorepipe.sa_mask); |
cl349@5357 | 180 ignorepipe.sa_flags = 0; |
cl349@5357 | 181 sigaction(SIGPIPE, &ignorepipe, &oldact); |
cl349@5357 | 182 |
kaf24@5423 | 183 if (!xs_write_all(h->fd, &msg, sizeof(msg))) |
cl349@5357 | 184 goto fail; |
cl349@5357 | 185 |
cl349@5357 | 186 for (i = 0; i < num_vecs; i++) |
kaf24@5423 | 187 if (!xs_write_all(h->fd, iovec[i].iov_base, iovec[i].iov_len)) |
cl349@5357 | 188 goto fail; |
cl349@5357 | 189 |
cl349@5357 | 190 /* Watches can have fired before reply comes: daemon detects |
cl349@5357 | 191 * and re-transmits, so we can ignore this. */ |
cl349@5357 | 192 do { |
cl349@5357 | 193 free(ret); |
cl349@5357 | 194 ret = read_reply(h->fd, &msg.type, len); |
cl349@5357 | 195 if (!ret) |
cl349@5357 | 196 goto fail; |
cl349@5357 | 197 } while (msg.type == XS_WATCH_EVENT); |
cl349@5357 | 198 |
cl349@5357 | 199 sigaction(SIGPIPE, &oldact, NULL); |
cl349@5357 | 200 if (msg.type == XS_ERROR) { |
cl349@5357 | 201 saved_errno = get_error(ret); |
cl349@5357 | 202 free(ret); |
cl349@5357 | 203 errno = saved_errno; |
cl349@5357 | 204 return NULL; |
cl349@5357 | 205 } |
cl349@5357 | 206 |
kaf24@6043 | 207 if (msg.type != type) { |
kaf24@6043 | 208 free(ret); |
kaf24@6043 | 209 saved_errno = EBADF; |
kaf24@6043 | 210 goto close_fd; |
kaf24@6043 | 211 |
kaf24@6043 | 212 } |
cl349@5357 | 213 return ret; |
cl349@5357 | 214 |
cl349@5357 | 215 fail: |
cl349@5357 | 216 /* We're in a bad state, so close fd. */ |
cl349@5357 | 217 saved_errno = errno; |
cl349@5357 | 218 sigaction(SIGPIPE, &oldact, NULL); |
kaf24@6043 | 219 close_fd: |
cl349@5357 | 220 close(h->fd); |
cl349@5357 | 221 h->fd = -1; |
cl349@5357 | 222 errno = saved_errno; |
cl349@5357 | 223 return NULL; |
cl349@5357 | 224 } |
cl349@5357 | 225 |
cl349@5357 | 226 /* free(), but don't change errno. */ |
cl349@5357 | 227 static void free_no_errno(void *p) |
cl349@5357 | 228 { |
cl349@5357 | 229 int saved_errno = errno; |
cl349@5357 | 230 free(p); |
cl349@5357 | 231 errno = saved_errno; |
cl349@5357 | 232 } |
cl349@5357 | 233 |
cl349@5357 | 234 /* Simplified version of xs_talkv: single message. */ |
cl349@5357 | 235 static void *xs_single(struct xs_handle *h, enum xsd_sockmsg_type type, |
cl349@5357 | 236 const char *string, unsigned int *len) |
cl349@5357 | 237 { |
cl349@5357 | 238 struct iovec iovec; |
cl349@5357 | 239 |
cl349@5357 | 240 iovec.iov_base = (void *)string; |
cl349@5357 | 241 iovec.iov_len = strlen(string) + 1; |
cl349@5357 | 242 return xs_talkv(h, type, &iovec, 1, len); |
cl349@5357 | 243 } |
cl349@5357 | 244 |
cl349@5357 | 245 static bool xs_bool(char *reply) |
cl349@5357 | 246 { |
cl349@5357 | 247 if (!reply) |
cl349@5357 | 248 return false; |
cl349@5357 | 249 free(reply); |
cl349@5357 | 250 return true; |
cl349@5357 | 251 } |
cl349@5357 | 252 |
cl349@5357 | 253 char **xs_directory(struct xs_handle *h, const char *path, unsigned int *num) |
cl349@5357 | 254 { |
cl349@5357 | 255 char *strings, *p, **ret; |
cl349@5357 | 256 unsigned int len; |
cl349@5357 | 257 |
cl349@5357 | 258 strings = xs_single(h, XS_DIRECTORY, path, &len); |
cl349@5357 | 259 if (!strings) |
cl349@5357 | 260 return NULL; |
cl349@5357 | 261 |
cl349@5357 | 262 /* Count the strings. */ |
kaf24@5423 | 263 *num = xs_count_strings(strings, len); |
cl349@5357 | 264 |
cl349@5357 | 265 /* Transfer to one big alloc for easy freeing. */ |
cl349@5357 | 266 ret = malloc(*num * sizeof(char *) + len); |
cl349@5357 | 267 if (!ret) { |
cl349@5357 | 268 free_no_errno(strings); |
cl349@5357 | 269 return NULL; |
cl349@5357 | 270 } |
cl349@5357 | 271 memcpy(&ret[*num], strings, len); |
cl349@5357 | 272 free_no_errno(strings); |
cl349@5357 | 273 |
cl349@5357 | 274 strings = (char *)&ret[*num]; |
cl349@5357 | 275 for (p = strings, *num = 0; p < strings + len; p += strlen(p) + 1) |
cl349@5357 | 276 ret[(*num)++] = p; |
cl349@5357 | 277 return ret; |
cl349@5357 | 278 } |
cl349@5357 | 279 |
cl349@5522 | 280 /* Get the value of a single file, nul terminated. |
cl349@5357 | 281 * Returns a malloced value: call free() on it after use. |
cl349@5522 | 282 * len indicates length in bytes, not including the nul. |
cl349@5357 | 283 */ |
cl349@5357 | 284 void *xs_read(struct xs_handle *h, const char *path, unsigned int *len) |
cl349@5357 | 285 { |
cl349@5357 | 286 return xs_single(h, XS_READ, path, len); |
cl349@5357 | 287 } |
cl349@5357 | 288 |
cl349@5357 | 289 /* Write the value of a single file. |
cl349@5357 | 290 * Returns false on failure. createflags can be 0, O_CREAT, or O_CREAT|O_EXCL. |
cl349@5357 | 291 */ |
cl349@5357 | 292 bool xs_write(struct xs_handle *h, const char *path, |
cl349@5357 | 293 const void *data, unsigned int len, int createflags) |
cl349@5357 | 294 { |
cl349@5357 | 295 const char *flags; |
cl349@5357 | 296 struct iovec iovec[3]; |
cl349@5357 | 297 |
cl349@5357 | 298 /* Format: Flags (as string), path, data. */ |
cl349@5357 | 299 if (createflags == 0) |
cl349@5357 | 300 flags = XS_WRITE_NONE; |
cl349@5357 | 301 else if (createflags == O_CREAT) |
cl349@5357 | 302 flags = XS_WRITE_CREATE; |
cl349@5357 | 303 else if (createflags == (O_CREAT|O_EXCL)) |
cl349@5357 | 304 flags = XS_WRITE_CREATE_EXCL; |
cl349@5357 | 305 else { |
cl349@5357 | 306 errno = EINVAL; |
cl349@5357 | 307 return false; |
cl349@5357 | 308 } |
cl349@5357 | 309 |
cl349@5357 | 310 iovec[0].iov_base = (void *)path; |
cl349@5357 | 311 iovec[0].iov_len = strlen(path) + 1; |
cl349@5357 | 312 iovec[1].iov_base = (void *)flags; |
cl349@5357 | 313 iovec[1].iov_len = strlen(flags) + 1; |
cl349@5357 | 314 iovec[2].iov_base = (void *)data; |
cl349@5357 | 315 iovec[2].iov_len = len; |
cl349@5357 | 316 |
cl349@5357 | 317 return xs_bool(xs_talkv(h, XS_WRITE, iovec, ARRAY_SIZE(iovec), NULL)); |
cl349@5357 | 318 } |
cl349@5357 | 319 |
cl349@5357 | 320 /* Create a new directory. |
cl349@5357 | 321 * Returns false on failure. |
cl349@5357 | 322 */ |
cl349@5357 | 323 bool xs_mkdir(struct xs_handle *h, const char *path) |
cl349@5357 | 324 { |
cl349@5357 | 325 return xs_bool(xs_single(h, XS_MKDIR, path, NULL)); |
cl349@5357 | 326 } |
cl349@5357 | 327 |
cl349@5357 | 328 /* Destroy a file or directory (directories must be empty). |
cl349@5357 | 329 * Returns false on failure. |
cl349@5357 | 330 */ |
cl349@5357 | 331 bool xs_rm(struct xs_handle *h, const char *path) |
cl349@5357 | 332 { |
cl349@5357 | 333 return xs_bool(xs_single(h, XS_RM, path, NULL)); |
cl349@5357 | 334 } |
cl349@5357 | 335 |
cl349@5357 | 336 /* Get permissions of node (first element is owner). |
cl349@5357 | 337 * Returns malloced array, or NULL: call free() after use. |
cl349@5357 | 338 */ |
cl349@5357 | 339 struct xs_permissions *xs_get_permissions(struct xs_handle *h, |
cl349@5478 | 340 const char *path, unsigned int *num) |
cl349@5357 | 341 { |
cl349@5357 | 342 char *strings; |
cl349@5357 | 343 unsigned int len; |
cl349@5357 | 344 struct xs_permissions *ret; |
cl349@5357 | 345 |
cl349@5357 | 346 strings = xs_single(h, XS_GET_PERMS, path, &len); |
cl349@5357 | 347 if (!strings) |
cl349@5357 | 348 return NULL; |
cl349@5357 | 349 |
cl349@5357 | 350 /* Count the strings: each one perms then domid. */ |
kaf24@5423 | 351 *num = xs_count_strings(strings, len); |
cl349@5357 | 352 |
cl349@5357 | 353 /* Transfer to one big alloc for easy freeing. */ |
cl349@5357 | 354 ret = malloc(*num * sizeof(struct xs_permissions)); |
cl349@5357 | 355 if (!ret) { |
cl349@5357 | 356 free_no_errno(strings); |
cl349@5357 | 357 return NULL; |
cl349@5357 | 358 } |
cl349@5357 | 359 |
kaf24@5423 | 360 if (!xs_strings_to_perms(ret, *num, strings)) { |
cl349@5357 | 361 free_no_errno(ret); |
cl349@5357 | 362 ret = NULL; |
cl349@5357 | 363 } |
cl349@5357 | 364 |
cl349@5357 | 365 free(strings); |
cl349@5357 | 366 return ret; |
cl349@5357 | 367 } |
cl349@5357 | 368 |
cl349@5357 | 369 /* Set permissions of node (must be owner). |
cl349@5357 | 370 * Returns false on failure. |
cl349@5357 | 371 */ |
cl349@5357 | 372 bool xs_set_permissions(struct xs_handle *h, const char *path, |
cl349@5357 | 373 struct xs_permissions *perms, |
cl349@5357 | 374 unsigned int num_perms) |
cl349@5357 | 375 { |
cl349@5357 | 376 unsigned int i; |
cl349@5357 | 377 struct iovec iov[1+num_perms]; |
cl349@5357 | 378 |
cl349@5357 | 379 iov[0].iov_base = (void *)path; |
cl349@5357 | 380 iov[0].iov_len = strlen(path) + 1; |
cl349@5357 | 381 |
cl349@5357 | 382 for (i = 0; i < num_perms; i++) { |
cl349@5357 | 383 char buffer[MAX_STRLEN(domid_t)+1]; |
cl349@5357 | 384 |
kaf24@5423 | 385 if (!xs_perm_to_string(&perms[i], buffer)) |
cl349@5357 | 386 goto unwind; |
cl349@5357 | 387 |
cl349@5357 | 388 iov[i+1].iov_base = strdup(buffer); |
cl349@5357 | 389 iov[i+1].iov_len = strlen(buffer) + 1; |
cl349@5357 | 390 if (!iov[i+1].iov_base) |
cl349@5357 | 391 goto unwind; |
cl349@5357 | 392 } |
cl349@5357 | 393 |
cl349@5357 | 394 if (!xs_bool(xs_talkv(h, XS_SET_PERMS, iov, 1+num_perms, NULL))) |
cl349@5357 | 395 goto unwind; |
cl349@5357 | 396 for (i = 0; i < num_perms; i++) |
cl349@5357 | 397 free(iov[i+1].iov_base); |
cl349@5357 | 398 return true; |
cl349@5357 | 399 |
cl349@5357 | 400 unwind: |
cl349@5357 | 401 num_perms = i; |
cl349@5357 | 402 for (i = 0; i < num_perms; i++) |
cl349@5357 | 403 free_no_errno(iov[i+1].iov_base); |
cl349@5357 | 404 return false; |
cl349@5357 | 405 } |
cl349@5357 | 406 |
cl349@5357 | 407 /* Watch a node for changes (poll on fd to detect, or call read_watch()). |
cl349@5357 | 408 * When the node (or any child) changes, fd will become readable. |
cl349@5478 | 409 * Token is returned when watch is read, to allow matching. |
cl349@5357 | 410 * Returns false on failure. |
cl349@5357 | 411 */ |
cl349@5856 | 412 bool xs_watch(struct xs_handle *h, const char *path, const char *token) |
cl349@5357 | 413 { |
cl349@5856 | 414 struct iovec iov[2]; |
cl349@5357 | 415 |
cl349@5357 | 416 iov[0].iov_base = (void *)path; |
cl349@5357 | 417 iov[0].iov_len = strlen(path) + 1; |
cl349@5478 | 418 iov[1].iov_base = (void *)token; |
cl349@5478 | 419 iov[1].iov_len = strlen(token) + 1; |
cl349@5357 | 420 |
cl349@5357 | 421 return xs_bool(xs_talkv(h, XS_WATCH, iov, ARRAY_SIZE(iov), NULL)); |
cl349@5357 | 422 } |
cl349@5357 | 423 |
cl349@5357 | 424 /* Find out what node change was on (will block if nothing pending). |
cl349@5478 | 425 * Returns array of two pointers: path and token, or NULL. |
cl349@5478 | 426 * Call free() after use. |
cl349@5357 | 427 */ |
cl349@5478 | 428 char **xs_read_watch(struct xs_handle *h) |
cl349@5357 | 429 { |
cl349@5357 | 430 struct xsd_sockmsg msg; |
cl349@5478 | 431 char **ret; |
cl349@5357 | 432 |
cl349@5357 | 433 if (!read_all(h->fd, &msg, sizeof(msg))) |
cl349@5357 | 434 return NULL; |
cl349@5357 | 435 |
cl349@5357 | 436 assert(msg.type == XS_WATCH_EVENT); |
cl349@5478 | 437 ret = malloc(sizeof(char *)*2 + msg.len); |
cl349@5478 | 438 if (!ret) |
cl349@5357 | 439 return NULL; |
cl349@5357 | 440 |
cl349@5478 | 441 ret[0] = (char *)(ret + 2); |
cl349@5478 | 442 if (!read_all(h->fd, ret[0], msg.len)) { |
cl349@5478 | 443 free_no_errno(ret); |
cl349@5357 | 444 return NULL; |
cl349@5357 | 445 } |
cl349@5478 | 446 ret[1] = ret[0] + strlen(ret[0]) + 1; |
cl349@5478 | 447 return ret; |
cl349@5357 | 448 } |
cl349@5357 | 449 |
cl349@5357 | 450 /* Acknowledge watch on node. Watches must be acknowledged before |
cl349@5357 | 451 * any other watches can be read. |
cl349@5357 | 452 * Returns false on failure. |
cl349@5357 | 453 */ |
cl349@5478 | 454 bool xs_acknowledge_watch(struct xs_handle *h, const char *token) |
cl349@5357 | 455 { |
cl349@5478 | 456 return xs_bool(xs_single(h, XS_WATCH_ACK, token, NULL)); |
cl349@5357 | 457 } |
cl349@5357 | 458 |
cl349@5357 | 459 /* Remove a watch on a node. |
cl349@5357 | 460 * Returns false on failure (no watch on that node). |
cl349@5357 | 461 */ |
cl349@5478 | 462 bool xs_unwatch(struct xs_handle *h, const char *path, const char *token) |
cl349@5357 | 463 { |
cl349@5478 | 464 struct iovec iov[2]; |
cl349@5478 | 465 |
cl349@5478 | 466 iov[0].iov_base = (char *)path; |
cl349@5478 | 467 iov[0].iov_len = strlen(path) + 1; |
cl349@5478 | 468 iov[1].iov_base = (char *)token; |
cl349@5478 | 469 iov[1].iov_len = strlen(token) + 1; |
cl349@5478 | 470 |
cl349@5478 | 471 return xs_bool(xs_talkv(h, XS_UNWATCH, iov, ARRAY_SIZE(iov), NULL)); |
cl349@5357 | 472 } |
cl349@5357 | 473 |
cl349@5357 | 474 /* Start a transaction: changes by others will not be seen during this |
cl349@5357 | 475 * transaction, and changes will not be visible to others until end. |
cl349@5357 | 476 * Transaction only applies to the given subtree. |
cl349@5357 | 477 * You can only have one transaction at any time. |
cl349@5357 | 478 * Returns false on failure. |
cl349@5357 | 479 */ |
cl349@5357 | 480 bool xs_transaction_start(struct xs_handle *h, const char *subtree) |
cl349@5357 | 481 { |
cl349@5357 | 482 return xs_bool(xs_single(h, XS_TRANSACTION_START, subtree, NULL)); |
cl349@5357 | 483 } |
cl349@5357 | 484 |
cl349@5357 | 485 /* End a transaction. |
cl349@5357 | 486 * If abandon is true, transaction is discarded instead of committed. |
cl349@5357 | 487 * Returns false on failure, which indicates an error: transactions will |
cl349@5357 | 488 * not fail spuriously. |
cl349@5357 | 489 */ |
cl349@5357 | 490 bool xs_transaction_end(struct xs_handle *h, bool abort) |
cl349@5357 | 491 { |
cl349@5357 | 492 char abortstr[2]; |
cl349@5357 | 493 |
cl349@5357 | 494 if (abort) |
cl349@5357 | 495 strcpy(abortstr, "F"); |
cl349@5357 | 496 else |
cl349@5357 | 497 strcpy(abortstr, "T"); |
cl349@5357 | 498 return xs_bool(xs_single(h, XS_TRANSACTION_END, abortstr, NULL)); |
cl349@5357 | 499 } |
cl349@5357 | 500 |
cl349@5357 | 501 /* Introduce a new domain. |
cl349@5357 | 502 * This tells the store daemon about a shared memory page and event channel |
cl349@5357 | 503 * associated with a domain: the domain uses these to communicate. |
cl349@5357 | 504 */ |
cl349@5478 | 505 bool xs_introduce_domain(struct xs_handle *h, domid_t domid, unsigned long mfn, |
cl349@5478 | 506 unsigned int eventchn, const char *path) |
cl349@5357 | 507 { |
cl349@5357 | 508 char domid_str[MAX_STRLEN(domid)]; |
cl349@5357 | 509 char mfn_str[MAX_STRLEN(mfn)]; |
cl349@5357 | 510 char eventchn_str[MAX_STRLEN(eventchn)]; |
cl349@5357 | 511 struct iovec iov[4]; |
cl349@5357 | 512 |
cl349@5357 | 513 sprintf(domid_str, "%u", domid); |
cl349@5357 | 514 sprintf(mfn_str, "%lu", mfn); |
cl349@5357 | 515 sprintf(eventchn_str, "%u", eventchn); |
cl349@5357 | 516 |
cl349@5357 | 517 iov[0].iov_base = domid_str; |
cl349@5357 | 518 iov[0].iov_len = strlen(domid_str) + 1; |
cl349@5357 | 519 iov[1].iov_base = mfn_str; |
cl349@5357 | 520 iov[1].iov_len = strlen(mfn_str) + 1; |
cl349@5357 | 521 iov[2].iov_base = eventchn_str; |
cl349@5357 | 522 iov[2].iov_len = strlen(eventchn_str) + 1; |
cl349@5357 | 523 iov[3].iov_base = (char *)path; |
cl349@5357 | 524 iov[3].iov_len = strlen(path) + 1; |
cl349@5357 | 525 |
cl349@5357 | 526 return xs_bool(xs_talkv(h, XS_INTRODUCE, iov, ARRAY_SIZE(iov), NULL)); |
cl349@5357 | 527 } |
cl349@5357 | 528 |
cl349@5478 | 529 bool xs_release_domain(struct xs_handle *h, domid_t domid) |
cl349@5357 | 530 { |
cl349@5357 | 531 char domid_str[MAX_STRLEN(domid)]; |
cl349@5357 | 532 |
cl349@5357 | 533 sprintf(domid_str, "%u", domid); |
cl349@5357 | 534 |
cl349@5357 | 535 return xs_bool(xs_single(h, XS_RELEASE, domid_str, NULL)); |
cl349@5357 | 536 } |
cl349@5357 | 537 |
cl349@5357 | 538 bool xs_shutdown(struct xs_handle *h) |
cl349@5357 | 539 { |
cl349@5357 | 540 bool ret = xs_bool(xs_single(h, XS_SHUTDOWN, "", NULL)); |
cl349@5357 | 541 if (ret) { |
cl349@5357 | 542 char c; |
cl349@5357 | 543 /* Wait for it to actually shutdown. */ |
cl349@5357 | 544 read(h->fd, &c, 1); |
cl349@5357 | 545 } |
cl349@5357 | 546 return ret; |
cl349@5357 | 547 } |
cl349@5357 | 548 |
cl349@5357 | 549 /* Only useful for DEBUG versions */ |
cl349@5357 | 550 char *xs_debug_command(struct xs_handle *h, const char *cmd, |
cl349@5357 | 551 void *data, unsigned int len) |
cl349@5357 | 552 { |
cl349@5357 | 553 struct iovec iov[2]; |
cl349@5357 | 554 |
cl349@5357 | 555 iov[0].iov_base = (void *)cmd; |
cl349@5357 | 556 iov[0].iov_len = strlen(cmd) + 1; |
cl349@5357 | 557 iov[1].iov_base = data; |
cl349@5357 | 558 iov[1].iov_len = len; |
cl349@5357 | 559 |
cl349@5357 | 560 return xs_talkv(h, XS_DEBUG, iov, ARRAY_SIZE(iov), NULL); |
cl349@5357 | 561 } |