drm_bufs.c 43.3 KB
Newer Older
L
Linus Torvalds 已提交
1
/**
D
Dave Airlie 已提交
2
 * \file drm_bufs.c
L
Linus Torvalds 已提交
3
 * Generic buffer template
D
Dave Airlie 已提交
4
 *
L
Linus Torvalds 已提交
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
 * \author Rickard E. (Rik) Faith <faith@valinux.com>
 * \author Gareth Hughes <gareth@valinux.com>
 */

/*
 * Created: Thu Nov 23 03:10:50 2000 by gareth@valinux.com
 *
 * Copyright 1999, 2000 Precision Insight, Inc., Cedar Park, Texas.
 * Copyright 2000 VA Linux Systems, Inc., Sunnyvale, California.
 * All Rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice (including the next
 * paragraph) shall be included in all copies or substantial portions of the
 * Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * VA LINUX SYSTEMS AND/OR ITS SUPPLIERS BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 */

#include <linux/vmalloc.h>
#include "drmP.h"

39
resource_size_t drm_get_resource_start(struct drm_device *dev, unsigned int resource)
L
Linus Torvalds 已提交
40
{
D
Dave Airlie 已提交
41 42 43
	return pci_resource_start(dev->pdev, resource);
}
EXPORT_SYMBOL(drm_get_resource_start);
L
Linus Torvalds 已提交
44

45
resource_size_t drm_get_resource_len(struct drm_device *dev, unsigned int resource)
D
Dave Airlie 已提交
46 47 48
{
	return pci_resource_len(dev->pdev, resource);
}
D
Dave Airlie 已提交
49

D
Dave Airlie 已提交
50
EXPORT_SYMBOL(drm_get_resource_len);
L
Linus Torvalds 已提交
51

D
Dave Airlie 已提交
52
static struct drm_map_list *drm_find_matching_map(struct drm_device *dev,
53
						  struct drm_local_map *map)
D
Dave Airlie 已提交
54
{
D
Dave Airlie 已提交
55
	struct drm_map_list *entry;
56
	list_for_each_entry(entry, &dev->maplist, head) {
57
		if (entry->map && (entry->master == dev->primary->master) && (map->type == entry->map->type) &&
58
		    ((entry->map->offset == map->offset) ||
59
		     ((map->type == _DRM_SHM) && (map->flags&_DRM_CONTAINS_LOCK)))) {
60
			return entry;
D
Dave Airlie 已提交
61 62 63 64
		}
	}

	return NULL;
L
Linus Torvalds 已提交
65 66
}

67
static int drm_map_handle(struct drm_device *dev, struct drm_hash_item *hash,
A
Adrian Bunk 已提交
68
			  unsigned long user_token, int hashed_handle)
69
{
70
	int use_hashed_handle;
D
Dave Airlie 已提交
71
#if (BITS_PER_LONG == 64)
72 73 74 75 76 77
	use_hashed_handle = ((user_token & 0xFFFFFFFF00000000UL) || hashed_handle);
#elif (BITS_PER_LONG == 32)
	use_hashed_handle = hashed_handle;
#else
#error Unsupported long size. Neither 64 nor 32 bits.
#endif
78

79 80
	if (!use_hashed_handle) {
		int ret;
81
		hash->key = user_token >> PAGE_SHIFT;
82 83 84
		ret = drm_ht_insert_item(&dev->map_hash, hash);
		if (ret != -EINVAL)
			return ret;
85
	}
86 87
	return drm_ht_just_insert_please(&dev->map_hash, hash,
					 user_token, 32 - PAGE_SHIFT - 3,
88
					 0, DRM_MAP_HASH_OFFSET >> PAGE_SHIFT);
89
}
90

L
Linus Torvalds 已提交
91
/**
92 93
 * Core function to create a range of memory available for mapping by a
 * non-root process.
L
Linus Torvalds 已提交
94 95 96 97 98
 *
 * Adjusts the memory offset to its absolute value according to the mapping
 * type.  Adds the map to the map list drm_device::maplist. Adds MTRR's where
 * applicable and if supported by the kernel.
 */
99
static int drm_addmap_core(struct drm_device * dev, unsigned int offset,
100
			   unsigned int size, enum drm_map_type type,
D
Dave Airlie 已提交
101 102
			   enum drm_map_flags flags,
			   struct drm_map_list ** maplist)
