On Fri, May 01, 2020 at 05:29:12PM -0700, Alan Mikhak wrote:
> From: Alan Mikhak <alan.mik...@sifive.com>
> 
> Modify pci_epf_test_data_transfer() to also support slave DMA transfers.
> Adds a direction parameter so caller can specify one of the supported DMA
> transfer directions: DMA_MEM_TO_MEM, DMA_MEM_TO_DEV, and DMA_DEV_TO_MEM.
> For DMA_MEM_TO_MEM, the function calls dmaengine_prep_dma_memcpy() as it
> did before. For DMA_MEM_TO_DEV or DMA_DEV_TO_MEM direction, the function
> calls dmaengine_slave_config() to configure the slave channel before it
> calls dmaengine_prep_slave_single().
> 
> Modify existing callers to specify DMA_MEM_TO_MEM since that is the only
> possible option so far. Rename the phys_addr local variable in some of the
> callers for more readability. Tighten some of the timing function calls to
> avoid counting error print time in case of error.

Looks fine, but also needs a user. The last sentence sounds like a 
separate change.

> 
> Signed-off-by: Alan Mikhak <alan.mik...@sifive.com>
> ---
>  drivers/pci/endpoint/functions/pci-epf-test.c | 67 
> ++++++++++++++++++---------
>  1 file changed, 44 insertions(+), 23 deletions(-)
> 
> diff --git a/drivers/pci/endpoint/functions/pci-epf-test.c 
> b/drivers/pci/endpoint/functions/pci-epf-test.c
> index 60330f3e3751..1d026682febb 100644
> --- a/drivers/pci/endpoint/functions/pci-epf-test.c
> +++ b/drivers/pci/endpoint/functions/pci-epf-test.c
> @@ -104,25 +104,41 @@ static void pci_epf_test_dma_callback(void *param)
>   * The function returns '0' on success and negative value on failure.
>   */
>  static int pci_epf_test_data_transfer(struct pci_epf_test *epf_test,
> -                                   dma_addr_t dma_dst, dma_addr_t dma_src,
> -                                   size_t len)
> +                                   dma_addr_t dma_dst,
> +                                   dma_addr_t dma_src,
> +                                   size_t len,
> +                                   enum dma_transfer_direction dir)
>  {
>       enum dma_ctrl_flags flags = DMA_CTRL_ACK | DMA_PREP_INTERRUPT;
>       struct dma_chan *chan = epf_test->dma_chan;
>       struct pci_epf *epf = epf_test->epf;
> +     struct dma_slave_config sconf;
>       struct dma_async_tx_descriptor *tx;
>       struct device *dev = &epf->dev;
>       dma_cookie_t cookie;
> +     dma_addr_t buf;
>       int ret;
>  
>       if (IS_ERR_OR_NULL(chan)) {
> -             dev_err(dev, "Invalid DMA memcpy channel\n");
> +             dev_err(dev, "Invalid DMA channel\n");
>               return -EINVAL;
>       }
>  
> -     tx = dmaengine_prep_dma_memcpy(chan, dma_dst, dma_src, len, flags);
> +     if (dir == DMA_MEM_TO_MEM) {
> +             tx = dmaengine_prep_dma_memcpy(chan, dma_dst, dma_src,
> +                                            len, flags);
> +     } else {
> +             memset(&sconf, 0, sizeof(sconf));
> +             sconf.direction = dir;
> +             sconf.dst_addr = dma_dst;
> +             sconf.src_addr = dma_src;
> +             dmaengine_slave_config(chan, &sconf);
> +
> +             buf = (dir == DMA_MEM_TO_DEV) ? dma_dst : dma_src;
> +             tx = dmaengine_prep_slave_single(chan, buf, len, dir, flags);
> +     }
>       if (!tx) {
> -             dev_err(dev, "Failed to prepare DMA memcpy\n");
> +             dev_err(dev, "Failed to prepare DMA transfer\n");
>               return -EIO;
>       }
>  
> @@ -268,7 +284,6 @@ static int pci_epf_test_copy(struct pci_epf_test 
> *epf_test)
>               goto err_dst_addr;
>       }
>  
> -     ktime_get_ts64(&start);
>       use_dma = !!(reg->flags & FLAG_USE_DMA);
>       if (use_dma) {
>               if (!epf_test->dma_supported) {
> @@ -277,14 +292,18 @@ static int pci_epf_test_copy(struct pci_epf_test 
> *epf_test)
>                       goto err_map_addr;
>               }
>  
> +             ktime_get_ts64(&start);
>               ret = pci_epf_test_data_transfer(epf_test, dst_phys_addr,
> -                                              src_phys_addr, reg->size);
> +                                              src_phys_addr, reg->size,
> +                                              DMA_MEM_TO_MEM);
> +             ktime_get_ts64(&end);
>               if (ret)
>                       dev_err(dev, "Data transfer failed\n");
>       } else {
> +             ktime_get_ts64(&start);
>               memcpy(dst_addr, src_addr, reg->size);
> +             ktime_get_ts64(&end);
>       }
> -     ktime_get_ts64(&end);
>       pci_epf_test_print_rate("COPY", reg->size, &start, &end, use_dma);
>  
>  err_map_addr:
> @@ -310,7 +329,7 @@ static int pci_epf_test_read(struct pci_epf_test 
> *epf_test)
>       void *buf;
>       u32 crc32;
>       bool use_dma;
> -     phys_addr_t phys_addr;
> +     phys_addr_t src_phys_addr;
>       phys_addr_t dst_phys_addr;
>       struct timespec64 start, end;
>       struct pci_epf *epf = epf_test->epf;
> @@ -319,8 +338,9 @@ static int pci_epf_test_read(struct pci_epf_test 
> *epf_test)
>       struct device *dma_dev = epf->epc->dev.parent;
>       enum pci_barno test_reg_bar = epf_test->test_reg_bar;
>       struct pci_epf_test_reg *reg = epf_test->reg[test_reg_bar];
> +     enum dma_transfer_direction dir = DMA_MEM_TO_MEM;
>  
> -     src_addr = pci_epc_mem_alloc_addr(epc, &phys_addr, reg->size);
> +     src_addr = pci_epc_mem_alloc_addr(epc, &src_phys_addr, reg->size);
>       if (!src_addr) {
>               dev_err(dev, "Failed to allocate address\n");
>               reg->status = STATUS_SRC_ADDR_INVALID;
> @@ -328,7 +348,7 @@ static int pci_epf_test_read(struct pci_epf_test 
> *epf_test)
>               goto err;
>       }
>  
> -     ret = pci_epc_map_addr(epc, epf->func_no, phys_addr, reg->src_addr,
> +     ret = pci_epc_map_addr(epc, epf->func_no, src_phys_addr, reg->src_addr,
>                              reg->size);
>       if (ret) {
>               dev_err(dev, "Failed to map address\n");
> @@ -360,10 +380,10 @@ static int pci_epf_test_read(struct pci_epf_test 
> *epf_test)
>  
>               ktime_get_ts64(&start);
>               ret = pci_epf_test_data_transfer(epf_test, dst_phys_addr,
> -                                              phys_addr, reg->size);
> +                                              src_phys_addr, reg->size, dir);
> +             ktime_get_ts64(&end);
>               if (ret)
>                       dev_err(dev, "Data transfer failed\n");
> -             ktime_get_ts64(&end);
>  
>               dma_unmap_single(dma_dev, dst_phys_addr, reg->size,
>                                DMA_FROM_DEVICE);
> @@ -383,10 +403,10 @@ static int pci_epf_test_read(struct pci_epf_test 
> *epf_test)
>       kfree(buf);
>  
>  err_map_addr:
> -     pci_epc_unmap_addr(epc, epf->func_no, phys_addr);
> +     pci_epc_unmap_addr(epc, epf->func_no, src_phys_addr);
>  
>  err_addr:
> -     pci_epc_mem_free_addr(epc, phys_addr, src_addr, reg->size);
> +     pci_epc_mem_free_addr(epc, src_phys_addr, src_addr, reg->size);
>  
>  err:
>       return ret;
> @@ -398,7 +418,7 @@ static int pci_epf_test_write(struct pci_epf_test 
> *epf_test)
>       void __iomem *dst_addr;
>       void *buf;
>       bool use_dma;
> -     phys_addr_t phys_addr;
> +     phys_addr_t dst_phys_addr;
>       phys_addr_t src_phys_addr;
>       struct timespec64 start, end;
>       struct pci_epf *epf = epf_test->epf;
> @@ -407,8 +427,9 @@ static int pci_epf_test_write(struct pci_epf_test 
> *epf_test)
>       struct device *dma_dev = epf->epc->dev.parent;
>       enum pci_barno test_reg_bar = epf_test->test_reg_bar;
>       struct pci_epf_test_reg *reg = epf_test->reg[test_reg_bar];
> +     enum dma_transfer_direction dir = DMA_MEM_TO_MEM;
>  
> -     dst_addr = pci_epc_mem_alloc_addr(epc, &phys_addr, reg->size);
> +     dst_addr = pci_epc_mem_alloc_addr(epc, &dst_phys_addr, reg->size);
>       if (!dst_addr) {
>               dev_err(dev, "Failed to allocate address\n");
>               reg->status = STATUS_DST_ADDR_INVALID;
> @@ -416,7 +437,7 @@ static int pci_epf_test_write(struct pci_epf_test 
> *epf_test)
>               goto err;
>       }
>  
> -     ret = pci_epc_map_addr(epc, epf->func_no, phys_addr, reg->dst_addr,
> +     ret = pci_epc_map_addr(epc, epf->func_no, dst_phys_addr, reg->dst_addr,
>                              reg->size);
>       if (ret) {
>               dev_err(dev, "Failed to map address\n");
> @@ -450,11 +471,11 @@ static int pci_epf_test_write(struct pci_epf_test 
> *epf_test)
>               }
>  
>               ktime_get_ts64(&start);
> -             ret = pci_epf_test_data_transfer(epf_test, phys_addr,
> -                                              src_phys_addr, reg->size);
> +             ret = pci_epf_test_data_transfer(epf_test, dst_phys_addr,
> +                                              src_phys_addr, reg->size, dir);
> +             ktime_get_ts64(&end);
>               if (ret)
>                       dev_err(dev, "Data transfer failed\n");
> -             ktime_get_ts64(&end);
>  
>               dma_unmap_single(dma_dev, src_phys_addr, reg->size,
>                                DMA_TO_DEVICE);
> @@ -476,10 +497,10 @@ static int pci_epf_test_write(struct pci_epf_test 
> *epf_test)
>       kfree(buf);
>  
>  err_map_addr:
> -     pci_epc_unmap_addr(epc, epf->func_no, phys_addr);
> +     pci_epc_unmap_addr(epc, epf->func_no, dst_phys_addr);
>  
>  err_addr:
> -     pci_epc_mem_free_addr(epc, phys_addr, dst_addr, reg->size);
> +     pci_epc_mem_free_addr(epc, dst_phys_addr, dst_addr, reg->size);
>  
>  err:
>       return ret;
> -- 
> 2.7.4
> 

Reply via email to