drm_prime.c 20 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774
  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 <drm/drmP.h>
  31. #include <drm/drm_gem.h>
  32. #include "drm_internal.h"
  33. /*
  34. * DMA-BUF/GEM Object references and lifetime overview:
  35. *
  36. * On the export the dma_buf holds a reference to the exporting GEM
  37. * object. It takes this reference in handle_to_fd_ioctl, when it
  38. * first calls .prime_export and stores the exporting GEM object in
  39. * the dma_buf priv. This reference is released when the dma_buf
  40. * object goes away in the driver .release function.
  41. *
  42. * On the import the importing GEM object holds a reference to the
  43. * dma_buf (which in turn holds a ref to the exporting GEM object).
  44. * It takes that reference in the fd_to_handle ioctl.
  45. * It calls dma_buf_get, creates an attachment to it and stores the
  46. * attachment in the GEM object. When this attachment is destroyed
  47. * when the imported object is destroyed, we remove the attachment
  48. * and drop the reference to the dma_buf.
  49. *
  50. * Thus the chain of references always flows in one direction
  51. * (avoiding loops): importing_gem -> dmabuf -> exporting_gem
  52. *
  53. * Self-importing: if userspace is using PRIME as a replacement for flink
  54. * then it will get a fd->handle request for a GEM object that it created.
  55. * Drivers should detect this situation and return back the gem object
  56. * from the dma-buf private. Prime will do this automatically for drivers that
  57. * use the drm_gem_prime_{import,export} helpers.
  58. */
  59. struct drm_prime_member {
  60. struct list_head entry;
  61. struct dma_buf *dma_buf;
  62. uint32_t handle;
  63. };
  64. struct drm_prime_attachment {
  65. struct sg_table *sgt;
  66. enum dma_data_direction dir;
  67. };
  68. static int drm_prime_add_buf_handle(struct drm_prime_file_private *prime_fpriv,
  69. struct dma_buf *dma_buf, uint32_t handle)
  70. {
  71. struct drm_prime_member *member;
  72. member = kmalloc(sizeof(*member), GFP_KERNEL);
  73. if (!member)
  74. return -ENOMEM;
  75. get_dma_buf(dma_buf);
  76. member->dma_buf = dma_buf;
  77. member->handle = handle;
  78. list_add(&member->entry, &prime_fpriv->head);
  79. return 0;
  80. }
  81. static struct dma_buf *drm_prime_lookup_buf_by_handle(struct drm_prime_file_private *prime_fpriv,
  82. uint32_t handle)
  83. {
  84. struct drm_prime_member *member;
  85. list_for_each_entry(member, &prime_fpriv->head, entry) {
  86. if (member->handle == handle)
  87. return member->dma_buf;
  88. }
  89. return NULL;
  90. }
  91. static int drm_prime_lookup_buf_handle(struct drm_prime_file_private *prime_fpriv,
  92. struct dma_buf *dma_buf,
  93. uint32_t *handle)
  94. {
  95. struct drm_prime_member *member;
  96. list_for_each_entry(member, &prime_fpriv->head, entry) {
  97. if (member->dma_buf == dma_buf) {
  98. *handle = member->handle;
  99. return 0;
  100. }
  101. }
  102. return -ENOENT;
  103. }
  104. static int drm_gem_map_attach(struct dma_buf *dma_buf,
  105. struct device *target_dev,
  106. struct dma_buf_attachment *attach)
  107. {
  108. struct drm_prime_attachment *prime_attach;
  109. struct drm_gem_object *obj = dma_buf->priv;
  110. struct drm_device *dev = obj->dev;
  111. prime_attach = kzalloc(sizeof(*prime_attach), GFP_KERNEL);
  112. if (!prime_attach)
  113. return -ENOMEM;
  114. prime_attach->dir = DMA_NONE;
  115. attach->priv = prime_attach;
  116. if (!dev->driver->gem_prime_pin)
  117. return 0;
  118. return dev->driver->gem_prime_pin(obj);
  119. }
  120. static void drm_gem_map_detach(struct dma_buf *dma_buf,
  121. struct dma_buf_attachment *attach)
  122. {
  123. struct drm_prime_attachment *prime_attach = attach->priv;
  124. struct drm_gem_object *obj = dma_buf->priv;
  125. struct drm_device *dev = obj->dev;
  126. struct sg_table *sgt;
  127. if (dev->driver->gem_prime_unpin)
  128. dev->driver->gem_prime_unpin(obj);
  129. if (!prime_attach)
  130. return;
  131. sgt = prime_attach->sgt;
  132. if (sgt) {
  133. if (prime_attach->dir != DMA_NONE)
  134. dma_unmap_sg(attach->dev, sgt->sgl, sgt->nents,
  135. prime_attach->dir);
  136. sg_free_table(sgt);
  137. }
  138. kfree(sgt);
  139. kfree(prime_attach);
  140. attach->priv = NULL;
  141. }
  142. void drm_prime_remove_buf_handle_locked(struct drm_prime_file_private *prime_fpriv,
  143. struct dma_buf *dma_buf)
  144. {
  145. struct drm_prime_member *member, *safe;
  146. list_for_each_entry_safe(member, safe, &prime_fpriv->head, entry) {
  147. if (member->dma_buf == dma_buf) {
  148. dma_buf_put(dma_buf);
  149. list_del(&member->entry);
  150. kfree(member);
  151. }
  152. }
  153. }
  154. static struct sg_table *drm_gem_map_dma_buf(struct dma_buf_attachment *attach,
  155. enum dma_data_direction dir)
  156. {
  157. struct drm_prime_attachment *prime_attach = attach->priv;
  158. struct drm_gem_object *obj = attach->dmabuf->priv;
  159. struct sg_table *sgt;
  160. if (WARN_ON(dir == DMA_NONE || !prime_attach))
  161. return ERR_PTR(-EINVAL);
  162. /* return the cached mapping when possible */
  163. if (prime_attach->dir == dir)
  164. return prime_attach->sgt;
  165. /*
  166. * two mappings with different directions for the same attachment are
  167. * not allowed
  168. */
  169. if (WARN_ON(prime_attach->dir != DMA_NONE))
  170. return ERR_PTR(-EBUSY);
  171. sgt = obj->dev->driver->gem_prime_get_sg_table(obj);
  172. if (!IS_ERR(sgt)) {
  173. if (!dma_map_sg(attach->dev, sgt->sgl, sgt->nents, dir)) {
  174. sg_free_table(sgt);
  175. kfree(sgt);
  176. sgt = ERR_PTR(-ENOMEM);
  177. } else {
  178. prime_attach->sgt = sgt;
  179. prime_attach->dir = dir;
  180. }
  181. }
  182. return sgt;
  183. }
  184. static void drm_gem_unmap_dma_buf(struct dma_buf_attachment *attach,
  185. struct sg_table *sgt,
  186. enum dma_data_direction dir)
  187. {
  188. /* nothing to be done here */
  189. }
  190. /**
  191. * drm_gem_dmabuf_release - dma_buf release implementation for GEM
  192. * @dma_buf: buffer to be released
  193. *
  194. * Generic release function for dma_bufs exported as PRIME buffers. GEM drivers
  195. * must use this in their dma_buf ops structure as the release callback.
  196. */
  197. void drm_gem_dmabuf_release(struct dma_buf *dma_buf)
  198. {
  199. struct drm_gem_object *obj = dma_buf->priv;
  200. /* drop the reference on the export fd holds */
  201. drm_gem_object_unreference_unlocked(obj);
  202. }
  203. EXPORT_SYMBOL(drm_gem_dmabuf_release);
  204. static void *drm_gem_dmabuf_vmap(struct dma_buf *dma_buf)
  205. {
  206. struct drm_gem_object *obj = dma_buf->priv;
  207. struct drm_device *dev = obj->dev;
  208. return dev->driver->gem_prime_vmap(obj);
  209. }
  210. static void drm_gem_dmabuf_vunmap(struct dma_buf *dma_buf, void *vaddr)
  211. {
  212. struct drm_gem_object *obj = dma_buf->priv;
  213. struct drm_device *dev = obj->dev;
  214. dev->driver->gem_prime_vunmap(obj, vaddr);
  215. }
  216. static void *drm_gem_dmabuf_kmap_atomic(struct dma_buf *dma_buf,
  217. unsigned long page_num)
  218. {
  219. return NULL;
  220. }
  221. static void drm_gem_dmabuf_kunmap_atomic(struct dma_buf *dma_buf,
  222. unsigned long page_num, void *addr)
  223. {
  224. }
  225. static void *drm_gem_dmabuf_kmap(struct dma_buf *dma_buf,
  226. unsigned long page_num)
  227. {
  228. return NULL;
  229. }
  230. static void drm_gem_dmabuf_kunmap(struct dma_buf *dma_buf,
  231. unsigned long page_num, void *addr)
  232. {
  233. }
  234. static int drm_gem_dmabuf_mmap(struct dma_buf *dma_buf,
  235. struct vm_area_struct *vma)
  236. {
  237. struct drm_gem_object *obj = dma_buf->priv;
  238. struct drm_device *dev = obj->dev;
  239. if (!dev->driver->gem_prime_mmap)
  240. return -ENOSYS;
  241. return dev->driver->gem_prime_mmap(obj, vma);
  242. }
  243. static const struct dma_buf_ops drm_gem_prime_dmabuf_ops = {
  244. .attach = drm_gem_map_attach,
  245. .detach = drm_gem_map_detach,
  246. .map_dma_buf = drm_gem_map_dma_buf,
  247. .unmap_dma_buf = drm_gem_unmap_dma_buf,
  248. .release = drm_gem_dmabuf_release,
  249. .kmap = drm_gem_dmabuf_kmap,
  250. .kmap_atomic = drm_gem_dmabuf_kmap_atomic,
  251. .kunmap = drm_gem_dmabuf_kunmap,
  252. .kunmap_atomic = drm_gem_dmabuf_kunmap_atomic,
  253. .mmap = drm_gem_dmabuf_mmap,
  254. .vmap = drm_gem_dmabuf_vmap,
  255. .vunmap = drm_gem_dmabuf_vunmap,
  256. };
  257. /**
  258. * DOC: PRIME Helpers
  259. *
  260. * Drivers can implement @gem_prime_export and @gem_prime_import in terms of
  261. * simpler APIs by using the helper functions @drm_gem_prime_export and
  262. * @drm_gem_prime_import. These functions implement dma-buf support in terms of
  263. * five lower-level driver callbacks:
  264. *
  265. * Export callbacks:
  266. *
  267. * - @gem_prime_pin (optional): prepare a GEM object for exporting
  268. *
  269. * - @gem_prime_get_sg_table: provide a scatter/gather table of pinned pages
  270. *
  271. * - @gem_prime_vmap: vmap a buffer exported by your driver
  272. *
  273. * - @gem_prime_vunmap: vunmap a buffer exported by your driver
  274. *
  275. * Import callback:
  276. *
  277. * - @gem_prime_import_sg_table (import): produce a GEM object from another
  278. * driver's scatter/gather table
  279. */
  280. /**
  281. * drm_gem_prime_export - helper library implementation of the export callback
  282. * @dev: drm_device to export from
  283. * @obj: GEM object to export
  284. * @flags: flags like DRM_CLOEXEC
  285. *
  286. * This is the implementation of the gem_prime_export functions for GEM drivers
  287. * using the PRIME helpers.
  288. */
  289. struct dma_buf *drm_gem_prime_export(struct drm_device *dev,
  290. struct drm_gem_object *obj, int flags)
  291. {
  292. DEFINE_DMA_BUF_EXPORT_INFO(exp_info);
  293. exp_info.ops = &drm_gem_prime_dmabuf_ops;
  294. exp_info.size = obj->size;
  295. exp_info.flags = flags;
  296. exp_info.priv = obj;
  297. if (dev->driver->gem_prime_res_obj)
  298. exp_info.resv = dev->driver->gem_prime_res_obj(obj);
  299. return dma_buf_export(&exp_info);
  300. }
  301. EXPORT_SYMBOL(drm_gem_prime_export);
  302. static struct dma_buf *export_and_register_object(struct drm_device *dev,
  303. struct drm_gem_object *obj,
  304. uint32_t flags)
  305. {
  306. struct dma_buf *dmabuf;
  307. /* prevent races with concurrent gem_close. */
  308. if (obj->handle_count == 0) {
  309. dmabuf = ERR_PTR(-ENOENT);
  310. return dmabuf;
  311. }
  312. dmabuf = dev->driver->gem_prime_export(dev, obj, flags);
  313. if (IS_ERR(dmabuf)) {
  314. /* normally the created dma-buf takes ownership of the ref,
  315. * but if that fails then drop the ref
  316. */
  317. return dmabuf;
  318. }
  319. /*
  320. * Note that callers do not need to clean up the export cache
  321. * since the check for obj->handle_count guarantees that someone
  322. * will clean it up.
  323. */
  324. obj->dma_buf = dmabuf;
  325. get_dma_buf(obj->dma_buf);
  326. /* Grab a new ref since the callers is now used by the dma-buf */
  327. drm_gem_object_reference(obj);
  328. return dmabuf;
  329. }
  330. /**
  331. * drm_gem_prime_handle_to_fd - PRIME export function for GEM drivers
  332. * @dev: dev to export the buffer from
  333. * @file_priv: drm file-private structure
  334. * @handle: buffer handle to export
  335. * @flags: flags like DRM_CLOEXEC
  336. * @prime_fd: pointer to storage for the fd id of the create dma-buf
  337. *
  338. * This is the PRIME export function which must be used mandatorily by GEM
  339. * drivers to ensure correct lifetime management of the underlying GEM object.
  340. * The actual exporting from GEM object to a dma-buf is done through the
  341. * gem_prime_export driver callback.
  342. */
  343. int drm_gem_prime_handle_to_fd(struct drm_device *dev,
  344. struct drm_file *file_priv, uint32_t handle,
  345. uint32_t flags,
  346. int *prime_fd)
  347. {
  348. struct drm_gem_object *obj;
  349. int ret = 0;
  350. struct dma_buf *dmabuf;
  351. mutex_lock(&file_priv->prime.lock);
  352. obj = drm_gem_object_lookup(dev, file_priv, handle);
  353. if (!obj) {
  354. ret = -ENOENT;
  355. goto out_unlock;
  356. }
  357. dmabuf = drm_prime_lookup_buf_by_handle(&file_priv->prime, handle);
  358. if (dmabuf) {
  359. get_dma_buf(dmabuf);
  360. goto out_have_handle;
  361. }
  362. mutex_lock(&dev->object_name_lock);
  363. /* re-export the original imported object */
  364. if (obj->import_attach) {
  365. dmabuf = obj->import_attach->dmabuf;
  366. get_dma_buf(dmabuf);
  367. goto out_have_obj;
  368. }
  369. if (obj->dma_buf) {
  370. get_dma_buf(obj->dma_buf);
  371. dmabuf = obj->dma_buf;
  372. goto out_have_obj;
  373. }
  374. dmabuf = export_and_register_object(dev, obj, flags);
  375. if (IS_ERR(dmabuf)) {
  376. /* normally the created dma-buf takes ownership of the ref,
  377. * but if that fails then drop the ref
  378. */
  379. ret = PTR_ERR(dmabuf);
  380. mutex_unlock(&dev->object_name_lock);
  381. goto out;
  382. }
  383. out_have_obj:
  384. /*
  385. * If we've exported this buffer then cheat and add it to the import list
  386. * so we get the correct handle back. We must do this under the
  387. * protection of dev->object_name_lock to ensure that a racing gem close
  388. * ioctl doesn't miss to remove this buffer handle from the cache.
  389. */
  390. ret = drm_prime_add_buf_handle(&file_priv->prime,
  391. dmabuf, handle);
  392. mutex_unlock(&dev->object_name_lock);
  393. if (ret)
  394. goto fail_put_dmabuf;
  395. out_have_handle:
  396. ret = dma_buf_fd(dmabuf, flags);
  397. /*
  398. * We must _not_ remove the buffer from the handle cache since the newly
  399. * created dma buf is already linked in the global obj->dma_buf pointer,
  400. * and that is invariant as long as a userspace gem handle exists.
  401. * Closing the handle will clean out the cache anyway, so we don't leak.
  402. */
  403. if (ret < 0) {
  404. goto fail_put_dmabuf;
  405. } else {
  406. *prime_fd = ret;
  407. ret = 0;
  408. }
  409. goto out;
  410. fail_put_dmabuf:
  411. dma_buf_put(dmabuf);
  412. out:
  413. drm_gem_object_unreference_unlocked(obj);
  414. out_unlock:
  415. mutex_unlock(&file_priv->prime.lock);
  416. return ret;
  417. }
  418. EXPORT_SYMBOL(drm_gem_prime_handle_to_fd);
  419. /**
  420. * drm_gem_prime_import - helper library implementation of the import callback
  421. * @dev: drm_device to import into
  422. * @dma_buf: dma-buf object to import
  423. *
  424. * This is the implementation of the gem_prime_import functions for GEM drivers
  425. * using the PRIME helpers.
  426. */
  427. struct drm_gem_object *drm_gem_prime_import(struct drm_device *dev,
  428. struct dma_buf *dma_buf)
  429. {
  430. struct dma_buf_attachment *attach;
  431. struct sg_table *sgt;
  432. struct drm_gem_object *obj;
  433. int ret;
  434. if (!dev->driver->gem_prime_import_sg_table)
  435. return ERR_PTR(-EINVAL);
  436. if (dma_buf->ops == &drm_gem_prime_dmabuf_ops) {
  437. obj = dma_buf->priv;
  438. if (obj->dev == dev) {
  439. /*
  440. * Importing dmabuf exported from out own gem increases
  441. * refcount on gem itself instead of f_count of dmabuf.
  442. */
  443. drm_gem_object_reference(obj);
  444. return obj;
  445. }
  446. }
  447. attach = dma_buf_attach(dma_buf, dev->dev);
  448. if (IS_ERR(attach))
  449. return ERR_CAST(attach);
  450. get_dma_buf(dma_buf);
  451. sgt = dma_buf_map_attachment(attach, DMA_BIDIRECTIONAL);
  452. if (IS_ERR(sgt)) {
  453. ret = PTR_ERR(sgt);
  454. goto fail_detach;
  455. }
  456. obj = dev->driver->gem_prime_import_sg_table(dev, attach, sgt);
  457. if (IS_ERR(obj)) {
  458. ret = PTR_ERR(obj);
  459. goto fail_unmap;
  460. }
  461. obj->import_attach = attach;
  462. return obj;
  463. fail_unmap:
  464. dma_buf_unmap_attachment(attach, sgt, DMA_BIDIRECTIONAL);
  465. fail_detach:
  466. dma_buf_detach(dma_buf, attach);
  467. dma_buf_put(dma_buf);
  468. return ERR_PTR(ret);
  469. }
  470. EXPORT_SYMBOL(drm_gem_prime_import);
  471. /**
  472. * drm_gem_prime_fd_to_handle - PRIME import function for GEM drivers
  473. * @dev: dev to export the buffer from
  474. * @file_priv: drm file-private structure
  475. * @prime_fd: fd id of the dma-buf which should be imported
  476. * @handle: pointer to storage for the handle of the imported buffer object
  477. *
  478. * This is the PRIME import function which must be used mandatorily by GEM
  479. * drivers to ensure correct lifetime management of the underlying GEM object.
  480. * The actual importing of GEM object from the dma-buf is done through the
  481. * gem_import_export driver callback.
  482. */
  483. int drm_gem_prime_fd_to_handle(struct drm_device *dev,
  484. struct drm_file *file_priv, int prime_fd,
  485. uint32_t *handle)
  486. {
  487. struct dma_buf *dma_buf;
  488. struct drm_gem_object *obj;
  489. int ret;
  490. dma_buf = dma_buf_get(prime_fd);
  491. if (IS_ERR(dma_buf))
  492. return PTR_ERR(dma_buf);
  493. mutex_lock(&file_priv->prime.lock);
  494. ret = drm_prime_lookup_buf_handle(&file_priv->prime,
  495. dma_buf, handle);
  496. if (ret == 0)
  497. goto out_put;
  498. /* never seen this one, need to import */
  499. mutex_lock(&dev->object_name_lock);
  500. obj = dev->driver->gem_prime_import(dev, dma_buf);
  501. if (IS_ERR(obj)) {
  502. ret = PTR_ERR(obj);
  503. goto out_unlock;
  504. }
  505. if (obj->dma_buf) {
  506. WARN_ON(obj->dma_buf != dma_buf);
  507. } else {
  508. obj->dma_buf = dma_buf;
  509. get_dma_buf(dma_buf);
  510. }
  511. /* drm_gem_handle_create_tail unlocks dev->object_name_lock. */
  512. ret = drm_gem_handle_create_tail(file_priv, obj, handle);
  513. drm_gem_object_unreference_unlocked(obj);
  514. if (ret)
  515. goto out_put;
  516. ret = drm_prime_add_buf_handle(&file_priv->prime,
  517. dma_buf, *handle);
  518. if (ret)
  519. goto fail;
  520. mutex_unlock(&file_priv->prime.lock);
  521. dma_buf_put(dma_buf);
  522. return 0;
  523. fail:
  524. /* hmm, if driver attached, we are relying on the free-object path
  525. * to detach.. which seems ok..
  526. */
  527. drm_gem_handle_delete(file_priv, *handle);
  528. out_unlock:
  529. mutex_unlock(&dev->object_name_lock);
  530. out_put:
  531. dma_buf_put(dma_buf);
  532. mutex_unlock(&file_priv->prime.lock);
  533. return ret;
  534. }
  535. EXPORT_SYMBOL(drm_gem_prime_fd_to_handle);
  536. int drm_prime_handle_to_fd_ioctl(struct drm_device *dev, void *data,
  537. struct drm_file *file_priv)
  538. {
  539. struct drm_prime_handle *args = data;
  540. uint32_t flags;
  541. if (!drm_core_check_feature(dev, DRIVER_PRIME))
  542. return -EINVAL;
  543. if (!dev->driver->prime_handle_to_fd)
  544. return -ENOSYS;
  545. /* check flags are valid */
  546. if (args->flags & ~DRM_CLOEXEC)
  547. return -EINVAL;
  548. /* we only want to pass DRM_CLOEXEC which is == O_CLOEXEC */
  549. flags = args->flags & DRM_CLOEXEC;
  550. return dev->driver->prime_handle_to_fd(dev, file_priv,
  551. args->handle, flags, &args->fd);
  552. }
  553. int drm_prime_fd_to_handle_ioctl(struct drm_device *dev, void *data,
  554. struct drm_file *file_priv)
  555. {
  556. struct drm_prime_handle *args = data;
  557. if (!drm_core_check_feature(dev, DRIVER_PRIME))
  558. return -EINVAL;
  559. if (!dev->driver->prime_fd_to_handle)
  560. return -ENOSYS;
  561. return dev->driver->prime_fd_to_handle(dev, file_priv,
  562. args->fd, &args->handle);
  563. }
  564. /**
  565. * drm_prime_pages_to_sg - converts a page array into an sg list
  566. * @pages: pointer to the array of page pointers to convert
  567. * @nr_pages: length of the page vector
  568. *
  569. * This helper creates an sg table object from a set of pages
  570. * the driver is responsible for mapping the pages into the
  571. * importers address space for use with dma_buf itself.
  572. */
  573. struct sg_table *drm_prime_pages_to_sg(struct page **pages, unsigned int nr_pages)
  574. {
  575. struct sg_table *sg = NULL;
  576. int ret;
  577. sg = kmalloc(sizeof(struct sg_table), GFP_KERNEL);
  578. if (!sg) {
  579. ret = -ENOMEM;
  580. goto out;
  581. }
  582. ret = sg_alloc_table_from_pages(sg, pages, nr_pages, 0,
  583. nr_pages << PAGE_SHIFT, GFP_KERNEL);
  584. if (ret)
  585. goto out;
  586. return sg;
  587. out:
  588. kfree(sg);
  589. return ERR_PTR(ret);
  590. }
  591. EXPORT_SYMBOL(drm_prime_pages_to_sg);
  592. /**
  593. * drm_prime_sg_to_page_addr_arrays - convert an sg table into a page array
  594. * @sgt: scatter-gather table to convert
  595. * @pages: array of page pointers to store the page array in
  596. * @addrs: optional array to store the dma bus address of each page
  597. * @max_pages: size of both the passed-in arrays
  598. *
  599. * Exports an sg table into an array of pages and addresses. This is currently
  600. * required by the TTM driver in order to do correct fault handling.
  601. */
  602. int drm_prime_sg_to_page_addr_arrays(struct sg_table *sgt, struct page **pages,
  603. dma_addr_t *addrs, int max_pages)
  604. {
  605. unsigned count;
  606. struct scatterlist *sg;
  607. struct page *page;
  608. u32 len;
  609. int pg_index;
  610. dma_addr_t addr;
  611. pg_index = 0;
  612. for_each_sg(sgt->sgl, sg, sgt->nents, count) {
  613. len = sg->length;
  614. page = sg_page(sg);
  615. addr = sg_dma_address(sg);
  616. while (len > 0) {
  617. if (WARN_ON(pg_index >= max_pages))
  618. return -1;
  619. pages[pg_index] = page;
  620. if (addrs)
  621. addrs[pg_index] = addr;
  622. page++;
  623. addr += PAGE_SIZE;
  624. len -= PAGE_SIZE;
  625. pg_index++;
  626. }
  627. }
  628. return 0;
  629. }
  630. EXPORT_SYMBOL(drm_prime_sg_to_page_addr_arrays);
  631. /**
  632. * drm_prime_gem_destroy - helper to clean up a PRIME-imported GEM object
  633. * @obj: GEM object which was created from a dma-buf
  634. * @sg: the sg-table which was pinned at import time
  635. *
  636. * This is the cleanup functions which GEM drivers need to call when they use
  637. * @drm_gem_prime_import to import dma-bufs.
  638. */
  639. void drm_prime_gem_destroy(struct drm_gem_object *obj, struct sg_table *sg)
  640. {
  641. struct dma_buf_attachment *attach;
  642. struct dma_buf *dma_buf;
  643. attach = obj->import_attach;
  644. if (sg)
  645. dma_buf_unmap_attachment(attach, sg, DMA_BIDIRECTIONAL);
  646. dma_buf = attach->dmabuf;
  647. dma_buf_detach(attach->dmabuf, attach);
  648. /* remove the reference */
  649. dma_buf_put(dma_buf);
  650. }
  651. EXPORT_SYMBOL(drm_prime_gem_destroy);
  652. void drm_prime_init_file_private(struct drm_prime_file_private *prime_fpriv)
  653. {
  654. INIT_LIST_HEAD(&prime_fpriv->head);
  655. mutex_init(&prime_fpriv->lock);
  656. }
  657. void drm_prime_destroy_file_private(struct drm_prime_file_private *prime_fpriv)
  658. {
  659. /* by now drm_gem_release should've made sure the list is empty */
  660. WARN_ON(!list_empty(&prime_fpriv->head));
  661. }