drm_prime.c 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881
  1. /*
  2. * Copyright © 2012 Red Hat
  3. *
  4. * Permission is hereby granted, free of charge, to any person obtaining a
  5. * copy of this software and associated documentation files (the "Software"),
  6. * to deal in the Software without restriction, including without limitation
  7. * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  8. * and/or sell copies of the Software, and to permit persons to whom the
  9. * Software is furnished to do so, subject to the following conditions:
  10. *
  11. * The above copyright notice and this permission notice (including the next
  12. * paragraph) shall be included in all copies or substantial portions of the
  13. * Software.
  14. *
  15. * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16. * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17. * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  18. * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19. * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20. * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  21. * IN THE SOFTWARE.
  22. *
  23. * Authors:
  24. * Dave Airlie <airlied@redhat.com>
  25. * Rob Clark <rob.clark@linaro.org>
  26. *
  27. */
  28. #include <linux/export.h>
  29. #include <linux/dma-buf.h>
  30. #include <linux/rbtree.h>
  31. #include <drm/drm_prime.h>
  32. #include <drm/drm_gem.h>
  33. #include <drm/drmP.h>
  34. #include "drm_internal.h"
  35. /*
  36. * DMA-BUF/GEM Object references and lifetime overview:
  37. *
  38. * On the export the dma_buf holds a reference to the exporting GEM
  39. * object. It takes this reference in handle_to_fd_ioctl, when it
  40. * first calls .prime_export and stores the exporting GEM object in
  41. * the dma_buf priv. This reference needs to be released when the
  42. * final reference to the &dma_buf itself is dropped and its
  43. * &dma_buf_ops.release function is called. For GEM-based drivers,
  44. * the dma_buf should be exported using drm_gem_dmabuf_export() and
  45. * then released by drm_gem_dmabuf_release().
  46. *
  47. * On the import the importing GEM object holds a reference to the
  48. * dma_buf (which in turn holds a ref to the exporting GEM object).
  49. * It takes that reference in the fd_to_handle ioctl.
  50. * It calls dma_buf_get, creates an attachment to it and stores the
  51. * attachment in the GEM object. When this attachment is destroyed
  52. * when the imported object is destroyed, we remove the attachment
  53. * and drop the reference to the dma_buf.
  54. *
  55. * When all the references to the &dma_buf are dropped, i.e. when
  56. * userspace has closed both handles to the imported GEM object (through the
  57. * FD_TO_HANDLE IOCTL) and closed the file descriptor of the exported
  58. * (through the HANDLE_TO_FD IOCTL) dma_buf, and all kernel-internal references
  59. * are also gone, then the dma_buf gets destroyed. This can also happen as a
  60. * part of the clean up procedure in the drm_release() function if userspace
  61. * fails to properly clean up. Note that both the kernel and userspace (by
  62. * keeeping the PRIME file descriptors open) can hold references onto a
  63. * &dma_buf.
  64. *
  65. * Thus the chain of references always flows in one direction
  66. * (avoiding loops): importing_gem -> dmabuf -> exporting_gem
  67. *
  68. * Self-importing: if userspace is using PRIME as a replacement for flink
  69. * then it will get a fd->handle request for a GEM object that it created.
  70. * Drivers should detect this situation and return back the gem object
  71. * from the dma-buf private. Prime will do this automatically for drivers that
  72. * use the drm_gem_prime_{import,export} helpers.
  73. */
  74. struct drm_prime_member {
  75. struct dma_buf *dma_buf;
  76. uint32_t handle;
  77. struct rb_node dmabuf_rb;
  78. struct rb_node handle_rb;
  79. };
  80. struct drm_prime_attachment {
  81. struct sg_table *sgt;
  82. enum dma_data_direction dir;
  83. };
  84. static int drm_prime_add_buf_handle(struct drm_prime_file_private *prime_fpriv,
  85. struct dma_buf *dma_buf, uint32_t handle)
  86. {
  87. struct drm_prime_member *member;
  88. struct rb_node **p, *rb;
  89. member = kmalloc(sizeof(*member), GFP_KERNEL);
  90. if (!member)
  91. return -ENOMEM;
  92. get_dma_buf(dma_buf);
  93. member->dma_buf = dma_buf;
  94. member->handle = handle;
  95. rb = NULL;
  96. p = &prime_fpriv->dmabufs.rb_node;
  97. while (*p) {
  98. struct drm_prime_member *pos;
  99. rb = *p;
  100. pos = rb_entry(rb, struct drm_prime_member, dmabuf_rb);
  101. if (dma_buf > pos->dma_buf)
  102. p = &rb->rb_right;
  103. else
  104. p = &rb->rb_left;
  105. }
  106. rb_link_node(&member->dmabuf_rb, rb, p);
  107. rb_insert_color(&member->dmabuf_rb, &prime_fpriv->dmabufs);
  108. rb = NULL;
  109. p = &prime_fpriv->handles.rb_node;
  110. while (*p) {
  111. struct drm_prime_member *pos;
  112. rb = *p;
  113. pos = rb_entry(rb, struct drm_prime_member, handle_rb);
  114. if (handle > pos->handle)
  115. p = &rb->rb_right;
  116. else
  117. p = &rb->rb_left;
  118. }
  119. rb_link_node(&member->handle_rb, rb, p);
  120. rb_insert_color(&member->handle_rb, &prime_fpriv->handles);
  121. return 0;
  122. }
  123. static struct dma_buf *drm_prime_lookup_buf_by_handle(struct drm_prime_file_private *prime_fpriv,
  124. uint32_t handle)
  125. {
  126. struct rb_node *rb;
  127. rb = prime_fpriv->handles.rb_node;
  128. while (rb) {
  129. struct drm_prime_member *member;
  130. member = rb_entry(rb, struct drm_prime_member, handle_rb);
  131. if (member->handle == handle)
  132. return member->dma_buf;
  133. else if (member->handle < handle)
  134. rb = rb->rb_right;
  135. else
  136. rb = rb->rb_left;
  137. }
  138. return NULL;
  139. }
  140. static int drm_prime_lookup_buf_handle(struct drm_prime_file_private *prime_fpriv,
  141. struct dma_buf *dma_buf,
  142. uint32_t *handle)
  143. {
  144. struct rb_node *rb;
  145. rb = prime_fpriv->dmabufs.rb_node;
  146. while (rb) {
  147. struct drm_prime_member *member;
  148. member = rb_entry(rb, struct drm_prime_member, dmabuf_rb);
  149. if (member->dma_buf == dma_buf) {
  150. *handle = member->handle;
  151. return 0;
  152. } else if (member->dma_buf < dma_buf) {
  153. rb = rb->rb_right;
  154. } else {
  155. rb = rb->rb_left;
  156. }
  157. }
  158. return -ENOENT;
  159. }
  160. static int drm_gem_map_attach(struct dma_buf *dma_buf,
  161. struct device *target_dev,
  162. struct dma_buf_attachment *attach)
  163. {
  164. struct drm_prime_attachment *prime_attach;
  165. struct drm_gem_object *obj = dma_buf->priv;
  166. struct drm_device *dev = obj->dev;
  167. prime_attach = kzalloc(sizeof(*prime_attach), GFP_KERNEL);
  168. if (!prime_attach)
  169. return -ENOMEM;
  170. prime_attach->dir = DMA_NONE;
  171. attach->priv = prime_attach;
  172. if (!dev->driver->gem_prime_pin)
  173. return 0;
  174. return dev->driver->gem_prime_pin(obj);
  175. }
  176. static void drm_gem_map_detach(struct dma_buf *dma_buf,
  177. struct dma_buf_attachment *attach)
  178. {
  179. struct drm_prime_attachment *prime_attach = attach->priv;
  180. struct drm_gem_object *obj = dma_buf->priv;
  181. struct drm_device *dev = obj->dev;
  182. struct sg_table *sgt;
  183. if (dev->driver->gem_prime_unpin)
  184. dev->driver->gem_prime_unpin(obj);
  185. if (!prime_attach)
  186. return;
  187. sgt = prime_attach->sgt;
  188. if (sgt) {
  189. if (prime_attach->dir != DMA_NONE)
  190. dma_unmap_sg(attach->dev, sgt->sgl, sgt->nents,
  191. prime_attach->dir);
  192. sg_free_table(sgt);
  193. }
  194. kfree(sgt);
  195. kfree(prime_attach);
  196. attach->priv = NULL;
  197. }
  198. void drm_prime_remove_buf_handle_locked(struct drm_prime_file_private *prime_fpriv,
  199. struct dma_buf *dma_buf)
  200. {
  201. struct rb_node *rb;
  202. rb = prime_fpriv->dmabufs.rb_node;
  203. while (rb) {
  204. struct drm_prime_member *member;
  205. member = rb_entry(rb, struct drm_prime_member, dmabuf_rb);
  206. if (member->dma_buf == dma_buf) {
  207. rb_erase(&member->handle_rb, &prime_fpriv->handles);
  208. rb_erase(&member->dmabuf_rb, &prime_fpriv->dmabufs);
  209. dma_buf_put(dma_buf);
  210. kfree(member);
  211. return;
  212. } else if (member->dma_buf < dma_buf) {
  213. rb = rb->rb_right;
  214. } else {
  215. rb = rb->rb_left;
  216. }
  217. }
  218. }
  219. static struct sg_table *drm_gem_map_dma_buf(struct dma_buf_attachment *attach,
  220. enum dma_data_direction dir)
  221. {
  222. struct drm_prime_attachment *prime_attach = attach->priv;
  223. struct drm_gem_object *obj = attach->dmabuf->priv;
  224. struct sg_table *sgt;
  225. if (WARN_ON(dir == DMA_NONE || !prime_attach))
  226. return ERR_PTR(-EINVAL);
  227. /* return the cached mapping when possible */
  228. if (prime_attach->dir == dir)
  229. return prime_attach->sgt;
  230. /*
  231. * two mappings with different directions for the same attachment are
  232. * not allowed
  233. */
  234. if (WARN_ON(prime_attach->dir != DMA_NONE))
  235. return ERR_PTR(-EBUSY);
  236. sgt = obj->dev->driver->gem_prime_get_sg_table(obj);
  237. if (!IS_ERR(sgt)) {
  238. if (!dma_map_sg(attach->dev, sgt->sgl, sgt->nents, dir)) {
  239. sg_free_table(sgt);
  240. kfree(sgt);
  241. sgt = ERR_PTR(-ENOMEM);
  242. } else {
  243. prime_attach->sgt = sgt;
  244. prime_attach->dir = dir;
  245. }
  246. }
  247. return sgt;
  248. }
  249. static void drm_gem_unmap_dma_buf(struct dma_buf_attachment *attach,
  250. struct sg_table *sgt,
  251. enum dma_data_direction dir)
  252. {
  253. /* nothing to be done here */
  254. }
  255. /**
  256. * drm_gem_dmabuf_export - dma_buf export implementation for GEM
  257. * @dev: parent device for the exported dmabuf
  258. * @exp_info: the export information used by dma_buf_export()
  259. *
  260. * This wraps dma_buf_export() for use by generic GEM drivers that are using
  261. * drm_gem_dmabuf_release(). In addition to calling dma_buf_export(), we take
  262. * a reference to the &drm_device and the exported &drm_gem_object (stored in
  263. * &dma_buf_export_info.priv) which is released by drm_gem_dmabuf_release().
  264. *
  265. * Returns the new dmabuf.
  266. */
  267. struct dma_buf *drm_gem_dmabuf_export(struct drm_device *dev,
  268. struct dma_buf_export_info *exp_info)
  269. {
  270. struct dma_buf *dma_buf;
  271. dma_buf = dma_buf_export(exp_info);
  272. if (IS_ERR(dma_buf))
  273. return dma_buf;
  274. drm_dev_ref(dev);
  275. drm_gem_object_get(exp_info->priv);
  276. return dma_buf;
  277. }
  278. EXPORT_SYMBOL(drm_gem_dmabuf_export);
  279. /**
  280. * drm_gem_dmabuf_release - dma_buf release implementation for GEM
  281. * @dma_buf: buffer to be released
  282. *
  283. * Generic release function for dma_bufs exported as PRIME buffers. GEM drivers
  284. * must use this in their dma_buf ops structure as the release callback.
  285. * drm_gem_dmabuf_release() should be used in conjunction with
  286. * drm_gem_dmabuf_export().
  287. */
  288. void drm_gem_dmabuf_release(struct dma_buf *dma_buf)
  289. {
  290. struct drm_gem_object *obj = dma_buf->priv;
  291. struct drm_device *dev = obj->dev;
  292. /* drop the reference on the export fd holds */
  293. drm_gem_object_put_unlocked(obj);
  294. drm_dev_unref(dev);
  295. }
  296. EXPORT_SYMBOL(drm_gem_dmabuf_release);
  297. static void *drm_gem_dmabuf_vmap(struct dma_buf *dma_buf)
  298. {
  299. struct drm_gem_object *obj = dma_buf->priv;
  300. struct drm_device *dev = obj->dev;
  301. return dev->driver->gem_prime_vmap(obj);
  302. }
  303. static void drm_gem_dmabuf_vunmap(struct dma_buf *dma_buf, void *vaddr)
  304. {
  305. struct drm_gem_object *obj = dma_buf->priv;
  306. struct drm_device *dev = obj->dev;
  307. dev->driver->gem_prime_vunmap(obj, vaddr);
  308. }
  309. static void *drm_gem_dmabuf_kmap_atomic(struct dma_buf *dma_buf,
  310. unsigned long page_num)
  311. {
  312. return NULL;
  313. }
  314. static void drm_gem_dmabuf_kunmap_atomic(struct dma_buf *dma_buf,
  315. unsigned long page_num, void *addr)
  316. {
  317. }
  318. static void *drm_gem_dmabuf_kmap(struct dma_buf *dma_buf,
  319. unsigned long page_num)
  320. {
  321. return NULL;
  322. }
  323. static void drm_gem_dmabuf_kunmap(struct dma_buf *dma_buf,
  324. unsigned long page_num, void *addr)
  325. {
  326. }
  327. static int drm_gem_dmabuf_mmap(struct dma_buf *dma_buf,
  328. struct vm_area_struct *vma)
  329. {
  330. struct drm_gem_object *obj = dma_buf->priv;
  331. struct drm_device *dev = obj->dev;
  332. if (!dev->driver->gem_prime_mmap)
  333. return -ENOSYS;
  334. return dev->driver->gem_prime_mmap(obj, vma);
  335. }
  336. static const struct dma_buf_ops drm_gem_prime_dmabuf_ops = {
  337. .attach = drm_gem_map_attach,
  338. .detach = drm_gem_map_detach,
  339. .map_dma_buf = drm_gem_map_dma_buf,
  340. .unmap_dma_buf = drm_gem_unmap_dma_buf,
  341. .release = drm_gem_dmabuf_release,
  342. .kmap = drm_gem_dmabuf_kmap,
  343. .kmap_atomic = drm_gem_dmabuf_kmap_atomic,
  344. .kunmap = drm_gem_dmabuf_kunmap,
  345. .kunmap_atomic = drm_gem_dmabuf_kunmap_atomic,
  346. .mmap = drm_gem_dmabuf_mmap,
  347. .vmap = drm_gem_dmabuf_vmap,
  348. .vunmap = drm_gem_dmabuf_vunmap,
  349. };
  350. /**
  351. * DOC: PRIME Helpers
  352. *
  353. * Drivers can implement @gem_prime_export and @gem_prime_import in terms of
  354. * simpler APIs by using the helper functions @drm_gem_prime_export and
  355. * @drm_gem_prime_import. These functions implement dma-buf support in terms of
  356. * six lower-level driver callbacks:
  357. *
  358. * Export callbacks:
  359. *
  360. * * @gem_prime_pin (optional): prepare a GEM object for exporting
  361. * * @gem_prime_get_sg_table: provide a scatter/gather table of pinned pages
  362. * * @gem_prime_vmap: vmap a buffer exported by your driver
  363. * * @gem_prime_vunmap: vunmap a buffer exported by your driver
  364. * * @gem_prime_mmap (optional): mmap a buffer exported by your driver
  365. *
  366. * Import callback:
  367. *
  368. * * @gem_prime_import_sg_table (import): produce a GEM object from another
  369. * driver's scatter/gather table
  370. */
  371. /**
  372. * drm_gem_prime_export - helper library implementation of the export callback
  373. * @dev: drm_device to export from
  374. * @obj: GEM object to export
  375. * @flags: flags like DRM_CLOEXEC and DRM_RDWR
  376. *
  377. * This is the implementation of the gem_prime_export functions for GEM drivers
  378. * using the PRIME helpers.
  379. */
  380. struct dma_buf *drm_gem_prime_export(struct drm_device *dev,
  381. struct drm_gem_object *obj,
  382. int flags)
  383. {
  384. struct dma_buf_export_info exp_info = {
  385. .exp_name = KBUILD_MODNAME, /* white lie for debug */
  386. .owner = dev->driver->fops->owner,
  387. .ops = &drm_gem_prime_dmabuf_ops,
  388. .size = obj->size,
  389. .flags = flags,
  390. .priv = obj,
  391. };
  392. if (dev->driver->gem_prime_res_obj)
  393. exp_info.resv = dev->driver->gem_prime_res_obj(obj);
  394. return drm_gem_dmabuf_export(dev, &exp_info);
  395. }
  396. EXPORT_SYMBOL(drm_gem_prime_export);
  397. static struct dma_buf *export_and_register_object(struct drm_device *dev,
  398. struct drm_gem_object *obj,
  399. uint32_t flags)
  400. {
  401. struct dma_buf *dmabuf;
  402. /* prevent races with concurrent gem_close. */
  403. if (obj->handle_count == 0) {
  404. dmabuf = ERR_PTR(-ENOENT);
  405. return dmabuf;
  406. }
  407. dmabuf = dev->driver->gem_prime_export(dev, obj, flags);
  408. if (IS_ERR(dmabuf)) {
  409. /* normally the created dma-buf takes ownership of the ref,
  410. * but if that fails then drop the ref
  411. */
  412. return dmabuf;
  413. }
  414. /*
  415. * Note that callers do not need to clean up the export cache
  416. * since the check for obj->handle_count guarantees that someone
  417. * will clean it up.
  418. */
  419. obj->dma_buf = dmabuf;
  420. get_dma_buf(obj->dma_buf);
  421. return dmabuf;
  422. }
  423. /**
  424. * drm_gem_prime_handle_to_fd - PRIME export function for GEM drivers
  425. * @dev: dev to export the buffer from
  426. * @file_priv: drm file-private structure
  427. * @handle: buffer handle to export
  428. * @flags: flags like DRM_CLOEXEC
  429. * @prime_fd: pointer to storage for the fd id of the create dma-buf
  430. *
  431. * This is the PRIME export function which must be used mandatorily by GEM
  432. * drivers to ensure correct lifetime management of the underlying GEM object.
  433. * The actual exporting from GEM object to a dma-buf is done through the
  434. * gem_prime_export driver callback.
  435. */
  436. int drm_gem_prime_handle_to_fd(struct drm_device *dev,
  437. struct drm_file *file_priv, uint32_t handle,
  438. uint32_t flags,
  439. int *prime_fd)
  440. {
  441. struct drm_gem_object *obj;
  442. int ret = 0;
  443. struct dma_buf *dmabuf;
  444. mutex_lock(&file_priv->prime.lock);
  445. obj = drm_gem_object_lookup(file_priv, handle);
  446. if (!obj) {
  447. ret = -ENOENT;
  448. goto out_unlock;
  449. }
  450. dmabuf = drm_prime_lookup_buf_by_handle(&file_priv->prime, handle);
  451. if (dmabuf) {
  452. get_dma_buf(dmabuf);
  453. goto out_have_handle;
  454. }
  455. mutex_lock(&dev->object_name_lock);
  456. /* re-export the original imported object */
  457. if (obj->import_attach) {
  458. dmabuf = obj->import_attach->dmabuf;
  459. get_dma_buf(dmabuf);
  460. goto out_have_obj;
  461. }
  462. if (obj->dma_buf) {
  463. get_dma_buf(obj->dma_buf);
  464. dmabuf = obj->dma_buf;
  465. goto out_have_obj;
  466. }
  467. dmabuf = export_and_register_object(dev, obj, flags);
  468. if (IS_ERR(dmabuf)) {
  469. /* normally the created dma-buf takes ownership of the ref,
  470. * but if that fails then drop the ref
  471. */
  472. ret = PTR_ERR(dmabuf);
  473. mutex_unlock(&dev->object_name_lock);
  474. goto out;
  475. }
  476. out_have_obj:
  477. /*
  478. * If we've exported this buffer then cheat and add it to the import list
  479. * so we get the correct handle back. We must do this under the
  480. * protection of dev->object_name_lock to ensure that a racing gem close
  481. * ioctl doesn't miss to remove this buffer handle from the cache.
  482. */
  483. ret = drm_prime_add_buf_handle(&file_priv->prime,
  484. dmabuf, handle);
  485. mutex_unlock(&dev->object_name_lock);
  486. if (ret)
  487. goto fail_put_dmabuf;
  488. out_have_handle:
  489. ret = dma_buf_fd(dmabuf, flags);
  490. /*
  491. * We must _not_ remove the buffer from the handle cache since the newly
  492. * created dma buf is already linked in the global obj->dma_buf pointer,
  493. * and that is invariant as long as a userspace gem handle exists.
  494. * Closing the handle will clean out the cache anyway, so we don't leak.
  495. */
  496. if (ret < 0) {
  497. goto fail_put_dmabuf;
  498. } else {
  499. *prime_fd = ret;
  500. ret = 0;
  501. }
  502. goto out;
  503. fail_put_dmabuf:
  504. dma_buf_put(dmabuf);
  505. out:
  506. drm_gem_object_put_unlocked(obj);
  507. out_unlock:
  508. mutex_unlock(&file_priv->prime.lock);
  509. return ret;
  510. }
  511. EXPORT_SYMBOL(drm_gem_prime_handle_to_fd);
  512. /**
  513. * drm_gem_prime_import - helper library implementation of the import callback
  514. * @dev: drm_device to import into
  515. * @dma_buf: dma-buf object to import
  516. *
  517. * This is the implementation of the gem_prime_import functions for GEM drivers
  518. * using the PRIME helpers.
  519. */
  520. struct drm_gem_object *drm_gem_prime_import(struct drm_device *dev,
  521. struct dma_buf *dma_buf)
  522. {
  523. struct dma_buf_attachment *attach;
  524. struct sg_table *sgt;
  525. struct drm_gem_object *obj;
  526. int ret;
  527. if (dma_buf->ops == &drm_gem_prime_dmabuf_ops) {
  528. obj = dma_buf->priv;
  529. if (obj->dev == dev) {
  530. /*
  531. * Importing dmabuf exported from out own gem increases
  532. * refcount on gem itself instead of f_count of dmabuf.
  533. */
  534. drm_gem_object_get(obj);
  535. return obj;
  536. }
  537. }
  538. if (!dev->driver->gem_prime_import_sg_table)
  539. return ERR_PTR(-EINVAL);
  540. attach = dma_buf_attach(dma_buf, dev->dev);
  541. if (IS_ERR(attach))
  542. return ERR_CAST(attach);
  543. get_dma_buf(dma_buf);
  544. sgt = dma_buf_map_attachment(attach, DMA_BIDIRECTIONAL);
  545. if (IS_ERR(sgt)) {
  546. ret = PTR_ERR(sgt);
  547. goto fail_detach;
  548. }
  549. obj = dev->driver->gem_prime_import_sg_table(dev, attach, sgt);
  550. if (IS_ERR(obj)) {
  551. ret = PTR_ERR(obj);
  552. goto fail_unmap;
  553. }
  554. obj->import_attach = attach;
  555. return obj;
  556. fail_unmap:
  557. dma_buf_unmap_attachment(attach, sgt, DMA_BIDIRECTIONAL);
  558. fail_detach:
  559. dma_buf_detach(dma_buf, attach);
  560. dma_buf_put(dma_buf);
  561. return ERR_PTR(ret);
  562. }
  563. EXPORT_SYMBOL(drm_gem_prime_import);
  564. /**
  565. * drm_gem_prime_fd_to_handle - PRIME import function for GEM drivers
  566. * @dev: dev to export the buffer from
  567. * @file_priv: drm file-private structure
  568. * @prime_fd: fd id of the dma-buf which should be imported
  569. * @handle: pointer to storage for the handle of the imported buffer object
  570. *
  571. * This is the PRIME import function which must be used mandatorily by GEM
  572. * drivers to ensure correct lifetime management of the underlying GEM object.
  573. * The actual importing of GEM object from the dma-buf is done through the
  574. * gem_import_export driver callback.
  575. */
  576. int drm_gem_prime_fd_to_handle(struct drm_device *dev,
  577. struct drm_file *file_priv, int prime_fd,
  578. uint32_t *handle)
  579. {
  580. struct dma_buf *dma_buf;
  581. struct drm_gem_object *obj;
  582. int ret;
  583. dma_buf = dma_buf_get(prime_fd);
  584. if (IS_ERR(dma_buf))
  585. return PTR_ERR(dma_buf);
  586. mutex_lock(&file_priv->prime.lock);
  587. ret = drm_prime_lookup_buf_handle(&file_priv->prime,
  588. dma_buf, handle);
  589. if (ret == 0)
  590. goto out_put;
  591. /* never seen this one, need to import */
  592. mutex_lock(&dev->object_name_lock);
  593. obj = dev->driver->gem_prime_import(dev, dma_buf);
  594. if (IS_ERR(obj)) {
  595. ret = PTR_ERR(obj);
  596. goto out_unlock;
  597. }
  598. if (obj->dma_buf) {
  599. WARN_ON(obj->dma_buf != dma_buf);
  600. } else {
  601. obj->dma_buf = dma_buf;
  602. get_dma_buf(dma_buf);
  603. }
  604. /* _handle_create_tail unconditionally unlocks dev->object_name_lock. */
  605. ret = drm_gem_handle_create_tail(file_priv, obj, handle);
  606. drm_gem_object_put_unlocked(obj);
  607. if (ret)
  608. goto out_put;
  609. ret = drm_prime_add_buf_handle(&file_priv->prime,
  610. dma_buf, *handle);
  611. mutex_unlock(&file_priv->prime.lock);
  612. if (ret)
  613. goto fail;
  614. dma_buf_put(dma_buf);
  615. return 0;
  616. fail:
  617. /* hmm, if driver attached, we are relying on the free-object path
  618. * to detach.. which seems ok..
  619. */
  620. drm_gem_handle_delete(file_priv, *handle);
  621. dma_buf_put(dma_buf);
  622. return ret;
  623. out_unlock:
  624. mutex_unlock(&dev->object_name_lock);
  625. out_put:
  626. mutex_unlock(&file_priv->prime.lock);
  627. dma_buf_put(dma_buf);
  628. return ret;
  629. }
  630. EXPORT_SYMBOL(drm_gem_prime_fd_to_handle);
  631. int drm_prime_handle_to_fd_ioctl(struct drm_device *dev, void *data,
  632. struct drm_file *file_priv)
  633. {
  634. struct drm_prime_handle *args = data;
  635. if (!drm_core_check_feature(dev, DRIVER_PRIME))
  636. return -EINVAL;
  637. if (!dev->driver->prime_handle_to_fd)
  638. return -ENOSYS;
  639. /* check flags are valid */
  640. if (args->flags & ~(DRM_CLOEXEC | DRM_RDWR))
  641. return -EINVAL;
  642. return dev->driver->prime_handle_to_fd(dev, file_priv,
  643. args->handle, args->flags, &args->fd);
  644. }
  645. int drm_prime_fd_to_handle_ioctl(struct drm_device *dev, void *data,
  646. struct drm_file *file_priv)
  647. {
  648. struct drm_prime_handle *args = data;
  649. if (!drm_core_check_feature(dev, DRIVER_PRIME))
  650. return -EINVAL;
  651. if (!dev->driver->prime_fd_to_handle)
  652. return -ENOSYS;
  653. return dev->driver->prime_fd_to_handle(dev, file_priv,
  654. args->fd, &args->handle);
  655. }
  656. /**
  657. * drm_prime_pages_to_sg - converts a page array into an sg list
  658. * @pages: pointer to the array of page pointers to convert
  659. * @nr_pages: length of the page vector
  660. *
  661. * This helper creates an sg table object from a set of pages
  662. * the driver is responsible for mapping the pages into the
  663. * importers address space for use with dma_buf itself.
  664. */
  665. struct sg_table *drm_prime_pages_to_sg(struct page **pages, unsigned int nr_pages)
  666. {
  667. struct sg_table *sg = NULL;
  668. int ret;
  669. sg = kmalloc(sizeof(struct sg_table), GFP_KERNEL);
  670. if (!sg) {
  671. ret = -ENOMEM;
  672. goto out;
  673. }
  674. ret = sg_alloc_table_from_pages(sg, pages, nr_pages, 0,
  675. nr_pages << PAGE_SHIFT, GFP_KERNEL);
  676. if (ret)
  677. goto out;
  678. return sg;
  679. out:
  680. kfree(sg);
  681. return ERR_PTR(ret);
  682. }
  683. EXPORT_SYMBOL(drm_prime_pages_to_sg);
  684. /**
  685. * drm_prime_sg_to_page_addr_arrays - convert an sg table into a page array
  686. * @sgt: scatter-gather table to convert
  687. * @pages: array of page pointers to store the page array in
  688. * @addrs: optional array to store the dma bus address of each page
  689. * @max_pages: size of both the passed-in arrays
  690. *
  691. * Exports an sg table into an array of pages and addresses. This is currently
  692. * required by the TTM driver in order to do correct fault handling.
  693. */
  694. int drm_prime_sg_to_page_addr_arrays(struct sg_table *sgt, struct page **pages,
  695. dma_addr_t *addrs, int max_pages)
  696. {
  697. unsigned count;
  698. struct scatterlist *sg;
  699. struct page *page;
  700. u32 len;
  701. int pg_index;
  702. dma_addr_t addr;
  703. pg_index = 0;
  704. for_each_sg(sgt->sgl, sg, sgt->nents, count) {
  705. len = sg->length;
  706. page = sg_page(sg);
  707. addr = sg_dma_address(sg);
  708. while (len > 0) {
  709. if (WARN_ON(pg_index >= max_pages))
  710. return -1;
  711. pages[pg_index] = page;
  712. if (addrs)
  713. addrs[pg_index] = addr;
  714. page++;
  715. addr += PAGE_SIZE;
  716. len -= PAGE_SIZE;
  717. pg_index++;
  718. }
  719. }
  720. return 0;
  721. }
  722. EXPORT_SYMBOL(drm_prime_sg_to_page_addr_arrays);
  723. /**
  724. * drm_prime_gem_destroy - helper to clean up a PRIME-imported GEM object
  725. * @obj: GEM object which was created from a dma-buf
  726. * @sg: the sg-table which was pinned at import time
  727. *
  728. * This is the cleanup functions which GEM drivers need to call when they use
  729. * @drm_gem_prime_import to import dma-bufs.
  730. */
  731. void drm_prime_gem_destroy(struct drm_gem_object *obj, struct sg_table *sg)
  732. {
  733. struct dma_buf_attachment *attach;
  734. struct dma_buf *dma_buf;
  735. attach = obj->import_attach;
  736. if (sg)
  737. dma_buf_unmap_attachment(attach, sg, DMA_BIDIRECTIONAL);
  738. dma_buf = attach->dmabuf;
  739. dma_buf_detach(attach->dmabuf, attach);
  740. /* remove the reference */
  741. dma_buf_put(dma_buf);
  742. }
  743. EXPORT_SYMBOL(drm_prime_gem_destroy);
  744. void drm_prime_init_file_private(struct drm_prime_file_private *prime_fpriv)
  745. {
  746. mutex_init(&prime_fpriv->lock);
  747. prime_fpriv->dmabufs = RB_ROOT;
  748. prime_fpriv->handles = RB_ROOT;
  749. }
  750. void drm_prime_destroy_file_private(struct drm_prime_file_private *prime_fpriv)
  751. {
  752. /* by now drm_gem_release should've made sure the list is empty */
  753. WARN_ON(!RB_EMPTY_ROOT(&prime_fpriv->dmabufs));
  754. }