udl_gem.c 4.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242
  1. /*
  2. * Copyright (C) 2012 Red Hat
  3. *
  4. * This file is subject to the terms and conditions of the GNU General Public
  5. * License v2. See the file COPYING in the main directory of this archive for
  6. * more details.
  7. */
  8. #include "drmP.h"
  9. #include "udl_drv.h"
  10. #include <linux/shmem_fs.h>
  11. struct udl_gem_object *udl_gem_alloc_object(struct drm_device *dev,
  12. size_t size)
  13. {
  14. struct udl_gem_object *obj;
  15. obj = kzalloc(sizeof(*obj), GFP_KERNEL);
  16. if (obj == NULL)
  17. return NULL;
  18. if (drm_gem_object_init(dev, &obj->base, size) != 0) {
  19. kfree(obj);
  20. return NULL;
  21. }
  22. return obj;
  23. }
  24. static int
  25. udl_gem_create(struct drm_file *file,
  26. struct drm_device *dev,
  27. uint64_t size,
  28. uint32_t *handle_p)
  29. {
  30. struct udl_gem_object *obj;
  31. int ret;
  32. u32 handle;
  33. size = roundup(size, PAGE_SIZE);
  34. obj = udl_gem_alloc_object(dev, size);
  35. if (obj == NULL)
  36. return -ENOMEM;
  37. ret = drm_gem_handle_create(file, &obj->base, &handle);
  38. if (ret) {
  39. drm_gem_object_release(&obj->base);
  40. kfree(obj);
  41. return ret;
  42. }
  43. drm_gem_object_unreference(&obj->base);
  44. *handle_p = handle;
  45. return 0;
  46. }
  47. int udl_dumb_create(struct drm_file *file,
  48. struct drm_device *dev,
  49. struct drm_mode_create_dumb *args)
  50. {
  51. args->pitch = args->width * ((args->bpp + 1) / 8);
  52. args->size = args->pitch * args->height;
  53. return udl_gem_create(file, dev,
  54. args->size, &args->handle);
  55. }
  56. int udl_dumb_destroy(struct drm_file *file, struct drm_device *dev,
  57. uint32_t handle)
  58. {
  59. return drm_gem_handle_delete(file, handle);
  60. }
  61. int udl_drm_gem_mmap(struct file *filp, struct vm_area_struct *vma)
  62. {
  63. int ret;
  64. ret = drm_gem_mmap(filp, vma);
  65. if (ret)
  66. return ret;
  67. vma->vm_flags &= ~VM_PFNMAP;
  68. vma->vm_flags |= VM_MIXEDMAP;
  69. return ret;
  70. }
  71. int udl_gem_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
  72. {
  73. struct udl_gem_object *obj = to_udl_bo(vma->vm_private_data);
  74. struct page *page;
  75. unsigned int page_offset;
  76. int ret = 0;
  77. page_offset = ((unsigned long)vmf->virtual_address - vma->vm_start) >>
  78. PAGE_SHIFT;
  79. if (!obj->pages)
  80. return VM_FAULT_SIGBUS;
  81. page = obj->pages[page_offset];
  82. ret = vm_insert_page(vma, (unsigned long)vmf->virtual_address, page);
  83. switch (ret) {
  84. case -EAGAIN:
  85. set_need_resched();
  86. case 0:
  87. case -ERESTARTSYS:
  88. return VM_FAULT_NOPAGE;
  89. case -ENOMEM:
  90. return VM_FAULT_OOM;
  91. default:
  92. return VM_FAULT_SIGBUS;
  93. }
  94. }
  95. int udl_gem_init_object(struct drm_gem_object *obj)
  96. {
  97. BUG();
  98. return 0;
  99. }
  100. static int udl_gem_get_pages(struct udl_gem_object *obj, gfp_t gfpmask)
  101. {
  102. int page_count, i;
  103. struct page *page;
  104. struct inode *inode;
  105. struct address_space *mapping;
  106. if (obj->pages)
  107. return 0;
  108. page_count = obj->base.size / PAGE_SIZE;
  109. BUG_ON(obj->pages != NULL);
  110. obj->pages = drm_malloc_ab(page_count, sizeof(struct page *));
  111. if (obj->pages == NULL)
  112. return -ENOMEM;
  113. inode = obj->base.filp->f_path.dentry->d_inode;
  114. mapping = inode->i_mapping;
  115. gfpmask |= mapping_gfp_mask(mapping);
  116. for (i = 0; i < page_count; i++) {
  117. page = shmem_read_mapping_page_gfp(mapping, i, gfpmask);
  118. if (IS_ERR(page))
  119. goto err_pages;
  120. obj->pages[i] = page;
  121. }
  122. return 0;
  123. err_pages:
  124. while (i--)
  125. page_cache_release(obj->pages[i]);
  126. drm_free_large(obj->pages);
  127. obj->pages = NULL;
  128. return PTR_ERR(page);
  129. }
  130. static void udl_gem_put_pages(struct udl_gem_object *obj)
  131. {
  132. int page_count = obj->base.size / PAGE_SIZE;
  133. int i;
  134. for (i = 0; i < page_count; i++)
  135. page_cache_release(obj->pages[i]);
  136. drm_free_large(obj->pages);
  137. obj->pages = NULL;
  138. }
  139. int udl_gem_vmap(struct udl_gem_object *obj)
  140. {
  141. int page_count = obj->base.size / PAGE_SIZE;
  142. int ret;
  143. ret = udl_gem_get_pages(obj, GFP_KERNEL);
  144. if (ret)
  145. return ret;
  146. obj->vmapping = vmap(obj->pages, page_count, 0, PAGE_KERNEL);
  147. if (!obj->vmapping)
  148. return -ENOMEM;
  149. return 0;
  150. }
  151. void udl_gem_vunmap(struct udl_gem_object *obj)
  152. {
  153. if (obj->vmapping)
  154. vunmap(obj->vmapping);
  155. udl_gem_put_pages(obj);
  156. }
  157. void udl_gem_free_object(struct drm_gem_object *gem_obj)
  158. {
  159. struct udl_gem_object *obj = to_udl_bo(gem_obj);
  160. if (obj->vmapping)
  161. udl_gem_vunmap(obj);
  162. if (obj->pages)
  163. udl_gem_put_pages(obj);
  164. if (gem_obj->map_list.map)
  165. drm_gem_free_mmap_offset(gem_obj);
  166. }
  167. /* the dumb interface doesn't work with the GEM straight MMAP
  168. interface, it expects to do MMAP on the drm fd, like normal */
  169. int udl_gem_mmap(struct drm_file *file, struct drm_device *dev,
  170. uint32_t handle, uint64_t *offset)
  171. {
  172. struct udl_gem_object *gobj;
  173. struct drm_gem_object *obj;
  174. int ret = 0;
  175. mutex_lock(&dev->struct_mutex);
  176. obj = drm_gem_object_lookup(dev, file, handle);
  177. if (obj == NULL) {
  178. ret = -ENOENT;
  179. goto unlock;
  180. }
  181. gobj = to_udl_bo(obj);
  182. ret = udl_gem_get_pages(gobj, GFP_KERNEL);
  183. if (ret)
  184. return ret;
  185. if (!gobj->base.map_list.map) {
  186. ret = drm_gem_create_mmap_offset(obj);
  187. if (ret)
  188. goto out;
  189. }
  190. *offset = (u64)gobj->base.map_list.hash.key << PAGE_SHIFT;
  191. out:
  192. drm_gem_object_unreference(&gobj->base);
  193. unlock:
  194. mutex_unlock(&dev->struct_mutex);
  195. return ret;
  196. }