L
Linus Torvalds 已提交
103
{
104
	struct drm_local_map *map;
D
Dave Airlie 已提交
105
	struct drm_map_list *list;
106
	drm_dma_handle_t *dmah;
107 108
	unsigned long user_token;
	int ret;
L
Linus Torvalds 已提交
109

D
Dave Airlie 已提交
110 111
	map = drm_alloc(sizeof(*map), DRM_MEM_MAPS);
	if (!map)
L
Linus Torvalds 已提交
112 113
		return -ENOMEM;

114 115 116 117
	map->offset = offset;
	map->size = size;
	map->flags = flags;
	map->type = type;
L
Linus Torvalds 已提交
118 119 120 121 122

	/* Only allow shared memory to be removable since we only keep enough
	 * book keeping information about shared memory to allow for removal
	 * when processes fork.
	 */
D
Dave Airlie 已提交
123 124
	if ((map->flags & _DRM_REMOVABLE) && map->type != _DRM_SHM) {
		drm_free(map, sizeof(*map), DRM_MEM_MAPS);
L
Linus Torvalds 已提交
125 126
		return -EINVAL;
	}
D
Dave Airlie 已提交
127 128 129 130
	DRM_DEBUG("offset = 0x%08lx, size = 0x%08lx, type = %d\n",
		  map->offset, map->size, map->type);
	if ((map->offset & (~PAGE_MASK)) || (map->size & (~PAGE_MASK))) {
		drm_free(map, sizeof(*map), DRM_MEM_MAPS);
L
Linus Torvalds 已提交
131 132
		return -EINVAL;
	}
D
Dave Airlie 已提交
133
	map->mtrr = -1;
L
Linus Torvalds 已提交
134 135
	map->handle = NULL;

D
Dave Airlie 已提交
136
	switch (map->type) {
L
Linus Torvalds 已提交
137 138
	case _DRM_REGISTERS:
	case _DRM_FRAME_BUFFER:
D
Dave Airlie 已提交
139
#if !defined(__sparc__) && !defined(__alpha__) && !defined(__ia64__) && !defined(__powerpc64__) && !defined(__x86_64__)
140
		if (map->offset + (map->size-1) < map->offset ||
D
Dave Airlie 已提交
141 142
		    map->offset < virt_to_phys(high_memory)) {
			drm_free(map, sizeof(*map), DRM_MEM_MAPS);
L
Linus Torvalds 已提交
143 144 145 146 147 148
			return -EINVAL;
		}
#endif
#ifdef __alpha__
		map->offset += dev->hose->mem_space->start;
#endif
D
Dave Airlie 已提交
149 150 151 152
		/* Some drivers preinitialize some maps, without the X Server
		 * needing to be aware of it.  Therefore, we just return success
		 * when the server tries to create a duplicate map.
		 */
153 154 155
		list = drm_find_matching_map(dev, map);
		if (list != NULL) {
			if (list->map->size != map->size) {
D
Dave Airlie 已提交
156
				DRM_DEBUG("Matching maps of type %d with "
D
Dave Airlie 已提交
157 158 159
					  "mismatched sizes, (%ld vs %ld)\n",
					  map->type, map->size,
					  list->map->size);
160
				list->map->size = map->size;
D
Dave Airlie 已提交
161 162 163
			}

			drm_free(map, sizeof(*map), DRM_MEM_MAPS);
164
			*maplist = list;
D
Dave Airlie 已提交
165 166 167
			return 0;
		}

L
Linus Torvalds 已提交
168
		if (drm_core_has_MTRR(dev)) {
D
Dave Airlie 已提交
169 170 171 172
			if (map->type == _DRM_FRAME_BUFFER ||
			    (map->flags & _DRM_WRITE_COMBINING)) {
				map->mtrr = mtrr_add(map->offset, map->size,
						     MTRR_TYPE_WRCOMB, 1);
L
Linus Torvalds 已提交
173 174
			}
		}
175
		if (map->type == _DRM_REGISTERS) {
176
			map->handle = ioremap(map->offset, map->size);
177 178 179 180 181
			if (!map->handle) {
				drm_free(map, sizeof(*map), DRM_MEM_MAPS);
				return -ENOMEM;
			}
		}
D
Dave Airlie 已提交
182

L
Linus Torvalds 已提交
183 184
		break;
	case _DRM_SHM:
185 186 187 188 189 190 191 192 193 194 195 196 197
		list = drm_find_matching_map(dev, map);
		if (list != NULL) {
			if(list->map->size != map->size) {
				DRM_DEBUG("Matching maps of type %d with "
					  "mismatched sizes, (%ld vs %ld)\n",
					  map->type, map->size, list->map->size);
				list->map->size = map->size;
			}

			drm_free(map, sizeof(*map), DRM_MEM_MAPS);
			*maplist = list;
			return 0;
		}
198
		map->handle = vmalloc_user(map->size);
D
Dave Airlie 已提交
199 200 201 202
		DRM_DEBUG("%lu %d %p\n",
			  map->size, drm_order(map->size), map->handle);
		if (!map->handle) {
			drm_free(map, sizeof(*map), DRM_MEM_MAPS);
L
Linus Torvalds 已提交
203 204 205
			return -ENOMEM;
		}
		map->offset = (unsigned long)map->handle;
D
Dave Airlie 已提交
206
		if (map->flags & _DRM_CONTAINS_LOCK) {
L
Linus Torvalds 已提交
207
			/* Prevent a 2nd X Server from creating a 2nd lock */
208
			if (dev->primary->master->lock.hw_lock != NULL) {
D
Dave Airlie 已提交
209 210
				vfree(map->handle);
				drm_free(map, sizeof(*map), DRM_MEM_MAPS);
L
Linus Torvalds 已提交
211 212
				return -EBUSY;
			}
213
			dev->sigdata.lock = dev->primary->master->lock.hw_lock = map->handle;	/* Pointer to lock */
L
Linus Torvalds 已提交
214 215
		}
		break;
216
	case _DRM_AGP: {
D
Dave Airlie 已提交
217
		struct drm_agp_mem *entry;
218 219 220 221 222 223
		int valid = 0;

		if (!drm_core_has_AGP(dev)) {
			drm_free(map, sizeof(*map), DRM_MEM_MAPS);
			return -EINVAL;
		}
L
Linus Torvalds 已提交
224
#ifdef __alpha__
225
		map->offset += dev->hose->mem_space->start;
L
Linus Torvalds 已提交
226
#endif
227 228 229 230 231
		/* In some cases (i810 driver), user space may have already
		 * added the AGP base itself, because dev->agp->base previously
		 * only got set during AGP enable.  So, only add the base
		 * address if the map's offset isn't already within the
		 * aperture.
232
		 */
233 234 235 236 237
		if (map->offset < dev->agp->base ||
		    map->offset > dev->agp->base +
		    dev->agp->agp_info.aper_size * 1024 * 1024 - 1) {
			map->offset += dev->agp->base;
		}
238 239 240 241 242 243 244 245
		map->mtrr = dev->agp->agp_mtrr;	/* for getmap */

		/* This assumes the DRM is in total control of AGP space.
		 * It's not always the case as AGP can be in the control
		 * of user space (i.e. i810 driver). So this loop will get
		 * skipped and we double check that dev->agp->memory is
		 * actually set as well as being invalid before EPERM'ing
		 */
246
		list_for_each_entry(entry, &dev->agp->memory, head) {
247 248 249 250 251
			if ((map->offset >= entry->bound) &&
			    (map->offset + map->size <= entry->bound + entry->pages * PAGE_SIZE)) {
				valid = 1;
				break;
			}
L
Linus Torvalds 已提交
252
		}
253
		if (!list_empty(&dev->agp->memory) && !valid) {
254 255 256 257 258
			drm_free(map, sizeof(*map), DRM_MEM_MAPS);
			return -EPERM;
		}
		DRM_DEBUG("AGP offset = 0x%08lx, size = 0x%08lx\n", map->offset, map->size);

J
Jesse Barnes 已提交
259 260 261
		break;
	case _DRM_GEM:
		DRM_ERROR("tried to rmmap GEM object\n");
L
Linus Torvalds 已提交
262
		break;
263
	}
L
Linus Torvalds 已提交
264 265 266 267 268
	case _DRM_SCATTER_GATHER:
		if (!dev->sg) {
			drm_free(map, sizeof(*map), DRM_MEM_MAPS);
			return -EINVAL;
		}
269
		map->offset += (unsigned long)dev->sg->virtual;
L
Linus Torvalds 已提交
270
		break;
D
Dave Airlie 已提交
271
	case _DRM_CONSISTENT:
D
Dave Airlie 已提交
272
		/* dma_addr_t is 64bit on i386 with CONFIG_HIGHMEM64G,
273
		 * As we're limiting the address to 2^32-1 (or less),
D
Dave Airlie 已提交
274 275
		 * casting it down to 32 bits is no problem, but we
		 * need to point to a 64bit variable first. */
276 277
		dmah = drm_pci_alloc(dev, map->size, map->size, 0xffffffffUL);
		if (!dmah) {
D
Dave Airlie 已提交
278 279 280
			drm_free(map, sizeof(*map), DRM_MEM_MAPS);
			return -ENOMEM;
		}
281 282 283
		map->handle = dmah->vaddr;
		map->offset = (unsigned long)dmah->busaddr;
		kfree(dmah);
D
Dave Airlie 已提交
284
		break;
L
Linus Torvalds 已提交
285
	default:
D
Dave Airlie 已提交
286
		drm_free(map, sizeof(*map), DRM_MEM_MAPS);
L
Linus Torvalds 已提交
287 288 289 290
		return -EINVAL;
	}

	list = drm_alloc(sizeof(*list), DRM_MEM_MAPS);
D
Dave Airlie 已提交
291
	if (!list) {
292
		if (map->type == _DRM_REGISTERS)
293
			iounmap(map->handle);
L
Linus Torvalds 已提交
294 295 296 297 298 299
		drm_free(map, sizeof(*map), DRM_MEM_MAPS);
		return -EINVAL;
	}
	memset(list, 0, sizeof(*list));
	list->map = map;

D
Dave Airlie 已提交
300
	mutex_lock(&dev->struct_mutex);
301
	list_add(&list->head, &dev->maplist);
302

303
	/* Assign a 32-bit handle */
D
Dave Airlie 已提交
304
	/* We do it here so that dev->struct_mutex protects the increment */
305 306
	user_token = (map->type == _DRM_SHM) ? (unsigned long)map->handle :
		map->offset;
307
	ret = drm_map_handle(dev, &list->hash, user_token, 0);
308
	if (ret) {
309
		if (map->type == _DRM_REGISTERS)
310
			iounmap(map->handle);
311 312 313 314 315 316
		drm_free(map, sizeof(*map), DRM_MEM_MAPS);
		drm_free(list, sizeof(*list), DRM_MEM_MAPS);
		mutex_unlock(&dev->struct_mutex);
		return ret;
	}

317
	list->user_token = list->hash.key << PAGE_SHIFT;
D
Dave Airlie 已提交
318
	mutex_unlock(&dev->struct_mutex);
L
Linus Torvalds 已提交
319

320
	list->master = dev->primary->master;
321
	*maplist = list;
322
	return 0;
323
	}
324

325
int drm_addmap(struct drm_device * dev, unsigned int offset,
326
	       unsigned int size, enum drm_map_type type,
327
	       enum drm_map_flags flags, struct drm_local_map ** map_ptr)
328
{
D
Dave Airlie 已提交
329
	struct drm_map_list *list;
330 331 332 333 334 335 336
	int rc;

	rc = drm_addmap_core(dev, offset, size, type, flags, &list);
	if (!rc)
		*map_ptr = list->map;
	return rc;
}
D
Dave Airlie 已提交
337

338 339
EXPORT_SYMBOL(drm_addmap);

340 341 342 343 344 345 346 347 348 349 350
/**
 * Ioctl to specify a range of memory that is available for mapping by a
 * non-root process.
 *
 * \param inode device inode.
 * \param file_priv DRM file private.
 * \param cmd command.
 * \param arg pointer to a drm_map structure.
 * \return zero on success or a negative value on error.
 *
 */
351 352
int drm_addmap_ioctl(struct drm_device *dev, void *data,
		     struct drm_file *file_priv)
353
{
354
	struct drm_map *map = data;
D
Dave Airlie 已提交
355
	struct drm_map_list *maplist;
356 357
	int err;

358
	if (!(capable(CAP_SYS_ADMIN) || map->type == _DRM_AGP || map->type == _DRM_SHM))
359 360
		return -EPERM;

361 362
	err = drm_addmap_core(dev, map->offset, map->size, map->type,
			      map->flags, &maplist);
363

D
Dave Airlie 已提交
364
	if (err)
365
		return err;
366

367
	/* avoid a warning on 64-bit, this casting isn't very nice, but the API is set so too late */
368
	map->handle = (void *)(unsigned long)maplist->user_token;
L
Linus Torvalds 已提交
369
	return 0;
D
Dave Airlie 已提交
370
}
L
Linus Torvalds 已提交
371 372 373 374 375 376 377 378 379

/**
 * Remove a map private from list and deallocate resources if the mapping
 * isn't in use.
 *
 * Searches the map on drm_device::maplist, removes it from the list, see if
 * its being used, and free any associate resource (such as MTRR's) if it's not
 * being on use.
 *
380
 * \sa drm_addmap
L
Linus Torvalds 已提交
381
 */
382
int drm_rmmap_locked(struct drm_device *dev, struct drm_local_map *map)
L
Linus Torvalds 已提交
383
{
D
Dave Airlie 已提交
384
	struct drm_map_list *r_list = NULL, *list_t;
D
Dave Airlie 已提交
385
	drm_dma_handle_t dmah;
386
	int found = 0;
387
	struct drm_master *master;
L
Linus Torvalds 已提交
388

D
Dave Airlie 已提交
389
	/* Find the list entry for the map and remove it */
390
	list_for_each_entry_safe(r_list, list_t, &dev->maplist, head) {
D
Dave Airlie 已提交
391
		if (r_list->map == map) {
392
			master = r_list->master;
393
			list_del(&r_list->head);
394 395
			drm_ht_remove_key(&dev->map_hash,
					  r_list->user_token >> PAGE_SHIFT);
396 397
			drm_free(r_list, sizeof(*r_list), DRM_MEM_MAPS);
			found = 1;
D
Dave Airlie 已提交
398 399
			break;
		}
L
Linus Torvalds 已提交
400 401
	}

402
	if (!found)
L
Linus Torvalds 已提交
403 404
		return -EINVAL;

D
Dave Airlie 已提交
405 406
	switch (map->type) {
	case _DRM_REGISTERS:
407
		iounmap(map->handle);
D
Dave Airlie 已提交
408 409 410 411
		/* FALLTHROUGH */
	case _DRM_FRAME_BUFFER:
		if (drm_core_has_MTRR(dev) && map->mtrr >= 0) {
			int retcode;
D
Dave Airlie 已提交
412 413
			retcode = mtrr_del(map->mtrr, map->offset, map->size);
			DRM_DEBUG("mtrr_del=%d\n", retcode);
L
Linus Torvalds 已提交
414
		}
D
Dave Airlie 已提交
415 416 417
		break;
	case _DRM_SHM:
		vfree(map->handle);
418 419 420 421 422
		if (master) {
			if (dev->sigdata.lock == master->lock.hw_lock)
				dev->sigdata.lock = NULL;
			master->lock.hw_lock = NULL;   /* SHM removed */
			master->lock.file_priv = NULL;
423
			wake_up_interruptible_all(&master->lock.lock_queue);
424
		}
D
Dave Airlie 已提交
425 426 427 428 429 430 431 432 433 434
		break;
	case _DRM_AGP:
	case _DRM_SCATTER_GATHER:
		break;
	case _DRM_CONSISTENT:
		dmah.vaddr = map->handle;
		dmah.busaddr = map->offset;
		dmah.size = map->size;
		__drm_pci_free(dev, &dmah);
		break;
J
Jesse Barnes 已提交
435 436 437
	case _DRM_GEM:
		DRM_ERROR("tried to rmmap GEM object\n");
		break;
L
Linus Torvalds 已提交
438
	}
D
Dave Airlie 已提交
439 440
	drm_free(map, sizeof(*map), DRM_MEM_MAPS);

L
Linus Torvalds 已提交
441 442
	return 0;
}
443
EXPORT_SYMBOL(drm_rmmap_locked);
D
Dave Airlie 已提交
444

