drm_prime.c 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772
  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. * six lower-level driver callbacks:
  264. *
  265. * Export callbacks:
  266. *
  267. * * @gem_prime_pin (optional): prepare a GEM object for exporting
  268. * * @gem_prime_get_sg_table: provide a scatter/gather table of pinned pages
  269. * * @gem_prime_vmap: vmap a buffer exported by your driver
  270. * * @gem_prime_vunmap: vunmap a buffer exported by your driver
  271. * * @gem_prime_mmap (optional): mmap a buffer exported by your driver
  272. *
  273. * Import callback:
  274. *
  275. * * @gem_prime_import_sg_table (import): produce a GEM object from another
  276. * driver's scatter/gather table
  277. */
  278. /**
  279. * drm_gem_prime_export - helper library implementation of the export callback
  280. * @dev: drm_device to export from
  281. * @obj: GEM object to export
  282. * @flags: flags like DRM_CLOEXEC
  283. *
  284. * This is the implementation of the gem_prime_export functions for GEM drivers
  285. * using the PRIME helpers.
  286. */
  287. struct dma_buf *drm_gem_prime_export(struct drm_device *dev,
  288. struct drm_gem_object *obj, int flags)
  289. {
  290. DEFINE_DMA_BUF_EXPORT_INFO(exp_info);
  291. exp_info.ops = &drm_gem_prime_dmabuf_ops;
  292. exp_info.size = obj->size;
  293. exp_info.flags = flags;
  294. exp_info.priv = obj;
  295. if (dev->driver->gem_prime_res_obj)
  296. exp_info.resv = dev->driver->gem_prime_res_obj(obj);
  297. return dma_buf_export(&exp_info);
  298. }
  299. EXPORT_SYMBOL(drm_gem_prime_export);
  300. static struct dma_buf *export_and_register_object(struct drm_device *dev,
  301. struct drm_gem_object *obj,
  302. uint32_t flags)
  303. {
  304. struct dma_buf *dmabuf;
  305. /* prevent races with concurrent gem_close. */
  306. if (obj->handle_count == 0) {
  307. dmabuf = ERR_PTR(-ENOENT);
  308. return dmabuf;
  309. }
  310. dmabuf = dev->driver->gem_prime_export(dev, obj, flags);
  311. if (IS_ERR(dmabuf)) {
  312. /* normally the created dma-buf takes ownership of the ref,
  313. * but if that fails then drop the ref
  314. */
  315. return dmabuf;
  316. }
  317. /*
  318. * Note that callers do not need to clean up the export cache
  319. * since the check for obj->handle_count guarantees that someone
  320. * will clean it up.
  321. */
  322. obj->dma_buf = dmabuf;
  323. get_dma_buf(obj->dma_buf);
  324. /* Grab a new ref since the callers is now used by the dma-buf */
  325. drm_gem_object_reference(obj);
  326. return dmabuf;
  327. }
  328. /**
  329. * drm_gem_prime_handle_to_fd - PRIME export function for GEM drivers
  330. * @dev: dev to export the buffer from
  331. * @file_priv: drm file-private structure
  332. * @handle: buffer handle to export
  333. * @flags: flags like DRM_CLOEXEC
  334. * @prime_fd: pointer to storage for the fd id of the create dma-buf
  335. *
  336. * This is the PRIME export function which must be used mandatorily by GEM
  337. * drivers to ensure correct lifetime management of the underlying GEM object.
  338. * The actual exporting from GEM object to a dma-buf is done through the
  339. * gem_prime_export driver callback.
  340. */
  341. int drm_gem_prime_handle_to_fd(struct drm_device *dev,
  342. struct drm_file *file_priv, uint32_t handle,
  343. uint32_t flags,
  344. int *prime_fd)
  345. {
  346. struct drm_gem_object *obj;
  347. int ret = 0;
  348. struct dma_buf *dmabuf;
  349. mutex_lock(&file_priv->prime.lock);
  350. obj = drm_gem_object_lookup(dev, file_priv, handle);
  351. if (!obj) {
  352. ret = -ENOENT;
  353. goto out_unlock;
  354. }
  355. dmabuf = drm_prime_lookup_buf_by_handle(&file_priv->prime, handle);
  356. if (dmabuf) {
  357. get_dma_buf(dmabuf);
  358. goto out_have_handle;
  359. }
  360. mutex_lock(&dev->object_name_lock);
  361. /* re-export the original imported object */
  362. if (obj->import_attach) {
  363. dmabuf = obj->import_attach->dmabuf;
  364. get_dma_buf(dmabuf);
  365. goto out_have_obj;
  366. }
  367. if (obj->dma_buf) {
  368. get_dma_buf(obj->dma_buf);
  369. dmabuf = obj->dma_buf;
  370. goto out_have_obj;
  371. }
  372. dmabuf = export_and_register_object(dev, obj, flags);
  373. if (IS_ERR(dmabuf)) {
  374. /* normally the created dma-buf takes ownership of the ref,
  375. * but if that fails then drop the ref
  376. */
  377. ret = PTR_ERR(dmabuf);
  378. mutex_unlock(&dev->object_name_lock);
  379. goto out;
  380. }
  381. out_have_obj:
  382. /*
  383. * If we've exported this buffer then cheat and add it to the import list
  384. * so we get the correct handle back. We must do this under the
  385. * protection of dev->object_name_lock to ensure that a racing gem close
  386. * ioctl doesn't miss to remove this buffer handle from the cache.
  387. */
  388. ret = drm_prime_add_buf_handle(&file_priv->prime,
  389. dmabuf, handle);
  390. mutex_unlock(&dev->object_name_lock);
  391. if (ret)
  392. goto fail_put_dmabuf;
  393. out_have_handle:
  394. ret = dma_buf_fd(dmabuf, flags);
  395. /*
  396. * We must _not_ remove the buffer from the handle cache since the newly
  397. * created dma buf is already linked in the global obj->dma_buf pointer,
  398. * and that is invariant as long as a userspace gem handle exists.
  399. * Closing the handle will clean out the cache anyway, so we don't leak.
  400. */
  401. if (ret < 0) {
  402. goto fail_put_dmabuf;
  403. } else {
  404. *prime_fd = ret;
  405. ret = 0;
  406. }
  407. goto out;
  408. fail_put_dmabuf:
  409. dma_buf_put(dmabuf);
  410. out:
  411. drm_gem_object_unreference_unlocked(obj);
  412. out_unlock:
  413. mutex_unlock(&file_priv->prime.lock);
  414. return ret;
  415. }
  416. EXPORT_SYMBOL(drm_gem_prime_handle_to_fd);
  417. /**
  418. * drm_gem_prime_import - helper library implementation of the import callback
  419. * @dev: drm_device to import into
  420. * @dma_buf: dma-buf object to import
  421. *
  422. * This is the implementation of the gem_prime_import functions for GEM drivers
  423. * using the PRIME helpers.
  424. */
  425. struct drm_gem_object *drm_gem_prime_import(struct drm_device *dev,
  426. struct dma_buf *dma_buf)
  427. {
  428. struct dma_buf_attachment *attach;
  429. struct sg_table *sgt;
  430. struct drm_gem_object *obj;
  431. int ret;
  432. if (dma_buf->ops == &drm_gem_prime_dmabuf_ops) {
  433. obj = dma_buf->priv;
  434. if (obj->dev == dev) {
  435. /*
  436. * Importing dmabuf exported from out own gem increases
  437. * refcount on gem itself instead of f_count of dmabuf.
  438. */
  439. drm_gem_object_reference(obj);
  440. return obj;
  441. }
  442. }
  443. if (!dev->driver->gem_prime_import_sg_table)
  444. return ERR_PTR(-EINVAL);
  445. attach = dma_buf_attach(dma_buf, dev->dev);
  446. if (IS_ERR(attach))
  447. return ERR_CAST(attach);
  448. get_dma_buf(dma_buf);
  449. sgt = dma_buf_map_attachment(attach, DMA_BIDIRECTIONAL);
  450. if (IS_ERR(sgt)) {
  451. ret = PTR_ERR(sgt);
  452. goto fail_detach;
  453. }
  454. obj = dev->driver->gem_prime_import_sg_table(dev, attach, sgt);
  455. if (IS_ERR(obj)) {
  456. ret = PTR_ERR(obj);
  457. goto fail_unmap;
  458. }
  459. obj->import_attach = attach;
  460. return obj;
  461. fail_unmap:
  462. dma_buf_unmap_attachment(attach, sgt, DMA_BIDIRECTIONAL);
  463. fail_detach:
  464. dma_buf_detach(dma_buf, attach);
  465. dma_buf_put(dma_buf);
  466. return ERR_PTR(ret);
  467. }
  468. EXPORT_SYMBOL(drm_gem_prime_import);
  469. /**
  470. * drm_gem_prime_fd_to_handle - PRIME import function for GEM drivers
  471. * @dev: dev to export the buffer from
  472. * @file_priv: drm file-private structure
  473. * @prime_fd: fd id of the dma-buf which should be imported
  474. * @handle: pointer to storage for the handle of the imported buffer object
  475. *
  476. * This is the PRIME import function which must be used mandatorily by GEM
  477. * drivers to ensure correct lifetime management of the underlying GEM object.
  478. * The actual importing of GEM object from the dma-buf is done through the
  479. * gem_import_export driver callback.
  480. */
  481. int drm_gem_prime_fd_to_handle(struct drm_device *dev,
  482. struct drm_file *file_priv, int prime_fd,
  483. uint32_t *handle)
  484. {
  485. struct dma_buf *dma_buf;
  486. struct drm_gem_object *obj;
  487. int ret;
  488. dma_buf = dma_buf_get(prime_fd);
  489. if (IS_ERR(dma_buf))
  490. return PTR_ERR(dma_buf);
  491. mutex_lock(&file_priv->prime.lock);
  492. ret = drm_prime_lookup_buf_handle(&file_priv->prime,
  493. dma_buf, handle);
  494. if (ret == 0)
  495. goto out_put;
  496. /* never seen this one, need to import */
  497. mutex_lock(&dev->object_name_lock);
  498. obj = dev->driver->gem_prime_import(dev, dma_buf);
  499. if (IS_ERR(obj)) {
  500. ret = PTR_ERR(obj);
  501. goto out_unlock;
  502. }
  503. if (obj->dma_buf) {
  504. WARN_ON(obj->dma_buf != dma_buf);
  505. } else {
  506. obj->dma_buf = dma_buf;
  507. get_dma_buf(dma_buf);
  508. }
  509. /* drm_gem_handle_create_tail unlocks dev->object_name_lock. */
  510. ret = drm_gem_handle_create_tail(file_priv, obj, handle);
  511. drm_gem_object_unreference_unlocked(obj);
  512. if (ret)
  513. goto out_put;
  514. ret = drm_prime_add_buf_handle(&file_priv->prime,
  515. dma_buf, *handle);
  516. if (ret)
  517. goto fail;
  518. mutex_unlock(&file_priv->prime.lock);
  519. dma_buf_put(dma_buf);
  520. return 0;
  521. fail:
  522. /* hmm, if driver attached, we are relying on the free-object path
  523. * to detach.. which seems ok..
  524. */
  525. drm_gem_handle_delete(file_priv, *handle);
  526. out_unlock:
  527. mutex_unlock(&dev->object_name_lock);
  528. out_put:
  529. dma_buf_put(dma_buf);
  530. mutex_unlock(&file_priv->prime.lock);
  531. return ret;
  532. }
  533. EXPORT_SYMBOL(drm_gem_prime_fd_to_handle);
  534. int drm_prime_handle_to_fd_ioctl(struct drm_device *dev, void *data,
  535. struct drm_file *file_priv)
  536. {
  537. struct drm_prime_handle *args = data;
  538. uint32_t flags;
  539. if (!drm_core_check_feature(dev, DRIVER_PRIME))
  540. return -EINVAL;
  541. if (!dev->driver->prime_handle_to_fd)
  542. return -ENOSYS;
  543. /* check flags are valid */
  544. if (args->flags & ~DRM_CLOEXEC)
  545. return -EINVAL;
  546. /* we only want to pass DRM_CLOEXEC which is == O_CLOEXEC */
  547. flags = args->flags & DRM_CLOEXEC;
  548. return dev->driver->prime_handle_to_fd(dev, file_priv,
  549. args->handle, flags, &args->fd);
  550. }
  551. int drm_prime_fd_to_handle_ioctl(struct drm_device *dev, void *data,
  552. struct drm_file *file_priv)
  553. {
  554. struct drm_prime_handle *args = data;
  555. if (!drm_core_check_feature(dev, DRIVER_PRIME))
  556. return -EINVAL;
  557. if (!dev->driver->prime_fd_to_handle)
  558. return -ENOSYS;
  559. return dev->driver->prime_fd_to_handle(dev, file_priv,
  560. args->fd, &args->handle);
  561. }
  562. /**
  563. * drm_prime_pages_to_sg - converts a page array into an sg list
  564. * @pages: pointer to the array of page pointers to convert
  565. * @nr_pages: length of the page vector
  566. *
  567. * This helper creates an sg table object from a set of pages
  568. * the driver is responsible for mapping the pages into the
  569. * importers address space for use with dma_buf itself.
  570. */
  571. struct sg_table *drm_prime_pages_to_sg(struct page **pages, unsigned int nr_pages)
  572. {
  573. struct sg_table *sg = NULL;
  574. int ret;
  575. sg = kmalloc(sizeof(struct sg_table), GFP_KERNEL);
  576. if (!sg) {
  577. ret = -ENOMEM;
  578. goto out;
  579. }
  580. ret = sg_alloc_table_from_pages(sg, pages, nr_pages, 0,
  581. nr_pages << PAGE_SHIFT, GFP_KERNEL);
  582. if (ret)
  583. goto out;
  584. return sg;
  585. out:
  586. kfree(sg);
  587. return ERR_PTR(ret);
  588. }
  589. EXPORT_SYMBOL(drm_prime_pages_to_sg);
  590. /**
  591. * drm_prime_sg_to_page_addr_arrays - convert an sg table into a page array
  592. * @sgt: scatter-gather table to convert
  593. * @pages: array of page pointers to store the page array in
  594. * @addrs: optional array to store the dma bus address of each page
  595. * @max_pages: size of both the passed-in arrays
  596. *
  597. * Exports an sg table into an array of pages and addresses. This is currently
  598. * required by the TTM driver in order to do correct fault handling.
  599. */
  600. int drm_prime_sg_to_page_addr_arrays(struct sg_table *sgt, struct page **pages,
  601. dma_addr_t *addrs, int max_pages)
  602. {
  603. unsigned count;
  604. struct scatterlist *sg;
  605. struct page *page;
  606. u32 len;
  607. int pg_index;
  608. dma_addr_t addr;
  609. pg_index = 0;
  610. for_each_sg(sgt->sgl, sg, sgt->nents, count) {
  611. len = sg->length;
  612. page = sg_page(sg);
  613. addr = sg_dma_address(sg);
  614. while (len > 0) {
  615. if (WARN_ON(pg_index >= max_pages))
  616. return -1;
  617. pages[pg_index] = page;
  618. if (addrs)
  619. addrs[pg_index] = addr;
  620. page++;
  621. addr += PAGE_SIZE;
  622. len -= PAGE_SIZE;
  623. pg_index++;
  624. }
  625. }
  626. return 0;
  627. }
  628. EXPORT_SYMBOL(drm_prime_sg_to_page_addr_arrays);
  629. /**
  630. * drm_prime_gem_destroy - helper to clean up a PRIME-imported GEM object
  631. * @obj: GEM object which was created from a dma-buf
  632. * @sg: the sg-table which was pinned at import time
  633. *
  634. * This is the cleanup functions which GEM drivers need to call when they use
  635. * @drm_gem_prime_import to import dma-bufs.
  636. */
  637. void drm_prime_gem_destroy(struct drm_gem_object *obj, struct sg_table *sg)
  638. {
  639. struct dma_buf_attachment *attach;
  640. struct dma_buf *dma_buf;
  641. attach = obj->import_attach;
  642. if (sg)
  643. dma_buf_unmap_attachment(attach, sg, DMA_BIDIRECTIONAL);
  644. dma_buf = attach->dmabuf;
  645. dma_buf_detach(attach->dmabuf, attach);
  646. /* remove the reference */
  647. dma_buf_put(dma_buf);
  648. }
  649. EXPORT_SYMBOL(drm_prime_gem_destroy);
  650. void drm_prime_init_file_private(struct drm_prime_file_private *prime_fpriv)
  651. {
  652. INIT_LIST_HEAD(&prime_fpriv->head);
  653. mutex_init(&prime_fpriv->lock);
  654. }
  655. void drm_prime_destroy_file_private(struct drm_prime_file_private *prime_fpriv)
  656. {
  657. /* by now drm_gem_release should've made sure the list is empty */
  658. WARN_ON(!list_empty(&prime_fpriv->head));
  659. }