1 /******************************************************************************
4 * This is the kernel equivalent of the "xs" library. We don't need everything
5 * and we use xenbus_comms for communication.
7 * Copyright (C) 2005 Rusty Russell, IBM Corporation
9 * This program is free software; you can redistribute it and/or
10 * modify it under the terms of the GNU General Public License version 2
11 * as published by the Free Software Foundation; or, when distributed
12 * separately from the Linux kernel or incorporated into other
13 * software packages, subject to the following license:
15 * Permission is hereby granted, free of charge, to any person obtaining a copy
16 * of this source file (the "Software"), to deal in the Software without
17 * restriction, including without limitation the rights to use, copy, modify,
18 * merge, publish, distribute, sublicense, and/or sell copies of the Software,
19 * and to permit persons to whom the Software is furnished to do so, subject to
20 * the following conditions:
22 * The above copyright notice and this permission notice shall be included in
23 * all copies or substantial portions of the Software.
25 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
26 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
27 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
28 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
29 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
30 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
34 #include <linux/unistd.h>
35 #include <linux/errno.h>
36 #include <linux/types.h>
37 #include <linux/uio.h>
38 #include <linux/kernel.h>
39 #include <linux/string.h>
40 #include <linux/err.h>
41 #include <linux/slab.h>
42 #include <linux/fcntl.h>
43 #include <linux/kthread.h>
44 #include <linux/rwsem.h>
45 #include <linux/module.h>
46 #include <linux/mutex.h>
47 #include <xen/xenbus.h>
49 #include "xenbus_comms.h"
51 struct xs_stored_msg
{
52 struct list_head list
;
54 struct xsd_sockmsg hdr
;
62 /* Queued watch events. */
64 struct xenbus_watch
*handle
;
66 unsigned int vec_size
;
72 /* A list of replies. Currently only one will ever be outstanding. */
73 struct list_head reply_list
;
74 spinlock_t reply_lock
;
75 wait_queue_head_t reply_waitq
;
78 * Mutex ordering: transaction_mutex -> watch_mutex -> request_mutex.
79 * response_mutex is never taken simultaneously with the other three.
81 * transaction_mutex must be held before incrementing
82 * transaction_count. The mutex is held when a suspend is in
83 * progress to prevent new transactions starting.
85 * When decrementing transaction_count to zero the wait queue
86 * should be woken up, the suspend code waits for count to
90 /* One request at a time. */
91 struct mutex request_mutex
;
93 /* Protect xenbus reader thread against save/restore. */
94 struct mutex response_mutex
;
96 /* Protect transactions against save/restore. */
97 struct mutex transaction_mutex
;
98 atomic_t transaction_count
;
99 wait_queue_head_t transaction_wq
;
101 /* Protect watch (de)register against save/restore. */
102 struct rw_semaphore watch_mutex
;
105 static struct xs_handle xs_state
;
107 /* List of registered watches, and a lock to protect it. */
108 static LIST_HEAD(watches
);
109 static DEFINE_SPINLOCK(watches_lock
);
111 /* List of pending watch callback events, and a lock to protect it. */
112 static LIST_HEAD(watch_events
);
113 static DEFINE_SPINLOCK(watch_events_lock
);
116 * Details of the xenwatch callback kernel thread. The thread waits on the
117 * watch_events_waitq for work to do (queued on watch_events list). When it
118 * wakes up it acquires the xenwatch_mutex before reading the list and
121 static pid_t xenwatch_pid
;
122 static DEFINE_MUTEX(xenwatch_mutex
);
123 static DECLARE_WAIT_QUEUE_HEAD(watch_events_waitq
);
125 static int get_error(const char *errorstring
)
129 for (i
= 0; strcmp(errorstring
, xsd_errors
[i
].errstring
) != 0; i
++) {
130 if (i
== ARRAY_SIZE(xsd_errors
) - 1) {
132 "XENBUS xen store gave: unknown error %s",
137 return xsd_errors
[i
].errnum
;
140 static void *read_reply(enum xsd_sockmsg_type
*type
, unsigned int *len
)
142 struct xs_stored_msg
*msg
;
145 spin_lock(&xs_state
.reply_lock
);
147 while (list_empty(&xs_state
.reply_list
)) {
148 spin_unlock(&xs_state
.reply_lock
);
149 /* XXX FIXME: Avoid synchronous wait for response here. */
150 wait_event(xs_state
.reply_waitq
,
151 !list_empty(&xs_state
.reply_list
));
152 spin_lock(&xs_state
.reply_lock
);
155 msg
= list_entry(xs_state
.reply_list
.next
,
156 struct xs_stored_msg
, list
);
157 list_del(&msg
->list
);
159 spin_unlock(&xs_state
.reply_lock
);
161 *type
= msg
->hdr
.type
;
164 body
= msg
->u
.reply
.body
;
171 static void transaction_start(void)
173 mutex_lock(&xs_state
.transaction_mutex
);
174 atomic_inc(&xs_state
.transaction_count
);
175 mutex_unlock(&xs_state
.transaction_mutex
);
178 static void transaction_end(void)
180 if (atomic_dec_and_test(&xs_state
.transaction_count
))
181 wake_up(&xs_state
.transaction_wq
);
184 static void transaction_suspend(void)
186 mutex_lock(&xs_state
.transaction_mutex
);
187 wait_event(xs_state
.transaction_wq
,
188 atomic_read(&xs_state
.transaction_count
) == 0);
191 static void transaction_resume(void)
193 mutex_unlock(&xs_state
.transaction_mutex
);
196 void *xenbus_dev_request_and_reply(struct xsd_sockmsg
*msg
)
199 struct xsd_sockmsg req_msg
= *msg
;
202 if (req_msg
.type
== XS_TRANSACTION_START
)
205 mutex_lock(&xs_state
.request_mutex
);
207 err
= xb_write(msg
, sizeof(*msg
) + msg
->len
);
209 msg
->type
= XS_ERROR
;
212 ret
= read_reply(&msg
->type
, &msg
->len
);
214 mutex_unlock(&xs_state
.request_mutex
);
216 if ((msg
->type
== XS_TRANSACTION_END
) ||
217 ((req_msg
.type
== XS_TRANSACTION_START
) &&
218 (msg
->type
== XS_ERROR
)))
223 EXPORT_SYMBOL(xenbus_dev_request_and_reply
);
225 /* Send message to xs, get kmalloc'ed reply. ERR_PTR() on error. */
226 static void *xs_talkv(struct xenbus_transaction t
,
227 enum xsd_sockmsg_type type
,
228 const struct kvec
*iovec
,
229 unsigned int num_vecs
,
232 struct xsd_sockmsg msg
;
241 for (i
= 0; i
< num_vecs
; i
++)
242 msg
.len
+= iovec
[i
].iov_len
;
244 mutex_lock(&xs_state
.request_mutex
);
246 err
= xb_write(&msg
, sizeof(msg
));
248 mutex_unlock(&xs_state
.request_mutex
);
252 for (i
= 0; i
< num_vecs
; i
++) {
253 err
= xb_write(iovec
[i
].iov_base
, iovec
[i
].iov_len
);
255 mutex_unlock(&xs_state
.request_mutex
);
260 ret
= read_reply(&msg
.type
, len
);
262 mutex_unlock(&xs_state
.request_mutex
);
267 if (msg
.type
== XS_ERROR
) {
268 err
= get_error(ret
);
270 return ERR_PTR(-err
);
273 if (msg
.type
!= type
) {
274 if (printk_ratelimit())
276 "XENBUS unexpected type [%d], expected [%d]\n",
279 return ERR_PTR(-EINVAL
);
284 /* Simplified version of xs_talkv: single message. */
285 static void *xs_single(struct xenbus_transaction t
,
286 enum xsd_sockmsg_type type
,
292 iovec
.iov_base
= (void *)string
;
293 iovec
.iov_len
= strlen(string
) + 1;
294 return xs_talkv(t
, type
, &iovec
, 1, len
);
297 /* Many commands only need an ack, don't care what it says. */
298 static int xs_error(char *reply
)
301 return PTR_ERR(reply
);
306 static unsigned int count_strings(const char *strings
, unsigned int len
)
311 for (p
= strings
, num
= 0; p
< strings
+ len
; p
+= strlen(p
) + 1)
317 /* Return the path to dir with /name appended. Buffer must be kfree()'ed. */
318 static char *join(const char *dir
, const char *name
)
322 if (strlen(name
) == 0)
323 buffer
= kasprintf(GFP_NOIO
| __GFP_HIGH
, "%s", dir
);
325 buffer
= kasprintf(GFP_NOIO
| __GFP_HIGH
, "%s/%s", dir
, name
);
326 return (!buffer
) ? ERR_PTR(-ENOMEM
) : buffer
;
329 static char **split(char *strings
, unsigned int len
, unsigned int *num
)
333 /* Count the strings. */
334 *num
= count_strings(strings
, len
);
336 /* Transfer to one big alloc for easy freeing. */
337 ret
= kmalloc(*num
* sizeof(char *) + len
, GFP_NOIO
| __GFP_HIGH
);
340 return ERR_PTR(-ENOMEM
);
342 memcpy(&ret
[*num
], strings
, len
);
345 strings
= (char *)&ret
[*num
];
346 for (p
= strings
, *num
= 0; p
< strings
+ len
; p
+= strlen(p
) + 1)
352 char **xenbus_directory(struct xenbus_transaction t
,
353 const char *dir
, const char *node
, unsigned int *num
)
355 char *strings
, *path
;
358 path
= join(dir
, node
);
360 return (char **)path
;
362 strings
= xs_single(t
, XS_DIRECTORY
, path
, &len
);
365 return (char **)strings
;
367 return split(strings
, len
, num
);
369 EXPORT_SYMBOL_GPL(xenbus_directory
);
371 /* Check if a path exists. Return 1 if it does. */
372 int xenbus_exists(struct xenbus_transaction t
,
373 const char *dir
, const char *node
)
378 d
= xenbus_directory(t
, dir
, node
, &dir_n
);
384 EXPORT_SYMBOL_GPL(xenbus_exists
);
386 /* Get the value of a single file.
387 * Returns a kmalloced value: call free() on it after use.
388 * len indicates length in bytes.
390 void *xenbus_read(struct xenbus_transaction t
,
391 const char *dir
, const char *node
, unsigned int *len
)
396 path
= join(dir
, node
);
400 ret
= xs_single(t
, XS_READ
, path
, len
);
404 EXPORT_SYMBOL_GPL(xenbus_read
);
406 /* Write the value of a single file.
407 * Returns -err on failure.
409 int xenbus_write(struct xenbus_transaction t
,
410 const char *dir
, const char *node
, const char *string
)
413 struct kvec iovec
[2];
416 path
= join(dir
, node
);
418 return PTR_ERR(path
);
420 iovec
[0].iov_base
= (void *)path
;
421 iovec
[0].iov_len
= strlen(path
) + 1;
422 iovec
[1].iov_base
= (void *)string
;
423 iovec
[1].iov_len
= strlen(string
);
425 ret
= xs_error(xs_talkv(t
, XS_WRITE
, iovec
, ARRAY_SIZE(iovec
), NULL
));
429 EXPORT_SYMBOL_GPL(xenbus_write
);
431 /* Create a new directory. */
432 int xenbus_mkdir(struct xenbus_transaction t
,
433 const char *dir
, const char *node
)
438 path
= join(dir
, node
);
440 return PTR_ERR(path
);
442 ret
= xs_error(xs_single(t
, XS_MKDIR
, path
, NULL
));
446 EXPORT_SYMBOL_GPL(xenbus_mkdir
);
448 /* Destroy a file or directory (directories must be empty). */
449 int xenbus_rm(struct xenbus_transaction t
, const char *dir
, const char *node
)
454 path
= join(dir
, node
);
456 return PTR_ERR(path
);
458 ret
= xs_error(xs_single(t
, XS_RM
, path
, NULL
));
462 EXPORT_SYMBOL_GPL(xenbus_rm
);
464 /* Start a transaction: changes by others will not be seen during this
465 * transaction, and changes will not be visible to others until end.
467 int xenbus_transaction_start(struct xenbus_transaction
*t
)
473 id_str
= xs_single(XBT_NIL
, XS_TRANSACTION_START
, "", NULL
);
474 if (IS_ERR(id_str
)) {
476 return PTR_ERR(id_str
);
479 t
->id
= simple_strtoul(id_str
, NULL
, 0);
483 EXPORT_SYMBOL_GPL(xenbus_transaction_start
);
485 /* End a transaction.
486 * If abandon is true, transaction is discarded instead of committed.
488 int xenbus_transaction_end(struct xenbus_transaction t
, int abort
)
494 strcpy(abortstr
, "F");
496 strcpy(abortstr
, "T");
498 err
= xs_error(xs_single(t
, XS_TRANSACTION_END
, abortstr
, NULL
));
504 EXPORT_SYMBOL_GPL(xenbus_transaction_end
);
506 /* Single read and scanf: returns -errno or num scanned. */
507 int xenbus_scanf(struct xenbus_transaction t
,
508 const char *dir
, const char *node
, const char *fmt
, ...)
514 val
= xenbus_read(t
, dir
, node
, NULL
);
519 ret
= vsscanf(val
, fmt
, ap
);
522 /* Distinctive errno. */
527 EXPORT_SYMBOL_GPL(xenbus_scanf
);
529 /* Single printf and write: returns -errno or 0. */
530 int xenbus_printf(struct xenbus_transaction t
,
531 const char *dir
, const char *node
, const char *fmt
, ...)
535 #define PRINTF_BUFFER_SIZE 4096
538 printf_buffer
= kmalloc(PRINTF_BUFFER_SIZE
, GFP_NOIO
| __GFP_HIGH
);
539 if (printf_buffer
== NULL
)
543 ret
= vsnprintf(printf_buffer
, PRINTF_BUFFER_SIZE
, fmt
, ap
);
546 BUG_ON(ret
> PRINTF_BUFFER_SIZE
-1);
547 ret
= xenbus_write(t
, dir
, node
, printf_buffer
);
549 kfree(printf_buffer
);
553 EXPORT_SYMBOL_GPL(xenbus_printf
);
555 /* Takes tuples of names, scanf-style args, and void **, NULL terminated. */
556 int xenbus_gather(struct xenbus_transaction t
, const char *dir
, ...)
563 while (ret
== 0 && (name
= va_arg(ap
, char *)) != NULL
) {
564 const char *fmt
= va_arg(ap
, char *);
565 void *result
= va_arg(ap
, void *);
568 p
= xenbus_read(t
, dir
, name
, NULL
);
574 if (sscanf(p
, fmt
, result
) == 0)
578 *(char **)result
= p
;
583 EXPORT_SYMBOL_GPL(xenbus_gather
);
585 static int xs_watch(const char *path
, const char *token
)
589 iov
[0].iov_base
= (void *)path
;
590 iov
[0].iov_len
= strlen(path
) + 1;
591 iov
[1].iov_base
= (void *)token
;
592 iov
[1].iov_len
= strlen(token
) + 1;
594 return xs_error(xs_talkv(XBT_NIL
, XS_WATCH
, iov
,
595 ARRAY_SIZE(iov
), NULL
));
598 static int xs_unwatch(const char *path
, const char *token
)
602 iov
[0].iov_base
= (char *)path
;
603 iov
[0].iov_len
= strlen(path
) + 1;
604 iov
[1].iov_base
= (char *)token
;
605 iov
[1].iov_len
= strlen(token
) + 1;
607 return xs_error(xs_talkv(XBT_NIL
, XS_UNWATCH
, iov
,
608 ARRAY_SIZE(iov
), NULL
));
611 static struct xenbus_watch
*find_watch(const char *token
)
613 struct xenbus_watch
*i
, *cmp
;
615 cmp
= (void *)simple_strtoul(token
, NULL
, 16);
617 list_for_each_entry(i
, &watches
, list
)
624 static void xs_reset_watches(void)
628 err
= xs_error(xs_single(XBT_NIL
, XS_RESET_WATCHES
, "", NULL
));
629 if (err
&& err
!= -EEXIST
)
630 printk(KERN_WARNING
"xs_reset_watches failed: %d\n", err
);
633 /* Register callback to watch this node. */
634 int register_xenbus_watch(struct xenbus_watch
*watch
)
636 /* Pointer in ascii is the token. */
637 char token
[sizeof(watch
) * 2 + 1];
640 sprintf(token
, "%lX", (long)watch
);
642 down_read(&xs_state
.watch_mutex
);
644 spin_lock(&watches_lock
);
645 BUG_ON(find_watch(token
));
646 list_add(&watch
->list
, &watches
);
647 spin_unlock(&watches_lock
);
649 err
= xs_watch(watch
->node
, token
);
652 spin_lock(&watches_lock
);
653 list_del(&watch
->list
);
654 spin_unlock(&watches_lock
);
657 up_read(&xs_state
.watch_mutex
);
661 EXPORT_SYMBOL_GPL(register_xenbus_watch
);
663 void unregister_xenbus_watch(struct xenbus_watch
*watch
)
665 struct xs_stored_msg
*msg
, *tmp
;
666 char token
[sizeof(watch
) * 2 + 1];
669 sprintf(token
, "%lX", (long)watch
);
671 down_read(&xs_state
.watch_mutex
);
673 spin_lock(&watches_lock
);
674 BUG_ON(!find_watch(token
));
675 list_del(&watch
->list
);
676 spin_unlock(&watches_lock
);
678 err
= xs_unwatch(watch
->node
, token
);
681 "XENBUS Failed to release watch %s: %i\n",
684 up_read(&xs_state
.watch_mutex
);
686 /* Make sure there are no callbacks running currently (unless
688 if (current
->pid
!= xenwatch_pid
)
689 mutex_lock(&xenwatch_mutex
);
691 /* Cancel pending watch events. */
692 spin_lock(&watch_events_lock
);
693 list_for_each_entry_safe(msg
, tmp
, &watch_events
, list
) {
694 if (msg
->u
.watch
.handle
!= watch
)
696 list_del(&msg
->list
);
697 kfree(msg
->u
.watch
.vec
);
700 spin_unlock(&watch_events_lock
);
702 if (current
->pid
!= xenwatch_pid
)
703 mutex_unlock(&xenwatch_mutex
);
705 EXPORT_SYMBOL_GPL(unregister_xenbus_watch
);
707 void xs_suspend(void)
709 transaction_suspend();
710 down_write(&xs_state
.watch_mutex
);
711 mutex_lock(&xs_state
.request_mutex
);
712 mutex_lock(&xs_state
.response_mutex
);
717 struct xenbus_watch
*watch
;
718 char token
[sizeof(watch
) * 2 + 1];
722 mutex_unlock(&xs_state
.response_mutex
);
723 mutex_unlock(&xs_state
.request_mutex
);
724 transaction_resume();
726 /* No need for watches_lock: the watch_mutex is sufficient. */
727 list_for_each_entry(watch
, &watches
, list
) {
728 sprintf(token
, "%lX", (long)watch
);
729 xs_watch(watch
->node
, token
);
732 up_write(&xs_state
.watch_mutex
);
735 void xs_suspend_cancel(void)
737 mutex_unlock(&xs_state
.response_mutex
);
738 mutex_unlock(&xs_state
.request_mutex
);
739 up_write(&xs_state
.watch_mutex
);
740 mutex_unlock(&xs_state
.transaction_mutex
);
743 static int xenwatch_thread(void *unused
)
745 struct list_head
*ent
;
746 struct xs_stored_msg
*msg
;
749 wait_event_interruptible(watch_events_waitq
,
750 !list_empty(&watch_events
));
752 if (kthread_should_stop())
755 mutex_lock(&xenwatch_mutex
);
757 spin_lock(&watch_events_lock
);
758 ent
= watch_events
.next
;
759 if (ent
!= &watch_events
)
761 spin_unlock(&watch_events_lock
);
763 if (ent
!= &watch_events
) {
764 msg
= list_entry(ent
, struct xs_stored_msg
, list
);
765 msg
->u
.watch
.handle
->callback(
767 (const char **)msg
->u
.watch
.vec
,
768 msg
->u
.watch
.vec_size
);
769 kfree(msg
->u
.watch
.vec
);
773 mutex_unlock(&xenwatch_mutex
);
779 static int process_msg(void)
781 struct xs_stored_msg
*msg
;
786 * We must disallow save/restore while reading a xenstore message.
787 * A partial read across s/r leaves us out of sync with xenstored.
790 err
= xb_wait_for_data_to_read();
793 mutex_lock(&xs_state
.response_mutex
);
794 if (xb_data_to_read())
796 /* We raced with save/restore: pending data 'disappeared'. */
797 mutex_unlock(&xs_state
.response_mutex
);
801 msg
= kmalloc(sizeof(*msg
), GFP_NOIO
| __GFP_HIGH
);
807 err
= xb_read(&msg
->hdr
, sizeof(msg
->hdr
));
813 body
= kmalloc(msg
->hdr
.len
+ 1, GFP_NOIO
| __GFP_HIGH
);
820 err
= xb_read(body
, msg
->hdr
.len
);
826 body
[msg
->hdr
.len
] = '\0';
828 if (msg
->hdr
.type
== XS_WATCH_EVENT
) {
829 msg
->u
.watch
.vec
= split(body
, msg
->hdr
.len
,
830 &msg
->u
.watch
.vec_size
);
831 if (IS_ERR(msg
->u
.watch
.vec
)) {
832 err
= PTR_ERR(msg
->u
.watch
.vec
);
837 spin_lock(&watches_lock
);
838 msg
->u
.watch
.handle
= find_watch(
839 msg
->u
.watch
.vec
[XS_WATCH_TOKEN
]);
840 if (msg
->u
.watch
.handle
!= NULL
) {
841 spin_lock(&watch_events_lock
);
842 list_add_tail(&msg
->list
, &watch_events
);
843 wake_up(&watch_events_waitq
);
844 spin_unlock(&watch_events_lock
);
846 kfree(msg
->u
.watch
.vec
);
849 spin_unlock(&watches_lock
);
851 msg
->u
.reply
.body
= body
;
852 spin_lock(&xs_state
.reply_lock
);
853 list_add_tail(&msg
->list
, &xs_state
.reply_list
);
854 spin_unlock(&xs_state
.reply_lock
);
855 wake_up(&xs_state
.reply_waitq
);
859 mutex_unlock(&xs_state
.response_mutex
);
863 static int xenbus_thread(void *unused
)
870 printk(KERN_WARNING
"XENBUS error %d while reading "
872 if (kthread_should_stop())
882 struct task_struct
*task
;
884 INIT_LIST_HEAD(&xs_state
.reply_list
);
885 spin_lock_init(&xs_state
.reply_lock
);
886 init_waitqueue_head(&xs_state
.reply_waitq
);
888 mutex_init(&xs_state
.request_mutex
);
889 mutex_init(&xs_state
.response_mutex
);
890 mutex_init(&xs_state
.transaction_mutex
);
891 init_rwsem(&xs_state
.watch_mutex
);
892 atomic_set(&xs_state
.transaction_count
, 0);
893 init_waitqueue_head(&xs_state
.transaction_wq
);
895 /* Initialize the shared memory rings to talk to xenstored */
896 err
= xb_init_comms();
900 task
= kthread_run(xenwatch_thread
, NULL
, "xenwatch");
902 return PTR_ERR(task
);
903 xenwatch_pid
= task
->pid
;
905 task
= kthread_run(xenbus_thread
, NULL
, "xenbus");
907 return PTR_ERR(task
);
909 /* shutdown watches for kexec boot */
910 if (xen_hvm_domain())