445
int drm_rmmap(struct drm_device *dev, struct drm_local_map *map)
D
Dave Airlie 已提交
446 447 448
{
	int ret;

D
Dave Airlie 已提交
449
	mutex_lock(&dev->struct_mutex);
D
Dave Airlie 已提交
450
	ret = drm_rmmap_locked(dev, map);
D
Dave Airlie 已提交
451
	mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
452 453 454

	return ret;
}
J
Jesse Barnes 已提交
455
EXPORT_SYMBOL(drm_rmmap);
456

D
Dave Airlie 已提交
457 458 459 460 461 462 463 464
/* The rmmap ioctl appears to be unnecessary.  All mappings are torn down on
 * the last close of the device, and this is necessary for cleanup when things
 * exit uncleanly.  Therefore, having userland manually remove mappings seems
 * like a pointless exercise since they're going away anyway.
 *
 * One use case might be after addmap is allowed for normal users for SHM and
 * gets used by drivers that the server doesn't need to care about.  This seems
 * unlikely.
465 466 467 468 469 470
 *
 * \param inode device inode.
 * \param file_priv DRM file private.
 * \param cmd command.
 * \param arg pointer to a struct drm_map structure.
 * \return zero on success or a negative value on error.
D
Dave Airlie 已提交
471
 */
472 473
int drm_rmmap_ioctl(struct drm_device *dev, void *data,
		    struct drm_file *file_priv)
474
{
475
	struct drm_map *request = data;
476
	struct drm_local_map *map = NULL;
D
Dave Airlie 已提交
477
	struct drm_map_list *r_list;
D
Dave Airlie 已提交
478
	int ret;
479

D
Dave Airlie 已提交
480
	mutex_lock(&dev->struct_mutex);
481
	list_for_each_entry(r_list, &dev->maplist, head) {
D
Dave Airlie 已提交
482
		if (r_list->map &&
483
		    r_list->user_token == (unsigned long)request->handle &&
D
Dave Airlie 已提交
484 485 486 487 488 489 490 491 492
		    r_list->map->flags & _DRM_REMOVABLE) {
			map = r_list->map;
			break;
		}
	}

	/* List has wrapped around to the head pointer, or its empty we didn't
	 * find anything.
	 */
493
	if (list_empty(&dev->maplist) || !map) {
D
Dave Airlie 已提交
494
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
495 496 497 498 499
		return -EINVAL;
	}

	/* Register and framebuffer maps are permanent */
	if ((map->type == _DRM_REGISTERS) || (map->type == _DRM_FRAME_BUFFER)) {
D
Dave Airlie 已提交
500
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
501 502 503 504 505
		return 0;
	}

	ret = drm_rmmap_locked(dev, map);

D
Dave Airlie 已提交
506
	mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
507 508

	return ret;
509
}
L
Linus Torvalds 已提交
510 511 512 513

/**
 * Cleanup after an error on one of the addbufs() functions.
 *
D
Dave Airlie 已提交
514
 * \param dev DRM device.
L
Linus Torvalds 已提交
515 516 517 518
 * \param entry buffer entry where the error occurred.
 *
 * Frees any pages and buffers associated with the given entry.
 */
519 520
static void drm_cleanup_buf_error(struct drm_device * dev,
				  struct drm_buf_entry * entry)
L
Linus Torvalds 已提交
521 522 523 524 525 526
{
	int i;

	if (entry->seg_count) {
		for (i = 0; i < entry->seg_count; i++) {
			if (entry->seglist[i]) {
D
Dave Airlie 已提交
527
				drm_pci_free(dev, entry->seglist[i]);
L
Linus Torvalds 已提交
528 529 530
			}
		}
		drm_free(entry->seglist,
D
Dave Airlie 已提交
531 532
			 entry->seg_count *
			 sizeof(*entry->seglist), DRM_MEM_SEGS);
L
Linus Torvalds 已提交
533 534 535 536

		entry->seg_count = 0;
	}

D
Dave Airlie 已提交
537 538
	if (entry->buf_count) {
		for (i = 0; i < entry->buf_count; i++) {
L
Linus Torvalds 已提交
539 540
			if (entry->buflist[i].dev_private) {
				drm_free(entry->buflist[i].dev_private,
D
Dave Airlie 已提交
541 542
					 entry->buflist[i].dev_priv_size,
					 DRM_MEM_BUFS);
L
Linus Torvalds 已提交
543 544 545
			}
		}
		drm_free(entry->buflist,
D
Dave Airlie 已提交
546 547
			 entry->buf_count *
			 sizeof(*entry->buflist), DRM_MEM_BUFS);
L
Linus Torvalds 已提交
548 549 550 551 552 553 554

		entry->buf_count = 0;
	}
}

#if __OS_HAS_AGP
/**
555
 * Add AGP buffers for DMA transfers.
L
Linus Torvalds 已提交
556
 *
557
 * \param dev struct drm_device to which the buffers are to be added.
558
 * \param request pointer to a struct drm_buf_desc describing the request.
L
Linus Torvalds 已提交
559
 * \return zero on success or a negative number on failure.
D
Dave Airlie 已提交
560
 *
L
Linus Torvalds 已提交
561 562 563 564
 * After some sanity checks creates a drm_buf structure for each buffer and
 * reallocates the buffer list of the same size order to accommodate the new
 * buffers.
 */
