/* SPDX-License-Identifier: GPL-2.0 */
  /*
   * Copyright (c) 2021, The Linux Foundation. All rights reserved.
   */
  #ifndef __LOG_MMIORW_H__
  #define __LOG_MMIORW_H__
  
  #include <linux/types.h>
  #include <linux/atomic.h>
  #include <linux/tracepoint-defs.h>
  
  /*
   * TODO - io.h is included in NVHE files and these tracepoints are getting
   * enabled for NVHE too. To avoid these tracepoints enabling in NHVE below
   * condition is introduced.
   * !(defined(__KVM_NVHE_HYPERVISOR__))
   */
  #if IS_ENABLED(CONFIG_TRACE_MMIO_ACCESS) && !(defined(__KVM_NVHE_HYPERVISOR__))
  DECLARE_TRACEPOINT(rwmmio_write);
  DECLARE_TRACEPOINT(rwmmio_read);
  DECLARE_TRACEPOINT(rwmmio_post_read);
  
  void __log_write_mmio(u64 val, u8 width, volatile void __iomem *addr);
  void __log_read_mmio(u8 width, const volatile void __iomem *addr);
  void __log_post_read_mmio(u64 val, u8 width, const volatile void __iomem *addr);
  
  #define log_write_mmio(val, width, addr)		\
  do {							\
  	if (tracepoint_enabled(rwmmio_write))		\
  		__log_write_mmio(val, width, addr);	\
  } while (0)
  
  #define log_read_mmio(width, addr)			\
  do {							\
  	if (tracepoint_enabled(rwmmio_read))		\
  		__log_read_mmio(width, addr);		\
  } while (0)
  
  #define log_post_read_mmio(val, width, addr)		\
  do {							\
  	if (tracepoint_enabled(rwmmio_post_read))	\
  		__log_post_read_mmio(val, width, addr);	\
  } while (0)
  
  #else
  static inline void log_write_mmio(u64 val, u8 width, volatile void __iomem *addr)
  { }
  static inline void log_read_mmio(u8 width, const volatile void __iomem *addr)
  { }
  static inline void log_post_read_mmio(u64 val, u8 width, const volatile void __iomem *addr)
  { }
  #endif /* CONFIG_TRACE_MMIO_ACCESS */
  
  #endif /* __LOG_MMIORW_H__  */