4babed899d5915e82a9cfeb2e01429afdabfcec0
[linux-2.6.git] / drivers / lguest / lguest_device.c
1 /*P:050 Lguest guests use a very simple method to describe devices.  It's a
2  * series of device descriptors contained just above the top of normal Guest
3  * memory.
4  *
5  * We use the standard "virtio" device infrastructure, which provides us with a
6  * console, a network and a block driver.  Each one expects some configuration
7  * information and a "virtqueue" or two to send and receive data. :*/
8 #include <linux/init.h>
9 #include <linux/bootmem.h>
10 #include <linux/lguest_launcher.h>
11 #include <linux/virtio.h>
12 #include <linux/virtio_config.h>
13 #include <linux/interrupt.h>
14 #include <linux/virtio_ring.h>
15 #include <linux/err.h>
16 #include <asm/io.h>
17 #include <asm/paravirt.h>
18 #include <asm/lguest_hcall.h>
19
20 /* The pointer to our (page) of device descriptions. */
21 static void *lguest_devices;
22
23 /* For Guests, device memory can be used as normal memory, so we cast away the
24  * __iomem to quieten sparse. */
25 static inline void *lguest_map(unsigned long phys_addr, unsigned long pages)
26 {
27         return (__force void *)ioremap_cache(phys_addr, PAGE_SIZE*pages);
28 }
29
30 static inline void lguest_unmap(void *addr)
31 {
32         iounmap((__force void __iomem *)addr);
33 }
34
35 /*D:100 Each lguest device is just a virtio device plus a pointer to its entry
36  * in the lguest_devices page. */
37 struct lguest_device {
38         struct virtio_device vdev;
39
40         /* The entry in the lguest_devices page for this device. */
41         struct lguest_device_desc *desc;
42 };
43
44 /* Since the virtio infrastructure hands us a pointer to the virtio_device all
45  * the time, it helps to have a curt macro to get a pointer to the struct
46  * lguest_device it's enclosed in.  */
47 #define to_lgdev(vd) container_of(vd, struct lguest_device, vdev)
48
49 /*D:130
50  * Device configurations
51  *
52  * The configuration information for a device consists of one or more
53  * virtqueues, a feature bitmap, and some configuration bytes.  The
54  * configuration bytes don't really matter to us: the Launcher sets them up, and
55  * the driver will look at them during setup.
56  *
57  * A convenient routine to return the device's virtqueue config array:
58  * immediately after the descriptor. */
59 static struct lguest_vqconfig *lg_vq(const struct lguest_device_desc *desc)
60 {
61         return (void *)(desc + 1);
62 }
63
64 /* The features come immediately after the virtqueues. */
65 static u8 *lg_features(const struct lguest_device_desc *desc)
66 {
67         return (void *)(lg_vq(desc) + desc->num_vq);
68 }
69
70 /* The config space comes after the two feature bitmasks. */
71 static u8 *lg_config(const struct lguest_device_desc *desc)
72 {
73         return lg_features(desc) + desc->feature_len * 2;
74 }
75
76 /* The total size of the config page used by this device (incl. desc) */
77 static unsigned desc_size(const struct lguest_device_desc *desc)
78 {
79         return sizeof(*desc)
80                 + desc->num_vq * sizeof(struct lguest_vqconfig)
81                 + desc->feature_len * 2
82                 + desc->config_len;
83 }
84
85 /* This gets the device's feature bits. */
86 static u32 lg_get_features(struct virtio_device *vdev)
87 {
88         unsigned int i;
89         u32 features = 0;
90         struct lguest_device_desc *desc = to_lgdev(vdev)->desc;
91         u8 *in_features = lg_features(desc);
92
93         /* We do this the slow but generic way. */
94         for (i = 0; i < min(desc->feature_len * 8, 32); i++)
95                 if (in_features[i / 8] & (1 << (i % 8)))
96                         features |= (1 << i);
97
98         return features;
99 }
100
101 /* The virtio core takes the features the Host offers, and copies the
102  * ones supported by the driver into the vdev->features array.  Once
103  * that's all sorted out, this routine is called so we can tell the
104  * Host which features we understand and accept. */
105 static void lg_finalize_features(struct virtio_device *vdev)
106 {
107         unsigned int i, bits;
108         struct lguest_device_desc *desc = to_lgdev(vdev)->desc;
109         /* Second half of bitmap is features we accept. */
110         u8 *out_features = lg_features(desc) + desc->feature_len;
111
112         /* Give virtio_ring a chance to accept features. */
113         vring_transport_features(vdev);
114
115         /* The vdev->feature array is a Linux bitmask: this isn't the
116          * same as a the simple array of bits used by lguest devices
117          * for features.  So we do this slow, manual conversion which is
118          * completely general. */
119         memset(out_features, 0, desc->feature_len);
120         bits = min_t(unsigned, desc->feature_len, sizeof(vdev->features)) * 8;
121         for (i = 0; i < bits; i++) {
122                 if (test_bit(i, vdev->features))
123                         out_features[i / 8] |= (1 << (i % 8));
124         }
125 }
126
127 /* Once they've found a field, getting a copy of it is easy. */
128 static void lg_get(struct virtio_device *vdev, unsigned int offset,
129                    void *buf, unsigned len)
130 {
131         struct lguest_device_desc *desc = to_lgdev(vdev)->desc;
132
133         /* Check they didn't ask for more than the length of the config! */
134         BUG_ON(offset + len > desc->config_len);
135         memcpy(buf, lg_config(desc) + offset, len);
136 }
137
138 /* Setting the contents is also trivial. */
139 static void lg_set(struct virtio_device *vdev, unsigned int offset,
140                    const void *buf, unsigned len)
141 {
142         struct lguest_device_desc *desc = to_lgdev(vdev)->desc;
143
144         /* Check they didn't ask for more than the length of the config! */
145         BUG_ON(offset + len > desc->config_len);
146         memcpy(lg_config(desc) + offset, buf, len);
147 }
148
149 /* The operations to get and set the status word just access the status field
150  * of the device descriptor. */
151 static u8 lg_get_status(struct virtio_device *vdev)
152 {
153         return to_lgdev(vdev)->desc->status;
154 }
155
156 /* To notify on status updates, we (ab)use the NOTIFY hypercall, with the
157  * descriptor address of the device.  A zero status means "reset". */
158 static void set_status(struct virtio_device *vdev, u8 status)
159 {
160         unsigned long offset = (void *)to_lgdev(vdev)->desc - lguest_devices;
161
162         /* We set the status. */
163         to_lgdev(vdev)->desc->status = status;
164         kvm_hypercall1(LHCALL_NOTIFY, (max_pfn << PAGE_SHIFT) + offset);
165 }
166
167 static void lg_set_status(struct virtio_device *vdev, u8 status)
168 {
169         BUG_ON(!status);
170         set_status(vdev, status);
171 }
172
173 static void lg_reset(struct virtio_device *vdev)
174 {
175         set_status(vdev, 0);
176 }
177
178 /*
179  * Virtqueues
180  *
181  * The other piece of infrastructure virtio needs is a "virtqueue": a way of
182  * the Guest device registering buffers for the other side to read from or
183  * write into (ie. send and receive buffers).  Each device can have multiple
184  * virtqueues: for example the console driver uses one queue for sending and
185  * another for receiving.
186  *
187  * Fortunately for us, a very fast shared-memory-plus-descriptors virtqueue
188  * already exists in virtio_ring.c.  We just need to connect it up.
189  *
190  * We start with the information we need to keep about each virtqueue.
191  */
192
193 /*D:140 This is the information we remember about each virtqueue. */
194 struct lguest_vq_info
195 {
196         /* A copy of the information contained in the device config. */
197         struct lguest_vqconfig config;
198
199         /* The address where we mapped the virtio ring, so we can unmap it. */
200         void *pages;
201 };
202
203 /* When the virtio_ring code wants to prod the Host, it calls us here and we
204  * make a hypercall.  We hand the physical address of the virtqueue so the Host
205  * knows which virtqueue we're talking about. */
206 static void lg_notify(struct virtqueue *vq)
207 {
208         /* We store our virtqueue information in the "priv" pointer of the
209          * virtqueue structure. */
210         struct lguest_vq_info *lvq = vq->priv;
211
212         kvm_hypercall1(LHCALL_NOTIFY, lvq->config.pfn << PAGE_SHIFT);
213 }
214
215 /* An extern declaration inside a C file is bad form.  Don't do it. */
216 extern void lguest_setup_irq(unsigned int irq);
217
218 /* This routine finds the first virtqueue described in the configuration of
219  * this device and sets it up.
220  *
221  * This is kind of an ugly duckling.  It'd be nicer to have a standard
222  * representation of a virtqueue in the configuration space, but it seems that
223  * everyone wants to do it differently.  The KVM coders want the Guest to
224  * allocate its own pages and tell the Host where they are, but for lguest it's
225  * simpler for the Host to simply tell us where the pages are.
226  *
227  * So we provide drivers with a "find the Nth virtqueue and set it up"
228  * function. */
229 static struct virtqueue *lg_find_vq(struct virtio_device *vdev,
230                                     unsigned index,
231                                     void (*callback)(struct virtqueue *vq),
232                                     const char *name)
233 {
234         struct lguest_device *ldev = to_lgdev(vdev);
235         struct lguest_vq_info *lvq;
236         struct virtqueue *vq;
237         int err;
238
239         /* We must have this many virtqueues. */
240         if (index >= ldev->desc->num_vq)
241                 return ERR_PTR(-ENOENT);
242
243         lvq = kmalloc(sizeof(*lvq), GFP_KERNEL);
244         if (!lvq)
245                 return ERR_PTR(-ENOMEM);
246
247         /* Make a copy of the "struct lguest_vqconfig" entry, which sits after
248          * the descriptor.  We need a copy because the config space might not
249          * be aligned correctly. */
250         memcpy(&lvq->config, lg_vq(ldev->desc)+index, sizeof(lvq->config));
251
252         printk("Mapping virtqueue %i addr %lx\n", index,
253                (unsigned long)lvq->config.pfn << PAGE_SHIFT);
254         /* Figure out how many pages the ring will take, and map that memory */
255         lvq->pages = lguest_map((unsigned long)lvq->config.pfn << PAGE_SHIFT,
256                                 DIV_ROUND_UP(vring_size(lvq->config.num,
257                                                         LGUEST_VRING_ALIGN),
258                                              PAGE_SIZE));
259         if (!lvq->pages) {
260                 err = -ENOMEM;
261                 goto free_lvq;
262         }
263
264         /* OK, tell virtio_ring.c to set up a virtqueue now we know its size
265          * and we've got a pointer to its pages. */
266         vq = vring_new_virtqueue(lvq->config.num, LGUEST_VRING_ALIGN,
267                                  vdev, lvq->pages, lg_notify, callback, name);
268         if (!vq) {
269                 err = -ENOMEM;
270                 goto unmap;
271         }
272
273         /* Make sure the interrupt is allocated. */
274         lguest_setup_irq(lvq->config.irq);
275
276         /* Tell the interrupt for this virtqueue to go to the virtio_ring
277          * interrupt handler. */
278         /* FIXME: We used to have a flag for the Host to tell us we could use
279          * the interrupt as a source of randomness: it'd be nice to have that
280          * back.. */
281         err = request_irq(lvq->config.irq, vring_interrupt, IRQF_SHARED,
282                           dev_name(&vdev->dev), vq);
283         if (err)
284                 goto destroy_vring;
285
286         /* Last of all we hook up our 'struct lguest_vq_info" to the
287          * virtqueue's priv pointer. */
288         vq->priv = lvq;
289         return vq;
290
291 destroy_vring:
292         vring_del_virtqueue(vq);
293 unmap:
294         lguest_unmap(lvq->pages);
295 free_lvq:
296         kfree(lvq);
297         return ERR_PTR(err);
298 }
299 /*:*/
300
301 /* Cleaning up a virtqueue is easy */
302 static void lg_del_vq(struct virtqueue *vq)
303 {
304         struct lguest_vq_info *lvq = vq->priv;
305
306         /* Release the interrupt */
307         free_irq(lvq->config.irq, vq);
308         /* Tell virtio_ring.c to free the virtqueue. */
309         vring_del_virtqueue(vq);
310         /* Unmap the pages containing the ring. */
311         lguest_unmap(lvq->pages);
312         /* Free our own queue information. */
313         kfree(lvq);
314 }
315
316 /* The ops structure which hooks everything together. */
317 static struct virtio_config_ops lguest_config_ops = {
318         .get_features = lg_get_features,
319         .finalize_features = lg_finalize_features,
320         .get = lg_get,
321         .set = lg_set,
322         .get_status = lg_get_status,
323         .set_status = lg_set_status,
324         .reset = lg_reset,
325         .find_vq = lg_find_vq,
326         .del_vq = lg_del_vq,
327 };
328
329 /* The root device for the lguest virtio devices.  This makes them appear as
330  * /sys/devices/lguest/0,1,2 not /sys/devices/0,1,2. */
331 static struct device *lguest_root;
332
333 /*D:120 This is the core of the lguest bus: actually adding a new device.
334  * It's a separate function because it's neater that way, and because an
335  * earlier version of the code supported hotplug and unplug.  They were removed
336  * early on because they were never used.
337  *
338  * As Andrew Tridgell says, "Untested code is buggy code".
339  *
340  * It's worth reading this carefully: we start with a pointer to the new device
341  * descriptor in the "lguest_devices" page, and the offset into the device
342  * descriptor page so we can uniquely identify it if things go badly wrong. */
343 static void add_lguest_device(struct lguest_device_desc *d,
344                               unsigned int offset)
345 {
346         struct lguest_device *ldev;
347
348         /* Start with zeroed memory; Linux's device layer seems to count on
349          * it. */
350         ldev = kzalloc(sizeof(*ldev), GFP_KERNEL);
351         if (!ldev) {
352                 printk(KERN_EMERG "Cannot allocate lguest dev %u type %u\n",
353                        offset, d->type);
354                 return;
355         }
356
357         /* This devices' parent is the lguest/ dir. */
358         ldev->vdev.dev.parent = lguest_root;
359         /* We have a unique device index thanks to the dev_index counter. */
360         ldev->vdev.id.device = d->type;
361         /* We have a simple set of routines for querying the device's
362          * configuration information and setting its status. */
363         ldev->vdev.config = &lguest_config_ops;
364         /* And we remember the device's descriptor for lguest_config_ops. */
365         ldev->desc = d;
366
367         /* register_virtio_device() sets up the generic fields for the struct
368          * virtio_device and calls device_register().  This makes the bus
369          * infrastructure look for a matching driver. */
370         if (register_virtio_device(&ldev->vdev) != 0) {
371                 printk(KERN_ERR "Failed to register lguest dev %u type %u\n",
372                        offset, d->type);
373                 kfree(ldev);
374         }
375 }
376
377 /*D:110 scan_devices() simply iterates through the device page.  The type 0 is
378  * reserved to mean "end of devices". */
379 static void scan_devices(void)
380 {
381         unsigned int i;
382         struct lguest_device_desc *d;
383
384         /* We start at the page beginning, and skip over each entry. */
385         for (i = 0; i < PAGE_SIZE; i += desc_size(d)) {
386                 d = lguest_devices + i;
387
388                 /* Once we hit a zero, stop. */
389                 if (d->type == 0)
390                         break;
391
392                 printk("Device at %i has size %u\n", i, desc_size(d));
393                 add_lguest_device(d, i);
394         }
395 }
396
397 /*D:105 Fairly early in boot, lguest_devices_init() is called to set up the
398  * lguest device infrastructure.  We check that we are a Guest by checking
399  * pv_info.name: there are other ways of checking, but this seems most
400  * obvious to me.
401  *
402  * So we can access the "struct lguest_device_desc"s easily, we map that memory
403  * and store the pointer in the global "lguest_devices".  Then we register a
404  * root device from which all our devices will hang (this seems to be the
405  * correct sysfs incantation).
406  *
407  * Finally we call scan_devices() which adds all the devices found in the
408  * lguest_devices page. */
409 static int __init lguest_devices_init(void)
410 {
411         if (strcmp(pv_info.name, "lguest") != 0)
412                 return 0;
413
414         lguest_root = root_device_register("lguest");
415         if (IS_ERR(lguest_root))
416                 panic("Could not register lguest root");
417
418         /* Devices are in a single page above top of "normal" mem */
419         lguest_devices = lguest_map(max_pfn<<PAGE_SHIFT, 1);
420
421         scan_devices();
422         return 0;
423 }
424 /* We do this after core stuff, but before the drivers. */
425 postcore_initcall(lguest_devices_init);
426
427 /*D:150 At this point in the journey we used to now wade through the lguest
428  * devices themselves: net, block and console.  Since they're all now virtio
429  * devices rather than lguest-specific, I've decided to ignore them.  Mostly,
430  * they're kind of boring.  But this does mean you'll never experience the
431  * thrill of reading the forbidden love scene buried deep in the block driver.
432  *
433  * "make Launcher" beckons, where we answer questions like "Where do Guests
434  * come from?", and "What do you do when someone asks for optimization?". */