565
int drm_addbufs_agp(struct drm_device * dev, struct drm_buf_desc * request)
L
Linus Torvalds 已提交
566
{
567 568
	struct drm_device_dma *dma = dev->dma;
	struct drm_buf_entry *entry;
D
Dave Airlie 已提交
569
	struct drm_agp_mem *agp_entry;
D
Dave Airlie 已提交
570
	struct drm_buf *buf;
L
Linus Torvalds 已提交
571 572 573 574 575 576 577 578 579
	unsigned long offset;
	unsigned long agp_offset;
	int count;
	int order;
	int size;
	int alignment;
	int page_order;
	int total;
	int byte_count;
580
	int i, valid;
D
Dave Airlie 已提交
581
	struct drm_buf **temp_buflist;
L
Linus Torvalds 已提交
582

D
Dave Airlie 已提交
583 584
	if (!dma)
		return -EINVAL;
L
Linus Torvalds 已提交
585

586 587
	count = request->count;
	order = drm_order(request->size);
L
Linus Torvalds 已提交
588 589
	size = 1 << order;

D
Dave Airlie 已提交
590 591
	alignment = (request->flags & _DRM_PAGE_ALIGN)
	    ? PAGE_ALIGN(size) : size;
L
Linus Torvalds 已提交
592 593 594 595
	page_order = order - PAGE_SHIFT > 0 ? order - PAGE_SHIFT : 0;
	total = PAGE_SIZE << page_order;

	byte_count = 0;
596
	agp_offset = dev->agp->base + request->agp_start;
L
Linus Torvalds 已提交
597

D
Dave Airlie 已提交
598 599 600
	DRM_DEBUG("count:      %d\n", count);
	DRM_DEBUG("order:      %d\n", order);
	DRM_DEBUG("size:       %d\n", size);
601
	DRM_DEBUG("agp_offset: %lx\n", agp_offset);
D
Dave Airlie 已提交
602 603 604
	DRM_DEBUG("alignment:  %d\n", alignment);
	DRM_DEBUG("page_order: %d\n", page_order);
	DRM_DEBUG("total:      %d\n", total);
L
Linus Torvalds 已提交
605

D
Dave Airlie 已提交
606 607 608 609
	if (order < DRM_MIN_ORDER || order > DRM_MAX_ORDER)
		return -EINVAL;
	if (dev->queue_count)
		return -EBUSY;	/* Not while in use */
L
Linus Torvalds 已提交
610

611 612
	/* Make sure buffers are located in AGP memory that we own */
	valid = 0;
613
	list_for_each_entry(agp_entry, &dev->agp->memory, head) {
614 615 616 617 618 619
		if ((agp_offset >= agp_entry->bound) &&
		    (agp_offset + total * count <= agp_entry->bound + agp_entry->pages * PAGE_SIZE)) {
			valid = 1;
			break;
		}
	}
620
	if (!list_empty(&dev->agp->memory) && !valid) {
621 622 623
		DRM_DEBUG("zone invalid\n");
		return -EINVAL;
	}
D
Dave Airlie 已提交
624 625 626
	spin_lock(&dev->count_lock);
	if (dev->buf_use) {
		spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
627 628
		return -EBUSY;
	}
D
Dave Airlie 已提交
629 630
	atomic_inc(&dev->buf_alloc);
	spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
631

D
Dave Airlie 已提交
632
	mutex_lock(&dev->struct_mutex);
L
Linus Torvalds 已提交
633
	entry = &dma->bufs[order];
D
Dave Airlie 已提交
634
	if (entry->buf_count) {
D
Dave Airlie 已提交
635
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
636 637
		atomic_dec(&dev->buf_alloc);
		return -ENOMEM;	/* May only call once for each order */
L
Linus Torvalds 已提交
638 639 640
	}

	if (count < 0 || count > 4096) {
D
Dave Airlie 已提交
641
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
642
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
643 644 645
		return -EINVAL;
	}

D
Dave Airlie 已提交
646 647 648
	entry->buflist = drm_alloc(count * sizeof(*entry->buflist),
				   DRM_MEM_BUFS);
	if (!entry->buflist) {
D
Dave Airlie 已提交
649
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
650
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
651 652
		return -ENOMEM;
	}
D
Dave Airlie 已提交
653
	memset(entry->buflist, 0, count * sizeof(*entry->buflist));
L
Linus Torvalds 已提交
654 655 656 657 658 659

	entry->buf_size = size;
	entry->page_order = page_order;

	offset = 0;

D
Dave Airlie 已提交
660 661 662 663 664 665
	while (entry->buf_count < count) {
		buf = &entry->buflist[entry->buf_count];
		buf->idx = dma->buf_count + entry->buf_count;
		buf->total = alignment;
		buf->order = order;
		buf->used = 0;
L
Linus Torvalds 已提交
666

D
Dave Airlie 已提交
667
		buf->offset = (dma->byte_count + offset);
L
Linus Torvalds 已提交
668 669
		buf->bus_address = agp_offset + offset;
		buf->address = (void *)(agp_offset + offset);
D
Dave Airlie 已提交
670
		buf->next = NULL;
L
Linus Torvalds 已提交
671 672
		buf->waiting = 0;
		buf->pending = 0;
D
Dave Airlie 已提交
673
		init_waitqueue_head(&buf->dma_wait);
674
		buf->file_priv = NULL;
L
Linus Torvalds 已提交
675 676

		buf->dev_priv_size = dev->driver->dev_priv_size;
D
Dave Airlie 已提交
677 678
		buf->dev_private = drm_alloc(buf->dev_priv_size, DRM_MEM_BUFS);
		if (!buf->dev_private) {
L
Linus Torvalds 已提交
679 680
			/* Set count correctly so we free the proper amount. */
			entry->buf_count = count;
D
Dave Airlie 已提交
681
			drm_cleanup_buf_error(dev, entry);
D
Dave Airlie 已提交
682
			mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
683
			atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
684 685
			return -ENOMEM;
		}
D
Dave Airlie 已提交
686
		memset(buf->dev_private, 0, buf->dev_priv_size);
L
Linus Torvalds 已提交
687

D
Dave Airlie 已提交
688
		DRM_DEBUG("buffer %d @ %p\n", entry->buf_count, buf->address);
L
Linus Torvalds 已提交
689 690 691 692 693 694

		offset += alignment;
		entry->buf_count++;
		byte_count += PAGE_SIZE << page_order;
	}

D
Dave Airlie 已提交
695
	DRM_DEBUG("byte_count: %d\n", byte_count);
L
Linus Torvalds 已提交
696

D
Dave Airlie 已提交
697 698 699 700 701
	temp_buflist = drm_realloc(dma->buflist,
				   dma->buf_count * sizeof(*dma->buflist),
				   (dma->buf_count + entry->buf_count)
				   * sizeof(*dma->buflist), DRM_MEM_BUFS);
	if (!temp_buflist) {
L
Linus Torvalds 已提交
702
		/* Free the entry because it isn't valid */
D
Dave Airlie 已提交
703
		drm_cleanup_buf_error(dev, entry);
D
Dave Airlie 已提交
704
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
705
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
706 707 708 709
		return -ENOMEM;
	}
	dma->buflist = temp_buflist;

D
Dave Airlie 已提交
710
	for (i = 0; i < entry->buf_count; i++) {
L
Linus Torvalds 已提交
711 712 713 714
		dma->buflist[i + dma->buf_count] = &entry->buflist[i];
	}

	dma->buf_count += entry->buf_count;
715 716
	dma->seg_count += entry->seg_count;
	dma->page_count += byte_count >> PAGE_SHIFT;
L
Linus Torvalds 已提交
717 718
	dma->byte_count += byte_count;

D
Dave Airlie 已提交
719 720
	DRM_DEBUG("dma->buf_count : %d\n", dma->buf_count);
	DRM_DEBUG("entry->buf_count : %d\n", entry->buf_count);
L
Linus Torvalds 已提交
721

D
Dave Airlie 已提交
722
	mutex_unlock(&dev->struct_mutex);
L
Linus Torvalds 已提交
723

724 725
	request->count = entry->buf_count;
	request->size = size;
L
Linus Torvalds 已提交
726 727 728

	dma->flags = _DRM_DMA_USE_AGP;

D
Dave Airlie 已提交
729
	atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
730 731
	return 0;
}
732
EXPORT_SYMBOL(drm_addbufs_agp);
D
Dave Airlie 已提交
733
#endif				/* __OS_HAS_AGP */
L
Linus Torvalds 已提交
734

