mmu_rb.h 1.7 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162
  1. /* SPDX-License-Identifier: GPL-2.0 or BSD-3-Clause */
  2. /*
  3. * Copyright(c) 2020 Cornelis Networks, Inc.
  4. * Copyright(c) 2016 Intel Corporation.
  5. */
  6. #ifndef _HFI1_MMU_RB_H
  7. #define _HFI1_MMU_RB_H
  8. #include "hfi.h"
  9. struct mmu_rb_node {
  10. unsigned long addr;
  11. unsigned long len;
  12. unsigned long __last;
  13. struct rb_node node;
  14. struct mmu_rb_handler *handler;
  15. struct list_head list;
  16. struct kref refcount;
  17. };
  18. /*
  19. * NOTE: filter, insert, invalidate, and evict must not sleep. Only remove is
  20. * allowed to sleep.
  21. */
  22. struct mmu_rb_ops {
  23. bool (*filter)(struct mmu_rb_node *node, unsigned long addr,
  24. unsigned long len);
  25. int (*insert)(void *ops_arg, struct mmu_rb_node *mnode);
  26. void (*remove)(void *ops_arg, struct mmu_rb_node *mnode);
  27. int (*invalidate)(void *ops_arg, struct mmu_rb_node *node);
  28. int (*evict)(void *ops_arg, struct mmu_rb_node *mnode,
  29. void *evict_arg, bool *stop);
  30. };
  31. struct mmu_rb_handler {
  32. struct mmu_notifier mn;
  33. struct rb_root_cached root;
  34. void *ops_arg;
  35. spinlock_t lock; /* protect the RB tree */
  36. struct mmu_rb_ops *ops;
  37. struct list_head lru_list;
  38. struct work_struct del_work;
  39. struct list_head del_list;
  40. struct workqueue_struct *wq;
  41. };
  42. int hfi1_mmu_rb_register(void *ops_arg,
  43. struct mmu_rb_ops *ops,
  44. struct workqueue_struct *wq,
  45. struct mmu_rb_handler **handler);
  46. void hfi1_mmu_rb_unregister(struct mmu_rb_handler *handler);
  47. int hfi1_mmu_rb_insert(struct mmu_rb_handler *handler,
  48. struct mmu_rb_node *mnode);
  49. void hfi1_mmu_rb_release(struct kref *refcount);
  50. void hfi1_mmu_rb_evict(struct mmu_rb_handler *handler, void *evict_arg);
  51. struct mmu_rb_node *hfi1_mmu_rb_get_first(struct mmu_rb_handler *handler,
  52. unsigned long addr,
  53. unsigned long len);
  54. #endif /* _HFI1_MMU_RB_H */