735
int drm_addbufs_pci(struct drm_device * dev, struct drm_buf_desc * request)
L
Linus Torvalds 已提交
736
{
737
	struct drm_device_dma *dma = dev->dma;
L
Linus Torvalds 已提交
738 739 740 741 742
	int count;
	int order;
	int size;
	int total;
	int page_order;
743
	struct drm_buf_entry *entry;
D
Dave Airlie 已提交
744
	drm_dma_handle_t *dmah;
D
Dave Airlie 已提交
745
	struct drm_buf *buf;
L
Linus Torvalds 已提交
746 747 748 749 750 751
	int alignment;
	unsigned long offset;
	int i;
	int byte_count;
	int page_count;
	unsigned long *temp_pagelist;
D
Dave Airlie 已提交
752
	struct drm_buf **temp_buflist;
L
Linus Torvalds 已提交
753

D
Dave Airlie 已提交
754 755
	if (!drm_core_check_feature(dev, DRIVER_PCI_DMA))
		return -EINVAL;
756

D
Dave Airlie 已提交
757 758
	if (!dma)
		return -EINVAL;
L
Linus Torvalds 已提交
759

760 761 762
	if (!capable(CAP_SYS_ADMIN))
		return -EPERM;

763 764
	count = request->count;
	order = drm_order(request->size);
L
Linus Torvalds 已提交
765 766
	size = 1 << order;

D
Dave Airlie 已提交
767 768
	DRM_DEBUG("count=%d, size=%d (%d), order=%d, queue_count=%d\n",
		  request->count, request->size, size, order, dev->queue_count);
L
Linus Torvalds 已提交
769

D
Dave Airlie 已提交
770 771 772 773
	if (order < DRM_MIN_ORDER || order > DRM_MAX_ORDER)
		return -EINVAL;
	if (dev->queue_count)
		return -EBUSY;	/* Not while in use */
L
Linus Torvalds 已提交
774

775
	alignment = (request->flags & _DRM_PAGE_ALIGN)
D
Dave Airlie 已提交
776
	    ? PAGE_ALIGN(size) : size;
L
Linus Torvalds 已提交
777 778 779
	page_order = order - PAGE_SHIFT > 0 ? order - PAGE_SHIFT : 0;
	total = PAGE_SIZE << page_order;

D
Dave Airlie 已提交
780 781 782
	spin_lock(&dev->count_lock);
	if (dev->buf_use) {
		spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
783 784
		return -EBUSY;
	}
D
Dave Airlie 已提交
785 786
	atomic_inc(&dev->buf_alloc);
	spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
787

D
Dave Airlie 已提交
788
	mutex_lock(&dev->struct_mutex);
L
Linus Torvalds 已提交
789
	entry = &dma->bufs[order];
D
Dave Airlie 已提交
790
	if (entry->buf_count) {
D
Dave Airlie 已提交
791
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
792
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
793 794 795 796
		return -ENOMEM;	/* May only call once for each order */
	}

	if (count < 0 || count > 4096) {
D
Dave Airlie 已提交
797
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
798
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
799 800 801
		return -EINVAL;
	}

D
Dave Airlie 已提交
802 803 804
	entry->buflist = drm_alloc(count * sizeof(*entry->buflist),
				   DRM_MEM_BUFS);
	if (!entry->buflist) {
D
Dave Airlie 已提交
805
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
806
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
807 808
		return -ENOMEM;
	}
D
Dave Airlie 已提交
809 810 811 812 813 814 815
	memset(entry->buflist, 0, count * sizeof(*entry->buflist));

	entry->seglist = drm_alloc(count * sizeof(*entry->seglist),
				   DRM_MEM_SEGS);
	if (!entry->seglist) {
		drm_free(entry->buflist,
			 count * sizeof(*entry->buflist), DRM_MEM_BUFS);
D
Dave Airlie 已提交
816
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
817
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
818 819
		return -ENOMEM;
	}
D
Dave Airlie 已提交
820
	memset(entry->seglist, 0, count * sizeof(*entry->seglist));
L
Linus Torvalds 已提交
821 822 823 824

	/* Keep the original pagelist until we know all the allocations
	 * have succeeded
	 */
D
Dave Airlie 已提交
825 826
	temp_pagelist = drm_alloc((dma->page_count + (count << page_order))
				  * sizeof(*dma->pagelist), DRM_MEM_PAGES);
L
Linus Torvalds 已提交
827
	if (!temp_pagelist) {
D
Dave Airlie 已提交
828 829 830 831
		drm_free(entry->buflist,
			 count * sizeof(*entry->buflist), DRM_MEM_BUFS);
		drm_free(entry->seglist,
			 count * sizeof(*entry->seglist), DRM_MEM_SEGS);
D
Dave Airlie 已提交
832
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
833
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
834 835 836
		return -ENOMEM;
	}
	memcpy(temp_pagelist,
D
Dave Airlie 已提交
837 838 839
	       dma->pagelist, dma->page_count * sizeof(*dma->pagelist));
	DRM_DEBUG("pagelist: %d entries\n",
		  dma->page_count + (count << page_order));
L
Linus Torvalds 已提交
840

D
Dave Airlie 已提交
841
	entry->buf_size = size;
L
Linus Torvalds 已提交
842 843 844 845
	entry->page_order = page_order;
	byte_count = 0;
	page_count = 0;

D
Dave Airlie 已提交
846
	while (entry->buf_count < count) {
D
Dave Airlie 已提交
847

D
Dave Airlie 已提交
848
		dmah = drm_pci_alloc(dev, PAGE_SIZE << page_order, 0x1000, 0xfffffffful);
D
Dave Airlie 已提交
849

D
Dave Airlie 已提交
850
		if (!dmah) {
L
Linus Torvalds 已提交
851 852 853 854
			/* Set count correctly so we free the proper amount. */
			entry->buf_count = count;
			entry->seg_count = count;
			drm_cleanup_buf_error(dev, entry);
D
Dave Airlie 已提交
855 856 857
			drm_free(temp_pagelist,
				 (dma->page_count + (count << page_order))
				 * sizeof(*dma->pagelist), DRM_MEM_PAGES);
D
Dave Airlie 已提交
858
			mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
859
			atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
860 861
			return -ENOMEM;
		}
D
Dave Airlie 已提交
862
		entry->seglist[entry->seg_count++] = dmah;
D
Dave Airlie 已提交
863 864 865
		for (i = 0; i < (1 << page_order); i++) {
			DRM_DEBUG("page %d @ 0x%08lx\n",
				  dma->page_count + page_count,
D
Dave Airlie 已提交
866
				  (unsigned long)dmah->vaddr + PAGE_SIZE * i);
L
Linus Torvalds 已提交
867
			temp_pagelist[dma->page_count + page_count++]
D
Dave Airlie 已提交
868
				= (unsigned long)dmah->vaddr + PAGE_SIZE * i;
L
Linus Torvalds 已提交
869
		}
D
Dave Airlie 已提交
870 871 872 873 874 875 876 877 878
		for (offset = 0;
		     offset + size <= total && entry->buf_count < count;
		     offset += alignment, ++entry->buf_count) {
			buf = &entry->buflist[entry->buf_count];
			buf->idx = dma->buf_count + entry->buf_count;
			buf->total = alignment;
			buf->order = order;
			buf->used = 0;
			buf->offset = (dma->byte_count + byte_count + offset);
D
Dave Airlie 已提交
879 880
			buf->address = (void *)(dmah->vaddr + offset);
			buf->bus_address = dmah->busaddr + offset;
D
Dave Airlie 已提交
881
			buf->next = NULL;
L
Linus Torvalds 已提交
882 883
			buf->waiting = 0;
			buf->pending = 0;
D
Dave Airlie 已提交
884
			init_waitqueue_head(&buf->dma_wait);
885
			buf->file_priv = NULL;
L
Linus Torvalds 已提交
886 887

			buf->dev_priv_size = dev->driver->dev_priv_size;
D
Dave Airlie 已提交
888 889 890
			buf->dev_private = drm_alloc(buf->dev_priv_size,
						     DRM_MEM_BUFS);
			if (!buf->dev_private) {
L
Linus Torvalds 已提交
891 892 893
				/* Set count correctly so we free the proper amount. */
				entry->buf_count = count;
				entry->seg_count = count;
D
Dave Airlie 已提交
894 895 896 897 898 899
				drm_cleanup_buf_error(dev, entry);
				drm_free(temp_pagelist,
					 (dma->page_count +
					  (count << page_order))
					 * sizeof(*dma->pagelist),
					 DRM_MEM_PAGES);
D
Dave Airlie 已提交
900
				mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
901
				atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
902 903
				return -ENOMEM;
			}
D
Dave Airlie 已提交
904
			memset(buf->dev_private, 0, buf->dev_priv_size);
L
Linus Torvalds 已提交
905

D
Dave Airlie 已提交
906 907
			DRM_DEBUG("buffer %d @ %p\n",
				  entry->buf_count, buf->address);
L
Linus Torvalds 已提交
908 909 910 911
		}
		byte_count += PAGE_SIZE << page_order;
	}

D
Dave Airlie 已提交
912 913 914 915
	temp_buflist = drm_realloc(dma->buflist,
				   dma->buf_count * sizeof(*dma->buflist),
				   (dma->buf_count + entry->buf_count)
				   * sizeof(*dma->buflist), DRM_MEM_BUFS);
L
Linus Torvalds 已提交
916 917
	if (!temp_buflist) {
		/* Free the entry because it isn't valid */
D
Dave Airlie 已提交
918 919 920 921
		drm_cleanup_buf_error(dev, entry);
		drm_free(temp_pagelist,
			 (dma->page_count + (count << page_order))
			 * sizeof(*dma->pagelist), DRM_MEM_PAGES);
D
Dave Airlie 已提交
922
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
923
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
924 925 926 927
		return -ENOMEM;
	}
	dma->buflist = temp_buflist;

D
Dave Airlie 已提交
928
	for (i = 0; i < entry->buf_count; i++) {
L
Linus Torvalds 已提交
929 930 931 932 933 934 935 936
		dma->buflist[i + dma->buf_count] = &entry->buflist[i];
	}

	/* No allocations failed, so now we can replace the orginal pagelist
	 * with the new one.
	 */
	if (dma->page_count) {
		drm_free(dma->pagelist,
D
Dave Airlie 已提交
937 938
			 dma->page_count * sizeof(*dma->pagelist),
			 DRM_MEM_PAGES);
L
Linus Torvalds 已提交
939 940 941 942 943 944 945 946
	}
	dma->pagelist = temp_pagelist;

	dma->buf_count += entry->buf_count;
	dma->seg_count += entry->seg_count;
	dma->page_count += entry->seg_count << page_order;
	dma->byte_count += PAGE_SIZE * (entry->seg_count << page_order);

D
Dave Airlie 已提交
947
	mutex_unlock(&dev->struct_mutex);
L
Linus Torvalds 已提交
948

949 950
	request->count = entry->buf_count;
	request->size = size;
L
Linus Torvalds 已提交
951

952 953 954
	if (request->flags & _DRM_PCI_BUFFER_RO)
		dma->flags = _DRM_DMA_USE_PCI_RO;

D
Dave Airlie 已提交
955
	atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
956 957 958
	return 0;

}
959
EXPORT_SYMBOL(drm_addbufs_pci);
L
Linus Torvalds 已提交
960

961
static int drm_addbufs_sg(struct drm_device * dev, struct drm_buf_desc * request)
L
Linus Torvalds 已提交
962
{
963 964
	struct drm_device_dma *dma = dev->dma;
	struct drm_buf_entry *entry;
D
Dave Airlie 已提交
965
	struct drm_buf *buf;
L
Linus Torvalds 已提交
966 967 968 969 970 971 972 973 974 975
	unsigned long offset;
	unsigned long agp_offset;
	int count;
	int order;
	int size;
	int alignment;
	int page_order;
	int total;
	int byte_count;
	int i;
D
Dave Airlie 已提交
976
	struct drm_buf **temp_buflist;
L
Linus Torvalds 已提交
977

D
Dave Airlie 已提交
978 979 980 981 982
	if (!drm_core_check_feature(dev, DRIVER_SG))
		return -EINVAL;

	if (!dma)
		return -EINVAL;
L
Linus Torvalds 已提交
983

984 985 986
	if (!capable(CAP_SYS_ADMIN))
		return -EPERM;

987 988
	count = request->count;
	order = drm_order(request->size);
L
Linus Torvalds 已提交
989 990
	size = 1 << order;

D
Dave Airlie 已提交
991 992
	alignment = (request->flags & _DRM_PAGE_ALIGN)
	    ? PAGE_ALIGN(size) : size;
L
Linus Torvalds 已提交
993 994 995 996
	page_order = order - PAGE_SHIFT > 0 ? order - PAGE_SHIFT : 0;
	total = PAGE_SIZE << page_order;

	byte_count = 0;
997
	agp_offset = request->agp_start;
L
Linus Torvalds 已提交
998

D
Dave Airlie 已提交
999 1000 1001 1002 1003 1004 1005
	DRM_DEBUG("count:      %d\n", count);
	DRM_DEBUG("order:      %d\n", order);
	DRM_DEBUG("size:       %d\n", size);
	DRM_DEBUG("agp_offset: %lu\n", agp_offset);
	DRM_DEBUG("alignment:  %d\n", alignment);
	DRM_DEBUG("page_order: %d\n", page_order);
	DRM_DEBUG("total:      %d\n", total);
L
Linus Torvalds 已提交
1006

D
Dave Airlie 已提交
1007 1008 1009 1010
	if (order < DRM_MIN_ORDER || order > DRM_MAX_ORDER)
		return -EINVAL;
	if (dev->queue_count)
		return -EBUSY;	/* Not while in use */
L
Linus Torvalds 已提交
1011

D
Dave Airlie 已提交
1012 1013 1014
	spin_lock(&dev->count_lock);
	if (dev->buf_use) {
		spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
1015 1016
		return -EBUSY;
	}
D
Dave Airlie 已提交
1017 1018
	atomic_inc(&dev->buf_alloc);
	spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
1019

D
Dave Airlie 已提交
1020
	mutex_lock(&dev->struct_mutex);
L
Linus Torvalds 已提交
1021
	entry = &dma->bufs[order];
D
Dave Airlie 已提交
1022
	if (entry->buf_count) {
D
Dave Airlie 已提交
1023
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1024 1025
		atomic_dec(&dev->buf_alloc);
		return -ENOMEM;	/* May only call once for each order */
L
Linus Torvalds 已提交
1026 1027 1028
	}

	if (count < 0 || count > 4096) {
D
Dave Airlie 已提交
1029
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1030
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
1031 1032 1033
		return -EINVAL;
	}

D
Dave Airlie 已提交
1034 1035 1036
	entry->buflist = drm_alloc(count * sizeof(*entry->buflist),
				   DRM_MEM_BUFS);
	if (!entry->buflist) {
D
Dave Airlie 已提交
1037
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1038
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
1039 1040
		return -ENOMEM;
	}
D
Dave Airlie 已提交
1041
	memset(entry->buflist, 0, count * sizeof(*entry->buflist));
L
Linus Torvalds 已提交
1042 1043 1044 1045 1046 1047

	entry->buf_size = size;
	entry->page_order = page_order;

	offset = 0;

D
Dave Airlie 已提交
1048 1049 1050 1051 1052 1053
	while (entry->buf_count < count) {
		buf = &entry->buflist[entry->buf_count];
		buf->idx = dma->buf_count + entry->buf_count;
		buf->total = alignment;
		buf->order = order;
		buf->used = 0;
L
Linus Torvalds 已提交
1054

D
Dave Airlie 已提交
1055
		buf->offset = (dma->byte_count + offset);
L
Linus Torvalds 已提交
1056
		buf->bus_address = agp_offset + offset;
D
Dave Airlie 已提交
1057
		buf->address = (void *)(agp_offset + offset
1058
					+ (unsigned long)dev->sg->virtual);
D
Dave Airlie 已提交
1059
		buf->next = NULL;
L
Linus Torvalds 已提交
1060 1061
		buf->waiting = 0;
		buf->pending = 0;
D
Dave Airlie 已提交
1062
		init_waitqueue_head(&buf->dma_wait);
1063
		buf->file_priv = NULL;
L
Linus Torvalds 已提交
1064 1065

		buf->dev_priv_size = dev->driver->dev_priv_size;
D
Dave Airlie 已提交
1066 1067
		buf->dev_private = drm_alloc(buf->dev_priv_size, DRM_MEM_BUFS);
		if (!buf->dev_private) {
L
Linus Torvalds 已提交
1068 1069
			/* Set count correctly so we free the proper amount. */
			entry->buf_count = count;
D
Dave Airlie 已提交
1070
			drm_cleanup_buf_error(dev, entry);
D
Dave Airlie 已提交
1071
			mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1072
			atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
1073 1074 1075
			return -ENOMEM;
		}

D
Dave Airlie 已提交
1076
		memset(buf->dev_private, 0, buf->dev_priv_size);
L
Linus Torvalds 已提交
1077

D
Dave Airlie 已提交
1078
		DRM_DEBUG("buffer %d @ %p\n", entry->buf_count, buf->address);
L
Linus Torvalds 已提交
1079 1080 1081 1082 1083 1084

		offset += alignment;
		entry->buf_count++;
		byte_count += PAGE_SIZE << page_order;
	}

D
Dave Airlie 已提交
1085
	DRM_DEBUG("byte_count: %d\n", byte_count);
L
Linus Torvalds 已提交
1086

D
Dave Airlie 已提交
1087 1088 1089 1090 1091
	temp_buflist = drm_realloc(dma->buflist,
				   dma->buf_count * sizeof(*dma->buflist),
				   (dma->buf_count + entry->buf_count)
				   * sizeof(*dma->buflist), DRM_MEM_BUFS);
	if (!temp_buflist) {
L
Linus Torvalds 已提交
1092
		/* Free the entry because it isn't valid */
D
Dave Airlie 已提交
1093
		drm_cleanup_buf_error(dev, entry);
D
Dave Airlie 已提交
1094
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1095
		atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
1096 1097 1098 1099
		return -ENOMEM;
	}
	dma->buflist = temp_buflist;

D
Dave Airlie 已提交
1100
	for (i = 0; i < entry->buf_count; i++) {
L
Linus Torvalds 已提交
1101 1102 1103 1104
		dma->buflist[i + dma->buf_count] = &entry->buflist[i];
	}

	dma->buf_count += entry->buf_count;
1105 1106
	dma->seg_count += entry->seg_count;
	dma->page_count += byte_count >> PAGE_SHIFT;
L
Linus Torvalds 已提交
1107 1108
	dma->byte_count += byte_count;

D
Dave Airlie 已提交
1109 1110
	DRM_DEBUG("dma->buf_count : %d\n", dma->buf_count);
	DRM_DEBUG("entry->buf_count : %d\n", entry->buf_count);
L
Linus Torvalds 已提交
1111

D
Dave Airlie 已提交
1112
	mutex_unlock(&dev->struct_mutex);
L
Linus Torvalds 已提交
1113

1114 1115
	request->count = entry->buf_count;
	request->size = size;
L
Linus Torvalds 已提交
1116 1117 1118

	dma->flags = _DRM_DMA_USE_SG;

D
Dave Airlie 已提交
1119
	atomic_dec(&dev->buf_alloc);
L
Linus Torvalds 已提交
1120 1121 1122
	return 0;
}

1123
static int drm_addbufs_fb(struct drm_device * dev, struct drm_buf_desc * request)
D
Dave Airlie 已提交
1124
{
1125 1126
	struct drm_device_dma *dma = dev->dma;
	struct drm_buf_entry *entry;
D
Dave Airlie 已提交
1127
	struct drm_buf *buf;
D
Dave Airlie 已提交
1128 1129 1130 1131 1132 1133 1134 1135 1136 1137
	unsigned long offset;
	unsigned long agp_offset;
	int count;
	int order;
	int size;
	int alignment;
	int page_order;
	int total;
	int byte_count;
	int i;
D
Dave Airlie 已提交
1138
	struct drm_buf **temp_buflist;
D
Dave Airlie 已提交
1139 1140 1141

	if (!drm_core_check_feature(dev, DRIVER_FB_DMA))
		return -EINVAL;
D
Dave Airlie 已提交
1142

D
Dave Airlie 已提交
1143 1144 1145
	if (!dma)
		return -EINVAL;

1146 1147 1148
	if (!capable(CAP_SYS_ADMIN))
		return -EPERM;

1149 1150
	count = request->count;
	order = drm_order(request->size);
D
Dave Airlie 已提交
1151 1152
	size = 1 << order;

1153
	alignment = (request->flags & _DRM_PAGE_ALIGN)
D
Dave Airlie 已提交
1154 1155 1156 1157 1158
	    ? PAGE_ALIGN(size) : size;
	page_order = order - PAGE_SHIFT > 0 ? order - PAGE_SHIFT : 0;
	total = PAGE_SIZE << page_order;

	byte_count = 0;
1159
	agp_offset = request->agp_start;
D
Dave Airlie 已提交
1160 1161 1162 1163 1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181

	DRM_DEBUG("count:      %d\n", count);
	DRM_DEBUG("order:      %d\n", order);
	DRM_DEBUG("size:       %d\n", size);
	DRM_DEBUG("agp_offset: %lu\n", agp_offset);
	DRM_DEBUG("alignment:  %d\n", alignment);
	DRM_DEBUG("page_order: %d\n", page_order);
	DRM_DEBUG("total:      %d\n", total);

	if (order < DRM_MIN_ORDER || order > DRM_MAX_ORDER)
		return -EINVAL;
	if (dev->queue_count)
		return -EBUSY;	/* Not while in use */

	spin_lock(&dev->count_lock);
	if (dev->buf_use) {
		spin_unlock(&dev->count_lock);
		return -EBUSY;
	}
	atomic_inc(&dev->buf_alloc);
	spin_unlock(&dev->count_lock);

D
Dave Airlie 已提交
1182
	mutex_lock(&dev->struct_mutex);
D
Dave Airlie 已提交
1183 1184
	entry = &dma->bufs[order];
	if (entry->buf_count) {
D
Dave Airlie 已提交
1185
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1186 1187 1188 1189 1190
		atomic_dec(&dev->buf_alloc);
		return -ENOMEM;	/* May only call once for each order */
	}

	if (count < 0 || count > 4096) {
D
Dave Airlie 已提交
1191
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1192 1193 1194 1195 1196 1197 1198
		atomic_dec(&dev->buf_alloc);
		return -EINVAL;
	}

	entry->buflist = drm_alloc(count * sizeof(*entry->buflist),
				   DRM_MEM_BUFS);
	if (!entry->buflist) {
D
Dave Airlie 已提交
1199
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1200 1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223
		atomic_dec(&dev->buf_alloc);
		return -ENOMEM;
	}
	memset(entry->buflist, 0, count * sizeof(*entry->buflist));

	entry->buf_size = size;
	entry->page_order = page_order;

	offset = 0;

	while (entry->buf_count < count) {
		buf = &entry->buflist[entry->buf_count];
		buf->idx = dma->buf_count + entry->buf_count;
		buf->total = alignment;
		buf->order = order;
		buf->used = 0;

		buf->offset = (dma->byte_count + offset);
		buf->bus_address = agp_offset + offset;
		buf->address = (void *)(agp_offset + offset);
		buf->next = NULL;
		buf->waiting = 0;
		buf->pending = 0;
		init_waitqueue_head(&buf->dma_wait);
1224
		buf->file_priv = NULL;
D
Dave Airlie 已提交
1225 1226 1227 1228 1229 1230 1231

		buf->dev_priv_size = dev->driver->dev_priv_size;
		buf->dev_private = drm_alloc(buf->dev_priv_size, DRM_MEM_BUFS);
		if (!buf->dev_private) {
			/* Set count correctly so we free the proper amount. */
			entry->buf_count = count;
			drm_cleanup_buf_error(dev, entry);
D
Dave Airlie 已提交
1232
			mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1233 1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253
			atomic_dec(&dev->buf_alloc);
			return -ENOMEM;
		}
		memset(buf->dev_private, 0, buf->dev_priv_size);

		DRM_DEBUG("buffer %d @ %p\n", entry->buf_count, buf->address);

		offset += alignment;
		entry->buf_count++;
		byte_count += PAGE_SIZE << page_order;
	}

	DRM_DEBUG("byte_count: %d\n", byte_count);

	temp_buflist = drm_realloc(dma->buflist,
				   dma->buf_count * sizeof(*dma->buflist),
				   (dma->buf_count + entry->buf_count)
				   * sizeof(*dma->buflist), DRM_MEM_BUFS);
	if (!temp_buflist) {
		/* Free the entry because it isn't valid */
		drm_cleanup_buf_error(dev, entry);
D
Dave Airlie 已提交
1254
		mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1255 1256 1257 1258 1259 1260 1261 1262 1263 1264
		atomic_dec(&dev->buf_alloc);
		return -ENOMEM;
	}
	dma->buflist = temp_buflist;

	for (i = 0; i < entry->buf_count; i++) {
		dma->buflist[i + dma->buf_count] = &entry->buflist[i];
	}

	dma->buf_count += entry->buf_count;
1265 1266
	dma->seg_count += entry->seg_count;
	dma->page_count += byte_count >> PAGE_SHIFT;
D
Dave Airlie 已提交
1267 1268 1269 1270 1271
	dma->byte_count += byte_count;

	DRM_DEBUG("dma->buf_count : %d\n", dma->buf_count);
	DRM_DEBUG("entry->buf_count : %d\n", entry->buf_count);

D
Dave Airlie 已提交
1272
	mutex_unlock(&dev->struct_mutex);
D
Dave Airlie 已提交
1273

1274 1275
	request->count = entry->buf_count;
	request->size = size;
D
Dave Airlie 已提交
1276 1277 1278 1279 1280 1281

	dma->flags = _DRM_DMA_USE_FB;

	atomic_dec(&dev->buf_alloc);
	return 0;
}
1282

D
Dave Airlie 已提交
1283

L
Linus Torvalds 已提交
1284 1285 1286 1287
/**
 * Add buffers for DMA transfers (ioctl).
 *
 * \param inode device inode.
1288
 * \param file_priv DRM file private.
L
Linus Torvalds 已提交
1289
 * \param cmd command.
1290
 * \param arg pointer to a struct drm_buf_desc request.
L
Linus Torvalds 已提交
1291 1292 1293 1294 1295 1296 1297
 * \return zero on success or a negative number on failure.
 *
 * According with the memory type specified in drm_buf_desc::flags and the
 * build options, it dispatches the call either to addbufs_agp(),
 * addbufs_sg() or addbufs_pci() for AGP, scatter-gather or consistent
 * PCI memory respectively.
 */
1298 1299
int drm_addbufs(struct drm_device *dev, void *data,
		struct drm_file *file_priv)
L
Linus Torvalds 已提交
1300
{
1301
	struct drm_buf_desc *request = data;
1302
	int ret;
D
Dave Airlie 已提交
1303

L
Linus Torvalds 已提交
1304 1305 1306 1307
	if (!drm_core_check_feature(dev, DRIVER_HAVE_DMA))
		return -EINVAL;

#if __OS_HAS_AGP
1308 1309
	if (request->flags & _DRM_AGP_BUFFER)
		ret = drm_addbufs_agp(dev, request);
L
Linus Torvalds 已提交
1310 1311
	else
#endif
1312 1313 1314 1315
	if (request->flags & _DRM_SG_BUFFER)
		ret = drm_addbufs_sg(dev, request);
	else if (request->flags & _DRM_FB_BUFFER)
		ret = drm_addbufs_fb(dev, request);
L
Linus Torvalds 已提交
1316
	else
1317
		ret = drm_addbufs_pci(dev, request);
1318 1319

	return ret;
L
Linus Torvalds 已提交
1320 1321 1322 1323 1324 1325 1326 1327 1328 1329
}

/**
 * Get information about the buffer mappings.
 *
 * This was originally mean for debugging purposes, or by a sophisticated
 * client library to determine how best to use the available buffers (e.g.,
 * large buffers can be used for image transfer).
 *
 * \param inode device inode.
1330
 * \param file_priv DRM file private.
L
Linus Torvalds 已提交
1331 1332 1333 1334 1335 1336 1337 1338
 * \param cmd command.
 * \param arg pointer to a drm_buf_info structure.
 * \return zero on success or a negative number on failure.
 *
 * Increments drm_device::buf_use while holding the drm_device::count_lock
 * lock, preventing of allocating more buffers after this call. Information
 * about each requested buffer is then copied into user space.
 */
1339 1340
int drm_infobufs(struct drm_device *dev, void *data,
		 struct drm_file *file_priv)
L
Linus Torvalds 已提交
1341
{
1342
	struct drm_device_dma *dma = dev->dma;
1343
	struct drm_buf_info *request = data;
L
Linus Torvalds 已提交
1344 1345 1346 1347 1348 1349
	int i;
	int count;

	if (!drm_core_check_feature(dev, DRIVER_HAVE_DMA))
		return -EINVAL;

D
Dave Airlie 已提交
1350 1351
	if (!dma)
		return -EINVAL;
L
Linus Torvalds 已提交
1352

D
Dave Airlie 已提交
1353 1354 1355
	spin_lock(&dev->count_lock);
	if (atomic_read(&dev->buf_alloc)) {
		spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
1356 1357 1358
		return -EBUSY;
	}
	++dev->buf_use;		/* Can't allocate more after this call */
D
Dave Airlie 已提交
1359
	spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
1360

D
Dave Airlie 已提交
1361 1362 1363
	for (i = 0, count = 0; i < DRM_MAX_ORDER + 1; i++) {
		if (dma->bufs[i].buf_count)
			++count;
L
Linus Torvalds 已提交
1364 1365
	}

D
Dave Airlie 已提交
1366
	DRM_DEBUG("count = %d\n", count);
L
Linus Torvalds 已提交
1367

1368
	if (request->count >= count) {
D
Dave Airlie 已提交
1369 1370
		for (i = 0, count = 0; i < DRM_MAX_ORDER + 1; i++) {
			if (dma->bufs[i].buf_count) {
1371
				struct drm_buf_desc __user *to =
1372
				    &request->list[count];
1373 1374
				struct drm_buf_entry *from = &dma->bufs[i];
				struct drm_freelist *list = &dma->bufs[i].freelist;
D
Dave Airlie 已提交
1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386
				if (copy_to_user(&to->count,
						 &from->buf_count,
						 sizeof(from->buf_count)) ||
				    copy_to_user(&to->size,
						 &from->buf_size,
						 sizeof(from->buf_size)) ||
				    copy_to_user(&to->low_mark,
						 &list->low_mark,
						 sizeof(list->low_mark)) ||
				    copy_to_user(&to->high_mark,
						 &list->high_mark,
						 sizeof(list->high_mark)))
L
Linus Torvalds 已提交
1387 1388
					return -EFAULT;

D
Dave Airlie 已提交
1389 1390 1391 1392 1393 1394
				DRM_DEBUG("%d %d %d %d %d\n",
					  i,
					  dma->bufs[i].buf_count,
					  dma->bufs[i].buf_size,
					  dma->bufs[i].freelist.low_mark,
					  dma->bufs[i].freelist.high_mark);
L
Linus Torvalds 已提交
1395 1396 1397 1398
				++count;
			}
		}
	}
1399
	request->count = count;
L
Linus Torvalds 已提交
1400 1401 1402 1403 1404 1405 1406 1407

	return 0;
}

/**
 * Specifies a low and high water mark for buffer allocation
 *
 * \param inode device inode.
1408
 * \param file_priv DRM file private.
L
Linus Torvalds 已提交
1409 1410 1411 1412 1413 1414 1415 1416 1417
 * \param cmd command.
 * \param arg a pointer to a drm_buf_desc structure.
 * \return zero on success or a negative number on failure.
 *
 * Verifies that the size order is bounded between the admissible orders and
 * updates the respective drm_device_dma::bufs entry low and high water mark.
 *
 * \note This ioctl is deprecated and mostly never used.
 */
1418 1419
int drm_markbufs(struct drm_device *dev, void *data,
		 struct drm_file *file_priv)
L
Linus Torvalds 已提交
1420
{
1421
	struct drm_device_dma *dma = dev->dma;
1422
	struct drm_buf_desc *request = data;
L
Linus Torvalds 已提交
1423
	int order;
1424
	struct drm_buf_entry *entry;
L
Linus Torvalds 已提交
1425 1426 1427 1428

	if (!drm_core_check_feature(dev, DRIVER_HAVE_DMA))
		return -EINVAL;

D
Dave Airlie 已提交
1429 1430
	if (!dma)
		return -EINVAL;
L
Linus Torvalds 已提交
1431

D
Dave Airlie 已提交
1432
	DRM_DEBUG("%d, %d, %d\n",
1433 1434
		  request->size, request->low_mark, request->high_mark);
	order = drm_order(request->size);
D
Dave Airlie 已提交
1435 1436
	if (order < DRM_MIN_ORDER || order > DRM_MAX_ORDER)
		return -EINVAL;
L
Linus Torvalds 已提交
1437 1438
	entry = &dma->bufs[order];

1439
	if (request->low_mark < 0 || request->low_mark > entry->buf_count)
L
Linus Torvalds 已提交
1440
		return -EINVAL;
1441
	if (request->high_mark < 0 || request->high_mark > entry->buf_count)
L
Linus Torvalds 已提交
1442 1443
		return -EINVAL;

1444 1445
	entry->freelist.low_mark = request->low_mark;
	entry->freelist.high_mark = request->high_mark;
L
Linus Torvalds 已提交
1446 1447 1448 1449 1450

	return 0;
}

/**
D
Dave Airlie 已提交
1451
 * Unreserve the buffers in list, previously reserved using drmDMA.
L
Linus Torvalds 已提交
1452 1453
 *
 * \param inode device inode.
1454
 * \param file_priv DRM file private.
L
Linus Torvalds 已提交
1455 1456 1457
 * \param cmd command.
 * \param arg pointer to a drm_buf_free structure.
 * \return zero on success or a negative number on failure.
D
Dave Airlie 已提交
1458
 *
L
Linus Torvalds 已提交
1459 1460 1461
 * Calls free_buffer() for each used buffer.
 * This function is primarily used for debugging.
 */
1462 1463
int drm_freebufs(struct drm_device *dev, void *data,
		 struct drm_file *file_priv)
L
Linus Torvalds 已提交
1464
{
1465
	struct drm_device_dma *dma = dev->dma;
1466
	struct drm_buf_free *request = data;
L
Linus Torvalds 已提交
1467 1468
	int i;
	int idx;
D
Dave Airlie 已提交
1469
	struct drm_buf *buf;
L
Linus Torvalds 已提交
1470 1471 1472 1473

	if (!drm_core_check_feature(dev, DRIVER_HAVE_DMA))
		return -EINVAL;

D
Dave Airlie 已提交
1474 1475
	if (!dma)
		return -EINVAL;
L
Linus Torvalds 已提交
1476

1477 1478 1479
	DRM_DEBUG("%d\n", request->count);
	for (i = 0; i < request->count; i++) {
		if (copy_from_user(&idx, &request->list[i], sizeof(idx)))
L
Linus Torvalds 已提交
1480
			return -EFAULT;
D
Dave Airlie 已提交
1481 1482 1483
		if (idx < 0 || idx >= dma->buf_count) {
			DRM_ERROR("Index %d (of %d max)\n",
				  idx, dma->buf_count - 1);
L
Linus Torvalds 已提交
1484 1485 1486
			return -EINVAL;
		}
		buf = dma->buflist[idx];
1487
		if (buf->file_priv != file_priv) {
D
Dave Airlie 已提交
1488
			DRM_ERROR("Process %d freeing buffer not owned\n",
1489
				  task_pid_nr(current));
L
Linus Torvalds 已提交
1490 1491
			return -EINVAL;
		}
D
Dave Airlie 已提交
1492
		drm_free_buffer(dev, buf);
L
Linus Torvalds 已提交
1493 1494 1495 1496 1497 1498 1499 1500 1501
	}

	return 0;
}

/**
 * Maps all of the DMA buffers into client-virtual space (ioctl).
 *
 * \param inode device inode.
1502
 * \param file_priv DRM file private.
L
Linus Torvalds 已提交
1503 1504 1505 1506
 * \param cmd command.
 * \param arg pointer to a drm_buf_map structure.
 * \return zero on success or a negative number on failure.
 *
1507 1508 1509 1510
 * Maps the AGP, SG or PCI buffer region with do_mmap(), and copies information
 * about each buffer into user space. For PCI buffers, it calls do_mmap() with
 * offset equal to 0, which drm_mmap() interpretes as PCI buffers and calls
 * drm_mmap_dma().
L
Linus Torvalds 已提交
1511
 */
1512 1513
int drm_mapbufs(struct drm_device *dev, void *data,
	        struct drm_file *file_priv)
L
Linus Torvalds 已提交
1514
{
1515
	struct drm_device_dma *dma = dev->dma;
L
Linus Torvalds 已提交
1516 1517 1518 1519
	int retcode = 0;
	const int zero = 0;
	unsigned long virtual;
	unsigned long address;
1520
	struct drm_buf_map *request = data;
L
Linus Torvalds 已提交
1521 1522 1523 1524 1525
	int i;

	if (!drm_core_check_feature(dev, DRIVER_HAVE_DMA))
		return -EINVAL;

D
Dave Airlie 已提交
1526 1527
	if (!dma)
		return -EINVAL;
L
Linus Torvalds 已提交
1528

D
Dave Airlie 已提交
1529 1530 1531
	spin_lock(&dev->count_lock);
	if (atomic_read(&dev->buf_alloc)) {
		spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
1532 1533 1534
		return -EBUSY;
	}
	dev->buf_use++;		/* Can't allocate more after this call */
D
Dave Airlie 已提交
1535
	spin_unlock(&dev->count_lock);
L
Linus Torvalds 已提交
1536

1537
	if (request->count >= dma->buf_count) {
D
Dave Airlie 已提交
1538
		if ((drm_core_has_AGP(dev) && (dma->flags & _DRM_DMA_USE_AGP))
D
Dave Airlie 已提交
1539
		    || (drm_core_check_feature(dev, DRIVER_SG)
D
Dave Airlie 已提交
1540 1541 1542
			&& (dma->flags & _DRM_DMA_USE_SG))
		    || (drm_core_check_feature(dev, DRIVER_FB_DMA)
			&& (dma->flags & _DRM_DMA_USE_FB))) {
1543
			struct drm_local_map *map = dev->agp_buffer_map;
1544
			unsigned long token = dev->agp_buffer_token;
L
Linus Torvalds 已提交
1545

D
Dave Airlie 已提交
1546
			if (!map) {
L
Linus Torvalds 已提交
1547 1548 1549
				retcode = -EINVAL;
				goto done;
			}
D
Dave Airlie 已提交
1550
			down_write(&current->mm->mmap_sem);
1551
			virtual = do_mmap(file_priv->filp, 0, map->size,
D
Dave Airlie 已提交
1552
					  PROT_READ | PROT_WRITE,
1553 1554
					  MAP_SHARED,
					  token);
D
Dave Airlie 已提交
1555
			up_write(&current->mm->mmap_sem);
L
Linus Torvalds 已提交
1556
		} else {
D
Dave Airlie 已提交
1557
			down_write(&current->mm->mmap_sem);
1558
			virtual = do_mmap(file_priv->filp, 0, dma->byte_count,
D
Dave Airlie 已提交
1559 1560 1561
					  PROT_READ | PROT_WRITE,
					  MAP_SHARED, 0);
			up_write(&current->mm->mmap_sem);
L
Linus Torvalds 已提交
1562
		}
D
Dave Airlie 已提交
1563
		if (virtual > -1024UL) {
L
Linus Torvalds 已提交
1564 1565 1566 1567
			/* Real error */
			retcode = (signed long)virtual;
			goto done;
		}
1568
		request->virtual = (void __user *)virtual;
L
Linus Torvalds 已提交
1569

D
Dave Airlie 已提交
1570
		for (i = 0; i < dma->buf_count; i++) {
1571
			if (copy_to_user(&request->list[i].idx,
D
Dave Airlie 已提交
1572
					 &dma->buflist[i]->idx,
1573
					 sizeof(request->list[0].idx))) {
L
Linus Torvalds 已提交
1574 1575 1576
				retcode = -EFAULT;
				goto done;
			}
1577
			if (copy_to_user(&request->list[i].total,
D
Dave Airlie 已提交
1578
					 &dma->buflist[i]->total,
1579
					 sizeof(request->list[0].total))) {
L
Linus Torvalds 已提交
1580 1581 1582
				retcode = -EFAULT;
				goto done;
			}
1583
			if (copy_to_user(&request->list[i].used,
D
Dave Airlie 已提交
1584
					 &zero, sizeof(zero))) {
L
Linus Torvalds 已提交
1585 1586 1587
				retcode = -EFAULT;
				goto done;
			}
D
Dave Airlie 已提交
1588
			address = virtual + dma->buflist[i]->offset;	/* *** */
1589
			if (copy_to_user(&request->list[i].address,
D
Dave Airlie 已提交
1590
					 &address, sizeof(address))) {
L
Linus Torvalds 已提交
1591 1592 1593 1594 1595
				retcode = -EFAULT;
				goto done;
			}
		}
	}
D
Dave Airlie 已提交
1596
      done:
1597 1598
	request->count = dma->buf_count;
	DRM_DEBUG("%d buffers, retcode = %d\n", request->count, retcode);
L
Linus Torvalds 已提交
1599 1600 1601 1602

	return retcode;
}

D
Dave Airlie 已提交
1603 1604 1605
/**
 * Compute size order.  Returns the exponent of the smaller power of two which
 * is greater or equal to given number.
D
Dave Airlie 已提交
1606
 *
D
Dave Airlie 已提交
1607 1608 1609 1610 1611
 * \param size size.
 * \return order.
 *
 * \todo Can be made faster.
 */
D
Dave Airlie 已提交
1612
int drm_order(unsigned long size)
D
Dave Airlie 已提交
1613 1614 1615 1616
{
	int order;
	unsigned long tmp;

D
Dave Airlie 已提交
1617
	for (order = 0, tmp = size >> 1; tmp; tmp >>= 1, order++) ;
D
Dave Airlie 已提交
1618 1619 1620 1621 1622 1623 1624

	if (size & (size - 1))
		++order;

	return order;
}
EXPORT_SYMBOL(drm_order);